Correct PQC memory allocation

Change-Id: Idc2f89cf7b8e1b25bbc100e4f3d21da4cc1d598a

diff --git a/ta/src/cmd_exec.c b/ta/src/cmd_exec.c
index 51fe988b5ecb6776ef3a37fc95886714180bd9eb..0ab792d187f98629401a453d039b8a5baf34ada9 100644 (file)
--- a/ta/src/cmd_exec.c
+++ b/ta/src/cmd_exec.c
@@ -777,8 +777,8 @@ TEE_Result KM_ExecCmdGenerateKEMKey(TEE_Param param[4])
                goto clean;
        }
 
-       public_key = TEE_Malloc(kem->length_public_key*sizeof(uint8_t), 0);
-       private_key = TEE_Malloc(kem->length_secret_key*sizeof(uint8_t), 0);
+       public_key = TEE_Malloc(kem->length_public_key, 0);
+       private_key = TEE_Malloc(kem->length_secret_key, 0);
        if (public_key == NULL || private_key == NULL) {
                LOG("Failed to allocate key buffers");
                ret = TEE_ERROR_OUT_OF_MEMORY;
@@ -3831,8 +3831,8 @@ TEE_Result KM_ExecCmdEncapsulateKey(TEE_Param param[4])
                goto clean;
        }
 
-       shared_secret = TEE_Malloc(kem->length_shared_secret*sizeof(uint8_t), 0);
-       ciphertext = TEE_Malloc(kem->length_ciphertext*sizeof(uint8_t), 0);
+       shared_secret = TEE_Malloc(kem->length_shared_secret, 0);
+       ciphertext = TEE_Malloc(kem->length_ciphertext, 0);
        if (shared_secret == NULL || ciphertext == NULL) {
                LOG("Failed to allocate buffers");
                ret = TEE_ERROR_OUT_OF_MEMORY;
@@ -3982,7 +3982,7 @@ TEE_Result KM_ExecCmdDecapsulateKey(TEE_Param param[4])
                goto clean;
        }
 
-       shared_secret = TEE_Malloc(kem->length_shared_secret*sizeof(uint8_t), 0);
+       shared_secret = TEE_Malloc(kem->length_shared_secret, 0);
        if (shared_secret == NULL) {
                LOG("Failed to allocate buffer");
                ret = TEE_ERROR_OUT_OF_MEMORY;