Properly handle a NULL returned from PyArena_New().
Klocwork #364. Will port to head.
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 @@
{
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;
}