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>

tests/suites/test_suite_pk.function

diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function
index d397374..7227f92 100644
--- a/tests/suites/test_suite_pk.function
+++ b/tests/suites/test_suite_pk.function
@@ -38,6 +38,10 @@
     size_t key_len;
     int ret;
 
+    if (curve == 0) {
+        return MBEDTLS_ERR_PK_BAD_INPUT_DATA;
+    }
+
     psa_set_key_type(&key_attr, PSA_KEY_TYPE_ECC_KEY_PAIR(curve));
     psa_set_key_bits(&key_attr, curve_bits);
     psa_set_key_usage_flags(&key_attr, PSA_KEY_USAGE_EXPORT);