| Manuel Pégourié-Gonnard | 0bad5c2 | 2013-01-26 15:30:46 +0100 | [diff] [blame] | 1 | /* | 
|  | 2 | *  Elliptic curve Diffie-Hellman | 
|  | 3 | * | 
| Bence Szépkúti | 1e14827 | 2020-08-07 13:07:28 +0200 | [diff] [blame] | 4 | *  Copyright The Mbed TLS Contributors | 
| Dave Rodgman | 16799db | 2023-11-02 19:47:20 +0000 | [diff] [blame] | 5 | *  SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later | 
| Manuel Pégourié-Gonnard | 0bad5c2 | 2013-01-26 15:30:46 +0100 | [diff] [blame] | 6 | */ | 
|  | 7 |  | 
|  | 8 | /* | 
|  | 9 | * References: | 
|  | 10 | * | 
| Xiaokang Qian | 4704147 | 2023-04-12 06:07:23 +0000 | [diff] [blame] | 11 | * SEC1 https://www.secg.org/sec1-v2.pdf | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 12 | * RFC 4492 | 
| Manuel Pégourié-Gonnard | 0bad5c2 | 2013-01-26 15:30:46 +0100 | [diff] [blame] | 13 | */ | 
|  | 14 |  | 
| Gilles Peskine | db09ef6 | 2020-06-03 01:43:33 +0200 | [diff] [blame] | 15 | #include "common.h" | 
| Manuel Pégourié-Gonnard | 0bad5c2 | 2013-01-26 15:30:46 +0100 | [diff] [blame] | 16 |  | 
| Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 17 | #if defined(MBEDTLS_ECDH_C) | 
| Manuel Pégourié-Gonnard | 0bad5c2 | 2013-01-26 15:30:46 +0100 | [diff] [blame] | 18 |  | 
| Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 19 | #include "mbedtls/ecdh.h" | 
| Hanno Becker | 91796d7 | 2018-12-17 18:10:51 +0000 | [diff] [blame] | 20 | #include "mbedtls/platform_util.h" | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 21 | #include "mbedtls/error.h" | 
| Jerry Yu | bdc7188 | 2021-09-14 19:30:36 +0800 | [diff] [blame] | 22 |  | 
| Rich Evans | 00ab470 | 2015-02-06 13:43:58 +0000 | [diff] [blame] | 23 | #include <string.h> | 
|  | 24 |  | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 25 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
|  | 26 | typedef mbedtls_ecdh_context mbedtls_ecdh_context_mbed; | 
|  | 27 | #endif | 
|  | 28 |  | 
| Gilles Peskine | 3081629 | 2019-02-22 12:31:25 +0100 | [diff] [blame] | 29 | static mbedtls_ecp_group_id mbedtls_ecdh_grp_id( | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 30 | const mbedtls_ecdh_context *ctx) | 
| Gilles Peskine | 3081629 | 2019-02-22 12:31:25 +0100 | [diff] [blame] | 31 | { | 
|  | 32 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 33 | return ctx->grp.id; | 
| Gilles Peskine | 3081629 | 2019-02-22 12:31:25 +0100 | [diff] [blame] | 34 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 35 | return ctx->grp_id; | 
| Gilles Peskine | 3081629 | 2019-02-22 12:31:25 +0100 | [diff] [blame] | 36 | #endif | 
|  | 37 | } | 
|  | 38 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 39 | int mbedtls_ecdh_can_do(mbedtls_ecp_group_id gid) | 
| Gilles Peskine | 20b3ef3 | 2019-02-11 18:41:27 +0100 | [diff] [blame] | 40 | { | 
|  | 41 | /* At this time, all groups support ECDH. */ | 
|  | 42 | (void) gid; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 43 | return 1; | 
| Gilles Peskine | 20b3ef3 | 2019-02-11 18:41:27 +0100 | [diff] [blame] | 44 | } | 
|  | 45 |  | 
| Ron Eldor | a84c1cb | 2017-10-10 19:04:27 +0300 | [diff] [blame] | 46 | #if !defined(MBEDTLS_ECDH_GEN_PUBLIC_ALT) | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 47 | /* | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 48 | * Generate public key (restartable version) | 
| Manuel Pégourié-Gonnard | c0edc96 | 2018-10-16 10:38:19 +0200 | [diff] [blame] | 49 | * | 
|  | 50 | * Note: this internal function relies on its caller preserving the value of | 
| Manuel Pégourié-Gonnard | ca29fdf | 2018-10-22 09:56:53 +0200 | [diff] [blame] | 51 | * the output parameter 'd' across continuation calls. This would not be | 
| Manuel Pégourié-Gonnard | c0edc96 | 2018-10-16 10:38:19 +0200 | [diff] [blame] | 52 | * acceptable for a public function but is OK here as we control call sites. | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 53 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 54 | static int ecdh_gen_public_restartable(mbedtls_ecp_group *grp, | 
|  | 55 | mbedtls_mpi *d, mbedtls_ecp_point *Q, | 
|  | 56 | int (*f_rng)(void *, unsigned char *, size_t), | 
|  | 57 | void *p_rng, | 
|  | 58 | mbedtls_ecp_restart_ctx *rs_ctx) | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 59 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 60 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 61 |  | 
| David Horstmann | 91e20a0 | 2022-10-06 19:11:28 +0100 | [diff] [blame] | 62 | int restarting = 0; | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 63 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 64 | restarting = (rs_ctx != NULL && rs_ctx->rsm != NULL); | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 65 | #endif | 
| David Horstmann | 91e20a0 | 2022-10-06 19:11:28 +0100 | [diff] [blame] | 66 | /* If multiplication is in progress, we already generated a privkey */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 67 | if (!restarting) { | 
|  | 68 | MBEDTLS_MPI_CHK(mbedtls_ecp_gen_privkey(grp, d, f_rng, p_rng)); | 
|  | 69 | } | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 70 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 71 | MBEDTLS_MPI_CHK(mbedtls_ecp_mul_restartable(grp, Q, d, &grp->G, | 
|  | 72 | f_rng, p_rng, rs_ctx)); | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 73 |  | 
|  | 74 | cleanup: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 75 | return ret; | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 76 | } | 
|  | 77 |  | 
|  | 78 | /* | 
|  | 79 | * Generate public key | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 80 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 81 | int mbedtls_ecdh_gen_public(mbedtls_ecp_group *grp, mbedtls_mpi *d, mbedtls_ecp_point *Q, | 
|  | 82 | int (*f_rng)(void *, unsigned char *, size_t), | 
|  | 83 | void *p_rng) | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 84 | { | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 85 | return ecdh_gen_public_restartable(grp, d, Q, f_rng, p_rng, NULL); | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 86 | } | 
| Ron Eldor | 936d284 | 2018-11-01 13:05:52 +0200 | [diff] [blame] | 87 | #endif /* !MBEDTLS_ECDH_GEN_PUBLIC_ALT */ | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 88 |  | 
| Ron Eldor | a84c1cb | 2017-10-10 19:04:27 +0300 | [diff] [blame] | 89 | #if !defined(MBEDTLS_ECDH_COMPUTE_SHARED_ALT) | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 90 | /* | 
|  | 91 | * Compute shared secret (SEC1 3.3.1) | 
|  | 92 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 93 | static int ecdh_compute_shared_restartable(mbedtls_ecp_group *grp, | 
|  | 94 | mbedtls_mpi *z, | 
|  | 95 | const mbedtls_ecp_point *Q, const mbedtls_mpi *d, | 
|  | 96 | int (*f_rng)(void *, unsigned char *, size_t), | 
|  | 97 | void *p_rng, | 
|  | 98 | mbedtls_ecp_restart_ctx *rs_ctx) | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 99 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 100 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 101 | mbedtls_ecp_point P; | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 102 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 103 | mbedtls_ecp_point_init(&P); | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 104 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 105 | MBEDTLS_MPI_CHK(mbedtls_ecp_mul_restartable(grp, &P, d, Q, | 
|  | 106 | f_rng, p_rng, rs_ctx)); | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 107 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 108 | if (mbedtls_ecp_is_zero(&P)) { | 
| Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 109 | ret = MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
| Paul Bakker | b548d77 | 2013-07-26 14:21:34 +0200 | [diff] [blame] | 110 | goto cleanup; | 
|  | 111 | } | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 112 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 113 | MBEDTLS_MPI_CHK(mbedtls_mpi_copy(z, &P.X)); | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 114 |  | 
|  | 115 | cleanup: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 116 | mbedtls_ecp_point_free(&P); | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 117 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 118 | return ret; | 
| Manuel Pégourié-Gonnard | 6545ca7 | 2013-01-26 16:05:22 +0100 | [diff] [blame] | 119 | } | 
|  | 120 |  | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 121 | /* | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 122 | * Compute shared secret (SEC1 3.3.1) | 
|  | 123 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 124 | int mbedtls_ecdh_compute_shared(mbedtls_ecp_group *grp, mbedtls_mpi *z, | 
|  | 125 | const mbedtls_ecp_point *Q, const mbedtls_mpi *d, | 
|  | 126 | int (*f_rng)(void *, unsigned char *, size_t), | 
|  | 127 | void *p_rng) | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 128 | { | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 129 | return ecdh_compute_shared_restartable(grp, z, Q, d, | 
|  | 130 | f_rng, p_rng, NULL); | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 131 | } | 
| Ron Eldor | 936d284 | 2018-11-01 13:05:52 +0200 | [diff] [blame] | 132 | #endif /* !MBEDTLS_ECDH_COMPUTE_SHARED_ALT */ | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 133 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 134 | static void ecdh_init_internal(mbedtls_ecdh_context_mbed *ctx) | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 135 | { | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 136 | mbedtls_ecp_group_init(&ctx->grp); | 
|  | 137 | mbedtls_mpi_init(&ctx->d); | 
|  | 138 | mbedtls_ecp_point_init(&ctx->Q); | 
|  | 139 | mbedtls_ecp_point_init(&ctx->Qp); | 
|  | 140 | mbedtls_mpi_init(&ctx->z); | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 141 |  | 
|  | 142 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 143 | mbedtls_ecp_restart_init(&ctx->rs); | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 144 | #endif | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 145 | } | 
|  | 146 |  | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 147 | /* | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 148 | * Initialize context | 
| Janos Follath | f61e486 | 2018-10-30 11:53:25 +0000 | [diff] [blame] | 149 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 150 | void mbedtls_ecdh_init(mbedtls_ecdh_context *ctx) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 151 | { | 
|  | 152 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 153 | ecdh_init_internal(ctx); | 
|  | 154 | mbedtls_ecp_point_init(&ctx->Vi); | 
|  | 155 | mbedtls_ecp_point_init(&ctx->Vf); | 
|  | 156 | mbedtls_mpi_init(&ctx->_d); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 157 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 158 | memset(ctx, 0, sizeof(mbedtls_ecdh_context)); | 
| Christoph M. Wintersteiger | d8c45d5 | 2019-02-20 17:16:53 +0000 | [diff] [blame] | 159 |  | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 160 | ctx->var = MBEDTLS_ECDH_VARIANT_NONE; | 
|  | 161 | #endif | 
|  | 162 | ctx->point_format = MBEDTLS_ECP_PF_UNCOMPRESSED; | 
|  | 163 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
|  | 164 | ctx->restart_enabled = 0; | 
|  | 165 | #endif | 
|  | 166 | } | 
|  | 167 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 168 | static int ecdh_setup_internal(mbedtls_ecdh_context_mbed *ctx, | 
|  | 169 | mbedtls_ecp_group_id grp_id) | 
| Janos Follath | f61e486 | 2018-10-30 11:53:25 +0000 | [diff] [blame] | 170 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 171 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Janos Follath | f61e486 | 2018-10-30 11:53:25 +0000 | [diff] [blame] | 172 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 173 | ret = mbedtls_ecp_group_load(&ctx->grp, grp_id); | 
|  | 174 | if (ret != 0) { | 
|  | 175 | return MBEDTLS_ERR_ECP_FEATURE_UNAVAILABLE; | 
| Janos Follath | f61e486 | 2018-10-30 11:53:25 +0000 | [diff] [blame] | 176 | } | 
|  | 177 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 178 | return 0; | 
| Janos Follath | f61e486 | 2018-10-30 11:53:25 +0000 | [diff] [blame] | 179 | } | 
|  | 180 |  | 
|  | 181 | /* | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 182 | * Setup context | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 183 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 184 | int mbedtls_ecdh_setup(mbedtls_ecdh_context *ctx, mbedtls_ecp_group_id grp_id) | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 185 | { | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 186 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 187 | return ecdh_setup_internal(ctx, grp_id); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 188 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 189 | switch (grp_id) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 190 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
| Christoph M. Wintersteiger | 3ff60bc | 2019-02-15 12:59:59 +0000 | [diff] [blame] | 191 | case MBEDTLS_ECP_DP_CURVE25519: | 
|  | 192 | ctx->point_format = MBEDTLS_ECP_PF_COMPRESSED; | 
|  | 193 | ctx->var = MBEDTLS_ECDH_VARIANT_EVEREST; | 
|  | 194 | ctx->grp_id = grp_id; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 195 | return mbedtls_everest_setup(&ctx->ctx.everest_ecdh, grp_id); | 
| Christoph M. Wintersteiger | 78c9c46 | 2018-12-06 17:16:32 +0000 | [diff] [blame] | 196 | #endif | 
| Christoph M. Wintersteiger | 3ff60bc | 2019-02-15 12:59:59 +0000 | [diff] [blame] | 197 | default: | 
|  | 198 | ctx->point_format = MBEDTLS_ECP_PF_UNCOMPRESSED; | 
|  | 199 | ctx->var = MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0; | 
|  | 200 | ctx->grp_id = grp_id; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 201 | ecdh_init_internal(&ctx->ctx.mbed_ecdh); | 
|  | 202 | return ecdh_setup_internal(&ctx->ctx.mbed_ecdh, grp_id); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 203 | } | 
|  | 204 | #endif | 
|  | 205 | } | 
|  | 206 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 207 | static void ecdh_free_internal(mbedtls_ecdh_context_mbed *ctx) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 208 | { | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 209 | mbedtls_ecp_group_free(&ctx->grp); | 
|  | 210 | mbedtls_mpi_free(&ctx->d); | 
|  | 211 | mbedtls_ecp_point_free(&ctx->Q); | 
|  | 212 | mbedtls_ecp_point_free(&ctx->Qp); | 
|  | 213 | mbedtls_mpi_free(&ctx->z); | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 214 |  | 
|  | 215 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 216 | mbedtls_ecp_restart_free(&ctx->rs); | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 217 | #endif | 
| Manuel Pégourié-Gonnard | 63533e4 | 2013-02-10 14:21:04 +0100 | [diff] [blame] | 218 | } | 
|  | 219 |  | 
| Manuel Pégourié-Gonnard | 23e4162 | 2017-05-18 12:35:37 +0200 | [diff] [blame] | 220 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
|  | 221 | /* | 
|  | 222 | * Enable restartable operations for context | 
|  | 223 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 224 | void mbedtls_ecdh_enable_restart(mbedtls_ecdh_context *ctx) | 
| Manuel Pégourié-Gonnard | 23e4162 | 2017-05-18 12:35:37 +0200 | [diff] [blame] | 225 | { | 
|  | 226 | ctx->restart_enabled = 1; | 
|  | 227 | } | 
|  | 228 | #endif | 
|  | 229 |  | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 230 | /* | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 231 | * Free context | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 232 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 233 | void mbedtls_ecdh_free(mbedtls_ecdh_context *ctx) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 234 | { | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 235 | if (ctx == NULL) { | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 236 | return; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 237 | } | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 238 |  | 
|  | 239 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 240 | mbedtls_ecp_point_free(&ctx->Vi); | 
|  | 241 | mbedtls_ecp_point_free(&ctx->Vf); | 
|  | 242 | mbedtls_mpi_free(&ctx->_d); | 
|  | 243 | ecdh_free_internal(ctx); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 244 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 245 | switch (ctx->var) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 246 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
|  | 247 | case MBEDTLS_ECDH_VARIANT_EVEREST: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 248 | mbedtls_everest_free(&ctx->ctx.everest_ecdh); | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 249 | break; | 
|  | 250 | #endif | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 251 | case MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 252 | ecdh_free_internal(&ctx->ctx.mbed_ecdh); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 253 | break; | 
|  | 254 | default: | 
|  | 255 | break; | 
|  | 256 | } | 
|  | 257 |  | 
|  | 258 | ctx->point_format = MBEDTLS_ECP_PF_UNCOMPRESSED; | 
|  | 259 | ctx->var = MBEDTLS_ECDH_VARIANT_NONE; | 
|  | 260 | ctx->grp_id = MBEDTLS_ECP_DP_NONE; | 
|  | 261 | #endif | 
|  | 262 | } | 
|  | 263 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 264 | static int ecdh_make_params_internal(mbedtls_ecdh_context_mbed *ctx, | 
|  | 265 | size_t *olen, int point_format, | 
|  | 266 | unsigned char *buf, size_t blen, | 
|  | 267 | int (*f_rng)(void *, | 
|  | 268 | unsigned char *, | 
|  | 269 | size_t), | 
|  | 270 | void *p_rng, | 
|  | 271 | int restart_enabled) | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 272 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 273 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 274 | size_t grp_len, pt_len; | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 275 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 276 | mbedtls_ecp_restart_ctx *rs_ctx = NULL; | 
| Ron Eldor | 936d284 | 2018-11-01 13:05:52 +0200 | [diff] [blame] | 277 | #endif | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 278 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 279 | if (ctx->grp.pbits == 0) { | 
|  | 280 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 281 | } | 
| Manuel Pégourié-Gonnard | f35b739 | 2013-02-11 22:12:39 +0100 | [diff] [blame] | 282 |  | 
| Ron Eldor | 19779c4 | 2018-11-05 16:58:13 +0200 | [diff] [blame] | 283 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 284 | if (restart_enabled) { | 
| Manuel Pégourié-Gonnard | 23e4162 | 2017-05-18 12:35:37 +0200 | [diff] [blame] | 285 | rs_ctx = &ctx->rs; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 286 | } | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 287 | #else | 
|  | 288 | (void) restart_enabled; | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 289 | #endif | 
|  | 290 |  | 
| Ron Eldor | 8493f80 | 2018-11-01 11:32:15 +0200 | [diff] [blame] | 291 |  | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 292 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 293 | if ((ret = ecdh_gen_public_restartable(&ctx->grp, &ctx->d, &ctx->Q, | 
|  | 294 | f_rng, p_rng, rs_ctx)) != 0) { | 
|  | 295 | return ret; | 
|  | 296 | } | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 297 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 298 | if ((ret = mbedtls_ecdh_gen_public(&ctx->grp, &ctx->d, &ctx->Q, | 
|  | 299 | f_rng, p_rng)) != 0) { | 
|  | 300 | return ret; | 
|  | 301 | } | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 302 | #endif /* MBEDTLS_ECP_RESTARTABLE */ | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 303 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 304 | if ((ret = mbedtls_ecp_tls_write_group(&ctx->grp, &grp_len, buf, | 
|  | 305 | blen)) != 0) { | 
|  | 306 | return ret; | 
|  | 307 | } | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 308 |  | 
|  | 309 | buf += grp_len; | 
|  | 310 | blen -= grp_len; | 
|  | 311 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 312 | if ((ret = mbedtls_ecp_tls_write_point(&ctx->grp, &ctx->Q, point_format, | 
|  | 313 | &pt_len, buf, blen)) != 0) { | 
|  | 314 | return ret; | 
|  | 315 | } | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 316 |  | 
|  | 317 | *olen = grp_len + pt_len; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 318 | return 0; | 
| Manuel Pégourié-Gonnard | 1372476 | 2013-02-10 15:01:54 +0100 | [diff] [blame] | 319 | } | 
|  | 320 |  | 
| Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 321 | /* | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 322 | * Setup and write the ServerKeyExchange parameters (RFC 4492) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 323 | *      struct { | 
|  | 324 | *          ECParameters    curve_params; | 
|  | 325 | *          ECPoint         public; | 
|  | 326 | *      } ServerECDHParams; | 
|  | 327 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 328 | int mbedtls_ecdh_make_params(mbedtls_ecdh_context *ctx, size_t *olen, | 
|  | 329 | unsigned char *buf, size_t blen, | 
|  | 330 | int (*f_rng)(void *, unsigned char *, size_t), | 
|  | 331 | void *p_rng) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 332 | { | 
|  | 333 | int restart_enabled = 0; | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 334 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
|  | 335 | restart_enabled = ctx->restart_enabled; | 
|  | 336 | #else | 
|  | 337 | (void) restart_enabled; | 
|  | 338 | #endif | 
|  | 339 |  | 
|  | 340 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 341 | return ecdh_make_params_internal(ctx, olen, ctx->point_format, buf, blen, | 
|  | 342 | f_rng, p_rng, restart_enabled); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 343 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 344 | switch (ctx->var) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 345 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
|  | 346 | case MBEDTLS_ECDH_VARIANT_EVEREST: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 347 | return mbedtls_everest_make_params(&ctx->ctx.everest_ecdh, olen, | 
|  | 348 | buf, blen, f_rng, p_rng); | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 349 | #endif | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 350 | case MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 351 | return ecdh_make_params_internal(&ctx->ctx.mbed_ecdh, olen, | 
|  | 352 | ctx->point_format, buf, blen, | 
|  | 353 | f_rng, p_rng, | 
|  | 354 | restart_enabled); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 355 | default: | 
|  | 356 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 357 | } | 
|  | 358 | #endif | 
|  | 359 | } | 
|  | 360 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 361 | static int ecdh_read_params_internal(mbedtls_ecdh_context_mbed *ctx, | 
|  | 362 | const unsigned char **buf, | 
|  | 363 | const unsigned char *end) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 364 | { | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 365 | return mbedtls_ecp_tls_read_point(&ctx->grp, &ctx->Qp, buf, | 
| Dave Rodgman | e4a6f5a | 2023-11-04 12:20:09 +0000 | [diff] [blame] | 366 | (size_t) (end - *buf)); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 367 | } | 
|  | 368 |  | 
|  | 369 | /* | 
| bootstrap-prime | 6dbbf44 | 2022-05-17 19:30:44 -0400 | [diff] [blame] | 370 | * Read the ServerKeyExchange parameters (RFC 4492) | 
| Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 371 | *      struct { | 
|  | 372 | *          ECParameters    curve_params; | 
|  | 373 | *          ECPoint         public; | 
|  | 374 | *      } ServerECDHParams; | 
|  | 375 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 376 | int mbedtls_ecdh_read_params(mbedtls_ecdh_context *ctx, | 
|  | 377 | const unsigned char **buf, | 
|  | 378 | const unsigned char *end) | 
| Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 379 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 380 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Janos Follath | f61e486 | 2018-10-30 11:53:25 +0000 | [diff] [blame] | 381 | mbedtls_ecp_group_id grp_id; | 
| Dave Rodgman | e4a6f5a | 2023-11-04 12:20:09 +0000 | [diff] [blame] | 382 | if ((ret = mbedtls_ecp_tls_read_group_id(&grp_id, buf, (size_t) (end - *buf))) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 383 | != 0) { | 
|  | 384 | return ret; | 
|  | 385 | } | 
| Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 386 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 387 | if ((ret = mbedtls_ecdh_setup(ctx, grp_id)) != 0) { | 
|  | 388 | return ret; | 
|  | 389 | } | 
| Janos Follath | f61e486 | 2018-10-30 11:53:25 +0000 | [diff] [blame] | 390 |  | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 391 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 392 | return ecdh_read_params_internal(ctx, buf, end); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 393 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 394 | switch (ctx->var) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 395 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
|  | 396 | case MBEDTLS_ECDH_VARIANT_EVEREST: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 397 | return mbedtls_everest_read_params(&ctx->ctx.everest_ecdh, | 
|  | 398 | buf, end); | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 399 | #endif | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 400 | case MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 401 | return ecdh_read_params_internal(&ctx->ctx.mbed_ecdh, | 
|  | 402 | buf, end); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 403 | default: | 
|  | 404 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 405 | } | 
|  | 406 | #endif | 
| Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 407 | } | 
| Manuel Pégourié-Gonnard | 0bad5c2 | 2013-01-26 15:30:46 +0100 | [diff] [blame] | 408 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 409 | static int ecdh_get_params_internal(mbedtls_ecdh_context_mbed *ctx, | 
|  | 410 | const mbedtls_ecp_keypair *key, | 
|  | 411 | mbedtls_ecdh_side side) | 
| Manuel Pégourié-Gonnard | cdff3cf | 2013-12-12 09:55:52 +0100 | [diff] [blame] | 412 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 413 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Manuel Pégourié-Gonnard | cdff3cf | 2013-12-12 09:55:52 +0100 | [diff] [blame] | 414 |  | 
| Manuel Pégourié-Gonnard | cdff3cf | 2013-12-12 09:55:52 +0100 | [diff] [blame] | 415 | /* If it's not our key, just import the public part as Qp */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 416 | if (side == MBEDTLS_ECDH_THEIRS) { | 
|  | 417 | return mbedtls_ecp_copy(&ctx->Qp, &key->Q); | 
|  | 418 | } | 
| Manuel Pégourié-Gonnard | cdff3cf | 2013-12-12 09:55:52 +0100 | [diff] [blame] | 419 |  | 
|  | 420 | /* Our key: import public (as Q) and private parts */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 421 | if (side != MBEDTLS_ECDH_OURS) { | 
|  | 422 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 423 | } | 
| Manuel Pégourié-Gonnard | cdff3cf | 2013-12-12 09:55:52 +0100 | [diff] [blame] | 424 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 425 | if ((ret = mbedtls_ecp_copy(&ctx->Q, &key->Q)) != 0 || | 
|  | 426 | (ret = mbedtls_mpi_copy(&ctx->d, &key->d)) != 0) { | 
|  | 427 | return ret; | 
|  | 428 | } | 
| Manuel Pégourié-Gonnard | cdff3cf | 2013-12-12 09:55:52 +0100 | [diff] [blame] | 429 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 430 | return 0; | 
| Manuel Pégourié-Gonnard | cdff3cf | 2013-12-12 09:55:52 +0100 | [diff] [blame] | 431 | } | 
|  | 432 |  | 
|  | 433 | /* | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 434 | * Get parameters from a keypair | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 435 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 436 | int mbedtls_ecdh_get_params(mbedtls_ecdh_context *ctx, | 
|  | 437 | const mbedtls_ecp_keypair *key, | 
|  | 438 | mbedtls_ecdh_side side) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 439 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 440 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 441 | if (side != MBEDTLS_ECDH_OURS && side != MBEDTLS_ECDH_THEIRS) { | 
|  | 442 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 443 | } | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 444 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 445 | if (mbedtls_ecdh_grp_id(ctx) == MBEDTLS_ECP_DP_NONE) { | 
| Gilles Peskine | 0b1b71d | 2018-11-07 22:10:59 +0100 | [diff] [blame] | 446 | /* This is the first call to get_params(). Set up the context | 
|  | 447 | * for use with the group. */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 448 | if ((ret = mbedtls_ecdh_setup(ctx, key->grp.id)) != 0) { | 
|  | 449 | return ret; | 
|  | 450 | } | 
|  | 451 | } else { | 
| Gilles Peskine | 0b1b71d | 2018-11-07 22:10:59 +0100 | [diff] [blame] | 452 | /* This is not the first call to get_params(). Check that the | 
|  | 453 | * current key's group is the same as the context's, which was set | 
|  | 454 | * from the first key's group. */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 455 | if (mbedtls_ecdh_grp_id(ctx) != key->grp.id) { | 
|  | 456 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 457 | } | 
| Gilles Peskine | 0b1b71d | 2018-11-07 22:10:59 +0100 | [diff] [blame] | 458 | } | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 459 |  | 
|  | 460 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 461 | return ecdh_get_params_internal(ctx, key, side); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 462 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 463 | switch (ctx->var) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 464 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
|  | 465 | case MBEDTLS_ECDH_VARIANT_EVEREST: | 
| Christoph M. Wintersteiger | 4936beb | 2018-12-12 17:26:41 +0000 | [diff] [blame] | 466 | { | 
| Christoph M. Wintersteiger | 2e724a1 | 2019-01-07 14:19:41 +0000 | [diff] [blame] | 467 | mbedtls_everest_ecdh_side s = side == MBEDTLS_ECDH_OURS ? | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 468 | MBEDTLS_EVEREST_ECDH_OURS : | 
|  | 469 | MBEDTLS_EVEREST_ECDH_THEIRS; | 
|  | 470 | return mbedtls_everest_get_params(&ctx->ctx.everest_ecdh, | 
|  | 471 | key, s); | 
| Christoph M. Wintersteiger | 4936beb | 2018-12-12 17:26:41 +0000 | [diff] [blame] | 472 | } | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 473 | #endif | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 474 | case MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 475 | return ecdh_get_params_internal(&ctx->ctx.mbed_ecdh, | 
|  | 476 | key, side); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 477 | default: | 
|  | 478 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 479 | } | 
|  | 480 | #endif | 
|  | 481 | } | 
|  | 482 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 483 | static int ecdh_make_public_internal(mbedtls_ecdh_context_mbed *ctx, | 
|  | 484 | size_t *olen, int point_format, | 
|  | 485 | unsigned char *buf, size_t blen, | 
|  | 486 | int (*f_rng)(void *, | 
|  | 487 | unsigned char *, | 
|  | 488 | size_t), | 
|  | 489 | void *p_rng, | 
|  | 490 | int restart_enabled) | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 491 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 492 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 493 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 494 | mbedtls_ecp_restart_ctx *rs_ctx = NULL; | 
| Ron Eldor | 936d284 | 2018-11-01 13:05:52 +0200 | [diff] [blame] | 495 | #endif | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 496 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 497 | if (ctx->grp.pbits == 0) { | 
|  | 498 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 499 | } | 
| Manuel Pégourié-Gonnard | f35b739 | 2013-02-11 22:12:39 +0100 | [diff] [blame] | 500 |  | 
| Ron Eldor | b430d9f | 2018-11-05 17:18:29 +0200 | [diff] [blame] | 501 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 502 | if (restart_enabled) { | 
| Manuel Pégourié-Gonnard | 23e4162 | 2017-05-18 12:35:37 +0200 | [diff] [blame] | 503 | rs_ctx = &ctx->rs; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 504 | } | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 505 | #else | 
|  | 506 | (void) restart_enabled; | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 507 | #endif | 
|  | 508 |  | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 509 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 510 | if ((ret = ecdh_gen_public_restartable(&ctx->grp, &ctx->d, &ctx->Q, | 
|  | 511 | f_rng, p_rng, rs_ctx)) != 0) { | 
|  | 512 | return ret; | 
|  | 513 | } | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 514 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 515 | if ((ret = mbedtls_ecdh_gen_public(&ctx->grp, &ctx->d, &ctx->Q, | 
|  | 516 | f_rng, p_rng)) != 0) { | 
|  | 517 | return ret; | 
|  | 518 | } | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 519 | #endif /* MBEDTLS_ECP_RESTARTABLE */ | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 520 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 521 | return mbedtls_ecp_tls_write_point(&ctx->grp, &ctx->Q, point_format, olen, | 
|  | 522 | buf, blen); | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 523 | } | 
|  | 524 |  | 
|  | 525 | /* | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 526 | * Setup and export the client public value | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 527 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 528 | int mbedtls_ecdh_make_public(mbedtls_ecdh_context *ctx, size_t *olen, | 
|  | 529 | unsigned char *buf, size_t blen, | 
|  | 530 | int (*f_rng)(void *, unsigned char *, size_t), | 
|  | 531 | void *p_rng) | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 532 | { | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 533 | int restart_enabled = 0; | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 534 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
|  | 535 | restart_enabled = ctx->restart_enabled; | 
|  | 536 | #endif | 
|  | 537 |  | 
|  | 538 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 539 | return ecdh_make_public_internal(ctx, olen, ctx->point_format, buf, blen, | 
|  | 540 | f_rng, p_rng, restart_enabled); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 541 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 542 | switch (ctx->var) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 543 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
|  | 544 | case MBEDTLS_ECDH_VARIANT_EVEREST: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 545 | return mbedtls_everest_make_public(&ctx->ctx.everest_ecdh, olen, | 
|  | 546 | buf, blen, f_rng, p_rng); | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 547 | #endif | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 548 | case MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 549 | return ecdh_make_public_internal(&ctx->ctx.mbed_ecdh, olen, | 
|  | 550 | ctx->point_format, buf, blen, | 
|  | 551 | f_rng, p_rng, | 
|  | 552 | restart_enabled); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 553 | default: | 
|  | 554 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 555 | } | 
|  | 556 | #endif | 
|  | 557 | } | 
|  | 558 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 559 | static int ecdh_read_public_internal(mbedtls_ecdh_context_mbed *ctx, | 
|  | 560 | const unsigned char *buf, size_t blen) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 561 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 562 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 563 | const unsigned char *p = buf; | 
|  | 564 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 565 | if ((ret = mbedtls_ecp_tls_read_point(&ctx->grp, &ctx->Qp, &p, | 
|  | 566 | blen)) != 0) { | 
|  | 567 | return ret; | 
|  | 568 | } | 
| Manuel Pégourié-Gonnard | 969ccc6 | 2014-03-26 19:53:25 +0100 | [diff] [blame] | 569 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 570 | if ((size_t) (p - buf) != blen) { | 
|  | 571 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 572 | } | 
| Manuel Pégourié-Gonnard | 969ccc6 | 2014-03-26 19:53:25 +0100 | [diff] [blame] | 573 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 574 | return 0; | 
| Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 575 | } | 
|  | 576 |  | 
| Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 577 | /* | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 578 | * Parse and import the client's public value | 
| Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 579 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 580 | int mbedtls_ecdh_read_public(mbedtls_ecdh_context *ctx, | 
|  | 581 | const unsigned char *buf, size_t blen) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 582 | { | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 583 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 584 | return ecdh_read_public_internal(ctx, buf, blen); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 585 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 586 | switch (ctx->var) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 587 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
|  | 588 | case MBEDTLS_ECDH_VARIANT_EVEREST: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 589 | return mbedtls_everest_read_public(&ctx->ctx.everest_ecdh, | 
|  | 590 | buf, blen); | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 591 | #endif | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 592 | case MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 593 | return ecdh_read_public_internal(&ctx->ctx.mbed_ecdh, | 
|  | 594 | buf, blen); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 595 | default: | 
|  | 596 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 597 | } | 
|  | 598 | #endif | 
|  | 599 | } | 
|  | 600 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 601 | static int ecdh_calc_secret_internal(mbedtls_ecdh_context_mbed *ctx, | 
|  | 602 | size_t *olen, unsigned char *buf, | 
|  | 603 | size_t blen, | 
|  | 604 | int (*f_rng)(void *, | 
|  | 605 | unsigned char *, | 
|  | 606 | size_t), | 
|  | 607 | void *p_rng, | 
|  | 608 | int restart_enabled) | 
| Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 609 | { | 
| Janos Follath | 24eed8d | 2019-11-22 13:21:35 +0000 | [diff] [blame] | 610 | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 611 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 612 | mbedtls_ecp_restart_ctx *rs_ctx = NULL; | 
| Ron Eldor | 936d284 | 2018-11-01 13:05:52 +0200 | [diff] [blame] | 613 | #endif | 
| Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 614 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 615 | if (ctx == NULL || ctx->grp.pbits == 0) { | 
|  | 616 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 617 | } | 
| Manuel Pégourié-Gonnard | f35b739 | 2013-02-11 22:12:39 +0100 | [diff] [blame] | 618 |  | 
| Ron Eldor | b430d9f | 2018-11-05 17:18:29 +0200 | [diff] [blame] | 619 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 620 | if (restart_enabled) { | 
| Manuel Pégourié-Gonnard | 23e4162 | 2017-05-18 12:35:37 +0200 | [diff] [blame] | 621 | rs_ctx = &ctx->rs; | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 622 | } | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 623 | #else | 
|  | 624 | (void) restart_enabled; | 
| Manuel Pégourié-Gonnard | 66ba48a | 2017-04-27 11:38:26 +0200 | [diff] [blame] | 625 | #endif | 
|  | 626 |  | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 627 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 628 | if ((ret = ecdh_compute_shared_restartable(&ctx->grp, &ctx->z, &ctx->Qp, | 
|  | 629 | &ctx->d, f_rng, p_rng, | 
|  | 630 | rs_ctx)) != 0) { | 
|  | 631 | return ret; | 
| Manuel Pégourié-Gonnard | e09d2f8 | 2013-09-02 14:29:09 +0200 | [diff] [blame] | 632 | } | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 633 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 634 | if ((ret = mbedtls_ecdh_compute_shared(&ctx->grp, &ctx->z, &ctx->Qp, | 
|  | 635 | &ctx->d, f_rng, p_rng)) != 0) { | 
|  | 636 | return ret; | 
| Ron Eldor | 2981d8f | 2018-11-05 18:07:10 +0200 | [diff] [blame] | 637 | } | 
|  | 638 | #endif /* MBEDTLS_ECP_RESTARTABLE */ | 
| Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 639 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 640 | if (mbedtls_mpi_size(&ctx->z) > blen) { | 
|  | 641 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
|  | 642 | } | 
| Paul Bakker | 41c83d3 | 2013-03-20 14:39:14 +0100 | [diff] [blame] | 643 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 644 | *olen = ctx->grp.pbits / 8 + ((ctx->grp.pbits % 8) != 0); | 
| Janos Follath | ab0f71a | 2019-02-20 10:48:49 +0000 | [diff] [blame] | 645 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 646 | if (mbedtls_ecp_get_type(&ctx->grp) == MBEDTLS_ECP_TYPE_MONTGOMERY) { | 
|  | 647 | return mbedtls_mpi_write_binary_le(&ctx->z, buf, *olen); | 
|  | 648 | } | 
| Janos Follath | ab0f71a | 2019-02-20 10:48:49 +0000 | [diff] [blame] | 649 |  | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 650 | return mbedtls_mpi_write_binary(&ctx->z, buf, *olen); | 
| Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 651 | } | 
|  | 652 |  | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 653 | /* | 
|  | 654 | * Derive and export the shared secret | 
|  | 655 | */ | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 656 | int mbedtls_ecdh_calc_secret(mbedtls_ecdh_context *ctx, size_t *olen, | 
|  | 657 | unsigned char *buf, size_t blen, | 
|  | 658 | int (*f_rng)(void *, unsigned char *, size_t), | 
|  | 659 | void *p_rng) | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 660 | { | 
|  | 661 | int restart_enabled = 0; | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 662 | #if defined(MBEDTLS_ECP_RESTARTABLE) | 
|  | 663 | restart_enabled = ctx->restart_enabled; | 
|  | 664 | #endif | 
|  | 665 |  | 
|  | 666 | #if defined(MBEDTLS_ECDH_LEGACY_CONTEXT) | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 667 | return ecdh_calc_secret_internal(ctx, olen, buf, blen, f_rng, p_rng, | 
|  | 668 | restart_enabled); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 669 | #else | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 670 | switch (ctx->var) { | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 671 | #if defined(MBEDTLS_ECDH_VARIANT_EVEREST_ENABLED) | 
|  | 672 | case MBEDTLS_ECDH_VARIANT_EVEREST: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 673 | return mbedtls_everest_calc_secret(&ctx->ctx.everest_ecdh, olen, | 
|  | 674 | buf, blen, f_rng, p_rng); | 
| Christoph M. Wintersteiger | c9f737b | 2018-10-25 13:03:05 +0100 | [diff] [blame] | 675 | #endif | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 676 | case MBEDTLS_ECDH_VARIANT_MBEDTLS_2_0: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 677 | return ecdh_calc_secret_internal(&ctx->ctx.mbed_ecdh, olen, buf, | 
|  | 678 | blen, f_rng, p_rng, | 
|  | 679 | restart_enabled); | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 680 | default: | 
| Gilles Peskine | 449bd83 | 2023-01-11 14:50:10 +0100 | [diff] [blame] | 681 | return MBEDTLS_ERR_ECP_BAD_INPUT_DATA; | 
| Janos Follath | 5a3e1bf | 2018-08-13 15:54:22 +0100 | [diff] [blame] | 682 | } | 
|  | 683 | #endif | 
|  | 684 | } | 
| Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 685 | #endif /* MBEDTLS_ECDH_C */ |