| /* | 
 |  *  Diffie-Hellman-Merkle key exchange | 
 |  * | 
 |  *  Copyright The Mbed TLS Contributors | 
 |  *  SPDX-License-Identifier: Apache-2.0 | 
 |  * | 
 |  *  Licensed under the Apache License, Version 2.0 (the "License"); you may | 
 |  *  not use this file except in compliance with the License. | 
 |  *  You may obtain a copy of the License at | 
 |  * | 
 |  *  http://www.apache.org/licenses/LICENSE-2.0 | 
 |  * | 
 |  *  Unless required by applicable law or agreed to in writing, software | 
 |  *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | 
 |  *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
 |  *  See the License for the specific language governing permissions and | 
 |  *  limitations under the License. | 
 |  */ | 
 | /* | 
 |  *  The following sources were referenced in the design of this implementation | 
 |  *  of the Diffie-Hellman-Merkle algorithm: | 
 |  * | 
 |  *  [1] Handbook of Applied Cryptography - 1997, Chapter 12 | 
 |  *      Menezes, van Oorschot and Vanstone | 
 |  * | 
 |  */ | 
 |  | 
 | #include "common.h" | 
 |  | 
 | #if defined(MBEDTLS_DHM_C) | 
 |  | 
 | #include "mbedtls/dhm.h" | 
 | #include "mbedtls/platform_util.h" | 
 | #include "mbedtls/error.h" | 
 |  | 
 | #include <string.h> | 
 |  | 
 | #if defined(MBEDTLS_PEM_PARSE_C) | 
 | #include "mbedtls/pem.h" | 
 | #endif | 
 |  | 
 | #if defined(MBEDTLS_ASN1_PARSE_C) | 
 | #include "mbedtls/asn1.h" | 
 | #endif | 
 |  | 
 | #include "mbedtls/platform.h" | 
 |  | 
 | #if !defined(MBEDTLS_DHM_ALT) | 
 |  | 
 | /* | 
 |  * helper to validate the mbedtls_mpi size and import it | 
 |  */ | 
 | static int dhm_read_bignum(mbedtls_mpi *X, | 
 |                            unsigned char **p, | 
 |                            const unsigned char *end) | 
 | { | 
 |     int ret, n; | 
 |  | 
 |     if (end - *p < 2) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 |     n = ((*p)[0] << 8) | (*p)[1]; | 
 |     (*p) += 2; | 
 |  | 
 |     if ((int) (end - *p) < n) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 |     if ((ret = mbedtls_mpi_read_binary(X, *p, n)) != 0) { | 
 |         return MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_READ_PARAMS_FAILED, ret); | 
 |     } | 
 |  | 
 |     (*p) += n; | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | /* | 
 |  * Verify sanity of parameter with regards to P | 
 |  * | 
 |  * Parameter should be: 2 <= public_param <= P - 2 | 
 |  * | 
 |  * This means that we need to return an error if | 
 |  *              public_param < 2 or public_param > P-2 | 
 |  * | 
 |  * For more information on the attack, see: | 
 |  *  http://www.cl.cam.ac.uk/~rja14/Papers/psandqs.pdf | 
 |  *  http://web.nvd.nist.gov/view/vuln/detail?vulnId=CVE-2005-2643 | 
 |  */ | 
 | static int dhm_check_range(const mbedtls_mpi *param, const mbedtls_mpi *P) | 
 | { | 
 |     mbedtls_mpi U; | 
 |     int ret = 0; | 
 |  | 
 |     mbedtls_mpi_init(&U); | 
 |  | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_sub_int(&U, P, 2)); | 
 |  | 
 |     if (mbedtls_mpi_cmp_int(param, 2) < 0 || | 
 |         mbedtls_mpi_cmp_mpi(param, &U) > 0) { | 
 |         ret = MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 | cleanup: | 
 |     mbedtls_mpi_free(&U); | 
 |     return ret; | 
 | } | 
 |  | 
 | void mbedtls_dhm_init(mbedtls_dhm_context *ctx) | 
 | { | 
 |     memset(ctx, 0, sizeof(mbedtls_dhm_context)); | 
 | } | 
 |  | 
 | size_t mbedtls_dhm_get_bitlen(const mbedtls_dhm_context *ctx) | 
 | { | 
 |     return mbedtls_mpi_bitlen(&ctx->P); | 
 | } | 
 |  | 
 | size_t mbedtls_dhm_get_len(const mbedtls_dhm_context *ctx) | 
 | { | 
 |     return mbedtls_mpi_size(&ctx->P); | 
 | } | 
 |  | 
 | int mbedtls_dhm_get_value(const mbedtls_dhm_context *ctx, | 
 |                           mbedtls_dhm_parameter param, | 
 |                           mbedtls_mpi *dest) | 
 | { | 
 |     const mbedtls_mpi *src = NULL; | 
 |     switch (param) { | 
 |         case MBEDTLS_DHM_PARAM_P: | 
 |             src = &ctx->P; | 
 |             break; | 
 |         case MBEDTLS_DHM_PARAM_G: | 
 |             src = &ctx->G; | 
 |             break; | 
 |         case MBEDTLS_DHM_PARAM_X: | 
 |             src = &ctx->X; | 
 |             break; | 
 |         case MBEDTLS_DHM_PARAM_GX: | 
 |             src = &ctx->GX; | 
 |             break; | 
 |         case MBEDTLS_DHM_PARAM_GY: | 
 |             src = &ctx->GY; | 
 |             break; | 
 |         case MBEDTLS_DHM_PARAM_K: | 
 |             src = &ctx->K; | 
 |             break; | 
 |         default: | 
 |             return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |     return mbedtls_mpi_copy(dest, src); | 
 | } | 
 |  | 
 | /* | 
 |  * Parse the ServerKeyExchange parameters | 
 |  */ | 
 | int mbedtls_dhm_read_params(mbedtls_dhm_context *ctx, | 
 |                             unsigned char **p, | 
 |                             const unsigned char *end) | 
 | { | 
 |     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
 |  | 
 |     if ((ret = dhm_read_bignum(&ctx->P,  p, end)) != 0 || | 
 |         (ret = dhm_read_bignum(&ctx->G,  p, end)) != 0 || | 
 |         (ret = dhm_read_bignum(&ctx->GY, p, end)) != 0) { | 
 |         return ret; | 
 |     } | 
 |  | 
 |     if ((ret = dhm_check_range(&ctx->GY, &ctx->P)) != 0) { | 
 |         return ret; | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | /* | 
 |  * Pick a random R in the range [2, M-2] for blinding or key generation. | 
 |  */ | 
 | static int dhm_random_below(mbedtls_mpi *R, const mbedtls_mpi *M, | 
 |                             int (*f_rng)(void *, unsigned char *, size_t), void *p_rng) | 
 | { | 
 |     int ret; | 
 |  | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_random(R, 3, M, f_rng, p_rng)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_sub_int(R, R, 1)); | 
 |  | 
 | cleanup: | 
 |     return ret; | 
 | } | 
 |  | 
 | static int dhm_make_common(mbedtls_dhm_context *ctx, int x_size, | 
 |                            int (*f_rng)(void *, unsigned char *, size_t), | 
 |                            void *p_rng) | 
 | { | 
 |     int ret = 0; | 
 |  | 
 |     if (mbedtls_mpi_cmp_int(&ctx->P, 0) == 0) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |     if (x_size < 0) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 |     if ((unsigned) x_size < mbedtls_mpi_size(&ctx->P)) { | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_fill_random(&ctx->X, x_size, f_rng, p_rng)); | 
 |     } else { | 
 |         /* Generate X as large as possible ( <= P - 2 ) */ | 
 |         ret = dhm_random_below(&ctx->X, &ctx->P, f_rng, p_rng); | 
 |         if (ret == MBEDTLS_ERR_MPI_NOT_ACCEPTABLE) { | 
 |             return MBEDTLS_ERR_DHM_MAKE_PARAMS_FAILED; | 
 |         } | 
 |         if (ret != 0) { | 
 |             return ret; | 
 |         } | 
 |     } | 
 |  | 
 |     /* | 
 |      * Calculate GX = G^X mod P | 
 |      */ | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_exp_mod(&ctx->GX, &ctx->G, &ctx->X, | 
 |                                         &ctx->P, &ctx->RP)); | 
 |  | 
 |     if ((ret = dhm_check_range(&ctx->GX, &ctx->P)) != 0) { | 
 |         return ret; | 
 |     } | 
 |  | 
 | cleanup: | 
 |     return ret; | 
 | } | 
 |  | 
 | /* | 
 |  * Setup and write the ServerKeyExchange parameters | 
 |  */ | 
 | int mbedtls_dhm_make_params(mbedtls_dhm_context *ctx, int x_size, | 
 |                             unsigned char *output, size_t *olen, | 
 |                             int (*f_rng)(void *, unsigned char *, size_t), | 
 |                             void *p_rng) | 
 | { | 
 |     int ret; | 
 |     size_t n1, n2, n3; | 
 |     unsigned char *p; | 
 |  | 
 |     ret = dhm_make_common(ctx, x_size, f_rng, p_rng); | 
 |     if (ret != 0) { | 
 |         goto cleanup; | 
 |     } | 
 |  | 
 |     /* | 
 |      * Export P, G, GX. RFC 5246 §4.4 states that "leading zero octets are | 
 |      * not required". We omit leading zeros for compactness. | 
 |      */ | 
 | #define DHM_MPI_EXPORT(X, n)                                          \ | 
 |     do {                                                                \ | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary((X),               \ | 
 |                                                  p + 2,               \ | 
 |                                                  (n)));           \ | 
 |         *p++ = MBEDTLS_BYTE_1(n);                                     \ | 
 |         *p++ = MBEDTLS_BYTE_0(n);                                     \ | 
 |         p += (n);                                                     \ | 
 |     } while (0) | 
 |  | 
 |     n1 = mbedtls_mpi_size(&ctx->P); | 
 |     n2 = mbedtls_mpi_size(&ctx->G); | 
 |     n3 = mbedtls_mpi_size(&ctx->GX); | 
 |  | 
 |     p = output; | 
 |     DHM_MPI_EXPORT(&ctx->P, n1); | 
 |     DHM_MPI_EXPORT(&ctx->G, n2); | 
 |     DHM_MPI_EXPORT(&ctx->GX, n3); | 
 |  | 
 |     *olen = p - output; | 
 |  | 
 | cleanup: | 
 |     if (ret != 0 && ret > -128) { | 
 |         ret = MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_MAKE_PARAMS_FAILED, ret); | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 | /* | 
 |  * Set prime modulus and generator | 
 |  */ | 
 | int mbedtls_dhm_set_group(mbedtls_dhm_context *ctx, | 
 |                           const mbedtls_mpi *P, | 
 |                           const mbedtls_mpi *G) | 
 | { | 
 |     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
 |  | 
 |     if ((ret = mbedtls_mpi_copy(&ctx->P, P)) != 0 || | 
 |         (ret = mbedtls_mpi_copy(&ctx->G, G)) != 0) { | 
 |         return MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_SET_GROUP_FAILED, ret); | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | /* | 
 |  * Import the peer's public value G^Y | 
 |  */ | 
 | int mbedtls_dhm_read_public(mbedtls_dhm_context *ctx, | 
 |                             const unsigned char *input, size_t ilen) | 
 | { | 
 |     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
 |  | 
 |     if (ilen < 1 || ilen > mbedtls_dhm_get_len(ctx)) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 |     if ((ret = mbedtls_mpi_read_binary(&ctx->GY, input, ilen)) != 0) { | 
 |         return MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_READ_PUBLIC_FAILED, ret); | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | /* | 
 |  * Create own private value X and export G^X | 
 |  */ | 
 | int mbedtls_dhm_make_public(mbedtls_dhm_context *ctx, int x_size, | 
 |                             unsigned char *output, size_t olen, | 
 |                             int (*f_rng)(void *, unsigned char *, size_t), | 
 |                             void *p_rng) | 
 | { | 
 |     int ret; | 
 |  | 
 |     if (olen < 1 || olen > mbedtls_dhm_get_len(ctx)) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 |     ret = dhm_make_common(ctx, x_size, f_rng, p_rng); | 
 |     if (ret == MBEDTLS_ERR_DHM_MAKE_PARAMS_FAILED) { | 
 |         return MBEDTLS_ERR_DHM_MAKE_PUBLIC_FAILED; | 
 |     } | 
 |     if (ret != 0) { | 
 |         goto cleanup; | 
 |     } | 
 |  | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary(&ctx->GX, output, olen)); | 
 |  | 
 | cleanup: | 
 |     if (ret != 0 && ret > -128) { | 
 |         ret = MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_MAKE_PUBLIC_FAILED, ret); | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 |  | 
 | /* | 
 |  * Use the blinding method and optimisation suggested in section 10 of: | 
 |  *  KOCHER, Paul C. Timing attacks on implementations of Diffie-Hellman, RSA, | 
 |  *  DSS, and other systems. In : Advances in Cryptology-CRYPTO'96. Springer | 
 |  *  Berlin Heidelberg, 1996. p. 104-113. | 
 |  */ | 
 | static int dhm_update_blinding(mbedtls_dhm_context *ctx, | 
 |                                int (*f_rng)(void *, unsigned char *, size_t), void *p_rng) | 
 | { | 
 |     int ret; | 
 |     mbedtls_mpi R; | 
 |  | 
 |     mbedtls_mpi_init(&R); | 
 |  | 
 |     /* | 
 |      * Don't use any blinding the first time a particular X is used, | 
 |      * but remember it to use blinding next time. | 
 |      */ | 
 |     if (mbedtls_mpi_cmp_mpi(&ctx->X, &ctx->pX) != 0) { | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_copy(&ctx->pX, &ctx->X)); | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_lset(&ctx->Vi, 1)); | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_lset(&ctx->Vf, 1)); | 
 |  | 
 |         return 0; | 
 |     } | 
 |  | 
 |     /* | 
 |      * Ok, we need blinding. Can we re-use existing values? | 
 |      * If yes, just update them by squaring them. | 
 |      */ | 
 |     if (mbedtls_mpi_cmp_int(&ctx->Vi, 1) != 0) { | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_mul_mpi(&ctx->Vi, &ctx->Vi, &ctx->Vi)); | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_mod_mpi(&ctx->Vi, &ctx->Vi, &ctx->P)); | 
 |  | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_mul_mpi(&ctx->Vf, &ctx->Vf, &ctx->Vf)); | 
 |         MBEDTLS_MPI_CHK(mbedtls_mpi_mod_mpi(&ctx->Vf, &ctx->Vf, &ctx->P)); | 
 |  | 
 |         return 0; | 
 |     } | 
 |  | 
 |     /* | 
 |      * We need to generate blinding values from scratch | 
 |      */ | 
 |  | 
 |     /* Vi = random( 2, P-2 ) */ | 
 |     MBEDTLS_MPI_CHK(dhm_random_below(&ctx->Vi, &ctx->P, f_rng, p_rng)); | 
 |  | 
 |     /* Vf = Vi^-X mod P | 
 |      * First compute Vi^-1 = R * (R Vi)^-1, (avoiding leaks from inv_mod), | 
 |      * then elevate to the Xth power. */ | 
 |     MBEDTLS_MPI_CHK(dhm_random_below(&R, &ctx->P, f_rng, p_rng)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mul_mpi(&ctx->Vf, &ctx->Vi, &R)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mod_mpi(&ctx->Vf, &ctx->Vf, &ctx->P)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_inv_mod(&ctx->Vf, &ctx->Vf, &ctx->P)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mul_mpi(&ctx->Vf, &ctx->Vf, &R)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mod_mpi(&ctx->Vf, &ctx->Vf, &ctx->P)); | 
 |  | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_exp_mod(&ctx->Vf, &ctx->Vf, &ctx->X, &ctx->P, &ctx->RP)); | 
 |  | 
 | cleanup: | 
 |     mbedtls_mpi_free(&R); | 
 |  | 
 |     return ret; | 
 | } | 
 |  | 
 | /* | 
 |  * Derive and export the shared secret (G^Y)^X mod P | 
 |  */ | 
 | int mbedtls_dhm_calc_secret(mbedtls_dhm_context *ctx, | 
 |                             unsigned char *output, size_t output_size, size_t *olen, | 
 |                             int (*f_rng)(void *, unsigned char *, size_t), | 
 |                             void *p_rng) | 
 | { | 
 |     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
 |     mbedtls_mpi GYb; | 
 |  | 
 |     if (f_rng == NULL) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 |     if (output_size < mbedtls_dhm_get_len(ctx)) { | 
 |         return MBEDTLS_ERR_DHM_BAD_INPUT_DATA; | 
 |     } | 
 |  | 
 |     if ((ret = dhm_check_range(&ctx->GY, &ctx->P)) != 0) { | 
 |         return ret; | 
 |     } | 
 |  | 
 |     mbedtls_mpi_init(&GYb); | 
 |  | 
 |     /* Blind peer's value */ | 
 |     MBEDTLS_MPI_CHK(dhm_update_blinding(ctx, f_rng, p_rng)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mul_mpi(&GYb, &ctx->GY, &ctx->Vi)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mod_mpi(&GYb, &GYb, &ctx->P)); | 
 |  | 
 |     /* Do modular exponentiation */ | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_exp_mod(&ctx->K, &GYb, &ctx->X, | 
 |                                         &ctx->P, &ctx->RP)); | 
 |  | 
 |     /* Unblind secret value */ | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mul_mpi(&ctx->K, &ctx->K, &ctx->Vf)); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_mod_mpi(&ctx->K, &ctx->K, &ctx->P)); | 
 |  | 
 |     /* Output the secret without any leading zero byte. This is mandatory | 
 |      * for TLS per RFC 5246 §8.1.2. */ | 
 |     *olen = mbedtls_mpi_size(&ctx->K); | 
 |     MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary(&ctx->K, output, *olen)); | 
 |  | 
 | cleanup: | 
 |     mbedtls_mpi_free(&GYb); | 
 |  | 
 |     if (ret != 0) { | 
 |         return MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_CALC_SECRET_FAILED, ret); | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | /* | 
 |  * Free the components of a DHM key | 
 |  */ | 
 | void mbedtls_dhm_free(mbedtls_dhm_context *ctx) | 
 | { | 
 |     if (ctx == NULL) { | 
 |         return; | 
 |     } | 
 |  | 
 |     mbedtls_mpi_free(&ctx->pX); | 
 |     mbedtls_mpi_free(&ctx->Vf); | 
 |     mbedtls_mpi_free(&ctx->Vi); | 
 |     mbedtls_mpi_free(&ctx->RP); | 
 |     mbedtls_mpi_free(&ctx->K); | 
 |     mbedtls_mpi_free(&ctx->GY); | 
 |     mbedtls_mpi_free(&ctx->GX); | 
 |     mbedtls_mpi_free(&ctx->X); | 
 |     mbedtls_mpi_free(&ctx->G); | 
 |     mbedtls_mpi_free(&ctx->P); | 
 |  | 
 |     mbedtls_platform_zeroize(ctx, sizeof(mbedtls_dhm_context)); | 
 | } | 
 |  | 
 | #if defined(MBEDTLS_ASN1_PARSE_C) | 
 | /* | 
 |  * Parse DHM parameters | 
 |  */ | 
 | int mbedtls_dhm_parse_dhm(mbedtls_dhm_context *dhm, const unsigned char *dhmin, | 
 |                           size_t dhminlen) | 
 | { | 
 |     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
 |     size_t len; | 
 |     unsigned char *p, *end; | 
 | #if defined(MBEDTLS_PEM_PARSE_C) | 
 |     mbedtls_pem_context pem; | 
 | #endif /* MBEDTLS_PEM_PARSE_C */ | 
 |  | 
 | #if defined(MBEDTLS_PEM_PARSE_C) | 
 |     mbedtls_pem_init(&pem); | 
 |  | 
 |     /* Avoid calling mbedtls_pem_read_buffer() on non-null-terminated string */ | 
 |     if (dhminlen == 0 || dhmin[dhminlen - 1] != '\0') { | 
 |         ret = MBEDTLS_ERR_PEM_NO_HEADER_FOOTER_PRESENT; | 
 |     } else { | 
 |         ret = mbedtls_pem_read_buffer(&pem, | 
 |                                       "-----BEGIN DH PARAMETERS-----", | 
 |                                       "-----END DH PARAMETERS-----", | 
 |                                       dhmin, NULL, 0, &dhminlen); | 
 |     } | 
 |  | 
 |     if (ret == 0) { | 
 |         /* | 
 |          * Was PEM encoded | 
 |          */ | 
 |         dhminlen = pem.buflen; | 
 |     } else if (ret != MBEDTLS_ERR_PEM_NO_HEADER_FOOTER_PRESENT) { | 
 |         goto exit; | 
 |     } | 
 |  | 
 |     p = (ret == 0) ? pem.buf : (unsigned char *) dhmin; | 
 | #else | 
 |     p = (unsigned char *) dhmin; | 
 | #endif /* MBEDTLS_PEM_PARSE_C */ | 
 |     end = p + dhminlen; | 
 |  | 
 |     /* | 
 |      *  DHParams ::= SEQUENCE { | 
 |      *      prime              INTEGER,  -- P | 
 |      *      generator          INTEGER,  -- g | 
 |      *      privateValueLength INTEGER OPTIONAL | 
 |      *  } | 
 |      */ | 
 |     if ((ret = mbedtls_asn1_get_tag(&p, end, &len, | 
 |                                     MBEDTLS_ASN1_CONSTRUCTED | MBEDTLS_ASN1_SEQUENCE)) != 0) { | 
 |         ret = MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_INVALID_FORMAT, ret); | 
 |         goto exit; | 
 |     } | 
 |  | 
 |     end = p + len; | 
 |  | 
 |     if ((ret = mbedtls_asn1_get_mpi(&p, end, &dhm->P)) != 0 || | 
 |         (ret = mbedtls_asn1_get_mpi(&p, end, &dhm->G)) != 0) { | 
 |         ret = MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_INVALID_FORMAT, ret); | 
 |         goto exit; | 
 |     } | 
 |  | 
 |     if (p != end) { | 
 |         /* This might be the optional privateValueLength. | 
 |          * If so, we can cleanly discard it */ | 
 |         mbedtls_mpi rec; | 
 |         mbedtls_mpi_init(&rec); | 
 |         ret = mbedtls_asn1_get_mpi(&p, end, &rec); | 
 |         mbedtls_mpi_free(&rec); | 
 |         if (ret != 0) { | 
 |             ret = MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_INVALID_FORMAT, ret); | 
 |             goto exit; | 
 |         } | 
 |         if (p != end) { | 
 |             ret = MBEDTLS_ERROR_ADD(MBEDTLS_ERR_DHM_INVALID_FORMAT, | 
 |                                     MBEDTLS_ERR_ASN1_LENGTH_MISMATCH); | 
 |             goto exit; | 
 |         } | 
 |     } | 
 |  | 
 |     ret = 0; | 
 |  | 
 | exit: | 
 | #if defined(MBEDTLS_PEM_PARSE_C) | 
 |     mbedtls_pem_free(&pem); | 
 | #endif | 
 |     if (ret != 0) { | 
 |         mbedtls_dhm_free(dhm); | 
 |     } | 
 |  | 
 |     return ret; | 
 | } | 
 |  | 
 | #if defined(MBEDTLS_FS_IO) | 
 | /* | 
 |  * Load all data from a file into a given buffer. | 
 |  * | 
 |  * The file is expected to contain either PEM or DER encoded data. | 
 |  * A terminating null byte is always appended. It is included in the announced | 
 |  * length only if the data looks like it is PEM encoded. | 
 |  */ | 
 | static int load_file(const char *path, unsigned char **buf, size_t *n) | 
 | { | 
 |     FILE *f; | 
 |     long size; | 
 |  | 
 |     if ((f = fopen(path, "rb")) == NULL) { | 
 |         return MBEDTLS_ERR_DHM_FILE_IO_ERROR; | 
 |     } | 
 |     /* The data loaded here is public, so don't bother disabling buffering. */ | 
 |  | 
 |     fseek(f, 0, SEEK_END); | 
 |     if ((size = ftell(f)) == -1) { | 
 |         fclose(f); | 
 |         return MBEDTLS_ERR_DHM_FILE_IO_ERROR; | 
 |     } | 
 |     fseek(f, 0, SEEK_SET); | 
 |  | 
 |     *n = (size_t) size; | 
 |  | 
 |     if (*n + 1 == 0 || | 
 |         (*buf = mbedtls_calloc(1, *n + 1)) == NULL) { | 
 |         fclose(f); | 
 |         return MBEDTLS_ERR_DHM_ALLOC_FAILED; | 
 |     } | 
 |  | 
 |     if (fread(*buf, 1, *n, f) != *n) { | 
 |         fclose(f); | 
 |  | 
 |         mbedtls_platform_zeroize(*buf, *n + 1); | 
 |         mbedtls_free(*buf); | 
 |  | 
 |         return MBEDTLS_ERR_DHM_FILE_IO_ERROR; | 
 |     } | 
 |  | 
 |     fclose(f); | 
 |  | 
 |     (*buf)[*n] = '\0'; | 
 |  | 
 |     if (strstr((const char *) *buf, "-----BEGIN ") != NULL) { | 
 |         ++*n; | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | /* | 
 |  * Load and parse DHM parameters | 
 |  */ | 
 | int mbedtls_dhm_parse_dhmfile(mbedtls_dhm_context *dhm, const char *path) | 
 | { | 
 |     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
 |     size_t n; | 
 |     unsigned char *buf; | 
 |  | 
 |     if ((ret = load_file(path, &buf, &n)) != 0) { | 
 |         return ret; | 
 |     } | 
 |  | 
 |     ret = mbedtls_dhm_parse_dhm(dhm, buf, n); | 
 |  | 
 |     mbedtls_platform_zeroize(buf, n); | 
 |     mbedtls_free(buf); | 
 |  | 
 |     return ret; | 
 | } | 
 | #endif /* MBEDTLS_FS_IO */ | 
 | #endif /* MBEDTLS_ASN1_PARSE_C */ | 
 | #endif /* MBEDTLS_DHM_ALT */ | 
 |  | 
 | #if defined(MBEDTLS_SELF_TEST) | 
 |  | 
 | #if defined(MBEDTLS_PEM_PARSE_C) | 
 | static const char mbedtls_test_dhm_params[] = | 
 |     "-----BEGIN DH PARAMETERS-----\r\n" | 
 |     "MIGHAoGBAJ419DBEOgmQTzo5qXl5fQcN9TN455wkOL7052HzxxRVMyhYmwQcgJvh\r\n" | 
 |     "1sa18fyfR9OiVEMYglOpkqVoGLN7qd5aQNNi5W7/C+VBdHTBJcGZJyyP5B3qcz32\r\n" | 
 |     "9mLJKudlVudV0Qxk5qUJaPZ/xupz0NyoVpviuiBOI1gNi8ovSXWzAgEC\r\n" | 
 |     "-----END DH PARAMETERS-----\r\n"; | 
 | #else /* MBEDTLS_PEM_PARSE_C */ | 
 | static const char mbedtls_test_dhm_params[] = { | 
 |     0x30, 0x81, 0x87, 0x02, 0x81, 0x81, 0x00, 0x9e, 0x35, 0xf4, 0x30, 0x44, | 
 |     0x3a, 0x09, 0x90, 0x4f, 0x3a, 0x39, 0xa9, 0x79, 0x79, 0x7d, 0x07, 0x0d, | 
 |     0xf5, 0x33, 0x78, 0xe7, 0x9c, 0x24, 0x38, 0xbe, 0xf4, 0xe7, 0x61, 0xf3, | 
 |     0xc7, 0x14, 0x55, 0x33, 0x28, 0x58, 0x9b, 0x04, 0x1c, 0x80, 0x9b, 0xe1, | 
 |     0xd6, 0xc6, 0xb5, 0xf1, 0xfc, 0x9f, 0x47, 0xd3, 0xa2, 0x54, 0x43, 0x18, | 
 |     0x82, 0x53, 0xa9, 0x92, 0xa5, 0x68, 0x18, 0xb3, 0x7b, 0xa9, 0xde, 0x5a, | 
 |     0x40, 0xd3, 0x62, 0xe5, 0x6e, 0xff, 0x0b, 0xe5, 0x41, 0x74, 0x74, 0xc1, | 
 |     0x25, 0xc1, 0x99, 0x27, 0x2c, 0x8f, 0xe4, 0x1d, 0xea, 0x73, 0x3d, 0xf6, | 
 |     0xf6, 0x62, 0xc9, 0x2a, 0xe7, 0x65, 0x56, 0xe7, 0x55, 0xd1, 0x0c, 0x64, | 
 |     0xe6, 0xa5, 0x09, 0x68, 0xf6, 0x7f, 0xc6, 0xea, 0x73, 0xd0, 0xdc, 0xa8, | 
 |     0x56, 0x9b, 0xe2, 0xba, 0x20, 0x4e, 0x23, 0x58, 0x0d, 0x8b, 0xca, 0x2f, | 
 |     0x49, 0x75, 0xb3, 0x02, 0x01, 0x02 | 
 | }; | 
 | #endif /* MBEDTLS_PEM_PARSE_C */ | 
 |  | 
 | static const size_t mbedtls_test_dhm_params_len = sizeof(mbedtls_test_dhm_params); | 
 |  | 
 | /* | 
 |  * Checkup routine | 
 |  */ | 
 | int mbedtls_dhm_self_test(int verbose) | 
 | { | 
 |     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
 |     mbedtls_dhm_context dhm; | 
 |  | 
 |     mbedtls_dhm_init(&dhm); | 
 |  | 
 |     if (verbose != 0) { | 
 |         mbedtls_printf("  DHM parameter load: "); | 
 |     } | 
 |  | 
 |     if ((ret = mbedtls_dhm_parse_dhm(&dhm, | 
 |                                      (const unsigned char *) mbedtls_test_dhm_params, | 
 |                                      mbedtls_test_dhm_params_len)) != 0) { | 
 |         if (verbose != 0) { | 
 |             mbedtls_printf("failed\n"); | 
 |         } | 
 |  | 
 |         ret = 1; | 
 |         goto exit; | 
 |     } | 
 |  | 
 |     if (verbose != 0) { | 
 |         mbedtls_printf("passed\n\n"); | 
 |     } | 
 |  | 
 | exit: | 
 |     mbedtls_dhm_free(&dhm); | 
 |  | 
 |     return ret; | 
 | } | 
 |  | 
 | #endif /* MBEDTLS_SELF_TEST */ | 
 |  | 
 | #endif /* MBEDTLS_DHM_C */ |