VMMCP: misc stuff
[akaros.git] / tests / vmm / vmrunkernel.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 <parlib/arch/arch.h>
7 #include <parlib/ros_debug.h>
8 #include <unistd.h>
9 #include <errno.h>
10 #include <dirent.h>
11 #include <stdlib.h>
12 #include <string.h>
13 #include <ros/syscall.h>
14 #include <sys/mman.h>
15 #include <vmm/coreboot_tables.h>
16 #include <vmm/vmm.h>
17 #include <vmm/acpi/acpi.h>
18 #include <ros/arch/mmu.h>
19 #include <ros/vmx.h>
20 #include <parlib/uthread.h>
21 #include <vmm/virtio.h>
22 #include <vmm/virtio_mmio.h>
23 #include <vmm/virtio_ids.h>
24 #include <vmm/virtio_config.h>
25
26 int msrio(struct vmctl *vcpu, uint32_t opcode);
27 /* Kind of sad what a total clusterf the pc world is. By 1999, you could just scan the hardware 
28  * and work it out. But 2005, that was no longer possible. How sad. 
29  * so we have to fake acpi to make it all work. !@#$!@#$#.
30  * This will be copied to memory at 0xe0000, so the kernel can find it.
31  */
32 /* assume they're all 256 bytes long just to make it easy. Just have pointers that point to aligned things. */
33
34 struct acpi_table_rsdp rsdp = {
35         .signature = "RSD PTR ",
36         .oem_id = "AKAROS",
37         .revision = 2,
38         .length = 36,
39 };
40
41 struct acpi_table_xsdt xsdt = {
42         .header = {
43                 .signature= "XSDT",
44                 // This is so stupid. Incredibly stupid.
45                 .revision = 0,
46                 .oem_id = "AKAROS",
47                 .oem_table_id = "ALPHABET",
48                 .oem_revision = 0,
49                 .asl_compiler_id = "RON ",
50                 .asl_compiler_revision = 0,
51         },
52 };
53 struct acpi_table_fadt fadt = {
54         .header = {
55                 .signature= "FADT",
56                 // This is so stupid. Incredibly stupid.
57                 .revision = 0,
58                 .oem_id = "AKAROS",
59                 .oem_table_id = "ALPHABET",
60                 .oem_revision = 0,
61                 .asl_compiler_id = "RON ",
62                 .asl_compiler_revision = 0,
63         },
64 };
65
66 /* This has to be dropped into memory, then the other crap just follows it.
67  */
68 struct acpi_table_madt madt = {
69         .header = {
70                 .signature = "APIC",
71                 .revision = 0,
72                 .oem_id = "AKAROS",
73                 .oem_table_id = "ALPHABET",
74                 .oem_revision = 0,
75                 .asl_compiler_id = "RON ",
76                 .asl_compiler_revision = 0,
77         },
78         
79         .address = 0xfee00000ULL,
80 };
81
82 struct acpi_madt_local_apic Apic0 = {.header = {.type = ACPI_MADT_TYPE_LOCAL_APIC, .length = sizeof(struct acpi_madt_local_apic)},
83                                      .processor_id = 0, .id = 0};
84 struct acpi_madt_io_apic Apic1 = {.header = {.type = ACPI_MADT_TYPE_IO_APIC, .length = sizeof(struct acpi_madt_io_apic)},
85                                   .id = 1, .address = 0xfec00000, .global_irq_base = 0};
86 struct acpi_madt_interrupt_override isor[] = {
87         /* I have no idea if it should be source irq 2, global 0, or global 2, source 0. Shit. */
88         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
89          .bus = 0, .source_irq = 2, .global_irq = 0, .inti_flags = 0},
90         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
91          .bus = 0, .source_irq = 1, .global_irq = 1, .inti_flags = 0},
92         //{.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
93          //.bus = 0, .source_irq = 2, .global_irq = 2, .inti_flags = 0},
94         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
95          .bus = 0, .source_irq = 3, .global_irq = 3, .inti_flags = 0},
96         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
97          .bus = 0, .source_irq = 4, .global_irq = 4, .inti_flags = 0},
98         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
99          .bus = 0, .source_irq = 5, .global_irq = 5, .inti_flags = 0},
100         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
101          .bus = 0, .source_irq = 6, .global_irq = 6, .inti_flags = 0},
102         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
103          .bus = 0, .source_irq = 7, .global_irq = 7, .inti_flags = 0},
104         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
105          .bus = 0, .source_irq = 8, .global_irq = 8, .inti_flags = 0},
106         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
107          .bus = 0, .source_irq = 9, .global_irq = 9, .inti_flags = 0},
108         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
109          .bus = 0, .source_irq = 10, .global_irq = 10, .inti_flags = 0},
110         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
111          .bus = 0, .source_irq = 11, .global_irq = 11, .inti_flags = 0},
112         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
113          .bus = 0, .source_irq = 12, .global_irq = 12, .inti_flags = 0},
114         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
115          .bus = 0, .source_irq = 13, .global_irq = 13, .inti_flags = 0},
116         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
117          .bus = 0, .source_irq = 14, .global_irq = 14, .inti_flags = 0},
118         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
119          .bus = 0, .source_irq = 15, .global_irq = 15, .inti_flags = 0},
120         // VMMCP routes irq 32 to gsi 17
121         {.header = {.type = ACPI_MADT_TYPE_INTERRUPT_OVERRIDE, .length = sizeof(struct acpi_madt_interrupt_override)},
122          .bus = 0, .source_irq = 32, .global_irq = 17, .inti_flags = 5},
123 };
124
125
126 /* this test will run the "kernel" in the negative address space. We hope. */
127 void *low1m;
128 uint8_t low4k[4096];
129 unsigned long long stack[1024];
130 volatile int shared = 0;
131 volatile int quit = 0;
132 int mcp = 1;
133 int virtioirq = 17;
134
135 /* total hack. If the vm runs away we want to get control again. */
136 unsigned int maxresume = (unsigned int) -1;
137
138 #define MiB 0x100000u
139 #define GiB (1u<<30)
140 #define GKERNBASE (16*MiB)
141 #define KERNSIZE (128*MiB+GKERNBASE)
142 uint8_t _kernel[KERNSIZE];
143
144 unsigned long long *p512, *p1, *p2m;
145
146 void **my_retvals;
147 int nr_threads = 3;
148 int debug = 0;
149 int resumeprompt = 0;
150 /* unlike Linux, this shared struct is for both host and guest. */
151 //      struct virtqueue *constoguest = 
152 //              vring_new_virtqueue(0, 512, 8192, 0, inpages, NULL, NULL, "test");
153 uint64_t virtio_mmio_base = 0x100000000;
154
155 void *consout(void *arg)
156 {
157         char *line, *consline, *outline;
158         static struct scatterlist out[] = { {NULL, sizeof(outline)}, };
159         static struct scatterlist in[] = { {NULL, sizeof(line)}, };
160         static struct scatterlist iov[32];
161         struct virtio_threadarg *a = arg;
162         static unsigned int inlen, outlen, conslen;
163         struct virtqueue *v = a->arg->virtio;
164         fprintf(stderr, "talk thread ..\n");
165         uint16_t head, gaveit = 0, gotitback = 0;
166         uint32_t vv;
167         int i;
168         int num;
169         if (debug) {
170                 fprintf(stderr, "----------------------- TT a %p\n", a);
171                 fprintf(stderr, "talk thread ttargs %x v %x\n", a, v);
172         }
173         
174         for(num = 0;;num++) {
175                 /* host: use any buffers we should have been sent. */
176                 head = wait_for_vq_desc(v, iov, &outlen, &inlen);
177                 if (debug)
178                         fprintf(stderr, "CCC: vq desc head %d, gaveit %d gotitback %d\n", head, gaveit, gotitback);
179                 for(i = 0; debug && i < outlen + inlen; i++)
180                         fprintf(stderr, "CCC: v[%d/%d] v %p len %d\n", i, outlen + inlen, iov[i].v, iov[i].length);
181                 /* host: if we got an output buffer, just output it. */
182                 for(i = 0; i < outlen; i++) {
183                         num++;
184                         int j;
185                         for (j = 0; j < iov[i].length; j++)
186                                 printf("%c", ((char *)iov[i].v)[j]);
187                 }
188                 
189                 if (debug)
190                         fprintf(stderr, "CCC: outlen is %d; inlen is %d\n", outlen, inlen);
191                 /* host: fill in the writeable buffers. */
192                 /* why we're getting these I don't know. */
193                 for (i = outlen; i < outlen + inlen; i++) {
194                         if (debug) fprintf(stderr, "CCC: send back empty writeable");
195                         iov[i].length = 0;
196                 }
197                 if (debug) fprintf(stderr, "CCC: call add_used\n");
198                 /* host: now ack that we used them all. */
199                 add_used(v, head, outlen+inlen);
200                 if (debug) fprintf(stderr, "CCC: DONE call add_used\n");
201         }
202         fprintf(stderr, "All done\n");
203         return NULL;
204 }
205
206 // FIXME. 
207 volatile int consdata = 0;
208
209 void *consin(void *arg)
210 {
211         struct virtio_threadarg *a = arg;
212         char *line, *outline;
213         static char consline[128];
214         static struct scatterlist iov[32];
215         static struct scatterlist out[] = { {NULL, sizeof(outline)}, };
216         static struct scatterlist in[] = { {NULL, sizeof(line)}, };
217
218         static unsigned int inlen, outlen, conslen;
219         struct virtqueue *v = a->arg->virtio;
220         fprintf(stderr, "consin thread ..\n");
221         uint16_t head, gaveit = 0, gotitback = 0;
222         uint32_t vv;
223         int i;
224         int num;
225         
226         if (debug) fprintf(stderr, "Spin on console being read, print num queues, halt\n");
227
228         for(num = 0;! quit;num++) {
229                 int debug = 1;
230                 /* host: use any buffers we should have been sent. */
231                 head = wait_for_vq_desc(v, iov, &outlen, &inlen);
232                 if (debug)
233                         fprintf(stderr, "vq desc head %d, gaveit %d gotitback %d\n", head, gaveit, gotitback);
234                 for(i = 0; debug && i < outlen + inlen; i++)
235                         fprintf(stderr, "v[%d/%d] v %p len %d\n", i, outlen + inlen, iov[i].v, iov[i].length);
236                 if (debug)
237                         fprintf(stderr, "outlen is %d; inlen is %d\n", outlen, inlen);
238                 /* host: fill in the writeable buffers. */
239                 for (i = outlen; i < outlen + inlen; i++) {
240                         /* host: read a line. */
241                         memset(consline, 0, 128);
242                         if (fgets(consline, 4096-256, stdin) == NULL) {
243                                 exit(0);
244                         } 
245                         if (debug) fprintf(stderr, "GOT A LINE:%s:\n", consline);
246                         if (strlen(consline) < 3 && consline[0] == 'q' ) {
247                                 quit = 1;
248                                 break;
249                         }
250
251                         memmove(iov[i].v, consline, strlen(consline)+ 1);
252                         iov[i].length = strlen(consline) + 1;
253                 }
254                 if (debug) fprintf(stderr, "call add_used\n");
255                 /* host: now ack that we used them all. */
256                 add_used(v, head, outlen+inlen);
257                 consdata = 1;
258                 if (debug) fprintf(stderr, "DONE call add_used\n");
259         }
260         fprintf(stderr, "All done\n");
261         return NULL;
262 }
263
264 static struct vqdev vqdev= {
265 name: "console",
266 dev: VIRTIO_ID_CONSOLE,
267 device_features: 0, /* Can't do it: linux console device does not support it. VIRTIO_F_VERSION_1*/
268 numvqs: 2,
269 vqs: {
270                 {name: "consin", maxqnum: 64, f: consin, arg: (void *)0},
271                 {name: "consout", maxqnum: 64, f: consout, arg: (void *)0},
272         }
273 };
274
275 void lowmem() {
276         __asm__ __volatile__ (".section .lowmem, \"aw\"\n\tlow: \n\t.=0x1000\n\t.align 0x100000\n\t.previous\n");
277 }
278
279 static uint8_t acpi_tb_checksum(uint8_t *buffer, uint32_t length)
280 {
281         uint8_t sum = 0;
282         uint8_t *end = buffer + length;
283         fprintf(stderr, "tbchecksum %p for %d", buffer, length);
284         while (buffer < end) {
285                 if (end - buffer < 2)
286                         fprintf(stderr, "%02x\n", sum);
287                 sum = (uint8_t)(sum + *(buffer++));
288         }
289         fprintf(stderr, " is %02x\n", sum);
290         return (sum);
291 }
292
293 static void gencsum(uint8_t *target, void *data, int len)
294 {
295         uint8_t csum;
296         // blast target to zero so it does not get counted (it might be in the struct we checksum) 
297         // And, yes, it is, goodness.
298         fprintf(stderr, "gencsum %p target %p source %d bytes\n", target, data, len);
299         *target = 0;
300         csum  = acpi_tb_checksum((uint8_t *)data, len);
301         *target = ~csum + 1;
302         fprintf(stderr, "Cmoputed is %02x\n", *target);
303 }
304
305 int main(int argc, char **argv)
306 {
307         void *a = (void *)0xe0000;
308         struct acpi_table_rsdp *r;
309         struct acpi_table_fadt *f;
310         struct acpi_table_madt *m;
311         struct acpi_table_xsdt *x;
312         uint64_t virtiobase = 0x100000000ULL;
313         // lowmem is a bump allocated pointer to 2M at the "physbase" of memory 
314         void *lowmem = (void *) 0x1000000;
315         struct vmctl vmctl;
316         int amt;
317         int vmmflags = 0; // Disabled probably forever. VMM_VMCALL_PRINTF;
318         uint64_t entry = 0x1200000, kerneladdress = 0x1200000;
319         int nr_gpcs = 1;
320         int fd = open("#c/vmctl", O_RDWR), ret;
321         void * xp;
322         int kfd = -1;
323         static char cmd[512];
324         int i;
325         uint8_t csum;
326         void *coreboot_tables = (void *) 0x1165000;
327 fprintf(stderr, "%p %p %p %p\n", PGSIZE, PGSHIFT, PML1_SHIFT, PML1_PTE_REACH);
328
329         // mmap is not working for us at present.
330         if ((uint64_t)_kernel > GKERNBASE) {
331                 fprintf(stderr, "kernel array @%p is above , GKERNBASE@%p sucks\n", _kernel, GKERNBASE);
332                 exit(1);
333         }
334         memset(_kernel, 0, sizeof(_kernel));
335         memset(lowmem, 0xff, 2*1048576);
336         memset(low4k, 0xff, 4096);
337         // avoid at all costs, requires too much instruction emulation.
338         //low4k[0x40e] = 0;
339         //low4k[0x40f] = 0xe0;
340
341         if (fd < 0) {
342                 perror("#cons/sysctl");
343                 exit(1);
344         }
345         argc--,argv++;
346         // switches ...
347         // Sorry, I don't much like the gnu opt parsing code.
348         while (1) {
349                 if (*argv[0] != '-')
350                         break;
351                 switch(argv[0][1]) {
352                 case 'd':
353                         debug++;
354                         break;
355                 case 'v':
356                         vmmflags |= VMM_VMCALL_PRINTF;
357                         break;
358                 case 'm':
359                         argc--,argv++;
360                         maxresume = strtoull(argv[0], 0, 0);
361                         break;
362                 case 'i':
363                         argc--,argv++;
364                         virtioirq = strtoull(argv[0], 0, 0);
365                         break;
366                 default:
367                         fprintf(stderr, "BMAFR\n");
368                         break;
369                 }
370                 argc--,argv++;
371         }
372         if (argc < 1) {
373                 fprintf(stderr, "Usage: %s vmimage [-n (no vmcall printf)] [coreboot_tables [loadaddress [entrypoint]]]\n", argv[0]);
374                 exit(1);
375         }
376         if (argc > 1)
377                 coreboot_tables = (void *) strtoull(argv[1], 0, 0);
378         if (argc > 2)
379                 kerneladdress = strtoull(argv[2], 0, 0);
380         if (argc > 3)
381                 entry = strtoull(argv[3], 0, 0);
382         kfd = open(argv[0], O_RDONLY);
383         if (kfd < 0) {
384                 perror(argv[0]);
385                 exit(1);
386         }
387         // read in the kernel.
388         xp = (void *)kerneladdress;
389         for(;;) {
390                 amt = read(kfd, xp, 1048576);
391                 if (amt < 0) {
392                         perror("read");
393                         exit(1);
394                 }
395                 if (amt == 0) {
396                         break;
397                 }
398                 xp += amt;
399         }
400         fprintf(stderr, "Read in %d bytes\n", xp-kerneladdress);
401         close(kfd);
402
403         // The low 1m so we can fill in bullshit like ACPI. */
404         low1m = mmap((int*)4096, MiB-4096, PROT_READ | PROT_WRITE,
405                          MAP_ANONYMOUS, -1, 0);
406         if (low1m != (void *)4096) {
407                 perror("Unable to mmap low 1m");
408                 exit(1);
409         }
410         memset(low1m, 0xff, MiB-4096);
411         r = a;
412         fprintf(stderr, "install rsdp to %p\n", r);
413         *r = rsdp;
414         a += sizeof(*r);
415         memmove(&r->xsdt_physical_address, &a, sizeof(a));
416         gencsum(&r->checksum, r, ACPI_RSDP_CHECKSUM_LENGTH);
417         if ((csum = acpi_tb_checksum((uint8_t *) r, ACPI_RSDP_CHECKSUM_LENGTH)) != 0) {
418                 fprintf(stderr, "RSDP has bad checksum; summed to %x\n", csum);
419                 exit(1);
420         }
421
422         /* Check extended checksum if table version >= 2 */
423         gencsum(&r->extended_checksum, r, ACPI_RSDP_XCHECKSUM_LENGTH);
424         if ((rsdp.revision >= 2) &&
425             (acpi_tb_checksum((uint8_t *) r, ACPI_RSDP_XCHECKSUM_LENGTH) != 0)) {
426                 fprintf(stderr, "RSDP has bad checksum v2\n");
427                 exit(1);
428         }
429
430         /* just leave a bunch of space for the xsdt. */
431         /* we need to zero the area since it has pointers. */
432         x = a;
433         a += sizeof(*x) + 8*sizeof(void *);
434         memset(x, 0, a - (void *)x);
435         fprintf(stderr, "install xsdt to %p\n", x);
436         *x = xsdt;
437         x->table_offset_entry[0] = 0;
438         x->table_offset_entry[1] = 0;
439         x->header.length = a - (void *)x;
440
441         f = a;
442         fprintf(stderr, "install fadt to %p\n", f);
443         *f = fadt;
444         x->table_offset_entry[2] = (uint64_t) f;
445         a += sizeof(*f);
446         f->header.length = a - (void *)f;
447         gencsum(&f->header.checksum, f, f->header.length);
448         if (acpi_tb_checksum((uint8_t *)f, f->header.length) != 0) {
449                 fprintf(stderr, "ffadt has bad checksum v2\n");
450                 exit(1);
451         }
452
453         m = a;
454         *m = madt;
455         x->table_offset_entry[3] = (uint64_t) m;
456         a += sizeof(*m);
457         fprintf(stderr, "install madt to %p\n", m);
458         memmove(a, &Apic0, sizeof(Apic0));
459         a += sizeof(Apic0);
460         memmove(a, &Apic1, sizeof(Apic1));
461         a += sizeof(Apic1);
462         memmove(a, &isor, sizeof(isor));
463         a += sizeof(isor);
464         m->header.length = a - (void *)m;
465         gencsum(&m->header.checksum, m, m->header.length);
466         if (acpi_tb_checksum((uint8_t *) m, m->header.length) != 0) {
467                 fprintf(stderr, "madt has bad checksum v2\n");
468                 exit(1);
469         }
470         fprintf(stderr, "allchecksums ok\n");
471
472         gencsum(&x->header.checksum, x, x->header.length);
473         if ((csum = acpi_tb_checksum((uint8_t *) x, x->header.length)) != 0) {
474                 fprintf(stderr, "XSDT has bad checksum; summed to %x\n", csum);
475                 exit(1);
476         }
477
478         hexdump(stdout, r, a-(void *)r);
479
480         if (ros_syscall(SYS_setup_vmm, nr_gpcs, vmmflags, 0, 0, 0, 0) != nr_gpcs) {
481                 perror("Guest pcore setup failed");
482                 exit(1);
483         }
484
485         fprintf(stderr, "Run with %d cores and vmmflags 0x%x\n", nr_gpcs, vmmflags);
486         mcp = 1;
487         if (mcp) {
488                 my_retvals = malloc(sizeof(void*) * nr_threads);
489                 if (!my_retvals)
490                         perror("Init threads/malloc");
491
492                 pthread_can_vcore_request(FALSE);       /* 2LS won't manage vcores */
493                 pthread_need_tls(FALSE);
494                 pthread_mcp_init();                                     /* gives us one vcore */
495                 vcore_request(nr_threads - 1);          /* ghetto incremental interface */
496                 for (int i = 0; i < nr_threads; i++) {
497                         xp = __procinfo.vcoremap;
498                         fprintf(stderr, "%p\n", __procinfo.vcoremap);
499                         fprintf(stderr, "Vcore %d mapped to pcore %d\n", i,
500                                 __procinfo.vcoremap[i].pcoreid);
501                 }
502         }
503
504         ret = syscall(33, 1);
505         if (ret < 0) {
506                 perror("vm setup");
507                 exit(1);
508         }
509         ret = posix_memalign((void **)&p512, 4096, 3*4096);
510         fprintf(stderr, "memalign is %p\n", p512);
511         if (ret) {
512                 perror("ptp alloc");
513                 exit(1);
514         }
515         p1 = &p512[512];
516         p2m = &p512[1024];
517         uint64_t kernbase = 0; //0xffffffff80000000;
518         uint64_t highkernbase = 0xffffffff80000000;
519         p512[PML4(kernbase)] = (unsigned long long)p1 | 7;
520         p1[PML3(kernbase)] = /*0x87; */(unsigned long long)p2m | 7;
521         p512[PML4(highkernbase)] = (unsigned long long)p1 | 7;
522         p1[PML3(highkernbase)] = /*0x87; */(unsigned long long)p2m | 7;
523 #define _2MiB (0x200000)
524
525         for (i = 0; i < 512; i++) {
526                 p2m[PML2(kernbase + i * _2MiB)] = 0x87 | i * _2MiB;
527         }
528
529         kernbase >>= (0+12);
530         kernbase <<= (0 + 12);
531         uint8_t *kernel = (void *)GKERNBASE;
532         //write_coreboot_table(coreboot_tables, ((void *)VIRTIOBASE) /*kernel*/, KERNSIZE + 1048576);
533         hexdump(stdout, coreboot_tables, 512);
534         fprintf(stderr, "kernbase for pml4 is 0x%llx and entry is %llx\n", kernbase, entry);
535         fprintf(stderr, "p512 %p p512[0] is 0x%lx p1 %p p1[0] is 0x%x\n", p512, p512[0], p1, p1[0]);
536         vmctl.interrupt = 0;
537         vmctl.command = REG_RSP_RIP_CR3;
538         vmctl.cr3 = (uint64_t) p512;
539         vmctl.regs.tf_rip = entry;
540         vmctl.regs.tf_rsp = (uint64_t) &stack[1024];
541         if (mcp) {
542                 /* set up virtio bits, which depend on threads being enabled. */
543                 register_virtio_mmio(&vqdev, virtio_mmio_base);
544         }
545         fprintf(stderr, "threads started\n");
546         fprintf(stderr, "Writing command :%s:\n", cmd);
547
548         ret = write(fd, &vmctl, sizeof(vmctl));
549         if (ret != sizeof(vmctl)) {
550                 perror(cmd);
551         }
552         while (1) {
553                 void showstatus(FILE *f, struct vmctl *v);
554                 int c;
555                 uint8_t byte;
556                 vmctl.command = REG_RIP;
557                 if (maxresume-- == 0) {
558                         debug = 1;
559                         resumeprompt = 1;
560                 }
561                 if (debug) {
562                         fprintf(stderr, "RIP %p, shutdown 0x%x\n", vmctl.regs.tf_rip, vmctl.shutdown);
563                         showstatus(stderr, &vmctl);
564                 }
565                 if (resumeprompt) {
566                         fprintf(stderr, "RESUME?\n");
567                         c = getchar();
568                         if (c == 'q')
569                                 break;
570                 }
571                 if (vmctl.shutdown == SHUTDOWN_EPT_VIOLATION) {
572                         uint64_t gpa, *regp, val;
573                         uint8_t regx;
574                         int store, size;
575                         int advance;
576                         if (decode(&vmctl, &gpa, &regx, &regp, &store, &size, &advance)) {
577                                 fprintf(stderr, "RIP %p, shutdown 0x%x\n", vmctl.regs.tf_rip, vmctl.shutdown);
578                                 showstatus(stderr, &vmctl);
579                                 quit = 1;
580                                 break;
581                         }
582                         if (debug) fprintf(stderr, "%p %p %p %p %p %p\n", gpa, regx, regp, store, size, advance);
583                         if ((gpa & ~0xfffULL) == virtiobase) {
584                                 if (debug) fprintf(stderr, "DO SOME VIRTIO\n");
585                                 // Lucky for us the various virtio ops are well-defined.
586                                 virtio_mmio(&vmctl, gpa, regx, regp, store);
587                         } else if ((gpa & 0xfee00000) == 0xfee00000) {
588                                 // until we fix our include mess, just put the proto here.
589                                 int apic(struct vmctl *v, uint64_t gpa, int destreg, uint64_t *regp, int store);
590                                 apic(&vmctl, gpa, regx, regp, store);
591                         } else if ((gpa & 0xfec00000) == 0xfec00000) {
592                                 // until we fix our include mess, just put the proto here.
593                                 int do_ioapic(struct vmctl *v, uint64_t gpa, int destreg, uint64_t *regp, int store);
594                                 do_ioapic(&vmctl, gpa, regx, regp, store);
595                         } else if (gpa < 4096) {
596                                 uint64_t val = 0;
597                                 memmove(&val, &low4k[gpa], size);
598                                 hexdump(stdout, &low4k[gpa], size);
599                                 fprintf(stderr, "Low 1m, code %p read @ %p, size %d, val %p\n", vmctl.regs.tf_rip, gpa, size, val);
600                                 memmove(regp, &low4k[gpa], size);
601                                 hexdump(stdout, regp, size);
602                         } else {
603                                 fprintf(stderr, "EPT violation: can't handle %p\n", gpa);
604                                 fprintf(stderr, "RIP %p, shutdown 0x%x\n", vmctl.regs.tf_rip, vmctl.shutdown);
605                                 fprintf(stderr, "Returning 0xffffffff\n");
606                                 showstatus(stderr, &vmctl);
607                                 // Just fill the whole register for now.
608                                 *regp = (uint64_t) -1;
609                         }
610                         vmctl.regs.tf_rip += advance;
611                         if (debug) fprintf(stderr, "Advance rip by %d bytes to %p\n", advance, vmctl.regs.tf_rip);
612                         vmctl.shutdown = 0;
613                         vmctl.gpa = 0;
614                         vmctl.command = REG_ALL;
615                 } else if (vmctl.shutdown == SHUTDOWN_UNHANDLED_EXIT_REASON) {
616                         switch(vmctl.ret_code){
617                         case  EXIT_REASON_VMCALL:
618                                 byte = vmctl.regs.tf_rdi;
619                                 printf("%c", byte);
620                                 if (byte == '\n') printf("%c", 'V');
621                                 vmctl.regs.tf_rip += 3;
622                                 break;
623                         case EXIT_REASON_EXTERNAL_INTERRUPT:
624                                 //debug = 1;
625                                 fprintf(stderr, "XINT 0x%x 0x%x\n", vmctl.intrinfo1, vmctl.intrinfo2);
626                                 vmctl.command = RESUME;
627                                 break;
628                         case EXIT_REASON_IO_INSTRUCTION:
629                                 fprintf(stderr, "IO @ %p\n", vmctl.regs.tf_rip);
630                                 io(&vmctl);
631                                 vmctl.shutdown = 0;
632                                 vmctl.gpa = 0;
633                                 vmctl.command = REG_ALL;
634                                 break;
635                         case EXIT_REASON_INTERRUPT_WINDOW:
636                                 if (consdata) {
637                                         if (debug) fprintf(stderr, "inject an interrupt\n");
638                                         virtio_mmio_set_vring_irq();
639                                         vmctl.interrupt = 0x80000000 | virtioirq;
640                                         vmctl.command = RESUME;
641                                         consdata = 0;
642                                 }
643                                 break;
644                         case EXIT_REASON_MSR_WRITE:
645                         case EXIT_REASON_MSR_READ:
646                                 fprintf(stderr, "Do an msr\n");
647                                 quit = msrio(&vmctl, vmctl.ret_code);
648                                 if (quit) {
649                                         fprintf(stderr, "MSR FAILED: RIP %p, shutdown 0x%x\n", vmctl.regs.tf_rip, vmctl.shutdown);
650                                         showstatus(stderr, &vmctl);
651                                 }
652                                 break;
653                         case EXIT_REASON_HLT:
654                                 fflush(stdout);
655                                 if (debug)fprintf(stderr, "\n================== Guest halted. =======================\n");
656                                 if (debug)fprintf(stderr, "Wait for cons data\n");
657                                 while (!consdata)
658                                         ;
659                                 //debug = 1;
660                                 if (debug)fprintf(stderr, "Resume with consdata ...\n");
661                                 vmctl.regs.tf_rip += 1;
662                                 ret = write(fd, &vmctl, sizeof(vmctl));
663                                 if (ret != sizeof(vmctl)) {
664                                         perror(cmd);
665                                 }
666                                 //fprintf(stderr, "RIP %p, shutdown 0x%x\n", vmctl.regs.tf_rip, vmctl.shutdown);
667                                 //showstatus(stderr, &vmctl);
668                                 break;
669                         default:
670                                 fprintf(stderr, "Don't know how to handle exit %d\n", vmctl.ret_code);
671                                 fprintf(stderr, "RIP %p, shutdown 0x%x\n", vmctl.regs.tf_rip, vmctl.shutdown);
672                                 showstatus(stderr, &vmctl);
673                                 quit = 1;
674                                 break;
675                         }
676                 }
677                 if (debug) fprintf(stderr, "at bottom of switch, quit is %d\n", quit);
678                 if (quit)
679                         break;
680                 if (consdata) {
681                         if (debug) fprintf(stderr, "inject an interrupt\n");
682                         fprintf(stderr, "XINT 0x%x 0x%x\n", vmctl.intrinfo1, vmctl.intrinfo2);
683                         if ((vmctl.intrinfo1 == 0) && (vmctl.regs.tf_rflags & 0x200)) {
684                                 vmctl.interrupt = 0x80000000 | virtioirq;
685                                 virtio_mmio_set_vring_irq();
686                                 consdata = 0;
687                                 //debug = 1;
688                         } else { 
689                                 fprintf(stderr, "Can't inject interrupt: IF is clear\n");
690                         }
691                         vmctl.command = RESUME;
692                 }
693                 if (debug) fprintf(stderr, "NOW DO A RESUME\n");
694                 ret = write(fd, &vmctl, sizeof(vmctl));
695                 if (ret != sizeof(vmctl)) {
696                         perror(cmd);
697                 }
698         }
699
700         /* later. 
701         for (int i = 0; i < nr_threads-1; i++) {
702                 int ret;
703                 if (pthread_join(my_threads[i], &my_retvals[i]))
704                         perror("pth_join failed");
705                 fprintf(stderr, "%d %d\n", i, ret);
706         }
707  */
708
709         fflush(stdout);
710         exit(0);
711 }