blob: d562cb156cf59d151083a6bb67ad7e8c180c0a0d [file] [log] [blame]
Gilles Peskinea3ed34f2021-01-05 21:11:16 +01001/*
Gilles Peskine0d980b82021-01-05 23:34:27 +01002 * Common code library for SSL test programs.
3 *
4 * In addition to the functions in this file, there is shared source code
5 * that cannot be compiled separately in "ssl_test_common_source.c".
Gilles Peskinea3ed34f2021-01-05 21:11:16 +01006 *
7 * Copyright The Mbed TLS Contributors
Dave Rodgman16799db2023-11-02 19:47:20 +00008 * SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
Gilles Peskinea3ed34f2021-01-05 21:11:16 +01009 */
10
Mateusz Starzyk6c2e9b62021-05-19 17:54:54 +020011#define MBEDTLS_ALLOW_PRIVATE_ACCESS
12
Gilles Peskinea3ed34f2021-01-05 21:11:16 +010013#include "ssl_test_lib.h"
14
Gilles Peskinee374b952021-02-03 00:05:19 +010015#if defined(MBEDTLS_TEST_HOOKS)
Paul Elliott17c119a2023-12-08 16:55:03 +000016#include "test/threading_helpers.h"
Gilles Peskinee374b952021-02-03 00:05:19 +010017#endif
18
Gilles Peskineab7ce962021-01-05 21:27:53 +010019#if !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE)
20
Valerio Settiacd32c02023-06-29 18:06:29 +020021#define ARRAY_LENGTH(x) (sizeof(x)/sizeof(x[0]))
22
Gilles Peskine449bd832023-01-11 14:50:10 +010023void my_debug(void *ctx, int level,
24 const char *file, int line,
25 const char *str)
Gilles Peskine504c1a32021-01-05 23:40:14 +010026{
27 const char *p, *basename;
28
29 /* Extract basename from file */
Gilles Peskine449bd832023-01-11 14:50:10 +010030 for (p = basename = file; *p != '\0'; p++) {
31 if (*p == '/' || *p == '\\') {
Gilles Peskine504c1a32021-01-05 23:40:14 +010032 basename = p + 1;
Gilles Peskine449bd832023-01-11 14:50:10 +010033 }
34 }
Gilles Peskine504c1a32021-01-05 23:40:14 +010035
Gilles Peskine449bd832023-01-11 14:50:10 +010036 mbedtls_fprintf((FILE *) ctx, "%s:%04d: |%d| %s",
37 basename, line, level, str);
38 fflush((FILE *) ctx);
Gilles Peskine504c1a32021-01-05 23:40:14 +010039}
40
Raoul Strackx9ed9bc92020-06-22 14:08:57 +020041#if defined(MBEDTLS_HAVE_TIME)
Gilles Peskine449bd832023-01-11 14:50:10 +010042mbedtls_time_t dummy_constant_time(mbedtls_time_t *time)
Gilles Peskine504c1a32021-01-05 23:40:14 +010043{
44 (void) time;
45 return 0x5af2a056;
46}
Raoul Strackx9ed9bc92020-06-22 14:08:57 +020047#endif
Gilles Peskine504c1a32021-01-05 23:40:14 +010048
Gilles Peskine8eb29432021-02-03 20:07:11 +010049#if !defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
Gilles Peskine449bd832023-01-11 14:50:10 +010050static int dummy_entropy(void *data, unsigned char *output, size_t len)
Gilles Peskine504c1a32021-01-05 23:40:14 +010051{
52 size_t i;
53 int ret;
54 (void) data;
55
Gilles Peskine449bd832023-01-11 14:50:10 +010056 ret = mbedtls_entropy_func(data, output, len);
57 for (i = 0; i < len; i++) {
Gilles Peskine504c1a32021-01-05 23:40:14 +010058 //replace result with pseudo random
59 output[i] = (unsigned char) rand();
60 }
Gilles Peskine449bd832023-01-11 14:50:10 +010061 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +010062}
Gilles Peskine8eb29432021-02-03 20:07:11 +010063#endif
Gilles Peskine504c1a32021-01-05 23:40:14 +010064
Gilles Peskine449bd832023-01-11 14:50:10 +010065void rng_init(rng_context_t *rng)
Gilles Peskinedaa94c42021-01-13 18:38:27 +010066{
Gilles Peskine8eb29432021-02-03 20:07:11 +010067#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
68 (void) rng;
Gilles Peskine449bd832023-01-11 14:50:10 +010069 psa_crypto_init();
Gilles Peskine8eb29432021-02-03 20:07:11 +010070#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
71
Gilles Peskineba749042021-01-13 20:02:03 +010072#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +010073 mbedtls_ctr_drbg_init(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +010074#elif defined(MBEDTLS_HMAC_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +010075 mbedtls_hmac_drbg_init(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +010076#else
77#error "No DRBG available"
78#endif
79
Gilles Peskine449bd832023-01-11 14:50:10 +010080 mbedtls_entropy_init(&rng->entropy);
Gilles Peskine8eb29432021-02-03 20:07:11 +010081#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskinedaa94c42021-01-13 18:38:27 +010082}
83
Gilles Peskine449bd832023-01-11 14:50:10 +010084int rng_seed(rng_context_t *rng, int reproducible, const char *pers)
Gilles Peskinedaa94c42021-01-13 18:38:27 +010085{
Gilles Peskineaaedbdc2021-02-03 13:55:22 +010086#if defined(MBEDTLS_USE_PSA_CRYPTO)
Gilles Peskine449bd832023-01-11 14:50:10 +010087 if (reproducible) {
88 mbedtls_fprintf(stderr,
89 "MBEDTLS_USE_PSA_CRYPTO does not support reproducible mode.\n");
90 return -1;
Gilles Peskineaaedbdc2021-02-03 13:55:22 +010091 }
92#endif
Gilles Peskine8eb29432021-02-03 20:07:11 +010093#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
94 /* The PSA crypto RNG does its own seeding. */
95 (void) rng;
96 (void) pers;
Gilles Peskine449bd832023-01-11 14:50:10 +010097 if (reproducible) {
98 mbedtls_fprintf(stderr,
99 "The PSA RNG does not support reproducible mode.\n");
100 return -1;
Gilles Peskine8eb29432021-02-03 20:07:11 +0100101 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100102 return 0;
Gilles Peskine8eb29432021-02-03 20:07:11 +0100103#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskine449bd832023-01-11 14:50:10 +0100104 int (*f_entropy)(void *, unsigned char *, size_t) =
105 (reproducible ? dummy_entropy : mbedtls_entropy_func);
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100106
Gilles Peskine449bd832023-01-11 14:50:10 +0100107 if (reproducible) {
108 srand(1);
109 }
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100110
Gilles Peskineba749042021-01-13 20:02:03 +0100111#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100112 int ret = mbedtls_ctr_drbg_seed(&rng->drbg,
113 f_entropy, &rng->entropy,
114 (const unsigned char *) pers,
115 strlen(pers));
Gilles Peskineba749042021-01-13 20:02:03 +0100116#elif defined(MBEDTLS_HMAC_DRBG_C)
Manuel Pégourié-Gonnardbef824d2023-03-17 12:50:01 +0100117#if defined(MBEDTLS_MD_CAN_SHA256)
Gilles Peskineba749042021-01-13 20:02:03 +0100118 const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA256;
Manuel Pégourié-Gonnardbef824d2023-03-17 12:50:01 +0100119#elif defined(MBEDTLS_MD_CAN_SHA512)
Gilles Peskineba749042021-01-13 20:02:03 +0100120 const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA512;
121#else
122#error "No message digest available for HMAC_DRBG"
123#endif
Gilles Peskine449bd832023-01-11 14:50:10 +0100124 int ret = mbedtls_hmac_drbg_seed(&rng->drbg,
125 mbedtls_md_info_from_type(md_type),
126 f_entropy, &rng->entropy,
127 (const unsigned char *) pers,
128 strlen(pers));
Gilles Peskine8eb29432021-02-03 20:07:11 +0100129#else /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */
Gilles Peskineba749042021-01-13 20:02:03 +0100130#error "No DRBG available"
Gilles Peskine8eb29432021-02-03 20:07:11 +0100131#endif /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */
Gilles Peskineba749042021-01-13 20:02:03 +0100132
Gilles Peskine449bd832023-01-11 14:50:10 +0100133 if (ret != 0) {
134 mbedtls_printf(" failed\n ! mbedtls_ctr_drbg_seed returned -0x%x\n",
135 (unsigned int) -ret);
136 return ret;
Gilles Peskinef1cb75f2021-01-13 18:46:01 +0100137 }
Gilles Peskine8eb29432021-02-03 20:07:11 +0100138#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100139
Gilles Peskine449bd832023-01-11 14:50:10 +0100140 return 0;
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100141}
142
Gilles Peskine449bd832023-01-11 14:50:10 +0100143void rng_free(rng_context_t *rng)
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100144{
Gilles Peskine8eb29432021-02-03 20:07:11 +0100145#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
146 (void) rng;
147 /* Deinitialize the PSA crypto subsystem. This deactivates all PSA APIs.
148 * This is ok because none of our applications try to do any crypto after
149 * deinitializing the RNG. */
Gilles Peskine449bd832023-01-11 14:50:10 +0100150 mbedtls_psa_crypto_free();
Gilles Peskine8eb29432021-02-03 20:07:11 +0100151#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
152
Gilles Peskineba749042021-01-13 20:02:03 +0100153#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100154 mbedtls_ctr_drbg_free(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +0100155#elif defined(MBEDTLS_HMAC_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100156 mbedtls_hmac_drbg_free(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +0100157#else
158#error "No DRBG available"
159#endif
160
Gilles Peskine449bd832023-01-11 14:50:10 +0100161 mbedtls_entropy_free(&rng->entropy);
Gilles Peskine8eb29432021-02-03 20:07:11 +0100162#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100163}
164
Gilles Peskine449bd832023-01-11 14:50:10 +0100165int rng_get(void *p_rng, unsigned char *output, size_t output_len)
Gilles Peskine535fb372021-01-13 18:59:46 +0100166{
Gilles Peskine8eb29432021-02-03 20:07:11 +0100167#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
168 (void) p_rng;
Gilles Peskine449bd832023-01-11 14:50:10 +0100169 return mbedtls_psa_get_random(MBEDTLS_PSA_RANDOM_STATE,
170 output, output_len);
Gilles Peskine8eb29432021-02-03 20:07:11 +0100171#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskine535fb372021-01-13 18:59:46 +0100172 rng_context_t *rng = p_rng;
Gilles Peskine8eb29432021-02-03 20:07:11 +0100173
Gilles Peskineba749042021-01-13 20:02:03 +0100174#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100175 return mbedtls_ctr_drbg_random(&rng->drbg, output, output_len);
Gilles Peskineba749042021-01-13 20:02:03 +0100176#elif defined(MBEDTLS_HMAC_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100177 return mbedtls_hmac_drbg_random(&rng->drbg, output, output_len);
Gilles Peskineba749042021-01-13 20:02:03 +0100178#else
179#error "No DRBG available"
180#endif
Gilles Peskine8eb29432021-02-03 20:07:11 +0100181
182#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskine535fb372021-01-13 18:59:46 +0100183}
184
Gilles Peskine449bd832023-01-11 14:50:10 +0100185int key_opaque_alg_parse(const char *arg, const char **alg1, const char **alg2)
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200186{
Gilles Peskine449bd832023-01-11 14:50:10 +0100187 char *separator;
188 if ((separator = strchr(arg, ',')) == NULL) {
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200189 return 1;
Gilles Peskine449bd832023-01-11 14:50:10 +0100190 }
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200191 *separator = '\0';
192
193 *alg1 = arg;
194 *alg2 = separator + 1;
195
Gilles Peskine449bd832023-01-11 14:50:10 +0100196 if (strcmp(*alg1, "rsa-sign-pkcs1") != 0 &&
197 strcmp(*alg1, "rsa-sign-pss") != 0 &&
198 strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
199 strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
200 strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
201 strcmp(*alg1, "rsa-decrypt") != 0 &&
202 strcmp(*alg1, "ecdsa-sign") != 0 &&
203 strcmp(*alg1, "ecdh") != 0) {
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200204 return 1;
Gilles Peskine449bd832023-01-11 14:50:10 +0100205 }
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200206
Gilles Peskine449bd832023-01-11 14:50:10 +0100207 if (strcmp(*alg2, "rsa-sign-pkcs1") != 0 &&
208 strcmp(*alg2, "rsa-sign-pss") != 0 &&
209 strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
210 strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
211 strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
212 strcmp(*alg2, "rsa-decrypt") != 0 &&
213 strcmp(*alg2, "ecdsa-sign") != 0 &&
214 strcmp(*alg2, "ecdh") != 0 &&
215 strcmp(*alg2, "none") != 0) {
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200216 return 1;
Gilles Peskine449bd832023-01-11 14:50:10 +0100217 }
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200218
219 return 0;
220}
221
Przemek Stekiel76a41f52022-05-04 13:55:23 +0200222#if defined(MBEDTLS_USE_PSA_CRYPTO)
Gilles Peskine449bd832023-01-11 14:50:10 +0100223int key_opaque_set_alg_usage(const char *alg1, const char *alg2,
224 psa_algorithm_t *psa_alg1,
225 psa_algorithm_t *psa_alg2,
226 psa_key_usage_t *usage,
227 mbedtls_pk_type_t key_type)
Przemek Stekiel01396a12022-05-02 13:41:53 +0200228{
Gilles Peskine449bd832023-01-11 14:50:10 +0100229 if (strcmp(alg1, "none") != 0) {
230 const char *algs[] = { alg1, alg2 };
Przemek Stekielcb20d202022-05-06 08:42:34 +0200231 psa_algorithm_t *psa_algs[] = { psa_alg1, psa_alg2 };
Przemek Stekiel01396a12022-05-02 13:41:53 +0200232
Gilles Peskine449bd832023-01-11 14:50:10 +0100233 for (int i = 0; i < 2; i++) {
234 if (strcmp(algs[i], "rsa-sign-pkcs1") == 0) {
235 *psa_algs[i] = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200236 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100237 } else if (strcmp(algs[i], "rsa-sign-pss") == 0) {
238 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200239 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100240 } else if (strcmp(algs[i], "rsa-sign-pss-sha256") == 0) {
241 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_256);
Ronald Cron50969e32022-09-16 15:54:33 +0200242 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100243 } else if (strcmp(algs[i], "rsa-sign-pss-sha384") == 0) {
244 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_384);
Ronald Cron50969e32022-09-16 15:54:33 +0200245 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100246 } else if (strcmp(algs[i], "rsa-sign-pss-sha512") == 0) {
247 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_512);
Ronald Cron50969e32022-09-16 15:54:33 +0200248 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100249 } else if (strcmp(algs[i], "rsa-decrypt") == 0) {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200250 *psa_algs[i] = PSA_ALG_RSA_PKCS1V15_CRYPT;
251 *usage |= PSA_KEY_USAGE_DECRYPT;
Gilles Peskine449bd832023-01-11 14:50:10 +0100252 } else if (strcmp(algs[i], "ecdsa-sign") == 0) {
253 *psa_algs[i] = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200254 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100255 } else if (strcmp(algs[i], "ecdh") == 0) {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200256 *psa_algs[i] = PSA_ALG_ECDH;
257 *usage |= PSA_KEY_USAGE_DERIVE;
Gilles Peskine449bd832023-01-11 14:50:10 +0100258 } else if (strcmp(algs[i], "none") == 0) {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200259 *psa_algs[i] = PSA_ALG_NONE;
260 }
261 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100262 } else {
263 if (key_type == MBEDTLS_PK_ECKEY) {
264 *psa_alg1 = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200265 *psa_alg2 = PSA_ALG_ECDH;
266 *usage = PSA_KEY_USAGE_SIGN_HASH | PSA_KEY_USAGE_DERIVE;
Gilles Peskine449bd832023-01-11 14:50:10 +0100267 } else if (key_type == MBEDTLS_PK_RSA) {
268 *psa_alg1 = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
269 *psa_alg2 = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200270 *usage = PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100271 } else {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200272 return 1;
273 }
Przemek Stekiel01396a12022-05-02 13:41:53 +0200274 }
275
276 return 0;
277}
Przemek Stekiel76a41f52022-05-04 13:55:23 +0200278#endif /* MBEDTLS_USE_PSA_CRYPTO */
Przemek Stekiel01396a12022-05-02 13:41:53 +0200279
Gilles Peskine504c1a32021-01-05 23:40:14 +0100280#if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
Gilles Peskine449bd832023-01-11 14:50:10 +0100281int ca_callback(void *data, mbedtls_x509_crt const *child,
282 mbedtls_x509_crt **candidates)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100283{
284 int ret = 0;
285 mbedtls_x509_crt *ca = (mbedtls_x509_crt *) data;
286 mbedtls_x509_crt *first;
287
288 /* This is a test-only implementation of the CA callback
289 * which always returns the entire list of trusted certificates.
290 * Production implementations managing a large number of CAs
291 * should use an efficient presentation and lookup for the
292 * set of trusted certificates (such as a hashtable) and only
293 * return those trusted certificates which satisfy basic
294 * parental checks, such as the matching of child `Issuer`
295 * and parent `Subject` field or matching key identifiers. */
296 ((void) child);
297
Gilles Peskine449bd832023-01-11 14:50:10 +0100298 first = mbedtls_calloc(1, sizeof(mbedtls_x509_crt));
299 if (first == NULL) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100300 ret = -1;
301 goto exit;
302 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100303 mbedtls_x509_crt_init(first);
Gilles Peskine504c1a32021-01-05 23:40:14 +0100304
Gilles Peskine449bd832023-01-11 14:50:10 +0100305 if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100306 ret = -1;
307 goto exit;
308 }
309
Gilles Peskine449bd832023-01-11 14:50:10 +0100310 while (ca->next != NULL) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100311 ca = ca->next;
Gilles Peskine449bd832023-01-11 14:50:10 +0100312 if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100313 ret = -1;
314 goto exit;
315 }
316 }
317
318exit:
319
Gilles Peskine449bd832023-01-11 14:50:10 +0100320 if (ret != 0) {
321 mbedtls_x509_crt_free(first);
322 mbedtls_free(first);
Gilles Peskine504c1a32021-01-05 23:40:14 +0100323 first = NULL;
324 }
325
326 *candidates = first;
Gilles Peskine449bd832023-01-11 14:50:10 +0100327 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100328}
329#endif /* MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK */
330
Gilles Peskine449bd832023-01-11 14:50:10 +0100331int delayed_recv(void *ctx, unsigned char *buf, size_t len)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100332{
333 static int first_try = 1;
334 int ret;
335
Gilles Peskine449bd832023-01-11 14:50:10 +0100336 if (first_try) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100337 first_try = 0;
Gilles Peskine449bd832023-01-11 14:50:10 +0100338 return MBEDTLS_ERR_SSL_WANT_READ;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100339 }
340
Gilles Peskine449bd832023-01-11 14:50:10 +0100341 ret = mbedtls_net_recv(ctx, buf, len);
342 if (ret != MBEDTLS_ERR_SSL_WANT_READ) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100343 first_try = 1; /* Next call will be a new operation */
Gilles Peskine449bd832023-01-11 14:50:10 +0100344 }
345 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100346}
347
Gilles Peskine449bd832023-01-11 14:50:10 +0100348int delayed_send(void *ctx, const unsigned char *buf, size_t len)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100349{
350 static int first_try = 1;
351 int ret;
352
Gilles Peskine449bd832023-01-11 14:50:10 +0100353 if (first_try) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100354 first_try = 0;
Gilles Peskine449bd832023-01-11 14:50:10 +0100355 return MBEDTLS_ERR_SSL_WANT_WRITE;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100356 }
357
Gilles Peskine449bd832023-01-11 14:50:10 +0100358 ret = mbedtls_net_send(ctx, buf, len);
359 if (ret != MBEDTLS_ERR_SSL_WANT_WRITE) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100360 first_try = 1; /* Next call will be a new operation */
Gilles Peskine449bd832023-01-11 14:50:10 +0100361 }
362 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100363}
364
365#if !defined(MBEDTLS_TIMING_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100366int idle(mbedtls_net_context *fd,
367 int idle_reason)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100368#else
Gilles Peskine449bd832023-01-11 14:50:10 +0100369int idle(mbedtls_net_context *fd,
370 mbedtls_timing_delay_context *timer,
371 int idle_reason)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100372#endif
373{
374 int ret;
375 int poll_type = 0;
376
Gilles Peskine449bd832023-01-11 14:50:10 +0100377 if (idle_reason == MBEDTLS_ERR_SSL_WANT_WRITE) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100378 poll_type = MBEDTLS_NET_POLL_WRITE;
Gilles Peskine449bd832023-01-11 14:50:10 +0100379 } else if (idle_reason == MBEDTLS_ERR_SSL_WANT_READ) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100380 poll_type = MBEDTLS_NET_POLL_READ;
Gilles Peskine449bd832023-01-11 14:50:10 +0100381 }
Gilles Peskine504c1a32021-01-05 23:40:14 +0100382#if !defined(MBEDTLS_TIMING_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100383 else {
384 return 0;
385 }
Gilles Peskine504c1a32021-01-05 23:40:14 +0100386#endif
387
Gilles Peskine449bd832023-01-11 14:50:10 +0100388 while (1) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100389 /* Check if timer has expired */
390#if defined(MBEDTLS_TIMING_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100391 if (timer != NULL &&
392 mbedtls_timing_get_delay(timer) == 2) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100393 break;
394 }
395#endif /* MBEDTLS_TIMING_C */
396
397 /* Check if underlying transport became available */
Gilles Peskine449bd832023-01-11 14:50:10 +0100398 if (poll_type != 0) {
399 ret = mbedtls_net_poll(fd, poll_type, 0);
400 if (ret < 0) {
401 return ret;
402 }
403 if (ret == poll_type) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100404 break;
Gilles Peskine449bd832023-01-11 14:50:10 +0100405 }
Gilles Peskine504c1a32021-01-05 23:40:14 +0100406 }
407 }
408
Gilles Peskine449bd832023-01-11 14:50:10 +0100409 return 0;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100410}
411
Gilles Peskine53dea742021-02-02 22:55:06 +0100412#if defined(MBEDTLS_TEST_HOOKS)
413
Gilles Peskine449bd832023-01-11 14:50:10 +0100414void test_hooks_init(void)
Gilles Peskine53dea742021-02-02 22:55:06 +0100415{
Gilles Peskine449bd832023-01-11 14:50:10 +0100416 mbedtls_test_info_reset();
Gilles Peskinee374b952021-02-03 00:05:19 +0100417
418#if defined(MBEDTLS_TEST_MUTEX_USAGE)
Gilles Peskine449bd832023-01-11 14:50:10 +0100419 mbedtls_test_mutex_usage_init();
Gilles Peskinee374b952021-02-03 00:05:19 +0100420#endif
Gilles Peskine53dea742021-02-02 22:55:06 +0100421}
422
Gilles Peskine449bd832023-01-11 14:50:10 +0100423int test_hooks_failure_detected(void)
Gilles Peskine53dea742021-02-02 22:55:06 +0100424{
Gilles Peskinee374b952021-02-03 00:05:19 +0100425#if defined(MBEDTLS_TEST_MUTEX_USAGE)
426 /* Errors are reported via mbedtls_test_info. */
Gilles Peskine449bd832023-01-11 14:50:10 +0100427 mbedtls_test_mutex_usage_check();
Gilles Peskinee374b952021-02-03 00:05:19 +0100428#endif
429
Gilles Peskine449bd832023-01-11 14:50:10 +0100430 if (mbedtls_test_info.result != MBEDTLS_TEST_RESULT_SUCCESS) {
431 return 1;
432 }
433 return 0;
Gilles Peskine53dea742021-02-02 22:55:06 +0100434}
435
Gilles Peskine449bd832023-01-11 14:50:10 +0100436void test_hooks_free(void)
Gilles Peskine53dea742021-02-02 22:55:06 +0100437{
Paul Elliottf25d8312023-11-23 18:49:43 +0000438#if defined(MBEDTLS_TEST_MUTEX_USAGE)
439 mbedtls_test_mutex_usage_end();
440#endif
Gilles Peskine53dea742021-02-02 22:55:06 +0100441}
442
443#endif /* MBEDTLS_TEST_HOOKS */
444
Valerio Settiacd32c02023-06-29 18:06:29 +0200445static const struct {
446 uint16_t tls_id;
447 const char *name;
448 uint8_t is_supported;
Valerio Setti5bdebb22023-07-07 10:52:18 +0200449} tls_id_group_name_table[] =
Valerio Settiacd32c02023-06-29 18:06:29 +0200450{
451#if defined(MBEDTLS_ECP_DP_SECP521R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_521)
Valerio Settideb67642023-07-03 14:26:04 +0200452 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP521R1, "secp521r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200453#else
Valerio Settideb67642023-07-03 14:26:04 +0200454 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP521R1, "secp521r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200455#endif
456#if defined(MBEDTLS_ECP_DP_BP512R1_ENABLED) || defined(PSA_WANT_ECC_BRAINPOOL_P_R1_512)
Valerio Settideb67642023-07-03 14:26:04 +0200457 { MBEDTLS_SSL_IANA_TLS_GROUP_BP512R1, "brainpoolP512r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200458#else
Valerio Settideb67642023-07-03 14:26:04 +0200459 { MBEDTLS_SSL_IANA_TLS_GROUP_BP512R1, "brainpoolP512r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200460#endif
461#if defined(MBEDTLS_ECP_DP_SECP384R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_384)
Valerio Settideb67642023-07-03 14:26:04 +0200462 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP384R1, "secp384r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200463#else
Valerio Settideb67642023-07-03 14:26:04 +0200464 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP384R1, "secp384r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200465#endif
466#if defined(MBEDTLS_ECP_DP_BP384R1_ENABLED) || defined(PSA_WANT_ECC_BRAINPOOL_P_R1_384)
Valerio Settideb67642023-07-03 14:26:04 +0200467 { MBEDTLS_SSL_IANA_TLS_GROUP_BP384R1, "brainpoolP384r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200468#else
Valerio Settideb67642023-07-03 14:26:04 +0200469 { MBEDTLS_SSL_IANA_TLS_GROUP_BP384R1, "brainpoolP384r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200470#endif
471#if defined(MBEDTLS_ECP_DP_SECP256R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_256)
Valerio Settideb67642023-07-03 14:26:04 +0200472 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256R1, "secp256r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200473#else
Valerio Settideb67642023-07-03 14:26:04 +0200474 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256R1, "secp256r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200475#endif
476#if defined(MBEDTLS_ECP_DP_SECP256K1_ENABLED) || defined(PSA_WANT_ECC_SECP_K1_256)
Valerio Settideb67642023-07-03 14:26:04 +0200477 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256K1, "secp256k1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200478#else
Valerio Settideb67642023-07-03 14:26:04 +0200479 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256K1, "secp256k1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200480#endif
481#if defined(MBEDTLS_ECP_DP_BP256R1_ENABLED) || defined(PSA_WANT_ECC_BRAINPOOL_P_R1_256)
Valerio Settideb67642023-07-03 14:26:04 +0200482 { MBEDTLS_SSL_IANA_TLS_GROUP_BP256R1, "brainpoolP256r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200483#else
Valerio Settideb67642023-07-03 14:26:04 +0200484 { MBEDTLS_SSL_IANA_TLS_GROUP_BP256R1, "brainpoolP256r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200485#endif
486#if defined(MBEDTLS_ECP_DP_SECP224R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_224)
Valerio Settideb67642023-07-03 14:26:04 +0200487 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224R1, "secp224r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200488#else
Valerio Settideb67642023-07-03 14:26:04 +0200489 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224R1, "secp224r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200490#endif
491#if defined(MBEDTLS_ECP_DP_SECP224K1_ENABLED) || defined(PSA_WANT_ECC_SECP_K1_224)
Valerio Settideb67642023-07-03 14:26:04 +0200492 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224K1, "secp224k1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200493#else
Valerio Settideb67642023-07-03 14:26:04 +0200494 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224K1, "secp224k1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200495#endif
496#if defined(MBEDTLS_ECP_DP_SECP192R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_192)
Valerio Settideb67642023-07-03 14:26:04 +0200497 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192R1, "secp192r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200498#else
Valerio Settideb67642023-07-03 14:26:04 +0200499 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192R1, "secp192r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200500#endif
501#if defined(MBEDTLS_ECP_DP_SECP192K1_ENABLED) || defined(PSA_WANT_ECC_SECP_K1_192)
Valerio Settideb67642023-07-03 14:26:04 +0200502 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192K1, "secp192k1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200503#else
Valerio Settideb67642023-07-03 14:26:04 +0200504 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192K1, "secp192k1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200505#endif
506#if defined(MBEDTLS_ECP_DP_CURVE25519_ENABLED) || defined(PSA_WANT_ECC_MONTGOMERY_255)
Valerio Settideb67642023-07-03 14:26:04 +0200507 { MBEDTLS_SSL_IANA_TLS_GROUP_X25519, "x25519", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200508#else
Valerio Settideb67642023-07-03 14:26:04 +0200509 { MBEDTLS_SSL_IANA_TLS_GROUP_X25519, "x25519", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200510#endif
511#if defined(MBEDTLS_ECP_DP_CURVE448_ENABLED) || defined(PSA_WANT_ECC_MONTGOMERY_448)
Valerio Settideb67642023-07-03 14:26:04 +0200512 { MBEDTLS_SSL_IANA_TLS_GROUP_X448, "x448", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200513#else
Valerio Settideb67642023-07-03 14:26:04 +0200514 { MBEDTLS_SSL_IANA_TLS_GROUP_X448, "x448", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200515#endif
Valerio Settideb67642023-07-03 14:26:04 +0200516#if defined(MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_SOME_EPHEMERAL_ENABLED) && \
517 defined(PSA_WANT_ALG_FFDH)
518 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE2048, "ffdhe2048", 1 },
519 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE3072, "ffdhe3072", 1 },
520 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE4096, "ffdhe4096", 1 },
521 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE6144, "ffdhe6144", 1 },
522 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE8192, "ffdhe8192", 1 },
523#else
524 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE2048, "ffdhe2048", 0 },
525 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE3072, "ffdhe3072", 0 },
526 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE4096, "ffdhe4096", 0 },
527 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE6144, "ffdhe6144", 0 },
528 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE8192, "ffdhe8192", 0 },
529#endif /* MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_SOME_EPHEMERAL_ENABLED && PSA_WANT_ALG_FFDH */
Valerio Settiacd32c02023-06-29 18:06:29 +0200530 { 0, NULL, 0 },
531};
532
533static uint16_t mbedtls_ssl_get_curve_tls_id_from_name(const char *name)
534{
535 if (name == NULL) {
536 return 0;
537 }
538
Valerio Setti5bdebb22023-07-07 10:52:18 +0200539 for (int i = 0; tls_id_group_name_table[i].tls_id != 0; i++) {
540 if (strcmp(tls_id_group_name_table[i].name, name) == 0) {
541 return tls_id_group_name_table[i].tls_id;
Valerio Settiacd32c02023-06-29 18:06:29 +0200542 }
543 }
544
545 return 0;
546}
547
Valerio Settidda00192023-07-10 10:22:51 +0200548static void mbedtls_ssl_print_supported_groups_list(void)
Valerio Settiacd32c02023-06-29 18:06:29 +0200549{
Valerio Setti5bdebb22023-07-07 10:52:18 +0200550 for (int i = 0; tls_id_group_name_table[i].tls_id != 0; i++) {
551 if (tls_id_group_name_table[i].is_supported == 1) {
552 mbedtls_printf("%s ", tls_id_group_name_table[i].name);
Valerio Settiacd32c02023-06-29 18:06:29 +0200553 }
554 }
Valerio Settiacd32c02023-06-29 18:06:29 +0200555}
556
Valerio Settidda00192023-07-10 10:22:51 +0200557int parse_groups(const char *groups, uint16_t *group_list, size_t group_list_len)
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200558{
Przemek Stekiel45255e42023-06-29 13:56:36 +0200559 char *p = (char *) groups;
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200560 char *q = NULL;
561 size_t i = 0;
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200562
563 if (strcmp(p, "none") == 0) {
564 group_list[0] = 0;
565 } else if (strcmp(p, "default") != 0) {
Przemek Stekiel68e75442023-07-06 11:21:39 +0200566 /* Leave room for a final NULL in group list */
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200567 while (i < group_list_len - 1 && *p != '\0') {
Valerio Settiacd32c02023-06-29 18:06:29 +0200568 uint16_t curve_tls_id;
Valerio Settideb67642023-07-03 14:26:04 +0200569 q = p;
Valerio Settiacd32c02023-06-29 18:06:29 +0200570
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200571 /* Terminate the current string */
572 while (*p != ',' && *p != '\0') {
573 p++;
574 }
575 if (*p == ',') {
576 *p++ = '\0';
577 }
578
Valerio Settiacd32c02023-06-29 18:06:29 +0200579 if ((curve_tls_id = mbedtls_ssl_get_curve_tls_id_from_name(q)) != 0) {
580 group_list[i++] = curve_tls_id;
Valerio Settideb67642023-07-03 14:26:04 +0200581 } else {
Valerio Settidda00192023-07-10 10:22:51 +0200582 mbedtls_printf("unknown group %s\n", q);
583 mbedtls_printf("supported groups: ");
584 mbedtls_ssl_print_supported_groups_list();
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200585 mbedtls_printf("\n");
586 return -1;
587 }
588 }
589
Przemek Stekiel45255e42023-06-29 13:56:36 +0200590 mbedtls_printf("Number of groups: %u\n", (unsigned int) i);
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200591
592 if (i == group_list_len - 1 && *p != '\0') {
Przemek Stekiel45255e42023-06-29 13:56:36 +0200593 mbedtls_printf("groups list too long, maximum %u",
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200594 (unsigned int) (group_list_len - 1));
595 return -1;
596 }
597
598 group_list[i] = 0;
599 }
600
601 return 0;
602}
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200603
Gilles Peskineab7ce962021-01-05 21:27:53 +0100604#endif /* !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE) */