Check return results for errors

diff --git a/Modules/_elementtree.c b/Modules/_elementtree.c
index ea5aa6cf08a2f8783f876769558325817ee9c4ef..a43fe2ec33b9433e0dc565a1382559312a69c80b 100644 (file)
--- a/Modules/_elementtree.c
+++ b/Modules/_elementtree.c
@@ -905,6 +905,9 @@ element_getiterator(ElementObject* self, PyObject* args)
     }
 
     args = PyTuple_New(2);
+    if (args == NULL)
+       return NULL;
+
     Py_INCREF(self); PyTuple_SET_ITEM(args, 0, (PyObject*) self);
     Py_INCREF(tag);  PyTuple_SET_ITEM(args, 1, (PyObject*) tag);
 
@@ -2596,6 +2599,8 @@ init_elementtree(void)
     /* python glue code */
 
     g = PyDict_New();
+    if (g == NULL)
+       return;
 
     PyDict_SetItemString(g, "__builtins__", PyEval_GetBuiltins());