static TEE_Result cmd_invoke_pta(uint32_t nParamTypes, TEE_Param pParams[TEE_NUM_PARAMS], uint32_t nCommandID) { const TEE_UUID uuid = PTA_INVOKE_TESTS_UUID; static TEE_TASessionHandle sess = TEE_HANDLE_NULL; uint32_t ret_orig; TEE_Result res; if (sess == TEE_HANDLE_NULL) { res = TEE_OpenTASession(&uuid, 0, 0, NULL, &sess, &ret_orig); if (res != TEE_SUCCESS) { EMSG("SDP basic test TA: TEE_OpenTASession() FAILED \n"); goto cleanup_return; } } res = TEE_InvokeTACommand(sess, 0, nCommandID, nParamTypes, pParams, &ret_orig); if (res != TEE_SUCCESS) { EMSG("SDP basic test TA: TEE_OpenTASession() FAILED %x/%d\n", res, ret_orig); } cleanup_return: if (res != TEE_SUCCESS) { TEE_CloseTASession(sess); sess = TEE_HANDLE_NULL; } return res; }
TEE_Result rpc_open(void *session_context, uint32_t param_types, TEE_Param params[4]) { TEE_TASessionHandle session; uint32_t orig; TEE_Result res; 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, 0, 0, NULL, &session, &orig); if (res != TEE_SUCCESS) return res; TEE_MemFill(params, 0, sizeof(TEE_Param) * 4); res = TEE_InvokeTACommand(session, 0, TA_SIMS_CMD_GET_COUNTER, types, par, &orig); if (res != TEE_SUCCESS) goto exit; exit: TEE_CloseTASession(session); return res; }
static TEE_Result rpc_call_cryp(uint32_t nParamTypes, TEE_Param pParams[4], uint32_t cmd) { TEE_TASessionHandle cryp_session; TEE_Result res; uint32_t origin; TEE_Param params[4]; uint32_t types = TEE_PARAM_TYPES(TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE); TEE_MemFill(params, 0, sizeof(TEE_Param) * 4); res = TEE_OpenTASession(&cryp_uuid, 0, types, params, &cryp_session, &origin); if (res != TEE_SUCCESS) { EMSG("rpc_sha256 - TEE_OpenTASession returned 0x%x\n", (unsigned int)res); return res; } res = TEE_InvokeTACommand(cryp_session, 0, cmd, nParamTypes, pParams, &origin); if (res != TEE_SUCCESS) { EMSG("rpc_sha256 - TEE_InvokeTACommand returned 0x%x\n", (unsigned int)res); } TEE_CloseTASession(cryp_session); return res; }
static TEE_Result rpc_call_cryp(bool sec_mem, uint32_t nParamTypes, TEE_Param pParams[4], uint32_t cmd) { TEE_TASessionHandle cryp_session; TEE_Result res; uint32_t origin; TEE_Param params[4]; size_t n; uint32_t types = TEE_PARAM_TYPES(TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE); TEE_MemFill(params, 0, sizeof(TEE_Param) * 4); res = TEE_OpenTASession(&cryp_uuid, 0, 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, 0, 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; }