test_disassembler.py revision fdddf771716a48857a1044abc7917886bf0bf719 
1from .base import TestBase
2
3from ..disassembler import Disassembler, Option_UseMarkup
4
5class TestDisassembler(TestBase):
6    def test_instantiate(self):
7         Disassembler('i686-apple-darwin9')
8
9    def test_basic(self):
10        sequence = '\x67\xe3\x81' # jcxz -127
11        triple = 'i686-apple-darwin9'
12
13        disassembler = Disassembler(triple)
14
15        count, s = disassembler.get_instruction(sequence)
16        self.assertEqual(count, 3)
17        self.assertEqual(s, '\tjcxz\t-127')
18
19    def test_get_instructions(self):
20        sequence = '\x67\xe3\x81\x01\xc7' # jcxz -127; addl %eax, %edi
21
22        disassembler = Disassembler('i686-apple-darwin9')
23
24        instructions = list(disassembler.get_instructions(sequence))
25        self.assertEqual(len(instructions), 2)
26
27        self.assertEqual(instructions[0], (0, 3, '\tjcxz\t-127'))
28        self.assertEqual(instructions[1], (3, 2, '\taddl\t%eax, %edi'))
29
30    def test_set_options(self):
31        sequence = '\x10\x40\x2d\xe9'
32        triple = 'arm-linux-android'
33
34        disassembler = Disassembler(triple)
35        disassembler.set_options(Option_UseMarkup)
36        count, s = disassembler.get_instruction(sequence)
37        print s
38        self.assertEqual(count, 4)
39        self.assertEqual(s, '\tpush\t{<reg:r4>, <reg:lr>}')
40