blob: 97d7555555866da8fafe95c84c411c13d6c56cce [file] [log] [blame]
Gilles Peskineb4063892019-07-27 21:36:44 +02001#!/usr/bin/env python3
2
3"""Mbed TLS configuration file manipulation library and tool
4
Fredrik Hessecc207bc2021-09-28 21:06:08 +02005Basic usage, to read the Mbed TLS configuration:
Gilles Peskineb4063892019-07-27 21:36:44 +02006 config = ConfigFile()
7 if 'MBEDTLS_RSA_C' in config: print('RSA is enabled')
8"""
9
Gilles Peskinecf425362022-10-10 22:52:30 +020010# Note that as long as Mbed TLS 2.28 LTS is maintained, the version of
11# this script in the mbedtls-2.28 branch must remain compatible with
12# Python 3.4. The version in development may only use more recent features
13# in parts that are not backported to 2.28.
14
Bence Szépkúti1e148272020-08-07 13:07:28 +020015## Copyright The Mbed TLS Contributors
Dave Rodgman16799db2023-11-02 19:47:20 +000016## SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
Gilles Peskineb4063892019-07-27 21:36:44 +020017##
Gilles Peskineb4063892019-07-27 21:36:44 +020018
Gilles Peskine208e4ec2019-07-29 23:43:20 +020019import os
Gilles Peskineb4063892019-07-27 21:36:44 +020020import re
21
Gabor Mezeie7742b32024-06-26 18:04:09 +020022from abc import ABCMeta
Gabor Mezei3678dee2024-06-04 19:58:43 +020023
Gilles Peskineb4063892019-07-27 21:36:44 +020024class Setting:
Bence Szépkútibb0cfeb2021-05-28 09:42:25 +020025 """Representation of one Mbed TLS mbedtls_config.h setting.
Gilles Peskineb4063892019-07-27 21:36:44 +020026
27 Fields:
28 * name: the symbol name ('MBEDTLS_xxx').
29 * value: the value of the macro. The empty string for a plain #define
30 with no value.
31 * active: True if name is defined, False if a #define for name is
Bence Szépkútibb0cfeb2021-05-28 09:42:25 +020032 present in mbedtls_config.h but commented out.
Gilles Peskine53d41ae2019-07-27 23:31:53 +020033 * section: the name of the section that contains this symbol.
Gilles Peskineb4063892019-07-27 21:36:44 +020034 """
Gabor Mezei92065ed2024-06-07 13:47:59 +020035 # pylint: disable=too-few-public-methods, too-many-arguments
Gabor Mezei3678dee2024-06-04 19:58:43 +020036 def __init__(self, active, name, value='', section=None, configfile=None):
Gilles Peskineb4063892019-07-27 21:36:44 +020037 self.active = active
38 self.name = name
39 self.value = value
Gilles Peskine53d41ae2019-07-27 23:31:53 +020040 self.section = section
Gabor Mezei3678dee2024-06-04 19:58:43 +020041 self.configfile = configfile
Gilles Peskineb4063892019-07-27 21:36:44 +020042
43class Config:
44 """Representation of the Mbed TLS configuration.
45
46 In the documentation of this class, a symbol is said to be *active*
47 if there is a #define for it that is not commented out, and *known*
48 if there is a #define for it whether commented out or not.
49
50 This class supports the following protocols:
Gilles Peskinec190c902019-08-01 23:31:05 +020051 * `name in config` is `True` if the symbol `name` is active, `False`
52 otherwise (whether `name` is inactive or not known).
53 * `config[name]` is the value of the macro `name`. If `name` is inactive,
54 raise `KeyError` (even if `name` is known).
Gilles Peskineb4063892019-07-27 21:36:44 +020055 * `config[name] = value` sets the value associated to `name`. `name`
56 must be known, but does not need to be set. This does not cause
57 name to become set.
58 """
59
Gabor Mezei92065ed2024-06-07 13:47:59 +020060 # pylint: disable=unused-argument
Gabor Mezeiee521b62024-06-07 13:50:41 +020061 def __init__(self):
Gilles Peskineb4063892019-07-27 21:36:44 +020062 self.settings = {}
63
64 def __contains__(self, name):
65 """True if the given symbol is active (i.e. set).
66
67 False if the given symbol is not set, even if a definition
68 is present but commented out.
69 """
70 return name in self.settings and self.settings[name].active
71
72 def all(self, *names):
73 """True if all the elements of names are active (i.e. set)."""
74 return all(self.__contains__(name) for name in names)
75
76 def any(self, *names):
77 """True if at least one symbol in names are active (i.e. set)."""
78 return any(self.__contains__(name) for name in names)
79
80 def known(self, name):
81 """True if a #define for name is present, whether it's commented out or not."""
82 return name in self.settings
83
84 def __getitem__(self, name):
85 """Get the value of name, i.e. what the preprocessor symbol expands to.
86
87 If name is not known, raise KeyError. name does not need to be active.
88 """
89 return self.settings[name].value
90
91 def get(self, name, default=None):
92 """Get the value of name. If name is inactive (not set), return default.
93
94 If a #define for name is present and not commented out, return
95 its expansion, even if this is the empty string.
96
97 If a #define for name is present but commented out, return default.
98 """
99 if name in self.settings:
100 return self.settings[name].value
101 else:
102 return default
103
104 def __setitem__(self, name, value):
105 """If name is known, set its value.
106
107 If name is not known, raise KeyError.
108 """
109 self.settings[name].value = value
110
111 def set(self, name, value=None):
112 """Set name to the given value and make it active.
113
114 If value is None and name is already known, don't change its value.
115 If value is None and name is not known, set its value to the empty
116 string.
117 """
118 if name in self.settings:
119 if value is not None:
120 self.settings[name].value = value
121 self.settings[name].active = True
122 else:
123 self.settings[name] = Setting(True, name, value=value)
124
125 def unset(self, name):
126 """Make name unset (inactive).
127
Gilles Peskine55cc4db2019-08-01 23:13:23 +0200128 name remains known if it was known before.
Gilles Peskineb4063892019-07-27 21:36:44 +0200129 """
Gilles Peskine55cc4db2019-08-01 23:13:23 +0200130 if name not in self.settings:
131 return
Gilles Peskineb4063892019-07-27 21:36:44 +0200132 self.settings[name].active = False
133
134 def adapt(self, adapter):
135 """Run adapter on each known symbol and (de)activate it accordingly.
136
137 `adapter` must be a function that returns a boolean. It is called as
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200138 `adapter(name, active, section)` for each setting, where `active` is
139 `True` if `name` is set and `False` if `name` is known but unset,
140 and `section` is the name of the section containing `name`. If
Gilles Peskineb4063892019-07-27 21:36:44 +0200141 `adapter` returns `True`, then set `name` (i.e. make it active),
142 otherwise unset `name` (i.e. make it known but inactive).
143 """
144 for setting in self.settings.values():
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200145 setting.active = adapter(setting.name, setting.active,
146 setting.section)
Gilles Peskineb4063892019-07-27 21:36:44 +0200147
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200148 def change_matching(self, regexs, enable):
149 """Change all symbols matching one of the regexs to the desired state."""
150 if not regexs:
151 return
152 regex = re.compile('|'.join(regexs))
153 for setting in self.settings.values():
154 if regex.search(setting.name):
155 setting.active = enable
156
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200157def is_full_section(section):
158 """Is this section affected by "config.py full" and friends?"""
Gabor Mezei3678dee2024-06-04 19:58:43 +0200159 return section is None or section.endswith('support') or section.endswith('modules')
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200160
161def realfull_adapter(_name, active, section):
Gilles Peskineba4162a2022-04-11 17:04:38 +0200162 """Activate all symbols found in the global and boolean feature sections.
163
164 This is intended for building the documentation, including the
165 documentation of settings that are activated by defining an optional
166 preprocessor macro.
167
168 Do not activate definitions in the section containing symbols that are
169 supposed to be defined and documented in their own module.
170 """
171 if section == 'Module configuration options':
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200172 return active
Gilles Peskineb4063892019-07-27 21:36:44 +0200173 return True
174
Gabor Mezei542fd382024-06-10 14:07:42 +0200175PSA_UNSUPPORTED_FEATURE = frozenset([
Gabor Mezei3678dee2024-06-04 19:58:43 +0200176 'PSA_WANT_ALG_CBC_MAC',
177 'PSA_WANT_ALG_XTS',
178 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR_DERIVE',
179 'PSA_WANT_KEY_TYPE_DH_KEY_PAIR_DERIVE'
180])
181
Gabor Mezei542fd382024-06-10 14:07:42 +0200182PSA_DEPRECATED_FEATURE = frozenset([
Gabor Mezei3678dee2024-06-04 19:58:43 +0200183 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR',
184 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR'
185])
186
Gabor Mezei542fd382024-06-10 14:07:42 +0200187PSA_UNSTABLE_FEATURE = frozenset([
Gabor Mezei3678dee2024-06-04 19:58:43 +0200188 'PSA_WANT_ECC_SECP_K1_224'
189])
190
Gabor Mezei9b0f9e72024-06-26 18:08:17 +0200191EXCLUDE_FROM_CRYPTO = PSA_UNSUPPORTED_FEATURE | \
192 PSA_DEPRECATED_FEATURE | \
193 PSA_UNSTABLE_FEATURE
Gabor Mezei542fd382024-06-10 14:07:42 +0200194
Gilles Peskinecfffc282020-04-12 13:55:45 +0200195# The goal of the full configuration is to have everything that can be tested
196# together. This includes deprecated or insecure options. It excludes:
197# * Options that require additional build dependencies or unusual hardware.
198# * Options that make testing less effective.
Gilles Peskinec9d04332020-04-16 20:50:17 +0200199# * Options that are incompatible with other options, or more generally that
200# interact with other parts of the code in such a way that a bulk enabling
201# is not a good way to test them.
Gilles Peskinecfffc282020-04-12 13:55:45 +0200202# * Options that remove features.
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200203EXCLUDE_FROM_FULL = frozenset([
Gilles Peskinecfffc282020-04-12 13:55:45 +0200204 #pylint: disable=line-too-long
Yanray Wanga8704672023-04-20 17:16:48 +0800205 'MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH', # interacts with CTR_DRBG_128_BIT_KEY
Gilles Peskinea8861e02023-09-05 20:20:51 +0200206 'MBEDTLS_AES_USE_HARDWARE_ONLY', # hardware dependency
Yanray Wang42be1ba2023-11-23 14:28:47 +0800207 'MBEDTLS_BLOCK_CIPHER_NO_DECRYPT', # incompatible with ECB in PSA, CBC/XTS/NIST_KW/DES
Gilles Peskinec9d04332020-04-16 20:50:17 +0200208 'MBEDTLS_CTR_DRBG_USE_128_BIT_KEY', # interacts with ENTROPY_FORCE_SHA256
Gilles Peskinecfffc282020-04-12 13:55:45 +0200209 'MBEDTLS_DEPRECATED_REMOVED', # conflicts with deprecated options
Gilles Peskine90581ee2020-04-12 14:02:47 +0200210 'MBEDTLS_DEPRECATED_WARNING', # conflicts with deprecated options
Gilles Peskinec9d04332020-04-16 20:50:17 +0200211 'MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED', # influences the use of ECDH in TLS
Janos Follath5b7c38f2023-08-01 08:51:12 +0100212 'MBEDTLS_ECP_WITH_MPI_UINT', # disables the default ECP and is experimental
Gilles Peskinec9d04332020-04-16 20:50:17 +0200213 'MBEDTLS_ENTROPY_FORCE_SHA256', # interacts with CTR_DRBG_128_BIT_KEY
Gilles Peskinecfffc282020-04-12 13:55:45 +0200214 'MBEDTLS_HAVE_SSE2', # hardware dependency
215 'MBEDTLS_MEMORY_BACKTRACE', # depends on MEMORY_BUFFER_ALLOC_C
216 'MBEDTLS_MEMORY_BUFFER_ALLOC_C', # makes sanitizers (e.g. ASan) less effective
217 'MBEDTLS_MEMORY_DEBUG', # depends on MEMORY_BUFFER_ALLOC_C
Gilles Peskinec9d04332020-04-16 20:50:17 +0200218 'MBEDTLS_NO_64BIT_MULTIPLICATION', # influences anything that uses bignum
Gilles Peskinecfffc282020-04-12 13:55:45 +0200219 'MBEDTLS_NO_DEFAULT_ENTROPY_SOURCES', # removes a feature
220 'MBEDTLS_NO_PLATFORM_ENTROPY', # removes a feature
Gilles Peskinec9d04332020-04-16 20:50:17 +0200221 'MBEDTLS_NO_UDBL_DIVISION', # influences anything that uses bignum
Gilles Peskineefaee9a2023-09-20 20:49:47 +0200222 'MBEDTLS_PSA_P256M_DRIVER_ENABLED', # influences SECP256R1 KeyGen/ECDH/ECDSA
Gilles Peskinecfffc282020-04-12 13:55:45 +0200223 'MBEDTLS_PLATFORM_NO_STD_FUNCTIONS', # removes a feature
David Horstmann6f8c95b2024-03-14 14:52:45 +0000224 'MBEDTLS_PSA_ASSUME_EXCLUSIVE_BUFFERS', # removes a feature
Gilles Peskinef08b3f82020-11-13 17:36:48 +0100225 'MBEDTLS_PSA_CRYPTO_EXTERNAL_RNG', # behavior change + build dependency
Ronald Cronc3623db2020-10-29 10:51:32 +0100226 'MBEDTLS_PSA_CRYPTO_KEY_ID_ENCODES_OWNER', # incompatible with USE_PSA_CRYPTO
Gilles Peskinecfffc282020-04-12 13:55:45 +0200227 'MBEDTLS_PSA_CRYPTO_SPM', # platform dependency (PSA SPM)
Gilles Peskinea08def92023-04-28 21:01:49 +0200228 'MBEDTLS_PSA_INJECT_ENTROPY', # conflicts with platform entropy sources
Gilles Peskinec9d04332020-04-16 20:50:17 +0200229 'MBEDTLS_RSA_NO_CRT', # influences the use of RSA in X.509 and TLS
Tom Cosgrove87fbfb52022-03-15 10:51:52 +0000230 'MBEDTLS_SHA256_USE_A64_CRYPTO_ONLY', # interacts with *_USE_A64_CRYPTO_IF_PRESENT
Dave Rodgman9be3cf02023-10-11 14:47:55 +0100231 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_ONLY', # interacts with *_USE_ARMV8_A_CRYPTO_IF_PRESENT
Tom Cosgrove87fbfb52022-03-15 10:51:52 +0000232 'MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY', # interacts with *_USE_A64_CRYPTO_IF_PRESENT
Dave Rodgman7cb635a2023-10-12 16:14:51 +0100233 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT', # setting *_USE_ARMV8_A_CRYPTO is sufficient
Manuel Pégourié-Gonnard6240def2020-07-10 09:35:54 +0200234 'MBEDTLS_TEST_CONSTANT_FLOW_MEMSAN', # build dependency (clang+memsan)
Manuel Pégourié-Gonnard73afa372020-08-19 10:27:38 +0200235 'MBEDTLS_TEST_CONSTANT_FLOW_VALGRIND', # build dependency (valgrind headers)
Hanno Beckere1113562019-06-12 13:59:14 +0100236 'MBEDTLS_X509_REMOVE_INFO', # removes a feature
Gabor Mezei542fd382024-06-10 14:07:42 +0200237 *PSA_UNSUPPORTED_FEATURE,
238 *PSA_DEPRECATED_FEATURE,
239 *PSA_UNSTABLE_FEATURE
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200240])
241
Gilles Peskine32e889d2020-04-12 23:43:28 +0200242def is_seamless_alt(name):
Gilles Peskinec34faba2020-04-20 15:44:14 +0200243 """Whether the xxx_ALT symbol should be included in the full configuration.
Gilles Peskine32e889d2020-04-12 23:43:28 +0200244
Gilles Peskinec34faba2020-04-20 15:44:14 +0200245 Include alternative implementations of platform functions, which are
Gilles Peskine32e889d2020-04-12 23:43:28 +0200246 configurable function pointers that default to the built-in function.
247 This way we test that the function pointers exist and build correctly
248 without changing the behavior, and tests can verify that the function
249 pointers are used by modifying those pointers.
250
251 Exclude alternative implementations of library functions since they require
252 an implementation of the relevant functions and an xxx_alt.h header.
253 """
Gilles Peskinea8861e02023-09-05 20:20:51 +0200254 if name in (
255 'MBEDTLS_PLATFORM_GMTIME_R_ALT',
256 'MBEDTLS_PLATFORM_SETUP_TEARDOWN_ALT',
257 'MBEDTLS_PLATFORM_MS_TIME_ALT',
258 'MBEDTLS_PLATFORM_ZEROIZE_ALT',
259 ):
Gilles Peskinec34faba2020-04-20 15:44:14 +0200260 # Similar to non-platform xxx_ALT, requires platform_alt.h
261 return False
Gilles Peskine32e889d2020-04-12 23:43:28 +0200262 return name.startswith('MBEDTLS_PLATFORM_')
263
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200264def include_in_full(name):
265 """Rules for symbols in the "full" configuration."""
Gabor Mezei542fd382024-06-10 14:07:42 +0200266 if name in EXCLUDE_FROM_FULL:
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200267 return False
268 if name.endswith('_ALT'):
Gilles Peskine32e889d2020-04-12 23:43:28 +0200269 return is_seamless_alt(name)
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200270 return True
271
272def full_adapter(name, active, section):
273 """Config adapter for "full"."""
274 if not is_full_section(section):
275 return active
276 return include_in_full(name)
277
Gilles Peskinecfffc282020-04-12 13:55:45 +0200278# The baremetal configuration excludes options that require a library or
279# operating system feature that is typically not present on bare metal
280# systems. Features that are excluded from "full" won't be in "baremetal"
281# either (unless explicitly turned on in baremetal_adapter) so they don't
282# need to be repeated here.
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200283EXCLUDE_FROM_BAREMETAL = frozenset([
Gilles Peskinecfffc282020-04-12 13:55:45 +0200284 #pylint: disable=line-too-long
Gilles Peskine98f8f952020-04-20 15:38:39 +0200285 'MBEDTLS_ENTROPY_NV_SEED', # requires a filesystem and FS_IO or alternate NV seed hooks
Gilles Peskinecfffc282020-04-12 13:55:45 +0200286 'MBEDTLS_FS_IO', # requires a filesystem
Gilles Peskinecfffc282020-04-12 13:55:45 +0200287 'MBEDTLS_HAVE_TIME', # requires a clock
288 'MBEDTLS_HAVE_TIME_DATE', # requires a clock
289 'MBEDTLS_NET_C', # requires POSIX-like networking
290 'MBEDTLS_PLATFORM_FPRINTF_ALT', # requires FILE* from stdio.h
Gilles Peskine98f8f952020-04-20 15:38:39 +0200291 'MBEDTLS_PLATFORM_NV_SEED_ALT', # requires a filesystem and ENTROPY_NV_SEED
292 'MBEDTLS_PLATFORM_TIME_ALT', # requires a clock and HAVE_TIME
293 'MBEDTLS_PSA_CRYPTO_SE_C', # requires a filesystem and PSA_CRYPTO_STORAGE_C
Gilles Peskinecfffc282020-04-12 13:55:45 +0200294 'MBEDTLS_PSA_CRYPTO_STORAGE_C', # requires a filesystem
295 'MBEDTLS_PSA_ITS_FILE_C', # requires a filesystem
296 'MBEDTLS_THREADING_C', # requires a threading interface
297 'MBEDTLS_THREADING_PTHREAD', # requires pthread
298 'MBEDTLS_TIMING_C', # requires a clock
Dave Rodgman9be3cf02023-10-11 14:47:55 +0100299 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection
Dave Rodgman5b89c552023-10-10 14:59:02 +0100300 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection
Dave Rodgmanbe7915a2023-10-11 10:46:38 +0100301 'MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200302])
303
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200304def keep_in_baremetal(name):
305 """Rules for symbols in the "baremetal" configuration."""
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200306 if name in EXCLUDE_FROM_BAREMETAL:
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200307 return False
308 return True
309
310def baremetal_adapter(name, active, section):
311 """Config adapter for "baremetal"."""
312 if not is_full_section(section):
313 return active
314 if name == 'MBEDTLS_NO_PLATFORM_ENTROPY':
Gilles Peskinecfffc282020-04-12 13:55:45 +0200315 # No OS-provided entropy source
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200316 return True
317 return include_in_full(name) and keep_in_baremetal(name)
318
Gilles Peskine120f29d2021-09-01 19:51:19 +0200319# This set contains options that are mostly for debugging or test purposes,
320# and therefore should be excluded when doing code size measurements.
321# Options that are their own module (such as MBEDTLS_ERROR_C) are not listed
322# and therefore will be included when doing code size measurements.
323EXCLUDE_FOR_SIZE = frozenset([
324 'MBEDTLS_DEBUG_C', # large code size increase in TLS
325 'MBEDTLS_SELF_TEST', # increases the size of many modules
326 'MBEDTLS_TEST_HOOKS', # only useful with the hosted test framework, increases code size
327])
328
329def baremetal_size_adapter(name, active, section):
330 if name in EXCLUDE_FOR_SIZE:
331 return False
332 return baremetal_adapter(name, active, section)
333
Gilles Peskine31987c62020-01-31 14:23:30 +0100334def include_in_crypto(name):
335 """Rules for symbols in a crypto configuration."""
336 if name.startswith('MBEDTLS_X509_') or \
337 name.startswith('MBEDTLS_SSL_') or \
338 name.startswith('MBEDTLS_KEY_EXCHANGE_'):
339 return False
340 if name in [
Gilles Peskinecfffc282020-04-12 13:55:45 +0200341 'MBEDTLS_DEBUG_C', # part of libmbedtls
342 'MBEDTLS_NET_C', # part of libmbedtls
Nayna Jainc9deb182020-11-16 19:03:12 +0000343 'MBEDTLS_PKCS7_C', # part of libmbedx509
Gilles Peskine31987c62020-01-31 14:23:30 +0100344 ]:
345 return False
Gabor Mezei542fd382024-06-10 14:07:42 +0200346 if name in EXCLUDE_FROM_CRYPTO:
347 return False
Gilles Peskine31987c62020-01-31 14:23:30 +0100348 return True
349
350def crypto_adapter(adapter):
351 """Modify an adapter to disable non-crypto symbols.
352
353 ``crypto_adapter(adapter)(name, active, section)`` is like
354 ``adapter(name, active, section)``, but unsets all X.509 and TLS symbols.
355 """
356 def continuation(name, active, section):
357 if not include_in_crypto(name):
358 return False
359 if adapter is None:
360 return active
361 return adapter(name, active, section)
362 return continuation
363
Gilles Peskineed5c21d2022-06-27 23:02:09 +0200364DEPRECATED = frozenset([
365 'MBEDTLS_PSA_CRYPTO_SE_C',
Gabor Mezei542fd382024-06-10 14:07:42 +0200366 *PSA_DEPRECATED_FEATURE
Gilles Peskineed5c21d2022-06-27 23:02:09 +0200367])
Gilles Peskine30de2e82020-04-20 21:39:22 +0200368def no_deprecated_adapter(adapter):
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200369 """Modify an adapter to disable deprecated symbols.
370
Gilles Peskine30de2e82020-04-20 21:39:22 +0200371 ``no_deprecated_adapter(adapter)(name, active, section)`` is like
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200372 ``adapter(name, active, section)``, but unsets all deprecated symbols
373 and sets ``MBEDTLS_DEPRECATED_REMOVED``.
374 """
375 def continuation(name, active, section):
376 if name == 'MBEDTLS_DEPRECATED_REMOVED':
377 return True
Gilles Peskineed5c21d2022-06-27 23:02:09 +0200378 if name in DEPRECATED:
379 return False
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200380 if adapter is None:
381 return active
382 return adapter(name, active, section)
383 return continuation
384
Paul Elliottfb81f772023-10-18 17:44:59 +0100385def no_platform_adapter(adapter):
386 """Modify an adapter to disable platform symbols.
387
388 ``no_platform_adapter(adapter)(name, active, section)`` is like
389 ``adapter(name, active, section)``, but unsets all platform symbols other
390 ``than MBEDTLS_PLATFORM_C.
391 """
392 def continuation(name, active, section):
393 # Allow MBEDTLS_PLATFORM_C but remove all other platform symbols.
394 if name.startswith('MBEDTLS_PLATFORM_') and name != 'MBEDTLS_PLATFORM_C':
395 return False
396 if adapter is None:
397 return active
398 return adapter(name, active, section)
399 return continuation
400
Gabor Mezei3678dee2024-06-04 19:58:43 +0200401class ConfigFile(metaclass=ABCMeta):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200402 """Representation of a configuration file."""
403
Gabor Mezei93a6d1f2024-06-26 18:01:09 +0200404 def __init__(self, default_path, name, filename=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200405 """Check if the config file exists."""
Gilles Peskineb4063892019-07-27 21:36:44 +0200406 if filename is None:
Gabor Mezei3678dee2024-06-04 19:58:43 +0200407 for candidate in default_path:
Gilles Peskinece674a92020-03-24 15:37:00 +0100408 if os.path.lexists(candidate):
409 filename = candidate
Gilles Peskine208e4ec2019-07-29 23:43:20 +0200410 break
Gilles Peskinece674a92020-03-24 15:37:00 +0100411 else:
Gabor Mezei92065ed2024-06-07 13:47:59 +0200412 raise ValueError(f'{name} configuration file not found: '
413 f'{filename if filename else default_path}')
Gilles Peskineb4063892019-07-27 21:36:44 +0200414
Gabor Mezei3678dee2024-06-04 19:58:43 +0200415 self.filename = filename
416 self.templates = []
417 self.current_section = None
418 self.inclusion_guard = None
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200419 self.modified = False
Gilles Peskineb4063892019-07-27 21:36:44 +0200420
421 _define_line_regexp = (r'(?P<indentation>\s*)' +
422 r'(?P<commented_out>(//\s*)?)' +
423 r'(?P<define>#\s*define\s+)' +
424 r'(?P<name>\w+)' +
425 r'(?P<arguments>(?:\((?:\w|\s|,)*\))?)' +
426 r'(?P<separator>\s*)' +
427 r'(?P<value>.*)')
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200428 _ifndef_line_regexp = r'#ifndef (?P<inclusion_guard>\w+)'
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200429 _section_line_regexp = (r'\s*/?\*+\s*[\\@]name\s+SECTION:\s*' +
430 r'(?P<section>.*)[ */]*')
431 _config_line_regexp = re.compile(r'|'.join([_define_line_regexp,
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200432 _ifndef_line_regexp,
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200433 _section_line_regexp]))
Gilles Peskineb4063892019-07-27 21:36:44 +0200434 def _parse_line(self, line):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200435 """Parse a line in the config file, save the templates representing the lines
436 and return the corresponding setting element.
437 """
Gilles Peskineb4063892019-07-27 21:36:44 +0200438 line = line.rstrip('\r\n')
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200439 m = re.match(self._config_line_regexp, line)
440 if m is None:
Gabor Mezei3678dee2024-06-04 19:58:43 +0200441 self.templates.append(line)
442 return None
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200443 elif m.group('section'):
444 self.current_section = m.group('section')
Gabor Mezei3678dee2024-06-04 19:58:43 +0200445 self.templates.append(line)
446 return None
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200447 elif m.group('inclusion_guard') and self.inclusion_guard is None:
448 self.inclusion_guard = m.group('inclusion_guard')
Gabor Mezei3678dee2024-06-04 19:58:43 +0200449 self.templates.append(line)
450 return None
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200451 else:
Gilles Peskineb4063892019-07-27 21:36:44 +0200452 active = not m.group('commented_out')
453 name = m.group('name')
454 value = m.group('value')
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200455 if name == self.inclusion_guard and value == '':
456 # The file double-inclusion guard is not an option.
Gabor Mezei3678dee2024-06-04 19:58:43 +0200457 self.templates.append(line)
458 return None
Gilles Peskineb4063892019-07-27 21:36:44 +0200459 template = (name,
460 m.group('indentation'),
461 m.group('define') + name +
462 m.group('arguments') + m.group('separator'))
Gabor Mezei3678dee2024-06-04 19:58:43 +0200463 self.templates.append(template)
Gilles Peskineb4063892019-07-27 21:36:44 +0200464
Gabor Mezei3678dee2024-06-04 19:58:43 +0200465 return (active, name, value, self.current_section)
466
467 def parse_file(self):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200468 """Parse the whole file and return the settings."""
Gabor Mezei3678dee2024-06-04 19:58:43 +0200469 with open(self.filename, 'r', encoding='utf-8') as file:
470 for line in file:
471 setting = self._parse_line(line)
472 if setting is not None:
473 yield setting
474 self.current_section = None
475
Gabor Mezeie7742b32024-06-26 18:04:09 +0200476 #pylint: disable=no-self-use
477 def _format_template(self, setting, indent, middle):
478 """Build a line for mbedtls_config.h for the given setting.
479
480 The line has the form "<indent>#define <name> <value>"
481 where <middle> is "#define <name> ".
482 """
483 value = setting.value
484 if value is None:
485 value = ''
486 # Normally the whitespace to separate the symbol name from the
487 # value is part of middle, and there's no whitespace for a symbol
488 # with no value. But if a symbol has been changed from having a
489 # value to not having one, the whitespace is wrong, so fix it.
490 if value:
491 if middle[-1] not in '\t ':
492 middle += ' '
493 else:
494 middle = middle.rstrip()
495 return ''.join([indent,
496 '' if setting.active else '//',
497 middle,
498 value]).rstrip()
Gabor Mezei3678dee2024-06-04 19:58:43 +0200499
500 def write_to_stream(self, settings, output):
501 """Write the whole configuration to output."""
502 for template in self.templates:
503 if isinstance(template, str):
504 line = template
505 else:
Gabor Mezeie7742b32024-06-26 18:04:09 +0200506 name, indent, middle = template
507 line = self._format_template(settings[name], indent, middle)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200508 output.write(line + '\n')
509
510 def write(self, settings, filename=None):
511 """Write the whole configuration to the file it was read from.
512
513 If filename is specified, write to this file instead.
514 """
515 if filename is None:
516 filename = self.filename
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200517
518 # Not modified so no need to write to the file
519 if not self.modified and filename == self.filename:
520 return
521
Gabor Mezei3678dee2024-06-04 19:58:43 +0200522 with open(filename, 'w', encoding='utf-8') as output:
523 self.write_to_stream(settings, output)
524
Gabor Mezeif77722d2024-06-28 16:49:33 +0200525class MbedTLSConfigFile(ConfigFile):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200526 """Representation of an MbedTLS configuration file."""
527
Gabor Mezei3678dee2024-06-04 19:58:43 +0200528 _path_in_tree = 'include/mbedtls/mbedtls_config.h'
529 default_path = [_path_in_tree,
530 os.path.join(os.path.dirname(__file__),
531 os.pardir,
532 _path_in_tree),
533 os.path.join(os.path.dirname(os.path.abspath(os.path.dirname(__file__))),
534 _path_in_tree)]
535
536 def __init__(self, filename=None):
Gabor Mezei93a6d1f2024-06-26 18:01:09 +0200537 super().__init__(self.default_path, 'Mbed TLS', filename)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200538 self.current_section = 'header'
539
Gabor Mezei3678dee2024-06-04 19:58:43 +0200540class CryptoConfigFile(ConfigFile):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200541 """Representation of an Crypto configuration file."""
542
Gabor Mezei3678dee2024-06-04 19:58:43 +0200543 _path_in_tree = 'tf-psa-crypto/include/psa/crypto_config.h'
544 default_path = [_path_in_tree,
545 os.path.join(os.path.dirname(__file__),
546 os.pardir,
547 _path_in_tree),
548 os.path.join(os.path.dirname(os.path.abspath(os.path.dirname(__file__))),
549 _path_in_tree)]
550
551 def __init__(self, filename=None):
Gabor Mezei93a6d1f2024-06-26 18:01:09 +0200552 super().__init__(self.default_path, 'Crypto', filename)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200553
Gabor Mezeif77722d2024-06-28 16:49:33 +0200554class MbedTLSConfig(Config):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200555 """Representation of the Mbed TLS configuration.
Gabor Mezei3678dee2024-06-04 19:58:43 +0200556
557 See the documentation of the `Config` class for methods to query
558 and modify the configuration.
559 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200560 def __init__(self, filename=None):
Gabor Mezei3678dee2024-06-04 19:58:43 +0200561 """Read the Mbed TLS configuration file."""
562 super().__init__()
Gabor Mezeif77722d2024-06-28 16:49:33 +0200563 self.configfile = MbedTLSConfigFile(filename)
Gabor Mezeiee521b62024-06-07 13:50:41 +0200564 self.settings.update({name: Setting(active, name, value, section)
Gabor Mezei92065ed2024-06-07 13:47:59 +0200565 for (active, name, value, section)
Gabor Mezeiee521b62024-06-07 13:50:41 +0200566 in self.configfile.parse_file()})
Gabor Mezei3678dee2024-06-04 19:58:43 +0200567
568 def set(self, name, value=None):
569 if name not in self.settings:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200570 self.configfile.templates.append((name, '', '#define ' + name + ' '))
571
Gabor Mezei3678dee2024-06-04 19:58:43 +0200572 super().set(name, value)
Gilles Peskineb4063892019-07-27 21:36:44 +0200573
574 def write(self, filename=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200575 """Write the whole configuration to the file it was read from.
576
577 If filename is specified, write to this file instead.
578 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200579 self.configfile.write(self.settings, filename)
Gilles Peskineb4063892019-07-27 21:36:44 +0200580
Gabor Mezeiee521b62024-06-07 13:50:41 +0200581 def filename(self):
582 return self.configfile.filename
Gabor Mezei3678dee2024-06-04 19:58:43 +0200583
584class CryptoConfig(Config):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200585 """Representation of the PSA crypto configuration.
Gabor Mezei3678dee2024-06-04 19:58:43 +0200586
587 See the documentation of the `Config` class for methods to query
588 and modify the configuration.
589 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200590 def __init__(self, filename=None):
Gabor Mezei3678dee2024-06-04 19:58:43 +0200591 """Read the PSA crypto configuration file."""
592 super().__init__()
Gabor Mezeiee521b62024-06-07 13:50:41 +0200593 self.configfile = CryptoConfigFile(filename)
594 self.settings.update({name: Setting(active, name, value, section)
Gabor Mezei92065ed2024-06-07 13:47:59 +0200595 for (active, name, value, section)
Gabor Mezeiee521b62024-06-07 13:50:41 +0200596 in self.configfile.parse_file()})
Gabor Mezei3678dee2024-06-04 19:58:43 +0200597
Gabor Mezeid723b512024-06-07 15:31:52 +0200598 def set(self, name, value='1'):
Gabor Mezei542fd382024-06-10 14:07:42 +0200599 if name in PSA_UNSUPPORTED_FEATURE:
Gabor Mezei92065ed2024-06-07 13:47:59 +0200600 raise ValueError(f'Feature is unsupported: \'{name}\'')
Gabor Mezei542fd382024-06-10 14:07:42 +0200601 if name in PSA_UNSTABLE_FEATURE:
Gabor Mezei92065ed2024-06-07 13:47:59 +0200602 raise ValueError(f'Feature is unstable: \'{name}\'')
Gabor Mezei3678dee2024-06-04 19:58:43 +0200603
Gabor Mezeid723b512024-06-07 15:31:52 +0200604 # The default value in the crypto config is '1'
605 if not value:
606 value = '1'
607
Gabor Mezei3678dee2024-06-04 19:58:43 +0200608 if name not in self.settings:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200609 self.configfile.templates.append((name, '', '#define ' + name + ' '))
610
Gabor Mezei3678dee2024-06-04 19:58:43 +0200611 super().set(name, value)
612
613 def write(self, filename=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200614 """Write the whole configuration to the file it was read from.
615
616 If filename is specified, write to this file instead.
617 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200618 self.configfile.write(self.settings, filename)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200619
Gabor Mezeiee521b62024-06-07 13:50:41 +0200620 def filename(self):
621 return self.configfile.filename
Gabor Mezei3678dee2024-06-04 19:58:43 +0200622
Gabor Mezeiee521b62024-06-07 13:50:41 +0200623class MultiConfig(Config):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200624 """Representation of MbedTLS and PSA crypto configuration
625
626 See the documentation of the `Config` class for methods to query
627 and modify the configuration.
628 """
Gabor Mezei3678dee2024-06-04 19:58:43 +0200629
Gabor Mezeiee521b62024-06-07 13:50:41 +0200630 def __init__(self, mbedtls_filename=None, crypto__filename=None):
631 super().__init__()
Gabor Mezeif77722d2024-06-28 16:49:33 +0200632 self.mbedtls_configfile = MbedTLSConfigFile(mbedtls_filename)
Gabor Mezeiee521b62024-06-07 13:50:41 +0200633 self.crypto_configfile = CryptoConfigFile(crypto__filename)
634 self.settings.update({name: Setting(active, name, value, section, configfile)
635 for configfile in [self.mbedtls_configfile, self.crypto_configfile]
636 for (active, name, value, section) in configfile.parse_file()})
Gabor Mezei3678dee2024-06-04 19:58:43 +0200637
638 _crypto_regexp = re.compile(r'$PSA_.*')
Gabor Mezeiee521b62024-06-07 13:50:41 +0200639 def _get_configfile(self, name):
640 """Find a config type for a setting name
641 """
642 if name in self.settings:
643 return self.settings[name].configfile
644 elif re.match(self._crypto_regexp, name):
645 return self.crypto_configfile
Gabor Mezei3678dee2024-06-04 19:58:43 +0200646 else:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200647 return self.mbedtls_configfile
Gabor Mezei3678dee2024-06-04 19:58:43 +0200648
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200649 def __setitem__(self, name, value):
650 super().__setitem__(name, value)
651 self.settings[name].configfile.modified = True
652
Gabor Mezei3678dee2024-06-04 19:58:43 +0200653 def set(self, name, value=None):
Gabor Mezeiee521b62024-06-07 13:50:41 +0200654 configfile = self._get_configfile(name)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200655
Gabor Mezeiee521b62024-06-07 13:50:41 +0200656 if configfile == self.crypto_configfile:
Gabor Mezei542fd382024-06-10 14:07:42 +0200657 if name in PSA_UNSUPPORTED_FEATURE:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200658 raise ValueError(f'Feature is unsupported: \'{name}\'')
Gabor Mezei542fd382024-06-10 14:07:42 +0200659 if name in PSA_UNSTABLE_FEATURE:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200660 raise ValueError(f'Feature is unstable: \'{name}\'')
661
Gabor Mezeid723b512024-06-07 15:31:52 +0200662 # The default value in the crypto config is '1'
663 if not value:
664 value = '1'
665
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200666 if name in self.settings:
667 setting = self.settings[name]
668 if not setting.active or (value is not None and setting.value != value):
669 configfile.modified = True
670 else:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200671 configfile.templates.append((name, '', '#define ' + name + ' '))
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200672 configfile.modified = True
Gabor Mezeiee521b62024-06-07 13:50:41 +0200673
674 super().set(name, value)
675
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200676 def unset(self, name):
677 if name in self.settings and self.settings[name].active:
678 self.settings[name].configfile.modified = True
679
680 super().unset(name)
681
682 def adapt(self, adapter):
683 # Determine if the config files will be modified
684 unmodified = {config for config in [self.mbedtls_configfile, self.crypto_configfile]
685 if not config.modified}
686 if unmodified:
687 for setting in self.settings.values():
688 if not setting.configfile.modified and \
689 setting.active != adapter(setting.name, setting.active, setting.section):
690 setting.configfile.modified = True
691 unmodified.remove(setting.configfile)
692 if not unmodified:
693 break
694
695 super().adapt(adapter)
696
697 def change_matching(self, regexs, enable):
698 # Determine if the config files will be modified
699 if regexs:
700 regex = re.compile('|'.join(regexs))
701 unmodified = {config for config in [self.mbedtls_configfile, self.crypto_configfile]
702 if not config.modified}
703 if unmodified:
704 for setting in self.settings.values():
705 if not setting.configfile.modified and \
706 setting.active != enable and \
707 regex.search(setting.name):
708 setting.configfile.modified = True
709 unmodified.remove(setting.configfile)
710 if not unmodified:
711 break
712
713 super().change_matching(regexs, enable)
714
Gabor Mezei3678dee2024-06-04 19:58:43 +0200715 def write(self, mbedtls_file=None, crypto_file=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200716 """Write the whole configuration to the file it was read from.
717
718 If mbedtls_file or crypto_file is specified, write the specific configuration
719 to the corresponding file instead.
720 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200721 self.mbedtls_configfile.write(self.settings, mbedtls_file)
722 self.crypto_configfile.write(self.settings, crypto_file)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200723
Gabor Mezeiee521b62024-06-07 13:50:41 +0200724 def filename(self, name=None):
725 if not name:
726 return [config.filename for config in [self.mbedtls_configfile, self.crypto_configfile]]
727
728 return self._get_configfile(name).filename
Gilles Peskineb4063892019-07-27 21:36:44 +0200729
730if __name__ == '__main__':
Gabor Mezei92065ed2024-06-07 13:47:59 +0200731 #pylint: disable=too-many-statements
Gilles Peskineb4063892019-07-27 21:36:44 +0200732 def main():
Bence Szépkútibb0cfeb2021-05-28 09:42:25 +0200733 """Command line mbedtls_config.h manipulation tool."""
Gilles Peskineb4063892019-07-27 21:36:44 +0200734 parser = argparse.ArgumentParser(description="""
Fredrik Hesse0ec8a902021-10-04 22:13:51 +0200735 Mbed TLS configuration file manipulation tool.
Gilles Peskineb4063892019-07-27 21:36:44 +0200736 """)
737 parser.add_argument('--file', '-f',
738 help="""File to read (and modify if requested).
739 Default: {}.
Gabor Mezeif77722d2024-06-28 16:49:33 +0200740 """.format(MbedTLSConfigFile.default_path))
Gabor Mezei3678dee2024-06-04 19:58:43 +0200741 parser.add_argument('--cryptofile', '-c',
742 help="""Crypto file to read (and modify if requested).
743 Default: {}.
744 """.format(CryptoConfigFile.default_path))
Gilles Peskineb4063892019-07-27 21:36:44 +0200745 parser.add_argument('--force', '-o',
Gilles Peskine435ce222019-08-01 23:13:47 +0200746 action='store_true',
Gilles Peskineb4063892019-07-27 21:36:44 +0200747 help="""For the set command, if SYMBOL is not
748 present, add a definition for it.""")
Gilles Peskinec190c902019-08-01 23:31:05 +0200749 parser.add_argument('--write', '-w', metavar='FILE',
Gilles Peskine40f103c2019-07-27 23:44:01 +0200750 help="""File to write to instead of the input file.""")
Gilles Peskineb4063892019-07-27 21:36:44 +0200751 subparsers = parser.add_subparsers(dest='command',
752 title='Commands')
753 parser_get = subparsers.add_parser('get',
754 help="""Find the value of SYMBOL
755 and print it. Exit with
756 status 0 if a #define for SYMBOL is
757 found, 1 otherwise.
758 """)
759 parser_get.add_argument('symbol', metavar='SYMBOL')
760 parser_set = subparsers.add_parser('set',
761 help="""Set SYMBOL to VALUE.
762 If VALUE is omitted, just uncomment
763 the #define for SYMBOL.
764 Error out of a line defining
765 SYMBOL (commented or not) is not
766 found, unless --force is passed.
767 """)
768 parser_set.add_argument('symbol', metavar='SYMBOL')
Gilles Peskine0c7fcd22019-08-01 23:14:00 +0200769 parser_set.add_argument('value', metavar='VALUE', nargs='?',
770 default='')
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200771 parser_set_all = subparsers.add_parser('set-all',
772 help="""Uncomment all #define
773 whose name contains a match for
774 REGEX.""")
775 parser_set_all.add_argument('regexs', metavar='REGEX', nargs='*')
Gilles Peskineb4063892019-07-27 21:36:44 +0200776 parser_unset = subparsers.add_parser('unset',
777 help="""Comment out the #define
778 for SYMBOL. Do nothing if none
779 is present.""")
780 parser_unset.add_argument('symbol', metavar='SYMBOL')
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200781 parser_unset_all = subparsers.add_parser('unset-all',
782 help="""Comment out all #define
783 whose name contains a match for
784 REGEX.""")
785 parser_unset_all.add_argument('regexs', metavar='REGEX', nargs='*')
Gilles Peskineb4063892019-07-27 21:36:44 +0200786
787 def add_adapter(name, function, description):
788 subparser = subparsers.add_parser(name, help=description)
789 subparser.set_defaults(adapter=function)
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200790 add_adapter('baremetal', baremetal_adapter,
791 """Like full, but exclude features that require platform
792 features such as file input-output.""")
Gilles Peskine120f29d2021-09-01 19:51:19 +0200793 add_adapter('baremetal_size', baremetal_size_adapter,
794 """Like baremetal, but exclude debugging features.
795 Useful for code size measurements.""")
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200796 add_adapter('full', full_adapter,
797 """Uncomment most features.
798 Exclude alternative implementations and platform support
799 options, as well as some options that are awkward to test.
800 """)
Gilles Peskine30de2e82020-04-20 21:39:22 +0200801 add_adapter('full_no_deprecated', no_deprecated_adapter(full_adapter),
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200802 """Uncomment most non-deprecated features.
803 Like "full", but without deprecated features.
804 """)
Paul Elliottfb81f772023-10-18 17:44:59 +0100805 add_adapter('full_no_platform', no_platform_adapter(full_adapter),
806 """Uncomment most non-platform features.
807 Like "full", but without platform features.
808 """)
Gilles Peskineb4063892019-07-27 21:36:44 +0200809 add_adapter('realfull', realfull_adapter,
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200810 """Uncomment all boolean #defines.
811 Suitable for generating documentation, but not for building.""")
Gilles Peskine31987c62020-01-31 14:23:30 +0100812 add_adapter('crypto', crypto_adapter(None),
813 """Only include crypto features. Exclude X.509 and TLS.""")
814 add_adapter('crypto_baremetal', crypto_adapter(baremetal_adapter),
815 """Like baremetal, but with only crypto features,
816 excluding X.509 and TLS.""")
817 add_adapter('crypto_full', crypto_adapter(full_adapter),
818 """Like full, but with only crypto features,
819 excluding X.509 and TLS.""")
Gilles Peskineb4063892019-07-27 21:36:44 +0200820
821 args = parser.parse_args()
Gabor Mezei3678dee2024-06-04 19:58:43 +0200822 config = MultiConfig(args.file, args.cryptofile)
Gilles Peskine90b30b62019-07-28 00:36:53 +0200823 if args.command is None:
824 parser.print_help()
825 return 1
826 elif args.command == 'get':
Gilles Peskineb4063892019-07-27 21:36:44 +0200827 if args.symbol in config:
828 value = config[args.symbol]
829 if value:
830 sys.stdout.write(value + '\n')
Gilles Peskinee22a4da2020-03-24 15:43:49 +0100831 return 0 if args.symbol in config else 1
Gilles Peskineb4063892019-07-27 21:36:44 +0200832 elif args.command == 'set':
Gilles Peskine98eb3652019-07-28 16:39:19 +0200833 if not args.force and args.symbol not in config.settings:
Gilles Peskineb4063892019-07-27 21:36:44 +0200834 sys.stderr.write("A #define for the symbol {} "
Gilles Peskine221df1e2019-08-01 23:14:29 +0200835 "was not found in {}\n"
Gabor Mezei3678dee2024-06-04 19:58:43 +0200836 .format(args.symbol, config.filename(args.symbol)))
Gilles Peskineb4063892019-07-27 21:36:44 +0200837 return 1
838 config.set(args.symbol, value=args.value)
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200839 elif args.command == 'set-all':
840 config.change_matching(args.regexs, True)
Gilles Peskineb4063892019-07-27 21:36:44 +0200841 elif args.command == 'unset':
842 config.unset(args.symbol)
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200843 elif args.command == 'unset-all':
844 config.change_matching(args.regexs, False)
Gilles Peskineb4063892019-07-27 21:36:44 +0200845 else:
846 config.adapt(args.adapter)
Gilles Peskine40f103c2019-07-27 23:44:01 +0200847 config.write(args.write)
Gilles Peskinee22a4da2020-03-24 15:43:49 +0100848 return 0
Gilles Peskineb4063892019-07-27 21:36:44 +0200849
850 # Import modules only used by main only if main is defined and called.
851 # pylint: disable=wrong-import-position
852 import argparse
853 import sys
854 sys.exit(main())