317 lines
12 KiB
Python
317 lines
12 KiB
Python
# Copyright (c) 2023 Yonatan Schachter
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from textwrap import dedent
|
|
import struct
|
|
|
|
from west.commands import WestCommand
|
|
from west import log
|
|
|
|
|
|
try:
|
|
from elftools.elf.elffile import ELFFile
|
|
from intelhex import IntelHex
|
|
MISSING_REQUIREMENTS = False
|
|
except ImportError:
|
|
MISSING_REQUIREMENTS = True
|
|
|
|
|
|
# Based on scripts/build/uf2conv.py
|
|
def convert_from_uf2(buf):
|
|
UF2_MAGIC_START0 = 0x0A324655 # First magic number ('UF2\n')
|
|
UF2_MAGIC_START1 = 0x9E5D5157 # Second magic number
|
|
numblocks = len(buf) // 512
|
|
curraddr = None
|
|
outp = []
|
|
for blockno in range(numblocks):
|
|
ptr = blockno * 512
|
|
block = buf[ptr:ptr + 512]
|
|
hd = struct.unpack(b'<IIIIIIII', block[0:32])
|
|
if hd[0] != UF2_MAGIC_START0 or hd[1] != UF2_MAGIC_START1:
|
|
log.inf('Skipping block at ' + ptr + '; bad magic')
|
|
continue
|
|
if hd[2] & 1:
|
|
# NO-flash flag set; skip block
|
|
continue
|
|
datalen = hd[4]
|
|
if datalen > 476:
|
|
log.die(f'Invalid UF2 data size at {ptr}')
|
|
newaddr = hd[3]
|
|
if curraddr is None:
|
|
curraddr = newaddr
|
|
padding = newaddr - curraddr
|
|
if padding < 0:
|
|
log.die(f'Block out of order at {ptr}')
|
|
if padding > 10*1024*1024:
|
|
log.die(f'More than 10M of padding needed at {ptr}')
|
|
if padding % 4 != 0:
|
|
log.die(f'Non-word padding size at {ptr}')
|
|
while padding > 0:
|
|
padding -= 4
|
|
outp += b'\x00\x00\x00\x00'
|
|
outp.append(block[32 : 32 + datalen])
|
|
curraddr = newaddr + datalen
|
|
return b''.join(outp)
|
|
|
|
|
|
class Bindesc(WestCommand):
|
|
EXTENSIONS = ['bin', 'hex', 'elf', 'uf2']
|
|
|
|
# Corresponds to the definitions in include/zephyr/bindesc.h.
|
|
# Do not change without syncing the definitions in both files!
|
|
TYPE_UINT = 0
|
|
TYPE_STR = 1
|
|
TYPE_BYTES = 2
|
|
MAGIC = 0xb9863e5a7ea46046
|
|
DESCRIPTORS_END = 0xffff
|
|
|
|
def __init__(self):
|
|
self.TAG_TO_NAME = {
|
|
# Corresponds to the definitions in include/zephyr/bindesc.h.
|
|
# Do not change without syncing the definitions in both files!
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0x800): 'APP_VERSION_STRING',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x801): 'APP_VERSION_MAJOR',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x802): 'APP_VERSION_MINOR',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x803): 'APP_VERSION_PATCHLEVEL',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x804): 'APP_VERSION_NUMBER',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0x900): 'KERNEL_VERSION_STRING',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x901): 'KERNEL_VERSION_MAJOR',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x902): 'KERNEL_VERSION_MINOR',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x903): 'KERNEL_VERSION_PATCHLEVEL',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0x904): 'KERNEL_VERSION_NUMBER',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0xa00): 'BUILD_TIME_YEAR',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0xa01): 'BUILD_TIME_MONTH',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0xa02): 'BUILD_TIME_DAY',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0xa03): 'BUILD_TIME_HOUR',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0xa04): 'BUILD_TIME_MINUTE',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0xa05): 'BUILD_TIME_SECOND',
|
|
self.bindesc_gen_tag(self.TYPE_UINT, 0xa06): 'BUILD_TIME_UNIX',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xa07): 'BUILD_DATE_TIME_STRING',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xa08): 'BUILD_DATE_STRING',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xa09): 'BUILD_TIME_STRING',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xb00): 'HOST_NAME',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xb01): 'C_COMPILER_NAME',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xb02): 'C_COMPILER_VERSION',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xb03): 'CXX_COMPILER_NAME',
|
|
self.bindesc_gen_tag(self.TYPE_STR, 0xb04): 'CXX_COMPILER_VERSION',
|
|
}
|
|
self.NAME_TO_TAG = {v: k for k, v in self.TAG_TO_NAME.items()}
|
|
|
|
super().__init__(
|
|
'bindesc',
|
|
'work with Binary Descriptors',
|
|
dedent('''
|
|
Work with Binary Descriptors - constant data objects
|
|
describing a binary image
|
|
'''))
|
|
|
|
def do_add_parser(self, parser_adder):
|
|
parser = parser_adder.add_parser(self.name,
|
|
help=self.help,
|
|
description=self.description)
|
|
|
|
subparsers = parser.add_subparsers(help='sub-command to run', required=True)
|
|
|
|
dump_parser = subparsers.add_parser('dump', help='Dump all binary descriptors in the image')
|
|
dump_parser.add_argument('file', type=str, help='Executable file')
|
|
dump_parser.add_argument('--file-type', type=str, choices=self.EXTENSIONS, help='File type')
|
|
dump_parser.add_argument('-b', '--big-endian', action='store_true',
|
|
help='Target CPU is big endian')
|
|
dump_parser.set_defaults(subcmd='dump', big_endian=False)
|
|
|
|
search_parser = subparsers.add_parser('search', help='Search for a specific descriptor')
|
|
search_parser.add_argument('descriptor', type=str, help='Descriptor name')
|
|
search_parser.add_argument('file', type=str, help='Executable file')
|
|
search_parser.add_argument('--file-type', type=str, choices=self.EXTENSIONS, help='File type')
|
|
search_parser.add_argument('-b', '--big-endian', action='store_true',
|
|
help='Target CPU is big endian')
|
|
search_parser.set_defaults(subcmd='search', big_endian=False)
|
|
|
|
custom_search_parser = subparsers.add_parser('custom_search',
|
|
help='Search for a custom descriptor')
|
|
custom_search_parser.add_argument('type', type=str, choices=['UINT', 'STR', 'BYTES'],
|
|
help='Descriptor type')
|
|
custom_search_parser.add_argument('id', type=str, help='Descriptor ID in hex')
|
|
custom_search_parser.add_argument('file', type=str, help='Executable file')
|
|
custom_search_parser.add_argument('--file-type', type=str, choices=self.EXTENSIONS,
|
|
help='File type')
|
|
custom_search_parser.add_argument('-b', '--big-endian', action='store_true',
|
|
help='Target CPU is big endian')
|
|
custom_search_parser.set_defaults(subcmd='custom_search', big_endian=False)
|
|
|
|
list_parser = subparsers.add_parser('list', help='List all known descriptors')
|
|
list_parser.set_defaults(subcmd='list', big_endian=False)
|
|
|
|
return parser
|
|
|
|
def dump(self, args):
|
|
image = self.get_image_data(args.file)
|
|
|
|
descriptors = self.parse_descriptors(image)
|
|
for tag, value in descriptors.items():
|
|
if tag in self.TAG_TO_NAME:
|
|
tag = self.TAG_TO_NAME[tag]
|
|
log.inf(f'{tag}', self.bindesc_repr(value))
|
|
|
|
def list(self, args):
|
|
for tag in self.TAG_TO_NAME.values():
|
|
log.inf(f'{tag}')
|
|
|
|
def common_search(self, args, search_term):
|
|
image = self.get_image_data(args.file)
|
|
|
|
descriptors = self.parse_descriptors(image)
|
|
|
|
if search_term in descriptors:
|
|
value = descriptors[search_term]
|
|
log.inf(self.bindesc_repr(value))
|
|
else:
|
|
log.die('Descriptor not found')
|
|
|
|
def search(self, args):
|
|
try:
|
|
search_term = self.NAME_TO_TAG[args.descriptor]
|
|
except KeyError:
|
|
log.die(f'Descriptor {args.descriptor} is invalid')
|
|
|
|
self.common_search(args, search_term)
|
|
|
|
def custom_search(self, args):
|
|
custom_type = {
|
|
'STR': self.TYPE_STR,
|
|
'UINT': self.TYPE_UINT,
|
|
'BYTES': self.TYPE_BYTES
|
|
}[args.type]
|
|
custom_tag = self.bindesc_gen_tag(custom_type, int(args.id, 16))
|
|
self.common_search(args, custom_tag)
|
|
|
|
def do_run(self, args, _):
|
|
if MISSING_REQUIREMENTS:
|
|
raise RuntimeError('one or more Python dependencies were missing; '
|
|
'see the getting started guide for details on '
|
|
'how to fix')
|
|
self.is_big_endian = args.big_endian
|
|
self.file_type = self.guess_file_type(args)
|
|
subcmd = getattr(self, args.subcmd)
|
|
subcmd(args)
|
|
|
|
def get_image_data(self, file_name):
|
|
if self.file_type == 'bin':
|
|
with open(file_name, 'rb') as bin_file:
|
|
return bin_file.read()
|
|
|
|
if self.file_type == 'hex':
|
|
return IntelHex(file_name).tobinstr()
|
|
|
|
if self.file_type == 'uf2':
|
|
with open(file_name, 'rb') as uf2_file:
|
|
return convert_from_uf2(uf2_file.read())
|
|
|
|
if self.file_type == 'elf':
|
|
with open(file_name, 'rb') as f:
|
|
elffile = ELFFile(f)
|
|
|
|
section = elffile.get_section_by_name('rom_start')
|
|
if section:
|
|
return section.data()
|
|
|
|
section = elffile.get_section_by_name('text')
|
|
if section:
|
|
return section.data()
|
|
|
|
log.die('No "rom_start" or "text" section found')
|
|
|
|
log.die('Unknown file type')
|
|
|
|
def parse_descriptors(self, image):
|
|
magic = struct.pack('>Q' if self.is_big_endian else 'Q', self.MAGIC)
|
|
index = image.find(magic)
|
|
if index == -1:
|
|
log.die('Could not find binary descriptor magic')
|
|
|
|
descriptors = {}
|
|
|
|
index += len(magic) # index points to first descriptor
|
|
current_tag = self.bytes_to_short(image[index:index+2])
|
|
while current_tag != self.DESCRIPTORS_END:
|
|
index += 2 # index points to length
|
|
length = self.bytes_to_short(image[index:index+2])
|
|
index += 2 # index points to data
|
|
data = image[index:index+length]
|
|
|
|
tag_type = self.bindesc_get_type(current_tag)
|
|
if tag_type == self.TYPE_STR:
|
|
decoded_data = data[:-1].decode('ascii')
|
|
elif tag_type == self.TYPE_UINT:
|
|
decoded_data = self.bytes_to_uint(data)
|
|
elif tag_type == self.TYPE_BYTES:
|
|
decoded_data = data
|
|
else:
|
|
log.die(f'Unknown type for tag 0x{current_tag:04x}')
|
|
|
|
key = f'0x{current_tag:04x}'
|
|
descriptors[key] = decoded_data
|
|
index += length
|
|
index = self.align(index, 4)
|
|
current_tag = self.bytes_to_short(image[index:index+2])
|
|
|
|
return descriptors
|
|
|
|
def guess_file_type(self, args):
|
|
if "file" not in args:
|
|
return None
|
|
|
|
# If file type is explicitly given, use it
|
|
if args.file_type is not None:
|
|
return args.file_type
|
|
|
|
# If the file has a known extension, use it
|
|
for extension in self.EXTENSIONS:
|
|
if args.file.endswith(f'.{extension}'):
|
|
return extension
|
|
|
|
with open(args.file, 'rb') as f:
|
|
header = f.read(1024)
|
|
|
|
# Try the elf magic
|
|
if header.startswith(b'\x7fELF'):
|
|
return 'elf'
|
|
|
|
# Try the uf2 magic
|
|
if header.startswith(b'UF2\n'):
|
|
return 'uf2'
|
|
|
|
try:
|
|
# if the file is textual it's probably hex
|
|
header.decode('ascii')
|
|
return 'hex'
|
|
except UnicodeDecodeError:
|
|
# Default to bin
|
|
return 'bin'
|
|
|
|
def bytes_to_uint(self, b):
|
|
return struct.unpack('>I' if self.is_big_endian else 'I', b)[0]
|
|
|
|
def bytes_to_short(self, b):
|
|
return struct.unpack('>H' if self.is_big_endian else 'H', b)[0]
|
|
|
|
@staticmethod
|
|
def bindesc_gen_tag(_type, _id):
|
|
return f'0x{(_type << 12 | _id):04x}'
|
|
|
|
@staticmethod
|
|
def bindesc_get_type(tag):
|
|
return tag >> 12
|
|
|
|
@staticmethod
|
|
def align(x, alignment):
|
|
return (x + alignment - 1) & (~(alignment - 1))
|
|
|
|
@staticmethod
|
|
def bindesc_repr(value):
|
|
if isinstance(value, str):
|
|
return f'"{value}"'
|
|
if isinstance(value, (int, bytes)):
|
|
return f'{value}'
|