mirror of
https://github.com/github/codeql.git
synced 2025-12-20 18:56:32 +01:00
47 lines
1019 B
Python
47 lines
1019 B
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
|
|
|
|
# Extended Iterable Unpacking -- PEP 3132
|
|
# https://www.python.org/dev/peps/pep-3132/
|
|
|
|
|
|
def extended_unpacking():
|
|
first, *rest, last = TAINTED_LIST
|
|
ensure_tainted(first, rest, last) # $ tainted
|
|
|
|
|
|
def also_allowed():
|
|
*a, = TAINTED_LIST
|
|
ensure_tainted(a) # $ tainted
|
|
|
|
# for b, *c in [(1, 2, 3), (4, 5, 6, 7)]:
|
|
# print(c)
|
|
# i=0; c=[2,3]
|
|
# i=1; c=[5,6,7]
|
|
|
|
for b, *c in [TAINTED_LIST, TAINTED_LIST]:
|
|
ensure_tainted(b, c) # $ tainted
|
|
|
|
|
|
def nested():
|
|
l = TAINTED_LIST
|
|
ll = [l,l]
|
|
|
|
[[x, *xs], ys] = ll
|
|
ensure_tainted(x, xs, ys) # $ tainted
|
|
|
|
|
|
# Make tests runable
|
|
|
|
extended_unpacking()
|
|
also_allowed()
|
|
nested()
|