haiku.git

#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, f;
	int rv;
	struct fd_set fdr, fdw, fde;
	struct timeval tv;
	int32 rtc;
	char path[PATH_MAX];
		
	test_banner("Select Test #2");
			
	s = socket(AF_INET, SOCK_DGRAM, 0);
	if (s < 0)
		err(s, "Socket creation failed");

	getcwd(path, PATH_MAX);
	sprintf(path, "%s/select_test2.c", path);
	f = open(path, O_RDWR);

	if (f > 0 && s > 0)	{
		printf("\nsocket and fd created.\n");
	} else {
		err(-1, "Failed to create socket or fd\n");
	}
	
	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("\nTest1\n=====\n\n");	
	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 (expecting 0) in %ld seconds\n", rv, rtc);

	FD_ZERO(&fdr);
	FD_SET(s, &fdr);
	FD_SET(f, &fdr);
	FD_ZERO(&fdw);
	FD_SET(s, &fdw);
	FD_ZERO(&fde);
	FD_SET(s, &fde);
	
	printf("\nTest2\n=====\n\n");
	printf("Trying without timeval and both sockets and files...\n");
	rv = select(f +1, &fdr, NULL, NULL, NULL);
	printf("select gave %d (expecting 2)\n", rv);
	
	if (rv > 0) {
		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");
		if (FD_ISSET(f, &fdr))
			printf("File is readable!\n");
	} else if (rv == 0) {
		printf("Timed out??? huh?!\n");
	} else {
		printf("errno = %d [%s]\n", errno, strerror(errno));
	}
	
	closesocket(s);
	close(f);
	
	printf("Test complete.\n");

	return (0);
}