sl@0: /* crypto/bio/bf_buff.c */
sl@0: /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
sl@0:  * All rights reserved.
sl@0:  *
sl@0:  * This package is an SSL implementation written
sl@0:  * by Eric Young (eay@cryptsoft.com).
sl@0:  * The implementation was written so as to conform with Netscapes SSL.
sl@0:  * 
sl@0:  * This library is free for commercial and non-commercial use as long as
sl@0:  * the following conditions are aheared to.  The following conditions
sl@0:  * apply to all code found in this distribution, be it the RC4, RSA,
sl@0:  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
sl@0:  * included with this distribution is covered by the same copyright terms
sl@0:  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
sl@0:  * 
sl@0:  * Copyright remains Eric Young's, and as such any Copyright notices in
sl@0:  * the code are not to be removed.
sl@0:  * If this package is used in a product, Eric Young should be given attribution
sl@0:  * as the author of the parts of the library used.
sl@0:  * This can be in the form of a textual message at program startup or
sl@0:  * in documentation (online or textual) provided with the package.
sl@0:  * 
sl@0:  * Redistribution and use in source and binary forms, with or without
sl@0:  * modification, are permitted provided that the following conditions
sl@0:  * are met:
sl@0:  * 1. Redistributions of source code must retain the copyright
sl@0:  *    notice, this list of conditions and the following disclaimer.
sl@0:  * 2. Redistributions in binary form must reproduce the above copyright
sl@0:  *    notice, this list of conditions and the following disclaimer in the
sl@0:  *    documentation and/or other materials provided with the distribution.
sl@0:  * 3. All advertising materials mentioning features or use of this software
sl@0:  *    must display the following acknowledgement:
sl@0:  *    "This product includes cryptographic software written by
sl@0:  *     Eric Young (eay@cryptsoft.com)"
sl@0:  *    The word 'cryptographic' can be left out if the rouines from the library
sl@0:  *    being used are not cryptographic related :-).
sl@0:  * 4. If you include any Windows specific code (or a derivative thereof) from 
sl@0:  *    the apps directory (application code) you must include an acknowledgement:
sl@0:  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
sl@0:  * 
sl@0:  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
sl@0:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
sl@0:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
sl@0:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
sl@0:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
sl@0:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
sl@0:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
sl@0:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
sl@0:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
sl@0:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
sl@0:  * SUCH DAMAGE.
sl@0:  * 
sl@0:  * The licence and distribution terms for any publically available version or
sl@0:  * derivative of this code cannot be changed.  i.e. this code cannot simply be
sl@0:  * copied and put under another distribution licence
sl@0:  * [including the GNU Public Licence.]
sl@0:  */
sl@0: 
sl@0: #include <stdio.h>
sl@0: #include <errno.h>
sl@0: #include "cryptlib.h"
sl@0: #include <openssl/bio.h>
sl@0: #include <openssl/evp.h>
sl@0: 
sl@0: static int linebuffer_write(BIO *h, const char *buf,int num);
sl@0: static int linebuffer_read(BIO *h, char *buf, int size);
sl@0: static int linebuffer_puts(BIO *h, const char *str);
sl@0: static int linebuffer_gets(BIO *h, char *str, int size);
sl@0: static long linebuffer_ctrl(BIO *h, int cmd, long arg1, void *arg2);
sl@0: static int linebuffer_new(BIO *h);
sl@0: static int linebuffer_free(BIO *data);
sl@0: static long linebuffer_callback_ctrl(BIO *h, int cmd, bio_info_cb *fp);
sl@0: 
sl@0: /* A 10k maximum should be enough for most purposes */
sl@0: #define DEFAULT_LINEBUFFER_SIZE	1024*10
sl@0: 
sl@0: /* #define DEBUG */
sl@0: 
sl@0: static BIO_METHOD methods_linebuffer=
sl@0: 	{
sl@0: 	BIO_TYPE_LINEBUFFER,
sl@0: 	"linebuffer",
sl@0: 	linebuffer_write,
sl@0: 	linebuffer_read,
sl@0: 	linebuffer_puts,
sl@0: 	linebuffer_gets,
sl@0: 	linebuffer_ctrl,
sl@0: 	linebuffer_new,
sl@0: 	linebuffer_free,
sl@0: 	linebuffer_callback_ctrl,
sl@0: 	};
sl@0: 
sl@0: BIO_METHOD *BIO_f_linebuffer(void)
sl@0: 	{
sl@0: 	return(&methods_linebuffer);
sl@0: 	}
sl@0: 
sl@0: typedef struct bio_linebuffer_ctx_struct
sl@0: 	{
sl@0: 	char *obuf;		/* the output char array */
sl@0: 	int obuf_size;		/* how big is the output buffer */
sl@0: 	int obuf_len;		/* how many bytes are in it */
sl@0: 	} BIO_LINEBUFFER_CTX;
sl@0: 
sl@0: static int linebuffer_new(BIO *bi)
sl@0: 	{
sl@0: 	BIO_LINEBUFFER_CTX *ctx;
sl@0: 
sl@0: 	ctx=(BIO_LINEBUFFER_CTX *)OPENSSL_malloc(sizeof(BIO_LINEBUFFER_CTX));
sl@0: 	if (ctx == NULL) return(0);
sl@0: 	ctx->obuf=(char *)OPENSSL_malloc(DEFAULT_LINEBUFFER_SIZE);
sl@0: 	if (ctx->obuf == NULL) { OPENSSL_free(ctx); return(0); }
sl@0: 	ctx->obuf_size=DEFAULT_LINEBUFFER_SIZE;
sl@0: 	ctx->obuf_len=0;
sl@0: 
sl@0: 	bi->init=1;
sl@0: 	bi->ptr=(char *)ctx;
sl@0: 	bi->flags=0;
sl@0: 	return(1);
sl@0: 	}
sl@0: 
sl@0: static int linebuffer_free(BIO *a)
sl@0: 	{
sl@0: 	BIO_LINEBUFFER_CTX *b;
sl@0: 
sl@0: 	if (a == NULL) return(0);
sl@0: 	b=(BIO_LINEBUFFER_CTX *)a->ptr;
sl@0: 	if (b->obuf != NULL) OPENSSL_free(b->obuf);
sl@0: 	OPENSSL_free(a->ptr);
sl@0: 	a->ptr=NULL;
sl@0: 	a->init=0;
sl@0: 	a->flags=0;
sl@0: 	return(1);
sl@0: 	}
sl@0: 	
sl@0: static int linebuffer_read(BIO *b, char *out, int outl)
sl@0: 	{
sl@0: 	int ret=0;
sl@0:  
sl@0: 	if (out == NULL) return(0);
sl@0: 	if (b->next_bio == NULL) return(0);
sl@0: 	ret=BIO_read(b->next_bio,out,outl);
sl@0: 	BIO_clear_retry_flags(b);
sl@0: 	BIO_copy_next_retry(b);
sl@0: 	return(ret);
sl@0: 	}
sl@0: 
sl@0: static int linebuffer_write(BIO *b, const char *in, int inl)
sl@0: 	{
sl@0: 	int i,num=0,foundnl;
sl@0: 	BIO_LINEBUFFER_CTX *ctx;
sl@0: 
sl@0: 	if ((in == NULL) || (inl <= 0)) return(0);
sl@0: 	ctx=(BIO_LINEBUFFER_CTX *)b->ptr;
sl@0: 	if ((ctx == NULL) || (b->next_bio == NULL)) return(0);
sl@0: 
sl@0: 	BIO_clear_retry_flags(b);
sl@0: 
sl@0: 	do
sl@0: 		{
sl@0: 		const char *p;
sl@0: 
sl@0: 		for(p = in; p < in + inl && *p != '\n'; p++)
sl@0: 			;
sl@0: 		if (*p == '\n')
sl@0: 			{
sl@0: 			p++;
sl@0: 			foundnl = 1;
sl@0: 			}
sl@0: 		else
sl@0: 			foundnl = 0;
sl@0: 
sl@0: 		/* If a NL was found and we already have text in the save
sl@0: 		   buffer, concatenate them and write */
sl@0: 		while ((foundnl || p - in > ctx->obuf_size - ctx->obuf_len)
sl@0: 			&& ctx->obuf_len > 0)
sl@0: 			{
sl@0: 			int orig_olen = ctx->obuf_len;
sl@0: 			
sl@0: 			i = ctx->obuf_size - ctx->obuf_len;
sl@0: 			if (p - in > 0)
sl@0: 				{
sl@0: 				if (i >= p - in)
sl@0: 					{
sl@0: 					memcpy(&(ctx->obuf[ctx->obuf_len]),
sl@0: 						in,p - in);
sl@0: 					ctx->obuf_len += p - in;
sl@0: 					inl -= p - in;
sl@0: 					num += p - in;
sl@0: 					in = p;
sl@0: 					}
sl@0: 				else
sl@0: 					{
sl@0: 					memcpy(&(ctx->obuf[ctx->obuf_len]),
sl@0: 						in,i);
sl@0: 					ctx->obuf_len += i;
sl@0: 					inl -= i;
sl@0: 					in += i;
sl@0: 					num += i;
sl@0: 					}
sl@0: 				}
sl@0: 
sl@0: #if 0
sl@0: BIO_write(b->next_bio, "<*<", 3);
sl@0: #endif
sl@0: 			i=BIO_write(b->next_bio,
sl@0: 				ctx->obuf, ctx->obuf_len);
sl@0: 			if (i <= 0)
sl@0: 				{
sl@0: 				ctx->obuf_len = orig_olen;
sl@0: 				BIO_copy_next_retry(b);
sl@0: 
sl@0: #if 0
sl@0: BIO_write(b->next_bio, ">*>", 3);
sl@0: #endif
sl@0: 				if (i < 0) return((num > 0)?num:i);
sl@0: 				if (i == 0) return(num);
sl@0: 				}
sl@0: #if 0
sl@0: BIO_write(b->next_bio, ">*>", 3);
sl@0: #endif
sl@0: 			if (i < ctx->obuf_len)
sl@0: 				memmove(ctx->obuf, ctx->obuf + i,
sl@0: 					ctx->obuf_len - i);
sl@0: 			ctx->obuf_len-=i;
sl@0: 			}
sl@0: 
sl@0: 		/* Now that the save buffer is emptied, let's write the input
sl@0: 		   buffer if a NL was found and there is anything to write. */
sl@0: 		if ((foundnl || p - in > ctx->obuf_size) && p - in > 0)
sl@0: 			{
sl@0: #if 0
sl@0: BIO_write(b->next_bio, "<*<", 3);
sl@0: #endif
sl@0: 			i=BIO_write(b->next_bio,in,p - in);
sl@0: 			if (i <= 0)
sl@0: 				{
sl@0: 				BIO_copy_next_retry(b);
sl@0: #if 0
sl@0: BIO_write(b->next_bio, ">*>", 3);
sl@0: #endif
sl@0: 				if (i < 0) return((num > 0)?num:i);
sl@0: 				if (i == 0) return(num);
sl@0: 				}
sl@0: #if 0
sl@0: BIO_write(b->next_bio, ">*>", 3);
sl@0: #endif
sl@0: 			num+=i;
sl@0: 			in+=i;
sl@0: 			inl-=i;
sl@0: 			}
sl@0: 		}
sl@0: 	while(foundnl && inl > 0);
sl@0: 	/* We've written as much as we can.  The rest of the input buffer, if
sl@0: 	   any, is text that doesn't and with a NL and therefore needs to be
sl@0: 	   saved for the next trip. */
sl@0: 	if (inl > 0)
sl@0: 		{
sl@0: 		memcpy(&(ctx->obuf[ctx->obuf_len]), in, inl);
sl@0: 		ctx->obuf_len += inl;
sl@0: 		num += inl;
sl@0: 		}
sl@0: 	return num;
sl@0: 	}
sl@0: 
sl@0: static long linebuffer_ctrl(BIO *b, int cmd, long num, void *ptr)
sl@0: 	{
sl@0: 	BIO *dbio;
sl@0: 	BIO_LINEBUFFER_CTX *ctx;
sl@0: 	long ret=1;
sl@0: 	char *p;
sl@0: 	int r;
sl@0: 	int obs;
sl@0: 
sl@0: 	ctx=(BIO_LINEBUFFER_CTX *)b->ptr;
sl@0: 
sl@0: 	switch (cmd)
sl@0: 		{
sl@0: 	case BIO_CTRL_RESET:
sl@0: 		ctx->obuf_len=0;
sl@0: 		if (b->next_bio == NULL) return(0);
sl@0: 		ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
sl@0: 		break;
sl@0: 	case BIO_CTRL_INFO:
sl@0: 		ret=(long)ctx->obuf_len;
sl@0: 		break;
sl@0: 	case BIO_CTRL_WPENDING:
sl@0: 		ret=(long)ctx->obuf_len;
sl@0: 		if (ret == 0)
sl@0: 			{
sl@0: 			if (b->next_bio == NULL) return(0);
sl@0: 			ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
sl@0: 			}
sl@0: 		break;
sl@0: 	case BIO_C_SET_BUFF_SIZE:
sl@0: 		obs=(int)num;
sl@0: 		p=ctx->obuf;
sl@0: 		if ((obs > DEFAULT_LINEBUFFER_SIZE) && (obs != ctx->obuf_size))
sl@0: 			{
sl@0: 			p=(char *)OPENSSL_malloc((int)num);
sl@0: 			if (p == NULL)
sl@0: 				goto malloc_error;
sl@0: 			}
sl@0: 		if (ctx->obuf != p)
sl@0: 			{
sl@0: 			if (ctx->obuf_len > obs)
sl@0: 				{
sl@0: 				ctx->obuf_len = obs;
sl@0: 				}
sl@0: 			memcpy(p, ctx->obuf, ctx->obuf_len);
sl@0: 			OPENSSL_free(ctx->obuf);
sl@0: 			ctx->obuf=p;
sl@0: 			ctx->obuf_size=obs;
sl@0: 			}
sl@0: 		break;
sl@0: 	case BIO_C_DO_STATE_MACHINE:
sl@0: 		if (b->next_bio == NULL) return(0);
sl@0: 		BIO_clear_retry_flags(b);
sl@0: 		ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
sl@0: 		BIO_copy_next_retry(b);
sl@0: 		break;
sl@0: 
sl@0: 	case BIO_CTRL_FLUSH:
sl@0: 		if (b->next_bio == NULL) return(0);
sl@0: 		if (ctx->obuf_len <= 0)
sl@0: 			{
sl@0: 			ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
sl@0: 			break;
sl@0: 			}
sl@0: 
sl@0: 		for (;;)
sl@0: 			{
sl@0: 			BIO_clear_retry_flags(b);
sl@0: 			if (ctx->obuf_len > 0)
sl@0: 				{
sl@0: 				r=BIO_write(b->next_bio,
sl@0: 					ctx->obuf, ctx->obuf_len);
sl@0: #if 0
sl@0: fprintf(stderr,"FLUSH %3d -> %3d\n",ctx->obuf_len,r);
sl@0: #endif
sl@0: 				BIO_copy_next_retry(b);
sl@0: 				if (r <= 0) return((long)r);
sl@0: 				if (r < ctx->obuf_len)
sl@0: 					memmove(ctx->obuf, ctx->obuf + r,
sl@0: 						ctx->obuf_len - r);
sl@0: 				ctx->obuf_len-=r;
sl@0: 				}
sl@0: 			else
sl@0: 				{
sl@0: 				ctx->obuf_len=0;
sl@0: 				ret=1;
sl@0: 				break;
sl@0: 				}
sl@0: 			}
sl@0: 		ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
sl@0: 		break;
sl@0: 	case BIO_CTRL_DUP:
sl@0: 		dbio=(BIO *)ptr;
sl@0: 		if (	!BIO_set_write_buffer_size(dbio,ctx->obuf_size))
sl@0: 			ret=0;
sl@0: 		break;
sl@0: 	default:
sl@0: 		if (b->next_bio == NULL) return(0);
sl@0: 		ret=BIO_ctrl(b->next_bio,cmd,num,ptr);
sl@0: 		break;
sl@0: 		}
sl@0: 	return(ret);
sl@0: malloc_error:
sl@0: 	BIOerr(BIO_F_LINEBUFFER_CTRL,ERR_R_MALLOC_FAILURE);
sl@0: 	return(0);
sl@0: 	}
sl@0: 
sl@0: static long linebuffer_callback_ctrl(BIO *b, int cmd, bio_info_cb *fp)
sl@0: 	{
sl@0: 	long ret=1;
sl@0: 
sl@0: 	if (b->next_bio == NULL) return(0);
sl@0: 	switch (cmd)
sl@0: 		{
sl@0: 	default:
sl@0: 		ret=BIO_callback_ctrl(b->next_bio,cmd,fp);
sl@0: 		break;
sl@0: 		}
sl@0: 	return(ret);
sl@0: 	}
sl@0: 
sl@0: static int linebuffer_gets(BIO *b, char *buf, int size)
sl@0: 	{
sl@0: 	if (b->next_bio == NULL) return(0);
sl@0: 	return(BIO_gets(b->next_bio,buf,size));
sl@0: 	}
sl@0: 
sl@0: static int linebuffer_puts(BIO *b, const char *str)
sl@0: 	{
sl@0: 	return(linebuffer_write(b,str,strlen(str)));
sl@0: 	}
sl@0: