Fix virtio net handling of the header.
[akaros.git] / user / parlib / signal.c
index df3c7b8..12b1d49 100644 (file)
@@ -1,5 +1,6 @@
 /* Copyright (c) 2013 The Regents of the University of California
  * Barret Rhoden <brho@cs.berkeley.edu>
+ * Kevin Klues <klueska@cs.berkeley.edu>
  * See LICENSE for details.
  *
  * POSIX signal handling glue.  All glibc programs link against parlib, so they
  *     only thing we do is allow the registration of signal handlers. 
  *     - Check each function for further notes.  */
 
-#include <signal.h>
+// Needed for sigmask functions...
+#define _GNU_SOURCE
+
 #include <stdio.h>
+#include <parlib/parlib.h>
+#include <parlib/signal.h>
+#include <parlib/uthread.h>
+#include <parlib/event.h>
+#include <errno.h>
+#include <parlib/assert.h>
+#include <ros/procinfo.h>
+#include <ros/syscall.h>
+#include <sys/mman.h>
+#include <parlib/stdio.h>
 
-#include <event.h>
-#include <assert.h>
+/* Forward declare our signal_ops functions. */
+static int __sigaltstack(__const struct sigaltstack *__restrict __ss,
+                         struct sigaltstack *__restrict __oss);
+static int __siginterrupt(int __sig, int __interrupt);
+static int __sigpending(sigset_t *__set);
+static int __sigprocmask(int __how, __const sigset_t *__restrict __set,
+                         sigset_t *__restrict __oset);
+static int __sigqueue(__pid_t __pid, int __sig, __const union sigval __val);
+static int __sigreturn(struct sigcontext *__scp);
+static int __sigstack(struct sigstack *__ss, struct sigstack *__oss);
+static int __sigsuspend(__const sigset_t *__set);
+static int __sigtimedwait(__const sigset_t *__restrict __set,
+                          siginfo_t *__restrict __info,
+                          __const struct timespec *__restrict __timeout);
+static int __sigwait(__const sigset_t *__restrict __set, int *__restrict __sig);
+static int __sigwaitinfo(__const sigset_t *__restrict __set,
+                         siginfo_t *__restrict __info);
+static int __sigself(int signo);
 
-struct sigaction sigactions[_NSIG - 1];
+/* The default definition of signal_ops (similar to sched_ops in uthread.c) */
+struct signal_ops default_signal_ops = {
+       .sigaltstack = __sigaltstack,
+       .siginterrupt = __siginterrupt,
+       .sigpending = __sigpending,
+       .sigprocmask = __sigprocmask,
+       .sigqueue = __sigqueue,
+       .sigreturn = __sigreturn,
+       .sigstack = __sigstack,
+       .sigsuspend = __sigsuspend,
+       .sigtimedwait = __sigtimedwait,
+       .sigwait = __sigwait,
+       .sigwaitinfo = __sigwaitinfo,
+       .sigself = __sigself
+};
 
-void handle_posix_signal(struct event_msg *ev_msg, unsigned int ev_type)
+/* This is the catch all akaros event->posix signal handler.  All posix signals
+ * are received in a single akaros event type.  They are then dispatched from
+ * this function to their proper posix signal handler */
+static void handle_event(struct event_msg *ev_msg, unsigned int ev_type,
+                         void *data)
 {
        int sig_nr;
-       struct sigaction *action;
        struct siginfo info = {0};
+       info.si_code = SI_USER;
+       struct user_context fake_uctx;
+
        assert(ev_msg);
        sig_nr = ev_msg->ev_arg1;
-       if (sig_nr > _NSIG - 1 || sig_nr < 0)
-               return;
-       action = &sigactions[sig_nr];
-       /* Would like a switch/case here, but they are pointers.  We can also get
-        * away with this check early since sa_handler and sa_sigaction are macros
-        * referencing the same union.  The man page isn't specific about whether or
-        * not you need to care about SA_SIGINFO when sending DFL/ERR/IGN. */
-       if (action->sa_handler == SIG_ERR)
-               return;
-       if (action->sa_handler == SIG_IGN)
-               return;
-       if (action->sa_handler == SIG_DFL)      /* aka, 0 */
-               return;
-       if (action->sa_flags & SA_SIGINFO) {
-               info.si_signo = sig_nr;
-               /* TODO: consider pid and whatnot */
-               action->sa_sigaction(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 {
-               action->sa_handler(sig_nr);
+               init_user_ctx(&fake_uctx, (uintptr_t)handle_event, get_stack_pointer());
+               trigger_posix_signal(sig_nr, &info, &fake_uctx);
        }
 }
 
@@ -64,121 +104,301 @@ void handle_posix_signal(struct event_msg *ev_msg, unsigned int ev_type)
 void init_posix_signals(void)
 {
        struct event_queue *posix_sig_ev_q;
-       ev_handlers[EV_POSIX_SIGNAL] = handle_posix_signal;
-       posix_sig_ev_q = get_big_event_q();
+
+       signal_ops = &default_signal_ops;
+       register_ev_handler(EV_POSIX_SIGNAL, handle_event, 0);
+       posix_sig_ev_q = get_eventq(EV_MBOX_UCQ);
        assert(posix_sig_ev_q);
-       posix_sig_ev_q->ev_flags = EVENT_IPI | EVENT_INDIR | EVENT_FALLBACK;
+       posix_sig_ev_q->ev_flags = EVENT_IPI | EVENT_INDIR | EVENT_SPAM_INDIR |
+                                  EVENT_WAKEUP;
        register_kevent_q(posix_sig_ev_q, EV_POSIX_SIGNAL);
 }
 
-/* Will need to do these if we have signal masks (sigprocmask style) */
-int sigaddset(sigset_t *__set, int __signo)
+/* Swap the contents of two user contexts (not just their pointers). */
+static void swap_user_contexts(struct user_context *c1, struct user_context *c2)
 {
-       return 0;
+       struct user_context temp_ctx;
+
+       temp_ctx = *c1;
+       *c1 = *c2;
+       *c2 = temp_ctx;
 }
 
-int sigdelset(sigset_t *__set, int __signo)
+/* Helper for checking a stack pointer.  It's possible the context we're
+ * injecting signals into is complete garbage, so using the SP is a little
+ * dangerous. */
+static bool stack_ptr_is_sane(uintptr_t sp)
 {
-       return 0;
+       if ((sp < PGSIZE) || (sp > ULIM))
+               return FALSE;
+       return TRUE;
 }
 
-int sigismember(__const sigset_t *__set, int __signo)
+static bool uth_is_handling_sigs(struct uthread *uth)
 {
-       return 0;
+       return uth->sigstate.data ? TRUE : FALSE;
 }
 
-/* Would need a layer/interposition to ignore blocked signals when they come in,
- * and then to manually play them when they are unblocked, like how x86 does
- * with the IRR and the ISR for interrupt delivery. */
-int sigprocmask(int __how, __const sigset_t *__restrict __set,
-                sigset_t *__restrict __oset)
+/* 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. */
+static void __prep_sighandler(struct uthread *uthread,
+                              void (*entry)(void),
+                              struct siginfo *info)
 {
-       return 0;
+       uintptr_t stack;
+       struct user_context *ctx;
+
+       if (uthread->flags & UTHREAD_SAVED) {
+               ctx = &uthread->u_ctx;
+               stack = get_user_ctx_stack(ctx) - sizeof(struct sigdata);
+               assert(stack_ptr_is_sane(stack));
+               uthread->sigstate.data = (struct sigdata*)stack;
+               if (uthread->flags & UTHREAD_FPSAVED) {
+                       uthread->sigstate.data->as = uthread->as;
+                       uthread->flags &= ~UTHREAD_FPSAVED;
+               }
+       } else {
+               assert(current_uthread == uthread);
+               ctx = &vcpd_of(vcore_id())->uthread_ctx;
+               stack = get_user_ctx_stack(ctx) - sizeof(struct sigdata);
+               assert(stack_ptr_is_sane(stack));
+               uthread->sigstate.data = (struct sigdata*)stack;
+               save_fp_state(&uthread->sigstate.data->as);
+       }
+       if (info != NULL)
+               uthread->sigstate.data->info = *info;
+
+       init_user_ctx(&uthread->sigstate.data->u_ctx, (uintptr_t)entry, stack);
+       swap_user_contexts(ctx, &uthread->sigstate.data->u_ctx);
 }
 
-/* Could do this with a loop on delivery of the signal, sleeping and getting
- * woken up by the kernel on any event, like we do with async syscalls. */
-int sigsuspend(__const sigset_t *__set)
+/* Restore the context saved as the result of running a signal handler on a
+ * uthread. This context will execute the next time the uthread is run. */
+static void __restore_after_sighandler(struct uthread *uthread)
 {
-       return 0;
+       uthread->u_ctx = uthread->sigstate.data->u_ctx;
+       uthread->flags |= UTHREAD_SAVED;
+       switch (uthread->u_ctx.type) {
+       case ROS_HW_CTX:
+       case ROS_VM_CTX:
+               uthread->as = uthread->sigstate.data->as;
+               uthread->flags |= UTHREAD_FPSAVED;
+               break;
+       }
+       uthread->sigstate.data = NULL;
 }
 
-int sigaction(int __sig, __const struct sigaction *__restrict __act,
-              struct sigaction *__restrict __oact)
+/* Callback when yielding a pthread after upon completion of a sighandler.  We
+ * didn't save the current context on yeild, but that's ok because here we
+ * restore the original saved context of the pthread and then treat this like a
+ * normal voluntary yield. */
+static void __exit_sighandler_cb(struct uthread *uthread, void *junk)
 {
-       if (__sig > _NSIG - 1 || __sig < 0)
-               return -1;
-       if (__oact) {
-               *__oact = sigactions[__sig];
+       __restore_after_sighandler(uthread);
+       uthread_paused(uthread);
+}
+
+/* Run a specific sighandler from the top of the sigstate stack. The 'info'
+ * struct is prepopulated before the call is triggered as the result of a
+ * reflected fault. */
+static void __run_sighandler(void)
+{
+       struct uthread *uthread = current_uthread;
+       int signo = uthread->sigstate.data->info.si_signo;
+
+       __sigdelset(&uthread->sigstate.pending, signo);
+       trigger_posix_signal(signo, &uthread->sigstate.data->info,
+                            &uthread->sigstate.data->u_ctx);
+       uthread_yield(FALSE, __exit_sighandler_cb, 0);
+}
+
+/* Run through all pending sighandlers and trigger them with a NULL info
+ * field. These handlers are triggered as the result of thread directed
+ * signals (i.e. not interprocess signals), and thus don't require individual
+ * 'info' structs. */
+static void __run_all_sighandlers(void)
+{
+       struct uthread *uthread = current_uthread;
+       sigset_t andset = uthread->sigstate.pending & (~uthread->sigstate.mask);
+
+       for (int i = 1; i < _NSIG; i++) {
+               if (__sigismember(&andset, i)) {
+                       __sigdelset(&uthread->sigstate.pending, i);
+                       trigger_posix_signal(i, NULL, &uthread->sigstate.data->u_ctx);
+               }
        }
-       if (!__act)
-               return 0;
-       sigactions[__sig] = *__act;
+       uthread_yield(FALSE, __exit_sighandler_cb, 0);
+}
+
+int uthread_signal(struct uthread *uthread, int signo)
+{
+       // Slightly racy with clearing of mask when triggering the signal, but
+       // that's OK, as signals are inherently racy since they don't queue up.
+       return sigaddset(&uthread->sigstate.pending, signo);
+}
+
+/* If there are any pending signals, prep the uthread to run it's signal
+ * handler. The next time the uthread is run, it will pop into it's signal
+ * handler context instead of its original saved context. Once the signal
+ * handler is complete, the original context will be restored and restarted. */
+void uthread_prep_pending_signals(struct uthread *uthread)
+{
+       if (!uth_is_handling_sigs(uthread) && uthread->sigstate.pending) {
+               sigset_t andset = uthread->sigstate.pending & (~uthread->sigstate.mask);
+
+               if (!__sigisemptyset(&andset))
+                       __prep_sighandler(uthread, __run_all_sighandlers, NULL);
+       }
+}
+
+/* If the given signal is unmasked, prep the uthread to run it's signal
+ * handler, but don't run it yet. In either case, make the uthread runnable
+ * again. Once the signal handler is complete, the original context will be
+ * restored and restarted. */
+void uthread_prep_signal_from_fault(struct uthread *uthread,
+                                    int signo, int code, void *addr)
+{
+       if (!__sigismember(&uthread->sigstate.mask, signo)) {
+               struct siginfo info = {0};
+
+               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);
+                       exit(-1);
+               }
+               info.si_signo = signo;
+               info.si_code = code;
+               info.si_addr = addr;
+               __prep_sighandler(uthread, __run_sighandler, &info);
+       }
+}
+
+/* This is managed by vcore / 2LS code */
+static int __sigaltstack(__const struct sigaltstack *__restrict __ss,
+                         struct sigaltstack *__restrict __oss)
+{
        return 0;
 }
 
-/* Not really possible or relevant - you'd need to walk/examine the event UCQ */
-int sigpending(sigset_t *__set)
+/* Akaros can't have signals interrupt syscalls to need a restart, though we can
+ * re-wake-up the process while it is waiting for its syscall. */
+static int __siginterrupt(int __sig, int __interrupt)
 {
        return 0;
 }
 
-/* Can be done similar to sigsuspend */
-int sigwait(__const sigset_t *__restrict __set, int *__restrict __sig)
+/* Not really possible or relevant - you'd need to walk/examine the event UCQ */
+static int __sigpending(sigset_t *__set)
 {
        return 0;
 }
 
-/* Can be done similar to sigsuspend */
-int sigwaitinfo(__const sigset_t *__restrict __set,
-                siginfo_t *__restrict __info)
+static int __sigprocmask(int __how, __const sigset_t *__restrict __set,
+                         sigset_t *__restrict __oset)
 {
+       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) &&
+                    (__how != SIG_UNBLOCK)) {
+               errno = EINVAL;
+               return -1;
+       }
+
+       if (__oset)
+               *__oset = *sigmask;
+       if (__set) {
+               switch (__how) {
+                       case SIG_BLOCK:
+                               *sigmask = *sigmask | *__set;
+                               break;
+                       case SIG_SETMASK:
+                               *sigmask = *__set;
+                               break;
+                       case SIG_UNBLOCK:
+                               *sigmask = *sigmask & ~(*__set);
+                               break;
+               }
+       }
        return 0;
 }
 
-/* Can be done similar to sigsuspend, with an extra alarm syscall */
-int sigtimedwait(__const sigset_t *__restrict __set,
-                 siginfo_t *__restrict __info,
-                 __const struct timespec *__restrict __timeout)
+/* Needs support with trigger_posix_signal to deal with passing values with
+ * POSIX signals. */
+static int __sigqueue(__pid_t __pid, int __sig, __const union sigval __val)
 {
        return 0;
 }
 
-/* Needs support with handle_posix_signal to deal with passing values with POSIX
- * signals. */
-int sigqueue(__pid_t __pid, int __sig, __const union sigval __val)
+/* Linux specific, and not really needed for us */
+static int __sigreturn(struct sigcontext *__scp)
 {
        return 0;
 }
 
-/* Old BSD interface, deprecated */
-int sigvec(int __sig, __const struct sigvec *__vec, struct sigvec *__ovec)
+/* This is managed by vcore / 2LS code */
+static int __sigstack(struct sigstack *__ss, struct sigstack *__oss)
 {
        return 0;
 }
 
-/* Linux specific, and not really needed for us */
-int sigreturn(struct sigcontext *__scp)
+/* Could do this with a loop on delivery of the signal, sleeping and getting
+ * woken up by the kernel on any event, like we do with async syscalls. */
+static int __sigsuspend(__const sigset_t *__set)
 {
        return 0;
 }
 
-/* Akaros can't have signals interrupt syscalls to need a restart, though we can
- * re-wake-up the process while it is waiting for its syscall. */
-int siginterrupt(int __sig, int __interrupt)
+/* Can be done similar to sigsuspend, with an extra alarm syscall */
+static int __sigtimedwait(__const sigset_t *__restrict __set,
+                          siginfo_t *__restrict __info,
+                          __const struct timespec *__restrict __timeout)
 {
        return 0;
 }
 
-/* This is managed by vcore / 2LS code */
-int sigstack(struct sigstack *__ss, struct sigstack *__oss)
+/* Can be done similar to sigsuspend */
+static int __sigwait(__const sigset_t *__restrict __set, int *__restrict __sig)
 {
        return 0;
 }
 
-/* This is managed by vcore / 2LS code */
-int sigaltstack(__const struct sigaltstack *__restrict __ss,
-                struct sigaltstack *__restrict __oss)
+/* Can be done similar to sigsuspend */
+static int __sigwaitinfo(__const sigset_t *__restrict __set,
+                         siginfo_t *__restrict __info)
 {
        return 0;
 }
+
+static int __sigself(int signo)
+{
+       int ret;
+
+       if (in_vcore_context())
+               return kill(getpid(), signo);
+
+       ret = uthread_signal(current_uthread, signo);
+
+       void cb(struct uthread *uthread, void *arg)
+       {
+               uthread_paused(uthread);
+       }
+       if (ret == 0)
+               uthread_yield(TRUE, cb, 0);
+       return ret;
+}