Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | """Mbed TLS configuration file manipulation library and tool |
| 4 | |
Fredrik Hesse | cc207bc | 2021-09-28 21:06:08 +0200 | [diff] [blame] | 5 | Basic usage, to read the Mbed TLS configuration: |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 6 | config = ConfigFile() |
| 7 | if 'MBEDTLS_RSA_C' in config: print('RSA is enabled') |
| 8 | """ |
| 9 | |
Gilles Peskine | cf42536 | 2022-10-10 22:52:30 +0200 | [diff] [blame] | 10 | # 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úti | 1e14827 | 2020-08-07 13:07:28 +0200 | [diff] [blame] | 15 | ## Copyright The Mbed TLS Contributors |
Dave Rodgman | 16799db | 2023-11-02 19:47:20 +0000 | [diff] [blame] | 16 | ## SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 17 | ## |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 18 | |
Gilles Peskine | 208e4ec | 2019-07-29 23:43:20 +0200 | [diff] [blame] | 19 | import os |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 20 | import re |
| 21 | |
Gabor Mezei | e7742b3 | 2024-06-26 18:04:09 +0200 | [diff] [blame^] | 22 | from abc import ABCMeta |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 23 | |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 24 | class Setting: |
Bence Szépkúti | bb0cfeb | 2021-05-28 09:42:25 +0200 | [diff] [blame] | 25 | """Representation of one Mbed TLS mbedtls_config.h setting. |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 26 | |
| 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úti | bb0cfeb | 2021-05-28 09:42:25 +0200 | [diff] [blame] | 32 | present in mbedtls_config.h but commented out. |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 33 | * section: the name of the section that contains this symbol. |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 34 | """ |
Gabor Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 35 | # pylint: disable=too-few-public-methods, too-many-arguments |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 36 | def __init__(self, active, name, value='', section=None, configfile=None): |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 37 | self.active = active |
| 38 | self.name = name |
| 39 | self.value = value |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 40 | self.section = section |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 41 | self.configfile = configfile |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 42 | |
| 43 | class 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 Peskine | c190c90 | 2019-08-01 23:31:05 +0200 | [diff] [blame] | 51 | * `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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 55 | * `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 Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 60 | # pylint: disable=unused-argument |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 61 | def __init__(self): |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 62 | 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 Peskine | 55cc4db | 2019-08-01 23:13:23 +0200 | [diff] [blame] | 128 | name remains known if it was known before. |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 129 | """ |
Gilles Peskine | 55cc4db | 2019-08-01 23:13:23 +0200 | [diff] [blame] | 130 | if name not in self.settings: |
| 131 | return |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 132 | 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 Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 138 | `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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 141 | `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 Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 145 | setting.active = adapter(setting.name, setting.active, |
| 146 | setting.section) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 147 | |
Gilles Peskine | 8e90cf4 | 2021-05-27 22:12:57 +0200 | [diff] [blame] | 148 | 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 Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 157 | def is_full_section(section): |
| 158 | """Is this section affected by "config.py full" and friends?""" |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 159 | return section is None or section.endswith('support') or section.endswith('modules') |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 160 | |
| 161 | def realfull_adapter(_name, active, section): |
Gilles Peskine | ba4162a | 2022-04-11 17:04:38 +0200 | [diff] [blame] | 162 | """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 Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 172 | return active |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 173 | return True |
| 174 | |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 175 | PSA_UNSUPPORTED_FEATURE = frozenset([ |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 176 | '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 Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 182 | PSA_DEPRECATED_FEATURE = frozenset([ |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 183 | 'PSA_WANT_KEY_TYPE_ECC_KEY_PAIR', |
| 184 | 'PSA_WANT_KEY_TYPE_RSA_KEY_PAIR' |
| 185 | ]) |
| 186 | |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 187 | PSA_UNSTABLE_FEATURE = frozenset([ |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 188 | 'PSA_WANT_ECC_SECP_K1_224' |
| 189 | ]) |
| 190 | |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 191 | EXCLUDE_FROM_CRYPTO = frozenset( |
| 192 | PSA_UNSUPPORTED_FEATURE | |
| 193 | PSA_DEPRECATED_FEATURE | |
| 194 | PSA_UNSTABLE_FEATURE |
| 195 | ) |
| 196 | |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 197 | # 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 Peskine | c9d0433 | 2020-04-16 20:50:17 +0200 | [diff] [blame] | 201 | # * 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 Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 204 | # * Options that remove features. |
Gilles Peskine | bbaa2b7 | 2020-04-12 13:33:57 +0200 | [diff] [blame] | 205 | EXCLUDE_FROM_FULL = frozenset([ |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 206 | #pylint: disable=line-too-long |
Yanray Wang | a870467 | 2023-04-20 17:16:48 +0800 | [diff] [blame] | 207 | 'MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH', # interacts with CTR_DRBG_128_BIT_KEY |
Gilles Peskine | a8861e0 | 2023-09-05 20:20:51 +0200 | [diff] [blame] | 208 | 'MBEDTLS_AES_USE_HARDWARE_ONLY', # hardware dependency |
Yanray Wang | 42be1ba | 2023-11-23 14:28:47 +0800 | [diff] [blame] | 209 | 'MBEDTLS_BLOCK_CIPHER_NO_DECRYPT', # incompatible with ECB in PSA, CBC/XTS/NIST_KW/DES |
Gilles Peskine | c9d0433 | 2020-04-16 20:50:17 +0200 | [diff] [blame] | 210 | 'MBEDTLS_CTR_DRBG_USE_128_BIT_KEY', # interacts with ENTROPY_FORCE_SHA256 |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 211 | 'MBEDTLS_DEPRECATED_REMOVED', # conflicts with deprecated options |
Gilles Peskine | 90581ee | 2020-04-12 14:02:47 +0200 | [diff] [blame] | 212 | 'MBEDTLS_DEPRECATED_WARNING', # conflicts with deprecated options |
Gilles Peskine | c9d0433 | 2020-04-16 20:50:17 +0200 | [diff] [blame] | 213 | 'MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED', # influences the use of ECDH in TLS |
Janos Follath | 5b7c38f | 2023-08-01 08:51:12 +0100 | [diff] [blame] | 214 | 'MBEDTLS_ECP_WITH_MPI_UINT', # disables the default ECP and is experimental |
Gilles Peskine | c9d0433 | 2020-04-16 20:50:17 +0200 | [diff] [blame] | 215 | 'MBEDTLS_ENTROPY_FORCE_SHA256', # interacts with CTR_DRBG_128_BIT_KEY |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 216 | '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 Peskine | c9d0433 | 2020-04-16 20:50:17 +0200 | [diff] [blame] | 220 | 'MBEDTLS_NO_64BIT_MULTIPLICATION', # influences anything that uses bignum |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 221 | 'MBEDTLS_NO_DEFAULT_ENTROPY_SOURCES', # removes a feature |
| 222 | 'MBEDTLS_NO_PLATFORM_ENTROPY', # removes a feature |
Gilles Peskine | c9d0433 | 2020-04-16 20:50:17 +0200 | [diff] [blame] | 223 | 'MBEDTLS_NO_UDBL_DIVISION', # influences anything that uses bignum |
Gilles Peskine | efaee9a | 2023-09-20 20:49:47 +0200 | [diff] [blame] | 224 | 'MBEDTLS_PSA_P256M_DRIVER_ENABLED', # influences SECP256R1 KeyGen/ECDH/ECDSA |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 225 | 'MBEDTLS_PLATFORM_NO_STD_FUNCTIONS', # removes a feature |
David Horstmann | 6f8c95b | 2024-03-14 14:52:45 +0000 | [diff] [blame] | 226 | 'MBEDTLS_PSA_ASSUME_EXCLUSIVE_BUFFERS', # removes a feature |
Gilles Peskine | f08b3f8 | 2020-11-13 17:36:48 +0100 | [diff] [blame] | 227 | 'MBEDTLS_PSA_CRYPTO_EXTERNAL_RNG', # behavior change + build dependency |
Ronald Cron | c3623db | 2020-10-29 10:51:32 +0100 | [diff] [blame] | 228 | 'MBEDTLS_PSA_CRYPTO_KEY_ID_ENCODES_OWNER', # incompatible with USE_PSA_CRYPTO |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 229 | 'MBEDTLS_PSA_CRYPTO_SPM', # platform dependency (PSA SPM) |
Gilles Peskine | a08def9 | 2023-04-28 21:01:49 +0200 | [diff] [blame] | 230 | 'MBEDTLS_PSA_INJECT_ENTROPY', # conflicts with platform entropy sources |
Gilles Peskine | c9d0433 | 2020-04-16 20:50:17 +0200 | [diff] [blame] | 231 | 'MBEDTLS_RSA_NO_CRT', # influences the use of RSA in X.509 and TLS |
Tom Cosgrove | 87fbfb5 | 2022-03-15 10:51:52 +0000 | [diff] [blame] | 232 | 'MBEDTLS_SHA256_USE_A64_CRYPTO_ONLY', # interacts with *_USE_A64_CRYPTO_IF_PRESENT |
Dave Rodgman | 9be3cf0 | 2023-10-11 14:47:55 +0100 | [diff] [blame] | 233 | 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_ONLY', # interacts with *_USE_ARMV8_A_CRYPTO_IF_PRESENT |
Tom Cosgrove | 87fbfb5 | 2022-03-15 10:51:52 +0000 | [diff] [blame] | 234 | 'MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY', # interacts with *_USE_A64_CRYPTO_IF_PRESENT |
Dave Rodgman | 7cb635a | 2023-10-12 16:14:51 +0100 | [diff] [blame] | 235 | 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT', # setting *_USE_ARMV8_A_CRYPTO is sufficient |
Manuel Pégourié-Gonnard | 6240def | 2020-07-10 09:35:54 +0200 | [diff] [blame] | 236 | 'MBEDTLS_TEST_CONSTANT_FLOW_MEMSAN', # build dependency (clang+memsan) |
Manuel Pégourié-Gonnard | 73afa37 | 2020-08-19 10:27:38 +0200 | [diff] [blame] | 237 | 'MBEDTLS_TEST_CONSTANT_FLOW_VALGRIND', # build dependency (valgrind headers) |
Hanno Becker | e111356 | 2019-06-12 13:59:14 +0100 | [diff] [blame] | 238 | 'MBEDTLS_X509_REMOVE_INFO', # removes a feature |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 239 | *PSA_UNSUPPORTED_FEATURE, |
| 240 | *PSA_DEPRECATED_FEATURE, |
| 241 | *PSA_UNSTABLE_FEATURE |
Gilles Peskine | bbaa2b7 | 2020-04-12 13:33:57 +0200 | [diff] [blame] | 242 | ]) |
| 243 | |
Gilles Peskine | 32e889d | 2020-04-12 23:43:28 +0200 | [diff] [blame] | 244 | def is_seamless_alt(name): |
Gilles Peskine | c34faba | 2020-04-20 15:44:14 +0200 | [diff] [blame] | 245 | """Whether the xxx_ALT symbol should be included in the full configuration. |
Gilles Peskine | 32e889d | 2020-04-12 23:43:28 +0200 | [diff] [blame] | 246 | |
Gilles Peskine | c34faba | 2020-04-20 15:44:14 +0200 | [diff] [blame] | 247 | Include alternative implementations of platform functions, which are |
Gilles Peskine | 32e889d | 2020-04-12 23:43:28 +0200 | [diff] [blame] | 248 | 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 Peskine | a8861e0 | 2023-09-05 20:20:51 +0200 | [diff] [blame] | 256 | 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 Peskine | c34faba | 2020-04-20 15:44:14 +0200 | [diff] [blame] | 262 | # Similar to non-platform xxx_ALT, requires platform_alt.h |
| 263 | return False |
Gilles Peskine | 32e889d | 2020-04-12 23:43:28 +0200 | [diff] [blame] | 264 | return name.startswith('MBEDTLS_PLATFORM_') |
| 265 | |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 266 | def include_in_full(name): |
| 267 | """Rules for symbols in the "full" configuration.""" |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 268 | if name in EXCLUDE_FROM_FULL: |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 269 | return False |
| 270 | if name.endswith('_ALT'): |
Gilles Peskine | 32e889d | 2020-04-12 23:43:28 +0200 | [diff] [blame] | 271 | return is_seamless_alt(name) |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 272 | return True |
| 273 | |
| 274 | def 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 Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 280 | # 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 Peskine | bbaa2b7 | 2020-04-12 13:33:57 +0200 | [diff] [blame] | 285 | EXCLUDE_FROM_BAREMETAL = frozenset([ |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 286 | #pylint: disable=line-too-long |
Gilles Peskine | 98f8f95 | 2020-04-20 15:38:39 +0200 | [diff] [blame] | 287 | 'MBEDTLS_ENTROPY_NV_SEED', # requires a filesystem and FS_IO or alternate NV seed hooks |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 288 | 'MBEDTLS_FS_IO', # requires a filesystem |
Gilles Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 289 | '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 Peskine | 98f8f95 | 2020-04-20 15:38:39 +0200 | [diff] [blame] | 293 | '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 Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 296 | '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 Rodgman | 9be3cf0 | 2023-10-11 14:47:55 +0100 | [diff] [blame] | 301 | 'MBEDTLS_SHA256_USE_A64_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection |
Dave Rodgman | 5b89c55 | 2023-10-10 14:59:02 +0100 | [diff] [blame] | 302 | 'MBEDTLS_SHA256_USE_ARMV8_A_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection |
Dave Rodgman | be7915a | 2023-10-11 10:46:38 +0100 | [diff] [blame] | 303 | 'MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT', # requires an OS for runtime-detection |
Gilles Peskine | bbaa2b7 | 2020-04-12 13:33:57 +0200 | [diff] [blame] | 304 | ]) |
| 305 | |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 306 | def keep_in_baremetal(name): |
| 307 | """Rules for symbols in the "baremetal" configuration.""" |
Gilles Peskine | bbaa2b7 | 2020-04-12 13:33:57 +0200 | [diff] [blame] | 308 | if name in EXCLUDE_FROM_BAREMETAL: |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 309 | return False |
| 310 | return True |
| 311 | |
| 312 | def 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 Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 317 | # No OS-provided entropy source |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 318 | return True |
| 319 | return include_in_full(name) and keep_in_baremetal(name) |
| 320 | |
Gilles Peskine | 120f29d | 2021-09-01 19:51:19 +0200 | [diff] [blame] | 321 | # 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. |
| 325 | EXCLUDE_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 | |
| 331 | def 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 Peskine | 31987c6 | 2020-01-31 14:23:30 +0100 | [diff] [blame] | 336 | def 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 Peskine | cfffc28 | 2020-04-12 13:55:45 +0200 | [diff] [blame] | 343 | 'MBEDTLS_DEBUG_C', # part of libmbedtls |
| 344 | 'MBEDTLS_NET_C', # part of libmbedtls |
Nayna Jain | c9deb18 | 2020-11-16 19:03:12 +0000 | [diff] [blame] | 345 | 'MBEDTLS_PKCS7_C', # part of libmbedx509 |
Gilles Peskine | 31987c6 | 2020-01-31 14:23:30 +0100 | [diff] [blame] | 346 | ]: |
| 347 | return False |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 348 | if name in EXCLUDE_FROM_CRYPTO: |
| 349 | return False |
Gilles Peskine | 31987c6 | 2020-01-31 14:23:30 +0100 | [diff] [blame] | 350 | return True |
| 351 | |
| 352 | def 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 Peskine | ed5c21d | 2022-06-27 23:02:09 +0200 | [diff] [blame] | 366 | DEPRECATED = frozenset([ |
| 367 | 'MBEDTLS_PSA_CRYPTO_SE_C', |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 368 | *PSA_DEPRECATED_FEATURE |
Gilles Peskine | ed5c21d | 2022-06-27 23:02:09 +0200 | [diff] [blame] | 369 | ]) |
Gilles Peskine | 30de2e8 | 2020-04-20 21:39:22 +0200 | [diff] [blame] | 370 | def no_deprecated_adapter(adapter): |
Gilles Peskine | be1d609 | 2020-04-12 14:17:16 +0200 | [diff] [blame] | 371 | """Modify an adapter to disable deprecated symbols. |
| 372 | |
Gilles Peskine | 30de2e8 | 2020-04-20 21:39:22 +0200 | [diff] [blame] | 373 | ``no_deprecated_adapter(adapter)(name, active, section)`` is like |
Gilles Peskine | be1d609 | 2020-04-12 14:17:16 +0200 | [diff] [blame] | 374 | ``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 Peskine | ed5c21d | 2022-06-27 23:02:09 +0200 | [diff] [blame] | 380 | if name in DEPRECATED: |
| 381 | return False |
Gilles Peskine | be1d609 | 2020-04-12 14:17:16 +0200 | [diff] [blame] | 382 | if adapter is None: |
| 383 | return active |
| 384 | return adapter(name, active, section) |
| 385 | return continuation |
| 386 | |
Paul Elliott | fb81f77 | 2023-10-18 17:44:59 +0100 | [diff] [blame] | 387 | def 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 Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 403 | class ConfigFile(metaclass=ABCMeta): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 404 | """Representation of a configuration file.""" |
| 405 | |
Gabor Mezei | 93a6d1f | 2024-06-26 18:01:09 +0200 | [diff] [blame] | 406 | def __init__(self, default_path, name, filename=None): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 407 | """Check if the config file exists.""" |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 408 | if filename is None: |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 409 | for candidate in default_path: |
Gilles Peskine | ce674a9 | 2020-03-24 15:37:00 +0100 | [diff] [blame] | 410 | if os.path.lexists(candidate): |
| 411 | filename = candidate |
Gilles Peskine | 208e4ec | 2019-07-29 23:43:20 +0200 | [diff] [blame] | 412 | break |
Gilles Peskine | ce674a9 | 2020-03-24 15:37:00 +0100 | [diff] [blame] | 413 | else: |
Gabor Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 414 | raise ValueError(f'{name} configuration file not found: ' |
| 415 | f'{filename if filename else default_path}') |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 416 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 417 | self.filename = filename |
| 418 | self.templates = [] |
| 419 | self.current_section = None |
| 420 | self.inclusion_guard = None |
Gabor Mezei | 8a64d8e | 2024-06-10 15:23:43 +0200 | [diff] [blame] | 421 | self.modified = False |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 422 | |
| 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 Peskine | 9ba9c21 | 2024-05-23 15:03:43 +0200 | [diff] [blame] | 430 | _ifndef_line_regexp = r'#ifndef (?P<inclusion_guard>\w+)' |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 431 | _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 Peskine | 9ba9c21 | 2024-05-23 15:03:43 +0200 | [diff] [blame] | 434 | _ifndef_line_regexp, |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 435 | _section_line_regexp])) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 436 | def _parse_line(self, line): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 437 | """Parse a line in the config file, save the templates representing the lines |
| 438 | and return the corresponding setting element. |
| 439 | """ |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 440 | line = line.rstrip('\r\n') |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 441 | m = re.match(self._config_line_regexp, line) |
| 442 | if m is None: |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 443 | self.templates.append(line) |
| 444 | return None |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 445 | elif m.group('section'): |
| 446 | self.current_section = m.group('section') |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 447 | self.templates.append(line) |
| 448 | return None |
Gilles Peskine | 9ba9c21 | 2024-05-23 15:03:43 +0200 | [diff] [blame] | 449 | elif m.group('inclusion_guard') and self.inclusion_guard is None: |
| 450 | self.inclusion_guard = m.group('inclusion_guard') |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 451 | self.templates.append(line) |
| 452 | return None |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 453 | else: |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 454 | active = not m.group('commented_out') |
| 455 | name = m.group('name') |
| 456 | value = m.group('value') |
Gilles Peskine | 9ba9c21 | 2024-05-23 15:03:43 +0200 | [diff] [blame] | 457 | if name == self.inclusion_guard and value == '': |
| 458 | # The file double-inclusion guard is not an option. |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 459 | self.templates.append(line) |
| 460 | return None |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 461 | template = (name, |
| 462 | m.group('indentation'), |
| 463 | m.group('define') + name + |
| 464 | m.group('arguments') + m.group('separator')) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 465 | self.templates.append(template) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 466 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 467 | return (active, name, value, self.current_section) |
| 468 | |
| 469 | def parse_file(self): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 470 | """Parse the whole file and return the settings.""" |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 471 | 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 Mezei | e7742b3 | 2024-06-26 18:04:09 +0200 | [diff] [blame^] | 478 | #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 Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 501 | |
| 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 Mezei | e7742b3 | 2024-06-26 18:04:09 +0200 | [diff] [blame^] | 508 | name, indent, middle = template |
| 509 | line = self._format_template(settings[name], indent, middle) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 510 | 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 Mezei | 8a64d8e | 2024-06-10 15:23:43 +0200 | [diff] [blame] | 519 | |
| 520 | # Not modified so no need to write to the file |
| 521 | if not self.modified and filename == self.filename: |
| 522 | return |
| 523 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 524 | with open(filename, 'w', encoding='utf-8') as output: |
| 525 | self.write_to_stream(settings, output) |
| 526 | |
| 527 | class MbedtlsConfigFile(ConfigFile): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 528 | """Representation of an MbedTLS configuration file.""" |
| 529 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 530 | _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 Mezei | 93a6d1f | 2024-06-26 18:01:09 +0200 | [diff] [blame] | 539 | super().__init__(self.default_path, 'Mbed TLS', filename) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 540 | self.current_section = 'header' |
| 541 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 542 | class CryptoConfigFile(ConfigFile): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 543 | """Representation of an Crypto configuration file.""" |
| 544 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 545 | _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 Mezei | 93a6d1f | 2024-06-26 18:01:09 +0200 | [diff] [blame] | 554 | super().__init__(self.default_path, 'Crypto', filename) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 555 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 556 | class MbedtlsConfig(Config): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 557 | """Representation of the Mbed TLS configuration. |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 558 | |
| 559 | See the documentation of the `Config` class for methods to query |
| 560 | and modify the configuration. |
| 561 | """ |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 562 | def __init__(self, filename=None): |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 563 | """Read the Mbed TLS configuration file.""" |
| 564 | super().__init__() |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 565 | self.configfile = MbedtlsConfigFile(filename) |
| 566 | self.settings.update({name: Setting(active, name, value, section) |
Gabor Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 567 | for (active, name, value, section) |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 568 | in self.configfile.parse_file()}) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 569 | |
| 570 | def set(self, name, value=None): |
| 571 | if name not in self.settings: |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 572 | self.configfile.templates.append((name, '', '#define ' + name + ' ')) |
| 573 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 574 | super().set(name, value) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 575 | |
| 576 | def write(self, filename=None): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 577 | """Write the whole configuration to the file it was read from. |
| 578 | |
| 579 | If filename is specified, write to this file instead. |
| 580 | """ |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 581 | self.configfile.write(self.settings, filename) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 582 | |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 583 | def filename(self): |
| 584 | return self.configfile.filename |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 585 | |
| 586 | class CryptoConfig(Config): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 587 | """Representation of the PSA crypto configuration. |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 588 | |
| 589 | See the documentation of the `Config` class for methods to query |
| 590 | and modify the configuration. |
| 591 | """ |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 592 | def __init__(self, filename=None): |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 593 | """Read the PSA crypto configuration file.""" |
| 594 | super().__init__() |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 595 | self.configfile = CryptoConfigFile(filename) |
| 596 | self.settings.update({name: Setting(active, name, value, section) |
Gabor Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 597 | for (active, name, value, section) |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 598 | in self.configfile.parse_file()}) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 599 | |
Gabor Mezei | d723b51 | 2024-06-07 15:31:52 +0200 | [diff] [blame] | 600 | def set(self, name, value='1'): |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 601 | if name in PSA_UNSUPPORTED_FEATURE: |
Gabor Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 602 | raise ValueError(f'Feature is unsupported: \'{name}\'') |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 603 | if name in PSA_UNSTABLE_FEATURE: |
Gabor Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 604 | raise ValueError(f'Feature is unstable: \'{name}\'') |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 605 | |
Gabor Mezei | d723b51 | 2024-06-07 15:31:52 +0200 | [diff] [blame] | 606 | # The default value in the crypto config is '1' |
| 607 | if not value: |
| 608 | value = '1' |
| 609 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 610 | if name not in self.settings: |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 611 | self.configfile.templates.append((name, '', '#define ' + name + ' ')) |
| 612 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 613 | super().set(name, value) |
| 614 | |
| 615 | def write(self, filename=None): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 616 | """Write the whole configuration to the file it was read from. |
| 617 | |
| 618 | If filename is specified, write to this file instead. |
| 619 | """ |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 620 | self.configfile.write(self.settings, filename) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 621 | |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 622 | def filename(self): |
| 623 | return self.configfile.filename |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 624 | |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 625 | class MultiConfig(Config): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 626 | """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 Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 631 | |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 632 | 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 Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 639 | |
| 640 | _crypto_regexp = re.compile(r'$PSA_.*') |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 641 | 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 Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 648 | else: |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 649 | return self.mbedtls_configfile |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 650 | |
Gabor Mezei | 8a64d8e | 2024-06-10 15:23:43 +0200 | [diff] [blame] | 651 | def __setitem__(self, name, value): |
| 652 | super().__setitem__(name, value) |
| 653 | self.settings[name].configfile.modified = True |
| 654 | |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 655 | def set(self, name, value=None): |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 656 | configfile = self._get_configfile(name) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 657 | |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 658 | if configfile == self.crypto_configfile: |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 659 | if name in PSA_UNSUPPORTED_FEATURE: |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 660 | raise ValueError(f'Feature is unsupported: \'{name}\'') |
Gabor Mezei | 542fd38 | 2024-06-10 14:07:42 +0200 | [diff] [blame] | 661 | if name in PSA_UNSTABLE_FEATURE: |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 662 | raise ValueError(f'Feature is unstable: \'{name}\'') |
| 663 | |
Gabor Mezei | d723b51 | 2024-06-07 15:31:52 +0200 | [diff] [blame] | 664 | # The default value in the crypto config is '1' |
| 665 | if not value: |
| 666 | value = '1' |
| 667 | |
Gabor Mezei | 8a64d8e | 2024-06-10 15:23:43 +0200 | [diff] [blame] | 668 | 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 Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 673 | configfile.templates.append((name, '', '#define ' + name + ' ')) |
Gabor Mezei | 8a64d8e | 2024-06-10 15:23:43 +0200 | [diff] [blame] | 674 | configfile.modified = True |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 675 | |
| 676 | super().set(name, value) |
| 677 | |
Gabor Mezei | 8a64d8e | 2024-06-10 15:23:43 +0200 | [diff] [blame] | 678 | 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 Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 717 | def write(self, mbedtls_file=None, crypto_file=None): |
Gabor Mezei | 62a9bd0 | 2024-06-07 13:44:40 +0200 | [diff] [blame] | 718 | """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 Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 723 | self.mbedtls_configfile.write(self.settings, mbedtls_file) |
| 724 | self.crypto_configfile.write(self.settings, crypto_file) |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 725 | |
Gabor Mezei | ee521b6 | 2024-06-07 13:50:41 +0200 | [diff] [blame] | 726 | 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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 731 | |
| 732 | if __name__ == '__main__': |
Gabor Mezei | 92065ed | 2024-06-07 13:47:59 +0200 | [diff] [blame] | 733 | #pylint: disable=too-many-statements |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 734 | def main(): |
Bence Szépkúti | bb0cfeb | 2021-05-28 09:42:25 +0200 | [diff] [blame] | 735 | """Command line mbedtls_config.h manipulation tool.""" |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 736 | parser = argparse.ArgumentParser(description=""" |
Fredrik Hesse | 0ec8a90 | 2021-10-04 22:13:51 +0200 | [diff] [blame] | 737 | Mbed TLS configuration file manipulation tool. |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 738 | """) |
| 739 | parser.add_argument('--file', '-f', |
| 740 | help="""File to read (and modify if requested). |
| 741 | Default: {}. |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 742 | """.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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 747 | parser.add_argument('--force', '-o', |
Gilles Peskine | 435ce22 | 2019-08-01 23:13:47 +0200 | [diff] [blame] | 748 | action='store_true', |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 749 | help="""For the set command, if SYMBOL is not |
| 750 | present, add a definition for it.""") |
Gilles Peskine | c190c90 | 2019-08-01 23:31:05 +0200 | [diff] [blame] | 751 | parser.add_argument('--write', '-w', metavar='FILE', |
Gilles Peskine | 40f103c | 2019-07-27 23:44:01 +0200 | [diff] [blame] | 752 | help="""File to write to instead of the input file.""") |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 753 | 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 Peskine | 0c7fcd2 | 2019-08-01 23:14:00 +0200 | [diff] [blame] | 771 | parser_set.add_argument('value', metavar='VALUE', nargs='?', |
| 772 | default='') |
Gilles Peskine | 8e90cf4 | 2021-05-27 22:12:57 +0200 | [diff] [blame] | 773 | 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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 778 | 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 Peskine | 8e90cf4 | 2021-05-27 22:12:57 +0200 | [diff] [blame] | 783 | 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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 788 | |
| 789 | def add_adapter(name, function, description): |
| 790 | subparser = subparsers.add_parser(name, help=description) |
| 791 | subparser.set_defaults(adapter=function) |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 792 | add_adapter('baremetal', baremetal_adapter, |
| 793 | """Like full, but exclude features that require platform |
| 794 | features such as file input-output.""") |
Gilles Peskine | 120f29d | 2021-09-01 19:51:19 +0200 | [diff] [blame] | 795 | add_adapter('baremetal_size', baremetal_size_adapter, |
| 796 | """Like baremetal, but exclude debugging features. |
| 797 | Useful for code size measurements.""") |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 798 | 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 Peskine | 30de2e8 | 2020-04-20 21:39:22 +0200 | [diff] [blame] | 803 | add_adapter('full_no_deprecated', no_deprecated_adapter(full_adapter), |
Gilles Peskine | be1d609 | 2020-04-12 14:17:16 +0200 | [diff] [blame] | 804 | """Uncomment most non-deprecated features. |
| 805 | Like "full", but without deprecated features. |
| 806 | """) |
Paul Elliott | fb81f77 | 2023-10-18 17:44:59 +0100 | [diff] [blame] | 807 | 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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 811 | add_adapter('realfull', realfull_adapter, |
Gilles Peskine | 53d41ae | 2019-07-27 23:31:53 +0200 | [diff] [blame] | 812 | """Uncomment all boolean #defines. |
| 813 | Suitable for generating documentation, but not for building.""") |
Gilles Peskine | 31987c6 | 2020-01-31 14:23:30 +0100 | [diff] [blame] | 814 | 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 Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 822 | |
| 823 | args = parser.parse_args() |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 824 | config = MultiConfig(args.file, args.cryptofile) |
Gilles Peskine | 90b30b6 | 2019-07-28 00:36:53 +0200 | [diff] [blame] | 825 | if args.command is None: |
| 826 | parser.print_help() |
| 827 | return 1 |
| 828 | elif args.command == 'get': |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 829 | if args.symbol in config: |
| 830 | value = config[args.symbol] |
| 831 | if value: |
| 832 | sys.stdout.write(value + '\n') |
Gilles Peskine | e22a4da | 2020-03-24 15:43:49 +0100 | [diff] [blame] | 833 | return 0 if args.symbol in config else 1 |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 834 | elif args.command == 'set': |
Gilles Peskine | 98eb365 | 2019-07-28 16:39:19 +0200 | [diff] [blame] | 835 | if not args.force and args.symbol not in config.settings: |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 836 | sys.stderr.write("A #define for the symbol {} " |
Gilles Peskine | 221df1e | 2019-08-01 23:14:29 +0200 | [diff] [blame] | 837 | "was not found in {}\n" |
Gabor Mezei | 3678dee | 2024-06-04 19:58:43 +0200 | [diff] [blame] | 838 | .format(args.symbol, config.filename(args.symbol))) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 839 | return 1 |
| 840 | config.set(args.symbol, value=args.value) |
Gilles Peskine | 8e90cf4 | 2021-05-27 22:12:57 +0200 | [diff] [blame] | 841 | elif args.command == 'set-all': |
| 842 | config.change_matching(args.regexs, True) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 843 | elif args.command == 'unset': |
| 844 | config.unset(args.symbol) |
Gilles Peskine | 8e90cf4 | 2021-05-27 22:12:57 +0200 | [diff] [blame] | 845 | elif args.command == 'unset-all': |
| 846 | config.change_matching(args.regexs, False) |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 847 | else: |
| 848 | config.adapt(args.adapter) |
Gilles Peskine | 40f103c | 2019-07-27 23:44:01 +0200 | [diff] [blame] | 849 | config.write(args.write) |
Gilles Peskine | e22a4da | 2020-03-24 15:43:49 +0100 | [diff] [blame] | 850 | return 0 |
Gilles Peskine | b406389 | 2019-07-27 21:36:44 +0200 | [diff] [blame] | 851 | |
| 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()) |