riscv: Implement setjmp/longjmp
Being able to do setjmp and longjmp can be quite useful for tests.
Implement the functions for riscv.
Signed-off-by: Andrew Jones <andrew.jones@linux.dev>
diff --git a/lib/riscv/setjmp.S b/lib/riscv/setjmp.S
new file mode 100644
index 0000000..38b0f1c
--- /dev/null
+++ b/lib/riscv/setjmp.S
@@ -0,0 +1,50 @@
+/* SPDX-License-Identifier: GPL-2.0 */
+#define __ASSEMBLY__
+#include <asm/asm.h>
+
+.section .text
+
+.balign 4
+.global setjmp
+setjmp:
+ REG_S ra, (0 * SZREG)(a0)
+ REG_S s0, (1 * SZREG)(a0)
+ REG_S s1, (2 * SZREG)(a0)
+ REG_S s2, (3 * SZREG)(a0)
+ REG_S s3, (4 * SZREG)(a0)
+ REG_S s4, (5 * SZREG)(a0)
+ REG_S s5, (6 * SZREG)(a0)
+ REG_S s6, (7 * SZREG)(a0)
+ REG_S s7, (8 * SZREG)(a0)
+ REG_S s8, (9 * SZREG)(a0)
+ REG_S s9, (10 * SZREG)(a0)
+ REG_S s10, (11 * SZREG)(a0)
+ REG_S s11, (12 * SZREG)(a0)
+ REG_S sp, (13 * SZREG)(a0)
+ REG_S gp, (14 * SZREG)(a0)
+ REG_S tp, (15 * SZREG)(a0)
+ li a0, 0
+ ret
+
+.balign 4
+.global longjmp
+longjmp:
+ REG_L ra, (0 * SZREG)(a0)
+ REG_L s0, (1 * SZREG)(a0)
+ REG_L s1, (2 * SZREG)(a0)
+ REG_L s2, (3 * SZREG)(a0)
+ REG_L s3, (4 * SZREG)(a0)
+ REG_L s4, (5 * SZREG)(a0)
+ REG_L s5, (6 * SZREG)(a0)
+ REG_L s6, (7 * SZREG)(a0)
+ REG_L s7, (8 * SZREG)(a0)
+ REG_L s8, (9 * SZREG)(a0)
+ REG_L s9, (10 * SZREG)(a0)
+ REG_L s10, (11 * SZREG)(a0)
+ REG_L s11, (12 * SZREG)(a0)
+ REG_L sp, (13 * SZREG)(a0)
+ REG_L gp, (14 * SZREG)(a0)
+ REG_L tp, (15 * SZREG)(a0)
+ seqz a0, a1
+ add a0, a0, a1
+ ret
diff --git a/lib/setjmp.h b/lib/setjmp.h
index 6afdf66..f878ad8 100644
--- a/lib/setjmp.h
+++ b/lib/setjmp.h
@@ -8,7 +8,11 @@
#define _LIBCFLAT_SETJMP_H_
typedef struct jmp_buf_tag {
+#if defined(__i386__) || defined(__x86_64__)
long int regs[8];
+#elif defined(__riscv)
+ long int regs[16];
+#endif
} jmp_buf[1];
extern int setjmp (struct jmp_buf_tag env[1]);
diff --git a/riscv/Makefile b/riscv/Makefile
index 734441f..28b0415 100644
--- a/riscv/Makefile
+++ b/riscv/Makefile
@@ -36,6 +36,7 @@
cflatobjs += lib/riscv/mmu.o
cflatobjs += lib/riscv/processor.o
cflatobjs += lib/riscv/sbi.o
+cflatobjs += lib/riscv/setjmp.o
cflatobjs += lib/riscv/setup.o
cflatobjs += lib/riscv/smp.o
cflatobjs += lib/riscv/stack.o
