Subversion Repositories HelenOS

/*

 * Copyright (c) 2005 Jakub Jermar

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * - Redistributions of source code must retain the above copyright

 *   notice, this list of conditions and the following disclaimer.

 * - Redistributions in binary form must reproduce the above copyright

 *   notice, this list of conditions and the following disclaimer in the

 *   documentation and/or other materials provided with the distribution.

 * - The name of the author may not be used to endorse or promote products

 *   derived from this software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

/** @addtogroup genericconsole

 * @{

 */

/**

 * @file  kconsole.c

 * @brief Kernel console.

 *

 * This file contains kernel thread managing the kernel console.

 *

 */

#include <console/kconsole.h>

#include <console/console.h>

#include <console/chardev.h>

#include <console/cmd.h>

#include <print.h>

#include <panic.h>

#include <arch/types.h>

#include <adt/list.h>

#include <arch.h>

#include <macros.h>

#include <debug.h>

#include <func.h>

#include <string.h>

#include <macros.h>

#include <sysinfo/sysinfo.h>

#include <ddi/device.h>

#include <symtab.h>

#include <errno.h>

#include <putchar.h>

#include <string.h>

/** Simple kernel console.

 *

 * The console is realized by kernel thread kconsole.

 * It doesn't understand any useful command on its own,

 * but makes it possible for other kernel subsystems to

 * register their own commands.

 */

/** Locking.

 *

 * There is a list of cmd_info_t structures. This list

 * is protected by cmd_lock spinlock. Note that specially

 * the link elements of cmd_info_t are protected by

 * this lock.

 *

 * Each cmd_info_t also has its own lock, which protects

 * all elements thereof except the link element.

 *

 * cmd_lock must be acquired before any cmd_info lock.

 * When locking two cmd info structures, structure with

 * lower address must be locked first.

 */

SPINLOCK_INITIALIZE(cmd_lock);  /**< Lock protecting command list. */

LIST_INITIALIZE(cmd_head);      /**< Command list. */

static wchar_t history[KCONSOLE_HISTORY][MAX_CMDLINE] = {};

static count_t history_pos = 0;

/** Initialize kconsole data structures

 *

 * This is the most basic initialization, almost no

 * other kernel subsystem is ready yet.

 *

 */

void kconsole_init(void)

{

	unsigned int i;

	cmd_init();

	for (i = 0; i < KCONSOLE_HISTORY; i++)

		history[i][0] = 0;

}

/** Register kconsole command.

 *

 * @param cmd Structure describing the command.

 *

 * @return False on failure, true on success.

 *

 */

bool cmd_register(cmd_info_t *cmd)

{

	link_t *cur;

	spinlock_lock(&cmd_lock);

	/*

	 * Make sure the command is not already listed.

	 */

	for (cur = cmd_head.next; cur != &cmd_head; cur = cur->next) {

		cmd_info_t *hlp = list_get_instance(cur, cmd_info_t, link);

		if (hlp == cmd) {

			/* The command is already there. */

			spinlock_unlock(&cmd_lock);

			return false;

		}

		/* Avoid deadlock. */

		if (hlp < cmd) {

			spinlock_lock(&hlp->lock);

			spinlock_lock(&cmd->lock);

		} else {

			spinlock_lock(&cmd->lock);

			spinlock_lock(&hlp->lock);

		}

		if (str_cmp(hlp->name, cmd->name) == 0) {

			/* The command is already there. */

			spinlock_unlock(&hlp->lock);

			spinlock_unlock(&cmd->lock);

			spinlock_unlock(&cmd_lock);

			return false;

		}

		spinlock_unlock(&hlp->lock);

		spinlock_unlock(&cmd->lock);

	}

	/*

	 * Now the command can be added.

	 */

	list_append(&cmd->link, &cmd_head);

	spinlock_unlock(&cmd_lock);

	return true;

}

/** Print count times a character */

static void print_cc(wchar_t ch, count_t count)

{

	count_t i;

	for (i = 0; i < count; i++)

		putchar(ch);

}

/** Try to find a command beginning with prefix */

static const char *cmdtab_search_one(const char *name, link_t **startpos)

{

	count_t namelen = str_length(name);

	spinlock_lock(&cmd_lock);

	if (*startpos == NULL)

		*startpos = cmd_head.next;

	for (; *startpos != &cmd_head; *startpos = (*startpos)->next) {

		cmd_info_t *hlp = list_get_instance(*startpos, cmd_info_t, link);

		const char *curname = hlp->name;

		if (str_length(curname) < namelen)

			continue;

		if (str_lcmp(curname, name, namelen) == 0) {

			spinlock_unlock(&cmd_lock);

			return (curname + str_lsize(curname, namelen));

		}

	}

	spinlock_unlock(&cmd_lock);

	return NULL;

}

/** Command completion of the commands

 *

 * @param name String to match, changed to hint on exit

 * @param size Input buffer size

 *

 * @return Number of found matches

 *

 */

static int cmdtab_compl(char *input, size_t size)

{

	const char *name = input;

	count_t found = 0;

	link_t *pos = NULL;

	const char *hint;

	char output[MAX_CMDLINE];

	output[0] = 0;

	while ((hint = cmdtab_search_one(name, &pos))) {

		if ((found == 0) || (str_length(output) > str_length(hint)))

			str_ncpy(output, hint, MAX_CMDLINE);

		pos = pos->next;

		found++;

	}

	if ((found > 1) && (str_length(output) != 0)) {

		printf("\n");

		pos = NULL;

		while ((hint = cmdtab_search_one(name, &pos))) {

			cmd_info_t *hlp = list_get_instance(pos, cmd_info_t, link);

			printf("%s (%s)\n", hlp->name, hlp->description);

			pos = pos->next;

		}

	}

	if (found > 0)

		str_ncpy(input, output, size);

	return found;

}

static wchar_t *clever_readline(const char *prompt, indev_t *indev)

{

	printf("%s> ", prompt);

	count_t position = 0;

	wchar_t *current = history[history_pos];

	current[0] = 0;

	while (true) {

		wchar_t ch = _getc(indev);

		if (ch == '\n') {

			/* Enter */

			putchar(ch);

			break;

		}

		if (ch == '\b') {

			/* Backspace */

			if (position == 0)

				continue;

			if (wstr_remove(current, position - 1)) {

				putchar('\b');

				printf("%ls", current + position);

				position--;

				print_cc('\b', wstr_length(current) - position);

				continue;

			}

		}

		if (ch == '\t') {

			/* Tab completion */

			/* Move to the end of the word */

			for (; (current[position] != 0) && (!isspace(current[position]));

			    position++)

				putchar(current[position]);

			if (position == 0)

				continue;

			/* Find the beginning of the word

			   and copy it to tmp */

			count_t beg;

			for (beg = position - 1; (beg > 0) && (!isspace(current[beg]));

			    beg--);

			if (isspace(current[beg]))

				beg++;

			char tmp[STR_BOUNDS(MAX_CMDLINE)];

			wstr_nstr(tmp, current + beg, position - beg + 1);

			int found;

			if (beg == 0) {

				/* Command completion */

				found = cmdtab_compl(tmp, STR_BOUNDS(MAX_CMDLINE));

			} else {

				/* Symbol completion */

				found = symtab_compl(tmp, STR_BOUNDS(MAX_CMDLINE));

			}

			if (found == 0)

				continue;

			size_t off = 0;

			count_t i = 0;

			while ((ch = str_decode(tmp, &off, STR_NO_LIMIT)) != 0) {

				if (!wstr_linsert(current, ch, position + i, MAX_CMDLINE))

					break;

				i++;

			}

			if ((str_length(tmp) > 0) || (found == 1)) {

				/* We have a hint */

				printf("%ls", current + position);

				print_cc('\b', wstr_length(current) - position);

				position += str_length(tmp);

				if ((found == 1) && (position == wstr_length(current))) {

					if (wstr_linsert(current, ' ', position, MAX_CMDLINE)) {

						printf("%ls", current + position);

						position++;

					}

				}

			} else {

				/* No unique hint, list was printed */

				printf("%s> ", prompt);

				printf("%ls", current);

				position += str_length(tmp);

			}

			print_cc('\b', wstr_length(current) - position);

			continue;

		}

		if (ch == 0x1b) {

			/* Special command */

			wchar_t mod = _getc(indev);

			wchar_t ch = _getc(indev);

			if ((mod != 0x5b) && (mod != 0x4f))

				continue;

			if ((ch == 0x33) && (_getc(indev) == 0x7e)) {

				/* Delete */

				if (position == wstr_length(current))

					continue;

				if (wstr_remove(current, position)) {

					putchar('\b');

					printf("%ls", current + position);

					position--;

					print_cc('\b', wstr_length(current) - position);

				}

			} else if (ch == 0x48) {

				/* Home */

				print_cc('\b', position);

				position = 0;

			} else if (ch == 0x46) {

				/* End */

				printf("%ls", current + position);

				position = wstr_length(current);

			} else if (ch == 0x44) {

				/* Left */

				if (position > 0) {

					putchar('\b');

					position--;

				}

			} else if (ch == 0x43) {

				/* Right */

				if (position < wstr_length(current)) {

					putchar(current[position]);

					position++;

				}

			} else if ((ch == 0x41) || (ch == 0x42)) {

				/* Up, down */

				print_cc('\b', position);

				print_cc(' ', wstr_length(current));

				print_cc('\b', wstr_length(current));

				if (ch == 0x41) {

					/* Up */

					if (history_pos == 0)

						history_pos = KCONSOLE_HISTORY - 1;

					else

						history_pos--;

				} else {

					/* Down */

					history_pos++;

					history_pos = history_pos % KCONSOLE_HISTORY;

				}

				current = history[history_pos];

				printf("%ls", current);

				position = wstr_length(current);

			}

			continue;

		}

		if (wstr_linsert(current, ch, position, MAX_CMDLINE)) {

			printf("%ls", current + position);

			position++;

			print_cc('\b', wstr_length(current) - position);

		}

	}

	if (wstr_length(current) > 0) {

		history_pos++;

		history_pos = history_pos % KCONSOLE_HISTORY;

	}

	return current;

}

bool kconsole_check_poll(void)

{

	return check_poll(stdin);

}

static bool parse_int_arg(const char *text, size_t len, unative_t *result)

{

	bool isaddr = false;

	bool isptr = false;

	/* If we get a name, try to find it in symbol table */

	if (text[0] == '&') {

		isaddr = true;

		text++;

		len--;

	} else if (text[0] == '*') {

		isptr = true;

		text++;

		len--;

	}

	if ((text[0] < '0') || (text[0] > '9')) {

		char symname[MAX_SYMBOL_NAME];

		str_ncpy(symname, text, min(len + 1, MAX_SYMBOL_NAME));

		uintptr_t symaddr;

		int rc = symtab_addr_lookup(symname, &symaddr);

		switch (rc) {

		case ENOENT:

			printf("Symbol %s not found.\n", symname);

			return false;

		case EOVERFLOW:

			printf("Duplicate symbol %s.\n", symname);

			symtab_print_search(symname);

			return false;

		case ENOTSUP:

			printf("No symbol information available.\n");

			return false;

		}

		if (isaddr)

			*result = (unative_t) symaddr;

		else if (isptr)

			*result = **((unative_t **) symaddr);

		else

			*result = *((unative_t *) symaddr);

	} else {

		/* It's a number - convert it */

		*result = atoi(text);

		if (isptr)

			*result = *((unative_t *) *result);

	}

	return true;

}

/** Parse argument.

 *

 * Find start and end positions of command line argument.

 *

 * @param cmdline Command line as read from the input device.

 * @param size    Size (in bytes) of the string.

 * @param start   On entry, 'start' contains pointer to the offset

 *                of the first unprocessed character of cmdline.

 *                On successful exit, it marks beginning of the next argument.

 * @param end     Undefined on entry. On exit, 'end' is the offset of the first

 *                character behind the next argument.

 *

 * @return False on failure, true on success.

 *

 */

static bool parse_argument(const char *cmdline, size_t size, size_t *start, size_t *end)

{

	ASSERT(start != NULL);

	ASSERT(end != NULL);

	bool found_start = false;

	size_t offset = *start;

	size_t prev = *start;

	wchar_t ch;

	while ((ch = str_decode(cmdline, &offset, size)) != 0) {

		if (!found_start) {

			if (!isspace(ch)) {

				*start = prev;

				found_start = true;

			}

		} else {

			if (isspace(ch))

				break;

		}

		prev = offset;

	}

	*end = offset;

	return found_start;

}

/** Parse command line.

 *

 * @param cmdline Command line as read from input device. 

 * @param size    Size (in bytes) of the string.

 *

 * @return Structure describing the command.

 *

 */

static cmd_info_t *parse_cmdline(const char *cmdline, size_t size)

{

	size_t start = 0;

	size_t end = 0;

	if (!parse_argument(cmdline, size, &start, &end)) {

		/* Command line did not contain alphanumeric word. */

		return NULL;

	}

	spinlock_lock(&cmd_lock);

	cmd_info_t *cmd = NULL;

	link_t *cur;

	for (cur = cmd_head.next; cur != &cmd_head; cur = cur->next) {

		cmd_info_t *hlp = list_get_instance(cur, cmd_info_t, link);

		spinlock_lock(&hlp->lock);

		if (str_lcmp(hlp->name, cmdline + start,

		    max(str_length(hlp->name),

		    str_nlength(cmdline + start, (count_t) (end - start) - 1))) == 0) {

			cmd = hlp;

			break;

		}

		spinlock_unlock(&hlp->lock);

	}

	spinlock_unlock(&cmd_lock);

	if (!cmd) {

		/* Unknown command. */

		printf("Unknown command.\n");

		return NULL;

	}

	/* cmd == hlp is locked */

	/*

	 * The command line must be further analyzed and

	 * the parameters therefrom must be matched and

	 * converted to those specified in the cmd info

	 * structure.

	 */

	bool error = false;

	count_t i;

	for (i = 0; i < cmd->argc; i++) {

		start = end;

		if (!parse_argument(cmdline, size, &start, &end)) {

			printf("Too few arguments.\n");

			spinlock_unlock(&cmd->lock);

			return NULL;

		}

		char *buf;

		switch (cmd->argv[i].type) {

		case ARG_TYPE_STRING:

			buf = (char *) cmd->argv[i].buffer;

			str_ncpy(buf, cmdline + start,

			    min((end - start) + 1, cmd->argv[i].len));

			break;

		case ARG_TYPE_INT:

			if (!parse_int_arg(cmdline + start, end - start,

			    &cmd->argv[i].intval))

				error = true;

			break;

		case ARG_TYPE_VAR:

			if ((start < end - 1) && (cmdline[start] == '"')) {

				if (cmdline[end - 1] == '"') {

					buf = (char *) cmd->argv[i].buffer;

					str_ncpy(buf, cmdline + start + 1,

					    min((end - start) - 1, cmd->argv[i].len));

					cmd->argv[i].intval = (unative_t) buf;

					cmd->argv[i].vartype = ARG_TYPE_STRING;

				} else {

					printf("Wrong synxtax.\n");

					error = true;

				}

			} else if (parse_int_arg(cmdline + start,

			    end - start, &cmd->argv[i].intval)) {

				cmd->argv[i].vartype = ARG_TYPE_INT;

			} else {

				printf("Unrecognized variable argument.\n");

				error = true;

			}

			break;

		case ARG_TYPE_INVALID:

		default:

			printf("Invalid argument type\n");

			error = true;

			break;

		}

	}

	if (error) {

		spinlock_unlock(&cmd->lock);

		return NULL;

	}

	start = end;

	if (parse_argument(cmdline, size, &start, &end)) {

		printf("Too many arguments.\n");

		spinlock_unlock(&cmd->lock);

		return NULL;

	}

	spinlock_unlock(&cmd->lock);

	return cmd;

}

/** Kernel console prompt.

 *

 * @param prompt Kernel console prompt (e.g kconsole/panic).

 * @param msg    Message to display in the beginning.

 * @param kcon   Wait for keypress to show the prompt

 *               and never exit.

 *

 */

void kconsole(char *prompt, char *msg, bool kcon)

{

	if (!stdin) {

		LOG("No stdin for kernel console");

		return;

	}

	if (msg)

		printf("%s", msg);

	if (kcon)

		_getc(stdin);

	else

		printf("Type \"exit\" to leave the console.\n");

	while (true) {

		wchar_t *tmp = clever_readline((char *) prompt, stdin);

		count_t len = wstr_length(tmp);

		if (!len)

			continue;

		char cmdline[STR_BOUNDS(MAX_CMDLINE)];

		wstr_nstr(cmdline, tmp, STR_BOUNDS(MAX_CMDLINE));

		if ((!kcon) && (len == 4) && (str_lcmp(cmdline, "exit", 4) == 0))

			break;

		cmd_info_t *cmd_info = parse_cmdline(cmdline, STR_BOUNDS(MAX_CMDLINE));

		if (!cmd_info)

			continue;

		(void) cmd_info->func(cmd_info->argv);

	}

}

/** Kernel console managing thread.

 *

 */

void kconsole_thread(void *data)

{

	kconsole("kconsole", "Kernel console ready (press any key to activate)\n", true);

}

/** @}

 */