]>
Commit | Line | Data |
---|---|---|
3e575651 SL |
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. | |
d3819813 | 8 | * |
3e575651 SL |
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). | |
d3819813 | 15 | * |
3e575651 SL |
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. | |
d3819813 | 22 | * |
3e575651 SL |
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 :-). | |
d3819813 | 37 | * 4. If you include any Windows specific code (or a derivative thereof) from |
3e575651 SL |
38 | * the apps directory (application code) you must include an acknowledgement: |
39 | * "This product includes software written by Tim Hudson (tjh@cryptsoft.com)" | |
d3819813 | 40 | * |
3e575651 SL |
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. | |
d3819813 | 52 | * |
3e575651 SL |
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 | ||
64 | static int mem_write(BIO *h, const char *buf, int num); | |
65 | static int mem_read(BIO *h, char *buf, int size); | |
66 | static int mem_puts(BIO *h, const char *str); | |
67 | static int mem_gets(BIO *h, char *str, int size); | |
68 | static long mem_ctrl(BIO *h, int cmd, long arg1, void *arg2); | |
69 | static int mem_new(BIO *h); | |
70 | static int mem_free(BIO *data); | |
d3819813 MTL |
71 | static BIO_METHOD mem_method = { |
72 | BIO_TYPE_MEM, | |
73 | "memory buffer", | |
74 | mem_write, | |
75 | mem_read, | |
76 | mem_puts, | |
77 | mem_gets, | |
78 | mem_ctrl, | |
79 | mem_new, | |
80 | mem_free, | |
81 | NULL, | |
82 | }; | |
3e575651 | 83 | |
d3819813 MTL |
84 | /* |
85 | * bio->num is used to hold the value to return on 'empty', if it is 0, | |
86 | * should_retry is not set | |
87 | */ | |
3e575651 SL |
88 | |
89 | BIO_METHOD *BIO_s_mem(void) | |
d3819813 MTL |
90 | { |
91 | return (&mem_method); | |
92 | } | |
3e575651 | 93 | |
62f0afa2 MTL |
94 | |
95 | BIO *BIO_new_mem_buf(const void *buf, int len) | |
3e575651 | 96 | { |
d3819813 MTL |
97 | BIO *ret; |
98 | BUF_MEM *b; | |
99 | size_t sz; | |
100 | ||
101 | if (!buf) { | |
102 | BIOerr(BIO_F_BIO_NEW_MEM_BUF, BIO_R_NULL_PARAMETER); | |
103 | return NULL; | |
104 | } | |
105 | sz = (len < 0) ? strlen(buf) : (size_t)len; | |
106 | if (!(ret = BIO_new(BIO_s_mem()))) | |
107 | return NULL; | |
108 | b = (BUF_MEM *)ret->ptr; | |
62f0afa2 MTL |
109 | /* Cast away const and trust in the MEM_RDONLY flag. */ |
110 | b->data = (void *)buf; | |
d3819813 MTL |
111 | b->length = sz; |
112 | b->max = sz; | |
113 | ret->flags |= BIO_FLAGS_MEM_RDONLY; | |
114 | /* Since this is static data retrying wont help */ | |
115 | ret->num = 0; | |
116 | return ret; | |
3e575651 SL |
117 | } |
118 | ||
119 | static int mem_new(BIO *bi) | |
d3819813 MTL |
120 | { |
121 | BUF_MEM *b; | |
3e575651 | 122 | |
d3819813 MTL |
123 | if ((b = BUF_MEM_new()) == NULL) |
124 | return (0); | |
125 | bi->shutdown = 1; | |
126 | bi->init = 1; | |
127 | bi->num = -1; | |
128 | bi->ptr = (char *)b; | |
129 | return (1); | |
130 | } | |
3e575651 SL |
131 | |
132 | static int mem_free(BIO *a) | |
d3819813 MTL |
133 | { |
134 | if (a == NULL) | |
135 | return (0); | |
136 | if (a->shutdown) { | |
137 | if ((a->init) && (a->ptr != NULL)) { | |
138 | BUF_MEM *b; | |
139 | b = (BUF_MEM *)a->ptr; | |
140 | if (a->flags & BIO_FLAGS_MEM_RDONLY) | |
141 | b->data = NULL; | |
142 | BUF_MEM_free(b); | |
143 | a->ptr = NULL; | |
144 | } | |
145 | } | |
146 | return (1); | |
147 | } | |
148 | ||
3e575651 | 149 | static int mem_read(BIO *b, char *out, int outl) |
d3819813 MTL |
150 | { |
151 | int ret = -1; | |
152 | BUF_MEM *bm; | |
3e575651 | 153 | |
d3819813 MTL |
154 | bm = (BUF_MEM *)b->ptr; |
155 | BIO_clear_retry_flags(b); | |
156 | ret = (outl >= 0 && (size_t)outl > bm->length) ? (int)bm->length : outl; | |
157 | if ((out != NULL) && (ret > 0)) { | |
158 | memcpy(out, bm->data, ret); | |
159 | bm->length -= ret; | |
160 | if (b->flags & BIO_FLAGS_MEM_RDONLY) | |
161 | bm->data += ret; | |
162 | else { | |
163 | memmove(&(bm->data[0]), &(bm->data[ret]), bm->length); | |
164 | } | |
165 | } else if (bm->length == 0) { | |
166 | ret = b->num; | |
167 | if (ret != 0) | |
168 | BIO_set_retry_read(b); | |
169 | } | |
170 | return (ret); | |
171 | } | |
3e575651 SL |
172 | |
173 | static int mem_write(BIO *b, const char *in, int inl) | |
d3819813 MTL |
174 | { |
175 | int ret = -1; | |
176 | int blen; | |
177 | BUF_MEM *bm; | |
3e575651 | 178 | |
d3819813 MTL |
179 | bm = (BUF_MEM *)b->ptr; |
180 | if (in == NULL) { | |
181 | BIOerr(BIO_F_MEM_WRITE, BIO_R_NULL_PARAMETER); | |
182 | goto end; | |
183 | } | |
3e575651 | 184 | |
d3819813 MTL |
185 | if (b->flags & BIO_FLAGS_MEM_RDONLY) { |
186 | BIOerr(BIO_F_MEM_WRITE, BIO_R_WRITE_TO_READ_ONLY_BIO); | |
187 | goto end; | |
188 | } | |
3e575651 | 189 | |
d3819813 MTL |
190 | BIO_clear_retry_flags(b); |
191 | blen = bm->length; | |
192 | if (BUF_MEM_grow_clean(bm, blen + inl) != (blen + inl)) | |
193 | goto end; | |
194 | memcpy(&(bm->data[blen]), in, inl); | |
195 | ret = inl; | |
196 | end: | |
197 | return (ret); | |
198 | } | |
3e575651 SL |
199 | |
200 | static long mem_ctrl(BIO *b, int cmd, long num, void *ptr) | |
d3819813 MTL |
201 | { |
202 | long ret = 1; | |
203 | char **pptr; | |
3e575651 | 204 | |
d3819813 | 205 | BUF_MEM *bm = (BUF_MEM *)b->ptr; |
3e575651 | 206 | |
d3819813 MTL |
207 | switch (cmd) { |
208 | case BIO_CTRL_RESET: | |
209 | if (bm->data != NULL) { | |
210 | /* For read only case reset to the start again */ | |
211 | if (b->flags & BIO_FLAGS_MEM_RDONLY) { | |
212 | bm->data -= bm->max - bm->length; | |
213 | bm->length = bm->max; | |
214 | } else { | |
215 | memset(bm->data, 0, bm->max); | |
216 | bm->length = 0; | |
217 | } | |
218 | } | |
219 | break; | |
220 | case BIO_CTRL_EOF: | |
221 | ret = (long)(bm->length == 0); | |
222 | break; | |
223 | case BIO_C_SET_BUF_MEM_EOF_RETURN: | |
224 | b->num = (int)num; | |
225 | break; | |
226 | case BIO_CTRL_INFO: | |
227 | ret = (long)bm->length; | |
228 | if (ptr != NULL) { | |
229 | pptr = (char **)ptr; | |
230 | *pptr = (char *)&(bm->data[0]); | |
231 | } | |
232 | break; | |
233 | case BIO_C_SET_BUF_MEM: | |
234 | mem_free(b); | |
235 | b->shutdown = (int)num; | |
236 | b->ptr = ptr; | |
237 | break; | |
238 | case BIO_C_GET_BUF_MEM_PTR: | |
239 | if (ptr != NULL) { | |
240 | pptr = (char **)ptr; | |
241 | *pptr = (char *)bm; | |
242 | } | |
243 | break; | |
244 | case BIO_CTRL_GET_CLOSE: | |
245 | ret = (long)b->shutdown; | |
246 | break; | |
247 | case BIO_CTRL_SET_CLOSE: | |
248 | b->shutdown = (int)num; | |
249 | break; | |
3e575651 | 250 | |
d3819813 MTL |
251 | case BIO_CTRL_WPENDING: |
252 | ret = 0L; | |
253 | break; | |
254 | case BIO_CTRL_PENDING: | |
255 | ret = (long)bm->length; | |
256 | break; | |
257 | case BIO_CTRL_DUP: | |
258 | case BIO_CTRL_FLUSH: | |
259 | ret = 1; | |
260 | break; | |
261 | case BIO_CTRL_PUSH: | |
262 | case BIO_CTRL_POP: | |
263 | default: | |
264 | ret = 0; | |
265 | break; | |
266 | } | |
267 | return (ret); | |
268 | } | |
3e575651 SL |
269 | |
270 | static int mem_gets(BIO *bp, char *buf, int size) | |
d3819813 MTL |
271 | { |
272 | int i, j; | |
273 | int ret = -1; | |
274 | char *p; | |
275 | BUF_MEM *bm = (BUF_MEM *)bp->ptr; | |
3e575651 | 276 | |
d3819813 MTL |
277 | BIO_clear_retry_flags(bp); |
278 | j = bm->length; | |
279 | if ((size - 1) < j) | |
280 | j = size - 1; | |
281 | if (j <= 0) { | |
282 | *buf = '\0'; | |
283 | return 0; | |
284 | } | |
285 | p = bm->data; | |
286 | for (i = 0; i < j; i++) { | |
287 | if (p[i] == '\n') { | |
288 | i++; | |
289 | break; | |
290 | } | |
291 | } | |
3e575651 | 292 | |
d3819813 MTL |
293 | /* |
294 | * i is now the max num of bytes to copy, either j or up to | |
295 | * and including the first newline | |
296 | */ | |
3e575651 | 297 | |
d3819813 MTL |
298 | i = mem_read(bp, buf, i); |
299 | if (i > 0) | |
300 | buf[i] = '\0'; | |
301 | ret = i; | |
302 | return (ret); | |
303 | } | |
3e575651 SL |
304 | |
305 | static int mem_puts(BIO *bp, const char *str) | |
d3819813 MTL |
306 | { |
307 | int n, ret; | |
3e575651 | 308 | |
d3819813 MTL |
309 | n = strlen(str); |
310 | ret = mem_write(bp, str, n); | |
311 | /* memory semantics is that it will always work */ | |
312 | return (ret); | |
313 | } |