aboutsummaryrefslogtreecommitdiff
path: root/client/src/injection/mapper.cpp
blob: 7fcb8b3f272729f3305b900adde79ca4a78125b6 (plain) (blame)
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
#include "../include.h"
#include "../client/client.h"
#include "../util/util.h"
#include "process.h"
#include "mapper.h"

void mmap::thread(tcp::client& client) {
	while (client.mapper_data.imports.empty()) {
		std::this_thread::sleep_for(std::chrono::milliseconds(100));
	}

	util::fetch_processes();

	auto needle = std::find_if(util::process_list.begin(), util::process_list.end(), [&](util::process& proc) {
		return proc.name() == "notepad++.exe";
	});

	while (needle == util::process_list.end()) {
		std::this_thread::sleep_for(std::chrono::seconds(5));
		util::fetch_processes();
		io::logger->info("waiting for process..");
		needle = std::find_if(util::process_list.begin(), util::process_list.end(), [&](util::process& proc) {
			return proc.name() == "notepad++.exe";
		});
	}

	needle->open();
	needle->enum_modules();

	auto image = needle->allocate(client.mapper_data.image_size, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
	if (!image) {
		io::logger->error("failed to allocate memory for image.");
		return;
	}

	io::logger->info("image base : {:x}", image);

	auto imports = nlohmann::json::parse(client.mapper_data.imports);

	nlohmann::json final_imports;
	for (auto& [key, value] : imports.items()) {
		auto mod = key;
		std::transform(mod.begin(), mod.end(), mod.begin(), ::tolower);

		auto base = needle->load(mod);
		if (!base) {
			io::logger->error("failed to load {}", mod);
			continue;
		}

		for (auto& i : value) {
			auto name = i.get<std::string>();

			auto func = needle->module_export(mod, name);

			final_imports[name] = func;
		}
	}

	nlohmann::json resp;
	resp["alloc"] = image;

	client.write(tcp::packet_t(resp.dump(), tcp::packet_type::write, client.session_id, tcp::packet_id::image));

	auto proc_imports = final_imports.dump();
	client.stream(proc_imports);

	io::logger->info("please wait...");
	while (client.mapper_data.image.empty()) {
		std::this_thread::sleep_for(std::chrono::seconds(1));
	}

	if (!needle->write(image, client.mapper_data.image.data(), client.mapper_data.image.size())) {
		io::logger->error("failed to write image.");
		return;
	}

	auto entry = image + client.mapper_data.entry;

	io::logger->info("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 = needle->allocate(shellcode.size(), MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
	if (!needle->write(code, shellcode.data(), shellcode.size())) {
		io::logger->error("failed to write shellcode.");
		return;
	}

	io::logger->info("shellcode : {:x}", code);

	needle->thread(code);

	needle->free(code, shellcode.size());

	needle->close();

	io::logger->info("done");

	std::cin.get();
}