mirror of https://github.com/zeldaret/botw.git
81 lines
2.6 KiB
Python
Executable File
81 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import struct
|
|
from typing import Optional
|
|
|
|
import cxxfilt
|
|
from colorama import Fore, Style
|
|
|
|
import util.elf
|
|
from util import utils
|
|
|
|
|
|
def find_vtable(symtab, class_name: str) -> Optional[str]:
|
|
name_offset = len("vtable for ")
|
|
for sym in symtab.iter_symbols():
|
|
if not sym.name.startswith("_ZTV"):
|
|
continue
|
|
if cxxfilt.demangle(sym.name)[name_offset:] == class_name:
|
|
return sym.name
|
|
return None
|
|
|
|
|
|
def bold(s) -> str:
|
|
return Style.BRIGHT + str(s) + Style.NORMAL
|
|
|
|
|
|
def dump_table(name: str) -> None:
|
|
try:
|
|
symbols = util.elf.build_symbol_table(util.elf.my_symtab)
|
|
decomp_symbols = {fn.decomp_name for fn in utils.get_functions() if fn.decomp_name}
|
|
|
|
offset, size = util.elf.get_symbol_file_offset_and_size(util.elf.my_elf, util.elf.my_symtab, name)
|
|
util.elf.my_elf.stream.seek(offset)
|
|
vtable_bytes = util.elf.my_elf.stream.read(size)
|
|
|
|
if not vtable_bytes:
|
|
utils.fail(
|
|
"empty vtable; has the key function been implemented? (https://lld.llvm.org/missingkeyfunction.html)")
|
|
|
|
print(f"{Fore.WHITE}{Style.BRIGHT}{cxxfilt.demangle(name)}{Style.RESET_ALL}")
|
|
print(f"{Fore.YELLOW}{Style.BRIGHT}vtable @ 0x0{Style.RESET_ALL}")
|
|
|
|
assert size % 8 == 0
|
|
for i in range(size // 8):
|
|
word: int = struct.unpack_from("<Q", vtable_bytes, 8 * i)[0]
|
|
name = symbols.get(word, None)
|
|
if word == 0:
|
|
pass
|
|
elif name is not None:
|
|
demangled_name: str = cxxfilt.demangle(name)
|
|
color = Fore.GREEN if name in decomp_symbols else Fore.BLUE
|
|
print(f"{color}{bold(demangled_name)}{Style.RESET_ALL}")
|
|
print(f" {name}")
|
|
elif word & (1 << 63):
|
|
offset = -struct.unpack_from("<q", vtable_bytes, 8 * i)[0]
|
|
print()
|
|
print(f"{Fore.YELLOW}{Style.BRIGHT}vtable @ {offset:#x}{Style.RESET_ALL}")
|
|
else:
|
|
print(f"{Fore.RED}unknown data: {word:016x}{Style.RESET_ALL}")
|
|
|
|
except KeyError:
|
|
utils.fail("could not find symbol")
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("symbol_name", help="Name of the vtable symbol (_ZTV...) or class name")
|
|
args = parser.parse_args()
|
|
|
|
symbol_name: str = args.symbol_name
|
|
|
|
if not symbol_name.startswith("_ZTV"):
|
|
symbol_name = find_vtable(util.elf.my_symtab, args.symbol_name)
|
|
|
|
dump_table(symbol_name)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|