Skip to content

Commit 0dc612f

Browse files
committed
C#: Re-factor the TaintTracking test.
1 parent b5b0d60 commit 0dc612f

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

csharp/ql/test/library-tests/cil/dataflow/TaintTracking.ql

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,16 +6,16 @@ import semmle.code.csharp.dataflow.TaintTracking3
66
import semmle.code.csharp.dataflow.TaintTracking4
77
import semmle.code.csharp.dataflow.TaintTracking5
88

9-
class FlowConfig extends TaintTracking::Configuration {
10-
FlowConfig() { this = "FlowConfig" }
9+
module FlowConfig implements DataFlow::ConfigSig {
10+
predicate isSource(DataFlow::Node source) { source.asExpr() instanceof Literal }
1111

12-
override predicate isSource(DataFlow::Node source) { source.asExpr() instanceof Literal }
13-
14-
override predicate isSink(DataFlow::Node sink) {
12+
predicate isSink(DataFlow::Node sink) {
1513
exists(LocalVariable decl | sink.asExpr() = decl.getInitializer())
1614
}
1715
}
1816

19-
from FlowConfig config, DataFlow::Node source, DataFlow::Node sink
20-
where config.hasFlow(source, sink)
17+
module Flow = TaintTracking::Global<FlowConfig>;
18+
19+
from DataFlow::Node source, DataFlow::Node sink
20+
where Flow::flow(source, sink)
2121
select source, sink

0 commit comments

Comments
 (0)