111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
|
import sys
|
||
|
|
import os
|
||
|
|
import struct
|
||
|
|
script_folder, script_name = os.path.split(os.path.realpath(__file__))
|
||
|
|
sys.path.append(os.path.join(script_folder, "lib"))
|
||
|
|
import gfh
|
||
|
|
import cert
|
||
|
|
|
||
|
|
def get_file_sizeb(file_path):
|
||
|
|
if not os.path.isfile(file_path):
|
||
|
|
return 0
|
||
|
|
file_handle = open(file_path, "rb")
|
||
|
|
file_handle.seek(0, 2)
|
||
|
|
file_size = file_handle.tell()
|
||
|
|
file_handle.close()
|
||
|
|
return file_size
|
||
|
|
|
||
|
|
def concatb(file1_path, file2_path):
|
||
|
|
file1_size = get_file_sizeb(file1_path)
|
||
|
|
file2_size = get_file_sizeb(file2_path)
|
||
|
|
file1 = open(file1_path, "ab+")
|
||
|
|
file2 = open(file2_path, "rb")
|
||
|
|
file1.write(file2.read(file2_size))
|
||
|
|
file2.close()
|
||
|
|
file1.close()
|
||
|
|
|
||
|
|
class tool_auth:
|
||
|
|
def __init__(self, out_path, tool_auth_path):
|
||
|
|
self.m_out_path = out_path
|
||
|
|
if not os.path.exists(self.m_out_path):
|
||
|
|
os.makedirs(self.m_out_path)
|
||
|
|
self.m_tool_auth_path = tool_auth_path
|
||
|
|
self.m_gfh = gfh.image_gfh()
|
||
|
|
self.m_sig_handler = None
|
||
|
|
def create_gfh(self, gfh_config):
|
||
|
|
self.m_gfh.load_ini(gfh_config)
|
||
|
|
return
|
||
|
|
def sign(self, key_ini_path):
|
||
|
|
#tool auth contains only gfh and signature, no extra content
|
||
|
|
self.m_gfh.finalize(0, key_ini_path)
|
||
|
|
#write tbs_tool_auth
|
||
|
|
tbs_toolauth_file_path = os.path.join(self.m_out_path, "tbs_toolauth.bin")
|
||
|
|
tbs_tool_auth_file = open(tbs_toolauth_file_path, "wb")
|
||
|
|
tbs_tool_auth_file.write(self.m_gfh.pack())
|
||
|
|
tbs_tool_auth_file.close()
|
||
|
|
print "===tool_auth signing==="
|
||
|
|
if self.m_gfh.get_sig_type() == "SINGLE":
|
||
|
|
self.m_sig_handler = cert.sig_single(self.m_gfh.get_pad_type())
|
||
|
|
self.m_sig_handler.set_out_path(self.m_out_path)
|
||
|
|
self.m_sig_handler.create(key_ini_path, tbs_toolauth_file_path)
|
||
|
|
self.m_sig_handler.sign()
|
||
|
|
sig_name = "toolauth.sig"
|
||
|
|
sig_file_path = os.path.join(self.m_out_path, sig_name)
|
||
|
|
self.m_sig_handler.output(self.m_out_path, sig_name)
|
||
|
|
#create final toolauth file
|
||
|
|
if os.path.isfile(self.m_tool_auth_path):
|
||
|
|
os.remove(self.m_tool_auth_path)
|
||
|
|
concatb(self.m_tool_auth_path, tbs_toolauth_file_path)
|
||
|
|
concatb(self.m_tool_auth_path, sig_file_path)
|
||
|
|
else:
|
||
|
|
print "unknown signature type"
|
||
|
|
#clean up
|
||
|
|
os.remove(tbs_toolauth_file_path)
|
||
|
|
os.remove(sig_file_path)
|
||
|
|
return
|
||
|
|
|
||
|
|
def main():
|
||
|
|
#parameter parsing
|
||
|
|
idx = 1
|
||
|
|
key_ini_path = ""
|
||
|
|
gfh_config_ini_path = ""
|
||
|
|
while idx < len(sys.argv):
|
||
|
|
if sys.argv[idx][0] == '-':
|
||
|
|
if sys.argv[idx][1] == 'i':
|
||
|
|
print "key: " + sys.argv[idx + 1]
|
||
|
|
key_ini_path = sys.argv[idx + 1]
|
||
|
|
idx += 2
|
||
|
|
elif sys.argv[idx][1] == 'g':
|
||
|
|
print "gfh: " + sys.argv[idx + 1]
|
||
|
|
gfh_config_ini_path = sys.argv[idx + 1]
|
||
|
|
idx += 2
|
||
|
|
else:
|
||
|
|
print "unknown input"
|
||
|
|
idx += 2
|
||
|
|
else:
|
||
|
|
tool_auth_path = sys.argv[idx]
|
||
|
|
print "tool_auth_path: " + tool_auth_path
|
||
|
|
idx += 1
|
||
|
|
|
||
|
|
if not key_ini_path:
|
||
|
|
print "key path is not given!"
|
||
|
|
return -1
|
||
|
|
if not gfh_config_ini_path:
|
||
|
|
print "gfh config path is not given!"
|
||
|
|
return -1
|
||
|
|
if not tool_auth_path:
|
||
|
|
print "tool_auth path is not given!"
|
||
|
|
return -1
|
||
|
|
|
||
|
|
out_path = os.path.dirname(os.path.abspath(tool_auth_path))
|
||
|
|
|
||
|
|
tool_auth_obj = tool_auth(out_path, tool_auth_path)
|
||
|
|
tool_auth_obj.create_gfh(gfh_config_ini_path)
|
||
|
|
tool_auth_obj.sign(key_ini_path)
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|
||
|
|
|