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