blob: cd6a4b4f4abc85c079ce2904ee2b97d26be8fadf [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 Mezei542fd382024-06-10 14:07:42 +0200191EXCLUDE_FROM_CRYPTO = frozenset(
192 PSA_UNSUPPORTED_FEATURE |
193 PSA_DEPRECATED_FEATURE |
194 PSA_UNSTABLE_FEATURE
195)
196
Gilles Peskinecfffc282020-04-12 13:55:45 +0200197# The goal of the full configuration is to have everything that can be tested
198# together. This includes deprecated or insecure options. It excludes:
199# * Options that require additional build dependencies or unusual hardware.
200# * Options that make testing less effective.
Gilles Peskinec9d04332020-04-16 20:50:17 +0200201# * Options that are incompatible with other options, or more generally that
202# interact with other parts of the code in such a way that a bulk enabling
203# is not a good way to test them.
Gilles Peskinecfffc282020-04-12 13:55:45 +0200204# * Options that remove features.
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200205EXCLUDE_FROM_FULL = frozenset([
Gilles Peskinecfffc282020-04-12 13:55:45 +0200206 #pylint: disable=line-too-long
Yanray Wanga8704672023-04-20 17:16:48 +0800207 'MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH', # interacts with CTR_DRBG_128_BIT_KEY
Gilles Peskinea8861e02023-09-05 20:20:51 +0200208 'MBEDTLS_AES_USE_HARDWARE_ONLY', # hardware dependency
Yanray Wang42be1ba2023-11-23 14:28:47 +0800209 'MBEDTLS_BLOCK_CIPHER_NO_DECRYPT', # incompatible with ECB in PSA, CBC/XTS/NIST_KW/DES
Gilles Peskinec9d04332020-04-16 20:50:17 +0200210 'MBEDTLS_CTR_DRBG_USE_128_BIT_KEY', # interacts with ENTROPY_FORCE_SHA256
Gilles Peskinecfffc282020-04-12 13:55:45 +0200211 'MBEDTLS_DEPRECATED_REMOVED', # conflicts with deprecated options
Gilles Peskine90581ee2020-04-12 14:02:47 +0200212 'MBEDTLS_DEPRECATED_WARNING', # conflicts with deprecated options
Gilles Peskinec9d04332020-04-16 20:50:17 +0200213 'MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED', # influences the use of ECDH in TLS
Janos Follath5b7c38f2023-08-01 08:51:12 +0100214 'MBEDTLS_ECP_WITH_MPI_UINT', # disables the default ECP and is experimental
Gilles Peskinec9d04332020-04-16 20:50:17 +0200215 'MBEDTLS_ENTROPY_FORCE_SHA256', # interacts with CTR_DRBG_128_BIT_KEY
Gilles Peskinecfffc282020-04-12 13:55:45 +0200216 'MBEDTLS_HAVE_SSE2', # hardware dependency
217 'MBEDTLS_MEMORY_BACKTRACE', # depends on MEMORY_BUFFER_ALLOC_C
218 'MBEDTLS_MEMORY_BUFFER_ALLOC_C', # makes sanitizers (e.g. ASan) less effective
219 'MBEDTLS_MEMORY_DEBUG', # depends on MEMORY_BUFFER_ALLOC_C
Gilles Peskinec9d04332020-04-16 20:50:17 +0200220 'MBEDTLS_NO_64BIT_MULTIPLICATION', # influences anything that uses bignum
Gilles Peskinecfffc282020-04-12 13:55:45 +0200221 'MBEDTLS_NO_DEFAULT_ENTROPY_SOURCES', # removes a feature
222 'MBEDTLS_NO_PLATFORM_ENTROPY', # removes a feature
Gilles Peskinec9d04332020-04-16 20:50:17 +0200223 'MBEDTLS_NO_UDBL_DIVISION', # influences anything that uses bignum
Gilles Peskineefaee9a2023-09-20 20:49:47 +0200224 'MBEDTLS_PSA_P256M_DRIVER_ENABLED', # influences SECP256R1 KeyGen/ECDH/ECDSA
Gilles Peskinecfffc282020-04-12 13:55:45 +0200225 'MBEDTLS_PLATFORM_NO_STD_FUNCTIONS', # removes a feature
David Horstmann6f8c95b2024-03-14 14:52:45 +0000226 'MBEDTLS_PSA_ASSUME_EXCLUSIVE_BUFFERS', # removes a feature
Gilles Peskinef08b3f82020-11-13 17:36:48 +0100227 'MBEDTLS_PSA_CRYPTO_EXTERNAL_RNG', # behavior change + build dependency
Ronald Cronc3623db2020-10-29 10:51:32 +0100228 'MBEDTLS_PSA_CRYPTO_KEY_ID_ENCODES_OWNER', # incompatible with USE_PSA_CRYPTO
Gilles Peskinecfffc282020-04-12 13:55:45 +0200229 'MBEDTLS_PSA_CRYPTO_SPM', # platform dependency (PSA SPM)
Gilles Peskinea08def92023-04-28 21:01:49 +0200230 'MBEDTLS_PSA_INJECT_ENTROPY', # conflicts with platform entropy sources
Gilles Peskinec9d04332020-04-16 20:50:17 +0200231 'MBEDTLS_RSA_NO_CRT', # influences the use of RSA in X.509 and TLS
Tom Cosgrove87fbfb52022-03-15 10:51:52 +0000232 'MBEDTLS_SHA256_USE_A64_CRYPTO_ONLY', # interacts with *_USE_A64_CRYPTO_IF_PRESENT
Dave Rodgman9be3cf02023-10-11 14:47:55 +0100233 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_ONLY', # interacts with *_USE_ARMV8_A_CRYPTO_IF_PRESENT
Tom Cosgrove87fbfb52022-03-15 10:51:52 +0000234 'MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY', # interacts with *_USE_A64_CRYPTO_IF_PRESENT
Dave Rodgman7cb635a2023-10-12 16:14:51 +0100235 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT', # setting *_USE_ARMV8_A_CRYPTO is sufficient
Manuel Pégourié-Gonnard6240def2020-07-10 09:35:54 +0200236 'MBEDTLS_TEST_CONSTANT_FLOW_MEMSAN', # build dependency (clang+memsan)
Manuel Pégourié-Gonnard73afa372020-08-19 10:27:38 +0200237 'MBEDTLS_TEST_CONSTANT_FLOW_VALGRIND', # build dependency (valgrind headers)
Hanno Beckere1113562019-06-12 13:59:14 +0100238 'MBEDTLS_X509_REMOVE_INFO', # removes a feature
Gabor Mezei542fd382024-06-10 14:07:42 +0200239 *PSA_UNSUPPORTED_FEATURE,
240 *PSA_DEPRECATED_FEATURE,
241 *PSA_UNSTABLE_FEATURE
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200242])
243
Gilles Peskine32e889d2020-04-12 23:43:28 +0200244def is_seamless_alt(name):
Gilles Peskinec34faba2020-04-20 15:44:14 +0200245 """Whether the xxx_ALT symbol should be included in the full configuration.
Gilles Peskine32e889d2020-04-12 23:43:28 +0200246
Gilles Peskinec34faba2020-04-20 15:44:14 +0200247 Include alternative implementations of platform functions, which are
Gilles Peskine32e889d2020-04-12 23:43:28 +0200248 configurable function pointers that default to the built-in function.
249 This way we test that the function pointers exist and build correctly
250 without changing the behavior, and tests can verify that the function
251 pointers are used by modifying those pointers.
252
253 Exclude alternative implementations of library functions since they require
254 an implementation of the relevant functions and an xxx_alt.h header.
255 """
Gilles Peskinea8861e02023-09-05 20:20:51 +0200256 if name in (
257 'MBEDTLS_PLATFORM_GMTIME_R_ALT',
258 'MBEDTLS_PLATFORM_SETUP_TEARDOWN_ALT',
259 'MBEDTLS_PLATFORM_MS_TIME_ALT',
260 'MBEDTLS_PLATFORM_ZEROIZE_ALT',
261 ):
Gilles Peskinec34faba2020-04-20 15:44:14 +0200262 # Similar to non-platform xxx_ALT, requires platform_alt.h
263 return False
Gilles Peskine32e889d2020-04-12 23:43:28 +0200264 return name.startswith('MBEDTLS_PLATFORM_')
265
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200266def include_in_full(name):
267 """Rules for symbols in the "full" configuration."""
Gabor Mezei542fd382024-06-10 14:07:42 +0200268 if name in EXCLUDE_FROM_FULL:
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200269 return False
270 if name.endswith('_ALT'):
Gilles Peskine32e889d2020-04-12 23:43:28 +0200271 return is_seamless_alt(name)
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200272 return True
273
274def full_adapter(name, active, section):
275 """Config adapter for "full"."""
276 if not is_full_section(section):
277 return active
278 return include_in_full(name)
279
Gilles Peskinecfffc282020-04-12 13:55:45 +0200280# The baremetal configuration excludes options that require a library or
281# operating system feature that is typically not present on bare metal
282# systems. Features that are excluded from "full" won't be in "baremetal"
283# either (unless explicitly turned on in baremetal_adapter) so they don't
284# need to be repeated here.
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200285EXCLUDE_FROM_BAREMETAL = frozenset([
Gilles Peskinecfffc282020-04-12 13:55:45 +0200286 #pylint: disable=line-too-long
Gilles Peskine98f8f952020-04-20 15:38:39 +0200287 'MBEDTLS_ENTROPY_NV_SEED', # requires a filesystem and FS_IO or alternate NV seed hooks
Gilles Peskinecfffc282020-04-12 13:55:45 +0200288 'MBEDTLS_FS_IO', # requires a filesystem
Gilles Peskinecfffc282020-04-12 13:55:45 +0200289 'MBEDTLS_HAVE_TIME', # requires a clock
290 'MBEDTLS_HAVE_TIME_DATE', # requires a clock
291 'MBEDTLS_NET_C', # requires POSIX-like networking
292 'MBEDTLS_PLATFORM_FPRINTF_ALT', # requires FILE* from stdio.h
Gilles Peskine98f8f952020-04-20 15:38:39 +0200293 'MBEDTLS_PLATFORM_NV_SEED_ALT', # requires a filesystem and ENTROPY_NV_SEED
294 'MBEDTLS_PLATFORM_TIME_ALT', # requires a clock and HAVE_TIME
295 'MBEDTLS_PSA_CRYPTO_SE_C', # requires a filesystem and PSA_CRYPTO_STORAGE_C
Gilles Peskinecfffc282020-04-12 13:55:45 +0200296 'MBEDTLS_PSA_CRYPTO_STORAGE_C', # requires a filesystem
297 'MBEDTLS_PSA_ITS_FILE_C', # requires a filesystem
298 'MBEDTLS_THREADING_C', # requires a threading interface
299 'MBEDTLS_THREADING_PTHREAD', # requires pthread
300 'MBEDTLS_TIMING_C', # requires a clock
Dave Rodgman9be3cf02023-10-11 14:47:55 +0100301 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection
Dave Rodgman5b89c552023-10-10 14:59:02 +0100302 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection
Dave Rodgmanbe7915a2023-10-11 10:46:38 +0100303 'MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200304])
305
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200306def keep_in_baremetal(name):
307 """Rules for symbols in the "baremetal" configuration."""
Gilles Peskinebbaa2b72020-04-12 13:33:57 +0200308 if name in EXCLUDE_FROM_BAREMETAL:
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200309 return False
310 return True
311
312def baremetal_adapter(name, active, section):
313 """Config adapter for "baremetal"."""
314 if not is_full_section(section):
315 return active
316 if name == 'MBEDTLS_NO_PLATFORM_ENTROPY':
Gilles Peskinecfffc282020-04-12 13:55:45 +0200317 # No OS-provided entropy source
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200318 return True
319 return include_in_full(name) and keep_in_baremetal(name)
320
Gilles Peskine120f29d2021-09-01 19:51:19 +0200321# This set contains options that are mostly for debugging or test purposes,
322# and therefore should be excluded when doing code size measurements.
323# Options that are their own module (such as MBEDTLS_ERROR_C) are not listed
324# and therefore will be included when doing code size measurements.
325EXCLUDE_FOR_SIZE = frozenset([
326 'MBEDTLS_DEBUG_C', # large code size increase in TLS
327 'MBEDTLS_SELF_TEST', # increases the size of many modules
328 'MBEDTLS_TEST_HOOKS', # only useful with the hosted test framework, increases code size
329])
330
331def baremetal_size_adapter(name, active, section):
332 if name in EXCLUDE_FOR_SIZE:
333 return False
334 return baremetal_adapter(name, active, section)
335
Gilles Peskine31987c62020-01-31 14:23:30 +0100336def include_in_crypto(name):
337 """Rules for symbols in a crypto configuration."""
338 if name.startswith('MBEDTLS_X509_') or \
339 name.startswith('MBEDTLS_SSL_') or \
340 name.startswith('MBEDTLS_KEY_EXCHANGE_'):
341 return False
342 if name in [
Gilles Peskinecfffc282020-04-12 13:55:45 +0200343 'MBEDTLS_DEBUG_C', # part of libmbedtls
344 'MBEDTLS_NET_C', # part of libmbedtls
Nayna Jainc9deb182020-11-16 19:03:12 +0000345 'MBEDTLS_PKCS7_C', # part of libmbedx509
Gilles Peskine31987c62020-01-31 14:23:30 +0100346 ]:
347 return False
Gabor Mezei542fd382024-06-10 14:07:42 +0200348 if name in EXCLUDE_FROM_CRYPTO:
349 return False
Gilles Peskine31987c62020-01-31 14:23:30 +0100350 return True
351
352def crypto_adapter(adapter):
353 """Modify an adapter to disable non-crypto symbols.
354
355 ``crypto_adapter(adapter)(name, active, section)`` is like
356 ``adapter(name, active, section)``, but unsets all X.509 and TLS symbols.
357 """
358 def continuation(name, active, section):
359 if not include_in_crypto(name):
360 return False
361 if adapter is None:
362 return active
363 return adapter(name, active, section)
364 return continuation
365
Gilles Peskineed5c21d2022-06-27 23:02:09 +0200366DEPRECATED = frozenset([
367 'MBEDTLS_PSA_CRYPTO_SE_C',
Gabor Mezei542fd382024-06-10 14:07:42 +0200368 *PSA_DEPRECATED_FEATURE
Gilles Peskineed5c21d2022-06-27 23:02:09 +0200369])
Gilles Peskine30de2e82020-04-20 21:39:22 +0200370def no_deprecated_adapter(adapter):
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200371 """Modify an adapter to disable deprecated symbols.
372
Gilles Peskine30de2e82020-04-20 21:39:22 +0200373 ``no_deprecated_adapter(adapter)(name, active, section)`` is like
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200374 ``adapter(name, active, section)``, but unsets all deprecated symbols
375 and sets ``MBEDTLS_DEPRECATED_REMOVED``.
376 """
377 def continuation(name, active, section):
378 if name == 'MBEDTLS_DEPRECATED_REMOVED':
379 return True
Gilles Peskineed5c21d2022-06-27 23:02:09 +0200380 if name in DEPRECATED:
381 return False
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200382 if adapter is None:
383 return active
384 return adapter(name, active, section)
385 return continuation
386
Paul Elliottfb81f772023-10-18 17:44:59 +0100387def no_platform_adapter(adapter):
388 """Modify an adapter to disable platform symbols.
389
390 ``no_platform_adapter(adapter)(name, active, section)`` is like
391 ``adapter(name, active, section)``, but unsets all platform symbols other
392 ``than MBEDTLS_PLATFORM_C.
393 """
394 def continuation(name, active, section):
395 # Allow MBEDTLS_PLATFORM_C but remove all other platform symbols.
396 if name.startswith('MBEDTLS_PLATFORM_') and name != 'MBEDTLS_PLATFORM_C':
397 return False
398 if adapter is None:
399 return active
400 return adapter(name, active, section)
401 return continuation
402
Gabor Mezei3678dee2024-06-04 19:58:43 +0200403class ConfigFile(metaclass=ABCMeta):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200404 """Representation of a configuration file."""
405
Gabor Mezei93a6d1f2024-06-26 18:01:09 +0200406 def __init__(self, default_path, name, filename=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200407 """Check if the config file exists."""
Gilles Peskineb4063892019-07-27 21:36:44 +0200408 if filename is None:
Gabor Mezei3678dee2024-06-04 19:58:43 +0200409 for candidate in default_path:
Gilles Peskinece674a92020-03-24 15:37:00 +0100410 if os.path.lexists(candidate):
411 filename = candidate
Gilles Peskine208e4ec2019-07-29 23:43:20 +0200412 break
Gilles Peskinece674a92020-03-24 15:37:00 +0100413 else:
Gabor Mezei92065ed2024-06-07 13:47:59 +0200414 raise ValueError(f'{name} configuration file not found: '
415 f'{filename if filename else default_path}')
Gilles Peskineb4063892019-07-27 21:36:44 +0200416
Gabor Mezei3678dee2024-06-04 19:58:43 +0200417 self.filename = filename
418 self.templates = []
419 self.current_section = None
420 self.inclusion_guard = None
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200421 self.modified = False
Gilles Peskineb4063892019-07-27 21:36:44 +0200422
423 _define_line_regexp = (r'(?P<indentation>\s*)' +
424 r'(?P<commented_out>(//\s*)?)' +
425 r'(?P<define>#\s*define\s+)' +
426 r'(?P<name>\w+)' +
427 r'(?P<arguments>(?:\((?:\w|\s|,)*\))?)' +
428 r'(?P<separator>\s*)' +
429 r'(?P<value>.*)')
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200430 _ifndef_line_regexp = r'#ifndef (?P<inclusion_guard>\w+)'
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200431 _section_line_regexp = (r'\s*/?\*+\s*[\\@]name\s+SECTION:\s*' +
432 r'(?P<section>.*)[ */]*')
433 _config_line_regexp = re.compile(r'|'.join([_define_line_regexp,
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200434 _ifndef_line_regexp,
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200435 _section_line_regexp]))
Gilles Peskineb4063892019-07-27 21:36:44 +0200436 def _parse_line(self, line):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200437 """Parse a line in the config file, save the templates representing the lines
438 and return the corresponding setting element.
439 """
Gilles Peskineb4063892019-07-27 21:36:44 +0200440 line = line.rstrip('\r\n')
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200441 m = re.match(self._config_line_regexp, line)
442 if m is None:
Gabor Mezei3678dee2024-06-04 19:58:43 +0200443 self.templates.append(line)
444 return None
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200445 elif m.group('section'):
446 self.current_section = m.group('section')
Gabor Mezei3678dee2024-06-04 19:58:43 +0200447 self.templates.append(line)
448 return None
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200449 elif m.group('inclusion_guard') and self.inclusion_guard is None:
450 self.inclusion_guard = m.group('inclusion_guard')
Gabor Mezei3678dee2024-06-04 19:58:43 +0200451 self.templates.append(line)
452 return None
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200453 else:
Gilles Peskineb4063892019-07-27 21:36:44 +0200454 active = not m.group('commented_out')
455 name = m.group('name')
456 value = m.group('value')
Gilles Peskine9ba9c212024-05-23 15:03:43 +0200457 if name == self.inclusion_guard and value == '':
458 # The file double-inclusion guard is not an option.
Gabor Mezei3678dee2024-06-04 19:58:43 +0200459 self.templates.append(line)
460 return None
Gilles Peskineb4063892019-07-27 21:36:44 +0200461 template = (name,
462 m.group('indentation'),
463 m.group('define') + name +
464 m.group('arguments') + m.group('separator'))
Gabor Mezei3678dee2024-06-04 19:58:43 +0200465 self.templates.append(template)
Gilles Peskineb4063892019-07-27 21:36:44 +0200466
Gabor Mezei3678dee2024-06-04 19:58:43 +0200467 return (active, name, value, self.current_section)
468
469 def parse_file(self):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200470 """Parse the whole file and return the settings."""
Gabor Mezei3678dee2024-06-04 19:58:43 +0200471 with open(self.filename, 'r', encoding='utf-8') as file:
472 for line in file:
473 setting = self._parse_line(line)
474 if setting is not None:
475 yield setting
476 self.current_section = None
477
Gabor Mezeie7742b32024-06-26 18:04:09 +0200478 #pylint: disable=no-self-use
479 def _format_template(self, setting, indent, middle):
480 """Build a line for mbedtls_config.h for the given setting.
481
482 The line has the form "<indent>#define <name> <value>"
483 where <middle> is "#define <name> ".
484 """
485 value = setting.value
486 if value is None:
487 value = ''
488 # Normally the whitespace to separate the symbol name from the
489 # value is part of middle, and there's no whitespace for a symbol
490 # with no value. But if a symbol has been changed from having a
491 # value to not having one, the whitespace is wrong, so fix it.
492 if value:
493 if middle[-1] not in '\t ':
494 middle += ' '
495 else:
496 middle = middle.rstrip()
497 return ''.join([indent,
498 '' if setting.active else '//',
499 middle,
500 value]).rstrip()
Gabor Mezei3678dee2024-06-04 19:58:43 +0200501
502 def write_to_stream(self, settings, output):
503 """Write the whole configuration to output."""
504 for template in self.templates:
505 if isinstance(template, str):
506 line = template
507 else:
Gabor Mezeie7742b32024-06-26 18:04:09 +0200508 name, indent, middle = template
509 line = self._format_template(settings[name], indent, middle)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200510 output.write(line + '\n')
511
512 def write(self, settings, filename=None):
513 """Write the whole configuration to the file it was read from.
514
515 If filename is specified, write to this file instead.
516 """
517 if filename is None:
518 filename = self.filename
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200519
520 # Not modified so no need to write to the file
521 if not self.modified and filename == self.filename:
522 return
523
Gabor Mezei3678dee2024-06-04 19:58:43 +0200524 with open(filename, 'w', encoding='utf-8') as output:
525 self.write_to_stream(settings, output)
526
527class MbedtlsConfigFile(ConfigFile):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200528 """Representation of an MbedTLS configuration file."""
529
Gabor Mezei3678dee2024-06-04 19:58:43 +0200530 _path_in_tree = 'include/mbedtls/mbedtls_config.h'
531 default_path = [_path_in_tree,
532 os.path.join(os.path.dirname(__file__),
533 os.pardir,
534 _path_in_tree),
535 os.path.join(os.path.dirname(os.path.abspath(os.path.dirname(__file__))),
536 _path_in_tree)]
537
538 def __init__(self, filename=None):
Gabor Mezei93a6d1f2024-06-26 18:01:09 +0200539 super().__init__(self.default_path, 'Mbed TLS', filename)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200540 self.current_section = 'header'
541
Gabor Mezei3678dee2024-06-04 19:58:43 +0200542class CryptoConfigFile(ConfigFile):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200543 """Representation of an Crypto configuration file."""
544
Gabor Mezei3678dee2024-06-04 19:58:43 +0200545 _path_in_tree = 'tf-psa-crypto/include/psa/crypto_config.h'
546 default_path = [_path_in_tree,
547 os.path.join(os.path.dirname(__file__),
548 os.pardir,
549 _path_in_tree),
550 os.path.join(os.path.dirname(os.path.abspath(os.path.dirname(__file__))),
551 _path_in_tree)]
552
553 def __init__(self, filename=None):
Gabor Mezei93a6d1f2024-06-26 18:01:09 +0200554 super().__init__(self.default_path, 'Crypto', filename)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200555
Gabor Mezei3678dee2024-06-04 19:58:43 +0200556class MbedtlsConfig(Config):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200557 """Representation of the Mbed TLS configuration.
Gabor Mezei3678dee2024-06-04 19:58:43 +0200558
559 See the documentation of the `Config` class for methods to query
560 and modify the configuration.
561 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200562 def __init__(self, filename=None):
Gabor Mezei3678dee2024-06-04 19:58:43 +0200563 """Read the Mbed TLS configuration file."""
564 super().__init__()
Gabor Mezeiee521b62024-06-07 13:50:41 +0200565 self.configfile = MbedtlsConfigFile(filename)
566 self.settings.update({name: Setting(active, name, value, section)
Gabor Mezei92065ed2024-06-07 13:47:59 +0200567 for (active, name, value, section)
Gabor Mezeiee521b62024-06-07 13:50:41 +0200568 in self.configfile.parse_file()})
Gabor Mezei3678dee2024-06-04 19:58:43 +0200569
570 def set(self, name, value=None):
571 if name not in self.settings:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200572 self.configfile.templates.append((name, '', '#define ' + name + ' '))
573
Gabor Mezei3678dee2024-06-04 19:58:43 +0200574 super().set(name, value)
Gilles Peskineb4063892019-07-27 21:36:44 +0200575
576 def write(self, filename=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200577 """Write the whole configuration to the file it was read from.
578
579 If filename is specified, write to this file instead.
580 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200581 self.configfile.write(self.settings, filename)
Gilles Peskineb4063892019-07-27 21:36:44 +0200582
Gabor Mezeiee521b62024-06-07 13:50:41 +0200583 def filename(self):
584 return self.configfile.filename
Gabor Mezei3678dee2024-06-04 19:58:43 +0200585
586class CryptoConfig(Config):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200587 """Representation of the PSA crypto configuration.
Gabor Mezei3678dee2024-06-04 19:58:43 +0200588
589 See the documentation of the `Config` class for methods to query
590 and modify the configuration.
591 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200592 def __init__(self, filename=None):
Gabor Mezei3678dee2024-06-04 19:58:43 +0200593 """Read the PSA crypto configuration file."""
594 super().__init__()
Gabor Mezeiee521b62024-06-07 13:50:41 +0200595 self.configfile = CryptoConfigFile(filename)
596 self.settings.update({name: Setting(active, name, value, section)
Gabor Mezei92065ed2024-06-07 13:47:59 +0200597 for (active, name, value, section)
Gabor Mezeiee521b62024-06-07 13:50:41 +0200598 in self.configfile.parse_file()})
Gabor Mezei3678dee2024-06-04 19:58:43 +0200599
Gabor Mezeid723b512024-06-07 15:31:52 +0200600 def set(self, name, value='1'):
Gabor Mezei542fd382024-06-10 14:07:42 +0200601 if name in PSA_UNSUPPORTED_FEATURE:
Gabor Mezei92065ed2024-06-07 13:47:59 +0200602 raise ValueError(f'Feature is unsupported: \'{name}\'')
Gabor Mezei542fd382024-06-10 14:07:42 +0200603 if name in PSA_UNSTABLE_FEATURE:
Gabor Mezei92065ed2024-06-07 13:47:59 +0200604 raise ValueError(f'Feature is unstable: \'{name}\'')
Gabor Mezei3678dee2024-06-04 19:58:43 +0200605
Gabor Mezeid723b512024-06-07 15:31:52 +0200606 # The default value in the crypto config is '1'
607 if not value:
608 value = '1'
609
Gabor Mezei3678dee2024-06-04 19:58:43 +0200610 if name not in self.settings:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200611 self.configfile.templates.append((name, '', '#define ' + name + ' '))
612
Gabor Mezei3678dee2024-06-04 19:58:43 +0200613 super().set(name, value)
614
615 def write(self, filename=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200616 """Write the whole configuration to the file it was read from.
617
618 If filename is specified, write to this file instead.
619 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200620 self.configfile.write(self.settings, filename)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200621
Gabor Mezeiee521b62024-06-07 13:50:41 +0200622 def filename(self):
623 return self.configfile.filename
Gabor Mezei3678dee2024-06-04 19:58:43 +0200624
Gabor Mezeiee521b62024-06-07 13:50:41 +0200625class MultiConfig(Config):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200626 """Representation of MbedTLS and PSA crypto configuration
627
628 See the documentation of the `Config` class for methods to query
629 and modify the configuration.
630 """
Gabor Mezei3678dee2024-06-04 19:58:43 +0200631
Gabor Mezeiee521b62024-06-07 13:50:41 +0200632 def __init__(self, mbedtls_filename=None, crypto__filename=None):
633 super().__init__()
634 self.mbedtls_configfile = MbedtlsConfigFile(mbedtls_filename)
635 self.crypto_configfile = CryptoConfigFile(crypto__filename)
636 self.settings.update({name: Setting(active, name, value, section, configfile)
637 for configfile in [self.mbedtls_configfile, self.crypto_configfile]
638 for (active, name, value, section) in configfile.parse_file()})
Gabor Mezei3678dee2024-06-04 19:58:43 +0200639
640 _crypto_regexp = re.compile(r'$PSA_.*')
Gabor Mezeiee521b62024-06-07 13:50:41 +0200641 def _get_configfile(self, name):
642 """Find a config type for a setting name
643 """
644 if name in self.settings:
645 return self.settings[name].configfile
646 elif re.match(self._crypto_regexp, name):
647 return self.crypto_configfile
Gabor Mezei3678dee2024-06-04 19:58:43 +0200648 else:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200649 return self.mbedtls_configfile
Gabor Mezei3678dee2024-06-04 19:58:43 +0200650
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200651 def __setitem__(self, name, value):
652 super().__setitem__(name, value)
653 self.settings[name].configfile.modified = True
654
Gabor Mezei3678dee2024-06-04 19:58:43 +0200655 def set(self, name, value=None):
Gabor Mezeiee521b62024-06-07 13:50:41 +0200656 configfile = self._get_configfile(name)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200657
Gabor Mezeiee521b62024-06-07 13:50:41 +0200658 if configfile == self.crypto_configfile:
Gabor Mezei542fd382024-06-10 14:07:42 +0200659 if name in PSA_UNSUPPORTED_FEATURE:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200660 raise ValueError(f'Feature is unsupported: \'{name}\'')
Gabor Mezei542fd382024-06-10 14:07:42 +0200661 if name in PSA_UNSTABLE_FEATURE:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200662 raise ValueError(f'Feature is unstable: \'{name}\'')
663
Gabor Mezeid723b512024-06-07 15:31:52 +0200664 # The default value in the crypto config is '1'
665 if not value:
666 value = '1'
667
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200668 if name in self.settings:
669 setting = self.settings[name]
670 if not setting.active or (value is not None and setting.value != value):
671 configfile.modified = True
672 else:
Gabor Mezeiee521b62024-06-07 13:50:41 +0200673 configfile.templates.append((name, '', '#define ' + name + ' '))
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200674 configfile.modified = True
Gabor Mezeiee521b62024-06-07 13:50:41 +0200675
676 super().set(name, value)
677
Gabor Mezei8a64d8e2024-06-10 15:23:43 +0200678 def unset(self, name):
679 if name in self.settings and self.settings[name].active:
680 self.settings[name].configfile.modified = True
681
682 super().unset(name)
683
684 def adapt(self, adapter):
685 # Determine if the config files will be modified
686 unmodified = {config for config in [self.mbedtls_configfile, self.crypto_configfile]
687 if not config.modified}
688 if unmodified:
689 for setting in self.settings.values():
690 if not setting.configfile.modified and \
691 setting.active != adapter(setting.name, setting.active, setting.section):
692 setting.configfile.modified = True
693 unmodified.remove(setting.configfile)
694 if not unmodified:
695 break
696
697 super().adapt(adapter)
698
699 def change_matching(self, regexs, enable):
700 # Determine if the config files will be modified
701 if regexs:
702 regex = re.compile('|'.join(regexs))
703 unmodified = {config for config in [self.mbedtls_configfile, self.crypto_configfile]
704 if not config.modified}
705 if unmodified:
706 for setting in self.settings.values():
707 if not setting.configfile.modified and \
708 setting.active != enable and \
709 regex.search(setting.name):
710 setting.configfile.modified = True
711 unmodified.remove(setting.configfile)
712 if not unmodified:
713 break
714
715 super().change_matching(regexs, enable)
716
Gabor Mezei3678dee2024-06-04 19:58:43 +0200717 def write(self, mbedtls_file=None, crypto_file=None):
Gabor Mezei62a9bd02024-06-07 13:44:40 +0200718 """Write the whole configuration to the file it was read from.
719
720 If mbedtls_file or crypto_file is specified, write the specific configuration
721 to the corresponding file instead.
722 """
Gabor Mezeiee521b62024-06-07 13:50:41 +0200723 self.mbedtls_configfile.write(self.settings, mbedtls_file)
724 self.crypto_configfile.write(self.settings, crypto_file)
Gabor Mezei3678dee2024-06-04 19:58:43 +0200725
Gabor Mezeiee521b62024-06-07 13:50:41 +0200726 def filename(self, name=None):
727 if not name:
728 return [config.filename for config in [self.mbedtls_configfile, self.crypto_configfile]]
729
730 return self._get_configfile(name).filename
Gilles Peskineb4063892019-07-27 21:36:44 +0200731
732if __name__ == '__main__':
Gabor Mezei92065ed2024-06-07 13:47:59 +0200733 #pylint: disable=too-many-statements
Gilles Peskineb4063892019-07-27 21:36:44 +0200734 def main():
Bence Szépkútibb0cfeb2021-05-28 09:42:25 +0200735 """Command line mbedtls_config.h manipulation tool."""
Gilles Peskineb4063892019-07-27 21:36:44 +0200736 parser = argparse.ArgumentParser(description="""
Fredrik Hesse0ec8a902021-10-04 22:13:51 +0200737 Mbed TLS configuration file manipulation tool.
Gilles Peskineb4063892019-07-27 21:36:44 +0200738 """)
739 parser.add_argument('--file', '-f',
740 help="""File to read (and modify if requested).
741 Default: {}.
Gabor Mezei3678dee2024-06-04 19:58:43 +0200742 """.format(MbedtlsConfigFile.default_path))
743 parser.add_argument('--cryptofile', '-c',
744 help="""Crypto file to read (and modify if requested).
745 Default: {}.
746 """.format(CryptoConfigFile.default_path))
Gilles Peskineb4063892019-07-27 21:36:44 +0200747 parser.add_argument('--force', '-o',
Gilles Peskine435ce222019-08-01 23:13:47 +0200748 action='store_true',
Gilles Peskineb4063892019-07-27 21:36:44 +0200749 help="""For the set command, if SYMBOL is not
750 present, add a definition for it.""")
Gilles Peskinec190c902019-08-01 23:31:05 +0200751 parser.add_argument('--write', '-w', metavar='FILE',
Gilles Peskine40f103c2019-07-27 23:44:01 +0200752 help="""File to write to instead of the input file.""")
Gilles Peskineb4063892019-07-27 21:36:44 +0200753 subparsers = parser.add_subparsers(dest='command',
754 title='Commands')
755 parser_get = subparsers.add_parser('get',
756 help="""Find the value of SYMBOL
757 and print it. Exit with
758 status 0 if a #define for SYMBOL is
759 found, 1 otherwise.
760 """)
761 parser_get.add_argument('symbol', metavar='SYMBOL')
762 parser_set = subparsers.add_parser('set',
763 help="""Set SYMBOL to VALUE.
764 If VALUE is omitted, just uncomment
765 the #define for SYMBOL.
766 Error out of a line defining
767 SYMBOL (commented or not) is not
768 found, unless --force is passed.
769 """)
770 parser_set.add_argument('symbol', metavar='SYMBOL')
Gilles Peskine0c7fcd22019-08-01 23:14:00 +0200771 parser_set.add_argument('value', metavar='VALUE', nargs='?',
772 default='')
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200773 parser_set_all = subparsers.add_parser('set-all',
774 help="""Uncomment all #define
775 whose name contains a match for
776 REGEX.""")
777 parser_set_all.add_argument('regexs', metavar='REGEX', nargs='*')
Gilles Peskineb4063892019-07-27 21:36:44 +0200778 parser_unset = subparsers.add_parser('unset',
779 help="""Comment out the #define
780 for SYMBOL. Do nothing if none
781 is present.""")
782 parser_unset.add_argument('symbol', metavar='SYMBOL')
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200783 parser_unset_all = subparsers.add_parser('unset-all',
784 help="""Comment out all #define
785 whose name contains a match for
786 REGEX.""")
787 parser_unset_all.add_argument('regexs', metavar='REGEX', nargs='*')
Gilles Peskineb4063892019-07-27 21:36:44 +0200788
789 def add_adapter(name, function, description):
790 subparser = subparsers.add_parser(name, help=description)
791 subparser.set_defaults(adapter=function)
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200792 add_adapter('baremetal', baremetal_adapter,
793 """Like full, but exclude features that require platform
794 features such as file input-output.""")
Gilles Peskine120f29d2021-09-01 19:51:19 +0200795 add_adapter('baremetal_size', baremetal_size_adapter,
796 """Like baremetal, but exclude debugging features.
797 Useful for code size measurements.""")
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200798 add_adapter('full', full_adapter,
799 """Uncomment most features.
800 Exclude alternative implementations and platform support
801 options, as well as some options that are awkward to test.
802 """)
Gilles Peskine30de2e82020-04-20 21:39:22 +0200803 add_adapter('full_no_deprecated', no_deprecated_adapter(full_adapter),
Gilles Peskinebe1d6092020-04-12 14:17:16 +0200804 """Uncomment most non-deprecated features.
805 Like "full", but without deprecated features.
806 """)
Paul Elliottfb81f772023-10-18 17:44:59 +0100807 add_adapter('full_no_platform', no_platform_adapter(full_adapter),
808 """Uncomment most non-platform features.
809 Like "full", but without platform features.
810 """)
Gilles Peskineb4063892019-07-27 21:36:44 +0200811 add_adapter('realfull', realfull_adapter,
Gilles Peskine53d41ae2019-07-27 23:31:53 +0200812 """Uncomment all boolean #defines.
813 Suitable for generating documentation, but not for building.""")
Gilles Peskine31987c62020-01-31 14:23:30 +0100814 add_adapter('crypto', crypto_adapter(None),
815 """Only include crypto features. Exclude X.509 and TLS.""")
816 add_adapter('crypto_baremetal', crypto_adapter(baremetal_adapter),
817 """Like baremetal, but with only crypto features,
818 excluding X.509 and TLS.""")
819 add_adapter('crypto_full', crypto_adapter(full_adapter),
820 """Like full, but with only crypto features,
821 excluding X.509 and TLS.""")
Gilles Peskineb4063892019-07-27 21:36:44 +0200822
823 args = parser.parse_args()
Gabor Mezei3678dee2024-06-04 19:58:43 +0200824 config = MultiConfig(args.file, args.cryptofile)
Gilles Peskine90b30b62019-07-28 00:36:53 +0200825 if args.command is None:
826 parser.print_help()
827 return 1
828 elif args.command == 'get':
Gilles Peskineb4063892019-07-27 21:36:44 +0200829 if args.symbol in config:
830 value = config[args.symbol]
831 if value:
832 sys.stdout.write(value + '\n')
Gilles Peskinee22a4da2020-03-24 15:43:49 +0100833 return 0 if args.symbol in config else 1
Gilles Peskineb4063892019-07-27 21:36:44 +0200834 elif args.command == 'set':
Gilles Peskine98eb3652019-07-28 16:39:19 +0200835 if not args.force and args.symbol not in config.settings:
Gilles Peskineb4063892019-07-27 21:36:44 +0200836 sys.stderr.write("A #define for the symbol {} "
Gilles Peskine221df1e2019-08-01 23:14:29 +0200837 "was not found in {}\n"
Gabor Mezei3678dee2024-06-04 19:58:43 +0200838 .format(args.symbol, config.filename(args.symbol)))
Gilles Peskineb4063892019-07-27 21:36:44 +0200839 return 1
840 config.set(args.symbol, value=args.value)
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200841 elif args.command == 'set-all':
842 config.change_matching(args.regexs, True)
Gilles Peskineb4063892019-07-27 21:36:44 +0200843 elif args.command == 'unset':
844 config.unset(args.symbol)
Gilles Peskine8e90cf42021-05-27 22:12:57 +0200845 elif args.command == 'unset-all':
846 config.change_matching(args.regexs, False)
Gilles Peskineb4063892019-07-27 21:36:44 +0200847 else:
848 config.adapt(args.adapter)
Gilles Peskine40f103c2019-07-27 23:44:01 +0200849 config.write(args.write)
Gilles Peskinee22a4da2020-03-24 15:43:49 +0100850 return 0
Gilles Peskineb4063892019-07-27 21:36:44 +0200851
852 # Import modules only used by main only if main is defined and called.
853 # pylint: disable=wrong-import-position
854 import argparse
855 import sys
856 sys.exit(main())