Subversion Repositories HelenOS

/*

 * Copyright (c) 2006 Josef Cejka

 * 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 console

 * @{

 */

/** @file

 */

#include <libc.h>

#include <fb.h>

#include <ipc/ipc.h>

#include <kbd.h>

#include <io/keycode.h>

#include <ipc/fb.h>

#include <ipc/services.h>

#include <errno.h>

#include <keybuffer.h>

#include <ipc/console.h>

#include <unistd.h>

#include <async.h>

#include <libadt/fifo.h>

#include <sys/mman.h>

#include <stdio.h>

#include <string.h>

#include <sysinfo.h>

#include <event.h>

#include <devmap.h>

#include "console.h"

#include "gcons.h"

#include "screenbuffer.h"

#define NAME  "console"

#define MAX_DEVICE_NAME  32

/** Phone to the keyboard driver. */

static int kbd_phone;

/** Information about framebuffer */

struct {

	int phone;      /**< Framebuffer phone */

	ipcarg_t cols;  /**< Framebuffer columns */

	ipcarg_t rows;  /**< Framebuffer rows */

} fb_info;

typedef struct {

	size_t index;             /**< Console index */

	size_t refcount;          /**< Connection reference count */

	dev_handle_t dev_handle;  /**< Device handle */

	keybuffer_t keybuffer;    /**< Buffer for incoming keys. */

	screenbuffer_t scr;       /**< Screenbuffer for saving screen

	                               contents and related settings. */

} console_t;

/** Array of data for virtual consoles */

static console_t consoles[CONSOLE_COUNT];

static console_t *active_console = &consoles[0];

static console_t *prev_console = &consoles[0];

static console_t *kernel_console = &consoles[KERNEL_CONSOLE];

/** Pointer to memory shared with framebufer used for

    faster virtual console switching */

static keyfield_t *interbuffer = NULL;

/** Information on row-span yet unsent to FB driver. */

struct {

	size_t col;  /**< Leftmost column of the span. */

	size_t row;  /**< Row where the span lies. */

	size_t cnt;  /**< Width of the span. */

} fb_pending;

/** Pending input structure. */

typedef struct {

	link_t link;

	console_t *cons;      /**< Console waiting for input */

	ipc_callid_t rid;     /**< Call ID waiting for input */

	ipc_callid_t callid;  /**< Call ID waiting for IPC_DATA_READ */

	size_t pos;           /**< Position of the last stored data */

	size_t size;          /**< Size of ther buffer */

	char *data;           /**< Already stored data */

} pending_input_t;

LIST_INITIALIZE(pending_input);

/** Process pending input requests */

static void process_pending_input(void)

{

	async_serialize_start();

	link_t *cur;

loop:

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

		pending_input_t *pr = list_get_instance(cur, pending_input_t, link);

		console_event_t ev;

		if (keybuffer_pop(&pr->cons->keybuffer, &ev)) {

			if (pr->data != NULL) {

				if (ev.type == KEY_PRESS) {

					pr->data[pr->pos] = ev.c;

					pr->pos++;

				}

			} else {

				ipc_answer_4(pr->rid, EOK, ev.type, ev.key, ev.mods, ev.c);

				list_remove(cur);

				free(pr);

				goto loop;

			}

		}

		if ((pr->data != NULL) && (pr->pos == pr->size)) {

			(void) ipc_data_read_finalize(pr->callid, pr->data, pr->size);

			ipc_answer_1(pr->rid, EOK, pr->size);

			free(pr->data);

			list_remove(cur);

			free(pr);

			goto loop;

		}

	}

	async_serialize_end();

}

static void curs_visibility(bool visible)

{

	async_msg_1(fb_info.phone, FB_CURSOR_VISIBILITY, visible); 

}

static void curs_hide_sync(void)

{

	ipc_call_sync_1_0(fb_info.phone, FB_CURSOR_VISIBILITY, false); 

}

static void curs_goto(size_t x, size_t y)

{

	async_msg_2(fb_info.phone, FB_CURSOR_GOTO, x, y);

}

static void screen_clear(void)

{

	async_msg_0(fb_info.phone, FB_CLEAR);

}

static void screen_yield(void)

{

	ipc_call_sync_0_0(fb_info.phone, FB_SCREEN_YIELD);

}

static void screen_reclaim(void)

{

	ipc_call_sync_0_0(fb_info.phone, FB_SCREEN_RECLAIM);

}

static void kbd_yield(void)

{

	ipc_call_sync_0_0(kbd_phone, KBD_YIELD);

}

static void kbd_reclaim(void)

{

	ipc_call_sync_0_0(kbd_phone, KBD_RECLAIM);

}

static void set_style(int style)

{

	async_msg_1(fb_info.phone, FB_SET_STYLE, style);

}

static void set_color(int fgcolor, int bgcolor, int flags)

{

	async_msg_3(fb_info.phone, FB_SET_COLOR, fgcolor, bgcolor, flags);

}

static void set_rgb_color(int fgcolor, int bgcolor)

{

	async_msg_2(fb_info.phone, FB_SET_RGB_COLOR, fgcolor, bgcolor); 

}

static void set_attrs(attrs_t *attrs)

{

	switch (attrs->t) {

	case at_style:

		set_style(attrs->a.s.style);

		break;

	case at_idx:

		set_color(attrs->a.i.fg_color, attrs->a.i.bg_color,

		    attrs->a.i.flags);

		break;

	case at_rgb:

		set_rgb_color(attrs->a.r.fg_color, attrs->a.r.bg_color);

		break;

	}

}

/** Send an area of screenbuffer to the FB driver. */

static void fb_update_area(console_t *cons, ipcarg_t x0, ipcarg_t y0, ipcarg_t width, ipcarg_t height)

{

	if (interbuffer) {

		ipcarg_t x;

		ipcarg_t y;

		for (y = 0; y < height; y++) {

			for (x = 0; x < width; x++) {

				interbuffer[y * width + x] =

				    *get_field_at(&cons->scr, x0 + x, y0 + y);

			}

		}

		async_req_4_0(fb_info.phone, FB_DRAW_TEXT_DATA,

		    x0, y0, width, height);

	}

}

/** Flush pending cells to FB. */

static void fb_pending_flush(void)

{

	if (fb_pending.cnt > 0) {

		fb_update_area(active_console, fb_pending.col,

		    fb_pending.row, fb_pending.cnt, 1);

		fb_pending.cnt = 0;

	}

}

/** Mark a character cell as changed.

 *

 * This adds the cell to the pending rowspan if possible. Otherwise

 * the old span is flushed first.

 *

 */

static void cell_mark_changed(size_t col, size_t row)

{

	if (fb_pending.cnt != 0) {

		if ((col != fb_pending.col + fb_pending.cnt)

		    || (row != fb_pending.row)) {

			fb_pending_flush();

		}

	}

	if (fb_pending.cnt == 0) {

		fb_pending.col = col;

		fb_pending.row = row;

	}

	fb_pending.cnt++;

}

/** Print a character to the active VC with buffering. */

static void fb_putchar(wchar_t c, ipcarg_t col, ipcarg_t row)

{

	async_msg_3(fb_info.phone, FB_PUTCHAR, c, col, row);

}

/** Process a character from the client (TTY emulation). */

static void write_char(console_t *cons, wchar_t ch)

{

	switch (ch) {

	case '\n':

		fb_pending_flush();

		cons->scr.position_y++;

		cons->scr.position_x = 0;

		break;

	case '\r':

		break;

	case '\t':

		cons->scr.position_x += 8;

		cons->scr.position_x -= cons->scr.position_x % 8;

		break;

	case '\b':

		if (cons->scr.position_x == 0)

			break;

		cons->scr.position_x--;

		if (cons == active_console)

			cell_mark_changed(cons->scr.position_x, cons->scr.position_y);

		screenbuffer_putchar(&cons->scr, ' ');

		break;

	default:

		if (cons == active_console)

			cell_mark_changed(cons->scr.position_x, cons->scr.position_y);

		screenbuffer_putchar(&cons->scr, ch);

		cons->scr.position_x++;

	}

	if (cons->scr.position_x >= cons->scr.size_x)

		cons->scr.position_y++;

	if (cons->scr.position_y >= cons->scr.size_y) {

		fb_pending_flush();

		cons->scr.position_y = cons->scr.size_y - 1;

		screenbuffer_clear_line(&cons->scr, cons->scr.top_line);

		cons->scr.top_line = (cons->scr.top_line + 1) % cons->scr.size_y;

		if (cons == active_console)

			async_msg_1(fb_info.phone, FB_SCROLL, 1);

	}

	cons->scr.position_x = cons->scr.position_x % cons->scr.size_x;

}

/** Switch to new console */

static void change_console(console_t *cons)

{

	if (cons == active_console)

		return;

	fb_pending_flush();

	if (cons == kernel_console) {

		async_serialize_start();

		curs_hide_sync();

		gcons_in_kernel();

		screen_yield();

		kbd_yield();

		async_serialize_end();

		if (__SYSCALL0(SYS_DEBUG_ENABLE_CONSOLE)) {

			prev_console = active_console;

			active_console = kernel_console;

		} else

			cons = active_console;

	}

	if (cons != kernel_console) {

		size_t x;

		size_t y;

		int rc = 0;

		async_serialize_start();

		if (active_console == kernel_console) {

			screen_reclaim();

			kbd_reclaim();

			gcons_redraw_console();

		}

		active_console = cons;

		gcons_change_console(cons->index);

		set_attrs(&cons->scr.attrs);

		curs_visibility(false);

		if (interbuffer) {

			for (y = 0; y < cons->scr.size_y; y++) {

				for (x = 0; x < cons->scr.size_x; x++) {

					interbuffer[y * cons->scr.size_x + x] =

					    *get_field_at(&cons->scr, x, y);

				}

			}

			/* This call can preempt, but we are already at the end */

			rc = async_req_4_0(fb_info.phone, FB_DRAW_TEXT_DATA,

			    0, 0, cons->scr.size_x,

			    cons->scr.size_y);

		}

		if ((!interbuffer) || (rc != 0)) {

			set_attrs(&cons->scr.attrs);

			screen_clear();

			for (y = 0; y < cons->scr.size_y; y++)

				for (x = 0; x < cons->scr.size_x; x++) {

					keyfield_t *field = get_field_at(&cons->scr, x, y);

					if (!attrs_same(cons->scr.attrs, field->attrs))

						set_attrs(&field->attrs);

					cons->scr.attrs = field->attrs;

					if ((field->character == ' ') &&

					    (attrs_same(field->attrs, cons->scr.attrs)))

						continue;

					fb_putchar(field->character, x, y);

				}

		}

		curs_goto(cons->scr.position_x, cons->scr.position_y);

		curs_visibility(cons->scr.is_cursor_visible);

		async_serialize_end();

	}

}

/** Handler for keyboard */

static void keyboard_events(ipc_callid_t iid, ipc_call_t *icall)

{

	/* Ignore parameters, the connection is already opened */

	while (true) {

		ipc_call_t call;

		ipc_callid_t callid = async_get_call(&call);

		int retval;

		console_event_t ev;

		switch (IPC_GET_METHOD(call)) {

		case IPC_M_PHONE_HUNGUP:

			/* TODO: Handle hangup */

			return;

		case KBD_EVENT:

			/* Got event from keyboard driver. */

			retval = 0;

			ev.type = IPC_GET_ARG1(call);

			ev.key = IPC_GET_ARG2(call);

			ev.mods = IPC_GET_ARG3(call);

			ev.c = IPC_GET_ARG4(call);

			if ((ev.key >= KC_F1) && (ev.key < KC_F1 +

			    CONSOLE_COUNT) && ((ev.mods & KM_CTRL) == 0)) {

				if (ev.key == KC_F1 + KERNEL_CONSOLE)

					change_console(kernel_console);

				else

					change_console(&consoles[ev.key - KC_F1]);

				break;

			}

			keybuffer_push(&active_console->keybuffer, &ev);

			break;

		default:

			retval = ENOENT;

		}

		ipc_answer_0(callid, retval);

	}

}

static void cons_write(console_t *cons, ipc_callid_t rid, ipc_call_t *request)

{

	ipc_callid_t callid;

	size_t size;

	if (!ipc_data_write_receive(&callid, &size)) {

		ipc_answer_0(callid, EINVAL);

		ipc_answer_0(rid, EINVAL);

		return;

	}

	char *buf = (char *) malloc(size);

	if (buf == NULL) {

		ipc_answer_0(callid, ENOMEM);

		ipc_answer_0(rid, ENOMEM);

		return;

	}

	(void) ipc_data_write_finalize(callid, buf, size);

	async_serialize_start();

	size_t off = 0;

	while (off < size) {

		wchar_t ch = str_decode(buf, &off, size);

		write_char(cons, ch);

	}

	if (cons == active_console)

		curs_goto(cons->scr.position_x, cons->scr.position_y);

	async_serialize_end();

	gcons_notify_char(cons->index);

	ipc_answer_1(rid, EOK, size);

	free(buf);

}

static void cons_read(console_t *cons, ipc_callid_t rid, ipc_call_t *request)

{

	ipc_callid_t callid;

	size_t size;

	if (!ipc_data_read_receive(&callid, &size)) {

		ipc_answer_0(callid, EINVAL);

		ipc_answer_0(rid, EINVAL);

		return;

	}

	char *buf = (char *) malloc(size);

	if (buf == NULL) {

		ipc_answer_0(callid, ENOMEM);

		ipc_answer_0(rid, ENOMEM);

		return;

	}

	async_serialize_start();

	size_t pos = 0;

	console_event_t ev;

	while ((keybuffer_pop(&cons->keybuffer, &ev)) && (pos < size)) {

		if (ev.type == KEY_PRESS) {

			buf[pos] = ev.c;

			pos++;

		}

	}

	if (pos == size) {

		(void) ipc_data_read_finalize(callid, buf, size);

		ipc_answer_1(rid, EOK, size);

		free(buf);

	} else {

		pending_input_t *pr = (pending_input_t *) malloc(sizeof(pending_input_t));

		if (!pr) {

			ipc_answer_0(callid, ENOMEM);

			ipc_answer_0(rid, ENOMEM);

			free(buf);

			async_serialize_end();

			return;

		}

		pr->cons = cons;

		pr->rid = rid;

		pr->callid = callid;

		pr->pos = pos;

		pr->size = size;

		pr->data = buf;

		list_append(&pr->link, &pending_input);

	}

	async_serialize_end();

}

static void cons_get_event(console_t *cons, ipc_callid_t rid, ipc_call_t *request)

{

	async_serialize_start();

	console_event_t ev;

	if (keybuffer_pop(&cons->keybuffer, &ev)) {

		ipc_answer_4(rid, EOK, ev.type, ev.key, ev.mods, ev.c);

	} else {

		pending_input_t *pr = (pending_input_t *) malloc(sizeof(pending_input_t));

		if (!pr) {

			ipc_answer_0(rid, ENOMEM);

			async_serialize_end();

			return;

		}

		pr->cons = cons;

		pr->rid = rid;

		pr->callid = 0;

		pr->data = NULL;

		list_append(&pr->link, &pending_input);

	}

	async_serialize_end();

}

/** Default thread for new connections */

static void client_connection(ipc_callid_t iid, ipc_call_t *icall)

{

	console_t *cons = NULL;

	size_t i;

	for (i = 0; i < CONSOLE_COUNT; i++) {

		if (i == KERNEL_CONSOLE)

			continue;

		if (consoles[i].dev_handle == (dev_handle_t) IPC_GET_ARG1(*icall)) {

			cons = &consoles[i];

			break;

		}

	}

	if (cons == NULL) {

		ipc_answer_0(iid, ENOENT);

		return;

	}

	ipc_callid_t callid;

	ipc_call_t call;

	ipcarg_t arg1;

	ipcarg_t arg2;

	ipcarg_t arg3;

	async_serialize_start();

	if (cons->refcount == 0)

		gcons_notify_connect(cons->index);

	cons->refcount++;

	/* Accept the connection */

	ipc_answer_0(iid, EOK);

	while (true) {

		async_serialize_end();

		callid = async_get_call(&call);

		async_serialize_start();

		arg1 = 0;

		arg2 = 0;

		arg3 = 0;

		switch (IPC_GET_METHOD(call)) {

		case IPC_M_PHONE_HUNGUP:

			cons->refcount--;

			if (cons->refcount == 0)

				gcons_notify_disconnect(cons->index);

			return;

		case VFS_READ:

			async_serialize_end();

			cons_read(cons, callid, &call);

			async_serialize_start();

			continue;

		case VFS_WRITE:

			async_serialize_end();

			cons_write(cons, callid, &call);

			async_serialize_start();

			continue;

		case VFS_SYNC:

			fb_pending_flush();

			if (cons == active_console) {

				async_req_0_0(fb_info.phone, FB_FLUSH);

				curs_goto(cons->scr.position_x, cons->scr.position_y);

			}

			break;

		case CONSOLE_CLEAR:

			/* Send message to fb */

			if (cons == active_console)

				async_msg_0(fb_info.phone, FB_CLEAR);

			screenbuffer_clear(&cons->scr);

			break;

		case CONSOLE_GOTO:

			screenbuffer_goto(&cons->scr,

			    IPC_GET_ARG1(call), IPC_GET_ARG2(call));

			if (cons == active_console)

				curs_goto(IPC_GET_ARG1(call),

				    IPC_GET_ARG2(call));

			break;

		case CONSOLE_GET_SIZE:

			arg1 = fb_info.cols;

			arg2 = fb_info.rows;

			break;

		case CONSOLE_SET_STYLE:

			fb_pending_flush();

			arg1 = IPC_GET_ARG1(call);

			screenbuffer_set_style(&cons->scr, arg1);

			if (cons == active_console)

				set_style(arg1);

			break;

		case CONSOLE_SET_COLOR:

			fb_pending_flush();

			arg1 = IPC_GET_ARG1(call);

			arg2 = IPC_GET_ARG2(call);

			arg3 = IPC_GET_ARG3(call);

			screenbuffer_set_color(&cons->scr, arg1, arg2, arg3);

			if (cons == active_console)

				set_color(arg1, arg2, arg3);

			break;

		case CONSOLE_SET_RGB_COLOR:

			fb_pending_flush();

			arg1 = IPC_GET_ARG1(call);

			arg2 = IPC_GET_ARG2(call);

			screenbuffer_set_rgb_color(&cons->scr, arg1, arg2);

			if (cons == active_console)

				set_rgb_color(arg1, arg2);

			break;

		case CONSOLE_CURSOR_VISIBILITY:

			fb_pending_flush();

			arg1 = IPC_GET_ARG1(call);

			cons->scr.is_cursor_visible = arg1;

			if (cons == active_console)

				curs_visibility(arg1);

			break;

		case CONSOLE_GET_EVENT:

			async_serialize_end();

			cons_get_event(cons, callid, &call);

			async_serialize_start();

			continue;

		case CONSOLE_KCON_ENABLE:

			change_console(kernel_console);

			break;

		}

		ipc_answer_3(callid, EOK, arg1, arg2, arg3);

	}

}

static void interrupt_received(ipc_callid_t callid, ipc_call_t *call)

{

	change_console(prev_console);

}

static bool console_init(void)

{

	/* Connect to keyboard driver */

	kbd_phone = ipc_connect_me_to_blocking(PHONE_NS, SERVICE_KEYBOARD, 0, 0);

	if (kbd_phone < 0) {

		printf(NAME ": Failed to connect to keyboard service\n");

		return false;

	}

	ipcarg_t phonehash;

	if (ipc_connect_to_me(kbd_phone, SERVICE_CONSOLE, 0, 0, &phonehash) != 0) {

		printf(NAME ": Failed to create callback from keyboard service\n");

		return false;

	}

	async_set_pending(process_pending_input);

	async_new_connection(phonehash, 0, NULL, keyboard_events);

	/* Connect to framebuffer driver */

	fb_info.phone = ipc_connect_me_to_blocking(PHONE_NS, SERVICE_VIDEO, 0, 0);

	if (fb_info.phone < 0) {

		printf(NAME ": Failed to connect to video service\n");

		return -1;

	}

	/* Register driver */

	int rc = devmap_driver_register(NAME, client_connection);

	if (rc < 0) {

		printf(NAME ": Unable to register driver (%d)\n", rc);

		return false;

	}

	/* Initialize gcons */

	gcons_init(fb_info.phone);

	/* Synchronize, the gcons could put something in queue */

	async_req_0_0(fb_info.phone, FB_FLUSH);

	async_req_0_2(fb_info.phone, FB_GET_CSIZE, &fb_info.cols, &fb_info.rows);

	/* Set up shared memory buffer. */

	size_t ib_size = sizeof(keyfield_t) * fb_info.cols * fb_info.rows;

	interbuffer = as_get_mappable_page(ib_size);

	if (as_area_create(interbuffer, ib_size, AS_AREA_READ |

	    AS_AREA_WRITE | AS_AREA_CACHEABLE) != interbuffer)

		interbuffer = NULL;

	if (interbuffer) {

		if (ipc_share_out_start(fb_info.phone, interbuffer,

		    AS_AREA_READ) != EOK) {

			as_area_destroy(interbuffer);

			interbuffer = NULL;

		}

	}

	fb_pending.cnt = 0;

	/* Inititalize consoles */

	size_t i;

	for (i = 0; i < CONSOLE_COUNT; i++) {

		if (i != KERNEL_CONSOLE) {

			if (screenbuffer_init(&consoles[i].scr,

			    fb_info.cols, fb_info.rows) == NULL) {

				printf(NAME ": Unable to allocate screen buffer %u\n", i);

				return false;

			}

			screenbuffer_clear(&consoles[i].scr);

			keybuffer_init(&consoles[i].keybuffer);

			consoles[i].index = i;

			consoles[i].refcount = 0;

			char vc[MAX_DEVICE_NAME];

			snprintf(vc, MAX_DEVICE_NAME, "vc%u", i);

			if (devmap_device_register(vc, &consoles[i].dev_handle) != EOK) {

				devmap_hangup_phone(DEVMAP_DRIVER);

				printf(NAME ": Unable to register device %s\n", vc);

				return false;

			}

		}

	}

	/* Disable kernel output to the console */

	__SYSCALL0(SYS_DEBUG_DISABLE_CONSOLE);

	/* Initialize the screen */

	gcons_redraw_console();

	set_rgb_color(DEFAULT_FOREGROUND, DEFAULT_BACKGROUND);

	screen_clear();

	curs_goto(0, 0);

	curs_visibility(active_console->scr.is_cursor_visible);

	/* Receive kernel notifications */

	if (event_subscribe(EVENT_KCONSOLE, 0) != EOK)

		printf(NAME ": Error registering kconsole notifications\n");

	async_set_interrupt_received(interrupt_received);

	return true;

}

int main(int argc, char *argv[])

{

	printf(NAME ": HelenOS Console service\n");

	if (!console_init())

		return -1;

	printf(NAME ": Accepting connections\n");

	async_manager();

	return 0;

}

/** @}

 */