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