1 // SPDX-License-Identifier: BSD-2-Clause
2 /*
3  * Copyright (c) 2016-2022, Linaro Limited
4  */
5 
6 #include <gen-asm-defines.h>
7 #include <kernel/boot.h>
8 #include <kernel/thread.h>
9 #include <kernel/thread_private.h>
10 #include <sm/pm.h>
11 #include <sm/sm.h>
12 #include <types_ext.h>
13 
14 DEFINES
15 {
16 #ifdef ARM32
17 	DEFINE(SM_NSEC_CTX_R0, offsetof(struct sm_nsec_ctx, r0));
18 	DEFINE(SM_NSEC_CTX_R8, offsetof(struct sm_nsec_ctx, r8));
19 	DEFINE(SM_SEC_CTX_R0, offsetof(struct sm_sec_ctx, r0));
20 	DEFINE(SM_SEC_CTX_MON_LR, offsetof(struct sm_sec_ctx, mon_lr));
21 	DEFINE(SM_CTX_SEC_SIZE, sizeof(struct sm_sec_ctx));
22 	DEFINE(SM_CTX_SIZE, sizeof(struct sm_ctx));
23 	DEFINE(SM_CTX_NSEC, offsetof(struct sm_ctx, nsec));
24 	DEFINE(SM_CTX_SEC, offsetof(struct sm_ctx, sec));
25 
26 	DEFINE(THREAD_SVC_REG_R0, offsetof(struct thread_svc_regs, r0));
27 	DEFINE(THREAD_SVC_REG_R5, offsetof(struct thread_svc_regs, r5));
28 	DEFINE(THREAD_SVC_REG_R6, offsetof(struct thread_svc_regs, r6));
29 
30 	/* struct thread_ctx */
31 	DEFINE(THREAD_CTX_STACK_VA_END, offsetof(struct thread_ctx,
32 						 stack_va_end));
33 
34 	/* struct thread_ctx_regs */
35 	DEFINE(THREAD_CTX_REGS_USR_SP,
36 	       offsetof(struct thread_ctx_regs, usr_sp));
37 	DEFINE(THREAD_CTX_REGS_PC, offsetof(struct thread_ctx_regs, pc));
38 	DEFINE(THREAD_CTX_REGS_CPSR, offsetof(struct thread_ctx_regs, cpsr));
39 
40 	/* struct thread_core_local */
41 	DEFINE(THREAD_CORE_LOCAL_R0, offsetof(struct thread_core_local, r[0]));
42 	DEFINE(THREAD_CORE_LOCAL_SM_PM_CTX_PHYS,
43 	       offsetof(struct thread_core_local, sm_pm_ctx_phys));
44 	DEFINE(THREAD_CORE_LOCAL_SIZE, sizeof(struct thread_core_local));
45 
46 	DEFINE(SM_PM_CTX_SIZE, sizeof(struct sm_pm_ctx));
47 #endif /*ARM32*/
48 
49 #ifdef ARM64
50 	DEFINE(THREAD_SMC_ARGS_X0, offsetof(struct thread_smc_args, a0));
51 	DEFINE(THREAD_SMC_ARGS_SIZE, sizeof(struct thread_smc_args));
52 
53 	DEFINE(THREAD_SVC_REG_X0, offsetof(struct thread_svc_regs, x0));
54 	DEFINE(THREAD_SVC_REG_X2, offsetof(struct thread_svc_regs, x2));
55 	DEFINE(THREAD_SVC_REG_X5, offsetof(struct thread_svc_regs, x5));
56 	DEFINE(THREAD_SVC_REG_X6, offsetof(struct thread_svc_regs, x6));
57 	DEFINE(THREAD_SVC_REG_X30, offsetof(struct thread_svc_regs, x30));
58 	DEFINE(THREAD_SVC_REG_ELR, offsetof(struct thread_svc_regs, elr));
59 	DEFINE(THREAD_SVC_REG_SPSR, offsetof(struct thread_svc_regs, spsr));
60 	DEFINE(THREAD_SVC_REG_SP_EL0, offsetof(struct thread_svc_regs, sp_el0));
61 #ifdef CFG_TA_PAUTH
62 	DEFINE(THREAD_SVC_REG_APIAKEY_HI, offsetof(struct thread_svc_regs,
63 						   apiakey_hi));
64 #endif
65 	DEFINE(THREAD_SVC_REG_SIZE, sizeof(struct thread_svc_regs));
66 
67 	/* struct thread_abort_regs */
68 	DEFINE(THREAD_ABT_REG_X0, offsetof(struct thread_abort_regs, x0));
69 	DEFINE(THREAD_ABT_REG_X2, offsetof(struct thread_abort_regs, x2));
70 	DEFINE(THREAD_ABT_REG_X30, offsetof(struct thread_abort_regs, x30));
71 	DEFINE(THREAD_ABT_REG_SPSR, offsetof(struct thread_abort_regs, spsr));
72 	DEFINE(THREAD_ABT_REGS_SIZE, sizeof(struct thread_abort_regs));
73 #if defined(CFG_TA_PAUTH) || defined(CFG_CORE_PAUTH)
74 	DEFINE(THREAD_ABT_REGS_APIAKEY_HI, offsetof(struct thread_abort_regs,
75 						    apiakey_hi));
76 #endif
77 
78 	/* struct thread_ctx */
79 	DEFINE(THREAD_CTX_KERN_SP, offsetof(struct thread_ctx, kern_sp));
80 	DEFINE(THREAD_CTX_STACK_VA_END, offsetof(struct thread_ctx,
81 						 stack_va_end));
82 #if defined(CFG_CORE_PAUTH)
83 	DEFINE(THREAD_CTX_KEYS, offsetof(struct thread_ctx, keys));
84 #endif
85 
86 	/* struct thread_ctx_regs */
87 	DEFINE(THREAD_CTX_REGS_SP, offsetof(struct thread_ctx_regs, sp));
88 	DEFINE(THREAD_CTX_REGS_X0, offsetof(struct thread_ctx_regs, x[0]));
89 	DEFINE(THREAD_CTX_REGS_X1, offsetof(struct thread_ctx_regs, x[1]));
90 	DEFINE(THREAD_CTX_REGS_X2, offsetof(struct thread_ctx_regs, x[2]));
91 	DEFINE(THREAD_CTX_REGS_X4, offsetof(struct thread_ctx_regs, x[4]));
92 	DEFINE(THREAD_CTX_REGS_X19, offsetof(struct thread_ctx_regs, x[19]));
93 	DEFINE(THREAD_CTX_REGS_TPIDR_EL0, offsetof(struct thread_ctx_regs,
94 						   tpidr_el0));
95 #if defined(CFG_TA_PAUTH) || defined(CFG_CORE_PAUTH)
96 	DEFINE(THREAD_CTX_REGS_APIAKEY_HI, offsetof(struct thread_ctx_regs,
97 						    apiakey_hi));
98 #endif
99 
100 	/* struct thread_user_mode_rec */
101 	DEFINE(THREAD_USER_MODE_REC_CTX_REGS_PTR,
102 	       offsetof(struct thread_user_mode_rec, ctx_regs_ptr));
103 	DEFINE(THREAD_USER_MODE_REC_EXIT_STATUS0_PTR,
104 	       offsetof(struct thread_user_mode_rec, exit_status0_ptr));
105 	DEFINE(THREAD_USER_MODE_REC_X19,
106 	       offsetof(struct thread_user_mode_rec, x[0]));
107 	DEFINE(THREAD_USER_MODE_REC_SIZE, sizeof(struct thread_user_mode_rec));
108 
109 	/* struct thread_core_local */
110 	DEFINE(THREAD_CORE_LOCAL_X0, offsetof(struct thread_core_local, x[0]));
111 	DEFINE(THREAD_CORE_LOCAL_X2, offsetof(struct thread_core_local, x[2]));
112 	DEFINE(THREAD_CORE_LOCAL_KCODE_OFFSET,
113 	       offsetof(struct thread_core_local, kcode_offset));
114 #ifdef CFG_CORE_WORKAROUND_SPECTRE_BP_SEC
115 	DEFINE(THREAD_CORE_LOCAL_BHB_LOOP_COUNT,
116 	       offsetof(struct thread_core_local, bhb_loop_count));
117 #endif
118 #if defined(CFG_CORE_PAUTH)
119 	DEFINE(THREAD_CORE_LOCAL_KEYS,
120 	       offsetof(struct thread_core_local, keys));
121 #endif
122 #endif /*ARM64*/
123 
124 	/* struct thread_ctx */
125 	DEFINE(THREAD_CTX_SIZE, sizeof(struct thread_ctx));
126 #ifdef CFG_CORE_FFA
127 	DEFINE(THREAD_CTX_TSD_RPC_TARGET_INFO,
128 	       offsetof(struct thread_ctx, tsd.rpc_target_info))
129 #endif
130 
131 	/* struct thread_core_local */
132 	DEFINE(THREAD_CORE_LOCAL_TMP_STACK_VA_END,
133 		offsetof(struct thread_core_local, tmp_stack_va_end));
134 	DEFINE(THREAD_CORE_LOCAL_CURR_THREAD,
135 		offsetof(struct thread_core_local, curr_thread));
136 	DEFINE(THREAD_CORE_LOCAL_FLAGS,
137 		offsetof(struct thread_core_local, flags));
138 	DEFINE(THREAD_CORE_LOCAL_ABT_STACK_VA_END,
139 		offsetof(struct thread_core_local, abt_stack_va_end));
140 
141 	DEFINE(STACK_TMP_GUARD, STACK_CANARY_SIZE / 2 + STACK_TMP_OFFS);
142 
143 	/* struct core_mmu_config */
144 	DEFINE(CORE_MMU_CONFIG_SIZE, sizeof(struct core_mmu_config));
145 	DEFINE(CORE_MMU_CONFIG_LOAD_OFFSET,
146 	       offsetof(struct core_mmu_config, load_offset));
147 
148 	/* struct boot_embdata */
149 	DEFINE(BOOT_EMBDATA_HASHES_OFFSET,
150 	       offsetof(struct boot_embdata, hashes_offset));
151 	DEFINE(BOOT_EMBDATA_HASHES_LEN,
152 	       offsetof(struct boot_embdata, hashes_len));
153 	DEFINE(BOOT_EMBDATA_RELOC_OFFSET,
154 	       offsetof(struct boot_embdata, reloc_offset));
155 	DEFINE(BOOT_EMBDATA_RELOC_LEN,
156 	       offsetof(struct boot_embdata, reloc_len));
157 }
158