mirror of
https://github.com/github/codeql.git
synced 2025-12-18 18:10:39 +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.
33 lines
1.1 KiB
Plaintext
33 lines
1.1 KiB
Plaintext
import python
|
|
import semmle.python.dataflow.TaintTracking
|
|
import semmle.python.web.HttpRequest
|
|
import semmle.python.security.strings.Untrusted
|
|
|
|
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 = "fail"
|
|
// if expected_taint = has_taint then test_res = "✓" else test_res = "✕"
|
|
select arg.getLocation().toString(), test_res, call.getScope().(Function).getName(), arg.toString(),
|
|
taint_string
|