commit 30ebe11f869ef6c04396e77bc091ca2d31a45c17
author Valerio Setti <vsetti@baylibre.com> Thu Nov 17 16:23:34 2022 +0100
committer Valerio Setti <vsetti@baylibre.com> Thu Nov 17 16:35:02 2022 +0100
tree a94602f6d620423953d16a2596869c12a33b7ce9
parent a98836476782db7fc3883b02c25707fe3da2cd3a

tls: psa_pake: add a check on read size on both rounds

Signed-off-by: Valerio Setti <vsetti@baylibre.com>

library/ssl_tls.c

diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index 35262cb..06a5ec5 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -8231,6 +8231,9 @@
         }
     }
 
+    if ( input_offset != len )
+        return PSA_ERROR_INVALID_ARGUMENT;
+
     return( 0 );
 }
 
@@ -8286,6 +8289,9 @@
         input_offset += length;
     }
 
+    if ( input_offset != len )
+        return PSA_ERROR_INVALID_ARGUMENT;
+
     return( 0 );
 }