Mirror
/
cpython
mirror of https://github.com/python/cpython
4
1
Fork 0
Code Issues Releases Wiki Activity

Issue #18408: random_seed() now raises a MemoryError on memory allocation 

failure
This commit is contained in:
Victor Stinner 2013-07-15 20:00:36 +02:00
parent 775b2dd778
commit a4ced86f03
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Modules/_randommodule.c
View File

@ -250,8 +250,10 @@ random_seed(RandomObject *self, PyObject *args)
/* Convert seed to byte sequence. */
key_as_bytes = (unsigned char *)PyMem_Malloc((size_t)4 * keyused);
if (key_as_bytes == NULL)
if (key_as_bytes == NULL) {
PyErr_NoMemory();
goto Done;
}
res = _PyLong_AsByteArray((PyLongObject *)n,
key_as_bytes, keyused * 4,
1, /* little-endian */
@ -264,6 +266,7 @@ random_seed(RandomObject *self, PyObject *args)
/* Fill array of unsigned longs from byte sequence. */
key = (unsigned long *)PyMem_Malloc(sizeof(unsigned long) * keyused);
if (key == NULL) {
PyErr_NoMemory();
PyMem_Free(key_as_bytes);
goto Done;
}