70 lines
3.0 KiB
Python
70 lines
3.0 KiB
Python
import json
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from h8536.model import Instruction
|
|
from h8536.pseudocode import PseudocodeOptions, generate_pseudocode
|
|
from h8536.render import format_listing, write_json
|
|
from h8536.rom import Rom
|
|
from h8536.serial_reconstruction import analyze_serial_reconstruction
|
|
|
|
|
|
def ins(address: int, mnemonic: str, operands: str = "", references: list[int] | None = None) -> Instruction:
|
|
return Instruction(address, b"", mnemonic, operands, references=references or [])
|
|
|
|
|
|
class SerialReconstructionIntegrationTest(unittest.TestCase):
|
|
def test_listing_json_and_pseudocode_include_candidate_frame_reconstruction(self):
|
|
instructions = {
|
|
0x3000: ins(0x3000, "MOV:G.B", "@H'F858, R0", [0xF858]),
|
|
0x3004: ins(0x3004, "MOV:G.B", "R0, @SCI1_TDR", [0xFEDB]),
|
|
0x3008: ins(0x3008, "MOV:G.B", "#1, @H'F9C2", [0xF9C2]),
|
|
0x3010: ins(0x3010, "MOV:E.B", "#H'5A, R0", []),
|
|
0x3014: ins(0x3014, "XOR.B", "@H'F858, R0", [0xF858]),
|
|
0x3018: ins(0x3018, "XOR.B", "@H'F859, R0", [0xF859]),
|
|
0x301C: ins(0x301C, "XOR.B", "@H'F85A, R0", [0xF85A]),
|
|
0x3020: ins(0x3020, "XOR.B", "@H'F85B, R0", [0xF85B]),
|
|
0x3024: ins(0x3024, "XOR.B", "@H'F85C, R0", [0xF85C]),
|
|
0x3028: ins(0x3028, "MOV:G.B", "R0, @H'F85D", [0xF85D]),
|
|
0x3030: ins(0x3030, "MOV:G.B", "@H'F9C2, R1", [0xF9C2]),
|
|
0x3034: ins(0x3034, "MOV:G.B", "@(-H'07A8,R1), R0", []),
|
|
0x3038: ins(0x3038, "MOV:G.B", "R0, @SCI1_TDR", [0xFEDB]),
|
|
0x303C: ins(0x303C, "ADD:Q.B", "#1, @H'F9C2", [0xF9C2]),
|
|
0x3040: ins(0x3040, "CMP:G.B", "#6, @H'F9C2", [0xF9C2]),
|
|
}
|
|
analysis = analyze_serial_reconstruction(instructions)
|
|
|
|
listing = format_listing(
|
|
Path("rom.bin"),
|
|
Rom(b"\xFF" * 0x20),
|
|
instructions,
|
|
{},
|
|
{},
|
|
"min",
|
|
traced=False,
|
|
serial_reconstruction=analysis,
|
|
)
|
|
self.assertIn("; Serial Protocol Reconstruction", listing)
|
|
self.assertIn("TX candidate: 6 bytes", listing)
|
|
self.assertIn("candidate/evidence-supported SCI1 6-byte TX frame", listing)
|
|
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
path = Path(tmp) / "out.json"
|
|
write_json(path, instructions, {}, {}, serial_reconstruction=analysis)
|
|
payload = json.loads(path.read_text(encoding="utf-8"))
|
|
|
|
self.assertEqual(payload["serial_reconstruction"]["candidates"][0]["frame_length"], 6)
|
|
tdr_instruction = next(item for item in payload["instructions"] if item["address"] == 0x3038)
|
|
self.assertIn("serial_reconstruction", tdr_instruction)
|
|
|
|
pseudocode = generate_pseudocode(
|
|
payload,
|
|
options=PseudocodeOptions(include_addresses=False, include_asm=False, structured=False),
|
|
)
|
|
self.assertIn("candidate/evidence-supported SCI1 6-byte TX frame", pseudocode)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|