commit b093650033a2ede358f17d599668fd3746b10ace
author Andrzej Kurek <andrzej.kurek@arm.com> Fri Sep 16 07:13:00 2022 -0400
committer Andrzej Kurek <andrzej.kurek@arm.com> Fri Sep 16 07:13:00 2022 -0400
tree a18f1c62d1c3e97fc7e90d66ac3d941a22e43458
parent d37850404adfb8064f6e590197983c7c852daa30

Add proper capacity calculation for EC J-PAKE to PMS KDF

Signed-off-by: Andrzej Kurek <andrzej.kurek@arm.com>

library/psa_crypto.c

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index cbdc912..981b6f8 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -5175,7 +5175,8 @@
         return( PSA_ERROR_NOT_SUPPORTED );
     }
 #if defined(MBEDTLS_PSA_BUILTIN_ALG_HKDF_EXTRACT)
-    if( PSA_ALG_IS_HKDF_EXTRACT( kdf_alg ) )
+    if( PSA_ALG_IS_HKDF_EXTRACT( kdf_alg ) ||
+        PSA_ALG_IS_TLS12_ECJPAKE_TO_PMS ( kdf_alg ))
         operation->capacity = hash_size;
     else
 #endif /* MBEDTLS_PSA_BUILTIN_ALG_HKDF_EXTRACT */