1532ed618SSoby Mathew /* 283ec7e45SBoyan Karatotev * Copyright (c) 2013-2025, Arm Limited and Contributors. All rights reserved. 3532ed618SSoby Mathew * 482cb2c1aSdp-arm * SPDX-License-Identifier: BSD-3-Clause 5532ed618SSoby Mathew */ 6532ed618SSoby Mathew 7a0fee747SAntonio Nino Diaz #ifndef CONTEXT_MGMT_H 8a0fee747SAntonio Nino Diaz #define CONTEXT_MGMT_H 9532ed618SSoby Mathew 10b10d4499SJeenu Viswambharan #include <assert.h> 111634cae8SAntonio Nino Diaz #include <context.h> 12b10d4499SJeenu Viswambharan #include <stdint.h> 13532ed618SSoby Mathew 1409d40e0eSAntonio Nino Diaz #include <arch.h> 1509d40e0eSAntonio Nino Diaz 16532ed618SSoby Mathew /******************************************************************************* 17532ed618SSoby Mathew * Forward declarations 18532ed618SSoby Mathew ******************************************************************************/ 19532ed618SSoby Mathew struct entry_point_info; 20532ed618SSoby Mathew 21532ed618SSoby Mathew /******************************************************************************* 22532ed618SSoby Mathew * Function & variable prototypes 23532ed618SSoby Mathew ******************************************************************************/ 24532ed618SSoby Mathew void cm_init(void); 25532ed618SSoby Mathew void *cm_get_context_by_index(unsigned int cpu_idx, 26532ed618SSoby Mathew unsigned int security_state); 27532ed618SSoby Mathew void cm_set_context_by_index(unsigned int cpu_idx, 28532ed618SSoby Mathew void *context, 29532ed618SSoby Mathew unsigned int security_state); 30532ed618SSoby Mathew void *cm_get_context(uint32_t security_state); 31532ed618SSoby Mathew void cm_set_context(void *context, uint32_t security_state); 32532ed618SSoby Mathew void cm_init_my_context(const struct entry_point_info *ep); 3325d819a3SVarun Wadekar void cm_setup_context(cpu_context_t *ctx, const struct entry_point_info *ep); 34532ed618SSoby Mathew void cm_prepare_el3_exit(uint32_t security_state); 358b95e848SZelalem Aweke void cm_prepare_el3_exit_ns(void); 36e33b78a6SSoby Mathew 37402b3cf8SJulius Werner #ifdef __aarch64__ 3824a70738SBoyan Karatotev #if IMAGE_BL31 3983ec7e45SBoyan Karatotev void cm_manage_extensions_el3(unsigned int my_idx); 40*6eafc060SBoyan Karatotev void cm_manage_extensions_per_world(void); 4124a70738SBoyan Karatotev #endif 424087ed6cSJayanth Dodderi Chidanand 43a0674ab0SJayanth Dodderi Chidanand #if (CTX_INCLUDE_EL2_REGS && IMAGE_BL31) 4428f39f02SMax Shvetsov void cm_el2_sysregs_context_save(uint32_t security_state); 4528f39f02SMax Shvetsov void cm_el2_sysregs_context_restore(uint32_t security_state); 46a0674ab0SJayanth Dodderi Chidanand #else 47532ed618SSoby Mathew void cm_el1_sysregs_context_save(uint32_t security_state); 48532ed618SSoby Mathew void cm_el1_sysregs_context_restore(uint32_t security_state); 49a0674ab0SJayanth Dodderi Chidanand #endif /* (CTX_INCLUDE_EL2_REGS && IMAGE_BL31) */ 50a0674ab0SJayanth Dodderi Chidanand 51532ed618SSoby Mathew void cm_set_elr_el3(uint32_t security_state, uintptr_t entrypoint); 52532ed618SSoby Mathew void cm_set_elr_spsr_el3(uint32_t security_state, 53532ed618SSoby Mathew uintptr_t entrypoint, uint32_t spsr); 54532ed618SSoby Mathew void cm_write_scr_el3_bit(uint32_t security_state, 55532ed618SSoby Mathew uint32_t bit_pos, 56532ed618SSoby Mathew uint32_t value); 57532ed618SSoby Mathew void cm_set_next_eret_context(uint32_t security_state); 58f1be00daSLouis Mayencourt u_register_t cm_get_scr_el3(uint32_t security_state); 59532ed618SSoby Mathew 60532ed618SSoby Mathew /* Inline definitions */ 61532ed618SSoby Mathew 62532ed618SSoby Mathew /******************************************************************************* 63532ed618SSoby Mathew * This function is used to program the context that's used for exception 64532ed618SSoby Mathew * return. This initializes the SP_EL3 to a pointer to a 'cpu_context' set for 65532ed618SSoby Mathew * the required security state 66532ed618SSoby Mathew ******************************************************************************/ 67532ed618SSoby Mathew static inline void cm_set_next_context(void *context) 68532ed618SSoby Mathew { 69aa61368eSAntonio Nino Diaz #if ENABLE_ASSERTIONS 70532ed618SSoby Mathew uint64_t sp_mode; 71532ed618SSoby Mathew 72532ed618SSoby Mathew /* 73532ed618SSoby Mathew * Check that this function is called with SP_EL0 as the stack 74532ed618SSoby Mathew * pointer 75532ed618SSoby Mathew */ 76532ed618SSoby Mathew __asm__ volatile("mrs %0, SPSel\n" 77532ed618SSoby Mathew : "=r" (sp_mode)); 78532ed618SSoby Mathew 79532ed618SSoby Mathew assert(sp_mode == MODE_SP_EL0); 80aa61368eSAntonio Nino Diaz #endif /* ENABLE_ASSERTIONS */ 81532ed618SSoby Mathew 82532ed618SSoby Mathew __asm__ volatile("msr spsel, #1\n" 83532ed618SSoby Mathew "mov sp, %0\n" 84532ed618SSoby Mathew "msr spsel, #0\n" 85532ed618SSoby Mathew : : "r" (context)); 86532ed618SSoby Mathew } 87f3b4914bSYatharth Kochar 88f3b4914bSYatharth Kochar #else 89f3b4914bSYatharth Kochar void *cm_get_next_context(void); 902ed7b71eSEtienne Carriere void cm_set_next_context(void *context); 9183ec7e45SBoyan Karatotev static inline void cm_manage_extensions_el3(unsigned int cpu_idx) {} 92402b3cf8SJulius Werner #endif /* __aarch64__ */ 93f3b4914bSYatharth Kochar 94a0fee747SAntonio Nino Diaz #endif /* CONTEXT_MGMT_H */ 95