tspd_main.c 22.2 KB
Newer Older
1
/*
2
 * Copyright (c) 2013-2017, ARM Limited and Contributors. All rights reserved.
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *
 * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * Neither the name of ARM nor the names of its contributors may be used
 * to endorse or promote products derived from this software without specific
 * prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */


/*******************************************************************************
 * This is the Secure Payload Dispatcher (SPD). The dispatcher is meant to be a
 * plug-in component to the Secure Monitor, registered as a runtime service. The
 * SPD is expected to be a functional extension of the Secure Payload (SP) that
 * executes in Secure EL1. The Secure Monitor will delegate all SMCs targeting
 * the Trusted OS/Applications range to the dispatcher. The SPD will either
 * handle the request locally or delegate it to the Secure Payload. It is also
 * responsible for initialising and maintaining communication with the SP.
 ******************************************************************************/
#include <arch_helpers.h>
42
43
44
#include <assert.h>
#include <bl_common.h>
#include <bl31.h>
45
#include <context_mgmt.h>
46
47
48
#include <debug.h>
#include <errno.h>
#include <platform.h>
49
#include <runtime_svc.h>
50
#include <stddef.h>
51
#include <string.h>
52
#include <tsp.h>
53
#include <uuid.h>
54
#include "tspd_private.h"
55
56

/*******************************************************************************
57
58
 * Address of the entrypoint vector table in the Secure Payload. It is
 * initialised once on the primary core after a cold boot.
59
 ******************************************************************************/
60
tsp_vectors_t *tsp_vectors;
61
62
63
64

/*******************************************************************************
 * Array to keep track of per-cpu Secure Payload state
 ******************************************************************************/
65
tsp_context_t tspd_sp_context[TSPD_CORE_COUNT];
66

67

68
69
70
71
72
/* TSP UID */
DEFINE_SVC_UUID(tsp_uuid,
		0x5b3056a0, 0x3291, 0x427b, 0x98, 0x11,
		0x71, 0x68, 0xca, 0x50, 0xf3, 0xfa);

73
int32_t tspd_init(void);
74

75
76
77
78
79
80
/*
 * This helper function handles Secure EL1 preemption. The preemption could be
 * due Non Secure interrupts or EL3 interrupts. In both the cases we context
 * switch to the normal world and in case of EL3 interrupts, it will again be
 * routed to EL3 which will get handled at the exception vectors.
 */
81
82
83
uint64_t tspd_handle_sp_preemption(void *handle)
{
	cpu_context_t *ns_cpu_context;
84

85
86
87
88
89
90
91
	assert(handle == cm_get_context(SECURE));
	cm_el1_sysregs_context_save(SECURE);
	/* Get a reference to the non-secure context */
	ns_cpu_context = cm_get_context(NON_SECURE);
	assert(ns_cpu_context);

	/*
92
93
94
95
96
97
98
99
	 * To allow Secure EL1 interrupt handler to re-enter TSP while TSP
	 * is preempted, the secure system register context which will get
	 * overwritten must be additionally saved. This is currently done
	 * by the TSPD S-EL1 interrupt handler.
	 */

	/*
	 * Restore non-secure state.
100
101
102
103
	 */
	cm_el1_sysregs_context_restore(NON_SECURE);
	cm_set_next_eret_context(NON_SECURE);

104
	/*
105
	 * The TSP was preempted during execution of a Yielding SMC Call.
106
107
	 * Return back to the normal world with SMC_PREEMPTED as error
	 * code in x0.
108
	 */
109
110
	SMC_RET1(ns_cpu_context, SMC_PREEMPTED);
}
111

112
113
114
/*******************************************************************************
 * This function is the handler registered for S-EL1 interrupts by the TSPD. It
 * validates the interrupt and upon success arranges entry into the TSP at
115
 * 'tsp_sel1_intr_entry()' for handling the interrupt.
116
117
118
119
120
121
122
123
124
125
126
127
128
 ******************************************************************************/
static uint64_t tspd_sel1_interrupt_handler(uint32_t id,
					    uint32_t flags,
					    void *handle,
					    void *cookie)
{
	uint32_t linear_id;
	tsp_context_t *tsp_ctx;

	/* Check the security state when the exception was generated */
	assert(get_interrupt_src_ss(flags) == NON_SECURE);

	/* Sanity check the pointer to this cpu's context */
129
	assert(handle == cm_get_context(NON_SECURE));
130
131
132
133
134

	/* Save the non-secure context before entering the TSP */
	cm_el1_sysregs_context_save(NON_SECURE);

	/* Get a reference to this cpu's TSP context */
135
	linear_id = plat_my_core_pos();
136
	tsp_ctx = &tspd_sp_context[linear_id];
137
	assert(&tsp_ctx->cpu_ctx == cm_get_context(SECURE));
138
139
140
141
142

	/*
	 * Determine if the TSP was previously preempted. Its last known
	 * context has to be preserved in this case.
	 * The TSP should return control to the TSPD after handling this
143
144
145
146
147
	 * S-EL1 interrupt. Preserve essential EL3 context to allow entry into
	 * the TSP at the S-EL1 interrupt entry point using the 'cpu_context'
	 * structure. There is no need to save the secure system register
	 * context since the TSP is supposed to preserve it during S-EL1
	 * interrupt handling.
148
	 */
149
	if (get_yield_smc_active_flag(tsp_ctx->state)) {
150
151
152
153
		tsp_ctx->saved_spsr_el3 = SMC_GET_EL3(&tsp_ctx->cpu_ctx,
						      CTX_SPSR_EL3);
		tsp_ctx->saved_elr_el3 = SMC_GET_EL3(&tsp_ctx->cpu_ctx,
						     CTX_ELR_EL3);
154
#if TSP_NS_INTR_ASYNC_PREEMPT
155
156
157
		/*Need to save the previously interrupted secure context */
		memcpy(&tsp_ctx->sp_ctx, &tsp_ctx->cpu_ctx, TSPD_SP_CTX_SIZE);
#endif
158
159
160
	}

	cm_el1_sysregs_context_restore(SECURE);
161
	cm_set_elr_spsr_el3(SECURE, (uint64_t) &tsp_vectors->sel1_intr_entry,
162
		    SPSR_64(MODE_EL1, MODE_SP_ELX, DISABLE_ALL_EXCEPTIONS));
163

164
165
166
	cm_set_next_eret_context(SECURE);

	/*
167
168
169
170
171
	 * Tell the TSP that it has to handle a S-EL1 interrupt synchronously.
	 * Also the instruction in normal world where the interrupt was
	 * generated is passed for debugging purposes. It is safe to retrieve
	 * this address from ELR_EL3 as the secure context will not take effect
	 * until el3_exit().
172
	 */
173
	SMC_RET2(&tsp_ctx->cpu_ctx, TSP_HANDLE_SEL1_INTR_AND_RETURN, read_elr_el3());
174
}
175

176
#if TSP_NS_INTR_ASYNC_PREEMPT
177
/*******************************************************************************
178
179
180
 * This function is the handler registered for Non secure interrupts by the
 * TSPD. It validates the interrupt and upon success arranges entry into the
 * normal world for handling the interrupt.
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
 ******************************************************************************/
static uint64_t tspd_ns_interrupt_handler(uint32_t id,
					    uint32_t flags,
					    void *handle,
					    void *cookie)
{
	/* Check the security state when the exception was generated */
	assert(get_interrupt_src_ss(flags) == SECURE);

	/*
	 * Disable the routing of NS interrupts from secure world to EL3 while
	 * interrupted on this core.
	 */
	disable_intr_rm_local(INTR_TYPE_NS, SECURE);

	return tspd_handle_sp_preemption(handle);
}
#endif

200
201
202
203
204
205
206
/*******************************************************************************
 * Secure Payload Dispatcher setup. The SPD finds out the SP entrypoint and type
 * (aarch32/aarch64) if not already known and initialises the context for entry
 * into the SP for its initialisation.
 ******************************************************************************/
int32_t tspd_setup(void)
{
Vikram Kanigiri's avatar
Vikram Kanigiri committed
207
	entry_point_info_t *tsp_ep_info;
208
209
	uint32_t linear_id;

210
	linear_id = plat_my_core_pos();
211
212
213
214
215
216

	/*
	 * Get information about the Secure Payload (BL32) image. Its
	 * absence is a critical failure.  TODO: Add support to
	 * conditionally include the SPD service
	 */
Vikram Kanigiri's avatar
Vikram Kanigiri committed
217
218
219
220
221
222
223
	tsp_ep_info = bl31_plat_get_next_image_ep_info(SECURE);
	if (!tsp_ep_info) {
		WARN("No TSP provided by BL2 boot loader, Booting device"
			" without TSP initialization. SMC`s destined for TSP"
			" will return SMC_UNK\n");
		return 1;
	}
224

225
226
227
228
229
	/*
	 * If there's no valid entry point for SP, we return a non-zero value
	 * signalling failure initializing the service. We bail out without
	 * registering any handlers
	 */
Vikram Kanigiri's avatar
Vikram Kanigiri committed
230
	if (!tsp_ep_info->pc)
231
232
		return 1;

233
	/*
234
	 * We could inspect the SP image and determine its execution
235
236
237
	 * state i.e whether AArch32 or AArch64. Assuming it's AArch64
	 * for the time being.
	 */
Vikram Kanigiri's avatar
Vikram Kanigiri committed
238
239
240
241
	tspd_init_tsp_ep_state(tsp_ep_info,
				TSP_AARCH64,
				tsp_ep_info->pc,
				&tspd_sp_context[linear_id]);
242

243
244
245
#if TSP_INIT_ASYNC
	bl31_set_next_image_type(SECURE);
#else
246
247
248
249
250
	/*
	 * All TSPD initialization done. Now register our init function with
	 * BL31 for deferred invocation
	 */
	bl31_register_bl32_init(&tspd_init);
251
#endif
Vikram Kanigiri's avatar
Vikram Kanigiri committed
252
	return 0;
253
254
255
256
257
258
259
260
261
}

/*******************************************************************************
 * This function passes control to the Secure Payload image (BL32) for the first
 * time on the primary cpu after a cold boot. It assumes that a valid secure
 * context has already been created by tspd_setup() which can be directly used.
 * It also assumes that a valid non-secure context has been initialised by PSCI
 * so it does not need to save and restore any non-secure state. This function
 * performs a synchronous entry into the Secure payload. The SP passes control
262
 * back to this routine through a SMC.
263
 ******************************************************************************/
264
int32_t tspd_init(void)
265
{
266
	uint32_t linear_id = plat_my_core_pos();
267
	tsp_context_t *tsp_ctx = &tspd_sp_context[linear_id];
Vikram Kanigiri's avatar
Vikram Kanigiri committed
268
	entry_point_info_t *tsp_entry_point;
269
	uint64_t rc;
Vikram Kanigiri's avatar
Vikram Kanigiri committed
270
271
272
273
274
275
276
277

	/*
	 * Get information about the Secure Payload (BL32) image. Its
	 * absence is a critical failure.
	 */
	tsp_entry_point = bl31_plat_get_next_image_ep_info(SECURE);
	assert(tsp_entry_point);

278
	cm_init_my_context(tsp_entry_point);
279

280
	/*
281
282
	 * Arrange for an entry into the test secure payload. It will be
	 * returned via TSP_ENTRY_DONE case
283
	 */
284
285
	rc = tspd_synchronous_sp_entry(tsp_ctx);
	assert(rc != 0);
286

287
288
289
	return rc;
}

290

291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
/*******************************************************************************
 * This function is responsible for handling all SMCs in the Trusted OS/App
 * range from the non-secure state as defined in the SMC Calling Convention
 * Document. It is also responsible for communicating with the Secure payload
 * to delegate work and return results back to the non-secure state. Lastly it
 * will also return any information that the secure payload needs to do the
 * work assigned to it.
 ******************************************************************************/
uint64_t tspd_smc_handler(uint32_t smc_fid,
			 uint64_t x1,
			 uint64_t x2,
			 uint64_t x3,
			 uint64_t x4,
			 void *cookie,
			 void *handle,
			 uint64_t flags)
{
308
	cpu_context_t *ns_cpu_context;
309
	uint32_t linear_id = plat_my_core_pos(), ns;
310
	tsp_context_t *tsp_ctx = &tspd_sp_context[linear_id];
311
312
313
314
	uint64_t rc;
#if TSP_INIT_ASYNC
	entry_point_info_t *next_image_info;
#endif
315
316
317
318
319
320

	/* Determine which security state this SMC originated from */
	ns = is_caller_non_secure(flags);

	switch (smc_fid) {

321
322
323
324
325
326
327
328
329
	/*
	 * This function ID is used by TSP to indicate that it was
	 * preempted by a normal world IRQ.
	 *
	 */
	case TSP_PREEMPTED:
		if (ns)
			SMC_RET1(handle, SMC_UNK);

330
		return tspd_handle_sp_preemption(handle);
331

332
333
	/*
	 * This function ID is used only by the TSP to indicate that it has
334
335
	 * finished handling a S-EL1 interrupt or was preempted by a higher
	 * priority pending EL3 interrupt. Execution should resume
336
337
	 * in the normal world.
	 */
338
	case TSP_HANDLED_S_EL1_INTR:
339
340
341
		if (ns)
			SMC_RET1(handle, SMC_UNK);

342
		assert(handle == cm_get_context(SECURE));
343
344
345
346
347

		/*
		 * Restore the relevant EL3 state which saved to service
		 * this SMC.
		 */
348
		if (get_yield_smc_active_flag(tsp_ctx->state)) {
349
350
351
352
353
354
			SMC_SET_EL3(&tsp_ctx->cpu_ctx,
				    CTX_SPSR_EL3,
				    tsp_ctx->saved_spsr_el3);
			SMC_SET_EL3(&tsp_ctx->cpu_ctx,
				    CTX_ELR_EL3,
				    tsp_ctx->saved_elr_el3);
355
#if TSP_NS_INTR_ASYNC_PREEMPT
356
357
358
359
360
361
362
			/*
			 * Need to restore the previously interrupted
			 * secure context.
			 */
			memcpy(&tsp_ctx->cpu_ctx, &tsp_ctx->sp_ctx,
				TSPD_SP_CTX_SIZE);
#endif
363
364
365
		}

		/* Get a reference to the non-secure context */
366
		ns_cpu_context = cm_get_context(NON_SECURE);
367
368
369
370
371
372
373
374
375
376
377
378
		assert(ns_cpu_context);

		/*
		 * Restore non-secure state. There is no need to save the
		 * secure system register context since the TSP was supposed
		 * to preserve it during S-EL1 interrupt handling.
		 */
		cm_el1_sysregs_context_restore(NON_SECURE);
		cm_set_next_eret_context(NON_SECURE);

		SMC_RET0((uint64_t) ns_cpu_context);

379
380
381
382
383
384
385
386
387
388
389
390
	/*
	 * This function ID is used only by the SP to indicate it has
	 * finished initialising itself after a cold boot
	 */
	case TSP_ENTRY_DONE:
		if (ns)
			SMC_RET1(handle, SMC_UNK);

		/*
		 * Stash the SP entry points information. This is done
		 * only once on the primary cpu
		 */
391
392
		assert(tsp_vectors == NULL);
		tsp_vectors = (tsp_vectors_t *) x1;
393

394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
		if (tsp_vectors) {
			set_tsp_pstate(tsp_ctx->state, TSP_PSTATE_ON);

			/*
			 * TSP has been successfully initialized. Register power
			 * managemnt hooks with PSCI
			 */
			psci_register_spd_pm_hook(&tspd_pm);

			/*
			 * Register an interrupt handler for S-EL1 interrupts
			 * when generated during code executing in the
			 * non-secure state.
			 */
			flags = 0;
			set_interrupt_rm_flag(flags, NON_SECURE);
			rc = register_interrupt_type_handler(INTR_TYPE_S_EL1,
						tspd_sel1_interrupt_handler,
						flags);
			if (rc)
				panic();
415

416
#if TSP_NS_INTR_ASYNC_PREEMPT
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
			/*
			 * Register an interrupt handler for NS interrupts when
			 * generated during code executing in secure state are
			 * routed to EL3.
			 */
			flags = 0;
			set_interrupt_rm_flag(flags, SECURE);

			rc = register_interrupt_type_handler(INTR_TYPE_NS,
						tspd_ns_interrupt_handler,
						flags);
			if (rc)
				panic();

			/*
432
			 * Disable the NS interrupt locally.
433
434
435
			 */
			disable_intr_rm_local(INTR_TYPE_NS, SECURE);
#endif
436
437
438
439
440
441
442
443
444
445
446
447
448
449
		}


#if TSP_INIT_ASYNC
		/* Save the Secure EL1 system register context */
		assert(cm_get_context(SECURE) == &tsp_ctx->cpu_ctx);
		cm_el1_sysregs_context_save(SECURE);

		/* Program EL3 registers to enable entry into the next EL */
		next_image_info = bl31_plat_get_next_image_ep_info(NON_SECURE);
		assert(next_image_info);
		assert(NON_SECURE ==
				GET_SECURITY_STATE(next_image_info->h.attr));

450
		cm_init_my_context(next_image_info);
451
452
453
		cm_prepare_el3_exit(NON_SECURE);
		SMC_RET0(cm_get_context(NON_SECURE));
#else
454
455
456
457
458
459
		/*
		 * SP reports completion. The SPD must have initiated
		 * the original request through a synchronous entry
		 * into the SP. Jump back to the original C runtime
		 * context.
		 */
460
		tspd_synchronous_sp_exit(tsp_ctx, x1);
461
#endif
462
463
	/*
	 * This function ID is used only by the SP to indicate it has finished
464
	 * aborting a preempted Yielding SMC Call.
465
466
	 */
	case TSP_ABORT_DONE:
467

468
	/*
469
	 * These function IDs are used only by the SP to indicate it has
470
471
472
473
474
475
476
477
478
479
	 * finished:
	 * 1. turning itself on in response to an earlier psci
	 *    cpu_on request
	 * 2. resuming itself after an earlier psci cpu_suspend
	 *    request.
	 */
	case TSP_ON_DONE:
	case TSP_RESUME_DONE:

	/*
480
	 * These function IDs are used only by the SP to indicate it has
481
482
483
484
485
486
487
488
	 * finished:
	 * 1. suspending itself after an earlier psci cpu_suspend
	 *    request.
	 * 2. turning itself off in response to an earlier psci
	 *    cpu_off request.
	 */
	case TSP_OFF_DONE:
	case TSP_SUSPEND_DONE:
489
490
	case TSP_SYSTEM_OFF_DONE:
	case TSP_SYSTEM_RESET_DONE:
491
492
493
494
495
496
497
498
499
		if (ns)
			SMC_RET1(handle, SMC_UNK);

		/*
		 * SP reports completion. The SPD must have initiated the
		 * original request through a synchronous entry into the SP.
		 * Jump back to the original C runtime context, and pass x1 as
		 * return value to the caller
		 */
500
		tspd_synchronous_sp_exit(tsp_ctx, x1);
501

502
503
504
505
506
		/*
		 * Request from non-secure client to perform an
		 * arithmetic operation or response from secure
		 * payload to an earlier request.
		 */
507
508
509
510
511
	case TSP_FAST_FID(TSP_ADD):
	case TSP_FAST_FID(TSP_SUB):
	case TSP_FAST_FID(TSP_MUL):
	case TSP_FAST_FID(TSP_DIV):

512
513
514
515
	case TSP_YIELD_FID(TSP_ADD):
	case TSP_YIELD_FID(TSP_SUB):
	case TSP_YIELD_FID(TSP_MUL):
	case TSP_YIELD_FID(TSP_DIV):
516
517
518
519
520
521
522
		if (ns) {
			/*
			 * This is a fresh request from the non-secure client.
			 * The parameters are in x1 and x2. Figure out which
			 * registers need to be preserved, save the non-secure
			 * state and send the request to the secure payload.
			 */
523
			assert(handle == cm_get_context(NON_SECURE));
524
525

			/* Check if we are already preempted */
526
			if (get_yield_smc_active_flag(tsp_ctx->state))
527
528
				SMC_RET1(handle, SMC_UNK);

529
530
531
			cm_el1_sysregs_context_save(NON_SECURE);

			/* Save x1 and x2 for use by TSP_GET_ARGS call below */
532
			store_tsp_args(tsp_ctx, x1, x2);
533
534
535
536
537
538
539
540
541
542
543
544
545

			/*
			 * We are done stashing the non-secure context. Ask the
			 * secure payload to do the work now.
			 */

			/*
			 * Verify if there is a valid context to use, copy the
			 * operation type and parameters to the secure context
			 * and jump to the fast smc entry point in the secure
			 * payload. Entry into S-EL1 will take place upon exit
			 * from this function.
			 */
546
			assert(&tsp_ctx->cpu_ctx == cm_get_context(SECURE));
547
548
549
550
551
552
553

			/* Set appropriate entry for SMC.
			 * We expect the TSP to manage the PSTATE.I and PSTATE.F
			 * flags as appropriate.
			 */
			if (GET_SMC_TYPE(smc_fid) == SMC_TYPE_FAST) {
				cm_set_elr_el3(SECURE, (uint64_t)
554
						&tsp_vectors->fast_smc_entry);
555
			} else {
556
				set_yield_smc_active_flag(tsp_ctx->state);
557
				cm_set_elr_el3(SECURE, (uint64_t)
558
						&tsp_vectors->yield_smc_entry);
559
#if TSP_NS_INTR_ASYNC_PREEMPT
560
561
				/*
				 * Enable the routing of NS interrupts to EL3
562
563
				 * during processing of a Yielding SMC Call on
				 * this core.
564
565
566
				 */
				enable_intr_rm_local(INTR_TYPE_NS, SECURE);
#endif
567
568
			}

569
570
			cm_el1_sysregs_context_restore(SECURE);
			cm_set_next_eret_context(SECURE);
571
			SMC_RET3(&tsp_ctx->cpu_ctx, smc_fid, x1, x2);
572
573
574
		} else {
			/*
			 * This is the result from the secure client of an
575
			 * earlier request. The results are in x1-x3. Copy it
576
577
578
			 * into the non-secure context, save the secure state
			 * and return to the non-secure state.
			 */
579
			assert(handle == cm_get_context(SECURE));
580
581
582
			cm_el1_sysregs_context_save(SECURE);

			/* Get a reference to the non-secure context */
583
			ns_cpu_context = cm_get_context(NON_SECURE);
584
585
586
587
588
			assert(ns_cpu_context);

			/* Restore non-secure state */
			cm_el1_sysregs_context_restore(NON_SECURE);
			cm_set_next_eret_context(NON_SECURE);
589
590
			if (GET_SMC_TYPE(smc_fid) == SMC_TYPE_YIELD) {
				clr_yield_smc_active_flag(tsp_ctx->state);
591
#if TSP_NS_INTR_ASYNC_PREEMPT
592
593
				/*
				 * Disable the routing of NS interrupts to EL3
594
595
				 * after processing of a Yielding SMC Call on
				 * this core is finished.
596
597
598
599
600
				 */
				disable_intr_rm_local(INTR_TYPE_NS, SECURE);
#endif
			}

601
			SMC_RET3(ns_cpu_context, x1, x2, x3);
602
603
604
		}

		break;
605
	/*
606
607
	 * Request from the non-secure world to abort a preempted Yielding SMC
	 * Call.
608
609
610
611
612
613
614
615
	 */
	case TSP_FID_ABORT:
		/* ABORT should only be invoked by normal world */
		if (!ns) {
			assert(0);
			break;
		}

616
617
618
		assert(handle == cm_get_context(NON_SECURE));
		cm_el1_sysregs_context_save(NON_SECURE);

619
		/* Abort the preempted SMC request */
620
		if (!tspd_abort_preempted_smc(tsp_ctx)) {
621
622
623
			/*
			 * If there was no preempted SMC to abort, return
			 * SMC_UNK.
624
625
626
627
			 *
			 * Restoring the NON_SECURE context is not necessary as
			 * the synchronous entry did not take place if the
			 * return code of tspd_abort_preempted_smc is zero.
628
			 */
629
630
631
			cm_set_next_eret_context(NON_SECURE);
			break;
		}
632

633
634
635
		cm_el1_sysregs_context_restore(NON_SECURE);
		cm_set_next_eret_context(NON_SECURE);
		SMC_RET0(handle);
636

637
638
		/*
		 * Request from non secure world to resume the preempted
639
		 * Yielding SMC Call.
640
641
		 */
	case TSP_FID_RESUME:
642
643
644
645
646
		/* RESUME should be invoked only by normal world */
		if (!ns) {
			assert(0);
			break;
		}
647

648
649
650
651
652
		/*
		 * This is a resume request from the non-secure client.
		 * save the non-secure state and send the request to
		 * the secure payload.
		 */
653
		assert(handle == cm_get_context(NON_SECURE));
654

655
		/* Check if we are already preempted before resume */
656
		if (!get_yield_smc_active_flag(tsp_ctx->state))
657
			SMC_RET1(handle, SMC_UNK);
658

659
		cm_el1_sysregs_context_save(NON_SECURE);
660

661
662
663
664
		/*
		 * We are done stashing the non-secure context. Ask the
		 * secure payload to do the work now.
		 */
665
#if TSP_NS_INTR_ASYNC_PREEMPT
666
667
		/*
		 * Enable the routing of NS interrupts to EL3 during resumption
668
		 * of a Yielding SMC Call on this core.
669
670
671
672
673
		 */
		enable_intr_rm_local(INTR_TYPE_NS, SECURE);
#endif


674

675
676
677
678
679
680
		/* We just need to return to the preempted point in
		 * TSP and the execution will resume as normal.
		 */
		cm_el1_sysregs_context_restore(SECURE);
		cm_set_next_eret_context(SECURE);
		SMC_RET0(&tsp_ctx->cpu_ctx);
681

682
683
684
685
686
687
688
689
690
691
		/*
		 * This is a request from the secure payload for more arguments
		 * for an ongoing arithmetic operation requested by the
		 * non-secure world. Simply return the arguments from the non-
		 * secure client in the original call.
		 */
	case TSP_GET_ARGS:
		if (ns)
			SMC_RET1(handle, SMC_UNK);

692
693
		get_tsp_args(tsp_ctx, x1, x2);
		SMC_RET2(handle, x1, x2);
694

695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
	case TOS_CALL_COUNT:
		/*
		 * Return the number of service function IDs implemented to
		 * provide service to non-secure
		 */
		SMC_RET1(handle, TSP_NUM_FID);

	case TOS_UID:
		/* Return TSP UID to the caller */
		SMC_UUID_RET(handle, tsp_uuid);

	case TOS_CALL_VERSION:
		/* Return the version of current implementation */
		SMC_RET2(handle, TSP_VERSION_MAJOR, TSP_VERSION_MINOR);

710
	default:
711
		break;
712
713
	}

714
	SMC_RET1(handle, SMC_UNK);
715
716
}

717
/* Define a SPD runtime service descriptor for fast SMC calls */
718
DECLARE_RT_SVC(
719
	tspd_fast,
720
721
722
723
724
725
726

	OEN_TOS_START,
	OEN_TOS_END,
	SMC_TYPE_FAST,
	tspd_setup,
	tspd_smc_handler
);
727

728
/* Define a SPD runtime service descriptor for Yielding SMC Calls */
729
730
731
732
733
DECLARE_RT_SVC(
	tspd_std,

	OEN_TOS_START,
	OEN_TOS_END,
734
	SMC_TYPE_YIELD,
735
736
737
	NULL,
	tspd_smc_handler
);