1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
"""
Appropriately names state manager weak functions, given a .s file with
correctly named vtables.
Call with:
python ./tools/custom/apply_state_mgr_fns.py PATH_TO_S_FILE PATH_TO_SYMBOLS_FILE
"""
from dataclasses import dataclass
import pathlib
import re
import sys
from typing import Callable, List
@dataclass
class Vtable:
# A vtable pattern with a match named class_name
vtable_pat: re.Pattern
# A factory for function names
functions: Callable[[str], List[str | None]]
def L(items):
"""Join items and add a length prefix"""
s = None
if isinstance(items, list):
s = "".join(items)
else:
s = items
return str(len(s)) + s
sFStateMgr_c = Vtable(
re.compile("__vt__[0-9]+sFStateMgr_c<(?P<class_name>[A-Za-z0-9_]+),20sStateMethodUsr_FI_c>"),
lambda name: [
None,
None,
"__dt__" + L(["sFStateMgr_c<", name, ",20sStateMethodUsr_FI_c>"]) + "Fv",
"initializeState__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "Fv",
"executeState__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "Fv",
"finalizeState__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "Fv",
"changeState__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "FRC12sStateIDIf_c",
"refreshState__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "Fv",
"getState__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "CFv",
"getNewStateID__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "CFv",
"getStateID__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "CFv",
"getOldStateID__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "CFv",
]
)
sStateMgr_c = Vtable(
re.compile("__vt__[0-9]+sStateMgr_c<(?P<class_name>[A-Za-z0-9_]+),20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"),
lambda name: [
None,
None,
"__dt__" + L(["sStateMgr_c<", name, ",20sStateMethodUsr_FI_c,12sFStateFct_c,13sStateIDChk_c>"]) + "Fv",
# the rest are dupes
]
)
sFStateFct_c = Vtable(
re.compile("__vt__[0-9]+sFStateFct_c<(?P<class_name>[A-Za-z0-9_]+)>"),
lambda name: [
None,
None,
"__dt__" + L(["sFStateFct_c<", name, ">"]) + "Fv",
"build__" + L(["sFStateFct_c<", name, ">"]) + "FRC12sStateIDIf_c",
"dispose__" + L(["sFStateFct_c<", name, ">"]) + "FRP10sStateIf_c",
]
)
sFState_c = Vtable(
re.compile("__vt__[0-9]+sFState_c<(?P<class_name>[A-Za-z0-9_]+)>"),
lambda name: [
None,
None,
"__dt__" + L(["sFState_c<", name, ">"]) + "Fv",
"initialize__" + L(["sFState_c<", name, ">"]) + "Fv",
"execute__" + L(["sFState_c<", name, ">"]) + "Fv",
"finalize__" + L(["sFState_c<", name, ">"]) + "Fv",
]
)
sFStateID_c = Vtable(
re.compile("__vt__[0-9]+sFStateID_c<(?P<class_name>[A-Za-z0-9_]+)>"),
lambda name: [
None,
None,
"__dt__" + L(["sFStateID_c<", name, ">"]) + "Fv",
None,
None,
None,
None,
"isSameName__" + L(["sFStateID_c<", name, ">"]) + "CFPCc",
None,
None,
"initializeState__" + L(["sFStateID_c<", name, ">"]) + "CFR" + name,
"executeState__" + L(["sFStateID_c<", name, ">"]) + "CFR" + name,
"finalizeState__" + L(["sFStateID_c<", name, ">"]) + "CFR" + name,
]
)
vtables = [
sFStateMgr_c,
sStateMgr_c,
sFStateFct_c,
sFState_c,
sFStateID_c,
]
def parse_data(file):
renames = []
lines: List[str] = file.splitlines()
i = 0
while i < len(lines):
if lines[i].startswith(".obj"):
obj_name = lines[i].rsplit(",", maxsplit=1)[0][5:]
if obj_name.startswith('"') and obj_name.endswith('"'):
obj_name = obj_name[1:-1]
for tab in vtables:
if (match := tab.vtable_pat.match(obj_name)):
class_name = match.group("class_name")
fns = tab.functions(class_name)
for new_name in fns:
i += 1
if new_name is not None:
old_name = lines[i].split(" ")[-1]
if old_name.startswith('"') and old_name.endswith('"'):
old_name = old_name[1:-1]
renames.append((old_name, new_name))
break
i += 1
return renames
def main():
s_file = sys.argv[1]
sym_file = sys.argv[2]
s_text = pathlib.Path(s_file).read_text()
renames = parse_data(s_text)
sym_text = pathlib.Path(sym_file).read_text()
for (old, new) in renames:
if old != new:
sym_text = sym_text.replace(old + " = .text", new + " = .text")
pathlib.Path(sym_file).write_text(sym_text)
if __name__ == "__main__":
main()
|
