1 // SPDX-License-Identifier: GPL-2.0
3 * Copyright (C) 2019 Linaro Ltd.
6 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
8 #include <linux/kernel.h>
9 #include <linux/slab.h>
10 #include <linux/tee_drv.h>
11 #include <linux/uuid.h>
12 #include "optee_private.h"
17 * [out] memref[0] Array of device UUIDs
20 * TEE_SUCCESS - Invoke command success
21 * TEE_ERROR_BAD_PARAMETERS - Incorrect input param
22 * TEE_ERROR_SHORT_BUFFER - Output buffer size less than required
24 #define PTA_CMD_GET_DEVICES 0x0
26 static int optee_ctx_match(struct tee_ioctl_version_data *ver, const void *data)
28 if (ver->impl_id == TEE_IMPL_ID_OPTEE)
34 static int get_devices(struct tee_context *ctx, u32 session,
35 struct tee_shm *device_shm, u32 *shm_size)
38 struct tee_ioctl_invoke_arg inv_arg;
39 struct tee_param param[4];
41 memset(&inv_arg, 0, sizeof(inv_arg));
42 memset(¶m, 0, sizeof(param));
44 /* Invoke PTA_CMD_GET_DEVICES function */
45 inv_arg.func = PTA_CMD_GET_DEVICES;
46 inv_arg.session = session;
47 inv_arg.num_params = 4;
49 /* Fill invoke cmd params */
50 param[0].attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT;
51 param[0].u.memref.shm = device_shm;
52 param[0].u.memref.size = *shm_size;
53 param[0].u.memref.shm_offs = 0;
55 ret = tee_client_invoke_func(ctx, &inv_arg, param);
56 if ((ret < 0) || ((inv_arg.ret != TEEC_SUCCESS) &&
57 (inv_arg.ret != TEEC_ERROR_SHORT_BUFFER))) {
58 pr_err("PTA_CMD_GET_DEVICES invoke function err: %x\n",
63 *shm_size = param[0].u.memref.size;
68 static int optee_register_device(const uuid_t *device_uuid)
70 struct tee_client_device *optee_device = NULL;
73 optee_device = kzalloc(sizeof(*optee_device), GFP_KERNEL);
77 optee_device->dev.bus = &tee_bus_type;
78 if (dev_set_name(&optee_device->dev, "optee-ta-%pUb", device_uuid)) {
82 uuid_copy(&optee_device->id.uuid, device_uuid);
84 rc = device_register(&optee_device->dev);
86 pr_err("device registration failed, err: %d\n", rc);
93 int optee_enumerate_devices(void)
95 const uuid_t pta_uuid =
96 UUID_INIT(0x7011a688, 0xddde, 0x4053,
97 0xa5, 0xa9, 0x7b, 0x3c, 0x4d, 0xdf, 0x13, 0xb8);
98 struct tee_ioctl_open_session_arg sess_arg;
99 struct tee_shm *device_shm = NULL;
100 const uuid_t *device_uuid = NULL;
101 struct tee_context *ctx = NULL;
102 u32 shm_size = 0, idx, num_devices = 0;
105 memset(&sess_arg, 0, sizeof(sess_arg));
107 /* Open context with OP-TEE driver */
108 ctx = tee_client_open_context(NULL, optee_ctx_match, NULL, NULL);
112 /* Open session with device enumeration pseudo TA */
113 memcpy(sess_arg.uuid, pta_uuid.b, TEE_IOCTL_UUID_LEN);
114 sess_arg.clnt_login = TEE_IOCTL_LOGIN_PUBLIC;
115 sess_arg.num_params = 0;
117 rc = tee_client_open_session(ctx, &sess_arg, NULL);
118 if ((rc < 0) || (sess_arg.ret != TEEC_SUCCESS)) {
119 /* Device enumeration pseudo TA not found */
124 rc = get_devices(ctx, sess_arg.session, NULL, &shm_size);
125 if (rc < 0 || !shm_size)
128 device_shm = tee_shm_alloc(ctx, shm_size,
129 TEE_SHM_MAPPED | TEE_SHM_DMA_BUF);
130 if (IS_ERR(device_shm)) {
131 pr_err("tee_shm_alloc failed\n");
132 rc = PTR_ERR(device_shm);
136 rc = get_devices(ctx, sess_arg.session, device_shm, &shm_size);
140 device_uuid = tee_shm_get_va(device_shm, 0);
141 if (IS_ERR(device_uuid)) {
142 pr_err("tee_shm_get_va failed\n");
143 rc = PTR_ERR(device_uuid);
147 num_devices = shm_size / sizeof(uuid_t);
149 for (idx = 0; idx < num_devices; idx++) {
150 rc = optee_register_device(&device_uuid[idx]);
156 tee_shm_free(device_shm);
158 tee_client_close_session(ctx, sess_arg.session);
160 tee_client_close_context(ctx);