Fix: custom signal handler not working

kernel/include/process.h

@@ -47,7 +47,7 @@ typedef struct process {
   int32_t current_signal;
   signal_handler_t sighandlers[SIGNAL_NUM];
 
-  trapframe_t *regs;
+  trapframe_t *regs, *retregs;
 } process_t;
 
 // thread_func_t, cast from

kernel/include/signal.h

@@ -9,3 +9,4 @@
 typedef void (*signal_handler_t)();
 
 void run_signal(int32_t SIGNAL);
+void end_signal(void);

kernel/lib/process.c

@@ -108,6 +108,8 @@ process_t *fork_process(const process_t *from)
   process_t *ret = _make_process();
   memcpy(ret->stack, from->stack, from->stack_size);
   memcpy(ret->mem, from->mem, from->mem_size);
+  memcpy(ret->sigstack, from->sigstack, from->sigstack_size);
+  memcpy(ret->sighandlers, from->sighandlers, sizeof(from->sighandlers));
   *ret->regs = (trapframe_t){
     .x0 = 0, // fork return value
     .fp = from->regs->fp - (uint64_t)from->stack + (uint64_t)ret->stack,

kernel/lib/signal.c

@@ -27,11 +27,9 @@ void _sigkill_default_handler()
 
 void run_signal(int32_t SIGNAL)
 {
-  LOG("run_signal"); DEBUG_SIGNAL(SIGNAL);
   process_t *process = get_current()->process;
   process->current_signal = SIGNAL;
-  trapframe_t *regs = kmalloc(sizeof(trapframe_t));
-  memcpy(regs, process->regs, sizeof(trapframe_t));
+  LOG("run_signal"); LOG(process->sighandlers[SIGNAL]); DEBUG_SIGNAL(SIGNAL);
 
   if (!process->sighandlers[SIGNAL]) {
     switch (SIGNAL) {
@@ -40,20 +38,39 @@ void run_signal(int32_t SIGNAL)
       break;
     default:
       LOG("No default handler for this signal"); DEBUG_SIGNAL(SIGNAL);
-      return;
     }
+    return;
   }
 
-  *regs = (trapframe_t){
+  DEBUG_SIGNAL(process->sighandlers[SIGNAL]);
+
+  process->retregs = process->regs;
+  process->regs = kmalloc(sizeof(trapframe_t));
+  memcpy(process->regs, process->retregs, sizeof(trapframe_t));
+  *process->regs = (trapframe_t){
     .x0 = (uint64_t)process->sighandlers[SIGNAL],
     .elr_el1 = (uint64_t)_sighandler_wrapper,
     .sp_el0 = ((uint64_t)process->sigstack + process->sigstack_size - 64),
   };
 
   disable_interrupt();
-  process->current_signal = 0x0;
+  process->current_signal = SIGNAL;
   process->sigstate[SIGNAL] = false;
-  user_exec(regs);
+  user_exec(process->regs);
+
+  __builtin_unreachable();
+}
+
+void end_signal()
+{
+  DEBUG_SIGNAL("end_signal");
+  process_t *process = get_current()->process;
+
+  disable_interrupt();
+  process->current_signal = 0x0;
+  kfree(process->regs);
+  process->regs = process->retregs;
+  user_exec(process->regs);
 
   __builtin_unreachable();
 }

kernel/lib/syscall.c

@@ -5,6 +5,7 @@
 #include <process.h>
 #include <mbox.h>
 #include <exec.h>
+#include <signal.h>
 
 void syscall_handler(trapframe_t *regs)
 {
@@ -110,8 +111,4 @@ void sys_kill(int32_t pid, int32_t SIGNAL)
 }
 
 void sys_sigreturn(void)
-{
-  DEBUG_SIGNAL("sigreturn");
-  thread_t *th = get_current();
-  user_exec(th->process->regs);
-}
+{ end_signal(); }

kernel/main.c

@@ -42,8 +42,11 @@ void _init(void *dtb)
 
 void main(void *dtb)
 {
+  INFOR("kernel running");
+
   _init(dtb);
 
+  INFOR("Run user process syscall.img");
   run_thread(run_process_by_name, (uint64_t)&"syscall.img");
 
   schedule();