bf_lbuf.c
上传用户:yisoukefu
上传日期:2020-08-09
资源大小:39506k
文件大小:10k
源码类别:

其他游戏

开发平台:

Visual C++

  1. /* crypto/bio/bf_buff.c */
  2. /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
  3.  * All rights reserved.
  4.  *
  5.  * This package is an SSL implementation written
  6.  * by Eric Young (eay@cryptsoft.com).
  7.  * The implementation was written so as to conform with Netscapes SSL.
  8.  * 
  9.  * This library is free for commercial and non-commercial use as long as
  10.  * the following conditions are aheared to.  The following conditions
  11.  * apply to all code found in this distribution, be it the RC4, RSA,
  12.  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
  13.  * included with this distribution is covered by the same copyright terms
  14.  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  15.  * 
  16.  * Copyright remains Eric Young's, and as such any Copyright notices in
  17.  * the code are not to be removed.
  18.  * If this package is used in a product, Eric Young should be given attribution
  19.  * as the author of the parts of the library used.
  20.  * This can be in the form of a textual message at program startup or
  21.  * in documentation (online or textual) provided with the package.
  22.  * 
  23.  * Redistribution and use in source and binary forms, with or without
  24.  * modification, are permitted provided that the following conditions
  25.  * are met:
  26.  * 1. Redistributions of source code must retain the copyright
  27.  *    notice, this list of conditions and the following disclaimer.
  28.  * 2. Redistributions in binary form must reproduce the above copyright
  29.  *    notice, this list of conditions and the following disclaimer in the
  30.  *    documentation and/or other materials provided with the distribution.
  31.  * 3. All advertising materials mentioning features or use of this software
  32.  *    must display the following acknowledgement:
  33.  *    "This product includes cryptographic software written by
  34.  *     Eric Young (eay@cryptsoft.com)"
  35.  *    The word 'cryptographic' can be left out if the rouines from the library
  36.  *    being used are not cryptographic related :-).
  37.  * 4. If you include any Windows specific code (or a derivative thereof) from 
  38.  *    the apps directory (application code) you must include an acknowledgement:
  39.  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  40.  * 
  41.  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  42.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51.  * SUCH DAMAGE.
  52.  * 
  53.  * The licence and distribution terms for any publically available version or
  54.  * derivative of this code cannot be changed.  i.e. this code cannot simply be
  55.  * copied and put under another distribution licence
  56.  * [including the GNU Public Licence.]
  57.  */
  58. #include <stdio.h>
  59. #include <errno.h>
  60. #include "cryptlib.h"
  61. #include <openssl/bio.h>
  62. #include <openssl/evp.h>
  63. static int linebuffer_write(BIO *h, const char *buf,int num);
  64. static int linebuffer_read(BIO *h, char *buf, int size);
  65. static int linebuffer_puts(BIO *h, const char *str);
  66. static int linebuffer_gets(BIO *h, char *str, int size);
  67. static long linebuffer_ctrl(BIO *h, int cmd, long arg1, void *arg2);
  68. static int linebuffer_new(BIO *h);
  69. static int linebuffer_free(BIO *data);
  70. static long linebuffer_callback_ctrl(BIO *h, int cmd, bio_info_cb *fp);
  71. /* A 10k maximum should be enough for most purposes */
  72. #define DEFAULT_LINEBUFFER_SIZE 1024*10
  73. /* #define DEBUG */
  74. static BIO_METHOD methods_linebuffer=
  75. {
  76. BIO_TYPE_LINEBUFFER,
  77. "linebuffer",
  78. linebuffer_write,
  79. linebuffer_read,
  80. linebuffer_puts,
  81. linebuffer_gets,
  82. linebuffer_ctrl,
  83. linebuffer_new,
  84. linebuffer_free,
  85. linebuffer_callback_ctrl,
  86. };
  87. BIO_METHOD *BIO_f_linebuffer(void)
  88. {
  89. return(&methods_linebuffer);
  90. }
  91. typedef struct bio_linebuffer_ctx_struct
  92. {
  93. char *obuf; /* the output char array */
  94. int obuf_size; /* how big is the output buffer */
  95. int obuf_len; /* how many bytes are in it */
  96. } BIO_LINEBUFFER_CTX;
  97. static int linebuffer_new(BIO *bi)
  98. {
  99. BIO_LINEBUFFER_CTX *ctx;
  100. ctx=(BIO_LINEBUFFER_CTX *)OPENSSL_malloc(sizeof(BIO_LINEBUFFER_CTX));
  101. if (ctx == NULL) return(0);
  102. ctx->obuf=(char *)OPENSSL_malloc(DEFAULT_LINEBUFFER_SIZE);
  103. if (ctx->obuf == NULL) { OPENSSL_free(ctx); return(0); }
  104. ctx->obuf_size=DEFAULT_LINEBUFFER_SIZE;
  105. ctx->obuf_len=0;
  106. bi->init=1;
  107. bi->ptr=(char *)ctx;
  108. bi->flags=0;
  109. return(1);
  110. }
  111. static int linebuffer_free(BIO *a)
  112. {
  113. BIO_LINEBUFFER_CTX *b;
  114. if (a == NULL) return(0);
  115. b=(BIO_LINEBUFFER_CTX *)a->ptr;
  116. if (b->obuf != NULL) OPENSSL_free(b->obuf);
  117. OPENSSL_free(a->ptr);
  118. a->ptr=NULL;
  119. a->init=0;
  120. a->flags=0;
  121. return(1);
  122. }
  123. static int linebuffer_read(BIO *b, char *out, int outl)
  124. {
  125. int ret=0;
  126.  
  127. if (out == NULL) return(0);
  128. if (b->next_bio == NULL) return(0);
  129. ret=BIO_read(b->next_bio,out,outl);
  130. BIO_clear_retry_flags(b);
  131. BIO_copy_next_retry(b);
  132. return(ret);
  133. }
  134. static int linebuffer_write(BIO *b, const char *in, int inl)
  135. {
  136. int i,num=0,foundnl;
  137. BIO_LINEBUFFER_CTX *ctx;
  138. if ((in == NULL) || (inl <= 0)) return(0);
  139. ctx=(BIO_LINEBUFFER_CTX *)b->ptr;
  140. if ((ctx == NULL) || (b->next_bio == NULL)) return(0);
  141. BIO_clear_retry_flags(b);
  142. do
  143. {
  144. const char *p;
  145. for(p = in; p < in + inl && *p != 'n'; p++)
  146. ;
  147. if (*p == 'n')
  148. {
  149. p++;
  150. foundnl = 1;
  151. }
  152. else
  153. foundnl = 0;
  154. /* If a NL was found and we already have text in the save
  155.    buffer, concatenate them and write */
  156. while ((foundnl || p - in > ctx->obuf_size - ctx->obuf_len)
  157. && ctx->obuf_len > 0)
  158. {
  159. int orig_olen = ctx->obuf_len;
  160. i = ctx->obuf_size - ctx->obuf_len;
  161. if (p - in > 0)
  162. {
  163. if (i >= p - in)
  164. {
  165. memcpy(&(ctx->obuf[ctx->obuf_len]),
  166. in,p - in);
  167. ctx->obuf_len += p - in;
  168. inl -= p - in;
  169. num += p - in;
  170. in = p;
  171. }
  172. else
  173. {
  174. memcpy(&(ctx->obuf[ctx->obuf_len]),
  175. in,i);
  176. ctx->obuf_len += i;
  177. inl -= i;
  178. in += i;
  179. num += i;
  180. }
  181. }
  182. #if 0
  183. BIO_write(b->next_bio, "<*<", 3);
  184. #endif
  185. i=BIO_write(b->next_bio,
  186. ctx->obuf, ctx->obuf_len);
  187. if (i <= 0)
  188. {
  189. ctx->obuf_len = orig_olen;
  190. BIO_copy_next_retry(b);
  191. #if 0
  192. BIO_write(b->next_bio, ">*>", 3);
  193. #endif
  194. if (i < 0) return((num > 0)?num:i);
  195. if (i == 0) return(num);
  196. }
  197. #if 0
  198. BIO_write(b->next_bio, ">*>", 3);
  199. #endif
  200. if (i < ctx->obuf_len)
  201. memmove(ctx->obuf, ctx->obuf + i,
  202. ctx->obuf_len - i);
  203. ctx->obuf_len-=i;
  204. }
  205. /* Now that the save buffer is emptied, let's write the input
  206.    buffer if a NL was found and there is anything to write. */
  207. if ((foundnl || p - in > ctx->obuf_size) && p - in > 0)
  208. {
  209. #if 0
  210. BIO_write(b->next_bio, "<*<", 3);
  211. #endif
  212. i=BIO_write(b->next_bio,in,p - in);
  213. if (i <= 0)
  214. {
  215. BIO_copy_next_retry(b);
  216. #if 0
  217. BIO_write(b->next_bio, ">*>", 3);
  218. #endif
  219. if (i < 0) return((num > 0)?num:i);
  220. if (i == 0) return(num);
  221. }
  222. #if 0
  223. BIO_write(b->next_bio, ">*>", 3);
  224. #endif
  225. num+=i;
  226. in+=i;
  227. inl-=i;
  228. }
  229. }
  230. while(foundnl && inl > 0);
  231. /* We've written as much as we can.  The rest of the input buffer, if
  232.    any, is text that doesn't and with a NL and therefore needs to be
  233.    saved for the next trip. */
  234. if (inl > 0)
  235. {
  236. memcpy(&(ctx->obuf[ctx->obuf_len]), in, inl);
  237. ctx->obuf_len += inl;
  238. num += inl;
  239. }
  240. return num;
  241. }
  242. static long linebuffer_ctrl(BIO *b, int cmd, long num, void *ptr)
  243. {
  244. BIO *dbio;
  245. BIO_LINEBUFFER_CTX *ctx;
  246. long ret=1;
  247. char *p;
  248. int r;
  249. int obs;
  250. ctx=(BIO_LINEBUFFER_CTX *)b->ptr;
  251. switch (cmd)
  252. {
  253. case BIO_CTRL_RESET:
  254. ctx->obuf_len=0;
  255. if (b->next_bio == NULL) return(0);
  256. ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
  257. break;
  258. case BIO_CTRL_INFO:
  259. ret=(long)ctx->obuf_len;
  260. break;
  261. case BIO_CTRL_WPENDING:
  262. ret=(long)ctx->obuf_len;
  263. if (ret == 0)
  264. {
  265. if (b->next_bio == NULL) return(0);
  266. ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
  267. }
  268. break;
  269. case BIO_C_SET_BUFF_SIZE:
  270. obs=(int)num;
  271. p=ctx->obuf;
  272. if ((obs > DEFAULT_LINEBUFFER_SIZE) && (obs != ctx->obuf_size))
  273. {
  274. p=(char *)OPENSSL_malloc((int)num);
  275. if (p == NULL)
  276. goto malloc_error;
  277. }
  278. if (ctx->obuf != p)
  279. {
  280. if (ctx->obuf_len > obs)
  281. {
  282. ctx->obuf_len = obs;
  283. }
  284. memcpy(p, ctx->obuf, ctx->obuf_len);
  285. OPENSSL_free(ctx->obuf);
  286. ctx->obuf=p;
  287. ctx->obuf_size=obs;
  288. }
  289. break;
  290. case BIO_C_DO_STATE_MACHINE:
  291. if (b->next_bio == NULL) return(0);
  292. BIO_clear_retry_flags(b);
  293. ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
  294. BIO_copy_next_retry(b);
  295. break;
  296. case BIO_CTRL_FLUSH:
  297. if (b->next_bio == NULL) return(0);
  298. if (ctx->obuf_len <= 0)
  299. {
  300. ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
  301. break;
  302. }
  303. for (;;)
  304. {
  305. BIO_clear_retry_flags(b);
  306. if (ctx->obuf_len > 0)
  307. {
  308. r=BIO_write(b->next_bio,
  309. ctx->obuf, ctx->obuf_len);
  310. #if 0
  311. fprintf(stderr,"FLUSH %3d -> %3dn",ctx->obuf_len,r);
  312. #endif
  313. BIO_copy_next_retry(b);
  314. if (r <= 0) return((long)r);
  315. if (r < ctx->obuf_len)
  316. memmove(ctx->obuf, ctx->obuf + r,
  317. ctx->obuf_len - r);
  318. ctx->obuf_len-=r;
  319. }
  320. else
  321. {
  322. ctx->obuf_len=0;
  323. ret=1;
  324. break;
  325. }
  326. }
  327. ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
  328. break;
  329. case BIO_CTRL_DUP:
  330. dbio=(BIO *)ptr;
  331. if ( !BIO_set_write_buffer_size(dbio,ctx->obuf_size))
  332. ret=0;
  333. break;
  334. default:
  335. if (b->next_bio == NULL) return(0);
  336. ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
  337. break;
  338. }
  339. return(ret);
  340. malloc_error:
  341. BIOerr(BIO_F_LINEBUFFER_CTRL,ERR_R_MALLOC_FAILURE);
  342. return(0);
  343. }
  344. static long linebuffer_callback_ctrl(BIO *b, int cmd, bio_info_cb *fp)
  345. {
  346. long ret=1;
  347. if (b->next_bio == NULL) return(0);
  348. switch (cmd)
  349. {
  350. default:
  351. ret=BIO_callback_ctrl(b->next_bio,cmd,fp);
  352. break;
  353. }
  354. return(ret);
  355. }
  356. static int linebuffer_gets(BIO *b, char *buf, int size)
  357. {
  358. if (b->next_bio == NULL) return(0);
  359. return(BIO_gets(b->next_bio,buf,size));
  360. }
  361. static int linebuffer_puts(BIO *b, const char *str)
  362. {
  363. return(linebuffer_write(b,str,strlen(str)));
  364. }