angr 9.2.87__py3-none-win_amd64.whl → 9.2.89__py3-none-win_amd64.whl

tests/analyses/test_slicing.py

@@ -1,164 +0,0 @@
-#!/usr/bin/env python3
-# pylint: disable=missing-class-docstring,no-self-use,line-too-long
-__package__ = __package__ or "tests.analyses"  # pylint:disable=redefined-builtin
-
-import logging
-import os
-import unittest
-
-import time
-import angr
-from angr.utils.constants import DEFAULT_STATEMENT
-
-from ..common import bin_location, broken
-
-
-log = logging.getLogger(__name__)
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestSlicing(unittest.TestCase):
-    def test_find_exits(self):
-        slicing_test = angr.Project(
-            os.path.join(test_location, "x86_64", "cfg_1"),
-            use_sim_procedures=True,
-            default_analysis_mode="symbolic",
-            auto_load_libs=False,
-        )
-
-        log.info("Unit test for BackwardSlice._find_exits()")
-        cfg = slicing_test.analyses.CFGEmulated(context_sensitivity_level=2, keep_state=True)
-        cdg = slicing_test.analyses.CDG(cfg)
-        ddg = slicing_test.analyses.DDG(cfg)
-
-        source = cfg.get_any_node(0x40059E)
-
-        # Test the conditional exit
-        target = cfg.get_any_node(0x400594)
-        bs_1 = slicing_test.analyses.BackwardSlice(cfg, cdg, ddg, targets=[(target, -1)], no_construct=True)
-        all_exits = bs_1._find_exits(source, target)
-
-        assert all_exits == {18: [0x400594], DEFAULT_STATEMENT: None}
-
-        # Test the default exit
-        target = cfg.get_any_node(0x4005A4)
-        bs_2 = slicing_test.analyses.BackwardSlice(cfg, cdg, ddg, targets=[(target, -1)], no_construct=True)
-        all_exits = bs_2._find_exits(source, target)
-
-        assert all_exits == {18: [0x400594], DEFAULT_STATEMENT: [0x4005A4]}
-
-    def test_control_flow_slicing(self):
-        slicing_test = angr.Project(
-            os.path.join(test_location, "x86_64", "cfg_1"),
-            use_sim_procedures=True,
-            default_analysis_mode="symbolic",
-            auto_load_libs=False,
-        )
-        log.info("Control Flow Slicing")
-        start = time.time()
-        cfg = slicing_test.analyses.CFGEmulated(context_sensitivity_level=2)
-        end = time.time()
-        duration = end - start
-        log.info("CFG generation is done in %f seconds.", duration)
-
-        target = cfg.get_any_node(0x400594)
-        bs = slicing_test.analyses.BackwardSlice(cfg, None, None, targets=[(target, -1)], control_flow_slice=True)
-        anno_cfg = bs.annotated_cfg()
-        assert anno_cfg.get_whitelisted_statements(0x40057C) is None
-        assert anno_cfg.get_whitelisted_statements(0x400594) is None
-        assert anno_cfg.get_whitelisted_statements(0x4005A4) == []
-
-    @broken
-    def test_backward_slice(self):
-        # TODO: Fix this test case
-
-        slicing_test = angr.Project(
-            os.path.join(test_location, "x86_64", "cfg_1"),
-            use_sim_procedures=True,
-            default_analysis_mode="symbolic",
-            auto_load_libs=False,
-        )
-
-        log.info("Control Flow Slicing")
-
-        cfg = slicing_test.analyses.CFGEmulated(
-            context_sensitivity_level=2, keep_state=True, state_add_options=angr.sim_options.refs
-        )
-        cdg = slicing_test.analyses.CDG(cfg=cfg)
-        ddg = slicing_test.analyses.DDG(cfg=cfg)
-
-        target = cfg.get_any_node(0x4005D3)
-        bs = slicing_test.analyses.BackwardSlice(cfg, cdg, ddg, targets=[(target, -1)], control_flow_slice=False)
-        anno_cfg = bs.annotated_cfg()
-        assert anno_cfg.get_whitelisted_statements(0x40057C) == [2, 3, 7, 20, 21]
-        assert anno_cfg.get_whitelisted_statements(0x400594) == [1, 17, 18, 19, 20]
-        assert anno_cfg.get_whitelisted_statements(0x4005A4) == []
-        assert anno_cfg.get_whitelisted_statements(0x4005CD) == [1, 2, 3, 5, 6, 11, 12, 13, 14, 15, 16, 17, 18, 19]
-
-    def test_last_branching_statement(self):
-        slicing_test = angr.Project(
-            os.path.join(test_location, "armel", "fauxware"),
-            use_sim_procedures=True,
-            auto_load_libs=False,
-        )
-        log.info("Testing _search_for_last_branching_statement.")
-
-        # The IRSB:
-
-        # ------ IMark(0x86dc, 4, 0) ------
-        # t1 = GET:I32(r0)
-        # PUT(pc) = 0x000086e0
-        # ------ IMark(0x86e0, 4, 0) ------
-        # t14 = GET:I32(r11)
-        # t13 = Sub32(t14,0x00000024)
-        # STle(t13) = t1
-        # PUT(pc) = 0x000086e4
-        # ------ IMark(0x86e4, 4, 0) ------
-        # t15 = t13
-        # t8 = LDle:I32(t15)
-        # PUT(r3) = t8
-        # ------ IMark(0x86e8, 4, 0) ------
-        # PUT(cc_op) = 0x00000002
-        # PUT(cc_dep1) = t8
-        # PUT(cc_dep2) = 0x00000000
-        # PUT(cc_ndep) = 0x00000000
-        # PUT(pc) = 0x000086ec
-        # ------ IMark(0x86ec, 4, 0) ------
-        # t26 = CmpEQ32(t8,0x00000000)
-        # t25 = 1Uto32(t26)
-        # t27 = 32to1(t25)
-        # if (t27) { PUT(68) = 0x86f8; Ijk_Boring }
-
-        target_state = slicing_test.factory.blank_state(addr=0x86DC)
-        simgr = slicing_test.factory.simgr(target_state)
-        simgr.step()
-        target = simgr.active[0]
-        log.debug("IRSB:")
-        for line in target.scratch.irsb._pp_str().split("\n"):
-            log.debug(line)
-
-        bs = slicing_test.analyses.BackwardSlice(None, None, None, targets=[(target, -1)], no_construct=True)
-
-        stmt_idx, tmp = bs._last_branching_statement(target.scratch.irsb.statements)
-
-        assert stmt_idx == 22
-        assert tmp == 27
-
-    def test_fauxware(self):
-        b = angr.Project(os.path.join(test_location, "x86_64", "fauxware"), auto_load_libs=False)
-        cfg = b.analyses.CFGEmulated(
-            keep_state=True, state_add_options=angr.sim_options.refs, context_sensitivity_level=2
-        )
-        cdg = b.analyses.CDG(cfg)
-        ddg = b.analyses.DDG(cfg)
-        target_func = cfg.kb.functions.function(name="exit")
-        target_node = cfg.get_any_node(target_func.addr)
-
-        bs = b.analyses.BackwardSlice(cfg, cdg=cdg, ddg=ddg, targets=[(target_node, -1)])
-
-        # Make sure dbg_repr() works
-        log.debug(bs.dbg_repr())
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/analyses/test_stack_pointer_tracker.py

@@ -1,74 +0,0 @@
-#!/usr/bin/env python3
-# pylint: disable=missing-class-docstring,no-self-use,line-too-long
-__package__ = __package__ or "tests.analyses"  # pylint:disable=redefined-builtin
-
-import logging
-import os
-import unittest
-
-import angr
-
-from ..common import bin_location
-
-
-test_location = os.path.join(bin_location, "tests")
-
-
-def run_tracker(track_mem, use_bp):
-    p = angr.Project(os.path.join(test_location, "x86_64", "fauxware"), auto_load_libs=False)
-    p.analyses.CFGFast()
-    main = p.kb.functions["main"]
-    sp = p.arch.sp_offset
-    regs = {sp}
-    if use_bp:
-        bp = p.arch.bp_offset
-        regs.add(bp)
-    sptracker = p.analyses.StackPointerTracker(main, regs, track_memory=track_mem)
-    sp_result = sptracker.offset_after(0x4007D4, sp)
-    if use_bp:
-        bp_result = sptracker.offset_after(0x4007D4, bp)
-        return sp_result, bp_result
-    else:
-        return sp_result
-
-
-def init_tracker(track_mem):
-    p = angr.Project(os.path.join(test_location, "x86_64", "fauxware"), auto_load_libs=False)
-    p.analyses.CFGFast()
-    main = p.kb.functions["main"]
-    sp = p.arch.sp_offset
-    regs = {sp}
-    sptracker = p.analyses.StackPointerTracker(main, regs, track_memory=track_mem)
-    return sptracker, sp
-
-
-class TestStackPointerTracker(unittest.TestCase):
-    def test_stack_pointer_tracker(self):
-        sp_result, bp_result = run_tracker(track_mem=True, use_bp=True)
-        assert sp_result == 8
-        assert bp_result == 0
-
-    def test_stack_pointer_tracker_no_mem(self):
-        sp_result, bp_result = run_tracker(track_mem=False, use_bp=True)
-        assert sp_result == 8
-        assert bp_result is None
-
-    def test_stack_pointer_tracker_just_sp(self):
-        sp_result = run_tracker(track_mem=False, use_bp=False)
-        assert sp_result is None
-
-    def test_stack_pointer_tracker_offset_block(self):
-        sptracker, sp = init_tracker(track_mem=False)
-        sp_result = sptracker.offset_after_block(0x40071D, sp)
-        assert sp_result is not None
-        sp_result = sptracker.offset_after_block(0x400700, sp)
-        assert sp_result is None
-        sp_result = sptracker.offset_before_block(0x40071D, sp)
-        assert sp_result is not None
-        sp_result = sptracker.offset_before_block(0x400700, sp)
-        assert sp_result is None
-
-
-if __name__ == "__main__":
-    logging.getLogger("angr.analyses.stack_pointer_tracker").setLevel(logging.INFO)
-    unittest.main()

tests/analyses/test_static_hooker.py

@@ -1,28 +0,0 @@
-#!/usr/bin/env python3
-# pylint: disable=missing-class-docstring,no-self-use,line-too-long
-__package__ = __package__ or "tests.analyses"  # pylint:disable=redefined-builtin
-
-import os
-import unittest
-
-import angr
-
-from ..common import bin_location
-
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestStaticHooker(unittest.TestCase):
-    def test_static_hooker(self):
-        test_file = os.path.join(test_location, "x86_64", "static")
-        p = angr.Project(test_file, auto_load_libs=False)
-        sh = p.analyses.StaticHooker("libc.so.6")
-
-        assert 4197616 in sh.results
-        assert type(sh.results[4197616]) is angr.SIM_PROCEDURES["glibc"]["__libc_start_main"]
-        assert type(p.hooked_by(4197616)) is angr.SIM_PROCEDURES["glibc"]["__libc_start_main"]
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/analyses/test_typehoon.py

@@ -1,55 +0,0 @@
-#!/usr/bin/env python3
-# pylint:disable=missing-class-docstring,no-self-use
-__package__ = __package__ or "tests.analyses"  # pylint:disable=redefined-builtin
-
-import os
-import unittest
-
-import angr
-
-from ..common import bin_location
-
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestTypehoon(unittest.TestCase):
-    def test_smoketest(self):
-        p = angr.Project(os.path.join(test_location, "x86_64", "linked_list"), auto_load_libs=False)
-        cfg = p.analyses.CFG(data_references=True, normalize=True)
-
-        main_func = cfg.kb.functions["sum"]
-
-        vr = p.analyses.VariableRecoveryFast(main_func)
-        p.analyses.CompleteCallingConventions()
-
-        # import pprint
-        tcons = vr.type_constraints
-        # pprint.pprint(vr._outstates[0x4005b2].typevars._typevars)
-        # pprint.pprint(tcons)
-
-        _ = p.analyses.Typehoon(tcons)
-        # pprint.pprint(t.simtypes_solution)
-
-        # convert function blocks to AIL blocks
-        # clinic = p.analyses.Clinic(main_func)
-
-        # t = p.analyses.Typehoon(main_func) #, clinic)
-        # print(t)
-
-    def test_type_inference_byte_pointer_cast(self):
-        proj = angr.Project(os.path.join(test_location, "i386", "type_inference_1"), auto_load_libs=False)
-        cfg = proj.analyses.CFG(data_references=True, normalize=True)
-        main_func = cfg.kb.functions["main"]
-        proj.analyses.VariableRecoveryFast(main_func)
-        proj.analyses.CompleteCallingConventions()
-
-        dec = proj.analyses.Decompiler(main_func)
-        assert "->field_0 = 10;" in dec.codegen.text
-        assert "->field_4 = 20;" in dec.codegen.text
-        assert "->field_8 = 808464432;" in dec.codegen.text
-        assert "->field_c = 0;" in dec.codegen.text
-
-
-if __name__ == "__main__":
-    unittest.main()