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

using namespace std;

uintptr_t FindAddr(HANDLE hProc, uintptr_t ptr, std::vector<unsigned int> offsets)
{
	uintptr_t addr = ptr;
	for (unsigned int i = 0; i < offsets.size(); ++i)
	{
		ReadProcessMemory(hProc, (BYTE*)addr, &addr, sizeof(addr), 0);
		addr += offsets[i];
	}
	return addr;
}

DWORD GetProcId(const wchar_t* procName)
{
	DWORD procId = 0;
	HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
	if (hSnap != INVALID_HANDLE_VALUE)
	{
		PROCESSENTRY32 procEntry;
		procEntry.dwSize = sizeof(procEntry);

		if (Process32First(hSnap, &procEntry))
		{
			do
			{
				if (!_wcsicmp(procEntry.szExeFile, procName))
				{
					procId = procEntry.th32ProcessID;
					break;
				}
			} while (Process32Next(hSnap, &procEntry));

		}
	}
	CloseHandle(hSnap);
	return procId;
}

uintptr_t GetModuleBaseAddress(DWORD procId, const wchar_t* modName)
{
	uintptr_t modBaseAddr = 0;
	HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE | TH32CS_SNAPMODULE32, procId);
	if (hSnap != INVALID_HANDLE_VALUE)
	{
		MODULEENTRY32 modEntry;
		modEntry.dwSize = sizeof(modEntry);
		if (Module32First(hSnap, &modEntry))
		{
			do
			{
				if (!_wcsicmp(modEntry.szModule, modName))
				{
					modBaseAddr = (uintptr_t)modEntry.modBaseAddr;
					break;
				}
			} while (Module32Next(hSnap, &modEntry));
		}
	}
	CloseHandle(hSnap);
	return modBaseAddr;
}

int main() {
	DWORD procID;
	uintptr_t ModuleBase;
	HANDLE hProcess;

	procID = GetProcId(L"Minecraft.Windows.exe");
	if (procID == NULL) {
		cout << "Unable to locate process, make sure Minecraft is running!\n";
	}
	else
	{
		cout << "Minecraft Process ID: " << procID << "\n";
		hProcess = OpenProcess(PROCESS_ALL_ACCESS, NULL, procID);
		ModuleBase = GetModuleBaseAddress(procID, L"Minecraft.Window.exe");

		for (;;) {
			//Memory writing

			uintptr_t localPlayerAddr = FindAddr(hProcess, ModuleBase + 0x02FC77D0, { 0xA8, 0x10, 0x40, 0x0 });
			uintptr_t gamemodeAddr = FindAddr(hProcess, localPlayerAddr, { 0x1D94 });

			int gmVal = 1;

			WriteProcessMemory(hProcess, (BYTE*)gamemodeAddr, &gmVal, sizeof(gmVal), 0); //Creative Mode
		}

	}

	return 0;
}