commit 398c503f6ff6b46f56d323d4e603285447d9d813
author Przemek Stekiel <przemyslaw.stekiel@mobica.com> Wed May 11 14:05:40 2022 +0200
committer Przemek Stekiel <przemyslaw.stekiel@mobica.com> Wed May 11 14:05:40 2022 +0200
tree 3ba7eca14fee20754748833cbee8650b76a6af9d
parent 6786a87ccdcfb8cb479a2392c2ca606ab09fdb1c

generate_psa_tests.py: adapt OpFail test generator for HKDF-Exract/Expand algs

Signed-off-by: Przemek Stekiel <przemyslaw.stekiel@mobica.com>

tests/scripts/generate_psa_tests.py

diff --git a/tests/scripts/generate_psa_tests.py b/tests/scripts/generate_psa_tests.py
index 492810b..d346631 100755
--- a/tests/scripts/generate_psa_tests.py
+++ b/tests/scripts/generate_psa_tests.py
@@ -78,6 +78,11 @@
     """
     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)