mirror of
https://github.com/UpsilonNumworks/Upsilon.git
synced 2026-03-26 17:20:53 +01:00
67 lines
2.8 KiB
Python
67 lines
2.8 KiB
Python
#coding=utf-8
|
|
# Translate ELF file into DfuSe file
|
|
|
|
import sys
|
|
import struct
|
|
import zlib # for CRC-32 calculation
|
|
import subprocess
|
|
import re
|
|
import argparse
|
|
|
|
# arm-none-eabi-objdump -h -w file.elf
|
|
# arm-none-eabi-objcopy -O binary -j .data file.elf file.bin
|
|
|
|
def loadable_sections(elf_file):
|
|
objdump_section_headers_pattern = re.compile("^\s+\d+\s+(\.[\w\.]+)\s+([0-9a-f]+)\s+([0-9a-f]+)\s+([0-9a-f]+)\s+([0-9a-f]+).*LOAD", flags=re.MULTILINE)
|
|
objdump_output = subprocess.check_output(["arm-none-eabi-objdump", "-h", "-w", elf_file])
|
|
sections = []
|
|
for (name, size, vma, lma, offset) in re.findall(objdump_section_headers_pattern, objdump_output):
|
|
int_size = int(size,16)
|
|
if (int_size > 0):
|
|
sections.append({'name': name, 'size': int_size, 'vma': int(vma,16), 'lma': int(lma,16), 'offset':int(offset,16)})
|
|
return sections
|
|
|
|
def generate_dfu_file(targets, usb_vid_pid, dfu_file):
|
|
data = b''
|
|
for t, target in enumerate(targets):
|
|
target_data = b''
|
|
for image in target:
|
|
# Pad the image to 8 bytes, this seems to be needed
|
|
pad = (8 - len(image['data']) % 8 ) % 8
|
|
image['data'] = image['data'] + bytes(bytearray(8)[0:pad])
|
|
target_data += struct.pack('<2I', image['address'], len(image['data'])) + image['data']
|
|
target_data = struct.pack('<6sBI255s2I', b'Target', 0, 1, b'ST...', len(target_data), len(target)) + target_data
|
|
data += target_data
|
|
data = struct.pack('<5sBIB', b'DfuSe', 1, len(data) + 11, len(targets)) + data
|
|
v, d = map(lambda x: int(x,0) & 0xFFFF, usb_vid_pid.split(':'))
|
|
data += struct.pack('<4H3sB', 0, d, v, 0x011a, b'UFD', 16)
|
|
crc = (0xFFFFFFFF & -zlib.crc32(data) - 1)
|
|
data += struct.pack('<I', crc)
|
|
open(dfu_file, 'wb').write(data)
|
|
|
|
def bin_file(section):
|
|
return "section" + section['name'] + ".bin"
|
|
|
|
def print_sections(sections):
|
|
for s in sections:
|
|
print("%s-%s: %s, %s" % (hex(s['lma']), hex(s['lma'] + s['size'] - 1), s['name'], "{:,} bytes".format(s['size'])))
|
|
|
|
def elf2dfu(elf_file, usb_vid_pid, dfu_file, verbose):
|
|
sections = loadable_sections(elf_file)
|
|
if verbose:
|
|
print_sections(sections)
|
|
for section in sections:
|
|
subprocess.call(["arm-none-eabi-objcopy", "-O", "binary", "-j", section['name'], elf_file, bin_file(section)])
|
|
targets = map(lambda s: {'address': s['lma'], 'name': s['name'], 'data': open(bin_file(s)).read()}, sections)
|
|
generate_dfu_file([targets], usb_vid_pid, dfu_file)
|
|
for section in sections:
|
|
subprocess.call(["rm", bin_file(section)])
|
|
|
|
parser = argparse.ArgumentParser(description="Convert an ELF file to DfuSe.")
|
|
parser.add_argument('elf_file', help='Input ELF file')
|
|
parser.add_argument('dfu_file', help='Output DfuSe file')
|
|
parser.add_argument('-v', '--verbose', action="store_true", help='Show verbose output')
|
|
|
|
args = parser.parse_args()
|
|
elf2dfu(args.elf_file, "0x0483:0xa291", args.dfu_file, args.verbose)
|