check return value of PyString_FromStringAndSize for NULL (closes #24734)

benjaminp · benjaminp · commit 0287f2f7cbee · 2015-07-27T07:47:21.000-07:00
Patch by Pankaj Sharma.
diff --git a/Python/compile.c b/Python/compile.c
@@ -1979,9 +1979,12 @@ compiler_import(struct compiler *c, stmt_ty s)
             identifier tmp = alias->name;
             const char *base = PyString_AS_STRING(alias->name);
             char *dot = strchr(base, '.');
-            if (dot)
+            if (dot) {
                 tmp = PyString_FromStringAndSize(base,
                                                  dot - base);
+                if (tmp == NULL)
+                    return 0;
+            }
             r = compiler_nameop(c, tmp, Store);
             if (dot) {
                 Py_DECREF(tmp);
