Rust: Make improvements to getArgString based on PR comments

github · Nov 12, 2024 · d9d3027 · d9d3027
1 parent cd2038a
commit d9d3027
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/rust/ql/test/utils/InlineFlowTest.qll b/rust/ql/test/utils/InlineFlowTest.qll
@@ -26,8 +26,13 @@ private module FlowTestImpl implements InputSig<Location, RustDataFlow> {
     result = src.asExpr().(CallExpr).getArgList().getArg(0).toString()
   }
 
+  bindingset[src, sink]
   string getArgString(DataFlow::Node src, DataFlow::Node sink) {
-    (if exists(getSourceArgString(src)) then result = getSourceArgString(src) else result = "") and
+    (
+      result = getSourceArgString(src)
+      or
+      not exists(getSourceArgString(src)) and result = ""
+    ) and
     exists(sink)
   }