#include <Windows.h>
#include <iostream>
#include <TlHelp32.h>

const char* PROCESS_NAME = "wwzRetailEgs.exe";
const unsigned short POLL_RATE = 100;

const unsigned int AMMO_ADDRESS = 0x7FF7E78FEC73;
const char* AMMO_OP_CODES = "\x89\x41\x08";
const char* NEW_AMMO_OP_CODES = "\x0\x0\x0";
const unsigned short AMMO_OP_CODE_LENGTH = 3;

DWORD get_process_id_by_process_name(const char* process_name_)
{
	PROCESSENTRY32 process_entry = { sizeof(PROCESSENTRY32) };
	HANDLE processes_snapshot;
	DWORD process_id = 0;

	processes_snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
	if (Process32First(processes_snapshot, &process_entry))
	{
		do
		{
			if (!strcmp(process_entry.szExeFile, process_name_))
			{
				process_id = process_entry.th32ProcessID;
				break;
			}
		} while (Process32Next(processes_snapshot, &process_entry));
	}

	CloseHandle(processes_snapshot);

	return process_id;
}

int main(int argc_, char** argv_)
{
	std::cout << "Looking for game..." << std::endl;
	DWORD process_id = NULL;
	while (!process_id)
	{
		process_id = get_process_id_by_process_name(PROCESS_NAME);
		Sleep(POLL_RATE);
	}
	std::cout << "Game found." << std::endl;

	HANDLE process_handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, process_id);
	if (process_handle)
	{
		std::cout << "Handle obtained to game." << std::endl;
	}
	else
	{
		std::cout << "Handle not obtained to game." << std::endl;
		exit(1);
	}

	bool hack_enabled = false;
	bool need_to_update_hack = false;
	bool need_to_update_screen = true;

	while (true)
	{
		if (need_to_update_screen)
		{
			system("cls");
			std::cout << "[F9] - " << (hack_enabled ? "(on )" : "(off)") << " Ammo Hack" << std::endl;
			need_to_update_screen = false;
		}

		if (GetAsyncKeyState(VK_F9))
		{
			hack_enabled = !hack_enabled;
			need_to_update_hack = true;
			need_to_update_screen = true;
		}

		if (hack_enabled && need_to_update_hack)
		{
			WriteProcessMemory(process_handle, (LPVOID)AMMO_ADDRESS, NEW_AMMO_OP_CODES, AMMO_OP_CODE_LENGTH, NULL);
			need_to_update_hack = false;
		}
		else if (need_to_update_hack)
		{
			WriteProcessMemory(process_handle, (LPVOID)AMMO_ADDRESS, AMMO_OP_CODES, AMMO_OP_CODE_LENGTH, NULL);
			need_to_update_hack = false;
		}

		Sleep(POLL_RATE);
	}

}