Add testcase

This commit is contained in:
Rasmus Wriedt Larsen
2019-10-10 15:31:42 +02:00
parent 36bb5f54ce
commit bf197b9f20
4 changed files with 74 additions and 0 deletions

View File

@@ -0,0 +1,45 @@
import python
import semmle.python.security.TaintTracking
import semmle.python.security.strings.Untrusted
class FooSource extends TaintSource {
FooSource() { this.(CallNode).getFunction().(NameNode).getId() = "foo_source" }
override predicate isSourceOf(TaintKind kind) { kind instanceof UntrustedStringKind }
override string toString() { result = "FooSource" }
}
class FooSink extends TaintSink {
FooSink() {
exists(CallNode call |
call.getFunction().(NameNode).getId() = "foo_sink" and
call.getAnArg() = this
)
}
override predicate sinks(TaintKind kind) { kind instanceof UntrustedStringKind }
override string toString() { result = "FooSink" }
}
class FooConfig extends TaintTracking::Configuration {
FooConfig() { this = "FooConfig" }
override predicate isSource(TaintTracking::Source source) { source instanceof FooSource }
override predicate isSink(TaintTracking::Sink sink) { sink instanceof FooSink }
}
class BarSink extends TaintSink {
BarSink() {
exists(CallNode call |
call.getFunction().(NameNode).getId() = "bar_sink" and
call.getAnArg() = this
)
}
override predicate sinks(TaintKind kind) { kind instanceof UntrustedStringKind }
override string toString() { result = "BarSink" }
}

View File

@@ -0,0 +1 @@
| test.py:16:9:16:20 | foo_source() | test.py:17:14:17:14 | x |

View File

@@ -0,0 +1,6 @@
import python
import Config
from FooConfig config, TaintedPathSource src, TaintedPathSink sink
where config.hasFlowPath(src, sink)
select src.getSource(), sink.getSink()

View File

@@ -0,0 +1,22 @@
def foo_source():
return 'foo'
def foo_sink(x):
if x == 'foo':
print('fire the foo missiles')
def bar_sink(x):
if x == 'bar':
print('fire the bar missiles')
def should_report():
x = foo_source()
foo_sink(x)
def should_not_report():
x = foo_source()
bar_sink(x)