win_authority.cpp

#include "shared.h"

#include "Commctrl.h"
#include "ShlObj.h"
#include "Shellapi.h"

void up_privilege() {
	HANDLE hToken;
	TOKEN_PRIVILEGES tkp;

	OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken);

	LookupPrivilegeValueA(NULL, "SeDebugPrivileges", &tkp.Privileges[0].Luid);

	tkp.PrivilegeCount = 1;
	tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

	AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, NULL, 0);
}

int Sys_IsAdmin() {
	int b;
	SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;
	PSID AdministratorsGroup;
	b = AllocateAndInitializeSid(
		&NtAuthority,
		2,
		SECURITY_BUILTIN_DOMAIN_RID,
		DOMAIN_ALIAS_RID_ADMINS,
		0, 0, 0, 0, 0, 0,
		&AdministratorsGroup);
	if (b)
	{
		if (!CheckTokenMembership(NULL, AdministratorsGroup, &b))
		{
			b = FALSE;
		}
		FreeSid(AdministratorsGroup);
	}

	return(b);
}

extern char sys_cmdline[MAX_STRING_CHARS];

#define ERROR_MSG(x) do \
	MessageBox(0, va("Call of Duty caught an error whilst trying to do something.\nFile: %s, Line: %d\nErrorcode: %d (0x%x)\nError: %s" \
				"Please report this error at the forums.",__FILE__,__LINE__,x,x,GetLastErrorAsString().c_str()), __TITLE, MB_ICONERROR | MB_OK); \
								while(0)

void Sys_ElevateProgram() {
	if (Sys_IsAdmin())
		return; //we already are admin

	char *fn;

	Sys_GetModulePathInfo(NULL, NULL, &fn, NULL);

	char *arg = va("allowdupe %s", sys_cmdline);

	DWORD err = ERROR_SUCCESS;
	ShellExecuteA(NULL, "runas", fn, arg, ".", SW_SHOWNORMAL | SW_SHOW);
	if ((err = GetLastError()) != ERROR_SUCCESS)
		if (err != ERROR_CANCELLED)
			ERROR_MSG(err);

	Com_Quit_f();
}

void Sys_RestartProgram(bool bAsAdmin) {
	char *fn;

	Sys_GetModulePathInfo(NULL, NULL, &fn, NULL);

	char *arg = va("allowdupe %s", sys_cmdline);

	DWORD err = ERROR_SUCCESS;
	ShellExecuteA(NULL, bAsAdmin ? "runas" : "open", fn, arg, ".", SW_SHOWNORMAL | SW_SHOW);
	if ((err = GetLastError()) != ERROR_SUCCESS)
		if (err != ERROR_CANCELLED)
			ERROR_MSG(err);

quit:
	Com_Quit_f();
}

typedef struct {
	int ver;
	char *md5;
} cod_v_pair;

cod_v_pair cod_v_info[] = {
	{ COD_1, "7b54e0259b24d32c16467ac454228418" }, // Modified, proper entry points, loads cod1x.dll instead of mss32.dll. f19da0d804bb20231572ee17225af339 - polish 7b54e0259b24d32c16467ac454228418 - md5 of english .exe
	{ COD_1_SP, "9fa83933bbf659050a2f213c217b624c" },
	//{ COD_1, "766345d1ceaf79caf7fe88a214b2f3ec" }, //from spect (different codmp entry points RVA import)
	{ CODUO_51, "928dd08dc169bd85fdd12d2db28def70" },
	{ COD_5_SP_STEAM, "acdf185fe7767d711fc99cb57df46044" },
	{ COD_5_STEAM, "4f4596b1cdb21f9eb62e6683ecf48dc6" },
	{ COD_5, "4bdf293d8e6fb32208d1b0942a1ba6bc" },
	{ COD_UNKNOWN, NULL }

};


int WINAPI WinMain_upd(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow) {

	char *tempFileName = va("codmp_%u.download", rand());
	bool simple_download(const char *url, const char *local);
	if (!simple_download("http://cod1.eu/bin/codmp_1.bin", tempFileName)) {
		MessageBoxA(NULL, "Failed to download patch CoDMP.exe! Run the program as Administrator!\n", __TITLE, MB_ICONERROR | MB_OK);
		ExitProcess(0);
	}
#if 1
	char cur_fn[MAX_PATH + 1];
	GetModuleFileNameA(NULL, cur_fn, MAX_PATH);

	MoveFile(cur_fn, "cod_tmp_del_file");
	DeleteFileA("cod_tmp_del_file");
	MoveFileA(tempFileName, cur_fn);
	Sys_RestartProgram(true);
	ExitProcess(0);
#endif
}

bool determine_cod_version() {
	FILE *fp = NULL;

	char szFileName[MAX_PATH + 1];
	GetModuleFileName(NULL, szFileName, MAX_PATH);

	if (GetLastError() != ERROR_SUCCESS) {
		MessageBoxA(NULL, "Too long filepath/name", "", 0);
		return false;
	}

	char *fn;
	Sys_GetModulePathInfo(NULL, NULL, &fn, NULL);

	fp = fopen(fn, "rb");

	if (!fp) {
		MessageBoxA(NULL, va("Error reading '%s'\nRun Call of Duty as Administrator.", fn), __TITLE, MB_OK);
		return false;
	}

	BYTE *buf = NULL;
	size_t fs = 0;
	fseek(fp, 0, SEEK_END);
	fs = ftell(fp);
	rewind(fp);

	buf = new BYTE[fs];

	if (fread(buf, 1, fs, fp) != fs) {
		MessageBox(NULL, va("Reading error '%s'\n", fn), __TITLE, MB_OK | MB_ICONERROR);
		delete[] buf;
		fclose(fp);
		return false;
	}


	std::string hash = GetHashText(buf, fs, HashMd5);

	delete[] buf;
	fclose(fp);
	fp = NULL;

	for (int i = 0; cod_v_info[i].md5 != NULL; i++) {
		if (!strncmp(cod_v_info[i].md5, hash.c_str(), 32)) {
			codversion = cod_v_info[i].ver;
			break;
		} else if (!strncmp(hash.c_str(), "766345d1ceaf79caf7fe88a214b2f3ec", 32)) { // If hash != [that] update DLL?
			__call(0x528B38, (int)WinMain_upd);
			break;
		}
	}

	return (codversion != COD_UNKNOWN);
}

bool find_cod_version() {
	if (!is_addr_safe(0x566C18))
		return false;

	char *l_1 = (char*)0x566C18;

	if (!l_1)
		return false;

	if (!strcmp(l_1, "1.1")) {
		codversion = COD_1;
		return true;
	}
	return false;
}

bool has_authority() {
	up_privilege();

	if (Sys_IsAdmin())
		return true;

	Sys_ElevateProgram();

	return false;
}