commit 7a18f9645c94cfffcf78e06b2631daa88c4546ba
author Gilles Peskine <Gilles.Peskine@arm.com> Mon Feb 12 16:48:11 2024 +0100
committer Gilles Peskine <Gilles.Peskine@arm.com> Thu Feb 15 11:57:46 2024 +0100
tree 858c711d27bc7aebd98c5c641f00f4a8cdf1bc36
parent f0765fa06a093df0a931974faccbff3582941541

psa_generate_key_ext: RSA: support custom public exponent

Signed-off-by: Gilles Peskine <Gilles.Peskine@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 05b49bd..34baa1b 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -1293,7 +1293,13 @@
         TEST_EQUAL(p[1], 0);
         TEST_EQUAL(p[2], 1);
     } else {
-        TEST_MEMORY_COMPARE(p, len, e_arg->x, e_arg->len);
+        const uint8_t *expected = e_arg->x;
+        size_t expected_len = e_arg->len;
+        while (expected_len > 0 && *expected == 0) {
+            ++expected;
+            --expected_len;
+        }
+        TEST_MEMORY_COMPARE(p, len, expected, expected_len);
     }
     ok = 1;
 
@@ -9960,6 +9966,12 @@
     TEST_EQUAL(psa_get_key_type(&got_attributes), type);
     TEST_EQUAL(psa_get_key_bits(&got_attributes), bits);
 
+#if defined(PSA_WANT_KEY_TYPE_RSA_KEY_PAIR_GENERATE)
+    if (type == PSA_KEY_TYPE_RSA_KEY_PAIR) {
+        TEST_ASSERT(rsa_test_e(key, bits, method_data));
+    }
+#endif
+
     /* Do something with the key according to its type and permitted usage. */
     if (!mbedtls_test_psa_exercise_key(key, usage, alg)) {
         goto exit;