blob: 17330ae8d7a45259d6aef71601b8aef84ecdc598 [file] [log] [blame]
Kevin O'Connor202024a2009-01-17 10:41:28 -05001#!/usr/bin/env python
Kevin O'Connor5b8f8092009-09-20 19:47:45 -04002# Script to analyze code and arrange ld sections.
Kevin O'Connor202024a2009-01-17 10:41:28 -05003#
Kevin O'Connor1a4885e2010-09-15 21:28:31 -04004# Copyright (C) 2008-2010 Kevin O'Connor <kevin@koconnor.net>
Kevin O'Connor202024a2009-01-17 10:41:28 -05005#
6# This file may be distributed under the terms of the GNU GPLv3 license.
7
8import sys
9
Kevin O'Connor5b8f8092009-09-20 19:47:45 -040010# LD script headers/trailers
11COMMONHEADER = """
12/* DO NOT EDIT! This is an autogenerated file. See tools/layoutrom.py. */
13OUTPUT_FORMAT("elf32-i386")
14OUTPUT_ARCH("i386")
15SECTIONS
16{
17"""
18COMMONTRAILER = """
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040019
20 /* Discard regular data sections to force a link error if
21 * code attempts to access data not marked with VAR16 (or other
22 * appropriate macro)
23 */
24 /DISCARD/ : {
25 *(.text*) *(.data*) *(.bss*) *(.rodata*)
Kevin O'Connor90ebed42012-06-21 20:54:53 -040026 *(COMMON) *(.discard*) *(.eh_frame) *(.note*)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040027 }
Kevin O'Connor5b8f8092009-09-20 19:47:45 -040028}
29"""
30
Kevin O'Connorc0693942009-06-10 21:56:01 -040031
32######################################################################
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040033# Determine section locations
Kevin O'Connorc0693942009-06-10 21:56:01 -040034######################################################################
35
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040036# Align 'pos' to 'alignbytes' offset
37def alignpos(pos, alignbytes):
38 mask = alignbytes - 1
39 return (pos + mask) & ~mask
40
41# Determine the final addresses for a list of sections that end at an
Kevin O'Connor5b8f8092009-09-20 19:47:45 -040042# address.
Kevin O'Connor46b82622012-05-13 12:10:30 -040043def setSectionsStart(sections, endaddr, minalign=1, segoffset=0):
Kevin O'Connor5b8f8092009-09-20 19:47:45 -040044 totspace = 0
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040045 for section in sections:
46 if section.align > minalign:
47 minalign = section.align
48 totspace = alignpos(totspace, section.align) + section.size
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040049 startaddr = (endaddr - totspace) / minalign * minalign
50 curaddr = startaddr
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040051 for section in sections:
52 curaddr = alignpos(curaddr, section.align)
53 section.finalloc = curaddr
Kevin O'Connor46b82622012-05-13 12:10:30 -040054 section.finalsegloc = curaddr - segoffset
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040055 curaddr += section.size
Kevin O'Connor46b82622012-05-13 12:10:30 -040056 return startaddr, minalign
Kevin O'Connorc0693942009-06-10 21:56:01 -040057
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040058# The 16bit code can't exceed 64K of space.
59BUILD_BIOS_ADDR = 0xf0000
60BUILD_BIOS_SIZE = 0x10000
Kevin O'Connor46b82622012-05-13 12:10:30 -040061BUILD_ROM_START = 0xc0000
Kevin O'Connor2b0fb8c2013-08-07 23:03:47 -040062BUILD_LOWRAM_END = 0xa0000
Kevin O'Connor6d152642013-02-19 21:35:20 -050063# Space to reserve in f-segment for dynamic allocations
64BUILD_MIN_BIOSTABLE = 2048
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040065
66# Layout the 16bit code. This ensures sections with fixed offset
67# requirements are placed in the correct location. It also places the
68# 16bit code as high as possible in the f-segment.
69def fitSections(sections, fillsections):
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040070 # fixedsections = [(addr, section), ...]
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040071 fixedsections = []
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040072 for section in sections:
73 if section.name.startswith('.fixedaddr.'):
74 addr = int(section.name[11:], 16)
Kevin O'Connor46b82622012-05-13 12:10:30 -040075 section.finalloc = addr + BUILD_BIOS_ADDR
76 section.finalsegloc = addr
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040077 fixedsections.append((addr, section))
78 if section.align != 1:
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040079 print "Error: Fixed section %s has non-zero alignment (%d)" % (
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040080 section.name, section.align)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040081 sys.exit(1)
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040082 fixedsections.sort()
83 firstfixed = fixedsections[0][0]
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040084
85 # Find freespace in fixed address area
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040086 # fixedAddr = [(freespace, section), ...]
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040087 fixedAddr = []
88 for i in range(len(fixedsections)):
89 fixedsectioninfo = fixedsections[i]
90 addr, section = fixedsectioninfo
91 if i == len(fixedsections) - 1:
92 nextaddr = BUILD_BIOS_SIZE
93 else:
94 nextaddr = fixedsections[i+1][0]
Kevin O'Connor1a4885e2010-09-15 21:28:31 -040095 avail = nextaddr - addr - section.size
96 fixedAddr.append((avail, section))
97 fixedAddr.sort()
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -040098
99 # Attempt to fit other sections into fixed area
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400100 canrelocate = [(section.size, section.align, section.name, section)
101 for section in fillsections]
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400102 canrelocate.sort()
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400103 canrelocate = [section for size, align, name, section in canrelocate]
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400104 totalused = 0
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400105 for freespace, fixedsection in fixedAddr:
Kevin O'Connor46b82622012-05-13 12:10:30 -0400106 addpos = fixedsection.finalsegloc + fixedsection.size
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400107 totalused += fixedsection.size
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400108 nextfixedaddr = addpos + freespace
109# print "Filling section %x uses %d, next=%x, available=%d" % (
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400110# fixedsection.finalloc, fixedsection.size, nextfixedaddr, freespace)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400111 while 1:
112 canfit = None
113 for fitsection in canrelocate:
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400114 if addpos + fitsection.size > nextfixedaddr:
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400115 # Can't fit and nothing else will fit.
116 break
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400117 fitnextaddr = alignpos(addpos, fitsection.align) + fitsection.size
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400118# print "Test %s - %x vs %x" % (
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400119# fitsection.name, fitnextaddr, nextfixedaddr)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400120 if fitnextaddr > nextfixedaddr:
121 # This item can't fit.
122 continue
123 canfit = (fitnextaddr, fitsection)
124 if canfit is None:
125 break
126 # Found a section that can fit.
127 fitnextaddr, fitsection = canfit
128 canrelocate.remove(fitsection)
Kevin O'Connor46b82622012-05-13 12:10:30 -0400129 fitsection.finalloc = addpos + BUILD_BIOS_ADDR
130 fitsection.finalsegloc = addpos
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400131 addpos = fitnextaddr
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400132 totalused += fitsection.size
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400133# print " Adding %s (size %d align %d) pos=%x avail=%d" % (
134# fitsection[2], fitsection[0], fitsection[1]
135# , fitnextaddr, nextfixedaddr - fitnextaddr)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400136
137 # Report stats
138 total = BUILD_BIOS_SIZE-firstfixed
139 slack = total - totalused
140 print ("Fixed space: 0x%x-0x%x total: %d slack: %d"
141 " Percent slack: %.1f%%" % (
142 firstfixed, BUILD_BIOS_SIZE, total, slack,
143 (float(slack) / total) * 100.0))
144
Kevin O'Connor46b82622012-05-13 12:10:30 -0400145 return firstfixed + BUILD_BIOS_ADDR
146
147# Return the subset of sections with a given category
148def getSectionsCategory(sections, category):
149 return [section for section in sections if section.category == category]
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400150
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400151# Return the subset of sections with a given name prefix
Kevin O'Connor46b82622012-05-13 12:10:30 -0400152def getSectionsPrefix(sections, prefix):
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400153 return [section for section in sections
Kevin O'Connor46b82622012-05-13 12:10:30 -0400154 if section.name.startswith(prefix)]
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400155
Kevin O'Connor46b82622012-05-13 12:10:30 -0400156# The sections (and associated information) to be placed in output rom
157class LayoutInfo:
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500158 genreloc = None
Kevin O'Connor46b82622012-05-13 12:10:30 -0400159 sections16 = sec16_start = sec16_align = None
160 sections32seg = sec32seg_start = sec32seg_align = None
161 sections32flat = sec32flat_start = sec32flat_align = None
162 sections32init = sec32init_start = sec32init_align = None
163 sections32low = sec32low_start = sec32low_align = None
Kevin O'Connor41953492013-02-18 23:09:01 -0500164 sections32fseg = sec32fseg_start = sec32fseg_align = None
Kevin O'Connor6d152642013-02-19 21:35:20 -0500165 zonefseg_start = zonefseg_end = None
Kevin O'Connorc9243442013-02-17 13:58:28 -0500166 zonelow_base = final_sec32low_start = None
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500167 exportsyms = varlowsyms = None
Kevin O'Connor46b82622012-05-13 12:10:30 -0400168
169# Determine final memory addresses for sections
Kevin O'Connor2b0fb8c2013-08-07 23:03:47 -0400170def doLayout(sections, config):
Kevin O'Connor46b82622012-05-13 12:10:30 -0400171 li = LayoutInfo()
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400172 # Determine 16bit positions
Kevin O'Connor46b82622012-05-13 12:10:30 -0400173 li.sections16 = getSectionsCategory(sections, '16')
174 textsections = getSectionsPrefix(li.sections16, '.text.')
Kevin O'Connor805ede22012-02-08 20:23:36 -0500175 rodatasections = (
Kevin O'Connor46b82622012-05-13 12:10:30 -0400176 getSectionsPrefix(li.sections16, '.rodata.str1.1')
177 + getSectionsPrefix(li.sections16, '.rodata.__func__.')
178 + getSectionsPrefix(li.sections16, '.rodata.__PRETTY_FUNCTION__.'))
179 datasections = getSectionsPrefix(li.sections16, '.data16.')
180 fixedsections = getSectionsPrefix(li.sections16, '.fixedaddr.')
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400181
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400182 firstfixed = fitSections(fixedsections, textsections)
183 remsections = [s for s in textsections+rodatasections+datasections
184 if s.finalloc is None]
Kevin O'Connor46b82622012-05-13 12:10:30 -0400185 li.sec16_start, li.sec16_align = setSectionsStart(
186 remsections, firstfixed, segoffset=BUILD_BIOS_ADDR)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400187
188 # Determine 32seg positions
Kevin O'Connor46b82622012-05-13 12:10:30 -0400189 li.sections32seg = getSectionsCategory(sections, '32seg')
190 textsections = getSectionsPrefix(li.sections32seg, '.text.')
Kevin O'Connor805ede22012-02-08 20:23:36 -0500191 rodatasections = (
Kevin O'Connor46b82622012-05-13 12:10:30 -0400192 getSectionsPrefix(li.sections32seg, '.rodata.str1.1')
193 + getSectionsPrefix(li.sections32seg, '.rodata.__func__.')
194 + getSectionsPrefix(li.sections32seg, '.rodata.__PRETTY_FUNCTION__.'))
195 datasections = getSectionsPrefix(li.sections32seg, '.data32seg.')
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400196
Kevin O'Connor46b82622012-05-13 12:10:30 -0400197 li.sec32seg_start, li.sec32seg_align = setSectionsStart(
198 textsections + rodatasections + datasections, li.sec16_start
199 , segoffset=BUILD_BIOS_ADDR)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400200
Kevin O'Connor41953492013-02-18 23:09:01 -0500201 # Determine "fseg memory" data positions
202 li.sections32fseg = getSectionsCategory(sections, '32fseg')
203
204 li.sec32fseg_start, li.sec32fseg_align = setSectionsStart(
205 li.sections32fseg, li.sec32seg_start, 16
206 , segoffset=BUILD_BIOS_ADDR)
207
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400208 # Determine 32flat runtime positions
Kevin O'Connor46b82622012-05-13 12:10:30 -0400209 li.sections32flat = getSectionsCategory(sections, '32flat')
210 textsections = getSectionsPrefix(li.sections32flat, '.text.')
211 rodatasections = getSectionsPrefix(li.sections32flat, '.rodata')
212 datasections = getSectionsPrefix(li.sections32flat, '.data.')
213 bsssections = getSectionsPrefix(li.sections32flat, '.bss.')
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400214
Kevin O'Connor46b82622012-05-13 12:10:30 -0400215 li.sec32flat_start, li.sec32flat_align = setSectionsStart(
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400216 textsections + rodatasections + datasections + bsssections
Kevin O'Connor41953492013-02-18 23:09:01 -0500217 , li.sec32fseg_start, 16)
Kevin O'Connor6d152642013-02-19 21:35:20 -0500218 li.zonefseg_end = li.sec32flat_start
219 li.zonefseg_start = BUILD_BIOS_ADDR
220 if li.zonefseg_start + BUILD_MIN_BIOSTABLE > li.zonefseg_end:
221 # Not enough ZoneFSeg space - force a minimum space.
222 li.zonefseg_end = li.sec32fseg_start
223 li.zonefseg_start = li.zonefseg_end - BUILD_MIN_BIOSTABLE
224 li.sec32flat_start, li.sec32flat_align = setSectionsStart(
225 textsections + rodatasections + datasections + bsssections
226 , li.zonefseg_start, 16)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400227
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400228 # Determine 32flat init positions
Kevin O'Connor46b82622012-05-13 12:10:30 -0400229 li.sections32init = getSectionsCategory(sections, '32init')
230 textsections = getSectionsPrefix(li.sections32init, '.text.')
231 rodatasections = getSectionsPrefix(li.sections32init, '.rodata')
232 datasections = getSectionsPrefix(li.sections32init, '.data.')
233 bsssections = getSectionsPrefix(li.sections32init, '.bss.')
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400234
Kevin O'Connor46b82622012-05-13 12:10:30 -0400235 li.sec32init_start, li.sec32init_align = setSectionsStart(
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400236 textsections + rodatasections + datasections + bsssections
Kevin O'Connor46b82622012-05-13 12:10:30 -0400237 , li.sec32flat_start, 16)
238
239 # Determine "low memory" data positions
240 li.sections32low = getSectionsCategory(sections, '32low')
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500241 sec32low_end = li.sec32init_start
Kevin O'Connor2b0fb8c2013-08-07 23:03:47 -0400242 if config.get('CONFIG_MALLOC_UPPERMEMORY'):
243 final_sec32low_end = min(BUILD_BIOS_ADDR, li.sec32flat_start)
244 zonelow_base = final_sec32low_end - 64*1024
245 li.zonelow_base = max(BUILD_ROM_START, alignpos(zonelow_base, 2*1024))
246 else:
247 final_sec32low_end = BUILD_LOWRAM_END
248 li.zonelow_base = final_sec32low_end - 64*1024
Kevin O'Connor3be89a12013-02-23 16:07:00 -0500249 relocdelta = final_sec32low_end - sec32low_end
Kevin O'Connor46b82622012-05-13 12:10:30 -0400250 li.sec32low_start, li.sec32low_align = setSectionsStart(
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500251 li.sections32low, sec32low_end, 16
Kevin O'Connorc9243442013-02-17 13:58:28 -0500252 , segoffset=li.zonelow_base - relocdelta)
Kevin O'Connorc91da7a2012-06-08 21:14:19 -0400253 li.final_sec32low_start = li.sec32low_start + relocdelta
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400254
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400255 # Print statistics
Kevin O'Connor46b82622012-05-13 12:10:30 -0400256 size16 = BUILD_BIOS_ADDR + BUILD_BIOS_SIZE - li.sec16_start
257 size32seg = li.sec16_start - li.sec32seg_start
Kevin O'Connor41953492013-02-18 23:09:01 -0500258 size32fseg = li.sec32seg_start - li.sec32fseg_start
259 size32flat = li.sec32fseg_start - li.sec32flat_start
Kevin O'Connor46b82622012-05-13 12:10:30 -0400260 size32init = li.sec32flat_start - li.sec32init_start
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500261 sizelow = sec32low_end - li.sec32low_start
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400262 print "16bit size: %d" % size16
263 print "32bit segmented size: %d" % size32seg
264 print "32bit flat size: %d" % size32flat
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400265 print "32bit flat init size: %d" % size32init
Kevin O'Connor46b82622012-05-13 12:10:30 -0400266 print "Lowmem size: %d" % sizelow
Kevin O'Connor41953492013-02-18 23:09:01 -0500267 print "f-segment var size: %d" % size32fseg
Kevin O'Connor46b82622012-05-13 12:10:30 -0400268 return li
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400269
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400270
271######################################################################
272# Linker script output
273######################################################################
274
275# Write LD script includes for the given cross references
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500276def outXRefs(sections, useseg=0, exportsyms=[], forcedelta=0):
Kevin O'Connora3c48f52013-02-05 22:36:13 -0500277 xrefs = dict([(symbol.name, symbol) for symbol in exportsyms])
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400278 out = ""
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400279 for section in sections:
280 for reloc in section.relocs:
281 symbol = reloc.symbol
Kevin O'Connora3c48f52013-02-05 22:36:13 -0500282 if (symbol.section is not None
283 and (symbol.section.fileid != section.fileid
284 or symbol.name != reloc.symbolname)):
285 xrefs[reloc.symbolname] = symbol
286 for symbolname, symbol in xrefs.items():
287 loc = symbol.section.finalloc
288 if useseg:
289 loc = symbol.section.finalsegloc
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500290 out += "%s = 0x%x ;\n" % (symbolname, loc + forcedelta + symbol.offset)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400291 return out
292
293# Write LD script includes for the given sections using relative offsets
Kevin O'Connor46b82622012-05-13 12:10:30 -0400294def outRelSections(sections, startsym, useseg=0):
295 sections = [(section.finalloc, section) for section in sections
296 if section.finalloc is not None]
297 sections.sort()
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400298 out = ""
Kevin O'Connor46b82622012-05-13 12:10:30 -0400299 for addr, section in sections:
300 loc = section.finalloc
301 if useseg:
302 loc = section.finalsegloc
303 out += ". = ( 0x%x - %s ) ;\n" % (loc, startsym)
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400304 if section.name == '.rodata.str1.1':
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400305 out += "_rodata = . ;\n"
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400306 out += "*(%s)\n" % (section.name,)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400307 return out
308
Kevin O'Connor46b82622012-05-13 12:10:30 -0400309# Build linker script output for a list of relocations.
310def strRelocs(outname, outrel, relocs):
311 relocs.sort()
312 return (" %s_start = ABSOLUTE(.) ;\n" % (outname,)
313 + "".join(["LONG(0x%x - %s)\n" % (pos, outrel)
314 for pos in relocs])
315 + " %s_end = ABSOLUTE(.) ;\n" % (outname,))
Kevin O'Connor871e0a02009-12-30 12:14:53 -0500316
Kevin O'Connor46b82622012-05-13 12:10:30 -0400317# Find all relocations in the given sections with the given attributes
318def getRelocs(sections, type=None, category=None, notcategory=None):
319 out = []
320 for section in sections:
321 for reloc in section.relocs:
322 if reloc.symbol.section is None:
323 continue
324 destcategory = reloc.symbol.section.category
325 if ((type is None or reloc.type == type)
326 and (category is None or destcategory == category)
327 and (notcategory is None or destcategory != notcategory)):
328 out.append(section.finalloc + reloc.offset)
329 return out
330
331# Return the start address and minimum alignment for a set of sections
332def getSectionsStart(sections, defaddr=0):
333 return min([section.finalloc for section in sections
334 if section.finalloc is not None] or [defaddr])
335
336# Output the linker scripts for all required sections.
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500337def writeLinkerScripts(li, out16, out32seg, out32flat):
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400338 # Write 16bit linker script
Kevin O'Connor46b82622012-05-13 12:10:30 -0400339 out = outXRefs(li.sections16, useseg=1) + """
Kevin O'Connorc9243442013-02-17 13:58:28 -0500340 zonelow_base = 0x%x ;
341 _zonelow_seg = 0x%x ;
Kevin O'Connor46b82622012-05-13 12:10:30 -0400342
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400343 code16_start = 0x%x ;
344 .text16 code16_start : {
Kevin O'Connor46b82622012-05-13 12:10:30 -0400345%s
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400346 }
Kevin O'Connorc9243442013-02-17 13:58:28 -0500347""" % (li.zonelow_base,
348 li.zonelow_base / 16,
Kevin O'Connor46b82622012-05-13 12:10:30 -0400349 li.sec16_start - BUILD_BIOS_ADDR,
350 outRelSections(li.sections16, 'code16_start', useseg=1))
351 outfile = open(out16, 'wb')
352 outfile.write(COMMONHEADER + out + COMMONTRAILER)
353 outfile.close()
Kevin O'Connor871e0a02009-12-30 12:14:53 -0500354
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400355 # Write 32seg linker script
Kevin O'Connor46b82622012-05-13 12:10:30 -0400356 out = outXRefs(li.sections32seg, useseg=1) + """
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400357 code32seg_start = 0x%x ;
358 .text32seg code32seg_start : {
Kevin O'Connor46b82622012-05-13 12:10:30 -0400359%s
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400360 }
Kevin O'Connor46b82622012-05-13 12:10:30 -0400361""" % (li.sec32seg_start - BUILD_BIOS_ADDR,
362 outRelSections(li.sections32seg, 'code32seg_start', useseg=1))
363 outfile = open(out32seg, 'wb')
364 outfile.write(COMMONHEADER + out + COMMONTRAILER)
365 outfile.close()
Kevin O'Connor871e0a02009-12-30 12:14:53 -0500366
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400367 # Write 32flat linker script
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500368 sections32all = (li.sections32flat + li.sections32init + li.sections32fseg)
Kevin O'Connor46b82622012-05-13 12:10:30 -0400369 sec32all_start = li.sec32low_start
Kevin O'Connor402fd9c2010-09-15 00:26:19 -0400370 relocstr = ""
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500371 if li.genreloc:
Kevin O'Connor402fd9c2010-09-15 00:26:19 -0400372 # Generate relocations
Kevin O'Connor46b82622012-05-13 12:10:30 -0400373 absrelocs = getRelocs(
374 li.sections32init, type='R_386_32', category='32init')
375 relrelocs = getRelocs(
376 li.sections32init, type='R_386_PC32', notcategory='32init')
377 initrelocs = getRelocs(
378 li.sections32flat + li.sections32low + li.sections16
Kevin O'Connor41953492013-02-18 23:09:01 -0500379 + li.sections32seg + li.sections32fseg, category='32init')
Kevin O'Connor46b82622012-05-13 12:10:30 -0400380 relocstr = (strRelocs("_reloc_abs", "code32init_start", absrelocs)
381 + strRelocs("_reloc_rel", "code32init_start", relrelocs)
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500382 + strRelocs("_reloc_init", "code32flat_start", initrelocs))
383 numrelocs = len(absrelocs + relrelocs + initrelocs)
Kevin O'Connor46b82622012-05-13 12:10:30 -0400384 sec32all_start -= numrelocs * 4
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500385 out = outXRefs(li.sections32low, exportsyms=li.varlowsyms
386 , forcedelta=li.final_sec32low_start-li.sec32low_start)
387 out += outXRefs(sections32all, exportsyms=li.exportsyms) + """
Kevin O'Connor402fd9c2010-09-15 00:26:19 -0400388 _reloc_min_align = 0x%x ;
Kevin O'Connor6d152642013-02-19 21:35:20 -0500389 zonefseg_start = 0x%x ;
390 zonefseg_end = 0x%x ;
Kevin O'Connorc9243442013-02-17 13:58:28 -0500391 zonelow_base = 0x%x ;
392 final_varlow_start = 0x%x ;
Kevin O'Connor46b82622012-05-13 12:10:30 -0400393
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400394 code32flat_start = 0x%x ;
395 .text code32flat_start : {
Kevin O'Connor46b82622012-05-13 12:10:30 -0400396%s
Kevin O'Connorc9243442013-02-17 13:58:28 -0500397 varlow_start = ABSOLUTE(.) ;
Kevin O'Connor46b82622012-05-13 12:10:30 -0400398%s
Kevin O'Connorc9243442013-02-17 13:58:28 -0500399 varlow_end = ABSOLUTE(.) ;
Kevin O'Connor402fd9c2010-09-15 00:26:19 -0400400 code32init_start = ABSOLUTE(.) ;
Kevin O'Connor46b82622012-05-13 12:10:30 -0400401%s
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400402 code32init_end = ABSOLUTE(.) ;
Kevin O'Connor46b82622012-05-13 12:10:30 -0400403%s
Kevin O'Connor41953492013-02-18 23:09:01 -0500404%s
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400405 . = ( 0x%x - code32flat_start ) ;
406 *(.text32seg)
407 . = ( 0x%x - code32flat_start ) ;
408 *(.text16)
409 code32flat_end = ABSOLUTE(.) ;
410 } :text
Kevin O'Connora3c48f52013-02-05 22:36:13 -0500411""" % (li.sec32init_align,
Kevin O'Connor6d152642013-02-19 21:35:20 -0500412 li.zonefseg_start,
413 li.zonefseg_end,
Kevin O'Connorc9243442013-02-17 13:58:28 -0500414 li.zonelow_base,
Kevin O'Connorc91da7a2012-06-08 21:14:19 -0400415 li.final_sec32low_start,
Kevin O'Connor46b82622012-05-13 12:10:30 -0400416 sec32all_start,
417 relocstr,
418 outRelSections(li.sections32low, 'code32flat_start'),
419 outRelSections(li.sections32init, 'code32flat_start'),
420 outRelSections(li.sections32flat, 'code32flat_start'),
Kevin O'Connor41953492013-02-18 23:09:01 -0500421 outRelSections(li.sections32fseg, 'code32flat_start'),
Kevin O'Connor46b82622012-05-13 12:10:30 -0400422 li.sec32seg_start,
423 li.sec16_start)
424 out = COMMONHEADER + out + COMMONTRAILER + """
Kevin O'Connora3c48f52013-02-05 22:36:13 -0500425ENTRY(entry_elf)
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400426PHDRS
427{
428 text PT_LOAD AT ( code32flat_start ) ;
429}
Kevin O'Connora3c48f52013-02-05 22:36:13 -0500430"""
Kevin O'Connor46b82622012-05-13 12:10:30 -0400431 outfile = open(out32flat, 'wb')
432 outfile.write(out)
433 outfile.close()
Kevin O'Connorc0693942009-06-10 21:56:01 -0400434
435
436######################################################################
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400437# Detection of init code
438######################################################################
439
Kevin O'Connor2af52da2013-03-08 19:36:28 -0500440def markRuntime(section, sections, chain=[]):
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400441 if (section is None or not section.keep or section.category is not None
442 or '.init.' in section.name or section.fileid != '32flat'):
443 return
Kevin O'Connor2af52da2013-03-08 19:36:28 -0500444 if '.data.varinit.' in section.name:
445 print "ERROR: %s is VARVERIFY32INIT but used from %s" % (
446 section.name, chain)
447 sys.exit(1)
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400448 section.category = '32flat'
449 # Recursively mark all sections this section points to
450 for reloc in section.relocs:
Kevin O'Connor2af52da2013-03-08 19:36:28 -0500451 markRuntime(reloc.symbol.section, sections, chain + [section.name])
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400452
Kevin O'Connor2af52da2013-03-08 19:36:28 -0500453def findInit(sections, genreloc):
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400454 # Recursively find and mark all "runtime" sections.
455 for section in sections:
Kevin O'Connor41953492013-02-18 23:09:01 -0500456 if ('.data.varlow.' in section.name or '.data.varfseg.' in section.name
457 or '.runtime.' in section.name or '.export.' in section.name):
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400458 markRuntime(section, sections)
459 for section in sections:
460 if section.category is not None:
461 continue
Kevin O'Connor2af52da2013-03-08 19:36:28 -0500462 if section.fileid == '32flat' and genreloc:
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400463 section.category = '32init'
464 else:
465 section.category = section.fileid
466
467
468######################################################################
Kevin O'Connorc0693942009-06-10 21:56:01 -0400469# Section garbage collection
470######################################################################
471
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500472CFUNCPREFIX = [('_cfunc16_', 0), ('_cfunc32seg_', 1), ('_cfunc32flat_', 2)]
473
Kevin O'Connorfdca4182010-01-01 12:46:54 -0500474# Find and keep the section associated with a symbol (if available).
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500475def keepsymbol(reloc, infos, pos, isxref):
476 symbolname = reloc.symbolname
477 mustbecfunc = 0
478 for symprefix, needpos in CFUNCPREFIX:
479 if symbolname.startswith(symprefix):
480 if needpos != pos:
481 return -1
482 symbolname = symbolname[len(symprefix):]
483 mustbecfunc = 1
484 break
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400485 symbol = infos[pos][1].get(symbolname)
486 if (symbol is None or symbol.section is None
487 or symbol.section.name.startswith('.discard.')):
Kevin O'Connorfdca4182010-01-01 12:46:54 -0500488 return -1
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500489 isdestcfunc = (symbol.section.name.startswith('.text.')
490 and not symbol.section.name.startswith('.text.asm.'))
491 if ((mustbecfunc and not isdestcfunc)
492 or (not mustbecfunc and isdestcfunc and isxref)):
493 return -1
494
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400495 reloc.symbol = symbol
496 keepsection(symbol.section, infos, pos)
Kevin O'Connorfdca4182010-01-01 12:46:54 -0500497 return 0
498
Kevin O'Connor5b8f8092009-09-20 19:47:45 -0400499# Note required section, and recursively set all referenced sections
500# as required.
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400501def keepsection(section, infos, pos=0):
502 if section.keep:
Kevin O'Connorc0693942009-06-10 21:56:01 -0400503 # Already kept - nothing to do.
504 return
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400505 section.keep = 1
Kevin O'Connorc0693942009-06-10 21:56:01 -0400506 # Keep all sections that this section points to
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400507 for reloc in section.relocs:
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500508 ret = keepsymbol(reloc, infos, pos, 0)
Kevin O'Connorfdca4182010-01-01 12:46:54 -0500509 if not ret:
Kevin O'Connorc0693942009-06-10 21:56:01 -0400510 continue
511 # Not in primary sections - it may be a cross 16/32 reference
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500512 ret = keepsymbol(reloc, infos, (pos+1)%3, 1)
Kevin O'Connorfdca4182010-01-01 12:46:54 -0500513 if not ret:
Kevin O'Connor871e0a02009-12-30 12:14:53 -0500514 continue
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500515 ret = keepsymbol(reloc, infos, (pos+2)%3, 1)
Kevin O'Connorfdca4182010-01-01 12:46:54 -0500516 if not ret:
517 continue
Kevin O'Connorc0693942009-06-10 21:56:01 -0400518
Kevin O'Connor5b8f8092009-09-20 19:47:45 -0400519# Determine which sections are actually referenced and need to be
520# placed into the output file.
Kevin O'Connor871e0a02009-12-30 12:14:53 -0500521def gc(info16, info32seg, info32flat):
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400522 # infos = ((sections16, symbols16), (sect32seg, sym32seg)
523 # , (sect32flat, sym32flat))
524 infos = (info16, info32seg, info32flat)
Kevin O'Connorc0693942009-06-10 21:56:01 -0400525 # Start by keeping sections that are globally visible.
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400526 for section in info16[0]:
527 if section.name.startswith('.fixedaddr.') or '.export.' in section.name:
Kevin O'Connor871e0a02009-12-30 12:14:53 -0500528 keepsection(section, infos)
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400529 return [section for section in info16[0]+info32seg[0]+info32flat[0]
530 if section.keep]
Kevin O'Connorc0693942009-06-10 21:56:01 -0400531
532
533######################################################################
534# Startup and input parsing
535######################################################################
536
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400537class Section:
538 name = size = alignment = fileid = relocs = None
Kevin O'Connor46b82622012-05-13 12:10:30 -0400539 finalloc = finalsegloc = category = keep = None
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400540class Reloc:
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500541 offset = type = symbolname = symbol = None
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400542class Symbol:
543 name = offset = section = None
544
Kevin O'Connorc0693942009-06-10 21:56:01 -0400545# Read in output from objdump
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400546def parseObjDump(file, fileid):
547 # sections = [section, ...]
Kevin O'Connorc0693942009-06-10 21:56:01 -0400548 sections = []
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400549 sectionmap = {}
550 # symbols[symbolname] = symbol
Kevin O'Connorc0693942009-06-10 21:56:01 -0400551 symbols = {}
Kevin O'Connorc0693942009-06-10 21:56:01 -0400552
553 state = None
554 for line in file.readlines():
555 line = line.rstrip()
556 if line == 'Sections:':
557 state = 'section'
558 continue
559 if line == 'SYMBOL TABLE:':
560 state = 'symbol'
561 continue
Kevin O'Connor6c2e7812010-09-13 18:04:02 -0400562 if line.startswith('RELOCATION RECORDS FOR ['):
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400563 sectionname = line[24:-2]
564 if sectionname.startswith('.debug_'):
565 # Skip debugging sections (to reduce parsing time)
566 state = None
567 continue
Kevin O'Connorc0693942009-06-10 21:56:01 -0400568 state = 'reloc'
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400569 relocsection = sectionmap[sectionname]
Kevin O'Connorc0693942009-06-10 21:56:01 -0400570 continue
571
572 if state == 'section':
573 try:
574 idx, name, size, vma, lma, fileoff, align = line.split()
575 if align[:3] != '2**':
576 continue
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400577 section = Section()
578 section.name = name
579 section.size = int(size, 16)
580 section.align = 2**int(align[3:])
581 section.fileid = fileid
582 section.relocs = []
583 sections.append(section)
584 sectionmap[name] = section
585 except ValueError:
Kevin O'Connorc0693942009-06-10 21:56:01 -0400586 pass
587 continue
588 if state == 'symbol':
589 try:
Kevin O'Connor90ebed42012-06-21 20:54:53 -0400590 parts = line[17:].split()
591 if len(parts) == 3:
592 sectionname, size, name = parts
593 elif len(parts) == 4 and parts[2] == '.hidden':
594 sectionname, size, hidden, name = parts
595 else:
596 continue
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400597 symbol = Symbol()
598 symbol.size = int(size, 16)
599 symbol.offset = int(line[:8], 16)
600 symbol.name = name
601 symbol.section = sectionmap.get(sectionname)
602 symbols[name] = symbol
603 except ValueError:
Kevin O'Connorc0693942009-06-10 21:56:01 -0400604 pass
605 continue
606 if state == 'reloc':
607 try:
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400608 off, type, symbolname = line.split()
609 reloc = Reloc()
610 reloc.offset = int(off, 16)
611 reloc.type = type
Kevin O'Connorf3fe3aa2010-12-05 12:38:33 -0500612 reloc.symbolname = symbolname
Kevin O'Connor67863be2010-12-24 10:23:10 -0500613 reloc.symbol = symbols.get(symbolname)
614 if reloc.symbol is None:
615 # Some binutils (2.20.1) give section name instead
616 # of a symbol - create a dummy symbol.
617 reloc.symbol = symbol = Symbol()
618 symbol.size = 0
619 symbol.offset = 0
620 symbol.name = symbolname
621 symbol.section = sectionmap.get(symbolname)
622 symbols[symbolname] = symbol
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400623 relocsection.relocs.append(reloc)
624 except ValueError:
Kevin O'Connorc0693942009-06-10 21:56:01 -0400625 pass
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400626 return sections, symbols
Kevin O'Connorc0693942009-06-10 21:56:01 -0400627
Kevin O'Connor2b0fb8c2013-08-07 23:03:47 -0400628# Parser for constants in simple C header files.
629def scanconfig(file):
630 f = open(file, 'rb')
631 opts = {}
632 for l in f.readlines():
633 parts = l.split()
634 if len(parts) != 3:
635 continue
636 if parts[0] != '#define':
637 continue
638 value = parts[2]
639 if value.isdigit() or (value.startswith('0x') and value[2:].isdigit()):
640 value = int(value, 0)
641 opts[parts[1]] = value
642 return opts
643
Kevin O'Connorc0693942009-06-10 21:56:01 -0400644def main():
645 # Get output name
Kevin O'Connor2b0fb8c2013-08-07 23:03:47 -0400646 in16, in32seg, in32flat, cfgfile, out16, out32seg, out32flat = sys.argv[1:]
Kevin O'Connorc0693942009-06-10 21:56:01 -0400647
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400648 # Read in the objdump information
Kevin O'Connorc0693942009-06-10 21:56:01 -0400649 infile16 = open(in16, 'rb')
Kevin O'Connor871e0a02009-12-30 12:14:53 -0500650 infile32seg = open(in32seg, 'rb')
651 infile32flat = open(in32flat, 'rb')
Kevin O'Connorc0693942009-06-10 21:56:01 -0400652
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400653 # infoX = (sections, symbols)
654 info16 = parseObjDump(infile16, '16')
655 info32seg = parseObjDump(infile32seg, '32seg')
656 info32flat = parseObjDump(infile32flat, '32flat')
Kevin O'Connorc0693942009-06-10 21:56:01 -0400657
Kevin O'Connor2b0fb8c2013-08-07 23:03:47 -0400658 # Read kconfig config file
659 config = scanconfig(cfgfile)
660
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400661 # Figure out which sections to keep.
Kevin O'Connor1a4885e2010-09-15 21:28:31 -0400662 sections = gc(info16, info32seg, info32flat)
Kevin O'Connorc0693942009-06-10 21:56:01 -0400663
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400664 # Separate 32bit flat into runtime and init parts
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500665 genreloc = '_reloc_abs_start' in info32flat[1]
Kevin O'Connor2af52da2013-03-08 19:36:28 -0500666 findInit(sections, genreloc)
Kevin O'Connord1b4f962010-09-15 21:38:16 -0400667
Kevin O'Connor41953492013-02-18 23:09:01 -0500668 # Note "low memory" and "fseg memory" parts
Kevin O'Connorc9243442013-02-17 13:58:28 -0500669 for section in getSectionsPrefix(sections, '.data.varlow.'):
Kevin O'Connor46b82622012-05-13 12:10:30 -0400670 section.category = '32low'
Kevin O'Connor41953492013-02-18 23:09:01 -0500671 for section in getSectionsPrefix(sections, '.data.varfseg.'):
672 section.category = '32fseg'
Kevin O'Connor46b82622012-05-13 12:10:30 -0400673
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400674 # Determine the final memory locations of each kept section.
Kevin O'Connor2b0fb8c2013-08-07 23:03:47 -0400675 li = doLayout(sections, config)
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500676 li.genreloc = genreloc
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400677
Kevin O'Connora3c48f52013-02-05 22:36:13 -0500678 # Exported symbols
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500679 li.exportsyms = [symbol for symbol in info16[1].values()
680 if (symbol.section is not None
681 and '.export.' in symbol.section.name
682 and symbol.name != symbol.section.name)]
683 li.varlowsyms = [symbol for symbol in info32flat[1].values()
684 if (symbol.section is not None
685 and symbol.section.finalloc is not None
686 and '.data.varlow.' in symbol.section.name
687 and symbol.name != symbol.section.name)]
Kevin O'Connora3c48f52013-02-05 22:36:13 -0500688
Kevin O'Connor9ba1dea2010-05-01 09:50:13 -0400689 # Write out linker script files.
Kevin O'Connor6afc6f82013-02-19 01:02:50 -0500690 writeLinkerScripts(li, out16, out32seg, out32flat)
Kevin O'Connorc0693942009-06-10 21:56:01 -0400691
Kevin O'Connor202024a2009-01-17 10:41:28 -0500692if __name__ == '__main__':
693 main()