Commit f76231f8 authored by Mat M's avatar Mat M Committed by Serhiy Storchaka

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

parent 8acaa31e
...@@ -1777,8 +1777,10 @@ fast_save_enter(PicklerObject *self, PyObject *obj) ...@@ -1777,8 +1777,10 @@ fast_save_enter(PicklerObject *self, PyObject *obj)
} }
} }
key = PyLong_FromVoidPtr(obj); key = PyLong_FromVoidPtr(obj);
if (key == NULL) if (key == NULL) {
self->fast_nesting = -1;
return 0; return 0;
}
if (PyDict_GetItemWithError(self->fast_memo, key)) { if (PyDict_GetItemWithError(self->fast_memo, key)) {
Py_DECREF(key); Py_DECREF(key);
PyErr_Format(PyExc_ValueError, PyErr_Format(PyExc_ValueError,
...@@ -1789,6 +1791,8 @@ fast_save_enter(PicklerObject *self, PyObject *obj) ...@@ -1789,6 +1791,8 @@ fast_save_enter(PicklerObject *self, PyObject *obj)
return 0; return 0;
} }
if (PyErr_Occurred()) { if (PyErr_Occurred()) {
Py_DECREF(key);
self->fast_nesting = -1;
return 0; return 0;
} }
if (PyDict_SetItem(self->fast_memo, key, Py_None) < 0) { if (PyDict_SetItem(self->fast_memo, key, Py_None) < 0) {
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment