commit f8c94a811a45d4e0291dbeba8aa622f1b115c064
author Andrzej Kurek <andrzej.kurek@arm.com> Wed Nov 07 08:18:52 2018 -0500
committer Andrzej Kurek <andrzej.kurek@arm.com> Thu Nov 22 13:37:14 2018 -0500
tree 11ad70fd6ce3e0681a01cb56646fe0dff6234682
parent 6d49ae92233a6bd1442ecc5aff4ccec4c82719cc

pk_wrap: check if curve conversion is successful

library/pk_wrap.c

diff --git a/library/pk_wrap.c b/library/pk_wrap.c
index 3e150a2..6007a23 100644
--- a/library/pk_wrap.c
+++ b/library/pk_wrap.c
@@ -677,6 +677,9 @@
     psa_algorithm_t psa_sig_md = PSA_ALG_ECDSA( translate_md_to_psa( md_alg ) );
     psa_ecc_curve_t curve = mbedtls_ecc_group_to_psa ( ( (mbedtls_ecdsa_context *) ctx )->grp.id );
 
+    if( curve == 0 )
+        return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
+
     memset( &signature, 0, sizeof( mbedtls_asn1_buf ) );
     key.pk_info = &pk_info;
     key.pk_ctx = ctx;