Remove more than half the stack usage of CFG traversal

avast · Feb 6, 2019 · 4dd9bea · 4dd9bea
1 parent 7febc9c
commit 4dd9bea
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/llvmir2hll/graphs/cfg/cfg_traversal.cpp b/src/llvmir2hll/graphs/cfg/cfg_traversal.cpp
@@ -168,7 +168,7 @@ bool CFGTraversal::getCurrRetVal() const {
 */
 bool CFGTraversal::performTraversalImpl(ShPtr<CFG::Node> node,
 		CFG::stmt_iterator stmtIter) {
-	if (stmtIter != node->stmt_end()) {
+	while (stmtIter != node->stmt_end()) {
 		// We're not at the end of the node, so check the statement under
 		// stmtIter.
 
@@ -182,7 +182,7 @@ bool CFGTraversal::performTraversalImpl(ShPtr<CFG::Node> node,
 			return getCurrRetVal();
 		}
 
-		return performTraversalImpl(node, ++stmtIter);
+		++stmtIter;
 	}
 
 	// We have reached the end of the node, so check node's successors.
@@ -202,7 +202,7 @@ bool CFGTraversal::performTraversalImpl(ShPtr<CFG::Node> node,
 bool CFGTraversal::performReverseTraversalImpl(ShPtr<CFG::Node> node,
 		CFG::stmt_reverse_iterator stmtRIter) {
 
-	if (stmtRIter != node->stmt_rend()) {
+	while (stmtRIter != node->stmt_rend()) {
 		// We're not at the start of the node, so check the statement under
 		// stmtRIter.
 
@@ -216,7 +216,7 @@ bool CFGTraversal::performReverseTraversalImpl(ShPtr<CFG::Node> node,
 			return getCurrRetVal();
 		}
 
-		return performReverseTraversalImpl(node, ++stmtRIter);
+		++stmtRIter;
 	}
 
 	// We have reached the start of the node, so check node's predecessors.