Pipat Methavanitpong | 8ca381f | 2019-07-30 15:22:53 +0900 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
Jerome Forissier | 1bb9298 | 2017-12-15 14:27:02 +0100 | [diff] [blame] | 2 | # SPDX-License-Identifier: BSD-2-Clause |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 3 | # |
| 4 | # Copyright (c) 2017, Linaro Limited |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 5 | # |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 6 | |
| 7 | import argparse |
| 8 | import array |
Maxim Uvarov | 8355f92 | 2020-04-09 15:19:28 +0100 | [diff] [blame^] | 9 | from elftools.elf.elffile import ELFFile |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 10 | import os |
| 11 | import re |
Maxim Uvarov | 8355f92 | 2020-04-09 15:19:28 +0100 | [diff] [blame^] | 12 | import struct |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 13 | import uuid |
Jerome Forissier | 509a980 | 2017-08-01 10:14:57 +0200 | [diff] [blame] | 14 | import zlib |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 15 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 16 | |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 17 | def get_args(): |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 18 | parser = argparse.ArgumentParser( |
| 19 | description='Converts a Trusted ' |
| 20 | 'Application ELF file into a C source file, ready for ' |
| 21 | 'inclusion in the TEE binary as an "early TA".') |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 22 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 23 | parser.add_argument('--out', required=True, |
| 24 | help='Name of the output C file') |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 25 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 26 | parser.add_argument( |
| 27 | '--ta', |
| 28 | required=True, |
| 29 | help='Path to the TA binary. File name has to be: <uuid>.* ' |
| 30 | 'such as: 8aaaf200-2450-11e4-abe2-0002a5d5c51b.stripped.elf') |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 31 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 32 | parser.add_argument( |
| 33 | '--compress', |
| 34 | dest="compress", |
| 35 | action="store_true", |
| 36 | help='Compress the TA using the DEFLATE ' |
| 37 | 'algorithm') |
Jerome Forissier | 509a980 | 2017-08-01 10:14:57 +0200 | [diff] [blame] | 38 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 39 | return parser.parse_args() |
| 40 | |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 41 | |
Maxim Uvarov | 8355f92 | 2020-04-09 15:19:28 +0100 | [diff] [blame^] | 42 | def get_name(obj): |
| 43 | # Symbol or section .name can be a byte array or a string, we want a string |
| 44 | try: |
| 45 | name = obj.name.decode() |
| 46 | except (UnicodeDecodeError, AttributeError): |
| 47 | name = obj.name |
| 48 | return name |
| 49 | |
| 50 | |
| 51 | def ta_get_flags(ta_f): |
| 52 | with open(ta_f, 'rb') as f: |
| 53 | elffile = ELFFile(f) |
| 54 | |
| 55 | for s in elffile.iter_sections(): |
| 56 | if get_name(s) == '.ta_head': |
| 57 | return struct.unpack('<16x4xI', s.data()[:24])[0] |
| 58 | |
| 59 | raise Exception('.ta_head section not found') |
| 60 | |
| 61 | |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 62 | def main(): |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 63 | args = get_args() |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 64 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 65 | ta_uuid = uuid.UUID(re.sub(r'\..*', '', os.path.basename(args.ta))) |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 66 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 67 | with open(args.ta, 'rb') as ta: |
| 68 | bytes = ta.read() |
| 69 | uncompressed_size = len(bytes) |
| 70 | if args.compress: |
| 71 | bytes = zlib.compress(bytes) |
| 72 | size = len(bytes) |
Jerome Forissier | 509a980 | 2017-08-01 10:14:57 +0200 | [diff] [blame] | 73 | |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 74 | f = open(args.out, 'w') |
| 75 | f.write('/* Generated from ' + args.ta + ' by ' + |
| 76 | os.path.basename(__file__) + ' */\n\n') |
| 77 | f.write('#include <compiler.h>\n') |
| 78 | f.write('#include <kernel/early_ta.h>\n\n') |
| 79 | f.write('__extension__ const struct early_ta __early_ta_' + |
| 80 | ta_uuid.hex + |
| 81 | '\n__early_ta __aligned(__alignof__(struct early_ta)) = {\n') |
Maxim Uvarov | 8355f92 | 2020-04-09 15:19:28 +0100 | [diff] [blame^] | 82 | f.write('\t.flags = 0x{:04x},\n'.format(ta_get_flags(args.ta))) |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 83 | f.write('\t.uuid = {\n') |
| 84 | f.write('\t\t.timeLow = 0x{:08x},\n'.format(ta_uuid.time_low)) |
| 85 | f.write('\t\t.timeMid = 0x{:04x},\n'.format(ta_uuid.time_mid)) |
| 86 | f.write('\t\t.timeHiAndVersion = ' + |
| 87 | '0x{:04x},\n'.format(ta_uuid.time_hi_version)) |
| 88 | f.write('\t\t.clockSeqAndNode = {\n') |
| 89 | csn = '{0:02x}{1:02x}{2:012x}'.format(ta_uuid.clock_seq_hi_variant, |
| 90 | ta_uuid.clock_seq_low, ta_uuid.node) |
| 91 | f.write('\t\t\t') |
| 92 | f.write(', '.join('0x' + csn[i:i + 2] for i in range(0, len(csn), 2))) |
| 93 | f.write('\n\t\t},\n\t},\n') |
| 94 | f.write('\t.size = {:d},\n'.format(size)) |
| 95 | if args.compress: |
| 96 | f.write('\t.uncompressed_size = ' |
| 97 | '{:d},\n'.format(uncompressed_size)) |
| 98 | f.write('\t.ta = {\n') |
| 99 | i = 0 |
| 100 | while i < size: |
| 101 | if i % 8 == 0: |
| 102 | f.write('\t\t') |
Pipat Methavanitpong | 8ca381f | 2019-07-30 15:22:53 +0900 | [diff] [blame] | 103 | f.write(hex(bytes[i]) + ',') |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 104 | i = i + 1 |
| 105 | if i % 8 == 0 or i == size: |
| 106 | f.write('\n') |
| 107 | else: |
| 108 | f.write(' ') |
| 109 | f.write('\t},\n') |
| 110 | f.write('};\n') |
| 111 | f.close() |
| 112 | |
Jerome Forissier | d0c6361 | 2017-07-25 18:17:11 +0200 | [diff] [blame] | 113 | |
| 114 | if __name__ == "__main__": |
Jerome Forissier | c51c4e1 | 2018-11-14 11:02:49 +0100 | [diff] [blame] | 115 | main() |