From 398c503f6ff6b46f56d323d4e603285447d9d813 Mon Sep 17 00:00:00 2001 From: Przemek Stekiel Date: Wed, 11 May 2022 14:05:40 +0200 Subject: [PATCH] generate_psa_tests.py: adapt OpFail test generator for HKDF-Exract/Expand algs Signed-off-by: Przemek Stekiel --- tests/scripts/generate_psa_tests.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/scripts/generate_psa_tests.py b/tests/scripts/generate_psa_tests.py index 492810bf0..d34663188 100755 --- a/tests/scripts/generate_psa_tests.py +++ b/tests/scripts/generate_psa_tests.py @@ -78,6 +78,11 @@ def automatic_dependencies(*expressions: str) -> List[str]: """ used = set() for expr in expressions: + # HKDF_EXTRACT and HKDF_EXPAND algs depend on HKDF + if "HKDF_EXTRACT" in expr: + expr = expr.replace("HKDF_EXTRACT", "HKDF") + if "HKDF_EXPAND" in expr: + expr = expr.replace("HKDF_EXPAND", "HKDF") used.update(re.findall(r'PSA_(?:ALG|ECC_FAMILY|KEY_TYPE)_\w+', expr)) used.difference_update(SYMBOLS_WITHOUT_DEPENDENCY) return sorted(psa_want_symbol(name) for name in used)