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