146 lines
3.8 KiB
Python
146 lines
3.8 KiB
Python
import re
|
|
import gdb
|
|
import sys
|
|
import struct
|
|
|
|
|
|
class CoordPrinter(object):
|
|
def __init__(self, val):
|
|
self.val = val
|
|
|
|
def to_string(self):
|
|
return self.val
|
|
|
|
def display_hint(self):
|
|
return 'array'
|
|
|
|
class _iterator(object):
|
|
def __init__(self, data):
|
|
self.data = data
|
|
|
|
def __iter__(self):
|
|
indices = self.data.type.range()
|
|
for i in range(indices[0], indices[1] + 1):
|
|
res = (str(i), self.data[i])
|
|
yield res
|
|
raise StopIteration()
|
|
|
|
|
|
def children(self):
|
|
return self._iterator(self.val['data'])
|
|
|
|
|
|
class ViewPrinter(object):
|
|
def __init__(self, val):
|
|
self.val = val
|
|
|
|
def to_string(self):
|
|
return self.val
|
|
|
|
def display_hint(self):
|
|
return 'array'
|
|
|
|
class _iterator(object):
|
|
def __init__(self, first, last):
|
|
self.cursor = first
|
|
self.last = last
|
|
self.index = 0
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def __next__(self):
|
|
if self.cursor == self.last:
|
|
raise StopIteration
|
|
|
|
val = self.cursor.dereference()
|
|
res = ("[%d]" % self.index, val)
|
|
|
|
self.cursor = self.cursor + 1
|
|
self.index = self.index + 1
|
|
|
|
return res
|
|
|
|
def children(self):
|
|
return self._iterator(
|
|
self.val["m_begin"],
|
|
self.val["m_end"]
|
|
)
|
|
|
|
class EndianValuePrinter(object):
|
|
val: gdb.Type
|
|
|
|
def __init__(self, val: gdb.Value):
|
|
self.val = val
|
|
# The type should be something like:
|
|
# "cruft::endian::value<cruft::endian::BIG,u64>"
|
|
order_name = str(val.type.template_argument(0))
|
|
|
|
# Fuck knows how to use the GDB API. It's easier to just search for a
|
|
# differentiator that's unlikely to be present in anything else.
|
|
if '::BIG' in order_name:
|
|
self.order = '>'
|
|
elif '::LITTLE' in order_name:
|
|
self.order = '<'
|
|
else:
|
|
raise RuntimeError("Unknown endianness")
|
|
|
|
# Find a format specifier for the size of the underlying value
|
|
self.size = {
|
|
1: 'B',
|
|
2: 'H',
|
|
4: 'I',
|
|
8: 'Q',
|
|
}[self.val.type.sizeof]
|
|
|
|
def to_string (self):
|
|
# Construct format specifiers for converting to bytes, then from
|
|
# bytes, so that we can perform the byte reversal.
|
|
to_bytes = f"{self.order}{self.size}"
|
|
from_bytes = f"={self.size}"
|
|
|
|
raw = int(self.val["raw"])
|
|
native = struct.unpack(from_bytes, struct.pack(to_bytes, raw))[0]
|
|
|
|
# Print the value and some annotations so that the user isn't misled
|
|
# about what value they're looking at.
|
|
return f"{native} (0x{native:x}, from 0x{raw:x})"
|
|
|
|
|
|
def build_cruft_dict():
|
|
pretty_printers_dict[re.compile('^cruft::point.*$') ] = lambda val: CoordPrinter(val)
|
|
pretty_printers_dict[re.compile('^cruft::vector.*$')] = lambda val: CoordPrinter(val)
|
|
pretty_printers_dict[re.compile('^cruft::extent.*$')] = lambda val: CoordPrinter(val)
|
|
pretty_printers_dict[re.compile('^cruft::colour.*$')] = lambda val: CoordPrinter(val)
|
|
|
|
pretty_printers_dict[re.compile('^cruft::view')] = lambda val: ViewPrinter(val)
|
|
|
|
pretty_printers_dict[re.compile('^cruft::endian::value')] = lambda val: EndianValuePrinter(val)
|
|
|
|
|
|
def lookup(val):
|
|
type = val.type
|
|
|
|
if type.code == gdb.TYPE_CODE_REF:
|
|
type = type.target()
|
|
|
|
type = type.unqualified().strip_typedefs()
|
|
|
|
typename = type.tag
|
|
if typename == None:
|
|
return None
|
|
|
|
for function in pretty_printers_dict:
|
|
if function.search(typename):
|
|
return pretty_printers_dict[function](val)
|
|
|
|
return None
|
|
|
|
|
|
def register_cruft_printers():
|
|
gdb.pretty_printers.append(lookup)
|
|
|
|
|
|
pretty_printers_dict = {}
|
|
build_cruft_dict()
|