Fix memory leaks (thanks @amlweems)

Signed-off-by: Juanjo Alvarez <juanjo@sourced.tech>

Author: Juanjo Alvarez

bblfsh/pyuast.c

@@ -16,8 +16,13 @@ static PyObject *AttributeValue(const void *node, const char *prop) {
 }
 
 static const char *String(const void *node, const char *prop) {
+  const char *retval = NULL;
   PyObject *o = Attribute(node, prop);
-  return o ? PyUnicode_AsUTF8(o) : NULL;
+  if (o != NULL) {
+    retval = PyUnicode_AsUTF8(o);
+    Py_DECREF(o);
+  }
+  return retval;
 }
 
 static size_t Size(const void *node, const char *prop) {
@@ -26,8 +31,14 @@ static size_t Size(const void *node, const char *prop) {
 }
 
 static PyObject *ItemAt(PyObject *object, int index) {
+  PyObject *retval = NULL;
   PyObject *seq = PySequence_Fast(object, "expected a sequence");
-  return PyList_GET_ITEM(seq, index);
+  if (seq != NULL) {
+    retval = PyList_GET_ITEM(seq, index);
+    Py_DECREF(seq);
+  }
+
+  return retval;
 }
 
 
@@ -68,8 +79,13 @@ static const char *PropertyKeyAt(const void *node, int index) {
     return NULL;
   }
 
+  const char *retval = NULL;
   PyObject *keys = PyMapping_Keys(properties);
-  return keys ? PyUnicode_AsUTF8(ItemAt(keys, index)) : NULL;
+  if (keys != NULL) {
+    retval = PyUnicode_AsUTF8(ItemAt(keys, index));
+    Py_DECREF(keys);
+  }
+  return retval;
 }
 
 static const char *PropertyValueAt(const void *node, int index) {
@@ -78,8 +94,13 @@ static const char *PropertyValueAt(const void *node, int index) {
     return NULL;
   }
 
+  const char *retval = NULL;
   PyObject *values = PyMapping_Values(properties);
-  return values ? PyUnicode_AsUTF8(ItemAt(values, index)) : NULL;
+  if (values != NULL) {
+    retval = PyUnicode_AsUTF8(ItemAt(values, index));
+    Py_DECREF(values);
+  }
+  return retval;
 }
 
 static uint32_t PositionValue(const void* node, const char *prop, const char *field) {
@@ -275,7 +296,9 @@ static PyObject *PyFilter(PyObject *self, PyObject *args)
     PyList_SET_ITEM(list, i, node);
   }
   NodesFree(nodes);
-  return PySeqIter_New(list);
+  PyObject *iter = PySeqIter_New(list);
+  Py_DECREF(list);
+  return iter;
 }
 
 static PyMethodDef extension_methods[] = {

bblfsh/test.py

@@ -188,6 +188,20 @@ def _validate_resp(self, resp):
         # Sometimes its fully qualified, sometimes is just "Node"... ditto
         assert(resp.uast.__class__.__name__.endswith('Node'))
 
+    def testManyFilters(self):
+        root = self.client.parse(__file__).uast
+        root.properties['k1'] = 'v2'
+        root.properties['k2'] = 'v1'
+
+        import resource
+        before = resource.getrusage(resource.RUSAGE_SELF)
+        for _ in range(100):
+            filter(root, "//*[@roleIdentifier]")
+        after = resource.getrusage(resource.RUSAGE_SELF)
+
+        # Check that memory usage has not doubled after running the filter
+        self.assertLess(after[2] / before[2], 2.0)
+
     def _validate_filter(self, resp):
         results = filter(resp.uast, "//Import[@roleImport and @roleDeclaration]//alias")
         self.assertEqual(next(results).token, "os")

setup.py

@@ -5,7 +5,7 @@
 from setuptools import setup, find_packages, Extension
 from setuptools.command.build_ext import build_ext
 
-LIBUAST_VERSION = "v1.6.0"
+LIBUAST_VERSION = "v1.8.2"
 SDK_VERSION = "v1.8.0"
 SDK_MAJOR = SDK_VERSION.split('.')[0]
 PYTHON = "python3"
@@ -134,7 +134,7 @@ def main():
         },
         name="bblfsh",
         description="Fetches Universal Abstract Syntax Trees from Babelfish.",
-        version="2.8.1",
+        version="2.8.2",
         license="Apache 2.0",
         author="source{d}",
         author_email="language-analysis@sourced.tech",