commit 0050b918f03f27bc01b1c32c47ddeb0533568cb0
author Tuvshinzaya Erdenekhuu <tuvshinzaya.erdenekhuu@arm.com> Tue Aug 30 13:15:55 2022 +0100
committer Tuvshinzaya Erdenekhuu <tuvshinzaya.erdenekhuu@arm.com> Wed Aug 31 10:14:25 2022 +0100
tree 69858aa85b88679132471f2a11a5c749ba3da113
parent ed99ed34f3a108ad61aaf9a1c5afd0562b733186

Added new tests in pk_invalid_param

Signed-off-by: Tuvshinzaya Erdenekhuu <tuvshinzaya.erdenekhuu@arm.com>

tests/suites/test_suite_pk.function

diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function
index 6268225..2c08a5e 100644
--- a/tests/suites/test_suite_pk.function
+++ b/tests/suites/test_suite_pk.function
@@ -303,7 +303,6 @@
 void pk_invalid_param()
 {
     mbedtls_pk_context ctx;
-    mbedtls_md_type_t md_alg_none = MBEDTLS_MD_NONE;
     mbedtls_pk_type_t pk_type = 0;
     unsigned char buf[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
     size_t buf_size = sizeof( buf );
@@ -311,21 +310,37 @@
     mbedtls_pk_init( &ctx );
 
     TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
-                mbedtls_pk_verify_restartable( &ctx, md_alg_none,
+                mbedtls_pk_verify_restartable( &ctx, MBEDTLS_MD_NONE,
                                                NULL, buf_size,
                                                buf, buf_size,
                                                NULL ) );
     TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
+                mbedtls_pk_verify_restartable( &ctx, MBEDTLS_MD_SHA256,
+                                               NULL, 0,
+                                               buf, buf_size,
+                                               NULL ) );
+    TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
                 mbedtls_pk_verify_ext( pk_type, NULL,
-                                       &ctx, md_alg_none,
+                                       &ctx, MBEDTLS_MD_NONE,
                                        NULL, buf_size,
                                        buf, buf_size ) );
     TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
-                mbedtls_pk_sign_restartable( &ctx, md_alg_none,
+                mbedtls_pk_verify_ext( pk_type, NULL,
+                                       &ctx, MBEDTLS_MD_SHA256,
+                                       NULL, 0,
+                                       buf, buf_size ) );
+    TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
+                mbedtls_pk_sign_restartable( &ctx, MBEDTLS_MD_NONE,
                                              NULL, buf_size,
                                              buf, buf_size, &buf_size,
                                              NULL, NULL,
                                              NULL ) );
+    TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
+                mbedtls_pk_sign_restartable( &ctx, MBEDTLS_MD_SHA256,
+                                             NULL, 0,
+                                             buf, buf_size, &buf_size,
+                                             NULL, NULL,
+                                             NULL ) );
 exit:
     mbedtls_pk_free( &ctx );
 }