From 18b6adf9b2f581da04cf5cd6149b84408763ea6e Mon Sep 17 00:00:00 2001 From: Neal Norwitz Date: Sun, 23 Jul 2006 07:50:36 +0000 Subject: Handle more mem alloc issues found with failmalloc --- Python/future.c | 4 +++- Python/symtable.c | 2 ++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/Python/future.c b/Python/future.c index 560077d..3b3ca1d 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 c010b7a..439a243 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; -- cgit v0.12