1 // SPDX-License-Identifier: GPL-2.0
3 * Copyright (C) 2019-2021 Linaro Ltd.
6 * Sumit Garg <sumit.garg@linaro.org>
10 #include <linux/key-type.h>
11 #include <linux/module.h>
12 #include <linux/slab.h>
13 #include <linux/string.h>
14 #include <linux/tee_drv.h>
15 #include <linux/uuid.h>
17 #include <keys/trusted_tee.h>
19 #define DRIVER_NAME "trusted-key-tee"
22 * Get random data for symmetric key
24 * [out] memref[0] Random data
26 #define TA_CMD_GET_RANDOM 0x0
29 * Seal trusted key using hardware unique key
31 * [in] memref[0] Plain key
32 * [out] memref[1] Sealed key datablob
34 #define TA_CMD_SEAL 0x1
37 * Unseal trusted key using hardware unique key
39 * [in] memref[0] Sealed key datablob
40 * [out] memref[1] Plain key
42 #define TA_CMD_UNSEAL 0x2
45 * struct trusted_key_tee_private - TEE Trusted key private data
46 * @dev: TEE based Trusted key device.
47 * @ctx: TEE context handler.
48 * @session_id: Trusted key TA session identifier.
49 * @shm_pool: Memory pool shared with TEE device.
51 struct trusted_key_tee_private {
53 struct tee_context *ctx;
55 struct tee_shm *shm_pool;
58 static struct trusted_key_tee_private pvt_data;
61 * Have the TEE seal(encrypt) the symmetric key
63 static int trusted_tee_seal(struct trusted_key_payload *p, char *datablob)
66 struct tee_ioctl_invoke_arg inv_arg;
67 struct tee_param param[4];
68 struct tee_shm *reg_shm_in = NULL, *reg_shm_out = NULL;
70 memset(&inv_arg, 0, sizeof(inv_arg));
71 memset(¶m, 0, sizeof(param));
73 reg_shm_in = tee_shm_register_kernel_buf(pvt_data.ctx, p->key,
75 if (IS_ERR(reg_shm_in)) {
76 dev_err(pvt_data.dev, "key shm register failed\n");
77 return PTR_ERR(reg_shm_in);
80 reg_shm_out = tee_shm_register_kernel_buf(pvt_data.ctx, p->blob,
82 if (IS_ERR(reg_shm_out)) {
83 dev_err(pvt_data.dev, "blob shm register failed\n");
84 ret = PTR_ERR(reg_shm_out);
88 inv_arg.func = TA_CMD_SEAL;
89 inv_arg.session = pvt_data.session_id;
90 inv_arg.num_params = 4;
92 param[0].attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT;
93 param[0].u.memref.shm = reg_shm_in;
94 param[0].u.memref.size = p->key_len;
95 param[0].u.memref.shm_offs = 0;
96 param[1].attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT;
97 param[1].u.memref.shm = reg_shm_out;
98 param[1].u.memref.size = sizeof(p->blob);
99 param[1].u.memref.shm_offs = 0;
101 ret = tee_client_invoke_func(pvt_data.ctx, &inv_arg, param);
102 if ((ret < 0) || (inv_arg.ret != 0)) {
103 dev_err(pvt_data.dev, "TA_CMD_SEAL invoke err: %x\n",
107 p->blob_len = param[1].u.memref.size;
112 tee_shm_free(reg_shm_out);
114 tee_shm_free(reg_shm_in);
120 * Have the TEE unseal(decrypt) the symmetric key
122 static int trusted_tee_unseal(struct trusted_key_payload *p, char *datablob)
125 struct tee_ioctl_invoke_arg inv_arg;
126 struct tee_param param[4];
127 struct tee_shm *reg_shm_in = NULL, *reg_shm_out = NULL;
129 memset(&inv_arg, 0, sizeof(inv_arg));
130 memset(¶m, 0, sizeof(param));
132 reg_shm_in = tee_shm_register_kernel_buf(pvt_data.ctx, p->blob,
134 if (IS_ERR(reg_shm_in)) {
135 dev_err(pvt_data.dev, "blob shm register failed\n");
136 return PTR_ERR(reg_shm_in);
139 reg_shm_out = tee_shm_register_kernel_buf(pvt_data.ctx, p->key,
141 if (IS_ERR(reg_shm_out)) {
142 dev_err(pvt_data.dev, "key shm register failed\n");
143 ret = PTR_ERR(reg_shm_out);
147 inv_arg.func = TA_CMD_UNSEAL;
148 inv_arg.session = pvt_data.session_id;
149 inv_arg.num_params = 4;
151 param[0].attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT;
152 param[0].u.memref.shm = reg_shm_in;
153 param[0].u.memref.size = p->blob_len;
154 param[0].u.memref.shm_offs = 0;
155 param[1].attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT;
156 param[1].u.memref.shm = reg_shm_out;
157 param[1].u.memref.size = sizeof(p->key);
158 param[1].u.memref.shm_offs = 0;
160 ret = tee_client_invoke_func(pvt_data.ctx, &inv_arg, param);
161 if ((ret < 0) || (inv_arg.ret != 0)) {
162 dev_err(pvt_data.dev, "TA_CMD_UNSEAL invoke err: %x\n",
166 p->key_len = param[1].u.memref.size;
171 tee_shm_free(reg_shm_out);
173 tee_shm_free(reg_shm_in);
179 * Have the TEE generate random symmetric key
181 static int trusted_tee_get_random(unsigned char *key, size_t key_len)
184 struct tee_ioctl_invoke_arg inv_arg;
185 struct tee_param param[4];
186 struct tee_shm *reg_shm = NULL;
188 memset(&inv_arg, 0, sizeof(inv_arg));
189 memset(¶m, 0, sizeof(param));
191 reg_shm = tee_shm_register_kernel_buf(pvt_data.ctx, key, key_len);
192 if (IS_ERR(reg_shm)) {
193 dev_err(pvt_data.dev, "key shm register failed\n");
194 return PTR_ERR(reg_shm);
197 inv_arg.func = TA_CMD_GET_RANDOM;
198 inv_arg.session = pvt_data.session_id;
199 inv_arg.num_params = 4;
201 param[0].attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT;
202 param[0].u.memref.shm = reg_shm;
203 param[0].u.memref.size = key_len;
204 param[0].u.memref.shm_offs = 0;
206 ret = tee_client_invoke_func(pvt_data.ctx, &inv_arg, param);
207 if ((ret < 0) || (inv_arg.ret != 0)) {
208 dev_err(pvt_data.dev, "TA_CMD_GET_RANDOM invoke err: %x\n",
212 ret = param[0].u.memref.size;
215 tee_shm_free(reg_shm);
220 static int optee_ctx_match(struct tee_ioctl_version_data *ver, const void *data)
222 if (ver->impl_id == TEE_IMPL_ID_OPTEE)
228 static int trusted_key_probe(struct device *dev)
230 struct tee_client_device *rng_device = to_tee_client_device(dev);
232 struct tee_ioctl_open_session_arg sess_arg;
234 memset(&sess_arg, 0, sizeof(sess_arg));
236 pvt_data.ctx = tee_client_open_context(NULL, optee_ctx_match, NULL,
238 if (IS_ERR(pvt_data.ctx))
241 memcpy(sess_arg.uuid, rng_device->id.uuid.b, TEE_IOCTL_UUID_LEN);
242 sess_arg.clnt_login = TEE_IOCTL_LOGIN_REE_KERNEL;
243 sess_arg.num_params = 0;
245 ret = tee_client_open_session(pvt_data.ctx, &sess_arg, NULL);
246 if ((ret < 0) || (sess_arg.ret != 0)) {
247 dev_err(dev, "tee_client_open_session failed, err: %x\n",
252 pvt_data.session_id = sess_arg.session;
254 ret = register_key_type(&key_type_trusted);
263 tee_client_close_session(pvt_data.ctx, pvt_data.session_id);
265 tee_client_close_context(pvt_data.ctx);
270 static int trusted_key_remove(struct device *dev)
272 unregister_key_type(&key_type_trusted);
273 tee_client_close_session(pvt_data.ctx, pvt_data.session_id);
274 tee_client_close_context(pvt_data.ctx);
279 static const struct tee_client_device_id trusted_key_id_table[] = {
280 {UUID_INIT(0xf04a0fe7, 0x1f5d, 0x4b9b,
281 0xab, 0xf7, 0x61, 0x9b, 0x85, 0xb4, 0xce, 0x8c)},
284 MODULE_DEVICE_TABLE(tee, trusted_key_id_table);
286 static struct tee_client_driver trusted_key_driver = {
287 .id_table = trusted_key_id_table,
290 .bus = &tee_bus_type,
291 .probe = trusted_key_probe,
292 .remove = trusted_key_remove,
296 static int trusted_tee_init(void)
298 return driver_register(&trusted_key_driver.driver);
301 static void trusted_tee_exit(void)
303 driver_unregister(&trusted_key_driver.driver);
306 struct trusted_key_ops trusted_key_tee_ops = {
307 .migratable = 0, /* non-migratable */
308 .init = trusted_tee_init,
309 .seal = trusted_tee_seal,
310 .unseal = trusted_tee_unseal,
311 .get_random = trusted_tee_get_random,
312 .exit = trusted_tee_exit,