commit fbe742b2d0ffe63ea3443b198e96b685bf388ba5
author Thomas Daubney <thomas.daubney@arm.com> Fri Jul 28 16:17:38 2023 +0100
committer Thomas Daubney <thomas.daubney@arm.com> Fri Jul 28 16:17:38 2023 +0100
tree 9a6856a07192af932c92fb6ceb5fb2d028553d94
parent ce14124f7c47b3955e43a34408cc7826763e5b4e

Add extra check to one-shot operation results

Signed-off-by: Thomas Daubney <thomas.daubney@arm.com>

programs/psa/psa_hash.c

diff --git a/programs/psa/psa_hash.c b/programs/psa/psa_hash.c
index ae4e9a5..f73e1f3 100644
--- a/programs/psa/psa_hash.c
+++ b/programs/psa/psa_hash.c
@@ -148,7 +148,7 @@
         return EXIT_FAILURE;
     }
 
-    if (memcmp(hash, sample_hash, sample_hash_len) != 0)
+    if (memcmp(hash, sample_hash, sample_hash_len) != 0 || hash_length != sample_hash_len)
     {
         mbedtls_printf("One-shot hash operation gave the wrong result!\n\n");
         psa_hash_abort(&hash_operation);