#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <aitsched.h>

intptr_t Kill;

void *event(sched_task_t *arg)
{
	printf("Event::\n");
	return NULL;
}

void *eventlo(sched_task_t *arg)
{
	printf("EventLOW::\n");
	return NULL;
}

void *timer(sched_task_t *arg)
{
	printf("Timer %p sec::\n", TASK_ARG(arg));
	return NULL;
}

void *r(sched_task_t *arg)
{
	printf("read:: bytes\n");
	Kill++;
	return NULL;
}

void *w(sched_task_t *arg)
{
	printf("write::\n");
	return NULL;
}

void *once(sched_task_t *arg)
{
	printf("once::\n");
	return NULL;
}

void *aioread(sched_task_t *arg);
void *aiowrite(sched_task_t *arg)
{
	char *ole = malloc(BUFSIZ);

	printf("AIO write[%d]:: %d bytes\n%p\n", TASK_FD(arg), (int) TASK_DATLEN(arg), 
			TASK_DATA(arg));
	free(TASK_DATA(arg));

	schedAIORead(TASK_ROOT(arg), aioread, NULL, TASK_FD(arg), ole, BUFSIZ);
	return NULL;
}

void *aioread(sched_task_t *arg)
{
	char *ole = malloc(BUFSIZ);
	int len;

	printf("AIO read[%d]:: %d bytes\n%s\n-------\n", TASK_FD(arg), (int) TASK_DATLEN(arg), 
			(char*) TASK_DATA(arg));

	if (TASK_ARG(arg)) {
//		write((int) TASK_ARG(arg), TASK_DATA(arg), TASK_DATLEN(arg));

		len = strlcpy(ole, "++++++BAHURA OR CULTURE .... A CULTURE OR BAHURA :-)\n", BUFSIZ);
		printf("sched Write len=%d %p\n", len, ole);
		schedAIOWrite(TASK_ROOT(arg), aiowrite, TASK_ARG(arg), TASK_FD(arg), ole,  len);
				
	}
	free(TASK_DATA(arg));
	return NULL;
}

void sig(int s)
{
	switch (s) {
		case SIGTERM:
			Kill++;
			break;
	}
}

int
main(int argc, char **argv)
{
	sched_root_task_t *root;
	int f, fd;
	struct sockaddr_in sin;
	struct timespec ts = { 20, 0 };
//	struct timespec p = { 0, 10000000 };
	struct sigaction sa;
	char *ole = malloc(BUFSIZ);

	sa.sa_handler = sig;
	sigemptyset(&sa.sa_mask);
	sigaction(SIGTERM, &sa, NULL);

	f = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
	if (f == -1)
		return 1;
	sin.sin_len = sizeof sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(2345);
	sin.sin_addr.s_addr = INADDR_ANY;
	if (bind(f, (struct sockaddr*) &sin, sizeof sin) == -1)
		return 1;

	fd = open("test_aio.dat", O_CREAT | O_RDWR, 0644);
	if (fd == -1)
		return 1;
	printf("fd=%d\n", fd);

	root = schedBegin();
	if (!root) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 1;
	}

	if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 4;
	} else
		ts.tv_sec = 15;
	if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 4;
	} else
		ts.tv_sec = 10;

	if (!schedEvent(root, event, "piuk", 1234, NULL, 0)) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 2;
	}

	if (!schedEventLo(root, eventlo, "piuk", 1111, NULL, 0)) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 3;
	}

	if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 4;
	}

	if (!schedRead(root, r, "rrr", f, NULL, 0)) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 5;
	}

	if (!schedWrite(root, w, "www", f, NULL, 0)) {
		printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
		return 6;
	}

	if (!schedAIORead(root, aioread, (void*) f, fd, ole, BUFSIZ))
		printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());

	schedCallOnce(root, once, "000000", 42, NULL, 0);

//	schedPolling(root, &p, NULL);
	schedRun(root, &Kill);
	schedEnd(&root);

	close(fd);
	close(f);
	return 0;
}