fix integer overflow in unicode case operations (closes #22643)

This commit is contained in:
Benjamin Peterson 2014-10-15 11:47:36 -04:00
parent 77a75b3db1
commit e1bd38c03c
3 changed files with 13 additions and 0 deletions

View file

@ -9484,6 +9484,11 @@ case_operation(PyObject *self,
kind = PyUnicode_KIND(self);
data = PyUnicode_DATA(self);
length = PyUnicode_GET_LENGTH(self);
if (length > PY_SSIZE_T_MAX / 3 ||
length > PY_SIZE_MAX / (3 * sizeof(Py_UCS4))) {
PyErr_SetString(PyExc_OverflowError, "string is too long");
return NULL;
}
tmp = PyMem_MALLOC(sizeof(Py_UCS4) * 3 * length);
if (tmp == NULL)
return PyErr_NoMemory();