source: neercs/trunk/src/client.c @ 4063 Tweet

/* 
 *  neercs        console-based window manager
 *  Copyright (c) 2006 Sam Hocevar <sam@zoy.org>
 *                2008 Jean-Yves Lamoureux <jylam@lnxscene.org>
 *                2008 Pascal Terjan <pterjan@linuxfr.org>
 *                All Rights Reserved
 *
 *  $Id: client.c 3939 2009-11-18 14:40:39Z jylam $
 *
 *  This program is free software. It comes without any warranty, to
 *  the extent permitted by applicable law. You can redistribute it
 *  and/or modify it under the terms of the Do What The Fuck You Want
 *  To Public License, Version 2, as published by Sam Hocevar. See
 *  http://sam.zoy.org/wtfpl/COPYING for more details.
 */

#include "config.h"

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <signal.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <time.h>
#include <pwd.h>

#include <errno.h>
#include <caca.h>
#include <caca.h>

#include "neercs.h"

#define NEERCS_RECV_BUFSIZE 128*1024


int start_client(struct screen_list * screen_list)
{
    char *sess = NULL;
    create_socket(screen_list, SOCK_CLIENT);
    while ((sess = connect_socket(screen_list, SOCK_SERVER)) == NULL)
        usleep(100);
    free(sess);

    /* Create main canvas and associated caca window */
    screen_list->cv = caca_create_canvas(0, 0);
    screen_list->dp = caca_create_display(screen_list->cv);

    if (!screen_list->dp)
        return -3;

    caca_set_display_time(screen_list->dp, screen_list->delay * 1000);
    caca_set_cursor(screen_list->dp, 1);

    request_attach(screen_list);
    
    return 0;
}

int send_event(caca_event_t ev, struct screen_list *screen_list)
{
    
    enum caca_event_type t;
    
    t = caca_get_event_type(&ev);
    
    if (t & CACA_EVENT_KEY_PRESS)
    {
        char buf[16];
        int bytes;
        bytes =
        snprintf(buf, sizeof(buf) - 1, "KEY %d",
                 caca_get_event_key_ch(&ev));
        buf[bytes] = '\0';
        debug("Sending key press to server: %s", buf);
        debug("Sending '%s', %d bytes\n", buf, strlen(buf));
        return write(screen_list->comm.socket[SOCK_SERVER], buf, strlen(buf)+1) <= 0;
    }
    else if (t & CACA_EVENT_RESIZE)
    {
        
        char buf[32];
        int bytes;
        bytes = snprintf(buf, sizeof(buf) - 1, "RESIZE %10d %10d",
                         caca_get_event_resize_width(&ev),
                         caca_get_event_resize_height(&ev));
        buf[bytes] = '\0';
        debug("Sending '%s', %d bytes\n", buf, strlen(buf));
        return write(screen_list->comm.socket[SOCK_SERVER], buf, strlen(buf)+1) <= 0;
    }
    else if (t & CACA_EVENT_MOUSE_PRESS)
    {
        char buf[52];
        int bytes;
        bytes = snprintf(buf, sizeof(buf) - 1, "MOUSEP %10d %10d %10d",
                                caca_get_mouse_x (screen_list->dp),
                            caca_get_mouse_y (screen_list->dp),
                                caca_get_event_mouse_button(&ev));
        buf[bytes] = '\0';
        debug("Sending '%s', %d bytes\n", buf, strlen(buf));
        return write(screen_list->comm.socket[SOCK_SERVER], buf, strlen(buf)+1) <= 0;
    }
    else if (t & CACA_EVENT_QUIT)
        return write(screen_list->comm.socket[SOCK_SERVER], "QUIT",
                     strlen("QUIT")) <= 0;
    
    return 0;
}

int send_delay(struct screen_list *screen_list)
{
    debug("Sending DELAY\n");
    char buf[18];
    int bytes;
    bytes = snprintf(buf, sizeof(buf) - 1, "DELAY %10d", screen_list->delay);
    buf[bytes] = '\0';
    return write(screen_list->comm.socket[SOCK_SERVER], buf, strlen(buf)) <= 0;
}


void mainloop(struct screen_list *screen_list)
{
    char *buf = NULL;

    screen_list->last_key_time = get_us();

    for (;;)
    {
        caca_event_t ev;
        int ret = 0;
        ssize_t n;
        if (!screen_list)
            goto end;
        if (!buf)
            buf = malloc(NEERCS_RECV_BUFSIZE);
        if (!buf)
        {
            debug("Failed to allocate memory");
            goto end;
        }
        if (screen_list->comm.socket[SOCK_CLIENT]
            && (n =
                read(screen_list->comm.socket[SOCK_CLIENT], buf,
                     NEERCS_RECV_BUFSIZE - 1)) > 0)
        {
            buf[n] = 0;
            debug("Received from server: '%s' (%d bytes)", buf, n);
            if (!strncmp("DETACH", buf, 6))
            {
                /* ret = 1; Not used */
                break;
            }
            else if (!strncmp("UPDATE ", buf, 7))
            {
                int x, y;
                ssize_t l2 = 0, lb = 0;
                char *buf2;
                size_t l = 0;
                /* FIXME check the length before calling atoi */
                x = atoi(buf + 8);
                y = atoi(buf + 19);

                /* 0 means we have valid data but incomplete, so read the rest 
                 */
                while (l == 0)
                {
                    buf2 = realloc(buf, l2 + NEERCS_RECV_BUFSIZE);
                    if (!buf2)
                    {
                        debug("Failed to allocate memory");
                        goto end;
                    }
                    buf = buf2;
                    fcntl(screen_list->comm.socket[SOCK_CLIENT], F_SETFL, 0);
                    lb = read(screen_list->comm.socket[SOCK_CLIENT], buf + l2,
                              NEERCS_RECV_BUFSIZE - 1);
                    if (lb < 0)
                    {
                        debug
                            ("Failed to read the end of the refresh message (%s)",
                             strerror(errno));
                        l = -1;
                    }
                    else
                    {
                        l2 += lb;
#if defined HAVE_CACA_DIRTY_RECTANGLES
                        l = caca_import_area_from_memory(screen_list->cv, x, y,
                                                         buf, l2, "caca");
#else
                        l = caca_import_from_memory(screen_list->cv, buf, l2,
                                               "caca");
#endif
                    }
                }
                fcntl(screen_list->comm.socket[SOCK_CLIENT], F_SETFL, O_NONBLOCK);
            }
            else if (!strncmp("REFRESH ", buf, 8))
            {
                int dt, x, y;
                /* FIXME check the length before calling atoi */
                x = atoi(buf + 8);
                y = atoi(buf + 19);
                caca_gotoxy(screen_list->cv, x, y);
                caca_refresh_display(screen_list->dp);
                dt = caca_get_display_time(screen_list->dp);

                /* Adjust refresh delay so that the server do not compute
                   useless things */
                if (dt > 2 * 1000 * screen_list->delay
                    && screen_list->delay <= 100)
                {
                    screen_list->delay *= 2;
                    send_delay(screen_list);
                }
                else if (dt < screen_list->delay * 1000 * 1.2 &&
                         screen_list->delay >=
                         3 * screen_list->requested_delay / 2)
                {
                    screen_list->delay = 2 * screen_list->delay / 3;
                    send_delay(screen_list);
                }
            }
            else if (!strncmp("CURSOR ", buf, 7))
            {
                caca_set_cursor(screen_list->dp, atoi(buf + 7));
            }
            else if (!strncmp("TITLE ", buf, 6))
            {
                caca_set_display_title(screen_list->dp, buf + 6);
                caca_refresh_display(screen_list->dp);
            }
            else
            {
                debug("Unknown message received from server: %s", buf);
            }
        }

        if (ret)
        {
            debug("ret1\n");
            break;
        }
        ret = caca_get_event(screen_list->dp,
                             CACA_EVENT_KEY_PRESS
                             | CACA_EVENT_MOUSE_PRESS
                             | CACA_EVENT_MOUSE_RELEASE
                             | CACA_EVENT_MOUSE_MOTION
                             | CACA_EVENT_RESIZE
                             | CACA_EVENT_QUIT, &ev, 10000);
        if (ret)
            ret = send_event(ev, screen_list);

        if (ret)
        {
            debug("ret2\n");
            break;
        }
    }

  end:
    if (buf)
        free(buf);

}