#include <errno.h>

#include "smud.h"

#include "smud_threads.h"

#define LISTEN_PORT 6500

#define LISTEN_QUEUE_LENGTH 6

char fileNames[NUM_FILES][30] = {

	"core.games.txt",

	"sdktools.games.txt",

	"sdktools.games.ep2.txt",

	"sdktools.games.l4d.txt",

	"sm-cstrike.games.txt",

	"sm-tf2.games.txt",

};

void *fileLocations[NUM_FILES];

int fileLength[NUM_FILES];

int main(int argc, char **argv)

{

	ThreadPool *pool;

	struct protoent *pProtocol;

	struct sockaddr_in serverAddress;

	struct sockaddr_in clientAddress;

	int serverSocket;

	int clientSocket;

	int addressLen;

	int opts;

	int file;

	char filename[100];

	struct stat sbuf;

#if defined DEBUG

	fprintf(stdout, "Loading Gamedata files into memory\n");

#endif

	for (int i=0; i<NUM_FILES; i++)

	{

		snprintf(filename, sizeof(filename), "./md5/%s", fileNames[i]);

		file = open(filename, O_RDWR);

		if (!file)

		{

			fprintf(stderr, "Could not find file: %s", filename);

			return 1;

		}

		if (stat(filename, &sbuf) == -1)

		{

			fprintf(stderr, "Could not stat file: %s (error: %s)", filename, strerror(errno));

			return 1;

		}

		if ((fileLocations[i] = mmap(NULL, sbuf.st_size, PROT_READ, MAP_PRIVATE, file, 0)) == MAP_FAILED)

		{

			fprintf(stderr, "Could not mmap file: %s (error: %s)", filename, strerror(errno));

			return 1;

		}

		fileLength[i] = sbuf.st_size;

#if defined DEBUG

		fprintf(stdout, "Initialised file of %s of length %i\n", fileNames[i], fileLength[i]);

#endif

	}

#if defined DEBUG

	fprintf(stdout, "Initializing Thread Pool\n");

#endif

	pool = new ThreadPool();

	if (!pool->Start())

	{

		fprintf(stderr, "Could not initialize thread pool!\n");

		return 1;

	}

#if defined DEBUG

	printf("Create Server Socket\n");

#endif

	memset(&serverAddress, 0, sizeof(serverAddress));

	serverAddress.sin_family = AF_INET;

	serverAddress.sin_addr.s_addr = INADDR_ANY;

	serverAddress.sin_port = htons(LISTEN_PORT);

	pProtocol = getprotobyname("tcp");

	if (pProtocol == NULL)

	{

		fprintf(stderr, "Could not get tcp proto: %s", strerror(errno));

		return 1;

	}

	serverSocket = socket(AF_INET, SOCK_STREAM, pProtocol->p_proto);

	if (serverSocket < 0) 

	{

		fprintf(stderr, "Could not open socket: %s", strerror(errno));

		return 1;

	}

	opts = 1;

	setsockopt(serverSocket, SOL_SOCKET, SO_REUSEADDR, &opts, sizeof(opts));

	if (bind(serverSocket, (struct sockaddr *)&serverAddress, sizeof(serverAddress)) < 0) 

	{

		fprintf(stderr, "Could not bind socket: %s", strerror(errno));

		return 1;

	}

	if (listen(serverSocket, LISTEN_QUEUE_LENGTH) < 0) 

	{

		fprintf(stderr, "Could not listen on socket: %s", strerror(errno));

		return 1;

	}

	fprintf(stdout, "Server has started.\n");

	while (1)

	{

		addressLen = sizeof(clientAddress);

		clientSocket = accept(serverSocket,

							  (struct sockaddr *)&clientAddress,

							  (socklen_t *)&addressLen);

		if (clientSocket < 0)

		{

			fprintf(stderr, "Could not accept client: %s", strerror(errno));

			continue;

		}

		opts = fcntl(clientSocket, F_GETFL, 0);

		if (fcntl(clientSocket, F_SETFL, opts|O_NONBLOCK) < 0)

		{

			fprintf(stderr, "Could not non-block client: %s", strerror(errno));

			closesocket(clientSocket);

			continue;

		}

#if defined DEBUG

		fprintf(stdout,

				"Accepting connection from client (sock %d, ip %s)",

				clientSocket,

				inet_ntoa(clientAddress.sin_addr));

#endif

		pool->AddConnection(clientSocket);

	}

	delete pool;

}