commit e0fe25be1ecbdf4abd1b0edd4aabacc4d75dec41
author Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> Mon Mar 25 23:26:42 2019 -0700
committer GitHub <noreply@github.com> Mon Mar 25 23:26:42 2019 -0700
tree 083880c823f50dad1e443e4a34afcf2b7d5f16ee
parent eb94e5b3ec01a34af1087c4c85568b5d20325892

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

(cherry picked from commit 0523c39e7720b82b38ad793d3f1a5681adcdf873)

Co-authored-by: Zackery Spytz <zspytz@gmail.com>

Modules/itertoolsmodule.c

diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c
index 1113fb6..bff9df6 100644
--- a/Modules/itertoolsmodule.c
+++ b/Modules/itertoolsmodule.c
@@ -4027,6 +4027,7 @@
     lz = (countobject *)type->tp_alloc(type, 0);
     if (lz == NULL) {
         Py_XDECREF(long_cnt);
+        Py_DECREF(long_step);
         return NULL;
     }
     lz->cnt = cnt;