feedc0de/boost_unordered
1
0
Fork 0
forked from boostorg/unordered
Code Pull Requests Activity

Add fancy pointer support to GDB pretty-printers

Browse Source
This commit is contained in:
Braden Ganetsky
2024-08-16 17:20:09 -05:00
parent f415d17e58
commit ae364ea311
1 changed files with 31 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
extra/boost_unordered_printers.py
View File

@ -24,12 +24,23 @@ class BoostUnorderedHelpers:
return i return i
return 32 return 32
class BoostUnorderedPointerCustomizationPoint:
def __init__(self, any_ptr):
vis = gdb.default_visualizer(any_ptr)
if vis is None:
self.to_address = lambda ptr: ptr
self.next = lambda ptr, offset: ptr + offset
else:
self.to_address = lambda ptr: ptr if (ptr.type.code == gdb.TYPE_CODE_PTR) else type(vis).boost_to_address(ptr)
self.next = lambda ptr, offset: type(vis).boost_next(ptr, offset)
class BoostUnorderedFcaPrinter: class BoostUnorderedFcaPrinter:
def __init__(self, val): def __init__(self, val):
self.val = val self.val = val
self.name = f"{self.val.type.strip_typedefs()}".split("<")[0] self.name = f"{self.val.type.strip_typedefs()}".split("<")[0]
self.name = self.name.replace("boost::unordered::", "boost::") self.name = self.name.replace("boost::unordered::", "boost::")
self.is_map = self.name.endswith("map") self.is_map = self.name.endswith("map")
self.cpo = BoostUnorderedPointerCustomizationPoint(self.val["table_"]["buckets_"]["buckets"])
def to_string(self): def to_string(self):
size = self.val["table_"]["size_"] size = self.val["table_"]["size_"]
@ -38,12 +49,6 @@ class BoostUnorderedFcaPrinter:
def display_hint(self): def display_hint(self):
return "map" return "map"
def to_address(self, arg):
return arg # TODO: Add fancy pointer support
def next(self, arg, n):
return arg + n # TODO: Add fancy pointer support
def children(self): def children(self):
def generator(): def generator():
grouped_buckets = self.val["table_"]["buckets_"] grouped_buckets = self.val["table_"]["buckets_"]
@ -54,8 +59,8 @@ class BoostUnorderedFcaPrinter:
count = 0 count = 0
while bucket_index != size: while bucket_index != size:
current_bucket = self.next(self.to_address(buckets), bucket_index) current_bucket = self.cpo.next(self.cpo.to_address(buckets), bucket_index)
node = self.to_address(current_bucket.dereference()["next"]) node = self.cpo.to_address(current_bucket.dereference()["next"])
while node != 0: while node != 0:
value = node.dereference()["buf"]["t_"] value = node.dereference()["buf"]["t_"]
if self.is_map: if self.is_map:
@ -67,7 +72,7 @@ class BoostUnorderedFcaPrinter:
yield "", count yield "", count
yield "", value yield "", value
count += 1 count += 1
node = self.to_address(node.dereference()["next"]) node = self.cpo.to_address(node.dereference()["next"])
bucket_index += 1 bucket_index += 1
return generator() return generator()
@ -75,19 +80,17 @@ class BoostUnorderedFcaPrinter:
class BoostUnorderedFcaIteratorPrinter: class BoostUnorderedFcaIteratorPrinter:
def __init__(self, val): def __init__(self, val):
self.val = val self.val = val
self.cpo = BoostUnorderedPointerCustomizationPoint(self.val["p"])
def to_string(self): def to_string(self):
if self.valid(): if self.valid():
value = self.to_address(self.val["p"]).dereference()["buf"]["t_"] value = self.cpo.to_address(self.val["p"]).dereference()["buf"]["t_"]
return f"iterator = {{ {value} }}" return f"iterator = {{ {value} }}"
else: else:
return "iterator = { end iterator }" return "iterator = { end iterator }"
def to_address(self, arg):
return arg # TODO: Add fancy pointer support
def valid(self): def valid(self):
return (self.to_address(self.val["p"]) != 0) and (self.to_address(self.val["itb"]["p"]) != 0) return (self.cpo.to_address(self.val["p"]) != 0) and (self.cpo.to_address(self.val["itb"]["p"]) != 0)
class BoostUnorderedFoaPrinter: class BoostUnorderedFoaPrinter:
def __init__(self, val): def __init__(self, val):
@ -95,6 +98,7 @@ class BoostUnorderedFoaPrinter:
self.name = f"{self.val.type.strip_typedefs()}".split("<")[0] self.name = f"{self.val.type.strip_typedefs()}".split("<")[0]
self.name = self.name.replace("boost::unordered::", "boost::") self.name = self.name.replace("boost::unordered::", "boost::")
self.is_map = self.name.endswith("map") self.is_map = self.name.endswith("map")
self.cpo = BoostUnorderedPointerCustomizationPoint(self.val["table_"]["arrays"]["groups_"])
def to_string(self): def to_string(self):
size = BoostUnorderedHelpers.maybe_unwrap_atomic(self.val["table_"]["size_ctrl"]["size"]) size = BoostUnorderedHelpers.maybe_unwrap_atomic(self.val["table_"]["size_ctrl"]["size"])
@ -103,12 +107,6 @@ class BoostUnorderedFoaPrinter:
def display_hint(self): def display_hint(self):
return "map" return "map"
def to_address(self, arg):
return arg # TODO: Add fancy pointer support
def next(self, arg, n):
return arg + n # TODO: Add fancy pointer support
def is_regular_layout(self, group): def is_regular_layout(self, group):
typename = group["m"].type.strip_typedefs() typename = group["m"].type.strip_typedefs()
array_size = typename.sizeof // typename.target().sizeof array_size = typename.sizeof // typename.target().sizeof
@ -143,8 +141,8 @@ class BoostUnorderedFoaPrinter:
def children(self): def children(self):
def generator(): def generator():
table = self.val["table_"] table = self.val["table_"]
groups = self.to_address(table["arrays"]["groups_"]) groups = self.cpo.to_address(table["arrays"]["groups_"])
elements = self.to_address(table["arrays"]["elements_"]) elements = self.cpo.to_address(table["arrays"]["elements_"])
pc_ = groups.cast(gdb.lookup_type("unsigned char").pointer()) pc_ = groups.cast(gdb.lookup_type("unsigned char").pointer())
p_ = elements p_ = elements
@ -158,7 +156,7 @@ class BoostUnorderedFoaPrinter:
# This if block mirrors the condition in the begin() call # This if block mirrors the condition in the begin() call
if (not first_time) or (self.match_occupied(groups.dereference()) & 1): if (not first_time) or (self.match_occupied(groups.dereference()) & 1):
pointer = BoostUnorderedHelpers.maybe_unwrap_foa_element(p_) pointer = BoostUnorderedHelpers.maybe_unwrap_foa_element(p_)
value = self.to_address(pointer).dereference() value = self.cpo.to_address(pointer).dereference()
if self.is_map: if self.is_map:
first = value["first"] first = value["first"]
second = value["second"] second = value["second"]
@ -171,40 +169,39 @@ class BoostUnorderedFoaPrinter:
first_time = False first_time = False
n0 = pc_.cast(gdb.lookup_type("uintptr_t")) % groups.dereference().type.sizeof n0 = pc_.cast(gdb.lookup_type("uintptr_t")) % groups.dereference().type.sizeof
pc_ = self.next(pc_, -n0) pc_ = self.cpo.next(pc_, -n0)
mask = (self.match_occupied(pc_.cast(groups.type).dereference()) >> (n0+1)) << (n0+1) mask = (self.match_occupied(pc_.cast(groups.type).dereference()) >> (n0+1)) << (n0+1)
while mask == 0: while mask == 0:
pc_ = self.next(pc_, groups.dereference().type.sizeof) pc_ = self.cpo.next(pc_, groups.dereference().type.sizeof)
p_ = self.next(p_, groups.dereference()["N"]) p_ = self.cpo.next(p_, groups.dereference()["N"])
mask = self.match_occupied(pc_.cast(groups.type).dereference()) mask = self.match_occupied(pc_.cast(groups.type).dereference())
n = BoostUnorderedHelpers.countr_zero(mask) n = BoostUnorderedHelpers.countr_zero(mask)
if self.is_sentinel(pc_.cast(groups.type).dereference(), n): if self.is_sentinel(pc_.cast(groups.type).dereference(), n):
p_ = 0 p_ = 0
else: else:
pc_ = self.next(pc_, n) pc_ = self.cpo.next(pc_, n)
p_ = self.next(p_, n - n0) p_ = self.cpo.next(p_, n - n0)
return generator() return generator()
class BoostUnorderedFoaIteratorPrinter: class BoostUnorderedFoaIteratorPrinter:
def __init__(self, val): def __init__(self, val):
self.val = val self.val = val
self.cpo = BoostUnorderedPointerCustomizationPoint(self.val["p_"])
def to_string(self): def to_string(self):
if self.valid(): if self.valid():
pointer = BoostUnorderedHelpers.maybe_unwrap_foa_element(self.val["p_"]) element = self.cpo.to_address(self.val["p_"])
value = self.to_address(pointer).dereference() pointer = BoostUnorderedHelpers.maybe_unwrap_foa_element(element)
value = self.cpo.to_address(pointer).dereference()
return f"iterator = {{ {value} }}" return f"iterator = {{ {value} }}"
else: else:
return "iterator = { end iterator }" return "iterator = { end iterator }"
def to_address(self, arg):
return arg # TODO: Add fancy pointer support
def valid(self): def valid(self):
return (self.to_address(self.val["p_"]) != 0) and (self.to_address(self.val["pc_"]) != 0) return (self.cpo.to_address(self.val["p_"]) != 0) and (self.cpo.to_address(self.val["pc_"]) != 0)
def boost_unordered_build_pretty_printer(): def boost_unordered_build_pretty_printer():
pp = gdb.printing.RegexpCollectionPrettyPrinter("boost_unordered") pp = gdb.printing.RegexpCollectionPrettyPrinter("boost_unordered")