Files
device_xiaomi_rosemary/dtbo/append_certs.py
Kshitij Gupta eb5043a68f rosemary: Add custom dtbo mk rules and append certs
- rosemary's LK appears to be checking for certs added to dtbo during
  signing. We unfortunately do not have the private keys needed to
  generate an image that the bootloader would accept.
- dd out the 2 der certs and append them appropriately with some padding
  to make the bootloader happy. (Extracted from MIUI 12 dtbo.img)

Change-Id: I26402236bcea3516fb98f5d48857106ae30b54c1
Signed-off-by: Kshitij Gupta <kshitijgm@gmail.com>
2023-04-25 17:59:38 +02:00

43 lines
1.3 KiB
Python
Executable File

#!/usr/bin/env python
import argparse
from os import stat
def padding_file(input_file, align_num):
"""Fill 0 to make input_file's size a multiple of align_num."""
filesize = stat(input_file).st_size
file1 = open(input_file, 'ab+')
padding = filesize % align_num
if padding != 0:
padding = align_num - padding
for _ in range(padding):
file1.write("\x00")
file1.close()
def append_file(img_file, cert_file, alignment):
"""Append provided cert_file to the end of target img_file."""
padding_file(img_file, alignment)
file1 = open(img_file, 'ab+')
file2 = open(cert_file, 'rb')
file1.write(file2.read())
file1.close()
file2.close()
def main():
parser = argparse.ArgumentParser(description='Sign an image with provided der certs')
parser.add_argument('--alignment', type=int, help='Alignment for calculating padding')
parser.add_argument('--cert1', type=str, help='Certificate - cert1.der')
parser.add_argument('--cert2', type=str, help='Certificate - cert2.der')
parser.add_argument('--dtbo', type=str, help='Path to the DTBO img')
args = parser.parse_args()
append_file(args.dtbo, args.cert1, args.alignment)
append_file(args.dtbo, args.cert2, args.alignment)
if __name__ == '__main__':
main()