commit b8f97a1f3f79908240604d097c9e2e4e4b0f6887
author Gabor Mezei <gabor.mezei@arm.com> Wed Jan 24 16:58:40 2024 +0100
committer Gabor Mezei <gabor.mezei@arm.com> Wed Feb 28 15:17:17 2024 +0000
tree 8f9355c4e6238846895a536374040e76255dfe17
parent 212eb088848e6bfc9bb1b8c8750cadf1c79c7b20

Add test wrapper functions for cipher buffer protection

Signed-off-by: Gabor Mezei <gabor.mezei@arm.com>

tests/scripts/generate_psa_wrappers.py

diff --git a/tests/scripts/generate_psa_wrappers.py b/tests/scripts/generate_psa_wrappers.py
index 0918dcc..777f48b 100755
--- a/tests/scripts/generate_psa_wrappers.py
+++ b/tests/scripts/generate_psa_wrappers.py
@@ -145,7 +145,8 @@
         #pylint: disable=too-many-return-statements
         if function_name.startswith('psa_aead'):
             return True
-        if function_name == 'psa_cipher_encrypt':
+        if function_name in {'psa_cipher_encrypt', 'psa_cipher_decrypt',
+                             'psa_cipher_update', 'psa_cipher_finish'}:
             return True
         if function_name in ('psa_key_derivation_output_bytes',
                              'psa_key_derivation_input_bytes'):