mirror/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-01-30 15:32:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add testing of complete API of interruptible export public-key

Browse Source 
Signed-off-by: Waleed Elmelegy <waleed.elmelegy@arm.com>
This commit is contained in:
Waleed Elmelegy 2024-12-03 14:28:50 +00:00
parent a04e88adf0
commit e283ed9e20
1 changed files with 59 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
tf-psa-crypto/tests/suites/test_suite_psa_crypto.function
View File

@ -10449,6 +10449,11 @@ void iop_export_public_key(
psa_status_t expected_status = expected_status_arg;
psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
psa_export_public_key_iop_t export_key_operation = PSA_EXPORT_PUBLIC_KEY_IOP_INIT;
uint8_t output[PSA_KEY_EXPORT_ECC_PUBLIC_KEY_MAX_SIZE(PSA_VENDOR_ECC_MAX_CURVE_BITS)] = { 0 };
size_t output_len = 0;
uint8_t refrence_output[PSA_KEY_EXPORT_ECC_PUBLIC_KEY_MAX_SIZE(PSA_VENDOR_ECC_MAX_CURVE_BITS)] =
{ 0 };
size_t refrence_output_len = 0;
psa_status_t status;
PSA_ASSERT(psa_crypto_init());
@ -10465,6 +10470,15 @@ void iop_export_public_key(
status = psa_generate_key(&attributes, &iop_key);
TEST_EQUAL(status, PSA_SUCCESS);
/* Test calling complete() without calling setup() will fail. */
status = psa_export_public_key_iop_complete(&export_key_operation,
output,
sizeof(output),
&output_len);
TEST_EQUAL(status, PSA_ERROR_BAD_STATE);
PSA_ASSERT(psa_export_public_key_iop_abort(&export_key_operation));
status = psa_export_public_key_iop_setup(&export_key_operation, iop_key);
TEST_EQUAL(status, expected_status);
@ -10474,12 +10488,56 @@ void iop_export_public_key(
TEST_EQUAL(status, PSA_ERROR_BAD_STATE);
#endif
TEST_EQUAL(psa_export_public_key_iop_abort(&export_key_operation), PSA_SUCCESS);
PSA_ASSERT(psa_export_public_key_iop_abort(&export_key_operation));
/* Test that after calling abort operation is reset to it's fresh state */
status = psa_export_public_key_iop_setup(&export_key_operation, iop_key);
TEST_EQUAL(status, expected_status);
if (expected_status != PSA_SUCCESS) {
expected_status = PSA_ERROR_BAD_STATE;
}
do {
status = psa_export_public_key_iop_complete(&export_key_operation,
output,
sizeof(output),
&output_len);
} while (status == PSA_OPERATION_INCOMPLETE);
TEST_EQUAL(status, expected_status);
/* Test calling complete() 2 times consecutively will fail. */
status = psa_export_public_key_iop_complete(&export_key_operation,
output,
sizeof(output),
&output_len);
TEST_EQUAL(status, PSA_ERROR_BAD_STATE);
if (expected_status == PSA_SUCCESS) {
status = psa_export_public_key(iop_key,
refrence_output,
sizeof(refrence_output),
&refrence_output_len);
TEST_EQUAL(status, PSA_SUCCESS);
TEST_MEMORY_COMPARE(refrence_output, refrence_output_len, output, output_len);
/* Test psa_export_public_key_iop_complete() returns right error code when
output buffer is not enough. */
PSA_ASSERT(psa_export_public_key_iop_abort(&export_key_operation));
status = psa_export_public_key_iop_setup(&export_key_operation, iop_key);
TEST_EQUAL(status, PSA_SUCCESS);
do {
status = psa_export_public_key_iop_complete(&export_key_operation,
output,
refrence_output_len-1,
&output_len);
} while (status == PSA_OPERATION_INCOMPLETE);
TEST_EQUAL(status, PSA_ERROR_BUFFER_TOO_SMALL);
}
exit:
psa_export_public_key_iop_abort(&export_key_operation);
psa_destroy_key(iop_key);