commit 0d001ef3da8461d8cf529371843287bd54bc9740
author Neil Armstrong <narmstrong@baylibre.com> Wed Jun 08 17:42:52 2022 +0200
committer Neil Armstrong <narmstrong@baylibre.com> Wed Aug 31 10:49:18 2022 +0200
tree ad428aad31d30c11df488ef95d6c77cc32ef1be9
parent 645cccd6a836d5dcb0f523da3b4e6550260e44c8

Check more parameters of psa_pake_output/psa_pake_input

Signed-off-by: Neil Armstrong <narmstrong@baylibre.com>

library/psa_crypto_pake.c

diff --git a/library/psa_crypto_pake.c b/library/psa_crypto_pake.c
index b8a08a1..7f147b7 100644
--- a/library/psa_crypto_pake.c
+++ b/library/psa_crypto_pake.c
@@ -328,6 +328,9 @@
         step != PSA_PAKE_STEP_ZK_PROOF )
         return( PSA_ERROR_INVALID_ARGUMENT );
 
+    if( output == NULL || output_size == 0 || output_length == NULL )
+        return( PSA_ERROR_INVALID_ARGUMENT );
+
 #if defined(MBEDTLS_PSA_BUILTIN_ALG_ECJPAKE)
     if( operation->alg == PSA_ALG_JPAKE )
     {
@@ -508,6 +511,9 @@
         step != PSA_PAKE_STEP_ZK_PROOF )
         return( PSA_ERROR_INVALID_ARGUMENT );
 
+    if( input == NULL || input_length == 0 )
+        return( PSA_ERROR_INVALID_ARGUMENT );
+
 #if defined(MBEDTLS_PSA_BUILTIN_ALG_ECJPAKE)
     if( operation->alg == PSA_ALG_JPAKE )
     {