Python: Add test for missing relative import in namespace packages

This commit is contained in:
Taus
2026-03-11 16:01:47 +00:00
parent a2f45f1b5b
commit 48bf4fd82a
4 changed files with 43 additions and 0 deletions

View File

@@ -0,0 +1,5 @@
from . import helper
def use_relative():
tainted = source()
helper.process(tainted)

View File

@@ -0,0 +1,2 @@
def process(value):
sink(value) #$ MISSING: prints=source

View File

@@ -0,0 +1,35 @@
import python
import semmle.python.dataflow.new.DataFlow
import semmle.python.dataflow.new.TaintTracking
import utils.test.InlineExpectationsTest
private module TestConfig implements DataFlow::ConfigSig {
predicate isSource(DataFlow::Node node) {
node.(DataFlow::CallCfgNode).getFunction().asCfgNode().(NameNode).getId() = "source"
}
predicate isSink(DataFlow::Node node) {
exists(DataFlow::CallCfgNode call |
call.getFunction().asCfgNode().(NameNode).getId() = "sink" and
node = call.getArg(0)
)
}
}
private module TestFlow = TaintTracking::Global<TestConfig>;
module FlowTest implements TestSig {
string getARelevantTag() { result = "prints" }
predicate hasActualResult(Location location, string element, string tag, string value) {
exists(DataFlow::Node sink |
TestFlow::flow(_, sink) and
tag = "prints" and
location = sink.getLocation() and
value = "source" and
element = sink.toString()
)
}
}
import MakeTest<FlowTest>