source: neercs/trunk/src/attach.c @ 2995 Tweet

#include <errno.h>
#include <fcntl.h>
#include <glob.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include <sys/socket.h>
#include <sys/un.h>
#include <sys/types.h>
#include <sys/stat.h>

#include <caca.h>

#include "config.h"
#include "neercs.h"

char * build_socket_path(char *socket_dir, char *session_name, enum socket_type socktype)
{
    char *path, *dir;
    path = (char *)malloc(PATH_MAX+1);
    dir = socket_dir;
    if(!dir)
        dir = getenv("NEERCSDIR");
    if(!dir)
        dir = getenv("TMPDIR");
    if(!dir)
        dir = "/tmp";
    if(path)
        snprintf(path, PATH_MAX+1, "%s/neercs.%s%s.sock", dir, session_name, socktype?"":".srv");
    return path;
}

int create_socket(struct screen_list* screen_list, enum socket_type socktype)
{
    int sock;
    struct sockaddr_un myaddr;

    sock = socket(AF_UNIX, SOCK_DGRAM, 0);

    if(sock < 0)
    {
        perror("create_socket:socket");
        return errno;
    }

    memset(&myaddr, 0, sizeof(struct sockaddr_un));

    myaddr.sun_family = AF_UNIX;
    strncpy(myaddr.sun_path, screen_list->socket_path[socktype], sizeof(myaddr.sun_path) - 1);

    unlink(screen_list->socket_path[socktype]);

    if(bind(sock, (struct sockaddr *)&myaddr, sizeof(struct sockaddr_un)) < 0)
    {
        free(screen_list->socket_path[socktype]);
        screen_list->socket_path[socktype] = NULL;
        close(sock);
        perror("create_socket:bind");
        return errno;
    }
    fcntl(sock, F_SETFL, O_NONBLOCK);

    debug("Listening on %s (%d)", screen_list->socket_path[socktype], sock);

    screen_list->socket[socktype] = sock;

    return 0;
}

char ** list_sockets(char *socket_dir, char *session_name)
{
    char *pattern, *dir;
    glob_t globbuf;

    globbuf.gl_pathv = NULL;

    pattern = (char *)malloc(PATH_MAX+1);

    dir = socket_dir;
    if(!dir)
        dir = getenv("NEERCSDIR");
    if(!dir)
        dir = getenv("TMPDIR");
    if(!dir)
        dir = "/tmp";

    if(!pattern)
        return globbuf.gl_pathv;

    if(session_name && strlen(session_name)+strlen(dir)+13<PATH_MAX)
        sprintf(pattern, "%s/neercs.%s.srv.sock", dir, session_name);
    else
        snprintf(pattern, PATH_MAX, "%s/neercs.*.srv.sock", dir);
    pattern[PATH_MAX] = '\0';

    debug("Looking for sockets in the form %s", pattern);

    glob(pattern, 0, NULL, &globbuf);

    free(pattern);

    return globbuf.gl_pathv;
}

char * connect_socket(struct screen_list* screen_list, enum socket_type socktype)
{
    int sock;
    struct sockaddr_un addr;
    char *p, *s;

    debug("Connecting to %s", screen_list->socket_path[socktype]);

    /* Open the socket */
    if ((sock = socket(AF_UNIX, SOCK_DGRAM, 0)) < 0) {
        perror("connect_server:socket");
        return NULL;
    }

    memset(&addr,0,sizeof(addr));
    addr.sun_family = AF_UNIX;
    strcpy(addr.sun_path,screen_list->socket_path[socktype]);
    if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0)
    {
        fprintf(stderr, "Failed to connect to %s: %s\n", screen_list->socket_path[socktype], strerror(errno));
        close(sock);
        return NULL;
    }
    fcntl(sock, F_SETFL, O_NONBLOCK);

    screen_list->socket[socktype] = sock;

    if(socktype == SOCK_SERVER)
    {
        p = strrchr(screen_list->socket_path[socktype], '/');
        p+=8; /* skip neercs. */
        s = strdup(p);
        p = strrchr(s, '.');
        *p = '\0'; /* drop .sock */
        p = strrchr(s, '.');
        *p = '\0'; /* drop .srv */
        p = strdup(s);
        free(s);
        return p;
    }
    else
        return NULL;
}

int request_attach(struct screen_list* screen_list)
{
    char buf[32];
    int bytes;

    bytes = snprintf(buf, sizeof(buf)-1, "ATTACH %10d%c%10d",
                     caca_get_canvas_width(screen_list->cv),
                     ' ',
                     caca_get_canvas_height(screen_list->cv));
    buf[bytes] = '\0';
    debug("Requesting attach: %s", buf);
    return write(screen_list->socket[SOCK_SERVER], buf, strlen(buf)) <= 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);
        return write(screen_list->socket[SOCK_SERVER], buf, strlen(buf)) <= 0;
    }
    else if(t & CACA_EVENT_RESIZE)
    {
        char buf[32];
        int bytes;
        bytes = snprintf(buf, sizeof(buf)-1, "RESIZE %10d%c%10d",
                         caca_get_event_resize_width(&ev),
                         ' ',
                         caca_get_event_resize_height(&ev));
        buf[bytes] = '\0';
        debug("Sending resize to server: %s", buf);
        return write(screen_list->socket[SOCK_SERVER], buf, strlen(buf)) <= 0;
    }
    else if(t & CACA_EVENT_QUIT)
        return write(screen_list->socket[SOCK_SERVER], "QUIT", strlen("QUIT")) <= 0;

    return 0;
}