2238dcc393
This is an ongoing series of commits that are reformatting our Python code. Reformatting is done with `black` (23.1.0). If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run `git checkout --ours <yourfile>` and then reformat it with black. RFC: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Differential revision: https://reviews.llvm.org/D151460
91 lines
2.4 KiB
Python
91 lines
2.4 KiB
Python
"""
|
|
Use lldb Python API to disassemble raw machine code bytes
|
|
"""
|
|
|
|
from io import StringIO
|
|
import sys
|
|
|
|
import lldb
|
|
from lldbsuite.test.decorators import *
|
|
from lldbsuite.test.lldbtest import *
|
|
from lldbsuite.test import lldbutil
|
|
|
|
|
|
class Disassemble_VST1_64(TestBase):
|
|
@no_debug_info_test
|
|
@skipIfLLVMTargetMissing("ARM")
|
|
def test_disassemble_invalid_vst_1_64_raw_data(self):
|
|
"""Test disassembling invalid vst1.64 raw bytes with the API."""
|
|
# Create a target from the debugger.
|
|
target = self.dbg.CreateTargetWithFileAndTargetTriple(
|
|
"", "thumbv7-apple-macosx"
|
|
)
|
|
self.assertTrue(target, VALID_TARGET)
|
|
|
|
raw_bytes = bytearray(
|
|
[
|
|
0xF0,
|
|
0xB5,
|
|
0x03,
|
|
0xAF,
|
|
0x2D,
|
|
0xE9,
|
|
0x00,
|
|
0x0D,
|
|
0xAD,
|
|
0xF1,
|
|
0x40,
|
|
0x04,
|
|
0x24,
|
|
0xF0,
|
|
0x0F,
|
|
0x04,
|
|
0xA5,
|
|
0x46,
|
|
]
|
|
)
|
|
|
|
assembly = """
|
|
push {r4, r5, r6, r7, lr}
|
|
add r7, sp, #0xc
|
|
push.w {r8, r10, r11}
|
|
sub.w r4, sp, #0x40
|
|
bic r4, r4, #0xf
|
|
mov sp, r4
|
|
"""
|
|
|
|
def split(s):
|
|
return [x.strip() for x in s.strip().splitlines()]
|
|
|
|
insts = target.GetInstructions(lldb.SBAddress(), raw_bytes)
|
|
|
|
if self.TraceOn():
|
|
print()
|
|
for i in insts:
|
|
print("Disassembled %s" % str(i))
|
|
|
|
sio = StringIO()
|
|
insts.Print(sio)
|
|
self.assertEqual(split(assembly), split(sio.getvalue()))
|
|
|
|
self.assertEqual(insts.GetSize(), len(split(assembly)))
|
|
|
|
for i, asm in enumerate(split(assembly)):
|
|
inst = insts.GetInstructionAtIndex(i)
|
|
sio = StringIO()
|
|
inst.Print(sio)
|
|
self.assertEqual(asm, sio.getvalue().strip())
|
|
|
|
raw_bytes = bytearray([0x04, 0xF9, 0xED, 0x82])
|
|
|
|
insts = target.GetInstructions(lldb.SBAddress(), raw_bytes)
|
|
|
|
inst = insts.GetInstructionAtIndex(0)
|
|
|
|
if self.TraceOn():
|
|
print()
|
|
print("Raw bytes: ", [hex(x) for x in raw_bytes])
|
|
print("Disassembled%s" % str(inst))
|
|
|
|
self.assertEqual(inst.GetMnemonic(target), "vst1.64")
|