bpo-38823: Fix refleak in marshal init error path (GH-17260)

(cherry picked from commit 33b671e) Co-authored-by: Brandt Bucher <brandtbucher@gmail.com>
python · Nov 20, 2019 · 2ea4c37 · 2ea4c37
1 parent 1322439
commit 2ea4c37
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Python/marshal.c b/Python/marshal.c
@@ -1829,6 +1829,9 @@ PyMarshal_Init(void)
     PyObject *mod = PyModule_Create(&marshalmodule);
     if (mod == NULL)
         return NULL;
-    PyModule_AddIntConstant(mod, "version", Py_MARSHAL_VERSION);
+    if (PyModule_AddIntConstant(mod, "version", Py_MARSHAL_VERSION) < 0) {
+        Py_DECREF(mod);
+        return NULL;
+    }
     return mod;
 }