blob: d3ac526f7e0afc4e8e242d3c26158aba8189de8a [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
Gilles Peskinea3ed34f2021-01-05 21:11:16 +010012#include "ssl_test_lib.h"
13
Gilles Peskinee374b952021-02-03 00:05:19 +010014#if defined(MBEDTLS_TEST_HOOKS)
Paul Elliott17c119a2023-12-08 16:55:03 +000015#include "test/threading_helpers.h"
Gilles Peskinee374b952021-02-03 00:05:19 +010016#endif
17
Gilles Peskineab7ce962021-01-05 21:27:53 +010018#if !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE)
19
Valerio Settiacd32c02023-06-29 18:06:29 +020020#define ARRAY_LENGTH(x) (sizeof(x)/sizeof(x[0]))
21
Gilles Peskine449bd832023-01-11 14:50:10 +010022void my_debug(void *ctx, int level,
23 const char *file, int line,
24 const char *str)
Gilles Peskine504c1a32021-01-05 23:40:14 +010025{
26 const char *p, *basename;
27
28 /* Extract basename from file */
Gilles Peskine449bd832023-01-11 14:50:10 +010029 for (p = basename = file; *p != '\0'; p++) {
30 if (*p == '/' || *p == '\\') {
Gilles Peskine504c1a32021-01-05 23:40:14 +010031 basename = p + 1;
Gilles Peskine449bd832023-01-11 14:50:10 +010032 }
33 }
Gilles Peskine504c1a32021-01-05 23:40:14 +010034
Gilles Peskine449bd832023-01-11 14:50:10 +010035 mbedtls_fprintf((FILE *) ctx, "%s:%04d: |%d| %s",
36 basename, line, level, str);
37 fflush((FILE *) ctx);
Gilles Peskine504c1a32021-01-05 23:40:14 +010038}
39
Raoul Strackx9ed9bc92020-06-22 14:08:57 +020040#if defined(MBEDTLS_HAVE_TIME)
Gilles Peskine449bd832023-01-11 14:50:10 +010041mbedtls_time_t dummy_constant_time(mbedtls_time_t *time)
Gilles Peskine504c1a32021-01-05 23:40:14 +010042{
43 (void) time;
44 return 0x5af2a056;
45}
Raoul Strackx9ed9bc92020-06-22 14:08:57 +020046#endif
Gilles Peskine504c1a32021-01-05 23:40:14 +010047
Gilles Peskine8eb29432021-02-03 20:07:11 +010048#if !defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
Gilles Peskine449bd832023-01-11 14:50:10 +010049static int dummy_entropy(void *data, unsigned char *output, size_t len)
Gilles Peskine504c1a32021-01-05 23:40:14 +010050{
51 size_t i;
52 int ret;
53 (void) data;
54
Gilles Peskine449bd832023-01-11 14:50:10 +010055 ret = mbedtls_entropy_func(data, output, len);
56 for (i = 0; i < len; i++) {
Gilles Peskine504c1a32021-01-05 23:40:14 +010057 //replace result with pseudo random
58 output[i] = (unsigned char) rand();
59 }
Gilles Peskine449bd832023-01-11 14:50:10 +010060 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +010061}
Gilles Peskine8eb29432021-02-03 20:07:11 +010062#endif
Gilles Peskine504c1a32021-01-05 23:40:14 +010063
Gilles Peskine449bd832023-01-11 14:50:10 +010064void rng_init(rng_context_t *rng)
Gilles Peskinedaa94c42021-01-13 18:38:27 +010065{
Gilles Peskine8eb29432021-02-03 20:07:11 +010066#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
67 (void) rng;
Gilles Peskine449bd832023-01-11 14:50:10 +010068 psa_crypto_init();
Gilles Peskine8eb29432021-02-03 20:07:11 +010069#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
70
Gilles Peskineba749042021-01-13 20:02:03 +010071#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +010072 mbedtls_ctr_drbg_init(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +010073#elif defined(MBEDTLS_HMAC_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +010074 mbedtls_hmac_drbg_init(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +010075#else
76#error "No DRBG available"
77#endif
78
Gilles Peskine449bd832023-01-11 14:50:10 +010079 mbedtls_entropy_init(&rng->entropy);
Gilles Peskine8eb29432021-02-03 20:07:11 +010080#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskinedaa94c42021-01-13 18:38:27 +010081}
82
Gilles Peskine449bd832023-01-11 14:50:10 +010083int rng_seed(rng_context_t *rng, int reproducible, const char *pers)
Gilles Peskinedaa94c42021-01-13 18:38:27 +010084{
Gilles Peskineaaedbdc2021-02-03 13:55:22 +010085#if defined(MBEDTLS_USE_PSA_CRYPTO)
Gilles Peskine449bd832023-01-11 14:50:10 +010086 if (reproducible) {
87 mbedtls_fprintf(stderr,
88 "MBEDTLS_USE_PSA_CRYPTO does not support reproducible mode.\n");
89 return -1;
Gilles Peskineaaedbdc2021-02-03 13:55:22 +010090 }
91#endif
Gilles Peskine8eb29432021-02-03 20:07:11 +010092#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
93 /* The PSA crypto RNG does its own seeding. */
94 (void) rng;
95 (void) pers;
Gilles Peskine449bd832023-01-11 14:50:10 +010096 if (reproducible) {
97 mbedtls_fprintf(stderr,
98 "The PSA RNG does not support reproducible mode.\n");
99 return -1;
Gilles Peskine8eb29432021-02-03 20:07:11 +0100100 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100101 return 0;
Gilles Peskine8eb29432021-02-03 20:07:11 +0100102#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskine449bd832023-01-11 14:50:10 +0100103 int (*f_entropy)(void *, unsigned char *, size_t) =
104 (reproducible ? dummy_entropy : mbedtls_entropy_func);
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100105
Gilles Peskine449bd832023-01-11 14:50:10 +0100106 if (reproducible) {
107 srand(1);
108 }
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100109
Gilles Peskineba749042021-01-13 20:02:03 +0100110#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100111 int ret = mbedtls_ctr_drbg_seed(&rng->drbg,
112 f_entropy, &rng->entropy,
113 (const unsigned char *) pers,
114 strlen(pers));
Gilles Peskineba749042021-01-13 20:02:03 +0100115#elif defined(MBEDTLS_HMAC_DRBG_C)
Manuel Pégourié-Gonnardbef824d2023-03-17 12:50:01 +0100116#if defined(MBEDTLS_MD_CAN_SHA256)
Gilles Peskineba749042021-01-13 20:02:03 +0100117 const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA256;
Manuel Pégourié-Gonnardbef824d2023-03-17 12:50:01 +0100118#elif defined(MBEDTLS_MD_CAN_SHA512)
Gilles Peskineba749042021-01-13 20:02:03 +0100119 const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA512;
120#else
121#error "No message digest available for HMAC_DRBG"
122#endif
Gilles Peskine449bd832023-01-11 14:50:10 +0100123 int ret = mbedtls_hmac_drbg_seed(&rng->drbg,
124 mbedtls_md_info_from_type(md_type),
125 f_entropy, &rng->entropy,
126 (const unsigned char *) pers,
127 strlen(pers));
Gilles Peskine8eb29432021-02-03 20:07:11 +0100128#else /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */
Gilles Peskineba749042021-01-13 20:02:03 +0100129#error "No DRBG available"
Gilles Peskine8eb29432021-02-03 20:07:11 +0100130#endif /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */
Gilles Peskineba749042021-01-13 20:02:03 +0100131
Gilles Peskine449bd832023-01-11 14:50:10 +0100132 if (ret != 0) {
133 mbedtls_printf(" failed\n ! mbedtls_ctr_drbg_seed returned -0x%x\n",
134 (unsigned int) -ret);
135 return ret;
Gilles Peskinef1cb75f2021-01-13 18:46:01 +0100136 }
Gilles Peskine8eb29432021-02-03 20:07:11 +0100137#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100138
Gilles Peskine449bd832023-01-11 14:50:10 +0100139 return 0;
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100140}
141
Gilles Peskine449bd832023-01-11 14:50:10 +0100142void rng_free(rng_context_t *rng)
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100143{
Gilles Peskine8eb29432021-02-03 20:07:11 +0100144#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
145 (void) rng;
146 /* Deinitialize the PSA crypto subsystem. This deactivates all PSA APIs.
147 * This is ok because none of our applications try to do any crypto after
148 * deinitializing the RNG. */
Gilles Peskine449bd832023-01-11 14:50:10 +0100149 mbedtls_psa_crypto_free();
Gilles Peskine8eb29432021-02-03 20:07:11 +0100150#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
151
Gilles Peskineba749042021-01-13 20:02:03 +0100152#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100153 mbedtls_ctr_drbg_free(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +0100154#elif defined(MBEDTLS_HMAC_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100155 mbedtls_hmac_drbg_free(&rng->drbg);
Gilles Peskineba749042021-01-13 20:02:03 +0100156#else
157#error "No DRBG available"
158#endif
159
Gilles Peskine449bd832023-01-11 14:50:10 +0100160 mbedtls_entropy_free(&rng->entropy);
Gilles Peskine8eb29432021-02-03 20:07:11 +0100161#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskinedaa94c42021-01-13 18:38:27 +0100162}
163
Gilles Peskine449bd832023-01-11 14:50:10 +0100164int rng_get(void *p_rng, unsigned char *output, size_t output_len)
Gilles Peskine535fb372021-01-13 18:59:46 +0100165{
Gilles Peskine8eb29432021-02-03 20:07:11 +0100166#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
167 (void) p_rng;
Gilles Peskine449bd832023-01-11 14:50:10 +0100168 return mbedtls_psa_get_random(MBEDTLS_PSA_RANDOM_STATE,
169 output, output_len);
Gilles Peskine8eb29432021-02-03 20:07:11 +0100170#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskine535fb372021-01-13 18:59:46 +0100171 rng_context_t *rng = p_rng;
Gilles Peskine8eb29432021-02-03 20:07:11 +0100172
Gilles Peskineba749042021-01-13 20:02:03 +0100173#if defined(MBEDTLS_CTR_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100174 return mbedtls_ctr_drbg_random(&rng->drbg, output, output_len);
Gilles Peskineba749042021-01-13 20:02:03 +0100175#elif defined(MBEDTLS_HMAC_DRBG_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100176 return mbedtls_hmac_drbg_random(&rng->drbg, output, output_len);
Gilles Peskineba749042021-01-13 20:02:03 +0100177#else
178#error "No DRBG available"
179#endif
Gilles Peskine8eb29432021-02-03 20:07:11 +0100180
181#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
Gilles Peskine535fb372021-01-13 18:59:46 +0100182}
183
Gilles Peskine449bd832023-01-11 14:50:10 +0100184int key_opaque_alg_parse(const char *arg, const char **alg1, const char **alg2)
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200185{
Gilles Peskine449bd832023-01-11 14:50:10 +0100186 char *separator;
187 if ((separator = strchr(arg, ',')) == NULL) {
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200188 return 1;
Gilles Peskine449bd832023-01-11 14:50:10 +0100189 }
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200190 *separator = '\0';
191
192 *alg1 = arg;
193 *alg2 = separator + 1;
194
Gilles Peskine449bd832023-01-11 14:50:10 +0100195 if (strcmp(*alg1, "rsa-sign-pkcs1") != 0 &&
196 strcmp(*alg1, "rsa-sign-pss") != 0 &&
197 strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
198 strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
199 strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
200 strcmp(*alg1, "rsa-decrypt") != 0 &&
201 strcmp(*alg1, "ecdsa-sign") != 0 &&
202 strcmp(*alg1, "ecdh") != 0) {
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200203 return 1;
Gilles Peskine449bd832023-01-11 14:50:10 +0100204 }
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200205
Gilles Peskine449bd832023-01-11 14:50:10 +0100206 if (strcmp(*alg2, "rsa-sign-pkcs1") != 0 &&
207 strcmp(*alg2, "rsa-sign-pss") != 0 &&
208 strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
209 strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
210 strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
211 strcmp(*alg2, "rsa-decrypt") != 0 &&
212 strcmp(*alg2, "ecdsa-sign") != 0 &&
213 strcmp(*alg2, "ecdh") != 0 &&
214 strcmp(*alg2, "none") != 0) {
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200215 return 1;
Gilles Peskine449bd832023-01-11 14:50:10 +0100216 }
Przemek Stekiel85d692d2022-04-25 12:42:55 +0200217
218 return 0;
219}
220
Przemek Stekiel76a41f52022-05-04 13:55:23 +0200221#if defined(MBEDTLS_USE_PSA_CRYPTO)
Gilles Peskine449bd832023-01-11 14:50:10 +0100222int key_opaque_set_alg_usage(const char *alg1, const char *alg2,
223 psa_algorithm_t *psa_alg1,
224 psa_algorithm_t *psa_alg2,
225 psa_key_usage_t *usage,
226 mbedtls_pk_type_t key_type)
Przemek Stekiel01396a12022-05-02 13:41:53 +0200227{
Gilles Peskine449bd832023-01-11 14:50:10 +0100228 if (strcmp(alg1, "none") != 0) {
229 const char *algs[] = { alg1, alg2 };
Przemek Stekielcb20d202022-05-06 08:42:34 +0200230 psa_algorithm_t *psa_algs[] = { psa_alg1, psa_alg2 };
Przemek Stekiel01396a12022-05-02 13:41:53 +0200231
Gilles Peskine449bd832023-01-11 14:50:10 +0100232 for (int i = 0; i < 2; i++) {
233 if (strcmp(algs[i], "rsa-sign-pkcs1") == 0) {
234 *psa_algs[i] = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200235 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100236 } else if (strcmp(algs[i], "rsa-sign-pss") == 0) {
237 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200238 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100239 } else if (strcmp(algs[i], "rsa-sign-pss-sha256") == 0) {
240 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_256);
Ronald Cron50969e32022-09-16 15:54:33 +0200241 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100242 } else if (strcmp(algs[i], "rsa-sign-pss-sha384") == 0) {
243 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_384);
Ronald Cron50969e32022-09-16 15:54:33 +0200244 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100245 } else if (strcmp(algs[i], "rsa-sign-pss-sha512") == 0) {
246 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_512);
Ronald Cron50969e32022-09-16 15:54:33 +0200247 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100248 } else if (strcmp(algs[i], "rsa-decrypt") == 0) {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200249 *psa_algs[i] = PSA_ALG_RSA_PKCS1V15_CRYPT;
250 *usage |= PSA_KEY_USAGE_DECRYPT;
Gilles Peskine449bd832023-01-11 14:50:10 +0100251 } else if (strcmp(algs[i], "ecdsa-sign") == 0) {
252 *psa_algs[i] = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200253 *usage |= PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100254 } else if (strcmp(algs[i], "ecdh") == 0) {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200255 *psa_algs[i] = PSA_ALG_ECDH;
256 *usage |= PSA_KEY_USAGE_DERIVE;
Gilles Peskine449bd832023-01-11 14:50:10 +0100257 } else if (strcmp(algs[i], "none") == 0) {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200258 *psa_algs[i] = PSA_ALG_NONE;
259 }
260 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100261 } else {
262 if (key_type == MBEDTLS_PK_ECKEY) {
263 *psa_alg1 = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200264 *psa_alg2 = PSA_ALG_ECDH;
265 *usage = PSA_KEY_USAGE_SIGN_HASH | PSA_KEY_USAGE_DERIVE;
Gilles Peskine449bd832023-01-11 14:50:10 +0100266 } else if (key_type == MBEDTLS_PK_RSA) {
267 *psa_alg1 = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
268 *psa_alg2 = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
Przemek Stekielcb20d202022-05-06 08:42:34 +0200269 *usage = PSA_KEY_USAGE_SIGN_HASH;
Gilles Peskine449bd832023-01-11 14:50:10 +0100270 } else {
Przemek Stekielcb20d202022-05-06 08:42:34 +0200271 return 1;
272 }
Przemek Stekiel01396a12022-05-02 13:41:53 +0200273 }
274
275 return 0;
276}
Przemek Stekiel76a41f52022-05-04 13:55:23 +0200277#endif /* MBEDTLS_USE_PSA_CRYPTO */
Przemek Stekiel01396a12022-05-02 13:41:53 +0200278
Gilles Peskine504c1a32021-01-05 23:40:14 +0100279#if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
Gilles Peskine449bd832023-01-11 14:50:10 +0100280int ca_callback(void *data, mbedtls_x509_crt const *child,
281 mbedtls_x509_crt **candidates)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100282{
283 int ret = 0;
284 mbedtls_x509_crt *ca = (mbedtls_x509_crt *) data;
285 mbedtls_x509_crt *first;
286
287 /* This is a test-only implementation of the CA callback
288 * which always returns the entire list of trusted certificates.
289 * Production implementations managing a large number of CAs
290 * should use an efficient presentation and lookup for the
291 * set of trusted certificates (such as a hashtable) and only
292 * return those trusted certificates which satisfy basic
293 * parental checks, such as the matching of child `Issuer`
294 * and parent `Subject` field or matching key identifiers. */
295 ((void) child);
296
Gilles Peskine449bd832023-01-11 14:50:10 +0100297 first = mbedtls_calloc(1, sizeof(mbedtls_x509_crt));
298 if (first == NULL) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100299 ret = -1;
300 goto exit;
301 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100302 mbedtls_x509_crt_init(first);
Gilles Peskine504c1a32021-01-05 23:40:14 +0100303
Gilles Peskine449bd832023-01-11 14:50:10 +0100304 if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100305 ret = -1;
306 goto exit;
307 }
308
Gilles Peskine449bd832023-01-11 14:50:10 +0100309 while (ca->next != NULL) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100310 ca = ca->next;
Gilles Peskine449bd832023-01-11 14:50:10 +0100311 if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100312 ret = -1;
313 goto exit;
314 }
315 }
316
317exit:
318
Gilles Peskine449bd832023-01-11 14:50:10 +0100319 if (ret != 0) {
320 mbedtls_x509_crt_free(first);
321 mbedtls_free(first);
Gilles Peskine504c1a32021-01-05 23:40:14 +0100322 first = NULL;
323 }
324
325 *candidates = first;
Gilles Peskine449bd832023-01-11 14:50:10 +0100326 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100327}
328#endif /* MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK */
329
Gilles Peskine449bd832023-01-11 14:50:10 +0100330int delayed_recv(void *ctx, unsigned char *buf, size_t len)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100331{
332 static int first_try = 1;
333 int ret;
334
Gilles Peskine449bd832023-01-11 14:50:10 +0100335 if (first_try) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100336 first_try = 0;
Gilles Peskine449bd832023-01-11 14:50:10 +0100337 return MBEDTLS_ERR_SSL_WANT_READ;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100338 }
339
Gilles Peskine449bd832023-01-11 14:50:10 +0100340 ret = mbedtls_net_recv(ctx, buf, len);
341 if (ret != MBEDTLS_ERR_SSL_WANT_READ) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100342 first_try = 1; /* Next call will be a new operation */
Gilles Peskine449bd832023-01-11 14:50:10 +0100343 }
344 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100345}
346
Gilles Peskine449bd832023-01-11 14:50:10 +0100347int delayed_send(void *ctx, const unsigned char *buf, size_t len)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100348{
349 static int first_try = 1;
350 int ret;
351
Gilles Peskine449bd832023-01-11 14:50:10 +0100352 if (first_try) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100353 first_try = 0;
Gilles Peskine449bd832023-01-11 14:50:10 +0100354 return MBEDTLS_ERR_SSL_WANT_WRITE;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100355 }
356
Gilles Peskine449bd832023-01-11 14:50:10 +0100357 ret = mbedtls_net_send(ctx, buf, len);
358 if (ret != MBEDTLS_ERR_SSL_WANT_WRITE) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100359 first_try = 1; /* Next call will be a new operation */
Gilles Peskine449bd832023-01-11 14:50:10 +0100360 }
361 return ret;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100362}
363
364#if !defined(MBEDTLS_TIMING_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100365int idle(mbedtls_net_context *fd,
366 int idle_reason)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100367#else
Gilles Peskine449bd832023-01-11 14:50:10 +0100368int idle(mbedtls_net_context *fd,
369 mbedtls_timing_delay_context *timer,
370 int idle_reason)
Gilles Peskine504c1a32021-01-05 23:40:14 +0100371#endif
372{
373 int ret;
374 int poll_type = 0;
375
Gilles Peskine449bd832023-01-11 14:50:10 +0100376 if (idle_reason == MBEDTLS_ERR_SSL_WANT_WRITE) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100377 poll_type = MBEDTLS_NET_POLL_WRITE;
Gilles Peskine449bd832023-01-11 14:50:10 +0100378 } else if (idle_reason == MBEDTLS_ERR_SSL_WANT_READ) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100379 poll_type = MBEDTLS_NET_POLL_READ;
Gilles Peskine449bd832023-01-11 14:50:10 +0100380 }
Gilles Peskine504c1a32021-01-05 23:40:14 +0100381#if !defined(MBEDTLS_TIMING_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100382 else {
383 return 0;
384 }
Gilles Peskine504c1a32021-01-05 23:40:14 +0100385#endif
386
Gilles Peskine449bd832023-01-11 14:50:10 +0100387 while (1) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100388 /* Check if timer has expired */
389#if defined(MBEDTLS_TIMING_C)
Gilles Peskine449bd832023-01-11 14:50:10 +0100390 if (timer != NULL &&
391 mbedtls_timing_get_delay(timer) == 2) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100392 break;
393 }
394#endif /* MBEDTLS_TIMING_C */
395
396 /* Check if underlying transport became available */
Gilles Peskine449bd832023-01-11 14:50:10 +0100397 if (poll_type != 0) {
398 ret = mbedtls_net_poll(fd, poll_type, 0);
399 if (ret < 0) {
400 return ret;
401 }
402 if (ret == poll_type) {
Gilles Peskine504c1a32021-01-05 23:40:14 +0100403 break;
Gilles Peskine449bd832023-01-11 14:50:10 +0100404 }
Gilles Peskine504c1a32021-01-05 23:40:14 +0100405 }
406 }
407
Gilles Peskine449bd832023-01-11 14:50:10 +0100408 return 0;
Gilles Peskine504c1a32021-01-05 23:40:14 +0100409}
410
Gilles Peskine53dea742021-02-02 22:55:06 +0100411#if defined(MBEDTLS_TEST_HOOKS)
412
Gilles Peskine449bd832023-01-11 14:50:10 +0100413void test_hooks_init(void)
Gilles Peskine53dea742021-02-02 22:55:06 +0100414{
Gilles Peskine449bd832023-01-11 14:50:10 +0100415 mbedtls_test_info_reset();
Gilles Peskinee374b952021-02-03 00:05:19 +0100416
417#if defined(MBEDTLS_TEST_MUTEX_USAGE)
Gilles Peskine449bd832023-01-11 14:50:10 +0100418 mbedtls_test_mutex_usage_init();
Gilles Peskinee374b952021-02-03 00:05:19 +0100419#endif
Gilles Peskine53dea742021-02-02 22:55:06 +0100420}
421
Gilles Peskine449bd832023-01-11 14:50:10 +0100422int test_hooks_failure_detected(void)
Gilles Peskine53dea742021-02-02 22:55:06 +0100423{
Gilles Peskinee374b952021-02-03 00:05:19 +0100424#if defined(MBEDTLS_TEST_MUTEX_USAGE)
425 /* Errors are reported via mbedtls_test_info. */
Gilles Peskine449bd832023-01-11 14:50:10 +0100426 mbedtls_test_mutex_usage_check();
Gilles Peskinee374b952021-02-03 00:05:19 +0100427#endif
428
Paul Elliott4580d4d2023-10-27 18:41:02 +0100429 if (mbedtls_test_get_result() != MBEDTLS_TEST_RESULT_SUCCESS) {
Gilles Peskine449bd832023-01-11 14:50:10 +0100430 return 1;
431 }
432 return 0;
Gilles Peskine53dea742021-02-02 22:55:06 +0100433}
434
Gilles Peskine449bd832023-01-11 14:50:10 +0100435void test_hooks_free(void)
Gilles Peskine53dea742021-02-02 22:55:06 +0100436{
Paul Elliottf25d8312023-11-23 18:49:43 +0000437#if defined(MBEDTLS_TEST_MUTEX_USAGE)
438 mbedtls_test_mutex_usage_end();
439#endif
Gilles Peskine53dea742021-02-02 22:55:06 +0100440}
441
442#endif /* MBEDTLS_TEST_HOOKS */
443
Valerio Settiacd32c02023-06-29 18:06:29 +0200444static const struct {
445 uint16_t tls_id;
446 const char *name;
447 uint8_t is_supported;
Valerio Setti5bdebb22023-07-07 10:52:18 +0200448} tls_id_group_name_table[] =
Valerio Settiacd32c02023-06-29 18:06:29 +0200449{
450#if defined(MBEDTLS_ECP_DP_SECP521R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_521)
Valerio Settideb67642023-07-03 14:26:04 +0200451 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP521R1, "secp521r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200452#else
Valerio Settideb67642023-07-03 14:26:04 +0200453 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP521R1, "secp521r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200454#endif
455#if defined(MBEDTLS_ECP_DP_BP512R1_ENABLED) || defined(PSA_WANT_ECC_BRAINPOOL_P_R1_512)
Valerio Settideb67642023-07-03 14:26:04 +0200456 { MBEDTLS_SSL_IANA_TLS_GROUP_BP512R1, "brainpoolP512r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200457#else
Valerio Settideb67642023-07-03 14:26:04 +0200458 { MBEDTLS_SSL_IANA_TLS_GROUP_BP512R1, "brainpoolP512r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200459#endif
460#if defined(MBEDTLS_ECP_DP_SECP384R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_384)
Valerio Settideb67642023-07-03 14:26:04 +0200461 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP384R1, "secp384r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200462#else
Valerio Settideb67642023-07-03 14:26:04 +0200463 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP384R1, "secp384r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200464#endif
465#if defined(MBEDTLS_ECP_DP_BP384R1_ENABLED) || defined(PSA_WANT_ECC_BRAINPOOL_P_R1_384)
Valerio Settideb67642023-07-03 14:26:04 +0200466 { MBEDTLS_SSL_IANA_TLS_GROUP_BP384R1, "brainpoolP384r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200467#else
Valerio Settideb67642023-07-03 14:26:04 +0200468 { MBEDTLS_SSL_IANA_TLS_GROUP_BP384R1, "brainpoolP384r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200469#endif
470#if defined(MBEDTLS_ECP_DP_SECP256R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_256)
Valerio Settideb67642023-07-03 14:26:04 +0200471 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256R1, "secp256r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200472#else
Valerio Settideb67642023-07-03 14:26:04 +0200473 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256R1, "secp256r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200474#endif
475#if defined(MBEDTLS_ECP_DP_SECP256K1_ENABLED) || defined(PSA_WANT_ECC_SECP_K1_256)
Valerio Settideb67642023-07-03 14:26:04 +0200476 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256K1, "secp256k1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200477#else
Valerio Settideb67642023-07-03 14:26:04 +0200478 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP256K1, "secp256k1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200479#endif
480#if defined(MBEDTLS_ECP_DP_BP256R1_ENABLED) || defined(PSA_WANT_ECC_BRAINPOOL_P_R1_256)
Valerio Settideb67642023-07-03 14:26:04 +0200481 { MBEDTLS_SSL_IANA_TLS_GROUP_BP256R1, "brainpoolP256r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200482#else
Valerio Settideb67642023-07-03 14:26:04 +0200483 { MBEDTLS_SSL_IANA_TLS_GROUP_BP256R1, "brainpoolP256r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200484#endif
485#if defined(MBEDTLS_ECP_DP_SECP224R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_224)
Valerio Settideb67642023-07-03 14:26:04 +0200486 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224R1, "secp224r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200487#else
Valerio Settideb67642023-07-03 14:26:04 +0200488 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224R1, "secp224r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200489#endif
490#if defined(MBEDTLS_ECP_DP_SECP224K1_ENABLED) || defined(PSA_WANT_ECC_SECP_K1_224)
Valerio Settideb67642023-07-03 14:26:04 +0200491 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224K1, "secp224k1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200492#else
Valerio Settideb67642023-07-03 14:26:04 +0200493 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP224K1, "secp224k1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200494#endif
495#if defined(MBEDTLS_ECP_DP_SECP192R1_ENABLED) || defined(PSA_WANT_ECC_SECP_R1_192)
Valerio Settideb67642023-07-03 14:26:04 +0200496 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192R1, "secp192r1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200497#else
Valerio Settideb67642023-07-03 14:26:04 +0200498 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192R1, "secp192r1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200499#endif
500#if defined(MBEDTLS_ECP_DP_SECP192K1_ENABLED) || defined(PSA_WANT_ECC_SECP_K1_192)
Valerio Settideb67642023-07-03 14:26:04 +0200501 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192K1, "secp192k1", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200502#else
Valerio Settideb67642023-07-03 14:26:04 +0200503 { MBEDTLS_SSL_IANA_TLS_GROUP_SECP192K1, "secp192k1", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200504#endif
505#if defined(MBEDTLS_ECP_DP_CURVE25519_ENABLED) || defined(PSA_WANT_ECC_MONTGOMERY_255)
Valerio Settideb67642023-07-03 14:26:04 +0200506 { MBEDTLS_SSL_IANA_TLS_GROUP_X25519, "x25519", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200507#else
Valerio Settideb67642023-07-03 14:26:04 +0200508 { MBEDTLS_SSL_IANA_TLS_GROUP_X25519, "x25519", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200509#endif
510#if defined(MBEDTLS_ECP_DP_CURVE448_ENABLED) || defined(PSA_WANT_ECC_MONTGOMERY_448)
Valerio Settideb67642023-07-03 14:26:04 +0200511 { MBEDTLS_SSL_IANA_TLS_GROUP_X448, "x448", 1 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200512#else
Valerio Settideb67642023-07-03 14:26:04 +0200513 { MBEDTLS_SSL_IANA_TLS_GROUP_X448, "x448", 0 },
Valerio Settiacd32c02023-06-29 18:06:29 +0200514#endif
Valerio Settideb67642023-07-03 14:26:04 +0200515#if defined(MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_SOME_EPHEMERAL_ENABLED) && \
516 defined(PSA_WANT_ALG_FFDH)
Valerio Settie8683ce2024-01-17 12:34:06 +0100517#if defined(PSA_WANT_DH_RFC7919_2048)
Valerio Settideb67642023-07-03 14:26:04 +0200518 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE2048, "ffdhe2048", 1 },
Valerio Settie8683ce2024-01-17 12:34:06 +0100519#else /* PSA_WANT_DH_RFC7919_2048 */
Valerio Settideb67642023-07-03 14:26:04 +0200520 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE2048, "ffdhe2048", 0 },
Valerio Settie8683ce2024-01-17 12:34:06 +0100521#endif /* PSA_WANT_DH_RFC7919_2048 */
522#if defined(PSA_WANT_DH_RFC7919_3072)
523 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE3072, "ffdhe3072", 1 },
524#else /* PSA_WANT_DH_RFC7919_3072 */
Valerio Settideb67642023-07-03 14:26:04 +0200525 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE3072, "ffdhe3072", 0 },
Valerio Settie8683ce2024-01-17 12:34:06 +0100526#endif /* PSA_WANT_DH_RFC7919_3072 */
527#if defined(PSA_WANT_DH_RFC7919_4096)
528 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE4096, "ffdhe4096", 1 },
529#else /* PSA_WANT_DH_RFC7919_4096 */
Valerio Settideb67642023-07-03 14:26:04 +0200530 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE4096, "ffdhe4096", 0 },
Valerio Settie8683ce2024-01-17 12:34:06 +0100531#endif /* PSA_WANT_DH_RFC7919_4096 */
532#if defined(PSA_WANT_DH_RFC7919_6144)
533 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE6144, "ffdhe6144", 1 },
534#else /* PSA_WANT_DH_RFC7919_6144 */
Valerio Settideb67642023-07-03 14:26:04 +0200535 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE6144, "ffdhe6144", 0 },
Valerio Settie8683ce2024-01-17 12:34:06 +0100536#endif /* PSA_WANT_DH_RFC7919_6144 */
537#if defined(PSA_WANT_DH_RFC7919_8192)
538 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE8192, "ffdhe8192", 1 },
539#else /* PSA_WANT_DH_RFC7919_8192 */
Valerio Settideb67642023-07-03 14:26:04 +0200540 { MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE8192, "ffdhe8192", 0 },
Valerio Settie8683ce2024-01-17 12:34:06 +0100541#endif /* PSA_WANT_DH_RFC7919_8192 */
Valerio Settideb67642023-07-03 14:26:04 +0200542#endif /* MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_SOME_EPHEMERAL_ENABLED && PSA_WANT_ALG_FFDH */
Valerio Settiacd32c02023-06-29 18:06:29 +0200543 { 0, NULL, 0 },
544};
545
546static uint16_t mbedtls_ssl_get_curve_tls_id_from_name(const char *name)
547{
548 if (name == NULL) {
549 return 0;
550 }
551
Valerio Setti5bdebb22023-07-07 10:52:18 +0200552 for (int i = 0; tls_id_group_name_table[i].tls_id != 0; i++) {
553 if (strcmp(tls_id_group_name_table[i].name, name) == 0) {
554 return tls_id_group_name_table[i].tls_id;
Valerio Settiacd32c02023-06-29 18:06:29 +0200555 }
556 }
557
558 return 0;
559}
560
Valerio Settidda00192023-07-10 10:22:51 +0200561static void mbedtls_ssl_print_supported_groups_list(void)
Valerio Settiacd32c02023-06-29 18:06:29 +0200562{
Valerio Setti5bdebb22023-07-07 10:52:18 +0200563 for (int i = 0; tls_id_group_name_table[i].tls_id != 0; i++) {
564 if (tls_id_group_name_table[i].is_supported == 1) {
565 mbedtls_printf("%s ", tls_id_group_name_table[i].name);
Valerio Settiacd32c02023-06-29 18:06:29 +0200566 }
567 }
Valerio Settiacd32c02023-06-29 18:06:29 +0200568}
569
Valerio Settidda00192023-07-10 10:22:51 +0200570int parse_groups(const char *groups, uint16_t *group_list, size_t group_list_len)
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200571{
Przemek Stekiel45255e42023-06-29 13:56:36 +0200572 char *p = (char *) groups;
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200573 char *q = NULL;
574 size_t i = 0;
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200575
576 if (strcmp(p, "none") == 0) {
577 group_list[0] = 0;
578 } else if (strcmp(p, "default") != 0) {
Przemek Stekiel68e75442023-07-06 11:21:39 +0200579 /* Leave room for a final NULL in group list */
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200580 while (i < group_list_len - 1 && *p != '\0') {
Valerio Settiacd32c02023-06-29 18:06:29 +0200581 uint16_t curve_tls_id;
Valerio Settideb67642023-07-03 14:26:04 +0200582 q = p;
Valerio Settiacd32c02023-06-29 18:06:29 +0200583
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200584 /* Terminate the current string */
585 while (*p != ',' && *p != '\0') {
586 p++;
587 }
588 if (*p == ',') {
589 *p++ = '\0';
590 }
591
Valerio Settiacd32c02023-06-29 18:06:29 +0200592 if ((curve_tls_id = mbedtls_ssl_get_curve_tls_id_from_name(q)) != 0) {
593 group_list[i++] = curve_tls_id;
Valerio Settideb67642023-07-03 14:26:04 +0200594 } else {
Valerio Settidda00192023-07-10 10:22:51 +0200595 mbedtls_printf("unknown group %s\n", q);
596 mbedtls_printf("supported groups: ");
597 mbedtls_ssl_print_supported_groups_list();
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200598 mbedtls_printf("\n");
599 return -1;
600 }
601 }
602
Przemek Stekiel45255e42023-06-29 13:56:36 +0200603 mbedtls_printf("Number of groups: %u\n", (unsigned int) i);
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200604
605 if (i == group_list_len - 1 && *p != '\0') {
Przemek Stekiel45255e42023-06-29 13:56:36 +0200606 mbedtls_printf("groups list too long, maximum %u",
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200607 (unsigned int) (group_list_len - 1));
608 return -1;
609 }
610
611 group_list[i] = 0;
612 }
613
614 return 0;
615}
Przemek Stekielff9fcbc2023-06-05 12:32:55 +0200616
Gilles Peskineab7ce962021-01-05 21:27:53 +0100617#endif /* !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE) */