From 2fa7b9cc34e62ce23d00a32af21e325b807f2d98 Mon Sep 17 00:00:00 2001 From: Yanfeng Liu Date: Tue, 16 Jul 2024 14:41:04 +0800 Subject: [PATCH] riscv/trap: fix sp restore logic This fixes stack pointer restore logic to avoid parent stack corruption by forked child in PROTECTED build. Signed-off-by: Yanfeng Liu --- arch/risc-v/src/common/riscv_exception_common.S | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/arch/risc-v/src/common/riscv_exception_common.S b/arch/risc-v/src/common/riscv_exception_common.S index 90359470314..7bcf2d2d72a 100644 --- a/arch/risc-v/src/common/riscv_exception_common.S +++ b/arch/risc-v/src/common/riscv_exception_common.S @@ -231,7 +231,11 @@ return_from_exception: load_ctx sp +#ifdef CONFIG_ARCH_KERNEL_STACK REGLOAD sp, REG_SP(sp) /* restore original sp */ +#else + addi sp, sp, XCPTCONTEXT_SIZE +#endif /* Return from exception */