commit 8a3d2348596d3972eb57a759bce1f6c799dc4066
author Gilles Peskine <Gilles.Peskine@arm.com> Thu Dec 03 21:06:15 2020 +0100
committer Manuel Pégourié-Gonnard <manuel.pegourie-gonnard@arm.com> Mon Dec 07 10:02:13 2020 +0100
tree 3f384b3ea787be207c6f490c9fc23809cc339ff6
parent a2971ea62cfdd930ec89ab99f00800550d50398b

Fail the test case immediately if cipher_reset_key fails

Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>

tests/suites/test_suite_cipher.function

diff --git a/tests/suites/test_suite_cipher.function b/tests/suites/test_suite_cipher.function
index 5e9a1e3..1d98f3d 100644
--- a/tests/suites/test_suite_cipher.function
+++ b/tests/suites/test_suite_cipher.function
@@ -27,7 +27,7 @@
  * individual ciphers, and it doesn't work with the PSA wrappers. So don't do
  * it, and instead start with a fresh context.
  */
-static void cipher_reset_key( mbedtls_cipher_context_t *ctx, int cipher_id,
+static int cipher_reset_key( mbedtls_cipher_context_t *ctx, int cipher_id,
         int use_psa, size_t tag_len, const data_t *key, int direction )
 {
     mbedtls_cipher_free( ctx );
@@ -52,8 +52,10 @@
 
     TEST_ASSERT( 0 == mbedtls_cipher_setkey( ctx, key->x, 8 * key->len,
                                              direction ) );
+    return( 1 );
+
 exit:
-    ;
+    return( 0 );
 }
 
 /*
@@ -1195,8 +1197,9 @@
     /*
      * Prepare context for decryption
      */
-    cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
-                      MBEDTLS_DECRYPT );
+    if( ! cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
+                            MBEDTLS_DECRYPT ) )
+        goto exit;
 
     /*
      * prepare buffer for decryption
@@ -1264,8 +1267,9 @@
     if( strcmp( result, "FAIL" ) != 0 )
     {
         /* prepare context for encryption */
-        cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
-                          MBEDTLS_ENCRYPT );
+        if( ! cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
+                                MBEDTLS_ENCRYPT ) )
+            goto exit;
 
         /*
          * Compute size of output buffer according to documentation
@@ -1327,8 +1331,9 @@
     /*
      * Prepare context for decryption
      */
-    cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
-                      MBEDTLS_DECRYPT );
+    if( ! cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
+                            MBEDTLS_DECRYPT ) )
+        goto exit;
 
     /*
      * Prepare pointers for decryption
@@ -1387,8 +1392,9 @@
     if( strcmp( result, "FAIL" ) != 0 )
     {
         /* prepare context for encryption */
-        cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
-                          MBEDTLS_ENCRYPT );
+        if( ! cipher_reset_key( &ctx, cipher_id, use_psa, tag->len, key,
+                                MBEDTLS_ENCRYPT ) )
+            goto exit;
 
         /* prepare buffers for encryption */
 #if defined(MBEDTLS_USE_PSA_CRYPTO)