From cd96473fd3f86415dd243b6ea1f7d3b9ded582ec Mon Sep 17 00:00:00 2001 From: Valerio Setti Date: Mon, 8 Apr 2024 13:44:41 +0200 Subject: [PATCH] tests: update Makefile to generate tests/src/test_keys.h Signed-off-by: Valerio Setti --- tests/data_files/Makefile | 3 +++ tests/scripts/generate_test_keys.py | 15 +++++++++++---- 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/tests/data_files/Makefile b/tests/data_files/Makefile index bbbfa9cd9c..3dab20eb3c 100644 --- a/tests/data_files/Makefile +++ b/tests/data_files/Makefile @@ -2184,6 +2184,9 @@ TEST_CERTS_H_INPUT_FILES=test-ca2.crt \ --string TEST_CLI_KEY_RSA_PEM=cli-rsa.key \ --binary TEST_CLI_KEY_RSA_DER=cli-rsa.key.der +../src/test_keys.h: ../scripts/generate_test_keys.py + ../scripts/generate_test_keys.py --output $@ + ################################################################ #### Diffie-Hellman parameters ################################################################ diff --git a/tests/scripts/generate_test_keys.py b/tests/scripts/generate_test_keys.py index 0a67a784cb..9920933692 100755 --- a/tests/scripts/generate_test_keys.py +++ b/tests/scripts/generate_test_keys.py @@ -9,6 +9,7 @@ generating the required key at run time. This helps speeding up testing.""" import os from typing import Iterator import re +import argparse import scripts_path # pylint: disable=unused-import from mbedtls_dev.asymmetric_key_data import ASYMMETRIC_KEY_DATA @@ -90,11 +91,17 @@ def get_look_up_table_entry(key_type: str, group_id_or_keybits: str, yield " {0}, sizeof({0}) }},".format(pub_array_name) def main() -> None: - # Remove output file if already existing. - if os.path.exists(OUTPUT_HEADER_FILE): - os.remove(OUTPUT_HEADER_FILE) + argparser = argparse.ArgumentParser() + argparser.add_argument("--output", required=True, help="Output file") + args = argparser.parse_args() - output_file = open(OUTPUT_HEADER_FILE, 'at') + output_file = args.output + # Remove output file if already existing. + if os.path.exists(output_file): + print("Warning: {} already existing, it will be overwritten.", output_file) + os.remove(output_file) + + output_file = open(output_file, 'at') output_file.write( "/*********************************************************************************\n" + " * This file was automatically generated from tests/scripts/generate_test_keys.py.\n" +