]> granicus.if.org Git - python/commitdiff
[3.6] bpo-30626: Fix error handling in PyImport_Import(). (GH-2103) (#2221)
authorSerhiy Storchaka <storchaka@gmail.com>
Thu, 15 Jun 2017 18:15:11 +0000 (21:15 +0300)
committerGitHub <noreply@github.com>
Thu, 15 Jun 2017 18:15:11 +0000 (21:15 +0300)
In rare circumstances PyImport_Import() could return NULL without raising
an error.
(cherry picked from commit 145541c)

Python/import.c

index cd865a5423666ed32959c62534bf38167f697314..a23a1026e1116917f4a3f5252f673afa540c67a4 100644 (file)
@@ -1785,9 +1785,13 @@ PyImport_Import(PyObject *module_name)
     Py_DECREF(r);
 
     modules = PyImport_GetModuleDict();
-    r = PyDict_GetItem(modules, module_name);
-    if (r != NULL)
+    r = PyDict_GetItemWithError(modules, module_name);
+    if (r != NULL) {
         Py_INCREF(r);
+    }
+    else if (!PyErr_Occurred()) {
+        PyErr_SetObject(PyExc_KeyError, module_name);
+    }
 
   err:
     Py_XDECREF(globals);