Added server support for both x64 and x32 images with automatic selection.

6 files changed, 126 insertions, 20 deletions

diff --git a/client/src/client/client.h b/client/src/client/client.h
index d840538..91cb494 100644
--- a/client/src/client/client.h
+++ b/client/src/client/client.h
@@ -18,6 +18,7 @@ struct game_data_t {
 	std::string name;
 	std::string version;
 	std::string process_name;
+	bool x64;
 	uint8_t id;
 };
 
diff --git a/client/src/injection/mapper.cpp b/client/src/injection/mapper.cpp
index 1951ab2..c7f771c 100644
--- a/client/src/injection/mapper.cpp
+++ b/client/src/injection/mapper.cpp
@@ -9,11 +9,116 @@ void mmap::thread(tcp::client& client) {
 		std::this_thread::sleep_for(std::chrono::seconds(1));
 	}
 
+	if (client.selected_game.x64) {
+		map64(client);
+
+		return;
+	}
+
+	map32(client);
+}
+
+void mmap::map32(tcp::client& client) {
+	util::system_data_t dat;
+	util::fetch_system_data(dat);
+
+	auto needle = std::find_if(dat.processes.begin(), dat.processes.end(), [&](util::process_data_t& dat) {
+		return dat.name == client.selected_game.process_name;
+		});
+
+	if (needle == dat.processes.end()) {
+		io::log_error("failed to find process.");
+		return;
+	}
+
+	util::process<uint32_t> proc(*needle);
+
+	if (!proc.open()) {
+		return;
+	}
+
+	if (!proc.enum_modules()) {
+		io::log_error("failed to enum {} modules", proc.name());
+		return;
+	}
+
+	auto image = proc.allocate(client.mapper_data.image_size, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
+	if (!image) {
+		io::log_error("failed to allocate memory for image.");
+		return;
+	}
+
+	io::log("image base : {:x}", image);
+
+	auto imports = nlohmann::json::parse(client.mapper_data.imports);
+
+	nlohmann::json final_imports;
+	for (auto& [key, value] : imports.items()) {
+		for (auto& i : value) {
+			auto name = i.get<std::string>();
+
+			final_imports[name] = proc.module_export(proc.map(key), name);
+		}
+	}
+	imports.clear();
+
+	nlohmann::json resp;
+	resp["alloc"] = image;
+	resp["id"] = client.selected_game.process_name;
+	resp["x64"] = client.selected_game.x64;
+
+	client.write(tcp::packet_t(resp.dump(), tcp::packet_type::write, client.session_id, tcp::packet_id::image));
+	resp.clear();
+
+	client.stream(final_imports.dump());
+	final_imports.clear();
+
+	io::log("please wait...");
+	while (client.state != tcp::client_state::image_ready) {
+		std::this_thread::sleep_for(std::chrono::seconds(1));
+	}
+
+	if (!proc.write(image, client.mapper_data.image.data(), client.mapper_data.image.size())) {
+		io::log_error("failed to write image.");
+		return;
+	}
+	client.mapper_data.image.clear();
+
+	auto entry = image + client.mapper_data.entry;
+
+	io::log("entry : {:x}", entry);
+
+	static std::vector<uint8_t> shellcode = { 0x55, 0x89, 0xE5, 0x6A, 0x00, 0x6A, 0x01, 0x68, 0xEF, 0xBE,
+		0xAD, 0xDE, 0xB8, 0xEF, 0xBE, 0xAD, 0xDE, 0xFF, 0xD0, 0x89, 0xEC, 0x5D, 0xC3 };
+
+	*reinterpret_cast<uint32_t*>(&shellcode[8]) = image;
+	*reinterpret_cast<uint32_t*>(&shellcode[13]) = entry;
+
+	auto code = proc.allocate(shellcode.size(), MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
+	if (!proc.write(code, shellcode.data(), shellcode.size())) {
+		io::log_error("failed to write shellcode.");
+		return;
+	}
+
+	io::log("shellcode : {:x}", code);
+
+	proc.thread(code);
+
+	proc.free(code, shellcode.size());
+
+	proc.close();
+
+	client.state = tcp::client_state::injected;
+
+	io::log("done");
+}
+
+void mmap::map64(tcp::client& client) {
 	util::system_data_t dat;
 	util::fetch_system_data(dat);
 
 	auto needle = std::find_if(dat.processes.begin(), dat.processes.end(), [&](util::process_data_t& dat) {
-		return dat.name == "sublime_text.exe";
+		return dat.name == client.selected_game.process_name;
 	});
 
 	if (needle == dat.processes.end()) {
@@ -54,7 +159,8 @@ void mmap::thread(tcp::client& client) {
 
 	nlohmann::json resp;
 	resp["alloc"] = image;
-	resp["id"] = client.selected_game.id;
+	resp["id"] = client.selected_game.process_name;
+	resp["x64"] = client.selected_game.x64;
 
 	client.write(tcp::packet_t(resp.dump(), tcp::packet_type::write, client.session_id, tcp::packet_id::image));
 	resp.clear();
@@ -77,15 +183,9 @@ void mmap::thread(tcp::client& client) {
 
 	io::log("entry : {:x}", entry);
 
-	/*static std::vector<uint8_t> shellcode = { 0x55, 0x89, 0xE5, 0x6A, 0x00, 0x6A, 0x01, 0x68, 0xEF, 0xBE,
-		0xAD, 0xDE, 0xB8, 0xEF, 0xBE, 0xAD, 0xDE, 0xFF, 0xD0, 0x89, 0xEC, 0x5D, 0xC3 };*/
-
 	static std::vector<uint8_t> shellcode = { 0x48, 0x83, 0xEC, 0x28, 0x48, 0xB9, 0x00, 0x00, 0x00, 0x00,
-		0x00, 0x00, 0x00, 0x00, 0x48, 0xC7, 0xC2,0x01, 0x00, 0x00, 0x00, 0x4D, 0x31, 0xC0,
-		0x48, 0xB8, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF, 0xD0, 0x48, 0x83, 0xC4, 0x28, 0xC3 };
-
-	/**reinterpret_cast<uint32_t*>(&shellcode[8]) = image;
-	*reinterpret_cast<uint32_t*>(&shellcode[13]) = entry;*/
+	0x00, 0x00, 0x00, 0x00, 0x48, 0xC7, 0xC2,0x01, 0x00, 0x00, 0x00, 0x4D, 0x31, 0xC0,
+	0x48, 0xB8, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF, 0xD0, 0x48, 0x83, 0xC4, 0x28, 0xC3 };
 
 	*reinterpret_cast<uint64_t*>(&shellcode[6]) = image;
 	*reinterpret_cast<uint64_t*>(&shellcode[26]) = entry;
@@ -102,8 +202,6 @@ void mmap::thread(tcp::client& client) {
 
 	proc.free(code, shellcode.size());
 
-	//proc.free(image, client.mapper_data.image_size);
-
 	proc.close();
 
 	client.state = tcp::client_state::injected;
diff --git a/client/src/injection/mapper.h b/client/src/injection/mapper.h
index 0ce7b8f..bdc3565 100644
--- a/client/src/injection/mapper.h
+++ b/client/src/injection/mapper.h
@@ -2,4 +2,7 @@
 
 namespace mmap {
 	void thread(tcp::client& client);
+
+	void map32(tcp::client& client);
+	void map64(tcp::client& client);
 };
\ No newline at end of file
diff --git a/client/src/main.cpp b/client/src/main.cpp
index 83b96fe..71a064f 100644
--- a/client/src/main.cpp
+++ b/client/src/main.cpp
@@ -9,8 +9,6 @@
 #include "util/apiset.h"
 
 int main(int argc, char* argv[]) {
-	io::log("{:x}", g_syscalls());
-
 	tcp::client client;
 
 	std::thread t{ tcp::client::monitor, std::ref(client) };
@@ -86,8 +84,9 @@ int main(int argc, char* argv[]) {
 					std::string version = value["version"];
 					std::string process = value["process"];
 					uint8_t id = value["id"];
+					bool x64 = value["x64"];
 
-					client.games.emplace_back(game_data_t{ key, version, process, id });
+					client.games.emplace_back(game_data_t{ key, version, process, x64, id });
 				}
 
 				io::log("logged in.");
@@ -165,7 +164,8 @@ int main(int argc, char* argv[]) {
 			client.selected_game = *it;
 
 			nlohmann::json j;
-			j["id"] = id;
+			j["id"] = client.selected_game.process_name;
+			j["x64"] = client.selected_game.x64;
 
 			int ret = client.write(tcp::packet_t(j.dump(), tcp::packet_type::write,
 				client.session_id,
diff --git a/client/src/util/io.h b/client/src/util/io.h
index 400c089..03d6964 100644
--- a/client/src/util/io.h
+++ b/client/src/util/io.h
@@ -7,9 +7,10 @@
 namespace io {
 	template<typename... Args>
 	void log(const std::string_view str, Args... params) {
-		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_GREEN);
+		static auto handle = GetStdHandle(STD_OUTPUT_HANDLE);
+		SetConsoleTextAttribute(handle, FOREGROUND_GREEN);
 		fmt::print("$> ");
-		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_RED);
+		SetConsoleTextAttribute(handle, FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_RED);
 
 		std::string msg{str};
 		msg.append("\n");
@@ -19,9 +20,10 @@ namespace io {
 
 	template<typename... Args>
 	void log_error(const std::string_view str, Args... params) {
-		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED);
+		static auto handle = GetStdHandle(STD_OUTPUT_HANDLE);
+		SetConsoleTextAttribute(handle, FOREGROUND_RED);
 		fmt::print("$> ");
-		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_RED);
+		SetConsoleTextAttribute(handle, FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_RED);
 
 		std::string msg{str};
 		msg.append("\n");
diff --git a/client/src/util/syscalls.cpp b/client/src/util/syscalls.cpp
index e877932..96036a3 100644
--- a/client/src/util/syscalls.cpp
+++ b/client/src/util/syscalls.cpp
@@ -9,6 +9,8 @@ syscalls::syscalls() {
 	m_call_table = VirtualAlloc(0, 0x100000, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
 	std::memset(m_call_table, 0x90, 0x100000);
 
+	io::log("syscalls call table : {:#x}", uintptr_t(m_call_table));
+
 	static auto nt = pe::virtual_image("ntdll.dll");
 	for (auto& exp : nt.exports()) {
 		auto addr = exp.second;