From 19eb87d857f6d1bad1a3619d9d7534820f82e5f2 Mon Sep 17 00:00:00 2001 From: Serhiy Storchaka Date: Tue, 26 Sep 2017 09:11:27 +0300 Subject: [2.7] bpo-31579: Fixed a possible leak in enumerate() with large indices. (GH-3753). (#3761) (cherry picked from commit 0e950dd22b075b4809c84afda8aede02b76ac0fa) --- Objects/enumobject.c | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/Objects/enumobject.c b/Objects/enumobject.c index 1ef381f..8f86a5b 100644 --- a/Objects/enumobject.c +++ b/Objects/enumobject.c @@ -87,19 +87,25 @@ enum_next_long(enumobject *en, PyObject* next_item) if (en->en_longindex == NULL) { en->en_longindex = PyInt_FromSsize_t(PY_SSIZE_T_MAX); - if (en->en_longindex == NULL) + if (en->en_longindex == NULL) { + Py_DECREF(next_item); return NULL; + } } if (one == NULL) { one = PyInt_FromLong(1); - if (one == NULL) + if (one == NULL) { + Py_DECREF(next_item); return NULL; + } } next_index = en->en_longindex; assert(next_index != NULL); stepped_up = PyNumber_Add(next_index, one); - if (stepped_up == NULL) + if (stepped_up == NULL) { + Py_DECREF(next_item); return NULL; + } en->en_longindex = stepped_up; if (result->ob_refcnt == 1) { -- cgit v0.12