]>
Commit | Line | Data |
---|---|---|
c911e5da | 1 | /* |
38fc02a7 | 2 | * Copyright 2018-2021 The OpenSSL Project Authors. All Rights Reserved. |
c911e5da | 3 | * |
909f1a2e | 4 | * Licensed under the Apache License 2.0 (the "License"). You may not use |
c911e5da BE |
5 | * this file except in compliance with the License. You can obtain a copy |
6 | * in the file LICENSE in the source distribution or at | |
7 | * https://www.openssl.org/source/license.html | |
8 | */ | |
0800318a | 9 | #define OPENSSL_SUPPRESS_DEPRECATED |
c911e5da BE |
10 | #include <stdio.h> |
11 | #include <string.h> | |
12 | #include <openssl/bio.h> | |
13 | ||
14 | #include "testutil.h" | |
15 | ||
16 | #define MAXCOUNT 5 | |
17 | static int my_param_count; | |
18 | static BIO *my_param_b[MAXCOUNT]; | |
19 | static int my_param_oper[MAXCOUNT]; | |
20 | static const char *my_param_argp[MAXCOUNT]; | |
21 | static int my_param_argi[MAXCOUNT]; | |
22 | static long my_param_argl[MAXCOUNT]; | |
23 | static long my_param_ret[MAXCOUNT]; | |
0800318a TM |
24 | static size_t my_param_len[MAXCOUNT]; |
25 | static size_t my_param_processed[MAXCOUNT]; | |
c911e5da | 26 | |
0800318a TM |
27 | static long my_bio_cb_ex(BIO *b, int oper, const char *argp, size_t len, |
28 | int argi, long argl, int ret, size_t *processed) | |
29 | { | |
30 | if (my_param_count >= MAXCOUNT) | |
31 | return -1; | |
32 | my_param_b[my_param_count] = b; | |
33 | my_param_oper[my_param_count] = oper; | |
34 | my_param_argp[my_param_count] = argp; | |
35 | my_param_argi[my_param_count] = argi; | |
36 | my_param_argl[my_param_count] = argl; | |
37 | my_param_ret[my_param_count] = ret; | |
38 | my_param_len[my_param_count] = len; | |
39 | my_param_processed[my_param_count] = processed != NULL ? *processed : 0; | |
40 | ||
41 | my_param_count++; | |
42 | return ret; | |
43 | } | |
44 | ||
45 | static int test_bio_callback_ex(void) | |
46 | { | |
47 | int ok = 0; | |
48 | BIO *bio; | |
49 | int i; | |
50 | char test1[] = "test"; | |
51 | const size_t test1len = sizeof(test1) - 1; | |
52 | char test2[] = "hello"; | |
53 | const size_t test2len = sizeof(test2) - 1; | |
54 | char buf[16]; | |
55 | ||
56 | my_param_count = 0; | |
57 | ||
58 | bio = BIO_new(BIO_s_mem()); | |
59 | if (bio == NULL) | |
60 | goto err; | |
61 | ||
62 | BIO_set_callback_ex(bio, my_bio_cb_ex); | |
63 | i = BIO_write(bio, test1, test1len); | |
64 | if (!TEST_int_eq(i, test1len) | |
65 | || !TEST_int_eq(my_param_count, 2) | |
66 | || !TEST_ptr_eq(my_param_b[0], bio) | |
67 | || !TEST_int_eq(my_param_oper[0], BIO_CB_WRITE) | |
68 | || !TEST_ptr_eq(my_param_argp[0], test1) | |
69 | || !TEST_size_t_eq(my_param_len[0], test1len) | |
70 | || !TEST_long_eq(my_param_argl[0], 0L) | |
71 | || !TEST_int_eq((int)my_param_ret[0], 1) | |
72 | || !TEST_ptr_eq(my_param_b[1], bio) | |
73 | || !TEST_int_eq(my_param_oper[1], BIO_CB_WRITE | BIO_CB_RETURN) | |
74 | || !TEST_ptr_eq(my_param_argp[1], test1) | |
75 | || !TEST_size_t_eq(my_param_len[1], test1len) | |
76 | || !TEST_long_eq(my_param_argl[1], 0L) | |
77 | || !TEST_size_t_eq(my_param_processed[1], test1len) | |
78 | || !TEST_int_eq((int)my_param_ret[1], 1)) | |
79 | goto err; | |
80 | ||
81 | my_param_count = 0; | |
82 | i = BIO_read(bio, buf, sizeof(buf)); | |
83 | if (!TEST_mem_eq(buf, i, test1, test1len) | |
84 | || !TEST_int_eq(my_param_count, 2) | |
85 | || !TEST_ptr_eq(my_param_b[0], bio) | |
86 | || !TEST_int_eq(my_param_oper[0], BIO_CB_READ) | |
87 | || !TEST_ptr_eq(my_param_argp[0], buf) | |
88 | || !TEST_size_t_eq(my_param_len[0], sizeof(buf)) | |
89 | || !TEST_long_eq(my_param_argl[0], 0L) | |
90 | || !TEST_int_eq((int)my_param_ret[0], 1) | |
91 | || !TEST_ptr_eq(my_param_b[1], bio) | |
92 | || !TEST_int_eq(my_param_oper[1], BIO_CB_READ | BIO_CB_RETURN) | |
93 | || !TEST_ptr_eq(my_param_argp[1], buf) | |
94 | || !TEST_size_t_eq(my_param_len[1], sizeof(buf)) | |
95 | || !TEST_long_eq(my_param_argl[1], 0L) | |
96 | || !TEST_size_t_eq(my_param_processed[1], test1len) | |
97 | || !TEST_int_eq((int)my_param_ret[1], 1)) | |
98 | goto err; | |
99 | ||
100 | /* By default a mem bio returns -1 if it has run out of data */ | |
101 | my_param_count = 0; | |
102 | i = BIO_read(bio, buf, sizeof(buf)); | |
103 | if (!TEST_int_eq(i, -1) | |
104 | || !TEST_int_eq(my_param_count, 2) | |
105 | || !TEST_ptr_eq(my_param_b[0], bio) | |
106 | || !TEST_int_eq(my_param_oper[0], BIO_CB_READ) | |
107 | || !TEST_ptr_eq(my_param_argp[0], buf) | |
108 | || !TEST_size_t_eq(my_param_len[0], sizeof(buf)) | |
109 | || !TEST_long_eq(my_param_argl[0], 0L) | |
110 | || !TEST_int_eq((int)my_param_ret[0], 1) | |
111 | || !TEST_ptr_eq(my_param_b[1], bio) | |
112 | || !TEST_int_eq(my_param_oper[1], BIO_CB_READ | BIO_CB_RETURN) | |
113 | || !TEST_ptr_eq(my_param_argp[1], buf) | |
114 | || !TEST_size_t_eq(my_param_len[1], sizeof(buf)) | |
115 | || !TEST_long_eq(my_param_argl[1], 0L) | |
116 | || !TEST_size_t_eq(my_param_processed[1], 0) | |
117 | || !TEST_int_eq((int)my_param_ret[1], -1)) | |
118 | goto err; | |
119 | ||
120 | /* Force the mem bio to return 0 if it has run out of data */ | |
121 | my_param_count = 0; | |
122 | i = BIO_set_mem_eof_return(bio, 0); | |
123 | if (!TEST_int_eq(i, 1) | |
124 | || !TEST_int_eq(my_param_count, 2) | |
125 | || !TEST_ptr_eq(my_param_b[0], bio) | |
126 | || !TEST_int_eq(my_param_oper[0], BIO_CB_CTRL) | |
127 | || !TEST_ptr_eq(my_param_argp[0], NULL) | |
128 | || !TEST_int_eq(my_param_argi[0], BIO_C_SET_BUF_MEM_EOF_RETURN) | |
129 | || !TEST_long_eq(my_param_argl[0], 0L) | |
130 | || !TEST_int_eq((int)my_param_ret[0], 1) | |
131 | || !TEST_ptr_eq(my_param_b[1], bio) | |
132 | || !TEST_int_eq(my_param_oper[1], BIO_CB_CTRL | BIO_CB_RETURN) | |
133 | || !TEST_ptr_eq(my_param_argp[1], NULL) | |
134 | || !TEST_int_eq(my_param_argi[1], BIO_C_SET_BUF_MEM_EOF_RETURN) | |
135 | || !TEST_long_eq(my_param_argl[1], 0L) | |
136 | || !TEST_int_eq((int)my_param_ret[1], 1)) | |
137 | goto err; | |
138 | my_param_count = 0; | |
139 | i = BIO_read(bio, buf, sizeof(buf)); | |
140 | if (!TEST_int_eq(i, 0) | |
141 | || !TEST_int_eq(my_param_count, 2) | |
142 | || !TEST_ptr_eq(my_param_b[0], bio) | |
143 | || !TEST_int_eq(my_param_oper[0], BIO_CB_READ) | |
144 | || !TEST_ptr_eq(my_param_argp[0], buf) | |
145 | || !TEST_size_t_eq(my_param_len[0], sizeof(buf)) | |
146 | || !TEST_long_eq(my_param_argl[0], 0L) | |
147 | || !TEST_int_eq((int)my_param_ret[0], 1) | |
148 | || !TEST_ptr_eq(my_param_b[1], bio) | |
149 | || !TEST_int_eq(my_param_oper[1], BIO_CB_READ | BIO_CB_RETURN) | |
150 | || !TEST_ptr_eq(my_param_argp[1], buf) | |
151 | || !TEST_size_t_eq(my_param_len[1], sizeof(buf)) | |
152 | || !TEST_long_eq(my_param_argl[1], 0L) | |
153 | || !TEST_size_t_eq(my_param_processed[1], 0) | |
154 | || !TEST_int_eq((int)my_param_ret[1], 0)) | |
155 | goto err; | |
156 | ||
157 | my_param_count = 0; | |
158 | i = BIO_puts(bio, test2); | |
159 | if (!TEST_int_eq(i, 5) | |
160 | || !TEST_int_eq(my_param_count, 2) | |
161 | || !TEST_ptr_eq(my_param_b[0], bio) | |
162 | || !TEST_int_eq(my_param_oper[0], BIO_CB_PUTS) | |
163 | || !TEST_ptr_eq(my_param_argp[0], test2) | |
164 | || !TEST_int_eq(my_param_argi[0], 0) | |
165 | || !TEST_long_eq(my_param_argl[0], 0L) | |
166 | || !TEST_int_eq((int)my_param_ret[0], 1) | |
167 | || !TEST_ptr_eq(my_param_b[1], bio) | |
168 | || !TEST_int_eq(my_param_oper[1], BIO_CB_PUTS | BIO_CB_RETURN) | |
169 | || !TEST_ptr_eq(my_param_argp[1], test2) | |
170 | || !TEST_int_eq(my_param_argi[1], 0) | |
171 | || !TEST_long_eq(my_param_argl[1], 0L) | |
172 | || !TEST_size_t_eq(my_param_processed[1], test2len) | |
173 | || !TEST_int_eq((int)my_param_ret[1], 1)) | |
174 | goto err; | |
175 | ||
176 | my_param_count = 0; | |
177 | i = BIO_free(bio); | |
178 | if (!TEST_int_eq(i, 1) | |
179 | || !TEST_int_eq(my_param_count, 1) | |
180 | || !TEST_ptr_eq(my_param_b[0], bio) | |
181 | || !TEST_int_eq(my_param_oper[0], BIO_CB_FREE) | |
182 | || !TEST_ptr_eq(my_param_argp[0], NULL) | |
183 | || !TEST_int_eq(my_param_argi[0], 0) | |
184 | || !TEST_long_eq(my_param_argl[0], 0L) | |
185 | || !TEST_int_eq((int)my_param_ret[0], 1)) | |
186 | goto finish; | |
187 | ||
188 | ok = 1; | |
189 | goto finish; | |
190 | ||
191 | err: | |
192 | BIO_free(bio); | |
193 | ||
194 | finish: | |
195 | /* This helps finding memory leaks with ASAN */ | |
196 | memset(my_param_b, 0, sizeof(my_param_b)); | |
197 | memset(my_param_argp, 0, sizeof(my_param_argp)); | |
198 | return ok; | |
199 | } | |
200 | ||
201 | #ifndef OPENSSL_NO_DEPRECATED_3_0 | |
c911e5da BE |
202 | static long my_bio_callback(BIO *b, int oper, const char *argp, int argi, |
203 | long argl, long ret) | |
204 | { | |
205 | if (my_param_count >= MAXCOUNT) | |
206 | return -1; | |
207 | my_param_b[my_param_count] = b; | |
208 | my_param_oper[my_param_count] = oper; | |
209 | my_param_argp[my_param_count] = argp; | |
210 | my_param_argi[my_param_count] = argi; | |
211 | my_param_argl[my_param_count] = argl; | |
212 | my_param_ret[my_param_count] = ret; | |
213 | my_param_count++; | |
214 | return ret; | |
215 | } | |
216 | ||
217 | static int test_bio_callback(void) | |
218 | { | |
219 | int ok = 0; | |
220 | BIO *bio; | |
221 | int i; | |
52d78cc5 MC |
222 | char test1[] = "test"; |
223 | const int test1len = sizeof(test1) - 1; | |
224 | char test2[] = "hello"; | |
225 | const int test2len = sizeof(test2) - 1; | |
226 | char buf[16]; | |
c911e5da BE |
227 | |
228 | my_param_count = 0; | |
229 | ||
230 | bio = BIO_new(BIO_s_mem()); | |
231 | if (bio == NULL) | |
232 | goto err; | |
233 | ||
234 | BIO_set_callback(bio, my_bio_callback); | |
52d78cc5 MC |
235 | i = BIO_write(bio, test1, test1len); |
236 | if (!TEST_int_eq(i, test1len) | |
c911e5da BE |
237 | || !TEST_int_eq(my_param_count, 2) |
238 | || !TEST_ptr_eq(my_param_b[0], bio) | |
239 | || !TEST_int_eq(my_param_oper[0], BIO_CB_WRITE) | |
240 | || !TEST_ptr_eq(my_param_argp[0], test1) | |
52d78cc5 | 241 | || !TEST_int_eq(my_param_argi[0], test1len) |
c911e5da BE |
242 | || !TEST_long_eq(my_param_argl[0], 0L) |
243 | || !TEST_long_eq(my_param_ret[0], 1L) | |
244 | || !TEST_ptr_eq(my_param_b[1], bio) | |
245 | || !TEST_int_eq(my_param_oper[1], BIO_CB_WRITE | BIO_CB_RETURN) | |
246 | || !TEST_ptr_eq(my_param_argp[1], test1) | |
52d78cc5 | 247 | || !TEST_int_eq(my_param_argi[1], test1len) |
c911e5da | 248 | || !TEST_long_eq(my_param_argl[1], 0L) |
52d78cc5 | 249 | || !TEST_long_eq(my_param_ret[1], (long)test1len)) |
c911e5da BE |
250 | goto err; |
251 | ||
52d78cc5 MC |
252 | my_param_count = 0; |
253 | i = BIO_read(bio, buf, sizeof(buf)); | |
254 | if (!TEST_mem_eq(buf, i, test1, test1len) | |
255 | || !TEST_int_eq(my_param_count, 2) | |
256 | || !TEST_ptr_eq(my_param_b[0], bio) | |
257 | || !TEST_int_eq(my_param_oper[0], BIO_CB_READ) | |
258 | || !TEST_ptr_eq(my_param_argp[0], buf) | |
259 | || !TEST_int_eq(my_param_argi[0], sizeof(buf)) | |
260 | || !TEST_long_eq(my_param_argl[0], 0L) | |
261 | || !TEST_long_eq(my_param_ret[0], 1L) | |
262 | || !TEST_ptr_eq(my_param_b[1], bio) | |
263 | || !TEST_int_eq(my_param_oper[1], BIO_CB_READ | BIO_CB_RETURN) | |
264 | || !TEST_ptr_eq(my_param_argp[1], buf) | |
265 | || !TEST_int_eq(my_param_argi[1], sizeof(buf)) | |
266 | || !TEST_long_eq(my_param_argl[1], 0L) | |
267 | || !TEST_long_eq(my_param_ret[1], (long)test1len)) | |
268 | goto err; | |
269 | ||
270 | /* By default a mem bio returns -1 if it has run out of data */ | |
271 | my_param_count = 0; | |
272 | i = BIO_read(bio, buf, sizeof(buf)); | |
273 | if (!TEST_int_eq(i, -1) | |
274 | || !TEST_int_eq(my_param_count, 2) | |
275 | || !TEST_ptr_eq(my_param_b[0], bio) | |
276 | || !TEST_int_eq(my_param_oper[0], BIO_CB_READ) | |
277 | || !TEST_ptr_eq(my_param_argp[0], buf) | |
278 | || !TEST_int_eq(my_param_argi[0], sizeof(buf)) | |
279 | || !TEST_long_eq(my_param_argl[0], 0L) | |
280 | || !TEST_long_eq(my_param_ret[0], 1L) | |
281 | || !TEST_ptr_eq(my_param_b[1], bio) | |
282 | || !TEST_int_eq(my_param_oper[1], BIO_CB_READ | BIO_CB_RETURN) | |
283 | || !TEST_ptr_eq(my_param_argp[1], buf) | |
284 | || !TEST_int_eq(my_param_argi[1], sizeof(buf)) | |
285 | || !TEST_long_eq(my_param_argl[1], 0L) | |
286 | || !TEST_long_eq(my_param_ret[1], -1L)) | |
287 | goto err; | |
288 | ||
289 | /* Force the mem bio to return 0 if it has run out of data */ | |
290 | BIO_set_mem_eof_return(bio, 0); | |
291 | my_param_count = 0; | |
292 | i = BIO_read(bio, buf, sizeof(buf)); | |
293 | if (!TEST_int_eq(i, 0) | |
294 | || !TEST_int_eq(my_param_count, 2) | |
295 | || !TEST_ptr_eq(my_param_b[0], bio) | |
296 | || !TEST_int_eq(my_param_oper[0], BIO_CB_READ) | |
297 | || !TEST_ptr_eq(my_param_argp[0], buf) | |
298 | || !TEST_int_eq(my_param_argi[0], sizeof(buf)) | |
299 | || !TEST_long_eq(my_param_argl[0], 0L) | |
300 | || !TEST_long_eq(my_param_ret[0], 1L) | |
301 | || !TEST_ptr_eq(my_param_b[1], bio) | |
302 | || !TEST_int_eq(my_param_oper[1], BIO_CB_READ | BIO_CB_RETURN) | |
303 | || !TEST_ptr_eq(my_param_argp[1], buf) | |
304 | || !TEST_int_eq(my_param_argi[1], sizeof(buf)) | |
305 | || !TEST_long_eq(my_param_argl[1], 0L) | |
306 | || !TEST_long_eq(my_param_ret[1], 0L)) | |
307 | goto err; | |
308 | ||
309 | my_param_count = 0; | |
c911e5da BE |
310 | i = BIO_puts(bio, test2); |
311 | if (!TEST_int_eq(i, 5) | |
52d78cc5 MC |
312 | || !TEST_int_eq(my_param_count, 2) |
313 | || !TEST_ptr_eq(my_param_b[0], bio) | |
314 | || !TEST_int_eq(my_param_oper[0], BIO_CB_PUTS) | |
315 | || !TEST_ptr_eq(my_param_argp[0], test2) | |
316 | || !TEST_int_eq(my_param_argi[0], 0) | |
317 | || !TEST_long_eq(my_param_argl[0], 0L) | |
318 | || !TEST_long_eq(my_param_ret[0], 1L) | |
319 | || !TEST_ptr_eq(my_param_b[1], bio) | |
320 | || !TEST_int_eq(my_param_oper[1], BIO_CB_PUTS | BIO_CB_RETURN) | |
321 | || !TEST_ptr_eq(my_param_argp[1], test2) | |
322 | || !TEST_int_eq(my_param_argi[1], 0) | |
323 | || !TEST_long_eq(my_param_argl[1], 0L) | |
324 | || !TEST_long_eq(my_param_ret[1], (long)test2len)) | |
c911e5da BE |
325 | goto err; |
326 | ||
52d78cc5 | 327 | my_param_count = 0; |
c911e5da | 328 | i = BIO_free(bio); |
c911e5da | 329 | if (!TEST_int_eq(i, 1) |
52d78cc5 MC |
330 | || !TEST_int_eq(my_param_count, 1) |
331 | || !TEST_ptr_eq(my_param_b[0], bio) | |
332 | || !TEST_int_eq(my_param_oper[0], BIO_CB_FREE) | |
333 | || !TEST_ptr_eq(my_param_argp[0], NULL) | |
334 | || !TEST_int_eq(my_param_argi[0], 0) | |
335 | || !TEST_long_eq(my_param_argl[0], 0L) | |
336 | || !TEST_long_eq(my_param_ret[0], 1L)) | |
c911e5da BE |
337 | goto finish; |
338 | ||
339 | ok = 1; | |
340 | goto finish; | |
341 | ||
342 | err: | |
343 | BIO_free(bio); | |
344 | ||
345 | finish: | |
346 | /* This helps finding memory leaks with ASAN */ | |
347 | memset(my_param_b, 0, sizeof(my_param_b)); | |
348 | memset(my_param_argp, 0, sizeof(my_param_argp)); | |
349 | return ok; | |
350 | } | |
0800318a | 351 | #endif |
c911e5da BE |
352 | |
353 | int setup_tests(void) | |
354 | { | |
0800318a TM |
355 | ADD_TEST(test_bio_callback_ex); |
356 | #ifndef OPENSSL_NO_DEPRECATED_3_0 | |
c911e5da | 357 | ADD_TEST(test_bio_callback); |
0800318a | 358 | #endif |
c911e5da BE |
359 | return 1; |
360 | } |