mbedtls_psa_ffdh_set_prime_generator: use switch instead if-else

Signed-off-by: Przemek Stekiel <przemyslaw.stekiel@mobica.com>
This commit is contained in:
Przemek Stekiel 2023-04-28 14:20:27 +02:00
parent 4ce523256b
commit 75095cce74

View File

@ -64,33 +64,39 @@ static psa_status_t mbedtls_psa_ffdh_set_prime_generator(size_t key_size,
static const unsigned char dhm_G_8192[] = static const unsigned char dhm_G_8192[] =
MBEDTLS_DHM_RFC7919_FFDHE8192_G_BIN; MBEDTLS_DHM_RFC7919_FFDHE8192_G_BIN;
if (key_size == sizeof(dhm_P_2048)) { switch(key_size) {
dhm_P = dhm_P_2048; case sizeof(dhm_P_2048):
dhm_G = dhm_G_2048; dhm_P = dhm_P_2048;
dhm_size_P = sizeof(dhm_P_2048); dhm_G = dhm_G_2048;
dhm_size_G = sizeof(dhm_G_2048); dhm_size_P = sizeof(dhm_P_2048);
} else if (key_size == sizeof(dhm_P_3072)) { dhm_size_G = sizeof(dhm_G_2048);
dhm_P = dhm_P_3072; break;
dhm_G = dhm_G_3072; case sizeof(dhm_P_3072):
dhm_size_P = sizeof(dhm_P_3072); dhm_P = dhm_P_3072;
dhm_size_G = sizeof(dhm_G_3072); dhm_G = dhm_G_3072;
} else if (key_size == sizeof(dhm_P_4096)) { dhm_size_P = sizeof(dhm_P_3072);
dhm_P = dhm_P_4096; dhm_size_G = sizeof(dhm_G_3072);
dhm_G = dhm_G_4096; break;
dhm_size_P = sizeof(dhm_P_4096); case sizeof(dhm_P_4096):
dhm_size_G = sizeof(dhm_G_4096); dhm_P = dhm_P_4096;
} else if (key_size == sizeof(dhm_P_6144)) { dhm_G = dhm_G_4096;
dhm_P = dhm_P_6144; dhm_size_P = sizeof(dhm_P_4096);
dhm_G = dhm_G_6144; dhm_size_G = sizeof(dhm_G_4096);
dhm_size_P = sizeof(dhm_P_6144); break;
dhm_size_G = sizeof(dhm_G_6144); case sizeof(dhm_P_6144):
} else if (key_size == sizeof(dhm_P_8192)) { dhm_P = dhm_P_6144;
dhm_P = dhm_P_8192; dhm_G = dhm_G_6144;
dhm_G = dhm_G_8192; dhm_size_P = sizeof(dhm_P_6144);
dhm_size_P = sizeof(dhm_P_8192); dhm_size_G = sizeof(dhm_G_6144);
dhm_size_G = sizeof(dhm_G_8192); break;
} else { case sizeof(dhm_P_8192):
return PSA_ERROR_INVALID_ARGUMENT; 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) { if (P != NULL) {