summaryrefslogtreecommitdiff
path: root/Python/frozenmain.c
diff options
context:
space:
mode:
Diffstat (limited to 'Python/frozenmain.c')
-rw-r--r--Python/frozenmain.c11
1 files changed, 9 insertions, 2 deletions
diff --git a/Python/frozenmain.c b/Python/frozenmain.c
index f08caf23b9..98f46ebdb2 100644
--- a/Python/frozenmain.c
+++ b/Python/frozenmain.c
@@ -20,9 +20,16 @@ Py_FrozenMain(int argc, char **argv)
int inspect = 0;
int unbuffered = 0;
char *oldloc;
- wchar_t **argv_copy = PyMem_Malloc(sizeof(wchar_t*)*argc);
+ wchar_t **argv_copy;
/* We need a second copies, as Python might modify the first one. */
- wchar_t **argv_copy2 = PyMem_Malloc(sizeof(wchar_t*)*argc);
+ wchar_t **argv_copy2;
+
+ argv_copy = PyMem_Malloc(sizeof(wchar_t*)*argc);
+ argv_copy2 = PyMem_Malloc(sizeof(wchar_t*)*argc);
+ if (!argv_copy || !argv_copy2) {
+ fprintf(stderr, "out of memory\n");
+ return 1;
+ }
Py_FrozenFlag = 1; /* Suppress errors from getpath.c */