Handle memory allocation failure. Found by Adam Olsen

5becac57 · Neal Norwitz · 400aedac · 5becac57
Commit 5becac57 authored Mar 15, 2008 by Neal Norwitz
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

Python/symtable.c Python/symtable.c +3 -2

No files found.
--- a/Python/symtable.c
+++ b/Python/symtable.c
@@ -27,8 +27,9 @@ PySTEntry_New(struct symtable *st, identifier name, _Py_block_ty block,
 	k = PyLong_FromVoidPtr(key);
 	if (k == NULL)
 		goto fail;
-	ste = (PySTEntryObject *)PyObject_New(PySTEntryObject,
-					      &PySTEntry_Type);
+	ste = PyObject_New(PySTEntryObject, &PySTEntry_Type);
+	if (ste == NULL)
+		goto fail;
 	ste->ste_table = st;
 	ste->ste_id = k;
 	ste->ste_tmpname = 0;