vmrunkernel: get the page table setup right
[akaros.git] / tests / evq_block.c
1 /* Copyright (c) 2015 Google, Inc.
2  * Barret Rhoden <brho@cs.berkeley.edu>
3  * See LICENSE for details.
4  *
5  * Very basic test for blocking a uthread on event queues. */
6
7 #include <stdio.h>
8 #include <unistd.h>
9 #include <parlib/event.h>
10 #include <parlib/timing.h>
11 #include <parlib/uthread.h>
12 #include <benchutil/alarm.h>
13
14 static struct event_queue *get_ectlr_evq(void)
15 {
16         struct event_queue *ev_q = get_eventq(EV_MBOX_UCQ);
17         evq_attach_wakeup_ctlr(ev_q);
18         return ev_q;
19 }
20
21 void trampoline_handler(struct event_queue *ev_q)
22 {
23         printf("Got event on evp %p\n", ev_q);
24         evq_wakeup_handler(ev_q);
25 }
26
27 int main(int argc, char **argv)
28 {
29         uint64_t now;
30         int ctlfd1, timerfd1;
31         int ctlfd2, timerfd2;
32         /* these need to just exist somewhere.  don't free them. */
33         struct event_queue *evq1 = get_ectlr_evq();
34         struct event_queue *evq2 = get_ectlr_evq();
35         evq1->ev_flags |= EVENT_INDIR | EVENT_SPAM_INDIR | EVENT_WAKEUP;
36         evq2->ev_flags |= EVENT_INDIR | EVENT_SPAM_INDIR | EVENT_WAKEUP;
37         /* hack in our own handler for debugging */
38         evq1->ev_handler = trampoline_handler;
39         evq2->ev_handler = trampoline_handler;
40
41         if (devalarm_get_fds(&ctlfd1, &timerfd1, 0))
42                 return -1;
43         if (devalarm_get_fds(&ctlfd2, &timerfd2, 0))
44                 return -1;
45         if (devalarm_set_evq(timerfd1, evq1, 0))
46                 return -1;
47         if (devalarm_set_evq(timerfd2, evq2, 0))
48                 return -1;
49         now = read_tsc();
50         /* with this setup and the early sleep, two fires, then one.  but we'll
51          * process one first, since that's the first one on the list passed to
52          * blockon */
53         if (devalarm_set_time(timerfd1, now + sec2tsc(4)))
54                 return -1;
55         if (devalarm_set_time(timerfd2, now + sec2tsc(2)))
56                 return -1;
57
58         /* if we remove this, two will fire first and wake us up.  if we don't exit
59          * right away, one will eventually fire and do nothing. */
60         uthread_sleep(5);
61         /* then the actual usage: */
62         struct event_msg msg;
63         struct event_queue *which;
64         uth_blockon_evqs(&msg, &which, 2, evq1, evq2);
65         printf("Got message type %d on evq %s (%p)\n", msg.ev_type,
66                which == evq1 ? "one" : "two", which);
67         return 0;
68 }