SYS_abort_syscall (XCC)
[akaros.git] / kern / src / rendez.c
1 /* Copyright (c) 2013 The Regents of the University of California
2  * Barret Rhoden <brho@cs.berkeley.edu>
3  * See LICENSE for details.
4  *
5  * Plan9 style Rendezvous (http://plan9.bell-labs.com/sys/doc/sleep.html)
6  *
7  * We implement it with CVs, and it can handle multiple sleepers/wakers. */
8
9 #include <rendez.h>
10 #include <kthread.h>
11 #include <alarm.h>
12 #include <assert.h>
13 #include <smp.h>
14 #include <err.h>
15
16 void rendez_init(struct rendez *rv)
17 {
18         cv_init_irqsave(&rv->cv);
19 }
20
21 void rendez_sleep(struct rendez *rv, int (*cond)(void*), void *arg)
22 {
23         int8_t irq_state = 0;
24         struct cv_lookup_elm cle;
25         /* Do a quick check before registering and sleeping.  this is the 'check,
26          * signal, check again' pattern, where the first check is an optimization.
27          * Many rendezes will already be satisfied, so we want to avoid excessive
28          * locking associated with reg/dereg. */
29         cv_lock_irqsave(&rv->cv, &irq_state);
30         if (cond(arg)) {
31                 cv_unlock_irqsave(&rv->cv, &irq_state);
32                 return;
33         }
34         __reg_abortable_cv(&cle, &rv->cv);
35         /* Mesa-style semantics, which is definitely what you want.  See the
36          * discussion at the end of the URL above. */
37         while (!cond(arg)) {
38                 /* it's okay if we miss the ABORT flag; we hold the cv lock, so an
39                  * aborter's broadcast is waiting until we unlock. */
40                 if (should_abort(&cle)) {
41                         cv_unlock_irqsave(&rv->cv, &irq_state);
42                         dereg_abortable_cv(&cle);
43                         error("syscall aborted");
44                 }
45                 cv_wait(&rv->cv);
46                 cpu_relax();
47         }
48         cv_unlock_irqsave(&rv->cv, &irq_state);
49         dereg_abortable_cv(&cle);
50 }
51
52 /* Force a wakeup of all waiters on the rv, including non-timeout users.  For
53  * those, they will just wake up, see the condition is still false (probably)
54  * and go back to sleep. */
55 static void rendez_alarm_handler(struct alarm_waiter *awaiter)
56 {
57         struct rendez *rv = (struct rendez*)awaiter->data;
58         rendez_wakeup(rv);
59 }
60
61 /* Like sleep, but it will timeout in 'msec' milliseconds. */
62 void rendez_sleep_timeout(struct rendez *rv, int (*cond)(void*), void *arg,
63                           unsigned int msec)
64 {
65         int8_t irq_state = 0;
66         struct alarm_waiter awaiter;
67         struct cv_lookup_elm cle;
68         struct timer_chain *pcpui_tchain = &per_cpu_info[core_id()].tchain;
69
70         assert((int)msec > 0);
71         /* Doing this cond check early, but then unlocking again.  Mostly just to
72          * avoid weird issues with the CV lock and the alarm tchain lock. */
73         cv_lock_irqsave(&rv->cv, &irq_state);
74         if (cond(arg)) {
75                 cv_unlock_irqsave(&rv->cv, &irq_state);
76                 return;
77         }
78         cv_unlock_irqsave(&rv->cv, &irq_state);
79         /* The handler will call rendez_wake, but won't mess with the condition
80          * state.  It's enough to break us out of cv_wait() to see .on_tchain. */
81         init_awaiter(&awaiter, rendez_alarm_handler);
82         awaiter.data = rv;
83         set_awaiter_rel(&awaiter, msec * 1000);
84         /* Set our alarm on this cpu's tchain.  Note that when we sleep in cv_wait,
85          * we could be migrated, and later on we could be unsetting the alarm
86          * remotely. */
87         set_alarm(pcpui_tchain, &awaiter);
88         cv_lock_irqsave(&rv->cv, &irq_state);
89         __reg_abortable_cv(&cle, &rv->cv);
90         /* We could wake early for a few reasons.  Legit wakeups after a changed
91          * condition (and we should exit), other alarms with different timeouts (and
92          * we should go back to sleep), etc.  Note it is possible for our alarm to
93          * fire immediately upon setting it: before we even cv_lock. */
94         while (!cond(arg) && awaiter.on_tchain) {
95                 if (should_abort(&cle)) {
96                         cv_unlock_irqsave(&rv->cv, &irq_state);
97                         unset_alarm(pcpui_tchain, &awaiter);
98                         dereg_abortable_cv(&cle);
99                         error("syscall aborted");
100                 }
101                 cv_wait(&rv->cv);
102                 cpu_relax();
103         }
104         cv_unlock_irqsave(&rv->cv, &irq_state);
105         dereg_abortable_cv(&cle);
106         /* Turn off our alarm.  If it already fired, this is a no-op.  Note this
107          * could be cross-core. */
108         unset_alarm(pcpui_tchain, &awaiter);
109 }
110
111 /* plan9 rendez returned a pointer to the proc woken up.  we return "true" if we
112  * woke someone up. */
113 bool rendez_wakeup(struct rendez *rv)
114 {
115         int8_t irq_state = 0;
116         bool ret;
117         /* The plan9 style "one sleeper, one waker" could get by with a signal here.
118          * But we want to make sure all potential waiters are woken up. */
119         cv_lock_irqsave(&rv->cv, &irq_state);
120         ret = rv->cv.nr_waiters ? TRUE : FALSE;
121         __cv_broadcast(&rv->cv);
122         cv_unlock_irqsave(&rv->cv, &irq_state);
123         return ret;
124 }