project64/Source/Project64/ValidateBinary.cpp

425 lines
11 KiB
C++

#include "stdafx.h"
#include <Wininet.h>
#pragma comment(lib, "Wininet.lib")
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 VALIDATE_DEBUG
WriteTraceF(TraceValidate,"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);
char Subkey2[] = "\xF9\x1C\x09\x12\x03\x16\x13\x17\x19\x0C\x1A\x7C\x02"; // "SOFTWARE\PJ64";
ValidateDecryptString(Subkey2,sizeof(Subkey2) - 1);
CRegistry Registry;
BOOL bRes = Registry.Open(HKEY_LOCAL_MACHINE,Subkey2,KEY_ALL_ACCESS,true);
DWORD Length = 0;
if (bRes)
{
Length = Registry.GetValueSize("data");
if (Length == -1)
{
Length = 0;
bRes = false;
}
if (bRes && Length > 0)
{
DWORD Type;
RunData = AUTO_PTR<BYTE>(new BYTE[Length]);
bRes = Registry.GetValue("data",RunData.get(),Length,&Type);
if (!bRes && Type != REG_BINARY)
{
bRes = false;
}
}
}
if (!bRes)
{
bRes = Registry.Open(HKEY_CLASSES_ROOT,Subkey,KEY_ALL_ACCESS,true);
if (bRes)
{
Length = Registry.GetValueSize("data");
if (Length == -1)
{
Length = 0;
bRes = false;
}
if (bRes)
{
DWORD Type;
RunData = AUTO_PTR<BYTE>(new BYTE[Length]);
bRes = Registry.GetValue("data",RunData.get(),Length,&Type);
if (!bRes && Type != REG_BINARY)
{
bRes = false;
}
}
}
}
if (bRes && Length > 0)
{
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 VALIDATE_DEBUG
WriteTraceF(TraceValidate,"v2: %d",RunItems[i].RunTimes);
#endif
if (RunItems[i].RunTimes >= MAX_BAD_DATA)
{
DefaultResult = false;
}
break;
}
}
}
//test to see if file has already been disabled
HINTERNET hSession = InternetOpen("project64", INTERNET_OPEN_TYPE_PRECONFIG,NULL, NULL, NULL);
if (hSession == NULL)
{
#ifdef VALIDATE_DEBUG
WriteTrace(TraceValidate,"v3");
#endif
_Settings->SaveBool(Beta_IsValidExe,DefaultResult);
return;
}
#ifdef VALIDATE_BIN_LOCAL
char Site[] = { "www.local.pj64-emu.com" };
ValidateEncryptString(Site,sizeof(Site) - 1);
#else
char Site[] = { "\xDD\x00\x00\x59\x5e\x1a\x5c\x02\x19\x48\x08\x18\x5b\x4d\x0c\x02" }; //"www.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 VALIDATE_DEBUG
WriteTrace(TraceValidate,"v4");
#endif
_Settings->SaveBool(Beta_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 VALIDATE_DEBUG
WriteTrace(TraceValidate,"v5");
#endif
_Settings->SaveBool(Beta_IsValidExe,DefaultResult);
InternetCloseHandle (hRequest);
return;
}
stdstr ComputerName;
Length = 260;
ComputerName.resize(Length);
GetComputerName((char *)ComputerName.c_str(),&Length);
ComputerName.ToLower();
stdstr_f PostInfo("1,%s,%s,%s,%s,%s,%s",VALIDATE_BIN_APP,File_md5.hex_digest(),ComputerName.c_str(),VersionInfo(VERSION_PRODUCT_VERSION).c_str(),_Settings->LoadString(Beta_UserName).c_str(),_Settings->LoadString(Beta_EmailAddress).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 (size_t 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 VALIDATE_DEBUG
WriteTrace(TraceValidate,"v6");
#endif
_Settings->SaveBool(Beta_IsValidExe,DefaultResult);
InternetCloseHandle (hRequest);
return;
}
#ifdef VALIDATE_DEBUG
{
bool bGotHeaders = true;
DWORD dwBufferLength = 300;
std::auto_ptr<BYTE> Headers(new BYTE[dwBufferLength]);
if (!HttpQueryInfo(hRequest,HTTP_QUERY_RAW_HEADERS,Headers.get(),&dwBufferLength,NULL))
{
if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
{
Headers.reset(new BYTE[dwBufferLength]);
if (!HttpQueryInfo(hRequest,HTTP_QUERY_RAW_HEADERS,Headers.get(),&dwBufferLength,NULL))
{
bGotHeaders = false;
}
}
}
if (bGotHeaders)
{
for (LPCTSTR Header = (LPCTSTR)Headers.get(); Header != NULL && Header[0] != 0; Header += strlen(Header) + 1)
{
WriteTraceF(TraceValidate,"v6a: Header: %s",Header);
}
} else {
WriteTrace(TraceValidate,"v6b: Failed to get headers");
}
}
#endif
std::auto_ptr<BYTE> WebSiteData;
DWORD dwRead = 0;
ULONG DataLen = 0;
do{
BYTE SiteData[0x1000];
memset(SiteData,0,sizeof(SiteData));
if (!InternetReadFile(hRequest,SiteData,sizeof(SiteData),&dwRead))
{
#ifdef VALIDATE_DEBUG
WriteTrace(TraceValidate,"v7");
#endif
_Settings->SaveBool(Beta_IsValidExe,DefaultResult);
InternetCloseHandle (hRequest);
return;
}
if (dwRead == 0)
{
break;
}
std::auto_ptr<BYTE> NewSiteData(new BYTE[DataLen + dwRead + 1]);
if (DataLen > 0)
{
memcpy(NewSiteData.get(),WebSiteData.get(),DataLen);
}
memcpy(NewSiteData.get() + DataLen,SiteData,dwRead);
NewSiteData.get()[DataLen + dwRead] = 0;
WebSiteData = NewSiteData;
DataLen += dwRead;
} while (dwRead > 0);
#ifdef VALIDATE_DEBUG
WriteTraceF(TraceValidate,"v8: %s",WebSiteData.get());
#endif
MD5 Result_md5(WebSiteData.get(),DataLen);
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 VALIDATE_DEBUG
WriteTraceF(TraceValidate,"v9: %s",Result_md5.hex_digest());
#endif
//if good MD5
if (_stricmp(Result_md5.hex_digest(),"FB2CDD258756A5472BD24BABF2EC9F66") == 0) // 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 (_stricmp(Result_md5.hex_digest(),"9030FF575A9B687DC868B966CB7C02D4") == 0) // 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;
}
}
bSaveRunInfo = true;
if (bSaveRunInfo)
{
bRes = Registry.Open(HKEY_CLASSES_ROOT,Subkey,KEY_ALL_ACCESS,true);
if (bRes)
{
bRes = Registry.SetValue("data",(BYTE *)RunData.get(),RunItemCount * sizeof(INVALID_RUN_ITEM),REG_BINARY);
}
if (bRes)
{
bRes = Registry.Open(HKEY_LOCAL_MACHINE,Subkey2,KEY_ALL_ACCESS);
if (bRes)
{
Registry.DeleteValue("data");
}
}else {
bRes = Registry.Open(HKEY_LOCAL_MACHINE,Subkey2,KEY_ALL_ACCESS,true);
if (bRes)
{
bRes = Registry.SetValue("data",(BYTE *)RunData.get(),RunItemCount * sizeof(INVALID_RUN_ITEM),REG_BINARY);
}
}
}
_Settings->SaveBool(Beta_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
}