test_suite_pk: use predefined RSA keys in pk_setup_for_type()

Signed-off-by: Valerio Setti <valerio.setti@nordicsemi.no>
This commit is contained in:
Valerio Setti 2024-04-16 15:00:52 +02:00
parent 1f6dab7c24
commit 4d14581d87

View File

@ -498,20 +498,13 @@ static int pk_setup_for_type(mbedtls_pk_type_t pk_type, int want_pair,
#if defined(MBEDTLS_RSA_C) #if defined(MBEDTLS_RSA_C)
case MBEDTLS_PK_RSA: case MBEDTLS_PK_RSA:
{ {
TEST_EQUAL(mbedtls_pk_setup(pk, mbedtls_pk_info_from_type(pk_type)), 0);
*psa_type = PSA_KEY_TYPE_RSA_KEY_PAIR; *psa_type = PSA_KEY_TYPE_RSA_KEY_PAIR;
if (want_pair) { TEST_EQUAL(pk_setup(pk, pk_type, MBEDTLS_RSA_GEN_KEY_MIN_BITS), 0);
TEST_EQUAL(pk_setup(pk, pk_type, MBEDTLS_RSA_GEN_KEY_MIN_BITS), 0); if (!want_pair) {
} else {
unsigned char N[PSA_BITS_TO_BYTES(MBEDTLS_RSA_GEN_KEY_MIN_BITS)] = { 0xff };
N[sizeof(N) - 1] = 0x03;
const unsigned char E[1] = { 0x03 };
mbedtls_rsa_context *rsa = mbedtls_pk_rsa(*pk); mbedtls_rsa_context *rsa = mbedtls_pk_rsa(*pk);
TEST_EQUAL(mbedtls_rsa_import_raw(rsa, mbedtls_mpi_free(&rsa->D);
N, sizeof(N), mbedtls_mpi_free(&rsa->P);
NULL, 0, NULL, 0, NULL, 0, mbedtls_mpi_free(&rsa->Q);
E, sizeof(E)), 0);
TEST_EQUAL(mbedtls_rsa_complete(rsa), 0);
} }
break; break;
} }