commit 602d307ac5e8a2da38a193dca3bdfef5994dfe67
author Zackery Spytz <zspytz@gmail.com> Fri Dec 07 05:17:43 2018 -0700
committer Serhiy Storchaka <storchaka@gmail.com> Fri Dec 07 14:17:43 2018 +0200
tree 8e876e0142683a7564a391dc6396bc6d33f08639
parent 2db190bb356d00422087e1286637887efb8d97c5

bpo-35436: Add missing PyErr_NoMemory() calls and other minor bug fixes. (GH-11015) (GH-11020)

(cherry picked from commit 4c49da0cb7434c676d70b9ccf38aca82ac0d64a9)

Python/ast.c

diff --git a/Python/ast.c b/Python/ast.c
index 07227c2..1e182c7 100644
--- a/Python/ast.c
+++ b/Python/ast.c
@@ -4092,6 +4092,9 @@
     }
     /* Create a duplicate without underscores. */
     dup = PyMem_Malloc(strlen(s) + 1);
+    if (dup == NULL) {
+        return PyErr_NoMemory();
+    }
     end = dup;
     for (; *s; s++) {
         if (*s != '_') {
@@ -4326,8 +4329,10 @@
     len = expr_end - expr_start;
     /* Allocate 3 extra bytes: open paren, close paren, null byte. */
     str = PyMem_RawMalloc(len + 3);
-    if (str == NULL)
+    if (str == NULL) {
+        PyErr_NoMemory();
         return NULL;
+    }
 
     str[0] = '(';
     memcpy(str+1, expr_start, len);