893664777c5f9cfc5cb03f71976777a3f46e0d5d
[openssl.git] / crypto / bio / bss_mem.c
1 /* crypto/bio/bss_mem.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 "bio.h"
63
64 #ifndef NOPROTO
65 static int mem_write(BIO *h,char *buf,int num);
66 static int mem_read(BIO *h,char *buf,int size);
67 static int mem_puts(BIO *h,char *str);
68 static int mem_gets(BIO *h,char *str,int size);
69 static long mem_ctrl(BIO *h,int cmd,long arg1,char *arg2);
70 static int mem_new(BIO *h);
71 static int mem_free(BIO *data);
72 #else
73 static int mem_write();
74 static int mem_read();
75 static int mem_puts();
76 static int mem_gets();
77 static long mem_ctrl();
78 static int mem_new();
79 static int mem_free();
80 #endif
81
82 static BIO_METHOD mem_method=
83         {
84         BIO_TYPE_MEM,
85         "memory buffer",
86         mem_write,
87         mem_read,
88         mem_puts,
89         mem_gets,
90         mem_ctrl,
91         mem_new,
92         mem_free,
93         };
94
95 /* bio->num is used to hold the value to return on 'empty', if it is
96  * 0, should_retry is not set */
97
98 BIO_METHOD *BIO_s_mem(void)
99         {
100         return(&mem_method);
101         }
102
103 static int mem_new(BIO *bi)
104         {
105         BUF_MEM *b;
106
107         if ((b=BUF_MEM_new()) == NULL)
108                 return(0);
109         bi->shutdown=1;
110         bi->init=1;
111         bi->num= -1;
112         bi->ptr=(char *)b;
113         return(1);
114         }
115
116 static int mem_free(BIO *a)
117         {
118         if (a == NULL) return(0);
119         if (a->shutdown)
120                 {
121                 if ((a->init) && (a->ptr != NULL))
122                         {
123                         BUF_MEM_free((BUF_MEM *)a->ptr);
124                         a->ptr=NULL;
125                         }
126                 }
127         return(1);
128         }
129         
130 static int mem_read(BIO *b, char *out, int outl)
131         {
132         int ret= -1;
133         BUF_MEM *bm;
134         int i;
135         char *from,*to;
136
137         bm=(BUF_MEM *)b->ptr;
138         BIO_clear_retry_flags(b);
139         ret=(outl > bm->length)?bm->length:outl;
140         if ((out != NULL) && (ret > 0))
141                 {
142                 memcpy(out,bm->data,ret);
143                 bm->length-=ret;
144                 /* memmove(&(bm->data[0]),&(bm->data[ret]), bm->length); */
145                 from=(char *)&(bm->data[ret]);
146                 to=(char *)&(bm->data[0]);
147                 for (i=0; i<bm->length; i++)
148                         to[i]=from[i];
149                 }
150         else if (bm->length == 0)
151                 {
152                 if (b->num != 0)
153                         BIO_set_retry_read(b);
154                 ret= b->num;
155                 }
156         return(ret);
157         }
158
159 static int mem_write(BIO *b, char *in, int inl)
160         {
161         int ret= -1;
162         int blen;
163         BUF_MEM *bm;
164
165         bm=(BUF_MEM *)b->ptr;
166         if (in == NULL)
167                 {
168                 BIOerr(BIO_F_MEM_WRITE,BIO_R_NULL_PARAMETER);
169                 goto end;
170                 }
171
172         BIO_clear_retry_flags(b);
173         blen=bm->length;
174         if (BUF_MEM_grow(bm,blen+inl) != (blen+inl))
175                 goto end;
176         memcpy(&(bm->data[blen]),in,inl);
177         ret=inl;
178 end:
179         return(ret);
180         }
181
182 static long mem_ctrl(BIO *b, int cmd, long num, char *ptr)
183         {
184         long ret=1;
185         char **pptr;
186
187         BUF_MEM *bm=(BUF_MEM *)b->ptr;
188
189         switch (cmd)
190                 {
191         case BIO_CTRL_RESET:
192                 if (bm->data != NULL)
193                         memset(bm->data,0,bm->max);
194                 bm->length=0;
195                 break;
196         case BIO_CTRL_EOF:
197                 ret=(long)(bm->length == 0);
198                 break;
199         case BIO_C_SET_BUF_MEM_EOF_RETURN:
200                 b->num=(int)num;
201                 break;
202         case BIO_CTRL_INFO:
203                 ret=(long)bm->length;
204                 if (ptr != NULL)
205                         {
206                         pptr=(char **)ptr;
207                         *pptr=(char *)&(bm->data[0]);
208                         }
209                 break;
210         case BIO_C_SET_BUF_MEM:
211                 mem_free(b);
212                 b->shutdown=(int)num;
213                 b->ptr=ptr;
214                 break;
215         case BIO_C_GET_BUF_MEM_PTR:
216                 if (ptr != NULL)
217                         {
218                         pptr=(char **)ptr;
219                         *pptr=(char *)bm;
220                         }
221                 break;
222         case BIO_CTRL_GET_CLOSE:
223                 ret=(long)b->shutdown;
224                 break;
225         case BIO_CTRL_SET_CLOSE:
226                 b->shutdown=(int)num;
227                 break;
228
229         case BIO_CTRL_WPENDING:
230                 ret=0L;
231                 break;
232         case BIO_CTRL_PENDING:
233                 ret=(long)bm->length;
234                 break;
235         case BIO_CTRL_DUP:
236         case BIO_CTRL_FLUSH:
237                 ret=1;
238                 break;
239         case BIO_CTRL_PUSH:
240         case BIO_CTRL_POP:
241         default:
242                 ret=0;
243                 break;
244                 }
245         return(ret);
246         }
247
248 static int mem_gets(BIO *bp, char *buf, int size)
249         {
250         int i,j;
251         int ret= -1;
252         char *p;
253         BUF_MEM *bm=(BUF_MEM *)bp->ptr;
254
255         BIO_clear_retry_flags(bp);
256         j=bm->length;
257         if (j <= 0) return(0);
258         p=bm->data;
259         for (i=0; i<j; i++)
260                 {
261                 if (p[i] == '\n') break;
262                 }
263         if (i == j)
264                 {
265                 BIO_set_retry_read(bp);
266                 /* return(-1);  change the semantics 0.6.6a */ 
267                 }
268         else
269                 i++;
270         /* i is the max to copy */
271         if ((size-1) < i) i=size-1;
272         i=mem_read(bp,buf,i);
273         if (i > 0) buf[i]='\0';
274         ret=i;
275         return(ret);
276         }
277
278 static int mem_puts(BIO *bp, char *str)
279         {
280         int n,ret;
281
282         n=strlen(str);
283         ret=mem_write(bp,str,n);
284         /* memory semantics is that it will always work */
285         return(ret);
286         }
287