From 602d307ac5e8a2da38a193dca3bdfef5994dfe67 Mon Sep 17 00:00:00 2001
From: Zackery Spytz <zspytz@gmail.com>
Date: Fri, 7 Dec 2018 05:17:43 -0700
Subject: bpo-35436: Add missing PyErr_NoMemory() calls and other minor bug
 fixes. (GH-11015) (GH-11020)

(cherry picked from commit 4c49da0cb7434c676d70b9ccf38aca82ac0d64a9)
---
 Parser/tokenizer.c | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'Parser/tokenizer.c')

diff --git a/Parser/tokenizer.c b/Parser/tokenizer.c
index fbc98880c9..1a36f1c384 100644
--- a/Parser/tokenizer.c
+++ b/Parser/tokenizer.c
@@ -953,6 +953,11 @@ tok_nextc(struct tok_state *tok)
                 buflen = PyBytes_GET_SIZE(u);
                 buf = PyBytes_AS_STRING(u);
                 newtok = PyMem_MALLOC(buflen+1);
+                if (newtok == NULL) {
+                    Py_DECREF(u);
+                    tok->done = E_NOMEM;
+                    return EOF;
+                }
                 strcpy(newtok, buf);
                 Py_DECREF(u);
             }
-- 
cgit v1.2.1