From: Neal Norwitz <nnorwitz@gmail.com>
Date: Sun, 23 Jul 2006 07:50:36 +0000 (+0000)
Subject: Handle more mem alloc issues found with failmalloc
X-Git-Tag: v2.5b3~198
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=18b6adf9b2f581da04cf5cd6149b84408763ea6e;p=python

Handle more mem alloc issues found with failmalloc
---

diff --git a/Python/future.c b/Python/future.c
index 560077d4d0..3b3ca1d7ba 100644
--- a/Python/future.c
+++ b/Python/future.c
@@ -121,8 +121,10 @@ PyFuture_FromAST(mod_ty mod, const char *filename)
 	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;
 
diff --git a/Python/symtable.c b/Python/symtable.c
index c010b7a9b2..439a24349d 100644
--- a/Python/symtable.c
+++ b/Python/symtable.c
@@ -529,6 +529,8 @@ update_symbols(PyObject *symbols, PyObject *scope,
 		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;