Mercurial > lcfOS
annotate python/ppci/objectfile.py @ 384:94f5b719ad0b
Small refactor
| author | Windel Bouwman |
|---|---|
| date | Sun, 27 Apr 2014 17:50:25 +0200 |
| parents | 0c44e494ef58 |
| children | d056b552d3f4 |
| rev | line source |
|---|---|
| 334 | 1 |
| 2 """ | |
| 3 Object files are used to store assembled code. Information contained | |
| 4 is code, symbol table and relocation information. | |
| 5 """ | |
| 6 | |
| 377 | 7 import json |
| 8 import binascii | |
| 384 | 9 from . import CompilerError, make_num |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
10 |
| 334 | 11 class Symbol: |
| 335 | 12 def __init__(self, name, value, section): |
| 334 | 13 self.name = name |
| 14 self.value = value | |
| 335 | 15 self.section = section |
| 16 | |
| 17 def __repr__(self): | |
| 18 return 'SYM {}, val={} sec={}'.format(self.name, self.value, self.section) | |
| 334 | 19 |
| 377 | 20 def __eq__(self, other): |
| 21 return (self.name, self.value, self.section) == \ | |
| 22 (other.name, other.value, other.section) | |
| 23 | |
| 334 | 24 |
| 25 class Relocation: | |
| 342 | 26 """ Represents a relocation entry. A relocation always has a symbol to refer to |
| 27 and a relocation type """ | |
| 335 | 28 def __init__(self, sym, offset, typ, section): |
| 29 self.sym = sym | |
| 30 self.offset = offset | |
| 31 self.typ = typ | |
| 32 self.section = section | |
| 33 | |
| 34 def __repr__(self): | |
| 35 return 'RELOC {} off={} t={} sec={}'.format(self.sym, self.offset, self.typ, self.section) | |
| 334 | 36 |
| 377 | 37 def __eq__(self, other): |
| 38 return (self.sym, self.offset, self.typ, self.section) ==\ | |
| 39 (other.sym, other.offset, other.typ, other.section) | |
| 40 | |
| 334 | 41 |
| 42 class Section: | |
| 43 def __init__(self, name): | |
| 44 self.name = name | |
| 335 | 45 self.address = 0 |
| 46 self.data = bytearray() | |
| 47 | |
| 48 def add_data(self, data): | |
| 49 self.data += data | |
| 50 | |
| 51 @property | |
| 52 def Size(self): | |
| 53 return len(self.data) | |
| 54 | |
| 55 def __repr__(self): | |
| 56 return 'SECTION {}'.format(self.name) | |
| 334 | 57 |
| 377 | 58 def __eq__(self, other): |
| 59 return (self.name == other.name) and (self.address == other.address) \ | |
| 60 and (self.data == other.data) | |
| 61 | |
| 334 | 62 |
| 63 class ObjectFile: | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
64 """ Container for sections with compiled code or data. |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
65 Also contains symbols and relocation entries """ |
| 334 | 66 def __init__(self): |
| 67 self.symbols = {} | |
| 68 self.sections = {} | |
| 69 self.relocations = [] | |
| 70 | |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
71 def find_symbol(self, name): |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
72 return self.symbols[name] |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
73 |
| 335 | 74 def add_symbol(self, name, value, section): |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
75 if name in self.symbols: |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
76 raise CompilerError('{} already defined'.format(name)) |
|
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
77 assert section in self.sections |
| 335 | 78 sym = Symbol(name, value, section) |
| 334 | 79 self.symbols[name] = sym |
| 80 return sym | |
| 335 | 81 |
| 82 def add_relocation(self, sym_name, offset, typ, section): | |
| 342 | 83 assert type(sym_name) is str, str(sym_name) |
|
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
335
diff
changeset
|
84 assert section in self.sections |
| 377 | 85 # assert sym_name in self.symbols |
| 335 | 86 reloc = Relocation(sym_name, offset, typ, section) |
| 87 self.relocations.append(reloc) | |
| 88 return reloc | |
| 89 | |
| 90 def get_section(self, name): | |
| 91 if not name in self.sections: | |
| 92 self.sections[name] = Section(name) | |
| 93 return self.sections[name] | |
| 377 | 94 |
| 95 def __eq__(self, other): | |
| 96 return (self.symbols == other.symbols) and \ | |
| 97 (self.sections == other.sections) and \ | |
| 98 (self.relocations == other.relocations) | |
| 99 | |
| 100 def save(self, f): | |
| 101 save_object(self, f) | |
| 102 | |
| 103 | |
| 104 def save_object(o, f): | |
| 105 json.dump(serialize(o), f, indent=2, sort_keys=True) | |
| 106 | |
| 107 | |
| 108 def load_object(f): | |
| 109 return deserialize(json.load(f)) | |
| 110 | |
| 111 | |
| 112 def serialize(x): | |
| 113 res = {} | |
| 114 if isinstance(x, ObjectFile): | |
| 115 res['sections'] = [] | |
| 116 for sname in sorted(x.sections.keys()): | |
| 117 s = x.sections[sname] | |
| 118 res['sections'].append(serialize(s)) | |
| 119 res['symbols'] = [] | |
| 120 for sname in sorted(x.symbols.keys()): | |
| 121 s = x.symbols[sname] | |
| 122 res['symbols'].append(serialize(s)) | |
| 123 res['relocations'] = [] | |
| 124 for reloc in x.relocations: | |
| 125 res['relocations'].append(serialize(reloc)) | |
| 126 elif isinstance(x, Section): | |
| 127 res['name'] = x.name | |
| 128 res['address'] = hex(x.address) | |
| 129 res['data'] = binascii.hexlify(x.data).decode('ascii') | |
| 130 elif isinstance(x, Symbol): | |
| 131 res['name'] = x.name | |
| 132 res['value'] = hex(x.value) | |
| 133 res['section'] = x.section | |
| 134 elif isinstance(x, Relocation): | |
| 135 res['symbol'] = x.sym | |
| 136 res['offset'] = hex(x.offset) | |
| 137 res['type'] = x.typ | |
| 138 res['section'] = x.section | |
| 139 return res | |
| 140 | |
| 141 | |
| 142 def deserialize(d): | |
| 143 obj = ObjectFile() | |
| 144 for section in d['sections']: | |
| 145 so = obj.get_section(section['name']) | |
| 384 | 146 so.address = make_num(section['address']) |
| 377 | 147 so.data = bytearray(binascii.unhexlify(section['data'].encode('ascii'))) |
| 148 for reloc in d['relocations']: | |
| 384 | 149 obj.add_relocation(reloc['symbol'], make_num(reloc['offset']), |
| 377 | 150 reloc['type'], reloc['section']) |
| 151 for sym in d['symbols']: | |
| 384 | 152 obj.add_symbol(sym['name'], make_num(sym['value']), sym['section']) |
| 377 | 153 return obj |
| 154 |