1 /* 2 * Copyright (c) 2013-2019, ARM Limited and Contributors. All rights reserved. 3 * 4 * SPDX-License-Identifier: BSD-3-Clause 5 */ 6 7 #ifndef CONTEXT_H 8 #define CONTEXT_H 9 10 #include <lib/utils_def.h> 11 12 /******************************************************************************* 13 * Constants that allow assembler code to access members of and the 'gp_regs' 14 * structure at their correct offsets. 15 ******************************************************************************/ 16 #define CTX_GPREGS_OFFSET U(0x0) 17 #define CTX_GPREG_X0 U(0x0) 18 #define CTX_GPREG_X1 U(0x8) 19 #define CTX_GPREG_X2 U(0x10) 20 #define CTX_GPREG_X3 U(0x18) 21 #define CTX_GPREG_X4 U(0x20) 22 #define CTX_GPREG_X5 U(0x28) 23 #define CTX_GPREG_X6 U(0x30) 24 #define CTX_GPREG_X7 U(0x38) 25 #define CTX_GPREG_X8 U(0x40) 26 #define CTX_GPREG_X9 U(0x48) 27 #define CTX_GPREG_X10 U(0x50) 28 #define CTX_GPREG_X11 U(0x58) 29 #define CTX_GPREG_X12 U(0x60) 30 #define CTX_GPREG_X13 U(0x68) 31 #define CTX_GPREG_X14 U(0x70) 32 #define CTX_GPREG_X15 U(0x78) 33 #define CTX_GPREG_X16 U(0x80) 34 #define CTX_GPREG_X17 U(0x88) 35 #define CTX_GPREG_X18 U(0x90) 36 #define CTX_GPREG_X19 U(0x98) 37 #define CTX_GPREG_X20 U(0xa0) 38 #define CTX_GPREG_X21 U(0xa8) 39 #define CTX_GPREG_X22 U(0xb0) 40 #define CTX_GPREG_X23 U(0xb8) 41 #define CTX_GPREG_X24 U(0xc0) 42 #define CTX_GPREG_X25 U(0xc8) 43 #define CTX_GPREG_X26 U(0xd0) 44 #define CTX_GPREG_X27 U(0xd8) 45 #define CTX_GPREG_X28 U(0xe0) 46 #define CTX_GPREG_X29 U(0xe8) 47 #define CTX_GPREG_LR U(0xf0) 48 #define CTX_GPREG_SP_EL0 U(0xf8) 49 #define CTX_GPREGS_END U(0x100) 50 51 /******************************************************************************* 52 * Constants that allow assembler code to access members of and the 'el3_state' 53 * structure at their correct offsets. Note that some of the registers are only 54 * 32-bits wide but are stored as 64-bit values for convenience 55 ******************************************************************************/ 56 #define CTX_EL3STATE_OFFSET (CTX_GPREGS_OFFSET + CTX_GPREGS_END) 57 #define CTX_SCR_EL3 U(0x0) 58 #define CTX_ESR_EL3 U(0x8) 59 #define CTX_RUNTIME_SP U(0x10) 60 #define CTX_SPSR_EL3 U(0x18) 61 #define CTX_ELR_EL3 U(0x20) 62 #define CTX_PMCR_EL0 U(0x28) 63 #define CTX_EL3STATE_END U(0x30) 64 65 /******************************************************************************* 66 * Constants that allow assembler code to access members of and the 67 * 'el1_sys_regs' structure at their correct offsets. Note that some of the 68 * registers are only 32-bits wide but are stored as 64-bit values for 69 * convenience 70 ******************************************************************************/ 71 #define CTX_SYSREGS_OFFSET (CTX_EL3STATE_OFFSET + CTX_EL3STATE_END) 72 #define CTX_SPSR_EL1 U(0x0) 73 #define CTX_ELR_EL1 U(0x8) 74 #define CTX_SCTLR_EL1 U(0x10) 75 #define CTX_ACTLR_EL1 U(0x18) 76 #define CTX_CPACR_EL1 U(0x20) 77 #define CTX_CSSELR_EL1 U(0x28) 78 #define CTX_SP_EL1 U(0x30) 79 #define CTX_ESR_EL1 U(0x38) 80 #define CTX_TTBR0_EL1 U(0x40) 81 #define CTX_TTBR1_EL1 U(0x48) 82 #define CTX_MAIR_EL1 U(0x50) 83 #define CTX_AMAIR_EL1 U(0x58) 84 #define CTX_TCR_EL1 U(0x60) 85 #define CTX_TPIDR_EL1 U(0x68) 86 #define CTX_TPIDR_EL0 U(0x70) 87 #define CTX_TPIDRRO_EL0 U(0x78) 88 #define CTX_PAR_EL1 U(0x80) 89 #define CTX_FAR_EL1 U(0x88) 90 #define CTX_AFSR0_EL1 U(0x90) 91 #define CTX_AFSR1_EL1 U(0x98) 92 #define CTX_CONTEXTIDR_EL1 U(0xa0) 93 #define CTX_VBAR_EL1 U(0xa8) 94 95 /* 96 * If the platform is AArch64-only, there is no need to save and restore these 97 * AArch32 registers. 98 */ 99 #if CTX_INCLUDE_AARCH32_REGS 100 #define CTX_SPSR_ABT U(0xb0) /* Align to the next 16 byte boundary */ 101 #define CTX_SPSR_UND U(0xb8) 102 #define CTX_SPSR_IRQ U(0xc0) 103 #define CTX_SPSR_FIQ U(0xc8) 104 #define CTX_DACR32_EL2 U(0xd0) 105 #define CTX_IFSR32_EL2 U(0xd8) 106 #define CTX_AARCH32_END U(0xe0) /* Align to the next 16 byte boundary */ 107 #else 108 #define CTX_AARCH32_END U(0xb0) /* Align to the next 16 byte boundary */ 109 #endif /* CTX_INCLUDE_AARCH32_REGS */ 110 111 /* 112 * If the timer registers aren't saved and restored, we don't have to reserve 113 * space for them in the context 114 */ 115 #if NS_TIMER_SWITCH 116 #define CTX_CNTP_CTL_EL0 (CTX_AARCH32_END + U(0x0)) 117 #define CTX_CNTP_CVAL_EL0 (CTX_AARCH32_END + U(0x8)) 118 #define CTX_CNTV_CTL_EL0 (CTX_AARCH32_END + U(0x10)) 119 #define CTX_CNTV_CVAL_EL0 (CTX_AARCH32_END + U(0x18)) 120 #define CTX_CNTKCTL_EL1 (CTX_AARCH32_END + U(0x20)) 121 #define CTX_TIMER_SYSREGS_END (CTX_AARCH32_END + U(0x30)) /* Align to the next 16 byte boundary */ 122 #else 123 #define CTX_TIMER_SYSREGS_END CTX_AARCH32_END 124 #endif /* NS_TIMER_SWITCH */ 125 126 /* 127 * End of system registers. 128 */ 129 #define CTX_SYSREGS_END CTX_TIMER_SYSREGS_END 130 131 /******************************************************************************* 132 * Constants that allow assembler code to access members of and the 'fp_regs' 133 * structure at their correct offsets. 134 ******************************************************************************/ 135 #define CTX_FPREGS_OFFSET (CTX_SYSREGS_OFFSET + CTX_SYSREGS_END) 136 #if CTX_INCLUDE_FPREGS 137 #define CTX_FP_Q0 U(0x0) 138 #define CTX_FP_Q1 U(0x10) 139 #define CTX_FP_Q2 U(0x20) 140 #define CTX_FP_Q3 U(0x30) 141 #define CTX_FP_Q4 U(0x40) 142 #define CTX_FP_Q5 U(0x50) 143 #define CTX_FP_Q6 U(0x60) 144 #define CTX_FP_Q7 U(0x70) 145 #define CTX_FP_Q8 U(0x80) 146 #define CTX_FP_Q9 U(0x90) 147 #define CTX_FP_Q10 U(0xa0) 148 #define CTX_FP_Q11 U(0xb0) 149 #define CTX_FP_Q12 U(0xc0) 150 #define CTX_FP_Q13 U(0xd0) 151 #define CTX_FP_Q14 U(0xe0) 152 #define CTX_FP_Q15 U(0xf0) 153 #define CTX_FP_Q16 U(0x100) 154 #define CTX_FP_Q17 U(0x110) 155 #define CTX_FP_Q18 U(0x120) 156 #define CTX_FP_Q19 U(0x130) 157 #define CTX_FP_Q20 U(0x140) 158 #define CTX_FP_Q21 U(0x150) 159 #define CTX_FP_Q22 U(0x160) 160 #define CTX_FP_Q23 U(0x170) 161 #define CTX_FP_Q24 U(0x180) 162 #define CTX_FP_Q25 U(0x190) 163 #define CTX_FP_Q26 U(0x1a0) 164 #define CTX_FP_Q27 U(0x1b0) 165 #define CTX_FP_Q28 U(0x1c0) 166 #define CTX_FP_Q29 U(0x1d0) 167 #define CTX_FP_Q30 U(0x1e0) 168 #define CTX_FP_Q31 U(0x1f0) 169 #define CTX_FP_FPSR U(0x200) 170 #define CTX_FP_FPCR U(0x208) 171 #if CTX_INCLUDE_AARCH32_REGS 172 #define CTX_FP_FPEXC32_EL2 U(0x210) 173 #define CTX_FPREGS_END U(0x220) /* Align to the next 16 byte boundary */ 174 #else 175 #define CTX_FPREGS_END U(0x210) /* Align to the next 16 byte boundary */ 176 #endif 177 #else 178 #define CTX_FPREGS_END U(0) 179 #endif 180 181 /******************************************************************************* 182 * Registers related to CVE-2018-3639 183 ******************************************************************************/ 184 #define CTX_CVE_2018_3639_OFFSET (CTX_FPREGS_OFFSET + CTX_FPREGS_END) 185 #define CTX_CVE_2018_3639_DISABLE U(0) 186 #define CTX_CVE_2018_3639_END U(0x10) /* Align to the next 16 byte boundary */ 187 188 /******************************************************************************* 189 * Registers related to ARMv8.3-PAuth. 190 ******************************************************************************/ 191 #define CTX_PAUTH_REGS_OFFSET (CTX_CVE_2018_3639_OFFSET + CTX_CVE_2018_3639_END) 192 #if CTX_INCLUDE_PAUTH_REGS 193 #define CTX_PACIAKEY_LO U(0x0) 194 #define CTX_PACIAKEY_HI U(0x8) 195 #define CTX_PACIBKEY_LO U(0x10) 196 #define CTX_PACIBKEY_HI U(0x18) 197 #define CTX_PACDAKEY_LO U(0x20) 198 #define CTX_PACDAKEY_HI U(0x28) 199 #define CTX_PACDBKEY_LO U(0x30) 200 #define CTX_PACDBKEY_HI U(0x38) 201 #define CTX_PACGAKEY_LO U(0x40) 202 #define CTX_PACGAKEY_HI U(0x48) 203 #define CTX_PACGAKEY_END U(0x50) 204 #define CTX_PAUTH_REGS_END U(0x60) /* Align to the next 16 byte boundary */ 205 #else 206 #define CTX_PAUTH_REGS_END U(0) 207 #endif /* CTX_INCLUDE_PAUTH_REGS */ 208 209 #ifndef __ASSEMBLER__ 210 211 #include <stdint.h> 212 213 #include <lib/cassert.h> 214 215 /* 216 * Common constants to help define the 'cpu_context' structure and its 217 * members below. 218 */ 219 #define DWORD_SHIFT U(3) 220 #define DEFINE_REG_STRUCT(name, num_regs) \ 221 typedef struct name { \ 222 uint64_t _regs[num_regs]; \ 223 } __aligned(16) name##_t 224 225 /* Constants to determine the size of individual context structures */ 226 #define CTX_GPREG_ALL (CTX_GPREGS_END >> DWORD_SHIFT) 227 #define CTX_SYSREG_ALL (CTX_SYSREGS_END >> DWORD_SHIFT) 228 #if CTX_INCLUDE_FPREGS 229 # define CTX_FPREG_ALL (CTX_FPREGS_END >> DWORD_SHIFT) 230 #endif 231 #define CTX_EL3STATE_ALL (CTX_EL3STATE_END >> DWORD_SHIFT) 232 #define CTX_CVE_2018_3639_ALL (CTX_CVE_2018_3639_END >> DWORD_SHIFT) 233 #if CTX_INCLUDE_PAUTH_REGS 234 # define CTX_PAUTH_REGS_ALL (CTX_PAUTH_REGS_END >> DWORD_SHIFT) 235 #endif 236 237 /* 238 * AArch64 general purpose register context structure. Usually x0-x18, 239 * lr are saved as the compiler is expected to preserve the remaining 240 * callee saved registers if used by the C runtime and the assembler 241 * does not touch the remaining. But in case of world switch during 242 * exception handling, we need to save the callee registers too. 243 */ 244 DEFINE_REG_STRUCT(gp_regs, CTX_GPREG_ALL); 245 246 /* 247 * AArch64 EL1 system register context structure for preserving the 248 * architectural state during switches from one security state to 249 * another in EL1. 250 */ 251 DEFINE_REG_STRUCT(el1_sys_regs, CTX_SYSREG_ALL); 252 253 /* 254 * AArch64 floating point register context structure for preserving 255 * the floating point state during switches from one security state to 256 * another. 257 */ 258 #if CTX_INCLUDE_FPREGS 259 DEFINE_REG_STRUCT(fp_regs, CTX_FPREG_ALL); 260 #endif 261 262 /* 263 * Miscellaneous registers used by EL3 firmware to maintain its state 264 * across exception entries and exits 265 */ 266 DEFINE_REG_STRUCT(el3_state, CTX_EL3STATE_ALL); 267 268 /* Function pointer used by CVE-2018-3639 dynamic mitigation */ 269 DEFINE_REG_STRUCT(cve_2018_3639, CTX_CVE_2018_3639_ALL); 270 271 /* Registers associated to ARMv8.3-PAuth */ 272 #if CTX_INCLUDE_PAUTH_REGS 273 DEFINE_REG_STRUCT(pauth, CTX_PAUTH_REGS_ALL); 274 #endif 275 276 /* 277 * Macros to access members of any of the above structures using their 278 * offsets 279 */ 280 #define read_ctx_reg(ctx, offset) ((ctx)->_regs[(offset) >> DWORD_SHIFT]) 281 #define write_ctx_reg(ctx, offset, val) (((ctx)->_regs[(offset) >> DWORD_SHIFT]) \ 282 = (uint64_t) (val)) 283 284 /* 285 * Top-level context structure which is used by EL3 firmware to 286 * preserve the state of a core at EL1 in one of the two security 287 * states and save enough EL3 meta data to be able to return to that 288 * EL and security state. The context management library will be used 289 * to ensure that SP_EL3 always points to an instance of this 290 * structure at exception entry and exit. Each instance will 291 * correspond to either the secure or the non-secure state. 292 */ 293 typedef struct cpu_context { 294 gp_regs_t gpregs_ctx; 295 el3_state_t el3state_ctx; 296 el1_sys_regs_t sysregs_ctx; 297 #if CTX_INCLUDE_FPREGS 298 fp_regs_t fpregs_ctx; 299 #endif 300 cve_2018_3639_t cve_2018_3639_ctx; 301 #if CTX_INCLUDE_PAUTH_REGS 302 pauth_t pauth_ctx; 303 #endif 304 } cpu_context_t; 305 306 /* Macros to access members of the 'cpu_context_t' structure */ 307 #define get_el3state_ctx(h) (&((cpu_context_t *) h)->el3state_ctx) 308 #if CTX_INCLUDE_FPREGS 309 # define get_fpregs_ctx(h) (&((cpu_context_t *) h)->fpregs_ctx) 310 #endif 311 #define get_sysregs_ctx(h) (&((cpu_context_t *) h)->sysregs_ctx) 312 #define get_gpregs_ctx(h) (&((cpu_context_t *) h)->gpregs_ctx) 313 #define get_cve_2018_3639_ctx(h) (&((cpu_context_t *) h)->cve_2018_3639_ctx) 314 #if CTX_INCLUDE_PAUTH_REGS 315 # define get_pauth_ctx(h) (&((cpu_context_t *) h)->pauth_ctx) 316 #endif 317 318 /* 319 * Compile time assertions related to the 'cpu_context' structure to 320 * ensure that the assembler and the compiler view of the offsets of 321 * the structure members is the same. 322 */ 323 CASSERT(CTX_GPREGS_OFFSET == __builtin_offsetof(cpu_context_t, gpregs_ctx), \ 324 assert_core_context_gp_offset_mismatch); 325 CASSERT(CTX_SYSREGS_OFFSET == __builtin_offsetof(cpu_context_t, sysregs_ctx), \ 326 assert_core_context_sys_offset_mismatch); 327 #if CTX_INCLUDE_FPREGS 328 CASSERT(CTX_FPREGS_OFFSET == __builtin_offsetof(cpu_context_t, fpregs_ctx), \ 329 assert_core_context_fp_offset_mismatch); 330 #endif 331 CASSERT(CTX_EL3STATE_OFFSET == __builtin_offsetof(cpu_context_t, el3state_ctx), \ 332 assert_core_context_el3state_offset_mismatch); 333 CASSERT(CTX_CVE_2018_3639_OFFSET == __builtin_offsetof(cpu_context_t, cve_2018_3639_ctx), \ 334 assert_core_context_cve_2018_3639_offset_mismatch); 335 #if CTX_INCLUDE_PAUTH_REGS 336 CASSERT(CTX_PAUTH_REGS_OFFSET == __builtin_offsetof(cpu_context_t, pauth_ctx), \ 337 assert_core_context_pauth_offset_mismatch); 338 #endif 339 340 /* 341 * Helper macro to set the general purpose registers that correspond to 342 * parameters in an aapcs_64 call i.e. x0-x7 343 */ 344 #define set_aapcs_args0(ctx, x0) do { \ 345 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X0, x0); \ 346 } while (0) 347 #define set_aapcs_args1(ctx, x0, x1) do { \ 348 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X1, x1); \ 349 set_aapcs_args0(ctx, x0); \ 350 } while (0) 351 #define set_aapcs_args2(ctx, x0, x1, x2) do { \ 352 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X2, x2); \ 353 set_aapcs_args1(ctx, x0, x1); \ 354 } while (0) 355 #define set_aapcs_args3(ctx, x0, x1, x2, x3) do { \ 356 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X3, x3); \ 357 set_aapcs_args2(ctx, x0, x1, x2); \ 358 } while (0) 359 #define set_aapcs_args4(ctx, x0, x1, x2, x3, x4) do { \ 360 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X4, x4); \ 361 set_aapcs_args3(ctx, x0, x1, x2, x3); \ 362 } while (0) 363 #define set_aapcs_args5(ctx, x0, x1, x2, x3, x4, x5) do { \ 364 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X5, x5); \ 365 set_aapcs_args4(ctx, x0, x1, x2, x3, x4); \ 366 } while (0) 367 #define set_aapcs_args6(ctx, x0, x1, x2, x3, x4, x5, x6) do { \ 368 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X6, x6); \ 369 set_aapcs_args5(ctx, x0, x1, x2, x3, x4, x5); \ 370 } while (0) 371 #define set_aapcs_args7(ctx, x0, x1, x2, x3, x4, x5, x6, x7) do { \ 372 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X7, x7); \ 373 set_aapcs_args6(ctx, x0, x1, x2, x3, x4, x5, x6); \ 374 } while (0) 375 376 /******************************************************************************* 377 * Function prototypes 378 ******************************************************************************/ 379 void el1_sysregs_context_save(el1_sys_regs_t *regs); 380 void el1_sysregs_context_restore(el1_sys_regs_t *regs); 381 #if CTX_INCLUDE_FPREGS 382 void fpregs_context_save(fp_regs_t *regs); 383 void fpregs_context_restore(fp_regs_t *regs); 384 #endif 385 386 #endif /* __ASSEMBLER__ */ 387 388 #endif /* CONTEXT_H */ 389