#include <cstdio>
#include <cerrno>
#include <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;
}