commit 097147540d7b0fb4526d63334a6c2fd6beee35b9
author Yanray Wang <yanray.wang@arm.com> Wed Mar 01 16:31:46 2023 +0800
committer Yanray Wang <yanray.wang@arm.com> Wed Mar 15 16:42:58 2023 +0800
tree 87562791ecca66f29ff874f9920e1d853ed470f0
parent 12cb39661cd8f2daceabc711c2d985c81a53a010

test_suite_rsa.function: add tests

Signed-off-by: Yanray Wang <yanray.wang@arm.com>

tests/suites/test_suite_rsa.function

diff --git a/tests/suites/test_suite_rsa.function b/tests/suites/test_suite_rsa.function
index a2fe6c8..deb156e 100644
--- a/tests/suites/test_suite_rsa.function
+++ b/tests/suites/test_suite_rsa.function
@@ -26,11 +26,17 @@
                                        MBEDTLS_MD_NONE),
                MBEDTLS_ERR_RSA_INVALID_PADDING);
 
+    TEST_EQUAL(mbedtls_rsa_get_padding_mode(&ctx), MBEDTLS_RSA_PKCS_V15);
+    TEST_EQUAL(mbedtls_rsa_get_hash_id(&ctx), MBEDTLS_MD_NONE);
+
     TEST_EQUAL(mbedtls_rsa_set_padding(&ctx,
                                        MBEDTLS_RSA_PKCS_V21,
                                        invalid_hash_id),
                MBEDTLS_ERR_RSA_INVALID_PADDING);
 
+    TEST_EQUAL(mbedtls_rsa_get_padding_mode(&ctx), MBEDTLS_RSA_PKCS_V15);
+    TEST_EQUAL(mbedtls_rsa_get_hash_id(&ctx), MBEDTLS_MD_NONE);
+
     TEST_EQUAL(mbedtls_rsa_pkcs1_sign(&ctx, NULL,
                                       NULL, MBEDTLS_MD_NONE,
                                       buf_len,
@@ -58,6 +64,9 @@
                                        MBEDTLS_RSA_PKCS_V15,
                                        MBEDTLS_MD_NONE),
                MBEDTLS_ERR_RSA_INVALID_PADDING);
+
+    TEST_EQUAL(mbedtls_rsa_get_padding_mode(&ctx), MBEDTLS_RSA_PKCS_V15);
+    TEST_EQUAL(mbedtls_rsa_get_hash_id(&ctx), MBEDTLS_MD_NONE);
 #endif
 
 #if defined(MBEDTLS_PKCS1_V15)
@@ -91,6 +100,9 @@
                                        MBEDTLS_RSA_PKCS_V21,
                                        MBEDTLS_MD_NONE),
                MBEDTLS_ERR_RSA_INVALID_PADDING);
+
+    TEST_EQUAL(mbedtls_rsa_get_padding_mode(&ctx), MBEDTLS_RSA_PKCS_V15);
+    TEST_EQUAL(mbedtls_rsa_get_hash_id(&ctx), MBEDTLS_MD_NONE);
 #endif
 
 #if defined(MBEDTLS_PKCS1_V21)
@@ -159,6 +171,25 @@
 /* END_CASE */
 
 /* BEGIN_CASE */
+void rsa_ctx_param_validation(int padding_mode, int hash_id)
+{
+    mbedtls_rsa_context ctx;
+
+    mbedtls_rsa_init(&ctx);
+
+    TEST_ASSERT(mbedtls_rsa_set_padding(&ctx,
+                                        padding_mode,
+                                        hash_id) == 0);
+
+    TEST_ASSERT(mbedtls_rsa_get_padding_mode(&ctx) == padding_mode);
+    TEST_ASSERT(mbedtls_rsa_get_hash_id(&ctx) == hash_id);
+
+exit:
+    mbedtls_rsa_free(&ctx);
+}
+/* END_CASE */
+
+/* BEGIN_CASE */
 void mbedtls_rsa_pkcs1_sign(data_t *message_str, int padding_mode,
                             int digest, int mod, char *input_P,
                             char *input_Q, char *input_N, char *input_E,