7ffd10bd78176fc2aa288b7a3d4abfb1287ad6eb
[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     int ret = 1;
30 #ifndef OPENSSL_NO_ENGINE
31     rand_engine_lock = CRYPTO_THREAD_lock_new();
32     ret &= rand_engine_lock != NULL;
33 #endif
34     rand_meth_lock = CRYPTO_THREAD_lock_new();
35     ret &= rand_meth_lock != NULL;
36     return ret;
37 }
38
39 int RAND_set_rand_method(const RAND_METHOD *meth)
40 {
41     if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
42         return 0;
43
44     CRYPTO_THREAD_write_lock(rand_meth_lock);
45 #ifndef OPENSSL_NO_ENGINE
46     ENGINE_finish(funct_ref);
47     funct_ref = NULL;
48 #endif
49     default_RAND_meth = meth;
50     CRYPTO_THREAD_unlock(rand_meth_lock);
51     return 1;
52 }
53
54 const RAND_METHOD *RAND_get_rand_method(void)
55 {
56     const RAND_METHOD *tmp_meth = NULL;
57
58     if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
59         return NULL;
60
61     CRYPTO_THREAD_write_lock(rand_meth_lock);
62     if (!default_RAND_meth) {
63 #ifndef OPENSSL_NO_ENGINE
64         ENGINE *e = ENGINE_get_default_RAND();
65         if (e) {
66             default_RAND_meth = ENGINE_get_RAND(e);
67             if (default_RAND_meth == NULL) {
68                 ENGINE_finish(e);
69                 e = NULL;
70             }
71         }
72         if (e)
73             funct_ref = e;
74         else
75 #endif
76             default_RAND_meth = RAND_OpenSSL();
77     }
78     tmp_meth = default_RAND_meth;
79     CRYPTO_THREAD_unlock(rand_meth_lock);
80     return tmp_meth;
81 }
82
83 #ifndef OPENSSL_NO_ENGINE
84 int RAND_set_rand_engine(ENGINE *engine)
85 {
86     const RAND_METHOD *tmp_meth = NULL;
87
88     if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
89         return 0;
90
91     if (engine) {
92         if (!ENGINE_init(engine))
93             return 0;
94         tmp_meth = ENGINE_get_RAND(engine);
95         if (tmp_meth == NULL) {
96             ENGINE_finish(engine);
97             return 0;
98         }
99     }
100     CRYPTO_THREAD_write_lock(rand_engine_lock);
101     /* This function releases any prior ENGINE so call it first */
102     RAND_set_rand_method(tmp_meth);
103     funct_ref = engine;
104     CRYPTO_THREAD_unlock(rand_engine_lock);
105     return 1;
106 }
107 #endif
108
109 void rand_cleanup_int(void)
110 {
111     const RAND_METHOD *meth = default_RAND_meth;
112     if (meth && meth->cleanup)
113         meth->cleanup();
114     RAND_set_rand_method(NULL);
115     CRYPTO_THREAD_lock_free(rand_meth_lock);
116 #ifndef OPENSSL_NO_ENGINE
117     CRYPTO_THREAD_lock_free(rand_engine_lock);
118 #endif
119 }
120
121 void RAND_seed(const void *buf, int num)
122 {
123     const RAND_METHOD *meth = RAND_get_rand_method();
124     if (meth && meth->seed)
125         meth->seed(buf, num);
126 }
127
128 void RAND_add(const void *buf, int num, double entropy)
129 {
130     const RAND_METHOD *meth = RAND_get_rand_method();
131     if (meth && meth->add)
132         meth->add(buf, num, entropy);
133 }
134
135 int RAND_bytes(unsigned char *buf, int num)
136 {
137     const RAND_METHOD *meth = RAND_get_rand_method();
138     if (meth && meth->bytes)
139         return meth->bytes(buf, num);
140     return (-1);
141 }
142
143 #if OPENSSL_API_COMPAT < 0x10100000L
144 int RAND_pseudo_bytes(unsigned char *buf, int num)
145 {
146     const RAND_METHOD *meth = RAND_get_rand_method();
147     if (meth && meth->pseudorand)
148         return meth->pseudorand(buf, num);
149     return (-1);
150 }
151 #endif
152
153 int RAND_status(void)
154 {
155     const RAND_METHOD *meth = RAND_get_rand_method();
156     if (meth && meth->status)
157         return meth->status();
158     return 0;
159 }