artulab.com Git - openbsd/blob

   1 #!/usr/bin/python
   2
   3 import lldb
   4 import struct
   5
   6
   7 class OperatingSystemPlugIn(object):
   8     """Class that provides data for an instance of a LLDB 'OperatingSystemPython' plug-in class"""
   9
  10     def __init__(self, process):
  11         '''Initialization needs a valid.SBProcess object.
  12
  13         This plug-in will get created after a live process is valid and has stopped for the
  14         first time.'''
  15         self.process = None
  16         self.registers = None
  17         self.threads = None
  18         if isinstance(process, lldb.SBProcess) and process.IsValid():
  19             self.process = process
  20             self.threads = None  # Will be an dictionary containing info for each thread
  21
  22     def get_target(self):
  23         # NOTE: Don't use "lldb.target" when trying to get your target as the "lldb.target"
  24         # tracks the current target in the LLDB command interpreter which isn't the
  25         # correct thing to use for this plug-in.
  26         return self.process.target
  27
  28     def create_thread(self, tid, context):
  29         if tid == 0x444444444:
  30             thread_info = {
  31                 'tid': tid,
  32                 'name': 'four',
  33                 'queue': 'queue4',
  34                 'state': 'stopped',
  35                 'stop_reason': 'none'}
  36             self.threads.append(thread_info)
  37             return thread_info
  38         return None
  39
  40     def get_thread_info(self):
  41         if not self.threads:
  42             # The sample dictionary below shows the values that can be returned for a thread
  43             # tid => thread ID (mandatory)
  44             # name => thread name (optional key/value pair)
  45             # queue => thread dispatch queue name (optional key/value pair)
  46             # state => thred state (mandatory, set to 'stopped' for now)
  47             # stop_reason => thread stop reason. (mandatory, usually set to 'none')
  48             #  Possible values include:
  49             #   'breakpoint' if the thread is stopped at a breakpoint
  50             #   'none' thread is just stopped because the process is stopped
  51             #   'trace' the thread just single stepped
  52             #   The usual value for this while threads are in memory is 'none'
  53             # register_data_addr => the address of the register data in memory (optional key/value pair)
  54             #   Specifying this key/value pair for a thread will avoid a call to get_register_data()
  55             #   and can be used when your registers are in a thread context structure that is contiguous
  56             #   in memory. Don't specify this if your register layout in memory doesn't match the layout
  57             # described by the dictionary returned from a call to the
  58             # get_register_info() method.
  59             self.threads = [{'tid': 0x111111111,
  60                              'name': 'one',
  61                              'queue': 'queue1',
  62                              'state': 'stopped',
  63                              'stop_reason': 'breakpoint'},
  64                             {'tid': 0x222222222,
  65                              'name': 'two',
  66                              'queue': 'queue2',
  67                              'state': 'stopped',
  68                              'stop_reason': 'none'},
  69                             {'tid': 0x333333333,
  70                              'name': 'three',
  71                              'queue': 'queue3',
  72                              'state': 'stopped',
  73                              'stop_reason': 'trace'}]
  74         return self.threads
  75
  76     def get_register_info(self):
  77         if self.registers is None:
  78             self.registers = dict()
  79             self.registers['sets'] = ['GPR']
  80             self.registers['registers'] = [
  81                 {'name': 'rax', 'bitsize': 64, 'offset': 0, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 0, 'dwarf': 0},
  82                 {'name': 'rbx', 'bitsize': 64, 'offset': 8, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 3, 'dwarf': 3},
  83                 {'name': 'rcx', 'bitsize': 64, 'offset': 16, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 2, 'dwarf': 2, 'generic': 'arg4', 'alt-name': 'arg4', },
  84                 {'name': 'rdx', 'bitsize': 64, 'offset': 24, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 1, 'dwarf': 1, 'generic': 'arg3', 'alt-name': 'arg3', },
  85                 {'name': 'rdi', 'bitsize': 64, 'offset': 32, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 5, 'dwarf': 5, 'generic': 'arg1', 'alt-name': 'arg1', },
  86                 {'name': 'rsi', 'bitsize': 64, 'offset': 40, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 4, 'dwarf': 4, 'generic': 'arg2', 'alt-name': 'arg2', },
  87                 {'name': 'rbp', 'bitsize': 64, 'offset': 48, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 6, 'dwarf': 6, 'generic': 'fp', 'alt-name': 'fp', },
  88                 {'name': 'rsp', 'bitsize': 64, 'offset': 56, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 7, 'dwarf': 7, 'generic': 'sp', 'alt-name': 'sp', },
  89                 {'name': 'r8', 'bitsize': 64, 'offset': 64, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 8, 'dwarf': 8, 'generic': 'arg5', 'alt-name': 'arg5', },
  90                 {'name': 'r9', 'bitsize': 64, 'offset': 72, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 9, 'dwarf': 9, 'generic': 'arg6', 'alt-name': 'arg6', },
  91                 {'name': 'r10', 'bitsize': 64, 'offset': 80, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 10, 'dwarf': 10},
  92                 {'name': 'r11', 'bitsize': 64, 'offset': 88, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 11, 'dwarf': 11},
  93                 {'name': 'r12', 'bitsize': 64, 'offset': 96, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 12, 'dwarf': 12},
  94                 {'name': 'r13', 'bitsize': 64, 'offset': 104, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 13, 'dwarf': 13},
  95                 {'name': 'r14', 'bitsize': 64, 'offset': 112, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 14, 'dwarf': 14},
  96                 {'name': 'r15', 'bitsize': 64, 'offset': 120, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 15, 'dwarf': 15},
  97                 {'name': 'rip', 'bitsize': 64, 'offset': 128, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'gcc': 16, 'dwarf': 16, 'generic': 'pc', 'alt-name': 'pc'},
  98                 {'name': 'rflags', 'bitsize': 64, 'offset': 136, 'encoding': 'uint', 'format': 'hex', 'set': 0, 'generic': 'flags', 'alt-name': 'flags'},
  99                 {'name': 'cs', 'bitsize': 64, 'offset': 144, 'encoding': 'uint', 'format': 'hex', 'set': 0},
 100                 {'name': 'fs', 'bitsize': 64, 'offset': 152, 'encoding': 'uint', 'format': 'hex', 'set': 0},
 101                 {'name': 'gs', 'bitsize': 64, 'offset': 160, 'encoding': 'uint', 'format': 'hex', 'set': 0},
 102             ]
 103         return self.registers
 104
 105     def get_register_data(self, tid):
 106         return struct.pack(
 107             '21Q',
 108             tid + 1,
 109             tid + 2,
 110             tid + 3,
 111             tid + 4,
 112             tid + 5,
 113             tid + 6,
 114             tid + 7,
 115             tid + 8,
 116             tid + 9,
 117             tid + 10,
 118             tid + 11,
 119             tid + 12,
 120             tid + 13,
 121             tid + 14,
 122             tid + 15,
 123             tid + 16,
 124             tid + 17,
 125             tid + 18,
 126             tid + 19,
 127             tid + 20,
 128             tid + 21)