Mbed-TLS/mbedtls-framework
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls-framework.git synced 2026-08-05 02:37:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #82 from Mbed-TLS/dev/paul-elliott-arm/add_iop_key_agreement_exercise_key

Browse Source 
Add iop key agreement to psa exercise key
This commit is contained in:
Paul Elliott
2024-12-18 16:34:30 +00:00
committed by GitHub
parent 7e6a3d255d dc256d0934
commit b9065f1717
1 changed files with 41 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/src/psa_exercise_key.c
+41
View File
@@ -703,6 +703,8 @@ psa_status_t mbedtls_test_psa_raw_key_agreement_with_self(
mbedtls_svc_key_id_t shared_secret_id = MBEDTLS_SVC_KEY_ID_INIT;
psa_key_attributes_t shared_secret_attributes = PSA_KEY_ATTRIBUTES_INIT;
psa_key_agreement_iop_t iop_operation = psa_key_agreement_iop_init();
#endif /* MBEDTLS_VERSION_MAJOR >= 4 */
psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
@@ -779,6 +781,44 @@ psa_status_t mbedtls_test_psa_raw_key_agreement_with_self(
PSA_ASSERT(status);
}
#if defined(MBEDTLS_ECP_RESTARTABLE) && defined(MBEDTLS_PSA_BUILTIN_ALG_ECDH)
psa_destroy_key(shared_secret_id);
if (PSA_KEY_TYPE_IS_ECC(private_key_type)) {
psa_interruptible_set_max_ops(1);
status = psa_key_agreement_iop_setup(&iop_operation, key, public_key,
public_key_length, alg,
&shared_secret_attributes);
if (key_destroyable && status == PSA_ERROR_INVALID_HANDLE) {
/* The key has been destroyed. */
status = PSA_SUCCESS;
goto exit;
} else if (status == PSA_SUCCESS) {
do {
status = psa_key_agreement_iop_complete(&iop_operation, &shared_secret_id);
} while (status == PSA_OPERATION_INCOMPLETE);
if (key_destroyable && status == PSA_ERROR_INVALID_HANDLE) {
/* The key has been destroyed. */
status = PSA_SUCCESS;
}
PSA_ASSERT(status);
}
} else {
TEST_EQUAL(psa_key_agreement_iop_setup(&iop_operation, key, public_key,
public_key_length, alg,
&shared_secret_attributes),
PSA_ERROR_INVALID_ARGUMENT);
}
#endif // defined(MBEDTLS_ECP_RESTARTABLE) && defined(MBEDTLS_PSA_BUILTIN_ALG_ECDH)
#endif /* MBEDTLS_VERSION_MAJOR >= 4 */
exit:
@@ -796,6 +836,7 @@ exit:
psa_destroy_key(shared_secret_id);
mbedtls_free(exported);
psa_key_agreement_iop_abort(&iop_operation);
#endif /* MBEDTLS_VERSION_MAJOR >= 4 */
mbedtls_free(public_key);