mirror of
https://github.com/zeldaret/tp
synced 2026-05-23 15:01:53 -04:00
LagoLunatic
ae9fa8922a
* Fix functionvalue weak func order * Add script to help diffing weak function order * Improve alink weak func order (44 -> 15) * Improve m_Do_ext weak order (12 -> 7) * Remove old hack * Fix d_a_tbox weak func order * Fix d_a_npc_ks weak func order * Fix error when trying to diff weak func order for ShieldD due to hidden function alignment gap symbols
110 lines
3.3 KiB
Python
Executable File
110 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from pathlib import Path
|
|
import re
|
|
import subprocess
|
|
from argparse import ArgumentParser
|
|
|
|
def get_symbols(o_path: Path, diff_data: bool):
|
|
output = subprocess.check_output(["readelf", "-Ws", o_path]).decode("ascii")
|
|
symbols = []
|
|
for line in output.split("\n")[3:]:
|
|
if line == "":
|
|
continue
|
|
words = line.split()
|
|
if len(words) == 7 and words[-1] == "UND":
|
|
continue
|
|
_, offset, size, sym_type, scope, vis, section_index, name = words
|
|
|
|
if diff_data:
|
|
# Only diff data.
|
|
if sym_type == "FUNC":
|
|
continue
|
|
if sym_type in ["FILE", "NOTYPE", "SECTION"]:
|
|
continue
|
|
if vis == "HIDDEN":
|
|
continue
|
|
if re.search(r"^@\d+$", name):
|
|
continue
|
|
if re.search(r"^lbl_[0-9a-f]+_(?:data|bss)_[0-9a-f]+$", name):
|
|
continue
|
|
match = re.search(r"^(\S+)\$\d+$", name)
|
|
if match:
|
|
name = match.group(1)
|
|
else:
|
|
# Only diff functions.
|
|
if sym_type != "FUNC":
|
|
continue
|
|
if vis == "HIDDEN":
|
|
continue
|
|
|
|
symbols.append((sym_type, int(section_index), int(offset, 16), name))
|
|
|
|
symbols.sort()
|
|
symbol_names = [sym[-1] for sym in symbols]
|
|
return symbol_names
|
|
|
|
|
|
def print_symbols_with_unmatched_order_for_object(relative_o_path: str, version: str, diff_data: bool):
|
|
target_o = Path("build") / version / "obj" / relative_o_path
|
|
base_o = Path("build") / version / "src" / relative_o_path
|
|
if not target_o.exists():
|
|
rel_name = relative_o_path.split("/")[-1].split(".")[0]
|
|
target_o = Path("build") / version / rel_name / "obj" / relative_o_path
|
|
|
|
subprocess.check_output(["ninja", base_o])
|
|
|
|
target_symbols = get_symbols(target_o, diff_data)
|
|
base_symbols = get_symbols(base_o, diff_data)
|
|
target_symbols_set = set(target_symbols)
|
|
base_symbols = [sym for sym in base_symbols if sym in target_symbols_set]
|
|
base_idx = 0
|
|
matched_count = 0
|
|
unmatched_count = 0
|
|
for target_sym in target_symbols:
|
|
if base_idx == len(base_symbols):
|
|
base_sym = None
|
|
else:
|
|
base_sym = base_symbols[base_idx]
|
|
|
|
if target_sym == base_sym:
|
|
base_idx += 1
|
|
matched_count += 1
|
|
else:
|
|
base_idx = base_symbols.index(target_sym)
|
|
base_idx += 1
|
|
unmatched_count += 1
|
|
print(target_sym)
|
|
|
|
print("====================================")
|
|
print("Number of order differences:", unmatched_count)
|
|
|
|
|
|
def main():
|
|
parser = ArgumentParser(description="Print differences in weak function order for an object.")
|
|
parser.add_argument(
|
|
"o_path",
|
|
type=str,
|
|
default="d/actor/d_a_alink.o",
|
|
nargs='?',
|
|
help="""relative path to the object file to diff (e.g. d/actor/d_a_alink.o).""",
|
|
)
|
|
parser.add_argument(
|
|
"-v",
|
|
"--version",
|
|
type=str,
|
|
default="GZ2E01",
|
|
help="version to build",
|
|
)
|
|
parser.add_argument(
|
|
"--data",
|
|
action="store_true",
|
|
help="""diffs data instead of functions.""",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
print_symbols_with_unmatched_order_for_object(args.o_path, args.version, args.data)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|