#define _GNU_SOURCE

#include <Conn.h>

/* Global variables */
static unsigned short	debug = 20;
static unsigned int	ipv4conns = 500, ipv6conns = 500;

static FILE	*Logf = NULL;
static char	*log_file = "c.log";
static int	port = 9000;
static unsigned int	answers = 0;

struct priv
{
	unsigned int	dummy;
};


static void c_connected(struct Conn *C)
{
	struct priv	*p;

	Log(4, "%s(A connection was estabilished on slot %d)\n",
		__FUNCTION__, C->slot);

	C->private = malloc(sizeof(struct priv));
	if (C->private == NULL) {
		Log(0, "cannot alloc memory!\n");
		exit(1);
	}

	p = (struct priv *) C->private;
	p->dummy = 0;
}

static void c_close(struct Conn *C)
{
	Log(5, "%s: id=%llu will close...\n",
		__FUNCTION__, Conn_getid(C));
	free(C->private);
}

static int c_data_cb(struct Conn *C, char *line)
{
	struct priv *p;
	char *dump;

	p = (struct priv *) C->private;

	if (debug >= 8) {
		dump = Conn_dump(line, strlen(line));
		Log(8, "data: recv: %s\n", dump);
		free(dump);
	}

	p->dummy++;

	answers++;

	return 0;
}

static void c_data(struct Conn *C)
{
	Conn_for_every_line(C, c_data_cb);

	Conn_close(C);
}

static void c_error(struct Conn *C)
{
	Log(0, "%s id=%llu [%s]\n",
		__FUNCTION__, Conn_getid(C), Conn_strerror());
}

int main(void)
{
	int ret;
	struct timeval start, end;
	unsigned int elap;
	unsigned int i;
	struct Conn *C4, *C6;

	Logf = fopen(log_file, "a+");
	if (!Logf) {
		fprintf(stderr, "Cannot open log file [%s] [%s]\n",
			log_file, strerror(errno));
		return 1;
	}
	if (debug > 0)
		setlinebuf(Logf);

	Conn_debug(Logf, debug);

	Log(0, "Starting...\n");
	Log(0, "\tPort=%d\n", port);
	Log(0, "\tLogFile=%s Debug=%d\n", log_file, debug);

	ret = Conn_init(0);
	if (ret == -1) {
		Log(0, "%s", Conn_strerror());
		return 1;
	}

	Conn_connected_cb = c_connected;
	Conn_data_cb = c_data;
	Conn_close_cb = c_close;
	Conn_error_cb = c_error;

	gettimeofday(&start, NULL);

	answers = 0;

	/* ipv4 */
	for (i = 0; i < ipv4conns; i++) {
		C4 = Conn_connect(PF_INET, SOCK_STREAM, "127.0.0.1", port);
		if (C4 == NULL) {
			Log(0, "Error calling Conn_connect [%s]!\n",
				Conn_strerror());
			return 1;
		}
	}

	/* ipv6 */
	for (i = 0; i < ipv6conns; i++) {
		C6 = Conn_connect(PF_INET6, SOCK_STREAM, "::1", port);
		if (C6 == NULL) {
			Log(0, "Error calling Conn_connect6 [%s]!\n",
				Conn_strerror());
			return 1;
		}
	}

	while (1) {
		ret = Conn_poll(1000);
		if (ret == -1) {
			Log(0, "Error in poll [%s]!\n",
				Conn_strerror());
			break;
		}

		printf("%u answer(s)\n", answers);

		if (answers == ipv4conns + ipv6conns)
			break;
	}

	gettimeofday(&end, NULL);

	elap = 1000000 * (end.tv_sec - start.tv_sec) + end.tv_usec - start.tv_usec;
	Log(0, "Finish in %ld.%06ld. ipv4conns=%d. ipv6conns=%d. Total=%d. answers=%d"
		" Conns per sec = %d\n",
		elap / 1000000, elap % 1000000,
		ipv4conns, ipv6conns, ipv4conns + ipv6conns,
		answers, (ipv4conns + ipv6conns) / (elap / 1000000 + 1));

	Conn_shutdown();

	fclose(Logf);

	return 0;
}