bpo-36430: Fix a possible reference leak in itertools.count(). (GH-12551)

0523c39e · Zackery Spytz · Serhiy Storchaka · 3e700e4c · 0523c39e · 0523c39e
Commit 0523c39e authored Mar 26, 2019 by Zackery Spytz Committed by Serhiy Storchaka Mar 26, 2019
Showing with 2 additions and 0 deletions

Misc/NEWS.d/next/Core and Builtins/2019-03-25-23-37-26.bpo-36430.sd9xxQ.rst ...ore and Builtins/2019-03-25-23-37-26.bpo-36430.sd9xxQ.rst +1 -0

Modules/itertoolsmodule.c Modules/itertoolsmodule.c +1 -0

No files found.
--- a/Misc/NEWS.d/next/Core and Builtins/2019-03-25-23-37-26.bpo-36430.sd9xxQ.rst
+++ b/Misc/NEWS.d/next/Core and Builtins/2019-03-25-23-37-26.bpo-36430.sd9xxQ.rst
+Fix a possible reference leak in :func:`itertools.count`.
--- a/Modules/itertoolsmodule.c
+++ b/Modules/itertoolsmodule.c
@@ -4089,6 +4089,7 @@ itertools_count_impl(PyTypeObject *type, PyObject *long_cnt,
    lz = (countobject *)type->tp_alloc(type, 0);
    if (lz == NULL) {
        Py_XDECREF(long_cnt);
+        Py_DECREF(long_step);
        return NULL;
    }
    lz->cnt = cnt;