Fix issue #116 - memory leaks in exprnode.c

cjdrake · Mar 17, 2015 · cb56a58 · cb56a58
1 parent aac235d
commit cb56a58
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pyeda/boolalg/exprnodemodule.c b/pyeda/boolalg/exprnodemodule.c
@@ -520,10 +520,13 @@ ExprNode_restrict(ExprNode *self, PyObject *args)
     }
 
     if ((ex = BoolExpr_Restrict(self->ex, var2ex)) == NULL) {
+        BoolExprDict_Del(var2ex);
         PyErr_SetString(Error, "BoolExpr_Restrict failed");
         return NULL;
     }
 
+    BoolExprDict_Del(var2ex);
+
     if (ex == self->ex) {
         BoolExpr_DecRef(ex);
         Py_INCREF((PyObject *) self);
@@ -568,10 +571,13 @@ ExprNode_compose(ExprNode *self, PyObject *args)
     }
 
     if ((ex = BoolExpr_Compose(self->ex, var2ex)) == NULL) {
+        BoolExprDict_Del(var2ex);
         PyErr_SetString(Error, "BoolExpr_Compose failed");
         return NULL;
     }
 
+    BoolExprDict_Del(var2ex);
+
     if (ex == self->ex) {
         BoolExpr_DecRef(ex);
         Py_INCREF((PyObject *) self);