mirror of
https://github.com/github/codeql.git
synced 2025-12-19 18:33:16 +01:00
f07a7bf8cf
Will need subsequent PRs fixing up test failures (due to deprecated methods moving around), but other than that everything should be straight-forward.
32 lines
1.0 KiB
Plaintext
32 lines
1.0 KiB
Plaintext
import python
|
|
import semmle.python.dataflow.TaintTracking
|
|
import Taint
|
|
|
|
from
|
|
Call call, Expr arg, boolean expected_taint, boolean has_taint, string test_res,
|
|
string taint_string
|
|
where
|
|
call.getLocation().getFile().getShortName() = "test.py" and
|
|
(
|
|
call.getFunc().(Name).getId() = "ensure_tainted" and
|
|
expected_taint = true
|
|
or
|
|
call.getFunc().(Name).getId() = "ensure_not_tainted" and
|
|
expected_taint = false
|
|
) and
|
|
arg = call.getAnArg() and
|
|
(
|
|
not exists(TaintedNode tainted | tainted.getAstNode() = arg) and
|
|
taint_string = "<NO TAINT>" and
|
|
has_taint = false
|
|
or
|
|
exists(TaintedNode tainted | tainted.getAstNode() = arg |
|
|
taint_string = tainted.getTaintKind().toString()
|
|
) and
|
|
has_taint = true
|
|
) and
|
|
if expected_taint = has_taint then test_res = "ok" else test_res = "failure"
|
|
// if expected_taint = has_taint then test_res = "✓" else test_res = "✕"
|
|
select arg.getLocation().toString(), call.getScope().(Function).getName(), arg.toString(),
|
|
taint_string, test_res
|