1 /* crypto/bio/bf_buff.c */
2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
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.
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).
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.
23 * Redistribution and use in source and binary forms, with or without
24 * modification, are permitted provided that the following conditions
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)"
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
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.]
62 #include <openssl/bio.h>
63 #include <openssl/evp.h>
65 static int buffer_write(BIO *h,char *buf,int num);
66 static int buffer_read(BIO *h,char *buf,int size);
67 static int buffer_puts(BIO *h,char *str);
68 static int buffer_gets(BIO *h,char *str,int size);
69 static long buffer_ctrl(BIO *h,int cmd,long arg1,char *arg2);
70 static int buffer_new(BIO *h);
71 static int buffer_free(BIO *data);
72 #define DEFAULT_BUFFER_SIZE 1024
74 static BIO_METHOD methods_buffer=
87 BIO_METHOD *BIO_f_buffer(void)
89 return(&methods_buffer);
92 static int buffer_new(BIO *bi)
94 BIO_F_BUFFER_CTX *ctx;
96 ctx=(BIO_F_BUFFER_CTX *)Malloc(sizeof(BIO_F_BUFFER_CTX));
97 if (ctx == NULL) return(0);
98 ctx->ibuf=(char *)Malloc(DEFAULT_BUFFER_SIZE);
99 if (ctx->ibuf == NULL) { Free(ctx); return(0); }
100 ctx->obuf=(char *)Malloc(DEFAULT_BUFFER_SIZE);
101 if (ctx->obuf == NULL) { Free(ctx->ibuf); Free(ctx); return(0); }
102 ctx->ibuf_size=DEFAULT_BUFFER_SIZE;
103 ctx->obuf_size=DEFAULT_BUFFER_SIZE;
115 static int buffer_free(BIO *a)
119 if (a == NULL) return(0);
120 b=(BIO_F_BUFFER_CTX *)a->ptr;
121 if (b->ibuf != NULL) Free(b->ibuf);
122 if (b->obuf != NULL) Free(b->obuf);
130 static int buffer_read(BIO *b, char *out, int outl)
133 BIO_F_BUFFER_CTX *ctx;
135 if (out == NULL) return(0);
136 ctx=(BIO_F_BUFFER_CTX *)b->ptr;
138 if ((ctx == NULL) || (b->next_bio == NULL)) return(0);
140 BIO_clear_retry_flags(b);
144 /* If there is stuff left over, grab it */
147 if (i > outl) i=outl;
148 memcpy(out,&(ctx->ibuf[ctx->ibuf_off]),i);
152 if (outl == i) return(num);
157 /* We may have done a partial read. try to do more.
158 * We have nothing in the buffer.
159 * If we get an error and have read some data, just return it
160 * and let them retry to get the error again.
161 * copy direct to parent address space */
162 if (outl > ctx->ibuf_size)
166 i=BIO_read(b->next_bio,out,outl);
169 BIO_copy_next_retry(b);
170 if (i < 0) return((num > 0)?num:i);
171 if (i == 0) return(num);
174 if (outl == i) return(num);
181 /* we are going to be doing some buffering */
182 i=BIO_read(b->next_bio,ctx->ibuf,ctx->ibuf_size);
185 BIO_copy_next_retry(b);
186 if (i < 0) return((num > 0)?num:i);
187 if (i == 0) return(num);
192 /* Lets re-read using ourselves :-) */
196 static int buffer_write(BIO *b, char *in, int inl)
199 BIO_F_BUFFER_CTX *ctx;
201 if ((in == NULL) || (inl <= 0)) return(0);
202 ctx=(BIO_F_BUFFER_CTX *)b->ptr;
203 if ((ctx == NULL) || (b->next_bio == NULL)) return(0);
205 BIO_clear_retry_flags(b);
207 i=ctx->obuf_size-(ctx->obuf_len+ctx->obuf_off);
208 /* add to buffer and return */
211 memcpy(&(ctx->obuf[ctx->obuf_len]),in,inl);
216 /* stuff already in buffer, so add to it first, then flush */
217 if (ctx->obuf_len != 0)
219 if (i > 0) /* lets fill it up if we can */
221 memcpy(&(ctx->obuf[ctx->obuf_len]),in,i);
227 /* we now have a full buffer needing flushing */
230 i=BIO_write(b->next_bio,&(ctx->obuf[ctx->obuf_off]),
234 BIO_copy_next_retry(b);
236 if (i < 0) return((num > 0)?num:i);
237 if (i == 0) return(num);
241 if (ctx->obuf_len == 0) break;
244 /* we only get here if the buffer has been flushed and we
245 * still have stuff to write */
248 /* we now have inl bytes to write */
249 while (inl >= ctx->obuf_size)
251 i=BIO_write(b->next_bio,in,inl);
254 BIO_copy_next_retry(b);
255 if (i < 0) return((num > 0)?num:i);
256 if (i == 0) return(num);
261 if (inl == 0) return(num);
264 /* copy the rest into the buffer since we have only a small
269 static long buffer_ctrl(BIO *b, int cmd, long num, char *ptr)
272 BIO_F_BUFFER_CTX *ctx;
278 ctx=(BIO_F_BUFFER_CTX *)b->ptr;
287 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
290 ret=(long)ctx->obuf_len;
292 case BIO_C_GET_BUFF_NUM_LINES:
295 for (i=ctx->ibuf_off; i<ctx->ibuf_len; i++)
297 if (p1[i] == '\n') ret++;
300 case BIO_CTRL_WPENDING:
301 ret=(long)ctx->obuf_len;
303 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
305 case BIO_CTRL_PENDING:
306 ret=(long)ctx->ibuf_len;
308 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
310 case BIO_C_SET_BUFF_READ_DATA:
311 if (num > ctx->ibuf_size)
314 if (p1 == NULL) goto malloc_error;
315 if (ctx->ibuf != NULL) Free(ctx->ibuf);
319 ctx->ibuf_len=(int)num;
320 memcpy(ctx->ibuf,ptr,(int)num);
323 case BIO_C_SET_BUFF_SIZE:
332 else /* if (*ip == 1) */
345 if ((ibs > DEFAULT_BUFFER_SIZE) && (ibs != ctx->ibuf_size))
347 p1=(char *)Malloc((int)num);
348 if (p1 == NULL) goto malloc_error;
350 if ((obs > DEFAULT_BUFFER_SIZE) && (obs != ctx->obuf_size))
352 p2=(char *)Malloc((int)num);
355 if (p1 != ctx->ibuf) Free(p1);
376 case BIO_C_DO_STATE_MACHINE:
377 BIO_clear_retry_flags(b);
378 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
379 BIO_copy_next_retry(b);
383 if (ctx->obuf_len <= 0)
385 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
391 BIO_clear_retry_flags(b);
392 if (ctx->obuf_len > ctx->obuf_off)
394 r=BIO_write(b->next_bio,
395 &(ctx->obuf[ctx->obuf_off]),
396 ctx->obuf_len-ctx->obuf_off);
398 fprintf(stderr,"FLUSH [%3d] %3d -> %3d\n",ctx->obuf_off,ctx->obuf_len-ctx->obuf_off,r);
400 BIO_copy_next_retry(b);
401 if (r <= 0) return((long)r);
412 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
416 if ( !BIO_set_read_buffer_size(dbio,ctx->ibuf_size) ||
417 !BIO_set_write_buffer_size(dbio,ctx->obuf_size))
421 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
426 BIOerr(BIO_F_BUFFER_CTRL,ERR_R_MALLOC_FAILURE);
430 static int buffer_gets(BIO *b, char *buf, int size)
432 BIO_F_BUFFER_CTX *ctx;
436 ctx=(BIO_F_BUFFER_CTX *)b->ptr;
437 size--; /* reserve space for a '\0' */
438 BIO_clear_retry_flags(b);
442 if (ctx->ibuf_len > 0)
444 p= &(ctx->ibuf[ctx->ibuf_off]);
446 for (i=0; (i<ctx->ibuf_len) && (i<size); i++)
460 if ((flag) || (i == size))
466 else /* read another chunk */
468 i=BIO_read(b->next_bio,ctx->ibuf,ctx->ibuf_size);
471 BIO_copy_next_retry(b);
472 if (i < 0) return((num > 0)?num:i);
473 if (i == 0) return(num);
481 static int buffer_puts(BIO *b, char *str)
483 return(BIO_write(b,str,strlen(str)));