commit a7cb845705163ee3c2fa396da5737df5b3daac65
author Valerio Setti <valerio.setti@nordicsemi.no> Mon May 22 18:39:43 2023 +0200
committer Valerio Setti <valerio.setti@nordicsemi.no> Mon May 22 18:39:43 2023 +0200
tree a40138bb08b091246c70dceec785a8f78bc94414
parent f57007dd1edb3f323cf396c66204a69349509d6f

pk: add checks for the returned ECC family

Signed-off-by: Valerio Setti <valerio.setti@nordicsemi.no>

library/pk.c

diff --git a/library/pk.c b/library/pk.c
index 826c29a..9c4aa16 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -224,6 +224,9 @@
 
     pk->ec_family = mbedtls_ecc_group_to_psa(ecp_keypair->grp.id,
                                              &pk->ec_bits);
+    if (pk->ec_family == 0) {
+        return MBEDTLS_ERR_PK_BAD_INPUT_DATA;
+    }
 
     return 0;
 }