73 lines
2.8 KiB
Python
73 lines
2.8 KiB
Python
import unittest
|
|
|
|
from h8536.indirect import analyze_indirect_flow, indirect_comment_for_instruction
|
|
from h8536.model import Instruction
|
|
from h8536.render import format_listing, write_json
|
|
from h8536.rom import Rom
|
|
import json
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
|
|
class IndirectFlowTest(unittest.TestCase):
|
|
def test_detects_indexed_pointer_table_before_indirect_jump(self):
|
|
data = bytearray([0xFF] * 0x240)
|
|
data[0x0200:0x0206] = bytes.fromhex("01200300FFFF")
|
|
instructions = {
|
|
0x0100: Instruction(0x0100, b"", "MOV:G.W", "@(H'0200,R4), R1"),
|
|
0x0104: Instruction(0x0104, b"", "JMP", "@R1", kind="jump", fallthrough=False),
|
|
0x0120: Instruction(0x0120, b"\x19", "RTS", kind="return", fallthrough=False),
|
|
0x0300: Instruction(0x0300, b"\x19", "RTS", kind="return", fallthrough=False),
|
|
}
|
|
|
|
analysis = analyze_indirect_flow(Rom(bytes(data)), instructions, {0x0120: "loc_0120"})
|
|
site = analysis["sites"][0]
|
|
|
|
self.assertEqual(site["address"], 0x0104)
|
|
self.assertEqual(site["target_register"], "R1")
|
|
self.assertEqual(site["table"]["base"], 0x0200)
|
|
self.assertEqual(site["table"]["entry_count"], 2)
|
|
self.assertEqual(site["table"]["decoded_target_count"], 2)
|
|
self.assertIn("pointer table H'0200", indirect_comment_for_instruction(analysis, 0x0104))
|
|
|
|
def test_records_unknown_indirect_call_without_prior_table_load(self):
|
|
instructions = {
|
|
0x0100: Instruction(0x0100, b"", "JSR", "@R0", kind="call"),
|
|
}
|
|
|
|
analysis = analyze_indirect_flow(Rom(bytes([0xFF] * 0x200)), instructions)
|
|
|
|
self.assertEqual(analysis["sites"][0]["confidence"], "unknown")
|
|
self.assertIn("target not resolved", analysis["sites"][0]["summary"])
|
|
|
|
def test_listing_and_json_include_indirect_flow_metadata(self):
|
|
instructions = {
|
|
0x0100: Instruction(0x0100, b"", "JSR", "@R0", kind="call"),
|
|
}
|
|
analysis = analyze_indirect_flow(Rom(bytes([0xFF] * 0x200)), instructions)
|
|
|
|
listing = format_listing(
|
|
Path("rom.bin"),
|
|
Rom(bytes([0xFF] * 0x200)),
|
|
instructions,
|
|
{},
|
|
{},
|
|
"min",
|
|
traced=True,
|
|
indirect_flow=analysis,
|
|
)
|
|
|
|
self.assertIn("target not resolved", listing)
|
|
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
path = Path(tmp) / "out.json"
|
|
write_json(path, instructions, {}, {}, indirect_flow=analysis)
|
|
payload = json.loads(path.read_text(encoding="utf-8"))
|
|
|
|
self.assertEqual(payload["indirect_flow"]["sites"][0]["address"], 0x0100)
|
|
self.assertEqual(payload["instructions"][0]["indirect_flow"]["confidence"], "unknown")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|