d89738b401d8983e13bd52f806d6034cd91c4f8b
[akaros.git] / kern / src / monitor.c
1 // Simple command-line kernel monitor useful for
2 // controlling the kernel and exploring the system interactively.
3
4 #ifdef __SHARC__
5 #pragma nosharc
6 #endif
7
8 #include <arch/arch.h>
9 #include <stab.h>
10 #include <smp.h>
11 #include <arch/console.h>
12
13 #include <stdio.h>
14 #include <string.h>
15 #include <assert.h>
16 #include <monitor.h>
17 #include <trap.h>
18 #include <pmap.h>
19 #include <kdebug.h>
20 #include <testing.h>
21 #include <manager.h>
22 #include <schedule.h>
23 #include <resource.h>
24 #include <kdebug.h>
25 #include <syscall.h>
26 #include <kmalloc.h>
27 #include <elf.h>
28
29 #include <ros/timer.h>
30 #include <ros/memlayout.h>
31
32 #define CMDBUF_SIZE     80      // enough for one VGA text line
33
34 typedef struct command {
35         const char *NTS name;
36         const char *NTS desc;
37         // return -1 to force monitor to exit
38         int (*func)(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf);
39 } command_t;
40
41 static command_t (RO commands)[] = {
42         { "help", "Display this list of commands", mon_help },
43         { "kerninfo", "Display information about the kernel", mon_kerninfo },
44         { "backtrace", "Dump a backtrace", mon_backtrace },
45         { "reboot", "Take a ride to the South Bay", mon_reboot },
46         { "showmapping", "Shows VA->PA mappings", mon_showmapping},
47         { "setmapperm", "Sets permissions on a VA->PA mapping", mon_setmapperm},
48         { "cpuinfo", "Prints CPU diagnostics", mon_cpuinfo},
49         { "ps", "Prints process list", mon_ps},
50         { "nanwan", "Meet Nanwan!!", mon_nanwan},
51         { "bin_ls", "List files in /bin", mon_bin_ls},
52         { "bin_run", "Create and run a program from /bin", mon_bin_run},
53         { "manager", "Run the manager", mon_manager},
54         { "procinfo", "Show information about processes", mon_procinfo},
55         { "exit", "Leave the monitor", mon_exit},
56         { "kfunc", "Run a kernel function directly (!!!)", mon_kfunc},
57         { "notify", "Notify a process.  Vcoreid will skip their prefs", mon_notify},
58         { "measure", "Run a specific measurement", mon_measure},
59         { "trace", "Run a specific measurement", mon_trace},
60         { "monitor", "Run the monitor on another core", mon_monitor},
61         { "fs", "Filesystem Diagnostics", mon_fs},
62 };
63 #define NCOMMANDS (sizeof(commands)/sizeof(commands[0]))
64
65 /***** Implementations of basic kernel monitor commands *****/
66
67 int mon_help(int argc, char **argv, trapframe_t *tf)
68 {
69         int i;
70
71         for (i = 0; i < NCOMMANDS; i++)
72                 cprintf("%s - %s\n", commands[i].name, commands[i].desc);
73         return 0;
74 }
75
76 int mon_ps(int argc, char** argv, trapframe_t *tf)
77 {
78         print_allpids();
79         return 0;
80 }
81
82 int mon_kerninfo(int argc, char **argv, trapframe_t *tf)
83 {
84         extern char (RO SNT _start)[], (RO SNT etext)[], (RO SNT edata)[], (RO SNT end)[];
85
86         cprintf("Special kernel symbols:\n");
87         cprintf("  _start %08x (virt)  %08x (phys)\n", _start, (uint32_t)(_start - KERNBASE));
88         cprintf("  etext  %08x (virt)  %08x (phys)\n", etext, (uint32_t)(etext - KERNBASE));
89         cprintf("  edata  %08x (virt)  %08x (phys)\n", edata, (uint32_t)(edata - KERNBASE));
90         cprintf("  end    %08x (virt)  %08x (phys)\n", end, (uint32_t)(end - KERNBASE));
91         cprintf("Kernel executable memory footprint: %dKB\n",
92                 (uint32_t)(end-_start+1023)/1024);
93         return 0;
94 }
95
96 #if 0
97 zra: not called
98 static char RO* function_of(uint32_t address)
99 {
100         extern stab_t (RO stab)[], (RO estab)[];
101         extern char (RO stabstr)[];
102         stab_t* symtab;
103         stab_t* best_symtab = 0;
104         uint32_t best_func = 0;
105
106         // ugly and unsorted
107         for (symtab = stab; symtab < estab; symtab++) {
108                 // only consider functions, type = N_FUN
109                 if ((symtab->n_type == N_FUN) &&
110                     (symtab->n_value <= address) &&
111                         (symtab->n_value > best_func)) {
112                         best_func = symtab->n_value;
113                         best_symtab = symtab;
114                 }
115         }
116         // maybe the first stab really is the right one...  we'll see.
117         if (best_symtab == 0)
118                 return "Function not found!";
119         return stabstr + best_symtab->n_strx;
120 }
121 #endif
122
123 int mon_backtrace(int argc, char **argv, trapframe_t *tf)
124 {
125         backtrace();
126         return 0;
127 }
128
129 int mon_reboot(int argc, char **argv, trapframe_t *tf)
130 {
131         cprintf("[Scottish Accent]: She's goin' down, Cap'n!\n");
132         reboot();
133
134         // really, should never see this
135         cprintf("Sigh....\n");
136         return 0;
137 }
138
139 int mon_showmapping(int argc, char **argv, trapframe_t *tf)
140 {
141         if (argc < 2) {
142                 cprintf("Shows virtual -> physical mappings for a virtual address range.\n");
143                 cprintf("Usage: showmapping START_ADDR [END_ADDR]\n");
144                 return 1;
145         }
146         pde_t* pgdir = (pde_t*)vpd;
147         pte_t *pte, *pde;
148         page_t* page;
149         uintptr_t start, i;
150         size_t size;
151         start = ROUNDDOWN(strtol(argv[1], 0, 16), PGSIZE);
152         size = (argc == 2) ? 1 : strtol(argv[2], 0, 16) - start;
153         if (size/PGSIZE > 512) {
154                 cprintf("Not going to do this for more than 512 items\n");
155                 return 1;
156         }
157
158         show_mapping(start,size);
159         return 0;
160 }
161
162 int mon_setmapperm(int argc, char **argv, trapframe_t *tf)
163 {
164 #ifndef __i386__
165         cprintf("I don't support this call yet!\n");
166         return 1;
167 #else
168         if (argc < 3) {
169                 cprintf("Sets VIRT_ADDR's mapping's permissions to PERMS (in hex)\n");
170                 cprintf("Only affects the lowest level PTE.  To adjust the PDE, do the math.\n");
171                 cprintf("Be careful with this around UVPT, VPT, and friends.\n");
172                 cprintf("Usage: setmapperm VIRT_ADDR PERMS\n");
173                 return 1;
174         }
175         pde_t*COUNT(PTSIZE) pgdir = (pde_t*COUNT(PTSIZE))vpd;
176         pte_t *pte, *pde;
177         page_t* page;
178         uintptr_t va;
179         va = ROUNDDOWN(strtol(argv[1], 0, 16), PGSIZE);
180         page = page_lookup(pgdir, (void*SNT)va, &pte);
181         if (!page) {
182                 cprintf("No such mapping\n");
183                 return 1;
184         }
185         pde = &pgdir[PDX(va)];
186         cprintf("   Virtual    Physical  Ps Dr Ac CD WT U W\n");
187         cprintf("------------------------------------------\n");
188         cprintf("%08p  %08p  %1d  %1d  %1d  %1d  %1d  %1d %1d\n", va, page2pa(page),
189                (*pte & PTE_PS) >> 7, (*pte & PTE_D) >> 6, (*pte & PTE_A) >> 5,
190                (*pte & PTE_PCD) >> 4, (*pte & PTE_PWT) >> 3, (*pte & *pde & PTE_U) >> 2,
191                (*pte & *pde & PTE_W) >> 1);
192         *pte = PTE_ADDR(*pte) | (*pte & PTE_PS) |
193                (PGOFF(strtol(argv[2], 0, 16)) & ~PTE_PS ) | PTE_P;
194         cprintf("%08p  %08p  %1d  %1d  %1d  %1d  %1d  %1d %1d\n", va, page2pa(page),
195                (*pte & PTE_PS) >> 7, (*pte & PTE_D) >> 6, (*pte & PTE_A) >> 5,
196                (*pte & PTE_PCD) >> 4, (*pte & PTE_PWT) >> 3, (*pte & *pde & PTE_U) >> 2,
197                (*pte & *pde & PTE_W) >> 1);
198         return 0;
199 #endif
200 }
201
202 int mon_cpuinfo(int argc, char **argv, trapframe_t *tf)
203 {
204         cprintf("Number of CPUs detected: %d\n", num_cpus);
205         cprintf("Calling CPU's ID: 0x%08x\n", core_id());
206
207         if (argc < 2)
208                 smp_call_function_self(test_print_info_handler, NULL, 0);
209         else
210                 smp_call_function_single(strtol(argv[1], 0, 10),
211                                          test_print_info_handler, NULL, 0);
212         return 0;
213 }
214
215 int mon_manager(int argc, char** argv, trapframe_t *tf)
216 {
217         manager();
218         panic("should never get here");
219         return 0;
220 }
221
222 int mon_nanwan(int argc, char **argv, trapframe_t *tf)
223 {
224         /* Borrowed with love from http://www.geocities.com/SoHo/7373/zoo.htm
225          * (http://www.ascii-art.com/).  Slightly modified to make it 25 lines tall.
226          */
227         printk("\n");
228         printk("             .-.  .-.\n");
229         printk("             |  \\/  |\n");
230         printk("            /,   ,_  `'-.\n");
231         printk("          .-|\\   /`\\     '. \n");
232         printk("        .'  0/   | 0\\  \\_  `\".  \n");
233         printk("     .-'  _,/    '--'.'|#''---'\n");
234         printk("      `--'  |       /   \\#\n");
235         printk("            |      /     \\#\n");
236         printk("            \\     ;|\\    .\\#\n");
237         printk("            |' ' //  \\   ::\\# \n");
238         printk("            \\   /`    \\   ':\\#\n");
239         printk("             `\"`       \\..   \\#\n");
240         printk("                        \\::.  \\#\n");
241         printk("                         \\::   \\#\n");
242         printk("                          \\'  .:\\#\n");
243         printk("                           \\  :::\\#\n");
244         printk("                            \\  '::\\#\n");
245         printk("                             \\     \\#\n");
246         printk("                              \\:.   \\#\n");
247         printk("                               \\::   \\#\n");
248         printk("                                \\'   .\\#\n");
249         printk("                             jgs \\   ::\\#\n");
250         printk("                                  \\      \n");
251         return 0;
252 }
253
254 int mon_bin_ls(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
255 {
256         struct dirent dir = {0};
257         struct file *bin_dir;
258         int retval = 0;
259
260         bin_dir = path_to_file("/bin");
261         if (!bin_dir) {
262                 printk("No /bin directory!\n");
263                 return 1;
264         }
265         printk("Files in /bin:\n-------------------------------\n");
266         do {
267                 retval = bin_dir->f_op->readdir(bin_dir, &dir); 
268                 printk("%s\n", dir.d_name);
269                 dir.d_off++;
270         } while (retval == 1);
271         /* TODO: (REF) need to dealloc when this hits 0, atomic/concurrent/etc */
272         atomic_dec(&bin_dir->f_refcnt);
273         assert(!bin_dir->f_refcnt);
274         bin_dir->f_op->release(bin_dir->f_inode, bin_dir);
275         return 0;
276 }
277
278 int mon_bin_run(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
279 {
280         if (argc != 2) {
281                 printk("Usage: bin_run FILENAME\n");
282                 return 1;
283         }
284         struct file *program;
285         int retval = 0;
286         char buf[6 + MAX_FILENAME_SZ] = "/bin/";        /* /bin/ + max + \0 */
287         strncpy(buf + 5, argv[1], MAX_FILENAME_SZ);
288         program = path_to_file(buf);
289         if (!program) {
290                 printk("No such program!\n");
291                 return 1;
292         }
293         char *p_argv[] = {0, 0, 0};
294         char *p_envp[] = {"LD_LIBRARY_PATH=/lib", 0};   /* for /bin/sh, i think */
295         p_argv[0] = file_name(program);
296         struct proc *p = proc_create(program, p_argv, p_envp);
297
298         spin_lock(&p->proc_lock);
299         __proc_set_state(p, PROC_RUNNABLE_S);
300         schedule_proc(p);
301         spin_unlock(&p->proc_lock);
302         kref_put(&p->kref); /* let go of the reference created in proc_create() */
303         /* TODO: (REF) need to dealloc when this hits 0, atomic/concurrent/etc */
304         atomic_dec(&program->f_refcnt);
305         /* Should never return from schedule (env_pop in there) also note you may
306          * not get the process you created, in the event there are others floating
307          * around that are runnable */
308         schedule();
309         return 0;
310 }
311
312 int mon_procinfo(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
313 {
314         if (argc < 2) {
315                 printk("Usage: procinfo OPTION\n");
316                 printk("\tidlecores: show idle core map\n");
317                 printk("\tall: show all active pids\n");
318                 printk("\trunnable: show proc_runnablelist\n");
319                 printk("\tresources: show resources wanted/granted for all procs\n");
320                 printk("\tpid NUM: show a lot of info for proc NUM\n");
321                 printk("\tunlock: unlock the lock for the ADDR (OMG!!!)\n");
322                 printk("\tkill NUM: destroy proc NUM\n");
323                 return 1;
324         }
325         if (!strcmp(argv[1], "idlecores")) {
326                 print_idlecoremap();
327         } else if (!strcmp(argv[1], "all")) {
328                 print_allpids();
329         } else if (!strcmp(argv[1], "runnable")) {
330                 dump_proclist(&proc_runnablelist);
331         } else if (!strcmp(argv[1], "resources")) {
332                 print_all_resources();
333         } else if (!strcmp(argv[1], "pid")) {
334                 if (argc != 3) {
335                         printk("Give me a pid number.\n");
336                         return 1;
337                 }
338                 print_proc_info(strtol(argv[2], 0, 0));
339         } else if (!strcmp(argv[1], "unlock")) {
340                 if (argc != 3) {
341                         printk("Gimme lock address!  Me want lock address!.\n");
342                         return 1;
343                 }
344                 spinlock_t *lock = (spinlock_t*)strtol(argv[2], 0, 16);
345                 if (!lock) {
346                         printk("Null address...\n");
347                         return 1;
348                 }
349                 spin_unlock(lock);
350         } else if (!strcmp(argv[1], "kill")) {
351                 if (argc != 3) {
352                         printk("Give me a pid number.\n");
353                         return 1;
354                 }
355                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
356                 if (!p) {
357                         printk("No such proc\n");
358                         return 1;
359                 }
360                 proc_destroy(p);
361                 kref_put(&p->kref);
362         } else {
363                 printk("Bad option\n");
364                 return 1;
365         }
366         return 0;
367 }
368
369 int mon_exit(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
370 {
371         return -1;
372 }
373
374 int mon_kfunc(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
375 {
376         #ifndef __i386__
377         printk("Only supported on x86 for now.  =(\n");
378         return -1;
379         #endif
380
381         void (*func)(void *arg, ...);
382
383         if (argc < 2) {
384                 printk("Usage: kfunc FUNCTION [arg1] [arg2] [etc]\n");
385                 printk("Arguments must be in hex.  Can take 6 args.\n");
386                 return 1;
387         }
388         func = debug_get_fn_addr(argv[1]);
389         if (!func) {
390                 printk("Function not found.\n");
391                 return 1;
392         }
393         /* Not elegant, but whatever.  maybe there's a better syntax, or we can do
394          * it with asm magic. */
395         switch (argc) {
396                 case 2: /* have to fake one arg */
397                         func((void*)0);
398                         break;
399                 case 3: /* the real first arg */
400                         func((void*)strtol(argv[2], 0, 16));
401                         break;
402                 case 4:
403                         func((void*)strtol(argv[2], 0, 16),
404                                     strtol(argv[3], 0, 16));
405                         break;
406                 case 5:
407                         func((void*)strtol(argv[2], 0, 16),
408                                     strtol(argv[3], 0, 16),
409                                     strtol(argv[4], 0, 16));
410                         break;
411                 case 6:
412                         func((void*)strtol(argv[2], 0, 16),
413                                     strtol(argv[3], 0, 16),
414                                     strtol(argv[4], 0, 16),
415                                     strtol(argv[5], 0, 16));
416                         break;
417                 case 7:
418                         func((void*)strtol(argv[2], 0, 16),
419                                     strtol(argv[3], 0, 16),
420                                     strtol(argv[4], 0, 16),
421                                     strtol(argv[5], 0, 16),
422                                     strtol(argv[6], 0, 16));
423                         break;
424                 case 8:
425                         func((void*)strtol(argv[2], 0, 16),
426                                     strtol(argv[3], 0, 16),
427                                     strtol(argv[4], 0, 16),
428                                     strtol(argv[5], 0, 16),
429                                     strtol(argv[6], 0, 16),
430                                     strtol(argv[7], 0, 16));
431                         break;
432                 default:
433                         printk("Bad number of arguments.\n");
434                         return -1;
435         }
436         return 0;
437 }
438
439 int mon_notify(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
440 {
441         struct proc *p;
442         unsigned int num;
443         uint32_t vcoreid;
444
445         if (argc < 3) {
446                 printk("Usage: notify PID NUM [VCOREID]\n");
447                 return 1;
448         }
449         p = pid2proc(strtol(argv[1], 0, 0));
450         if (!p) {
451                 printk("No such proc\n");
452                 return 1;
453         }
454         num = strtol(argv[2], 0, 0);
455         if (argc == 4) {
456                 vcoreid = strtol(argv[3], 0, 0);
457                 do_notify(p, vcoreid, num, 0);
458         } else {
459                 proc_notify(p, num, 0);
460         }
461         kref_put(&p->kref);
462         return 0;
463 }
464
465 /* Micro-benchmarky Measurements.  This is really fragile code that probably
466  * won't work perfectly, esp as the kernel evolves. */
467 int mon_measure(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
468 {
469         uint64_t begin = 0, diff = 0;
470         bool self_ipi_pending = FALSE;
471         uint32_t end_refcnt = 0;
472
473         if (argc < 2) {
474                 printk("Usage: measure OPTION\n");
475                 printk("\tkill PID : kill proc PID\n");
476                 printk("\tpreempt PID : preempt proc PID (no delay)\n");
477                 printk("\tpreempt PID [pcore] : preempt PID's pcore (no delay)\n");
478                 printk("\tpreempt-warn PID : warn-preempt proc PID (pending)\n");
479                 printk("\tpreempt-warn PID [pcore] : warn-preempt proc PID's pcore\n");
480                 printk("\tpreempt-raw PID : raw-preempt proc PID\n");
481                 printk("\tpreempt-raw PID [pcore] : raw-preempt proc PID's pcore\n");
482                 return 1;
483         }
484         if (!strcmp(argv[1], "kill")) {
485                 if (argc < 3) {
486                         printk("Give me a pid number.\n");
487                         return 1;
488                 }
489                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
490                 if (!p) {
491                         printk("No such proc\n");
492                         return 1;
493                 }
494                 begin = start_timing();
495 #ifdef __CONFIG_APPSERVER__
496                 printk("Warning: this will be inaccurate due to the appserver.\n");
497                 end_refcnt = atomic_read(&p->kref.refcount) -
498                              p->procinfo->num_vcores - 1;
499 #endif /* __CONFIG_APPSERVER__ */
500                 proc_destroy(p);
501                 kref_put(&p->kref);
502 #ifdef __CONFIG_APPSERVER__
503                 /* Won't be that accurate, since it's not actually going through the
504                  * __proc_free() path. */
505                 spin_on(atomic_read(&p->kref.refcount) != end_refcnt);  
506 #else
507                 /* this is a little ghetto. it's not fully free yet, but we are also
508                  * slowing it down by messing with it, esp with the busy waiting on a
509                  * hyperthreaded core. */
510                 spin_on(p->env_cr3);
511 #endif /* __CONFIG_APPSERVER__ */
512                 /* No noticeable difference using stop_timing instead of read_tsc() */
513                 diff = stop_timing(begin);
514         } else if (!strcmp(argv[1], "preempt")) {
515                 if (argc < 3) {
516                         printk("Give me a pid number.\n");
517                         return 1;
518                 }
519                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
520                 if (!p) {
521                         printk("No such proc\n");
522                         return 1;
523                 }
524                 if (argc == 4) { /* single core being preempted, warned but no delay */
525                         uint32_t pcoreid = strtol(argv[3], 0, 0);
526                         begin = start_timing();
527                         proc_preempt_core(p, pcoreid, 1000000); // 1 sec warning
528                         /* done when unmapped (right before abandoning) */
529                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
530                         diff = stop_timing(begin);
531                 } else { /* preempt all cores, warned but no delay */
532                         end_refcnt = atomic_read(&p->kref.refcount)
533                                      - p->procinfo->num_vcores;
534                         begin = start_timing();
535                         proc_preempt_all(p, 1000000);
536                         /* a little ghetto, implies no one is using p */
537                         spin_on(atomic_read(&p->kref.refcount) != end_refcnt);
538                         diff = stop_timing(begin);
539                 }
540                 kref_put(&p->kref);
541         } else if (!strcmp(argv[1], "preempt-warn")) {
542                 if (argc < 3) {
543                         printk("Give me a pid number.\n");
544                         return 1;
545                 }
546                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
547                 if (!p) {
548                         printk("No such proc\n");
549                         return 1;
550                 }
551                 if (argc == 4) { /* single core being preempted-warned */
552                         uint32_t pcoreid = strtol(argv[3], 0, 0);
553                         spin_lock(&p->proc_lock);
554                         uint32_t vcoreid = p->procinfo->pcoremap[pcoreid].vcoreid;
555                         if (!p->procinfo->pcoremap[pcoreid].valid) {
556                                 printk("Pick a mapped pcore\n");
557                                 spin_unlock(&p->proc_lock);
558                                 return 1;
559                         }
560                         begin = start_timing();
561                         __proc_preempt_warn(p, vcoreid, 1000000); // 1 sec
562                         spin_unlock(&p->proc_lock);
563                         /* done when unmapped (right before abandoning) */
564                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
565                         diff = stop_timing(begin);
566                 } else { /* preempt-warn all cores */
567                         printk("Warning, this won't work if they can't yield their "
568                                "last vcore, will stop at 1!\n");
569                         spin_lock(&p->proc_lock);
570                         begin = start_timing();
571                         __proc_preempt_warnall(p, 1000000);
572                         spin_unlock(&p->proc_lock);
573                         /* target cores do the unmapping / changing of the num_vcores */
574                         spin_on(p->procinfo->num_vcores > 1);
575                         diff = stop_timing(begin);
576                 }
577                 kref_put(&p->kref);
578         } else if (!strcmp(argv[1], "preempt-raw")) {
579                 if (argc < 3) {
580                         printk("Give me a pid number.\n");
581                         return 1;
582                 }
583                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
584                 if (!p) {
585                         printk("No such proc\n");
586                         return 1;
587                 }
588                 if (argc == 4) { /* single core preempted, no warning or waiting */
589                         uint32_t pcoreid = strtol(argv[3], 0, 0);
590                         spin_lock(&p->proc_lock);
591                         if (!p->procinfo->pcoremap[pcoreid].valid) {
592                                 printk("Pick a mapped pcore\n");
593                                 spin_unlock(&p->proc_lock);
594                                 return 1;
595                         }
596                         begin = start_timing();
597                         self_ipi_pending = __proc_preempt_core(p, pcoreid);
598                         spin_unlock(&p->proc_lock);
599                         __proc_kmsg_pending(p, self_ipi_pending);
600                         /* done when unmapped (right before abandoning) */
601                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
602                         diff = stop_timing(begin);
603                         /* TODO: (RMS), if num_vcores == 0, RUNNABLE_M, schedule */
604                 } else { /* preempt all cores, no warning or waiting */
605                         spin_lock(&p->proc_lock);
606                         end_refcnt = atomic_read(&p->kref.refcount)
607                                      - p->procinfo->num_vcores;
608                         begin = start_timing();
609                         self_ipi_pending = __proc_preempt_all(p);
610                         /* TODO: (RMS), RUNNABLE_M, schedule */
611                         spin_unlock(&p->proc_lock);
612                         __proc_kmsg_pending(p, self_ipi_pending);
613                         /* a little ghetto, implies no one else is using p */
614                         spin_on(atomic_read(&p->kref.refcount) != end_refcnt);
615                         diff = stop_timing(begin);
616                 }
617                 kref_put(&p->kref);
618         } else {
619                 printk("Bad option\n");
620                 return 1;
621         }
622         printk("[Tired Giraffe Accent] Took %llu usec (%llu nsec) to finish.\n",
623                diff * 1000000 / system_timing.tsc_freq,
624                diff * 1000000000 / system_timing.tsc_freq);
625         return 0;
626 }
627
628 int mon_trace(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
629 {
630         if (argc < 2) {
631                 printk("Usage: trace OPTION\n");
632                 printk("\tsyscall start [silent] [pid]: starts tracing\n");
633                 printk("\tsyscall stop: stops tracing, prints if it was silent\n");
634                 return 1;
635         }
636         if (!strcmp(argv[1], "syscall")) {
637                 if (argc < 3) {
638                         printk("Need a start or stop.\n");
639                         return 1;
640                 }
641                 if (!strcmp(argv[2], "start")) {
642                         bool all = TRUE;
643                         bool silent = FALSE;
644                         struct proc *p = NULL;
645                         if (argc >= 4) {
646                                 silent = (bool)strtol(argv[3], 0, 0);
647                         }
648                         if (argc >= 5) {
649                                 all = FALSE;
650                                 p = pid2proc(strtol(argv[4], 0, 0));
651                                 if (!p) {
652                                         printk("No such process\n");
653                                         return 1;
654                                 }
655                         }
656                         systrace_start(silent);
657                         if (systrace_reg(all, p))
658                                 printk("No room to trace more processes\n");
659                 } else if (!strcmp(argv[2], "stop")) {
660                         /* Stop and print for all processes */
661                         systrace_stop();
662                         systrace_print(TRUE, 0);
663                         systrace_clear_buffer();
664                 }
665         } else if (!strcmp(argv[1], "opt2")) {
666                 if (argc != 3) {
667                         printk("ERRRRRRRRRR.\n");
668                         return 1;
669                 }
670                 print_proc_info(strtol(argv[2], 0, 0));
671         } else {
672                 printk("Bad option\n");
673                 return 1;
674         }
675         return 0;
676 }
677
678 int mon_monitor(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
679 {
680         if (argc < 2) {
681                 printk("Usage: monitor COREID\n");
682                 return 1;
683         }
684         uint32_t core = strtol(argv[1], 0, 0);
685         if (core >= num_cpus) {
686                 printk("No such core!  Maybe it's in another cell...\n");
687                 return 1;
688         }
689         void run_mon(struct trapframe *tf, uint32_t srcid, void *a0, void *a1,
690                      void *a2)
691         {
692                 monitor(0); // TODO consider passing the tf
693         }
694         send_kernel_message(core, run_mon, 0, 0, 0, KMSG_ROUTINE);
695         return 0;
696 }
697
698 /***** Kernel monitor command interpreter *****/
699
700 #define WHITESPACE "\t\r\n "
701 #define MAXARGS 16
702
703 static int runcmd(char *NTS real_buf, trapframe_t *tf) {
704         char * buf = NTEXPAND(real_buf);
705         int argc;
706         char *NTS argv[MAXARGS];
707         int i;
708
709         // Parse the command buffer into whitespace-separated arguments
710         argc = 0;
711         argv[argc] = 0;
712         while (1) {
713                 // gobble whitespace
714                 while (*buf && strchr(WHITESPACE, *buf))
715                         *buf++ = 0;
716                 if (*buf == 0)
717                         break;
718
719                 // save and scan past next arg
720                 if (argc == MAXARGS-1) {
721                         cprintf("Too many arguments (max %d)\n", MAXARGS);
722                         return 0;
723                 }
724                 //This will get fucked at runtime..... in the ASS
725                 argv[argc++] = buf;
726                 while (*buf && !strchr(WHITESPACE, *buf))
727                         buf++;
728         }
729         argv[argc] = 0;
730
731         // Lookup and invoke the command
732         if (argc == 0)
733                 return 0;
734         for (i = 0; i < NCOMMANDS; i++) {
735                 if (strcmp(argv[0], commands[i].name) == 0)
736                         return commands[i].func(argc, argv, tf);
737         }
738         cprintf("Unknown command '%s'\n", argv[0]);
739         return 0;
740 }
741
742 void monitor(trapframe_t *tf) {
743         static spinlock_t monitor_lock = SPINLOCK_INITIALIZER;
744         spin_lock_irqsave(&monitor_lock);
745
746         char *buf;
747
748         /* they are always disabled, since we have this irqsave lock */
749         if (irq_is_enabled())
750                 printk("Entering Nanwan's Dungeon on Core %d (Ints on):\n", core_id());
751         else
752                 printk("Entering Nanwan's Dungeon on Core %d (Ints off):\n", core_id());
753         printk("Type 'help' for a list of commands.\n");
754
755         if (tf != NULL)
756                 print_trapframe(tf);
757
758         spin_unlock_irqsave(&monitor_lock);
759         while (1) {
760                 spin_lock_irqsave(&monitor_lock);
761                 buf = readline("ROS(Core %d)> ", core_id());
762                 if (buf != NULL) {
763                         spin_unlock_irqsave(&monitor_lock);
764                         if (runcmd(buf, tf) < 0)
765                                 break;
766                 }
767         }
768 }
769
770 int mon_fs(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
771 {
772         /* this assumes one mounted FS at the NS root */
773         struct super_block *sb = default_ns.root->mnt_root->d_inode->i_sb;
774         struct file *i;
775         if (argc < 2) {
776                 printk("Usage: fs OPTION\n");
777                 printk("\topen: show all open files\n");
778                 printk("\tpid: proc PID's fs crap placeholder\n");
779                 return 1;
780         }
781         if (!strcmp(argv[1], "open")) {
782                 printk("Open Files:\n----------------------------\n");
783                 TAILQ_FOREACH(i, &sb->s_files, f_list)
784                         printk("File: %08p, %s, Refs: %d\n", i, file_name(i), i->f_refcnt);
785         } else if (!strcmp(argv[1], "pid")) {
786                 if (argc != 3) {
787                         printk("Give me a pid number.\n");
788                         return 1;
789                 }
790                 /* whatever.  placeholder. */
791         } else {
792                 printk("Bad option\n");
793                 return 1;
794         }
795         return 0;
796 }