From 6d2ca07e72d096e0d0936f9b9935bd2b1f9da88f Mon Sep 17 00:00:00 2001
From: Valerio Setti <vsetti@baylibre.com>
Date: Wed, 7 Jan 2026 11:02:38 +0100
Subject: [PATCH] scripts: generate_test_keys: minor improvements

- better guard for the "ecp.h" header file;
- use already existing Mbed TLS' macro for unused variables and place it
  at the beginning of the declaration of "predefined_keys".

Signed-off-by: Valerio Setti <vsetti@baylibre.com>
---
 scripts/generate_test_keys.py | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/scripts/generate_test_keys.py b/scripts/generate_test_keys.py
index 78551c9cf..7cec7b391 100755
--- a/scripts/generate_test_keys.py
+++ b/scripts/generate_test_keys.py
@@ -100,9 +100,11 @@ def write_output_file(output_file_name: str, arrays: str, look_up_table: str):
 #ifndef TEST_TEST_KEYS_H
 #define TEST_TEST_KEYS_H
 
-#if !defined(MBEDTLS_VERSION_MAJOR) || MBEDTLS_VERSION_MAJOR >= 4
+#if TF_PSA_CRYPTO_VERSION_MAJOR >= 1
+#include <tf_psa_crypto_common.h>
 #include <mbedtls/private/ecp.h>
 #else
+#include <common.h>
 #include <mbedtls/ecp.h>
 #endif
 
@@ -118,13 +120,7 @@ struct predefined_key_element {{
     size_t pub_key_len;
 }};
 
-#if defined(__GNUC__) || defined(__clang__)
-#define UNUSED __attribute__((unused))
-#else
-#define UNUSED
-#endif
-
-static struct predefined_key_element predefined_keys[] UNUSED = {{
+MBEDTLS_MAYBE_UNUSED static struct predefined_key_element predefined_keys[] = {{
 {}
 }};