Set MemoryError when alloc fails

diff --git a/Python/compile.c b/Python/compile.c
index accd3b70bc0ce676f3c3e57329d29c788708793c..84f52e659ab6970af695df903009b3c02600cc95 100644 (file)
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -1075,12 +1075,16 @@ compiler_enter_scope(struct compiler *c, identifier name, void *key,
        struct compiler_unit *u;
 
        u = PyObject_Malloc(sizeof(struct compiler_unit));
+       if (!u) {
+                PyErr_NoMemory();
+                return 0;
+       }
         memset(u, 0, sizeof(struct compiler_unit));
        u->u_argcount = 0;
        u->u_ste = PySymtable_Lookup(c->c_st, key);
        if (!u->u_ste) {
                 compiler_unit_free(u);
-               return 0;
+                return 0;
        }
        Py_INCREF(name);
        u->u_name = name;
@@ -1163,8 +1167,10 @@ compiler_new_block(struct compiler *c)
 
        u = c->u;
        b = (basicblock *)PyObject_Malloc(sizeof(basicblock));
-       if (b == NULL)
+       if (b == NULL) {
+               PyErr_NoMemory();
                return NULL;
+       }
        memset((void *)b, 0, sizeof(basicblock));
        assert (b->b_next == NULL);
        b->b_list = u->u_blocks;
@@ -3747,8 +3753,10 @@ assemble_init(struct assembler *a, int nblocks, int firstlineno)
                return 0;
        a->a_postorder = (basicblock **)PyObject_Malloc(
                                             sizeof(basicblock *) * nblocks);
-       if (!a->a_postorder)
+       if (!a->a_postorder) {
+               PyErr_NoMemory();
                return 0;
+       }
        return 1;
 }