提交 87975cfa 编写于 作者: R Richard Levitte

Make getting and setting the RAND default method thread safe

Reviewed-by: NMatt Caswell <matt@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/3137)
上级 005f6766
...@@ -12,27 +12,50 @@ ...@@ -12,27 +12,50 @@
#include "internal/cryptlib.h" #include "internal/cryptlib.h"
#include <openssl/opensslconf.h> #include <openssl/opensslconf.h>
#include "internal/rand.h" #include "internal/rand.h"
#include <openssl/engine.h> #include <openssl/engine.h>
#include "internal/thread_once.h"
#ifndef OPENSSL_NO_ENGINE #ifndef OPENSSL_NO_ENGINE
/* non-NULL if default_RAND_meth is ENGINE-provided */ /* non-NULL if default_RAND_meth is ENGINE-provided */
static ENGINE *funct_ref = NULL; static ENGINE *funct_ref = NULL;
static CRYPTO_RWLOCK *rand_engine_lock = NULL;
#endif #endif
static const RAND_METHOD *default_RAND_meth = NULL; static const RAND_METHOD *default_RAND_meth = NULL;
static CRYPTO_RWLOCK *rand_meth_lock = NULL;
static CRYPTO_ONCE rand_lock_init = CRYPTO_ONCE_STATIC_INIT;
DEFINE_RUN_ONCE_STATIC(do_rand_lock_init)
{
#ifndef OPENSSL_NO_ENGINE
rand_engine_lock = CRYPTO_THREAD_lock_new();
#endif
rand_meth_lock = CRYPTO_THREAD_lock_new();
return rand_engine_lock != NULL && rand_meth_lock != NULL;
}
int RAND_set_rand_method(const RAND_METHOD *meth) int RAND_set_rand_method(const RAND_METHOD *meth)
{ {
if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
return 0;
CRYPTO_THREAD_write_lock(rand_meth_lock);
#ifndef OPENSSL_NO_ENGINE #ifndef OPENSSL_NO_ENGINE
ENGINE_finish(funct_ref); ENGINE_finish(funct_ref);
funct_ref = NULL; funct_ref = NULL;
#endif #endif
default_RAND_meth = meth; default_RAND_meth = meth;
CRYPTO_THREAD_unlock(rand_meth_lock);
return 1; return 1;
} }
const RAND_METHOD *RAND_get_rand_method(void) const RAND_METHOD *RAND_get_rand_method(void)
{ {
const RAND_METHOD *tmp_meth = NULL;
if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
return NULL;
CRYPTO_THREAD_write_lock(rand_meth_lock);
if (!default_RAND_meth) { if (!default_RAND_meth) {
#ifndef OPENSSL_NO_ENGINE #ifndef OPENSSL_NO_ENGINE
ENGINE *e = ENGINE_get_default_RAND(); ENGINE *e = ENGINE_get_default_RAND();
...@@ -49,13 +72,19 @@ const RAND_METHOD *RAND_get_rand_method(void) ...@@ -49,13 +72,19 @@ const RAND_METHOD *RAND_get_rand_method(void)
#endif #endif
default_RAND_meth = RAND_OpenSSL(); default_RAND_meth = RAND_OpenSSL();
} }
return default_RAND_meth; tmp_meth = default_RAND_meth;
CRYPTO_THREAD_unlock(rand_meth_lock);
return tmp_meth;
} }
#ifndef OPENSSL_NO_ENGINE #ifndef OPENSSL_NO_ENGINE
int RAND_set_rand_engine(ENGINE *engine) int RAND_set_rand_engine(ENGINE *engine)
{ {
const RAND_METHOD *tmp_meth = NULL; const RAND_METHOD *tmp_meth = NULL;
if (!RUN_ONCE(&rand_lock_init, do_rand_lock_init))
return 0;
if (engine) { if (engine) {
if (!ENGINE_init(engine)) if (!ENGINE_init(engine))
return 0; return 0;
...@@ -65,9 +94,11 @@ int RAND_set_rand_engine(ENGINE *engine) ...@@ -65,9 +94,11 @@ int RAND_set_rand_engine(ENGINE *engine)
return 0; return 0;
} }
} }
CRYPTO_THREAD_write_lock(rand_engine_lock);
/* This function releases any prior ENGINE so call it first */ /* This function releases any prior ENGINE so call it first */
RAND_set_rand_method(tmp_meth); RAND_set_rand_method(tmp_meth);
funct_ref = engine; funct_ref = engine;
CRYPTO_THREAD_unlock(rand_engine_lock);
return 1; return 1;
} }
#endif #endif
...@@ -78,6 +109,8 @@ void rand_cleanup_int(void) ...@@ -78,6 +109,8 @@ void rand_cleanup_int(void)
if (meth && meth->cleanup) if (meth && meth->cleanup)
meth->cleanup(); meth->cleanup();
RAND_set_rand_method(NULL); RAND_set_rand_method(NULL);
CRYPTO_THREAD_lock_free(rand_meth_lock);
CRYPTO_THREAD_lock_free(rand_engine_lock);
} }
void RAND_seed(const void *buf, int num) void RAND_seed(const void *buf, int num)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册