commit e87804920a7100a7e12ff49fcb31962acc871422
author Neil Armstrong <narmstrong@baylibre.com> Thu Mar 03 16:54:16 2022 +0100
committer Neil Armstrong <narmstrong@baylibre.com> Thu Mar 03 16:54:16 2022 +0100
tree f8ded3b08f405dfee0128c30a68be7a620c02bfa
parent b556a426561ef676ca309d3c336aab0138c1a9a6

Use new PSA to mbedtls PK error mapping functions in rsa_decrypt_wrap()

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

library/pk_wrap.c

diff --git a/library/pk_wrap.c b/library/pk_wrap.c
index 3320078..aad56a4 100644
--- a/library/pk_wrap.c
+++ b/library/pk_wrap.c
@@ -266,7 +266,7 @@
                              &key_id );
     if( status != PSA_SUCCESS )
     {
-        ret = mbedtls_psa_err_translate_pk( status );
+        ret = mbedtls_pk_error_from_psa( status );
         goto cleanup;
     }
 
@@ -274,14 +274,7 @@
                                      NULL, 0, output, osize, olen );
     if( status != PSA_SUCCESS )
     {
-        if ( status == PSA_ERROR_INVALID_PADDING )
-        {
-            ret = MBEDTLS_ERR_RSA_INVALID_PADDING;
-        }
-        else
-        {
-            ret = mbedtls_psa_err_translate_pk( status );
-        }
+        ret = mbedtls_pk_error_from_psa_rsa( status );
         goto cleanup;
     }
 
@@ -290,7 +283,7 @@
 cleanup:
     status = psa_destroy_key( key_id );
     if( ret == 0 && status != PSA_SUCCESS )
-        ret = mbedtls_psa_err_translate_pk( status );
+        ret = mbedtls_pk_error_from_psa( status );
 
     return( ret );
 }