crypto/openssl-0.9/crypto/bio/bf_lbuf.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 linebuffer_write(BIO *h, const char *buf,int num);
  66 static int linebuffer_read(BIO *h, char *buf, int size);
  67 static int linebuffer_puts(BIO *h, const char *str);
  68 static int linebuffer_gets(BIO *h, char *str, int size);
  69 static long linebuffer_ctrl(BIO *h, int cmd, long arg1, void *arg2);
  70 static int linebuffer_new(BIO *h);
  71 static int linebuffer_free(BIO *data);
  72 static long linebuffer_callback_ctrl(BIO *h, int cmd, bio_info_cb *fp);
  73
  74 /* A 10k maximum should be enough for most purposes */
  75 #define DEFAULT_LINEBUFFER_SIZE 1024*10
  76
  77 /* #define DEBUG */
  78
  79 static BIO_METHOD methods_linebuffer=
  80         {
  81         BIO_TYPE_LINEBUFFER,
  82         "linebuffer",
  83         linebuffer_write,
  84         linebuffer_read,
  85         linebuffer_puts,
  86         linebuffer_gets,
  87         linebuffer_ctrl,
  88         linebuffer_new,
  89         linebuffer_free,
  90         linebuffer_callback_ctrl,
  91         };
  92
  93 BIO_METHOD *BIO_f_linebuffer(void)
  94         {
  95         return(&methods_linebuffer);
  96         }
  97
  98 typedef struct bio_linebuffer_ctx_struct
  99         {
 100         char *obuf;             /* the output char array */
 101         int obuf_size;          /* how big is the output buffer */
 102         int obuf_len;           /* how many bytes are in it */
 103         } BIO_LINEBUFFER_CTX;
 104
 105 static int linebuffer_new(BIO *bi)
 106         {
 107         BIO_LINEBUFFER_CTX *ctx;
 108
 109         ctx=(BIO_LINEBUFFER_CTX *)OPENSSL_malloc(sizeof(BIO_LINEBUFFER_CTX));
 110         if (ctx == NULL) return(0);
 111         ctx->obuf=(char *)OPENSSL_malloc(DEFAULT_LINEBUFFER_SIZE);
 112         if (ctx->obuf == NULL) { OPENSSL_free(ctx); return(0); }
 113         ctx->obuf_size=DEFAULT_LINEBUFFER_SIZE;
 114         ctx->obuf_len=0;
 115
 116         bi->init=1;
 117         bi->ptr=(char *)ctx;
 118         bi->flags=0;
 119         return(1);
 120         }
 121
 122 static int linebuffer_free(BIO *a)
 123         {
 124         BIO_LINEBUFFER_CTX *b;
 125
 126         if (a == NULL) return(0);
 127         b=(BIO_LINEBUFFER_CTX *)a->ptr;
 128         if (b->obuf != NULL) OPENSSL_free(b->obuf);
 129         OPENSSL_free(a->ptr);
 130         a->ptr=NULL;
 131         a->init=0;
 132         a->flags=0;
 133         return(1);
 134         }
 135
 136 static int linebuffer_read(BIO *b, char *out, int outl)
 137         {
 138         int ret=0;
 139
 140         if (out == NULL) return(0);
 141         if (b->next_bio == NULL) return(0);
 142         ret=BIO_read(b->next_bio,out,outl);
 143         BIO_clear_retry_flags(b);
 144         BIO_copy_next_retry(b);
 145         return(ret);
 146         }
 147
 148 static int linebuffer_write(BIO *b, const char *in, int inl)
 149         {
 150         int i,num=0,foundnl;
 151         BIO_LINEBUFFER_CTX *ctx;
 152
 153         if ((in == NULL) || (inl <= 0)) return(0);
 154         ctx=(BIO_LINEBUFFER_CTX *)b->ptr;
 155         if ((ctx == NULL) || (b->next_bio == NULL)) return(0);
 156
 157         BIO_clear_retry_flags(b);
 158
 159         do
 160                 {
 161                 const char *p;
 162
 163                 for(p = in; p < in + inl && *p != '\n'; p++)
 164                         ;
 165                 if (*p == '\n')
 166                         {
 167                         p++;
 168                         foundnl = 1;
 169                         }
 170                 else
 171                         foundnl = 0;
 172
 173                 /* If a NL was found and we already have text in the save
 174                    buffer, concatenate them and write */
 175                 while ((foundnl || p - in > ctx->obuf_size - ctx->obuf_len)
 176                         && ctx->obuf_len > 0)
 177                         {
 178                         int orig_olen = ctx->obuf_len;
 179
 180                         i = ctx->obuf_size - ctx->obuf_len;
 181                         if (p - in > 0)
 182                                 {
 183                                 if (i >= p - in)
 184                                         {
 185                                         memcpy(&(ctx->obuf[ctx->obuf_len]),
 186                                                 in,p - in);
 187                                         ctx->obuf_len += p - in;
 188                                         inl -= p - in;
 189                                         num += p - in;
 190                                         in = p;
 191                                         }
 192                                 else
 193                                         {
 194                                         memcpy(&(ctx->obuf[ctx->obuf_len]),
 195                                                 in,i);
 196                                         ctx->obuf_len += i;
 197                                         inl -= i;
 198                                         in += i;
 199                                         num += i;
 200                                         }
 201                                 }
 202
 203 #if 0
 204 BIO_write(b->next_bio, "<*<", 3);
 205 #endif
 206                         i=BIO_write(b->next_bio,
 207                                 ctx->obuf, ctx->obuf_len);
 208                         if (i <= 0)
 209                                 {
 210                                 ctx->obuf_len = orig_olen;
 211                                 BIO_copy_next_retry(b);
 212
 213 #if 0
 214 BIO_write(b->next_bio, ">*>", 3);
 215 #endif
 216                                 if (i < 0) return((num > 0)?num:i);
 217                                 if (i == 0) return(num);
 218                                 }
 219 #if 0
 220 BIO_write(b->next_bio, ">*>", 3);
 221 #endif
 222                         if (i < ctx->obuf_len)
 223                                 memmove(ctx->obuf, ctx->obuf + i,
 224                                         ctx->obuf_len - i);
 225                         ctx->obuf_len-=i;
 226                         }
 227
 228                 /* Now that the save buffer is emptied, let's write the input
 229                    buffer if a NL was found and there is anything to write. */
 230                 if ((foundnl || p - in > ctx->obuf_size) && p - in > 0)
 231                         {
 232 #if 0
 233 BIO_write(b->next_bio, "<*<", 3);
 234 #endif
 235                         i=BIO_write(b->next_bio,in,p - in);
 236                         if (i <= 0)
 237                                 {
 238                                 BIO_copy_next_retry(b);
 239 #if 0
 240 BIO_write(b->next_bio, ">*>", 3);
 241 #endif
 242                                 if (i < 0) return((num > 0)?num:i);
 243                                 if (i == 0) return(num);
 244                                 }
 245 #if 0
 246 BIO_write(b->next_bio, ">*>", 3);
 247 #endif
 248                         num+=i;
 249                         in+=i;
 250                         inl-=i;
 251                         }
 252                 }
 253         while(foundnl && inl > 0);
 254         /* We've written as much as we can.  The rest of the input buffer, if
 255            any, is text that doesn't and with a NL and therefore needs to be
 256            saved for the next trip. */
 257         if (inl > 0)
 258                 {
 259                 memcpy(&(ctx->obuf[ctx->obuf_len]), in, inl);
 260                 ctx->obuf_len += inl;
 261                 num += inl;
 262                 }
 263         return num;
 264         }
 265
 266 static long linebuffer_ctrl(BIO *b, int cmd, long num, void *ptr)
 267         {
 268         BIO *dbio;
 269         BIO_LINEBUFFER_CTX *ctx;
 270         long ret=1;
 271         char *p;
 272         int r;
 273         int obs;
 274
 275         ctx=(BIO_LINEBUFFER_CTX *)b->ptr;
 276
 277         switch (cmd)
 278                 {
 279         case BIO_CTRL_RESET:
 280                 ctx->obuf_len=0;
 281                 if (b->next_bio == NULL) return(0);
 282                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
 283                 break;
 284         case BIO_CTRL_INFO:
 285                 ret=(long)ctx->obuf_len;
 286                 break;
 287         case BIO_CTRL_WPENDING:
 288                 ret=(long)ctx->obuf_len;
 289                 if (ret == 0)
 290                         {
 291                         if (b->next_bio == NULL) return(0);
 292                         ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
 293                         }
 294                 break;
 295         case BIO_C_SET_BUFF_SIZE:
 296                 obs=(int)num;
 297                 p=ctx->obuf;
 298                 if ((obs > DEFAULT_LINEBUFFER_SIZE) && (obs != ctx->obuf_size))
 299                         {
 300                         p=(char *)OPENSSL_malloc((int)num);
 301                         if (p == NULL)
 302                                 goto malloc_error;
 303                         }
 304                 if (ctx->obuf != p)
 305                         {
 306                         if (ctx->obuf_len > obs)
 307                                 {
 308                                 ctx->obuf_len = obs;
 309                                 }
 310                         memcpy(p, ctx->obuf, ctx->obuf_len);
 311                         OPENSSL_free(ctx->obuf);
 312                         ctx->obuf=p;
 313                         ctx->obuf_size=obs;
 314                         }
 315                 break;
 316         case BIO_C_DO_STATE_MACHINE:
 317                 if (b->next_bio == NULL) return(0);
 318                 BIO_clear_retry_flags(b);
 319                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
 320                 BIO_copy_next_retry(b);
 321                 break;
 322
 323         case BIO_CTRL_FLUSH:
 324                 if (b->next_bio == NULL) return(0);
 325                 if (ctx->obuf_len <= 0)
 326                         {
 327                         ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
 328                         break;
 329                         }
 330
 331                 for (;;)
 332                         {
 333                         BIO_clear_retry_flags(b);
 334                         if (ctx->obuf_len > 0)
 335                                 {
 336                                 r=BIO_write(b->next_bio,
 337                                         ctx->obuf, ctx->obuf_len);
 338 #if 0
 339 fprintf(stderr,"FLUSH %3d -> %3d\n",ctx->obuf_len,r);
 340 #endif
 341                                 BIO_copy_next_retry(b);
 342                                 if (r <= 0) return((long)r);
 343                                 if (r < ctx->obuf_len)
 344                                         memmove(ctx->obuf, ctx->obuf + r,
 345                                                 ctx->obuf_len - r);
 346                                 ctx->obuf_len-=r;
 347                                 }
 348                         else
 349                                 {
 350                                 ctx->obuf_len=0;
 351                                 ret=1;
 352                                 break;
 353                                 }
 354                         }
 355                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
 356                 break;
 357         case BIO_CTRL_DUP:
 358                 dbio=(BIO *)ptr;
 359                 if (    !BIO_set_write_buffer_size(dbio,ctx->obuf_size))
 360                         ret=0;
 361                 break;
 362         default:
 363                 if (b->next_bio == NULL) return(0);
 364                 ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
 365                 break;
 366                 }
 367         return(ret);
 368 malloc_error:
 369         BIOerr(BIO_F_LINEBUFFER_CTRL,ERR_R_MALLOC_FAILURE);
 370         return(0);
 371         }
 372
 373 static long linebuffer_callback_ctrl(BIO *b, int cmd, bio_info_cb *fp)
 374         {
 375         long ret=1;
 376
 377         if (b->next_bio == NULL) return(0);
 378         switch (cmd)
 379                 {
 380         default:
 381                 ret=BIO_callback_ctrl(b->next_bio,cmd,fp);
 382                 break;
 383                 }
 384         return(ret);
 385         }
 386
 387 static int linebuffer_gets(BIO *b, char *buf, int size)
 388         {
 389         if (b->next_bio == NULL) return(0);
 390         return(BIO_gets(b->next_bio,buf,size));
 391         }
 392
 393 static int linebuffer_puts(BIO *b, const char *str)
 394         {
 395         return(linebuffer_write(b,str,strlen(str)));
 396         }
 397