mirror of
https://github.com/Mbed-TLS/mbedtls-framework.git
synced 2026-10-05 08:17:22 +00:00
Gilles Peskine
2e1c8c9510
This fixes an error when `tests/Makefile` in Mbed TLS invokes `../framwework/scripts/generate_test_cert_macros.py`: this needs paths that are either absolute or relative to the current directory (which is `tests`), rather than relative to the repository root. Signed-off-by: Gilles Peskine <[email protected]>
112 lines
4.2 KiB
Python
Executable File
112 lines
4.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Generate `tests/src/test_certs.h` which includes certficaties/keys/certificate list for testing.
|
|
"""
|
|
|
|
#
|
|
# Copyright The Mbed TLS Contributors
|
|
# SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
|
|
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import jinja2
|
|
from mbedtls_framework.build_tree import guess_project_root
|
|
|
|
TESTS_DIR = os.path.join(guess_project_root(), 'tests')
|
|
DATA_DIR_RELATIVE = 'framework/data_files'
|
|
DATA_DIR_ABSOLUTE = os.path.join(guess_project_root(), DATA_DIR_RELATIVE)
|
|
|
|
INPUT_ARGS = [
|
|
("string", "TEST_CA_CRT_EC_PEM", "test-ca2.crt"),
|
|
("binary", "TEST_CA_CRT_EC_DER", "test-ca2.crt.der"),
|
|
("string", "TEST_CA_KEY_EC_PEM", "test-ca2.key.enc"),
|
|
("password", "TEST_CA_PWD_EC_PEM", "PolarSSLTest"),
|
|
("binary", "TEST_CA_KEY_EC_DER", "test-ca2.key.der"),
|
|
("string", "TEST_CA_CRT_RSA_SHA256_PEM", "test-ca-sha256.crt"),
|
|
("binary", "TEST_CA_CRT_RSA_SHA256_DER", "test-ca-sha256.crt.der"),
|
|
("string", "TEST_CA_CRT_RSA_SHA1_PEM", "test-ca-sha1.crt"),
|
|
("binary", "TEST_CA_CRT_RSA_SHA1_DER", "test-ca-sha1.crt.der"),
|
|
("string", "TEST_CA_KEY_RSA_PEM", "test-ca.key"),
|
|
("password", "TEST_CA_PWD_RSA_PEM", "PolarSSLTest"),
|
|
("binary", "TEST_CA_KEY_RSA_DER", "test-ca.key.der"),
|
|
("string", "TEST_SRV_CRT_EC_PEM", "server5.crt"),
|
|
("binary", "TEST_SRV_CRT_EC_DER", "server5.crt.der"),
|
|
("string", "TEST_SRV_KEY_EC_PEM", "server5.key"),
|
|
("binary", "TEST_SRV_KEY_EC_DER", "server5.key.der"),
|
|
("string", "TEST_SRV_CRT_RSA_SHA256_PEM", "server2-sha256.crt"),
|
|
("binary", "TEST_SRV_CRT_RSA_SHA256_DER", "server2-sha256.crt.der"),
|
|
("string", "TEST_SRV_CRT_RSA_SHA1_PEM", "server2.crt"),
|
|
("binary", "TEST_SRV_CRT_RSA_SHA1_DER", "server2.crt.der"),
|
|
("string", "TEST_SRV_KEY_RSA_PEM", "server2.key"),
|
|
("binary", "TEST_SRV_KEY_RSA_DER", "server2.key.der"),
|
|
("string", "TEST_CLI_CRT_EC_PEM", "cli2.crt"),
|
|
("binary", "TEST_CLI_CRT_EC_DER", "cli2.crt.der"),
|
|
("string", "TEST_CLI_KEY_EC_PEM", "cli2.key"),
|
|
("binary", "TEST_CLI_KEY_EC_DER", "cli2.key.der"),
|
|
("string", "TEST_CLI_CRT_RSA_PEM", "cli-rsa-sha256.crt"),
|
|
("binary", "TEST_CLI_CRT_RSA_DER", "cli-rsa-sha256.crt.der"),
|
|
("string", "TEST_CLI_KEY_RSA_PEM", "cli-rsa.key"),
|
|
("binary", "TEST_CLI_KEY_RSA_DER", "cli-rsa.key.der"),
|
|
]
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
default_output_path = os.path.join(TESTS_DIR, 'include', 'test', 'test_certs.h')
|
|
parser.add_argument('--output', type=str, default=default_output_path)
|
|
parser.add_argument('--list-dependencies', action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
if args.list_dependencies:
|
|
files_list = [arg[2] for arg in INPUT_ARGS
|
|
if arg[0] != "password"]
|
|
print(" ".join(os.path.join(DATA_DIR_ABSOLUTE, filename)
|
|
for filename in files_list))
|
|
return
|
|
|
|
generate(INPUT_ARGS, output=args.output)
|
|
|
|
#pylint: disable=dangerous-default-value, unused-argument
|
|
def generate(values=[], output=None):
|
|
"""Generate C header file.
|
|
"""
|
|
template_loader = jinja2.FileSystemLoader(DATA_DIR_ABSOLUTE)
|
|
template_env = jinja2.Environment(
|
|
loader=template_loader, lstrip_blocks=True, trim_blocks=True,
|
|
keep_trailing_newline=True)
|
|
|
|
def read_as_c_array(filename):
|
|
with open(os.path.join(DATA_DIR_ABSOLUTE, filename), 'rb') as f:
|
|
data = f.read(12)
|
|
while data:
|
|
yield ', '.join(['{:#04x}'.format(b) for b in data])
|
|
data = f.read(12)
|
|
|
|
def read_lines(filename):
|
|
with open(os.path.join(DATA_DIR_ABSOLUTE, filename)) as f:
|
|
try:
|
|
for line in f:
|
|
yield line.strip()
|
|
except:
|
|
print(filename)
|
|
raise
|
|
|
|
def put_to_column(value, position=0):
|
|
return ' '*position + value
|
|
|
|
template_env.filters['read_as_c_array'] = read_as_c_array
|
|
template_env.filters['read_lines'] = read_lines
|
|
template_env.filters['put_to_column'] = put_to_column
|
|
|
|
template = template_env.get_template('test_certs.h.jinja2')
|
|
|
|
with open(output, 'w') as f:
|
|
f.write(template.render(data_dir=DATA_DIR_RELATIVE,
|
|
macros=values))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|