Exemplo n.º 1
0
def test_recursive_call(set_up_test_data):
    EXPECTED = TEST_RESOURCES_DIR / "expect_recursive_chains.csv"
    outfile = TEST_DATA_DIR / "recursive_chains.csv"
    cmd = [QUERY_FC,
           "--calls", CALLS_FILE,
           "--from_function", "recursive_call",
           "--to_function", "recursive_call",
           "--out", outfile]
    assert subprocess.run(cmd).returncode == 0
    assert EXPECTED.exists()
    df_expected = pd.read_csv(EXPECTED)
    df_generated = pd.read_csv(outfile)
    df_diff = test_utils.df_difference(df_expected, df_generated)
    assert df_diff.empty, test_utils.df_to_string(df_diff)
Exemplo n.º 2
0
def test_from_function_with_fname(set_up_test_data):
    EXPECTED = TEST_RESOURCES_DIR / "expect_single_chain_right.csv"
    outfile = TEST_DATA_DIR / "single_chain_right.csv"
    cmd = [QUERY_FC,
           "--calls", CALLS_FILE,
           "--from_function", "test-chain.c:chain1",
           "--to_function", "chain3",
           "--direction", "right",
           "--out", outfile]
    assert subprocess.run(cmd).returncode == 0
    assert EXPECTED.exists()
    df_expected = pd.read_csv(EXPECTED)
    df_generated = pd.read_csv(outfile)
    df_diff = test_utils.df_difference(df_expected, df_generated)
    assert df_diff.empty, test_utils.df_to_string(df_diff)
Exemplo n.º 3
0
def check_function_calls_from(target_bc_file_name, cpp=False):
    target_bc_file = TEST_RESOURCES_DIR / target_bc_file_name
    assert Path(target_bc_file).exists()

    outfile = TEST_DATA_DIR / "calls.csv"
    if target_bc_file_name.endswith(".bclist"):
        target_bc_file = "@%s" % target_bc_file

    if cpp:
        cmd = [
            CG_BIN,
            "-cpp_linked_bitcode",
            target_bc_file,
            "-o",
            outfile,
            target_bc_file,
        ]
    else:
        cmd = [
            CG_BIN,
            "-o",
            outfile,
            target_bc_file,
        ]
    assert subprocess.run(cmd).returncode == 0
    assert Path(outfile).exists()

    # Check the generated contents matches expected
    assert EXPECTED.exists()
    df_expected = pd.read_csv(EXPECTED)
    caller_regex = "^%s" % target_bc_file_name.split(".")[0]
    df_expected = df_regex_filter(df_expected, "caller_filename", caller_regex)
    df_generated = pd.read_csv(outfile)
    # Remove entries where caller or callee function name begins with '__cxx'.
    # These are library functions we want to ignore in the tests:
    df_generated = df_generated[
        ~df_generated["caller_function"].str.startswith("__cxx", na=False)
        & ~df_generated["callee_function"].str.startswith("__cxx", na=False)]
    df_diff = test_utils.df_difference(df_expected, df_generated)
    assert df_diff.empty, test_utils.df_to_string(df_diff)
Exemplo n.º 4
0
def test_function_specific(set_up_test_data):
    cmd = [QUERY_FC,
           "--calls", CALLS_FILE_DUPL,
           "--from_function", "test-chain.c:chain2",
           "--to_function", "chain[0-9]",
           "--direction", "both"]
    assert subprocess.run(cmd).returncode == 0

    EXPECTED = TEST_RESOURCES_DIR / "expect_function_filename_specific.csv"
    outfile = TEST_DATA_DIR / "function_filename_specific.csv"
    cmd = [QUERY_FC,
           "--calls", CALLS_FILE_DUPL,
           "--from_function", "test-chain-alt.c:chain2",
           "--to_function", "chain[0-9]",
           "--direction", "left",
           "--out", outfile]
    assert subprocess.run(cmd).returncode == 0
    assert EXPECTED.exists()
    df_expected = pd.read_csv(EXPECTED)
    df_generated = pd.read_csv(outfile)
    df_diff = test_utils.df_difference(df_expected, df_generated)
    assert df_diff.empty, test_utils.df_to_string(df_diff)
Exemplo n.º 5
0
def test_csv_graph_duplicate(set_up_test_data):
    callgraph_csv = TEST_DATA_DIR / "calls.csv"
    generate_call_graph_from("test-duplicate-path.bclist", callgraph_csv)

    query_out = TEST_DATA_DIR / "graph.csv"
    cmd = [
        QUERY_CG, "--csv", callgraph_csv, "--function", "main", "--depth",
        "10", "--out", query_out
    ]
    assert subprocess.run(cmd).returncode == 0
    assert Path(query_out).exists()

    query_out_inverse = TEST_DATA_DIR / "graph_inverse.csv"
    cmd = [
        QUERY_CG, "--csv", callgraph_csv, "--function", "printf", "--depth",
        "10", "--inverse", "--out", query_out_inverse
    ]

    assert subprocess.run(cmd).returncode == 0
    assert Path(query_out_inverse).exists()

    # When 'depth' covers the entire callgraph, the output from
    # the two above commands should be the same, except for column
    # 'call_depth': below, we remove that column from both outputs and
    # compare the two dataframes

    df_csv = pd.read_csv(query_out)
    assert not df_csv.empty
    df_csv = df_csv.drop('call_depth', 1)
    df_csv = df_csv.sort_values(by=['caller_line'])

    df_csv_inv = pd.read_csv(query_out_inverse)
    assert not df_csv_inv.empty
    df_csv_inv = df_csv_inv.drop('call_depth', 1)
    df_csv_inv = df_csv_inv.sort_values(by=['caller_line'])

    df_diff = test_utils.df_difference(df_csv, df_csv_inv)
    assert df_diff.empty, test_utils.df_to_string(df_diff)