[Python-checkins] commit of r41751 - python/trunk/Python/compile.c

[neal.norwitz]

Author: neal.norwitz
Date: Sun Dec 18 05:42:47 2005
New Revision: 41751

Modified:
   python/trunk/Python/compile.c
Log:
Set MemoryError when alloc fails

Modified: python/trunk/Python/compile.c
==============================================================================
--- python/trunk/Python/compile.c	(original)
+++ python/trunk/Python/compile.c	Sun Dec 18 05:42:47 2005
@@ -1075,12 +1075,16 @@
 	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 @@
 
 	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 @@
 		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;
 }