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