From 0f54727bf48a5441388465e427d6150c01bd9d61 Mon Sep 17 00:00:00 2001 From: Ryan Everett Date: Thu, 25 Jan 2024 11:55:23 +0000 Subject: [PATCH] Restructure wrapper script Signed-off-by: Ryan Everett --- tests/scripts/generate_psa_wrappers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/scripts/generate_psa_wrappers.py b/tests/scripts/generate_psa_wrappers.py index 75a20fa30c..816dd5f53b 100755 --- a/tests/scripts/generate_psa_wrappers.py +++ b/tests/scripts/generate_psa_wrappers.py @@ -143,7 +143,9 @@ class PSAWrapperGenerator(c_wrapper_generator.Base): """Whether the specified buffer argument to a PSA function should be copied. """ # Proof-of-concept: just instrument one function for now - if function_name == 'psa_cipher_encrypt' or function_name == 'psa_key_derivation_output_bytes' or function_name == 'psa_key_derivation_input_bytes': + if function_name == 'psa_cipher_encrypt': + return True + if function_name == 'psa_key_derivation_output_bytes' or function_name == 'psa_key_derivation_input_bytes': return True if function_name in ('psa_import_key', 'psa_export_key',