kfs_run and friends use the FS and s/kfs/bin/
[akaros.git] / kern / src / monitor.c
1 // Simple command-line kernel monitor useful for
2 // controlling the kernel and exploring the system interactively.
3
4 #ifdef __SHARC__
5 #pragma nosharc
6 #endif
7
8 #include <arch/arch.h>
9 #include <stab.h>
10 #include <smp.h>
11 #include <arch/console.h>
12
13 #include <stdio.h>
14 #include <string.h>
15 #include <assert.h>
16 #include <monitor.h>
17 #include <trap.h>
18 #include <pmap.h>
19 #include <kdebug.h>
20 #include <testing.h>
21 #include <kfs.h>
22 #include <manager.h>
23 #include <schedule.h>
24 #include <resource.h>
25 #include <kdebug.h>
26 #include <syscall.h>
27 #include <kmalloc.h>
28 #include <elf.h>
29
30 #include <ros/timer.h>
31 #include <ros/memlayout.h>
32
33 #define CMDBUF_SIZE     80      // enough for one VGA text line
34
35 typedef struct command {
36         const char *NTS name;
37         const char *NTS desc;
38         // return -1 to force monitor to exit
39         int (*func)(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf);
40 } command_t;
41
42 static command_t (RO commands)[] = {
43         { "help", "Display this list of commands", mon_help },
44         { "kerninfo", "Display information about the kernel", mon_kerninfo },
45         { "backtrace", "Dump a backtrace", mon_backtrace },
46         { "reboot", "Take a ride to the South Bay", mon_reboot },
47         { "showmapping", "Shows VA->PA mappings", mon_showmapping},
48         { "setmapperm", "Sets permissions on a VA->PA mapping", mon_setmapperm},
49         { "cpuinfo", "Prints CPU diagnostics", mon_cpuinfo},
50         { "ps", "Prints process list", mon_ps},
51         { "nanwan", "Meet Nanwan!!", mon_nanwan},
52         { "bin_ls", "List files in /bin", mon_bin_ls},
53         { "bin_run", "Create and run a program from /bin", mon_bin_run},
54         { "kfs_cat", "Dumps text from a file from KFS", mon_kfs_cat},
55         { "manager", "Run the manager", mon_manager},
56         { "procinfo", "Show information about processes", mon_procinfo},
57         { "exit", "Leave the monitor", mon_exit},
58         { "kfunc", "Run a kernel function directly (!!!)", mon_kfunc},
59         { "notify", "Notify a process.  Vcoreid will skip their prefs", mon_notify},
60         { "measure", "Run a specific measurement", mon_measure},
61         { "trace", "Run a specific measurement", mon_trace},
62         { "monitor", "Run the monitor on another core", mon_monitor},
63 };
64 #define NCOMMANDS (sizeof(commands)/sizeof(commands[0]))
65
66 /***** Implementations of basic kernel monitor commands *****/
67
68 int mon_help(int argc, char **argv, trapframe_t *tf)
69 {
70         int i;
71
72         for (i = 0; i < NCOMMANDS; i++)
73                 cprintf("%s - %s\n", commands[i].name, commands[i].desc);
74         return 0;
75 }
76
77 int mon_ps(int argc, char** argv, trapframe_t *tf)
78 {
79         print_allpids();
80         return 0;
81 }
82
83 int mon_kerninfo(int argc, char **argv, trapframe_t *tf)
84 {
85         extern char (RO SNT _start)[], (RO SNT etext)[], (RO SNT edata)[], (RO SNT end)[];
86
87         cprintf("Special kernel symbols:\n");
88         cprintf("  _start %08x (virt)  %08x (phys)\n", _start, (uint32_t)(_start - KERNBASE));
89         cprintf("  etext  %08x (virt)  %08x (phys)\n", etext, (uint32_t)(etext - KERNBASE));
90         cprintf("  edata  %08x (virt)  %08x (phys)\n", edata, (uint32_t)(edata - KERNBASE));
91         cprintf("  end    %08x (virt)  %08x (phys)\n", end, (uint32_t)(end - KERNBASE));
92         cprintf("Kernel executable memory footprint: %dKB\n",
93                 (uint32_t)(end-_start+1023)/1024);
94         return 0;
95 }
96
97 #if 0
98 zra: not called
99 static char RO* function_of(uint32_t address)
100 {
101         extern stab_t (RO stab)[], (RO estab)[];
102         extern char (RO stabstr)[];
103         stab_t* symtab;
104         stab_t* best_symtab = 0;
105         uint32_t best_func = 0;
106
107         // ugly and unsorted
108         for (symtab = stab; symtab < estab; symtab++) {
109                 // only consider functions, type = N_FUN
110                 if ((symtab->n_type == N_FUN) &&
111                     (symtab->n_value <= address) &&
112                         (symtab->n_value > best_func)) {
113                         best_func = symtab->n_value;
114                         best_symtab = symtab;
115                 }
116         }
117         // maybe the first stab really is the right one...  we'll see.
118         if (best_symtab == 0)
119                 return "Function not found!";
120         return stabstr + best_symtab->n_strx;
121 }
122 #endif
123
124 int mon_backtrace(int argc, char **argv, trapframe_t *tf)
125 {
126         backtrace();
127         return 0;
128 }
129
130 int mon_reboot(int argc, char **argv, trapframe_t *tf)
131 {
132         cprintf("[Scottish Accent]: She's goin' down, Cap'n!\n");
133         reboot();
134
135         // really, should never see this
136         cprintf("Sigh....\n");
137         return 0;
138 }
139
140 int mon_showmapping(int argc, char **argv, trapframe_t *tf)
141 {
142         if (argc < 2) {
143                 cprintf("Shows virtual -> physical mappings for a virtual address range.\n");
144                 cprintf("Usage: showmapping START_ADDR [END_ADDR]\n");
145                 return 1;
146         }
147         pde_t* pgdir = (pde_t*)vpd;
148         pte_t *pte, *pde;
149         page_t* page;
150         uintptr_t start, i;
151         size_t size;
152         start = ROUNDDOWN(strtol(argv[1], 0, 16), PGSIZE);
153         size = (argc == 2) ? 1 : strtol(argv[2], 0, 16) - start;
154         if (size/PGSIZE > 512) {
155                 cprintf("Not going to do this for more than 512 items\n");
156                 return 1;
157         }
158
159         show_mapping(start,size);
160         return 0;
161 }
162
163 int mon_setmapperm(int argc, char **argv, trapframe_t *tf)
164 {
165 #ifndef __i386__
166         cprintf("I don't support this call yet!\n");
167         return 1;
168 #else
169         if (argc < 3) {
170                 cprintf("Sets VIRT_ADDR's mapping's permissions to PERMS (in hex)\n");
171                 cprintf("Only affects the lowest level PTE.  To adjust the PDE, do the math.\n");
172                 cprintf("Be careful with this around UVPT, VPT, and friends.\n");
173                 cprintf("Usage: setmapperm VIRT_ADDR PERMS\n");
174                 return 1;
175         }
176         pde_t*COUNT(PTSIZE) pgdir = (pde_t*COUNT(PTSIZE))vpd;
177         pte_t *pte, *pde;
178         page_t* page;
179         uintptr_t va;
180         va = ROUNDDOWN(strtol(argv[1], 0, 16), PGSIZE);
181         page = page_lookup(pgdir, (void*SNT)va, &pte);
182         if (!page) {
183                 cprintf("No such mapping\n");
184                 return 1;
185         }
186         pde = &pgdir[PDX(va)];
187         cprintf("   Virtual    Physical  Ps Dr Ac CD WT U W\n");
188         cprintf("------------------------------------------\n");
189         cprintf("%08p  %08p  %1d  %1d  %1d  %1d  %1d  %1d %1d\n", va, page2pa(page),
190                (*pte & PTE_PS) >> 7, (*pte & PTE_D) >> 6, (*pte & PTE_A) >> 5,
191                (*pte & PTE_PCD) >> 4, (*pte & PTE_PWT) >> 3, (*pte & *pde & PTE_U) >> 2,
192                (*pte & *pde & PTE_W) >> 1);
193         *pte = PTE_ADDR(*pte) | (*pte & PTE_PS) |
194                (PGOFF(strtol(argv[2], 0, 16)) & ~PTE_PS ) | PTE_P;
195         cprintf("%08p  %08p  %1d  %1d  %1d  %1d  %1d  %1d %1d\n", va, page2pa(page),
196                (*pte & PTE_PS) >> 7, (*pte & PTE_D) >> 6, (*pte & PTE_A) >> 5,
197                (*pte & PTE_PCD) >> 4, (*pte & PTE_PWT) >> 3, (*pte & *pde & PTE_U) >> 2,
198                (*pte & *pde & PTE_W) >> 1);
199         return 0;
200 #endif
201 }
202
203 int mon_cpuinfo(int argc, char **argv, trapframe_t *tf)
204 {
205         cprintf("Number of CPUs detected: %d\n", num_cpus);
206         cprintf("Calling CPU's ID: 0x%08x\n", core_id());
207
208         if (argc < 2)
209                 smp_call_function_self(test_print_info_handler, NULL, 0);
210         else
211                 smp_call_function_single(strtol(argv[1], 0, 10),
212                                          test_print_info_handler, NULL, 0);
213         return 0;
214 }
215
216 int mon_manager(int argc, char** argv, trapframe_t *tf)
217 {
218         manager();
219         panic("should never get here");
220         return 0;
221 }
222
223 int mon_nanwan(int argc, char **argv, trapframe_t *tf)
224 {
225         /* Borrowed with love from http://www.geocities.com/SoHo/7373/zoo.htm
226          * (http://www.ascii-art.com/).  Slightly modified to make it 25 lines tall.
227          */
228         printk("\n");
229         printk("             .-.  .-.\n");
230         printk("             |  \\/  |\n");
231         printk("            /,   ,_  `'-.\n");
232         printk("          .-|\\   /`\\     '. \n");
233         printk("        .'  0/   | 0\\  \\_  `\".  \n");
234         printk("     .-'  _,/    '--'.'|#''---'\n");
235         printk("      `--'  |       /   \\#\n");
236         printk("            |      /     \\#\n");
237         printk("            \\     ;|\\    .\\#\n");
238         printk("            |' ' //  \\   ::\\# \n");
239         printk("            \\   /`    \\   ':\\#\n");
240         printk("             `\"`       \\..   \\#\n");
241         printk("                        \\::.  \\#\n");
242         printk("                         \\::   \\#\n");
243         printk("                          \\'  .:\\#\n");
244         printk("                           \\  :::\\#\n");
245         printk("                            \\  '::\\#\n");
246         printk("                             \\     \\#\n");
247         printk("                              \\:.   \\#\n");
248         printk("                               \\::   \\#\n");
249         printk("                                \\'   .\\#\n");
250         printk("                             jgs \\   ::\\#\n");
251         printk("                                  \\      \n");
252         return 0;
253 }
254
255 int mon_bin_ls(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
256 {
257         struct dirent dir = {0};
258         struct file *bin_dir;
259         int retval = 0;
260
261         bin_dir = path_to_file("/bin");
262         if (!bin_dir) {
263                 printk("No /bin directory!\n");
264                 return 1;
265         }
266         printk("Files in KFS /bin:\n-------------------------------\n");
267         do {
268                 retval = bin_dir->f_op->readdir(bin_dir, &dir); 
269                 printk("%s\n", dir.d_name);
270                 dir.d_off++;
271         } while (retval == 1);
272         /* TODO: (REF) need to dealloc when this hits 0, atomic/concurrent/etc */
273         atomic_dec(&bin_dir->f_refcnt);
274         assert(!bin_dir->f_refcnt);
275         bin_dir->f_op->release(bin_dir->f_inode, bin_dir);
276         return 0;
277 }
278
279 int mon_bin_run(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
280 {
281         if (argc != 2) {
282                 printk("Usage: kfs_run FILENAME\n");
283                 return 1;
284         }
285         struct file *program;
286         int retval = 0;
287         char buf[6 + MAX_FILENAME_SZ] = "/bin/";        /* /bin/ + max + \0 */
288         strncpy(buf + 5, argv[1], MAX_FILENAME_SZ);
289         program = path_to_file(buf);
290         if (!program) {
291                 printk("No such program!\n");
292                 return 1;
293         }
294         /* TODO: push this into proc_create */
295         struct proc *p = proc_create(NULL, 0);
296         char* p_argv[] = {0, 0, 0};
297         char* p_envp[] = {"LD_LIBRARY_PATH=/lib",0};
298         p_argv[0] = file_name(program);
299         procinfo_pack_args(p->procinfo, p_argv, p_envp);
300         assert(load_elf(p, program) == 0);
301
302         spin_lock(&p->proc_lock);
303         __proc_set_state(p, PROC_RUNNABLE_S);
304         schedule_proc(p);
305         spin_unlock(&p->proc_lock);
306         proc_decref(p, 1); /* let go of the reference created in proc_create() */
307         /* TODO: (REF) need to dealloc when this hits 0, atomic/concurrent/etc */
308         atomic_dec(&program->f_refcnt);
309         /* Should never return from schedule (env_pop in there) also note you may
310          * not get the process you created, in the event there are others floating
311          * around that are runnable */
312         schedule();
313         return 0;
314 }
315
316 int mon_kfs_cat(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
317 {
318         if (argc != 2) {
319                 printk("Usage: kfs_cat FILENAME\n");
320                 return 1;
321         }
322         int kfs_inode = kfs_lookup_path(argv[1]);
323         if (kfs_inode < 0) {
324                 printk("Bad filename!\n");
325                 return 1;
326         }
327         kfs_cat(kfs_inode);
328         return 0;
329 }
330
331 int mon_procinfo(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
332 {
333         if (argc < 2) {
334                 printk("Usage: procinfo OPTION\n");
335                 printk("\tidlecores: show idle core map\n");
336                 printk("\tall: show all active pids\n");
337                 printk("\trunnable: show proc_runnablelist\n");
338                 printk("\tresources: show resources wanted/granted for all procs\n");
339                 printk("\tpid NUM: show a lot of info for proc NUM\n");
340                 printk("\tunlock: unlock the lock for the ADDR (OMG!!!)\n");
341                 printk("\tkill NUM: destroy proc NUM\n");
342                 return 1;
343         }
344         if (!strcmp(argv[1], "idlecores")) {
345                 print_idlecoremap();
346         } else if (!strcmp(argv[1], "all")) {
347                 print_allpids();
348         } else if (!strcmp(argv[1], "runnable")) {
349                 dump_proclist(&proc_runnablelist);
350         } else if (!strcmp(argv[1], "resources")) {
351                 print_all_resources();
352         } else if (!strcmp(argv[1], "pid")) {
353                 if (argc != 3) {
354                         printk("Give me a pid number.\n");
355                         return 1;
356                 }
357                 print_proc_info(strtol(argv[2], 0, 0));
358         } else if (!strcmp(argv[1], "unlock")) {
359                 if (argc != 3) {
360                         printk("Gimme lock address!  Me want lock address!.\n");
361                         return 1;
362                 }
363                 spinlock_t *lock = (spinlock_t*)strtol(argv[2], 0, 16);
364                 if (!lock) {
365                         printk("Null address...\n");
366                         return 1;
367                 }
368                 spin_unlock(lock);
369         } else if (!strcmp(argv[1], "kill")) {
370                 if (argc != 3) {
371                         printk("Give me a pid number.\n");
372                         return 1;
373                 }
374                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
375                 if (!p) {
376                         printk("No such proc\n");
377                         return 1;
378                 }
379                 proc_destroy(p);
380                 proc_decref(p, 1);
381         } else {
382                 printk("Bad option\n");
383                 return 1;
384         }
385         return 0;
386 }
387
388 int mon_exit(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
389 {
390         return -1;
391 }
392
393 int mon_kfunc(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
394 {
395         #ifndef __i386__
396         printk("Only supported on x86 for now.  =(\n");
397         return -1;
398         #endif
399
400         void (*func)(void *arg, ...);
401
402         if (argc < 2) {
403                 printk("Usage: kfunc FUNCTION [arg1] [arg2] [etc]\n");
404                 printk("Arguments must be in hex.  Can take 6 args.\n");
405                 return 1;
406         }
407         func = debug_get_fn_addr(argv[1]);
408         if (!func) {
409                 printk("Function not found.\n");
410                 return 1;
411         }
412         /* Not elegant, but whatever.  maybe there's a better syntax, or we can do
413          * it with asm magic. */
414         switch (argc) {
415                 case 2: /* have to fake one arg */
416                         func((void*)0);
417                         break;
418                 case 3: /* the real first arg */
419                         func((void*)strtol(argv[2], 0, 16));
420                         break;
421                 case 4:
422                         func((void*)strtol(argv[2], 0, 16),
423                                     strtol(argv[3], 0, 16));
424                         break;
425                 case 5:
426                         func((void*)strtol(argv[2], 0, 16),
427                                     strtol(argv[3], 0, 16),
428                                     strtol(argv[4], 0, 16));
429                         break;
430                 case 6:
431                         func((void*)strtol(argv[2], 0, 16),
432                                     strtol(argv[3], 0, 16),
433                                     strtol(argv[4], 0, 16),
434                                     strtol(argv[5], 0, 16));
435                         break;
436                 case 7:
437                         func((void*)strtol(argv[2], 0, 16),
438                                     strtol(argv[3], 0, 16),
439                                     strtol(argv[4], 0, 16),
440                                     strtol(argv[5], 0, 16),
441                                     strtol(argv[6], 0, 16));
442                         break;
443                 case 8:
444                         func((void*)strtol(argv[2], 0, 16),
445                                     strtol(argv[3], 0, 16),
446                                     strtol(argv[4], 0, 16),
447                                     strtol(argv[5], 0, 16),
448                                     strtol(argv[6], 0, 16),
449                                     strtol(argv[7], 0, 16));
450                         break;
451                 default:
452                         printk("Bad number of arguments.\n");
453                         return -1;
454         }
455         return 0;
456 }
457
458 int mon_notify(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
459 {
460         struct proc *p;
461         unsigned int num;
462         uint32_t vcoreid;
463
464         if (argc < 3) {
465                 printk("Usage: notify PID NUM [VCOREID]\n");
466                 return 1;
467         }
468         p = pid2proc(strtol(argv[1], 0, 0));
469         if (!p) {
470                 printk("No such proc\n");
471                 return 1;
472         }
473         num = strtol(argv[2], 0, 0);
474         if (argc == 4) {
475                 vcoreid = strtol(argv[3], 0, 0);
476                 do_notify(p, vcoreid, num, 0);
477         } else {
478                 proc_notify(p, num, 0);
479         }
480         proc_decref(p, 1);
481         return 0;
482 }
483
484 /* Micro-benchmarky Measurements */
485 int mon_measure(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
486 {
487         uint64_t begin = 0, diff = 0;
488         bool self_ipi_pending = FALSE;
489         uint32_t end_refcnt = 0;
490
491         if (argc < 2) {
492                 printk("Usage: measure OPTION\n");
493                 printk("\tkill PID : kill proc PID\n");
494                 printk("\tpreempt PID : preempt proc PID (no delay)\n");
495                 printk("\tpreempt PID [pcore] : preempt PID's pcore (no delay)\n");
496                 printk("\tpreempt-warn PID : warn-preempt proc PID (pending)\n");
497                 printk("\tpreempt-warn PID [pcore] : warn-preempt proc PID's pcore\n");
498                 printk("\tpreempt-raw PID : raw-preempt proc PID\n");
499                 printk("\tpreempt-raw PID [pcore] : raw-preempt proc PID's pcore\n");
500                 return 1;
501         }
502         if (!strcmp(argv[1], "kill")) {
503                 if (argc < 3) {
504                         printk("Give me a pid number.\n");
505                         return 1;
506                 }
507                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
508                 if (!p) {
509                         printk("No such proc\n");
510                         return 1;
511                 }
512                 begin = start_timing();
513 #ifdef __CONFIG_APPSERVER__
514                 printk("Warning: this will be inaccurate due to the appserver.\n");
515                 end_refcnt = p->env_refcnt - p->procinfo->num_vcores - 1;
516 #endif /* __CONFIG_APPSERVER__ */
517                 proc_destroy(p);
518                 proc_decref(p, 1);
519 #ifdef __CONFIG_APPSERVER__
520                 /* Won't be that accurate, since it's not actually going through the
521                  * __proc_free() path. */
522                 spin_on(p->env_refcnt != end_refcnt);   
523 #else
524                 /* this is a little ghetto. it's not fully free yet, but we are also
525                  * slowing it down by messing with it, esp with the busy waiting on a
526                  * hyperthreaded core. */
527                 spin_on(p->env_cr3);
528 #endif /* __CONFIG_APPSERVER__ */
529                 /* No noticeable difference using stop_timing instead of read_tsc() */
530                 diff = stop_timing(begin);
531         } else if (!strcmp(argv[1], "preempt")) {
532                 if (argc < 3) {
533                         printk("Give me a pid number.\n");
534                         return 1;
535                 }
536                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
537                 if (!p) {
538                         printk("No such proc\n");
539                         return 1;
540                 }
541                 if (argc == 4) { /* single core being preempted, warned but no delay */
542                         uint32_t pcoreid = strtol(argv[3], 0, 0);
543                         begin = start_timing();
544                         proc_preempt_core(p, pcoreid, 1000000); // 1 sec warning
545                         /* done when unmapped (right before abandoning) */
546                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
547                         diff = stop_timing(begin);
548                 } else { /* preempt all cores, warned but no delay */
549                         end_refcnt = p->env_refcnt - p->procinfo->num_vcores;
550                         begin = start_timing();
551                         proc_preempt_all(p, 1000000);
552                         /* a little ghetto, implies no one is using p */
553                         spin_on(p->env_refcnt != end_refcnt);
554                         diff = stop_timing(begin);
555                 }
556                 proc_decref(p, 1);
557         } else if (!strcmp(argv[1], "preempt-warn")) {
558                 if (argc < 3) {
559                         printk("Give me a pid number.\n");
560                         return 1;
561                 }
562                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
563                 if (!p) {
564                         printk("No such proc\n");
565                         return 1;
566                 }
567                 if (argc == 4) { /* single core being preempted-warned */
568                         uint32_t pcoreid = strtol(argv[3], 0, 0);
569                         spin_lock(&p->proc_lock);
570                         uint32_t vcoreid = p->procinfo->pcoremap[pcoreid].vcoreid;
571                         if (!p->procinfo->pcoremap[pcoreid].valid) {
572                                 printk("Pick a mapped pcore\n");
573                                 spin_unlock(&p->proc_lock);
574                                 return 1;
575                         }
576                         begin = start_timing();
577                         __proc_preempt_warn(p, vcoreid, 1000000); // 1 sec
578                         spin_unlock(&p->proc_lock);
579                         /* done when unmapped (right before abandoning) */
580                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
581                         diff = stop_timing(begin);
582                 } else { /* preempt-warn all cores */
583                         printk("Warning, this won't work if they can't yield their "
584                                "last vcore, will stop at 1!\n");
585                         spin_lock(&p->proc_lock);
586                         begin = start_timing();
587                         __proc_preempt_warnall(p, 1000000);
588                         spin_unlock(&p->proc_lock);
589                         /* target cores do the unmapping / changing of the num_vcores */
590                         spin_on(p->procinfo->num_vcores > 1);
591                         diff = stop_timing(begin);
592                 }
593                 proc_decref(p, 1);
594         } else if (!strcmp(argv[1], "preempt-raw")) {
595                 if (argc < 3) {
596                         printk("Give me a pid number.\n");
597                         return 1;
598                 }
599                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
600                 if (!p) {
601                         printk("No such proc\n");
602                         return 1;
603                 }
604                 if (argc == 4) { /* single core preempted, no warning or waiting */
605                         uint32_t pcoreid = strtol(argv[3], 0, 0);
606                         spin_lock(&p->proc_lock);
607                         if (!p->procinfo->pcoremap[pcoreid].valid) {
608                                 printk("Pick a mapped pcore\n");
609                                 spin_unlock(&p->proc_lock);
610                                 return 1;
611                         }
612                         begin = start_timing();
613                         self_ipi_pending = __proc_preempt_core(p, pcoreid);
614                         spin_unlock(&p->proc_lock);
615                         __proc_kmsg_pending(p, self_ipi_pending);
616                         /* done when unmapped (right before abandoning) */
617                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
618                         diff = stop_timing(begin);
619                         /* TODO: (RMS), if num_vcores == 0, RUNNABLE_M, schedule */
620                 } else { /* preempt all cores, no warning or waiting */
621                         spin_lock(&p->proc_lock);
622                         end_refcnt = p->env_refcnt - p->procinfo->num_vcores;
623                         begin = start_timing();
624                         self_ipi_pending = __proc_preempt_all(p);
625                         /* TODO: (RMS), RUNNABLE_M, schedule */
626                         spin_unlock(&p->proc_lock);
627                         __proc_kmsg_pending(p, self_ipi_pending);
628                         /* a little ghetto, implies no one else is using p */
629                         spin_on(p->env_refcnt != end_refcnt);
630                         diff = stop_timing(begin);
631                 }
632                 proc_decref(p, 1);
633         } else {
634                 printk("Bad option\n");
635                 return 1;
636         }
637         printk("[Tired Giraffe Accent] Took %llu usec (%llu nsec) to finish.\n",
638                diff * 1000000 / system_timing.tsc_freq,
639                diff * 1000000000 / system_timing.tsc_freq);
640         return 0;
641 }
642
643 int mon_trace(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
644 {
645         if (argc < 2) {
646                 printk("Usage: trace OPTION\n");
647                 printk("\tsyscall start [silent] [pid]: starts tracing\n");
648                 printk("\tsyscall stop: stops tracing, prints if it was silent\n");
649                 return 1;
650         }
651         if (!strcmp(argv[1], "syscall")) {
652                 if (argc < 3) {
653                         printk("Need a start or stop.\n");
654                         return 1;
655                 }
656                 if (!strcmp(argv[2], "start")) {
657                         bool all = TRUE;
658                         bool silent = FALSE;
659                         struct proc *p = NULL;
660                         if (argc >= 4) {
661                                 silent = (bool)strtol(argv[3], 0, 0);
662                         }
663                         if (argc >= 5) {
664                                 all = FALSE;
665                                 p = pid2proc(strtol(argv[4], 0, 0));
666                                 if (!p) {
667                                         printk("No such process\n");
668                                         return 1;
669                                 }
670                         }
671                         systrace_start(silent);
672                         if (systrace_reg(all, p))
673                                 printk("No room to trace more processes\n");
674                 } else if (!strcmp(argv[2], "stop")) {
675                         /* Stop and print for all processes */
676                         systrace_stop();
677                         systrace_print(TRUE, 0);
678                         systrace_clear_buffer();
679                 }
680         } else if (!strcmp(argv[1], "opt2")) {
681                 if (argc != 3) {
682                         printk("ERRRRRRRRRR.\n");
683                         return 1;
684                 }
685                 print_proc_info(strtol(argv[2], 0, 0));
686         } else {
687                 printk("Bad option\n");
688                 return 1;
689         }
690         return 0;
691 }
692
693 int mon_monitor(int argc, char *NTS *NT COUNT(argc) argv, trapframe_t *tf)
694 {
695         if (argc < 2) {
696                 printk("Usage: monitor COREID\n");
697                 return 1;
698         }
699         uint32_t core = strtol(argv[1], 0, 0);
700         if (core >= num_cpus) {
701                 printk("No such core!  Maybe it's in another cell...\n");
702                 return 1;
703         }
704         void run_mon(struct trapframe *tf, uint32_t srcid, void *a0, void *a1,
705                      void *a2)
706         {
707                 monitor(0); // TODO consider passing the tf
708         }
709         send_kernel_message(core, run_mon, 0, 0, 0, KMSG_ROUTINE);
710         return 0;
711 }
712
713 /***** Kernel monitor command interpreter *****/
714
715 #define WHITESPACE "\t\r\n "
716 #define MAXARGS 16
717
718 static int runcmd(char *NTS real_buf, trapframe_t *tf) {
719         char * buf = NTEXPAND(real_buf);
720         int argc;
721         char *NTS argv[MAXARGS];
722         int i;
723
724         // Parse the command buffer into whitespace-separated arguments
725         argc = 0;
726         argv[argc] = 0;
727         while (1) {
728                 // gobble whitespace
729                 while (*buf && strchr(WHITESPACE, *buf))
730                         *buf++ = 0;
731                 if (*buf == 0)
732                         break;
733
734                 // save and scan past next arg
735                 if (argc == MAXARGS-1) {
736                         cprintf("Too many arguments (max %d)\n", MAXARGS);
737                         return 0;
738                 }
739                 //This will get fucked at runtime..... in the ASS
740                 argv[argc++] = buf;
741                 while (*buf && !strchr(WHITESPACE, *buf))
742                         buf++;
743         }
744         argv[argc] = 0;
745
746         // Lookup and invoke the command
747         if (argc == 0)
748                 return 0;
749         for (i = 0; i < NCOMMANDS; i++) {
750                 if (strcmp(argv[0], commands[i].name) == 0)
751                         return commands[i].func(argc, argv, tf);
752         }
753         cprintf("Unknown command '%s'\n", argv[0]);
754         return 0;
755 }
756
757 void monitor(trapframe_t *tf) {
758         static spinlock_t monitor_lock = SPINLOCK_INITIALIZER;
759         spin_lock_irqsave(&monitor_lock);
760
761         char *buf;
762
763         /* they are always disabled, since we have this irqsave lock */
764         if (irq_is_enabled())
765                 printk("Entering Nanwan's Dungeon on Core %d (Ints on):\n", core_id());
766         else
767                 printk("Entering Nanwan's Dungeon on Core %d (Ints off):\n", core_id());
768         printk("Type 'help' for a list of commands.\n");
769
770         if (tf != NULL)
771                 print_trapframe(tf);
772
773         spin_unlock_irqsave(&monitor_lock);
774         while (1) {
775                 spin_lock_irqsave(&monitor_lock);
776                 buf = readline("ROS(Core %d)> ", core_id());
777                 if (buf != NULL) {
778                         spin_unlock_irqsave(&monitor_lock);
779                         if (runcmd(buf, tf) < 0)
780                                 break;
781                 }
782         }
783 }