1 /* 2 * Copyright (c) 2013-2025, Arm Limited and Contributors. All rights reserved. 3 * 4 * SPDX-License-Identifier: BSD-3-Clause 5 */ 6 7 #ifndef CONTEXT_MGMT_H 8 #define CONTEXT_MGMT_H 9 10 #include <assert.h> 11 #include <context.h> 12 #include <stdint.h> 13 14 #include <arch.h> 15 16 /******************************************************************************* 17 * Forward declarations 18 ******************************************************************************/ 19 struct entry_point_info; 20 21 /******************************************************************************* 22 * Function & variable prototypes 23 ******************************************************************************/ 24 void cm_init(void); 25 void *cm_get_context_by_index(unsigned int cpu_idx, 26 unsigned int security_state); 27 void cm_set_context_by_index(unsigned int cpu_idx, 28 void *context, 29 unsigned int security_state); 30 void *cm_get_context(uint32_t security_state); 31 void cm_set_context(void *context, uint32_t security_state); 32 void cm_init_my_context(const struct entry_point_info *ep); 33 void cm_setup_context(cpu_context_t *ctx, const struct entry_point_info *ep); 34 void cm_prepare_el3_exit(uint32_t security_state); 35 void cm_prepare_el3_exit_ns(void); 36 37 #if !IMAGE_BL1 38 void cm_init_context_by_index(unsigned int cpu_idx, 39 const struct entry_point_info *ep); 40 #endif /* !IMAGE_BL1 */ 41 42 #ifdef __aarch64__ 43 #if IMAGE_BL31 44 void cm_manage_extensions_el3(unsigned int my_idx); 45 void manage_extensions_nonsecure_per_world(void); 46 void cm_el3_arch_init_per_world(per_world_context_t *per_world_ctx); 47 void cm_handle_asymmetric_features(void); 48 #endif 49 50 #if (CTX_INCLUDE_EL2_REGS && IMAGE_BL31) 51 void cm_el2_sysregs_context_save(uint32_t security_state); 52 void cm_el2_sysregs_context_restore(uint32_t security_state); 53 #else 54 void cm_el1_sysregs_context_save(uint32_t security_state); 55 void cm_el1_sysregs_context_restore(uint32_t security_state); 56 #endif /* (CTX_INCLUDE_EL2_REGS && IMAGE_BL31) */ 57 58 void cm_set_elr_el3(uint32_t security_state, uintptr_t entrypoint); 59 void cm_set_elr_spsr_el3(uint32_t security_state, 60 uintptr_t entrypoint, uint32_t spsr); 61 void cm_write_scr_el3_bit(uint32_t security_state, 62 uint32_t bit_pos, 63 uint32_t value); 64 void cm_set_next_eret_context(uint32_t security_state); 65 u_register_t cm_get_scr_el3(uint32_t security_state); 66 67 /* Inline definitions */ 68 69 /******************************************************************************* 70 * This function is used to program the context that's used for exception 71 * return. This initializes the SP_EL3 to a pointer to a 'cpu_context' set for 72 * the required security state 73 ******************************************************************************/ 74 static inline void cm_set_next_context(void *context) 75 { 76 #if ENABLE_ASSERTIONS 77 uint64_t sp_mode; 78 79 /* 80 * Check that this function is called with SP_EL0 as the stack 81 * pointer 82 */ 83 __asm__ volatile("mrs %0, SPSel\n" 84 : "=r" (sp_mode)); 85 86 assert(sp_mode == MODE_SP_EL0); 87 #endif /* ENABLE_ASSERTIONS */ 88 89 __asm__ volatile("msr spsel, #1\n" 90 "mov sp, %0\n" 91 "msr spsel, #0\n" 92 : : "r" (context)); 93 } 94 95 #else 96 void *cm_get_next_context(void); 97 void cm_set_next_context(void *context); 98 static inline void cm_manage_extensions_el3(unsigned int cpu_idx) {} 99 static inline void manage_extensions_nonsecure_per_world(void) {} 100 static inline void cm_handle_asymmetric_features(void) {} 101 #endif /* __aarch64__ */ 102 103 #endif /* CONTEXT_MGMT_H */ 104