Macintosh-Tools/lpch
1
0
Fork 0
mirror of https://github.com/elliotnunn/lpch.git synced 2024-05-29 02:41:27 +00:00
Code Issues Projects Releases Wiki Activity

Mutate token stream to make sense

Browse Source
This commit is contained in:
Elliot Nunn 2019-09-06 17:05:09 +08:00
parent 37f7f31dcc
commit 0fffe2c90f
1 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
patch_rip.py
View File

@ -212,10 +212,20 @@ for num, data, annotations in lpch_list:
tokens.append(tok) tokens.append(tok)
if tok[0] == 'end': break if tok[0] == 'end': break
# Mutate the tokens list to merge the 'prev=' tokens
# (do it here, not coded yet)
for i in reversed(range(len(tokens) - 1)):
if tokens[i+1][0].startswith('prev='):
assert tokens[i][0] == 'distance'
tokens[i] = (tokens[i][0] + tokens[i+1][0][4:],) + tokens[i][1:]
del tokens[i+1]
# From here on, a 'distance' token can be treated as 'distance=module_end'
if args.pt: if args.pt:
daccum = 0 daccum = 0
for i, (a, b) in enumerate(tokens): for i, (a, b) in enumerate(tokens):
if a == 'distance': if a.startswith('distance'):
daccum += b daccum += b
print('%02d'%i, a, hex(b), '='+hex(daccum)) print('%02d'%i, a, hex(b), '='+hex(daccum))
elif b is None: elif b is None:
@ -232,23 +242,30 @@ for num, data, annotations in lpch_list:
modules.append(Mod()) modules.append(Mod())
modules[-1].start = 0 modules[-1].start = 0
i = 0 for tok, arg in tokens:
while tokens[i][0] != 'end': if tok == 'skipjt':
a, b = tokens[i]; i += 1 jt_offset += arg
if a == 'distance': if tok.startswith('distance'):
cur_offset += b cur_offset += arg
if tokens[i][0] == 'prev=entry_not_module':
i += 1 if tok == 'distance': # to end of module
modules[-1].entry_points.append(Ent()) modules[-1].stop = cur_offset
modules[-1].entry_points[-1].offset = cur_offset modules[-1].jt_entry = jt_offset
elif tokens[i][0] == 'prev=ref_list_head': modules.append(Mod())
i += 1 modules[-1].start = cur_offset
modules[-1].references.append(cur_offset)
else: jt_offset += 1
modules[-1].stop = cur_offset
modules.append(Mod()) if tok == 'distance=entry_not_module':
modules[-1].start = cur_offset modules[-1].entry_points.append(Ent())
modules[-1].entry_points[-1].offset = cur_offset
modules[-1].entry_points[-1].jt_entry = jt_offset
jt_offset += 1
if tok == 'distance=ref_list_head':
modules[-1].references.append(cur_offset)
modules.pop() modules.pop()