diff --git a/Lib/test/test_import/__init__.py b/Lib/test/test_import/__init__.py index ddef27d9cd4..5a610ba3126 100644 --- a/Lib/test/test_import/__init__.py +++ b/Lib/test/test_import/__init__.py @@ -400,6 +400,18 @@ class ImportTests(unittest.TestCase): self.assertEqual(str(cm.exception), "cannot import name 'does_not_exist' from '' (unknown location)") + @cpython_only + def test_issue31492(self): + # There shouldn't be an assertion failure in case of failing to import + # from a module with a bad __name__ attribute, or in case of failing + # to access an attribute of such a module. + with swap_attr(os, '__name__', None): + with self.assertRaises(ImportError): + from os import does_not_exist + + with self.assertRaises(AttributeError): + os.does_not_exist + def test_concurrency(self): sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'data')) try: diff --git a/Misc/NEWS.d/next/Core and Builtins/2017-09-16-22-49-16.bpo-31492.RtyteL.rst b/Misc/NEWS.d/next/Core and Builtins/2017-09-16-22-49-16.bpo-31492.RtyteL.rst new file mode 100644 index 00000000000..a8704738d1a --- /dev/null +++ b/Misc/NEWS.d/next/Core and Builtins/2017-09-16-22-49-16.bpo-31492.RtyteL.rst @@ -0,0 +1,3 @@ +Fix assertion failures in case of failing to import from a module with a bad +``__name__`` attribute, and in case of failing to access an attribute of such +a module. Patch by Oren Milman. diff --git a/Objects/moduleobject.c b/Objects/moduleobject.c index 2be49fbda38..29732633da1 100644 --- a/Objects/moduleobject.c +++ b/Objects/moduleobject.c @@ -687,14 +687,11 @@ module_getattro(PyModuleObject *m, PyObject *name) if (m->md_dict) { _Py_IDENTIFIER(__name__); mod_name = _PyDict_GetItemId(m->md_dict, &PyId___name__); - if (mod_name) { + if (mod_name && PyUnicode_Check(mod_name)) { PyErr_Format(PyExc_AttributeError, "module '%U' has no attribute '%U'", mod_name, name); return NULL; } - else if (PyErr_Occurred()) { - PyErr_Clear(); - } } PyErr_Format(PyExc_AttributeError, "module has no attribute '%U'", name); diff --git a/Python/ceval.c b/Python/ceval.c index 8cc5094a3f4..cf0c6c9ae2f 100644 --- a/Python/ceval.c +++ b/Python/ceval.c @@ -4930,6 +4930,10 @@ import_from(PyObject *v, PyObject *name) if (pkgname == NULL) { goto error; } + if (!PyUnicode_Check(pkgname)) { + Py_CLEAR(pkgname); + goto error; + } fullmodname = PyUnicode_FromFormat("%U.%U", pkgname, name); if (fullmodname == NULL) { Py_DECREF(pkgname);