#include <cstdio></cstdio>

#include <cerrno></cerrno>

#include <cstring></cstring>

#include "tunnel.h"

bool parseAddress(const char *str, Address &address) {

	address = Address();

	sscanf( str, "%hhu.%hhu.%hhu.%hhu:%hu",

			&address.ip[0], &address.ip[1], &address.ip[2], &address.ip[3],

			&address.port );

	if (!address.port) {

		printf("Wrong ip-address: [%s]\n", str);

		return false;

	}

	return true;

}

int main(int argc, char **argv) {

	// parse args

	unsigned long long bytesPerSecond = 0;

	Address localUdp, remoteUdp, localTcp, remoteTcp;

	if (argc == 6) {

		sscanf(argv[2], "%llu", &bytesPerSecond);

		if (!bytesPerSecond)

			{ printf("Wrong bytes per second: %llu\n", bytesPerSecond); return 1; }

		parseAddress(argv[3], localUdp);

		parseAddress(argv[4], remoteUdp);

		parseAddress(argv[5], localTcp);

		if (!localUdp.port || !remoteUdp.port || !localTcp.port) return 1;

	} else

	if (argc == 5) {

		sscanf(argv[2], "%llu", &bytesPerSecond);

		if (!bytesPerSecond)

			{ printf("Wrong bytes per second: %llu\n", bytesPerSecond); return 1; }

		parseAddress(argv[3], localUdp);

		parseAddress(argv[4], remoteTcp);

		if (!localUdp.port || !remoteTcp.port) return 1;

	} else {

		printf("Usage: %s keyfile bytesPerSecond localUdp:port remoteUdp:port localTcp:port\n", argv[0]);

		printf("   or: %s keyfile bytesPerSecond localUdp:port remoteTcp:port\n", argv[0]);

		return 1;

	}

	// load key

	FILE *f = fopen(argv[1], "r");

	if (!f) {

		printf("Cannot open keyfile: [%s]\n", argv[1]);

		return 1;

	}

	int maxSize = 1024*1024;

	char *key = (char*)calloc(1, maxSize);

	char *c = key, *e = c + maxSize - 1;

	int i = 0;

	while((i = fgetc(f)) > 0 && c < e) *c++ = i;

	if (i != EOF || errno) {

		printf("Cannot read keyfile: [%s]\n", argv[1]);

		memset(key, 0, maxSize);

		free(key);

		fclose(f);

	}

	fclose(f);

	// init tunnel

	Tunnel tunnel;

	tunnel.initSpeed(bytesPerSecond);

	if ( !tunnel.initUdpServer(localUdp, remoteTcp, key) ) {

		memset(key, 0, maxSize);

		free(key);

		return 1;

	}

	memset(key, 0, maxSize);

	free(key);

	if ( localTcp.port && remoteUdp.port

	  && !tunnel.initTcpServer(localTcp, remoteUdp) )

		return 1;

	tunnel.run();

	return 0;

}