| # Copyright 2018,2020 NXP |
| # |
| # SPDX-License-Identifier: Apache-2.0 |
| # |
| |
| """ |
| Convert files that we get from OEF to header files |
| |
| """ |
| |
| import glob |
| import os |
| import sys |
| import time |
| |
| LOCAL_TIME = time.localtime() |
| DATETIME_STR = time.strftime("%Y:%m:%d", LOCAL_TIME) + "_" + time.strftime("%H:%M", LOCAL_TIME) |
| |
| AUTO_GEN_INFO = """This file is auto generated (%s) using: |
| <simw-top>/scripts/parse_oef_keys.py""" % (DATETIME_STR,) |
| |
| USAGE_INFO = """Usage info: |
| python parse_oef_keys.py <directory_path> <header> <footer> <macro_prefix> <output_filename> |
| |
| directory_path => Keys folder path. |
| header => header for the c file. Optional argument. |
| footer => footer for the c file. Optional argument. |
| macro_prefix => Prefix of the macro. Optional argument. default: SSS_AUTH_SE050_OEF. |
| output_filename => Output file name for c and python file. Optional argument. |
| default: ex_sss_tp_scp03_keys.h, tp_scp03_keys.py""" |
| |
| DEFAULT_MACRO_PREFIX = "SSS_AUTH_SE050_OEF" |
| |
| LICENSE_TEXT = """ Copyright 2019,2020 NXP |
| |
| This software is owned or controlled by NXP and may only be used |
| strictly in accordance with the applicable license terms. By expressly |
| accepting such terms or by downloading, installing, activating and/or |
| otherwise using the software, you are agreeing that you have read, and |
| that you agree to comply with and are bound by, such license terms. If |
| you do not agree to be bound by the applicable license terms, then you |
| may not retain, install, activate or otherwise use the software.""" |
| |
| CLANG_OFF = """/* ************************************************************************** */ |
| /* Includes */ |
| /* ************************************************************************** */ |
| |
| /* ************************************************************************** */ |
| /* Defines */ |
| /* ************************************************************************** */ |
| |
| /* clang-format off */""" |
| |
| CLANG_ON = """/* clang-format on */ |
| |
| /* ************************************************************************** */ |
| /* Structures and Typedefs */ |
| /* ************************************************************************** */ |
| |
| /* ************************************************************************** */ |
| /* Global Variables */ |
| /* ************************************************************************** */ |
| |
| /* ************************************************************************** */ |
| /* Functions */ |
| /* ************************************************************************** */""" |
| |
| DEFAULT_OUTPUT_FILE_NAME = "tp_scp03_keys" |
| |
| |
| def read_data(filename): |
| """ |
| Read key value from the file |
| :param filename: file name to read key |
| :return: Key value |
| """ |
| f = open(filename, 'r') |
| file_data = f.read() |
| key_data = "" |
| for i in range(0, len(file_data), 2): |
| if i == 0: |
| key_data += "0x" |
| else: |
| key_data += ", 0x" |
| key_data += file_data[i:i + 2] |
| f.close() |
| return key_data |
| |
| |
| def write_data_to_c(key_data_dict, output_file): |
| """ |
| Write ``output_file`` with keys. |
| :param key_data_dict: Key values |
| :param output_file: Destination c file to write key value |
| :return: None |
| """ |
| |
| if len(sys.argv) >= 5: |
| |
| if os.path.isfile(sys.argv[2]): |
| header_file = open(sys.argv[2], 'r') |
| header_data = header_file.read() |
| header_file.close() |
| else: |
| header_data = sys.argv[2] |
| |
| if os.path.isfile(sys.argv[3]): |
| footer_file = open(sys.argv[3], 'r') |
| footer_data = footer_file.read() |
| footer_file.close() |
| else: |
| footer_data = sys.argv[3] |
| macro = sys.argv[4] |
| else: |
| macro = DEFAULT_MACRO_PREFIX |
| timestamp_str = time.strftime("_%Y%m%d_", LOCAL_TIME) + \ |
| time.strftime("%H%M_", LOCAL_TIME) |
| header_data = "#ifndef %s\n#define %s\n" % (macro + timestamp_str, |
| macro + timestamp_str,) |
| footer_data = "#endif /* %s */" % (macro + timestamp_str,) |
| |
| usage_info = AUTO_GEN_INFO.replace("\n", "\n * ") |
| usage_info += "\n *\n * " |
| usage_info += USAGE_INFO.replace("\n", "\n * ") |
| license_info = "/*" + LICENSE_TEXT.replace("\n", "\n *") + "\n */" |
| data_to_write = license_info + "\n\n\n/* " + usage_info + "\n */\n\n\n" \ |
| + header_data + "\n" + CLANG_OFF + "\n\n" |
| |
| for key_data in key_data_dict: |
| data_to_write += "#define " + macro + "_" + str( |
| key_data) + " " * 5 + "\\\n" + " " * 4 + "{" + " " * 59 + "\\\n" + " " * 8 |
| key_data_len = len(key_data_dict[key_data]) |
| data_to_write += key_data_dict[key_data][ |
| 0:int(key_data_len / 2)] + " " * 9 + "\\\n" + " " * 8 |
| data_to_write += key_data_dict[key_data][(int(key_data_len / 2) + 1):] + " " * 8 |
| data_to_write += " \\\n }\n\n" |
| |
| data_to_write += CLANG_ON + "\n\n" + footer_data |
| |
| out_file = open(output_file, 'w') |
| out_file.write(data_to_write) |
| out_file.close() |
| |
| |
| def write_data_to_python(key_data_dict, output_file_name): |
| """ |
| Append keys to the output file |
| :param key_data_dict: Key values |
| :param output_file_name: Destination python file to write key value |
| :return: None |
| """ |
| |
| if len(sys.argv) >= 3: |
| macro = sys.argv[2] |
| else: |
| macro = DEFAULT_MACRO_PREFIX |
| |
| license_info = "#" + LICENSE_TEXT.replace("\n", "\n#") + "\n#\n\n\n" |
| data_to_write = license_info |
| python_usage_info = "# " + AUTO_GEN_INFO.replace("\n", "\n# ") |
| python_usage_info += "\n#\n# " |
| python_usage_info += USAGE_INFO.replace("\n", "\n# ") |
| data_to_write += python_usage_info + "\n\n\n" |
| |
| for key_data in key_data_dict: |
| data_to_write += macro + "_" + str(key_data) + " = (\n " |
| key_data_len = len(key_data_dict[key_data]) |
| data_to_write += key_data_dict[key_data][0:int(key_data_len / 2)] + "\n " |
| data_to_write += key_data_dict[key_data][(int(key_data_len / 2) + 1):] |
| data_to_write += ",\n )\n\n" |
| |
| out_file = open(output_file_name, 'w') |
| out_file.write(data_to_write) |
| out_file.close() |
| |
| |
| def main(): # pylint: disable=too-many-locals |
| """ |
| Parse keys from input directory and write in C and python files. |
| :return: None |
| """ |
| keys_dir = sys.argv[1] |
| enc_keys_file_list = list() |
| mac_keys_file_list = list() |
| dec_keys_file_list = list() |
| key_file_list = glob.glob(keys_dir + os.sep + "*.key") |
| for key_file in key_file_list: |
| if "_AuthKey_" in key_file: |
| enc_keys_file_list.append(key_file) |
| elif "_MacKey_" in key_file: |
| mac_keys_file_list.append(key_file) |
| elif "_EncKey_" in key_file: |
| dec_keys_file_list.append(key_file) |
| |
| key_data = {} |
| for enc_file in enc_keys_file_list: |
| match_key = enc_file.split("_AuthKey_", 2)[1].split("_NXP", 2)[0] |
| key_data[match_key + "_KEY_ENC"] = read_data(enc_file) |
| for mac_file in [x for x in mac_keys_file_list if match_key in x]: |
| key_data[match_key + "_KEY_MAC"] = read_data(mac_file) |
| for dec_file in [x for x in dec_keys_file_list if match_key in x]: |
| key_data[match_key + "_KEY_DEK"] = read_data(dec_file) |
| |
| c_output_dir = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", |
| "sss", "ex", "inc") |
| python_output_dir = os.path.join(os.path.abspath(os.path.dirname(__file__)), |
| "..", "pycli", "src", "sss") |
| |
| if len(sys.argv) >= 6: |
| output_file_name = sys.argv[5] |
| if output_file_name[-2:] == ".h": |
| c_output_file = output_file_name |
| python_output_file = output_file_name[:-2] + ".py" |
| elif output_file_name[-3:] == ".py": |
| c_output_file = output_file_name[:-3] + ".h" |
| python_output_file = output_file_name |
| else: |
| c_output_file = output_file_name + ".h" |
| python_output_file = output_file_name + ".py" |
| else: |
| c_output_file = "ex_sss_" + DEFAULT_OUTPUT_FILE_NAME + ".h" |
| python_output_file = DEFAULT_OUTPUT_FILE_NAME + ".py" |
| |
| write_data_to_c(key_data, c_output_dir + os.sep + c_output_file) |
| write_data_to_python(key_data, python_output_dir + os.sep + python_output_file) |
| |
| |
| def usage(): |
| """ |
| Display usage information |
| :return: None |
| """ |
| print(USAGE_INFO) |
| |
| |
| if __name__ == "__main__": |
| if len(sys.argv) >= 2: |
| main() |
| else: |
| usage() |
