class TestFastDisasmMulti(object): def get_disasm(self): z = BasicZ80Disassembler() parent = Basic6502Disassembler() # Use the Z80 processor for style type 64 parent.fast.chunk_type_processor[1] = z.fast.chunk_processor return parent def setup(self): self.editor = MockHexEditor() guess = FileGuess("../test_data/pytest.atr") self.editor.load(guess) self.disasm = self.get_disasm() self.fast = self.disasm.fast def test_ranges(self): self.editor.find_segment("boot code at $0800") s = self.editor.segment r = fast_get_entire_style_ranges(s, user=user_bit_mask, split_comments=[]) print(r) assert r == [((0, 238), 0), ((238, 268), 2), ((268, 332), 0), ((332, 464), 1), ((464, 512), 0)] info_sections = self.fast.get_all(s.rawdata.unindexed_data, s.origin, 0, r)
class TestFastDisasm(object): def get_disasm(self): return Basic6502Disassembler() def setup(self): self.editor = MockHexEditor() guess = FileGuess("../test_data/pytest.atr") self.editor.load(guess) self.disasm = self.get_disasm() self.fast = self.disasm.fast def test_ranges(self): # force the use of the normal disassembler for the other style self.disasm.fast.chunk_type_processor[ 64] = self.disasm.fast.chunk_processor self.editor.find_segment("02: robots I") s = self.editor.segment r = fast_get_entire_style_ranges(s, user=user_bit_mask, split_comments=[]) print(r) assert r == [((0, 497), 0), ((497, 524), 1), ((524, 602), 0), ((602, 690), 1), ((690, 1004), 0), ((1004, 1024), 1), ((1024, 1536), 0), ((1536, 1710), 1), ((1710, 1792), 0), ((1792, 1954), 1), ((1954, 2048), 0)] info_all = self.fast.get_all(s.rawdata.unindexed_data, s.origin, 0) #print info_all.instructions[0:20] info_sections = self.fast.get_all(s.rawdata.unindexed_data, s.origin, 0, r) # for i in range(info_sections.num_instructions): # print info_sections[i].instruction assert len(info_all.instructions) == len(info_sections.instructions) assert np.all(info_all.index_to_row - info_sections.index_to_row == 0)
class TestChunkBreak(object): def get_disasm(self): disasm = Basic6502Disassembler() disasm.add_chunk_processor("data", 1) disasm.add_chunk_processor("antic_dl", 2) disasm.add_chunk_processor("jumpman_level", 3) disasm.add_chunk_processor("jumpman_harvest", 4) return disasm def setup(self): self.disasm = self.get_disasm() self.editor = MockHexEditor() guess = FileGuess("../test_data/pytest.atr") self.editor.load(guess) def test_simple(self): self.editor.find_segment("chunk type changes") s = self.editor.segment r = fast_get_entire_style_ranges(s, user=user_bit_mask) print(r) info = self.disasm.disassemble_segment(s) inst = info.instructions for i in range(info.num_instructions): print(info[i].instruction) assert info[0].instruction.startswith("DEX") assert info[2].instruction.startswith("RTS") assert info[4].instruction == "00" assert info[5].instruction.startswith("7070707070;") def test_recompile(self): self.editor.find_segment("modified boot") s = self.editor.segment info = self.disasm.disassemble_segment(s) disasm = self.disasm.get_disassembled_text() with open("%s.s" % s.name, "w") as fh: fh.write("\n".join(disasm) + "\n") text1 = self.disasm.get_atasm_lst_text() with open("%s.omnivore-lst" % s.name, "w") as fh: fh.write("\n".join(text1) + "\n") def test_bad(self): data = np.fromstring( "\x8dp0L\xaa8\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00pppM\x00p\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\x8d\x8d\x06\x16\x8e\r", dtype=np.uint8) style = np.empty(len(data), dtype=np.uint8) style[0:17] = 0 style[17:] = 2 raw = SegmentData(data, style) s = DefaultSegment(raw, 0x3bef) info = self.disasm.disassemble_segment(s) inst = info.instructions for i in range(info.num_instructions): print(info[i].instruction) text = self.disasm.get_disassembled_text() print("\n".join(text)) text = self.disasm.get_atasm_lst_text() print("\n".join(text))
class TestChunkBreak(object): def get_disasm(self): disasm = Basic6502Disassembler() disasm.add_chunk_processor("data", 1) disasm.add_chunk_processor("antic_dl", 2) disasm.add_chunk_processor("jumpman_level", 3) disasm.add_chunk_processor("jumpman_harvest", 4) return disasm def setup(self): self.disasm = self.get_disasm() self.editor = MockHexEditor() guess = FileGuess("../test_data/pytest.atr") self.editor.load(guess) def test_simple(self): self.editor.find_segment("chunk type changes") s = self.editor.segment r = fast_get_entire_style_ranges(s, user=user_bit_mask) print(r) info = self.disasm.disassemble_segment(s) inst = info.instructions for i in range(info.num_instructions): print(info[i].instruction) assert info[0].instruction.startswith("DEX") assert info[2].instruction.startswith("RTS") assert info[4].instruction == "00" assert info[5].instruction.startswith("7070707070;") def test_recompile(self): self.editor.find_segment("modified boot") s = self.editor.segment info = self.disasm.disassemble_segment(s) disasm = self.disasm.get_disassembled_text() with open("%s.s" % s.name, "w") as fh: fh.write("\n".join(disasm) + "\n") text1 = self.disasm.get_atasm_lst_text() with open("%s.omnivore-lst" % s.name, "w") as fh: fh.write("\n".join(text1) + "\n") def test_bad(self): data = np.fromstring("\x8dp0L\xaa8\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00pppM\x00p\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\r\x8d\x8d\x06\x16\x8e\r", dtype=np.uint8) style = np.empty(len(data), dtype=np.uint8) style[0:17] = 0 style[17:] = 2 raw = SegmentData(data, style) s = DefaultSegment(raw, 0x3bef) info = self.disasm.disassemble_segment(s) inst = info.instructions for i in range(info.num_instructions): print(info[i].instruction) text = self.disasm.get_disassembled_text() print("\n".join(text)) text = self.disasm.get_atasm_lst_text() print("\n".join(text))
class TestFastDisasm(object): def get_disasm(self): return Basic6502Disassembler() def setup(self): self.editor = MockHexEditor() guess = FileGuess("../test_data/pytest.atr") self.editor.load(guess) self.disasm = self.get_disasm() self.fast = self.disasm.fast def test_ranges(self): # force the use of the normal disassembler for the other style self.disasm.fast.chunk_type_processor[64] = self.disasm.fast.chunk_processor self.editor.find_segment("02: robots I") s = self.editor.segment r = fast_get_entire_style_ranges(s, user=user_bit_mask, split_comments=[]) print(r) assert r == [ ((0, 497), 0), ((497, 524), 1), ((524, 602), 0), ((602, 690), 1), ((690, 1004), 0), ((1004, 1024), 1), ((1024, 1536), 0), ((1536, 1710), 1), ((1710, 1792), 0), ((1792, 1954), 1), ((1954, 2048), 0)] info_all = self.fast.get_all(s.rawdata.unindexed_data, s.origin, 0) #print info_all.instructions[0:20] info_sections = self.fast.get_all(s.rawdata.unindexed_data, s.origin, 0, r) # for i in range(info_sections.num_instructions): # print info_sections[i].instruction assert len(info_all.instructions) == len(info_sections.instructions) assert np.all(info_all.index_to_row - info_sections.index_to_row == 0)