source: nanoscale/branches/1.0/client.c @ 6295

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <sys/select.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <getopt.h>

#include "clientlib.h"

/*
 * min()/max() macros that also do
 * strict type-checking.. See the
 * "unnecessary" pointer comparison.
 */
#define min(x,y) ({                             \
            typeof(x) _x = (x);                 \
            typeof(y) _y = (y);                 \
            (void) (&_x == &_y);                \
            _x < _y ? _x : _y; })

#define max(x,y) ({                             \
            typeof(x) _x = (x);                 \
            typeof(y) _y = (y);                 \
            (void) (&_x == &_y);                \
            _x > _y ? _x : _y; })

void help(const char *argv0)
{
    fprintf(stderr,
            "Syntax: %s addr:port load\n",
            argv0);
    exit(1);
}

int connect_fd = -1;

void 
activity(int signum)
{
    write(connect_fd, "hello\n\r", 7);
    write(1,".",1);
}


int 
main(int argc, char *argv[])
{
    int status;
    int connect_port = -1;

    if (argc != 3) {
        help(argv[0]);
    }

    char addr[100];
    if (sscanf(argv[1], "%100[^:\n]:%d\n", addr, &connect_port) != 2) {
        help(argv[0]);
    }
    int memory = strtoul(argv[2],0,0);

    connect_fd = connect_renderer(addr, connect_port, memory);

    if (connect_fd == -1) {
        return 1;
    }

    printf("Connection accepted.\n");

#ifdef DEBUGGING
    if (signal(SIGALRM, activity) == SIG_ERR) {
        perror("signal");
    }

    struct itimerval itvalue = {
        {0,500000}, {0,500000}
    };
    status = setitimer(ITIMER_REAL, &itvalue, NULL);
#endif

    fd_set rfds;
    FD_ZERO(&rfds);
    while(1) {
        char buffer[100];
        FD_SET(0,&rfds);
        FD_SET(connect_fd,&rfds);
        status = select(connect_fd+1, &rfds, NULL, NULL, NULL);
        if (FD_ISSET(0,&rfds)) {
            status = read(0,buffer,sizeof(buffer));
            if (status <= 0) {
                fprintf(stderr,"Goodbye\n");
                exit(0);
            } else {
                status = write(connect_fd, buffer, status);
                if (status < 0) {
                    perror("write");
                }
            }
        } else if (FD_ISSET(connect_fd,&rfds)) {
            status = read(connect_fd,buffer,sizeof(buffer));
            if (status <= 0) {
                fprintf(stderr,"Server closed connection\n");
                exit(0);
            } else {
                status = write(1, buffer, status);
                if (status < 0) {
                    perror("write");
                }
            }
        } else {
            fprintf(stderr,"Strange case for select: %d\n", status);
            exit(1);
        }
    }

    return 0;
}