Fix leaked proc refcnt in showmapping
[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 #include <arch/arch.h>
5 #include <stab.h>
6 #include <smp.h>
7 #include <arch/console.h>
8
9 #include <stdio.h>
10 #include <string.h>
11 #include <assert.h>
12 #include <monitor.h>
13 #include <trap.h>
14 #include <pmap.h>
15 #include <kdebug.h>
16 #include <testing.h>
17 #include <manager.h>
18 #include <schedule.h>
19 #include <kdebug.h>
20 #include <syscall.h>
21 #include <kmalloc.h>
22 #include <elf.h>
23 #include <event.h>
24 #include <trap.h>
25 #include <time.h>
26 #include <percpu.h>
27
28 #include <ros/memlayout.h>
29 #include <ros/event.h>
30
31 #define CMDBUF_SIZE     80      // enough for one VGA text line
32
33 typedef struct command {
34         const char *name;
35         const char *desc;
36         // return -1 to force monitor to exit
37         int (*func)(int argc, char **argv, struct hw_trapframe *hw_tf);
38 } command_t;
39
40 static command_t commands[] = {
41         { "help", "Display this list of commands", mon_help },
42         { "kerninfo", "Display information about the kernel", mon_kerninfo },
43         { "backtrace", "Dump a backtrace", mon_backtrace },
44         { "bt", "Dump a backtrace", mon_backtrace },
45         { "reboot", "Take a ride to the South Bay", mon_reboot },
46         { "showmapping", "Shows VA->PA mappings", mon_showmapping},
47         { "sm", "Shows VA->PA mappings", mon_sm},
48         { "cpuinfo", "Prints CPU diagnostics", mon_cpuinfo},
49         { "ps", "Prints process list", mon_ps},
50         { "nanwan", "Meet Nanwan!!", mon_nanwan},
51         { "bin_ls", "List files in /bin", mon_bin_ls},
52         { "bin_run", "Create and run a program from /bin", mon_bin_run},
53         { "manager", "Run the manager", mon_manager},
54         { "procinfo", "Show information about processes", mon_procinfo},
55         { "pip", "Shorthand for procinfo pid", mon_pip},
56         { "kill", "Kills a process", mon_kill},
57         { "exit", "Leave the monitor", mon_exit},
58         { "e", "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         { "sh", "Try to run a shell (bash)", mon_shell},
66         { "bash", "Try to run a shell (bash)", mon_shell},
67         { "bb", "Try to run a shell (bash)", mon_shell},
68         { "alarm", "Alarm Diagnostics", mon_alarm},
69         { "msr", "read/write msr: msr msr [value]", mon_msr},
70         { "db", "Misc debugging", mon_db},
71         { "px", "Toggle printx", mon_px},
72         { "kpfret", "Attempt to idle after a kernel fault", mon_kpfret},
73         { "ks", "Kernel scheduler hacks", mon_ks},
74         { "coreinfo", "Print diagnostics for a core", mon_coreinfo},
75 };
76 #define NCOMMANDS (sizeof(commands)/sizeof(commands[0]))
77
78 /***** Implementations of basic kernel monitor commands *****/
79
80 int mon_help(int argc, char **argv, struct hw_trapframe *hw_tf)
81 {
82         int i;
83
84         for (i = 0; i < NCOMMANDS; i++)
85                 cprintf("%s - %s\n", commands[i].name, commands[i].desc);
86         return 0;
87 }
88
89 int mon_ps(int argc, char** argv, struct hw_trapframe *hw_tf)
90 {
91         print_allpids();
92         return 0;
93 }
94
95 int mon_kerninfo(int argc, char **argv, struct hw_trapframe *hw_tf)
96 {
97         extern char _start[], etext[], end[];
98
99         cprintf("Special kernel symbols:\n");
100         cprintf("  _start %016x (virt)  %016x (phys)\n", _start, (uintptr_t)(_start - KERNBASE));
101         cprintf("  etext  %016x (virt)  %016x (phys)\n", etext, (uintptr_t)(etext - 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 static int __backtrace(int argc, char **argv, struct hw_trapframe *hw_tf)
109 {
110         uintptr_t pc, fp;
111         if (argc == 1) {
112                 backtrace();
113                 return 0;
114         }
115         if (argc != 3) {
116                 printk("Need either no arguments, or two (PC and FP) in hex\n");
117                 return 1;
118         }
119         pc = strtol(argv[1], 0, 16);
120         fp = strtol(argv[2], 0, 16);
121         printk("Backtrace from instruction %p, with frame pointer %p\n", pc, fp);
122         backtrace_frame(pc, fp);
123         return 0;
124 }
125
126 int mon_backtrace(int argc, char **argv, struct hw_trapframe *hw_tf)
127 {
128         return __backtrace(argc, argv, hw_tf);
129 }
130
131 int mon_reboot(int argc, char **argv, struct hw_trapframe *hw_tf)
132 {
133         cprintf("[Scottish Accent]: She's goin' down, Cap'n!\n");
134         reboot();
135
136         // really, should never see this
137         cprintf("Sigh....\n");
138         return 0;
139 }
140
141 static int __showmapping(int argc, char **argv, struct hw_trapframe *hw_tf)
142 {
143         struct proc *p = NULL;
144         uintptr_t start;
145         size_t size;
146         pgdir_t pgdir;
147         pid_t pid;
148
149         if (argc < 3) {
150                 printk("Shows virtual -> physical mappings for a virt addr range.\n");
151                 printk("Usage: showmapping PID START_ADDR [END_ADDR]\n");
152                 printk("    PID == 0 for the boot pgdir\n");
153                 return 1;
154         }
155         pid = strtol(argv[1], 0, 10);
156         if (!pid) {
157                 pgdir = boot_pgdir;
158         } else {
159                 p = pid2proc(pid);
160                 if (!p) {
161                         printk("No proc with pid %d\n", pid);
162                         return 1;
163                 }
164                 pgdir = p->env_pgdir;
165         }
166         start = ROUNDDOWN(strtol(argv[2], 0, 16), PGSIZE);
167         size = (argc == 3) ? 1 : strtol(argv[3], 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         show_mapping(pgdir, start, size);
173         if (p)
174                 proc_decref(p);
175         return 0;
176 }
177
178 int mon_showmapping(int argc, char **argv, struct hw_trapframe *hw_tf)
179 {
180         return __showmapping(argc, argv, hw_tf);
181 }
182
183 int mon_sm(int argc, char **argv, struct hw_trapframe *hw_tf)
184 {
185         return __showmapping(argc, argv, hw_tf);
186 }
187
188 static spinlock_t print_info_lock = SPINLOCK_INITIALIZER_IRQSAVE;
189
190 static void print_info_handler(struct hw_trapframe *hw_tf, void *data)
191 {
192         uint64_t tsc = read_tsc();
193
194         spin_lock_irqsave(&print_info_lock);
195         cprintf("----------------------------\n");
196         cprintf("This is Core %d\n", core_id());
197         cprintf("Timestamp = %lld\n", tsc);
198 #ifdef CONFIG_X86
199         cprintf("Hardware core %d\n", hw_core_id());
200         cprintf("MTRR_DEF_TYPE = 0x%08x\n", read_msr(IA32_MTRR_DEF_TYPE));
201         cprintf("MTRR Phys0 Base = 0x%016llx, Mask = 0x%016llx\n",
202                 read_msr(0x200), read_msr(0x201));
203         cprintf("MTRR Phys1 Base = 0x%016llx, Mask = 0x%016llx\n",
204                 read_msr(0x202), read_msr(0x203));
205         cprintf("MTRR Phys2 Base = 0x%016llx, Mask = 0x%016llx\n",
206                 read_msr(0x204), read_msr(0x205));
207         cprintf("MTRR Phys3 Base = 0x%016llx, Mask = 0x%016llx\n",
208                 read_msr(0x206), read_msr(0x207));
209         cprintf("MTRR Phys4 Base = 0x%016llx, Mask = 0x%016llx\n",
210                 read_msr(0x208), read_msr(0x209));
211         cprintf("MTRR Phys5 Base = 0x%016llx, Mask = 0x%016llx\n",
212                 read_msr(0x20a), read_msr(0x20b));
213         cprintf("MTRR Phys6 Base = 0x%016llx, Mask = 0x%016llx\n",
214                 read_msr(0x20c), read_msr(0x20d));
215         cprintf("MTRR Phys7 Base = 0x%016llx, Mask = 0x%016llx\n",
216                 read_msr(0x20e), read_msr(0x20f));
217 #endif // CONFIG_X86
218         cprintf("----------------------------\n");
219         spin_unlock_irqsave(&print_info_lock);
220 }
221
222 static bool print_all_info(void)
223 {
224         cprintf("\nCORE 0 asking all cores to print info:\n");
225         smp_call_function_all(print_info_handler, NULL, 0);
226         cprintf("\nDone!\n");
227         return true;
228 }
229
230 int mon_cpuinfo(int argc, char **argv, struct hw_trapframe *hw_tf)
231 {
232         cprintf("Number of Cores detected: %d\n", num_cores);
233         cprintf("Calling CPU's ID: 0x%08x\n", core_id());
234
235         if (argc < 2)
236                 smp_call_function_self(print_info_handler, NULL, 0);
237         else
238                 smp_call_function_single(strtol(argv[1], 0, 10),
239                                          print_info_handler, NULL, 0);
240         return 0;
241 }
242
243 int mon_manager(int argc, char** argv, struct hw_trapframe *hw_tf)
244 {
245         manager();
246         panic("should never get here");
247         return 0;
248 }
249
250 int mon_nanwan(int argc, char **argv, struct hw_trapframe *hw_tf)
251 {
252         /* Borrowed with love from http://www.geocities.com/SoHo/7373/zoo.htm
253          * (http://www.ascii-art.com/).  Slightly modified to make it 25 lines tall.
254          */
255         printk("\n");
256         printk("             .-.  .-.\n");
257         printk("             |  \\/  |\n");
258         printk("            /,   ,_  `'-.\n");
259         printk("          .-|\\   /`\\     '. \n");
260         printk("        .'  0/   | 0\\  \\_  `\".  \n");
261         printk("     .-'  _,/    '--'.'|#''---'\n");
262         printk("      `--'  |       /   \\#\n");
263         printk("            |      /     \\#\n");
264         printk("            \\     ;|\\    .\\#\n");
265         printk("            |' ' //  \\   ::\\# \n");
266         printk("            \\   /`    \\   ':\\#\n");
267         printk("             `\"`       \\..   \\#\n");
268         printk("                        \\::.  \\#\n");
269         printk("                         \\::   \\#\n");
270         printk("                          \\'  .:\\#\n");
271         printk("                           \\  :::\\#\n");
272         printk("                            \\  '::\\#\n");
273         printk("                             \\     \\#\n");
274         printk("                              \\:.   \\#\n");
275         printk("                               \\::   \\#\n");
276         printk("                                \\'   .\\#\n");
277         printk("                             jgs \\   ::\\#\n");
278         printk("                                  \\      \n");
279         return 0;
280 }
281
282 int mon_bin_ls(int argc, char **argv, struct hw_trapframe *hw_tf)
283 {
284         struct dirent dir = {0};
285         struct file *bin_dir;
286         int retval = 0;
287
288         bin_dir = do_file_open("/bin", O_READ, 0);
289         if (!bin_dir) {
290                 printk("No /bin directory!\n");
291                 return 1;
292         }
293         printk("Files in /bin:\n-------------------------------\n");
294         do {
295                 retval = bin_dir->f_op->readdir(bin_dir, &dir);
296                 printk("%s\n", dir.d_name);
297         } while (retval == 1);
298         kref_put(&bin_dir->f_kref);
299         return 0;
300 }
301
302 int mon_bin_run(int argc, char **argv, struct hw_trapframe *hw_tf)
303 {
304         if (argc < 2) {
305                 printk("Usage: bin_run FILENAME\n");
306                 return 1;
307         }
308         struct file *program;
309         int retval = 0;
310         char buf[5 + MAX_FILENAME_SZ + 1] = "/bin/";    /* /bin/ + max + \0 */
311
312         strlcpy(buf, "/bin/", sizeof(buf));
313         if (strlcat(buf, argv[1], sizeof(buf)) > sizeof(buf)) {
314                 printk("Filename '%s' too long!\n", argv[1]);
315                 return 1;
316         }
317         program = do_file_open(buf, O_READ, 0);
318         if (!program) {
319                 printk("No such program!\n");
320                 return 1;
321         }
322         char **p_argv = kmalloc(sizeof(char*) * argc, 0);       /* bin_run's argc */
323         for (int i = 0; i < argc - 1; i++)
324                 p_argv[i] = argv[i + 1];
325         p_argv[argc - 1] = 0;
326         /* super ugly: we need to stash current, so that proc_create doesn't pick up
327          * on random processes running here and assuming they are the parent */
328         struct proc *old_cur = current;
329         current = 0;
330         struct proc *p = proc_create(program, p_argv, NULL);
331         current = old_cur;
332         kfree(p_argv);
333         proc_wakeup(p);
334         proc_decref(p); /* let go of the reference created in proc_create() */
335         kref_put(&program->f_kref);
336         /* Make a scheduling decision.  You might not get the process you created,
337          * in the event there are others floating around that are runnable */
338         run_scheduler();
339         /* want to idle, so we un the process we just selected.  this is a bit
340          * hackish, but so is the monitor. */
341         smp_idle();
342         assert(0);
343         return 0;
344 }
345
346 int mon_procinfo(int argc, char **argv, struct hw_trapframe *hw_tf)
347 {
348         int verbosity = 0;
349
350         if (argc < 2) {
351                 printk("Usage: procinfo OPTION\n");
352                 printk("\tall: show all active pids\n");
353                 printk("\tpid NUM: show a lot of info for proc NUM\n");
354                 printk("\tunlock: unlock the lock for the ADDR (OMG!!!)\n");
355                 printk("\tkill NUM: destroy proc NUM\n");
356                 return 1;
357         }
358         if (!strcmp(argv[1], "all")) {
359                 print_allpids();
360         } else if (!strcmp(argv[1], "pid")) {
361                 if (argc < 3) {
362                         printk("Give me a pid number.\n");
363                         return 1;
364                 }
365                 if (argc >= 4)
366                         verbosity = strtol(argv[3], 0, 0);
367                 print_proc_info(strtol(argv[2], 0, 0), verbosity);
368         } else if (!strcmp(argv[1], "unlock")) {
369                 if (argc != 3) {
370                         printk("Gimme lock address!  Me want lock address!.\n");
371                         return 1;
372                 }
373                 spinlock_t *lock = (spinlock_t*)strtol(argv[2], 0, 16);
374                 if (!lock) {
375                         printk("Null address...\n");
376                         return 1;
377                 }
378                 spin_unlock(lock);
379         } else if (!strcmp(argv[1], "kill")) {
380                 if (argc != 3) {
381                         printk("Give me a pid number.\n");
382                         return 1;
383                 }
384                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
385                 if (!p) {
386                         printk("No such proc\n");
387                         return 1;
388                 }
389                 proc_destroy(p);
390                 proc_decref(p);
391         } else {
392                 printk("Bad option\n");
393                 return 1;
394         }
395         return 0;
396 }
397
398 int mon_pip(int argc, char **argv, struct hw_trapframe *hw_tf)
399 {
400         int verbosity = 0;
401
402         if (argc < 2) {
403                 printk("Give me a pid number.\n");
404                 return 1;
405         }
406         if (argc >= 3)
407                 verbosity = strtol(argv[2], 0, 0);
408         print_proc_info(strtol(argv[1], 0, 0), verbosity);
409         return 0;
410 }
411
412 int mon_kill(int argc, char **argv, struct hw_trapframe *hw_tf)
413 {
414         struct proc *p;
415
416         if (argc < 2) {
417                 printk("Usage: kill PID\n");
418                 return 1;
419         }
420         p = pid2proc(strtol(argv[1], 0, 0));
421         if (!p) {
422                 printk("No such proc\n");
423                 return 1;
424         }
425         p->exitcode = 1;        /* typical EXIT_FAILURE */
426         proc_destroy(p);
427         proc_decref(p);
428         return 0;
429 }
430
431 int mon_exit(int argc, char **argv, struct hw_trapframe *hw_tf)
432 {
433         return -1;
434 }
435
436 int mon_kfunc(int argc, char **argv, struct hw_trapframe *hw_tf)
437 {
438         long ret;
439         long (*func)(void *arg, ...);
440
441         if (argc < 2) {
442                 printk("Usage: kfunc FUNCTION [arg1] [arg2] [etc]\n");
443                 printk("Use 0x with hex arguments.  Can take 6 args.\n");
444                 return 1;
445         }
446         func = (void*)get_symbol_addr(argv[1]);
447         if (!func) {
448                 printk("Function not found.\n");
449                 return 1;
450         }
451         /* Not elegant, but whatever.  maybe there's a better syntax, or we can do
452          * it with asm magic. */
453         switch (argc) {
454         case 2: /* have to fake one arg */
455                 ret = func((void*)0);
456                 break;
457         case 3: /* the real first arg */
458                 ret = func((void*)strtol(argv[2], 0, 0));
459                 break;
460         case 4:
461                 ret = func((void*)strtol(argv[2], 0, 0),
462                                   strtol(argv[3], 0, 0));
463                 break;
464         case 5:
465                 ret = func((void*)strtol(argv[2], 0, 0),
466                                   strtol(argv[3], 0, 0),
467                                   strtol(argv[4], 0, 0));
468                 break;
469         case 6:
470                 ret = func((void*)strtol(argv[2], 0, 0),
471                                   strtol(argv[3], 0, 0),
472                                   strtol(argv[4], 0, 0),
473                                   strtol(argv[5], 0, 0));
474                 break;
475         case 7:
476                 ret = func((void*)strtol(argv[2], 0, 0),
477                                   strtol(argv[3], 0, 0),
478                                   strtol(argv[4], 0, 0),
479                                   strtol(argv[5], 0, 0),
480                                   strtol(argv[6], 0, 0));
481                 break;
482         case 8:
483                 ret = func((void*)strtol(argv[2], 0, 0),
484                                   strtol(argv[3], 0, 0),
485                                   strtol(argv[4], 0, 0),
486                                   strtol(argv[5], 0, 0),
487                                   strtol(argv[6], 0, 0),
488                                   strtol(argv[7], 0, 0));
489                 break;
490         default:
491                 printk("Bad number of arguments.\n");
492                 return -1;
493         }
494         printk("%s (might have) returned %p\n", argv[1], ret);
495         return 0;
496 }
497
498 /* Sending a vcoreid forces an event and an IPI/notification */
499 int mon_notify(int argc, char **argv, struct hw_trapframe *hw_tf)
500 {
501         struct proc *p;
502         uint32_t vcoreid;
503         struct event_msg msg = {0};
504
505         if (argc < 3) {
506                 printk("Usage: notify PID NUM [VCOREID]\n");
507                 return 1;
508         }
509         p = pid2proc(strtol(argv[1], 0, 0));
510         if (!p) {
511                 printk("No such proc\n");
512                 return 1;
513         }
514         msg.ev_type = strtol(argv[2], 0, 0);
515         if (argc == 4) {
516                 vcoreid = strtol(argv[3], 0, 0);
517                 /* This will go to the private mbox */
518                 post_vcore_event(p, &msg, vcoreid, EVENT_VCORE_PRIVATE);
519                 proc_notify(p, vcoreid);
520         } else {
521                 /* o/w, try and do what they want */
522                 send_kernel_event(p, &msg, 0);
523         }
524         proc_decref(p);
525         return 0;
526 }
527
528 /* Micro-benchmarky Measurements.  This is really fragile code that probably
529  * won't work perfectly, esp as the kernel evolves. */
530 int mon_measure(int argc, char **argv, struct hw_trapframe *hw_tf)
531 {
532         uint64_t begin = 0, diff = 0;
533         uint32_t end_refcnt = 0;
534
535         if (argc < 2) {
536                 printk("Usage: measure OPTION\n");
537                 printk("\tkill PID : kill proc PID\n");
538                 printk("\tpreempt PID : preempt proc PID (no delay)\n");
539                 printk("\tpreempt PID [pcore] : preempt PID's pcore (no delay)\n");
540                 printk("\tpreempt-warn PID : warn-preempt proc PID (pending)\n");
541                 printk("\tpreempt-warn PID [pcore] : warn-preempt proc PID's pcore\n");
542                 printk("\tpreempt-raw PID : raw-preempt proc PID\n");
543                 printk("\tpreempt-raw PID [pcore] : raw-preempt proc PID's pcore\n");
544                 return 1;
545         }
546         if (!strcmp(argv[1], "kill")) {
547                 if (argc < 3) {
548                         printk("Give me a pid number.\n");
549                         return 1;
550                 }
551                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
552                 if (!p) {
553                         printk("No such proc\n");
554                         return 1;
555                 }
556                 begin = start_timing();
557 #ifdef CONFIG_APPSERVER
558                 printk("Warning: this will be inaccurate due to the appserver.\n");
559                 end_refcnt = kref_refcnt(&p->p_kref) - p->procinfo->num_vcores - 1;
560 #endif /* CONFIG_APPSERVER */
561                 proc_destroy(p);
562                 proc_decref(p);
563 #ifdef CONFIG_APPSERVER
564                 /* Won't be that accurate, since it's not actually going through the
565                  * __proc_free() path. */
566                 spin_on(kref_refcnt(&p->p_kref) != end_refcnt);
567 #else
568                 /* this is a little ghetto. it's not fully free yet, but we are also
569                  * slowing it down by messing with it, esp with the busy waiting on a
570                  * hyperthreaded core. */
571                 spin_on(p->env_cr3);
572 #endif /* CONFIG_APPSERVER */
573                 /* No noticeable difference using stop_timing instead of read_tsc() */
574                 diff = stop_timing(begin);
575         } else if (!strcmp(argv[1], "preempt")) {
576                 if (argc < 3) {
577                         printk("Give me a pid number.\n");
578                         return 1;
579                 }
580                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
581                 if (!p) {
582                         printk("No such proc\n");
583                         return 1;
584                 }
585                 if (argc == 4) { /* single core being preempted, warned but no delay */
586                         uint32_t pcoreid = strtol(argv[3], 0, 0);
587                         begin = start_timing();
588                         if (proc_preempt_core(p, pcoreid, 1000000)) {
589                                 __sched_put_idle_core(p, pcoreid);
590                                 /* done when unmapped (right before abandoning) */
591                                 spin_on(p->procinfo->pcoremap[pcoreid].valid);
592                         } else {
593                                 printk("Core %d was not mapped to proc\n", pcoreid);
594                         }
595                         diff = stop_timing(begin);
596                 } else { /* preempt all cores, warned but no delay */
597                         end_refcnt = kref_refcnt(&p->p_kref) - p->procinfo->num_vcores;
598                         begin = start_timing();
599                         proc_preempt_all(p, 1000000);
600                         /* a little ghetto, implies no one is using p */
601                         spin_on(kref_refcnt(&p->p_kref) != end_refcnt);
602                         diff = stop_timing(begin);
603                 }
604                 proc_decref(p);
605         } else if (!strcmp(argv[1], "preempt-warn")) {
606                 if (argc < 3) {
607                         printk("Give me a pid number.\n");
608                         return 1;
609                 }
610                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
611                 if (!p) {
612                         printk("No such proc\n");
613                         return 1;
614                 }
615                 printk("Careful: if this hangs, then the process isn't responding.\n");
616                 if (argc == 4) { /* single core being preempted-warned */
617                         uint32_t pcoreid = strtol(argv[3], 0, 0);
618                         spin_lock(&p->proc_lock);
619                         uint32_t vcoreid = p->procinfo->pcoremap[pcoreid].vcoreid;
620                         if (!p->procinfo->pcoremap[pcoreid].valid) {
621                                 printk("Pick a mapped pcore\n");
622                                 spin_unlock(&p->proc_lock);
623                                 return 1;
624                         }
625                         begin = start_timing();
626                         __proc_preempt_warn(p, vcoreid, 1000000); // 1 sec
627                         spin_unlock(&p->proc_lock);
628                         /* done when unmapped (right before abandoning) */
629                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
630                         diff = stop_timing(begin);
631                 } else { /* preempt-warn all cores */
632                         printk("Warning, this won't work if they can't yield their "
633                                "last vcore, will stop at 1!\n");
634                         spin_lock(&p->proc_lock);
635                         begin = start_timing();
636                         __proc_preempt_warnall(p, 1000000);
637                         spin_unlock(&p->proc_lock);
638                         /* target cores do the unmapping / changing of the num_vcores */
639                         spin_on(p->procinfo->num_vcores > 1);
640                         diff = stop_timing(begin);
641                 }
642                 proc_decref(p);
643         } else if (!strcmp(argv[1], "preempt-raw")) {
644                 if (argc < 3) {
645                         printk("Give me a pid number.\n");
646                         return 1;
647                 }
648                 struct proc *p = pid2proc(strtol(argv[2], 0, 0));
649                 if (!p) {
650                         printk("No such proc\n");
651                         return 1;
652                 }
653                 if (argc == 4) { /* single core preempted, no warning or waiting */
654                         uint32_t pcoreid = strtol(argv[3], 0, 0);
655                         spin_lock(&p->proc_lock);
656                         if (!p->procinfo->pcoremap[pcoreid].valid) {
657                                 printk("Pick a mapped pcore\n");
658                                 spin_unlock(&p->proc_lock);
659                                 return 1;
660                         }
661                         begin = start_timing();
662                         __proc_preempt_core(p, pcoreid);
663                         if (!p->procinfo->num_vcores)
664                                 __proc_set_state(p, PROC_RUNNABLE_M);
665                         spin_unlock(&p->proc_lock);
666                         /* ghetto, since the ksched should be calling all of this */
667                         __sched_put_idle_core(p, pcoreid);
668                         /* done when unmapped (right before abandoning) */
669                         spin_on(p->procinfo->pcoremap[pcoreid].valid);
670                         diff = stop_timing(begin);
671                 } else { /* preempt all cores, no warning or waiting */
672                         spin_lock(&p->proc_lock);
673                         uint32_t pc_arr[p->procinfo->num_vcores];
674                         uint32_t num_revoked;
675                         end_refcnt = kref_refcnt(&p->p_kref) - p->procinfo->num_vcores;
676                         begin = start_timing();
677                         num_revoked = __proc_preempt_all(p, pc_arr);
678                         __proc_set_state(p, PROC_RUNNABLE_M);
679                         spin_unlock(&p->proc_lock);
680                         if (num_revoked)
681                                 __sched_put_idle_cores(p, pc_arr, num_revoked);
682                         /* a little ghetto, implies no one else is using p */
683                         spin_on(kref_refcnt(&p->p_kref) != end_refcnt);
684                         diff = stop_timing(begin);
685                 }
686                 proc_decref(p);
687         } else {
688                 printk("Bad option\n");
689                 return 1;
690         }
691         printk("[Tired Giraffe Accent] Took %llu usec (%llu nsec) to finish.\n",
692                tsc2usec(diff), tsc2nsec(diff));
693         return 0;
694 }
695
696 static bool mon_verbose_trace = FALSE;
697 static DEFINE_PERCPU(bool, mon_nmi_trace);
698
699 static void emit_hwtf_backtrace(struct hw_trapframe *hw_tf)
700 {
701         char *fn_name;
702
703         if (mon_verbose_trace) {
704                 print_trapframe(hw_tf);
705                 backtrace_hwtf(hw_tf);
706         }
707         fn_name = get_fn_name(get_hwtf_pc(hw_tf));
708         printk("Core %d is at %p (%s)\n", core_id(), get_hwtf_pc(hw_tf),
709                fn_name);
710         kfree(fn_name);
711 }
712
713 static void emit_vmtf_backtrace(struct vm_trapframe *vm_tf)
714 {
715         if (mon_verbose_trace)
716                 print_vmtrapframe(vm_tf);
717         printk("Core %d is at %p\n", core_id(), get_vmtf_pc(vm_tf));
718 }
719
720 /* This is dangerous and could cause a deadlock, since it runs in NMI context.
721  * It's only for monitor debugging, so YMMV.  We pass the type since the kernel
722  * doesn't deal in contexts (yet) */
723 void emit_monitor_backtrace(int type, void *tf)
724 {
725         struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
726
727         if (!PERCPU_VAR(mon_nmi_trace))
728                 return;
729         /* To prevent a spew of output during a lot of perf NMIs, we'll turn off the
730          * monitor output as soon as any NMI hits our core. */
731         PERCPU_VAR(mon_nmi_trace) = FALSE;
732         /* Temporarily disable deadlock detection when we print.  We could
733          * deadlock if we were printing when we NMIed. */
734         pcpui->__lock_checking_enabled--;
735         if (type == ROS_HW_CTX)
736                 emit_hwtf_backtrace((struct hw_trapframe*)tf);
737         else
738                 emit_vmtf_backtrace((struct vm_trapframe*)tf);
739         print_kmsgs(core_id());
740         pcpui->__lock_checking_enabled++;
741 }
742
743
744 int mon_trace(int argc, char **argv, struct hw_trapframe *hw_tf)
745 {
746         int core;
747         if (argc < 2) {
748                 printk("Usage: trace OPTION\n");
749                 printk("\tsyscall start [silent (0 or non-zero, NOT the word silent)] [pid]: starts tracing\n");
750                 printk("\tsyscall stop: stops tracing.\n");
751                 printk("\tcoretf COREID: prints PC, -1 for all cores, verbose => TF\n");
752                 printk("\tpcpui [type [coreid]]: runs pcpui trace ring handlers\n");
753                 printk("\tpcpui-reset [noclear]: resets/clears pcpui trace ring\n");
754                 printk("\tverbose: toggles verbosity, depends on trace command\n");
755                 return 1;
756         }
757         if (!strcmp(argv[1], "syscall")) {
758                 if (argc < 3) {
759                         printk("Need a start or stop.\n");
760                         return 1;
761                 }
762                 if (!strcmp(argv[2], "start")) {
763                         systrace_loud = TRUE;
764                 } else if (!strcmp(argv[2], "stop")) {
765                         systrace_loud = FALSE;
766                 } else {
767                         printk("Need a start or stop.\n");
768                         return 1;
769                 }
770         } else if (!strcmp(argv[1], "coretf")) {
771                 if (argc != 3) {
772                         printk("Need a coreid, fool.\n");
773                         return 1;
774                 }
775                 core = strtol(argv[2], 0, 0);
776                 if (core < 0) {
777                         printk("Sending NMIs to all cores:\n");
778                         for (int i = 0; i < num_cores; i++) {
779                                 _PERCPU_VAR(mon_nmi_trace, i) = TRUE;
780                                 send_nmi(i);
781                                 udelay(1000000);
782                         }
783                 } else {
784                         printk("Sending NMI core %d:\n", core);
785                         if (core >= num_cores) {
786                                 printk("No such core!  Maybe it's in another cell...\n");
787                                 return 1;
788                         }
789                         _PERCPU_VAR(mon_nmi_trace, core) = TRUE;
790                         send_nmi(core);
791                 }
792                 udelay(1000000);
793         } else if (!strcmp(argv[1], "pcpui")) {
794                 int pcpui_type, pcpui_coreid;
795                 if (argc >= 3)
796                         pcpui_type = strtol(argv[2], 0, 0);
797                 else
798                         pcpui_type = 0;
799                 printk("\nRunning PCPUI Trace Ring handlers for type %d\n", pcpui_type);
800                 if (argc >= 4) {
801                         pcpui_coreid = strtol(argv[3], 0, 0);
802                         pcpui_tr_foreach(pcpui_coreid, pcpui_type);
803                 } else {
804                         pcpui_tr_foreach_all(pcpui_type);
805                 }
806         } else if (!strcmp(argv[1], "pcpui-reset")) {
807                 if (argc >= 3) {
808                         printk("\nResetting all PCPUI Trace Rings\n");
809                         pcpui_tr_reset_all();
810                 } else {
811                         printk("\nResetting and clearing all PCPUI Trace Rings\n");
812                         pcpui_tr_reset_and_clear_all();
813                 }
814         } else if (!strcmp(argv[1], "verbose")) {
815                 if (mon_verbose_trace) {
816                         printk("Turning trace verbosity off\n");
817                         mon_verbose_trace = FALSE;
818                 } else {
819                         printk("Turning trace verbosity on\n");
820                         mon_verbose_trace = TRUE;
821                 }
822         } else if (!strcmp(argv[1], "opt2")) {
823                 if (argc != 3) {
824                         printk("ERRRRRRRRRR.\n");
825                         return 1;
826                 }
827                 print_proc_info(strtol(argv[2], 0, 0), 0);
828         } else {
829                 printk("Bad option\n");
830                 return 1;
831         }
832         return 0;
833 }
834
835 int mon_monitor(int argc, char **argv, struct hw_trapframe *hw_tf)
836 {
837         if (argc < 2) {
838                 printk("Usage: monitor COREID\n");
839                 return 1;
840         }
841         uint32_t core = strtol(argv[1], 0, 0);
842         if (core >= num_cores) {
843                 printk("No such core!  Maybe it's in another cell...\n");
844                 return 1;
845         }
846         send_kernel_message(core, __run_mon, 0, 0, 0, KMSG_ROUTINE);
847         return 0;
848 }
849
850 /***** Kernel monitor command interpreter *****/
851
852 #define WHITESPACE "\t\r\n "
853 #define MAXARGS 16
854
855
856 int onecmd(int argc, char *argv[], struct hw_trapframe *hw_tf) {
857         int i;
858         if (!argc)
859                 return -1;
860         for (i = 0; i < NCOMMANDS; i++) {
861                 if (strcmp(argv[0], commands[i].name) == 0)
862                         return commands[i].func(argc, argv, hw_tf);
863         }
864         return -1;
865 }
866
867 void __run_mon(uint32_t srcid, long a0, long a1, long a2)
868 {
869         monitor(0);
870 }
871
872 static int runcmd(char *real_buf, struct hw_trapframe *hw_tf) {
873         char * buf = real_buf;
874         int argc;
875         char *argv[MAXARGS];
876         int i;
877
878         // Parse the command buffer into whitespace-separated arguments
879         argc = 0;
880         argv[argc] = 0;
881         while (1) {
882                 // gobble whitespace
883                 while (*buf && strchr(WHITESPACE, *buf))
884                         *buf++ = 0;
885                 if (*buf == 0)
886                         break;
887
888                 // save and scan past next arg
889                 if (argc == MAXARGS-1) {
890                         cprintf("Too many arguments (max %d)\n", MAXARGS);
891                         return 0;
892                 }
893                 //This will get fucked at runtime..... in the ASS
894                 argv[argc++] = buf;
895                 while (*buf && !strchr(WHITESPACE, *buf))
896                         buf++;
897         }
898         argv[argc] = 0;
899
900         // Lookup and invoke the command
901         if (argc == 0)
902                 return 0;
903         for (i = 0; i < NCOMMANDS; i++) {
904                 if (strcmp(argv[0], commands[i].name) == 0)
905                         return commands[i].func(argc, argv, hw_tf);
906         }
907         cprintf("Unknown command '%s'\n", argv[0]);
908         return 0;
909 }
910
911 void monitor(struct hw_trapframe *hw_tf)
912 {
913         #define MON_CMD_LENGTH 256
914         char buf[MON_CMD_LENGTH];
915         int cnt;
916         int coreid = core_id_early();
917
918         /* they are always disabled, since we have this irqsave lock */
919         if (irq_is_enabled())
920                 printk("Entering Nanwan's Dungeon on Core %d (Ints on):\n", coreid);
921         else
922                 printk("Entering Nanwan's Dungeon on Core %d (Ints off):\n", coreid);
923         printk("Type 'help' for a list of commands.\n");
924
925         if (hw_tf != NULL)
926                 print_trapframe(hw_tf);
927
928         while (1) {
929                 /* on occasion, the kernel monitor can migrate (like if you run
930                  * something that blocks / syncs and wakes up on another core) */
931                 cmb();
932                 cnt = readline(buf, MON_CMD_LENGTH, "ROS(Core %d)> ", core_id_early());
933                 if (cnt > 0) {
934                         buf[cnt] = 0;
935                         if (runcmd(buf, hw_tf) < 0)
936                                 break;
937                 }
938         }
939 }
940
941 static void pm_flusher(void *unused)
942 {
943         struct super_block *sb;
944         struct inode *inode;
945         unsigned long nr_pages;
946
947         /* could also put the delay between calls, or even within remove, during the
948          * WB phase. */
949         printk("GIANT WARNING: the pm_flusher is running and will never stop!\n");
950         while (1) {
951                 kthread_usleep(5000);
952                 TAILQ_FOREACH(sb, &super_blocks, s_list) {
953                         TAILQ_FOREACH(inode, &sb->s_inodes, i_sb_list) {
954                                 nr_pages = ROUNDUP(inode->i_size, PGSIZE) >> PGSHIFT;
955                                 if (nr_pages)
956                                         pm_remove_contig(inode->i_mapping, 0, nr_pages);
957                         }
958                 }
959         }
960 }
961
962 int mon_fs(int argc, char **argv, struct hw_trapframe *hw_tf)
963 {
964         /* this assumes one mounted FS at the NS root */
965         struct super_block *sb;
966         struct file *file;
967         struct inode *inode;
968         struct dentry *dentry;
969         if (argc < 2) {
970                 printk("Usage: fs OPTION\n");
971                 printk("\topen: show all open files\n");
972                 printk("\tinodes: show all inodes\n");
973                 printk("\tdentries [lru|prune]: show all dentries, opt LRU/prune\n");
974                 printk("\tls DIR: print the dir tree starting with DIR\n");
975                 printk("\tpid: proc PID's fs crap placeholder\n");
976                 printk("\tpmflusher: start a ktask to keep flushing all PMs\n");
977                 return 1;
978         }
979         if (!strcmp(argv[1], "open")) {
980                 printk("Open Files:\n----------------------------\n");
981                 TAILQ_FOREACH(sb, &super_blocks, s_list) {
982                         printk("Superblock for %s\n", sb->s_name);
983                         TAILQ_FOREACH(file, &sb->s_files, f_list)
984                                 printk("File: %p, %s, Refs: %d, Drefs: %d, Irefs: %d PM: %p\n",
985                                        file, file_name(file), kref_refcnt(&file->f_kref),
986                                        kref_refcnt(&file->f_dentry->d_kref),
987                                        kref_refcnt(&file->f_dentry->d_inode->i_kref),
988                                            file->f_mapping);
989                 }
990         } else if (!strcmp(argv[1], "inodes")) {
991                 printk("Mounted FS Inodes:\n----------------------------\n");
992                 TAILQ_FOREACH(sb, &super_blocks, s_list) {
993                         printk("Superblock for %s\n", sb->s_name);
994                         TAILQ_FOREACH(inode, &sb->s_inodes, i_sb_list) {
995                                 printk("Inode: %p, Refs: %d, Nlinks: %d, Size(B): %d\n",
996                                        inode, kref_refcnt(&inode->i_kref), inode->i_nlink,
997                                        inode->i_size);
998                                 TAILQ_FOREACH(dentry, &inode->i_dentry, d_alias)
999                                         printk("\t%s: Dentry: %p, Refs: %d\n",
1000                                                dentry->d_name.name, dentry,
1001                                                kref_refcnt(&dentry->d_kref));
1002                         }
1003                 }
1004         } else if (!strcmp(argv[1], "dentries")) {
1005                 printk("Dentry Cache:\n----------------------------\n");
1006                 TAILQ_FOREACH(sb, &super_blocks, s_list) {
1007                         printk("Superblock for %s\n", sb->s_name);
1008                         printk("DENTRY     FLAGS      REFCNT NAME\n");
1009                         printk("--------------------------------\n");
1010                         /* Hash helper */
1011                         void print_dcache_entry(void *item, void *opaque)
1012                         {
1013                                 struct dentry *d_i = (struct dentry*)item;
1014                                 printk("%p %p %02d     %s\n", d_i, d_i->d_flags,
1015                                        kref_refcnt(&d_i->d_kref), d_i->d_name.name);
1016                         }
1017                         hash_for_each(sb->s_dcache, print_dcache_entry, NULL);
1018                 }
1019                 if (argc < 3)
1020                         return 0;
1021                 if (!strcmp(argv[2], "lru")) {
1022                         printk("LRU lists:\n");
1023                         TAILQ_FOREACH(sb, &super_blocks, s_list) {
1024                                 printk("Superblock for %s\n", sb->s_name);
1025                                 TAILQ_FOREACH(dentry, &sb->s_lru_d, d_lru)
1026                                         printk("Dentry: %p, Name: %s\n", dentry,
1027                                                dentry->d_name.name);
1028                         }
1029                 } else if (!strcmp(argv[2], "prune")) {
1030                         printk("Pruning unused dentries\n");
1031                         TAILQ_FOREACH(sb, &super_blocks, s_list)
1032                                 dcache_prune(sb, FALSE);
1033                 }
1034         } else if (!strcmp(argv[1], "ls")) {
1035                 if (argc != 3) {
1036                         printk("Give me a dir.\n");
1037                         return 1;
1038                 }
1039                 if (argv[2][0] != '/') {
1040                         printk("Dear fellow giraffe lover, Use absolute paths.\n");
1041                         return 1;
1042                 }
1043                 ls_dash_r(argv[2]);
1044                 /* whatever.  placeholder. */
1045         } else if (!strcmp(argv[1], "pid")) {
1046                 if (argc != 3) {
1047                         printk("Give me a pid number.\n");
1048                         return 1;
1049                 }
1050                 /* whatever.  placeholder. */
1051         } else if (!strcmp(argv[1], "pmflusher")) {
1052                 ktask("pm_flusher", pm_flusher, 0);
1053         } else {
1054                 printk("Bad option\n");
1055                 return 1;
1056         }
1057         return 0;
1058 }
1059
1060 int mon_shell(int argc, char **argv, struct hw_trapframe *hw_tf)
1061 {
1062         char *l_argv[2] = {"/bin/bash", "bash"};
1063         return mon_bin_run(2, l_argv, hw_tf);
1064 }
1065
1066 int mon_alarm(int argc, char **argv, struct hw_trapframe *hw_tf)
1067 {
1068         if (argc < 2) {
1069                 printk("Usage: alarm OPTION\n");
1070                 printk("\tpcpu: print full alarm tchains from every core\n");
1071                 return 1;
1072         }
1073         if (!strcmp(argv[1], "pcpu")) {
1074                 print_pcpu_chains();
1075         } else {
1076                 printk("Bad option\n");
1077                 return 1;
1078         }
1079         return 0;
1080 }
1081
1082 static void show_msr(struct hw_trapframe *unused, void *v)
1083 {
1084         int core = core_id();
1085         uint64_t val;
1086         uint32_t msr = *(uint32_t *)v;
1087         val = read_msr(msr);
1088         printk("%d: %08x: %016llx\n", core, msr, val);
1089 }
1090
1091 struct set {
1092         uint32_t msr;
1093         uint64_t val;
1094 };
1095
1096 static void set_msr(struct hw_trapframe *unused, void *v)
1097 {
1098         int core = core_id();
1099         struct set *s = v;
1100         uint32_t msr = s->msr;
1101         uint64_t val = s->val;
1102         write_msr(msr, val);
1103         val = read_msr(msr);
1104         printk("%d: %08x: %016llx\n", core, msr, val);
1105 }
1106
1107 int mon_msr(int argc, char **argv, struct hw_trapframe *hw_tf)
1108 {
1109 #ifndef CONFIG_X86
1110         cprintf("Not on this architecture\n");
1111         return 1;
1112 #else
1113         uint64_t val;
1114         uint32_t msr;
1115         if (argc < 2 || argc > 3) {
1116                 printk("Usage: msr register [value]\n");
1117                 return 1;
1118         }
1119         msr = strtoul(argv[1], 0, 16);
1120         handler_wrapper_t *w;
1121         smp_call_function_all(show_msr, &msr, &w);
1122         smp_call_wait(w);
1123
1124         if (argc < 3)
1125                 return 0;
1126         /* somewhat bogus on 32 bit. */
1127         val = strtoul(argv[2], 0, 16);
1128
1129         struct set set;
1130         set.msr = msr;
1131         set.val = val;
1132         smp_call_function_all(set_msr, &set, &w);
1133         smp_call_wait(w);
1134         return 0;
1135 #endif
1136 }
1137
1138 int mon_db(int argc, char **argv, struct hw_trapframe *hw_tf)
1139 {
1140         pid_t pid = -1;
1141
1142         if (argc < 2) {
1143                 printk("Usage: db OPTION\n");
1144                 printk("\tsem [PID]: print all semaphore info\n");
1145                 printk("\taddr PID 0xADDR: for PID lookup ADDR's file/vmr info\n");
1146                 return 1;
1147         }
1148         if (!strcmp(argv[1], "sem")) {
1149                 if (argc > 2)
1150                         pid = strtol(argv[2], 0, 0);
1151                 print_all_sem_info(pid);
1152         } else if (!strcmp(argv[1], "addr")) {
1153                 if (argc < 4) {
1154                         printk("Usage: db addr PID 0xADDR\n");
1155                         return 1;
1156                 }
1157                 debug_addr_pid(strtol(argv[2], 0, 10), strtol(argv[3], 0, 16));
1158         } else {
1159                 printk("Bad option\n");
1160                 return 1;
1161         }
1162         return 0;
1163 }
1164
1165 int mon_px(int argc, char **argv, struct hw_trapframe *hw_tf)
1166 {
1167         set_printx(2);
1168         printk("Printxing is now %sabled\n", printx_on ? "en" : "dis");
1169         return 0;
1170 }
1171
1172 /* Super hack.  Given a kernel hw_tf, we hack the RIP to smp_idle, then return
1173  * to it.  Any locks or other stuff being done is completely lost, so you could
1174  * deadlock.  This gets out of the "we're totall screwed, but don't want to
1175  * reboot right now", typically caused by screw-ups from the monitor. */
1176 int mon_kpfret(int argc, char **argv, struct hw_trapframe *hw_tf)
1177 {
1178         struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
1179
1180         /* if monitor had a TF, try to use that */
1181         if (!hw_tf) {
1182                 if (argc < 2) {
1183                         printk("Usage: kpfret HW_TF\n");
1184                         return 1;
1185                 }
1186                 /* the hw_tf passed in is the one we got from monitor, which is 0 from
1187                  * panics. */
1188                 hw_tf = (struct hw_trapframe*)strtol(argv[1], 0, 16);
1189         }
1190
1191         if (!in_kernel(hw_tf)) {
1192                 printk("hw_tf %p was not a kernel tf!\n", hw_tf);
1193                 return -1;
1194         }
1195
1196 #ifdef CONFIG_X86
1197         hw_tf->tf_rip = (uintptr_t)smp_idle;
1198         dec_ktrap_depth(pcpui);
1199
1200         asm volatile("mov %0, %%rsp;"
1201                      "addq $0x10, %%rsp;"
1202                      "popq %%rax;"
1203                      "popq %%rbx;"
1204                      "popq %%rcx;"
1205                      "popq %%rdx;"
1206                      "popq %%rbp;"
1207                      "popq %%rsi;"
1208                      "popq %%rdi;"
1209                      "popq %%r8;"
1210                      "popq %%r9;"
1211                      "popq %%r10;"
1212                      "popq %%r11;"
1213                      "popq %%r12;"
1214                      "popq %%r13;"
1215                      "popq %%r14;"
1216                      "popq %%r15;"
1217                      "addq $0x10, %%rsp;"
1218                      "iretq;"
1219                                  : : "r"(hw_tf));
1220         assert(0);
1221 #else
1222         printk("KPF return not supported\n");
1223         return -1;
1224 #endif /* CONFIG_X86 */
1225 }
1226
1227 int mon_ks(int argc, char **argv, struct hw_trapframe *hw_tf)
1228 {
1229         if (argc < 2) {
1230 usage:
1231                 printk("Usage: ks OPTION\n");
1232                 printk("\tidles: show idle core map\n");
1233                 printk("\tdiag: scheduler diagnostic report\n");
1234                 printk("\tresources: show resources wanted/granted for all procs\n");
1235                 printk("\tsort: sorts the idlecoremap, 1..n\n");
1236                 printk("\tnc PCOREID: sets the next CG core allocated\n");
1237                 return 1;
1238         }
1239         if (!strcmp(argv[1], "idles")) {
1240                 print_idle_core_map();
1241         } else if (!strcmp(argv[1], "diag")) {
1242                 sched_diag();
1243         } else if (!strcmp(argv[1], "resources")) {
1244                 print_all_resources();
1245         } else if (!strcmp(argv[1], "sort")) {
1246                 sort_idle_cores();
1247         } else if (!strcmp(argv[1], "nc")) {
1248                 if (argc != 3) {
1249                         printk("Need a pcore number.\n");
1250                         return 1;
1251                 }
1252                 next_core_to_alloc(strtol(argv[2], 0, 0));
1253         } else {
1254                 printk("Bad option %s\n", argv[1]);
1255                 goto usage;
1256         }
1257         return 0;
1258 }
1259
1260 /* Prints info about a core.  Optional first arg == coreid. */
1261 int mon_coreinfo(int argc, char **argv, struct hw_trapframe *hw_tf)
1262 {
1263         struct per_cpu_info *pcpui;
1264         struct kthread *kth;
1265         int coreid = core_id();
1266
1267         if (argc >= 2)
1268                 coreid = strtol(argv[1], 0, 0);
1269         pcpui = &per_cpu_info[coreid];
1270         printk("Core %d:\n\tcur_proc %d\n\towning proc %d, owning vc %d\n",
1271                coreid, pcpui->cur_proc ? pcpui->cur_proc->pid : 0,
1272                pcpui->owning_proc ? pcpui->owning_proc->pid : 0,
1273                pcpui->owning_vcoreid != 0xdeadbeef ? pcpui->owning_vcoreid : 0);
1274         kth = pcpui->cur_kthread;
1275         if (kth) {
1276                 /* kth->proc is only used when the kthread is sleeping.  when it's
1277                  * running, we care about cur_proc.  if we're here, proc should be 0
1278                  * unless the kth is concurrently sleeping (we called this remotely) */
1279                 printk("\tkthread %p (%s), sysc %p (%d)\n", kth, kth->name,
1280                        kth->sysc, kth->sysc ? kth->sysc->num : -1);
1281         } else {
1282                 /* Can happen during early boot */
1283                 printk("\tNo kthread!\n");
1284         }
1285         return 0;
1286 }