use PyMem_NEW to detect overflow (closes #23362)

This commit is contained in:
Benjamin Peterson 2015-03-02 13:23:25 -05:00
parent b779bfba45
commit e5a853c390
1 changed files with 1 additions and 1 deletions

View File

@ -8501,7 +8501,7 @@ _PyUnicode_TranslateCharmap(PyObject *input,
/* allocate enough for a simple 1:1 translation without
replacements, if we need more, we'll resize */
osize = size;
output = PyMem_Malloc(osize * sizeof(Py_UCS4));
output = PyMem_NEW(Py_UCS4, osize);
opos = 0;
if (output == NULL) {
PyErr_NoMemory();