78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import sys
|
|
import os
|
|
import subprocess
|
|
from construct import *
|
|
from elftools.elf.elffile import ELFFile
|
|
from elftools.elf.constants import SH_FLAGS
|
|
|
|
prekernel_path = sys.argv[1]
|
|
kernel_path = sys.argv[2]
|
|
devtree_path = sys.argv[3]
|
|
out_path = sys.argv[4]
|
|
build_dir_path = os.path.dirname(os.path.abspath(sys.argv[4]))
|
|
tmp_dir_path = build_dir_path + "/tmp"
|
|
rawimage_header_path = tmp_dir_path + "/rawimage_header.bin"
|
|
|
|
# GNU-LD can't convert ELF object format to Binary on the fly.
|
|
# Checking if output is an ELF and convertining it to binary.
|
|
prekernel_header = []
|
|
with open(prekernel_path, "rb") as f:
|
|
prekernel_header = f.read(4)
|
|
|
|
if prekernel_header == b'\x7fELF':
|
|
kernel_vaddr = 0
|
|
kernel_vaddr_end = 0
|
|
|
|
with open(prekernel_path, 'rb') as elffile:
|
|
signature_section = None
|
|
for section in ELFFile(elffile).iter_sections():
|
|
if (section['sh_flags'] & SH_FLAGS.SHF_ALLOC) == SH_FLAGS.SHF_ALLOC:
|
|
kernel_vaddr_end = max(
|
|
kernel_vaddr_end, section['sh_addr'] + section['sh_size'])
|
|
|
|
output = subprocess.check_output(
|
|
"aarch64-elf-objcopy -O binary {0} {0}".format(prekernel_path), shell=True)
|
|
binary_blob_size = os.path.getsize(prekernel_path)
|
|
assert(binary_blob_size <= kernel_vaddr_end)
|
|
|
|
need_to_append = kernel_vaddr_end - binary_blob_size
|
|
with open(prekernel_path, 'r+b') as file:
|
|
_ = file.read()
|
|
padding = [0x0 for _ in range(need_to_append)]
|
|
file.write(bytearray(padding))
|
|
|
|
|
|
prekernel_size = os.path.getsize(prekernel_path)
|
|
rawimage_header_size = 8 * 8
|
|
kernel_size = os.path.getsize(kernel_path)
|
|
devtree_size = os.path.getsize(devtree_path)
|
|
ramdisk_size = 0
|
|
|
|
result = {
|
|
"kern_off": rawimage_header_size + prekernel_size,
|
|
"kern_size": kernel_size,
|
|
"devtree_off": rawimage_header_size + prekernel_size + kernel_size,
|
|
"devtree_size": devtree_size,
|
|
"ramdisk_off": 0,
|
|
"ramdisk_size": 0,
|
|
"rawimage_size": rawimage_header_size + prekernel_size + kernel_size + devtree_size + ramdisk_size,
|
|
"padding": 0,
|
|
}
|
|
|
|
rawimage_header = Struct(
|
|
"kern_off" / Int64ul,
|
|
"kern_size" / Int64ul,
|
|
"devtree_off" / Int64ul,
|
|
"devtree_size" / Int64ul,
|
|
"ramdisk_off" / Int64ul,
|
|
"ramdisk_size" / Int64ul,
|
|
"rawimage_size" / Int64ul,
|
|
"padding" / Int64ul,
|
|
).build(result)
|
|
|
|
with open(rawimage_header_path, "wb") as binfile:
|
|
binfile.write(bytes(rawimage_header))
|
|
|
|
output = subprocess.check_output("cat {0} {1} {2} {3} > {4}".format(
|
|
prekernel_path, rawimage_header_path, kernel_path, devtree_path, out_path), shell=True)
|