#include <windows.h>
#include <Wininet.h>
#include <common\md5.h>
#include "settings.h"
#include "support.h"
#include "Validate Binary.h"

#pragma comment(lib, "Wininet.lib")

#define DEBUG_VALIDATE

LPSTR ValidateDecryptString (LPSTR String, int Len)
{
	BYTE PreviousChar = 0xAA;
	for (int x = 0; x < Len; x++)
	{
		String[x] ^= PreviousChar;
		PreviousChar = String[x];
	}
	return String;
}

LPSTR ValidateEncryptString (LPSTR String, int Len)
{
	BYTE EncryptChar = 0xAA;
	for (int x = 0; x < Len; x++)
	{
		BYTE PreviousChar = String[x];
		String[x] ^= EncryptChar;
		EncryptChar = PreviousChar;
	}
	return String;
}

void TestValidBinaryThread ( )
{
	typedef struct 
	{
		BYTE File_md5[16];
		int  RunTimes;
	} INVALID_RUN_ITEM;

	enum { MAX_BAD_DATA = 50, FAILED_CHECK = 0x7FFFFFFF };
	
	AUTO_PTR<BYTE> RunData;
	INVALID_RUN_ITEM * RunItems = NULL;
	int RunItemCount = 0;
	bool DefaultResult = true;

	CPath ModuleFileName(CPath::MODULE_FILE);
	MD5 File_md5(ModuleFileName);
#ifdef DEBUG_VALIDATE
	WriteTraceF(TraceError,"v1: %s",File_md5.hex_digest());
#endif

	// see if already invalid
	char Subkey[] = "\x84\x5E\x1A\x5C\x02\x4E"; // ".pj64z";
	ValidateDecryptString(Subkey,sizeof(Subkey) - 1);

	{
		HKEY hKeyResults = 0;
		long lResult;

		lResult = RegOpenKey( HKEY_CLASSES_ROOT,Subkey,&hKeyResults);	
		if (lResult == ERROR_SUCCESS) 
		{
			DWORD Type, Length = 0;
			lResult = RegQueryValueEx(hKeyResults,"data",NULL,&Type,NULL,&Length);
			if (lResult == ERROR_SUCCESS && Type == REG_BINARY && Length > 0)
			{
				RunData = AUTO_PTR<BYTE>(new BYTE[Length]);
				lResult = RegQueryValueEx(hKeyResults,"data",NULL,&Type,RunData.get(),&Length);				
				if (lResult == ERROR_SUCCESS && Type == REG_BINARY)
				{
					RunItems = (INVALID_RUN_ITEM *)RunData.get();
					RunItemCount = Length / sizeof(INVALID_RUN_ITEM);

					for (int i = 0; i < RunItemCount; i ++)
					{
						if (memcmp(RunItems[i].File_md5,File_md5.raw_digest(),sizeof(RunItems[i].File_md5)) == 0)
						{
#ifdef DEBUG_VALIDATE
							WriteTraceF(TraceError,"v2: %d",RunItems[i].RunTimes);
#endif
							if (RunItems[i].RunTimes >= MAX_BAD_DATA)
							{
								DefaultResult = false;
							}
							break;
						}
					}
				}
			}		
			RegCloseKey(hKeyResults);
		}	
	}


	//test to see if file has already been disabled
	HINTERNET hSession = InternetOpen("project64", INTERNET_OPEN_TYPE_PRECONFIG,NULL, NULL, NULL);
	if (hSession == NULL)
	{
#ifdef DEBUG_VALIDATE
		WriteTrace(TraceError,"v3");
#endif
		_Settings->SaveDword(IsValidExe,DefaultResult);
		return;
	}

#ifdef VALIDATE_BIN_LOCAL
	char Site[] = { "local.pj64.net" };
	ValidateEncryptString(Site,sizeof(Site) - 1);
#else
	char Site[] = { "\xda\x1a\x5c\x02\x19\x48\x08\x18\x5b\x4d\x0c\x02" }; //"pj64-emu.com"
#endif
	ValidateDecryptString(Site,sizeof(Site) - 1);
	HINTERNET hConnect = InternetConnect(hSession, Site, 80, "", "", INTERNET_SERVICE_HTTP, 0, (LPARAM)0);

	if (hConnect == NULL)
	{
#ifdef DEBUG_VALIDATE
		WriteTrace(TraceError,"v4");
#endif
		_Settings->SaveDword(IsValidExe,DefaultResult);
		InternetCloseHandle (hSession);
		hSession = NULL;
		return;
	}

	char AcceptText[] = "\xDE\x11\x1D\x0C\x5B\x05"; // "text/*";
	ValidateDecryptString(AcceptText,sizeof(AcceptText)  - 1);

	LPCTSTR lpszAcceptTypes[] = 
	{ 
		AcceptText,
		NULL
	};

	char szPost[] = { "\xFA\x1F\x1C\x07" }; // "POST"
	ValidateDecryptString(szPost,sizeof(szPost)  - 1);

	// "index.php?option=com_validate_pj64"
	char SiteLoc[] = { "\xc3\x07\x0a\x01\x1d\x56\x5e\x18\x18\x4f\x50\x1f\x04\x1d\x06\x01\x53\x5e\x0c\x02\x32\x29\x17\x0d\x05\x0d\x05\x15\x11\x3a\x2f\x1a\x5c\x02" };
//	char SiteLoc[] = { "index.php?option=com_validate_pj64" };

//	ValidateEncryptString(SiteLoc,sizeof(SiteLoc)  - 1);
	ValidateDecryptString(SiteLoc,sizeof(SiteLoc)  - 1);


	HINTERNET hRequest = HttpOpenRequest(hConnect, szPost, SiteLoc, NULL, NULL, lpszAcceptTypes, INTERNET_FLAG_PRAGMA_NOCACHE, (LPARAM)0);
	if (hRequest == NULL)
	{
#ifdef DEBUG_VALIDATE
		WriteTrace(TraceError,"v5");
#endif
		_Settings->SaveDword(IsValidExe,DefaultResult);
		InternetCloseHandle (hRequest);
		return;
	}

	char ComputerName[256];
	DWORD Length = sizeof(ComputerName);
	GetComputerName(ComputerName,&Length);

	stdstr_f PostInfo("1,%s,%s,%s,%s,%s,%s",VALIDATE_BIN_APP,File_md5.hex_digest(),ComputerName,VersionInfo(VERSION_PRODUCT_VERSION).c_str(),_Settings->LoadString(BetaUserName).c_str(),_Settings->LoadString(BetaEmailAddress).c_str());
	
	//"Content-Type: application/x-www-form-urlencoded"
    char ContentType[] = { "\xE9\x2C\x01\x1A\x11\x0B\x1A\x59\x79\x2D\x09\x15\x5F\x1A\x41\x11\x00\x1C\x05\x0A\x02\x15\x1D\x06\x01\x41\x57\x55\x5A\x00\x00\x5A\x4B\x09\x1D\x1F\x40\x58\x07\x1E\x09\x0B\x0D\x0C\x0B\x01\x01" }; 
	ValidateDecryptString(ContentType,sizeof(ContentType) - 1);


	//encrypt PostData
	stdstr_f PostData("%s=","Data");
	BYTE * Input = (BYTE *)PostInfo.c_str();
	BYTE PreviousChar = 0xAA;
	for (int x = 0; x < PostInfo.size(); x++)
	{
		PostData += stdstr_f("%02X",(BYTE)(Input[x] ^ PreviousChar));
		PreviousChar = Input[x];
	}

	BOOL Success = HttpSendRequest(hRequest, ContentType, sizeof(ContentType) - 1, (LPVOID)PostData.c_str(), PostData.length());
	if (!Success)
	{
#ifdef DEBUG_VALIDATE
		WriteTrace(TraceError,"v6");
#endif
		_Settings->SaveDword(IsValidExe,DefaultResult);
		InternetCloseHandle (hRequest);
		return;
	}
	
	BYTE WebSiteData[4000];
	memset(WebSiteData,0,sizeof(WebSiteData));
	DWORD dwRead = 0;
	if (!InternetReadFile(hRequest,WebSiteData,sizeof(WebSiteData),&dwRead))
	{
#ifdef DEBUG_VALIDATE
		WriteTrace(TraceError,"v7");
#endif
		_Settings->SaveDword(IsValidExe,DefaultResult);
		InternetCloseHandle (hRequest);
		return;
	}
	WebSiteData[dwRead] = 0;
#ifdef DEBUG_VALIDATE
	WriteTraceF(TraceError,"v8: %s",WebSiteData);
#endif

	MD5 Result_md5(WebSiteData,dwRead);

	int  LastRunItem  = -1;
	bool bSaveRunInfo = false;
	for (int run = 0; run < RunItemCount; run ++)
	{
		if (memcmp(RunItems[run].File_md5,File_md5.raw_digest(),sizeof(RunItems[run].File_md5)) == 0)
		{
			LastRunItem = run;
			break;
		}
	}

#ifdef DEBUG_VALIDATE
	WriteTraceF(TraceError,"v9: %s",Result_md5.hex_digest());
#endif
	//if good MD5
	if (Result_md5.hex_digest() == "FB2CDD258756A5472BD24BABF2EC9F66") // Good Md5
	{
		if (LastRunItem > 0)
		{
			for (int i = (LastRunItem + 1); i < RunItemCount; i++)
			{
				memcpy(&RunItems[i - 1],&RunItems[i],sizeof(RunItems[i]));
			}
			RunItemCount -= 1;
			bSaveRunInfo  = true;
		}
		DefaultResult = true;
	} else if (Result_md5.hex_digest() == "9030FF575A9B687DC868B966CB7C02D4") // Bad MD5
	{
		if (LastRunItem > 0)
		{
			if (RunItems[LastRunItem].RunTimes != FAILED_CHECK)
			{
				RunItems[LastRunItem].RunTimes = FAILED_CHECK;
				bSaveRunInfo  = true;
			}
		} else {
			AUTO_PTR<BYTE> NewRunData(new BYTE[(RunItemCount + 1) * sizeof(INVALID_RUN_ITEM)]);
			INVALID_RUN_ITEM * NewRunItems = (INVALID_RUN_ITEM *)NewRunData.get();
			for (int i = 0; i < RunItemCount; i++)
			{
				NewRunItems[i] = RunItems[i];
			}
			LastRunItem = RunItemCount;
			memcpy(NewRunItems[LastRunItem].File_md5,File_md5.raw_digest(),sizeof(NewRunItems[LastRunItem].File_md5));
			NewRunItems[LastRunItem].RunTimes = FAILED_CHECK;
				
			RunData       = NewRunData;
			RunItems      = (INVALID_RUN_ITEM *)RunData.get();
			RunItemCount += 1; 
			bSaveRunInfo  = true;
		}
		DefaultResult = false;
	}
	else
	{
		if (LastRunItem >= 0)
		{
			if (RunItems[LastRunItem].RunTimes != FAILED_CHECK)
			{
				RunItems[LastRunItem].RunTimes += 1;
				if (RunItems[LastRunItem].RunTimes >= MAX_BAD_DATA)
				{
					DefaultResult = false;
					RunItems[LastRunItem].RunTimes = MAX_BAD_DATA;
				}
				bSaveRunInfo  = true;
			}
		} else {
			AUTO_PTR<BYTE> NewRunData(new BYTE[(RunItemCount + 1) * sizeof(INVALID_RUN_ITEM)]);
			INVALID_RUN_ITEM * NewRunItems = (INVALID_RUN_ITEM *)NewRunData.get();
			for (int i = 0; i < RunItemCount; i++)
			{
				NewRunItems[i] = RunItems[i];
			}
			LastRunItem = RunItemCount;
			memcpy(NewRunItems[LastRunItem].File_md5,File_md5.raw_digest(),sizeof(NewRunItems[LastRunItem].File_md5));
			NewRunItems[LastRunItem].RunTimes = 1;
				
			RunData       = NewRunData;
			RunItems      = (INVALID_RUN_ITEM *)RunData.get();
			RunItemCount += 1; 
			bSaveRunInfo  = true;
		}
		if (RunItems[LastRunItem].RunTimes >= MAX_BAD_DATA)
		{
			DefaultResult = false;
		}

	}
	
	if (bSaveRunInfo)
	{
		HKEY hKeyResults = 0;
		long lResult;
		DWORD Disposition = 0;
		lResult = RegCreateKeyEx( HKEY_CLASSES_ROOT, Subkey,0,"", REG_OPTION_NON_VOLATILE,
			KEY_ALL_ACCESS,NULL, &hKeyResults,&Disposition);
		if (lResult == ERROR_SUCCESS) 
		{
			RegSetValueEx(hKeyResults,"data",0,REG_BINARY,(BYTE *)RunData.get(),RunItemCount * sizeof(INVALID_RUN_ITEM));
			RegCloseKey(hKeyResults);
		}
	}
	_Settings->SaveDword(IsValidExe,DefaultResult);
}

void TestValidBinary ( )
{
#if defined(EXTERNAL_RELEASE) || defined(VALIDATE_BIN_LOCAL)
	static DWORD ThreadID = 0;
	if (ThreadID == 0)
	{
		HANDLE hThread = CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)TestValidBinaryThread,(LPVOID)NULL,0,&ThreadID);
		CloseHandle(hThread);
	}
#endif
}