properly handle malloc failure (closes #24044)

Patch by Christian Heimes.
This commit is contained in:
Benjamin Peterson 2015-04-23 17:04:36 -04:00
parent 893cce921c
commit 0823ffb2fb
2 changed files with 7 additions and 2 deletions

View file

@ -1924,8 +1924,10 @@ listsort(PyListObject *self, PyObject *args, PyObject *kwds)
keys = &ms.temparray[saved_ob_size+1];
else {
keys = PyMem_MALLOC(sizeof(PyObject *) * saved_ob_size);
if (keys == NULL)
return NULL;
if (keys == NULL) {
PyErr_NoMemory();
goto keyfunc_fail;
}
}
for (i = 0; i < saved_ob_size ; i++) {