commit a87601dc112e3fe308b04609f9706d3f138ba179
author valerio <valerio.setti@nordicsemi.no> Wed May 31 12:01:55 2023 +0200
committer valerio <valerio.setti@nordicsemi.no> Thu Jun 01 11:06:40 2023 +0200
tree 3cf67d4df91f3d04581380c5dcdaa15d14045b53
parent f6853a87512171975949b0b8d8ca1b5ae6aba14f

pk_internal: add support for opaque keys for getting EC curve ID

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

library/pk_internal.h

diff --git a/library/pk_internal.h b/library/pk_internal.h
index 8d4b005..21fb34a 100644
--- a/library/pk_internal.h
+++ b/library/pk_internal.h
@@ -84,11 +84,30 @@
 static inline mbedtls_ecp_group_id mbedtls_pk_get_group_id(const mbedtls_pk_context *pk)
 {
     mbedtls_ecp_group_id id;
+
+#if defined(MBEDTLS_USE_PSA_CRYPTO)
+    psa_key_attributes_t opaque_attrs = PSA_KEY_ATTRIBUTES_INIT;
+    psa_key_type_t opaque_key_type;
+    psa_ecc_family_t curve;
+
+    if (mbedtls_pk_get_type(pk) == MBEDTLS_PK_OPAQUE) {
+        if (psa_get_key_attributes(pk->priv_id, &opaque_attrs) != PSA_SUCCESS) {
+            return MBEDTLS_ECP_DP_NONE;
+        }
+        opaque_key_type = psa_get_key_type(&opaque_attrs);
+        curve = PSA_KEY_TYPE_ECC_GET_FAMILY(opaque_key_type);
+        id = mbedtls_ecc_group_of_psa(curve, psa_get_key_bits(&opaque_attrs), 0);
+        psa_reset_key_attributes(&opaque_attrs);
+    } else
+#endif /* MBEDTLS_USE_PSA_CRYPTO */
+    {
 #if defined(MBEDTLS_PK_USE_PSA_EC_DATA)
-    id = mbedtls_ecc_group_of_psa(pk->ec_family, pk->ec_bits, 0);
+        id = mbedtls_ecc_group_of_psa(pk->ec_family, pk->ec_bits, 0);
 #else /* MBEDTLS_PK_USE_PSA_EC_DATA */
-    id = mbedtls_pk_ec_ro(*pk)->grp.id;
+        id = mbedtls_pk_ec_ro(*pk)->grp.id;
 #endif /* MBEDTLS_PK_USE_PSA_EC_DATA */
+    }
+
     return id;
 }