commit 08b223443f5611e678f52fbb808d34be63b38e9a
author Tuvshinzaya Erdenekhuu <tuvshinzaya.erdenekhuu@arm.com> Thu Sep 01 16:18:00 2022 +0100
committer Tuvshinzaya Erdenekhuu <tuvshinzaya.erdenekhuu@arm.com> Thu Sep 01 16:18:00 2022 +0100
tree 24c415100bc953ff0b4cef44bfe8c8b8e8b724b9
parent fe7524de03914c1387818526e1dcecbdfca377f8

Add new tests

Signed-off-by: Tuvshinzaya Erdenekhuu <tuvshinzaya.erdenekhuu@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 5bba023..ad52e98 100644
--- a/tests/suites/test_suite_rsa.function
+++ b/tests/suites/test_suite_rsa.function
@@ -33,16 +33,27 @@
                 MBEDTLS_ERR_RSA_INVALID_PADDING );
 
     TEST_EQUAL( mbedtls_rsa_pkcs1_sign(&ctx, NULL,
-                                       NULL, md_alg_none,
+                                       NULL, MBEDTLS_MD_NONE,
                                        buf_len,
                                        NULL, buf),
                 MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 
-    TEST_EQUAL( mbedtls_rsa_pkcs1_verify(&ctx, md_alg_none,
+    TEST_EQUAL( mbedtls_rsa_pkcs1_sign(&ctx, NULL,
+                                       NULL, MBEDTLS_MD_SHA256,
+                                       0,
+                                       NULL, buf),
+                MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
+
+    TEST_EQUAL( mbedtls_rsa_pkcs1_verify(&ctx, MBEDTLS_MD_NONE,
                                          buf_len,
                                          NULL, buf),
                 MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 
+    TEST_EQUAL( mbedtls_rsa_pkcs1_verify(&ctx, MBEDTLS_MD_SHA256,
+                                         0,
+                                         NULL, buf),
+                MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
+
 #if !defined(MBEDTLS_PKCS1_V15)
     TEST_EQUAL( mbedtls_rsa_set_padding( &ctx,
                                          MBEDTLS_RSA_PKCS_V15,
@@ -52,16 +63,28 @@
 
 #if defined(MBEDTLS_PKCS1_V15)
     TEST_EQUAL( mbedtls_rsa_rsassa_pkcs1_v15_sign(&ctx, NULL,
-                                              NULL, md_alg_none,
+                                              NULL, MBEDTLS_MD_NONE,
                                               buf_len,
                                               NULL, buf),
                 MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 
-    TEST_EQUAL( mbedtls_rsa_rsassa_pkcs1_v15_verify(&ctx, md_alg_none,
+    TEST_EQUAL( mbedtls_rsa_rsassa_pkcs1_v15_sign(&ctx, NULL,
+                                              NULL, MBEDTLS_MD_SHA256,
+                                              0,
+                                              NULL, buf),
+                MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
+
+    TEST_EQUAL( mbedtls_rsa_rsassa_pkcs1_v15_verify(&ctx, MBEDTLS_MD_NONE,
                                                 buf_len,
                                                 NULL, buf),
                 MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 
+    TEST_EQUAL( mbedtls_rsa_rsassa_pkcs1_v15_verify(&ctx, MBEDTLS_MD_SHA256,
+                                                0,
+                                                NULL, buf),
+                MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
+
+
 #endif
 
 #if !defined(MBEDTLS_PKCS1_V21)
@@ -73,22 +96,38 @@
 
 #if defined(MBEDTLS_PKCS1_V21)
     TEST_EQUAL( mbedtls_rsa_rsassa_pss_sign_ext(&ctx, NULL, NULL,
-                                    md_alg_none, buf_len,
+                                    MBEDTLS_MD_NONE, buf_len,
                                     NULL, buf_len,
                                     buf ),
                 MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 
-    TEST_EQUAL( mbedtls_rsa_rsassa_pss_verify_ext(&ctx, md_alg_none,
+    TEST_EQUAL( mbedtls_rsa_rsassa_pss_sign_ext(&ctx, NULL, NULL,
+                                    MBEDTLS_MD_SHA256, 0,
+                                    NULL, buf_len,
+                                    buf ),
+                MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
+
+    TEST_EQUAL( mbedtls_rsa_rsassa_pss_verify_ext(&ctx, MBEDTLS_MD_NONE,
                                                   buf_len, NULL,
-                                                  md_alg_none,
+                                                  MBEDTLS_MD_NONE,
                                                   buf_len, buf),
                 MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 
-    TEST_EQUAL( mbedtls_rsa_rsassa_pss_verify(&ctx, md_alg_none,
+    TEST_EQUAL( mbedtls_rsa_rsassa_pss_verify_ext(&ctx, MBEDTLS_MD_SHA256,
+                                                  0, NULL,
+                                                  MBEDTLS_MD_NONE,
+                                                  buf_len, buf),
+                MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
+
+    TEST_EQUAL( mbedtls_rsa_rsassa_pss_verify(&ctx, MBEDTLS_MD_NONE,
                                               buf_len,
                                               NULL, buf),
                 MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 
+    TEST_EQUAL( mbedtls_rsa_rsassa_pss_verify(&ctx, MBEDTLS_MD_SHA256,
+                                              0,
+                                              NULL, buf),
+                MBEDTLS_ERR_RSA_BAD_INPUT_DATA );
 #endif
 
 exit: