blob: ce7fee65c272ab5fba88e94b38a6b7f1458c726c [file] [log] [blame]
Gilles Peskineb39e3ec2019-01-29 08:50:20 +01001#!/usr/bin/env python3
2
Andrzej Kurek629c4122022-10-17 08:34:40 -04003# Copyright (c) 2022, Arm Limited, All Rights Reserved.
Gilles Peskineb39e3ec2019-01-29 08:50:20 +01004# SPDX-License-Identifier: Apache-2.0
5#
6# Licensed under the Apache License, Version 2.0 (the "License"); you may
7# not use this file except in compliance with the License.
8# You may obtain a copy of the License at
9#
10# http://www.apache.org/licenses/LICENSE-2.0
11#
12# Unless required by applicable law or agreed to in writing, software
13# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
14# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15# See the License for the specific language governing permissions and
16# limitations under the License.
17#
18# This file is part of Mbed TLS (https://tls.mbed.org)
19
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010020"""
Andrzej Kurek01af84a2022-10-09 05:29:44 -040021Test Mbed TLS with a subset of algorithms.
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010022
Andrzej Kurek01af84a2022-10-09 05:29:44 -040023This script can be divided into several steps:
24
25First, include/mbedtls/mbedtls_config.h or a different config file passed
26in the arguments is parsed to extract any configuration options (collect_config_symbols).
27
28Then, test domains (groups of jobs, tests) are built based on predefined data
29collected in the DomainData class. Here, each domain has five major traits:
Andrzej Kurek629c4122022-10-17 08:34:40 -040030- domain name, can be used to run only specific tests via command-line;
Andrzej Kurek01af84a2022-10-09 05:29:44 -040031- configuration building method, described in detail below;
32- list of symbols passed to the configuration building method;
33- commands to be run on each job (only build, build and test, or any other custom);
34- optional list of symbols to be excluded from testing.
35
36The configuration building method can be one of the three following:
37
38- ComplementaryDomain - build a job for each passed symbol by disabling a single
39 symbol and its reverse dependencies (defined in REVERSE_DEPENDENCIES);
40
41- ExclusiveDomain - build a job where, for each passed symbol, only this particular
42 one is defined and other symbols from the list are unset. For each job look for
43 any non-standard symbols to set/unset in EXCLUSIVE_GROUPS. These are usually not
44 direct dependencies, but rather non-trivial results of other configs missing. Then
45 look for any unset symbols and handle their reverse dependencies.
46 Examples of EXCLUSIVE_GROUPS usage:
Andrzej Kurek629c4122022-10-17 08:34:40 -040047 - MBEDTLS_SHA256 job turns off all hashes except SHA256, however, when investigating
48 reverse dependencies, SHA224 is found to depend on SHA256, so it is disabled,
49 and then SHA256 is found to depend on SHA224, so it is also disabled. To handle
50 this, there's a field in EXCLUSIVE_GROUPS that states that in a SHA256 test SHA224
Andrzej Kurek01af84a2022-10-09 05:29:44 -040051 should also be enabled before processing reverse dependencies:
Andrzej Kurek629c4122022-10-17 08:34:40 -040052 'MBEDTLS_SHA256_C': ['+MBEDTLS_SHA224_C']
Andrzej Kurek01af84a2022-10-09 05:29:44 -040053 - MBEDTLS_SHA512_C job turns off all hashes except SHA512. MBEDTLS_SSL_COOKIE_C
54 requires either SHA256 or SHA384 to work, so it also has to be disabled.
55 This is not a dependency on SHA512_C, but a result of an exclusive domain
56 config building method. Relevant field:
Andrzej Kurek629c4122022-10-17 08:34:40 -040057 'MBEDTLS_SHA512_C': ['-MBEDTLS_SSL_COOKIE_C'],
Andrzej Kurek01af84a2022-10-09 05:29:44 -040058
59- DualDomain - combination of the two above - both complementary and exclusive domain
60 job generation code will be run. Currently only used for hashes.
61
62Lastly, the collected jobs are executed and (optionally) tested, with
63error reporting and coloring as configured in options. Each test starts with
64a full config without a couple of slowing down or unnecessary options
65(see set_reference_config), then the specific job config is derived.
66"""
Gilles Peskineb39e3ec2019-01-29 08:50:20 +010067import argparse
68import os
69import re
70import shutil
71import subprocess
72import sys
73import traceback
74
Andrzej Kurek3322c222022-10-04 15:02:41 -040075class Colors: # pylint: disable=too-few-public-methods
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +010076 """Minimalistic support for colored output.
77Each field of an object of this class is either None if colored output
78is not possible or not desired, or a pair of strings (start, stop) such
79that outputting start switches the text color to the desired color and
80stop switches the text color back to the default."""
81 red = None
82 green = None
83 bold_red = None
84 bold_green = None
85 def __init__(self, options=None):
Andrzej Kurek3322c222022-10-04 15:02:41 -040086 """Initialize color profile according to passed options."""
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +010087 if not options or options.color in ['no', 'never']:
88 want_color = False
89 elif options.color in ['yes', 'always']:
90 want_color = True
91 else:
92 want_color = sys.stderr.isatty()
93 if want_color:
94 # Assume ANSI compatible terminal
95 normal = '\033[0m'
96 self.red = ('\033[31m', normal)
97 self.green = ('\033[32m', normal)
98 self.bold_red = ('\033[1;31m', normal)
99 self.bold_green = ('\033[1;32m', normal)
100NO_COLORS = Colors(None)
101
102def log_line(text, prefix='depends.py:', suffix='', color=None):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100103 """Print a status message."""
Andrzej Kurek3322c222022-10-04 15:02:41 -0400104 if color is not None:
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100105 prefix = color[0] + prefix
106 suffix = suffix + color[1]
107 sys.stderr.write(prefix + ' ' + text + suffix + '\n')
Gilles Peskine46c82562019-01-29 18:42:55 +0100108 sys.stderr.flush()
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100109
Gilles Peskine54aa5c62019-01-29 18:46:34 +0100110def log_command(cmd):
111 """Print a trace of the specified command.
112cmd is a list of strings: a command name and its arguments."""
113 log_line(' '.join(cmd), prefix='+')
114
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100115def backup_config(options):
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400116 """Back up the library configuration file (mbedtls_config.h).
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100117If the backup file already exists, it is presumed to be the desired backup,
118so don't make another backup."""
119 if os.path.exists(options.config_backup):
120 options.own_backup = False
121 else:
122 options.own_backup = True
123 shutil.copy(options.config, options.config_backup)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100124
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100125def restore_config(options):
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400126 """Restore the library configuration file (mbedtls_config.h).
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100127Remove the backup file if it was saved earlier."""
128 if options.own_backup:
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100129 shutil.move(options.config_backup, options.config)
130 else:
131 shutil.copy(options.config_backup, options.config)
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100132
Andrzej Kureka44c5bc2022-10-16 12:52:20 -0400133def run_config_py(options, args):
Andrzej Kurek3322c222022-10-04 15:02:41 -0400134 """Run scripts/config.py with the specified arguments."""
135 cmd = ['scripts/config.py']
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400136 if options.config != 'include/mbedtls/mbedtls_config.h':
Gilles Peskine54aa5c62019-01-29 18:46:34 +0100137 cmd += ['--file', options.config]
138 cmd += args
139 log_command(cmd)
140 subprocess.check_call(cmd)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100141
Andrzej Kurek3322c222022-10-04 15:02:41 -0400142def set_reference_config(options):
143 """Change the library configuration file (mbedtls_config.h) to the reference state.
144The reference state is the one from which the tested configurations are
145derived."""
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400146 # Turn off options that are not relevant to the tests and slow them down.
Andrzej Kureka44c5bc2022-10-16 12:52:20 -0400147 run_config_py(options, ['full'])
Andrzej Kureka44c5bc2022-10-16 12:52:20 -0400148 run_config_py(options, ['unset', 'MBEDTLS_TEST_HOOKS'])
Andrzej Kurek3322c222022-10-04 15:02:41 -0400149
150def collect_config_symbols(options):
151 """Read the list of settings from mbedtls_config.h.
152Return them in a generator."""
153 with open(options.config, encoding="utf-8") as config_file:
154 rx = re.compile(r'\s*(?://\s*)?#define\s+(\w+)\s*(?:$|/[/*])')
155 for line in config_file:
156 m = re.match(rx, line)
157 if m:
158 yield m.group(1)
159
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100160class Job:
161 """A job builds the library in a specific configuration and runs some tests."""
162 def __init__(self, name, config_settings, commands):
163 """Build a job object.
164The job uses the configuration described by config_settings. This is a
165dictionary where the keys are preprocessor symbols and the values are
166booleans or strings. A boolean indicates whether or not to #define the
167symbol. With a string, the symbol is #define'd to that value.
168After setting the configuration, the job runs the programs specified by
169commands. This is a list of lists of strings; each list of string is a
170command name and its arguments and is passed to subprocess.call with
171shell=False."""
172 self.name = name
173 self.config_settings = config_settings
174 self.commands = commands
175
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100176 def announce(self, colors, what):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100177 '''Announce the start or completion of a job.
178If what is None, announce the start of the job.
179If what is True, announce that the job has passed.
180If what is False, announce that the job has failed.'''
181 if what is True:
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100182 log_line(self.name + ' PASSED', color=colors.green)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100183 elif what is False:
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100184 log_line(self.name + ' FAILED', color=colors.red)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100185 else:
186 log_line('starting ' + self.name)
187
Gilles Peskine54aa5c62019-01-29 18:46:34 +0100188 def configure(self, options):
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400189 '''Set library configuration options as required for the job.'''
Andrzej Kurek3322c222022-10-04 15:02:41 -0400190 set_reference_config(options)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100191 for key, value in sorted(self.config_settings.items()):
192 if value is True:
193 args = ['set', key]
194 elif value is False:
195 args = ['unset', key]
196 else:
197 args = ['set', key, value]
Andrzej Kureka44c5bc2022-10-16 12:52:20 -0400198 run_config_py(options, args)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100199
200 def test(self, options):
201 '''Run the job's build and test commands.
202Return True if all the commands succeed and False otherwise.
203If options.keep_going is false, stop as soon as one command fails. Otherwise
204run all the commands, except that if the first command fails, none of the
205other commands are run (typically, the first command is a build command
206and subsequent commands are tests that cannot run if the build failed).'''
207 built = False
208 success = True
209 for command in self.commands:
Gilles Peskine54aa5c62019-01-29 18:46:34 +0100210 log_command(command)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100211 ret = subprocess.call(command)
212 if ret != 0:
213 if command[0] not in ['make', options.make_command]:
214 log_line('*** [{}] Error {}'.format(' '.join(command), ret))
215 if not options.keep_going or not built:
216 return False
217 success = False
218 built = True
219 return success
220
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100221# If the configuration option A requires B, make sure that
Andrzej Kurek202932f2022-10-04 16:22:22 -0400222# B in REVERSE_DEPENDENCIES[A].
Gilles Peskine584c24a2019-01-29 19:30:40 +0100223# All the information here should be contained in check_config.h. This
224# file includes a copy because it changes rarely and it would be a pain
225# to extract automatically.
Andrzej Kurek202932f2022-10-04 16:22:22 -0400226REVERSE_DEPENDENCIES = {
Gilles Peskine34a15572019-01-29 23:12:28 +0100227 'MBEDTLS_AES_C': ['MBEDTLS_CTR_DRBG_C',
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400228 'MBEDTLS_NIST_KW_C'],
Gilles Peskine34a15572019-01-29 23:12:28 +0100229 'MBEDTLS_CHACHA20_C': ['MBEDTLS_CHACHAPOLY_C'],
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400230 'MBEDTLS_ECDSA_C': ['MBEDTLS_KEY_EXCHANGE_ECDHE_ECDSA_ENABLED',
231 'MBEDTLS_KEY_EXCHANGE_ECDH_ECDSA_ENABLED'],
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100232 'MBEDTLS_ECP_C': ['MBEDTLS_ECDSA_C',
233 'MBEDTLS_ECDH_C',
234 'MBEDTLS_ECJPAKE_C',
235 'MBEDTLS_KEY_EXCHANGE_ECDH_ECDSA_ENABLED',
236 'MBEDTLS_KEY_EXCHANGE_ECDH_RSA_ENABLED',
237 'MBEDTLS_KEY_EXCHANGE_ECDHE_PSK_ENABLED',
238 'MBEDTLS_KEY_EXCHANGE_ECDHE_RSA_ENABLED',
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400239 'MBEDTLS_KEY_EXCHANGE_ECDHE_ECDSA_ENABLED',
240 'MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED'],
Gilles Peskine584c24a2019-01-29 19:30:40 +0100241 'MBEDTLS_ECP_DP_SECP256R1_ENABLED': ['MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED'],
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100242 'MBEDTLS_PKCS1_V21': ['MBEDTLS_X509_RSASSA_PSS_SUPPORT'],
243 'MBEDTLS_PKCS1_V15': ['MBEDTLS_KEY_EXCHANGE_DHE_RSA_ENABLED',
244 'MBEDTLS_KEY_EXCHANGE_ECDHE_RSA_ENABLED',
245 'MBEDTLS_KEY_EXCHANGE_RSA_PSK_ENABLED',
246 'MBEDTLS_KEY_EXCHANGE_RSA_ENABLED'],
247 'MBEDTLS_RSA_C': ['MBEDTLS_X509_RSASSA_PSS_SUPPORT',
248 'MBEDTLS_KEY_EXCHANGE_DHE_RSA_ENABLED',
249 'MBEDTLS_KEY_EXCHANGE_ECDHE_RSA_ENABLED',
250 'MBEDTLS_KEY_EXCHANGE_RSA_PSK_ENABLED',
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400251 'MBEDTLS_KEY_EXCHANGE_RSA_ENABLED',
252 'MBEDTLS_KEY_EXCHANGE_ECDH_RSA_ENABLED'],
Gilles Peskine584c24a2019-01-29 19:30:40 +0100253 'MBEDTLS_SHA256_C': ['MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED',
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400254 'MBEDTLS_ENTROPY_FORCE_SHA256',
255 'MBEDTLS_SHA224_C',
256 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT',
Andrzej Kurek22b959d2022-10-16 12:51:41 -0400257 'MBEDTLS_SHA256_USE_A64_CRYPTO_ONLY',
258 'MBEDTLS_LMS_C',
259 'MBEDTLS_LMS_PRIVATE'],
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400260 'MBEDTLS_SHA512_C': ['MBEDTLS_SHA384_C',
261 'MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT',
262 'MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY'],
263 'MBEDTLS_SHA224_C': ['MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED',
264 'MBEDTLS_ENTROPY_FORCE_SHA256',
265 'MBEDTLS_SHA256_C',
266 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT',
267 'MBEDTLS_SHA256_USE_A64_CRYPTO_ONLY'],
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400268 'MBEDTLS_X509_RSASSA_PSS_SUPPORT': []
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100269}
270
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400271# If an option is tested in an exclusive test, alter the following defines.
Andrzej Kurek01af84a2022-10-09 05:29:44 -0400272# These are not necessarily dependencies, but just minimal required changes
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400273# if a given define is the only one enabled from an exclusive group.
Andrzej Kurek202932f2022-10-04 16:22:22 -0400274EXCLUSIVE_GROUPS = {
Andrzej Kurek65b2ac12022-10-14 08:09:16 -0400275 'MBEDTLS_SHA256_C': ['+MBEDTLS_SHA224_C'],
276 'MBEDTLS_SHA384_C': ['+MBEDTLS_SHA512_C'],
277 'MBEDTLS_SHA512_C': ['-MBEDTLS_SSL_COOKIE_C',
278 '-MBEDTLS_SSL_PROTO_TLS1_3'],
279 'MBEDTLS_ECP_DP_CURVE448_ENABLED': ['-MBEDTLS_ECDSA_C',
280 '-MBEDTLS_ECDSA_DETERMINISTIC',
281 '-MBEDTLS_KEY_EXCHANGE_ECDHE_ECDSA_ENABLED',
282 '-MBEDTLS_KEY_EXCHANGE_ECDH_ECDSA_ENABLED',
283 '-MBEDTLS_ECJPAKE_C',
284 '-MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED'],
285 'MBEDTLS_ECP_DP_CURVE25519_ENABLED': ['-MBEDTLS_ECDSA_C',
286 '-MBEDTLS_ECDSA_DETERMINISTIC',
287 '-MBEDTLS_KEY_EXCHANGE_ECDHE_ECDSA_ENABLED',
288 '-MBEDTLS_KEY_EXCHANGE_ECDH_ECDSA_ENABLED',
289 '-MBEDTLS_ECJPAKE_C',
290 '-MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED'],
291 'MBEDTLS_ARIA_C': ['-MBEDTLS_CMAC_C'],
292 'MBEDTLS_CAMELLIA_C': ['-MBEDTLS_CMAC_C'],
293 'MBEDTLS_CHACHA20_C': ['-MBEDTLS_CMAC_C', '-MBEDTLS_CCM_C', '-MBEDTLS_GCM_C'],
294 'MBEDTLS_DES_C': ['-MBEDTLS_CCM_C',
295 '-MBEDTLS_GCM_C',
296 '-MBEDTLS_SSL_TICKET_C',
297 '-MBEDTLS_SSL_CONTEXT_SERIALIZATION'],
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400298}
299def handle_exclusive_groups(config_settings, symbol):
300 """For every symbol tested in an exclusive group check if there are other
301defines to be altered. """
Andrzej Kurek202932f2022-10-04 16:22:22 -0400302 for dep in EXCLUSIVE_GROUPS.get(symbol, []):
Andrzej Kurek65b2ac12022-10-14 08:09:16 -0400303 unset = dep.startswith('-')
304 dep = dep[1:]
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400305 config_settings[dep] = not unset
306
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100307def turn_off_dependencies(config_settings):
308 """For every option turned off config_settings, also turn off what depends on it.
309An option O is turned off if config_settings[O] is False."""
310 for key, value in sorted(config_settings.items()):
311 if value is not False:
312 continue
Andrzej Kurek202932f2022-10-04 16:22:22 -0400313 for dep in REVERSE_DEPENDENCIES.get(key, []):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100314 config_settings[dep] = False
315
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400316class BaseDomain: # pylint: disable=too-few-public-methods, unused-argument
317 """A base class for all domains."""
318 def __init__(self, symbols, commands, exclude):
319 """Initialize the jobs container"""
320 self.jobs = []
321
322class ExclusiveDomain(BaseDomain): # pylint: disable=too-few-public-methods
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100323 """A domain consisting of a set of conceptually-equivalent settings.
324Establish a list of configuration symbols. For each symbol, run a test job
Andrzej Kurekfe469492022-10-06 16:57:38 -0400325with this symbol set and the others unset."""
Gilles Peskineb1284cf2019-01-29 18:56:03 +0100326 def __init__(self, symbols, commands, exclude=None):
327 """Build a domain for the specified list of configuration symbols.
Andrzej Kurekfe469492022-10-06 16:57:38 -0400328The domain contains a set of jobs that enable one of the elements
329of symbols and disable the others.
Gilles Peskineb1284cf2019-01-29 18:56:03 +0100330Each job runs the specified commands.
331If exclude is a regular expression, skip generated jobs whose description
332would match this regular expression."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400333 super().__init__(symbols, commands, exclude)
Andrzej Kurekfe469492022-10-06 16:57:38 -0400334 base_config_settings = {}
335 for symbol in symbols:
336 base_config_settings[symbol] = False
337 for symbol in symbols:
338 description = symbol
339 if exclude and re.match(exclude, description):
340 continue
341 config_settings = base_config_settings.copy()
342 config_settings[symbol] = True
343 handle_exclusive_groups(config_settings, symbol)
344 turn_off_dependencies(config_settings)
345 job = Job(description, config_settings, commands)
346 self.jobs.append(job)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100347
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400348class ComplementaryDomain(BaseDomain): # pylint: disable=too-few-public-methods
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100349 """A domain consisting of a set of loosely-related settings.
350Establish a list of configuration symbols. For each symbol, run a test job
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400351with this symbol unset.
352If exclude is a regular expression, skip generated jobs whose description
353would match this regular expression."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400354 def __init__(self, symbols, commands, exclude=None):
Gilles Peskineb1284cf2019-01-29 18:56:03 +0100355 """Build a domain for the specified list of configuration symbols.
356Each job in the domain disables one of the specified symbols.
357Each job runs the specified commands."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400358 super().__init__(symbols, commands, exclude)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100359 for symbol in symbols:
360 description = '!' + symbol
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400361 if exclude and re.match(exclude, description):
362 continue
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100363 config_settings = {symbol: False}
364 turn_off_dependencies(config_settings)
365 job = Job(description, config_settings, commands)
366 self.jobs.append(job)
367
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400368class DualDomain(ExclusiveDomain, ComplementaryDomain): # pylint: disable=too-few-public-methods
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400369 """A domain that contains both the ExclusiveDomain and BaseDomain tests.
Andrzej Kurekf4b18672022-10-14 07:57:00 -0400370Both parent class __init__ calls are performed in any order and
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400371each call adds respective jobs. The job array initialization is done once in
372BaseDomain, before the parent __init__ calls."""
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400373
Andrzej Kurek3322c222022-10-04 15:02:41 -0400374class CipherInfo: # pylint: disable=too-few-public-methods
Gilles Peskine34a15572019-01-29 23:12:28 +0100375 """Collect data about cipher.h."""
Andrzej Kurek3322c222022-10-04 15:02:41 -0400376 def __init__(self):
Gilles Peskine34a15572019-01-29 23:12:28 +0100377 self.base_symbols = set()
Andrzej Kurek3322c222022-10-04 15:02:41 -0400378 with open('include/mbedtls/cipher.h', encoding="utf-8") as fh:
Gilles Peskine34a15572019-01-29 23:12:28 +0100379 for line in fh:
380 m = re.match(r' *MBEDTLS_CIPHER_ID_(\w+),', line)
381 if m and m.group(1) not in ['NONE', 'NULL', '3DES']:
382 self.base_symbols.add('MBEDTLS_' + m.group(1) + '_C')
383
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100384class DomainData:
Andrzej Kurek3322c222022-10-04 15:02:41 -0400385 """A container for domains and jobs, used to structurize testing."""
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100386 def config_symbols_matching(self, regexp):
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400387 """List the mbedtls_config.h settings matching regexp."""
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100388 return [symbol for symbol in self.all_config_symbols
389 if re.match(regexp, symbol)]
390
391 def __init__(self, options):
392 """Gather data about the library and establish a list of domains to test."""
393 build_command = [options.make_command, 'CFLAGS=-Werror']
394 build_and_test = [build_command, [options.make_command, 'test']]
Andrzej Kurek3322c222022-10-04 15:02:41 -0400395 self.all_config_symbols = set(collect_config_symbols(options))
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100396 # Find hash modules by name.
397 hash_symbols = self.config_symbols_matching(r'MBEDTLS_(MD|RIPEMD|SHA)[0-9]+_C\Z')
398 # Find elliptic curve enabling macros by name.
399 curve_symbols = self.config_symbols_matching(r'MBEDTLS_ECP_DP_\w+_ENABLED\Z')
400 # Find key exchange enabling macros by name.
401 key_exchange_symbols = self.config_symbols_matching(r'MBEDTLS_KEY_EXCHANGE_\w+_ENABLED\Z')
Gilles Peskine34a15572019-01-29 23:12:28 +0100402 # Find cipher IDs (block permutations and stream ciphers --- chaining
403 # and padding modes are exercised separately) information by parsing
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400404 # cipher.h, as the information is not readily available in mbedtls_config.h.
Andrzej Kurek3322c222022-10-04 15:02:41 -0400405 cipher_info = CipherInfo()
Gilles Peskine34a15572019-01-29 23:12:28 +0100406 # Find block cipher chaining and padding mode enabling macros by name.
407 cipher_chaining_symbols = self.config_symbols_matching(r'MBEDTLS_CIPHER_MODE_\w+\Z')
408 cipher_padding_symbols = self.config_symbols_matching(r'MBEDTLS_CIPHER_PADDING_\w+\Z')
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100409 self.domains = {
Gilles Peskine34a15572019-01-29 23:12:28 +0100410 # Cipher IDs, chaining modes and padding modes. Run the test suites.
411 'cipher_id': ExclusiveDomain(cipher_info.base_symbols,
412 build_and_test),
413 'cipher_chaining': ExclusiveDomain(cipher_chaining_symbols,
414 build_and_test),
415 'cipher_padding': ExclusiveDomain(cipher_padding_symbols,
416 build_and_test),
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100417 # Elliptic curves. Run the test suites.
418 'curves': ExclusiveDomain(curve_symbols, build_and_test),
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400419 # Hash algorithms. Exclude three groups:
420 # - Exclusive domain of MD, RIPEMD, SHA1 (obsolete);
421 # - Exclusive domain of SHA224 (tested with and depends on SHA256);
422 # - Complementary domain of SHA224 and SHA384 - tested with and depend
423 # on SHA256 and SHA512, respectively.
Andrzej Kurek228b12c2022-10-06 18:52:44 -0400424 'hashes': DualDomain(hash_symbols, build_and_test,
Andrzej Kureka0cb4fa2022-10-14 07:06:43 -0400425 exclude=r'MBEDTLS_(MD|RIPEMD|SHA1_)' \
426 '|MBEDTLS_SHA224_'\
427 '|!MBEDTLS_(SHA224_|SHA384_)'),
Gilles Peskinec3b4dee2019-01-29 19:33:05 +0100428 # Key exchange types. Only build the library and the sample
429 # programs.
430 'kex': ExclusiveDomain(key_exchange_symbols,
431 [build_command + ['lib'],
432 build_command + ['-C', 'programs']]),
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100433 'pkalgs': ComplementaryDomain(['MBEDTLS_ECDSA_C',
434 'MBEDTLS_ECP_C',
435 'MBEDTLS_PKCS1_V21',
436 'MBEDTLS_PKCS1_V15',
437 'MBEDTLS_RSA_C',
438 'MBEDTLS_X509_RSASSA_PSS_SUPPORT'],
439 build_and_test),
440 }
441 self.jobs = {}
442 for domain in self.domains.values():
443 for job in domain.jobs:
444 self.jobs[job.name] = job
445
446 def get_jobs(self, name):
447 """Return the list of jobs identified by the given name.
448A name can either be the name of a domain or the name of one specific job."""
449 if name in self.domains:
450 return sorted(self.domains[name].jobs, key=lambda job: job.name)
451 else:
452 return [self.jobs[name]]
453
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100454def run(options, job, colors=NO_COLORS):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100455 """Run the specified job (a Job instance)."""
456 subprocess.check_call([options.make_command, 'clean'])
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100457 job.announce(colors, None)
Gilles Peskine54aa5c62019-01-29 18:46:34 +0100458 job.configure(options)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100459 success = job.test(options)
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100460 job.announce(colors, success)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100461 return success
462
Andrzej Kurek3322c222022-10-04 15:02:41 -0400463def run_tests(options, domain_data):
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100464 """Run the desired jobs.
465domain_data should be a DomainData instance that describes the available
466domains and jobs.
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400467Run the jobs listed in options.tasks."""
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100468 if not hasattr(options, 'config_backup'):
469 options.config_backup = options.config + '.bak'
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100470 colors = Colors(options)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100471 jobs = []
472 failures = []
473 successes = []
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400474 for name in options.tasks:
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100475 jobs += domain_data.get_jobs(name)
476 backup_config(options)
477 try:
478 for job in jobs:
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100479 success = run(options, job, colors=colors)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100480 if not success:
481 if options.keep_going:
482 failures.append(job.name)
483 else:
484 return False
485 else:
486 successes.append(job.name)
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100487 restore_config(options)
488 except:
489 # Restore the configuration, except in stop-on-error mode if there
490 # was an error, where we leave the failing configuration up for
491 # developer convenience.
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100492 if options.keep_going:
Gilles Peskinebf7537d2019-01-29 18:52:16 +0100493 restore_config(options)
494 raise
Gilles Peskinee85163b2019-01-29 18:50:03 +0100495 if successes:
496 log_line('{} passed'.format(' '.join(successes)), color=colors.bold_green)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100497 if failures:
Gilles Peskinee85163b2019-01-29 18:50:03 +0100498 log_line('{} FAILED'.format(' '.join(failures)), color=colors.bold_red)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100499 return False
500 else:
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100501 return True
502
Andrzej Kurek3322c222022-10-04 15:02:41 -0400503def main():
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100504 try:
Andrzej Kurek01af84a2022-10-09 05:29:44 -0400505 parser = argparse.ArgumentParser(
506 formatter_class=argparse.RawDescriptionHelpFormatter,
507 description=
508 "Test Mbed TLS with a subset of algorithms.\n\n"
509 "Example usage:\n"
Andrzej Kurek629c4122022-10-17 08:34:40 -0400510 r"./tests/scripts/depends.py \!MBEDTLS_SHA1_C MBEDTLS_SHA256_C""\n"
Andrzej Kurek01af84a2022-10-09 05:29:44 -0400511 "./tests/scripts/depends.py MBEDTLS_AES_C hashes\n"
512 "./tests/scripts/depends.py cipher_id cipher_chaining\n")
Gilles Peskine0fa7cbe2019-01-29 18:48:48 +0100513 parser.add_argument('--color', metavar='WHEN',
514 help='Colorize the output (always/auto/never)',
515 choices=['always', 'auto', 'never'], default='auto')
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100516 parser.add_argument('-c', '--config', metavar='FILE',
517 help='Configuration file to modify',
Andrzej Kureke05b17f2022-09-28 03:17:56 -0400518 default='include/mbedtls/mbedtls_config.h')
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100519 parser.add_argument('-C', '--directory', metavar='DIR',
520 help='Change to this directory before anything else',
521 default='.')
522 parser.add_argument('-k', '--keep-going',
523 help='Try all configurations even if some fail (default)',
524 action='store_true', dest='keep_going', default=True)
525 parser.add_argument('-e', '--no-keep-going',
526 help='Stop as soon as a configuration fails',
527 action='store_false', dest='keep_going')
528 parser.add_argument('--list-jobs',
529 help='List supported jobs and exit',
530 action='append_const', dest='list', const='jobs')
531 parser.add_argument('--list-domains',
532 help='List supported domains and exit',
533 action='append_const', dest='list', const='domains')
534 parser.add_argument('--make-command', metavar='CMD',
535 help='Command to run instead of make (e.g. gmake)',
536 action='store', default='make')
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400537 parser.add_argument('tasks', metavar='TASKS', nargs='*',
538 help='The domain(s) or job(s) to test (default: all).',
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100539 default=True)
540 options = parser.parse_args()
541 os.chdir(options.directory)
542 domain_data = DomainData(options)
Andrzej Kurekb8a97e72022-10-17 08:39:09 -0400543 if options.tasks is True:
544 options.tasks = sorted(domain_data.domains.keys())
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100545 if options.list:
Andrzej Kurek3322c222022-10-04 15:02:41 -0400546 for arg in options.list:
547 for domain_name in sorted(getattr(domain_data, arg).keys()):
548 print(domain_name)
549 sys.exit(0)
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100550 else:
Andrzej Kurek3322c222022-10-04 15:02:41 -0400551 sys.exit(0 if run_tests(options, domain_data) else 1)
552 except Exception: # pylint: disable=broad-except
Gilles Peskineb39e3ec2019-01-29 08:50:20 +0100553 traceback.print_exc()
Andrzej Kurek3322c222022-10-04 15:02:41 -0400554 sys.exit(3)
555
556if __name__ == '__main__':
557 main()