Query: Use index stack dfg in the DFG query (#1972)

* Query: Use index stack dfg in the DFG query * Configure stack usage
Fraunhofer-AISEC · Jan 24, 2025 · 81b9939 · 81b9939
1 parent e4845ec
commit 81b9939
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/cpg-analysis/src/main/kotlin/de/fraunhofer/aisec/cpg/query/Query.kt b/cpg-analysis/src/main/kotlin/de/fraunhofer/aisec/cpg/query/Query.kt
@@ -205,9 +205,16 @@ fun dataFlow(
     to: Node,
     collectFailedPaths: Boolean = true,
     findAllPossiblePaths: Boolean = true,
+    useIndexStack: Boolean = true,
 ): QueryTree<Boolean> {
     val evalRes =
-        from.followNextFullDFGEdgesUntilHit(collectFailedPaths, findAllPossiblePaths) { it == to }
+        from.followNextDFGEdgesUntilHit(
+            collectFailedPaths = collectFailedPaths,
+            findAllPossiblePaths = findAllPossiblePaths,
+            useIndexStack = useIndexStack,
+        ) {
+            it == to
+        }
     val allPaths = evalRes.fulfilled.map { QueryTree(it) }.toMutableList()
     if (collectFailedPaths) allPaths.addAll(evalRes.failed.map { QueryTree(it) })
     return QueryTree(
@@ -227,9 +234,15 @@ fun dataFlow(
     predicate: (Node) -> Boolean,
     collectFailedPaths: Boolean = true,
     findAllPossiblePaths: Boolean = true,
+    useIndexStack: Boolean = true,
 ): QueryTree<Boolean> {
     val evalRes =
-        from.followNextFullDFGEdgesUntilHit(collectFailedPaths, findAllPossiblePaths, predicate)
+        from.followNextDFGEdgesUntilHit(
+            collectFailedPaths = collectFailedPaths,
+            findAllPossiblePaths = findAllPossiblePaths,
+            useIndexStack = useIndexStack,
+            predicate = predicate,
+        )
     val allPaths = evalRes.fulfilled.map { QueryTree(it) }.toMutableList()
     if (collectFailedPaths) allPaths.addAll(evalRes.failed.map { QueryTree(it) })
     return QueryTree(