commit 0287f2f7cbeee90491b24f24da871c981074ac00
author Benjamin Peterson <benjamin@python.org> Mon Jul 27 07:47:21 2015 -0700
committer Benjamin Peterson <benjamin@python.org> Mon Jul 27 07:47:21 2015 -0700
tree 58c401b526dc1a8de358669cce18db19d837e0d2
parent 581234ee0d2612fbbc9e521e1bcb3a6d48050dc8

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

Patch by Pankaj Sharma.

Python/compile.c

diff --git a/Python/compile.c b/Python/compile.c
index e871d38..2900757 100644
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -1979,9 +1979,12 @@
             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);