detect overflow in combinations (closes #23366)

This commit is contained in:
Benjamin Peterson 2015-02-01 20:59:00 -05:00
parent 68389eaa3f
commit 4b40eeb339
3 changed files with 11 additions and 0 deletions

View file

@ -2326,6 +2326,10 @@ combinations_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
goto error;
}
if (r > PY_SSIZE_T_MAX/sizeof(Py_ssize_t)) {
PyErr_SetString(PyExc_OverflowError, "r is too big");
goto error;
}
indices = PyMem_Malloc(r * sizeof(Py_ssize_t));
if (indices == NULL) {
PyErr_NoMemory();