Add missing check of PyDict_SetItem()'s return value in _PyImport_FindExtensionObject()

CID 486649

Add missing check of PyDict_SetItem()'s return value in _PyImport_FindExtensionObject()
CID 486649
7bf9a4ea · Christian Heimes · f6e7dfa6 · 7bf9a4ea
Commit 7bf9a4ea authored Jul 20, 2013 by Christian Heimes
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

Python/import.c Python/import.c +4 -1

No files found.
--- a/Python/import.c
+++ b/Python/import.c
@@ -553,7 +553,10 @@ _PyImport_FindExtensionObject(PyObject *name, PyObject *filename)
        mod = def->m_base.m_init();
        if (mod == NULL)
            return NULL;
-        PyDict_SetItem(PyImport_GetModuleDict(), name, mod);
+        if (PyDict_SetItem(PyImport_GetModuleDict(), name, mod) == -1) {
+            Py_DECREF(mod);
+            return NULL;
+        }
        Py_DECREF(mod);
    }
    if (_PyState_AddModule(mod, def) < 0) {