commit f3e696dc1be31b80089a4e26b4d1af099179f25a
author Kusumit Ghoderao <Kusumit.Ghoderao@silabs.com> Fri Jul 28 13:30:50 2023 +0530
committer Kusumit Ghoderao <Kusumit.Ghoderao@silabs.com> Fri Jul 28 13:30:50 2023 +0530
tree 2b4ba1d342c15b0839ba615976a00fbabc5a6b62
parent be55b7e45a45a737727f20d8755edda6e718ee5f

Add ifdef for hmac and cmac set password

Signed-off-by: Kusumit Ghoderao <Kusumit.Ghoderao@silabs.com>

library/psa_crypto.c

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index ec99e16..947c2e2 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -6768,15 +6768,23 @@
         return PSA_ERROR_BAD_STATE;
     }
 
+#if defined(MBEDTLS_PSA_BUILTIN_ALG_PBKDF2_HMAC)
     if (PSA_ALG_IS_PBKDF2_HMAC(kdf_alg)) {
         psa_algorithm_t hash_alg = PSA_ALG_PBKDF2_HMAC_GET_HASH(kdf_alg);
         status = psa_pbkdf2_hmac_set_password(hash_alg, data, data_length,
                                               pbkdf2->password,
                                               &pbkdf2->password_length);
-    } else if (kdf_alg == PSA_ALG_PBKDF2_AES_CMAC_PRF_128) {
+    } else
+#endif /* MBEDTLS_PSA_BUILTIN_ALG_PBKDF2_HMAC */
+#if defined(MBEDTLS_PSA_BUILTIN_ALG_PBKDF2_AES_CMAC_PRF_128)
+    if (kdf_alg == PSA_ALG_PBKDF2_AES_CMAC_PRF_128) {
         status = psa_pbkdf2_cmac_set_password(data, data_length,
                                               pbkdf2->password,
                                               &pbkdf2->password_length);
+    } else
+#endif /* MBEDTLS_PSA_BUILTIN_ALG_PBKDF2_AES_CMAC_PRF_128 */
+    {
+        return PSA_ERROR_INVALID_ARGUMENT;
     }
 
     pbkdf2->state = PSA_PBKDF2_STATE_PASSWORD_SET;