commit f1743e2440e7f5e3b702195aeaf846e2cec5e33b
author Paul Elliott <paul.elliott@arm.com> Wed Feb 15 18:44:16 2023 +0000
committer Paul Elliott <paul.elliott@arm.com> Thu Feb 16 12:15:39 2023 +0000
tree 5eb0b035935cdb07f9b3c8d203240d53103d4d4e
parent ba70ad49446e3e814cb3dc0413b3956ade0e3265

Add verify call to max ops tests

Signed-off-by: Paul Elliott <paul.elliott@arm.com>

tests/suites/test_suite_psa_crypto.function

diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index fd355de..5379eaf 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -7484,8 +7484,13 @@
     psa_key_type_t key_type = key_type_arg;
     psa_algorithm_t alg = alg_arg;
     psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
+    size_t key_bits;
+    unsigned char *signature = NULL;
+    size_t signature_size;
     psa_sign_hash_interruptible_operation_t sign_operation =
         psa_sign_hash_interruptible_operation_init();
+    psa_verify_hash_interruptible_operation_t verify_operation =
+        psa_verify_hash_interruptible_operation_init();
 
     PSA_ASSERT(psa_crypto_init());
 
@@ -7494,8 +7499,17 @@
     psa_set_key_algorithm(&attributes, alg);
     psa_set_key_type(&attributes, key_type);
 
-    PSA_ASSERT(psa_import_key(&attributes, key_data->x, key_data->len,
-                              &key));
+    PSA_ASSERT(psa_import_key(&attributes, key_data->x, key_data->len, &key));
+    PSA_ASSERT(psa_get_key_attributes(key, &attributes));
+    key_bits = psa_get_key_bits(&attributes);
+
+    /* Allocate a buffer which has the size advertised by the
+     * library. */
+    signature_size = PSA_SIGN_OUTPUT_SIZE(key_type, key_bits, alg);
+
+    TEST_ASSERT(signature_size != 0);
+    TEST_LE_U(signature_size, PSA_SIGNATURE_MAX_SIZE);
+    ASSERT_ALLOC(signature, signature_size);
 
     /* Check that default max ops gets set if we don't set it. */
     PSA_ASSERT(psa_sign_hash_start(&sign_operation, key, alg,
@@ -7506,12 +7520,20 @@
 
     PSA_ASSERT(psa_sign_hash_abort(&sign_operation));
 
+    PSA_ASSERT(psa_verify_hash_start(&verify_operation, key, alg,
+                                     input_data->x, input_data->len,
+                                     signature, signature_size));
+
+    TEST_EQUAL(psa_interruptible_get_max_ops(),
+               PSA_INTERRUPTIBLE_MAX_OPS_UNLIMITED);
+
+    PSA_ASSERT(psa_verify_hash_abort(&verify_operation));
+
     /* Check that max ops gets set properly. */
 
     psa_interruptible_set_max_ops(0xbeef);
 
-    TEST_EQUAL(psa_interruptible_get_max_ops(),
-               0xbeef);
+    TEST_EQUAL(psa_interruptible_get_max_ops(), 0xbeef);
 
 exit:
     /*
@@ -7521,6 +7543,7 @@
     psa_reset_key_attributes(&attributes);
 
     psa_destroy_key(key);
+    mbedtls_free(signature);
     PSA_DONE();
 }
 /* END_CASE */