Subversion Repositories HelenOS

/*

 * Copyright (c) 2006 Ondrej Palkovsky

 * Copyright (c) 2008 Martin Decky

 * Copyright (c) 2008 Pavel Rimsky

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

 */

/**

 * @defgroup serial Serial console

 * @brief    Serial console services (putc, puts, clear screen, cursor goto,...)

 * @{

 */ 

/** @file

 */

#include <stdio.h>

#include <ipc/ipc.h>

#include <async.h>

#include <ipc/fb.h>

#include <bool.h>

#include <errno.h>

#include <console/color.h>

#include <console/style.h>

#include "../console/screenbuffer.h"

#include "main.h"

#include "serial_console.h"

#define MAX_CONTROL 20

static void serial_sgr(const unsigned int mode);

static int scr_width;

static int scr_height;

static bool color = true;	/** True if producing color output. */

static putc_function_t putc_function;

/* Allow only 1 connection */

static int client_connected = 0;

enum sgr_color_index {

	CI_BLACK	= 0,

	CI_RED		= 1,

	CI_GREEN	= 2,

	CI_BROWN	= 3,

	CI_BLUE		= 4,

	CI_MAGENTA	= 5,

	CI_CYAN		= 6,

	CI_WHITE	= 7,

};

enum sgr_command {

	SGR_RESET	= 0,

	SGR_BOLD	= 1,

	SGR_BLINK	= 5,

	SGR_REVERSE	= 7,

	SGR_NORMAL_INT	= 22,

	SGR_BLINK_OFF	= 25,

	SGR_REVERSE_OFF = 27,

	SGR_FGCOLOR	= 30,

	SGR_BGCOLOR	= 40

};

static int color_map[] = {

	[COLOR_BLACK]	= CI_BLACK,

	[COLOR_BLUE]	= CI_RED,

	[COLOR_GREEN]	= CI_GREEN,

	[COLOR_CYAN]	= CI_CYAN,

	[COLOR_RED]	= CI_RED,

	[COLOR_MAGENTA] = CI_MAGENTA,

	[COLOR_YELLOW]	= CI_BROWN,

	[COLOR_WHITE]	= CI_WHITE

};

void serial_puts(char *str)

{

	while (*str)

		putc_function(*(str++));

}

void serial_goto(const unsigned int row, const unsigned int col)

{

	if ((row > scr_height) || (col > scr_width))

		return;

	char control[MAX_CONTROL];

	snprintf(control, MAX_CONTROL, "\033[%u;%uf", row + 1, col + 1);

	serial_puts(control);

}

void serial_clrscr(void)

{

	/* Initialize graphic rendition attributes. */

	serial_sgr(SGR_RESET);

	if (color) {

		serial_sgr(SGR_FGCOLOR + CI_BLACK);

		serial_sgr(SGR_BGCOLOR + CI_WHITE);

	}

	serial_puts("\033[2J");

}

void serial_scroll(int i)

{

	if (i > 0) {

		serial_goto(scr_height - 1, 0);

		while (i--)

			serial_puts("\033D");

	} else if (i < 0) {

		serial_goto(0, 0);

		while (i++)

			serial_puts("\033M");

	}

}

/** ECMA-48 Set Graphics Rendition. */

static void serial_sgr(const unsigned int mode)

{

	char control[MAX_CONTROL];

	snprintf(control, MAX_CONTROL, "\033[%um", mode);

	serial_puts(control);

}

/** Set scrolling region. */

void serial_set_scroll_region(unsigned last_row)

{

	char control[MAX_CONTROL];

	snprintf(control, MAX_CONTROL, "\033[0;%ur", last_row);

	serial_puts(control);

}

void serial_cursor_disable(void)

{

	serial_puts("\033[?25l");

}

void serial_cursor_enable(void)

{

	serial_puts("\033[?25h");

}

void serial_console_init(putc_function_t putc_fn, uint32_t w, uint32_t h)

{

	scr_width = w;

	scr_height = h;

	putc_function = putc_fn;

}

static void serial_set_style(int style)

{

	if (style == STYLE_EMPHASIS) {

		if (color) {

			serial_sgr(SGR_RESET);

			serial_sgr(SGR_FGCOLOR + CI_RED);

			serial_sgr(SGR_BGCOLOR + CI_WHITE);

		}

		serial_sgr(SGR_BOLD);

	} else {

		if (color) {

			serial_sgr(SGR_RESET);

			serial_sgr(SGR_FGCOLOR + CI_BLACK);

			serial_sgr(SGR_BGCOLOR + CI_WHITE);

		}

		serial_sgr(SGR_NORMAL_INT);

	}

}

static void serial_set_idx(unsigned fgcolor, unsigned bgcolor,

    unsigned flags)

{

	if (color) {

		serial_sgr(SGR_RESET);

		serial_sgr(SGR_FGCOLOR + color_map[fgcolor]);

		serial_sgr(SGR_BGCOLOR + color_map[bgcolor]);

	} else {

		if (fgcolor < bgcolor)

			serial_sgr(SGR_RESET);

		else

			serial_sgr(SGR_REVERSE);

	}	

}

static void serial_set_rgb(uint32_t fgcolor, uint32_t bgcolor)

{

	if (fgcolor < bgcolor)

		serial_sgr(SGR_REVERSE_OFF);

	else

		serial_sgr(SGR_REVERSE);	

}

static void serial_set_attrs(const attrs_t *a)

{

	switch (a->t) {

	case at_style: serial_set_style(a->a.s.style); break;

	case at_rgb: serial_set_rgb(a->a.r.fg_color, a->a.r.bg_color); break;

	case at_idx: serial_set_idx(a->a.i.fg_color,

	    a->a.i.bg_color, a->a.i.flags); break;

	default: break;

	}

}

static void draw_text_data(keyfield_t *data)

{

	int i, j;

	attrs_t *a0, *a1;

	serial_goto(0, 0);

	a0 = &data[0].attrs;

	serial_set_attrs(a0);

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

		for (j = 0; j < scr_width; j++) {

			a1 = &data[i * scr_width + j].attrs;

			if (!attrs_same(*a0, *a1))

				serial_set_attrs(a1);

			(*putc_function)(data[i * scr_width + j].character);

			a0 = a1;

		}

	}

}

int lastcol = 0;

int lastrow = 0;

#define FB_WRITE_BUF_SIZE 256

static char fb_write_buf[FB_WRITE_BUF_SIZE];

static void fb_write(ipc_callid_t rid, ipc_call_t *request)

{

	int row, col;

	ipc_callid_t callid;

	size_t len;

	size_t i;

	row = IPC_GET_ARG1(*request);

	col = IPC_GET_ARG2(*request);

	if ((col >= scr_width) || (row >= scr_height)) {

		ipc_answer_0(callid, EINVAL);

		ipc_answer_0(rid, EINVAL);

		return;

	}

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

		ipc_answer_0(callid, EINVAL);

		ipc_answer_0(rid, EINVAL);

		return;

	}

	if (len > FB_WRITE_BUF_SIZE)

		len = FB_WRITE_BUF_SIZE;

	if (len >= scr_width - col)

		len = scr_width - col;

	(void) ipc_data_write_finalize(callid, fb_write_buf, len);

	if ((lastcol != col) || (lastrow != row))

		serial_goto(row, col);

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

		(*putc_function)(fb_write_buf[i]);

	}

	lastcol = col + len;

	lastrow = row;

	ipc_answer_1(rid, EOK, len);

}

/**

 * Main function of the thread serving client connections.

 */

void serial_client_connection(ipc_callid_t iid, ipc_call_t *icall)

{

	int retval;

	ipc_callid_t callid;

	ipc_call_t call;

	keyfield_t *interbuf = NULL;

	size_t intersize = 0;

	char c;

	int newcol;

	int newrow;

	int fgcolor;

	int bgcolor;

	int flags;

	int style;

	int i;

	if (client_connected) {

		ipc_answer_0(iid, ELIMIT);

		return;

	}

	client_connected = 1;

	ipc_answer_0(iid, EOK);

	/* Clear the terminal, set scrolling region

	   to 0 - height rows. */

	serial_clrscr();

	serial_goto(0, 0);

	serial_set_scroll_region(scr_height);

	while (true) {

		callid = async_get_call(&call);

		switch (IPC_GET_METHOD(call)) {

		case IPC_M_PHONE_HUNGUP:

			client_connected = 0;

			ipc_answer_0(callid, EOK);

			return;

		case IPC_M_SHARE_OUT:

			/* We accept one area for data interchange */

			intersize = IPC_GET_ARG2(call);

			if (intersize >= scr_width * scr_height *

			    sizeof(*interbuf)) {

				receive_comm_area(callid, &call,

				    (void *) &interbuf);

				continue;

			}

			retval = EINVAL;

			break;

		case FB_DRAW_TEXT_DATA:

			if (!interbuf) {

				retval = EINVAL;

				break;

			}

			draw_text_data(interbuf);

			retval = 0;

			break;

		case FB_PUTCHAR:

			c = IPC_GET_ARG1(call);

			newrow = IPC_GET_ARG2(call);

			newcol = IPC_GET_ARG3(call);

			if ((lastcol != newcol) || (lastrow != newrow))

				serial_goto(newrow, newcol);

			lastcol = newcol + 1;

			lastrow = newrow;

			(*putc_function)(c);

			retval = 0;

			break;

		case FB_WRITE:

			fb_write(callid, &call);

			/* Message already answered */

			continue;

		case FB_CURSOR_GOTO:

			newrow = IPC_GET_ARG1(call);

			newcol = IPC_GET_ARG2(call);

			serial_goto(newrow, newcol);

			lastrow = newrow;

			lastcol = newcol;

			retval = 0;

			break;

		case FB_GET_CSIZE:

			ipc_answer_2(callid, EOK, scr_height, scr_width);

			continue;

		case FB_CLEAR:

			serial_clrscr();

			retval = 0;

			break;

		case FB_SET_STYLE:

			style = IPC_GET_ARG1(call);

			serial_set_style(style);

			retval = 0;

			break;

		case FB_SET_COLOR:

			fgcolor = IPC_GET_ARG1(call);

			bgcolor = IPC_GET_ARG2(call);

			flags = IPC_GET_ARG3(call);

			serial_set_idx(fgcolor, bgcolor, flags);

			retval = 0;

			break;

		case FB_SET_RGB_COLOR:

			fgcolor = IPC_GET_ARG1(call);

			bgcolor = IPC_GET_ARG2(call);

			serial_set_rgb(fgcolor, bgcolor);

			retval = 0;

			break;

		case FB_SCROLL:

			i = IPC_GET_ARG1(call);

			if ((i > scr_height) || (i < -scr_height)) {

				retval = EINVAL;

				break;

			}

			serial_scroll(i);

			serial_goto(lastrow, lastcol);

			retval = 0;

			break;

		case FB_CURSOR_VISIBILITY:

			if(IPC_GET_ARG1(call))

				serial_cursor_enable();

			else

				serial_cursor_disable();

			retval = 0;

			break;

		default:

			retval = ENOENT;

		}

		ipc_answer_0(callid, retval);

	}

}

/** 

 * @}

 */