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

bpo-31579: Fixed a possible leak in enumerate() with large indices. (#3753)

This commit is contained in:
Serhiy Storchaka 2017-09-26 08:14:58 +03:00 committed by GitHub
parent 4a2d00cb45
commit 0e950dd22b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Objects/enumobject.c
View File

@ -108,14 +108,18 @@ enum_next_long(enumobject *en, PyObject* next_item)
if (en->en_longindex == NULL) { if (en->en_longindex == NULL) {
en->en_longindex = PyLong_FromSsize_t(PY_SSIZE_T_MAX); en->en_longindex = PyLong_FromSsize_t(PY_SSIZE_T_MAX);
if (en->en_longindex == NULL) if (en->en_longindex == NULL) {
Py_DECREF(next_item);
return NULL; return NULL;
}
} }
next_index = en->en_longindex; next_index = en->en_longindex;
assert(next_index != NULL); assert(next_index != NULL);
stepped_up = PyNumber_Add(next_index, _PyLong_One); stepped_up = PyNumber_Add(next_index, _PyLong_One);
if (stepped_up == NULL) if (stepped_up == NULL) {
Py_DECREF(next_item);
return NULL; return NULL;
}
en->en_longindex = stepped_up; en->en_longindex = stepped_up;
if (result->ob_refcnt == 1) { if (result->ob_refcnt == 1) {