mconsole_kern.c 20.6 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/*
 * Copyright (C) 2001 Lennert Buytenhek (buytenh@gnu.org)
 * Copyright (C) 2001 - 2003 Jeff Dike (jdike@addtoit.com)
 * Licensed under the GPL
 */

#include "linux/kernel.h"
#include "linux/slab.h"
#include "linux/init.h"
#include "linux/notifier.h"
#include "linux/reboot.h"
#include "linux/utsname.h"
#include "linux/ctype.h"
#include "linux/interrupt.h"
#include "linux/sysrq.h"
#include "linux/workqueue.h"
#include "linux/module.h"
#include "linux/file.h"
#include "linux/fs.h"
#include "linux/namei.h"
#include "linux/proc_fs.h"
#include "linux/syscalls.h"
J
Jeff Dike 已提交
23 24
#include "linux/list.h"
#include "linux/mm.h"
25
#include "linux/console.h"
L
Linus Torvalds 已提交
26 27 28 29 30 31 32 33 34 35 36 37
#include "asm/irq.h"
#include "asm/uaccess.h"
#include "user_util.h"
#include "kern_util.h"
#include "kern.h"
#include "mconsole.h"
#include "mconsole_kern.h"
#include "irq_user.h"
#include "init.h"
#include "os.h"
#include "umid.h"
#include "irq_kern.h"
38
#include "choose-mode.h"
L
Linus Torvalds 已提交
39

J
Jeff Dike 已提交
40
static int do_unlink_socket(struct notifier_block *notifier,
L
Linus Torvalds 已提交
41 42 43 44 45 46 47 48 49 50 51
			    unsigned long what, void *data)
{
	return(mconsole_unlink_socket());
}


static struct notifier_block reboot_notifier = {
	.notifier_call		= do_unlink_socket,
	.priority		= 0,
};

J
Jeff Dike 已提交
52
/* Safe without explicit locking for now.  Tasklets provide their own
L
Linus Torvalds 已提交
53 54 55 56
 * locking, and the interrupt handler is safe because it can't interrupt
 * itself and it can only happen on CPU 0.
 */

57
static LIST_HEAD(mc_requests);
L
Linus Torvalds 已提交
58

59
static void mc_work_proc(struct work_struct *unused)
L
Linus Torvalds 已提交
60 61 62 63 64
{
	struct mconsole_entry *req;
	unsigned long flags;

	while(!list_empty(&mc_requests)){
65
		local_irq_save(flags);
J
Jeff Dike 已提交
66
		req = list_entry(mc_requests.next, struct mconsole_entry,
L
Linus Torvalds 已提交
67 68 69 70 71 72 73 74
				 list);
		list_del(&req->list);
		local_irq_restore(flags);
		req->request.cmd->handler(&req->request);
		kfree(req);
	}
}

75
static DECLARE_WORK(mconsole_work, mc_work_proc);
L
Linus Torvalds 已提交
76

A
Al Viro 已提交
77
static irqreturn_t mconsole_interrupt(int irq, void *dev_id)
L
Linus Torvalds 已提交
78 79 80 81
{
	/* long to avoid size mismatch warnings from gcc */
	long fd;
	struct mconsole_entry *new;
A
Al Viro 已提交
82
	static struct mc_request req;	/* that's OK */
L
Linus Torvalds 已提交
83 84 85 86 87 88

	fd = (long) dev_id;
	while (mconsole_get_request(fd, &req)){
		if(req.cmd->context == MCONSOLE_INTR)
			(*req.cmd->handler)(&req);
		else {
J
Jeff Dike 已提交
89
			new = kmalloc(sizeof(*new), GFP_NOWAIT);
L
Linus Torvalds 已提交
90 91 92 93
			if(new == NULL)
				mconsole_reply(&req, "Out of memory", 1, 0);
			else {
				new->request = req;
A
Al Viro 已提交
94
				new->request.regs = get_irq_regs()->regs;
L
Linus Torvalds 已提交
95 96 97 98 99 100 101 102 103 104 105 106 107 108
				list_add(&new->list, &mc_requests);
			}
		}
	}
	if(!list_empty(&mc_requests))
		schedule_work(&mconsole_work);
	reactivate_fd(fd, MCONSOLE_IRQ);
	return(IRQ_HANDLED);
}

void mconsole_version(struct mc_request *req)
{
	char version[256];

109 110 111
	sprintf(version, "%s %s %s %s %s", utsname()->sysname,
		utsname()->nodename, utsname()->release,
		utsname()->version, utsname()->machine);
L
Linus Torvalds 已提交
112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277
	mconsole_reply(req, version, 0, 0);
}

void mconsole_log(struct mc_request *req)
{
	int len;
	char *ptr = req->request.data;

	ptr += strlen("log ");

	len = req->len - (ptr - req->request.data);
	printk("%.*s", len, ptr);
	mconsole_reply(req, "", 0, 0);
}

/* This is a more convoluted version of mconsole_proc, which has some stability
 * problems; however, we need it fixed, because it is expected that UML users
 * mount HPPFS instead of procfs on /proc. And we want mconsole_proc to still
 * show the real procfs content, not the ones from hppfs.*/
#if 0
void mconsole_proc(struct mc_request *req)
{
	struct nameidata nd;
	struct file_system_type *proc;
	struct super_block *super;
	struct file *file;
	int n, err;
	char *ptr = req->request.data, *buf;

	ptr += strlen("proc");
	while(isspace(*ptr)) ptr++;

	proc = get_fs_type("proc");
	if(proc == NULL){
		mconsole_reply(req, "procfs not registered", 1, 0);
		goto out;
	}

	super = (*proc->get_sb)(proc, 0, NULL, NULL);
	put_filesystem(proc);
	if(super == NULL){
		mconsole_reply(req, "Failed to get procfs superblock", 1, 0);
		goto out;
	}
	up_write(&super->s_umount);

	nd.dentry = super->s_root;
	nd.mnt = NULL;
	nd.flags = O_RDONLY + 1;
	nd.last_type = LAST_ROOT;

	/* START: it was experienced that the stability problems are closed
	 * if commenting out these two calls + the below read cycle. To
	 * make UML crash again, it was enough to readd either one.*/
	err = link_path_walk(ptr, &nd);
	if(err){
		mconsole_reply(req, "Failed to look up file", 1, 0);
		goto out_kill;
	}

	file = dentry_open(nd.dentry, nd.mnt, O_RDONLY);
	if(IS_ERR(file)){
		mconsole_reply(req, "Failed to open file", 1, 0);
		goto out_kill;
	}
	/*END*/

	buf = kmalloc(PAGE_SIZE, GFP_KERNEL);
	if(buf == NULL){
		mconsole_reply(req, "Failed to allocate buffer", 1, 0);
		goto out_fput;
	}

	if((file->f_op != NULL) && (file->f_op->read != NULL)){
		do {
			n = (*file->f_op->read)(file, buf, PAGE_SIZE - 1,
						&file->f_pos);
			if(n >= 0){
				buf[n] = '\0';
				mconsole_reply(req, buf, 0, (n > 0));
			}
			else {
				mconsole_reply(req, "Read of file failed",
					       1, 0);
				goto out_free;
			}
		} while(n > 0);
	}
	else mconsole_reply(req, "", 0, 0);

 out_free:
	kfree(buf);
 out_fput:
	fput(file);
 out_kill:
	deactivate_super(super);
 out: ;
}
#endif

void mconsole_proc(struct mc_request *req)
{
	char path[64];
	char *buf;
	int len;
	int fd;
	int first_chunk = 1;
	char *ptr = req->request.data;

	ptr += strlen("proc");
	while(isspace(*ptr)) ptr++;
	snprintf(path, sizeof(path), "/proc/%s", ptr);

	fd = sys_open(path, 0, 0);
	if (fd < 0) {
		mconsole_reply(req, "Failed to open file", 1, 0);
		printk("open %s: %d\n",path,fd);
		goto out;
	}

	buf = kmalloc(PAGE_SIZE, GFP_KERNEL);
	if(buf == NULL){
		mconsole_reply(req, "Failed to allocate buffer", 1, 0);
		goto out_close;
	}

	for (;;) {
		len = sys_read(fd, buf, PAGE_SIZE-1);
		if (len < 0) {
			mconsole_reply(req, "Read of file failed", 1, 0);
			goto out_free;
		}
		/*Begin the file content on his own line.*/
		if (first_chunk) {
			mconsole_reply(req, "\n", 0, 1);
			first_chunk = 0;
		}
		if (len == PAGE_SIZE-1) {
			buf[len] = '\0';
			mconsole_reply(req, buf, 0, 1);
		} else {
			buf[len] = '\0';
			mconsole_reply(req, buf, 0, 0);
			break;
		}
	}

 out_free:
	kfree(buf);
 out_close:
	sys_close(fd);
 out:
	/* nothing */;
}

#define UML_MCONSOLE_HELPTEXT \
"Commands: \n\
    version - Get kernel version \n\
    help - Print this message \n\
    halt - Halt UML \n\
    reboot - Reboot UML \n\
    config <dev>=<config> - Add a new device to UML;  \n\
	same syntax as command line \n\
    config <dev> - Query the configuration of a device \n\
    remove <dev> - Remove a device from UML \n\
    sysrq <letter> - Performs the SysRq action controlled by the letter \n\
J
Jeff Dike 已提交
278
    cad - invoke the Ctrl-Alt-Del handler \n\
L
Linus Torvalds 已提交
279 280 281 282
    stop - pause the UML; it will do nothing until it receives a 'go' \n\
    go - continue the UML after a 'stop' \n\
    log <string> - make UML enter <string> into the kernel log\n\
    proc <file> - returns the contents of the UML's /proc/<file>\n\
283
    stack <pid> - returns the stack of the specified pid\n\
L
Linus Torvalds 已提交
284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317
"

void mconsole_help(struct mc_request *req)
{
	mconsole_reply(req, UML_MCONSOLE_HELPTEXT, 0, 0);
}

void mconsole_halt(struct mc_request *req)
{
	mconsole_reply(req, "", 0, 0);
	machine_halt();
}

void mconsole_reboot(struct mc_request *req)
{
	mconsole_reply(req, "", 0, 0);
	machine_restart(NULL);
}

void mconsole_cad(struct mc_request *req)
{
	mconsole_reply(req, "", 0, 0);
	ctrl_alt_del();
}

void mconsole_go(struct mc_request *req)
{
	mconsole_reply(req, "Not stopped", 1, 0);
}

void mconsole_stop(struct mc_request *req)
{
	deactivate_fd(req->originating_fd, MCONSOLE_IRQ);
	os_set_fd_block(req->originating_fd, 1);
A
Al Viro 已提交
318 319 320 321 322 323 324 325 326 327 328 329 330 331 332
	mconsole_reply(req, "stopped", 0, 0);
	while (mconsole_get_request(req->originating_fd, req)) {
		if (req->cmd->handler == mconsole_go)
			break;
		if (req->cmd->handler == mconsole_stop) {
			mconsole_reply(req, "Already stopped", 1, 0);
			continue;
		}
		if (req->cmd->handler == mconsole_sysrq) {
			struct pt_regs *old_regs;
			old_regs = set_irq_regs((struct pt_regs *)&req->regs);
			mconsole_sysrq(req);
			set_irq_regs(old_regs);
			continue;
		}
L
Linus Torvalds 已提交
333 334 335 336 337 338 339
		(*req->cmd->handler)(req);
	}
	os_set_fd_block(req->originating_fd, 0);
	reactivate_fd(req->originating_fd, MCONSOLE_IRQ);
	mconsole_reply(req, "", 0, 0);
}

J
Jeff Dike 已提交
340
static DEFINE_SPINLOCK(mc_devices_lock);
341
static LIST_HEAD(mconsole_devices);
L
Linus Torvalds 已提交
342 343 344

void mconsole_register_dev(struct mc_device *new)
{
J
Jeff Dike 已提交
345 346
	spin_lock(&mc_devices_lock);
	BUG_ON(!list_empty(&new->list));
L
Linus Torvalds 已提交
347
	list_add(&new->list, &mconsole_devices);
J
Jeff Dike 已提交
348
	spin_unlock(&mc_devices_lock);
L
Linus Torvalds 已提交
349 350 351 352 353 354 355 356 357 358 359 360 361 362 363
}

static struct mc_device *mconsole_find_dev(char *name)
{
	struct list_head *ele;
	struct mc_device *dev;

	list_for_each(ele, &mconsole_devices){
		dev = list_entry(ele, struct mc_device, list);
		if(!strncmp(name, dev->name, strlen(dev->name)))
			return(dev);
	}
	return(NULL);
}

J
Jeff Dike 已提交
364 365 366 367 368 369 370 371
#define UNPLUGGED_PER_PAGE \
	((PAGE_SIZE - sizeof(struct list_head)) / sizeof(unsigned long))

struct unplugged_pages {
	struct list_head list;
	void *pages[UNPLUGGED_PER_PAGE];
};

J
Jeff Dike 已提交
372
static DECLARE_MUTEX(plug_mem_mutex);
J
Jeff Dike 已提交
373 374 375 376
static unsigned long long unplugged_pages_count = 0;
static struct list_head unplugged_pages = LIST_HEAD_INIT(unplugged_pages);
static int unplug_index = UNPLUGGED_PER_PAGE;

377
static int mem_config(char *str, char **error_out)
J
Jeff Dike 已提交
378 379 380 381 382
{
	unsigned long long diff;
	int err = -EINVAL, i, add;
	char *ret;

383 384
	if(str[0] != '='){
		*error_out = "Expected '=' after 'mem'";
J
Jeff Dike 已提交
385
		goto out;
386
	}
J
Jeff Dike 已提交
387 388 389 390 391 392 393

	str++;
	if(str[0] == '-')
		add = 0;
	else if(str[0] == '+'){
		add = 1;
	}
394 395 396 397
	else {
		*error_out = "Expected increment to start with '-' or '+'";
		goto out;
	}
J
Jeff Dike 已提交
398 399 400

	str++;
	diff = memparse(str, &ret);
401 402
	if(*ret != '\0'){
		*error_out = "Failed to parse memory increment";
J
Jeff Dike 已提交
403
		goto out;
404
	}
J
Jeff Dike 已提交
405 406 407

	diff /= PAGE_SIZE;

J
Jeff Dike 已提交
408
	down(&plug_mem_mutex);
J
Jeff Dike 已提交
409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449
	for(i = 0; i < diff; i++){
		struct unplugged_pages *unplugged;
		void *addr;

		if(add){
			if(list_empty(&unplugged_pages))
				break;

			unplugged = list_entry(unplugged_pages.next,
					       struct unplugged_pages, list);
			if(unplug_index > 0)
				addr = unplugged->pages[--unplug_index];
			else {
				list_del(&unplugged->list);
				addr = unplugged;
				unplug_index = UNPLUGGED_PER_PAGE;
			}

			free_page((unsigned long) addr);
			unplugged_pages_count--;
		}
		else {
			struct page *page;

			page = alloc_page(GFP_ATOMIC);
			if(page == NULL)
				break;

			unplugged = page_address(page);
			if(unplug_index == UNPLUGGED_PER_PAGE){
				list_add(&unplugged->list, &unplugged_pages);
				unplug_index = 0;
			}
			else {
				struct list_head *entry = unplugged_pages.next;
				addr = unplugged;

				unplugged = list_entry(entry,
						       struct unplugged_pages,
						       list);
				err = os_drop_memory(addr, PAGE_SIZE);
450
				if(err){
J
Jeff Dike 已提交
451 452
					printk("Failed to release memory - "
					       "errno = %d\n", err);
453
					*error_out = "Failed to release memory";
J
Jeff Dike 已提交
454
					goto out_unlock;
455 456
				}
				unplugged->pages[unplug_index++] = addr;
J
Jeff Dike 已提交
457 458 459 460 461 462 463
			}

			unplugged_pages_count++;
		}
	}

	err = 0;
J
Jeff Dike 已提交
464 465
out_unlock:
	up(&plug_mem_mutex);
J
Jeff Dike 已提交
466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488
out:
	return err;
}

static int mem_get_config(char *name, char *str, int size, char **error_out)
{
	char buf[sizeof("18446744073709551615")];
	int len = 0;

	sprintf(buf, "%ld", uml_physmem);
	CONFIG_CHUNK(str, size, len, buf, 1);

	return len;
}

static int mem_id(char **str, int *start_out, int *end_out)
{
	*start_out = 0;
	*end_out = 0;

	return 0;
}

489
static int mem_remove(int n, char **error_out)
J
Jeff Dike 已提交
490
{
491
	*error_out = "Memory doesn't support the remove operation";
J
Jeff Dike 已提交
492 493 494 495
	return -EBUSY;
}

static struct mc_device mem_mc = {
J
Jeff Dike 已提交
496
	.list		= LIST_HEAD_INIT(mem_mc.list),
J
Jeff Dike 已提交
497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514
	.name		= "mem",
	.config		= mem_config,
	.get_config	= mem_get_config,
	.id		= mem_id,
	.remove		= mem_remove,
};

static int mem_mc_init(void)
{
	if(can_drop_memory())
		mconsole_register_dev(&mem_mc);
	else printk("Can't release memory to the host - memory hotplug won't "
		    "be supported\n");
	return 0;
}

__initcall(mem_mc_init);

L
Linus Torvalds 已提交
515 516
#define CONFIG_BUF_SIZE 64

J
Jeff Dike 已提交
517
static void mconsole_get_config(int (*get_config)(char *, char *, int,
L
Linus Torvalds 已提交
518 519 520 521 522 523 524 525 526 527 528 529
						  char **),
				struct mc_request *req, char *name)
{
	char default_buf[CONFIG_BUF_SIZE], *error, *buf;
	int n, size;

	if(get_config == NULL){
		mconsole_reply(req, "No get_config routine defined", 1, 0);
		return;
	}

	error = NULL;
530
	size = ARRAY_SIZE(default_buf);
L
Linus Torvalds 已提交
531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562
	buf = default_buf;

	while(1){
		n = (*get_config)(name, buf, size, &error);
		if(error != NULL){
			mconsole_reply(req, error, 1, 0);
			goto out;
		}

		if(n <= size){
			mconsole_reply(req, buf, 0, 0);
			goto out;
		}

		if(buf != default_buf)
			kfree(buf);

		size = n;
		buf = kmalloc(size, GFP_KERNEL);
		if(buf == NULL){
			mconsole_reply(req, "Failed to allocate buffer", 1, 0);
			return;
		}
	}
 out:
	if(buf != default_buf)
		kfree(buf);
}

void mconsole_config(struct mc_request *req)
{
	struct mc_device *dev;
563
	char *ptr = req->request.data, *name, *error_string = "";
L
Linus Torvalds 已提交
564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579
	int err;

	ptr += strlen("config");
	while(isspace(*ptr)) ptr++;
	dev = mconsole_find_dev(ptr);
	if(dev == NULL){
		mconsole_reply(req, "Bad configuration option", 1, 0);
		return;
	}

	name = &ptr[strlen(dev->name)];
	ptr = name;
	while((*ptr != '=') && (*ptr != '\0'))
		ptr++;

	if(*ptr == '='){
580 581
		err = (*dev->config)(name, &error_string);
		mconsole_reply(req, error_string, err, 0);
L
Linus Torvalds 已提交
582 583 584 585 586 587
	}
	else mconsole_get_config(dev->get_config, req, name);
}

void mconsole_remove(struct mc_request *req)
{
J
Jeff Dike 已提交
588
	struct mc_device *dev;
J
Jeff Dike 已提交
589
	char *ptr = req->request.data, *err_msg = "";
590
	char error[256];
J
Jeff Dike 已提交
591
	int err, start, end, n;
L
Linus Torvalds 已提交
592 593 594 595 596 597 598 599

	ptr += strlen("remove");
	while(isspace(*ptr)) ptr++;
	dev = mconsole_find_dev(ptr);
	if(dev == NULL){
		mconsole_reply(req, "Bad remove option", 1, 0);
		return;
	}
J
Jeff Dike 已提交
600

601 602 603 604 605 606 607 608 609 610 611 612 613 614
	ptr = &ptr[strlen(dev->name)];

	err = 1;
	n = (*dev->id)(&ptr, &start, &end);
	if(n < 0){
		err_msg = "Couldn't parse device number";
		goto out;
	}
	else if((n < start) || (n > end)){
		sprintf(error, "Invalid device number - must be between "
			"%d and %d", start, end);
		err_msg = error;
		goto out;
	}
J
Jeff Dike 已提交
615

616 617
	err_msg = NULL;
	err = (*dev->remove)(n, &err_msg);
618 619
	switch(err){
	case -ENODEV:
620 621
		if(err_msg == NULL)
			err_msg = "Device doesn't exist";
622 623
		break;
	case -EBUSY:
624 625
		if(err_msg == NULL)
			err_msg = "Device is currently open";
626 627 628 629 630
		break;
	default:
		break;
	}
out:
J
Jeff Dike 已提交
631
	mconsole_reply(req, err_msg, err, 0);
L
Linus Torvalds 已提交
632 633
}

634 635 636 637 638
struct mconsole_output {
	struct list_head list;
	struct mc_request *req;
};

J
Jeff Dike 已提交
639
static DEFINE_SPINLOCK(client_lock);
640 641 642 643 644 645 646 647 648 649 650 651 652 653
static LIST_HEAD(clients);
static char console_buf[MCONSOLE_MAX_DATA];
static int console_index = 0;

static void console_write(struct console *console, const char *string,
			  unsigned len)
{
	struct list_head *ele;
	int n;

	if(list_empty(&clients))
		return;

	while(1){
654
		n = min((size_t) len, ARRAY_SIZE(console_buf) - console_index);
655 656 657 658 659 660 661 662
		strncpy(&console_buf[console_index], string, n);
		console_index += n;
		string += n;
		len -= n;
		if(len == 0)
			return;

		list_for_each(ele, &clients){
663
			struct mconsole_output *entry;
664

665 666
			entry = list_entry(ele, struct mconsole_output, list);
			mconsole_reply_len(entry->req, console_buf,
667 668 669 670 671 672 673 674 675
					   console_index, 0, 1);
		}

		console_index = 0;
	}
}

static struct console mc_console = { .name	= "mc",
				     .write	= console_write,
676
				     .flags	= CON_ENABLED,
677 678 679 680 681 682 683 684 685 686 687 688 689
				     .index	= -1 };

static int mc_add_console(void)
{
	register_console(&mc_console);
	return 0;
}

late_initcall(mc_add_console);

static void with_console(struct mc_request *req, void (*proc)(void *),
			 void *arg)
{
690
	struct mconsole_output entry;
691 692
	unsigned long flags;

693
	entry.req = req;
J
Jeff Dike 已提交
694
	spin_lock_irqsave(&client_lock, flags);
695
	list_add(&entry.list, &clients);
J
Jeff Dike 已提交
696
	spin_unlock_irqrestore(&client_lock, flags);
697 698 699 700 701 702

	(*proc)(arg);

	mconsole_reply_len(req, console_buf, console_index, 0, 0);
	console_index = 0;

J
Jeff Dike 已提交
703
	spin_lock_irqsave(&client_lock, flags);
704
	list_del(&entry.list);
J
Jeff Dike 已提交
705
	spin_unlock_irqrestore(&client_lock, flags);
706 707
}

708 709 710 711
#ifdef CONFIG_MAGIC_SYSRQ
static void sysrq_proc(void *arg)
{
	char *op = arg;
A
Al Viro 已提交
712
	handle_sysrq(*op, NULL);
713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736
}

void mconsole_sysrq(struct mc_request *req)
{
	char *ptr = req->request.data;

	ptr += strlen("sysrq");
	while(isspace(*ptr)) ptr++;

	/* With 'b', the system will shut down without a chance to reply,
	 * so in this case, we reply first.
	 */
	if(*ptr == 'b')
		mconsole_reply(req, "", 0, 0);

	with_console(req, sysrq_proc, ptr);
}
#else
void mconsole_sysrq(struct mc_request *req)
{
	mconsole_reply(req, "Sysrq not compiled in", 1, 0);
}
#endif

737 738
#ifdef CONFIG_MODE_SKAS

739 740 741 742 743 744 745 746
static void stack_proc(void *arg)
{
	struct task_struct *from = current, *to = arg;

	to->thread.saved_task = from;
	switch_to(from, to, from);
}

747 748 749 750 751
/* Mconsole stack trace
 *  Added by Allan Graves, Jeff Dike
 *  Dumps a stacks registers to the linux console.
 *  Usage stack <pid>.
 */
752
static void do_stack_trace(struct mc_request *req)
753
{
754 755
	char *ptr = req->request.data;
	int pid_requested= -1;
756
	struct task_struct *from = NULL;
757 758
	struct task_struct *to = NULL;

759 760
	/* Would be nice:
	 * 1) Send showregs output to mconsole.
761 762 763
	 * 2) Add a way to stack dump all pids.
	 */

764 765
	ptr += strlen("stack");
	while(isspace(*ptr)) ptr++;
766

767 768 769 770 771 772
	/* Should really check for multiple pids or reject bad args here */
	/* What do the arguments in mconsole_reply mean? */
	if(sscanf(ptr, "%d", &pid_requested) == 0){
		mconsole_reply(req, "Please specify a pid", 1, 0);
		return;
	}
773

774
	from = current;
775

776
	to = find_task_by_pid(pid_requested);
777 778 779 780
	if((to == NULL) || (pid_requested == 0)) {
		mconsole_reply(req, "Couldn't find that pid", 1, 0);
		return;
	}
781
	with_console(req, stack_proc, to);
782
}
783
#endif /* CONFIG_MODE_SKAS */
784 785 786 787 788 789 790 791

void mconsole_stack(struct mc_request *req)
{
	/* This command doesn't work in TT mode, so let's check and then
	 * get out of here
	 */
	CHOOSE_MODE(mconsole_reply(req, "Sorry, this doesn't work in TT mode",
				   1, 0),
792
		    do_stack_trace(req));
793 794
}

L
Linus Torvalds 已提交
795 796 797
/* Changed by mconsole_setup, which is __setup, and called before SMP is
 * active.
 */
J
Jeff Dike 已提交
798
static char *notify_socket = NULL;
L
Linus Torvalds 已提交
799

800
static int mconsole_init(void)
L
Linus Torvalds 已提交
801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818
{
	/* long to avoid size mismatch warnings from gcc */
	long sock;
	int err;
	char file[256];

	if(umid_file_name("mconsole", file, sizeof(file))) return(-1);
	snprintf(mconsole_socket_name, sizeof(file), "%s", file);

	sock = os_create_unix_socket(file, sizeof(file), 1);
	if (sock < 0){
		printk("Failed to initialize management console\n");
		return(1);
	}

	register_reboot_notifier(&reboot_notifier);

	err = um_request_irq(MCONSOLE_IRQ, sock, IRQ_READ, mconsole_interrupt,
819
			     IRQF_DISABLED | IRQF_SHARED | IRQF_SAMPLE_RANDOM,
L
Linus Torvalds 已提交
820 821 822 823 824 825 826
			     "mconsole", (void *)sock);
	if (err){
		printk("Failed to get IRQ for management console\n");
		return(1);
	}

	if(notify_socket != NULL){
J
Jeff Dike 已提交
827
		notify_socket = kstrdup(notify_socket, GFP_KERNEL);
L
Linus Torvalds 已提交
828 829
		if(notify_socket != NULL)
			mconsole_notify(notify_socket, MCONSOLE_SOCKET,
J
Jeff Dike 已提交
830
					mconsole_socket_name,
L
Linus Torvalds 已提交
831 832 833 834 835
					strlen(mconsole_socket_name) + 1);
		else printk(KERN_ERR "mconsole_setup failed to strdup "
			    "string\n");
	}

J
Jeff Dike 已提交
836
	printk("mconsole (version %d) initialized on %s\n",
L
Linus Torvalds 已提交
837 838 839 840 841 842 843 844 845 846 847 848
	       MCONSOLE_VERSION, mconsole_socket_name);
	return(0);
}

__initcall(mconsole_init);

static int write_proc_mconsole(struct file *file, const char __user *buffer,
			       unsigned long count, void *data)
{
	char *buf;

	buf = kmalloc(count + 1, GFP_KERNEL);
J
Jeff Dike 已提交
849
	if(buf == NULL)
L
Linus Torvalds 已提交
850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872
		return(-ENOMEM);

	if(copy_from_user(buf, buffer, count)){
		count = -EFAULT;
		goto out;
	}

	buf[count] = '\0';

	mconsole_notify(notify_socket, MCONSOLE_USER_NOTIFY, buf, count);
 out:
	kfree(buf);
	return(count);
}

static int create_proc_mconsole(void)
{
	struct proc_dir_entry *ent;

	if(notify_socket == NULL) return(0);

	ent = create_proc_entry("mconsole", S_IFREG | 0200, NULL);
	if(ent == NULL){
873
		printk(KERN_INFO "create_proc_mconsole : create_proc_entry failed\n");
L
Linus Torvalds 已提交
874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924
		return(0);
	}

	ent->read_proc = NULL;
	ent->write_proc = write_proc_mconsole;
	return(0);
}

static DEFINE_SPINLOCK(notify_spinlock);

void lock_notify(void)
{
	spin_lock(&notify_spinlock);
}

void unlock_notify(void)
{
	spin_unlock(&notify_spinlock);
}

__initcall(create_proc_mconsole);

#define NOTIFY "=notify:"

static int mconsole_setup(char *str)
{
	if(!strncmp(str, NOTIFY, strlen(NOTIFY))){
		str += strlen(NOTIFY);
		notify_socket = str;
	}
	else printk(KERN_ERR "mconsole_setup : Unknown option - '%s'\n", str);
	return(1);
}

__setup("mconsole", mconsole_setup);

__uml_help(mconsole_setup,
"mconsole=notify:<socket>\n"
"    Requests that the mconsole driver send a message to the named Unix\n"
"    socket containing the name of the mconsole socket.  This also serves\n"
"    to notify outside processes when UML has booted far enough to respond\n"
"    to mconsole requests.\n\n"
);

static int notify_panic(struct notifier_block *self, unsigned long unused1,
			void *ptr)
{
	char *message = ptr;

	if(notify_socket == NULL) return(0);

J
Jeff Dike 已提交
925
	mconsole_notify(notify_socket, MCONSOLE_PANIC, message,
L
Linus Torvalds 已提交
926 927 928 929 930 931 932 933 934 935 936 937
			strlen(message) + 1);
	return(0);
}

static struct notifier_block panic_exit_notifier = {
	.notifier_call 		= notify_panic,
	.next 			= NULL,
	.priority 		= 1
};

static int add_notifier(void)
{
938 939
	atomic_notifier_chain_register(&panic_notifier_list,
			&panic_exit_notifier);
L
Linus Torvalds 已提交
940 941 942 943 944 945 946 947 948 949 950
	return(0);
}

__initcall(add_notifier);

char *mconsole_notify_socket(void)
{
	return(notify_socket);
}

EXPORT_SYMBOL(mconsole_notify_socket);