File tree Expand file tree Collapse file tree
powershell/ql/test/TestUtilities Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -7,6 +7,8 @@ import semmle.code.powershell.dataflow.DataFlow
77
88predicate defaultSource ( DataFlow:: Node src ) {
99 src .asStmt ( ) .getStmt ( ) .( Cmd ) .getCommandName ( ) = [ "Source" , "Taint" ]
10+ or
11+ src .asParameter ( ) .getName ( ) .matches ( [ "Source%" , "Taint%" ] )
1012}
1113
1214predicate defaultSink ( DataFlow:: Node sink ) {
@@ -15,5 +17,9 @@ predicate defaultSink(DataFlow::Node sink) {
1517
1618string 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}
You can’t perform that action at this time.
0 commit comments