dict.py

# 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 *

# 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")


def In(tainted):
    def captureIn1():
        sinkI1 = tainted
        SINK(sinkI1) #$ MISSING:captured
    captureIn1()

    def captureIn2():
        def m():
            sinkI2 = tainted
            SINK(sinkI2) #$ MISSING:captured
        m()
    captureIn2()

    # captureIn3 = lambda arg:(
    #     sinkI3 = tainted;
    #     check(sinkI3);
    #     return arg)
    # [ captureIn3(x) for x in " " ]

    def captureIn1NotCalled():
        nonSink0 = tainted
        SINK_F(nonSink0)

    def captureIn2NotCalled():
        def m():
            nonSink0 = tainted
            SINK_F(nonSink0)
    captureIn2NotCalled()

@expects(2)
def test_In():
    In(SOURCE)

def Out():
    sinkO1 = { "x": "" }
    def captureOut1():
        sinkO1["x"] = "source"
    captureOut1()
    SINK(sinkO1["x"]) #$ MISSING:captured

    sinkO2 = { "x": "" }
    def captureOut2():
        def m():
            sinkO2["x"] = "source"
        m()
    captureOut2()
    SINK(sinkO2["x"]) #$ MISSING:captured

    nonSink0 = { "x": "" }
    def captureOut1NotCalled():
        nonSink0["x"] = "source"
    SINK_F(nonSink0["x"])

    def captureOut2NotCalled():
        def m():
            nonSink0["x"] = "source"
    captureOut2NotCalled()
    SINK_F(nonSink0["x"])

@expects(4)
def test_Out():
    Out()

def Through(tainted):
    sinkO1 = { "x": "" }
    def captureOut1():
        sinkO1["x"] = tainted
    captureOut1()
    SINK(sinkO1["x"]) #$ MISSING:captured

    sinkO2 = { "x": "" }
    def captureOut2():
        def m():
            sinkO2["x"] = tainted
        m()
    captureOut2()
    SINK(sinkO2["x"]) #$ MISSING:captured

    nonSink0 = { "x": "" }
    def captureOut1NotCalled():
        nonSink0["x"] = tainted
    SINK_F(nonSink0["x"])

    def captureOut2NotCalled():
        def m():
            nonSink0["x"] = tainted
    captureOut2NotCalled()
    SINK_F(nonSink0["x"])

@expects(4)
def test_Through():
    Through(SOURCE)