hc
2023-12-09 b22da3d8526a935aa31e086e63f60ff3246cb61c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// SPDX-License-Identifier: BSD-2-Clause
/*
 * Copyright (c) 2021 Rockchip Electronics Co. Ltd.
 */
 
#include <tee_internal_api.h>
#include <tee_internal_api_extensions.h>
#include "rk_trng_api.h"
#include "rk_os_service.h"
 
/*
 * RK356x, RK3399 platform supported.
 * Not all platform supports this API. When not supports, it will returns
 * TEE_ERROR_NOT_SUPPORTED.
 */
TEE_Result rk_get_trng(uint8_t *buffer, uint32_t size)
{
   TEE_Result res = TEE_SUCCESS;
   uint32_t param_type;
   TEE_Param params[4];
   TEE_TASessionHandle session = TEE_HANDLE_NULL;
   uint32_t error_origin = 0;
   TEE_UUID uuid = RK_OS_SERVICE_UUID;
 
   param_type = TEE_PARAM_TYPES(TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE,
                    TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE);
 
   res = TEE_OpenTASession(&uuid, 0, param_type, params,
               &session, &error_origin);
   if (res != TEE_SUCCESS) {
       EMSG("TEE_OpenTASession failed with code 0x%x origin 0x%x",
            res, error_origin);
       session = TEE_HANDLE_NULL;
       return res;
   }
 
   param_type = TEE_PARAM_TYPES(TEE_PARAM_TYPE_VALUE_INOUT,
                    TEE_PARAM_TYPE_NONE,
                    TEE_PARAM_TYPE_NONE,
                    TEE_PARAM_TYPE_NONE);
 
   TEE_MemFill(params, 0, sizeof(params));
   params[0].value.a = (uint32_t)buffer;
   params[0].value.b = size;
 
   res = TEE_InvokeTACommand(session, 0, OS_SERVICE_CMD_GET_TRNG,
                 param_type, params, &error_origin);
   if (res != TEE_SUCCESS)
       EMSG("TEE_InvokeTACommand failed with code 0x%x origin 0x%x\n",
            res, error_origin);
 
   TEE_CloseTASession(session);
   session = TEE_HANDLE_NULL;
   return res;
}