192a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorcfrom .base import TestBase 292a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 392a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorcfrom ..disassembler import Disassembler 492a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 592a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorcclass TestDisassembler(TestBase): 692a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc def test_instantiate(self): 792a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc Disassembler('i686-apple-darwin9') 892a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 992a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc def test_basic(self): 1092a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc sequence = '\x67\xe3\x81' # jcxz -127 1192a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc triple = 'i686-apple-darwin9' 1292a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 1392a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc disassembler = Disassembler(triple) 1492a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 1592a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc count, s = disassembler.get_instruction(sequence) 1692a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc self.assertEqual(count, 3) 1792a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc self.assertEqual(s, '\tjcxz\t-127') 1892a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 1992a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc def test_get_instructions(self): 2092a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc sequence = '\x67\xe3\x81\x01\xc7' # jcxz -127; addl %eax, %edi 2192a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 2292a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc disassembler = Disassembler('i686-apple-darwin9') 2392a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 2492a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc instructions = list(disassembler.get_instructions(sequence)) 2592a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc self.assertEqual(len(instructions), 2) 2692a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc 2792a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc self.assertEqual(instructions[0], (0, 3, '\tjcxz\t-127')) 2892a3e9d63a4fd408fce76c1b2ba71b5a7fb04b8aGregory Szorc self.assertEqual(instructions[1], (3, 2, '\taddl\t%eax, %edi')) 29