def test_memory_layout(self):
spec = """
MEMORY flash LOCATION=0x08000000 SIZE=0x3000 {
DEFINESYMBOL(codestart)
SECTION(code)
DEFINESYMBOL(codeend)
}
MEMORY flash LOCATION=0x20000000 SIZE=0x3000 {
SECTION(data)
}
"""
memory_layout = layout.Layout.load(io.StringIO(spec))
arch = ExampleArch()
object1 = ObjectFile(arch)
object1.get_section('code', create=True).add_data(bytes([0] * 108))
object1.add_symbol(0, 'b', 'global', 24, 'code', 'object', 0)
object2 = ObjectFile(arch)
object2.get_section('code', create=True).add_data(bytes([0] * 100))
object2.get_section('data', create=True).add_data(bytes([0] * 100))
object2.add_symbol(0, 'a', 'global', 2, 'data', 'object', 0)
object2.add_symbol(1, 'c', 'global', 2, 'code', 'object', 0)
object3 = link([object1, object2], memory_layout)
self.assertEqual(0x20000000 + 2, object3.get_symbol_value('a'))
self.assertEqual(0x08000000 + 24, object3.get_symbol_value('b'))
self.assertEqual(0x08000000 + 110, object3.get_symbol_value('c'))
self.assertEqual(208, object3.get_section('code').size)
self.assertEqual(100, object3.get_section('data').size)
self.assertEqual(0x08000000, object3.get_symbol_value('codestart'))
self.assertEqual(0x08000000 + 208, object3.get_symbol_value('codeend'))
def test_undefined_reference(self):
arch = get_arch('arm')
object1 = ObjectFile(arch)
object1.get_section('.text', create=True)
object1.add_symbol(0, 'undef', 'global', None, None, 'object', 0)
object1.gen_relocation('rel8', 0, '.text', 0)
object2 = ObjectFile(arch)
with self.assertRaises(CompilerError):
link([object1, object2])
def test_save(self):
""" Test the generation of a windows exe file """
arch = get_arch('x86_64')
obj = ObjectFile(arch)
obj.get_section('code', create=True)
obj.get_section('data', create=True)
f = io.BytesIO()
# TODO:
ExeWriter().write(obj, f)
def test_rel8_relocation(self):
arch = get_arch('arm')
object1 = ObjectFile(arch)
object1.get_section('.text', create=True).add_data(bytes([0] * 100))
object1.add_symbol(10, 'a', 'global', None, None, 'object', 0)
object1.gen_relocation('rel8', 10, '.text', 0)
object2 = ObjectFile(arch)
object2.get_section('.text', create=True).add_data(bytes([0] * 100))
object2.add_symbol(0, 'a', 'global', 24, '.text', 'object', 0)
link([object1, object2])
def test_duplicate_symbol(self):
arch = get_arch('arm')
object1 = ObjectFile(arch)
object1.get_section('.text', create=True)
object1.add_symbol(0, 'a', 'global', 0, '.text', 'object', 0)
object2 = ObjectFile(arch)
object2.get_section('.text', create=True)
object2.add_symbol(0, 'a', 'global', 0, '.text', 'object', 0)
with self.assertRaises(CompilerError):
link([object1, object2])
def test_overlapping_sections(self):
""" Check that overlapping sections are detected """
obj = ObjectFile(get_arch('msp430'))
obj.get_section('s1', create=True).add_data(bytes(range(100)))
obj.get_section('s2', create=True).add_data(bytes(range(100)))
obj.add_image(Image('x', 0))
obj.get_image('x').add_section(obj.get_section('s1'))
obj.get_image('x').add_section(obj.get_section('s2'))
with self.assertRaisesRegex(ValueError, 'overlap'):
obj.get_image('x').data
def test_code_exceeds_memory(self):
""" Check the error that is given when code exceeds memory size """
arch = ExampleArch()
layout2 = layout.Layout()
m = layout.Memory('flash')
m.location = 0x0
m.size = 0x10
m.add_input(layout.Section('code'))
layout2.add_memory(m)
object1 = ObjectFile(arch)
object1.get_section('code', create=True).add_data(bytes([0] * 22))
with self.assertRaisesRegex(CompilerError, 'exceeds'):
link([object1], layout2)
class AsmTestCaseBase(unittest.TestCase):
""" Base testcase for assembly """
def setUp(self):
self.source = io.StringIO()
self.as_args = []
arch = get_arch(self.march)
self.obj = ObjectFile(arch)
self.ostream = BinaryOutputStream(self.obj)
self.ostream.select_section('code')
self.diag = DiagnosticsManager()
# Prep assembler!
self.assembler = arch.assembler
self.assembler.prepare()
def feed(self, line):
self.assembler.assemble(line, self.ostream, self.diag)
print(line, file=self.source)
def check(self, hexstr, layout=Layout()):
self.assembler.flush()
self.obj = link([self.obj], layout)
data = bytes(self.obj.get_section('code').data)
if hexstr is None:
gnu_assemble(self.source.getvalue(), as_args=self.as_args)
self.fail('Implement this test-case')
else:
self.assertSequenceEqual(bytes.fromhex(hexstr), data)
def test_normal_use(self):
arch = get_arch('example')
obj = ObjectFile(arch)
o = BinaryOutputStream(obj)
o.select_section('.text')
o.emit(Label('a'))
self.assertSequenceEqual(bytes(), obj.get_section('.text').data)
def test_offset_adjustment(self):
""" Test if offsets are correctly modified when linking debug info """
arch = get_arch('arm')
obj1 = ObjectFile(arch)
obj1.get_section('code', create=True).add_data(bytes(59))
obj2 = ObjectFile(arch)
obj2.get_section('code', create=True).add_data(bytes(59))
obj2.add_symbol(1, 'x', 'local', 5, 'code')
obj2.debug_info = debuginfo.DebugInfo()
loc = SourceLocation('a.txt', 1, 1, 22)
obj2.debug_info.add(
debuginfo.DebugLocation(loc, address=debuginfo.DebugAddress(1)))
obj = link([obj1, obj2], debug=True)
# Take into account alignment! So 60 + 5 = 65.
self.assertEqual(0, obj.debug_info.locations[0].address.symbol_id)
self.assertEqual(65, obj.get_symbol_id_value(0))
def test_symbol_values(self):
""" Check if values are correctly resolved """
arch = get_arch('arm')
object1 = ObjectFile(arch)
object1.get_section('.text', create=True).add_data(bytes([0] * 108))
object1.add_symbol(0, 'b', 'global', 24, '.text', 'object', 0)
object2 = ObjectFile(arch)
object2.get_section('.text', create=True).add_data(bytes([0] * 100))
object2.add_symbol(0, 'a', 'global', 2, '.text', 'object', 0)
layout1 = layout.Layout()
flash_mem = layout.Memory('flash')
flash_mem.location = 0x0
flash_mem.size = 0x1000
flash_mem.add_input(layout.SymbolDefinition('code_start'))
flash_mem.add_input(layout.Section('.text'))
flash_mem.add_input(layout.SymbolDefinition('code_end'))
layout1.add_memory(flash_mem)
object3 = link([object1, object2], layout1)
self.assertEqual(110, object3.get_symbol_value('a'))
self.assertEqual(24, object3.get_symbol_value('b'))
self.assertEqual(208, object3.get_section('.text').size)
self.assertEqual(0, object3.get_symbol_value('code_start'))
self.assertEqual(208, object3.get_symbol_value('code_end'))
def make_twins(self):
""" Make two object files that have equal contents """
arch = get_arch('arm')
object1 = ObjectFile(arch)
object2 = ObjectFile(arch)
object2.get_section('code', create=True).add_data(bytes(range(55)))
object1.get_section('code', create=True).add_data(bytes(range(55)))
object1.add_symbol(10, 'A', 'global', None, None, 'object', 0)
object2.add_symbol(10, 'A', 'global', None, None, 'object', 0)
object1.gen_relocation('rel8', 10, 'code', 0x2)
object2.gen_relocation('rel8', 10, 'code', 0x2)
object1.add_symbol(0, 'A2', 'global', 0x90, 'code', 'object', 0)
object2.add_symbol(0, 'A2', 'global', 0x90, 'code', 'object', 0)
object1.add_symbol(1, 'A3', 'global', 0x90, 'code', 'object', 0)
object2.add_symbol(1, 'A3', 'global', 0x90, 'code', 'object', 0)
object1.add_image(Image('a', 0x0))
object1.get_image('a').add_section(object1.get_section('code'))
object2.add_image(Image('a', 0x0))
object2.get_image('a').add_section(object2.get_section('code'))
return object1, object2