commit 7b6caffd70e83b90c50ab006e9b32ae3406d5aa3
author Fred Drake <fdrake@acm.org> Mon Jul 21 17:05:56 2003 +0000
committer Fred Drake <fdrake@acm.org> Mon Jul 21 17:05:56 2003 +0000
tree c0a1f14177789b7080311754fd86eecaed2cde90
parent 55512dccb47ce385c1ca8372a2fb75fe2aa6606c

Fix memory leak reported & discussed on the Python XML-SIG mailing list.
This patch was provided by Jeremy Kloth, and corresponds to pyexpat.c
1.77 in the PyXML CVS.

Modules/pyexpat.c

diff --git a/Modules/pyexpat.c b/Modules/pyexpat.c
index 5a8423e..5c3cf25 100644
--- a/Modules/pyexpat.c
+++ b/Modules/pyexpat.c
@@ -925,8 +925,10 @@
     if ((bytes = PyInt_FromLong(buf_size)) == NULL)
         goto finally;
 
-    if ((arg = PyTuple_New(1)) == NULL)
+    if ((arg = PyTuple_New(1)) == NULL) {
+        Py_DECREF(bytes);
         goto finally;
+    }
 
     PyTuple_SET_ITEM(arg, 0, bytes);
 
@@ -946,7 +948,6 @@
                      "read() returned too much data: "
                      "%i bytes requested, %i returned",
                      buf_size, len);
-        Py_DECREF(str);
         goto finally;
     }
     memcpy(buf, PyString_AsString(str), len);
@@ -987,8 +988,10 @@
     for (;;) {
         int bytes_read;
         void *buf = XML_GetBuffer(self->itself, BUF_SIZE);
-        if (buf == NULL)
+        if (buf == NULL) {
+            Py_DECREF(readmethod);
             return PyErr_NoMemory();
+        }
 
         if (fp) {
             bytes_read = fread(buf, sizeof(char), BUF_SIZE, fp);
@@ -999,16 +1002,21 @@
         }
         else {
             bytes_read = readinst(buf, BUF_SIZE, readmethod);
-            if (bytes_read < 0)
+            if (bytes_read < 0) {
+                Py_DECREF(readmethod);
                 return NULL;
+            }
         }
         rv = XML_ParseBuffer(self->itself, bytes_read, bytes_read == 0);
-        if (PyErr_Occurred())
+        if (PyErr_Occurred()) {
+            Py_XDECREF(readmethod);
             return NULL;
+        }
 
         if (!rv || bytes_read == 0)
             break;
     }
+    Py_XDECREF(readmethod);
     return get_parse_result(self, rv);
 }