Remove the need for a poke_pvcalarm function
[akaros.git] / user / utest / pvcalarm.c
1 #include <utest.h>
2 #include <vcore.h>
3 #include <uthread.h>
4 #include <event.h>
5 #include <pvcalarm.h>
6
7 TEST_SUITE("PVCALARMS");
8
9 void pvcalarm_vcore_entry()
10 {
11         uint32_t vcoreid= vcore_id();
12
13         /* Drop back into main thread  for core 0 */
14         if (current_uthread) {
15                 assert(vcoreid == 0);
16                 run_current_uthread();
17         }
18
19         /* Other vcores get here, so enable notifs so they can get their alarm
20          * events */
21         enable_notifs(vcoreid);
22         while(1);
23 }
24 struct schedule_ops pvcalarm_sched_ops = {
25         .sched_entry = pvcalarm_vcore_entry,
26 };
27 struct schedule_ops *sched_ops = &pvcalarm_sched_ops;
28
29 /* <--- Begin definition of test cases ---> */
30 bool test_pvcalarms(void) {
31         const int INTERVAL = 10000;
32         const int ITERATIONS = 100;
33         int count[max_vcores() - num_vcores()];
34         void pvcalarm_callback()
35         {
36                 __sync_fetch_and_add(&count[vcore_id()], 1);
37         }
38
39         struct uthread dummy = {0};
40         uthread_lib_init(&dummy);
41         vcore_request(max_vcores() - num_vcores());
42         
43         uint64_t now, then;
44         now = tsc2usec(read_tsc());
45         enable_pvcalarms(PVCALARM_PROF, INTERVAL, pvcalarm_callback);
46         for (int i=0; i<num_vcores(); i++)
47                 while(count[i] < ITERATIONS)
48                         cpu_relax();
49         then = tsc2usec(read_tsc());
50         disable_pvcalarms();
51         UT_ASSERT_M("Alarms finished too soon", then > (now + INTERVAL*count[0]));
52         UT_ASSERT_M("Alarms finished too late", then < (now + 2*INTERVAL*count[0]));
53         return true;
54 }
55
56 /* <--- End definition of test cases ---> */
57
58 struct utest utests[] = {
59         UTEST_REG(pvcalarms),
60 };
61 int num_utests = sizeof(utests) / sizeof(struct utest);
62
63 int main(int argc, char *argv[]) {
64         // Run test suite passing it all the args as whitelist of what tests to run.
65         char **whitelist = &argv[1];
66         int whitelist_len = argc - 1;
67         RUN_TEST_SUITE(utests, num_utests, whitelist, whitelist_len);
68 }