mbedtls_psa_ffdh_export_public_key: allow bigger output buffer
Signed-off-by: Przemek Stekiel <przemyslaw.stekiel@mobica.com>
This commit is contained in:
parent
da4fba64b8
commit
a4700fa69d
1 changed files with 5 additions and 3 deletions
|
@ -150,7 +150,9 @@ psa_status_t mbedtls_psa_ffdh_export_public_key(
|
||||||
mbedtls_mpi_init(&GX); mbedtls_mpi_init(&G);
|
mbedtls_mpi_init(&GX); mbedtls_mpi_init(&G);
|
||||||
mbedtls_mpi_init(&X); mbedtls_mpi_init(&P);
|
mbedtls_mpi_init(&X); mbedtls_mpi_init(&P);
|
||||||
|
|
||||||
status = mbedtls_psa_ffdh_set_prime_generator(data_size, &P, &G);
|
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) {
|
if (status != PSA_SUCCESS) {
|
||||||
goto cleanup;
|
goto cleanup;
|
||||||
|
@ -160,9 +162,9 @@ psa_status_t mbedtls_psa_ffdh_export_public_key(
|
||||||
key_buffer_size));
|
key_buffer_size));
|
||||||
|
|
||||||
MBEDTLS_MPI_CHK(mbedtls_mpi_exp_mod(&GX, &G, &X, &P, NULL));
|
MBEDTLS_MPI_CHK(mbedtls_mpi_exp_mod(&GX, &G, &X, &P, NULL));
|
||||||
MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary(&GX, data, data_size));
|
MBEDTLS_MPI_CHK(mbedtls_mpi_write_binary(&GX, data, key_len));
|
||||||
|
|
||||||
*data_length = data_size;
|
*data_length = key_len;
|
||||||
|
|
||||||
ret = 0;
|
ret = 0;
|
||||||
cleanup:
|
cleanup:
|
||||||
|
|
Loading…
Reference in a new issue