Files
ss/tools/custom/apply_objdiff_mappings.py
T
2025-02-28 14:01:46 +01:00

53 lines
1.5 KiB
Python

# Thanks KooShnoo! https://discord.com/channels/727908905392275526/1024871155804426310/1329716224971767891
"""
objdiff allows one to map symbols from the target binary to the source binary.
this script will take those symbols you have mapped, and write them to `symbols.txt`.
use case: say you have changed the signature/name of a function.
now, your source file says `foo_Ful`, but symbols.txt says `bar__Fl`.
objdiff allows you to map the `foo_Ful` in the target object to `bar__Fl` in the source object.
this script allows you to replace `foo_Ful` with `bar__Fl` in `symbols.txt`.
"""
import json
with open("./objdiff.json") as f:
objdiff_config = json.load(f)
# lol python has types. it's typython. typescript all over again
units: list = objdiff_config["units"]
symbol_mappings_list = list(
filter(None, [unit.get("symbol_mappings") for unit in units])
)
symbol_mappings = {}
for unit in units:
symbol_mapping = unit.get("symbol_mappings")
if symbol_mapping is None:
continue
symbol_mappings.update(symbol_mapping)
del unit['symbol_mappings']
with open("./config/SOUE01/symbols.txt") as f:
symbols = f.readlines()
for i, line in enumerate(symbols):
tokens = line.split()
old_symbol = tokens[0]
new_symbol = symbol_mappings.get(old_symbol)
if new_symbol is None:
continue
tokens[0] = new_symbol
symbols[i] = " ".join(tokens) + "\n"
with open("./config/SOUE01/symbols.txt", "w") as f:
f.writelines(symbols)
with open("./objdiff.json", "w") as f:
json.dump(objdiff_config, f)