886c06a5e3bc8a60884a497064807604b10cdcda
[akaros.git] / tests / vmmcp.c
1 #include <stdio.h> 
2 #include <pthread.h>
3 #include <sys/types.h>
4 #include <sys/stat.h>
5 #include <fcntl.h>
6 #include <arch/arch.h>
7 #include <unistd.h>
8 #include <errno.h>
9 #include <dirent.h>
10 #include <stdlib.h>
11 #include <string.h>
12 #include <ros/syscall.h>
13
14 unsigned long long stack[1024];
15 volatile int shared = 0;
16 int mcp = 1;
17 #define V(x, t) (*((volatile t*)(x)))
18
19 static void *fail(void*arg)
20 {
21
22         if (mcp)
23         while (V(&shared, int) < 31) {
24                 if (! (V(&shared, int) & 1))
25                         V(&shared, int) = V(&shared, int) + 1;
26 //              cpu_relax();
27         }
28         V(&shared, int) = 55;
29
30         __asm__ __volatile__("vmcall\n");
31         __asm__ __volatile__("mov $0xdeadbeef, %rbx; mov 5, %rax\n");   
32 }
33
34 unsigned long long *p512, *p1, *p2m;
35
36 void *talk_thread(void *arg)
37 {
38         printf("talk thread ..\n");
39         for(; V(&shared, int) < 32; ){
40                 if (V(&shared, int) & 1) {
41                         printf("shared %d\n", V(&shared, int) );
42                         V(&shared, int) = V(&shared, int) + 1;
43                 }
44                 cpu_relax();
45         }
46         printf("All done\n");
47         return NULL;
48 }
49
50 pthread_t *my_threads;
51 void **my_retvals;
52 int nr_threads = 2;
53
54 int main(int argc, char **argv)
55 {
56         int nr_gpcs = 1;
57         int fd = open("#c/sysctl", O_RDWR), ret;
58         void * x;
59         static char cmd[512];
60         if (fd < 0) {
61                 perror("#c/sysctl");
62                 exit(1);
63         }
64         if (ros_syscall(SYS_setup_vmm, nr_gpcs, 0, 0, 0, 0, 0) != nr_gpcs) {
65                 perror("Guest pcore setup failed");
66                 exit(1);
67         }
68
69         mcp = 1; //argc - 1;
70         if (mcp) {
71                 my_threads = malloc(sizeof(pthread_t) * nr_threads);
72                 my_retvals = malloc(sizeof(void*) * nr_threads);
73                 if (!(my_retvals && my_threads))
74                         perror("Init threads/malloc");
75
76                 pthread_can_vcore_request(FALSE);       /* 2LS won't manage vcores */
77                 pthread_need_tls(FALSE);
78                 pthread_lib_init();                                     /* gives us one vcore */
79                 vcore_request(nr_threads - 1);          /* ghetto incremental interface */
80                 for (int i = 0; i < nr_threads; i++) {
81                         x = __procinfo.vcoremap;
82                         printf("%p\n", __procinfo.vcoremap);
83                         printf("Vcore %d mapped to pcore %d\n", i,
84                                 __procinfo.vcoremap[i].pcoreid);
85                 }
86         }
87
88         if (mcp) {
89                 if (pthread_create(&my_threads[0], NULL, &talk_thread, NULL))
90                         perror("pth_create failed");
91 //              if (pthread_create(&my_threads[1], NULL, &fail, NULL))
92 //                      perror("pth_create failed");
93         }
94         printf("threads started\n");
95
96         if (0) for (int i = 0; i < nr_threads-1; i++) {
97                 int ret;
98                 if (pthread_join(my_threads[i], &my_retvals[i]))
99                         perror("pth_join failed");
100                 printf("%d %d\n", i, ret);
101         }
102         
103
104         ret = syscall(33, 1);
105         if (ret < 0) {
106                 perror("vm setup");
107                 exit(1);
108         }
109         ret = posix_memalign((void **)&p512, 4096, 3*4096);
110         if (ret) {
111                 perror("ptp alloc");
112                 exit(1);
113         }
114         p1 = &p512[512];
115         p2m = &p512[1024];
116         p512[0] = (unsigned long long)p1 | 7;
117         p1[0] = /*0x87; */(unsigned long long)p2m | 7;
118         p2m[0] = 0x87;
119         p2m[1] = 0x200000 | 0x87;
120         p2m[2] = 0x400000 | 0x87;
121         p2m[3] = 0x600000 | 0x87;
122
123         printf("p512 %p p512[0] is 0x%lx p1 %p p1[0] is 0x%x\n", p512, p512[0], p1, p1[0]);
124         sprintf(cmd, "V 0x%x 0x%x 0x%x", (unsigned long long)fail, (unsigned long long) &stack[1024], (unsigned long long) p512);
125         printf("Writing command :%s:\n", cmd);
126         ret = write(fd, cmd, strlen(cmd));
127         if (ret != strlen(cmd)) {
128                 perror(cmd);
129         }
130         printf("shared is %d\n", shared);
131
132         return 0;
133 }