From 0050b918f03f27bc01b1c32c47ddeb0533568cb0 Mon Sep 17 00:00:00 2001 From: Tuvshinzaya Erdenekhuu Date: Tue, 30 Aug 2022 13:15:55 +0100 Subject: [PATCH] Added new tests in pk_invalid_param Signed-off-by: Tuvshinzaya Erdenekhuu --- tests/suites/test_suite_pk.function | 23 +++++++++++++++++++---- 1 file changed, 19 insertions(+), 4 deletions(-) diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function index 62682256d1..2c08a5e799 100644 --- a/tests/suites/test_suite_pk.function +++ b/tests/suites/test_suite_pk.function @@ -303,7 +303,6 @@ exit: 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 @@ void pk_invalid_param() 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 ); }