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