mirror of
https://github.com/isledecomp/isle.git
synced 2024-11-22 07:37:59 -05:00
Virtual inheritance follow-up (#727)
This commit is contained in:
parent
0ed9f37587
commit
d9913ead83
2 changed files with 18 additions and 7 deletions
|
@ -358,24 +358,28 @@ def match_function(self, addr: int, name: str) -> bool:
|
|||
def match_vtable(
|
||||
self, addr: int, name: str, base_class: Optional[str] = None
|
||||
) -> bool:
|
||||
# Set up our potential match names
|
||||
bare_vftable = f"{name}::`vftable'"
|
||||
for_name = base_class if base_class is not None else name
|
||||
for_vftable = f"{name}::`vftable'{{for `{for_name}'}}"
|
||||
|
||||
# Only allow a match against "Class:`vftable'"
|
||||
# if this is the derived class.
|
||||
name = (
|
||||
f"{name}::`vftable'"
|
||||
if base_class is None or base_class == name
|
||||
else f"{name}::`vftable'{{for `{base_class}'}}"
|
||||
)
|
||||
if base_class is None or base_class == name:
|
||||
name_options = (for_vftable, bare_vftable)
|
||||
else:
|
||||
name_options = (for_vftable, for_vftable)
|
||||
|
||||
row = self._db.execute(
|
||||
"""
|
||||
SELECT recomp_addr
|
||||
FROM `symbols`
|
||||
WHERE orig_addr IS NULL
|
||||
AND name = ?
|
||||
AND (name = ? OR name = ?)
|
||||
AND (compare_type = ?)
|
||||
LIMIT 1
|
||||
""",
|
||||
(name, SymbolType.VTABLE.value),
|
||||
(*name_options, SymbolType.VTABLE.value),
|
||||
).fetchone()
|
||||
|
||||
if row is not None and self.set_pair(addr, row[0], SymbolType.VTABLE):
|
||||
|
|
|
@ -43,6 +43,13 @@ def set_decorated(self, name: str):
|
|||
self.node_type = SymbolType.VTABLE
|
||||
self.friendly_name = demangle_vtable(self.decorated_name)
|
||||
|
||||
elif self.decorated_name.startswith("??_8"):
|
||||
# This is the `vbtable' symbol for virtual inheritance.
|
||||
# Should be okay to reuse demangle_vtable. We still want to
|
||||
# remove things like "const" from the output.
|
||||
self.node_type = SymbolType.DATA
|
||||
self.friendly_name = demangle_vtable(self.decorated_name)
|
||||
|
||||
elif self.decorated_name.startswith("??_C@"):
|
||||
self.node_type = SymbolType.STRING
|
||||
(strlen, _) = demangle_string_const(self.decorated_name)
|
||||
|
|
Loading…
Reference in a new issue