source: neercs/trunk/src/attach.c @ 2588

#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 "neercs.h"

char * build_socket_path(char *socket_dir, char *session_name)
{
    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.sock", dir, session_name);
    return path;
}

int create_socket(struct screen_list* screen_list)
{
    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, sizeof(myaddr.sun_path) - 1);

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

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

    screen_list->socket = 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.sock", dir, session_name);
    else
        snprintf(pattern, PATH_MAX, "%s/neercs.*.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_server(char *socket_path, struct screen_list* screen_list)
{
    int sock;
    struct sockaddr_un addr;

    debug("Connecting to %s", socket_path);

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

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

    screen_list->socket_path = socket_path;
    screen_list->socket = sock;

    return strdup(socket_path); /* FIXME */
}

int request_attach(struct screen_list* screen_list)
{
    debug("Requesting attach");

    write(screen_list->socket, "ATTACH", strlen("ATTACH"));

    return 0;
}

int send_event(caca_event_t ev, int sock)
{
    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(sock, buf, strlen(buf)) <= 0;
    }
    else if(t & CACA_EVENT_RESIZE)
        return write(sock, "RESIZE", strlen("RESIZE")) <= 0;
    else if(t & CACA_EVENT_QUIT)
        return write(sock, "QUIT", strlen("QUIT")) <= 0;

    return 0;
}