commit f76231f89a7231fd486b37f728fbb4aab389e4d7
author Mat M <mathew1800@gmail.com> Mon Nov 13 02:50:16 2017 -0500
committer Serhiy Storchaka <storchaka@gmail.com> Mon Nov 13 09:50:16 2017 +0200
tree b620ee10abb077ca86699553cd227537a35fdd21
parent 8acaa31eec84010c2211b838cffa9f69a91a5240

bpo-32013: _pickle: Add missing Py_DECREF in error case in fast_save_enter() (#4384)

Modules/_pickle.c

diff --git a/Modules/_pickle.c b/Modules/_pickle.c
index b71fb93..4b7f1ed 100644
--- a/Modules/_pickle.c
+++ b/Modules/_pickle.c
@@ -1777,8 +1777,10 @@
             }
         }
         key = PyLong_FromVoidPtr(obj);
-        if (key == NULL)
+        if (key == NULL) {
+            self->fast_nesting = -1;
             return 0;
+        }
         if (PyDict_GetItemWithError(self->fast_memo, key)) {
             Py_DECREF(key);
             PyErr_Format(PyExc_ValueError,
@@ -1789,6 +1791,8 @@
             return 0;
         }
         if (PyErr_Occurred()) {
+            Py_DECREF(key);
+            self->fast_nesting = -1;
             return 0;
         }
         if (PyDict_SetItem(self->fast_memo, key, Py_None) < 0) {