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

tests/engines/test_actions.py

@@ -1,27 +0,0 @@
-#!/usr/bin/env python3
-import unittest
-
-import angr
-from angr import SimState, SIM_PROCEDURES
-from angr.engines import ProcedureEngine
-
-FAKE_ADDR = 0x100000
-
-
-# pylint: disable=missing-class-docstring
-# pylint: disable=no-self-use
-class TestActions(unittest.TestCase):
-    def test_procedure_actions(self):
-        s = SimState(arch="AMD64", add_options={angr.options.ADD_AUTO_REFS})
-
-        s.registers.store("rbx", 2)
-        proc = SIM_PROCEDURES["testing"]["retreg"](reg="rbx")
-        succ = ProcedureEngine(None).process(s, procedure=proc)
-        rbx = succ.artifacts["procedure"].ret_expr
-        assert type(rbx) is angr.state_plugins.SimActionObject
-        assert s.solver.eval(rbx) == 2
-        assert rbx.reg_deps == {s.arch.registers["rbx"][0]}
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/engines/test_hook.py

@@ -1,112 +0,0 @@
-#!/usr/bin/env python3
-# pylint: disable=missing-class-docstring,disable=no-self-use
-__package__ = __package__ or "tests.engines"  # pylint:disable=redefined-builtin
-
-import os
-import unittest
-
-import angr
-
-from ..common import bin_location
-
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestHook(unittest.TestCase):
-    def test_mips(self):
-        MAIN_END = 0x4007D8
-        INNER_LOOP = 0x40069C
-        OUTER_LOOP = 0x40076C
-
-        p = angr.Project(os.path.join(test_location, "mips", "test_loops"), auto_load_libs=False)
-        output = []
-
-        # hooking by a function decorator
-        @p.hook(INNER_LOOP)
-        def hook1(_):  # pylint:disable=unused-variable
-            output.append(1)
-
-        def hook2(state):
-            output.append(2)
-            num = state.solver.eval(state.regs.a1)
-            string = b"%d " % num
-            state.posix.get_fd(1).write_data(state.solver.BVV(string))
-
-        # a manual hook
-        p.hook(OUTER_LOOP, hook2, length=0x14)
-
-        s = p.factory.simulation_manager(p.factory.entry_state()).explore(find=[MAIN_END])
-
-        assert len(s.found) == 1
-        assert s.found[0].posix.dumps(1) == b"".join(b"%d " % x for x in range(100)) + b"\n"
-        assert output == [1] * 100 + [2] * 100
-        # print 'Executed %d blocks' % len(s._f.trace)
-
-    def test_zero_length_userhook(self):
-        # If a user hook does not overwrite any instruction (length = 0),
-        # we should not run the hook twice.
-        # jumpkind Ijk_NoHook is used for exactly this purpose.
-
-        class OneTimeHook:
-            def __init__(self):
-                self.ctr = 0
-
-            def one_time_hook(self, _):
-                self.ctr += 1
-                if self.ctr > 1:
-                    raise Exception("OneTimeHook is executed multiple times.")
-
-        # Amd64
-        # 0x0:	mov	qword ptr [1], rax
-        # 0x8:	jmp	0x1b
-        # 0xa:	mov	qword ptr [2], rax
-        # 0x12:	mov	qword ptr [3], rax
-        # 0x1a:	ret
-        # 0x1b:	jmp	0xa
-        shellcode = (
-            b"\x48\x89\x04\x25\x01\x00\x00\x00\xeb\x11\x48\x89\x04\x25\x02\x00\x00\x00\x48"
-            b"\x89\x04\x25\x03\x00\x00\x00\xc3\xeb\xed"
-        )
-        proj = angr.load_shellcode(shellcode, arch="amd64")
-
-        proj.hook(0x8, hook=OneTimeHook().one_time_hook, length=0)
-        s = proj.factory.simgr()
-        s.run()
-
-    def test_nonzero_length_userhook(self):
-        # If a user hook overwrites any instruction (length > 0), we should allow the execution of another hook that
-        # follows this hook immediately.
-
-        class TwoTimesHook:
-            def __init__(self):
-                self.addrs = []
-
-            def hook(self, state):
-                self.addrs.append(state.addr)
-
-            # Amd64
-            # 0x0:	mov	qword ptr [1], rax
-            # 0x8:	jmp	0x1b
-            # 0xa:	mov	qword ptr [2], rax
-            # 0x12:	mov	qword ptr [3], rax
-            # 0x1a:	ret
-            # 0x1b:	jmp	0xa
-
-        shellcode = (
-            b"\x48\x89\x04\x25\x01\x00\x00\x00\xeb\x11\x48\x89\x04\x25\x02\x00\x00\x00\x48"
-            b"\x89\x04\x25\x03\x00\x00\x00\xc3\xeb\xed"
-        )
-        proj = angr.load_shellcode(shellcode, arch="amd64")
-
-        hook = TwoTimesHook()
-        proj.hook(0x8, hook=hook.hook, length=2)
-        proj.hook(0xA, hook=hook.hook, length=7)
-        s = proj.factory.simgr()
-        s.run()
-
-        assert hook.addrs == [0x8, 0xA]
-
-
-if __name__ == "__main__":
-    unittest.main()