commit c1152e4a0f1f12f95f446798a2b91a2225fea19f
author Neil Armstrong <narmstrong@baylibre.com> Tue Mar 22 10:29:06 2022 +0100
committer Neil Armstrong <narmstrong@baylibre.com> Thu Apr 07 15:01:24 2022 +0200
tree 0d3dcb9b9d2ada80f6d270802d62eb05035432ee
parent 7e1b4a45fa44ef8ed90853447473afe3f90ab474

Handle and return translated PSA errors in mbedtls_pk_wrap_as_opaque()

Signed-off-by: Neil Armstrong <narmstrong@baylibre.com>

library/pk.c

diff --git a/library/pk.c b/library/pk.c
index 0ba334b..837d6a2 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -725,6 +725,7 @@
         psa_key_type_t key_type;
         size_t bits;
         int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
+        psa_status_t status;
 
         /* export the private key material in the format PSA wants */
         ec = mbedtls_pk_ec( *pk );
@@ -744,8 +745,9 @@
         psa_set_key_enrollment_algorithm( &attributes, PSA_ALG_ECDH );
 
         /* import private key into PSA */
-        if( PSA_SUCCESS != psa_import_key( &attributes, d, d_len, key ) )
-            return( MBEDTLS_ERR_PLATFORM_HW_ACCEL_FAILED );
+        status = psa_import_key( &attributes, d, d_len, key );
+        if( status != PSA_SUCCESS )
+            return( mbedtls_pk_error_from_psa( status ) );
 
         /* make PK context wrap the key slot */
         mbedtls_pk_free( pk );
@@ -783,7 +785,7 @@
         mbedtls_platform_zeroize( buf, sizeof( buf ) );
 
         if( status != PSA_SUCCESS )
-            return( MBEDTLS_ERR_PLATFORM_HW_ACCEL_FAILED );
+            return( mbedtls_pk_error_from_psa( status ) );
 
         /* make PK context wrap the key slot */
         mbedtls_pk_free( pk );