mirror of
https://github.com/github/codeql.git
synced 2026-01-02 09:16:34 +01:00
34 lines
1.2 KiB
Plaintext
34 lines
1.2 KiB
Plaintext
import python
|
|
import experimental.dataflow.DataFlow
|
|
private import experimental.dataflow.internal.DataFlowPrivate as DataFlowPrivate
|
|
|
|
/**
|
|
* A configuration to check routing of arguments through magic methods.
|
|
*/
|
|
class ArgumentRoutingConfig extends DataFlow::Configuration {
|
|
ArgumentRoutingConfig() { this = "ArgumentRoutingConfig" }
|
|
|
|
override predicate isSource(DataFlow::Node node) {
|
|
exists(AssignmentDefinition def, DataFlowPrivate::DataFlowCall call |
|
|
def.getVariable() = node.(DataFlow::EssaNode).getVar() and
|
|
def.getValue() = call.getNode() and
|
|
call.getNode().(CallNode).getFunction().(NameNode).getId().matches("With\\_%")
|
|
) and
|
|
node.(DataFlow::EssaNode).getVar().getName().matches("with\\_%")
|
|
}
|
|
|
|
override predicate isSink(DataFlow::Node node) {
|
|
exists(CallNode call |
|
|
call.getFunction().(NameNode).getId() = "SINK1" and
|
|
node.(DataFlow::CfgNode).getNode() = call.getAnArg()
|
|
)
|
|
}
|
|
}
|
|
|
|
from DataFlow::Node source, DataFlow::Node sink
|
|
where
|
|
source.getLocation().getFile().getBaseName() = "classes.py" and
|
|
sink.getLocation().getFile().getBaseName() = "classes.py" and
|
|
exists(ArgumentRoutingConfig cfg | cfg.hasFlow(source, sink))
|
|
select source, sink
|