Make arsc and local async calls use the same structure.
[akaros.git] / tests / msr_cycling_vcores.c
1 /* tests/msr_cycling_vcores.c
2  *
3  * This requests the max_vcores in the system, waits a bit, then gives them
4  * back, looping forever.  We can't give up all vcores, based on the current
5  * kernel, so we hold on to vcore0 to do the thinking. */
6
7 #include <ros/resource.h>
8 #include <parlib.h>
9 #include <rstdio.h>
10 #include <vcore.h>
11 #include <timing.h>
12 #include <mcs.h>
13 #include <uthread.h>
14
15 #ifdef __sparc_v8__
16 # define udelay(x) udelay((x)/2000)
17 #endif
18
19 mcs_barrier_t b;
20 uint64_t begin = 0, end = 0;
21
22 int main(int argc, char** argv)
23 {
24
25         /* don't forget to enable notifs on vcore0.  if you don't, the kernel will
26          * restart your _S with notifs disabled, which is a path to confusion. */
27         struct preempt_data *vcpd = &__procdata.vcore_preempt_data[0];
28         vcpd->notif_enabled = TRUE;
29
30         mcs_barrier_init(&b, max_vcores());
31
32         vcore_request(max_vcores());
33         printf("not enough vcores, going to try it manually\n");
34         sys_resource_req(RES_CORES, max_vcores(), 1, REQ_SOFT);
35         printf("We're screwed!\n");
36
37         /* should never make it here */
38         return -1;
39 }
40
41 void vcore_entry(void)
42 {
43         uint32_t vcoreid = vcore_id();
44
45         if (vcoreid) {
46                 mcs_barrier_wait(&b, vcoreid);
47                 udelay(5000000);
48                 if (vcoreid == 1)
49                         printf("Proc %d's vcores are yielding\n", getpid());
50                 sys_yield(0);
51         } else {
52                 /* trip the barrier here, all future times are in the loop */
53                 mcs_barrier_wait(&b, vcoreid);
54                 while (1) {
55                         udelay(15000000);
56                         printf("Proc %d requesting its cores again\n", getpid());
57                         begin = read_tsc();
58                         sys_resource_req(RES_CORES, max_vcores(), 1, REQ_SOFT);
59                         mcs_barrier_wait(&b, vcoreid);
60                         end = read_tsc();
61                         printf("Took %llu usec (%llu nsec) to get my yielded cores back.\n",
62                                udiff(begin, end), ndiff(begin, end));
63                         printf("[T]:010:%llu:%llu\n",
64                                udiff(begin, end), ndiff(begin, end));
65                 }
66         }
67         printf("We're screwed!\n");
68         exit(-1);
69 }