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

tests/procedures/libc/test_scanf.py

@@ -1,205 +0,0 @@
-#!/usr/bin/env python3
-__package__ = __package__ or "tests.procedures.libc"  # pylint:disable=redefined-builtin
-
-import os
-import string
-
-import unittest
-
-import angr
-import claripy
-
-from ...common import bin_location
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class Checker:
-    def __init__(
-        self, check_func, length=None, base=10, dummy: bool = False, multi: bool = False, delimiter: str = None
-    ):
-        self._check_func = check_func
-        self._length = length
-        self._base = base
-        self._dummy = dummy
-        self._multi = multi
-        self._delimiter = delimiter
-
-        if multi:
-            if not delimiter:
-                raise ValueError("Delimiter is required when multi is True.")
-            if not isinstance(check_func, list):
-                raise TypeError("You must provide a list of check functions when multi is True.")
-            self._parts = len(check_func)
-
-    def _extract_integer(self, s):
-        charset = string.digits if self._base == 10 else string.digits + "abcdefABCDEF"
-
-        component = ""
-
-        digit_start_pos = None
-
-        for i, c in enumerate(s):
-            if digit_start_pos is not None:
-                if c not in charset:
-                    component = s[:i]
-                    break
-            else:
-                if c in charset and s[i : i + 2] not in ("0x", "0X"):
-                    digit_start_pos = c
-
-        if not component:
-            component = s
-
-        return component
-
-    def check(self, path):
-        if self._dummy:
-            return True
-
-        if not isinstance(path.posix.stdin, angr.storage.file.SimPacketsStream):
-            raise TypeError("This test case only supports SimPacketsStream-type of stdin.")
-
-        if not self._multi:
-            stdin_input = path.posix.stdin.content[1][0]  # skip the first char used in switch
-        else:
-            stdin_input = claripy.Concat(*[part[0] for part in path.posix.stdin.content[1:]])
-        some_strings = path.solver.eval_upto(stdin_input, 1000, cast_to=bytes)
-        some_strings = [x.decode() for x in some_strings]
-
-        check_passes = False
-
-        for s in some_strings:
-            if self._length is not None:
-                s = s[: self._length]
-
-            if not self._multi:
-                # single part
-                component = self._extract_integer(s)
-                if self._check_func(component):
-                    check_passes = True
-                    break
-            else:
-                # multiple parts
-                substrs = s.split(self._delimiter)
-                if len(substrs) != len(self._check_func):
-                    continue
-
-                components = [self._extract_integer(substr) for substr in substrs]
-
-                if all(func(component) for func, component in zip(self._check_func, components)):
-                    check_passes = True
-                    break
-
-        return check_passes
-
-
-class TestScanf(unittest.TestCase):
-    def test_scanf(self):
-        test_bin = os.path.join(test_location, "x86_64", "scanf_test")
-        b = angr.Project(test_bin, auto_load_libs=False)
-
-        pg = b.factory.simulation_manager()
-
-        # find the end of main
-        expected_outputs = {
-            b"%%07x\n": Checker(lambda s: int(s, 16) == 0xAAAA, length=7, base=16),
-            b"%%07x and negative numbers\n": Checker(lambda s: int(s, 16) == -0xCDCD, length=7, base=16),
-            b"nope 0\n": Checker(None, dummy=True),
-            b"%%d\n": Checker(lambda s: int(s) == 133337),
-            b"%%d and negative numbers\n": Checker(lambda s: int(s) == 2**32 - 1337),
-            b"nope 1\n": Checker(None, dummy=True),
-            b"%%u\n": Checker(lambda s: int(s) == 0xAAAA),
-            b"%%u and negative numbers\n": Checker(lambda s: int(s) == 2**32 - 0xCDCD),
-            b"nope 2\n": Checker(None, dummy=True),
-            b"Unsupported switch\n": Checker(None, dummy=True),
-        }
-        pg.explore(find=0x4007F3, num_find=len(expected_outputs))
-
-        # check the outputs
-        total_outputs = 0
-        for path in pg.found:
-            test_output = path.posix.dumps(1)
-            if test_output in expected_outputs:
-                assert expected_outputs[test_output].check(path), "Test case failed. Output is %s." % test_output
-
-            total_outputs += 1
-
-        # check that all of the outputs were seen
-        assert total_outputs == len(expected_outputs)
-
-    def test_scanf_multi(self):
-        test_bin = os.path.join(test_location, "x86_64", "scanf_multi_test")
-        b = angr.Project(test_bin, auto_load_libs=False)
-
-        pg = b.factory.simulation_manager()
-
-        expected_outputs = {
-            b"%%04x.%%04x.%%04x\n": Checker(
-                [
-                    lambda x: int(x, 16) == 0xAAAA,
-                    lambda x: int(x, 16) == 0xBBBB,
-                    lambda x: int(x, 16) == 0xCCCC,
-                ],
-                base=16,
-                multi=True,
-                delimiter=".",
-            ),
-            b"%%04x.%%04x.%%04x and negative numbers\n": Checker(
-                [lambda x: int(x, 16) == -0xCD] * 3,
-                base=16,
-                multi=True,
-                delimiter=".",
-            ),
-            b"%%d.%%d.%%d\n": Checker(
-                [lambda x: int(x, 10) == 133337, lambda x: int(x, 10) == 1337, lambda x: int(x, 10) == 13337],
-                base=10,
-                multi=True,
-                delimiter=".",
-            ),
-            b"%%d.%%d.%%d and negative numbers\n": Checker(
-                [lambda x: int(x, 10) == 2**32 - 1337] * 3,
-                base=10,
-                multi=True,
-                delimiter=".",
-            ),
-            b"%%u\n": Checker(
-                [lambda x: int(x) == 0xAAAA, lambda x: int(x) == 0xBBBB, lambda x: int(x) == 0xCCCC],
-                base=10,
-                multi=True,
-                delimiter=".",
-            ),
-            b"%%u and negative numbers\n": Checker(
-                [lambda s: int(s) == 2**32 - 0xCDCD] * 3,
-                base=10,
-                multi=True,
-                delimiter=".",
-            ),
-            b"Unsupported switch\n": Checker(None, dummy=True),
-        }
-        pg.explore(
-            find=0x40083E,
-            avoid=(
-                0x4006DB,
-                0x400776,
-                0x40080B,
-            ),  # avoid all "nope N" branches
-            num_find=len(expected_outputs),
-        )
-
-        # check the outputs
-        total_outputs = 0
-        for path in pg.found:
-            path.posix.dumps(0)
-            test_output = path.posix.dumps(1)
-            if test_output in expected_outputs:
-                assert expected_outputs[test_output].check(path), "Test case failed. Output is %s." % test_output
-
-            total_outputs += 1
-
-        # check that all of the outputs were seen
-        assert total_outputs == len(expected_outputs)
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/procedures/libc/test_sprintf.py

@@ -1,44 +0,0 @@
-#!/usr/bin/env python3
-# pylint: disable=missing-class-docstring,no-self-use,line-too-long
-__package__ = __package__ or "tests.procedures.libc"  # pylint:disable=redefined-builtin
-
-import os
-import unittest
-
-import angr
-
-from ...common import bin_location
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestSprintf(unittest.TestCase):
-    def test_sprintf(self):
-        p = angr.Project(
-            os.path.join(test_location, "x86_64", "sprintf_test"),
-            auto_load_libs=False,
-        )
-        a = p.factory.simulation_manager().explore(find=0x4005C0)
-        state = a.found[0]
-
-        str1 = state.solver.eval(state.memory.load(0x600AD0, 13), cast_to=bytes)
-        assert str1 == b"Immediate: 3\n"
-
-        str2 = state.solver.eval(state.memory.load(0x600A70, 7), cast_to=bytes)
-        assert str2 == b"Int: 3\n"
-
-        str3 = state.solver.eval(state.memory.load(0x600AB0, 8), cast_to=bytes)
-        assert str3 == b"Char: c\n"
-
-        str4 = state.solver.eval(state.memory.load(0x600A50, 14), cast_to=bytes)
-        assert str4 == b"Uninit int: 0\n"
-
-        str5 = state.solver.eval(state.memory.load(0x600A90, 24), cast_to=bytes)
-        assert str5 == b"Str on stack: A string.\n"
-
-        str6 = state.solver.eval(state.memory.load(0x600A30, 21), cast_to=bytes)
-        assert str6 == b"Global str: GLOB_STR\n"
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/procedures/libc/test_sscanf.py

@@ -1,63 +0,0 @@
-#!/usr/bin/env python3
-__package__ = __package__ or "tests.procedures.libc"  # pylint:disable=redefined-builtin
-
-import os
-import subprocess
-import sys
-import unittest
-
-import angr
-
-from ...common import bin_location
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestSscanf(unittest.TestCase):
-    @unittest.skipUnless(sys.platform.startswith("linux"), "linux only")
-    def test_sscanf(self):
-        test_bin = os.path.join(test_location, "x86_64", "sscanf_test")
-        b = angr.Project(test_bin, auto_load_libs=False)
-        pg = b.factory.simulation_manager()
-        # find the end of main
-        expected_outputs = {
-            b"0x worked\n",
-            b"+0x worked\n",
-            b"base +16 worked\n",
-            b"base 16 worked\n",
-            b"-0x worked\n",
-            b"base -16 worked\n",
-            b"base 16 length 2 worked\n",
-            b"Nope x\n",
-            b"base 8 worked\n",
-            b"base +8 worked\n",
-            b"base +10 worked\n",
-            b"base 10 worked\n",
-            b"base -8 worked\n",
-            b"base -10 worked\n",
-            b"Nope u\n",
-            b"No switch\n",
-        }
-        pg.run()
-        assert len(pg.deadended) == len(expected_outputs)
-        assert len(pg.active) == 0
-        assert len(pg.errored) == 0
-
-        # check the outputs
-        pipe = subprocess.PIPE
-        for f in pg.deadended:
-            test_input = f.posix.dumps(0)
-            test_output = f.posix.dumps(1)
-            expected_outputs.remove(test_output)
-
-            # check the output works as expected
-            p = subprocess.Popen(test_bin, stdout=pipe, stderr=pipe, stdin=pipe)
-            ret = p.communicate(test_input)[0]
-            assert ret == test_output
-
-        # check that all of the outputs were seen
-        assert len(expected_outputs) == 0
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/procedures/libc/test_strcasecmp.py

@@ -1,37 +0,0 @@
-#!/usr/bin/env python3
-# pylint: disable=missing-class-docstring,no-self-use,line-too-long
-__package__ = __package__ or "tests.procedures.libc"  # pylint:disable=redefined-builtin
-
-import os
-import unittest
-
-import claripy
-
-import angr
-
-from ...common import bin_location
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestStrCaseCmp(unittest.TestCase):
-    def test_i386(self):
-        p = angr.Project(os.path.join(test_location, "i386", "test_strcasecmp"), auto_load_libs=False)
-        arg1 = claripy.BVS("arg1", 20 * 8)
-        s = p.factory.entry_state(args=("test_strcasecmp", arg1))
-        sm = p.factory.simulation_manager(s)
-        sm.explore()
-
-        sm.move("deadended", "found", filter_func=lambda s: b"Welcome" in s.posix.dumps(1))
-
-        assert len(sm.found) == 1
-
-        f = sm.found[0]
-        sol = f.solver.eval(arg1, cast_to=bytes)
-        assert b"\x00" in sol
-        assert sol[: sol.index(b"\x00")].lower() == b"letmein"
-        assert b"wchar works" in f.posix.dumps(1)
-
-
-if __name__ == "__main__":
-    unittest.main()