120 lines
3.6 KiB
Python
120 lines
3.6 KiB
Python
import gdb
|
|
import sys
|
|
|
|
|
|
# CSTRING ==============================================================================================================
|
|
class CStringPrinter:
|
|
def __init__(self, val):
|
|
self.val = val
|
|
|
|
def to_string(self):
|
|
return self.val['_cstr']
|
|
|
|
def children(self):
|
|
size = int(self.val['_size'])
|
|
start = self.val['_cstr']
|
|
return (('[{}]'.format(i), start[i]) for i in range(size))
|
|
|
|
|
|
# ALLOCATION ===========================================================================================================
|
|
class StringPrinter:
|
|
def __init__(self, val):
|
|
self.val = val
|
|
|
|
def to_string(self):
|
|
return str("fennec::allocation = {value}").format(value = self.val['_str']['_data'])
|
|
|
|
def children(self):
|
|
size = int(self.val['_str']['_capacity'])
|
|
start = self.val['_str']['_data']
|
|
return (('[{}]'.format(i), start[i]) for i in range(size))
|
|
|
|
# OPTIONAL =================================================================================================================
|
|
class OptionalPrinter:
|
|
"""Print a fennec::optional"""
|
|
|
|
def __init__(self, val):
|
|
self.val = val
|
|
|
|
def to_string(self):
|
|
is_initialized = self.val['_set']
|
|
if is_initialized:
|
|
return "Value = {}".format(self.val['_val'])
|
|
else:
|
|
return "empty"
|
|
|
|
|
|
# ALLOCATION ===========================================================================================================
|
|
class AllocationPrinter:
|
|
def __init__(self, val):
|
|
self.val = val
|
|
|
|
def to_string(self):
|
|
return "fennec::allocation"
|
|
|
|
def children(self):
|
|
size = int(self.val['_capacity'])
|
|
start = self.val['_data']
|
|
return (('[{}]'.format(i), start[i]) for i in range(size))
|
|
|
|
|
|
# LIST =================================================================================================================
|
|
class ListPrinter:
|
|
"""Print a fennec::list"""
|
|
|
|
class Iterator:
|
|
def __init__(self, val):
|
|
self.list = val
|
|
self.node = self.list['_root']
|
|
#self.index = 0
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def __next__(self):
|
|
if self.node == 18446744073709551615:
|
|
raise StopIteration
|
|
|
|
|
|
value = self.list['_table']['_data'][self.node]['value']['_val']
|
|
self.node = self.list['_table']['_data'][self.node]['next']
|
|
return value
|
|
#index = self.index
|
|
#self.index = self.index + 1
|
|
#value = self.list['_table']['_data'][self.node]['value']['_val']
|
|
#self.node = self.list['_table']['_data'][self.node]['next']
|
|
#return str("value[{index}] = {value}").format(index = index, value = value)
|
|
|
|
|
|
|
|
def __init__(self, val):
|
|
self.val = val
|
|
|
|
def to_string(self):
|
|
return "fennec::list"
|
|
|
|
def children(self):
|
|
return enumerate(self.Iterator(self.val))
|
|
|
|
|
|
# GDB CODE =============================================================================================================
|
|
|
|
|
|
def lookup_function(val):
|
|
type = str(val.type)
|
|
if type.startswith('const '):
|
|
type = type[6:]
|
|
if type.startswith('fennec::cstring') or type.startswith('fennec::wcstring'):
|
|
return CStringPrinter(val)
|
|
if type.startswith('fennec::string') or type.startswith('fennec::wstring'):
|
|
return StringPrinter(val)
|
|
if type.startswith('fennec::optional'):
|
|
return OptionalPrinter(val)
|
|
if type.startswith('fennec::list'):
|
|
return ListPrinter(val)
|
|
if type.startswith('fennec::allocation'):
|
|
return AllocationPrinter(val)
|
|
return None
|
|
|
|
|
|
gdb.pretty_printers.append(lookup_function) |