mirror of
https://github.com/github/codeql.git
synced 2025-12-20 10:46:30 +01:00
c054ba6a97
This provides variable capture in standard situations: - nested functions - lambdas There are some deficiencies: - we do not yet handle objects capturing variables. - we do not handle variables captured via the `nonlocal` keyword. This should be solved at the AST level, though, and then it should "just work". There are still inconsistencies in the case where a `SynthesizedCaptureNode` has a comprehensions as its enclosing callable. In this case, `TFunction(cn.getEnclosingCallable())` is not defined and so getEnclosingCallable does not exist for the `CaptureNode`.
107 lines
2.2 KiB
Python
107 lines
2.2 KiB
Python
# Here we test writing to a captured global variable via the `global` keyword (see `out`).
|
|
# We also test reading one captured variable and writing the value to another (see `through`).
|
|
|
|
# All functions starting with "test_" should run and execute `print("OK")` exactly once.
|
|
# This can be checked by running validTest.py.
|
|
|
|
import sys
|
|
import os
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname((__file__))))
|
|
from testlib import expects
|
|
|
|
# These are defined so that we can evaluate the test code.
|
|
NONSOURCE = "not a source"
|
|
SOURCE = "source"
|
|
|
|
def is_source(x):
|
|
return x == "source" or x == b"source" or x == 42 or x == 42.0 or x == 42j
|
|
|
|
|
|
def SINK(x):
|
|
if is_source(x):
|
|
print("OK")
|
|
else:
|
|
print("Unexpected flow", x)
|
|
|
|
|
|
def SINK_F(x):
|
|
if is_source(x):
|
|
print("Unexpected flow", x)
|
|
else:
|
|
print("OK")
|
|
|
|
|
|
sinkO1 = ""
|
|
sinkO2 = ""
|
|
nonSink1 = ""
|
|
nonSink2 = ""
|
|
|
|
def out():
|
|
def captureOut1():
|
|
global sinkO1
|
|
sinkO1 = SOURCE
|
|
captureOut1()
|
|
SINK(sinkO1) #$ captured
|
|
|
|
def captureOut2():
|
|
def m():
|
|
global sinkO2
|
|
sinkO2 = SOURCE
|
|
m()
|
|
captureOut2()
|
|
SINK(sinkO2) #$ captured
|
|
|
|
def captureOut1NotCalled():
|
|
global nonSink1
|
|
nonSink1 = SOURCE
|
|
SINK_F(nonSink1) #$ SPURIOUS: captured
|
|
|
|
def captureOut2NotCalled():
|
|
# notice that `m` is not called
|
|
def m():
|
|
global nonSink2
|
|
nonSink2 = SOURCE
|
|
captureOut2NotCalled()
|
|
SINK_F(nonSink2) #$ SPURIOUS: captured
|
|
|
|
@expects(4)
|
|
def test_out():
|
|
out()
|
|
|
|
sinkT1 = ""
|
|
sinkT2 = ""
|
|
nonSinkT1 = ""
|
|
nonSinkT2 = ""
|
|
def through(tainted):
|
|
def captureOut1():
|
|
global sinkT1
|
|
sinkT1 = tainted
|
|
captureOut1()
|
|
SINK(sinkT1) #$ captured
|
|
|
|
def captureOut2():
|
|
def m():
|
|
global sinkT2
|
|
sinkT2 = tainted
|
|
m()
|
|
captureOut2()
|
|
SINK(sinkT2) #$ captured
|
|
|
|
def captureOut1NotCalled():
|
|
global nonSinkT1
|
|
nonSinkT1 = tainted
|
|
SINK_F(nonSinkT1)
|
|
|
|
def captureOut2NotCalled():
|
|
# notice that `m` is not called
|
|
def m():
|
|
global nonSinkT2
|
|
nonSinkT2 = tainted
|
|
captureOut2NotCalled()
|
|
SINK_F(nonSinkT2)
|
|
|
|
@expects(4)
|
|
def test_through():
|
|
through(SOURCE)
|