Split uthread code into its own source file
[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 void ghetto_vcore_entry(void);
23 struct schedule_ops ghetto_sched_ops = {
24         0, /* init, */
25         ghetto_vcore_entry,
26         0, /* thread_create, */
27         0, /* thread_runnable, */
28         0, /* thread_yield, */
29         0, /* thread_exit, */
30         0, /* preempt_pending, */
31         0, /* spawn_thread, */
32 };
33 struct schedule_ops *sched_ops = &ghetto_sched_ops;
34
35 int main(int argc, char** argv)
36 {
37
38         /* don't forget to enable notifs on vcore0.  if you don't, the kernel will
39          * restart your _S with notifs disabled, which is a path to confusion. */
40         struct preempt_data *vcpd = &__procdata.vcore_preempt_data[0];
41         vcpd->notif_enabled = TRUE;
42
43         mcs_barrier_init(&b, max_vcores());
44
45         vcore_request(max_vcores());
46         printf("not enough vcores, going to try it manually\n");
47         sys_resource_req(RES_CORES, max_vcores(), 1, REQ_SOFT);
48         printf("We're screwed!\n");
49
50         /* should never make it here */
51         return -1;
52 }
53
54 void ghetto_vcore_entry(void)
55 {
56         uint32_t vcoreid = vcore_id();
57
58         if (vcoreid) {
59                 mcs_barrier_wait(&b, vcoreid);
60                 udelay(5000000);
61                 if (vcoreid == 1)
62                         printf("Proc %d's vcores are yielding\n", getpid());
63                 sys_yield(0);
64         } else {
65                 /* trip the barrier here, all future times are in the loop */
66                 mcs_barrier_wait(&b, vcoreid);
67                 while (1) {
68                         udelay(15000000);
69                         printf("Proc %d requesting its cores again\n", getpid());
70                         begin = read_tsc();
71                         sys_resource_req(RES_CORES, max_vcores(), 1, REQ_SOFT);
72                         mcs_barrier_wait(&b, vcoreid);
73                         end = read_tsc();
74                         printf("Took %llu usec (%llu nsec) to get my yielded cores back.\n",
75                                udiff(begin, end), ndiff(begin, end));
76                         printf("[T]:010:%llu:%llu\n",
77                                udiff(begin, end), ndiff(begin, end));
78                 }
79         }
80         printf("We're screwed!\n");
81         exit(-1);
82 }