mirror of
https://github.com/angr/angr
synced 2026-06-08 13:09:39 +00:00
Fish
78799acd95
* Preliminary implementation of a spilling CFG graph. * Fix some test cases. * Fix another test case. * Fixes. * Lint code. * Implement InEdgeView and OutEdgeView. * Implement InDegreeView and OutDegreeView. * Fixes. * Use (Node.addr, Node.size) as key. * Fix test cases. * More fixes. * Fix CFGENode support. * Fix the remaining issues. * Get rid of SpillingCFGGraph.reverse(). * Some refactoring. * Oops * Fix perf issue. * Fix set size change error during key iteration. * Fix db_batch_size assignment. Fix some test cases. * SpillingCFGNodeDict.__setstate__: Initialize self._all_keys. * Getting rid of the nodes dictionary. * Fix logic in cfg_emulated.py * Multiple fixes. * Type annotation fix. * Fix missing edges after merging CFGNodes. * Fix more logic. * Add Soot CFGNodes to the CFG model. * Lint and type check. * Add USE_SPILLING_CFGNODE_DICT and CFGNode.dirty. * Oops * Lint code. * No more pickling of CFGNodes. * Fix no_ret being None. * Lint code. * Some refactor; Destroy the fallback mechanism. * Fix node dirty bug after deserialization.
142 lines
3.8 KiB
Python
Executable File
142 lines
3.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# pylint: disable=missing-class-docstring,no-self-use,line-too-long
|
|
from __future__ import annotations
|
|
|
|
__package__ = __package__ or "tests.serialization" # pylint:disable=redefined-builtin
|
|
|
|
import gc
|
|
import os
|
|
import pickle
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
import angr
|
|
from angr.sim_variable import SimStackVariable
|
|
|
|
from tests.common import bin_location
|
|
|
|
test_location = os.path.join(bin_location, "tests")
|
|
|
|
internaltest_files = [
|
|
"argc_decide",
|
|
"argc_symbol",
|
|
"argv_test",
|
|
"counter",
|
|
"fauxware",
|
|
"fauxware.idb",
|
|
"manysum",
|
|
"pw",
|
|
"strlen",
|
|
"test_arrays",
|
|
"test_division",
|
|
"test_loops",
|
|
]
|
|
internaltest_arch = ["i386", "armel"]
|
|
|
|
|
|
def internaltest_vfg(p, cfg):
|
|
vfg = p.analyses.VFG(cfg=cfg)
|
|
v = angr.vaults.VaultDict()
|
|
state = v.dumps(vfg)
|
|
vfg2 = v.loads(state)
|
|
assert vfg.final_states == vfg2.final_states
|
|
assert set(vfg.graph.nodes()) == set(vfg2.graph.nodes())
|
|
|
|
|
|
def internaltest_cfg(p):
|
|
with tempfile.TemporaryFile() as state:
|
|
cfg = p.analyses.CFGEmulated()
|
|
pickle.dump(cfg.model, state, -1)
|
|
|
|
state.seek(0)
|
|
cfg2_model = pickle.load(state)
|
|
assert set(cfg.model.nodes()) == set(cfg2_model.nodes())
|
|
|
|
return cfg
|
|
|
|
|
|
def internaltest_cfgfast(p):
|
|
with tempfile.TemporaryFile() as state:
|
|
cfg = p.analyses.CFGFast()
|
|
|
|
# generate capstone blocks
|
|
main_function = cfg.functions.function(name="main")
|
|
for b in main_function.blocks:
|
|
_ = b.capstone
|
|
|
|
pickle.dump(cfg.model, state, -1)
|
|
|
|
state.seek(0)
|
|
cfg2_model = pickle.load(state)
|
|
assert set(cfg.model.nodes()) == set(cfg2_model.nodes())
|
|
|
|
|
|
def internaltest_project(fpath):
|
|
tpath = tempfile.mktemp()
|
|
shutil.copy(fpath, tpath)
|
|
|
|
p = angr.Project(tpath, auto_load_libs=False)
|
|
state = pickle.dumps(p, -1)
|
|
loaded_p = pickle.loads(state)
|
|
assert p is not loaded_p
|
|
assert p.arch == loaded_p.arch
|
|
assert p.filename == loaded_p.filename
|
|
assert p.entry == loaded_p.entry
|
|
|
|
simgr = loaded_p.factory.simulation_manager()
|
|
simgr.run(n=10)
|
|
assert len(simgr.errored) == 0
|
|
|
|
|
|
class TestSerialization(unittest.TestCase):
|
|
def test_analyses(self):
|
|
p = angr.Project(os.path.join(test_location, "i386", "fauxware"), load_options={"auto_load_libs": False})
|
|
cfg = p.analyses.CFG()
|
|
cfb = p.analyses.CFB(kb=cfg.kb)
|
|
vrf = p.analyses.VariableRecoveryFast(p.kb.functions["main"])
|
|
|
|
assert len(p.kb.functions) > 0
|
|
assert len(pickle.loads(pickle.dumps(p.kb, -1)).functions) > 0
|
|
|
|
state = pickle.dumps((p, cfg, cfb, vrf), -1)
|
|
del p
|
|
del cfg
|
|
del cfb
|
|
del vrf
|
|
|
|
gc.collect()
|
|
|
|
p, cfg, _cfb, _vrf = pickle.loads(state)
|
|
assert p.kb is not None
|
|
assert p.kb.functions is not None
|
|
assert cfg.kb is not None
|
|
assert len(p.kb.functions) > 0
|
|
|
|
def test_serialization(self):
|
|
for d in internaltest_arch:
|
|
for f in internaltest_files:
|
|
fpath = os.path.join(test_location, d, f)
|
|
if os.path.isfile(fpath) and os.access(fpath, os.X_OK):
|
|
internaltest_project(fpath)
|
|
|
|
p = angr.Project(os.path.join(test_location, "i386", "fauxware"), load_options={"auto_load_libs": False})
|
|
internaltest_cfgfast(p)
|
|
|
|
cfg = internaltest_cfg(p)
|
|
internaltest_vfg(p, cfg)
|
|
|
|
def test_simstackvariable_offest_too_large(self):
|
|
v0 = SimStackVariable(-0x8000_0001, 4, ident="s_0")
|
|
cmsg = v0.serialize_to_cmessage()
|
|
assert cmsg.offset == -0x7FFF_DEAD
|
|
|
|
v1 = SimStackVariable(0, 4, ident="s_1")
|
|
v1.offset = 0x8000_0000 # we gotta force it
|
|
cmsg = v1.serialize_to_cmessage()
|
|
assert cmsg.offset == 0x7FFF_DEAD
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|