user/libc: setjmp

2 files changed, 42 insertions, 10 deletions

diff --git a/src/user/lib/include/setjmp.h b/src/user/lib/include/setjmp.h
index 51c7fd2..298939c 100644
--- a/src/user/lib/include/setjmp.h
+++ b/src/user/lib/include/setjmp.h
@@ -1,24 +1,17 @@
 #pragma once
 #include <user/lib/panic.h>
 
-typedef char jmp_buf[1];
+typedef uint64_t jmp_buf[8]; /* rbx, rsp, rbp, r12, r13, r14, r15, rip */
 typedef char sigjmp_buf[1];
 
-static inline int setjmp(jmp_buf env) {
-	(void)env;
-	return 0;
-}
+int setjmp(jmp_buf env);
+_Noreturn void longjmp(jmp_buf env, int val);
 
 static inline int sigsetjmp(sigjmp_buf env, int savemask) {
 	(void)env; (void)savemask;
 	return 0;
 }
 
-static inline _Noreturn void longjmp(jmp_buf env, int val) {
-	(void)env; (void)val;
-	__libc_panic("unimplemented");
-}
-
 static inline _Noreturn void siglongjmp(sigjmp_buf env, int val) {
 	(void)env; (void)val;
 	__libc_panic("unimplemented");
diff --git a/src/user/lib/setjmp.s b/src/user/lib/setjmp.s
new file mode 100644
index 0000000..29292da
--- /dev/null
+++ b/src/user/lib/setjmp.s
@@ -0,0 +1,39 @@
+.section .text
+.global setjmp
+.type setjmp, @function
+// int setjmp(jmp_buf env);
+setjmp:
+	mov  %rbx,  0(%rdi)
+	mov  %rbp, 16(%rdi)
+	mov  %r12, 24(%rdi)
+	mov  %r13, 32(%rdi)
+	mov  %r14, 40(%rdi)
+	mov  %r15, 48(%rdi)
+	/* save registers as if after a ret */
+	lea 8(%rsp), %rax
+	mov %rax, 8(%rdi)
+	mov (%rsp), %rax
+	mov %rax, 56(%rdi)
+	xor %rax, %rax
+
+	mov   8(%rdi), %rsp
+	jmp *56(%rdi)
+
+	ret
+
+
+.global longjmp
+.type longjmp, @function
+// _Noreturn void longjmp(jmp_buf env, int val);
+longjmp:
+	mov %rsi, %rax
+	cmp $1, %rax /* carry set as for %rax - 1 - so, 1 only if %rax == 0 */
+	adc $0, %rax
+1:	mov   0(%rdi), %rbx
+	mov   8(%rdi), %rsp
+	mov  16(%rdi), %rbp
+	mov  24(%rdi), %r12
+	mov  32(%rdi), %r13
+	mov  40(%rdi), %r14
+	mov  48(%rdi), %r15
+	jmp *56(%rdi)