#ifndef MY_WINSOCK_FCTS
#define MY_WINSOCK_FCTS

#include <windows.h>
#include <winsock.h>

//Connect to a server, return socket
SOCKET ConnectServ(const char *servername, unsigned short portnum)
{
	struct sockaddr_in sa;
	struct hostent *hp;
	unsigned long addr;
	SOCKET s;
	
	if ( (addr = inet_addr(servername)) == INADDR_NONE )
		hp = gethostbyname(servername); //a DNS like battle.net
		
	else
		hp = gethostbyaddr((char*)&addr, sizeof(addr), AF_INET); //An IP address	
	
	if ( hp == NULL )
		return INVALID_SOCKET;
		
	memset(&sa, 0, sizeof(sa) );
	memcpy((char*)&sa.sin_addr, hp->h_addr, hp->h_length);
	sa.sin_family = hp->h_addrtype;
	sa.sin_port = htons((u_short)portnum);
	
	s = socket(hp->h_addrtype, SOCK_STREAM, 0);	
	if ( s == INVALID_SOCKET )
		return INVALID_SOCKET;
		
	if ( connect(s, (struct sockaddr *)&sa, sizeof(sa)) == SOCKET_ERROR )
	{
		closesocket(s);
		return INVALID_SOCKET;
	}
	
	return s;
}

#endif

// The following functions are exported from Arse Master.exe
// to be used by the Game DLL.

#ifndef ARSEMASTER_INTERFACE
#define ARSEMASTER_INTERFACE

#include "Arse Master.h"
#include "global.h"
#include "intern.h"

#define DllExport   __declspec( dllexport )
//Prototypes
extern "C" {
	DllExport void NewGame( int color );
	DllExport void PlayerMove( char *move, bool first );
	DllExport void Quit();	
} //extern "C"

/** Functions **/

//Use this function to create a new chess game.
//It runs The King.exe and initializes the pipes communication
DllExport void NewGame( int color )
{
	HANDLE hThread = 0;

	GameColor = color; //Put in global variable

	CArseMasterDlg am;
	HANDLE hReadPipe1, hReadPipe2, hWritePipe1, hWritePipe2;
	SECURITY_ATTRIBUTES SecAttr = {0};

	HANDLE hProcess = GetCurrentProcess();

	SecAttr.nLength = sizeof( SECURITY_ATTRIBUTES );
	SecAttr.bInheritHandle = TRUE;
	SecAttr.lpSecurityDescriptor = NULL;

	if ( !CreatePipe(&hReadPipe1,&hWritePipe1,&SecAttr,0) )
	{
		MessageBox(hWnd, "Error for pipes 1", Error, MB_OK | MB_ICONERROR);
		am.OnTerminate();
		return ;
	}

	if ( !DuplicateHandle(hProcess,hReadPipe1,hProcess,&MainRead,0,0,2) )
	{
		MessageBox(hWnd, "Error when duplicating first handle", Error, MB_OK | MB_ICONERROR);
		am.OnTerminate();
		return ;
	}

	CloseHandle(hReadPipe1);

	SecAttr.nLength = sizeof( SECURITY_ATTRIBUTES );
	SecAttr.bInheritHandle = TRUE;
	SecAttr.lpSecurityDescriptor = NULL;

	if ( !CreatePipe(&hReadPipe2,&hWritePipe2,&SecAttr,0) )
	{
		MessageBox(hWnd, "Error for pipes 2", Error, MB_OK | MB_ICONERROR);
		am.OnTerminate();
		return ;
	}

	if ( !DuplicateHandle(hProcess,hWritePipe2,hProcess,&MainWrite,0,0,2) )
	{
		MessageBox(hWnd, "Error when duplicating second handle", Error, MB_OK | MB_ICONERROR);
		am.OnTerminate();
		return ;
	}

	CloseHandle(hWritePipe2);

	PROCESS_INFORMATION pi = {0};
	STARTUPINFO si = {0};
	si.cb = sizeof(STARTUPINFO);
	si.dwFlags = STARTF_USESTDHANDLES;
	si.hStdInput = hReadPipe2;
	si.hStdOutput = hWritePipe1;
	si.hStdError = hWritePipe1;

	if ( !CreateProcess(TheKingPath, "-l- -a- -m-", 0, 0, TRUE, DETACHED_PROCESS | CREATE_NEW_PROCESS_GROUP, 0, 0, &si, &pi) )
	{
		MessageBox(hWnd, "Couldn't find The King.exe", Error, MB_OK | MB_ICONERROR);
		am.OnTerminate();
		return ;
	}

	CloseHandle(hWritePipe1);
	CloseHandle(hReadPipe2);

	/** Initiate communication **/
	#define LINES 13
	DWORD bytes;
	char init[13][60] = { {"xboard\n"},
			       {"post\n"},
			       {"new\n"},
			       {"level 0 1 0\n"},
			       {"cm_parm default\n"},
			       {"cm_parm opp=100 opn=100 opb=100 opr=100 opq=100\n"},
			       {"cm_parm myp=100 myn=100 myb=100 myr=100 myq=100\n"},
			       {"cm_parm cc=100 mob=100 ks=100 pp=100 pw=100\n"},
			       {"cm_parm cfd=0 sop=100 avd=0 rnd=0 sel=6 md=99\n"},
			       {"cm_parm tts=1048576 ttu=2\n"},
			       {"hard\n"},
			       {"cm_parm myk=877654 opk=298584\n"},
			       {"force\n"} };
			       
	for (int x = 0; x < LINES; x++)
		WriteFile(MainWrite,init[x], strlen(init[x]),&bytes,0);

	if ( color == WHITE ) {
		WriteFile(MainWrite,"force\n",6,&bytes,0);
		RandomFirstMove();
	}

	DWORD threadID;
	hThread = CreateThread(NULL,NULL,ReadPipeFunc,0,0,&threadID);
}

//The Game DLL uses this function to tell ArseMaster what move the
//other player just made.
//Move format: e2e4\0
//Or : a7a8q\0 (Other side reached, change into queen)
DllExport void PlayerMove( char* move, bool first )
{
	char movetmp[10];
	DWORD bytes;
	int len;

	len = strlen( move );	
	if ( len < 4 || len > 5 )
		return; //Invalid format
		
	strcpy(movetmp, move);
	movetmp[len] = '\n';
	WriteFile(MainWrite,movetmp,len+1,&bytes,0);

	char* white = "white\n";
	char* black = "black\n";

	if ( first )
	{
		WriteFile(MainWrite,white,6,&bytes,0);

		if ( GameColor == WHITE )
			WriteFile(MainWrite,white,6,&bytes,0);

		else
			WriteFile(MainWrite,black,6,&bytes,0);

		WriteFile(MainWrite,"go\n",3,&bytes,0);
	}
}

//Use this function to quit the current game (Game with other player is over).
//This will close The King.exe and the pipes handles, until the next NewGame.
DllExport void Quit()
{
	char* quit = "quit\n";
	DWORD bytes;

	WriteFile(MainWrite,quit,strlen(quit),&bytes,0);
	CloseHandle(MainWrite);
	CloseHandle(MainRead);
}

#endif

//Functions used by ArseMaster.

#ifndef INTERN_FUNCTIONS
#define INTERN_FUNCTIONS

#include <time.h>
#include <stdlib.h>
#include "global.h"

// Prototypes
DWORD WINAPI ReadPipeFunc(LPVOID param); //Thread to read datas from pipe
void RandomFirstMove(); //Make a random first move when playing white

/** Functions **/

//Thread to read datas from pipe
DWORD WINAPI ReadPipeFunc(LPVOID param)
{
	#define BUF_SIZE 10000
	char tmp[BUF_SIZE];
	DWORD bytes;
	char* result;
	char move[6];
	int len;

	while( ReadFile(MainRead,tmp,BUF_SIZE,&bytes,0) && bytes > 0 )
	{
		tmp[bytes] = '\0';
		result = strstr(tmp, "move ");
		if ( result )
		{
			result += 5;
			for (len = 0; result[len] != (char)0x0d; len++);
			strncpy(move, result, len);
			move[len] = '\0';
			MakeMove( move ); //Game DLL
		}
	}

	return 0;
}

//When creating a new game and we play white.
//Select a random move from a table and tell the Game DLL of it.
void RandomFirstMove()
{
	char moves[5][6] = { {"g1f3\n"},
			     {"c2c4\n"},
			     {"d2d4\n"},
			     {"e2e4\n"},
			     {"b1c3\n"} };
	char move[5];
	DWORD bytes;
			     
	srand( time(0) );
	int nb = rand() % 5;
	
	WriteFile(MainWrite,moves[nb],5,&bytes,0);
	strncpy(move, moves[nb], 5);
	move[4] = '\0';
	MakeMove( move ); //Tell Game Dll	
}

#endif

#ifndef GLOBAL_H
#define GLOBAL_H

HWND hWnd; //Handle to main window

//GameDll exported functions
HINSTANCE GameDll; //For LoadLibrary
typedef bool (WINAPI *fnStart)();
typedef void (WINAPI *fnMakeMove)( char* move );
fnStart Start;			//Pointer to Start function
fnMakeMove MakeMove;		//Pointer to MakeMove function

//Handles for pipes
HANDLE MainRead;
HANDLE MainWrite;

#define WHITE 1
#define BLACK 2

char* Error = "Error";
char path[300];
char TheKingPath[300]; //The directory that The King is in!

int GameColor;

#endif