The monitor doesn't cache coreid
[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 <console.h>
12 #include <arch/console.h>
13
14 #include <stdio.h>
15 #include <string.h>
16 #include <assert.h>
17 #include <monitor.h>
18 #include <trap.h>
19 #include <pmap.h>
20 #include <kdebug.h>
21 #include <testing.h>
22 #include <manager.h>
23 #include <schedule.h>
24 #include <kdebug.h>
25 #include <syscall.h>
26 #include <kmalloc.h>
27 #include <elf.h>
28 #include <event.h>
29 #include <trap.h>
30 #include <time.h>
31
32 #include <ros/memlayout.h>
33 #include <ros/event.h>
34
35 #define CMDBUF_SIZE     80      // enough for one VGA text line
36
37 typedef struct command {
38         const char *NTS name;
39         const char *NTS desc;
40         // return -1 to force monitor to exit
41         int (*func)(int argc, char **argv, struct hw_trapframe *hw_tf);
42 } command_t;
43
44 static command_t (RO commands)[] = {
45         { "help", "Display this list of commands", mon_help },
46         { "kerninfo", "Display information about the kernel", mon_kerninfo },
47         { "backtrace", "Dump a backtrace", mon_backtrace },
48         { "reboot", "Take a ride to the South Bay", mon_reboot },
49         { "showmapping", "Shows VA->PA mappings", mon_showmapping},
50         { "setmapperm", "Sets permissions on a VA->PA mapping", mon_setmapperm},
51         { "cpuinfo", "Prints CPU diagnostics", mon_cpuinfo},
52         { "ps", "Prints process list", mon_ps},
53         { "nanwan", "Meet Nanwan!!", mon_nanwan},
54         { "bin_ls", "List files in /bin", mon_bin_ls},
55         { "bin_run", "Create and run a program from /bin", mon_bin_run},
56         { "manager", "Run the manager", mon_manager},
57         { "procinfo", "Show information about processes", mon_procinfo},
58         { "exit", "Leave the monitor", mon_exit},
59         { "kfunc", "Run a kernel function directly (!!!)", mon_kfunc},
60         { "notify", "Notify a process.  Vcoreid will skip their prefs", mon_notify},
61         { "measure", "Run a specific measurement", mon_measure},
62         { "trace", "Run some tracing functions", mon_trace},
63         { "monitor", "Run the monitor on another core", mon_monitor},
64         { "fs", "Filesystem Diagnostics", mon_fs},
65         { "bb", "Try to run busybox (ash)", mon_bb},
66         { "alarm", "Alarm Diagnostics", mon_alarm},
67 };
68 #define NCOMMANDS (sizeof(commands)/sizeof(commands[0]))
69
70 /***** Implementations of basic kernel monitor commands *****/
71
72 int mon_help(int argc, char **argv, struct hw_trapframe *hw_tf)
73 {
74         int i;
75
76         for (i = 0; i < NCOMMANDS; i++)
77                 cprintf("%s - %s\n", commands[i].name, commands[i].desc);
78         return 0;
79 }
80
81 int mon_ps(int argc, char** argv, struct hw_trapframe *hw_tf)
82 {
83         print_allpids();
84         return 0;
85 }
86
87 int mon_kerninfo(int argc, char **argv, struct hw_trapframe *hw_tf)
88 {
89         extern char (RO SNT _start)[], (RO SNT etext)[], (RO SNT edata)[], (RO SNT end)[];
90
91         cprintf("Special kernel symbols:\n");
92         cprintf("  _start %016x (virt)  %016x (phys)\n", _start, (uintptr_t)(_start - KERNBASE));
93         cprintf("  etext  %016x (virt)  %016x (phys)\n", etext, (uintptr_t)(etext - KERNBASE));
94         cprintf("  edata  %016x (virt)  %016x (phys)\n", edata, (uintptr_t)(edata - KERNBASE));
95         cprintf("  end    %016x (virt)  %016x (phys)\n", end, (uintptr_t)(end - KERNBASE));
96         cprintf("Kernel executable memory footprint: %dKB\n",
97                 (uint32_t)(end-_start+1023)/1024);
98         return 0;
99 }
100
101 #if 0
102 zra: not called
103 static char RO* function_of(uint32_t address)
104 {
105         extern stab_t (RO stab)[], (RO estab)[];
106         extern char (RO stabstr)[];
107         stab_t* symtab;
108         stab_t* best_symtab = 0;
109         uint32_t best_func = 0;
110
111         // ugly and unsorted
112         for (symtab = stab; symtab < estab; symtab++) {
113                 // only consider functions, type = N_FUN
114                 if ((symtab->n_type == N_FUN) &&
115                     (symtab->n_value <= address) &&
116                         (symtab->n_value > best_func)) {
117                         best_func = symtab->n_value;
118                         best_symtab = symtab;
119                 }
120         }
121         // maybe the first stab really is the right one...  we'll see.
122         if (best_symtab == 0)
123                 return "Function not found!";
124         return stabstr + best_symtab->n_strx;
125 }
126 #endif
127
128 int mon_backtrace(int argc, char **argv, struct hw_trapframe *hw_tf)
129 {
130         uintptr_t pc, fp;
131         if (argc == 1) {
132                 backtrace();
133                 return 0;
134         }
135         if (argc != 3) {
136                 printk("Need either no arguments, or two (PC and FP) in hex\n");
137                 return 1;
138         }
139         pc = strtol(argv[1], 0, 16);
140         fp = strtol(argv[2], 0, 16);
141         printk("Backtrace from instruction %p, with frame pointer %p\n", pc, fp);
142         backtrace_frame(pc, fp);
143         return 0;
144 }
145
146 int mon_reboot(int argc, char **argv, struct hw_trapframe *hw_tf)
147 {
148         cprintf("[Scottish Accent]: She's goin' down, Cap'n!\n");
149         reboot();
150
151         // really, should never see this
152         cprintf("Sigh....\n");
153         return 0;
154 }
155
156 int mon_showmapping(int argc, char **argv, struct hw_trapframe *hw_tf)
157 {
158         if (argc < 2) {
159                 cprintf("Shows virtual -> physical mappings for a virt addr range.\n");
160                 cprintf("Usage: showmapping START_ADDR [END_ADDR]\n");
161                 return 1;
162         }
163         uintptr_t start;
164         size_t size;
165         start = ROUNDDOWN(strtol(argv[1], 0, 16), PGSIZE);
166         size = (argc == 2) ? 1 : strtol(argv[2], 0, 16) - start;
167         if (size/PGSIZE > 512) {
168                 cprintf("Not going to do this for more than 512 items\n");
169                 return 1;
170         }
171
172         show_mapping(start,size);
173         return 0;
174 }
175
176 int mon_setmapperm(int argc, char **argv, struct hw_trapframe *hw_tf)
177 {
178 #ifndef CONFIG_X86_32
179         cprintf("I don't support this call yet!\n");
180         return 1;
181 #else
182         if (argc < 3) {
183                 cprintf("Sets VIRT_ADDR's mapping's permissions to PERMS (in hex)\n");
184                 cprintf("Only affects the lowest level PTE.  To adjust the PDE, do the math.\n");
185                 cprintf("Be careful with this around UVPT, VPT, and friends.\n");
186                 cprintf("Usage: setmapperm VIRT_ADDR PERMS\n");
187                 return 1;
188         }
189         pde_t*COUNT(PTSIZE) pgdir = (pde_t*COUNT(PTSIZE))vpd;
190         pte_t *pte, *pde;
191         page_t* page;
192         uintptr_t va;
193         va = ROUNDDOWN(strtol(argv[1], 0, 16), PGSIZE);
194         page = page_lookup(pgdir, (void*SNT)va, &pte);
195         if (!page) {
196                 cprintf("No such mapping\n");
197                 return 1;
198         }
199         pde = &pgdir[PDX(va)];
200         cprintf("   Virtual    Physical  Ps Dr Ac CD WT U W\n");
201         cprintf("------------------------------------------\n");
202         cprintf("%p  %p  %1d  %1d  %1d  %1d  %1d  %1d %1d\n", va, page2pa(page),
203                (*pte & PTE_PS) >> 7, (*pte & PTE_D) >> 6, (*pte & PTE_A) >> 5,
204                (*pte & PTE_PCD) >> 4, (*pte & PTE_PWT) >> 3, (*pte & *pde & PTE_U) >> 2,
205                (*pte & *pde & PTE_W) >> 1);
206         *pte = PTE_ADDR(*pte) | (*pte & PTE_PS) |
207                (PGOFF(strtol(argv[2], 0, 16)) & ~PTE_PS ) | PTE_P;
208         cprintf("%p  %p  %1d  %1d  %1d  %1d  %1d  %1d %1d\n", va, page2pa(page),
209                (*pte & PTE_PS) >> 7, (*pte & PTE_D) >> 6, (*pte & PTE_A) >> 5,
210                (*pte & PTE_PCD) >> 4, (*pte & PTE_PWT) >> 3, (*pte & *pde & PTE_U) >> 2,
211                (*pte & *pde & PTE_W) >> 1);
212         return 0;
213 #endif
214 }
215
216 int mon_cpuinfo(int argc, char **argv, struct hw_trapframe *hw_tf)
217 {
218         cprintf("Number of CPUs detected: %d\n", num_cpus);
219         cprintf("Calling CPU's ID: 0x%08x\n", core_id());
220
221         if (argc < 2)
222                 smp_call_function_self(test_print_info_handler, NULL, 0);
223         else
224                 smp_call_function_single(strtol(argv[1], 0, 10),
225                                          test_print_info_handler, NULL, 0);
226         return 0;
227 }
228
229 int mon_manager(int argc, char** argv, struct hw_trapframe *hw_tf)
230 {
231         manager();
232         panic("should never get here");
233         return 0;
234 }
235
236 int mon_nanwan(int argc, char **argv, struct hw_trapframe *hw_tf)
237 {
238         /* Borrowed with love from http://www.geocities.com/SoHo/7373/zoo.htm
239          * (http://www.ascii-art.com/).  Slightly modified to make it 25 lines tall.
240          */
241         printk("\n");
242         printk("             .-.  .-.\n");
243         printk("             |  \\/  |\n");
244         printk("            /,   ,_  `'-.\n");
245         printk("          .-|\\   /`\\     '. \n");
246         printk("        .'  0/   | 0\\  \\_  `\".  \n");
247         printk("     .-'  _,/    '--'.'|#''---'\n");
248         printk("      `--'  |       /   \\#\n");
249         printk("            |      /     \\#\n");
250         printk("            \\     ;|\\    .\\#\n");
251         printk("            |' ' //  \\   ::\\# \n");
252         printk("            \\   /`    \\   ':\\#\n");
253         printk("             `\"`       \\..   \\#\n");
254         printk("                        \\::.  \\#\n");
255         printk("                         \\::   \\#\n");
256         printk("                          \\'  .:\\#\n");
257         printk("                           \\  :::\\#\n");
258         printk("                            \\  '::\\#\n");
259         printk("                             \\     \\#\n");
260         printk("                              \\:.   \\#\n");
261         printk("                               \\::   \\#\n");
262         printk("                                \\'   .\\#\n");
263         printk("                             jgs \\   ::\\#\n");
264         printk("                                  \\      \n");
265         return 0;
266 }
267
268 int mon_bin_ls(int argc, char **argv, struct hw_trapframe *hw_tf)
269 {
270         struct dirent dir = {0};
271         struct file *bin_dir;
272         int retval = 0;
273
274         bin_dir = do_file_open("/bin", 0, 0);
275         if (!bin_dir) {
276                 printk("No /bin directory!\n");
277                 return 1;
278         }
279         printk("Files in /bin:\n-------------------------------\n");
280         do {
281                 retval = bin_dir->f_op->readdir(bin_dir, &dir); 
282                 printk("%s\n", dir.d_name);
283         } while (retval == 1);
284         kref_put(&bin_dir->f_kref);
285         return 0;
286 }
287
288 int mon_bin_run(int argc, char **argv, struct hw_trapframe *hw_tf)
289 {
290         if (argc < 2) {
291                 printk("Usage: bin_run FILENAME\n");
292                 return 1;
293         }
294         struct file *program;
295         int retval = 0;
296         char buf[6 + MAX_FILENAME_SZ] = "/bin/";        /* /bin/ + max + \0 */
297         strncpy(buf + 5, argv[1], MAX_FILENAME_SZ);
298         program = do_file_open(buf, 0, 0);
299         if (!program) {
300                 printk("No such program!\n");
301                 return 1;
302         }
303         char **p_argv = kmalloc(sizeof(char*) * argc, 0);       /* bin_run's argc */
304         for (int i = 0; i < argc - 1; i++)
305                 p_argv[i] = argv[i + 1];
306         p_argv[argc - 1] = 0;
307         char *p_envp[] = {"LD_LIBRARY_PATH=/lib", 0};
308         struct proc *p = proc_create(program, p_argv, p_envp);
309         kfree(p_argv);
310         proc_wakeup(p);
311         proc_decref(p); /* let go of the reference created in proc_create() */
312         kref_put(&program->f_kref);
313         /* Make a scheduling decision.  You might not get the process you created,
314          * in the event there are others floating around that are runnable */
315         schedule();
316         /* want to idle, so we un the process we just selected.  this is a bit
317          * hackish, but so is the monitor. */
318         smp_idle();
319         assert(0);
320         return 0;
321 }
322
323 int mon_procinfo(int argc, char **argv, struct hw_trapframe *hw_tf)
324 {
325         int8_t irq_state = 0;
326         if (argc < 2) {
327                 printk("Usage: procinfo OPTION\n");
328                 printk("\tidlecores: show idle core map\n");
329                 printk("\tall: show all active pids\n");
330                 printk("\tsched: scheduler diagnostic report\n");
331                 printk("\tresources: show resources wanted/granted for all procs\n");
332                 printk("\tpid NUM: show a lot of info for proc NUM\n");
333                 printk("\tunlock: unlock the lock for the ADDR (OMG!!!)\n");
334                 printk("\tkill NUM: destroy proc NUM\n");
335                 return 1;
336         }
337         if (!strcmp(argv[1], "idlecores")) {
338                 print_idlecoremap();
339         } else if (!strcmp(argv[1], "all")) {
340                 print_allpids();
341         } else if (!strcmp(argv[1], "sched")) {
342                 sched_diag();
343         } else if (!strcmp(argv[1], "resources")) {
344                 print_all_resources();
345         } else if (!strcmp(argv[1], "pid")) {
346                 if (argc != 3) {
347                         printk("Give me a pid number.\n");
348                         return 1;
349                 }
350                 print_proc_info(strtol(argv[2], 0, 0));
351         } else if (!strcmp(argv[1], "unlock")) {
352                 if (argc != 3) {
353                         printk("Gimme lock address!  Me want lock address!.\n");
354                         return 1;
355                 }
356                 spinlock_t *lock = (spinlock_t*)strtol(argv[2], 0, 16);
357                 if (!lock) {
358                         printk("Null address...\n");
359                         return 1;
360                 }
361                 spin_unlock(lock);
362         } else if (!strcmp(argv[1], "kill")) {
363                 if (argc != 3) {
364                         printk("Give me a pid number.\n");
365                         return 1;
366                 }
367                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
368                 if (!p) {
369                         printk("No such proc\n");
370                         return 1;
371                 }
372                 enable_irqsave(&irq_state);
373                 proc_destroy(p);
374                 disable_irqsave(&irq_state);
375                 proc_decref(p);
376         } else {
377                 printk("Bad option\n");
378                 return 1;
379         }
380         return 0;
381 }
382
383 int mon_exit(int argc, char **argv, struct hw_trapframe *hw_tf)
384 {
385         return -1;
386 }
387
388 int mon_kfunc(int argc, char **argv, struct hw_trapframe *hw_tf)
389 {
390         void (*func)(void *arg, ...);
391
392         if (argc < 2) {
393                 printk("Usage: kfunc FUNCTION [arg1] [arg2] [etc]\n");
394                 printk("Arguments must be in hex.  Can take 6 args.\n");
395                 return 1;
396         }
397         func = (void*)get_symbol_addr(argv[1]);
398         if (!func) {
399                 printk("Function not found.\n");
400                 return 1;
401         }
402         /* Not elegant, but whatever.  maybe there's a better syntax, or we can do
403          * it with asm magic. */
404         switch (argc) {
405                 case 2: /* have to fake one arg */
406                         func((void*)0);
407                         break;
408                 case 3: /* the real first arg */
409                         func((void*)strtol(argv[2], 0, 16));
410                         break;
411                 case 4:
412                         func((void*)strtol(argv[2], 0, 16),
413                                     strtol(argv[3], 0, 16));
414                         break;
415                 case 5:
416                         func((void*)strtol(argv[2], 0, 16),
417                                     strtol(argv[3], 0, 16),
418                                     strtol(argv[4], 0, 16));
419                         break;
420                 case 6:
421                         func((void*)strtol(argv[2], 0, 16),
422                                     strtol(argv[3], 0, 16),
423                                     strtol(argv[4], 0, 16),
424                                     strtol(argv[5], 0, 16));
425                         break;
426                 case 7:
427                         func((void*)strtol(argv[2], 0, 16),
428                                     strtol(argv[3], 0, 16),
429                                     strtol(argv[4], 0, 16),
430                                     strtol(argv[5], 0, 16),
431                                     strtol(argv[6], 0, 16));
432                         break;
433                 case 8:
434                         func((void*)strtol(argv[2], 0, 16),
435                                     strtol(argv[3], 0, 16),
436                                     strtol(argv[4], 0, 16),
437                                     strtol(argv[5], 0, 16),
438                                     strtol(argv[6], 0, 16),
439                                     strtol(argv[7], 0, 16));
440                         break;
441                 default:
442                         printk("Bad number of arguments.\n");
443                         return -1;
444         }
445         return 0;
446 }
447
448 /* Sending a vcoreid forces an event and an IPI/notification */
449 int mon_notify(int argc, char **argv, struct hw_trapframe *hw_tf)
450 {
451         struct proc *p;
452         uint32_t vcoreid;
453         struct event_msg msg = {0};
454
455         if (argc < 3) {
456                 printk("Usage: notify PID NUM [VCOREID]\n");
457                 return 1;
458         }
459         p = pid2proc(strtol(argv[1], 0, 0));
460         if (!p) {
461                 printk("No such proc\n");
462                 return 1;
463         }
464         msg.ev_type = strtol(argv[2], 0, 0);
465         if (argc == 4) {
466                 vcoreid = strtol(argv[3], 0, 0);
467                 /* This will go to the private mbox */
468                 post_vcore_event(p, &msg, vcoreid, EVENT_VCORE_PRIVATE);
469                 proc_notify(p, vcoreid);
470         } else {
471                 /* o/w, try and do what they want */
472                 send_kernel_event(p, &msg, 0);
473         }
474         proc_decref(p);
475         return 0;
476 }
477
478 /* Micro-benchmarky Measurements.  This is really fragile code that probably
479  * won't work perfectly, esp as the kernel evolves. */
480 int mon_measure(int argc, char **argv, struct hw_trapframe *hw_tf)
481 {
482         uint64_t begin = 0, diff = 0;
483         uint32_t end_refcnt = 0;
484         int8_t irq_state = 0;
485
486         if (argc < 2) {
487                 printk("Usage: measure OPTION\n");
488                 printk("\tkill PID : kill proc PID\n");
489                 printk("\tpreempt PID : preempt proc PID (no delay)\n");
490                 printk("\tpreempt PID [pcore] : preempt PID's pcore (no delay)\n");
491                 printk("\tpreempt-warn PID : warn-preempt proc PID (pending)\n");
492                 printk("\tpreempt-warn PID [pcore] : warn-preempt proc PID's pcore\n");
493                 printk("\tpreempt-raw PID : raw-preempt proc PID\n");
494                 printk("\tpreempt-raw PID [pcore] : raw-preempt proc PID's pcore\n");
495                 return 1;
496         }
497         if (!strcmp(argv[1], "kill")) {
498                 if (argc < 3) {
499                         printk("Give me a pid number.\n");
500                         return 1;
501                 }
502                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
503                 if (!p) {
504                         printk("No such proc\n");
505                         return 1;
506                 }
507                 begin = start_timing();
508 #ifdef CONFIG_APPSERVER
509                 printk("Warning: this will be inaccurate due to the appserver.\n");
510                 end_refcnt = kref_refcnt(&p->p_kref) - p->procinfo->num_vcores - 1;
511 #endif /* CONFIG_APPSERVER */
512                 enable_irqsave(&irq_state);
513                 proc_destroy(p);
514                 disable_irqsave(&irq_state);
515                 proc_decref(p);
516 #ifdef CONFIG_APPSERVER
517                 /* Won't be that accurate, since it's not actually going through the
518                  * __proc_free() path. */
519                 spin_on(kref_refcnt(&p->p_kref) != end_refcnt); 
520 #else
521                 /* this is a little ghetto. it's not fully free yet, but we are also
522                  * slowing it down by messing with it, esp with the busy waiting on a
523                  * hyperthreaded core. */
524                 spin_on(p->env_cr3);
525 #endif /* CONFIG_APPSERVER */
526                 /* No noticeable difference using stop_timing instead of read_tsc() */
527                 diff = stop_timing(begin);
528         } else if (!strcmp(argv[1], "preempt")) {
529                 if (argc < 3) {
530                         printk("Give me a pid number.\n");
531                         return 1;
532                 }
533                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
534                 if (!p) {
535                         printk("No such proc\n");
536                         return 1;
537                 }
538                 if (argc == 4) { /* single core being preempted, warned but no delay */
539                         uint32_t pcoreid = strtol(argv[3], 0, 0);
540                         begin = start_timing();
541                         if (proc_preempt_core(p, pcoreid, 1000000)) {
542                                 __sched_put_idle_core(p, pcoreid);
543                                 /* done when unmapped (right before abandoning) */
544                                 spin_on(p->procinfo->pcoremap[pcoreid].valid);
545                         } else {
546                                 printk("Core %d was not mapped to proc\n", pcoreid);
547                         }
548                         diff = stop_timing(begin);
549                 } else { /* preempt all cores, warned but no delay */
550                         end_refcnt = kref_refcnt(&p->p_kref) - p->procinfo->num_vcores;
551                         begin = start_timing();
552                         proc_preempt_all(p, 1000000);
553                         /* a little ghetto, implies no one is using p */
554                         spin_on(kref_refcnt(&p->p_kref) != end_refcnt);
555                         diff = stop_timing(begin);
556                 }
557                 proc_decref(p);
558         } else if (!strcmp(argv[1], "preempt-warn")) {
559                 if (argc < 3) {
560                         printk("Give me a pid number.\n");
561                         return 1;
562                 }
563                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
564                 if (!p) {
565                         printk("No such proc\n");
566                         return 1;
567                 }
568                 printk("Careful: if this hangs, then the process isn't responding.\n");
569                 if (argc == 4) { /* single core being preempted-warned */
570                         uint32_t pcoreid = strtol(argv[3], 0, 0);
571                         spin_lock(&p->proc_lock);
572                         uint32_t vcoreid = p->procinfo->pcoremap[pcoreid].vcoreid;
573                         if (!p->procinfo->pcoremap[pcoreid].valid) {
574                                 printk("Pick a mapped pcore\n");
575                                 spin_unlock(&p->proc_lock);
576                                 return 1;
577                         }
578                         begin = start_timing();
579                         __proc_preempt_warn(p, vcoreid, 1000000); // 1 sec
580                         spin_unlock(&p->proc_lock);
581                         /* done when unmapped (right before abandoning) */
582                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
583                         diff = stop_timing(begin);
584                 } else { /* preempt-warn all cores */
585                         printk("Warning, this won't work if they can't yield their "
586                                "last vcore, will stop at 1!\n");
587                         spin_lock(&p->proc_lock);
588                         begin = start_timing();
589                         __proc_preempt_warnall(p, 1000000);
590                         spin_unlock(&p->proc_lock);
591                         /* target cores do the unmapping / changing of the num_vcores */
592                         spin_on(p->procinfo->num_vcores > 1);
593                         diff = stop_timing(begin);
594                 }
595                 proc_decref(p);
596         } else if (!strcmp(argv[1], "preempt-raw")) {
597                 if (argc < 3) {
598                         printk("Give me a pid number.\n");
599                         return 1;
600                 }
601                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
602                 if (!p) {
603                         printk("No such proc\n");
604                         return 1;
605                 }
606                 if (argc == 4) { /* single core preempted, no warning or waiting */
607                         uint32_t pcoreid = strtol(argv[3], 0, 0);
608                         spin_lock(&p->proc_lock);
609                         if (!p->procinfo->pcoremap[pcoreid].valid) {
610                                 printk("Pick a mapped pcore\n");
611                                 spin_unlock(&p->proc_lock);
612                                 return 1;
613                         }
614                         begin = start_timing();
615                         __proc_preempt_core(p, pcoreid);
616                         if (!p->procinfo->num_vcores)
617                                 __proc_set_state(p, PROC_RUNNABLE_M);
618                         spin_unlock(&p->proc_lock);
619                         /* ghetto, since the ksched should be calling all of this */
620                         __sched_put_idle_core(p, pcoreid);
621                         /* done when unmapped (right before abandoning) */
622                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
623                         diff = stop_timing(begin);
624                 } else { /* preempt all cores, no warning or waiting */
625                         spin_lock(&p->proc_lock);
626                         uint32_t pc_arr[p->procinfo->num_vcores];
627                         uint32_t num_revoked;
628                         end_refcnt = kref_refcnt(&p->p_kref) - p->procinfo->num_vcores;
629                         begin = start_timing();
630                         num_revoked = __proc_preempt_all(p, pc_arr);
631                         __proc_set_state(p, PROC_RUNNABLE_M);
632                         spin_unlock(&p->proc_lock);
633                         if (num_revoked)
634                                 __sched_put_idle_cores(p, pc_arr, num_revoked);
635                         /* a little ghetto, implies no one else is using p */
636                         spin_on(kref_refcnt(&p->p_kref) != end_refcnt);
637                         diff = stop_timing(begin);
638                 }
639                 proc_decref(p);
640         } else {
641                 printk("Bad option\n");
642                 return 1;
643         }
644         printk("[Tired Giraffe Accent] Took %llu usec (%llu nsec) to finish.\n",
645                tsc2usec(diff), tsc2nsec(diff));
646         return 0;
647 }
648
649 /* Used in various debug locations.  Not a kernel API or anything. */
650 bool mon_verbose_trace = FALSE;
651
652 int mon_trace(int argc, char **argv, struct hw_trapframe *hw_tf)
653 {
654         int core;
655         if (argc < 2) {
656                 printk("Usage: trace OPTION\n");
657                 printk("\tsyscall start [silent] [pid]: starts tracing\n");
658                 printk("\tsyscall stop: stops tracing, prints if it was silent\n");
659                 printk("\tcoretf COREID: prints PC, -1 for all cores, verbose => TF\n");
660                 printk("\tpcpui [type [coreid]]: runs pcpui trace ring handlers\n");
661                 printk("\tpcpui-reset [noclear]: resets/clears pcpui trace ring\n");
662                 printk("\tverbose: toggles verbosity, depends on trace command\n");
663                 return 1;
664         }
665         if (!strcmp(argv[1], "syscall")) {
666                 if (argc < 3) {
667                         printk("Need a start or stop.\n");
668                         return 1;
669                 }
670                 if (!strcmp(argv[2], "start")) {
671                         bool all = TRUE;
672                         bool silent = FALSE;
673                         struct proc *p = NULL;
674                         if (argc >= 4) {
675                                 silent = (bool)strtol(argv[3], 0, 0);
676                         }
677                         if (argc >= 5) {
678                                 all = FALSE;
679                                 p = pid2proc(strtol(argv[4], 0, 0));
680                                 if (!p) {
681                                         printk("No such process\n");
682                                         return 1;
683                                 }
684                         }
685                         systrace_start(silent);
686                         if (systrace_reg(all, p))
687                                 printk("No room to trace more processes\n");
688                 } else if (!strcmp(argv[2], "stop")) {
689                         /* Stop and print for all processes */
690                         systrace_stop();
691                         systrace_print(TRUE, 0);
692                         systrace_clear_buffer();
693                 }
694         } else if (!strcmp(argv[1], "coretf")) {
695                 if (argc != 3) {
696                         printk("Need a coreid, fool.\n");
697                         return 1;
698                 }
699                 core = strtol(argv[2], 0, 0);
700                 if (core < 0) {
701                         printk("Sending NMIs to all cores:\n");
702                         for (int i = 0; i < num_cpus; i++)
703                                 send_nmi(i);
704                 } else {
705                         printk("Sending NMI core %d:\n", core);
706                         if (core >= num_cpus) {
707                                 printk("No such core!  Maybe it's in another cell...\n");
708                                 return 1;
709                         }
710                         send_nmi(core);
711                 }
712                 udelay(1000000);
713         } else if (!strcmp(argv[1], "pcpui")) {
714                 int pcpui_type, pcpui_coreid;
715                 if (argc >= 3)
716                         pcpui_type = strtol(argv[2], 0, 0);
717                 else
718                         pcpui_type = 0;
719                 printk("\nRunning PCPUI Trace Ring handlers for type %d\n", pcpui_type);
720                 if (argc >= 4) {
721                         pcpui_coreid = strtol(argv[3], 0, 0); 
722                         pcpui_tr_foreach(pcpui_coreid, pcpui_type);
723                 } else {
724                         pcpui_tr_foreach_all(pcpui_type);
725                 }
726         } else if (!strcmp(argv[1], "pcpui-reset")) {
727                 if (argc >= 3) {
728                         printk("\nResetting all PCPUI Trace Rings\n");
729                         pcpui_tr_reset_all();
730                 } else {
731                         printk("\nResetting and clearing all PCPUI Trace Rings\n");
732                         pcpui_tr_reset_and_clear_all();
733                 }
734         } else if (!strcmp(argv[1], "verbose")) {
735                 if (mon_verbose_trace) {
736                         printk("Turning trace verbosity off\n");
737                         mon_verbose_trace = FALSE;
738                 } else {
739                         printk("Turning trace verbosity on\n");
740                         mon_verbose_trace = TRUE;
741                 }
742         } else if (!strcmp(argv[1], "opt2")) {
743                 if (argc != 3) {
744                         printk("ERRRRRRRRRR.\n");
745                         return 1;
746                 }
747                 print_proc_info(strtol(argv[2], 0, 0));
748         } else {
749                 printk("Bad option\n");
750                 return 1;
751         }
752         return 0;
753 }
754
755 int mon_monitor(int argc, char **argv, struct hw_trapframe *hw_tf)
756 {
757         if (argc < 2) {
758                 printk("Usage: monitor COREID\n");
759                 return 1;
760         }
761         uint32_t core = strtol(argv[1], 0, 0);
762         if (core >= num_cpus) {
763                 printk("No such core!  Maybe it's in another cell...\n");
764                 return 1;
765         }
766         send_kernel_message(core, __run_mon, 0, 0, 0, KMSG_ROUTINE);
767         return 0;
768 }
769
770 /***** Kernel monitor command interpreter *****/
771
772 #define WHITESPACE "\t\r\n "
773 #define MAXARGS 16
774
775 static int runcmd(char *NTS real_buf, struct hw_trapframe *hw_tf) {
776         char * buf = NTEXPAND(real_buf);
777         int argc;
778         char *NTS argv[MAXARGS];
779         int i;
780
781         // Parse the command buffer into whitespace-separated arguments
782         argc = 0;
783         argv[argc] = 0;
784         while (1) {
785                 // gobble whitespace
786                 while (*buf && strchr(WHITESPACE, *buf))
787                         *buf++ = 0;
788                 if (*buf == 0)
789                         break;
790
791                 // save and scan past next arg
792                 if (argc == MAXARGS-1) {
793                         cprintf("Too many arguments (max %d)\n", MAXARGS);
794                         return 0;
795                 }
796                 //This will get fucked at runtime..... in the ASS
797                 argv[argc++] = buf;
798                 while (*buf && !strchr(WHITESPACE, *buf))
799                         buf++;
800         }
801         argv[argc] = 0;
802
803         // Lookup and invoke the command
804         if (argc == 0)
805                 return 0;
806         for (i = 0; i < NCOMMANDS; i++) {
807                 if (strcmp(argv[0], commands[i].name) == 0)
808                         return commands[i].func(argc, argv, hw_tf);
809         }
810         cprintf("Unknown command '%s'\n", argv[0]);
811         return 0;
812 }
813
814 void monitor(struct hw_trapframe *hw_tf)
815 {
816         #define MON_CMD_LENGTH 256
817         char buf[MON_CMD_LENGTH];
818         int cnt;
819         int coreid = core_id_early();
820
821         /* they are always disabled, since we have this irqsave lock */
822         if (irq_is_enabled())
823                 printk("Entering Nanwan's Dungeon on Core %d (Ints on):\n", coreid);
824         else
825                 printk("Entering Nanwan's Dungeon on Core %d (Ints off):\n", coreid);
826         printk("Type 'help' for a list of commands.\n");
827
828         if (hw_tf != NULL)
829                 print_trapframe(hw_tf);
830
831         while (1) {
832                 /* on occasion, the kernel monitor can migrate (like if you run
833                  * something that blocks / syncs and wakes up on another core) */
834                 cmb();
835                 cnt = readline(buf, MON_CMD_LENGTH, "ROS(Core %d)> ", core_id_early());
836                 if (cnt > 0) {
837                         buf[cnt] = 0;
838                         if (runcmd(buf, hw_tf) < 0)
839                                 break;
840                 }
841         }
842 }
843
844 int mon_fs(int argc, char **argv, struct hw_trapframe *hw_tf)
845 {
846         /* this assumes one mounted FS at the NS root */
847         struct super_block *sb;
848         struct file *file;
849         struct inode *inode;
850         struct dentry *dentry;
851         if (argc < 2) {
852                 printk("Usage: fs OPTION\n");
853                 printk("\topen: show all open files\n");
854                 printk("\tinodes: show all inodes\n");
855                 printk("\tdentries [lru|prune]: show all dentries, opt LRU/prune\n");
856                 printk("\tls DIR: print the dir tree starting with DIR\n");
857                 printk("\tpid: proc PID's fs crap placeholder\n");
858                 return 1;
859         }
860         if (!strcmp(argv[1], "open")) {
861                 printk("Open Files:\n----------------------------\n");
862                 TAILQ_FOREACH(sb, &super_blocks, s_list) {
863                         printk("Superblock for %s\n", sb->s_name);
864                         TAILQ_FOREACH(file, &sb->s_files, f_list)
865                                 printk("File: %p, %s, Refs: %d, Drefs: %d, Irefs: %d\n", file,
866                                        file_name(file), kref_refcnt(&file->f_kref),
867                                        kref_refcnt(&file->f_dentry->d_kref),
868                                        kref_refcnt(&file->f_dentry->d_inode->i_kref));
869                 }
870         } else if (!strcmp(argv[1], "inodes")) {
871                 printk("Mounted FS Inodes:\n----------------------------\n");
872                 TAILQ_FOREACH(sb, &super_blocks, s_list) {
873                         printk("Superblock for %s\n", sb->s_name);
874                         TAILQ_FOREACH(inode, &sb->s_inodes, i_sb_list) {
875                                 printk("Inode: %p, Refs: %d, Nlinks: %d, Size(B): %d\n",
876                                        inode, kref_refcnt(&inode->i_kref), inode->i_nlink,
877                                        inode->i_size);
878                                 TAILQ_FOREACH(dentry, &inode->i_dentry, d_alias)
879                                         printk("\t%s: Dentry: %p, Refs: %d\n",
880                                                dentry->d_name.name, dentry,
881                                                kref_refcnt(&dentry->d_kref));
882                         }
883                 }
884         } else if (!strcmp(argv[1], "dentries")) {
885                 printk("Dentry Cache:\n----------------------------\n");
886                 TAILQ_FOREACH(sb, &super_blocks, s_list) {
887                         printk("Superblock for %s\n", sb->s_name);
888                         printk("DENTRY     FLAGS      REFCNT NAME\n");
889                         printk("--------------------------------\n");
890                         /* Hash helper */
891                         void print_dcache_entry(void *item)
892                         {
893                                 struct dentry *d_i = (struct dentry*)item;
894                                 printk("%p %p %02d     %s\n", d_i, d_i->d_flags,
895                                        kref_refcnt(&d_i->d_kref), d_i->d_name.name);
896                         }
897                         hash_for_each(sb->s_dcache, print_dcache_entry);
898                 }
899                 if (argc < 3)
900                         return 0;
901                 if (!strcmp(argv[2], "lru")) {
902                         printk("LRU lists:\n");
903                         TAILQ_FOREACH(sb, &super_blocks, s_list) {
904                                 printk("Superblock for %s\n", sb->s_name);
905                                 TAILQ_FOREACH(dentry, &sb->s_lru_d, d_lru)
906                                         printk("Dentry: %p, Name: %s\n", dentry,
907                                                dentry->d_name.name);
908                         }
909                 } else if (!strcmp(argv[2], "prune")) {
910                         printk("Pruning unused dentries\n");
911                         TAILQ_FOREACH(sb, &super_blocks, s_list)
912                                 dcache_prune(sb, FALSE);
913                 }
914         } else if (!strcmp(argv[1], "ls")) {
915                 if (argc != 3) {
916                         printk("Give me a dir.\n");
917                         return 1;
918                 }
919                 if (argv[2][0] != '/') {
920                         printk("Dear fellow giraffe lover, Use absolute paths.\n");
921                         return 1;
922                 }
923                 ls_dash_r(argv[2]);
924                 /* whatever.  placeholder. */
925         } else if (!strcmp(argv[1], "pid")) {
926                 if (argc != 3) {
927                         printk("Give me a pid number.\n");
928                         return 1;
929                 }
930                 /* whatever.  placeholder. */
931         } else {
932                 printk("Bad option\n");
933                 return 1;
934         }
935         return 0;
936 }
937
938 int mon_bb(int argc, char **argv, struct hw_trapframe *hw_tf)
939 {
940         char *l_argv[3] = {"", "busybox", "ash"};
941         return mon_bin_run(3, l_argv, hw_tf);
942 }
943
944 int mon_alarm(int argc, char **argv, struct hw_trapframe *hw_tf)
945 {
946         if (argc < 2) {
947                 printk("Usage: alarm OPTION\n");
948                 printk("\tpcpu: print full alarm tchains from every core\n");
949                 return 1;
950         }
951         if (!strcmp(argv[1], "pcpu")) {
952                 print_pcpu_chains();
953         } else {
954                 printk("Bad option\n");
955                 return 1;
956         }
957         return 0;
958 }