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

tiran · tiran · commit 74ba26a40c3f · 2013-07-20T14:52:18.000+02:00
CID 486649
diff --git a/Python/import.c b/Python/import.c
@@ -585,7 +585,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) {
