Commit 64adc7fe authored by Benjamin Peterson's avatar Benjamin Peterson

ensure no one tries to hash things before the random seed is found

parent 52c44306
...@@ -479,6 +479,10 @@ typedef struct { ...@@ -479,6 +479,10 @@ typedef struct {
} _Py_HashSecret_t; } _Py_HashSecret_t;
PyAPI_DATA(_Py_HashSecret_t) _Py_HashSecret; PyAPI_DATA(_Py_HashSecret_t) _Py_HashSecret;
#ifdef Py_DEBUG
PyAPI_DATA(int) _Py_HashSecret_Initialized;
#endif
/* Helper for passing objects to printf and the like */ /* Helper for passing objects to printf and the like */
#define PyObject_REPR(obj) _PyUnicode_AsString(PyObject_Repr(obj)) #define PyObject_REPR(obj) _PyUnicode_AsString(PyObject_Repr(obj))
......
...@@ -2565,6 +2565,7 @@ generic_hash(unsigned char *data, int len) ...@@ -2565,6 +2565,7 @@ generic_hash(unsigned char *data, int len)
register unsigned char *p; register unsigned char *p;
register long x; register long x;
assert(_Py_HashSecret_Initialized);
p = (unsigned char *) data; p = (unsigned char *) data;
x = _Py_HashSecret.prefix; x = _Py_HashSecret.prefix;
x ^= *p << 7; x ^= *p << 7;
......
...@@ -896,6 +896,7 @@ bytes_hash(PyBytesObject *a) ...@@ -896,6 +896,7 @@ bytes_hash(PyBytesObject *a)
register unsigned char *p; register unsigned char *p;
register long x; register long x;
assert(_Py_HashSecret_Initialized);
if (a->ob_shash != -1) if (a->ob_shash != -1)
return a->ob_shash; return a->ob_shash;
len = Py_SIZE(a); len = Py_SIZE(a);
......
...@@ -7341,6 +7341,7 @@ unicode_hash(PyUnicodeObject *self) ...@@ -7341,6 +7341,7 @@ unicode_hash(PyUnicodeObject *self)
Py_UNICODE *p; Py_UNICODE *p;
long x; long x;
assert(_Py_HashSecret_Initialized);
if (self->hash != -1) if (self->hash != -1)
return self->hash; return self->hash;
len = Py_SIZE(self); len = Py_SIZE(self);
......
...@@ -5,7 +5,11 @@ ...@@ -5,7 +5,11 @@
#include <fcntl.h> #include <fcntl.h>
#endif #endif
static int random_initialized = 0; #ifdef Py_DEBUG
int _Py_HashSecret_Initialized = 0;
#else
static int _Py_HashSecret_Initialized = 0;
#endif
#ifdef MS_WINDOWS #ifdef MS_WINDOWS
typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\ typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTA)(HCRYPTPROV *phProv,\
...@@ -246,11 +250,11 @@ _PyRandom_Init(void) ...@@ -246,11 +250,11 @@ _PyRandom_Init(void)
{ {
char *env; char *env;
void *secret = &_Py_HashSecret; void *secret = &_Py_HashSecret;
Py_ssize_t secret_size = sizeof(_Py_HashSecret); Py_ssize_t secret_size = sizeof(_Py_HashSecret_t);
if (random_initialized) if (_Py_HashSecret_Initialized)
return; return;
random_initialized = 1; _Py_HashSecret_Initialized = 1;
/* /*
By default, hash randomization is disabled, and only By default, hash randomization is disabled, and only
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment