commit dcfb38c21b323d9672aee196fbec5276f30f512d
author Andrew McNamara <andrewm@object-craft.com.au> Mon Jun 09 05:59:23 2003 +0000
committer Andrew McNamara <andrewm@object-craft.com.au> Mon Jun 09 05:59:23 2003 +0000
tree dd1c1852dfa50be39a581fbc7d1e5e070e71f4c9
parent 9067c8d85c4774118675436208cadf59bb048552

Fix potential leaks identified by Neal Norwitz.

Modules/_csv.c

diff --git a/Modules/_csv.c b/Modules/_csv.c
index ab9add2..89b0a0d 100644
--- a/Modules/_csv.c
+++ b/Modules/_csv.c
@@ -465,6 +465,8 @@
 {
 	if (self->field_size == 0) {
 		self->field_size = 4096;
+		if (self->field != NULL)
+			PyMem_Free(self->field);
 		self->field = PyMem_Malloc(self->field_size);
 	}
 	else {
@@ -739,6 +741,8 @@
         Py_XDECREF(self->dialect);
         Py_XDECREF(self->input_iter);
         Py_XDECREF(self->fields);
+        if (self->field != NULL)
+        	PyMem_Free(self->field);
 	PyObject_GC_Del(self);
 }
 
@@ -1002,6 +1006,8 @@
 	if (rec_len > self->rec_size) {
 		if (self->rec_size == 0) {
 			self->rec_size = (rec_len / MEM_INCR + 1) * MEM_INCR;
+			if (self->rec != NULL)
+				PyMem_Free(self->rec);
 			self->rec = PyMem_Malloc(self->rec_size);
 		}
 		else {
@@ -1191,6 +1197,8 @@
 {
         Py_XDECREF(self->dialect);
         Py_XDECREF(self->writeline);
+	if (self->rec != NULL)
+		PyMem_Free(self->rec);
 	PyObject_GC_Del(self);
 }