You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.6 KiB
66 lines
1.6 KiB
8 years ago
|
#!/usr/bin/env python3
|
||
|
import sys
|
||
|
import struct
|
||
|
import binascii
|
||
|
|
||
6 years ago
|
|
||
8 years ago
|
# encode vendor name, add length byte and padding to multiple of 4
|
||
8 years ago
|
def encode_vendor(vname):
|
||
7 years ago
|
vbin = vname.encode()
|
||
6 years ago
|
vbin = struct.pack("<B", len(vbin)) + vbin
|
||
|
vbin += b"\0" * (-len(vbin) & 3)
|
||
8 years ago
|
return vbin
|
||
|
|
||
7 years ago
|
|
||
8 years ago
|
def encode_pubkey(pubkey):
|
||
|
if len(pubkey) != 64:
|
||
6 years ago
|
raise Exception("Wrong public key length")
|
||
8 years ago
|
return binascii.unhexlify(pubkey)
|
||
8 years ago
|
|
||
7 years ago
|
|
||
7 years ago
|
def decode_vtrust(vtrust):
|
||
|
t = 0xFFFF
|
||
|
for i, b in enumerate(reversed(vtrust)):
|
||
6 years ago
|
if b != ".":
|
||
7 years ago
|
t &= ~(1 << i)
|
||
|
return t
|
||
|
|
||
|
|
||
8 years ago
|
def main():
|
||
|
if len(sys.argv) < 7:
|
||
6 years ago
|
print(
|
||
|
'Usage build_vendorheader "pubkey1hex:pubkey2hex:..." m version vendortrust vendorname vendorimage.toif vendorheader.bin'
|
||
|
)
|
||
8 years ago
|
return 1
|
||
|
|
||
6 years ago
|
keys = [encode_pubkey(x) for x in sys.argv[1].split(":")]
|
||
8 years ago
|
m = int(sys.argv[2])
|
||
6 years ago
|
(vmajor, vminor) = [int(x) for x in sys.argv[3].split(".")]
|
||
7 years ago
|
vtrust = decode_vtrust(sys.argv[4])
|
||
7 years ago
|
vname = sys.argv[5]
|
||
|
ifn = sys.argv[6]
|
||
|
ofn = sys.argv[7]
|
||
6 years ago
|
if not ifn.endswith(".toif"):
|
||
|
print("Must provide TOIF file")
|
||
8 years ago
|
return 2
|
||
|
|
||
8 years ago
|
expiry = 0
|
||
6 years ago
|
vheader = b"TRZV" + struct.pack(
|
||
|
"<IIBBBBH", 0, expiry, vmajor, vminor, m, len(keys), vtrust
|
||
|
)
|
||
|
vheader += 14 * b"\0"
|
||
8 years ago
|
for k in keys:
|
||
|
vheader += k
|
||
6 years ago
|
vheader += encode_vendor(vname) + open(ifn, "rb").read()
|
||
8 years ago
|
padding = 65 + (-len(vheader) - 65) & 511
|
||
6 years ago
|
vheader += b"\0" * padding
|
||
8 years ago
|
|
||
|
# put in length
|
||
6 years ago
|
vheader = vheader[0:4] + struct.pack("<I", len(vheader)) + vheader[8:]
|
||
8 years ago
|
|
||
6 years ago
|
with open(ofn, "wb") as f:
|
||
8 years ago
|
f.write(vheader)
|
||
|
|
||
7 years ago
|
|
||
8 years ago
|
main()
|