summaryrefslogtreecommitdiff
path: root/csgo-loader/csgo-client/RemoteCode/RemoteProcess.cpp
blob: abed829d96a5cc4af042c33b9f186b9c3bed757d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#include <RemoteCode/RemoteProcess.hpp>

namespace RemoteCode
{
	// RemoteModule implementation
	RemoteModule::RemoteModule(HANDLE Module) :
		m_Module(Module) {}

	// RemoteProcess implementation
	bool RemoteProcess::Start(const char *ProcessName)
	{
		void *Toolhelp = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);

		if(!Toolhelp)
			return false;

		PROCESSENTRY32 ProcessEntry{ sizeof PROCESSENTRY32 };

		if(!Process32First(Toolhelp, &ProcessEntry))
			return false;

		while(Process32Next(Toolhelp, &ProcessEntry))
		{
			if(strstr(ProcessName, ProcessEntry.szExeFile))
			{
				CloseHandle(Toolhelp);

				// swoo
				m_ProcessId = ProcessEntry.th32ProcessID;
				m_Process   = OpenProcess(PROCESS_ALL_ACCESS, false, ProcessEntry.th32ProcessID);

				if(!m_Process)
					ERROR_ASSERT(STR("[000G:%08x] There was an error with accessing a process."), GetLastError());

				return true;
			}
		}

		CloseHandle(Toolhelp);
		return false;
	}

	void RemoteProcess::ReadMemoryWrapper_Internal(void *Address, void *Data, size_t SizeOfData)
	{
		static auto ZwReadVirtualMemory = Syscalls->Find<long(__stdcall *)(void *, void *, void *, size_t, void *)>(FNV("ZwReadVirtualMemory"));
		
		NTSTATUS Status = ZwReadVirtualMemory(m_Process, Address, Data, SizeOfData, nullptr);
	
		if(NT_ERROR(Status))
			ERROR_ASSERT(STR("[00DF:%08x] There was an error with accessing a process."), Status);
	}

	void RemoteProcess::WriteMemoryWrapper_Internal(void *Address, void *Data, size_t SizeOfData)
	{
		static auto ZwWriteVirtualMemory = Syscalls->Find<long(__stdcall *)(void *, void *, void *, size_t, void *)>(FNV("ZwWriteVirtualMemory"));
		
		NTSTATUS Status = ZwWriteVirtualMemory(m_Process, Address, Data, SizeOfData, nullptr);
	
		if(NT_ERROR(Status))
			ERROR_ASSERT(STR("[00DF:%08x] There was an error with accessing a process."), Status);
	}

	void *RemoteProcess::Allocate(size_t AllocationSize)
	{
		static auto ZwAllocateVirtualMemory = Syscalls->Find<long(__stdcall *)(void *, void *, uint32_t, size_t *, uint32_t, uint32_t)>(FNV("ZwAllocateVirtualMemory"));
		
		// :b:invoke the :b:unction :b:oi
		void *AllocationAddress = nullptr;
		NTSTATUS Status         = ZwAllocateVirtualMemory(
			m_Process,
			&AllocationAddress,
			0,
			&AllocationSize,
			MEM_COMMIT | MEM_RESERVE,
			PAGE_EXECUTE_READWRITE
		);

		if(NT_ERROR(Status))
			ERROR_ASSERT(STR("[00DF:%08x] There was an error with accessing a process."), Status);

		return AllocationAddress;
	}

	RemoteModule RemoteProcess::FindModule(const char *ModuleName)
	{
		void *Toolhelp = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE | TH32CS_SNAPMODULE32, m_ProcessId);
		
		if(!Toolhelp)
			return RemoteModule{};
		
		MODULEENTRY32 ModuleEntry{ sizeof MODULEENTRY32 };
		
		if(!Module32First(Toolhelp, &ModuleEntry))
			return RemoteModule{};
		
		while(Module32Next(Toolhelp, &ModuleEntry))
		{
			if(strstr(ModuleEntry.szModule, ModuleName))
			{
				if(!ModuleEntry.hModule)
					ERROR_ASSERT(STR("[00DF:00001C00] An integrity check failed."));

				CloseHandle(Toolhelp);
				return RemoteModule(ModuleEntry.hModule);
			}
		}
		
		CloseHandle(Toolhelp);
		return RemoteModule{};
	}
}