diff options
| -rw-r--r-- | Python/import.c | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Python/import.c b/Python/import.c index 5af3651..390f9e3 100644 --- a/Python/import.c +++ b/Python/import.c @@ -796,14 +796,16 @@ parse_source_module(const char *pathname, FILE *fp) { PyCodeObject *co = NULL; mod_ty mod; - PyArena *arena = PyArena_New(); + PyArena *arena = PyArena_New(); + if (arena == NULL) + return NULL; mod = PyParser_ASTFromFile(fp, pathname, Py_file_input, 0, 0, 0, NULL, arena); if (mod) { co = PyAST_Compile(mod, pathname, NULL, arena); } - PyArena_Free(arena); + PyArena_Free(arena); return co; } |