mirror of
https://github.com/github/codeql.git
synced 2025-12-21 03:06:31 +01:00
76 lines
1.8 KiB
Python
76 lines
1.8 KiB
Python
# Add taintlib to PATH so it can be imported during runtime without any hassle
|
|
import sys; import os; sys.path.append(os.path.dirname(os.path.dirname((__file__))))
|
|
from taintlib import *
|
|
|
|
# This has no runtime impact, but allows autocomplete to work
|
|
from typing import TYPE_CHECKING
|
|
if TYPE_CHECKING:
|
|
from ..taintlib import *
|
|
|
|
|
|
# Actual tests
|
|
|
|
def unpacking():
|
|
l = TAINTED_LIST[0:3]
|
|
a, b, c = l
|
|
ensure_tainted(a, b, c) # $ tainted
|
|
|
|
|
|
def unpacking_to_list():
|
|
l = TAINTED_LIST[0:3]
|
|
[a, b, c] = l
|
|
ensure_tainted(a, b, c) # $ tainted
|
|
|
|
|
|
def nested():
|
|
l = TAINTED_LIST[0:3]
|
|
ll = [l, l, l]
|
|
|
|
# list
|
|
[[a1, a2, a3], b, c] = ll
|
|
ensure_tainted(a1, a2, a3, b, c) # $ tainted
|
|
|
|
# tuple
|
|
((a1, a2, a3), b, c) = ll
|
|
ensure_tainted(a1, a2, a3, b, c) # $ tainted
|
|
|
|
# mixed
|
|
[(a1, a2, a3), b, c] = ll
|
|
ensure_tainted(a1, a2, a3, b, c) # $ tainted
|
|
|
|
|
|
def unpack_from_set():
|
|
# no guarantee on ordering ... don't know why you would ever do this
|
|
a, b, c = {"foo", "bar", TAINTED_STRING}
|
|
# either all should be tainted, or none of them
|
|
ensure_tainted(a, b, c) # $ tainted
|
|
|
|
|
|
def contrived_1():
|
|
# A contrived example. Don't know why anyone would ever actually do this.
|
|
tainted_list = TAINTED_LIST[0:3]
|
|
no_taint_list = [1,2,3]
|
|
|
|
(a, b, c), (d, e, f) = tainted_list, no_taint_list
|
|
ensure_tainted(a, b, c) # $ tainted
|
|
ensure_not_tainted(d, e, f) # $ SPURIOUS: tainted
|
|
|
|
|
|
def contrived_2():
|
|
# A contrived example. Don't know why anyone would ever actually do this.
|
|
|
|
# Old taint tracking was only able to handle taint nested 2 levels in sequences,
|
|
# so would not mark a, b, c as tainted
|
|
[[[ (a, b, c) ]]] = [[[ TAINTED_LIST[0:3] ]]]
|
|
ensure_tainted(a, b, c) # $ tainted
|
|
|
|
|
|
# Make tests runable
|
|
|
|
unpacking()
|
|
unpacking_to_list()
|
|
nested()
|
|
unpack_from_set()
|
|
contrived_1()
|
|
contrived_2()
|