commit 5dbe17de36aea3646f9c0e5ce56aac3aa92dc41b
author David Horstmann <david.horstmann@arm.com> Tue Jun 27 10:30:28 2023 +0100
committer David Horstmann <david.horstmann@arm.com> Tue Jun 27 10:30:28 2023 +0100
tree 02e99e08b2e5090ca29362092a616941b680c35e
parent a5f7de1df2cb7cc3aa776e6053d4658c1c9eb946

Add PSA_JPAKE_FINISHED to EXPECTED_{IN,OUT}PUTS()

Signed-off-by: David Horstmann <david.horstmann@arm.com>

include/psa/crypto_extra.h

diff --git a/include/psa/crypto_extra.h b/include/psa/crypto_extra.h
index a7d98a0..94def5c 100644
--- a/include/psa/crypto_extra.h
+++ b/include/psa/crypto_extra.h
@@ -2024,8 +2024,10 @@
     psa_pake_step_t MBEDTLS_PRIVATE(step);
 };
 
-#define PSA_JPAKE_EXPECTED_INPUTS(round) (((round) == PSA_JPAKE_FIRST) ? 2 : 1)
-#define PSA_JPAKE_EXPECTED_OUTPUTS(round) (((round) == PSA_JPAKE_FIRST) ? 2 : 1)
+#define PSA_JPAKE_EXPECTED_INPUTS(round) ((round) == PSA_JPAKE_FINISHED ? 0 : \
+                                          ((round) == PSA_JPAKE_FIRST ? 2 : 1))
+#define PSA_JPAKE_EXPECTED_OUTPUTS(round) ((round) == PSA_JPAKE_FINISHED ? 0 : \
+                                           ((round) == PSA_JPAKE_FIRST ? 2 : 1))
 
 struct psa_pake_operation_s {
     /** Unique ID indicating which driver got assigned to do the