#include <stdio.h>
#include <kernel/OS.h>
#include <string.h>
#include <sys/time.h>
#include <errno.h>

#include "sys/socket.h"
#include "netinet/in.h"
#include "arpa/inet.h"
#include "sys/select.h"

#include "ufunc.h"

int main(int argc, char **argv)
{
        int s;
        int rv;
        struct fd_set fdr, fdw, fde;
        struct timeval tv;
        int32 rtc;
        
        test_banner("Select test");
                        
        s = socket(AF_INET, SOCK_DGRAM, 0);
        if (s < 0)
                err(s, "Socket creation failed");

        FD_ZERO(&fdr);
        FD_SET(s, &fdr);
        FD_ZERO(&fdw);
        FD_SET(s, &fdw);
        FD_ZERO(&fde);
        FD_SET(s, &fde);
        
        tv.tv_sec = 5;
        tv.tv_usec = 0;
        
        printf("Trying with timeval (5 secs)...\n");
        rtc = real_time_clock();
        rv = select(s + 1, &fdr, NULL, &fde, &tv);
        rtc = real_time_clock() - rtc;
        printf("select gave %d in %ld seconds\n", rv, rtc);
        if (rv < 0)
                printf("errno = %d [%s]\n", errno, strerror(errno));

        printf("resetting select fd's\n");
        FD_ZERO(&fdr);
        FD_SET(s, &fdr);
        FD_ZERO(&fdw);
        FD_SET(s, &fdw);
        FD_ZERO(&fde);
        FD_SET(s, &fde);
        
        printf("Trying without timeval (= NULL)\n");
        rv = select(s +1, &fdr, &fdw, &fde, NULL);
        printf("select gave %d\n", rv);
        
        if (FD_ISSET(s, &fdr))
                printf("Data to read\n");
        if (FD_ISSET(s, &fdw))
                printf("OK to write\n");
        if (FD_ISSET(s, &fde))
                printf("Exception!\n");
                
        closesocket(s);
        
        printf("Test complete.\n");

        return (0);
}