blob: bf401e067538963a0dc6d14cf234bba3b1965bec [file] [log] [blame]
Gilles Peskineb39e3ec2019-01-29 08:50:20 +01001#!/usr/bin/env python3
2
Dave Rodgman4eb44e42023-11-03 12:15:12 +00003# Copyright The Mbed TLS Contributors
Dave Rodgman16799db2023-11-02 19:47:20 +00004# SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
Gilles Peskineb39e3ec2019-01-29 08:50:20 +01005
Gilles Peskineb39e3ec2019-01-29 08:50:20 +01006"""
Andrzej Kurek01af84a2022-10-09 05:29:44 -04007Test Mbed TLS with a subset of algorithms.
Gilles Peskineb39e3ec2019-01-29 08:50:20 +01008
Andrzej Kurek01af84a2022-10-09 05:29:44 -04009This script can be divided into several steps:
10
11First, include/mbedtls/mbedtls_config.h or a different config file passed
Andrzej Kurek3f930122022-10-26 08:08:26 -040012in the arguments is parsed to extract any configuration options (using config.py).
Andrzej Kurek01af84a2022-10-09 05:29:44 -040013
14Then, test domains (groups of jobs, tests) are built based on predefined data
15collected in the DomainData class. Here, each domain has five major traits:
Andrzej Kurek629c4122022-10-17 08:34:40 -040016- domain name, can be used to run only specific tests via command-line;
Andrzej Kurek01af84a2022-10-09 05:29:44 -040017- configuration building method, described in detail below;
18- list of symbols passed to the configuration building method;
19- commands to be run on each job (only build, build and test, or any other custom);
20- optional list of symbols to be excluded from testing.
21
22The configuration building method can be one of the three following:
23
24- ComplementaryDomain - build a job for each passed symbol by disabling a single
25 symbol and its reverse dependencies (defined in REVERSE_DEPENDENCIES);
26
27- ExclusiveDomain - build a job where, for each passed symbol, only this particular
28 one is defined and other symbols from the list are unset. For each job look for
29 any non-standard symbols to set/unset in EXCLUSIVE_GROUPS. These are usually not
30 direct dependencies, but rather non-trivial results of other configs missing. Then
31 look for any unset symbols and handle their reverse dependencies.
32 Examples of EXCLUSIVE_GROUPS usage:
Ronald Crondfd501d2025-07-21 14:44:12 +020033 - PSA_WANT_ALG_SHA_512 job turns off all hashes except SHA512. MBEDTLS_SSL_COOKIE_C
Andrzej Kurek01af84a2022-10-09 05:29:44 -040034 requires either SHA256 or SHA384 to work, so it also has to be disabled.
Ronald Crondfd501d2025-07-21 14:44:12 +020035 This is not a dependency on SHA512, but a result of an exclusive domain
Andrzej Kurek01af84a2022-10-09 05:29:44 -040036 config building method. Relevant field:
Ronald Crondfd501d2025-07-21 14:44:12 +020037 'PSA_WANT_ALG_SHA_512': ['-MBEDTLS_SSL_COOKIE_C'],
Andrzej Kurek01af84a2022-10-09 05:29:44 -040038
39- DualDomain - combination of the two above - both complementary and exclusive domain
40 job generation code will be run. Currently only used for hashes.
41
42Lastly, the collected jobs are executed and (optionally) tested, with
43error reporting and coloring as configured in options. Each test starts with
44a full config without a couple of slowing down or unnecessary options
45(see set_reference_config), then the specific job config is derived.
46"""
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010047import argparse
48import os
49import re
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010050import subprocess
51import sys
52import traceback
Andrzej Kurek576803f2023-01-24 07:40:42 -050053from typing import Union
54
Andrzej Kurek3b0215d2023-01-23 07:19:22 -050055# Add the Mbed TLS Python library directory to the module search path
Andrzej Kurek3f930122022-10-26 08:08:26 -040056import scripts_path # pylint: disable=unused-import
57import config
Gabor Mezei4e10d6c2024-08-03 14:26:13 +020058from mbedtls_framework import c_build_helper
Gabor Mezei402381d2024-09-24 19:46:54 +020059from mbedtls_framework import crypto_knowledge
60from mbedtls_framework import psa_information
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010061
Andrzej Kurek3322c222022-10-04 15:02:41 -040062class Colors: # pylint: disable=too-few-public-methods
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +010063 """Minimalistic support for colored output.
64Each field of an object of this class is either None if colored output
65is not possible or not desired, or a pair of strings (start, stop) such
66that outputting start switches the text color to the desired color and
67stop switches the text color back to the default."""
68 red = None
69 green = None
Andrzej Kurek3f930122022-10-26 08:08:26 -040070 cyan = None
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +010071 bold_red = None
72 bold_green = None
73 def __init__(self, options=None):
Andrzej Kurek3322c222022-10-04 15:02:41 -040074 """Initialize color profile according to passed options."""
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +010075 if not options or options.color in ['no', 'never']:
76 want_color = False
77 elif options.color in ['yes', 'always']:
78 want_color = True
79 else:
80 want_color = sys.stderr.isatty()
81 if want_color:
82 # Assume ANSI compatible terminal
83 normal = '\033[0m'
84 self.red = ('\033[31m', normal)
85 self.green = ('\033[32m', normal)
Andrzej Kurek3f930122022-10-26 08:08:26 -040086 self.cyan = ('\033[36m', normal)
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +010087 self.bold_red = ('\033[1;31m', normal)
88 self.bold_green = ('\033[1;32m', normal)
89NO_COLORS = Colors(None)
90
91def log_line(text, prefix='depends.py:', suffix='', color=None):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010092 """Print a status message."""
Andrzej Kurek3322c222022-10-04 15:02:41 -040093 if color is not None:
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +010094 prefix = color[0] + prefix
95 suffix = suffix + color[1]
96 sys.stderr.write(prefix + ' ' + text + suffix + '\n')
Gilles Peskine46c82562019-01-29 18:42:55 +010097 sys.stderr.flush()
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010098
Gilles Peskine54aa5c62019-01-29 18:46:34 +010099def log_command(cmd):
100 """Print a trace of the specified command.
101cmd is a list of strings: a command name and its arguments."""
102 log_line(' '.join(cmd), prefix='+')
103
Andrzej Kurek3f930122022-10-26 08:08:26 -0400104def option_exists(conf, option):
Andrzej Kurek81cf5ad2023-02-06 10:48:43 +0100105 return option in conf.settings
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100106
Andrzej Kurek576803f2023-01-24 07:40:42 -0500107def set_config_option_value(conf, option, colors, value: Union[bool, str]):
108 """Set/unset a configuration option, optionally specifying a value.
109value can be either True/False (set/unset config option), or a string,
110which will make a symbol defined with a certain value."""
Andrzej Kurek3f930122022-10-26 08:08:26 -0400111 if not option_exists(conf, option):
Ronald Crond5da0202025-07-02 09:12:50 +0200112 if value is False:
Ronald Cron50f99ca2025-07-15 09:32:03 +0200113 log_line(
114 f'Warning, disabling {option} that does not exist in {conf.filename}',
115 color=colors.cyan
116 )
Ronald Crond5da0202025-07-02 09:12:50 +0200117 return True
Andrzej Kurek3f930122022-10-26 08:08:26 -0400118 log_line('Symbol {} was not found in {}'.format(option, conf.filename), color=colors.red)
119 return False
Andrzej Kurek3f930122022-10-26 08:08:26 -0400120
Andrzej Kurek3b0215d2023-01-23 07:19:22 -0500121 if value is False:
122 log_command(['config.py', 'unset', option])
123 conf.unset(option)
Andrzej Kurek72082dc2023-02-06 10:49:46 +0100124 elif value is True:
125 log_command(['config.py', 'set', option])
126 conf.set(option)
Andrzej Kurek3b0215d2023-01-23 07:19:22 -0500127 else:
Andrzej Kurek72082dc2023-02-06 10:49:46 +0100128 log_command(['config.py', 'set', option, value])
129 conf.set(option, value)
Andrzej Kurek3f930122022-10-26 08:08:26 -0400130 return True
131
Gabor Mezei9ce6d242024-06-19 17:47:05 +0200132def set_reference_config(conf, colors):
Andrzej Kurek3322c222022-10-04 15:02:41 -0400133 """Change the library configuration file (mbedtls_config.h) to the reference state.
134The reference state is the one from which the tested configurations are
135derived."""
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400136 # Turn off options that are not relevant to the tests and slow them down.
Andrzej Kurek3f930122022-10-26 08:08:26 -0400137 log_command(['config.py', 'full'])
138 conf.adapt(config.full_adapter)
Andrzej Kurek3b0215d2023-01-23 07:19:22 -0500139 set_config_option_value(conf, 'MBEDTLS_TEST_HOOKS', colors, False)
Andrzej Kurek3322c222022-10-04 15:02:41 -0400140
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100141class Job:
142 """A job builds the library in a specific configuration and runs some tests."""
143 def __init__(self, name, config_settings, commands):
144 """Build a job object.
145The job uses the configuration described by config_settings. This is a
146dictionary where the keys are preprocessor symbols and the values are
147booleans or strings. A boolean indicates whether or not to #define the
148symbol. With a string, the symbol is #define'd to that value.
149After setting the configuration, the job runs the programs specified by
150commands. This is a list of lists of strings; each list of string is a
151command name and its arguments and is passed to subprocess.call with
152shell=False."""
153 self.name = name
154 self.config_settings = config_settings
155 self.commands = commands
156
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100157 def announce(self, colors, what):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100158 '''Announce the start or completion of a job.
159If what is None, announce the start of the job.
160If what is True, announce that the job has passed.
161If what is False, announce that the job has failed.'''
162 if what is True:
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100163 log_line(self.name + ' PASSED', color=colors.green)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100164 elif what is False:
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100165 log_line(self.name + ' FAILED', color=colors.red)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100166 else:
Andrzej Kurek3f930122022-10-26 08:08:26 -0400167 log_line('starting ' + self.name, color=colors.cyan)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100168
Gabor Mezei9ce6d242024-06-19 17:47:05 +0200169 def configure(self, conf, colors):
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400170 '''Set library configuration options as required for the job.'''
Gabor Mezei9ce6d242024-06-19 17:47:05 +0200171 set_reference_config(conf, colors)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100172 for key, value in sorted(self.config_settings.items()):
Andrzej Kurek3b0215d2023-01-23 07:19:22 -0500173 ret = set_config_option_value(conf, key, colors, value)
Andrzej Kurek3f930122022-10-26 08:08:26 -0400174 if ret is False:
175 return False
176 return True
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100177
Gabor Mezei4e10d6c2024-08-03 14:26:13 +0200178 def _consistency_check(self):
179 '''Check if the testable option is consistent with the goal.
180
181 The purpose of this function to ensure that every option is set or unset according to
182 the settings.
183 '''
184 log_command(['consistency check'])
185 c_name = None
186 exe_name = None
187 header = '#include "mbedtls/build_info.h"\n'
188
189 # Generate a C error directive for each setting to test if it is active
190 for option, value in sorted(self.config_settings.items()):
191 header += '#if '
192 if value:
193 header += '!'
Gabor Mezeic9f01cf2024-09-26 13:02:01 +0200194 header += f'defined({option})\n'
195 header += f'#error "{option}"\n'
Gabor Mezei4e10d6c2024-08-03 14:26:13 +0200196 header += '#endif\n'
197 include_path = ['include', 'tf-psa-crypto/include',
198 'tf-psa-crypto/drivers/builtin/include']
199
200 try:
201 # Generate a C file, build and run it
202 c_file, c_name, exe_name = c_build_helper.create_c_file(self.name)
203 c_build_helper.generate_c_file(c_file, 'depends.py', header, lambda x: '')
204 c_file.close()
205 c_build_helper.compile_c_file(c_name, exe_name, include_path)
Gabor Mezei4e10d6c2024-08-03 14:26:13 +0200206 return True
207
208 except c_build_helper.CompileError as e:
209 # Read the command line output to find out which setting has been failed
210 failed = {m.group(1) for m in re.finditer('.*#error "(.*)"', e.message) if m}
211 log_line('Inconsistent config option(s):')
212 for option in sorted(failed):
213 log_line(' ' + option)
Gabor Mezei4e10d6c2024-08-03 14:26:13 +0200214 return False
215
216 finally:
217 c_build_helper.remove_file_if_exists(c_name)
218 c_build_helper.remove_file_if_exists(exe_name)
219
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100220 def test(self, options):
221 '''Run the job's build and test commands.
222Return True if all the commands succeed and False otherwise.
223If options.keep_going is false, stop as soon as one command fails. Otherwise
224run all the commands, except that if the first command fails, none of the
225other commands are run (typically, the first command is a build command
226and subsequent commands are tests that cannot run if the build failed).'''
Gabor Mezei4e10d6c2024-08-03 14:26:13 +0200227 if not self._consistency_check():
228 return False
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100229 built = False
230 success = True
231 for command in self.commands:
Gilles Peskine54aa5c62019-01-29 18:46:34 +0100232 log_command(command)
Gilles Peskine35cb3192024-02-12 14:16:05 +0100233 env = os.environ.copy()
234 if 'MBEDTLS_TEST_CONFIGURATION' in env:
235 env['MBEDTLS_TEST_CONFIGURATION'] += '-' + self.name
236 ret = subprocess.call(command, env=env)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100237 if ret != 0:
238 if command[0] not in ['make', options.make_command]:
239 log_line('*** [{}] Error {}'.format(' '.join(command), ret))
240 if not options.keep_going or not built:
241 return False
242 success = False
243 built = True
244 return success
245
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100246# If the configuration option A requires B, make sure that
Andrzej Kurek202932f2022-10-04 16:22:22 -0400247# B in REVERSE_DEPENDENCIES[A].
Gabor Mezei610e6e22024-08-03 14:33:21 +0200248# All the information here should be contained in check_config.h or check_crypto_config.h.
249# This file includes a copy because it changes rarely and it would be a pain
Gilles Peskine584c24a2019-01-29 19:30:40 +0100250# to extract automatically.
Andrzej Kurek202932f2022-10-04 16:22:22 -0400251REVERSE_DEPENDENCIES = {
Gabor Mezeib50043b2024-06-18 20:30:36 +0200252 'PSA_WANT_KEY_TYPE_AES': ['PSA_WANT_ALG_PBKDF2_AES_CMAC_PRF_128',
253 'MBEDTLS_CTR_DRBG_C',
Ronald Crondfd501d2025-07-21 14:44:12 +0200254 'MBEDTLS_NIST_KW_C'],
Gabor Mezeib50043b2024-06-18 20:30:36 +0200255 'PSA_WANT_KEY_TYPE_CHACHA20': ['PSA_WANT_ALG_CHACHA20_POLY1305',
Ronald Crondfd501d2025-07-21 14:44:12 +0200256 'PSA_WANT_ALG_STREAM_CIPHER'],
257 'PSA_WANT_ALG_CCM': ['PSA_WANT_ALG_CCM_STAR_NO_TAG'],
258 'PSA_WANT_ALG_CMAC': ['PSA_WANT_ALG_PBKDF2_AES_CMAC_PRF_128'],
Gabor Mezei610e6e22024-08-03 14:33:21 +0200259
Ronald Cronfeb5e262025-09-15 18:36:39 +0200260 'PSA_WANT_ECC_SECP_R1_256': ['PSA_WANT_ALG_JPAKE'],
Valerio Settifa648ba2025-08-06 11:02:25 +0200261
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200262 'PSA_WANT_ALG_ECDSA': ['PSA_WANT_ALG_DETERMINISTIC_ECDSA',
Ronald Cron3c6bbdd2025-09-15 18:28:06 +0200263 'MBEDTLS_KEY_EXCHANGE_ECDHE_ECDSA_ENABLED'],
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200264 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR_BASIC': [
265 'PSA_WANT_ALG_ECDSA',
Ronald Crona19ee282025-09-15 18:25:06 +0200266 'PSA_WANT_ALG_ECDH',
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200267 'PSA_WANT_ALG_JPAKE',
268 'PSA_WANT_KEY_TYPE_ECC_PUBLIC_KEY',
269 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR_IMPORT',
270 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR_EXPORT',
271 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR_DERIVE',
272 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR_GENERATE',
273 'MBEDTLS_ECP_RESTARTABLE',
274 'MBEDTLS_PK_PARSE_EC_EXTENDED',
275 'MBEDTLS_PK_PARSE_EC_COMPRESSED',
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200276 'MBEDTLS_KEY_EXCHANGE_ECDHE_PSK_ENABLED',
277 'MBEDTLS_KEY_EXCHANGE_ECDHE_RSA_ENABLED',
278 'MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_EPHEMERAL_ENABLED',
Ronald Cron6cfab282025-09-15 18:32:29 +0200279 'MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_EPHEMERAL_ENABLED'],
Ronald Cron2ad1e5c2025-09-15 18:30:07 +0200280 'PSA_WANT_ALG_JPAKE': ['MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED'],
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200281 'PSA_WANT_ALG_RSA_OAEP': ['PSA_WANT_ALG_RSA_PSS',
Ronald Crondfd501d2025-07-21 14:44:12 +0200282 'MBEDTLS_X509_RSASSA_PSS_SUPPORT'],
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200283 'PSA_WANT_ALG_RSA_PKCS1V15_CRYPT': ['PSA_WANT_ALG_RSA_PKCS1V15_SIGN',
Ronald Crondfd501d2025-07-21 14:44:12 +0200284 'MBEDTLS_KEY_EXCHANGE_ECDHE_RSA_ENABLED'],
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200285 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR_BASIC': [
286 'PSA_WANT_ALG_RSA_PKCS1V15_CRYPT',
287 'PSA_WANT_ALG_RSA_OAEP',
288 'PSA_WANT_KEY_TYPE_RSA_PUBLIC_KEY',
289 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR_IMPORT',
290 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR_EXPORT',
Ben Taylor9e360b82025-09-05 09:09:28 +0100291 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR_GENERATE'],
Gabor Mezei610e6e22024-08-03 14:33:21 +0200292
Gabor Mezeief013a62024-09-24 14:12:43 +0200293 'PSA_WANT_ALG_SHA_224': ['MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED',
Gabor Mezeief013a62024-09-24 14:12:43 +0200294 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_IF_PRESENT',
Ronald Crondfd501d2025-07-21 14:44:12 +0200295 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_ONLY'],
Gabor Mezeief013a62024-09-24 14:12:43 +0200296 'PSA_WANT_ALG_SHA_256': ['MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED',
Gabor Mezeief013a62024-09-24 14:12:43 +0200297 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_IF_PRESENT',
298 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_ONLY',
299 'MBEDTLS_LMS_C',
300 'MBEDTLS_LMS_PRIVATE',
Gabor Mezeief013a62024-09-24 14:12:43 +0200301 'PSA_WANT_ALG_TLS12_ECJPAKE_TO_PMS'],
Gabor Mezeief013a62024-09-24 14:12:43 +0200302 'PSA_WANT_ALG_SHA_512': ['MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT',
Ronald Crondfd501d2025-07-21 14:44:12 +0200303 'MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY'],
Ben Taylor1948c942025-03-04 09:11:11 +0000304 'PSA_WANT_ALG_ECB_NO_PADDING' : ['MBEDTLS_NIST_KW_C'],
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100305}
306
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400307# If an option is tested in an exclusive test, alter the following defines.
Andrzej Kurek01af84a2022-10-09 05:29:44 -0400308# These are not necessarily dependencies, but just minimal required changes
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400309# if a given define is the only one enabled from an exclusive group.
Andrzej Kurek202932f2022-10-04 16:22:22 -0400310EXCLUSIVE_GROUPS = {
Gabor Mezeief013a62024-09-24 14:12:43 +0200311 'PSA_WANT_ALG_SHA_512': ['-MBEDTLS_SSL_COOKIE_C',
312 '-MBEDTLS_SSL_TLS_C'],
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200313 'PSA_WANT_ECC_MONTGOMERY_448': ['-PSA_WANT_ALG_ECDSA',
314 '-PSA_WANT_ALG_JPAKE',],
315 'PSA_WANT_ECC_MONTGOMERY_255': ['-PSA_WANT_ALG_ECDSA',
316 '-PSA_WANT_ALG_JPAKE'],
Gabor Mezeib50043b2024-06-18 20:30:36 +0200317 'PSA_WANT_KEY_TYPE_ARIA': ['-PSA_WANT_ALG_CMAC',
318 '-PSA_WANT_ALG_CCM',
319 '-PSA_WANT_ALG_GCM',
320 '-MBEDTLS_SSL_TICKET_C',
321 '-MBEDTLS_SSL_CONTEXT_SERIALIZATION'],
322 'PSA_WANT_KEY_TYPE_CAMELLIA': ['-PSA_WANT_ALG_CMAC'],
323 'PSA_WANT_KEY_TYPE_CHACHA20': ['-PSA_WANT_ALG_CMAC',
324 '-PSA_WANT_ALG_CCM',
325 '-PSA_WANT_ALG_GCM',
326 '-PSA_WANT_ALG_ECB_NO_PADDING'],
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400327}
328def handle_exclusive_groups(config_settings, symbol):
329 """For every symbol tested in an exclusive group check if there are other
330defines to be altered. """
Andrzej Kurek202932f2022-10-04 16:22:22 -0400331 for dep in EXCLUSIVE_GROUPS.get(symbol, []):
Andrzej Kurek65b2ac12022-10-14 08:09:16 -0400332 unset = dep.startswith('-')
333 dep = dep[1:]
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400334 config_settings[dep] = not unset
335
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100336def turn_off_dependencies(config_settings, exclude=None):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100337 """For every option turned off config_settings, also turn off what depends on it.
Gabor Mezeifb061012024-09-24 18:58:14 +0200338
339 An option O is turned off if config_settings[O] is False.
340 Handle the dependencies recursively.
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100341
Gabor Mezei6763a552024-11-12 18:07:42 +0100342 If 'exclude' is a symbol, ensure its dependencies are not turned off while dependencies
343 of other settings are turned off.
Gabor Mezeifb061012024-09-24 18:58:14 +0200344 """
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100345
Gabor Mezei6763a552024-11-12 18:07:42 +0100346 # Determine recursively the settings that should not be turned off for the sake of 'exclude'.
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100347 excludes = set()
348 if exclude:
Gabor Mezeiaf198c22024-11-12 18:03:33 +0100349 revdep = set(REVERSE_DEPENDENCIES.get(exclude, []))
350 while revdep:
351 dep = revdep.pop()
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100352 excludes.add(dep)
Gabor Mezeiaf198c22024-11-12 18:03:33 +0100353 revdep.update(set(REVERSE_DEPENDENCIES.get(dep, [])) - excludes)
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100354
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100355 for key, value in sorted(config_settings.items()):
356 if value is not False:
357 continue
Gabor Mezeifb061012024-09-24 18:58:14 +0200358
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100359 # Save the processed settings to handle cross referencies.
Gabor Mezei6763a552024-11-12 18:07:42 +0100360 # Start with set of settings that we do not want to turn off.
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100361 history = excludes.copy()
362 revdep = set(REVERSE_DEPENDENCIES.get(key, [])) - excludes
Gabor Mezei8ec990b2024-07-31 17:14:04 +0200363 while revdep:
364 dep = revdep.pop()
Gabor Mezeifb061012024-09-24 18:58:14 +0200365 history.add(dep)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100366 config_settings[dep] = False
Gabor Mezeifb061012024-09-24 18:58:14 +0200367 # Do not add symbols which are already processed
368 revdep.update(set(REVERSE_DEPENDENCIES.get(dep, [])) - history)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100369
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400370class BaseDomain: # pylint: disable=too-few-public-methods, unused-argument
371 """A base class for all domains."""
372 def __init__(self, symbols, commands, exclude):
373 """Initialize the jobs container"""
374 self.jobs = []
375
376class ExclusiveDomain(BaseDomain): # pylint: disable=too-few-public-methods
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100377 """A domain consisting of a set of conceptually-equivalent settings.
378Establish a list of configuration symbols. For each symbol, run a test job
Andrzej Kurekfe469492022-10-06 16:57:38 -0400379with this symbol set and the others unset."""
Gilles Peskineb1284cf2019-01-29 18:56:03 +0100380 def __init__(self, symbols, commands, exclude=None):
381 """Build a domain for the specified list of configuration symbols.
Andrzej Kurekfe469492022-10-06 16:57:38 -0400382The domain contains a set of jobs that enable one of the elements
383of symbols and disable the others.
Gilles Peskineb1284cf2019-01-29 18:56:03 +0100384Each job runs the specified commands.
385If exclude is a regular expression, skip generated jobs whose description
386would match this regular expression."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400387 super().__init__(symbols, commands, exclude)
Andrzej Kurekfe469492022-10-06 16:57:38 -0400388 base_config_settings = {}
389 for symbol in symbols:
390 base_config_settings[symbol] = False
391 for symbol in symbols:
392 description = symbol
393 if exclude and re.match(exclude, description):
394 continue
395 config_settings = base_config_settings.copy()
396 config_settings[symbol] = True
397 handle_exclusive_groups(config_settings, symbol)
Gabor Mezei95be5fb2024-10-30 17:21:49 +0100398 turn_off_dependencies(config_settings, symbol)
Andrzej Kurekfe469492022-10-06 16:57:38 -0400399 job = Job(description, config_settings, commands)
400 self.jobs.append(job)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100401
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400402class ComplementaryDomain(BaseDomain): # pylint: disable=too-few-public-methods
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100403 """A domain consisting of a set of loosely-related settings.
404Establish a list of configuration symbols. For each symbol, run a test job
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400405with this symbol unset.
406If exclude is a regular expression, skip generated jobs whose description
407would match this regular expression."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400408 def __init__(self, symbols, commands, exclude=None):
Gilles Peskineb1284cf2019-01-29 18:56:03 +0100409 """Build a domain for the specified list of configuration symbols.
410Each job in the domain disables one of the specified symbols.
411Each job runs the specified commands."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400412 super().__init__(symbols, commands, exclude)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100413 for symbol in symbols:
414 description = '!' + symbol
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400415 if exclude and re.match(exclude, description):
416 continue
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100417 config_settings = {symbol: False}
418 turn_off_dependencies(config_settings)
419 job = Job(description, config_settings, commands)
420 self.jobs.append(job)
421
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400422class DualDomain(ExclusiveDomain, ComplementaryDomain): # pylint: disable=too-few-public-methods
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400423 """A domain that contains both the ExclusiveDomain and BaseDomain tests.
Andrzej Kurekf4b18672022-10-14 07:57:00 -0400424Both parent class __init__ calls are performed in any order and
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400425each call adds respective jobs. The job array initialization is done once in
426BaseDomain, before the parent __init__ calls."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400427
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100428class DomainData:
Andrzej Kurek3322c222022-10-04 15:02:41 -0400429 """A container for domains and jobs, used to structurize testing."""
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100430 def config_symbols_matching(self, regexp):
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400431 """List the mbedtls_config.h settings matching regexp."""
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100432 return [symbol for symbol in self.all_config_symbols
433 if re.match(regexp, symbol)]
434
Gabor Mezeibd8e8182024-11-06 11:23:23 +0100435 # pylint: disable=too-many-locals
Andrzej Kurek3f930122022-10-26 08:08:26 -0400436 def __init__(self, options, conf):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100437 """Gather data about the library and establish a list of domains to test."""
Ronald Cron31f63212025-08-29 16:12:40 +0200438 build_command = [options.make_command, '-f', 'scripts/legacy.make', 'CFLAGS=-Werror -O2']
439 build_and_test = [build_command, [options.make_command, '-f',
440 'scripts/legacy.make', 'test']]
Andrzej Kurek3f930122022-10-26 08:08:26 -0400441 self.all_config_symbols = set(conf.settings.keys())
Gabor Mezei402381d2024-09-24 19:46:54 +0200442 psa_info = psa_information.Information().constructors
443 algs = {crypto_knowledge.Algorithm(alg): symbol
444 for alg, symbol in ((alg, psa_information.psa_want_symbol(alg))
445 for alg in psa_info.algorithms)
446 if symbol in self.all_config_symbols}
Gabor Mezei06a060d2024-11-04 17:36:22 +0100447 cipher_algs = {alg
448 for alg in algs
449 if alg.can_do(crypto_knowledge.AlgorithmCategory.CIPHER)}
Gabor Mezei03cc5922024-11-04 17:28:46 +0100450 key_types = {crypto_knowledge.KeyType(expr): symbol
451 for key_type in psa_info.key_types
452 for expr, symbol in ((expr, psa_information.psa_want_symbol(key_type))
453 for expr in psa_info.generate_expressions([key_type]))
454 if symbol in self.all_config_symbols}
Gabor Mezei402381d2024-09-24 19:46:54 +0200455
Gabor Mezeief013a62024-09-24 14:12:43 +0200456 # Find hash modules by category.
457 hash_symbols = {symbol
458 for alg, symbol in algs.items()
459 if alg.can_do(crypto_knowledge.AlgorithmCategory.HASH)}
Gabor Mezei1c49cff2024-09-23 14:38:59 +0200460
Gabor Mezeief013a62024-09-24 14:12:43 +0200461 # Find elliptic curve enabling macros by name.
Valerio Settifa648ba2025-08-06 11:02:25 +0200462 curve_symbols = self.config_symbols_matching(r'PSA_WANT_ECC_\w+\Z')
Gabor Mezei1c49cff2024-09-23 14:38:59 +0200463
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100464 # Find key exchange enabling macros by name.
465 key_exchange_symbols = self.config_symbols_matching(r'MBEDTLS_KEY_EXCHANGE_\w+_ENABLED\Z')
Gabor Mezeib50043b2024-06-18 20:30:36 +0200466
Gabor Mezeifb6b0db2024-11-04 17:30:32 +0100467 # Find cipher key types
468 cipher_key_types = {symbol
469 for key_type, symbol in key_types.items()
470 for alg in cipher_algs
471 if key_type.can_do(alg)}
Gabor Mezei6f77baf2024-09-20 14:18:42 +0200472
Gabor Mezei5a610862024-10-30 17:19:00 +0100473 # Get cipher modes
474 cipher_chaining_symbols = {algs[cipher_alg] for cipher_alg in cipher_algs}
Gabor Mezei6f77baf2024-09-20 14:18:42 +0200475
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100476 self.domains = {
Gabor Mezeifb6b0db2024-11-04 17:30:32 +0100477 # Cipher key types
478 'cipher_id': ExclusiveDomain(cipher_key_types, build_and_test),
Gabor Mezei6f77baf2024-09-20 14:18:42 +0200479
Gabor Mezeia5f35292024-10-30 17:30:25 +0100480 # XTS is not yet supported via the PSA API.
481 # See https://github.com/Mbed-TLS/mbedtls/issues/6384
Gilles Peskine34a15572019-01-29 23:12:28 +0100482 'cipher_chaining': ExclusiveDomain(cipher_chaining_symbols,
Gabor Mezei6f77baf2024-09-20 14:18:42 +0200483 build_and_test,
484 exclude=r'PSA_WANT_ALG_XTS'),
485
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100486 # Elliptic curves. Run the test suites.
Gabor Mezeief013a62024-09-24 14:12:43 +0200487 'curves': ExclusiveDomain(curve_symbols, build_and_test),
Gabor Mezei069e3e62025-01-10 17:44:41 +0100488
Gabor Mezeief013a62024-09-24 14:12:43 +0200489 # Hash algorithms. Excluding exclusive domains of MD, RIPEMD, SHA1, SHA3*,
Ronald Cronefcec8c2025-09-02 17:22:35 +0200490 # SHA224 and SHA384 because the built-in entropy module is extensively used
Valerio Settiea8c88f2022-12-29 11:08:35 +0100491 # across various modules, but it depends on either SHA256 or SHA512.
492 # As a consequence an "exclusive" test of anything other than SHA256
Ronald Cronefcec8c2025-09-02 17:22:35 +0200493 # or SHA512 with the built-in entropy module enabled is not possible.
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400494 'hashes': DualDomain(hash_symbols, build_and_test,
Gabor Mezeief013a62024-09-24 14:12:43 +0200495 exclude=r'PSA_WANT_ALG_(?!SHA_(256|512))'),
496
Andrzej Kurek98682b52023-01-23 06:16:23 -0500497 # Key exchange types.
Andrzej Kurek1ff73362022-11-02 04:50:16 -0400498 'kex': ExclusiveDomain(key_exchange_symbols, build_and_test),
Gabor Mezeif13fd1e2024-09-25 15:49:09 +0200499
500 'pkalgs': ComplementaryDomain(['PSA_WANT_ALG_ECDSA',
501 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR_BASIC',
502 'PSA_WANT_ALG_RSA_OAEP',
503 'PSA_WANT_ALG_RSA_PKCS1V15_CRYPT',
504 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR_BASIC',
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100505 'MBEDTLS_X509_RSASSA_PSS_SUPPORT'],
506 build_and_test),
507 }
508 self.jobs = {}
509 for domain in self.domains.values():
510 for job in domain.jobs:
511 self.jobs[job.name] = job
512
513 def get_jobs(self, name):
514 """Return the list of jobs identified by the given name.
515A name can either be the name of a domain or the name of one specific job."""
516 if name in self.domains:
517 return sorted(self.domains[name].jobs, key=lambda job: job.name)
518 else:
519 return [self.jobs[name]]
520
Andrzej Kurek3f930122022-10-26 08:08:26 -0400521def run(options, job, conf, colors=NO_COLORS):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100522 """Run the specified job (a Job instance)."""
Ronald Cron31f63212025-08-29 16:12:40 +0200523 subprocess.check_call([options.make_command, '-f', 'scripts/legacy.make', 'clean'])
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100524 job.announce(colors, None)
Gabor Mezei9ce6d242024-06-19 17:47:05 +0200525 if not job.configure(conf, colors):
Andrzej Kurek3f930122022-10-26 08:08:26 -0400526 job.announce(colors, False)
527 return False
528 conf.write()
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100529 success = job.test(options)
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100530 job.announce(colors, success)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100531 return success
532
Andrzej Kurek3f930122022-10-26 08:08:26 -0400533def run_tests(options, domain_data, conf):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100534 """Run the desired jobs.
535domain_data should be a DomainData instance that describes the available
536domains and jobs.
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400537Run the jobs listed in options.tasks."""
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100538 colors = Colors(options)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100539 jobs = []
540 failures = []
541 successes = []
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400542 for name in options.tasks:
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100543 jobs += domain_data.get_jobs(name)
Gabor Mezei035d7c82024-06-19 15:46:21 +0200544 conf.backup()
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100545 try:
546 for job in jobs:
Andrzej Kurek3f930122022-10-26 08:08:26 -0400547 success = run(options, job, conf, colors=colors)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100548 if not success:
549 if options.keep_going:
550 failures.append(job.name)
551 else:
552 return False
553 else:
554 successes.append(job.name)
Gabor Mezei035d7c82024-06-19 15:46:21 +0200555 conf.restore()
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100556 except:
557 # Restore the configuration, except in stop-on-error mode if there
558 # was an error, where we leave the failing configuration up for
559 # developer convenience.
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100560 if options.keep_going:
Gabor Mezei035d7c82024-06-19 15:46:21 +0200561 conf.restore()
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100562 raise
Gilles Peskinee85163b2019-01-29 18:50:03 +0100563 if successes:
564 log_line('{} passed'.format(' '.join(successes)), color=colors.bold_green)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100565 if failures:
Gilles Peskinee85163b2019-01-29 18:50:03 +0100566 log_line('{} FAILED'.format(' '.join(failures)), color=colors.bold_red)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100567 return False
568 else:
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100569 return True
570
Andrzej Kurek3322c222022-10-04 15:02:41 -0400571def main():
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100572 try:
Andrzej Kurek01af84a2022-10-09 05:29:44 -0400573 parser = argparse.ArgumentParser(
574 formatter_class=argparse.RawDescriptionHelpFormatter,
575 description=
576 "Test Mbed TLS with a subset of algorithms.\n\n"
577 "Example usage:\n"
Ronald Crondfd501d2025-07-21 14:44:12 +0200578 r"./tests/scripts/depends.py \!PSA_WANT_ALG_SHA_1 PSA_WANT_ALG_SHA_256""\n"
579 "./tests/scripts/depends.py PSA_WANT_KEY_TYPE_AES hashes\n"
Andrzej Kurek01af84a2022-10-09 05:29:44 -0400580 "./tests/scripts/depends.py cipher_id cipher_chaining\n")
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100581 parser.add_argument('--color', metavar='WHEN',
582 help='Colorize the output (always/auto/never)',
583 choices=['always', 'auto', 'never'], default='auto')
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100584 parser.add_argument('-c', '--config', metavar='FILE',
585 help='Configuration file to modify',
Gabor Mezeif5408f02024-06-14 15:25:46 +0200586 default=config.MbedTLSConfigFile.default_path[0])
587 parser.add_argument('-r', '--crypto-config', metavar='FILE',
588 help='Crypto configuration file to modify',
589 default=config.CryptoConfigFile.default_path[0])
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100590 parser.add_argument('-C', '--directory', metavar='DIR',
591 help='Change to this directory before anything else',
592 default='.')
593 parser.add_argument('-k', '--keep-going',
594 help='Try all configurations even if some fail (default)',
595 action='store_true', dest='keep_going', default=True)
596 parser.add_argument('-e', '--no-keep-going',
597 help='Stop as soon as a configuration fails',
598 action='store_false', dest='keep_going')
599 parser.add_argument('--list-jobs',
600 help='List supported jobs and exit',
601 action='append_const', dest='list', const='jobs')
602 parser.add_argument('--list-domains',
603 help='List supported domains and exit',
604 action='append_const', dest='list', const='domains')
605 parser.add_argument('--make-command', metavar='CMD',
606 help='Command to run instead of make (e.g. gmake)',
607 action='store', default='make')
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400608 parser.add_argument('tasks', metavar='TASKS', nargs='*',
609 help='The domain(s) or job(s) to test (default: all).',
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100610 default=True)
611 options = parser.parse_args()
612 os.chdir(options.directory)
Gabor Mezeif5408f02024-06-14 15:25:46 +0200613 conf = config.CombinedConfig(config.MbedTLSConfigFile(options.config),
614 config.CryptoConfigFile(options.crypto_config))
Andrzej Kurek3f930122022-10-26 08:08:26 -0400615 domain_data = DomainData(options, conf)
616
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400617 if options.tasks is True:
618 options.tasks = sorted(domain_data.domains.keys())
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100619 if options.list:
Andrzej Kurek3322c222022-10-04 15:02:41 -0400620 for arg in options.list:
621 for domain_name in sorted(getattr(domain_data, arg).keys()):
622 print(domain_name)
623 sys.exit(0)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100624 else:
Andrzej Kurek3f930122022-10-26 08:08:26 -0400625 sys.exit(0 if run_tests(options, domain_data, conf) else 1)
Andrzej Kurek3322c222022-10-04 15:02:41 -0400626 except Exception: # pylint: disable=broad-except
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100627 traceback.print_exc()
Andrzej Kurek3322c222022-10-04 15:02:41 -0400628 sys.exit(3)
629
630if __name__ == '__main__':
631 main()