Revamped event interfaces (XCC)
[akaros.git] / tests / msr_nice_while.c
1 /* tests/msr_dumb_while.c
2  *
3  * This requests the max_vcores in the system, then just while loops in a
4  * userthread.  The pthread code will nicely yield if it detects an incoming
5  * preemption. */
6
7 #include <ros/event.h>
8 #include <stdlib.h>
9 #include <vcore.h>
10 #include <pthread.h>
11 #include <rassert.h>
12
13 void *while_thread(void *arg)
14 {
15         while (1);
16 }
17
18 int main(int argc, char** argv)
19 {
20         pthread_t *my_threads = malloc(sizeof(pthread_t) * max_vcores());
21
22         /* set up to receive the PREEMPT_PENDING event.  EVENT_VCORE_APPRO tells the
23          * kernel to send the msg to whichever vcore is appropriate. 
24          * TODO: (PIN) this ev_q needs to be pinned */
25         struct event_queue *ev_q = malloc(sizeof(struct event_queue));
26         ev_q->ev_mbox = &__procdata.vcore_preempt_data[0].ev_mbox;
27         ev_q->ev_flags = EVENT_IPI | EVENT_NOMSG | EVENT_VCORE_APPRO;
28         ev_q->ev_vcore = 0;
29         ev_q->ev_handler = 0;
30         __procdata.kernel_evts[EV_PREEMPT_PENDING] = ev_q;
31
32         /* actually only need one less, since the _S will be pthread 0 */
33         for (int i = 0; i < max_vcores() - 1; i++)
34                 pthread_create(&my_threads[i], NULL, &while_thread, NULL);
35
36         assert(num_vcores() == max_vcores());
37         while (1);
38
39         /* should never make it here */
40         return -1;
41 }