hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-05-01 03:35:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

Python: Fix CUSTOM_SOURCE dataflow regression test

Browse Source
This commit is contained in:
Rasmus Wriedt Larsen
2020-09-03 15:03:53 +02:00
parent febbe1229a
commit 29bf98ad26
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/ql/test/experimental/dataflow/regression/custom_dataflow.ql
View File

@@ -7,19 +7,20 @@
* hope to remove the false positive.
*/
import python
import experimental.dataflow.DataFlow
class CustomTestConfiguration extends DataFlow::Configuration {
CustomTestConfiguration() { this = "CustomTestConfiguration" }
override predicate isSource(DataFlow::Node node) {
node.(DataFlow::CfgNode).getNode().(NameNode).getId() = "CUSTOM_SOURCE"
node.asCfgNode().(NameNode).getId() = "CUSTOM_SOURCE"
}
override predicate isSink(DataFlow::Node node) {
exists(CallNode call |
call.getFunction().(NameNode).getId() in ["CUSTOM_SINK", "CUSTOM_SINK_F"] and
node.(DataFlow::CfgNode).getNode() = call.getAnArg()
node.asCfgNode() = call.getAnArg()
)
}
}