/* * Copyright (c) 2006-2021, RT-Thread Development Team * * SPDX-License-Identifier: Apache-2.0 * * Change Logs: * Date Author Notes * 2021-02-02 lizhirui first version * 2021-02-11 lizhirui fixed gp save/store bug * 2021-11-18 JasonHu add fpu registers save/restore */ #ifndef __STACKFRAME_H__ #define __STACKFRAME_H__ #define BYTES(idx) ((idx) * REGBYTES) #define FRAME_OFF_SSTATUS BYTES(2) #define FRAME_OFF_SP BYTES(32) #define FRAME_OFF_GP BYTES(3) #include "cpuport.h" #include "encoding.h" #ifdef ARCH_RISCV_FPU #define FPU_CTX_F0_OFF 0 /* offsetof(fpu_context_t, fpustatus.f[0]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F1_OFF 8 /* offsetof(fpu_context_t, fpustatus.f[1]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F2_OFF 16 /* offsetof(fpu_context_t, fpustatus.f[2]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F3_OFF 24 /* offsetof(fpu_context_t, fpustatus.f[3]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F4_OFF 32 /* offsetof(fpu_context_t, fpustatus.f[4]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F5_OFF 40 /* offsetof(fpu_context_t, fpustatus.f[5]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F6_OFF 48 /* offsetof(fpu_context_t, fpustatus.f[6]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F7_OFF 56 /* offsetof(fpu_context_t, fpustatus.f[7]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F8_OFF 64 /* offsetof(fpu_context_t, fpustatus.f[8]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F9_OFF 72 /* offsetof(fpu_context_t, fpustatus.f[9]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F10_OFF 80 /* offsetof(fpu_context_t, fpustatus.f[10]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F11_OFF 88 /* offsetof(fpu_context_t, fpustatus.f[11]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F12_OFF 96 /* offsetof(fpu_context_t, fpustatus.f[12]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F13_OFF 104 /* offsetof(fpu_context_t, fpustatus.f[13]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F14_OFF 112 /* offsetof(fpu_context_t, fpustatus.f[14]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F15_OFF 120 /* offsetof(fpu_context_t, fpustatus.f[15]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F16_OFF 128 /* offsetof(fpu_context_t, fpustatus.f[16]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F17_OFF 136 /* offsetof(fpu_context_t, fpustatus.f[17]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F18_OFF 144 /* offsetof(fpu_context_t, fpustatus.f[18]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F19_OFF 152 /* offsetof(fpu_context_t, fpustatus.f[19]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F20_OFF 160 /* offsetof(fpu_context_t, fpustatus.f[20]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F21_OFF 168 /* offsetof(fpu_context_t, fpustatus.f[21]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F22_OFF 176 /* offsetof(fpu_context_t, fpustatus.f[22]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F23_OFF 184 /* offsetof(fpu_context_t, fpustatus.f[23]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F24_OFF 192 /* offsetof(fpu_context_t, fpustatus.f[24]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F25_OFF 200 /* offsetof(fpu_context_t, fpustatus.f[25]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F26_OFF 208 /* offsetof(fpu_context_t, fpustatus.f[26]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F27_OFF 216 /* offsetof(fpu_context_t, fpustatus.f[27]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F28_OFF 224 /* offsetof(fpu_context_t, fpustatus.f[28]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F29_OFF 232 /* offsetof(fpu_context_t, fpustatus.f[29]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F30_OFF 240 /* offsetof(fpu_context_t, fpustatus.f[30]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #define FPU_CTX_F31_OFF 248 /* offsetof(fpu_context_t, fpustatus.f[31]) - offsetof(fpu_context_t, fpustatus.f[0]) */ #endif /* ARCH_RISCV_FPU */ /** * The register `tp` always save/restore when context switch, * we call `lwp_user_setting_save` when syscall enter, * call `lwp_user_setting_restore` when syscall exit * and modify context stack after `lwp_user_setting_restore` called * so that the `tp` can be the correct thread area value. */ .macro SAVE_ALL #ifdef ARCH_RISCV_FPU /* reserve float registers */ addi sp, sp, -CTX_FPU_REG_NR * REGBYTES #endif /* ARCH_RISCV_FPU */ /* save general registers */ addi sp, sp, -CTX_GENERAL_REG_NR * REGBYTES STORE x1, 1 * REGBYTES(sp) csrr x1, sstatus STORE x1, 2 * REGBYTES(sp) csrr x1, sepc STORE x1, 0 * REGBYTES(sp) STORE x3, 3 * REGBYTES(sp) STORE x4, 4 * REGBYTES(sp) /* save tp */ STORE x5, 5 * REGBYTES(sp) STORE x6, 6 * REGBYTES(sp) STORE x7, 7 * REGBYTES(sp) STORE x8, 8 * REGBYTES(sp) STORE x9, 9 * REGBYTES(sp) STORE x10, 10 * REGBYTES(sp) STORE x11, 11 * REGBYTES(sp) STORE x12, 12 * REGBYTES(sp) STORE x13, 13 * REGBYTES(sp) STORE x14, 14 * REGBYTES(sp) STORE x15, 15 * REGBYTES(sp) STORE x16, 16 * REGBYTES(sp) STORE x17, 17 * REGBYTES(sp) STORE x18, 18 * REGBYTES(sp) STORE x19, 19 * REGBYTES(sp) STORE x20, 20 * REGBYTES(sp) STORE x21, 21 * REGBYTES(sp) STORE x22, 22 * REGBYTES(sp) STORE x23, 23 * REGBYTES(sp) STORE x24, 24 * REGBYTES(sp) STORE x25, 25 * REGBYTES(sp) STORE x26, 26 * REGBYTES(sp) STORE x27, 27 * REGBYTES(sp) STORE x28, 28 * REGBYTES(sp) STORE x29, 29 * REGBYTES(sp) STORE x30, 30 * REGBYTES(sp) STORE x31, 31 * REGBYTES(sp) csrr t0, sscratch STORE t0, 32 * REGBYTES(sp) #ifdef ARCH_RISCV_FPU /* backup sp and adjust sp to save float registers */ mv t1, sp addi t1, t1, CTX_GENERAL_REG_NR * REGBYTES li t0, SSTATUS_FS csrs sstatus, t0 fsd f0, FPU_CTX_F0_OFF(t1) fsd f1, FPU_CTX_F1_OFF(t1) fsd f2, FPU_CTX_F2_OFF(t1) fsd f3, FPU_CTX_F3_OFF(t1) fsd f4, FPU_CTX_F4_OFF(t1) fsd f5, FPU_CTX_F5_OFF(t1) fsd f6, FPU_CTX_F6_OFF(t1) fsd f7, FPU_CTX_F7_OFF(t1) fsd f8, FPU_CTX_F8_OFF(t1) fsd f9, FPU_CTX_F9_OFF(t1) fsd f10, FPU_CTX_F10_OFF(t1) fsd f11, FPU_CTX_F11_OFF(t1) fsd f12, FPU_CTX_F12_OFF(t1) fsd f13, FPU_CTX_F13_OFF(t1) fsd f14, FPU_CTX_F14_OFF(t1) fsd f15, FPU_CTX_F15_OFF(t1) fsd f16, FPU_CTX_F16_OFF(t1) fsd f17, FPU_CTX_F17_OFF(t1) fsd f18, FPU_CTX_F18_OFF(t1) fsd f19, FPU_CTX_F19_OFF(t1) fsd f20, FPU_CTX_F20_OFF(t1) fsd f21, FPU_CTX_F21_OFF(t1) fsd f22, FPU_CTX_F22_OFF(t1) fsd f23, FPU_CTX_F23_OFF(t1) fsd f24, FPU_CTX_F24_OFF(t1) fsd f25, FPU_CTX_F25_OFF(t1) fsd f26, FPU_CTX_F26_OFF(t1) fsd f27, FPU_CTX_F27_OFF(t1) fsd f28, FPU_CTX_F28_OFF(t1) fsd f29, FPU_CTX_F29_OFF(t1) fsd f30, FPU_CTX_F30_OFF(t1) fsd f31, FPU_CTX_F31_OFF(t1) /* clr FS domain */ csrc sstatus, t0 /* clean status would clr sr_sd; */ li t0, SSTATUS_FS_CLEAN csrs sstatus, t0 #endif /* ARCH_RISCV_FPU */ .endm .macro RESTORE_ALL #ifdef ARCH_RISCV_FPU /* restore float register */ mv t2, sp addi t2, t2, CTX_GENERAL_REG_NR * REGBYTES /* skip all normal reg */ li t0, SSTATUS_FS csrs sstatus, t0 fld f0, FPU_CTX_F0_OFF(t2) fld f1, FPU_CTX_F1_OFF(t2) fld f2, FPU_CTX_F2_OFF(t2) fld f3, FPU_CTX_F3_OFF(t2) fld f4, FPU_CTX_F4_OFF(t2) fld f5, FPU_CTX_F5_OFF(t2) fld f6, FPU_CTX_F6_OFF(t2) fld f7, FPU_CTX_F7_OFF(t2) fld f8, FPU_CTX_F8_OFF(t2) fld f9, FPU_CTX_F9_OFF(t2) fld f10, FPU_CTX_F10_OFF(t2) fld f11, FPU_CTX_F11_OFF(t2) fld f12, FPU_CTX_F12_OFF(t2) fld f13, FPU_CTX_F13_OFF(t2) fld f14, FPU_CTX_F14_OFF(t2) fld f15, FPU_CTX_F15_OFF(t2) fld f16, FPU_CTX_F16_OFF(t2) fld f17, FPU_CTX_F17_OFF(t2) fld f18, FPU_CTX_F18_OFF(t2) fld f19, FPU_CTX_F19_OFF(t2) fld f20, FPU_CTX_F20_OFF(t2) fld f21, FPU_CTX_F21_OFF(t2) fld f22, FPU_CTX_F22_OFF(t2) fld f23, FPU_CTX_F23_OFF(t2) fld f24, FPU_CTX_F24_OFF(t2) fld f25, FPU_CTX_F25_OFF(t2) fld f26, FPU_CTX_F26_OFF(t2) fld f27, FPU_CTX_F27_OFF(t2) fld f28, FPU_CTX_F28_OFF(t2) fld f29, FPU_CTX_F29_OFF(t2) fld f30, FPU_CTX_F30_OFF(t2) fld f31, FPU_CTX_F31_OFF(t2) /* clr FS domain */ csrc sstatus, t0 /* clean status would clr sr_sd; */ li t0, SSTATUS_FS_CLEAN csrs sstatus, t0 #endif /* ARCH_RISCV_FPU */ /* restore general register */ /* resw ra to sepc */ LOAD x1, 0 * REGBYTES(sp) csrw sepc, x1 LOAD x1, 2 * REGBYTES(sp) csrw sstatus, x1 LOAD x1, 1 * REGBYTES(sp) LOAD x3, 3 * REGBYTES(sp) LOAD x4, 4 * REGBYTES(sp) /* restore tp */ LOAD x5, 5 * REGBYTES(sp) LOAD x6, 6 * REGBYTES(sp) LOAD x7, 7 * REGBYTES(sp) LOAD x8, 8 * REGBYTES(sp) LOAD x9, 9 * REGBYTES(sp) LOAD x10, 10 * REGBYTES(sp) LOAD x11, 11 * REGBYTES(sp) LOAD x12, 12 * REGBYTES(sp) LOAD x13, 13 * REGBYTES(sp) LOAD x14, 14 * REGBYTES(sp) LOAD x15, 15 * REGBYTES(sp) LOAD x16, 16 * REGBYTES(sp) LOAD x17, 17 * REGBYTES(sp) LOAD x18, 18 * REGBYTES(sp) LOAD x19, 19 * REGBYTES(sp) LOAD x20, 20 * REGBYTES(sp) LOAD x21, 21 * REGBYTES(sp) LOAD x22, 22 * REGBYTES(sp) LOAD x23, 23 * REGBYTES(sp) LOAD x24, 24 * REGBYTES(sp) LOAD x25, 25 * REGBYTES(sp) LOAD x26, 26 * REGBYTES(sp) LOAD x27, 27 * REGBYTES(sp) LOAD x28, 28 * REGBYTES(sp) LOAD x29, 29 * REGBYTES(sp) LOAD x30, 30 * REGBYTES(sp) LOAD x31, 31 * REGBYTES(sp) /* restore user sp */ LOAD sp, 32 * REGBYTES(sp) .endm .macro RESTORE_SYS_GP .option push .option norelax la gp, __global_pointer$ .option pop .endm .macro OPEN_INTERRUPT csrsi sstatus, 2 .endm .macro CLOSE_INTERRUPT csrci sstatus, 2 .endm #endif