xref: /optee_os/core/tee/tee_svc.c (revision 6069995790269f6a25a47324ec3753781f8e3745)
1b0104773SPascal Brand /*
2b0104773SPascal Brand  * Copyright (c) 2014, STMicroelectronics International N.V.
3b0104773SPascal Brand  * All rights reserved.
4b0104773SPascal Brand  *
5b0104773SPascal Brand  * Redistribution and use in source and binary forms, with or without
6b0104773SPascal Brand  * modification, are permitted provided that the following conditions are met:
7b0104773SPascal Brand  *
8b0104773SPascal Brand  * 1. Redistributions of source code must retain the above copyright notice,
9b0104773SPascal Brand  * this list of conditions and the following disclaimer.
10b0104773SPascal Brand  *
11b0104773SPascal Brand  * 2. Redistributions in binary form must reproduce the above copyright notice,
12b0104773SPascal Brand  * this list of conditions and the following disclaimer in the documentation
13b0104773SPascal Brand  * and/or other materials provided with the distribution.
14b0104773SPascal Brand  *
15b0104773SPascal Brand  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16b0104773SPascal Brand  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17b0104773SPascal Brand  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18b0104773SPascal Brand  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
19b0104773SPascal Brand  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
20b0104773SPascal Brand  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
21b0104773SPascal Brand  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
22b0104773SPascal Brand  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23b0104773SPascal Brand  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
24b0104773SPascal Brand  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
25b0104773SPascal Brand  * POSSIBILITY OF SUCH DAMAGE.
26b0104773SPascal Brand  */
272eb765fcSJens Wiklander #include <util.h>
28b0104773SPascal Brand #include <kernel/tee_common_otp.h>
29b0104773SPascal Brand #include <kernel/tee_common.h>
30b0104773SPascal Brand #include <kernel/tee_compat.h>
31b0104773SPascal Brand #include <tee_api_types.h>
32b0104773SPascal Brand #include <kernel/tee_ta_manager.h>
33b0104773SPascal Brand #include <utee_types.h>
34b0104773SPascal Brand #include <tee/tee_svc.h>
35ffe04039SJerome Forissier #include <tee/tee_cryp_utl.h>
36b0104773SPascal Brand #include <mm/tee_mmu.h>
37b0104773SPascal Brand #include <mm/tee_mm.h>
38b0104773SPascal Brand #include <kernel/tee_rpc.h>
39b0104773SPascal Brand #include <kernel/tee_rpc_types.h>
40b0104773SPascal Brand #include <kernel/tee_time.h>
41b0104773SPascal Brand 
42b0104773SPascal Brand #include <user_ta_header.h>
434de4bebcSJens Wiklander #include <trace.h>
444de4bebcSJens Wiklander #include <kernel/trace_ta.h>
45b0104773SPascal Brand #include <kernel/chip_services.h>
46b0104773SPascal Brand 
477c876f12SPascal Brand #if (CFG_TRACE_LEVEL == TRACE_FLOW) && defined(CFG_TEE_CORE_TA_TRACE)
484de4bebcSJens Wiklander void tee_svc_trace_syscall(int num)
494de4bebcSJens Wiklander {
504de4bebcSJens Wiklander 	/* #0 is syscall return, not really interesting */
514de4bebcSJens Wiklander 	if (num == 0)
524de4bebcSJens Wiklander 		return;
534de4bebcSJens Wiklander 	FMSG("syscall #%d", num);
544de4bebcSJens Wiklander }
554de4bebcSJens Wiklander #endif
564de4bebcSJens Wiklander 
577c876f12SPascal Brand void tee_svc_sys_log(const void *buf __unused, size_t len __unused)
58b0104773SPascal Brand {
597c876f12SPascal Brand #ifdef CFG_TEE_CORE_TA_TRACE
60b0104773SPascal Brand 	char *kbuf;
61b0104773SPascal Brand 
62b0104773SPascal Brand 	if (len == 0)
63b0104773SPascal Brand 		return;
64b0104773SPascal Brand 
65b0104773SPascal Brand 	kbuf = malloc(len);
66b0104773SPascal Brand 	if (kbuf == NULL)
67b0104773SPascal Brand 		return;
68c0346845SJens Wiklander 	*kbuf = '\0';
69b0104773SPascal Brand 
70b0104773SPascal Brand 	/* log as Info/Raw traces */
71b0104773SPascal Brand 	if (tee_svc_copy_from_user(NULL, kbuf, buf, len) == TEE_SUCCESS)
724de4bebcSJens Wiklander 		TAMSG_RAW("%.*s", (int)len, kbuf);
73b0104773SPascal Brand 
74b0104773SPascal Brand 	free(kbuf);
757c876f12SPascal Brand #endif
76b0104773SPascal Brand }
77b0104773SPascal Brand 
78b7fc217fSPascal Brand TEE_Result tee_svc_reserved(void)
79b7fc217fSPascal Brand {
80b7fc217fSPascal Brand 	return TEE_ERROR_GENERIC;
81b7fc217fSPascal Brand }
82b7fc217fSPascal Brand 
83197d17e7SSY Chiu TEE_Result tee_svc_not_supported(void)
84197d17e7SSY Chiu {
85197d17e7SSY Chiu 	return TEE_ERROR_NOT_SUPPORTED;
86197d17e7SSY Chiu }
87197d17e7SSY Chiu 
88cebdec51SJens Wiklander uint32_t tee_svc_sys_dummy(uint32_t *a __unused)
89b0104773SPascal Brand {
90b0104773SPascal Brand 	DMSG("tee_svc_sys_dummy: a 0x%x", (unsigned int)a);
91b0104773SPascal Brand 	return 0;
92b0104773SPascal Brand }
93b0104773SPascal Brand 
94cebdec51SJens Wiklander uint32_t tee_svc_sys_dummy_7args(uint32_t a1 __unused, uint32_t a2 __unused,
95cebdec51SJens Wiklander 				uint32_t a3 __unused, uint32_t a4 __unused,
96cebdec51SJens Wiklander 				uint32_t a5 __unused, uint32_t a6 __unused,
97cebdec51SJens Wiklander 				uint32_t a7 __unused)
98b0104773SPascal Brand {
99b0104773SPascal Brand 	DMSG("tee_svc_sys_dummy_7args: 0x%x, 0x%x, 0x%x, 0x%x, 0x%x, %x, %x\n",
100b0104773SPascal Brand 	     a1, a2, a3, a4, a5, a6, a7);
101b0104773SPascal Brand 	return 0;
102b0104773SPascal Brand }
103b0104773SPascal Brand 
104b0104773SPascal Brand uint32_t tee_svc_sys_nocall(void)
105b0104773SPascal Brand {
106b0104773SPascal Brand 	DMSG("No syscall");
107b0104773SPascal Brand 	return 0x1;
108b0104773SPascal Brand }
109b0104773SPascal Brand 
110b0104773SPascal Brand TEE_Result tee_svc_sys_get_property(uint32_t prop, tee_uaddr_t buf, size_t blen)
111b0104773SPascal Brand {
112b0104773SPascal Brand 	static const char api_vers[] = "1.0";
113b0104773SPascal Brand 	static const char descr[] = "Version N.N";
114b0104773SPascal Brand 	/*
115b0104773SPascal Brand 	 * Value 100 means:
116b0104773SPascal Brand 	 * System time based on REE-controlled timers. Can be tampered by the
117b0104773SPascal Brand 	 * REE.  The implementation must still guarantee that the system time
118b0104773SPascal Brand 	 * is monotonous, i.e., successive calls to TEE_GetSystemTime must
119b0104773SPascal Brand 	 * return increasing values of the system time.
120b0104773SPascal Brand 	 */
121b0104773SPascal Brand 	static const uint32_t sys_time_prot_lvl = 100;
122b0104773SPascal Brand 	static const uint32_t ta_time_prot_lvl = 100;
123b0104773SPascal Brand 	struct tee_ta_session *sess;
124b0104773SPascal Brand 	TEE_Result res;
125b0104773SPascal Brand 
126b0104773SPascal Brand 	res = tee_ta_get_current_session(&sess);
127b0104773SPascal Brand 	if (res != TEE_SUCCESS)
128b0104773SPascal Brand 		return res;
129b0104773SPascal Brand 
130b0104773SPascal Brand 	switch (prop) {
131b0104773SPascal Brand 	case UTEE_PROP_TEE_API_VERSION:
132b0104773SPascal Brand 		if (blen < sizeof(api_vers))
133b0104773SPascal Brand 			return TEE_ERROR_SHORT_BUFFER;
134b0104773SPascal Brand 		return tee_svc_copy_to_user(sess, (void *)buf, api_vers,
135b0104773SPascal Brand 					    sizeof(api_vers));
136b0104773SPascal Brand 
137b0104773SPascal Brand 	case UTEE_PROP_TEE_DESCR:
138b0104773SPascal Brand 		if (blen < sizeof(descr))
139b0104773SPascal Brand 			return TEE_ERROR_SHORT_BUFFER;
140b0104773SPascal Brand 		return tee_svc_copy_to_user(sess, (void *)buf, descr,
141b0104773SPascal Brand 					    sizeof(descr));
142b0104773SPascal Brand 
143b0104773SPascal Brand 	case UTEE_PROP_TEE_DEV_ID:
144b0104773SPascal Brand 		{
145b0104773SPascal Brand 			TEE_UUID uuid;
146b0104773SPascal Brand 			const size_t nslen = 4;
147b0104773SPascal Brand 			uint8_t data[4 +
148b0104773SPascal Brand 				     FVR_DIE_ID_NUM_REGS * sizeof(uint32_t)] = {
149b0104773SPascal Brand 			    'S', 'T', 'E', 'E' };
150b0104773SPascal Brand 
151b0104773SPascal Brand 			if (blen < sizeof(uuid))
152b0104773SPascal Brand 				return TEE_ERROR_SHORT_BUFFER;
153b0104773SPascal Brand 
154b0104773SPascal Brand 			if (tee_otp_get_die_id
155b0104773SPascal Brand 					(data + nslen, sizeof(data) - nslen))
156b0104773SPascal Brand 				return TEE_ERROR_BAD_STATE;
157b0104773SPascal Brand 
158ffe04039SJerome Forissier 			res = tee_hash_createdigest(TEE_ALG_SHA256, data,
159ffe04039SJerome Forissier 						    sizeof(data),
160ffe04039SJerome Forissier 						    (uint8_t *)&uuid,
161ffe04039SJerome Forissier 						    sizeof(uuid));
162b0104773SPascal Brand 			if (res != TEE_SUCCESS)
163b0104773SPascal Brand 				return TEE_ERROR_BAD_STATE;
164b0104773SPascal Brand 
165b0104773SPascal Brand 			/*
166b0104773SPascal Brand 			 * Changes the random value into and UUID as specifiec
167b0104773SPascal Brand 			 * in RFC 4122. The magic values are from the example
168b0104773SPascal Brand 			 * code in the RFC.
169b0104773SPascal Brand 			 *
170b0104773SPascal Brand 			 * TEE_UUID is defined slightly different from the RFC,
171b0104773SPascal Brand 			 * but close enough for our purpose.
172b0104773SPascal Brand 			 */
173b0104773SPascal Brand 
174b0104773SPascal Brand 			uuid.timeHiAndVersion &= 0x0fff;
175b0104773SPascal Brand 			uuid.timeHiAndVersion |= 5 << 12;
176b0104773SPascal Brand 
177b0104773SPascal Brand 			/* uuid.clock_seq_hi_and_reserved in the RFC */
178b0104773SPascal Brand 			uuid.clockSeqAndNode[0] &= 0x3f;
179b0104773SPascal Brand 			uuid.clockSeqAndNode[0] |= 0x80;
180b0104773SPascal Brand 
181b0104773SPascal Brand 			return tee_svc_copy_to_user(sess, (void *)buf, &uuid,
182b0104773SPascal Brand 						    sizeof(TEE_UUID));
183b0104773SPascal Brand 		}
184b0104773SPascal Brand 
185b0104773SPascal Brand 	case UTEE_PROP_TEE_SYS_TIME_PROT_LEVEL:
186b0104773SPascal Brand 		if (blen < sizeof(sys_time_prot_lvl))
187b0104773SPascal Brand 			return TEE_ERROR_SHORT_BUFFER;
188b0104773SPascal Brand 		return tee_svc_copy_to_user(sess, (void *)buf,
189b0104773SPascal Brand 					    &sys_time_prot_lvl,
190b0104773SPascal Brand 					    sizeof(sys_time_prot_lvl));
191b0104773SPascal Brand 
192b0104773SPascal Brand 	case UTEE_PROP_TEE_TA_TIME_PROT_LEVEL:
193b0104773SPascal Brand 		if (blen < sizeof(ta_time_prot_lvl))
194b0104773SPascal Brand 			return TEE_ERROR_SHORT_BUFFER;
195b0104773SPascal Brand 		return tee_svc_copy_to_user(sess, (void *)buf,
196b0104773SPascal Brand 					    &ta_time_prot_lvl,
197b0104773SPascal Brand 					    sizeof(ta_time_prot_lvl));
198b0104773SPascal Brand 
199b0104773SPascal Brand 	case UTEE_PROP_CLIENT_ID:
200b0104773SPascal Brand 		if (blen < sizeof(TEE_Identity))
201b0104773SPascal Brand 			return TEE_ERROR_SHORT_BUFFER;
202b0104773SPascal Brand 
203b0104773SPascal Brand 		return tee_svc_copy_to_user(sess, (void *)buf,
20437d6ae92SPascal Brand 			&sess->clnt_id, sizeof(TEE_Identity));
20537d6ae92SPascal Brand 
206b0104773SPascal Brand 	case UTEE_PROP_TA_APP_ID:
207b0104773SPascal Brand 		if (blen < sizeof(TEE_UUID))
208b0104773SPascal Brand 			return TEE_ERROR_SHORT_BUFFER;
209b0104773SPascal Brand 
210b0104773SPascal Brand 		return tee_svc_copy_to_user(sess, (void *)buf,
21137d6ae92SPascal Brand 			&sess->ctx->head->uuid, sizeof(TEE_UUID));
212b0104773SPascal Brand 
213b0104773SPascal Brand 	default:
214b0104773SPascal Brand 		break;
215b0104773SPascal Brand 	}
216b0104773SPascal Brand 	return TEE_ERROR_NOT_IMPLEMENTED;
217b0104773SPascal Brand }
218b0104773SPascal Brand 
219b0104773SPascal Brand /*
220b0104773SPascal Brand  * TA invokes some TA with parameter.
221b0104773SPascal Brand  * If some parameters are memory references:
222b0104773SPascal Brand  * - either the memref is inside TA private RAM: TA is not allowed to expose
223b0104773SPascal Brand  *   its private RAM: use a temporary memory buffer and copy the data.
224b0104773SPascal Brand  * - or the memref is not in the TA private RAM:
225b0104773SPascal Brand  *   - if the memref was mapped to the TA, TA is allowed to expose it.
226b0104773SPascal Brand  *   - if so, converts memref virtual address into a physical address.
227b0104773SPascal Brand  */
228b0104773SPascal Brand static TEE_Result tee_svc_copy_param(struct tee_ta_session *sess,
229b0104773SPascal Brand 				     struct tee_ta_session *called_sess,
230b0104773SPascal Brand 				     uint32_t param_types,
231b7fc217fSPascal Brand 				     TEE_Param callee_params[TEE_NUM_PARAMS],
232b0104773SPascal Brand 				     struct tee_ta_param *param,
233b0104773SPascal Brand 				     tee_paddr_t tmp_buf_pa[TEE_NUM_PARAMS],
234b0104773SPascal Brand 				     tee_mm_entry_t **mm)
235b0104773SPascal Brand {
236b0104773SPascal Brand 	size_t n;
237b0104773SPascal Brand 	TEE_Result res;
238b0104773SPascal Brand 	size_t req_mem = 0;
239b0104773SPascal Brand 	size_t s;
240b0104773SPascal Brand 	uint8_t *dst = 0;
241b0104773SPascal Brand 	tee_paddr_t dst_pa, src_pa = 0;
242b0104773SPascal Brand 	bool ta_private_memref[TEE_NUM_PARAMS];
243b0104773SPascal Brand 
244b7fc217fSPascal Brand 	/* fill 'param' input struct with caller params description buffer */
245b0104773SPascal Brand 	param->types = param_types;
246b7fc217fSPascal Brand 	if (!callee_params) {
247b0104773SPascal Brand 		if (param->types != 0)
248b0104773SPascal Brand 			return TEE_ERROR_BAD_PARAMETERS;
249b0104773SPascal Brand 		memset(param->params, 0, sizeof(param->params));
250b0104773SPascal Brand 	} else {
251b7fc217fSPascal Brand 		tee_svc_copy_from_user(sess, param->params, callee_params,
252b0104773SPascal Brand 				       sizeof(param->params));
253b0104773SPascal Brand 	}
254b0104773SPascal Brand 
255b0104773SPascal Brand 	if ((called_sess != NULL) &&
256b0104773SPascal Brand 		(called_sess->ctx->static_ta == NULL) &&
257b0104773SPascal Brand 		(called_sess->ctx->flags & TA_FLAG_USER_MODE) == 0) {
258b0104773SPascal Brand 		/*
259b0104773SPascal Brand 		 * kernel TA, borrow the mapping of the calling
260b0104773SPascal Brand 		 * during this call.
261b0104773SPascal Brand 		 */
262b0104773SPascal Brand 		called_sess->calling_sess = sess;
263b0104773SPascal Brand 		return TEE_SUCCESS;
264b0104773SPascal Brand 	}
265b0104773SPascal Brand 
266b0104773SPascal Brand 	for (n = 0; n < TEE_NUM_PARAMS; n++) {
267b0104773SPascal Brand 
268b0104773SPascal Brand 		ta_private_memref[n] = false;
269b0104773SPascal Brand 
270b0104773SPascal Brand 		switch (TEE_PARAM_TYPE_GET(param->types, n)) {
271b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_INPUT:
272b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_OUTPUT:
273b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_INOUT:
274b0104773SPascal Brand 			if (param->params[n].memref.buffer == NULL) {
275b0104773SPascal Brand 				if (param->params[n].memref.size != 0)
276b0104773SPascal Brand 					return TEE_ERROR_BAD_PARAMETERS;
277b0104773SPascal Brand 				break;
278b0104773SPascal Brand 			}
279b0104773SPascal Brand 			/* uTA cannot expose its private memory */
280b0104773SPascal Brand 			if (tee_mmu_is_vbuf_inside_ta_private(sess->ctx,
281106d8aa6SPascal Brand 				    param->params[n].memref.buffer,
282b0104773SPascal Brand 				    param->params[n].memref.size)) {
283b0104773SPascal Brand 
284a17acc4cSSabrina Ni 				s = ROUNDUP(param->params[n].memref.size,
285b0104773SPascal Brand 						sizeof(uint32_t));
286b0104773SPascal Brand 				/* Check overflow */
287b0104773SPascal Brand 				if (req_mem + s < req_mem)
288b0104773SPascal Brand 					return TEE_ERROR_BAD_PARAMETERS;
289b0104773SPascal Brand 				req_mem += s;
290b0104773SPascal Brand 				ta_private_memref[n] = true;
291b0104773SPascal Brand 				break;
292b0104773SPascal Brand 			}
293106d8aa6SPascal Brand 			if (tee_mmu_is_vbuf_intersect_ta_private(sess->ctx,
294106d8aa6SPascal Brand 				    param->params[n].memref.buffer,
295b0104773SPascal Brand 				    param->params[n].memref.size))
296b0104773SPascal Brand 				return TEE_ERROR_BAD_PARAMETERS;
297b0104773SPascal Brand 
298b0104773SPascal Brand 			if (tee_mmu_user_va2pa(sess->ctx,
299b0104773SPascal Brand 					(void *)param->params[n].memref.buffer,
3000e692b78SJens Wiklander 					&src_pa) != TEE_SUCCESS)
301b0104773SPascal Brand 				return TEE_ERROR_BAD_PARAMETERS;
302b0104773SPascal Brand 
303b0104773SPascal Brand 			param->param_attr[n] = tee_mmu_user_get_cache_attr(
304b0104773SPascal Brand 				sess->ctx,
305b0104773SPascal Brand 				(void *)param->params[n].memref.buffer);
306b0104773SPascal Brand 
307b0104773SPascal Brand 			param->params[n].memref.buffer = (void *)src_pa;
308b0104773SPascal Brand 			break;
309b0104773SPascal Brand 
310b0104773SPascal Brand 		default:
311b0104773SPascal Brand 			break;
312b0104773SPascal Brand 		}
313b0104773SPascal Brand 	}
314b0104773SPascal Brand 
315b0104773SPascal Brand 	if (req_mem == 0)
316b0104773SPascal Brand 		return TEE_SUCCESS;
317b0104773SPascal Brand 
318b0104773SPascal Brand 	/* Allocate section in secure DDR */
319b0104773SPascal Brand 	*mm = tee_mm_alloc(&tee_mm_sec_ddr, req_mem);
320b0104773SPascal Brand 	if (*mm == NULL) {
321b0104773SPascal Brand 		DMSG("tee_mm_alloc TEE_ERROR_GENERIC");
322b0104773SPascal Brand 		return TEE_ERROR_GENERIC;
323b0104773SPascal Brand 	}
324b0104773SPascal Brand 
325b0104773SPascal Brand 	/* Get the virtual address for the section in secure DDR */
326b0104773SPascal Brand 	res = tee_mmu_kmap(tee_mm_get_smem(*mm), req_mem, &dst);
327b0104773SPascal Brand 	if (res != TEE_SUCCESS)
328b0104773SPascal Brand 		return res;
329b0104773SPascal Brand 	dst_pa = tee_mm_get_smem(*mm);
330b0104773SPascal Brand 
331b0104773SPascal Brand 	for (n = 0; n < 4; n++) {
332b0104773SPascal Brand 
333b0104773SPascal Brand 		if (ta_private_memref[n] == false)
334b0104773SPascal Brand 			continue;
335b0104773SPascal Brand 
336a17acc4cSSabrina Ni 		s = ROUNDUP(param->params[n].memref.size, sizeof(uint32_t));
337b0104773SPascal Brand 
338b0104773SPascal Brand 		switch (TEE_PARAM_TYPE_GET(param->types, n)) {
339b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_INPUT:
340b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_INOUT:
341b0104773SPascal Brand 			if (param->params[n].memref.buffer != NULL) {
342b0104773SPascal Brand 				res = tee_svc_copy_from_user(sess, dst,
343b7fc217fSPascal Brand 						param->params[n].memref.buffer,
344b7fc217fSPascal Brand 						param->params[n].memref.size);
345b0104773SPascal Brand 				if (res != TEE_SUCCESS)
346b0104773SPascal Brand 					return res;
347b0104773SPascal Brand 				param->param_attr[n] =
348b0104773SPascal Brand 					tee_mmu_kmap_get_cache_attr(dst);
349b0104773SPascal Brand 				param->params[n].memref.buffer = (void *)dst_pa;
350b0104773SPascal Brand 				tmp_buf_pa[n] = dst_pa;
351b0104773SPascal Brand 				dst += s;
352b0104773SPascal Brand 				dst_pa += s;
353b0104773SPascal Brand 			}
354b0104773SPascal Brand 			break;
355b0104773SPascal Brand 
356b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_OUTPUT:
357b0104773SPascal Brand 			if (param->params[n].memref.buffer != NULL) {
358b0104773SPascal Brand 				param->param_attr[n] =
359b0104773SPascal Brand 					tee_mmu_kmap_get_cache_attr(dst);
360b0104773SPascal Brand 				param->params[n].memref.buffer = (void *)dst_pa;
361b0104773SPascal Brand 				tmp_buf_pa[n] = dst_pa;
362b0104773SPascal Brand 				dst += s;
363b0104773SPascal Brand 				dst_pa += s;
364b0104773SPascal Brand 			}
365b0104773SPascal Brand 			break;
366b0104773SPascal Brand 
367b0104773SPascal Brand 		default:
368b0104773SPascal Brand 			continue;
369b0104773SPascal Brand 		}
370b0104773SPascal Brand 	}
371b0104773SPascal Brand 
372b0104773SPascal Brand 	tee_mmu_kunmap(dst, req_mem);
373b0104773SPascal Brand 
374b0104773SPascal Brand 	return TEE_SUCCESS;
375b0104773SPascal Brand }
376b0104773SPascal Brand 
377b0104773SPascal Brand /*
378b0104773SPascal Brand  * Back from execution of service: update parameters passed from TA:
379b0104773SPascal Brand  * If some parameters were memory references:
380b0104773SPascal Brand  * - either the memref was temporary: copy back data and update size
381b0104773SPascal Brand  * - or it was the original TA memref: update only the size value.
382b0104773SPascal Brand  */
383b0104773SPascal Brand static TEE_Result tee_svc_update_out_param(
384b0104773SPascal Brand 		struct tee_ta_session *sess,
385b0104773SPascal Brand 		struct tee_ta_session *called_sess,
386b0104773SPascal Brand 		struct tee_ta_param *param,
387b0104773SPascal Brand 		tee_paddr_t tmp_buf_pa[TEE_NUM_PARAMS],
388b7fc217fSPascal Brand 		TEE_Param callee_params[TEE_NUM_PARAMS])
389b0104773SPascal Brand {
390b0104773SPascal Brand 	size_t n;
391b0104773SPascal Brand 	bool have_private_mem_map = (called_sess == NULL) ||
392b0104773SPascal Brand 		(called_sess->ctx->static_ta != NULL) ||
393b0104773SPascal Brand 		((called_sess->ctx->flags & TA_FLAG_USER_MODE) != 0);
394b0104773SPascal Brand 
395b0104773SPascal Brand 	tee_ta_set_current_session(sess);
396b0104773SPascal Brand 
397b0104773SPascal Brand 	for (n = 0; n < TEE_NUM_PARAMS; n++) {
398b0104773SPascal Brand 		switch (TEE_PARAM_TYPE_GET(param->types, n)) {
399b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_OUTPUT:
400b0104773SPascal Brand 		case TEE_PARAM_TYPE_MEMREF_INOUT:
401b0104773SPascal Brand 
402b0104773SPascal Brand 			/* outside TA private => memref is valid, update size */
403b0104773SPascal Brand 			if (!tee_mmu_is_vbuf_inside_ta_private(sess->ctx,
404106d8aa6SPascal Brand 					callee_params[n].memref.buffer,
405b0104773SPascal Brand 					param->params[n].memref.size)) {
406b7fc217fSPascal Brand 				callee_params[n].memref.size =
407b0104773SPascal Brand 					param->params[n].memref.size;
408b0104773SPascal Brand 				break;
409b0104773SPascal Brand 			}
410b0104773SPascal Brand 
411b0104773SPascal Brand 			/*
412b0104773SPascal Brand 			 * If we called a kernel TA the parameters are in shared
413b0104773SPascal Brand 			 * memory and no copy is needed.
414b0104773SPascal Brand 			 */
415b0104773SPascal Brand 			if (have_private_mem_map &&
416b0104773SPascal Brand 			    param->params[n].memref.size <=
417b7fc217fSPascal Brand 			    callee_params[n].memref.size) {
418b0104773SPascal Brand 				uint8_t *src = 0;
419b0104773SPascal Brand 				TEE_Result res;
420b0104773SPascal Brand 
421b0104773SPascal Brand 				/* FIXME: TA_RAM is already mapped ! */
422b0104773SPascal Brand 				res = tee_mmu_kmap(tmp_buf_pa[n],
423b0104773SPascal Brand 					param->params[n].memref.size, &src);
424b0104773SPascal Brand 				if (res != TEE_SUCCESS)
425b0104773SPascal Brand 					return TEE_ERROR_GENERIC;
426b0104773SPascal Brand 
427b0104773SPascal Brand 				res = tee_svc_copy_to_user(sess,
428b7fc217fSPascal Brand 							 callee_params[n].memref.
429b0104773SPascal Brand 							 buffer, src,
430b0104773SPascal Brand 							 param->params[n].
431b0104773SPascal Brand 							 memref.size);
432b0104773SPascal Brand 				if (res != TEE_SUCCESS)
433b0104773SPascal Brand 					return res;
434b0104773SPascal Brand 				tee_mmu_kunmap(src,
435b0104773SPascal Brand 					       param->params[n].memref.size);
436b0104773SPascal Brand 
437b0104773SPascal Brand 			}
438b7fc217fSPascal Brand 			callee_params[n].memref.size = param->params[n].memref.size;
439b0104773SPascal Brand 			break;
440b0104773SPascal Brand 
441b0104773SPascal Brand 		case TEE_PARAM_TYPE_VALUE_OUTPUT:
442b0104773SPascal Brand 		case TEE_PARAM_TYPE_VALUE_INOUT:
443b7fc217fSPascal Brand 			callee_params[n].value = param->params[n].value;
444b0104773SPascal Brand 			break;
445b0104773SPascal Brand 
446b0104773SPascal Brand 		default:
447b0104773SPascal Brand 			continue;
448b0104773SPascal Brand 		}
449b0104773SPascal Brand 	}
450b0104773SPascal Brand 
451b0104773SPascal Brand 	return TEE_SUCCESS;
452b0104773SPascal Brand }
453b0104773SPascal Brand 
454b0104773SPascal Brand /* Called when a TA calls an OpenSession on another TA */
455b0104773SPascal Brand TEE_Result tee_svc_open_ta_session(const TEE_UUID *dest,
456b0104773SPascal Brand 				   uint32_t cancel_req_to, uint32_t param_types,
457b0104773SPascal Brand 				   TEE_Param params[4],
458b0104773SPascal Brand 				   TEE_TASessionHandle *ta_sess,
459b0104773SPascal Brand 				   uint32_t *ret_orig)
460b0104773SPascal Brand {
461b0104773SPascal Brand 	TEE_Result res;
462b0104773SPascal Brand 	uint32_t ret_o = TEE_ORIGIN_TEE;
463b0104773SPascal Brand 	struct tee_ta_session *s = NULL;
464b0104773SPascal Brand 	struct tee_ta_session *sess;
465b0104773SPascal Brand 	tee_mm_entry_t *mm_param = NULL;
466b0104773SPascal Brand 
467b0104773SPascal Brand 	TEE_UUID *uuid = malloc(sizeof(TEE_UUID));
468b0104773SPascal Brand 	struct tee_ta_param *param = malloc(sizeof(struct tee_ta_param));
469b0104773SPascal Brand 	TEE_Identity *clnt_id = malloc(sizeof(TEE_Identity));
470b0104773SPascal Brand 	tee_paddr_t tmp_buf_pa[TEE_NUM_PARAMS];
471b0104773SPascal Brand 
472b0104773SPascal Brand 	if (uuid == NULL || param == NULL || clnt_id == NULL) {
473b0104773SPascal Brand 		res = TEE_ERROR_OUT_OF_MEMORY;
474b0104773SPascal Brand 		goto out_free_only;
475b0104773SPascal Brand 	}
476b0104773SPascal Brand 
477b0104773SPascal Brand 	memset(param, 0, sizeof(struct tee_ta_param));
478b0104773SPascal Brand 
479b0104773SPascal Brand 	res = tee_ta_get_current_session(&sess);
480b0104773SPascal Brand 	if (res != TEE_SUCCESS)
481b0104773SPascal Brand 		goto out_free_only;
482b0104773SPascal Brand 
483b0104773SPascal Brand 	res = tee_svc_copy_from_user(sess, uuid, dest, sizeof(TEE_UUID));
484b0104773SPascal Brand 	if (res != TEE_SUCCESS)
485b0104773SPascal Brand 		goto function_exit;
486b0104773SPascal Brand 
487b0104773SPascal Brand 	clnt_id->login = TEE_LOGIN_TRUSTED_APP;
488b0104773SPascal Brand 	memcpy(&clnt_id->uuid, &sess->ctx->head->uuid, sizeof(TEE_UUID));
489b0104773SPascal Brand 
490b0104773SPascal Brand 	res = tee_svc_copy_param(sess, NULL, param_types, params, param,
491b0104773SPascal Brand 				 tmp_buf_pa, &mm_param);
492b0104773SPascal Brand 	if (res != TEE_SUCCESS)
493b0104773SPascal Brand 		goto function_exit;
494b0104773SPascal Brand 
495b0104773SPascal Brand 	/*
496b0104773SPascal Brand 	 * Find session of a multi session TA or a static TA
497b0104773SPascal Brand 	 * In such a case, there is no need to ask the supplicant for the TA
498b0104773SPascal Brand 	 * code
499b0104773SPascal Brand 	 */
500b0104773SPascal Brand 	res = tee_ta_open_session(&ret_o, &s, &sess->ctx->open_sessions, uuid,
50127cbcc57SJens Wiklander 				  clnt_id, cancel_req_to, param);
502c0346845SJens Wiklander 	if (res != TEE_SUCCESS)
503b0104773SPascal Brand 		goto function_exit;
504b0104773SPascal Brand 
505b0104773SPascal Brand 	res = tee_svc_update_out_param(sess, NULL, param, tmp_buf_pa, params);
506b0104773SPascal Brand 
507b0104773SPascal Brand function_exit:
508b0104773SPascal Brand 	tee_ta_set_current_session(sess);
509b0104773SPascal Brand 
510b0104773SPascal Brand 	if (mm_param != NULL) {
511b0104773SPascal Brand 		TEE_Result res2;
512b0104773SPascal Brand 		void *va = 0;
513b0104773SPascal Brand 
514b0104773SPascal Brand 		res2 =
515b0104773SPascal Brand 		    tee_mmu_kmap_pa2va((void *)tee_mm_get_smem(mm_param), &va);
516b0104773SPascal Brand 		if (res2 == TEE_SUCCESS)
517b0104773SPascal Brand 			tee_mmu_kunmap(va, tee_mm_get_bytes(mm_param));
518b0104773SPascal Brand 	}
519b0104773SPascal Brand 	tee_mm_free(mm_param);
520b0104773SPascal Brand 	tee_svc_copy_to_user(sess, ta_sess, &s, sizeof(s));
521b0104773SPascal Brand 	tee_svc_copy_to_user(sess, ret_orig, &ret_o, sizeof(ret_o));
522b0104773SPascal Brand 
523b0104773SPascal Brand out_free_only:
524b0104773SPascal Brand 	free(param);
525b0104773SPascal Brand 	free(uuid);
526b0104773SPascal Brand 	free(clnt_id);
527b0104773SPascal Brand 	return res;
528b0104773SPascal Brand }
529b0104773SPascal Brand 
530b0104773SPascal Brand TEE_Result tee_svc_close_ta_session(TEE_TASessionHandle ta_sess)
531b0104773SPascal Brand {
532b0104773SPascal Brand 	TEE_Result res;
533b0104773SPascal Brand 	struct tee_ta_session *sess;
534*60699957SPascal Brand 	TEE_Identity clnt_id;
535b0104773SPascal Brand 
536b0104773SPascal Brand 	res = tee_ta_get_current_session(&sess);
537b0104773SPascal Brand 	if (res != TEE_SUCCESS)
538b0104773SPascal Brand 		return res;
539b0104773SPascal Brand 
540*60699957SPascal Brand 	clnt_id.login = TEE_LOGIN_TRUSTED_APP;
541*60699957SPascal Brand 	memcpy(&clnt_id.uuid, &sess->ctx->head->uuid, sizeof(TEE_UUID));
542b0104773SPascal Brand 
543*60699957SPascal Brand 	tee_ta_set_current_session(NULL);
544*60699957SPascal Brand 	res = tee_ta_close_session((uint32_t)ta_sess, &sess->ctx->open_sessions,
545*60699957SPascal Brand 				   &clnt_id);
546b0104773SPascal Brand 	tee_ta_set_current_session(sess);
547b0104773SPascal Brand 	return res;
548b0104773SPascal Brand }
549b0104773SPascal Brand 
550b0104773SPascal Brand TEE_Result tee_svc_invoke_ta_command(TEE_TASessionHandle ta_sess,
551b0104773SPascal Brand 				     uint32_t cancel_req_to, uint32_t cmd_id,
552b0104773SPascal Brand 				     uint32_t param_types, TEE_Param params[4],
553b0104773SPascal Brand 				     uint32_t *ret_orig)
554b0104773SPascal Brand {
555b0104773SPascal Brand 	TEE_Result res;
556b0104773SPascal Brand 	uint32_t ret_o = TEE_ORIGIN_TEE;
557b0104773SPascal Brand 	struct tee_ta_param param = { 0 };
558*60699957SPascal Brand 	TEE_Identity clnt_id;
559b0104773SPascal Brand 	struct tee_ta_session *sess;
560b0104773SPascal Brand 	struct tee_ta_session *called_sess = (struct tee_ta_session *)ta_sess;
561b0104773SPascal Brand 	tee_mm_entry_t *mm_param = NULL;
562b0104773SPascal Brand 	tee_paddr_t tmp_buf_pa[TEE_NUM_PARAMS];
563b0104773SPascal Brand 
564b0104773SPascal Brand 	res = tee_ta_get_current_session(&sess);
565b0104773SPascal Brand 	if (res != TEE_SUCCESS)
566b0104773SPascal Brand 		return res;
567b0104773SPascal Brand 
568b0104773SPascal Brand 	res =
569b0104773SPascal Brand 	    tee_ta_verify_session_pointer(called_sess,
570b0104773SPascal Brand 					  &sess->ctx->open_sessions);
571b0104773SPascal Brand 	if (res != TEE_SUCCESS)
572b0104773SPascal Brand 		return res;
573b0104773SPascal Brand 
574*60699957SPascal Brand 	clnt_id.login = TEE_LOGIN_TRUSTED_APP;
575*60699957SPascal Brand 	memcpy(&clnt_id.uuid, &sess->ctx->head->uuid, sizeof(TEE_UUID));
576*60699957SPascal Brand 
577b0104773SPascal Brand 	res = tee_svc_copy_param(sess, called_sess, param_types, params,
578b0104773SPascal Brand 				 &param, tmp_buf_pa, &mm_param);
579b0104773SPascal Brand 	if (res != TEE_SUCCESS)
580b0104773SPascal Brand 		goto function_exit;
581b0104773SPascal Brand 
582*60699957SPascal Brand 	res = tee_ta_invoke_command(&ret_o, called_sess, &clnt_id,
583*60699957SPascal Brand 				    cancel_req_to, cmd_id, &param);
584*60699957SPascal Brand 
585b0104773SPascal Brand 	if (res != TEE_SUCCESS)
586b0104773SPascal Brand 		goto function_exit;
587b0104773SPascal Brand 
588b0104773SPascal Brand 	res = tee_svc_update_out_param(sess, called_sess, &param, tmp_buf_pa,
589b0104773SPascal Brand 				       params);
590b0104773SPascal Brand 	if (res != TEE_SUCCESS)
591b0104773SPascal Brand 		goto function_exit;
592b0104773SPascal Brand 
593b0104773SPascal Brand function_exit:
594b0104773SPascal Brand 	tee_ta_set_current_session(sess);
595b0104773SPascal Brand 	called_sess->calling_sess = NULL; /* clear eventual borrowed mapping */
596b0104773SPascal Brand 
597b0104773SPascal Brand 	if (mm_param != NULL) {
598b0104773SPascal Brand 		TEE_Result res2;
599b0104773SPascal Brand 		void *va = 0;
600b0104773SPascal Brand 
601b0104773SPascal Brand 		res2 =
602b0104773SPascal Brand 		    tee_mmu_kmap_pa2va((void *)tee_mm_get_smem(mm_param), &va);
603b0104773SPascal Brand 		if (res2 == TEE_SUCCESS)
604b0104773SPascal Brand 			tee_mmu_kunmap(va, tee_mm_get_bytes(mm_param));
605b0104773SPascal Brand 	}
606b0104773SPascal Brand 	tee_mm_free(mm_param);
607b0104773SPascal Brand 	if (ret_orig)
608b0104773SPascal Brand 		tee_svc_copy_to_user(sess, ret_orig, &ret_o, sizeof(ret_o));
609b0104773SPascal Brand 	return res;
610b0104773SPascal Brand }
611b0104773SPascal Brand 
612b0104773SPascal Brand TEE_Result tee_svc_check_access_rights(uint32_t flags, const void *buf,
613b0104773SPascal Brand 				       size_t len)
614b0104773SPascal Brand {
615b0104773SPascal Brand 	TEE_Result res;
616b0104773SPascal Brand 	struct tee_ta_session *s;
617b0104773SPascal Brand 
618b0104773SPascal Brand 	res = tee_ta_get_current_session(&s);
619b0104773SPascal Brand 	if (res != TEE_SUCCESS)
620b0104773SPascal Brand 		return res;
621b0104773SPascal Brand 
622b0104773SPascal Brand 	return tee_mmu_check_access_rights(s->ctx, flags, (tee_uaddr_t)buf,
623b0104773SPascal Brand 					   len);
624b0104773SPascal Brand }
625b0104773SPascal Brand 
626b0104773SPascal Brand TEE_Result tee_svc_copy_from_user(struct tee_ta_session *sess, void *kaddr,
627b0104773SPascal Brand 				  const void *uaddr, size_t len)
628b0104773SPascal Brand {
629b0104773SPascal Brand 	TEE_Result res;
630b0104773SPascal Brand 	struct tee_ta_session *s;
631b0104773SPascal Brand 
632b0104773SPascal Brand 	if (sess == NULL) {
633b0104773SPascal Brand 		res = tee_ta_get_current_session(&s);
634b0104773SPascal Brand 		if (res != TEE_SUCCESS)
635b0104773SPascal Brand 			return res;
636b0104773SPascal Brand 	} else {
637b0104773SPascal Brand 		s = sess;
638b0104773SPascal Brand 		tee_ta_set_current_session(s);
639b0104773SPascal Brand 	}
640b0104773SPascal Brand 	res =
641b0104773SPascal Brand 	    tee_mmu_check_access_rights(s->ctx,
642b0104773SPascal Brand 					TEE_MEMORY_ACCESS_READ |
643b0104773SPascal Brand 					TEE_MEMORY_ACCESS_ANY_OWNER,
644b0104773SPascal Brand 					(tee_uaddr_t)uaddr, len);
645b0104773SPascal Brand 	if (res != TEE_SUCCESS)
646b0104773SPascal Brand 		return res;
647b0104773SPascal Brand 
648b0104773SPascal Brand 	memcpy(kaddr, uaddr, len);
649b0104773SPascal Brand 	return TEE_SUCCESS;
650b0104773SPascal Brand }
651b0104773SPascal Brand 
652b0104773SPascal Brand TEE_Result tee_svc_copy_to_user(struct tee_ta_session *sess, void *uaddr,
653b0104773SPascal Brand 				const void *kaddr, size_t len)
654b0104773SPascal Brand {
655b0104773SPascal Brand 	TEE_Result res;
656b0104773SPascal Brand 	struct tee_ta_session *s;
657b0104773SPascal Brand 
658b0104773SPascal Brand 	if (sess == NULL) {
659b0104773SPascal Brand 		res = tee_ta_get_current_session(&s);
660b0104773SPascal Brand 		if (res != TEE_SUCCESS)
661b0104773SPascal Brand 			return res;
662b0104773SPascal Brand 	} else {
663b0104773SPascal Brand 		s = sess;
664b0104773SPascal Brand 		tee_ta_set_current_session(s);
665b0104773SPascal Brand 	}
666b0104773SPascal Brand 
667b0104773SPascal Brand 	res =
668b0104773SPascal Brand 	    tee_mmu_check_access_rights(s->ctx,
669b0104773SPascal Brand 					TEE_MEMORY_ACCESS_WRITE |
670b0104773SPascal Brand 					TEE_MEMORY_ACCESS_ANY_OWNER,
671b0104773SPascal Brand 					(tee_uaddr_t)uaddr, len);
672b0104773SPascal Brand 	if (res != TEE_SUCCESS)
673b0104773SPascal Brand 		return res;
674b0104773SPascal Brand 
675b0104773SPascal Brand 	memcpy(uaddr, kaddr, len);
676b0104773SPascal Brand 	return TEE_SUCCESS;
677b0104773SPascal Brand }
678b0104773SPascal Brand 
679b0104773SPascal Brand static bool session_is_cancelled(struct tee_ta_session *s, TEE_Time *curr_time)
680b0104773SPascal Brand {
681b0104773SPascal Brand 	TEE_Time current_time;
682b0104773SPascal Brand 
683b0104773SPascal Brand 	if (s->cancel_mask)
684b0104773SPascal Brand 		return false;
685b0104773SPascal Brand 
686b0104773SPascal Brand 	if (s->cancel)
687b0104773SPascal Brand 		return true;
688b0104773SPascal Brand 
689b0104773SPascal Brand 	if (s->cancel_time.seconds == UINT32_MAX)
690b0104773SPascal Brand 		return false;
691b0104773SPascal Brand 
692b0104773SPascal Brand 	if (curr_time != NULL)
693b0104773SPascal Brand 		current_time = *curr_time;
694b0104773SPascal Brand 	else if (tee_time_get_sys_time(&current_time) != TEE_SUCCESS)
695b0104773SPascal Brand 		return false;
696b0104773SPascal Brand 
697b0104773SPascal Brand 	if (current_time.seconds > s->cancel_time.seconds ||
698b0104773SPascal Brand 	    (current_time.seconds == s->cancel_time.seconds &&
699b0104773SPascal Brand 	     current_time.millis >= s->cancel_time.millis)) {
700b0104773SPascal Brand 		return true;
701b0104773SPascal Brand 	}
702b0104773SPascal Brand 
703b0104773SPascal Brand 	return false;
704b0104773SPascal Brand }
705b0104773SPascal Brand 
706b0104773SPascal Brand TEE_Result tee_svc_get_cancellation_flag(bool *cancel)
707b0104773SPascal Brand {
708b0104773SPascal Brand 	TEE_Result res;
709b0104773SPascal Brand 	struct tee_ta_session *s = NULL;
710b0104773SPascal Brand 	bool c;
711b0104773SPascal Brand 
712b0104773SPascal Brand 	res = tee_ta_get_current_session(&s);
713b0104773SPascal Brand 	if (res != TEE_SUCCESS)
714b0104773SPascal Brand 		return res;
715b0104773SPascal Brand 
716b0104773SPascal Brand 	c = session_is_cancelled(s, NULL);
717b0104773SPascal Brand 
718b0104773SPascal Brand 	return tee_svc_copy_to_user(s, cancel, &c, sizeof(c));
719b0104773SPascal Brand }
720b0104773SPascal Brand 
721b0104773SPascal Brand TEE_Result tee_svc_unmask_cancellation(bool *old_mask)
722b0104773SPascal Brand {
723b0104773SPascal Brand 	TEE_Result res;
724b0104773SPascal Brand 	struct tee_ta_session *s = NULL;
725b0104773SPascal Brand 	bool m;
726b0104773SPascal Brand 
727b0104773SPascal Brand 	res = tee_ta_get_current_session(&s);
728b0104773SPascal Brand 	if (res != TEE_SUCCESS)
729b0104773SPascal Brand 		return res;
730b0104773SPascal Brand 
731b0104773SPascal Brand 	m = s->cancel_mask;
732b0104773SPascal Brand 	s->cancel_mask = false;
733b0104773SPascal Brand 	return tee_svc_copy_to_user(s, old_mask, &m, sizeof(m));
734b0104773SPascal Brand }
735b0104773SPascal Brand 
736b0104773SPascal Brand TEE_Result tee_svc_mask_cancellation(bool *old_mask)
737b0104773SPascal Brand {
738b0104773SPascal Brand 	TEE_Result res;
739b0104773SPascal Brand 	struct tee_ta_session *s = NULL;
740b0104773SPascal Brand 	bool m;
741b0104773SPascal Brand 
742b0104773SPascal Brand 	res = tee_ta_get_current_session(&s);
743b0104773SPascal Brand 	if (res != TEE_SUCCESS)
744b0104773SPascal Brand 		return res;
745b0104773SPascal Brand 
746b0104773SPascal Brand 	m = s->cancel_mask;
747b0104773SPascal Brand 	s->cancel_mask = true;
748b0104773SPascal Brand 	return tee_svc_copy_to_user(s, old_mask, &m, sizeof(m));
749b0104773SPascal Brand }
750b0104773SPascal Brand 
751b0104773SPascal Brand TEE_Result tee_svc_wait(uint32_t timeout)
752b0104773SPascal Brand {
753b0104773SPascal Brand 	TEE_Result res = TEE_SUCCESS;
754b0104773SPascal Brand 	uint32_t mytime = 0;
755b0104773SPascal Brand 	struct tee_ta_session *s;
756b0104773SPascal Brand 	TEE_Time base_time;
757b0104773SPascal Brand 	TEE_Time current_time;
758b0104773SPascal Brand 
759b0104773SPascal Brand 	res = tee_ta_get_current_session(&s);
760b0104773SPascal Brand 	if (res != TEE_SUCCESS)
761b0104773SPascal Brand 		return res;
762b0104773SPascal Brand 
763b0104773SPascal Brand 	res = tee_time_get_sys_time(&base_time);
764b0104773SPascal Brand 	if (res != TEE_SUCCESS)
765b0104773SPascal Brand 		return res;
766b0104773SPascal Brand 
767b0104773SPascal Brand 	while (true) {
768b0104773SPascal Brand 		res = tee_time_get_sys_time(&current_time);
769b0104773SPascal Brand 		if (res != TEE_SUCCESS)
770b0104773SPascal Brand 			return res;
771b0104773SPascal Brand 
772b0104773SPascal Brand 		if (session_is_cancelled(s, &current_time))
773b0104773SPascal Brand 			return TEE_ERROR_CANCEL;
774b0104773SPascal Brand 
775b0104773SPascal Brand 		mytime = (current_time.seconds - base_time.seconds) * 1000 +
776b0104773SPascal Brand 		    (int)current_time.millis - (int)base_time.millis;
777b0104773SPascal Brand 		if (mytime >= timeout)
778b0104773SPascal Brand 			return TEE_SUCCESS;
779b0104773SPascal Brand 
780d1aea08fSSY Chiu 		tee_time_wait(timeout - mytime);
781b0104773SPascal Brand 	}
782b0104773SPascal Brand 
783b0104773SPascal Brand 	return res;
784b0104773SPascal Brand }
785b0104773SPascal Brand 
786b0104773SPascal Brand TEE_Result tee_svc_get_time(enum utee_time_category cat, TEE_Time *mytime)
787b0104773SPascal Brand {
788b0104773SPascal Brand 	TEE_Result res, res2;
789b0104773SPascal Brand 	struct tee_ta_session *s = NULL;
790b0104773SPascal Brand 	TEE_Time t;
791b0104773SPascal Brand 
792b0104773SPascal Brand 	res = tee_ta_get_current_session(&s);
793b0104773SPascal Brand 	if (res != TEE_SUCCESS)
794b0104773SPascal Brand 		return res;
795b0104773SPascal Brand 
796b0104773SPascal Brand 	switch (cat) {
797b0104773SPascal Brand 	case UTEE_TIME_CAT_SYSTEM:
798b0104773SPascal Brand 		res = tee_time_get_sys_time(&t);
799b0104773SPascal Brand 		break;
800b0104773SPascal Brand 	case UTEE_TIME_CAT_TA_PERSISTENT:
801b0104773SPascal Brand 		res =
802b0104773SPascal Brand 		    tee_time_get_ta_time((const void *)&s->ctx->head->uuid, &t);
803b0104773SPascal Brand 		break;
804b0104773SPascal Brand 	case UTEE_TIME_CAT_REE:
805b0104773SPascal Brand 		res = tee_time_get_ree_time(&t);
806b0104773SPascal Brand 		break;
807b0104773SPascal Brand 	default:
808b0104773SPascal Brand 		res = TEE_ERROR_BAD_PARAMETERS;
809b0104773SPascal Brand 		break;
810b0104773SPascal Brand 	}
811b0104773SPascal Brand 
812b0104773SPascal Brand 	if (res == TEE_SUCCESS || res == TEE_ERROR_OVERFLOW) {
813b0104773SPascal Brand 		res2 = tee_svc_copy_to_user(s, mytime, &t, sizeof(t));
814b0104773SPascal Brand 		if (res2 != TEE_SUCCESS)
815b0104773SPascal Brand 			res = res2;
816b0104773SPascal Brand 	}
817b0104773SPascal Brand 
818b0104773SPascal Brand 	return res;
819b0104773SPascal Brand }
820b0104773SPascal Brand 
821b0104773SPascal Brand TEE_Result tee_svc_set_ta_time(const TEE_Time *mytime)
822b0104773SPascal Brand {
823b0104773SPascal Brand 	TEE_Result res;
824b0104773SPascal Brand 	struct tee_ta_session *s = NULL;
825b0104773SPascal Brand 	TEE_Time t;
826b0104773SPascal Brand 
827b0104773SPascal Brand 	res = tee_ta_get_current_session(&s);
828b0104773SPascal Brand 	if (res != TEE_SUCCESS)
829b0104773SPascal Brand 		return res;
830b0104773SPascal Brand 
831b0104773SPascal Brand 	res = tee_svc_copy_from_user(s, &t, mytime, sizeof(t));
832b0104773SPascal Brand 	if (res != TEE_SUCCESS)
833b0104773SPascal Brand 		return res;
834b0104773SPascal Brand 
835b0104773SPascal Brand 	return tee_time_set_ta_time((const void *)&s->ctx->head->uuid, &t);
836b0104773SPascal Brand }
837fa530828SPascal Brand 
838fa530828SPascal Brand #ifdef CFG_CACHE_API
839fa530828SPascal Brand TEE_Result tee_svc_cache_operation(void *va, size_t len,
840fa530828SPascal Brand 				   enum utee_cache_operation op)
841fa530828SPascal Brand {
842fa530828SPascal Brand 	TEE_Result res;
843fa530828SPascal Brand 	struct tee_ta_session *s = NULL;
844fa530828SPascal Brand 
845fa530828SPascal Brand 	res = tee_ta_get_current_session(&s);
846fa530828SPascal Brand 	if (res != TEE_SUCCESS)
847fa530828SPascal Brand 		return res;
848fa530828SPascal Brand 
849fa530828SPascal Brand 	if ((s->ctx->flags & TA_FLAG_CACHE_MAINTENANCE) == 0)
850fa530828SPascal Brand 		return TEE_ERROR_NOT_SUPPORTED;
851fa530828SPascal Brand 
852fa530828SPascal Brand 	return tee_uta_cache_operation(s, op, va, len);
853fa530828SPascal Brand }
854fa530828SPascal Brand #endif
855