rk: scripts: mkbootimg: support header_version and recovery_dtbo
Merge AOSP 26e01bbdc5cb ("Populate recovery DTBO offset correctly")
Change-Id: Icfc8957b5101d86a1c70bd9c6676824aaaeb75e4
Signed-off-by: Tao Huang <huangtao@rock-chips.com>
This commit is contained in:
@ -45,6 +45,22 @@ def pad_file(f, padding):
|
||||
f.write(pack(str(pad) + 'x'))
|
||||
|
||||
|
||||
def get_number_of_pages(image_size, page_size):
|
||||
"""calculates the number of pages required for the image"""
|
||||
return (image_size + page_size - 1) / page_size
|
||||
|
||||
|
||||
def get_recovery_dtbo_offset(args):
|
||||
"""calculates the offset of recovery_dtbo image in the boot image"""
|
||||
num_header_pages = 1 # header occupies a page
|
||||
num_kernel_pages = get_number_of_pages(filesize(args.kernel), args.pagesize)
|
||||
num_ramdisk_pages = get_number_of_pages(filesize(args.ramdisk), args.pagesize)
|
||||
num_second_pages = get_number_of_pages(filesize(args.second), args.pagesize)
|
||||
dtbo_offset = args.pagesize * (num_header_pages + num_kernel_pages +
|
||||
num_ramdisk_pages + num_second_pages)
|
||||
return dtbo_offset
|
||||
|
||||
|
||||
def write_header(args):
|
||||
BOOT_MAGIC = 'ANDROID!'.encode()
|
||||
args.output.write(pack('8s', BOOT_MAGIC))
|
||||
@ -57,7 +73,7 @@ def write_header(args):
|
||||
args.base + args.second_offset, # physical load addr
|
||||
args.base + args.tags_offset, # physical addr for kernel tags
|
||||
args.pagesize, # flash page size we assume
|
||||
0, # future expansion: MUST be 0
|
||||
args.header_version, # version of bootimage header
|
||||
(args.os_version << 11) | args.os_patch_level)) # os version and patch level
|
||||
args.output.write(pack('16s', args.board.encode())) # asciiz product name
|
||||
args.output.write(pack('512s', args.cmdline[:512].encode()))
|
||||
@ -66,10 +82,23 @@ def write_header(args):
|
||||
update_sha(sha, args.kernel)
|
||||
update_sha(sha, args.ramdisk)
|
||||
update_sha(sha, args.second)
|
||||
|
||||
if args.header_version > 0:
|
||||
update_sha(sha, args.recovery_dtbo)
|
||||
|
||||
img_id = pack('32s', sha.digest())
|
||||
|
||||
args.output.write(img_id)
|
||||
args.output.write(pack('1024s', args.cmdline[512:].encode()))
|
||||
|
||||
if args.header_version > 0:
|
||||
args.output.write(pack('I', filesize(args.recovery_dtbo))) # size in bytes
|
||||
if args.recovery_dtbo:
|
||||
args.output.write(pack('Q', get_recovery_dtbo_offset(args))) # recovery dtbo offset
|
||||
else:
|
||||
args.output.write(pack('Q', 0)) # Will be set to 0 for devices without a recovery dtbo
|
||||
args.output.write(pack('I', args.output.tell() + 4)) # size of boot header
|
||||
|
||||
pad_file(args.output, args.pagesize)
|
||||
return img_id
|
||||
|
||||
@ -132,6 +161,7 @@ def parse_cmdline():
|
||||
required=True)
|
||||
parser.add_argument('--ramdisk', help='path to the ramdisk', type=FileType('rb'))
|
||||
parser.add_argument('--second', help='path to the 2nd bootloader', type=FileType('rb'))
|
||||
parser.add_argument('--recovery_dtbo', help='path to the recovery DTBO', type=FileType('rb'))
|
||||
parser.add_argument('--cmdline', help='extra arguments to be passed on the '
|
||||
'kernel command line', default='', action=ValidateStrLenAction, maxlen=1536)
|
||||
parser.add_argument('--base', help='base address', type=parse_int, default=0x10000000)
|
||||
@ -150,6 +180,7 @@ def parse_cmdline():
|
||||
choices=[2**i for i in range(11,15)], default=2048)
|
||||
parser.add_argument('--id', help='print the image ID on standard output',
|
||||
action='store_true')
|
||||
parser.add_argument('--header_version', help='boot image header version', type=parse_int, default=0)
|
||||
parser.add_argument('-o', '--output', help='output file name', type=FileType('wb'),
|
||||
required=True)
|
||||
return parser.parse_args()
|
||||
@ -160,6 +191,8 @@ def write_data(args):
|
||||
write_padded_file(args.output, args.ramdisk, args.pagesize)
|
||||
write_padded_file(args.output, args.second, args.pagesize)
|
||||
|
||||
if args.header_version > 0:
|
||||
write_padded_file(args.output, args.recovery_dtbo, args.pagesize)
|
||||
|
||||
def main():
|
||||
args = parse_cmdline()
|
||||
|
||||
Reference in New Issue
Block a user