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);

    }

}

/**

 * @}

 */