1 // SPDX-License-Identifier: BSD-2-Clause
2 /*
3 * Copyright (c) 2018, Linaro Limited
4 */
5
6 #include <common.h>
7 #include <log.h>
8 #include <tee.h>
9 #include <linux/types.h>
10
11 #include "optee_msg.h"
12 #include "optee_msg_supplicant.h"
13 #include "optee_private.h"
14 #include "optee_smc.h"
15
cmd_shm_alloc(struct udevice * dev,struct optee_msg_arg * arg,void ** page_list)16 static void cmd_shm_alloc(struct udevice *dev, struct optee_msg_arg *arg,
17 void **page_list)
18 {
19 int rc;
20 struct tee_shm *shm;
21 void *pl;
22 u64 ph_ptr;
23
24 arg->ret_origin = TEE_ORIGIN_COMMS;
25
26 if (arg->num_params != 1 ||
27 arg->params[0].attr != OPTEE_MSG_ATTR_TYPE_VALUE_INPUT) {
28 arg->ret = TEE_ERROR_BAD_PARAMETERS;
29 return;
30 }
31
32 rc = __tee_shm_add(dev, 0, NULL, arg->params[0].u.value.b,
33 TEE_SHM_REGISTER | TEE_SHM_ALLOC, &shm);
34 if (rc) {
35 if (rc == -ENOMEM)
36 arg->ret = TEE_ERROR_OUT_OF_MEMORY;
37 else
38 arg->ret = TEE_ERROR_GENERIC;
39 return;
40 }
41
42 pl = optee_alloc_and_init_page_list(shm->addr, shm->size, &ph_ptr);
43 if (!pl) {
44 arg->ret = TEE_ERROR_OUT_OF_MEMORY;
45 tee_shm_free(shm);
46 return;
47 }
48
49 *page_list = pl;
50 arg->params[0].attr = OPTEE_MSG_ATTR_TYPE_TMEM_OUTPUT |
51 OPTEE_MSG_ATTR_NONCONTIG;
52 arg->params[0].u.tmem.buf_ptr = ph_ptr;
53 arg->params[0].u.tmem.size = shm->size;
54 arg->params[0].u.tmem.shm_ref = (ulong)shm;
55 arg->ret = TEE_SUCCESS;
56 }
57
cmd_shm_free(struct optee_msg_arg * arg)58 static void cmd_shm_free(struct optee_msg_arg *arg)
59 {
60 arg->ret_origin = TEE_ORIGIN_COMMS;
61
62 if (arg->num_params != 1 ||
63 arg->params[0].attr != OPTEE_MSG_ATTR_TYPE_VALUE_INPUT) {
64 arg->ret = TEE_ERROR_BAD_PARAMETERS;
65 return;
66 }
67
68 tee_shm_free((struct tee_shm *)(ulong)arg->params[0].u.value.b);
69 arg->ret = TEE_SUCCESS;
70 }
71
optee_suppl_cmd(struct udevice * dev,struct tee_shm * shm_arg,void ** page_list)72 void optee_suppl_cmd(struct udevice *dev, struct tee_shm *shm_arg,
73 void **page_list)
74 {
75 struct optee_msg_arg *arg = shm_arg->addr;
76
77 switch (arg->cmd) {
78 case OPTEE_MSG_RPC_CMD_SHM_ALLOC:
79 cmd_shm_alloc(dev, arg, page_list);
80 break;
81 case OPTEE_MSG_RPC_CMD_SHM_FREE:
82 cmd_shm_free(arg);
83 break;
84 case OPTEE_MSG_RPC_CMD_FS:
85 debug("REE FS storage isn't available\n");
86 arg->ret = TEE_ERROR_STORAGE_NOT_AVAILABLE;
87 break;
88 case OPTEE_MSG_RPC_CMD_RPMB:
89 optee_suppl_cmd_rpmb(dev, arg);
90 break;
91 default:
92 arg->ret = TEE_ERROR_NOT_IMPLEMENTED;
93 }
94
95 arg->ret_origin = TEE_ORIGIN_COMMS;
96 }
97