commit c6977b489929cf01a01a634fec183a5ff5bab8dc
author David Horstmann <david.horstmann@arm.com> Mon Nov 27 17:43:05 2023 +0000
committer David Horstmann <david.horstmann@arm.com> Mon Dec 11 17:58:56 2023 +0000
tree 817139c2f4c4dab8ac1164114e9c0e9bb62c9557
parent 8f35a4f0034a58a747c75dbd44066747c509d9a9

Copy input and output in psa_cipher_encrypt()

Signed-off-by: David Horstmann <david.horstmann@arm.com>

library/psa_crypto.c

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index ddc834f..36c5f75 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -4346,6 +4346,20 @@
     size_t default_iv_length = 0;
     psa_key_attributes_t attributes;
 
+    psa_crypto_local_input_t local_input = PSA_CRYPTO_LOCAL_INPUT_INIT;
+    status = psa_crypto_local_input_alloc(input, input_length, &local_input);
+    if (status != PSA_SUCCESS) {
+        goto exit;
+    }
+    input = local_input.buffer;
+
+    psa_crypto_local_output_t local_output = PSA_CRYPTO_LOCAL_OUTPUT_INIT;
+    status = psa_crypto_local_output_alloc(output, output_size, &local_output);
+    if (status != PSA_SUCCESS) {
+        goto exit;
+    }
+    output = local_output.buffer;
+
     if (!PSA_ALG_IS_CIPHER(alg)) {
         status = PSA_ERROR_INVALID_ARGUMENT;
         goto exit;
@@ -4401,6 +4415,9 @@
         *output_length = 0;
     }
 
+    psa_crypto_local_input_free(&local_input);
+    psa_crypto_local_output_free(&local_output);
+
     return status;
 }