Add missing #include.
[oweals/openssl.git] / crypto / bio / bf_buff.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
59 #include <stdio.h>
60 #include <errno.h>
61 #include "cryptlib.h"
62 #include <openssl/bio.h>
63 #include <openssl/evp.h>
64
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 static long buffer_callback_ctrl(BIO *h,int cmd, void (*fp)());
73 #define DEFAULT_BUFFER_SIZE     1024
74
75 static BIO_METHOD methods_buffer=
76         {
77         BIO_TYPE_BUFFER,
78         "buffer",
79         buffer_write,
80         buffer_read,
81         buffer_puts,
82         buffer_gets,
83         buffer_ctrl,
84         buffer_new,
85         buffer_free,
86         buffer_callback_ctrl,
87         };
88
89 BIO_METHOD *BIO_f_buffer(void)
90         {
91         return(&methods_buffer);
92         }
93
94 static int buffer_new(BIO *bi)
95         {
96         BIO_F_BUFFER_CTX *ctx;
97
98         ctx=(BIO_F_BUFFER_CTX *)Malloc(sizeof(BIO_F_BUFFER_CTX));
99         if (ctx == NULL) return(0);
100         ctx->ibuf=(char *)Malloc(DEFAULT_BUFFER_SIZE);
101         if (ctx->ibuf == NULL) { Free(ctx); return(0); }
102         ctx->obuf=(char *)Malloc(DEFAULT_BUFFER_SIZE);
103         if (ctx->obuf == NULL) { Free(ctx->ibuf); Free(ctx); return(0); }
104         ctx->ibuf_size=DEFAULT_BUFFER_SIZE;
105         ctx->obuf_size=DEFAULT_BUFFER_SIZE;
106         ctx->ibuf_len=0;
107         ctx->ibuf_off=0;
108         ctx->obuf_len=0;
109         ctx->obuf_off=0;
110
111         bi->init=1;
112         bi->ptr=(char *)ctx;
113         bi->flags=0;
114         return(1);
115         }
116
117 static int buffer_free(BIO *a)
118         {
119         BIO_F_BUFFER_CTX *b;
120
121         if (a == NULL) return(0);
122         b=(BIO_F_BUFFER_CTX *)a->ptr;
123         if (b->ibuf != NULL) Free(b->ibuf);
124         if (b->obuf != NULL) Free(b->obuf);
125         Free(a->ptr);
126         a->ptr=NULL;
127         a->init=0;
128         a->flags=0;
129         return(1);
130         }
131         
132 static int buffer_read(BIO *b, char *out, int outl)
133         {
134         int i,num=0;
135         BIO_F_BUFFER_CTX *ctx;
136
137         if (out == NULL) return(0);
138         ctx=(BIO_F_BUFFER_CTX *)b->ptr;
139
140         if ((ctx == NULL) || (b->next_bio == NULL)) return(0);
141         num=0;
142         BIO_clear_retry_flags(b);
143
144 start:
145         i=ctx->ibuf_len;
146         /* If there is stuff left over, grab it */
147         if (i != 0)
148                 {
149                 if (i > outl) i=outl;
150                 memcpy(out,&(ctx->ibuf[ctx->ibuf_off]),i);
151                 ctx->ibuf_off+=i;
152                 ctx->ibuf_len-=i;
153                 num+=i;
154                 if (outl == i)  return(num);
155                 outl-=i;
156                 out+=i;
157                 }
158
159         /* We may have done a partial read. try to do more.
160          * We have nothing in the buffer.
161          * If we get an error and have read some data, just return it
162          * and let them retry to get the error again.
163          * copy direct to parent address space */
164         if (outl > ctx->ibuf_size)
165                 {
166                 for (;;)
167                         {
168                         i=BIO_read(b->next_bio,out,outl);
169                         if (i <= 0)
170                                 {
171                                 BIO_copy_next_retry(b);
172                                 if (i < 0) return((num > 0)?num:i);
173                                 if (i == 0) return(num);
174                                 }
175                         num+=i;
176                         if (outl == i) return(num);
177                         out+=i;
178                         outl-=i;
179                         }
180                 }
181         /* else */
182
183         /* we are going to be doing some buffering */
184         i=BIO_read(b->next_bio,ctx->ibuf,ctx->ibuf_size);
185         if (i <= 0)
186                 {
187                 BIO_copy_next_retry(b);
188                 if (i < 0) return((num > 0)?num:i);
189                 if (i == 0) return(num);
190                 }
191         ctx->ibuf_off=0;
192         ctx->ibuf_len=i;
193
194         /* Lets re-read using ourselves :-) */
195         goto start;
196         }
197
198 static int buffer_write(BIO *b, char *in, int inl)
199         {
200         int i,num=0;
201         BIO_F_BUFFER_CTX *ctx;
202
203         if ((in == NULL) || (inl <= 0)) return(0);
204         ctx=(BIO_F_BUFFER_CTX *)b->ptr;
205         if ((ctx == NULL) || (b->next_bio == NULL)) return(0);
206
207         BIO_clear_retry_flags(b);
208 start:
209         i=ctx->obuf_size-(ctx->obuf_len+ctx->obuf_off);
210         /* add to buffer and return */
211         if (i >= inl)
212                 {
213                 memcpy(&(ctx->obuf[ctx->obuf_len]),in,inl);
214                 ctx->obuf_len+=inl;
215                 return(num+inl);
216                 }
217         /* else */
218         /* stuff already in buffer, so add to it first, then flush */
219         if (ctx->obuf_len != 0)
220                 {
221                 if (i > 0) /* lets fill it up if we can */
222                         {
223                         memcpy(&(ctx->obuf[ctx->obuf_len]),in,i);
224                         in+=i;
225                         inl-=i;
226                         num+=i;
227                         ctx->obuf_len+=i;
228                         }
229                 /* we now have a full buffer needing flushing */
230                 for (;;)
231                         {
232                         i=BIO_write(b->next_bio,&(ctx->obuf[ctx->obuf_off]),
233                                 ctx->obuf_len);
234                         if (i <= 0)
235                                 {
236                                 BIO_copy_next_retry(b);
237
238                                 if (i < 0) return((num > 0)?num:i);
239                                 if (i == 0) return(num);
240                                 }
241                         ctx->obuf_off+=i;
242                         ctx->obuf_len-=i;
243                         if (ctx->obuf_len == 0) break;
244                         }
245                 }
246         /* we only get here if the buffer has been flushed and we
247          * still have stuff to write */
248         ctx->obuf_off=0;
249
250         /* we now have inl bytes to write */
251         while (inl >= ctx->obuf_size)
252                 {
253                 i=BIO_write(b->next_bio,in,inl);
254                 if (i <= 0)
255                         {
256                         BIO_copy_next_retry(b);
257                         if (i < 0) return((num > 0)?num:i);
258                         if (i == 0) return(num);
259                         }
260                 num+=i;
261                 in+=i;
262                 inl-=i;
263                 if (inl == 0) return(num);
264                 }
265
266         /* copy the rest into the buffer since we have only a small 
267          * amount left */
268         goto start;
269         }
270
271 static long buffer_ctrl(BIO *b, int cmd, long num, char *ptr)
272         {
273         BIO *dbio;
274         BIO_F_BUFFER_CTX *ctx;
275         long ret=1;
276         char *p1,*p2;
277         int r,i,*ip;
278         int ibs,obs;
279
280         ctx=(BIO_F_BUFFER_CTX *)b->ptr;
281
282         switch (cmd)
283                 {
284         case BIO_CTRL_RESET:
285                 ctx->ibuf_off=0;
286                 ctx->ibuf_len=0;
287                 ctx->obuf_off=0;
288                 ctx->obuf_len=0;
289                 if (b->next_bio == NULL) return(0);
290                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
291                 break;
292         case BIO_CTRL_INFO:
293                 ret=(long)ctx->obuf_len;
294                 break;
295         case BIO_C_GET_BUFF_NUM_LINES:
296                 ret=0;
297                 p1=ctx->ibuf;
298                 for (i=ctx->ibuf_off; i<ctx->ibuf_len; i++)
299                         {
300                         if (p1[i] == '\n') ret++;
301                         }
302                 break;
303         case BIO_CTRL_WPENDING:
304                 ret=(long)ctx->obuf_len;
305                 if (ret == 0)
306                         {
307                         if (b->next_bio == NULL) return(0);
308                         ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
309                         }
310                 break;
311         case BIO_CTRL_PENDING:
312                 ret=(long)ctx->ibuf_len;
313                 if (ret == 0)
314                         {
315                         if (b->next_bio == NULL) return(0);
316                         ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
317                         }
318                 break;
319         case BIO_C_SET_BUFF_READ_DATA:
320                 if (num > ctx->ibuf_size)
321                         {
322                         p1=Malloc((int)num);
323                         if (p1 == NULL) goto malloc_error;
324                         if (ctx->ibuf != NULL) Free(ctx->ibuf);
325                         ctx->ibuf=p1;
326                         }
327                 ctx->ibuf_off=0;
328                 ctx->ibuf_len=(int)num;
329                 memcpy(ctx->ibuf,ptr,(int)num);
330                 ret=1;
331                 break;
332         case BIO_C_SET_BUFF_SIZE:
333                 if (ptr != NULL)
334                         {
335                         ip=(int *)ptr;
336                         if (*ip == 0)
337                                 {
338                                 ibs=(int)num;
339                                 obs=ctx->obuf_size;
340                                 }
341                         else /* if (*ip == 1) */
342                                 {
343                                 ibs=ctx->ibuf_size;
344                                 obs=(int)num;
345                                 }
346                         }
347                 else
348                         {
349                         ibs=(int)num;
350                         obs=(int)num;
351                         }
352                 p1=ctx->ibuf;
353                 p2=ctx->obuf;
354                 if ((ibs > DEFAULT_BUFFER_SIZE) && (ibs != ctx->ibuf_size))
355                         {
356                         p1=(char *)Malloc((int)num);
357                         if (p1 == NULL) goto malloc_error;
358                         }
359                 if ((obs > DEFAULT_BUFFER_SIZE) && (obs != ctx->obuf_size))
360                         {
361                         p2=(char *)Malloc((int)num);
362                         if (p2 == NULL)
363                                 {
364                                 if (p1 != ctx->ibuf) Free(p1);
365                                 goto malloc_error;
366                                 }
367                         }
368                 if (ctx->ibuf != p1)
369                         {
370                         Free(ctx->ibuf);
371                         ctx->ibuf=p1;
372                         ctx->ibuf_off=0;
373                         ctx->ibuf_len=0;
374                         ctx->ibuf_size=ibs;
375                         }
376                 if (ctx->obuf != p2)
377                         {
378                         Free(ctx->obuf);
379                         ctx->obuf=p2;
380                         ctx->obuf_off=0;
381                         ctx->obuf_len=0;
382                         ctx->obuf_size=obs;
383                         }
384                 break;
385         case BIO_C_DO_STATE_MACHINE:
386                 if (b->next_bio == NULL) return(0);
387                 BIO_clear_retry_flags(b);
388                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
389                 BIO_copy_next_retry(b);
390                 break;
391
392         case BIO_CTRL_FLUSH:
393                 if (b->next_bio == NULL) return(0);
394                 if (ctx->obuf_len <= 0)
395                         {
396                         ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
397                         break;
398                         }
399
400                 for (;;)
401                         {
402                         BIO_clear_retry_flags(b);
403                         if (ctx->obuf_len > ctx->obuf_off)
404                                 {
405                                 r=BIO_write(b->next_bio,
406                                         &(ctx->obuf[ctx->obuf_off]),
407                                         ctx->obuf_len-ctx->obuf_off);
408 #if 0
409 fprintf(stderr,"FLUSH [%3d] %3d -> %3d\n",ctx->obuf_off,ctx->obuf_len-ctx->obuf_off,r);
410 #endif
411                                 BIO_copy_next_retry(b);
412                                 if (r <= 0) return((long)r);
413                                 ctx->obuf_off+=r;
414                                 }
415                         else
416                                 {
417                                 ctx->obuf_len=0;
418                                 ctx->obuf_off=0;
419                                 ret=1;
420                                 break;
421                                 }
422                         }
423                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
424                 break;
425         case BIO_CTRL_DUP:
426                 dbio=(BIO *)ptr;
427                 if (    !BIO_set_read_buffer_size(dbio,ctx->ibuf_size) ||
428                         !BIO_set_write_buffer_size(dbio,ctx->obuf_size))
429                         ret=0;
430                 break;
431         default:
432                 if (b->next_bio == NULL) return(0);
433                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
434                 break;
435                 }
436         return(ret);
437 malloc_error:
438         BIOerr(BIO_F_BUFFER_CTRL,ERR_R_MALLOC_FAILURE);
439         return(0);
440         }
441
442 static long buffer_callback_ctrl(BIO *b, int cmd, void (*fp)())
443         {
444         long ret=1;
445
446         if (b->next_bio == NULL) return(0);
447         switch (cmd)
448                 {
449         default:
450                 ret=BIO_callback_ctrl(b->next_bio,cmd,fp);
451                 break;
452                 }
453         return(ret);
454         }
455
456 static int buffer_gets(BIO *b, char *buf, int size)
457         {
458         BIO_F_BUFFER_CTX *ctx;
459         int num=0,i,flag;
460         char *p;
461
462         ctx=(BIO_F_BUFFER_CTX *)b->ptr;
463         size--; /* reserve space for a '\0' */
464         BIO_clear_retry_flags(b);
465
466         for (;;)
467                 {
468                 if (ctx->ibuf_len > 0)
469                         {
470                         p= &(ctx->ibuf[ctx->ibuf_off]);
471                         flag=0;
472                         for (i=0; (i<ctx->ibuf_len) && (i<size); i++)
473                                 {
474                                 *(buf++)=p[i];
475                                 if (p[i] == '\n')
476                                         {
477                                         flag=1;
478                                         i++;
479                                         break;
480                                         }
481                                 }
482                         num+=i;
483                         size-=i;
484                         ctx->ibuf_len-=i;
485                         ctx->ibuf_off+=i;
486                         if ((flag) || (i == size))
487                                 {
488                                 *buf='\0';
489                                 return(num);
490                                 }
491                         }
492                 else    /* read another chunk */
493                         {
494                         i=BIO_read(b->next_bio,ctx->ibuf,ctx->ibuf_size);
495                         if (i <= 0)
496                                 {
497                                 BIO_copy_next_retry(b);
498                                 if (i < 0) return((num > 0)?num:i);
499                                 if (i == 0) return(num);
500                                 }
501                         ctx->ibuf_len=i;
502                         ctx->ibuf_off=0;
503                         }
504                 }
505         }
506
507 static int buffer_puts(BIO *b, char *str)
508         {
509         return(BIO_write(b,str,strlen(str)));
510         }
511