mirror of
https://github.com/Detanup01/gbe_fork.git
synced 2024-11-27 21:24:01 +08:00
296 lines
9.4 KiB
C
296 lines
9.4 KiB
C
/*
|
|
* PSA FFDH layer on top of Mbed TLS crypto
|
|
*/
|
|
/*
|
|
* Copyright The Mbed TLS Contributors
|
|
* SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
|
|
*/
|
|
|
|
#include "common.h"
|
|
|
|
#if defined(MBEDTLS_PSA_CRYPTO_C)
|
|
|
|
#include <psa/crypto.h>
|
|
#include "psa_crypto_core.h"
|
|
#include "psa_crypto_ffdh.h"
|
|
#include "psa_crypto_random_impl.h"
|
|
#include "mbedtls/platform.h"
|
|
#include "mbedtls/error.h"
|
|
|
|
#if defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_EXPORT) || \
|
|
defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_GENERATE) || \
|
|
defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_PUBLIC_KEY) || \
|
|
defined(MBEDTLS_PSA_BUILTIN_ALG_FFDH)
|
|
static psa_status_t mbedtls_psa_ffdh_set_prime_generator(size_t key_size,
|
|
mbedtls_mpi *P,
|
|
mbedtls_mpi *G)
|
|
{
|
|
const unsigned char *dhm_P = NULL;
|
|
const unsigned char *dhm_G = NULL;
|
|
size_t dhm_size_P = 0;
|
|
size_t dhm_size_G = 0;
|
|
int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
|
|
|
|
if (P == NULL && G == NULL) {
|
|
return PSA_ERROR_INVALID_ARGUMENT;
|
|
}
|
|
|
|
static const unsigned char dhm_P_2048[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE2048_P_BIN;
|
|
static const unsigned char dhm_P_3072[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE3072_P_BIN;
|
|
static const unsigned char dhm_P_4096[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE4096_P_BIN;
|
|
static const unsigned char dhm_P_6144[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE6144_P_BIN;
|
|
static const unsigned char dhm_P_8192[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE8192_P_BIN;
|
|
static const unsigned char dhm_G_2048[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE2048_G_BIN;
|
|
static const unsigned char dhm_G_3072[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE3072_G_BIN;
|
|
static const unsigned char dhm_G_4096[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE4096_G_BIN;
|
|
static const unsigned char dhm_G_6144[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE6144_G_BIN;
|
|
static const unsigned char dhm_G_8192[] =
|
|
MBEDTLS_DHM_RFC7919_FFDHE8192_G_BIN;
|
|
|
|
switch (key_size) {
|
|
case sizeof(dhm_P_2048):
|
|
dhm_P = dhm_P_2048;
|
|
dhm_G = dhm_G_2048;
|
|
dhm_size_P = sizeof(dhm_P_2048);
|
|
dhm_size_G = sizeof(dhm_G_2048);
|
|
break;
|
|
case sizeof(dhm_P_3072):
|
|
dhm_P = dhm_P_3072;
|
|
dhm_G = dhm_G_3072;
|
|
dhm_size_P = sizeof(dhm_P_3072);
|
|
dhm_size_G = sizeof(dhm_G_3072);
|
|
break;
|
|
case sizeof(dhm_P_4096):
|
|
dhm_P = dhm_P_4096;
|
|
dhm_G = dhm_G_4096;
|
|
dhm_size_P = sizeof(dhm_P_4096);
|
|
dhm_size_G = sizeof(dhm_G_4096);
|
|
break;
|
|
case sizeof(dhm_P_6144):
|
|
dhm_P = dhm_P_6144;
|
|
dhm_G = dhm_G_6144;
|
|
dhm_size_P = sizeof(dhm_P_6144);
|
|
dhm_size_G = sizeof(dhm_G_6144);
|
|
break;
|
|
case sizeof(dhm_P_8192):
|
|
dhm_P = dhm_P_8192;
|
|
dhm_G = dhm_G_8192;
|
|
dhm_size_P = sizeof(dhm_P_8192);
|
|
dhm_size_G = sizeof(dhm_G_8192);
|
|
break;
|
|
default:
|
|
return PSA_ERROR_INVALID_ARGUMENT;
|
|
}
|
|
|
|
if (P != NULL) {
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_read_binary(P, dhm_P,
|
|
dhm_size_P));
|
|
}
|
|
if (G != NULL) {
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_read_binary(G, dhm_G,
|
|
dhm_size_G));
|
|
}
|
|
|
|
cleanup:
|
|
if (ret != 0) {
|
|
return mbedtls_to_psa_error(ret);
|
|
}
|
|
|
|
return PSA_SUCCESS;
|
|
}
|
|
#endif /* MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_EXPORT ||
|
|
MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_GENERATE ||
|
|
MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_PUBLIC_KEY ||
|
|
MBEDTLS_PSA_BUILTIN_ALG_FFDH */
|
|
|
|
#if defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_EXPORT) || \
|
|
defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_PUBLIC_KEY)
|
|
psa_status_t mbedtls_psa_ffdh_export_public_key(
|
|
const psa_key_attributes_t *attributes,
|
|
const uint8_t *key_buffer,
|
|
size_t key_buffer_size,
|
|
uint8_t *data,
|
|
size_t data_size,
|
|
size_t *data_length)
|
|
{
|
|
int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
|
|
psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
|
|
mbedtls_mpi GX, G, X, P;
|
|
psa_key_type_t type = attributes->core.type;
|
|
|
|
if (PSA_KEY_TYPE_IS_PUBLIC_KEY(type)) {
|
|
if (key_buffer_size > data_size) {
|
|
return PSA_ERROR_BUFFER_TOO_SMALL;
|
|
}
|
|
memcpy(data, key_buffer, key_buffer_size);
|
|
memset(data + key_buffer_size, 0,
|
|
data_size - key_buffer_size);
|
|
*data_length = key_buffer_size;
|
|
return PSA_SUCCESS;
|
|
}
|
|
|
|
mbedtls_mpi_init(&GX); mbedtls_mpi_init(&G);
|
|
mbedtls_mpi_init(&X); mbedtls_mpi_init(&P);
|
|
|
|
size_t key_len = PSA_BITS_TO_BYTES(attributes->core.bits);
|
|
|
|
status = mbedtls_psa_ffdh_set_prime_generator(key_len, &P, &G);
|
|
|
|
if (status != PSA_SUCCESS) {
|
|
goto cleanup;
|
|
}
|
|
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_read_binary(&X, key_buffer,
|
|
key_buffer_size));
|
|
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_exp_mod(&GX, &G, &X, &P, NULL));
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary(&GX, data, key_len));
|
|
|
|
*data_length = key_len;
|
|
|
|
ret = 0;
|
|
cleanup:
|
|
mbedtls_mpi_free(&P); mbedtls_mpi_free(&G);
|
|
mbedtls_mpi_free(&X); mbedtls_mpi_free(&GX);
|
|
|
|
if (status == PSA_SUCCESS && ret != 0) {
|
|
status = mbedtls_to_psa_error(ret);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
#endif /* MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_EXPORT ||
|
|
MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_PUBLIC_KEY */
|
|
|
|
#if defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_GENERATE)
|
|
psa_status_t mbedtls_psa_ffdh_generate_key(
|
|
const psa_key_attributes_t *attributes,
|
|
uint8_t *key_buffer, size_t key_buffer_size, size_t *key_buffer_length)
|
|
{
|
|
mbedtls_mpi X, P;
|
|
int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
|
|
psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
|
|
mbedtls_mpi_init(&P); mbedtls_mpi_init(&X);
|
|
(void) attributes;
|
|
|
|
status = mbedtls_psa_ffdh_set_prime_generator(key_buffer_size, &P, NULL);
|
|
|
|
if (status != PSA_SUCCESS) {
|
|
goto cleanup;
|
|
}
|
|
|
|
/* RFC7919: Traditional finite field Diffie-Hellman has each peer choose their
|
|
secret exponent from the range [2, P-2].
|
|
Select random value in range [3, P-1] and decrease it by 1. */
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_random(&X, 3, &P, mbedtls_psa_get_random,
|
|
MBEDTLS_PSA_RANDOM_STATE));
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_sub_int(&X, &X, 1));
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary(&X, key_buffer, key_buffer_size));
|
|
*key_buffer_length = key_buffer_size;
|
|
|
|
cleanup:
|
|
mbedtls_mpi_free(&P); mbedtls_mpi_free(&X);
|
|
if (status == PSA_SUCCESS && ret != 0) {
|
|
return mbedtls_to_psa_error(ret);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
#endif /* MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_GENERATE */
|
|
|
|
#if defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_IMPORT)
|
|
psa_status_t mbedtls_psa_ffdh_import_key(
|
|
const psa_key_attributes_t *attributes,
|
|
const uint8_t *data, size_t data_length,
|
|
uint8_t *key_buffer, size_t key_buffer_size,
|
|
size_t *key_buffer_length, size_t *bits)
|
|
{
|
|
(void) attributes;
|
|
|
|
if (key_buffer_size < data_length) {
|
|
return PSA_ERROR_BUFFER_TOO_SMALL;
|
|
}
|
|
memcpy(key_buffer, data, data_length);
|
|
*key_buffer_length = data_length;
|
|
*bits = PSA_BYTES_TO_BITS(data_length);
|
|
|
|
return PSA_SUCCESS;
|
|
}
|
|
#endif /* MBEDTLS_PSA_BUILTIN_KEY_TYPE_DH_KEY_PAIR_IMPORT */
|
|
|
|
#if defined(MBEDTLS_PSA_BUILTIN_ALG_FFDH)
|
|
psa_status_t mbedtls_psa_ffdh_key_agreement(
|
|
const psa_key_attributes_t *attributes,
|
|
const uint8_t *peer_key,
|
|
size_t peer_key_length,
|
|
const uint8_t *key_buffer,
|
|
size_t key_buffer_size,
|
|
uint8_t *shared_secret,
|
|
size_t shared_secret_size,
|
|
size_t *shared_secret_length)
|
|
{
|
|
int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
|
|
psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
|
|
mbedtls_mpi P, G, X, GY, K;
|
|
const size_t calculated_shared_secret_size = peer_key_length;
|
|
|
|
if (peer_key_length != key_buffer_size ||
|
|
calculated_shared_secret_size > shared_secret_size) {
|
|
return PSA_ERROR_INVALID_ARGUMENT;
|
|
}
|
|
|
|
if (!PSA_KEY_TYPE_IS_DH_KEY_PAIR(psa_get_key_type(attributes))) {
|
|
return PSA_ERROR_INVALID_ARGUMENT;
|
|
}
|
|
|
|
mbedtls_mpi_init(&P); mbedtls_mpi_init(&G);
|
|
mbedtls_mpi_init(&X); mbedtls_mpi_init(&GY);
|
|
mbedtls_mpi_init(&K);
|
|
|
|
status = mbedtls_psa_ffdh_set_prime_generator(
|
|
PSA_BITS_TO_BYTES(attributes->core.bits), &P, &G);
|
|
|
|
if (status != PSA_SUCCESS) {
|
|
goto cleanup;
|
|
}
|
|
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_read_binary(&X, key_buffer,
|
|
key_buffer_size));
|
|
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_read_binary(&GY, peer_key,
|
|
peer_key_length));
|
|
|
|
/* Calculate shared secret public key: K = G^(XY) mod P = GY^X mod P */
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_exp_mod(&K, &GY, &X, &P, NULL));
|
|
|
|
MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary(&K, shared_secret,
|
|
calculated_shared_secret_size));
|
|
|
|
*shared_secret_length = calculated_shared_secret_size;
|
|
|
|
ret = 0;
|
|
|
|
cleanup:
|
|
mbedtls_mpi_free(&P); mbedtls_mpi_free(&G);
|
|
mbedtls_mpi_free(&X); mbedtls_mpi_free(&GY);
|
|
mbedtls_mpi_free(&K);
|
|
|
|
if (status == PSA_SUCCESS && ret != 0) {
|
|
status = mbedtls_to_psa_error(ret);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
#endif /* MBEDTLS_PSA_BUILTIN_ALG_FFDH */
|
|
|
|
#endif /* MBEDTLS_PSA_CRYPTO_C */
|