ec7044437e
Disables allowing the python argparse library from automatically shortening command line arguments, this prevents issues whereby a new command is added and code that wrongly uses the shortened command of an existing argument which is the same as the new command being added will silently change script behaviour. Signed-off-by: Jamie McCrae <jamie.mccrae@nordicsemi.no>
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (c) 2022, Nordic Semiconductor ASA
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
'''
|
|
Script to generate image information files.
|
|
|
|
This script creates a image information header which can be included by a
|
|
second build system.
|
|
This allows a second stage build system to use image information from a Zephyr
|
|
build by including the generated header.
|
|
|
|
Information included in the image information header:
|
|
- Number of segments in the image
|
|
- LMA address of each segment
|
|
- VMA address of each segment
|
|
- LMA adjusted of each segment if the LMA addresses has been adjusted after linking
|
|
- Size of each segment
|
|
'''
|
|
|
|
import argparse
|
|
import re
|
|
from elftools.elf.elffile import ELFFile
|
|
|
|
|
|
def write_header(filename, segments, adjusted_lma):
|
|
content = []
|
|
|
|
filename_we = re.sub(r'[\W]','_', filename).upper()
|
|
content.append(f'#ifndef {filename_we}_H')
|
|
content.append(f'#define {filename_we}_H')
|
|
content.append(f'')
|
|
content.append(f'#define SEGMENT_NUM {len(segments)}')
|
|
content.append(f'#define ADJUSTED_LMA {adjusted_lma}')
|
|
|
|
for idx, segment in enumerate(segments):
|
|
segment_header = segment['segment'].header
|
|
hex_lma_addr = hex(segment_header.p_paddr)
|
|
hex_vma_addr = hex(segment_header.p_vaddr)
|
|
hex_size = hex(segment_header.p_filesz)
|
|
|
|
content.append(f'')
|
|
content.append(f'#define SEGMENT_LMA_ADDRESS_{idx} {hex_lma_addr}')
|
|
content.append(f'#define SEGMENT_VMA_ADDRESS_{idx} {hex_vma_addr}')
|
|
content.append(f'#define SEGMENT_SIZE_{idx} {hex_size}')
|
|
|
|
content.append(f'')
|
|
content.append(f'#endif /* {filename_we}_H */')
|
|
|
|
with open(filename, 'w') as out_file:
|
|
out_file.write('\n'.join(content))
|
|
|
|
|
|
def read_segments(filename):
|
|
elffile = ELFFile(open(filename, 'rb'))
|
|
segments = list()
|
|
for segment_idx in range(elffile.num_segments()):
|
|
segments.insert(segment_idx, dict())
|
|
segments[segment_idx]['segment'] = elffile.get_segment(segment_idx)
|
|
return segments
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='''
|
|
Process ELF file and extract image information.
|
|
Create header file with extracted image information which can be included
|
|
in other build systems.''', allow_abbrev=False)
|
|
|
|
parser.add_argument('--header-file', required=True,
|
|
help="""Header file to write with image data.""")
|
|
parser.add_argument('--elf-file', required=True,
|
|
help="""ELF File to process.""")
|
|
parser.add_argument('--adjusted-lma', required=False, default=0,
|
|
help="""Adjusted LMA address value.""")
|
|
args = parser.parse_args()
|
|
|
|
segments = read_segments(args.elf_file)
|
|
write_header(args.header_file, segments, args.adjusted_lma)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|