Subversion Repositories HelenOS

Rev

Rev 2678 | Rev 3599 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

/*
 * 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 <keys.h>
#include <ipc/fb.h>
#include <ipc/services.h>
#include <errno.h>
#include <key_buffer.h>
#include <console.h>
#include <unistd.h>
#include <async.h>
#include <libadt/fifo.h>
#include <screenbuffer.h>
#include <sys/mman.h>
#include <stdio.h>

#include "gcons.h"

#define MAX_KEYREQUESTS_BUFFERED 32

#define NAME "console"

/** Index of currently used virtual console.
 */
int active_console = 0;

/** Information about framebuffer
 */
struct {
    int phone;      /**< Framebuffer phone */
    ipcarg_t rows;      /**< Framebuffer rows */
    ipcarg_t cols;      /**< Framebuffer columns */
} fb_info;

typedef struct {
    keybuffer_t keybuffer;      /**< Buffer for incoming keys. */
    /** Buffer for unsatisfied request for keys. */
    FIFO_CREATE_STATIC(keyrequests, ipc_callid_t,
        MAX_KEYREQUESTS_BUFFERED);  
    int keyrequest_counter;     /**< Number of requests in buffer. */
    int client_phone;       /**< Phone to connected client. */
    int used;           /**< 1 if this virtual console is
                     * connected to some client.*/
    screenbuffer_t screenbuffer;    /**< Screenbuffer for saving screen
                     * contents and related settings. */
} connection_t;

static connection_t connections[CONSOLE_COUNT]; /**< Array of data for virtual
                         * consoles */
static keyfield_t *interbuffer = NULL;      /**< Pointer to memory shared
                         * with framebufer used for
                         * faster virtual console
                         * switching */

static int kernel_pixmap = -1;  /**< Number of fb pixmap, where kernel
                 * console is stored */


/** Find unused virtual console.
 *
 */
static int find_free_connection(void) 
{
    int i;
    
    for (i = 0; i < CONSOLE_COUNT; i++) {
        if (!connections[i].used)
            return i;
    }
    return -1;
}

static void clrscr(void)
{
    async_msg_0(fb_info.phone, FB_CLEAR);
}

static void curs_visibility(int v)
{
    async_msg_1(fb_info.phone, FB_CURSOR_VISIBILITY, v); 
}

static void curs_goto(int row, int col)
{
    async_msg_2(fb_info.phone, FB_CURSOR_GOTO, row, col); 
}

static void set_style(style_t *style)
{
    async_msg_2(fb_info.phone, FB_SET_STYLE, style->fg_color,
        style->bg_color); 
}

static void set_style_col(int fgcolor, int bgcolor)
{
    async_msg_2(fb_info.phone, FB_SET_STYLE, fgcolor, bgcolor); 
}

static void prtchr(char c, int row, int col)
{
    async_msg_3(fb_info.phone, FB_PUTCHAR, c, row, col);
}

/** Check key and process special keys. 
 *
 *
 */
static void write_char(int console, char key)
{
    screenbuffer_t *scr = &(connections[console].screenbuffer);
    
    switch (key) {
    case '\n':
        scr->position_y++;
        scr->position_x = 0;
        break;
    case '\r':
        break;
    case '\t':
        scr->position_x += 8;
        scr->position_x -= scr->position_x % 8; 
        break;
    case '\b':
        if (scr->position_x == 0) 
            break;
        scr->position_x--;
        if (console == active_console)
            prtchr(' ', scr->position_y, scr->position_x);
        screenbuffer_putchar(scr, ' ');
        break;
    default:    
        if (console == active_console)
            prtchr(key, scr->position_y, scr->position_x);

        screenbuffer_putchar(scr, key);
        scr->position_x++;
    }
    
    scr->position_y += (scr->position_x >= scr->size_x);
    
    if (scr->position_y >= scr->size_y) {
        scr->position_y = scr->size_y - 1;
        screenbuffer_clear_line(scr, scr->top_line);
        scr->top_line = (scr->top_line + 1) % scr->size_y;
        if (console == active_console)
            async_msg_1(fb_info.phone, FB_SCROLL, 1);
    }
    
    scr->position_x = scr->position_x % scr->size_x;
    
    if (console == active_console)
        curs_goto(scr->position_y, scr->position_x);
    
}

/** Save current screen to pixmap, draw old pixmap
 *
 * @param oldpixmap Old pixmap
 * @return ID of pixmap of current screen
 */
static int switch_screens(int oldpixmap)
{
    int newpmap;
       
    /* Save screen */
    newpmap = async_req_0_0(fb_info.phone, FB_VP2PIXMAP);
    if (newpmap < 0)
        return -1;

    if (oldpixmap != -1) {
        /* Show old screen */
        async_msg_2(fb_info.phone, FB_VP_DRAW_PIXMAP, 0, oldpixmap);
        /* Drop old pixmap */
        async_msg_1(fb_info.phone, FB_DROP_PIXMAP, oldpixmap);
    }
    
    return newpmap;
}

/** Switch to new console */
static void change_console(int newcons)
{
    connection_t *conn;
    static int console_pixmap = -1;
    int i, j, rc;
    keyfield_t *field;
    style_t *style;

    if (newcons == active_console)
        return;

    if (newcons == KERNEL_CONSOLE) {
        if (active_console == KERNEL_CONSOLE)
            return;
        active_console = KERNEL_CONSOLE;
        curs_visibility(0);

        async_serialize_start();
        if (kernel_pixmap == -1) { 
            /* store/restore unsupported */
            set_style_col(DEFAULT_FOREGROUND, DEFAULT_BACKGROUND);
            clrscr();
        } else {
            gcons_in_kernel();
            console_pixmap = switch_screens(kernel_pixmap);
            kernel_pixmap = -1;
        }
        async_serialize_end();

        __SYSCALL0(SYS_DEBUG_ENABLE_CONSOLE);
        return;
    } 
    
    async_serialize_start();

    if (console_pixmap != -1) {
        kernel_pixmap = switch_screens(console_pixmap);
        console_pixmap = -1;
    }
    active_console = newcons;
    gcons_change_console(newcons);
    conn = &connections[active_console];

    set_style(&conn->screenbuffer.style);
    curs_visibility(0);
    if (interbuffer) {
        for (i = 0; i < conn->screenbuffer.size_x; i++)
            for (j = 0; j < conn->screenbuffer.size_y; j++) {
                unsigned int size_x;

                size_x = conn->screenbuffer.size_x; 
                interbuffer[i + j * size_x] =
                    *get_field_at(&conn->screenbuffer, i, j);
            }
        /* This call can preempt, but we are already at the end */
        rc = async_req_0_0(fb_info.phone, FB_DRAW_TEXT_DATA);       
    }
    
    if ((!interbuffer) || (rc != 0)) {
        set_style(&conn->screenbuffer.style);
        clrscr();
        style = &conn->screenbuffer.style;

        for (j = 0; j < conn->screenbuffer.size_y; j++) 
            for (i = 0; i < conn->screenbuffer.size_x; i++) {
                field = get_field_at(&conn->screenbuffer, i, j);
                if (!style_same(*style, field->style))
                    set_style(&field->style);
                style = &field->style;
                if ((field->character == ' ') &&
                    (style_same(field->style,
                    conn->screenbuffer.style)))
                    continue;

                prtchr(field->character, j, i);
            }
    }
    
    curs_goto(conn->screenbuffer.position_y,
        conn->screenbuffer.position_x);
    curs_visibility(conn->screenbuffer.is_cursor_visible);

    async_serialize_end();
}

/** Handler for keyboard */
static void keyboard_events(ipc_callid_t iid, ipc_call_t *icall)
{
    ipc_callid_t callid;
    ipc_call_t call;
    int retval;
    int c;
    connection_t *conn;
    int newcon;
    
    /* Ignore parameters, the connection is alread opened */
    while (1) {
        callid = async_get_call(&call);
        switch (IPC_GET_METHOD(call)) {
        case IPC_M_PHONE_HUNGUP:
            /* TODO: Handle hangup */
            return;
        case KBD_MS_LEFT:
            newcon = gcons_mouse_btn(IPC_GET_ARG1(call));
            if (newcon != -1)
                change_console(newcon);
            retval = 0;
            break;
        case KBD_MS_MOVE:
            gcons_mouse_move(IPC_GET_ARG1(call),
                IPC_GET_ARG2(call));
            retval = 0;
            break;
        case KBD_PUSHCHAR:
            /* got key from keyboard driver */
            
            retval = 0;
            c = IPC_GET_ARG1(call);
            /* switch to another virtual console */
            
            conn = &connections[active_console];
/*
 *          if ((c >= KBD_KEY_F1) && (c < KBD_KEY_F1 +
 *              CONSOLE_COUNT)) {
 */
            if ((c >= 0x101) && (c < 0x101 + CONSOLE_COUNT)) {
                if (c == 0x112)
                    change_console(KERNEL_CONSOLE);
                else
                    change_console(c - 0x101);
                break;
            }
            
            /* if client is awaiting key, send it */
            if (conn->keyrequest_counter > 0) {     
                conn->keyrequest_counter--;
                ipc_answer_1(fifo_pop(conn->keyrequests), EOK,
                    c);
                break;
            }
            
            keybuffer_push(&conn->keybuffer, c);
            retval = 0;
            
            break;
        default:
            retval = ENOENT;
        }
        ipc_answer_0(callid, retval);
    }
}

/** Default thread for new connections */
static void client_connection(ipc_callid_t iid, ipc_call_t *icall)
{
    ipc_callid_t callid;
    ipc_call_t call;
    int consnum;
    ipcarg_t arg1, arg2;
    connection_t *conn;

    if ((consnum = find_free_connection()) == -1) {
        ipc_answer_0(iid, ELIMIT);
        return;
    }
    conn = &connections[consnum];
    conn->used = 1;
    
    async_serialize_start();
    gcons_notify_connect(consnum);
    conn->client_phone = IPC_GET_ARG5(*icall);
    screenbuffer_clear(&conn->screenbuffer);
    
    /* Accept the connection */
    ipc_answer_0(iid, EOK);

    while (1) {
        async_serialize_end();
        callid = async_get_call(&call);
        async_serialize_start();

        arg1 = 0;
        arg2 = 0;
        switch (IPC_GET_METHOD(call)) {
        case IPC_M_PHONE_HUNGUP:
            gcons_notify_disconnect(consnum);
            
            /* Answer all pending requests */
            while (conn->keyrequest_counter > 0) {      
                conn->keyrequest_counter--;
                ipc_answer_0(fifo_pop(conn->keyrequests),
                    ENOENT);
                break;
            }
            conn->used = 0;
            return;
        case CONSOLE_PUTCHAR:
            write_char(consnum, IPC_GET_ARG1(call));
            gcons_notify_char(consnum);
            break;
        case CONSOLE_CLEAR:
            /* Send message to fb */
            if (consnum == active_console) {
                async_msg_0(fb_info.phone, FB_CLEAR); 
            }
            
            screenbuffer_clear(&conn->screenbuffer);
            
            break;
        case CONSOLE_GOTO:
            screenbuffer_goto(&conn->screenbuffer,
                IPC_GET_ARG2(call), IPC_GET_ARG1(call));
            if (consnum == active_console)
                curs_goto(IPC_GET_ARG1(call),
                    IPC_GET_ARG2(call));
            break;
        case CONSOLE_GETSIZE:
            arg1 = fb_info.rows;
            arg2 = fb_info.cols;
            break;
        case CONSOLE_FLUSH:
            if (consnum == active_console)
                async_req_0_0(fb_info.phone, FB_FLUSH);
            break;
        case CONSOLE_SET_STYLE:
            arg1 = IPC_GET_ARG1(call);
            arg2 = IPC_GET_ARG2(call);
            screenbuffer_set_style(&conn->screenbuffer, arg1,
                arg2);
            if (consnum == active_console)
                set_style_col(arg1, arg2);
            break;
        case CONSOLE_CURSOR_VISIBILITY:
            arg1 = IPC_GET_ARG1(call);
            conn->screenbuffer.is_cursor_visible = arg1;
            if (consnum == active_console)
                curs_visibility(arg1);
            break;
        case CONSOLE_GETCHAR:
            if (keybuffer_empty(&conn->keybuffer)) {
                /* buffer is empty -> store request */
                if (conn->keyrequest_counter <
                    MAX_KEYREQUESTS_BUFFERED) { 
                    fifo_push(conn->keyrequests, callid);
                    conn->keyrequest_counter++;
                } else {
                    /*
                     * No key available and too many
                     * requests => fail.
                    */
                    ipc_answer_0(callid, ELIMIT);
                }
                continue;
            }
            keybuffer_pop(&conn->keybuffer, (int *) &arg1);
            break;
        }
        ipc_answer_2(callid, EOK, arg1, arg2);
    }
}

int main(int argc, char *argv[])
{
    printf(NAME ": HelenOS Console service\n");
    
    ipcarg_t phonehash;
    int kbd_phone;
    int i;

    async_set_client_connection(client_connection);
    
    /* Connect to keyboard driver */

    kbd_phone = ipc_connect_me_to(PHONE_NS, SERVICE_KEYBOARD, 0, 0);
    while (kbd_phone < 0) {
        usleep(10000);
        kbd_phone = ipc_connect_me_to(PHONE_NS, SERVICE_KEYBOARD, 0, 0);
    }
    
    if (ipc_connect_to_me(kbd_phone, SERVICE_CONSOLE, 0, 0, &phonehash) != 0)
        return -1;
    async_new_connection(phonehash, 0, NULL, keyboard_events);
    
    /* Connect to framebuffer driver */
    
    fb_info.phone = ipc_connect_me_to(PHONE_NS, SERVICE_VIDEO, 0, 0);
    while (fb_info.phone < 0) {
        usleep(10000);
        fb_info.phone = ipc_connect_me_to(PHONE_NS, SERVICE_VIDEO, 0, 0);
    }
    
    /* Save old kernel screen */
    kernel_pixmap = switch_screens(-1);

    /* Initialize gcons */
    gcons_init(fb_info.phone);
    /* Synchronize, the gcons can have something in queue */
    async_req_0_0(fb_info.phone, FB_FLUSH);
    /* Enable double buffering */
    async_msg_2(fb_info.phone, FB_VIEWPORT_DB, (sysarg_t) -1, 1);
    
    async_req_0_2(fb_info.phone, FB_GET_CSIZE, &fb_info.rows,
        &fb_info.cols); 
    set_style_col(DEFAULT_FOREGROUND, DEFAULT_BACKGROUND);
    clrscr();
    
    /* Init virtual consoles */
    for (i = 0; i < CONSOLE_COUNT; i++) {
        connections[i].used = 0;
        keybuffer_init(&connections[i].keybuffer);
        
        connections[i].keyrequests.head = 0;
        connections[i].keyrequests.tail = 0;
        connections[i].keyrequests.items = MAX_KEYREQUESTS_BUFFERED;
        connections[i].keyrequest_counter = 0;
        
        if (screenbuffer_init(&connections[i].screenbuffer,
            fb_info.cols, fb_info.rows) == NULL) {
            /* FIXME: handle error */
            return -1;
        }
    }
    connections[KERNEL_CONSOLE].used = 1;
    
    interbuffer = mmap(NULL,
        sizeof(keyfield_t) * fb_info.cols * fb_info.rows,
        PROTO_READ | PROTO_WRITE, MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
    if (!interbuffer) {
        if (ipc_share_out_start(fb_info.phone, interbuffer,
            AS_AREA_READ) != EOK) {
            munmap(interbuffer,
                sizeof(keyfield_t) * fb_info.cols * fb_info.rows);
            interbuffer = NULL;
        }
    }

    curs_goto(0, 0);
    curs_visibility(
        connections[active_console].screenbuffer.is_cursor_visible);

    /* Register at NS */
    if (ipc_connect_to_me(PHONE_NS, SERVICE_CONSOLE, 0, 0, &phonehash) != 0)
        return -1;
    
    // FIXME: avoid connectiong to itself, keep using klog
    // printf(NAME ": Accepting connections\n");
    async_manager();

    return 0;   
}
 
/** @}
 */