Issue #18408: Fix compiler_import() to handle PyUnicode_Substring() failure properly

diff --git a/Python/compile.c b/Python/compile.c
index 4fc75759260f6e4cdcb112b1a79e9da80d0cee23..d11e3abeaa4ca2fc9f1b743039ea03bdc2a72926 100644 (file)
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -2316,8 +2316,11 @@ compiler_import(struct compiler *c, stmt_ty s)
             identifier tmp = alias->name;
             Py_ssize_t dot = PyUnicode_FindChar(
                 alias->name, '.', 0, PyUnicode_GET_LENGTH(alias->name), 1);
-            if (dot != -1)
+            if (dot != -1) {
                 tmp = PyUnicode_Substring(alias->name, 0, dot);
+                if (tmp == NULL)
+                    return 0;
+            }
             r = compiler_nameop(c, tmp, Store);
             if (dot != -1) {
                 Py_DECREF(tmp);