def test_assemble_bss():
a = Assembler()
try:
a.assemble(src_bss)
except TypeError:
raised = True
else:
raised = False
assert not raised
assert a.offsets[BSS] == 4 # 1 word * 4B
def test_assemble():
a = Assembler()
a.assemble(src)
assert a.symbols.has_sym('start')
assert a.symbols.has_sym('end')
assert a.symbols.get_sym('start') == (REL, TEXT, 0)
assert a.symbols.get_sym('end') == (REL, TEXT, 4)
assert len(b''.join(a.sections[TEXT])) == 16 # 4 instructions * 4B
assert len(a.sections[DATA]) == 0
assert a.offsets[BSS] == 0
def test_parse_line():
a = Assembler()
lines = iter(src.splitlines())
assert a.parse_line(next(lines)) == (None, '.set', (
'const',
'123',
))
assert a.parse_line(next(lines)) == (None, '.set', (
'const_left',
'976',
))
assert a.parse_line(next(lines)) == None
assert a.parse_line(next(lines)) == ('start', 'wait', ('42', ))
assert a.parse_line(next(lines)) == (None, 'ld', (
'r0',
'r1',
'0',
))
assert a.parse_line(next(lines)) == (None, 'st', (
'r0',
'r1',
'0',
))
assert a.parse_line(next(lines)) == (None, 'halt', ())
assert a.parse_line(next(lines)) == ('end', None, ())
assert a.parse_line(next(lines)) == (
None, '.data', ()
) # test left-aligned directive is not treated as label
def test_assemble_uppercase_opcode():
a = Assembler()
try:
a.assemble(" WAIT 42")
except ValueError as e:
if str(e) != "Unknown opcode or directive: WAIT":
# re-raise failures we didn't expect
raise
raised = True
else:
raised = False
assert not raised
def test_assemble_test_regressions_from_evaluation():
line = " reg_wr (0x3ff48400 + 0x10), 1, 1, 1"
a = Assembler()
raised = False
try:
a.assemble(line)
except ValueError as e:
if str(
e
) == 'invalid register base': # ensure we trapped the expected Exception
raised = True
assert not raised
def test_assemble_global():
a = Assembler()
a.assemble(src_global)
assert a.symbols.has_sym('counter')
assert a.symbols.has_sym('internal')
assert a.symbols.has_sym('entry')
exported_symbols = a.symbols.export()
assert exported_symbols == [(0, 'counter'),
(2, 'entry')] # internal not exported
exported_symbols = a.symbols.export(True) # include non-global symbols
assert exported_symbols == [(0, 'counter'), (1, 'internal'), (2, 'entry')]
def test_assemble_bss_with_value():
lines = """\
.bss
.long 3 #non-zero value not allowed in bss section
"""
a = Assembler()
try:
a.assemble(lines)
except ValueError as e:
if str(e) != "attempt to store non-zero value in section .bss":
raise # re-raise failures we didn't expect
raised = True
else:
raised = False
assert raised
def test_support_multiple_statements_per_line():
src = """
label: nop; nop;
wait 42
"""
lines = Assembler().parse(src.splitlines())
assert lines == [('label', 'nop', ()), (None, 'nop', ()),
(None, 'wait', ('42', ))]
def test_assemble_evalulate_expressions():
src_w_expr = """\
.set shft, 2
.set loops, (1 << shft)
entry:
move r0, 1+1
move r1, loops
move r2, (shft + 10) * 2
move r3, entry << 2
"""
a = Assembler()
a.assemble(src_w_expr)
assert a.symbols.has_sym('shft')
assert a.symbols.has_sym('loops')
assert a.symbols.has_sym('entry')
assert a.symbols.get_sym('shft') == (ABS, None, 2)
assert a.symbols.get_sym('loops') == (ABS, None, 4)
assert a.symbols.get_sym('entry') == (REL, TEXT, 0)
def test_assemble_optional_comment_removal():
line = " move r1, 123 # comment"
a = Assembler()
# first assemble as normal (comments will be removed by default)
a.assemble(line)
# now assemble with comment removal disabled
try:
a.assemble(line, remove_comments=False)
except ValueError as e:
raised = True
else:
raised = False
assert raised
def test_parse_line():
a = Assembler()
lines = src.splitlines()
# note: line number = index + 1
assert a.parse_line(lines[0]) == None
assert a.parse_line(lines[1]) == ('start', 'wait', ('42', ))
assert a.parse_line(lines[2]) == (None, 'ld', (
'r0',
'r1',
'0',
))
assert a.parse_line(lines[3]) == (None, 'st', (
'r0',
'r1',
'0',
))
assert a.parse_line(lines[4]) == (None, 'halt', ())
assert a.parse_line(lines[5]) == ('end', None, ())
def test_parse_labels_correctly():
"""
description of what defines a label
https://sourceware.org/binutils/docs/as/Statements.html
https://sourceware.org/binutils/docs/as/Labels.html
"""
a = Assembler()
assert a.parse_line('') is None
assert a.parse_line('label: .set const, 42') == ('label', '.set', (
'const',
'42',
))
assert a.parse_line('label:.set const, 42') == ('label', '.set', (
'const',
'42',
))
assert a.parse_line('label:') == ('label', None, ())
assert a.parse_line(' label:') == ('label', None, ())
assert a.parse_line(' label: ') == ('label', None, ())
assert a.parse_line('nop ') == (None, 'nop', ())
assert a.parse_line('.set c, 1 ') == (None, '.set', (
'c',
'1',
))
assert a.parse_line('invalid : nop') == (
None, 'invalid', (': nop', )) # no whitespace between label and colon
assert a.parse_line('.string "hello world"') == (None, '.string',
('"hello world"', ))
assert a.parse_line('.string "hello : world"') == (None, '.string',
('"hello : world"', )
) # colon in string
assert a.parse_line('label::') == ('label', ':', ())
assert a.parse_line('label: :') == ('label', ':', ())
assert a.parse_line('a_label:') == ('a_label', None, ())
assert a.parse_line('$label:') == ('$label', None, ())
assert a.parse_line('.label:') == ('.label', None, ())
assert a.parse_line('&label:') == (None, '&label:', ()
) # & not a valid char in a label
def test_parse():
a = Assembler()
lines = remove_comments(src)
result = a.parse(lines)
assert None not in result