Use an IRQ alarm in rendez_sleep_timeout()
[akaros.git] / kern / src / rendez.c
index 3421193..80c29fa 100644 (file)
@@ -40,8 +40,7 @@ void rendez_sleep(struct rendez *rv, int (*cond)(void*), void *arg)
                if (should_abort(&cle)) {
                        cv_unlock_irqsave(&rv->cv, &irq_state);
                        dereg_abortable_cv(&cle);
-                       set_errno(EINTR);
-                       error("syscall aborted");
+                       error(EINTR, "syscall aborted");
                }
                cv_wait(&rv->cv);
                cpu_relax();
@@ -53,22 +52,25 @@ void rendez_sleep(struct rendez *rv, int (*cond)(void*), void *arg)
 /* Force a wakeup of all waiters on the rv, including non-timeout users.  For
  * those, they will just wake up, see the condition is still false (probably)
  * and go back to sleep. */
-static void rendez_alarm_handler(struct alarm_waiter *awaiter)
+static void rendez_alarm_handler(struct alarm_waiter *awaiter,
+                                 struct hw_trapframe *hw_tf)
 {
        struct rendez *rv = (struct rendez*)awaiter->data;
+
        rendez_wakeup(rv);
 }
 
-/* Like sleep, but it will timeout in 'msec' milliseconds. */
+/* Like sleep, but it will timeout in 'usec' microseconds. */
 void rendez_sleep_timeout(struct rendez *rv, int (*cond)(void*), void *arg,
-                          unsigned int msec)
+                          uint64_t usec)
 {
        int8_t irq_state = 0;
        struct alarm_waiter awaiter;
        struct cv_lookup_elm cle;
        struct timer_chain *pcpui_tchain = &per_cpu_info[core_id()].tchain;
 
-       assert((int)msec > 0);
+       if (!usec)
+               return;
        /* Doing this cond check early, but then unlocking again.  Mostly just to
         * avoid weird issues with the CV lock and the alarm tchain lock. */
        cv_lock_irqsave(&rv->cv, &irq_state);
@@ -78,10 +80,14 @@ void rendez_sleep_timeout(struct rendez *rv, int (*cond)(void*), void *arg,
        }
        cv_unlock_irqsave(&rv->cv, &irq_state);
        /* The handler will call rendez_wake, but won't mess with the condition
-        * state.  It's enough to break us out of cv_wait() to see .on_tchain. */
-       init_awaiter(&awaiter, rendez_alarm_handler);
+        * state.  It's enough to break us out of cv_wait() to see .on_tchain.  But
+        * for this to work, we need to be an IRQ alarm handler.  If an RKM fired,
+        * then we'd be off the tchain, but it didn't actually run yet.  Also, since
+        * all we're doing is poking a rendez, we might as well just do it from IRQ
+        * ctx instead of mucking with an extra RKM. */
+       init_awaiter_irq(&awaiter, rendez_alarm_handler);
        awaiter.data = rv;
-       set_awaiter_rel(&awaiter, msec * 1000);
+       set_awaiter_rel(&awaiter, usec);
        /* Set our alarm on this cpu's tchain.  Note that when we sleep in cv_wait,
         * we could be migrated, and later on we could be unsetting the alarm
         * remotely. */
@@ -97,8 +103,7 @@ void rendez_sleep_timeout(struct rendez *rv, int (*cond)(void*), void *arg,
                        cv_unlock_irqsave(&rv->cv, &irq_state);
                        unset_alarm(pcpui_tchain, &awaiter);
                        dereg_abortable_cv(&cle);
-                       set_errno(EINTR);
-                       error("syscall aborted");
+                       error(EINTR, "syscall aborted");
                }
                cv_wait(&rv->cv);
                cpu_relax();