ZOOKEEPER-792: zkpython memory leak (Lei Zhang via henryr)

git-svn-id: https://svn.apache.org/repos/asf/hadoop/zookeeper/trunk@987982 13f79535-47bb-0310-9956-ffa450edef68

CHANGES.txt

@@ -78,6 +78,8 @@ BUGFIXES:
   ZOOKEEPER-795. eventThread isn't shutdown after a connection 
   "session expired" event coming (Sergey Doroshenko and Ben via mahadev)
 
+  ZOOKEEPER-792. zkpython memory leak (Lei Zhang via henryr)
+
 IMPROVEMENTS:
   ZOOKEEPER-724. Improve junit test integration - log harness information 
   (phunt via mahadev)

src/contrib/zkpython/src/c/zookeeper.c

@@ -1011,6 +1011,7 @@ static PyObject *pyzoo_create(PyObject *self, PyObject *args)
   }
   zhandle_t *zh = zhandles[zkhid];
   int err = zoo_create(zh, path, values, valuelen, &aclv, flags, realbuf, maxbuf_len);
+  free_acls(&aclv);
   if (err != ZOK) {
     PyErr_SetString(err_to_exception(err), zerror(err));
     return NULL;
@@ -1183,7 +1184,7 @@ static PyObject *pyzoo_get(PyObject *self, PyObject *args)
   }
   buffer = malloc(sizeof(char)*buffer_len);
   if (buffer == NULL) {
-    free(pw);
+    free_pywatcher(pw);
     PyErr_SetString(PyExc_MemoryError, "buffer could not be allocated in pyzoo_get");
     return NULL;
   }
@@ -1193,14 +1194,17 @@ static PyObject *pyzoo_get(PyObject *self, PyObject *args)
                      pw, buffer, 
                      &buffer_len, &stat);
  
-  PyObject *stat_dict = build_stat( &stat );
-
   if (err != ZOK) {
     PyErr_SetString(err_to_exception(err), zerror(err));
+    free_pywatcher(pw);
+    free(buffer);
     return NULL;
   }
+
+  PyObject *stat_dict = build_stat( &stat );
   PyObject *ret = Py_BuildValue( "(s#,N)", buffer,buffer_len, stat_dict );
   free(buffer);
+
   return ret;
 }