mirror of
https://github.com/angr/angr
synced 2026-06-08 13:09:39 +00:00
pre-commit-ci[bot]
563fb5f862
* [pre-commit.ci] pre-commit autoupdate updates: - [github.com/astral-sh/ruff-pre-commit: v0.14.11 → v0.14.13](https://github.com/astral-sh/ruff-pre-commit/compare/v0.14.11...v0.14.13) - [github.com/psf/black-pre-commit-mirror: 25.12.0 → 26.1.0](https://github.com/psf/black-pre-commit-mirror/compare/25.12.0...26.1.0) * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
122 lines
5.1 KiB
Python
122 lines
5.1 KiB
Python
# pylint:disable=no-self-use
|
|
from __future__ import annotations
|
|
|
|
__package__ = __package__ or "tests.gui.decompilation_workflows" # pylint:disable=redefined-builtin
|
|
|
|
import os
|
|
import unittest
|
|
|
|
import angr
|
|
from angr.sim_type import SimTypeInt, TypeRef
|
|
from tests.common import bin_location, print_decompilation_result
|
|
|
|
test_location = os.path.join(bin_location, "tests")
|
|
|
|
|
|
class TestDecompilationWorkflows(unittest.TestCase):
|
|
"""
|
|
Tests for decompilation workflows in angr management (or any other GUI if anyone cares enough to create).
|
|
"""
|
|
|
|
def test_decompiling_a_function_multiple_times(self):
|
|
bin_path = os.path.join(test_location, "x86_64", "decompiler", "struct_access")
|
|
proj = angr.Project(bin_path, auto_load_libs=False)
|
|
|
|
proj.analyses.CFGFast(normalize=True)
|
|
func = proj.kb.functions["main"]
|
|
dec = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec.codegen is not None and dec.codegen.text is not None
|
|
print_decompilation_result(dec)
|
|
|
|
# decompile again, using decompilation cache
|
|
dec_2 = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec_2.codegen is not None and dec_2.codegen.text is not None
|
|
print_decompilation_result(dec_2)
|
|
|
|
assert dec.codegen.text == dec_2.codegen.text, "Decompilation results should be identical on multiple runs."
|
|
|
|
def test_decompiling_function_with_renamed_struct_name(self):
|
|
bin_path = os.path.join(test_location, "x86_64", "decompiler", "struct_access")
|
|
proj = angr.Project(bin_path, auto_load_libs=False)
|
|
|
|
proj.analyses.CFGFast(normalize=True)
|
|
func = proj.kb.functions["main"]
|
|
dec = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec.codegen is not None and dec.codegen.text is not None
|
|
print_decompilation_result(dec)
|
|
|
|
assert dec._variable_kb is not None
|
|
types = dec._variable_kb.variables["main"].types
|
|
# let's rename a struct field
|
|
new_type_name = "my_awesome_type"
|
|
t = types["struct_0"]
|
|
assert isinstance(t, TypeRef)
|
|
assert len(t.type.fields) == 2
|
|
t.type.name = new_type_name
|
|
|
|
# decompile again, using decompilation cache
|
|
dec_2 = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec_2.codegen is not None and dec_2.codegen.text is not None
|
|
print_decompilation_result(dec_2)
|
|
|
|
assert new_type_name in dec_2.codegen.text, "Decompilation results should reflect the renamed struct type."
|
|
|
|
def test_decompiling_function_with_renamed_struct_fields(self):
|
|
bin_path = os.path.join(test_location, "x86_64", "decompiler", "struct_access")
|
|
proj = angr.Project(bin_path, auto_load_libs=False)
|
|
|
|
proj.analyses.CFGFast(normalize=True)
|
|
func = proj.kb.functions["main"]
|
|
dec = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec.codegen is not None and dec.codegen.text is not None
|
|
print_decompilation_result(dec)
|
|
|
|
assert dec._variable_kb is not None
|
|
types = dec._variable_kb.variables["main"].types
|
|
# let's rename a struct field
|
|
t = types["struct_0"]
|
|
assert isinstance(t, TypeRef)
|
|
assert len(t.type.fields) == 2
|
|
new_field_name = "my_new_field_120"
|
|
t.type.fields[new_field_name] = t.type.fields["field_120"]
|
|
del t.type.fields["field_120"]
|
|
|
|
# decompile again, using decompilation cache
|
|
dec_2 = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec_2.codegen is not None and dec_2.codegen.text is not None
|
|
print_decompilation_result(dec_2)
|
|
|
|
assert new_field_name in dec_2.codegen.text, "Decompilation results should reflect the renamed struct field."
|
|
|
|
def test_decompiling_function_with_retyped_struct_field(self):
|
|
bin_path = os.path.join(test_location, "x86_64", "decompiler", "struct_access")
|
|
proj = angr.Project(bin_path, auto_load_libs=False)
|
|
|
|
proj.analyses.CFGFast(normalize=True)
|
|
func = proj.kb.functions["main"]
|
|
dec = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec.codegen is not None and dec.codegen.text is not None
|
|
print_decompilation_result(dec)
|
|
assert "struct struct_1 *field_120;" in dec.codegen.text
|
|
|
|
assert dec._variable_kb is not None
|
|
types = dec._variable_kb.variables["main"].types
|
|
# let's type a struct field
|
|
t = types["struct_0"]
|
|
assert isinstance(t, TypeRef)
|
|
assert len(t.type.fields) == 2
|
|
assert "field_120" in t.type.fields
|
|
t.type.fields["field_120"] = SimTypeInt(signed=True).with_arch(proj.arch)
|
|
|
|
# decompile again, using decompilation cache
|
|
dec_2 = proj.analyses.Decompiler(func, cfg=proj.kb.cfgs["CFGFast"])
|
|
assert dec_2.codegen is not None and dec_2.codegen.text is not None
|
|
print_decompilation_result(dec_2)
|
|
|
|
assert "int field_120;" in dec_2.codegen.text, "Decompilation results should reflect the retyped struct field."
|
|
assert "struct_1" not in dec_2.codegen.text
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|