Fix virtio net handling of the header.
[akaros.git] / user / parlib / signal.c
index 7d3553d..12b1d49 100644 (file)
@@ -83,10 +83,21 @@ static void handle_event(struct event_msg *ev_msg, unsigned int ev_type,
        int sig_nr;
        struct siginfo info = {0};
        info.si_code = SI_USER;
+       struct user_context fake_uctx;
 
        assert(ev_msg);
        sig_nr = ev_msg->ev_arg1;
-       trigger_posix_signal(sig_nr, &info, 0);
+       /* We're handling a process-wide signal, but signal handlers will want a
+        * user context.  They operate on the model that some thread got the signal,
+        * but that didn't happen on Akaros.  If we happen to have a current
+        * uthread, we can use that - perhaps that's what the user wants.  If not,
+        * we'll build a fake one representing our current call stack. */
+       if (current_uthread) {
+               trigger_posix_signal(sig_nr, &info, get_cur_uth_ctx());
+       } else {
+               init_user_ctx(&fake_uctx, (uintptr_t)handle_event, get_stack_pointer());
+               trigger_posix_signal(sig_nr, &info, &fake_uctx);
+       }
 }
 
 /* Called from uthread_slim_init() */
@@ -123,6 +134,11 @@ static bool stack_ptr_is_sane(uintptr_t sp)
        return TRUE;
 }
 
+static bool uth_is_handling_sigs(struct uthread *uth)
+{
+       return uth->sigstate.data ? TRUE : FALSE;
+}
+
 /* Prep a uthread to run a signal handler.  The original context of the uthread
  * is saved on its stack, and a new context is set up to run the signal handler
  * the next time the uthread is run. */
@@ -228,7 +244,7 @@ int uthread_signal(struct uthread *uthread, int signo)
  * handler is complete, the original context will be restored and restarted. */
 void uthread_prep_pending_signals(struct uthread *uthread)
 {
-       if (!uthread->sigstate.data && uthread->sigstate.pending) {
+       if (!uth_is_handling_sigs(uthread) && uthread->sigstate.pending) {
                sigset_t andset = uthread->sigstate.pending & (~uthread->sigstate.mask);
 
                if (!__sigisemptyset(&andset))
@@ -246,7 +262,7 @@ void uthread_prep_signal_from_fault(struct uthread *uthread,
        if (!__sigismember(&uthread->sigstate.mask, signo)) {
                struct siginfo info = {0};
 
-               if (uthread->sigstate.data) {
+               if (uth_is_handling_sigs(uthread)) {
                        printf("Uthread sighandler faulted, signal: %d\n", signo);
                        /* uthread.c already copied out the faulting ctx into the uth */
                        print_user_context(&uthread->u_ctx);
@@ -282,7 +298,20 @@ static int __sigpending(sigset_t *__set)
 static int __sigprocmask(int __how, __const sigset_t *__restrict __set,
                          sigset_t *__restrict __oset)
 {
-       sigset_t *sigmask = &current_uthread->sigstate.mask;
+       sigset_t *sigmask;
+
+       /* Signal handlers might call sigprocmask, with the intent of affecting the
+        * uthread's sigmask.  Process-wide signal handlers run on behalf of the
+        * entire process and aren't bound to a uthread, which means sigprocmask
+        * won't work.  We can tell we're running one of these handlers since we are
+        * in vcore context.  Uthread signals (e.g. pthread_kill()) run from uthread
+        * context. */
+       if (in_vcore_context()) {
+               errno = ENOENT;
+               return -1;
+       }
+
+       sigmask = &current_uthread->sigstate.mask;
 
        if (__set && (__how != SIG_BLOCK) &&
                     (__how != SIG_SETMASK) &&
@@ -358,7 +387,12 @@ static int __sigwaitinfo(__const sigset_t *__restrict __set,
 
 static int __sigself(int signo)
 {
-       int ret = uthread_signal(current_uthread, signo);
+       int ret;
+
+       if (in_vcore_context())
+               return kill(getpid(), signo);
+
+       ret = uthread_signal(current_uthread, signo);
 
        void cb(struct uthread *uthread, void *arg)
        {
@@ -366,4 +400,5 @@ static int __sigself(int signo)
        }
        if (ret == 0)
                uthread_yield(TRUE, cb, 0);
+       return ret;
 }