// SPDX-License-Identifier: BSD-2-Clause
|
/*
|
* Copyright (c) 2014, STMicroelectronics International N.V.
|
* All rights reserved.
|
*/
|
#include <stdint.h>
|
|
#include <ta_rpc.h>
|
#include <tee_api.h>
|
#include <trace.h>
|
#include <ta_crypt.h>
|
#include <ta_sims_test.h>
|
|
static TEE_UUID cryp_uuid = TA_CRYPT_UUID;
|
|
static TEE_Result rpc_call_cryp(bool sec_mem, uint32_t nParamTypes,
|
TEE_Param pParams[4], uint32_t cmd)
|
{
|
TEE_TASessionHandle cryp_session = TEE_HANDLE_NULL;
|
TEE_Result res = TEE_ERROR_GENERIC;
|
uint32_t origin = 0;
|
TEE_Param params[4] = { };
|
size_t n = 0;
|
uint32_t types =
|
TEE_PARAM_TYPES(TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE,
|
TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE);
|
|
res = TEE_OpenTASession(&cryp_uuid, TEE_TIMEOUT_INFINITE, types,
|
params, &cryp_session, &origin);
|
|
if (res != TEE_SUCCESS) {
|
EMSG("rpc_sha256 - TEE_OpenTASession returned 0x%x\n",
|
(unsigned int)res);
|
return res;
|
}
|
|
types = nParamTypes;
|
if (sec_mem) {
|
TEE_MemFill(params, 0, sizeof(params));
|
for (n = 0; n < 4; n++) {
|
switch (TEE_PARAM_TYPE_GET(types, n)) {
|
case TEE_PARAM_TYPE_VALUE_INPUT:
|
case TEE_PARAM_TYPE_VALUE_INOUT:
|
params[n].value = pParams[n].value;
|
break;
|
|
case TEE_PARAM_TYPE_MEMREF_INPUT:
|
case TEE_PARAM_TYPE_MEMREF_OUTPUT:
|
case TEE_PARAM_TYPE_MEMREF_INOUT:
|
params[n].memref.buffer =
|
TEE_Malloc(pParams[n].memref.size, 0);
|
if (!params[n].memref.buffer)
|
TEE_Panic(0);
|
params[n].memref.size = pParams[n].memref.size;
|
if (TEE_PARAM_TYPE_GET(types, n) !=
|
TEE_PARAM_TYPE_MEMREF_OUTPUT)
|
TEE_MemMove(params[n].memref.buffer,
|
pParams[n].memref.buffer,
|
pParams[n].memref.size);
|
break;
|
default:
|
break;
|
}
|
}
|
} else {
|
TEE_MemMove(params, pParams, sizeof(params));
|
}
|
|
res = TEE_InvokeTACommand(cryp_session, TEE_TIMEOUT_INFINITE, cmd,
|
types, params, &origin);
|
if (res != TEE_SUCCESS) {
|
EMSG("rpc_call_cryp - TEE_InvokeTACommand returned 0x%x\n",
|
(unsigned int)res);
|
}
|
|
TEE_CloseTASession(cryp_session);
|
|
if (sec_mem) {
|
for (n = 0; n < 4; n++) {
|
switch (TEE_PARAM_TYPE_GET(types, n)) {
|
case TEE_PARAM_TYPE_VALUE_INOUT:
|
case TEE_PARAM_TYPE_VALUE_OUTPUT:
|
pParams[n].value = params[n].value;
|
break;
|
|
case TEE_PARAM_TYPE_MEMREF_INPUT:
|
case TEE_PARAM_TYPE_MEMREF_OUTPUT:
|
case TEE_PARAM_TYPE_MEMREF_INOUT:
|
if (TEE_PARAM_TYPE_GET(types, n) !=
|
TEE_PARAM_TYPE_MEMREF_INPUT)
|
TEE_MemMove(pParams[n].memref.buffer,
|
params[n].memref.buffer,
|
params[n].memref.size);
|
pParams[n].memref.size = params[n].memref.size;
|
TEE_Free(params[n].memref.buffer);
|
break;
|
default:
|
break;
|
}
|
}
|
|
}
|
|
return res;
|
}
|
|
TEE_Result rpc_sha224(bool sec_mem, uint32_t nParamTypes, TEE_Param pParams[4])
|
{
|
return rpc_call_cryp(sec_mem, nParamTypes, pParams,
|
TA_CRYPT_CMD_SHA224);
|
}
|
|
TEE_Result rpc_sha256(bool sec_mem, uint32_t nParamTypes, TEE_Param pParams[4])
|
{
|
return rpc_call_cryp(sec_mem, nParamTypes, pParams,
|
TA_CRYPT_CMD_SHA256);
|
}
|
|
TEE_Result rpc_aes256ecb_encrypt(bool sec_mem, uint32_t nParamTypes,
|
TEE_Param pParams[4])
|
{
|
return rpc_call_cryp(sec_mem, nParamTypes, pParams,
|
TA_CRYPT_CMD_AES256ECB_ENC);
|
}
|
|
TEE_Result rpc_aes256ecb_decrypt(bool sec_mem, uint32_t nParamTypes,
|
TEE_Param pParams[4])
|
{
|
return rpc_call_cryp(sec_mem, nParamTypes, pParams,
|
TA_CRYPT_CMD_AES256ECB_DEC);
|
}
|
|
TEE_Result rpc_open(void *session_context, uint32_t param_types,
|
TEE_Param params[4])
|
{
|
TEE_TASessionHandle session = TEE_HANDLE_NULL;
|
uint32_t orig = 0;
|
TEE_Result res = TEE_ERROR_GENERIC;
|
TEE_UUID uuid = TA_SIMS_TEST_UUID;
|
uint32_t types =
|
TEE_PARAM_TYPES(TEE_PARAM_TYPE_VALUE_OUTPUT, TEE_PARAM_TYPE_NONE,
|
TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE);
|
TEE_Param par[4] = { };
|
|
(void)session_context;
|
(void)param_types;
|
|
res = TEE_OpenTASession(&uuid, TEE_TIMEOUT_INFINITE, 0, NULL, &session,
|
&orig);
|
|
if (res != TEE_SUCCESS)
|
return res;
|
|
TEE_MemFill(params, 0, sizeof(TEE_Param) * 4);
|
res =
|
TEE_InvokeTACommand(session, TEE_TIMEOUT_INFINITE,
|
TA_SIMS_CMD_GET_COUNTER, types, par, &orig);
|
|
if (res != TEE_SUCCESS)
|
goto exit;
|
|
exit:
|
TEE_CloseTASession(session);
|
|
return res;
|
}
|
