1 /*
2  * Copyright (c) 2013-2017, ARM Limited and Contributors. All rights reserved.
3  *
4  * SPDX-License-Identifier: BSD-3-Clause
5  */
6 
7 #include <assert.h>
8 #include <string.h>
9 
10 #include <arch_helpers.h>
11 #include <common/bl_common.h>
12 #include <lib/el3_runtime/context_mgmt.h>
13 #include <lib/utils.h>
14 
15 #include "opteed_private.h"
16 
17 /*******************************************************************************
18  * Given a OPTEE entrypoint info pointer, entry point PC, register width,
19  * cpu id & pointer to a context data structure, this function will
20  * initialize OPTEE context and entry point info for OPTEE.
21  ******************************************************************************/
opteed_init_optee_ep_state(struct entry_point_info * optee_entry_point,uint32_t rw,uint64_t pc,uint64_t pageable_part,uint64_t mem_limit,uint64_t dt_addr,optee_context_t * optee_ctx)22 void opteed_init_optee_ep_state(struct entry_point_info *optee_entry_point,
23 				uint32_t rw, uint64_t pc,
24 				uint64_t pageable_part, uint64_t mem_limit,
25 				uint64_t dt_addr, optee_context_t *optee_ctx)
26 {
27 	uint32_t ep_attr;
28 
29 	/* Passing a NULL context is a critical programming error */
30 	assert(optee_ctx);
31 	assert(optee_entry_point);
32 	assert(pc);
33 
34 	/* Associate this context with the cpu specified */
35 	optee_ctx->mpidr = read_mpidr_el1();
36 	optee_ctx->state = 0;
37 	set_optee_pstate(optee_ctx->state, OPTEE_PSTATE_OFF);
38 
39 	cm_set_context(&optee_ctx->cpu_ctx, SECURE);
40 
41 	/* initialise an entrypoint to set up the CPU context */
42 	ep_attr = SECURE | EP_ST_ENABLE;
43 	if (read_sctlr_el3() & SCTLR_EE_BIT)
44 		ep_attr |= EP_EE_BIG;
45 	SET_PARAM_HEAD(optee_entry_point, PARAM_EP, VERSION_1, ep_attr);
46 	optee_entry_point->pc = pc;
47 	if (rw == OPTEE_AARCH64)
48 		optee_entry_point->spsr = SPSR_64(MODE_EL1, MODE_SP_ELX,
49 						  DISABLE_ALL_EXCEPTIONS);
50 	else
51 		optee_entry_point->spsr = SPSR_MODE32(MODE32_svc, SPSR_T_ARM,
52 						      SPSR_E_LITTLE,
53 						      DAIF_FIQ_BIT |
54 							DAIF_IRQ_BIT |
55 							DAIF_ABT_BIT);
56 	zeromem(&optee_entry_point->args, sizeof(optee_entry_point->args));
57 	optee_entry_point->args.arg0 = pageable_part;
58 	optee_entry_point->args.arg1 = mem_limit;
59 	optee_entry_point->args.arg2 = dt_addr;
60 }
61 
62 /*******************************************************************************
63  * This function takes an OPTEE context pointer and:
64  * 1. Applies the S-EL1 system register context from optee_ctx->cpu_ctx.
65  * 2. Saves the current C runtime state (callee saved registers) on the stack
66  *    frame and saves a reference to this state.
67  * 3. Calls el3_exit() so that the EL3 system and general purpose registers
68  *    from the optee_ctx->cpu_ctx are used to enter the OPTEE image.
69  ******************************************************************************/
opteed_synchronous_sp_entry(optee_context_t * optee_ctx)70 uint64_t opteed_synchronous_sp_entry(optee_context_t *optee_ctx)
71 {
72 	uint64_t rc;
73 
74 	assert(optee_ctx != NULL);
75 	assert(optee_ctx->c_rt_ctx == 0);
76 
77 	/* Apply the Secure EL1 system register context and switch to it */
78 	assert(cm_get_context(SECURE) == &optee_ctx->cpu_ctx);
79 	cm_el1_sysregs_context_restore(SECURE);
80 	cm_set_next_eret_context(SECURE);
81 
82 	rc = opteed_enter_sp(&optee_ctx->c_rt_ctx);
83 #if ENABLE_ASSERTIONS
84 	optee_ctx->c_rt_ctx = 0;
85 #endif
86 
87 	return rc;
88 }
89 
90 
91 /*******************************************************************************
92  * This function takes an OPTEE context pointer and:
93  * 1. Saves the S-EL1 system register context tp optee_ctx->cpu_ctx.
94  * 2. Restores the current C runtime state (callee saved registers) from the
95  *    stack frame using the reference to this state saved in opteed_enter_sp().
96  * 3. It does not need to save any general purpose or EL3 system register state
97  *    as the generic smc entry routine should have saved those.
98  ******************************************************************************/
opteed_synchronous_sp_exit(optee_context_t * optee_ctx,uint64_t ret)99 void opteed_synchronous_sp_exit(optee_context_t *optee_ctx, uint64_t ret)
100 {
101 	assert(optee_ctx != NULL);
102 	/* Save the Secure EL1 system register context */
103 	assert(cm_get_context(SECURE) == &optee_ctx->cpu_ctx);
104 	cm_el1_sysregs_context_save(SECURE);
105 
106 	assert(optee_ctx->c_rt_ctx != 0);
107 	opteed_exit_sp(optee_ctx->c_rt_ctx, ret);
108 
109 	/* Should never reach here */
110 	assert(0);
111 }
112