Python: Allow provenance in additional taint steps

This commit is contained in:
Rasmus Lerchedahl Petersen
2024-05-16 14:04:10 +02:00
parent ea3cc51286
commit 20ea9255a1
2 changed files with 5 additions and 1 deletions

View File

@@ -29,6 +29,8 @@ private module Cached {
or
any(AdditionalTaintStep a).step(nodeFrom, nodeTo) and
model = "AdditionalTaintStep"
or
any(AdditionalTaintStep a).step(nodeFrom, nodeTo, model)
}
/**

View File

@@ -48,5 +48,7 @@ class AdditionalTaintStep extends Unit {
* Holds if the step from `nodeFrom` to `nodeTo` should be considered a taint
* step for all configurations.
*/
abstract predicate step(DataFlow::Node nodeFrom, DataFlow::Node nodeTo);
predicate step(DataFlow::Node nodeFrom, DataFlow::Node nodeTo) { none() }
predicate step(DataFlow::Node nodeFrom, DataFlow::Node nodeTo, string model) { none() }
}