Add weak platform independent PRNG to test framework.
[oweals/openssl.git] / crypto / ex_data.c
1 /*
2  * Copyright 1995-2018 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the Apache License 2.0 (the "License").  You may not use
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  */
9
10 #include "internal/cryptlib_int.h"
11 #include "internal/thread_once.h"
12
13 int do_ex_data_init(OPENSSL_CTX *ctx)
14 {
15     OSSL_EX_DATA_GLOBAL *global = openssl_ctx_get_ex_data_global(ctx);
16
17     if (global == NULL)
18         return 0;
19
20     global->ex_data_lock = CRYPTO_THREAD_lock_new();
21     return global->ex_data_lock != NULL;
22 }
23
24 /*
25  * Return the EX_CALLBACKS from the |ex_data| array that corresponds to
26  * a given class.  On success, *holds the lock.*
27  */
28 static EX_CALLBACKS *get_and_lock(OPENSSL_CTX *ctx, int class_index)
29 {
30     EX_CALLBACKS *ip;
31     OSSL_EX_DATA_GLOBAL *global = NULL;
32
33     if (class_index < 0 || class_index >= CRYPTO_EX_INDEX__COUNT) {
34         CRYPTOerr(CRYPTO_F_GET_AND_LOCK, ERR_R_PASSED_INVALID_ARGUMENT);
35         return NULL;
36     }
37
38     global = openssl_ctx_get_ex_data_global(ctx);
39     if (global == NULL || global->ex_data_lock == NULL) {
40         /*
41          * This can happen in normal operation when using CRYPTO_mem_leaks().
42          * The CRYPTO_mem_leaks() function calls OPENSSL_cleanup() which cleans
43          * up the locks. Subsequently the BIO that CRYPTO_mem_leaks() uses gets
44          * freed, which also attempts to free the ex_data. However
45          * CRYPTO_mem_leaks() ensures that the ex_data is freed early (i.e.
46          * before OPENSSL_cleanup() is called), so if we get here we can safely
47          * ignore this operation. We just treat it as an error.
48          */
49          return NULL;
50     }
51
52     ip = &global->ex_data[class_index];
53     CRYPTO_THREAD_write_lock(global->ex_data_lock);
54     return ip;
55 }
56
57 static void cleanup_cb(EX_CALLBACK *funcs)
58 {
59     OPENSSL_free(funcs);
60 }
61
62 /*
63  * Release all "ex_data" state to prevent memory leaks. This can't be made
64  * thread-safe without overhauling a lot of stuff, and shouldn't really be
65  * called under potential race-conditions anyway (it's for program shutdown
66  * after all).
67  */
68 void crypto_cleanup_all_ex_data_int(OPENSSL_CTX *ctx)
69 {
70     int i;
71     OSSL_EX_DATA_GLOBAL *global = openssl_ctx_get_ex_data_global(ctx);
72
73     if (global == NULL)
74         return;
75
76     for (i = 0; i < CRYPTO_EX_INDEX__COUNT; ++i) {
77         EX_CALLBACKS *ip = &global->ex_data[i];
78
79         sk_EX_CALLBACK_pop_free(ip->meth, cleanup_cb);
80         ip->meth = NULL;
81     }
82
83     CRYPTO_THREAD_lock_free(global->ex_data_lock);
84     global->ex_data_lock = NULL;
85 }
86
87
88 /*
89  * Unregister a new index by replacing the callbacks with no-ops.
90  * Any in-use instances are leaked.
91  */
92 static void dummy_new(void *parent, void *ptr, CRYPTO_EX_DATA *ad, int idx,
93                      long argl, void *argp)
94 {
95 }
96
97 static void dummy_free(void *parent, void *ptr, CRYPTO_EX_DATA *ad, int idx,
98                        long argl, void *argp)
99 {
100 }
101
102 static int dummy_dup(CRYPTO_EX_DATA *to, const CRYPTO_EX_DATA *from,
103                      void *from_d, int idx,
104                      long argl, void *argp)
105 {
106     return 1;
107 }
108
109 int crypto_free_ex_index_ex(OPENSSL_CTX *ctx, int class_index, int idx)
110 {
111     EX_CALLBACKS *ip = get_and_lock(ctx, class_index);
112     EX_CALLBACK *a;
113     int toret = 0;
114     OSSL_EX_DATA_GLOBAL *global = openssl_ctx_get_ex_data_global(ctx);
115
116     if (global == NULL)
117         return 0;
118
119     ip = get_and_lock(ctx, class_index);
120     if (ip == NULL)
121         return 0;
122     if (idx < 0 || idx >= sk_EX_CALLBACK_num(ip->meth))
123         goto err;
124     a = sk_EX_CALLBACK_value(ip->meth, idx);
125     if (a == NULL)
126         goto err;
127     a->new_func = dummy_new;
128     a->dup_func = dummy_dup;
129     a->free_func = dummy_free;
130     toret = 1;
131 err:
132     CRYPTO_THREAD_unlock(global->ex_data_lock);
133     return toret;
134 }
135
136 int CRYPTO_free_ex_index(int class_index, int idx)
137 {
138     return crypto_free_ex_index_ex(NULL, class_index, idx);
139 }
140
141 /*
142  * Register a new index.
143  */
144 int crypto_get_ex_new_index_ex(OPENSSL_CTX *ctx, int class_index, long argl,
145                                void *argp, CRYPTO_EX_new *new_func,
146                                CRYPTO_EX_dup *dup_func,
147                                CRYPTO_EX_free *free_func)
148 {
149     int toret = -1;
150     EX_CALLBACK *a;
151     EX_CALLBACKS *ip;
152     OSSL_EX_DATA_GLOBAL *global = openssl_ctx_get_ex_data_global(ctx);
153
154     if (global == NULL)
155         return -1;
156
157     ip = get_and_lock(ctx, class_index);
158     if (ip == NULL)
159         return -1;
160
161     if (ip->meth == NULL) {
162         ip->meth = sk_EX_CALLBACK_new_null();
163         /* We push an initial value on the stack because the SSL
164          * "app_data" routines use ex_data index zero.  See RT 3710. */
165         if (ip->meth == NULL
166             || !sk_EX_CALLBACK_push(ip->meth, NULL)) {
167             CRYPTOerr(CRYPTO_F_CRYPTO_GET_EX_NEW_INDEX_EX, ERR_R_MALLOC_FAILURE);
168             goto err;
169         }
170     }
171
172     a = (EX_CALLBACK *)OPENSSL_malloc(sizeof(*a));
173     if (a == NULL) {
174         CRYPTOerr(CRYPTO_F_CRYPTO_GET_EX_NEW_INDEX_EX, ERR_R_MALLOC_FAILURE);
175         goto err;
176     }
177     a->argl = argl;
178     a->argp = argp;
179     a->new_func = new_func;
180     a->dup_func = dup_func;
181     a->free_func = free_func;
182
183     if (!sk_EX_CALLBACK_push(ip->meth, NULL)) {
184         CRYPTOerr(CRYPTO_F_CRYPTO_GET_EX_NEW_INDEX_EX, ERR_R_MALLOC_FAILURE);
185         OPENSSL_free(a);
186         goto err;
187     }
188     toret = sk_EX_CALLBACK_num(ip->meth) - 1;
189     (void)sk_EX_CALLBACK_set(ip->meth, toret, a);
190
191  err:
192     CRYPTO_THREAD_unlock(global->ex_data_lock);
193     return toret;
194 }
195
196 int CRYPTO_get_ex_new_index(int class_index, long argl, void *argp,
197                             CRYPTO_EX_new *new_func, CRYPTO_EX_dup *dup_func,
198                             CRYPTO_EX_free *free_func)
199 {
200     return crypto_get_ex_new_index_ex(NULL, class_index, argl, argp, new_func,
201                                       dup_func, free_func);
202 }
203
204 /*
205  * Initialise a new CRYPTO_EX_DATA for use in a particular class - including
206  * calling new() callbacks for each index in the class used by this variable
207  * Thread-safe by copying a class's array of "EX_CALLBACK" entries
208  * in the lock, then using them outside the lock. Note this only applies
209  * to the global "ex_data" state (ie. class definitions), not 'ad' itself.
210  */
211 int crypto_new_ex_data_ex(OPENSSL_CTX *ctx, int class_index, void *obj,
212                           CRYPTO_EX_DATA *ad)
213 {
214     int mx, i;
215     void *ptr;
216     EX_CALLBACK **storage = NULL;
217     EX_CALLBACK *stack[10];
218     EX_CALLBACKS *ip;
219     OSSL_EX_DATA_GLOBAL *global = openssl_ctx_get_ex_data_global(ctx);
220
221     if (global == NULL)
222         return 0;
223
224     ip = get_and_lock(ctx, class_index);
225     if (ip == NULL)
226         return 0;
227
228     ad->ctx = ctx;
229     ad->sk = NULL;
230
231     mx = sk_EX_CALLBACK_num(ip->meth);
232     if (mx > 0) {
233         if (mx < (int)OSSL_NELEM(stack))
234             storage = stack;
235         else
236             storage = OPENSSL_malloc(sizeof(*storage) * mx);
237         if (storage != NULL)
238             for (i = 0; i < mx; i++)
239                 storage[i] = sk_EX_CALLBACK_value(ip->meth, i);
240     }
241     CRYPTO_THREAD_unlock(global->ex_data_lock);
242
243     if (mx > 0 && storage == NULL) {
244         CRYPTOerr(CRYPTO_F_CRYPTO_NEW_EX_DATA_EX, ERR_R_MALLOC_FAILURE);
245         return 0;
246     }
247     for (i = 0; i < mx; i++) {
248         if (storage[i] != NULL && storage[i]->new_func != NULL) {
249             ptr = CRYPTO_get_ex_data(ad, i);
250             storage[i]->new_func(obj, ptr, ad, i,
251                                  storage[i]->argl, storage[i]->argp);
252         }
253     }
254     if (storage != stack)
255         OPENSSL_free(storage);
256     return 1;
257 }
258
259 int CRYPTO_new_ex_data(int class_index, void *obj, CRYPTO_EX_DATA *ad)
260 {
261     return crypto_new_ex_data_ex(NULL, class_index, obj, ad);
262 }
263
264 /*
265  * Duplicate a CRYPTO_EX_DATA variable - including calling dup() callbacks
266  * for each index in the class used by this variable
267  */
268 int CRYPTO_dup_ex_data(int class_index, CRYPTO_EX_DATA *to,
269                        const CRYPTO_EX_DATA *from)
270 {
271     int mx, j, i;
272     void *ptr;
273     EX_CALLBACK *stack[10];
274     EX_CALLBACK **storage = NULL;
275     EX_CALLBACKS *ip;
276     int toret = 0;
277     OSSL_EX_DATA_GLOBAL *global = openssl_ctx_get_ex_data_global(from->ctx);
278
279     if (global == NULL)
280         return 0;
281
282     to->ctx = from->ctx;
283     if (from->sk == NULL)
284         /* Nothing to copy over */
285         return 1;
286     if ((ip = get_and_lock(from->ctx, class_index)) == NULL)
287         return 0;
288
289     mx = sk_EX_CALLBACK_num(ip->meth);
290     j = sk_void_num(from->sk);
291     if (j < mx)
292         mx = j;
293     if (mx > 0) {
294         if (mx < (int)OSSL_NELEM(stack))
295             storage = stack;
296         else
297             storage = OPENSSL_malloc(sizeof(*storage) * mx);
298         if (storage != NULL)
299             for (i = 0; i < mx; i++)
300                 storage[i] = sk_EX_CALLBACK_value(ip->meth, i);
301     }
302     CRYPTO_THREAD_unlock(global->ex_data_lock);
303
304     if (mx == 0)
305         return 1;
306     if (storage == NULL) {
307         CRYPTOerr(CRYPTO_F_CRYPTO_DUP_EX_DATA, ERR_R_MALLOC_FAILURE);
308         return 0;
309     }
310     /*
311      * Make sure the ex_data stack is at least |mx| elements long to avoid
312      * issues in the for loop that follows; so go get the |mx|'th element
313      * (if it does not exist CRYPTO_get_ex_data() returns NULL), and assign
314      * to itself. This is normally a no-op; but ensures the stack is the
315      * proper size
316      */
317     if (!CRYPTO_set_ex_data(to, mx - 1, CRYPTO_get_ex_data(to, mx - 1)))
318         goto err;
319
320     for (i = 0; i < mx; i++) {
321         ptr = CRYPTO_get_ex_data(from, i);
322         if (storage[i] != NULL && storage[i]->dup_func != NULL)
323             if (!storage[i]->dup_func(to, from, &ptr, i,
324                                       storage[i]->argl, storage[i]->argp))
325                 goto err;
326         CRYPTO_set_ex_data(to, i, ptr);
327     }
328     toret = 1;
329  err:
330     if (storage != stack)
331         OPENSSL_free(storage);
332     return toret;
333 }
334
335
336 /*
337  * Cleanup a CRYPTO_EX_DATA variable - including calling free() callbacks for
338  * each index in the class used by this variable
339  */
340 void CRYPTO_free_ex_data(int class_index, void *obj, CRYPTO_EX_DATA *ad)
341 {
342     int mx, i;
343     EX_CALLBACKS *ip;
344     void *ptr;
345     EX_CALLBACK *f;
346     EX_CALLBACK *stack[10];
347     EX_CALLBACK **storage = NULL;
348     OSSL_EX_DATA_GLOBAL *global;
349
350     if ((ip = get_and_lock(ad->ctx, class_index)) == NULL)
351         goto err;
352     global = openssl_ctx_get_ex_data_global(ad->ctx);
353     if (global == NULL)
354         goto err;
355
356     mx = sk_EX_CALLBACK_num(ip->meth);
357     if (mx > 0) {
358         if (mx < (int)OSSL_NELEM(stack))
359             storage = stack;
360         else
361             storage = OPENSSL_malloc(sizeof(*storage) * mx);
362         if (storage != NULL)
363             for (i = 0; i < mx; i++)
364                 storage[i] = sk_EX_CALLBACK_value(ip->meth, i);
365     }
366     CRYPTO_THREAD_unlock(global->ex_data_lock);
367
368     for (i = 0; i < mx; i++) {
369         if (storage != NULL)
370             f = storage[i];
371         else {
372             CRYPTO_THREAD_write_lock(global->ex_data_lock);
373             f = sk_EX_CALLBACK_value(ip->meth, i);
374             CRYPTO_THREAD_unlock(global->ex_data_lock);
375         }
376         if (f != NULL && f->free_func != NULL) {
377             ptr = CRYPTO_get_ex_data(ad, i);
378             f->free_func(obj, ptr, ad, i, f->argl, f->argp);
379         }
380     }
381
382     if (storage != stack)
383         OPENSSL_free(storage);
384  err:
385     sk_void_free(ad->sk);
386     ad->sk = NULL;
387     ad->ctx = NULL;
388 }
389
390 /*
391  * Allocate a given CRYPTO_EX_DATA item using the class specific allocation
392  * function
393  */
394 int CRYPTO_alloc_ex_data(int class_index, void *obj, CRYPTO_EX_DATA *ad,
395                          int idx)
396 {
397     EX_CALLBACK *f;
398     EX_CALLBACKS *ip;
399     void *curval;
400     OSSL_EX_DATA_GLOBAL *global = openssl_ctx_get_ex_data_global(ad->ctx);
401
402     if (global == NULL)
403         return 0;
404
405     curval = CRYPTO_get_ex_data(ad, idx);
406
407     /* Already there, no need to allocate */
408     if (curval != NULL)
409         return 1;
410
411     ip = get_and_lock(ad->ctx, class_index);
412     if (ip == NULL)
413         return 0;
414     f = sk_EX_CALLBACK_value(ip->meth, idx);
415     CRYPTO_THREAD_unlock(global->ex_data_lock);
416
417     /*
418      * This should end up calling CRYPTO_set_ex_data(), which allocates
419      * everything necessary to support placing the new data in the right spot.
420      */
421     if (f->new_func == NULL)
422         return 0;
423
424     f->new_func(obj, curval, ad, idx, f->argl, f->argp);
425
426     return 1;
427 }
428
429 /*
430  * For a given CRYPTO_EX_DATA variable, set the value corresponding to a
431  * particular index in the class used by this variable
432  */
433 int CRYPTO_set_ex_data(CRYPTO_EX_DATA *ad, int idx, void *val)
434 {
435     int i;
436
437     if (ad->sk == NULL) {
438         if ((ad->sk = sk_void_new_null()) == NULL) {
439             CRYPTOerr(CRYPTO_F_CRYPTO_SET_EX_DATA, ERR_R_MALLOC_FAILURE);
440             return 0;
441         }
442     }
443
444     for (i = sk_void_num(ad->sk); i <= idx; ++i) {
445         if (!sk_void_push(ad->sk, NULL)) {
446             CRYPTOerr(CRYPTO_F_CRYPTO_SET_EX_DATA, ERR_R_MALLOC_FAILURE);
447             return 0;
448         }
449     }
450     sk_void_set(ad->sk, idx, val);
451     return 1;
452 }
453
454 /*
455  * For a given CRYPTO_EX_DATA_ variable, get the value corresponding to a
456  * particular index in the class used by this variable
457  */
458 void *CRYPTO_get_ex_data(const CRYPTO_EX_DATA *ad, int idx)
459 {
460     if (ad->sk == NULL || idx >= sk_void_num(ad->sk))
461         return NULL;
462     return sk_void_value(ad->sk, idx);
463 }
464
465 OPENSSL_CTX *crypto_ex_data_get_openssl_ctx(const CRYPTO_EX_DATA *ad)
466 {
467     return ad->ctx;
468 }