dc629022c3f24198c08fb9b73e00a94828e2e687
[openssl.git] / crypto / rand / rand_lib.c
1 /*
2  * Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (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 <stdio.h>
11 #include <time.h>
12 #include "internal/cryptlib.h"
13 #include <openssl/opensslconf.h>
14 #include "internal/rand.h"
15 #include <openssl/engine.h>
16 #include "internal/thread_once.h"
17
18 #ifndef OPENSSL_NO_ENGINE
19 /* non-NULL if default_RAND_meth is ENGINE-provided */
20 static ENGINE *funct_ref = NULL;
21 static CRYPTO_RWLOCK *rand_engine_lock = NULL;
22 #endif
23 static const RAND_METHOD *default_RAND_meth = NULL;
24 static CRYPTO_RWLOCK *rand_meth_lock = NULL;
25 static CRYPTO_ONCE rand_lock_init = CRYPTO_ONCE_STATIC_INIT;
26
27 DEFINE_RUN_ONCE_STATIC(do_rand_lock_init)
28 {
29 #ifndef OPENSSL_NO_ENGINE
30     rand_engine_lock = CRYPTO_THREAD_lock_new();
31 #endif
32     rand_meth_lock = CRYPTO_THREAD_lock_new();
33     return rand_engine_lock != NULL && rand_meth_lock != NULL;
34 }
35
36 int RAND_set_rand_method(const RAND_METHOD *meth)
37 {
38     if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
39         return 0;
40
41     CRYPTO_THREAD_write_lock(rand_meth_lock);
42 #ifndef OPENSSL_NO_ENGINE
43     ENGINE_finish(funct_ref);
44     funct_ref = NULL;
45 #endif
46     default_RAND_meth = meth;
47     CRYPTO_THREAD_unlock(rand_meth_lock);
48     return 1;
49 }
50
51 const RAND_METHOD *RAND_get_rand_method(void)
52 {
53     const RAND_METHOD *tmp_meth = NULL;
54
55     if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
56         return NULL;
57
58     CRYPTO_THREAD_write_lock(rand_meth_lock);
59     if (!default_RAND_meth) {
60 #ifndef OPENSSL_NO_ENGINE
61         ENGINE *e = ENGINE_get_default_RAND();
62         if (e) {
63             default_RAND_meth = ENGINE_get_RAND(e);
64             if (default_RAND_meth == NULL) {
65                 ENGINE_finish(e);
66                 e = NULL;
67             }
68         }
69         if (e)
70             funct_ref = e;
71         else
72 #endif
73             default_RAND_meth = RAND_OpenSSL();
74     }
75     tmp_meth = default_RAND_meth;
76     CRYPTO_THREAD_unlock(rand_meth_lock);
77     return tmp_meth;
78 }
79
80 #ifndef OPENSSL_NO_ENGINE
81 int RAND_set_rand_engine(ENGINE *engine)
82 {
83     const RAND_METHOD *tmp_meth = NULL;
84
85     if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
86         return 0;
87
88     if (engine) {
89         if (!ENGINE_init(engine))
90             return 0;
91         tmp_meth = ENGINE_get_RAND(engine);
92         if (tmp_meth == NULL) {
93             ENGINE_finish(engine);
94             return 0;
95         }
96     }
97     CRYPTO_THREAD_write_lock(rand_engine_lock);
98     /* This function releases any prior ENGINE so call it first */
99     RAND_set_rand_method(tmp_meth);
100     funct_ref = engine;
101     CRYPTO_THREAD_unlock(rand_engine_lock);
102     return 1;
103 }
104 #endif
105
106 void rand_cleanup_int(void)
107 {
108     const RAND_METHOD *meth = default_RAND_meth;
109     if (meth && meth->cleanup)
110         meth->cleanup();
111     RAND_set_rand_method(NULL);
112     CRYPTO_THREAD_lock_free(rand_meth_lock);
113     CRYPTO_THREAD_lock_free(rand_engine_lock);
114 }
115
116 void RAND_seed(const void *buf, int num)
117 {
118     const RAND_METHOD *meth = RAND_get_rand_method();
119     if (meth && meth->seed)
120         meth->seed(buf, num);
121 }
122
123 void RAND_add(const void *buf, int num, double entropy)
124 {
125     const RAND_METHOD *meth = RAND_get_rand_method();
126     if (meth && meth->add)
127         meth->add(buf, num, entropy);
128 }
129
130 int RAND_bytes(unsigned char *buf, int num)
131 {
132     const RAND_METHOD *meth = RAND_get_rand_method();
133     if (meth && meth->bytes)
134         return meth->bytes(buf, num);
135     return (-1);
136 }
137
138 #if OPENSSL_API_COMPAT < 0x10100000L
139 int RAND_pseudo_bytes(unsigned char *buf, int num)
140 {
141     const RAND_METHOD *meth = RAND_get_rand_method();
142     if (meth && meth->pseudorand)
143         return meth->pseudorand(buf, num);
144     return (-1);
145 }
146 #endif
147
148 int RAND_status(void)
149 {
150     const RAND_METHOD *meth = RAND_get_rand_method();
151     if (meth && meth->status)
152         return meth->status();
153     return 0;
154 }