1"""
2Use lldb Python API to disassemble raw machine code bytes
3"""
4
5from __future__ import print_function
6
7
8import re
9import lldb
10from lldbsuite.test.decorators import *
11from lldbsuite.test.lldbtest import *
12from lldbsuite.test import lldbutil
13
14
15class DisassembleRawDataTestCase(TestBase):
16
17    mydir = TestBase.compute_mydir(__file__)
18
19    @no_debug_info_test
20    @skipIfRemote
21    @skipIfReproducer # GetInstructions is not instrumented.
22    def test_disassemble_raw_data(self):
23        """Test disassembling raw bytes with the API."""
24        # Create a target from the debugger.
25        arch = self.getArchitecture()
26        if re.match("mips*el", arch):
27            target = self.dbg.CreateTargetWithFileAndTargetTriple("", "mipsel")
28            raw_bytes = bytearray([0x21, 0xf0, 0xa0, 0x03])
29        elif re.match("mips", arch):
30            target = self.dbg.CreateTargetWithFileAndTargetTriple("", "mips")
31            raw_bytes = bytearray([0x03, 0xa0, 0xf0, 0x21])
32        elif re.match("powerpc64le", arch):
33            target = self.dbg.CreateTargetWithFileAndTargetTriple("", "powerpc64le")
34            raw_bytes = bytearray([0x00, 0x00, 0x80, 0x38])
35        elif arch == "aarch64":
36            target = self.dbg.CreateTargetWithFileAndTargetTriple("", "aarch64")
37            raw_bytes = bytearray([0x60, 0x0c, 0x80, 0x52])
38        elif arch == "arm":
39            target = self.dbg.CreateTargetWithFileAndTargetTriple("", "arm")
40            raw_bytes = bytearray([0x63, 0x30, 0xa0, 0xe3])
41        else:
42            target = self.dbg.CreateTargetWithFileAndTargetTriple("", "x86_64")
43            raw_bytes = bytearray([0x48, 0x89, 0xe5])
44
45        self.assertTrue(target, VALID_TARGET)
46        insts = target.GetInstructions(lldb.SBAddress(0, target), raw_bytes)
47
48        inst = insts.GetInstructionAtIndex(0)
49
50        if self.TraceOn():
51            print()
52            print("Raw bytes:    ", [hex(x) for x in raw_bytes])
53            print("Disassembled%s" % str(inst))
54        if re.match("mips", arch):
55            self.assertEqual(inst.GetMnemonic(target), "move")
56            self.assertEqual(inst.GetOperands(target),
57                            '$' + "fp, " + '$' + "sp")
58        elif re.match("powerpc64le", arch):
59            self.assertEqual(inst.GetMnemonic(target), "li")
60            self.assertEqual(inst.GetOperands(target), "4, 0")
61        elif arch == "aarch64":
62            self.assertEqual(inst.GetMnemonic(target), "mov")
63            self.assertEqual(inst.GetOperands(target), "w0, #0x63")
64        elif arch == "arm":
65            self.assertEqual(inst.GetMnemonic(target), "mov")
66            self.assertEqual(inst.GetOperands(target), "r3, #99")
67        else:
68            self.assertEqual(inst.GetMnemonic(target), "movq")
69            self.assertEqual(inst.GetOperands(target),
70                            '%' + "rsp, " + '%' + "rbp")
71