Fix a memory leak by introducing destructor for InputQueryBuilder

NeuralNetworkVerification · Feb 5, 2025 · 6f707d7 · 6f707d7
1 parent 391e7ec
commit 6f707d7
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/src/input_parsers/InputQueryBuilder.cpp b/src/input_parsers/InputQueryBuilder.cpp
@@ -231,3 +231,42 @@ Equation *InputQueryBuilder::findEquationWithOutputVariable( Variable variable )
     }
     return NULL;
 }
+
+InputQueryBuilder::~InputQueryBuilder()
+{
+    for ( ReluConstraint *constraintPtr : _reluList )
+    {
+        delete constraintPtr;
+    }
+    _reluList = {};
+
+    for ( LeakyReluConstraint *constraintPtr : _leakyReluList )
+    {
+        delete constraintPtr;
+    }
+    _leakyReluList = {};
+
+    for ( SigmoidConstraint *constraintPtr : _sigmoidList )
+    {
+        delete constraintPtr;
+    }
+    _sigmoidList = {};
+
+    for ( MaxConstraint *constraintPtr : _maxList )
+    {
+        delete constraintPtr;
+    }
+    _maxList = {};
+
+    for ( AbsoluteValueConstraint *constraintPtr : _absList )
+    {
+        delete constraintPtr;
+    }
+    _absList = {};
+
+    for ( SignConstraint *constraintPtr : _signList )
+    {
+        delete constraintPtr;
+    }
+    _signList = {};
+}
diff --git a/src/input_parsers/InputQueryBuilder.h b/src/input_parsers/InputQueryBuilder.h
@@ -75,6 +75,7 @@ class InputQueryBuilder
     void generateQuery( IQuery &query );
 
     Equation *findEquationWithOutputVariable( Variable variable );
+    virtual ~InputQueryBuilder();
 };
 
 #endif // __InputQueryBuilder_h__