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_UNUSED 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 #define CTX_PMCR_EL0 U(0xb0) 95 96 /* 97 * If the platform is AArch64-only, there is no need to save and restore these 98 * AArch32 registers. 99 */ 100 #if CTX_INCLUDE_AARCH32_REGS 101 #define CTX_SPSR_ABT U(0xc0) /* Align to the next 16 byte boundary */ 102 #define CTX_SPSR_UND U(0xc8) 103 #define CTX_SPSR_IRQ U(0xd0) 104 #define CTX_SPSR_FIQ U(0xd8) 105 #define CTX_DACR32_EL2 U(0xe0) 106 #define CTX_IFSR32_EL2 U(0xe8) 107 #define CTX_AARCH32_END U(0xf0) /* Align to the next 16 byte boundary */ 108 #else 109 #define CTX_AARCH32_END U(0xc0) /* Align to the next 16 byte boundary */ 110 #endif /* CTX_INCLUDE_AARCH32_REGS */ 111 112 /* 113 * If the timer registers aren't saved and restored, we don't have to reserve 114 * space for them in the context 115 */ 116 #if NS_TIMER_SWITCH 117 #define CTX_CNTP_CTL_EL0 (CTX_AARCH32_END + U(0x0)) 118 #define CTX_CNTP_CVAL_EL0 (CTX_AARCH32_END + U(0x8)) 119 #define CTX_CNTV_CTL_EL0 (CTX_AARCH32_END + U(0x10)) 120 #define CTX_CNTV_CVAL_EL0 (CTX_AARCH32_END + U(0x18)) 121 #define CTX_CNTKCTL_EL1 (CTX_AARCH32_END + U(0x20)) 122 #define CTX_TIMER_SYSREGS_END (CTX_AARCH32_END + U(0x30)) /* Align to the next 16 byte boundary */ 123 #else 124 #define CTX_TIMER_SYSREGS_END CTX_AARCH32_END 125 #endif /* NS_TIMER_SWITCH */ 126 127 /* 128 * End of system registers. 129 */ 130 #define CTX_SYSREGS_END CTX_TIMER_SYSREGS_END 131 132 /******************************************************************************* 133 * Constants that allow assembler code to access members of and the 'fp_regs' 134 * structure at their correct offsets. 135 ******************************************************************************/ 136 #define CTX_FPREGS_OFFSET (CTX_SYSREGS_OFFSET + CTX_SYSREGS_END) 137 #if CTX_INCLUDE_FPREGS 138 #define CTX_FP_Q0 U(0x0) 139 #define CTX_FP_Q1 U(0x10) 140 #define CTX_FP_Q2 U(0x20) 141 #define CTX_FP_Q3 U(0x30) 142 #define CTX_FP_Q4 U(0x40) 143 #define CTX_FP_Q5 U(0x50) 144 #define CTX_FP_Q6 U(0x60) 145 #define CTX_FP_Q7 U(0x70) 146 #define CTX_FP_Q8 U(0x80) 147 #define CTX_FP_Q9 U(0x90) 148 #define CTX_FP_Q10 U(0xa0) 149 #define CTX_FP_Q11 U(0xb0) 150 #define CTX_FP_Q12 U(0xc0) 151 #define CTX_FP_Q13 U(0xd0) 152 #define CTX_FP_Q14 U(0xe0) 153 #define CTX_FP_Q15 U(0xf0) 154 #define CTX_FP_Q16 U(0x100) 155 #define CTX_FP_Q17 U(0x110) 156 #define CTX_FP_Q18 U(0x120) 157 #define CTX_FP_Q19 U(0x130) 158 #define CTX_FP_Q20 U(0x140) 159 #define CTX_FP_Q21 U(0x150) 160 #define CTX_FP_Q22 U(0x160) 161 #define CTX_FP_Q23 U(0x170) 162 #define CTX_FP_Q24 U(0x180) 163 #define CTX_FP_Q25 U(0x190) 164 #define CTX_FP_Q26 U(0x1a0) 165 #define CTX_FP_Q27 U(0x1b0) 166 #define CTX_FP_Q28 U(0x1c0) 167 #define CTX_FP_Q29 U(0x1d0) 168 #define CTX_FP_Q30 U(0x1e0) 169 #define CTX_FP_Q31 U(0x1f0) 170 #define CTX_FP_FPSR U(0x200) 171 #define CTX_FP_FPCR U(0x208) 172 #if CTX_INCLUDE_AARCH32_REGS 173 #define CTX_FP_FPEXC32_EL2 U(0x210) 174 #define CTX_FPREGS_END U(0x220) /* Align to the next 16 byte boundary */ 175 #else 176 #define CTX_FPREGS_END U(0x210) /* Align to the next 16 byte boundary */ 177 #endif 178 #else 179 #define CTX_FPREGS_END U(0) 180 #endif 181 182 /******************************************************************************* 183 * Registers related to CVE-2018-3639 184 ******************************************************************************/ 185 #define CTX_CVE_2018_3639_OFFSET (CTX_FPREGS_OFFSET + CTX_FPREGS_END) 186 #define CTX_CVE_2018_3639_DISABLE U(0) 187 #define CTX_CVE_2018_3639_END U(0x10) /* Align to the next 16 byte boundary */ 188 189 /******************************************************************************* 190 * Registers related to ARMv8.3-PAuth. 191 ******************************************************************************/ 192 #define CTX_PAUTH_REGS_OFFSET (CTX_CVE_2018_3639_OFFSET + CTX_CVE_2018_3639_END) 193 #if CTX_INCLUDE_PAUTH_REGS 194 #define CTX_PACIAKEY_LO U(0x0) 195 #define CTX_PACIAKEY_HI U(0x8) 196 #define CTX_PACIBKEY_LO U(0x10) 197 #define CTX_PACIBKEY_HI U(0x18) 198 #define CTX_PACDAKEY_LO U(0x20) 199 #define CTX_PACDAKEY_HI U(0x28) 200 #define CTX_PACDBKEY_LO U(0x30) 201 #define CTX_PACDBKEY_HI U(0x38) 202 #define CTX_PACGAKEY_LO U(0x40) 203 #define CTX_PACGAKEY_HI U(0x48) 204 #define CTX_PACGAKEY_END U(0x50) 205 #define CTX_PAUTH_REGS_END U(0x60) /* Align to the next 16 byte boundary */ 206 #else 207 #define CTX_PAUTH_REGS_END U(0) 208 #endif /* CTX_INCLUDE_PAUTH_REGS */ 209 210 #ifndef __ASSEMBLY__ 211 212 #include <stdint.h> 213 214 #include <lib/cassert.h> 215 216 /* 217 * Common constants to help define the 'cpu_context' structure and its 218 * members below. 219 */ 220 #define DWORD_SHIFT U(3) 221 #define DEFINE_REG_STRUCT(name, num_regs) \ 222 typedef struct name { \ 223 uint64_t _regs[num_regs]; \ 224 } __aligned(16) name##_t 225 226 /* Constants to determine the size of individual context structures */ 227 #define CTX_GPREG_ALL (CTX_GPREGS_END >> DWORD_SHIFT) 228 #define CTX_SYSREG_ALL (CTX_SYSREGS_END >> DWORD_SHIFT) 229 #if CTX_INCLUDE_FPREGS 230 # define CTX_FPREG_ALL (CTX_FPREGS_END >> DWORD_SHIFT) 231 #endif 232 #define CTX_EL3STATE_ALL (CTX_EL3STATE_END >> DWORD_SHIFT) 233 #define CTX_CVE_2018_3639_ALL (CTX_CVE_2018_3639_END >> DWORD_SHIFT) 234 #if CTX_INCLUDE_PAUTH_REGS 235 # define CTX_PAUTH_REGS_ALL (CTX_PAUTH_REGS_END >> DWORD_SHIFT) 236 #endif 237 238 /* 239 * AArch64 general purpose register context structure. Usually x0-x18, 240 * lr are saved as the compiler is expected to preserve the remaining 241 * callee saved registers if used by the C runtime and the assembler 242 * does not touch the remaining. But in case of world switch during 243 * exception handling, we need to save the callee registers too. 244 */ 245 DEFINE_REG_STRUCT(gp_regs, CTX_GPREG_ALL); 246 247 /* 248 * AArch64 EL1 system register context structure for preserving the 249 * architectural state during switches from one security state to 250 * another in EL1. 251 */ 252 DEFINE_REG_STRUCT(el1_sys_regs, CTX_SYSREG_ALL); 253 254 /* 255 * AArch64 floating point register context structure for preserving 256 * the floating point state during switches from one security state to 257 * another. 258 */ 259 #if CTX_INCLUDE_FPREGS 260 DEFINE_REG_STRUCT(fp_regs, CTX_FPREG_ALL); 261 #endif 262 263 /* 264 * Miscellaneous registers used by EL3 firmware to maintain its state 265 * across exception entries and exits 266 */ 267 DEFINE_REG_STRUCT(el3_state, CTX_EL3STATE_ALL); 268 269 /* Function pointer used by CVE-2018-3639 dynamic mitigation */ 270 DEFINE_REG_STRUCT(cve_2018_3639, CTX_CVE_2018_3639_ALL); 271 272 /* Registers associated to ARMv8.3-PAuth */ 273 #if CTX_INCLUDE_PAUTH_REGS 274 DEFINE_REG_STRUCT(pauth, CTX_PAUTH_REGS_ALL); 275 #endif 276 277 /* 278 * Macros to access members of any of the above structures using their 279 * offsets 280 */ 281 #define read_ctx_reg(ctx, offset) ((ctx)->_regs[(offset) >> DWORD_SHIFT]) 282 #define write_ctx_reg(ctx, offset, val) (((ctx)->_regs[(offset) >> DWORD_SHIFT]) \ 283 = (uint64_t) (val)) 284 285 /* 286 * Top-level context structure which is used by EL3 firmware to 287 * preserve the state of a core at EL1 in one of the two security 288 * states and save enough EL3 meta data to be able to return to that 289 * EL and security state. The context management library will be used 290 * to ensure that SP_EL3 always points to an instance of this 291 * structure at exception entry and exit. Each instance will 292 * correspond to either the secure or the non-secure state. 293 */ 294 typedef struct cpu_context { 295 gp_regs_t gpregs_ctx; 296 el3_state_t el3state_ctx; 297 el1_sys_regs_t sysregs_ctx; 298 #if CTX_INCLUDE_FPREGS 299 fp_regs_t fpregs_ctx; 300 #endif 301 cve_2018_3639_t cve_2018_3639_ctx; 302 #if CTX_INCLUDE_PAUTH_REGS 303 pauth_t pauth_ctx; 304 #endif 305 } cpu_context_t; 306 307 /* Macros to access members of the 'cpu_context_t' structure */ 308 #define get_el3state_ctx(h) (&((cpu_context_t *) h)->el3state_ctx) 309 #if CTX_INCLUDE_FPREGS 310 # define get_fpregs_ctx(h) (&((cpu_context_t *) h)->fpregs_ctx) 311 #endif 312 #define get_sysregs_ctx(h) (&((cpu_context_t *) h)->sysregs_ctx) 313 #define get_gpregs_ctx(h) (&((cpu_context_t *) h)->gpregs_ctx) 314 #define get_cve_2018_3639_ctx(h) (&((cpu_context_t *) h)->cve_2018_3639_ctx) 315 #if CTX_INCLUDE_PAUTH_REGS 316 # define get_pauth_ctx(h) (&((cpu_context_t *) h)->pauth_ctx) 317 #endif 318 319 /* 320 * Compile time assertions related to the 'cpu_context' structure to 321 * ensure that the assembler and the compiler view of the offsets of 322 * the structure members is the same. 323 */ 324 CASSERT(CTX_GPREGS_OFFSET == __builtin_offsetof(cpu_context_t, gpregs_ctx), \ 325 assert_core_context_gp_offset_mismatch); 326 CASSERT(CTX_SYSREGS_OFFSET == __builtin_offsetof(cpu_context_t, sysregs_ctx), \ 327 assert_core_context_sys_offset_mismatch); 328 #if CTX_INCLUDE_FPREGS 329 CASSERT(CTX_FPREGS_OFFSET == __builtin_offsetof(cpu_context_t, fpregs_ctx), \ 330 assert_core_context_fp_offset_mismatch); 331 #endif 332 CASSERT(CTX_EL3STATE_OFFSET == __builtin_offsetof(cpu_context_t, el3state_ctx), \ 333 assert_core_context_el3state_offset_mismatch); 334 CASSERT(CTX_CVE_2018_3639_OFFSET == __builtin_offsetof(cpu_context_t, cve_2018_3639_ctx), \ 335 assert_core_context_cve_2018_3639_offset_mismatch); 336 #if CTX_INCLUDE_PAUTH_REGS 337 CASSERT(CTX_PAUTH_REGS_OFFSET == __builtin_offsetof(cpu_context_t, pauth_ctx), \ 338 assert_core_context_pauth_offset_mismatch); 339 #endif 340 341 /* 342 * Helper macro to set the general purpose registers that correspond to 343 * parameters in an aapcs_64 call i.e. x0-x7 344 */ 345 #define set_aapcs_args0(ctx, x0) do { \ 346 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X0, x0); \ 347 } while (0) 348 #define set_aapcs_args1(ctx, x0, x1) do { \ 349 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X1, x1); \ 350 set_aapcs_args0(ctx, x0); \ 351 } while (0) 352 #define set_aapcs_args2(ctx, x0, x1, x2) do { \ 353 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X2, x2); \ 354 set_aapcs_args1(ctx, x0, x1); \ 355 } while (0) 356 #define set_aapcs_args3(ctx, x0, x1, x2, x3) do { \ 357 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X3, x3); \ 358 set_aapcs_args2(ctx, x0, x1, x2); \ 359 } while (0) 360 #define set_aapcs_args4(ctx, x0, x1, x2, x3, x4) do { \ 361 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X4, x4); \ 362 set_aapcs_args3(ctx, x0, x1, x2, x3); \ 363 } while (0) 364 #define set_aapcs_args5(ctx, x0, x1, x2, x3, x4, x5) do { \ 365 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X5, x5); \ 366 set_aapcs_args4(ctx, x0, x1, x2, x3, x4); \ 367 } while (0) 368 #define set_aapcs_args6(ctx, x0, x1, x2, x3, x4, x5, x6) do { \ 369 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X6, x6); \ 370 set_aapcs_args5(ctx, x0, x1, x2, x3, x4, x5); \ 371 } while (0) 372 #define set_aapcs_args7(ctx, x0, x1, x2, x3, x4, x5, x6, x7) do { \ 373 write_ctx_reg(get_gpregs_ctx(ctx), CTX_GPREG_X7, x7); \ 374 set_aapcs_args6(ctx, x0, x1, x2, x3, x4, x5, x6); \ 375 } while (0) 376 377 /******************************************************************************* 378 * Function prototypes 379 ******************************************************************************/ 380 void el1_sysregs_context_save(el1_sys_regs_t *regs); 381 void el1_sysregs_context_restore(el1_sys_regs_t *regs); 382 #if CTX_INCLUDE_FPREGS 383 void fpregs_context_save(fp_regs_t *regs); 384 void fpregs_context_restore(fp_regs_t *regs); 385 #endif 386 387 #endif /* __ASSEMBLY__ */ 388 389 #endif /* CONTEXT_H */ 390