mirror of
https://github.com/angr/angr
synced 2026-08-17 12:23:11 -04:00
d38cc5a019
* [pre-commit.ci] pre-commit autoupdate updates: - [github.com/astral-sh/ruff-pre-commit: v0.15.22 → v0.16.0](https://github.com/astral-sh/ruff-pre-commit/compare/v0.15.22...v0.16.0) * Apply fixes * Add values() --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Kevin Phoenix <kevin@kphoenix.us>
80 lines
3 KiB
Python
Executable file
80 lines
3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
from __future__ import annotations
|
|
|
|
__package__ = __package__ or "tests.analyses.decompiler" # pylint:disable=redefined-builtin
|
|
|
|
import os.path
|
|
import unittest
|
|
|
|
import angr
|
|
from angr import ailment
|
|
from angr.analyses.decompiler.optimization_passes.base_ptr_save_simplifier import (
|
|
BasePointerSaveSimplifier,
|
|
)
|
|
from tests.common import bin_location
|
|
|
|
test_location = os.path.join(bin_location, "tests")
|
|
|
|
|
|
def _get_block(clinic, addr):
|
|
for block in clinic.graph.nodes():
|
|
if block.addr == addr:
|
|
return block
|
|
return None
|
|
|
|
|
|
def check_bp_save_fauxware(arch):
|
|
p = angr.Project(os.path.join(test_location, arch, "fauxware"), auto_load_libs=False)
|
|
cfg = p.analyses.CFGFast(normalize=True)
|
|
main = p.kb.functions["main"]
|
|
optimization_passes = [BasePointerSaveSimplifier]
|
|
dra = p.analyses.Decompiler(main, cfg=cfg, optimization_passes=optimization_passes)
|
|
first_block_stmts = dra.codegen._sequence.nodes[0].nodes[0].statements
|
|
for stmt in first_block_stmts:
|
|
if isinstance(stmt, ailment.Stmt.Store):
|
|
assert not (isinstance(stmt.data, ailment.Expr.Register) and stmt.data.reg_offset == p.arch.bp_offset) or (
|
|
isinstance(stmt.data, ailment.Expr.StackBaseOffset) and stmt.data.offset == 0
|
|
)
|
|
|
|
|
|
# pylint: disable=missing-class-docstring
|
|
# pylint: disable=no-self-use
|
|
class TestBaseptrSaveSimplifier(unittest.TestCase):
|
|
def test_baseptr_save_simplifier_amd64(self):
|
|
# decompile all:main and make sure the first and the last blocks do not save or restore to rbp
|
|
bin_path = os.path.join(test_location, "x86_64", "all")
|
|
proj = angr.Project(bin_path, auto_load_libs=False, load_debug_info=True)
|
|
|
|
cfg = proj.analyses.CFG(data_references=True, normalize=True)
|
|
|
|
optimization_passes = [BasePointerSaveSimplifier]
|
|
main_func = cfg.functions["main"]
|
|
dec = proj.analyses.Decompiler(main_func, cfg=cfg, optimization_passes=optimization_passes)
|
|
|
|
entry_block = _get_block(dec.clinic, main_func.addr)
|
|
endpoint_block = _get_block(dec.clinic, next(iter(main_func.endpoints)).addr)
|
|
|
|
assert entry_block is not None
|
|
assert endpoint_block is not None
|
|
|
|
for stmt in entry_block.statements:
|
|
if isinstance(stmt, ailment.Stmt.Store) and isinstance(stmt.data, ailment.Expr.StackBaseOffset):
|
|
assert False, "Found a base-pointer saving statement in the first block."
|
|
|
|
for stmt in endpoint_block.statements:
|
|
if (
|
|
isinstance(stmt, ailment.Stmt.Assignment)
|
|
and isinstance(stmt.dst, ailment.Expr.Register)
|
|
and stmt.dst.reg_offset == proj.arch.bp_offset
|
|
):
|
|
assert False, "Found a base-pointer restoring statement in the last block."
|
|
|
|
def test_bp_save_amd64_fauxware(self):
|
|
check_bp_save_fauxware("x86_64")
|
|
|
|
def test_bp_save_armel_fauxware(self):
|
|
check_bp_save_fauxware("armel")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|