artulab.com Git - openbsd/blob

   1 """
   2 Use lldb Python API to disassemble raw machine code bytes
   3 """
   4
   5 from __future__ import print_function
   6
   7 from io import StringIO
   8 import sys
   9
  10 import lldb
  11 from lldbsuite.test.decorators import *
  12 from lldbsuite.test.lldbtest import *
  13 from lldbsuite.test import lldbutil
  14
  15
  16 class Disassemble_VST1_64(TestBase):
  17
  18     mydir = TestBase.compute_mydir(__file__)
  19
  20     @add_test_categories(['pyapi'])
  21     @no_debug_info_test
  22     @skipIfLLVMTargetMissing("ARM")
  23     def test_disassemble_invalid_vst_1_64_raw_data(self):
  24         """Test disassembling invalid vst1.64 raw bytes with the API."""
  25         # Create a target from the debugger.
  26         target = self.dbg.CreateTargetWithFileAndTargetTriple("", "thumbv7-apple-macosx")
  27         self.assertTrue(target, VALID_TARGET)
  28
  29         raw_bytes = bytearray([0xf0, 0xb5, 0x03, 0xaf,
  30                                0x2d, 0xe9, 0x00, 0x0d,
  31                                0xad, 0xf1, 0x40, 0x04,
  32                                0x24, 0xf0, 0x0f, 0x04,
  33                                0xa5, 0x46])
  34
  35         assembly = """
  36         push   {r4, r5, r6, r7, lr}
  37         add    r7, sp, #0xc
  38         push.w {r8, r10, r11}
  39         sub.w  r4, sp, #0x40
  40         bic    r4, r4, #0xf
  41         mov    sp, r4
  42         """
  43         def split(s):
  44             return [x.strip() for x in s.strip().splitlines()]
  45
  46         insts = target.GetInstructions(lldb.SBAddress(), raw_bytes)
  47
  48         if self.TraceOn():
  49             print()
  50             for i in insts:
  51                 print("Disassembled %s" % str(i))
  52
  53         if sys.version_info.major >= 3:
  54             sio = StringIO()
  55             insts.Print(sio)
  56             self.assertEqual(split(assembly), split(sio.getvalue()))
  57
  58         self.assertEqual(insts.GetSize(), len(split(assembly)))
  59
  60         if sys.version_info.major >= 3:
  61             for i,asm in enumerate(split(assembly)):
  62                 inst = insts.GetInstructionAtIndex(i)
  63                 sio = StringIO()
  64                 inst.Print(sio)
  65                 self.assertEqual(asm, sio.getvalue().strip())
  66
  67         raw_bytes = bytearray([0x04, 0xf9, 0xed, 0x82])
  68
  69         insts = target.GetInstructions(lldb.SBAddress(), raw_bytes)
  70
  71         inst = insts.GetInstructionAtIndex(0)
  72
  73         if self.TraceOn():
  74             print()
  75             print("Raw bytes:    ", [hex(x) for x in raw_bytes])
  76             print("Disassembled%s" % str(inst))
  77
  78         self.assertTrue(inst.GetMnemonic(target) == "vst1.64")