sporniket
commented
2 years ago """
draft assembly 68k parser
"""
import re
import sys
from enum import Enum
class BasicFragmentType(Enum):
WORD = 0
WHITE = 1
SPECIAL = 2
class BasicFragment():
def __init__(self, src):
self.src = src
self.type = BasicFragmentType.WHITE if re.match("\s+", src) else BasicFragmentType.SPECIAL if re.match("([^0-9A-Za-z_])", src) else BasicFragmentType.WORD
def __str__(self):
return f"('{self.src}', {self.type.name})"
class LineOfCode():
def __init__(self, src, fragments):
self.src = src
self.fragments = fragments
label=""
fragmentIndex = 0
while fragments[fragmentIndex].type != BasicFragmentType.WHITE:
if fragments[fragmentIndex].type == BasicFragmentType.SPECIAL and not fragments[fragmentIndex].src in (".","\\","@"):
break
label+= fragments[fragmentIndex].src
fragmentIndex += 1
self.label = label if len(label) > 0 else None
def __str__(self):
f = [f"{f}" for f in self.fragments]
lb = f"Label = {self.label}\n" if self.label is not None else "No label\n"
return f"=====[Line of code]=====\n{self.src}\n{lb}{f}\n---- ---- ---- ----"
class Parser():
def parseLine(self, src:str)->LineOfCode:
fragments = []
for preFragment in re.split("(\s+)", src):
if (re.match("^\s+$", preFragment)):
fragments+=[preFragment]
else:
fragments+=[f for f in re.split("([^0-9A-Za-z_])",preFragment) if len(f)>0]
return LineOfCode(src, [BasicFragment(f) for f in fragments])
def test_Parser_parseLine_can_find_label_when_there_is_one():
loc = Parser().parseLine("DestAssetFont0: dc.b 'font_0.dat',0")
print(loc)
assert loc.label == "DestAssetFont0"
loc = Parser().parseLine(".nextEntry\@ _Setcolor d5,#-1")
print(loc)
assert loc.label == ".nextEntry\@"
def test_Parser_parseLine_does_not_set_label_when_there_is_none():
loc = Parser().parseLine(" moveq #0,d5")
print(loc)
assert loc.label == None
def run(test):
print(f"----- {test} -----")
test()
if __name__ == "__main__":
if len(sys.argv) == 1:
print(f"=====[ START SELF TESTS ]=====")
run(test_Parser_parseLine_can_find_label_when_there_is_one)
run(test_Parser_parseLine_does_not_set_label_when_there_is_none)
else:
prsr = Parser()
for file in sys.argv[1:]:
with open(file) as f:
lines = f.readlines()
for line in lines:
print(prsr.conditionLine(line))
vasm is quite fine tool, except for the licence terms that prevents freedom #0 .
I should get the same binary from sporny-wrecking-ball when using vasm or my compiler