xref: /optee_os/core/tee/tee_svc.c (revision df24e6517b6454cf906c16979ea0e7546c5c99d5)
1 // SPDX-License-Identifier: BSD-2-Clause
2 /*
3  * Copyright (c) 2014, STMicroelectronics International N.V.
4  */
5 #include <kernel/chip_services.h>
6 #include <kernel/pseudo_ta.h>
7 #include <kernel/tee_common.h>
8 #include <kernel/tee_common_otp.h>
9 #include <kernel/tee_ta_manager.h>
10 #include <kernel/tee_time.h>
11 #include <kernel/trace_ta.h>
12 #include <mm/core_memprot.h>
13 #include <mm/mobj.h>
14 #include <mm/tee_mm.h>
15 #include <mm/tee_mmu.h>
16 #include <stdlib_ext.h>
17 #include <tee_api_types.h>
18 #include <tee/tee_cryp_utl.h>
19 #include <tee/tee_svc.h>
20 #include <trace.h>
21 #include <user_ta_header.h>
22 #include <utee_types.h>
23 #include <util.h>
24 
25 vaddr_t tee_svc_uref_base;
26 
27 void syscall_log(const void *buf __maybe_unused, size_t len __maybe_unused)
28 {
29 #ifdef CFG_TEE_CORE_TA_TRACE
30 	char *kbuf;
31 
32 	if (len == 0)
33 		return;
34 
35 	kbuf = malloc(len + 1);
36 	if (kbuf == NULL)
37 		return;
38 
39 	if (tee_svc_copy_from_user(kbuf, buf, len) == TEE_SUCCESS) {
40 		kbuf[len] = '\0';
41 		trace_ext_puts(kbuf);
42 	}
43 
44 	free_wipe(kbuf);
45 #endif
46 }
47 
48 TEE_Result syscall_not_supported(void)
49 {
50 	return TEE_ERROR_NOT_SUPPORTED;
51 }
52 
53 /* Configuration properties */
54 /* API implementation version */
55 static const char api_vers[] = TO_STR(CFG_TEE_API_VERSION);
56 
57 /* Implementation description (implementation-dependent) */
58 static const char descr[] = TO_STR(CFG_TEE_IMPL_DESCR);
59 
60 /*
61  * TA persistent time protection level
62  * 100: Persistent time based on an REE-controlled real-time clock
63  * and on the TEE Trusted Storage for the storage of origins (default).
64  * 1000: Persistent time based on a TEE-controlled real-time clock
65  * and the TEE Trusted Storage.
66  * The real-time clock MUST be out of reach of software attacks
67  * from the REE.
68  */
69 static const uint32_t ta_time_prot_lvl = 100;
70 
71 /* Elliptic Curve Cryptographic support */
72 #ifdef CFG_CRYPTO_ECC
73 static const bool crypto_ecc_en = 1;
74 #else
75 static const bool crypto_ecc_en;
76 #endif
77 
78 /*
79  * Trusted storage anti rollback protection level
80  * 0 (or missing): No antirollback protection (default)
81  * 100: Antirollback enforced at REE level
82  * 1000: Antirollback TEE-controlled hardware
83  */
84 #ifdef CFG_RPMB_FS
85 static const uint32_t ts_antiroll_prot_lvl = 1000;
86 #else
87 static const uint32_t ts_antiroll_prot_lvl;
88 #endif
89 
90 /* Trusted OS implementation version */
91 static const char trustedos_impl_version[] = TO_STR(TEE_IMPL_VERSION);
92 
93 /* Trusted OS implementation version (binary value) */
94 static const uint32_t trustedos_impl_bin_version; /* 0 by default */
95 
96 /* Trusted OS implementation manufacturer name */
97 static const char trustedos_manufacturer[] = TO_STR(CFG_TEE_MANUFACTURER);
98 
99 /* Trusted firmware version */
100 static const char fw_impl_version[] = TO_STR(CFG_TEE_FW_IMPL_VERSION);
101 
102 /* Trusted firmware version (binary value) */
103 static const uint32_t fw_impl_bin_version; /* 0 by default */
104 
105 /* Trusted firmware manufacturer name */
106 static const char fw_manufacturer[] = TO_STR(CFG_TEE_FW_MANUFACTURER);
107 
108 static TEE_Result get_prop_tee_dev_id(struct tee_ta_session *sess __unused,
109 				      void *buf, size_t *blen)
110 {
111 	TEE_Result res;
112 	TEE_UUID uuid;
113 	const size_t nslen = 5;
114 	uint8_t data[5 + FVR_DIE_ID_NUM_REGS * sizeof(uint32_t)] = {
115 	    'O', 'P', 'T', 'E', 'E' };
116 
117 	if (*blen < sizeof(uuid)) {
118 		*blen = sizeof(uuid);
119 		return TEE_ERROR_SHORT_BUFFER;
120 	}
121 	*blen = sizeof(uuid);
122 
123 	if (tee_otp_get_die_id(data + nslen, sizeof(data) - nslen))
124 		return TEE_ERROR_BAD_STATE;
125 
126 	res = tee_hash_createdigest(TEE_ALG_SHA256, data, sizeof(data),
127 				    (uint8_t *)&uuid, sizeof(uuid));
128 	if (res != TEE_SUCCESS)
129 		return TEE_ERROR_BAD_STATE;
130 
131 	/*
132 	 * Changes the random value into and UUID as specifiec
133 	 * in RFC 4122. The magic values are from the example
134 	 * code in the RFC.
135 	 *
136 	 * TEE_UUID is defined slightly different from the RFC,
137 	 * but close enough for our purpose.
138 	 */
139 
140 	uuid.timeHiAndVersion &= 0x0fff;
141 	uuid.timeHiAndVersion |= 5 << 12;
142 
143 	/* uuid.clock_seq_hi_and_reserved in the RFC */
144 	uuid.clockSeqAndNode[0] &= 0x3f;
145 	uuid.clockSeqAndNode[0] |= 0x80;
146 
147 	return tee_svc_copy_to_user(buf, &uuid, sizeof(TEE_UUID));
148 }
149 
150 static TEE_Result get_prop_tee_sys_time_prot_level(
151 			struct tee_ta_session *sess __unused,
152 			void *buf, size_t *blen)
153 {
154 	uint32_t prot;
155 
156 	if (*blen < sizeof(prot)) {
157 		*blen = sizeof(prot);
158 		return TEE_ERROR_SHORT_BUFFER;
159 	}
160 	*blen = sizeof(prot);
161 	prot = tee_time_get_sys_time_protection_level();
162 	return tee_svc_copy_to_user(buf, &prot, sizeof(prot));
163 }
164 
165 static TEE_Result get_prop_client_id(struct tee_ta_session *sess __unused,
166 				     void *buf, size_t *blen)
167 {
168 	if (*blen < sizeof(TEE_Identity)) {
169 		*blen = sizeof(TEE_Identity);
170 		return TEE_ERROR_SHORT_BUFFER;
171 	}
172 	*blen = sizeof(TEE_Identity);
173 	return tee_svc_copy_to_user(buf, &sess->clnt_id, sizeof(TEE_Identity));
174 }
175 
176 static TEE_Result get_prop_ta_app_id(struct tee_ta_session *sess,
177 				     void *buf, size_t *blen)
178 {
179 	if (*blen < sizeof(TEE_UUID)) {
180 		*blen = sizeof(TEE_UUID);
181 		return TEE_ERROR_SHORT_BUFFER;
182 	}
183 	*blen = sizeof(TEE_UUID);
184 	return tee_svc_copy_to_user(buf, &sess->ctx->uuid, sizeof(TEE_UUID));
185 }
186 
187 /* Properties of the set TEE_PROPSET_CURRENT_CLIENT */
188 const struct tee_props tee_propset_client[] = {
189 	{
190 		.name = "gpd.client.identity",
191 		.prop_type = USER_TA_PROP_TYPE_IDENTITY,
192 		.get_prop_func = get_prop_client_id
193 	},
194 };
195 
196 /* Properties of the set TEE_PROPSET_CURRENT_TA */
197 const struct tee_props tee_propset_ta[] = {
198 	{
199 		.name = "gpd.ta.appID",
200 		.prop_type = USER_TA_PROP_TYPE_UUID,
201 		.get_prop_func = get_prop_ta_app_id
202 	},
203 
204 	/*
205 	 * Following properties are processed directly in libutee:
206 	 *	TA_PROP_STR_SINGLE_INSTANCE
207 	 *	TA_PROP_STR_MULTI_SESSION
208 	 *	TA_PROP_STR_KEEP_ALIVE
209 	 *	TA_PROP_STR_DATA_SIZE
210 	 *	TA_PROP_STR_STACK_SIZE
211 	 *	TA_PROP_STR_VERSION
212 	 *	TA_PROP_STR_DESCRIPTION
213 	 *	USER_TA_PROP_TYPE_STRING,
214 	 *	TA_DESCRIPTION
215 	 */
216 };
217 
218 /* Properties of the set TEE_PROPSET_TEE_IMPLEMENTATION */
219 const struct tee_props tee_propset_tee[] = {
220 	{
221 		.name = "gpd.tee.apiversion",
222 		.prop_type = USER_TA_PROP_TYPE_STRING,
223 		.data = api_vers,
224 		.len = sizeof(api_vers),
225 	},
226 	{
227 		.name = "gpd.tee.description",
228 		.prop_type = USER_TA_PROP_TYPE_STRING,
229 		.data = descr, .len = sizeof(descr)
230 	},
231 	{
232 		.name = "gpd.tee.deviceID",
233 		.prop_type = USER_TA_PROP_TYPE_UUID,
234 		.get_prop_func = get_prop_tee_dev_id
235 	},
236 	{
237 		.name = "gpd.tee.systemTime.protectionLevel",
238 		.prop_type = USER_TA_PROP_TYPE_U32,
239 		.get_prop_func = get_prop_tee_sys_time_prot_level
240 	},
241 	{
242 		.name = "gpd.tee.TAPersistentTime.protectionLevel",
243 		.prop_type = USER_TA_PROP_TYPE_U32,
244 		.data = &ta_time_prot_lvl,
245 		.len = sizeof(ta_time_prot_lvl)
246 	},
247 	{
248 		.name = "gpd.tee.cryptography.ecc",
249 		.prop_type = USER_TA_PROP_TYPE_BOOL,
250 		.data = &crypto_ecc_en,
251 		.len = sizeof(crypto_ecc_en)
252 	},
253 	{
254 		.name = "gpd.tee.trustedStorage.antiRollback.protectionLevel",
255 		.prop_type = USER_TA_PROP_TYPE_U32,
256 		.data = &ts_antiroll_prot_lvl,
257 		.len = sizeof(ts_antiroll_prot_lvl)
258 	},
259 	{
260 		.name = "gpd.tee.trustedos.implementation.version",
261 		.prop_type = USER_TA_PROP_TYPE_STRING,
262 		.data = trustedos_impl_version,
263 		.len = sizeof(trustedos_impl_version)
264 	},
265 	{
266 		.name = "gpd.tee.trustedos.implementation.binaryversion",
267 		.prop_type = USER_TA_PROP_TYPE_U32,
268 		.data = &trustedos_impl_bin_version,
269 		.len = sizeof(trustedos_impl_bin_version)
270 	},
271 	{
272 		.name = "gpd.tee.trustedos.manufacturer",
273 		.prop_type = USER_TA_PROP_TYPE_STRING,
274 		.data = trustedos_manufacturer,
275 		.len = sizeof(trustedos_manufacturer)
276 	},
277 	{
278 		.name = "gpd.tee.firmware.implementation.version",
279 		.prop_type = USER_TA_PROP_TYPE_STRING,
280 		.data = fw_impl_version,
281 		.len = sizeof(fw_impl_version)
282 	},
283 	{
284 		.name = "gpd.tee.firmware.implementation.binaryversion",
285 		.prop_type = USER_TA_PROP_TYPE_U32,
286 		.data = &fw_impl_bin_version,
287 		.len = sizeof(fw_impl_bin_version)
288 	},
289 	{
290 		.name = "gpd.tee.firmware.manufacturer",
291 		.prop_type = USER_TA_PROP_TYPE_STRING,
292 		.data = fw_manufacturer,
293 		.len = sizeof(fw_manufacturer)
294 	},
295 
296 	/*
297 	 * Following properties are processed directly in libutee:
298 	 *	gpd.tee.arith.maxBigIntSize
299 	 */
300 };
301 
302 __weak const struct tee_vendor_props vendor_props_client;
303 __weak const struct tee_vendor_props vendor_props_ta;
304 __weak const struct tee_vendor_props vendor_props_tee;
305 
306 static void get_prop_set(unsigned long prop_set,
307 			 const struct tee_props **props,
308 			 size_t *size,
309 			 const struct tee_props **vendor_props,
310 			 size_t *vendor_size)
311 {
312 	if ((TEE_PropSetHandle)prop_set == TEE_PROPSET_CURRENT_CLIENT) {
313 		*props = tee_propset_client;
314 		*size = ARRAY_SIZE(tee_propset_client);
315 		*vendor_props = vendor_props_client.props;
316 		*vendor_size = vendor_props_client.len;
317 	} else if ((TEE_PropSetHandle)prop_set == TEE_PROPSET_CURRENT_TA) {
318 		*props = tee_propset_ta;
319 		*size = ARRAY_SIZE(tee_propset_ta);
320 		*vendor_props = vendor_props_ta.props;
321 		*vendor_size = vendor_props_ta.len;
322 	} else if ((TEE_PropSetHandle)prop_set ==
323 		   TEE_PROPSET_TEE_IMPLEMENTATION) {
324 		*props = tee_propset_tee;
325 		*size = ARRAY_SIZE(tee_propset_tee);
326 		*vendor_props = vendor_props_tee.props;
327 		*vendor_size = vendor_props_tee.len;
328 	} else {
329 		*props = NULL;
330 		*size = 0;
331 		*vendor_props = NULL;
332 		*vendor_size = 0;
333 	}
334 }
335 
336 static const struct tee_props *get_prop_struct(unsigned long prop_set,
337 					       unsigned long index)
338 {
339 	const struct tee_props *props;
340 	const struct tee_props *vendor_props;
341 	size_t size;
342 	size_t vendor_size;
343 
344 	get_prop_set(prop_set, &props, &size, &vendor_props, &vendor_size);
345 
346 	if (index < size)
347 		return &(props[index]);
348 	index -= size;
349 
350 	if (index < vendor_size)
351 		return &(vendor_props[index]);
352 
353 	return NULL;
354 }
355 
356 /*
357  * prop_set is part of TEE_PROPSET_xxx
358  * index is the index in the Property Set to retrieve
359  * if name is not NULL, the name of "index" property is returned
360  * if buf is not NULL, the property is returned
361  */
362 TEE_Result syscall_get_property(unsigned long prop_set,
363 				unsigned long index,
364 				void *name, uint32_t *name_len,
365 				void *buf, uint32_t *blen,
366 				uint32_t *prop_type)
367 {
368 	struct tee_ta_session *sess;
369 	TEE_Result res;
370 	TEE_Result res2;
371 	const struct tee_props *prop;
372 	uint32_t klen;
373 	size_t klen_size;
374 	uint32_t elen;
375 
376 	prop = get_prop_struct(prop_set, index);
377 	if (!prop)
378 		return TEE_ERROR_ITEM_NOT_FOUND;
379 
380 	res = tee_ta_get_current_session(&sess);
381 	if (res != TEE_SUCCESS)
382 		return res;
383 
384 	/* Get the property type */
385 	if (prop_type) {
386 		res = tee_svc_copy_to_user(prop_type, &prop->prop_type,
387 					   sizeof(*prop_type));
388 		if (res != TEE_SUCCESS)
389 			return res;
390 	}
391 
392 	/* Get the property */
393 	if (buf && blen) {
394 		res = tee_svc_copy_from_user(&klen, blen, sizeof(klen));
395 		if (res != TEE_SUCCESS)
396 			return res;
397 
398 		if (prop->get_prop_func) {
399 			klen_size = klen;
400 			res = prop->get_prop_func(sess, buf, &klen_size);
401 			klen = klen_size;
402 			res2 = tee_svc_copy_to_user(blen, &klen, sizeof(*blen));
403 		} else {
404 			if (klen < prop->len)
405 				res = TEE_ERROR_SHORT_BUFFER;
406 			else
407 				res = tee_svc_copy_to_user(buf, prop->data,
408 							   prop->len);
409 			res2 = tee_svc_copy_to_user(blen, &prop->len,
410 						    sizeof(*blen));
411 		}
412 		if (res2 != TEE_SUCCESS)
413 			return res2;
414 		if (res != TEE_SUCCESS)
415 			return res;
416 	}
417 
418 	/* Get the property name */
419 	if (name && name_len) {
420 		res = tee_svc_copy_from_user(&klen, name_len, sizeof(klen));
421 		if (res != TEE_SUCCESS)
422 			return res;
423 
424 		elen = strlen(prop->name) + 1;
425 
426 		if (klen < elen)
427 			res = TEE_ERROR_SHORT_BUFFER;
428 		else
429 			res = tee_svc_copy_to_user(name, prop->name, elen);
430 		res2 = tee_svc_copy_to_user(name_len, &elen, sizeof(*name_len));
431 		if (res2 != TEE_SUCCESS)
432 			return res2;
433 		if (res != TEE_SUCCESS)
434 			return res;
435 	}
436 
437 	return res;
438 }
439 
440 /*
441  * prop_set is part of TEE_PROPSET_xxx
442  */
443 TEE_Result syscall_get_property_name_to_index(unsigned long prop_set,
444 					      void *name,
445 					      unsigned long name_len,
446 					      uint32_t *index)
447 {
448 	TEE_Result res;
449 	struct tee_ta_session *sess;
450 	const struct tee_props *props;
451 	size_t size;
452 	const struct tee_props *vendor_props;
453 	size_t vendor_size;
454 	char *kname = 0;
455 	uint32_t i;
456 
457 	get_prop_set(prop_set, &props, &size, &vendor_props, &vendor_size);
458 	if (!props)
459 		return TEE_ERROR_ITEM_NOT_FOUND;
460 
461 	res = tee_ta_get_current_session(&sess);
462 	if (res != TEE_SUCCESS)
463 		goto out;
464 
465 	if (!name || !name_len) {
466 		res = TEE_ERROR_BAD_PARAMETERS;
467 		goto out;
468 	}
469 
470 	kname = malloc(name_len);
471 	if (!kname)
472 		return TEE_ERROR_OUT_OF_MEMORY;
473 	res = tee_svc_copy_from_user(kname, name, name_len);
474 	if (res != TEE_SUCCESS)
475 		goto out;
476 	kname[name_len - 1] = 0;
477 
478 	res = TEE_ERROR_ITEM_NOT_FOUND;
479 	for (i = 0; i < size; i++) {
480 		if (!strcmp(kname, props[i].name)) {
481 			res = tee_svc_copy_to_user(index, &i, sizeof(*index));
482 			goto out;
483 		}
484 	}
485 	for (i = size; i < size + vendor_size; i++) {
486 		if (!strcmp(kname, vendor_props[i - size].name)) {
487 			res = tee_svc_copy_to_user(index, &i, sizeof(*index));
488 			goto out;
489 		}
490 	}
491 
492 out:
493 	free_wipe(kname);
494 	return res;
495 }
496 
497 static TEE_Result utee_param_to_param(struct user_ta_ctx *utc,
498 				      struct tee_ta_param *p,
499 				      struct utee_params *up)
500 {
501 	size_t n = 0;
502 	uint32_t types = up->types;
503 
504 	p->types = types;
505 	for (n = 0; n < TEE_NUM_PARAMS; n++) {
506 		uintptr_t a = up->vals[n * 2];
507 		size_t b = up->vals[n * 2 + 1];
508 		uint32_t flags = TEE_MEMORY_ACCESS_READ |
509 				 TEE_MEMORY_ACCESS_ANY_OWNER;
510 
511 		switch (TEE_PARAM_TYPE_GET(types, n)) {
512 		case TEE_PARAM_TYPE_MEMREF_OUTPUT:
513 		case TEE_PARAM_TYPE_MEMREF_INOUT:
514 			flags |= TEE_MEMORY_ACCESS_WRITE;
515 			/*FALLTHROUGH*/
516 		case TEE_PARAM_TYPE_MEMREF_INPUT:
517 			p->u[n].mem.mobj = &mobj_virt;
518 			p->u[n].mem.offs = a;
519 			p->u[n].mem.size = b;
520 			if (tee_mmu_check_access_rights(&utc->uctx, flags, a,
521 							b))
522 				return TEE_ERROR_ACCESS_DENIED;
523 			break;
524 		case TEE_PARAM_TYPE_VALUE_INPUT:
525 		case TEE_PARAM_TYPE_VALUE_INOUT:
526 			p->u[n].val.a = a;
527 			p->u[n].val.b = b;
528 			break;
529 		default:
530 			memset(&p->u[n], 0, sizeof(p->u[n]));
531 			break;
532 		}
533 	}
534 
535 	return TEE_SUCCESS;
536 }
537 
538 static TEE_Result alloc_temp_sec_mem(size_t size, struct mobj **mobj,
539 				     uint8_t **va)
540 {
541 	/* Allocate section in secure DDR */
542 #ifdef CFG_PAGED_USER_TA
543 	*mobj = mobj_seccpy_shm_alloc(size);
544 #else
545 	*mobj = mobj_mm_alloc(mobj_sec_ddr, size, &tee_mm_sec_ddr);
546 #endif
547 	if (!*mobj)
548 		return TEE_ERROR_GENERIC;
549 
550 	*va = mobj_get_va(*mobj, 0);
551 	return TEE_SUCCESS;
552 }
553 
554 /*
555  * TA invokes some TA with parameter.
556  * If some parameters are memory references:
557  * - either the memref is inside TA private RAM: TA is not allowed to expose
558  *   its private RAM: use a temporary memory buffer and copy the data.
559  * - or the memref is not in the TA private RAM:
560  *   - if the memref was mapped to the TA, TA is allowed to expose it.
561  *   - if so, converts memref virtual address into a physical address.
562  */
563 static TEE_Result tee_svc_copy_param(struct tee_ta_session *sess,
564 				     struct tee_ta_session *called_sess,
565 				     struct utee_params *callee_params,
566 				     struct tee_ta_param *param,
567 				     void *tmp_buf_va[TEE_NUM_PARAMS],
568 				     size_t tmp_buf_size[TEE_NUM_PARAMS],
569 				     struct mobj **mobj_tmp)
570 {
571 	struct user_ta_ctx *utc = to_user_ta_ctx(sess->ctx);
572 	bool ta_private_memref[TEE_NUM_PARAMS] = { false, };
573 	TEE_Result res = TEE_SUCCESS;
574 	size_t dst_offs = 0;
575 	size_t req_mem = 0;
576 	uint8_t *dst = 0;
577 	void *va = NULL;
578 	size_t n = 0;
579 	size_t s = 0;
580 
581 	/* fill 'param' input struct with caller params description buffer */
582 	if (!callee_params) {
583 		memset(param, 0, sizeof(*param));
584 	} else {
585 		uint32_t flags = TEE_MEMORY_ACCESS_READ |
586 				 TEE_MEMORY_ACCESS_ANY_OWNER;
587 
588 		res = tee_mmu_check_access_rights(&utc->uctx, flags,
589 						  (uaddr_t)callee_params,
590 						  sizeof(struct utee_params));
591 		if (res != TEE_SUCCESS)
592 			return res;
593 		res = utee_param_to_param(utc, param, callee_params);
594 		if (res != TEE_SUCCESS)
595 			return res;
596 	}
597 
598 	if (called_sess && is_pseudo_ta_ctx(called_sess->ctx)) {
599 		/* pseudo TA borrows the mapping of the calling TA */
600 		return TEE_SUCCESS;
601 	}
602 
603 	/* All mobj in param are of type MOJB_TYPE_VIRT */
604 
605 	for (n = 0; n < TEE_NUM_PARAMS; n++) {
606 
607 		ta_private_memref[n] = false;
608 
609 		switch (TEE_PARAM_TYPE_GET(param->types, n)) {
610 		case TEE_PARAM_TYPE_MEMREF_INPUT:
611 		case TEE_PARAM_TYPE_MEMREF_OUTPUT:
612 		case TEE_PARAM_TYPE_MEMREF_INOUT:
613 			va = (void *)param->u[n].mem.offs;
614 			s = param->u[n].mem.size;
615 			if (!va) {
616 				if (s)
617 					return TEE_ERROR_BAD_PARAMETERS;
618 				break;
619 			}
620 			/* uTA cannot expose its private memory */
621 			if (tee_mmu_is_vbuf_inside_um_private(&utc->uctx, va,
622 							      s)) {
623 
624 				s = ROUNDUP(s, sizeof(uint32_t));
625 				if (ADD_OVERFLOW(req_mem, s, &req_mem))
626 					return TEE_ERROR_BAD_PARAMETERS;
627 				ta_private_memref[n] = true;
628 				break;
629 			}
630 
631 			res = tee_mmu_vbuf_to_mobj_offs(&utc->uctx, va, s,
632 							&param->u[n].mem.mobj,
633 							&param->u[n].mem.offs);
634 			if (res != TEE_SUCCESS)
635 				return res;
636 			break;
637 		default:
638 			break;
639 		}
640 	}
641 
642 	if (req_mem == 0)
643 		return TEE_SUCCESS;
644 
645 	res = alloc_temp_sec_mem(req_mem, mobj_tmp, &dst);
646 	if (res != TEE_SUCCESS)
647 		return res;
648 	dst_offs = 0;
649 
650 	for (n = 0; n < TEE_NUM_PARAMS; n++) {
651 
652 		if (!ta_private_memref[n])
653 			continue;
654 
655 		s = ROUNDUP(param->u[n].mem.size, sizeof(uint32_t));
656 
657 		switch (TEE_PARAM_TYPE_GET(param->types, n)) {
658 		case TEE_PARAM_TYPE_MEMREF_INPUT:
659 		case TEE_PARAM_TYPE_MEMREF_INOUT:
660 			va = (void *)param->u[n].mem.offs;
661 			if (va) {
662 				res = tee_svc_copy_from_user(dst, va,
663 						param->u[n].mem.size);
664 				if (res != TEE_SUCCESS)
665 					return res;
666 				param->u[n].mem.offs = dst_offs;
667 				param->u[n].mem.mobj = *mobj_tmp;
668 				tmp_buf_va[n] = dst;
669 				tmp_buf_size[n] = param->u[n].mem.size;
670 				dst += s;
671 				dst_offs += s;
672 			}
673 			break;
674 
675 		case TEE_PARAM_TYPE_MEMREF_OUTPUT:
676 			va = (void *)param->u[n].mem.offs;
677 			if (va) {
678 				param->u[n].mem.offs = dst_offs;
679 				param->u[n].mem.mobj = *mobj_tmp;
680 				tmp_buf_va[n] = dst;
681 				tmp_buf_size[n] = param->u[n].mem.size;
682 				dst += s;
683 				dst_offs += s;
684 			}
685 			break;
686 
687 		default:
688 			continue;
689 		}
690 	}
691 
692 	return TEE_SUCCESS;
693 }
694 
695 /*
696  * Back from execution of service: update parameters passed from TA:
697  * If some parameters were memory references:
698  * - either the memref was temporary: copy back data and update size
699  * - or it was the original TA memref: update only the size value.
700  */
701 static TEE_Result tee_svc_update_out_param(
702 		struct tee_ta_param *param,
703 		void *tmp_buf_va[TEE_NUM_PARAMS],
704 		size_t tmp_buf_size[TEE_NUM_PARAMS],
705 		struct utee_params *usr_param)
706 {
707 	size_t n;
708 	uint64_t *vals = usr_param->vals;
709 
710 	for (n = 0; n < TEE_NUM_PARAMS; n++) {
711 		switch (TEE_PARAM_TYPE_GET(param->types, n)) {
712 		case TEE_PARAM_TYPE_MEMREF_OUTPUT:
713 		case TEE_PARAM_TYPE_MEMREF_INOUT:
714 			/*
715 			 * Memory copy is only needed if there's a temporary
716 			 * buffer involved, tmp_buf_va[n] is only update if
717 			 * a temporary buffer is used. Otherwise only the
718 			 * size needs to be updated.
719 			 */
720 			if (tmp_buf_va[n] &&
721 			    param->u[n].mem.size <= vals[n * 2 + 1]) {
722 				void *src = tmp_buf_va[n];
723 				void *dst = (void *)(uintptr_t)vals[n * 2];
724 				TEE_Result res;
725 
726 				/*
727 				 * TA is allowed to return a size larger than
728 				 * the original size. However, in such cases no
729 				 * data should be synchronized as per TEE Client
730 				 * API spec.
731 				 */
732 				if (param->u[n].mem.size <= tmp_buf_size[n]) {
733 					res = tee_svc_copy_to_user(dst, src,
734 							param->u[n].mem.size);
735 					if (res != TEE_SUCCESS)
736 						return res;
737 				}
738 			}
739 			usr_param->vals[n * 2 + 1] = param->u[n].mem.size;
740 			break;
741 
742 		case TEE_PARAM_TYPE_VALUE_OUTPUT:
743 		case TEE_PARAM_TYPE_VALUE_INOUT:
744 			vals[n * 2] = param->u[n].val.a;
745 			vals[n * 2 + 1] = param->u[n].val.b;
746 			break;
747 
748 		default:
749 			continue;
750 		}
751 	}
752 
753 	return TEE_SUCCESS;
754 }
755 
756 /* Called when a TA calls an OpenSession on another TA */
757 TEE_Result syscall_open_ta_session(const TEE_UUID *dest,
758 			unsigned long cancel_req_to,
759 			struct utee_params *usr_param, uint32_t *ta_sess,
760 			uint32_t *ret_orig)
761 {
762 	TEE_Result res;
763 	uint32_t ret_o = TEE_ORIGIN_TEE;
764 	struct tee_ta_session *s = NULL;
765 	struct tee_ta_session *sess;
766 	struct mobj *mobj_param = NULL;
767 	TEE_UUID *uuid = malloc(sizeof(TEE_UUID));
768 	struct tee_ta_param *param = malloc(sizeof(struct tee_ta_param));
769 	TEE_Identity *clnt_id = malloc(sizeof(TEE_Identity));
770 	void *tmp_buf_va[TEE_NUM_PARAMS] = { NULL };
771 	size_t tmp_buf_size[TEE_NUM_PARAMS] = { 0 };
772 	struct user_ta_ctx *utc;
773 
774 	if (uuid == NULL || param == NULL || clnt_id == NULL) {
775 		res = TEE_ERROR_OUT_OF_MEMORY;
776 		goto out_free_only;
777 	}
778 
779 	memset(param, 0, sizeof(struct tee_ta_param));
780 
781 	res = tee_ta_get_current_session(&sess);
782 	if (res != TEE_SUCCESS)
783 		goto out_free_only;
784 	utc = to_user_ta_ctx(sess->ctx);
785 
786 	res = tee_svc_copy_from_user(uuid, dest, sizeof(TEE_UUID));
787 	if (res != TEE_SUCCESS)
788 		goto function_exit;
789 
790 	clnt_id->login = TEE_LOGIN_TRUSTED_APP;
791 	memcpy(&clnt_id->uuid, &sess->ctx->uuid, sizeof(TEE_UUID));
792 
793 	res = tee_svc_copy_param(sess, NULL, usr_param, param, tmp_buf_va,
794 				 tmp_buf_size, &mobj_param);
795 	if (res != TEE_SUCCESS)
796 		goto function_exit;
797 
798 	res = tee_ta_open_session(&ret_o, &s, &utc->open_sessions, uuid,
799 				  clnt_id, cancel_req_to, param);
800 	tee_mmu_set_ctx(&utc->uctx.ctx);
801 	if (res != TEE_SUCCESS)
802 		goto function_exit;
803 
804 	res = tee_svc_update_out_param(param, tmp_buf_va, tmp_buf_size,
805 				       usr_param);
806 
807 function_exit:
808 	mobj_put_wipe(mobj_param);
809 	if (res == TEE_SUCCESS)
810 		tee_svc_copy_to_user(ta_sess, &s->id, sizeof(s->id));
811 	tee_svc_copy_to_user(ret_orig, &ret_o, sizeof(ret_o));
812 
813 out_free_only:
814 	free_wipe(param);
815 	free_wipe(uuid);
816 	free_wipe(clnt_id);
817 	return res;
818 }
819 
820 TEE_Result syscall_close_ta_session(unsigned long ta_sess)
821 {
822 	TEE_Result res;
823 	struct tee_ta_session *sess;
824 	TEE_Identity clnt_id;
825 	struct tee_ta_session *s = NULL;
826 	struct user_ta_ctx *utc;
827 
828 	res = tee_ta_get_current_session(&sess);
829 	if (res != TEE_SUCCESS)
830 		return res;
831 	utc = to_user_ta_ctx(sess->ctx);
832 	s = tee_ta_find_session(ta_sess, &utc->open_sessions);
833 
834 	clnt_id.login = TEE_LOGIN_TRUSTED_APP;
835 	memcpy(&clnt_id.uuid, &sess->ctx->uuid, sizeof(TEE_UUID));
836 
837 	return tee_ta_close_session(s, &utc->open_sessions, &clnt_id);
838 }
839 
840 TEE_Result syscall_invoke_ta_command(unsigned long ta_sess,
841 			unsigned long cancel_req_to, unsigned long cmd_id,
842 			struct utee_params *usr_param, uint32_t *ret_orig)
843 {
844 	TEE_Result res;
845 	TEE_Result res2;
846 	uint32_t ret_o = TEE_ORIGIN_TEE;
847 	struct tee_ta_param param = { 0 };
848 	TEE_Identity clnt_id;
849 	struct tee_ta_session *sess;
850 	struct tee_ta_session *called_sess;
851 	struct mobj *mobj_param = NULL;
852 	void *tmp_buf_va[TEE_NUM_PARAMS] = { NULL };
853 	size_t tmp_buf_size[TEE_NUM_PARAMS] = { };
854 	struct user_ta_ctx *utc;
855 
856 	res = tee_ta_get_current_session(&sess);
857 	if (res != TEE_SUCCESS)
858 		return res;
859 	utc = to_user_ta_ctx(sess->ctx);
860 
861 	called_sess = tee_ta_get_session((uint32_t)ta_sess, true,
862 				&utc->open_sessions);
863 	if (!called_sess)
864 		return TEE_ERROR_BAD_PARAMETERS;
865 
866 	clnt_id.login = TEE_LOGIN_TRUSTED_APP;
867 	memcpy(&clnt_id.uuid, &sess->ctx->uuid, sizeof(TEE_UUID));
868 
869 	res = tee_svc_copy_param(sess, called_sess, usr_param, &param,
870 				 tmp_buf_va, tmp_buf_size, &mobj_param);
871 	if (res != TEE_SUCCESS)
872 		goto function_exit;
873 
874 	res = tee_ta_invoke_command(&ret_o, called_sess, &clnt_id,
875 				    cancel_req_to, cmd_id, &param);
876 	if (res == TEE_ERROR_TARGET_DEAD)
877 		goto function_exit;
878 
879 	res2 = tee_svc_update_out_param(&param, tmp_buf_va, tmp_buf_size,
880 					usr_param);
881 	if (res2 != TEE_SUCCESS) {
882 		/*
883 		 * Spec for TEE_InvokeTACommand() says:
884 		 * "If the return origin is different from
885 		 * TEE_ORIGIN_TRUSTED_APP, then the function has failed
886 		 * before it could reach the destination Trusted
887 		 * Application."
888 		 *
889 		 * But if we can't update params to the caller we have no
890 		 * choice we need to return some error to indicate that
891 		 * parameters aren't updated as expected.
892 		 */
893 		ret_o = TEE_ORIGIN_TEE;
894 		res = res2;
895 	}
896 
897 function_exit:
898 	tee_ta_put_session(called_sess);
899 	mobj_put_wipe(mobj_param);
900 	if (ret_orig)
901 		tee_svc_copy_to_user(ret_orig, &ret_o, sizeof(ret_o));
902 	return res;
903 }
904 
905 TEE_Result syscall_check_access_rights(unsigned long flags, const void *buf,
906 				       size_t len)
907 {
908 	struct tee_ta_session *s = NULL;
909 	TEE_Result res = TEE_SUCCESS;
910 
911 	res = tee_ta_get_current_session(&s);
912 	if (res != TEE_SUCCESS)
913 		return res;
914 
915 	return tee_mmu_check_access_rights(&to_user_ta_ctx(s->ctx)->uctx, flags,
916 					   (uaddr_t)buf, len);
917 }
918 
919 TEE_Result tee_svc_copy_from_user(void *kaddr, const void *uaddr, size_t len)
920 {
921 	struct tee_ta_session *s = NULL;
922 	TEE_Result res = TEE_SUCCESS;
923 
924 	res = tee_ta_get_current_session(&s);
925 	if (res != TEE_SUCCESS)
926 		return res;
927 
928 	res = tee_mmu_check_access_rights(&to_user_ta_ctx(s->ctx)->uctx,
929 					  TEE_MEMORY_ACCESS_READ |
930 					  TEE_MEMORY_ACCESS_ANY_OWNER,
931 					  (uaddr_t)uaddr, len);
932 	if (res != TEE_SUCCESS)
933 		return res;
934 
935 	memcpy(kaddr, uaddr, len);
936 	return TEE_SUCCESS;
937 }
938 
939 TEE_Result tee_svc_copy_to_user(void *uaddr, const void *kaddr, size_t len)
940 {
941 	struct tee_ta_session *s = NULL;
942 	TEE_Result res = TEE_SUCCESS;
943 
944 	res = tee_ta_get_current_session(&s);
945 	if (res != TEE_SUCCESS)
946 		return res;
947 
948 	res = tee_mmu_check_access_rights(&to_user_ta_ctx(s->ctx)->uctx,
949 					  TEE_MEMORY_ACCESS_WRITE |
950 					  TEE_MEMORY_ACCESS_ANY_OWNER,
951 					  (uaddr_t)uaddr, len);
952 	if (res != TEE_SUCCESS)
953 		return res;
954 
955 	memcpy(uaddr, kaddr, len);
956 	return TEE_SUCCESS;
957 }
958 
959 TEE_Result tee_svc_copy_kaddr_to_uref(uint32_t *uref, void *kaddr)
960 {
961 	uint32_t ref = tee_svc_kaddr_to_uref(kaddr);
962 
963 	return tee_svc_copy_to_user(uref, &ref, sizeof(ref));
964 }
965 
966 TEE_Result syscall_get_cancellation_flag(uint32_t *cancel)
967 {
968 	TEE_Result res;
969 	struct tee_ta_session *s = NULL;
970 	uint32_t c;
971 
972 	res = tee_ta_get_current_session(&s);
973 	if (res != TEE_SUCCESS)
974 		return res;
975 
976 	c = tee_ta_session_is_cancelled(s, NULL);
977 
978 	return tee_svc_copy_to_user(cancel, &c, sizeof(c));
979 }
980 
981 TEE_Result syscall_unmask_cancellation(uint32_t *old_mask)
982 {
983 	TEE_Result res;
984 	struct tee_ta_session *s = NULL;
985 	uint32_t m;
986 
987 	res = tee_ta_get_current_session(&s);
988 	if (res != TEE_SUCCESS)
989 		return res;
990 
991 	m = s->cancel_mask;
992 	s->cancel_mask = false;
993 	return tee_svc_copy_to_user(old_mask, &m, sizeof(m));
994 }
995 
996 TEE_Result syscall_mask_cancellation(uint32_t *old_mask)
997 {
998 	TEE_Result res;
999 	struct tee_ta_session *s = NULL;
1000 	uint32_t m;
1001 
1002 	res = tee_ta_get_current_session(&s);
1003 	if (res != TEE_SUCCESS)
1004 		return res;
1005 
1006 	m = s->cancel_mask;
1007 	s->cancel_mask = true;
1008 	return tee_svc_copy_to_user(old_mask, &m, sizeof(m));
1009 }
1010 
1011 TEE_Result syscall_wait(unsigned long timeout)
1012 {
1013 	TEE_Result res = TEE_SUCCESS;
1014 	uint32_t mytime = 0;
1015 	struct tee_ta_session *s;
1016 	TEE_Time base_time;
1017 	TEE_Time current_time;
1018 
1019 	res = tee_ta_get_current_session(&s);
1020 	if (res != TEE_SUCCESS)
1021 		return res;
1022 
1023 	res = tee_time_get_sys_time(&base_time);
1024 	if (res != TEE_SUCCESS)
1025 		return res;
1026 
1027 	while (true) {
1028 		res = tee_time_get_sys_time(&current_time);
1029 		if (res != TEE_SUCCESS)
1030 			return res;
1031 
1032 		if (tee_ta_session_is_cancelled(s, &current_time))
1033 			return TEE_ERROR_CANCEL;
1034 
1035 		mytime = (current_time.seconds - base_time.seconds) * 1000 +
1036 		    (int)current_time.millis - (int)base_time.millis;
1037 		if (mytime >= timeout)
1038 			return TEE_SUCCESS;
1039 
1040 		tee_time_wait(timeout - mytime);
1041 	}
1042 
1043 	return res;
1044 }
1045 
1046 TEE_Result syscall_get_time(unsigned long cat, TEE_Time *mytime)
1047 {
1048 	TEE_Result res, res2;
1049 	struct tee_ta_session *s = NULL;
1050 	TEE_Time t;
1051 
1052 	res = tee_ta_get_current_session(&s);
1053 	if (res != TEE_SUCCESS)
1054 		return res;
1055 
1056 	switch (cat) {
1057 	case UTEE_TIME_CAT_SYSTEM:
1058 		res = tee_time_get_sys_time(&t);
1059 		break;
1060 	case UTEE_TIME_CAT_TA_PERSISTENT:
1061 		res = tee_time_get_ta_time((const void *)&s->ctx->uuid, &t);
1062 		break;
1063 	case UTEE_TIME_CAT_REE:
1064 		res = tee_time_get_ree_time(&t);
1065 		break;
1066 	default:
1067 		res = TEE_ERROR_BAD_PARAMETERS;
1068 		break;
1069 	}
1070 
1071 	if (res == TEE_SUCCESS || res == TEE_ERROR_OVERFLOW) {
1072 		res2 = tee_svc_copy_to_user(mytime, &t, sizeof(t));
1073 		if (res2 != TEE_SUCCESS)
1074 			res = res2;
1075 	}
1076 
1077 	return res;
1078 }
1079 
1080 TEE_Result syscall_set_ta_time(const TEE_Time *mytime)
1081 {
1082 	TEE_Result res;
1083 	struct tee_ta_session *s = NULL;
1084 	TEE_Time t;
1085 
1086 	res = tee_ta_get_current_session(&s);
1087 	if (res != TEE_SUCCESS)
1088 		return res;
1089 
1090 	res = tee_svc_copy_from_user(&t, mytime, sizeof(t));
1091 	if (res != TEE_SUCCESS)
1092 		return res;
1093 
1094 	return tee_time_set_ta_time((const void *)&s->ctx->uuid, &t);
1095 }
1096