Skip to content

Commit d3b9e13

Browse files
committed
PS: Extend the set of sources in tests.
1 parent 3c18124 commit d3b9e13

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

powershell/ql/test/TestUtilities/InlineFlowTestUtil.qll

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ import semmle.code.powershell.dataflow.DataFlow
77

88
predicate defaultSource(DataFlow::Node src) {
99
src.asStmt().getStmt().(Cmd).getCommandName() = ["Source", "Taint"]
10+
or
11+
src.asParameter().getName().matches(["Source%", "Taint%"])
1012
}
1113

1214
predicate defaultSink(DataFlow::Node sink) {
@@ -15,5 +17,9 @@ predicate defaultSink(DataFlow::Node sink) {
1517

1618
string getSourceArgString(DataFlow::Node src) {
1719
defaultSource(src) and
18-
src.asStmt().getStmt().(Cmd).getAnArgument().(StringConstExpr).getValue().getValue() = result
20+
(
21+
src.asStmt().getStmt().(Cmd).getAnArgument().(StringConstExpr).getValue().getValue() = result
22+
or
23+
src.asParameter().getName().regexpCapture(["Source(.+)", "Taint(.+)"], 1) = result
24+
)
1925
}

0 commit comments

Comments
 (0)