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

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

#include "ufunc.h"

#define THREADS	2
#define TIME	10


int32 test_thread(void *data)
{
	int tnum = *(int*)data;
	int sock = 0;
	uint32 num = 0;
	int rv;
	struct sockaddr_in sa;
	bigtime_t tn;

	sa.sin_len = sizeof(sa);
	sa.sin_port = 0;
	sa.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
	sa.sin_family = AF_INET;
	memset(&sa.sin_zero, 0, sizeof(sa.sin_zero));
	
	printf("Thread %d, starting test...\n", tnum + 1);

	tn = real_time_clock();

	while (real_time_clock() - tn <= TIME) {
		sock = socket(AF_INET, SOCK_DGRAM , 0);
		if (sock < 0)
			err(sock, "Socket couldn't be created");
		rv = bind(sock, (struct sockaddr *)&sa, sizeof(sa));
		if (rv < 0)
			err(rv, "Socket could not be bound to an ephemereal port");
		closesocket(sock);
		num++;
	}

	printf( "Thread %d:\n"
		"       sockets created : %5ld\n"
		"       test time       : %5d seconds\n"
		"       average         : %5ld sockets/sec\n",
		tnum + 1, num, TIME, num / TIME);
}

int main(int argc, char **argv)
{
	thread_id t[THREADS];
	int i;
	status_t retval;

	test_banner("Socket creation and bind() test");
	
	for (i=0;i<THREADS;i++) {
		t[i] = spawn_thread(test_thread, "socket test thread", 
			B_NORMAL_PRIORITY, &i);
		if (t[i] >= 0)
			resume_thread(t[i]);
	}

	for (i=0;i<THREADS;i++) {
		wait_for_thread(t[i], &retval);
	}

	return (0);
}