commit e60b365a5e4a991a843d85b749632628b0b585e5
author Gilles Peskine <Gilles.Peskine@arm.com> Tue Feb 25 19:54:07 2020 +0100
committer Gilles Peskine <Gilles.Peskine@arm.com> Tue Feb 25 19:54:07 2020 +0100
tree defd1339c13e4744168467b466c249ad37966936
parent f02b984f863c1dd0013e12cb9b2c715a63c1823f

EC keys can have the type MBEDTLS_PK_ECKEY_DH too

programs/fuzz/fuzz_pubkey.c

diff --git a/programs/fuzz/fuzz_pubkey.c b/programs/fuzz/fuzz_pubkey.c
index 265ee2a..95b7521 100644
--- a/programs/fuzz/fuzz_pubkey.c
+++ b/programs/fuzz/fuzz_pubkey.c
@@ -39,7 +39,8 @@
         else
 #endif
 #if defined(MBEDTLS_ECP_C)
-        if( mbedtls_pk_get_type( &pk ) == MBEDTLS_PK_ECKEY )
+        if( mbedtls_pk_get_type( &pk ) == MBEDTLS_PK_ECKEY ||
+            mbedtls_pk_get_type( &pk ) == MBEDTLS_PK_ECKEY_DH )
         {
             mbedtls_ecp_keypair *ecp = mbedtls_pk_ec( pk );
             mbedtls_ecp_group_id grp_id = ecp->grp.id;