From 0d001ef3da8461d8cf529371843287bd54bc9740 Mon Sep 17 00:00:00 2001 From: Neil Armstrong Date: Wed, 8 Jun 2022 17:42:52 +0200 Subject: [PATCH] Check more parameters of psa_pake_output/psa_pake_input Signed-off-by: Neil Armstrong --- library/psa_crypto_pake.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/library/psa_crypto_pake.c b/library/psa_crypto_pake.c index b8a08a1a07..7f147b7cca 100644 --- a/library/psa_crypto_pake.c +++ b/library/psa_crypto_pake.c @@ -328,6 +328,9 @@ psa_status_t psa_pake_output( psa_pake_operation_t *operation, 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 @@ psa_status_t psa_pake_input( psa_pake_operation_t *operation, 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 ) {