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

tests/analyses/test_vtable.py

@@ -1,31 +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 TestVtable(unittest.TestCase):
-    def test_vtable_extraction_x86_64(self):
-        p = angr.Project(os.path.join(test_location, "x86_64", "cpp_classes"), auto_load_libs=False)
-        vtables_sizes = {0x403CB0: 24, 0x403CD8: 16, 0x403CF8: 16, 0x403D18: 16}
-        vtable_analysis = p.analyses.VtableFinder()
-        vtables = vtable_analysis.vtables_list
-
-        assert len(vtables) == 4
-
-        for vtable in vtables:
-            assert vtable.vaddr in [0x403CB0, 0x403CD8, 0x403CF8, 0x403D18]
-            assert vtables_sizes[vtable.vaddr] == vtable.size
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/analyses/test_xrefs.py

@@ -1,77 +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 angr.knowledge_plugins.xrefs import XRef, XRefType
-
-from ..common import bin_location
-
-
-test_location = os.path.join(bin_location, "tests")
-
-
-class TestXrefs(unittest.TestCase):
-    def test_lwip_udpecho_bm(self):
-        bin_path = os.path.join(test_location, "armel", "lwip_udpecho_bm.elf")
-        p = angr.Project(bin_path, auto_load_libs=False)
-        cfg = p.analyses.CFG(data_references=True)
-
-        func = cfg.functions[0x23C9]
-        state = p.factory.blank_state()
-        prop = p.analyses.Propagator(func=func, base_state=state)
-        _ = p.analyses.XRefs(func=func, replacements=prop.replacements)
-
-        timenow_cp_xrefs = p.kb.xrefs.get_xrefs_by_dst(0x23D4)  # the constant in the constant pool
-        timenow_xrefs = p.kb.xrefs.get_xrefs_by_dst(0x1FFF36F4)  # the value in .bss
-
-        assert len(timenow_cp_xrefs) == 1
-        assert next(iter(timenow_cp_xrefs)) == XRef(ins_addr=0x23C9, dst=0x23D4, xref_type=XRefType.Read)
-
-        assert len(timenow_xrefs) == 3
-        assert [x for x in timenow_xrefs if x.type == XRefType.Offset][0] == XRef(
-            ins_addr=0x23C9, dst=0x1FFF36F4, xref_type=XRefType.Offset
-        )
-        assert [x for x in timenow_xrefs if x.type == XRefType.Read][0] == XRef(
-            ins_addr=0x23CB, dst=0x1FFF36F4, xref_type=XRefType.Read
-        )
-        assert [x for x in timenow_xrefs if x.type == XRefType.Write][0] == XRef(
-            ins_addr=0x23CF, dst=0x1FFF36F4, xref_type=XRefType.Write
-        )
-
-    def test_lwip_udpecho_bm_the_better_way(self):
-        bin_path = os.path.join(test_location, "armel", "lwip_udpecho_bm.elf")
-        p = angr.Project(bin_path, auto_load_libs=False)
-        p.analyses.CFG(cross_references=True)
-
-        timenow_cp_xrefs = p.kb.xrefs.get_xrefs_by_dst(0x23D4)  # the constant in the constant pool
-        timenow_xrefs = p.kb.xrefs.get_xrefs_by_dst(0x1FFF36F4)  # the value in .bss
-
-        assert len(timenow_cp_xrefs) == 1
-        assert next(iter(timenow_cp_xrefs)) == XRef(ins_addr=0x23C9, dst=0x23D4, xref_type=XRefType.Read)
-        # sys_now (2), time_isr (3) == 5
-        assert len(timenow_xrefs) == 5
-
-    def test_p2im_drone_with_inits(self):
-        bin_path = os.path.join(test_location, "armel", "p2im_drone.elf")
-        proj = angr.Project(bin_path, auto_load_libs=False)
-        cfg = proj.analyses.CFG(data_references=True)
-
-        func = cfg.functions["Peripherals_Init"]
-        state = proj.factory.blank_state()
-        prop = proj.analyses.Propagator(func=func, base_state=state)
-
-        init_finder = proj.analyses.InitializationFinder(func=func, replacements=prop.replacements)
-        overlay_state = init_finder.overlay_state
-
-        cfg.do_full_xrefs(overlay_state=overlay_state)
-
-        h12c1_inst_xrefs = proj.kb.xrefs.get_xrefs_by_dst(0x20001500)
-        assert len(h12c1_inst_xrefs) == 5
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/common.py

@@ -1,128 +0,0 @@
-import os
-import pickle
-import sys
-import logging
-import subprocess
-from functools import lru_cache
-from typing import Optional, Sequence
-from tempfile import NamedTemporaryFile
-
-from unittest import skipIf, skipUnless, skip, SkipTest
-
-l = logging.getLogger("angr.tests.common")
-
-try:
-    import tracer
-except ImportError:
-    tracer = None
-
-bin_location = os.path.join(os.path.dirname(os.path.realpath(__file__)), "..", "..", "binaries")
-bin_priv_location = os.path.join(os.path.dirname(os.path.realpath(__file__)), "..", "..", "binaries-private")
-
-if not os.path.isdir(bin_location):
-    raise Exception(
-        "Can't find the angr/binaries repo for holding testcases. "
-        "It should be cloned into the same folder as the rest of your angr modules."
-    )
-
-
-def broken(func):
-    return skip(reason="Broken test method")(func)
-
-
-def requires_binaries_private(func):
-    return skipIf(
-        not os.path.exists(
-            os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "binaries-private"),
-        ),
-        "Skip this test since we do not have the binaries-private repo cloned on Travis CI.",
-    )(func)
-
-
-def slow_test(func):
-    func.speed = "slow"
-    slow_test_env = os.environ["SKIP_SLOW_TESTS"].lower() if "SKIP_SLOW_TESTS" in os.environ else ""
-    return skipIf(slow_test_env == "true" or slow_test_env == "1", "Skipping slow test")(func)
-
-
-def skip_if_not_linux(func):
-    return skipUnless(sys.platform.startswith("linux"), "Skipping Linux Test Cases")(func)
-
-
-TRACE_VERSION = 1
-
-
-def do_trace(proj, test_name, input_data, **kwargs):
-    """
-    trace, magic, crash_mode, crash_addr = load_cached_trace(proj, "test_blurble")
-    """
-    fname = os.path.join(
-        bin_location,
-        "tests_data",
-        "runner_traces",
-        f"{test_name}_{os.path.basename(proj.filename)}_{proj.arch.name}.p",
-    )
-
-    if os.path.isfile(fname):
-        try:
-            with open(fname, "rb") as f:
-                r = pickle.load(f)
-                if type(r) is tuple and len(r) == 2 and r[1] == TRACE_VERSION:
-                    return r[0]
-        except (pickle.UnpicklingError, UnicodeDecodeError):
-            print("Can't unpickle trace - rerunning")
-
-    if tracer is None:
-        raise SkipTest("Tracer is not installed and cached data is not present")
-
-    runner = tracer.QEMURunner(project=proj, input=input_data, **kwargs)
-    r = (runner.trace, runner.magic, runner.crash_mode, runner.crash_addr)
-    with open(fname, "wb") as f:
-        pickle.dump((r, TRACE_VERSION), f, -1)
-    return r
-
-
-@skipUnless(tracer, "tracer is not installed")
-def load_cgc_pov(pov_file: str) -> "tracer.TracerPoV":
-    return tracer.TracerPoV(pov_file)
-
-
-def compile_c(c_code: str, cflags: Optional[Sequence[str]], silent: bool = False) -> NamedTemporaryFile:
-    # pylint:disable=consider-using-with
-    """
-    Compile `c_code` and return the file containing the compiled output
-    """
-    dst = None
-    try:
-        dst = NamedTemporaryFile(delete=False)
-        dst.close()
-        src = NamedTemporaryFile(mode="x", delete=False, suffix=".c")
-        src.write(c_code)
-        src.close()
-
-        call_args = ["cc"] + (cflags or []) + ["-o", dst.name, src.name]
-        l.debug("Compiling with: %s", " ".join(call_args))
-        l.debug("Source:\n%s", c_code)
-        out = subprocess.DEVNULL if silent else None
-        subprocess.check_call(call_args, stderr=out, stdout=out)
-        return dst
-    except:
-        if dst and os.path.exists(dst.name):
-            os.remove(dst.name)
-        raise
-    finally:
-        if src and os.path.exists(src.name):
-            os.remove(src.name)
-
-
-@lru_cache
-def has_32_bit_compiler_support() -> bool:
-    """
-    Check if we are able to compile a 32-bit binary
-    """
-    try:
-        binary = compile_c("#include <stdlib.h>\nint main() { return 0; }", ["-m32"], True)
-        os.remove(binary.name)
-        return True
-    except subprocess.CalledProcessError:
-        return False

tests/engines/light/test_data.py

@@ -1,17 +0,0 @@
-#!/usr/bin/env python3
-from unittest import main, TestCase
-
-from angr.engines.light.data import SpOffset
-
-
-class TestSpOffset(TestCase):
-    def test_difference_between_two_sp_offset_is_concrete(self):
-        size = 8
-        first_offset = SpOffset(size, 10)
-        second_offset = SpOffset(size, 20)
-
-        self.assertEqual(first_offset - second_offset, -10)
-
-
-if __name__ == "__main__":
-    main()