diff options
Diffstat (limited to 'onlineupdate/source/update/updater/gen_cert_header.py')
-rwxr-xr-x | onlineupdate/source/update/updater/gen_cert_header.py | 33 |
1 files changed, 14 insertions, 19 deletions
diff --git a/onlineupdate/source/update/updater/gen_cert_header.py b/onlineupdate/source/update/updater/gen_cert_header.py index 5fff9be79084..5b70758db3f2 100755 --- a/onlineupdate/source/update/updater/gen_cert_header.py +++ b/onlineupdate/source/update/updater/gen_cert_header.py @@ -1,27 +1,22 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 import sys import binascii -def file_byte_generator(filename, block_size = 512): - with open(filename, "rb") as f: - while True: - block = f.read(block_size) - if block: - for byte in block: - yield byte - else: - break +def file_byte_generator(filename): + with open(filename, "rb") as f: + block = f.read() + return block def create_header(array_name, in_filename): - hexified = ["0x" + binascii.hexlify(byte) for byte in file_byte_generator(in_filename)] - print("const uint8_t " + array_name + "[] = {") - print(", ".join(hexified)) - print("};") - return 0 + hexified = ["0x" + binascii.hexlify(bytes([inp])).decode('ascii') for inp in file_byte_generator(in_filename)] + print("const uint8_t " + array_name + "[] = {") + print(", ".join(hexified)) + print("};") + return 0 if __name__ == '__main__': - if len(sys.argv) < 3: - print('ERROR: usage: gen_cert_header.py array_name in_filename') - sys.exit(1); - sys.exit(create_header(sys.argv[1], sys.argv[2])) + if len(sys.argv) < 3: + print('ERROR: usage: gen_cert_header.py array_name in_filename') + sys.exit(1); + sys.exit(create_header(sys.argv[1], sys.argv[2])) |