[Python-checkins] r50777 - python/trunk/Python/future.c python/trunk/Python/symtable.c

[neal.norwitz]

Author: neal.norwitz
Date: Sun Jul 23 09:50:36 2006
New Revision: 50777

Modified:
   python/trunk/Python/future.c
   python/trunk/Python/symtable.c
Log:
Handle more mem alloc issues found with failmalloc

Modified: python/trunk/Python/future.c
==============================================================================
--- python/trunk/Python/future.c	(original)
+++ python/trunk/Python/future.c	Sun Jul 23 09:50:36 2006
@@ -121,8 +121,10 @@
 	PyFutureFeatures *ff;
 
 	ff = (PyFutureFeatures *)PyObject_Malloc(sizeof(PyFutureFeatures));
-	if (ff == NULL)
+	if (ff == NULL) {
+		PyErr_NoMemory();
 		return NULL;
+	}
 	ff->ff_features = 0;
 	ff->ff_lineno = -1;
 

Modified: python/trunk/Python/symtable.c
==============================================================================
--- python/trunk/Python/symtable.c	(original)
+++ python/trunk/Python/symtable.c	Sun Jul 23 09:50:36 2006
@@ -529,6 +529,8 @@
 		i = PyInt_AS_LONG(w);
 		flags |= (i << SCOPE_OFF);
 		u = PyInt_FromLong(flags);
+		if (!u)
+			return 0;
 		if (PyDict_SetItem(symbols, name, u) < 0) {
 			Py_DECREF(u);
 			return 0;