Mercurial > lcfOS
annotate python/ppci/linker.py @ 374:72a3b646d543
Added if statement sample
| author | Windel Bouwman |
|---|---|
| date | Fri, 21 Mar 2014 15:27:18 +0100 |
| parents | 98ff43cfdd36 |
| children | 9667d78ba79e |
| rev | line source |
|---|---|
| 348 | 1 import logging |
| 335 | 2 import struct |
| 334 | 3 from .objectfile import ObjectFile |
| 335 | 4 from . import CompilerError |
| 365 | 5 from .bitfun import encode_imm32 |
| 335 | 6 |
| 7 def align(x, m): | |
| 8 while ((x % m) != 0): | |
| 9 x = x + 1 | |
| 10 return x | |
| 11 | |
| 12 def wrap_negative(x, bits): | |
| 13 b = struct.unpack('<I', struct.pack('<i', x))[0] | |
| 14 mask = (1 << bits) - 1 | |
| 15 return b & mask | |
| 16 | |
| 17 reloc_map = {} | |
| 18 | |
| 19 def reloc(t): | |
| 20 def f(c): | |
| 21 reloc_map[t] = c | |
| 22 return f | |
| 23 | |
| 24 | |
| 25 @reloc('lit_add_8') | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
26 def apply_lit8(reloc, sym, section, reloc_value): |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
27 assert sym.value % 4 == 0 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
28 offset = (sym.value - (align(reloc_value + 2, 4))) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
29 assert offset in range(0, 1024, 4), str(offset)+str( self.dst.sections) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
30 rel8 = offset >> 2 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
31 section.data[reloc.offset] = rel8 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
32 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
33 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
34 @reloc('wrap_new11') |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
35 def apply_wrap_new11(reloc, sym, section, reloc_value): |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
36 offset = sym.value - (align(reloc_value, 2) + 4) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
37 assert offset in range(-2048, 2046, 2) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
38 imm11 = wrap_negative(offset >> 1, 11) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
39 section.data[reloc.offset] = (imm11 & 0xff) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
40 section.data[reloc.offset + 1] |= (imm11 >> 8) & 0x7 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
41 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
42 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
43 @reloc('rel8') |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
44 def apply_rel8(reloc, sym, section, reloc_value): |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
45 assert sym.value % 2 == 0 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
46 offset = sym.value - (align(reloc_value, 2) + 4) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
47 assert offset in range(-256, 254, 2), str(offset) + str(reloc) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
48 imm8 = wrap_negative(offset >> 1, 8) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
49 section.data[reloc.offset] = imm8 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
50 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
51 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
52 @reloc('bl_imm11_imm10') |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
53 def apply_bl_imm11(reloc, sym, section, reloc_value): |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
54 assert sym.value % 2 == 0 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
55 offset = sym.value - (align(reloc_value, 2) + 4) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
56 assert offset in range(-16777216, 16777214, 2), str(offset) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
57 imm32 = wrap_negative(offset >> 1, 32) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
58 imm11 = imm32 & 0x7FF |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
59 imm10 = (imm32 >> 11) & 0x3FF |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
60 s = (imm32 >> 24) & 0x1 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
61 section.data[reloc.offset + 2] = imm11 & 0xFF |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
62 section.data[reloc.offset + 3] |= (imm11 >> 8) & 0x7 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
63 section.data[reloc.offset] = imm10 & 0xff |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
64 section.data[reloc.offset + 1] |= ((imm10 >> 8) & 0x3) | (s << 2) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
65 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
66 @reloc('b_imm11_imm6') |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
67 def apply_b_imm11_imm6(reloc, sym, section, reloc_value): |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
68 assert sym.value % 2 == 0 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
69 offset = sym.value - (align(reloc_value, 2) + 4) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
70 assert offset in range(-1048576, 1048574, 2), str(offset) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
71 imm32 = wrap_negative(offset >> 1, 32) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
72 imm11 = imm32 & 0x7FF |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
73 imm6 = (imm32 >> 11) & 0x3F |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
74 s = (imm32 >> 24) & 0x1 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
75 section.data[reloc.offset + 2] = imm11 & 0xFF |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
76 section.data[reloc.offset + 3] |= (imm11 >> 8) & 0x7 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
77 section.data[reloc.offset] |= imm6 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
78 section.data[reloc.offset + 1] |= (s << 2) |
| 335 | 79 |
| 345 | 80 # ARM reloc!! |
| 81 # TODO: move to target classes??? | |
| 82 @reloc('b_imm24') | |
| 83 def apply_b_imm24(reloc, sym, section, reloc_value): | |
| 84 assert sym.value % 4 == 0 | |
| 85 assert reloc_value % 4 == 0 | |
| 86 offset = (sym.value - (reloc_value + 8)) | |
| 87 rel24 = wrap_negative(offset >> 2, 24) | |
| 88 section.data[reloc.offset+2] = (rel24 >> 16) & 0xFF | |
| 89 section.data[reloc.offset+1] = (rel24 >> 8) & 0xFF | |
| 90 section.data[reloc.offset+0] = rel24 & 0xFF | |
| 91 | |
| 334 | 92 |
| 350 | 93 @reloc('ldr_imm12') |
| 94 def apply_ldr_imm12(reloc, sym, section, reloc_value): | |
| 95 assert sym.value % 4 == 0 | |
| 96 assert reloc_value % 4 == 0 | |
| 97 offset = (sym.value - (reloc_value + 8)) | |
| 98 U = 1 | |
| 99 if offset < 0: | |
| 100 offset = -offset | |
| 101 U = 0 | |
| 102 assert offset < 4096 | |
| 365 | 103 section.data[reloc.offset+2] |= (U << 7) |
| 350 | 104 section.data[reloc.offset+1] |= (offset >> 8) & 0xF |
| 105 section.data[reloc.offset+0] = offset & 0xFF | |
| 106 | |
| 354 | 107 @reloc('adr_imm12') |
| 108 def apply_adr_imm12(reloc, sym, section, reloc_value): | |
| 109 assert sym.value % 4 == 0 | |
| 110 assert reloc_value % 4 == 0 | |
| 111 offset = (sym.value - (reloc_value + 8)) | |
| 112 U = 2 | |
| 113 if offset < 0: | |
| 114 offset = -offset | |
| 115 U = 1 | |
| 116 assert offset < 4096 | |
| 365 | 117 offset = encode_imm32(offset) |
| 364 | 118 section.data[reloc.offset+2] |= (U << 6) |
| 354 | 119 section.data[reloc.offset+1] |= (offset >> 8) & 0xF |
| 120 section.data[reloc.offset+0] = offset & 0xFF | |
| 121 | |
| 364 | 122 @reloc('absaddr32') |
| 123 def apply_absaddr32(reloc, sym, section, reloc_value): | |
| 124 assert sym.value % 4 == 0 | |
| 125 assert reloc_value % 4 == 0 | |
| 126 offset = sym.value | |
| 127 section.data[reloc.offset+3] = (offset >> 24) & 0xFF | |
| 128 section.data[reloc.offset+2] = (offset >> 16) & 0xFF | |
| 129 section.data[reloc.offset+1] = (offset >> 8) & 0xFF | |
| 130 section.data[reloc.offset+0] = offset & 0xFF | |
| 131 | |
| 354 | 132 |
| 334 | 133 class Linker: |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
134 """ Merges the sections of several object files and |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
135 performs relocation """ |
| 348 | 136 def __init__(self): |
| 137 self.logger = logging.getLogger('Linker') | |
| 138 | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
139 def link(self, objs, layout={}): |
| 335 | 140 # Create new object file to store output: |
| 334 | 141 self.dst = ObjectFile() |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
142 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
143 # Create sections with address: |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
144 for section_name, address in layout.items(): |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
145 self.dst.get_section(section_name).address = address |
| 335 | 146 |
| 334 | 147 # First copy all sections into output sections: |
| 148 for iobj in objs: | |
| 335 | 149 offsets = {} |
| 150 # Merge sections: | |
| 151 for in_s in iobj.sections.values(): | |
| 152 out_s = self.dst.get_section(in_s.name) | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
153 # TODO: align section in other way: |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
154 while out_s.Size % 4 != 0: |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
155 out_s.add_data(bytes([0])) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
156 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
157 # Add new section: |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
158 offsets[in_s.name] = out_s.Size |
| 335 | 159 out_s.add_data(in_s.data) |
| 348 | 160 self.logger.debug('{} {}({})'.format(offsets[in_s.name], iobj, in_s.name)) |
| 335 | 161 |
| 162 # Merge symbols: | |
| 163 for sym in iobj.symbols.values(): | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
164 out_s = self.dst.get_section(sym.section) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
165 value = offsets[sym.section] + out_s.address + sym.value |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
166 self.dst.add_symbol(sym.name, value, sym.section) |
| 363 | 167 self.logger.debug('{} at 0x{:08X} in section {}'.format(sym.name, value, sym.section)) |
| 335 | 168 |
| 169 # Merge relocations: | |
| 170 for reloc in iobj.relocations: | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
171 offset = offsets[reloc.section] + reloc.offset |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
172 self.dst.add_relocation(reloc.sym, offset, reloc.typ, reloc.section) |
| 335 | 173 |
| 174 # Perform relocations: | |
| 175 for reloc in self.dst.relocations: | |
| 176 # Lookup symbol: | |
| 177 if reloc.sym not in self.dst.symbols: | |
| 178 raise CompilerError('Undefined reference "{}"'.format(reloc.sym)) | |
| 179 sym = self.dst.symbols[reloc.sym] | |
| 180 # patch up: | |
| 181 section = self.dst.get_section(reloc.section) | |
| 182 | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
183 # Determine location in memory of reloc patchup position: |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
184 reloc_value = section.address + reloc.offset |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
185 |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
186 if reloc.typ in reloc_map: |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
187 f = reloc_map[reloc.typ] |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
188 f(reloc, sym, section, reloc_value) |
| 335 | 189 else: |
| 190 raise NotImplementedError('Unknown relocation type {}'.format(reloc.typ)) | |
| 191 | |
| 334 | 192 return self.dst |