mirror of
https://github.com/github/codeql.git
synced 2025-12-30 23:58:15 +01:00
a82fa04d8a
This can't be seen on the example, but I went through quite a lot of iterations before arriving at this fairly simple solution.
37 lines
1.1 KiB
Plaintext
37 lines
1.1 KiB
Plaintext
private import python
|
|
private import experimental.dataflow.DataFlow
|
|
private import experimental.dataflow.TaintTracking
|
|
|
|
// Helpers modeling MyClass
|
|
/** A data-flow Node representing an instance of MyClass. */
|
|
abstract class MyClass extends DataFlow::Node { }
|
|
|
|
private DataFlow::Node myClassGetValue(MyClass qualifier, DataFlow::TypeTracker t) {
|
|
t.startInAttr("get_value") and
|
|
result = qualifier
|
|
or
|
|
exists(DataFlow::TypeTracker t2 | result = myClassGetValue(qualifier, t2).track(t2, t))
|
|
}
|
|
|
|
DataFlow::Node myClassGetValue(MyClass qualifier) {
|
|
result = myClassGetValue(qualifier, DataFlow::TypeTracker::end())
|
|
}
|
|
|
|
// Config
|
|
class SourceCall extends DataFlow::Node, MyClass {
|
|
SourceCall() { this.asCfgNode().(CallNode).getFunction().(NameNode).getId() = "source" }
|
|
}
|
|
|
|
class SharedConfig extends TaintTracking::Configuration {
|
|
SharedConfig() { this = "SharedConfig" }
|
|
|
|
override predicate isSource(DataFlow::Node source) { source instanceof SourceCall }
|
|
|
|
override predicate isSink(DataFlow::Node sink) {
|
|
exists(CallNode call |
|
|
call.getFunction().(NameNode).getId() = "sink" and
|
|
call.getArg(0) = sink.asCfgNode()
|
|
)
|
|
}
|
|
}
|