From 7caedef9a8c343b63cef6e971f4f87660520bb82 Mon Sep 17 00:00:00 2001
From: auth12 <67507608+auth12@users.noreply.github.com>
Date: Wed, 22 Jul 2020 08:37:58 -0700
Subject: Client injection. Process class implementation.

---
 client/src/injection/process.h | 54 +++++++++++++++++++++++++++++-------------
 1 file changed, 37 insertions(+), 17 deletions(-)

(limited to 'client/src/injection/process.h')

diff --git a/client/src/injection/process.h b/client/src/injection/process.h
index 574713a..ddbadf5 100644
--- a/client/src/injection/process.h
+++ b/client/src/injection/process.h
@@ -1,19 +1,39 @@
 #pragma once
 
-class process {
-	int m_id;
-	std::string m_name;
-
-	HANDLE m_handle = INVALID_HANDLE_VALUE;
-public:
-	process() = default;
-	process(const SYSTEM_PROCESS_INFORMATION* info);
-	~process();
-
-	bool open();
-	bool read(const uintptr_t addr, void* data, const size_t size);
-	bool write(const uintptr_t addr, void* data, const size_t size);
-
-	auto &get_name() { return m_name; }
-	auto &get_id() { return m_id; }
-};
\ No newline at end of file
+namespace util {
+	class process {
+		int m_id;
+		std::string m_name;
+		std::unordered_map<std::string, uintptr_t> m_modules;
+
+		HANDLE m_handle;
+	public:
+		process() : m_handle{ INVALID_HANDLE_VALUE } {};
+		process(const SYSTEM_PROCESS_INFORMATION* info);
+		~process();
+
+		bool open();
+		bool read(const uintptr_t addr, void* data, size_t size);
+		bool write(const uintptr_t addr, void* data, size_t size);
+		bool free(const uintptr_t addr, size_t size);
+		bool thread(const uintptr_t start);
+		bool enum_modules();
+
+		uintptr_t peb();
+		uintptr_t load(const std::string_view mod);
+		uintptr_t allocate(size_t size, uint32_t type, uint32_t protection);
+		uintptr_t module_export(const std::string_view name, const std::string_view func);
+
+		bool close();
+
+		operator bool() const { return m_handle != INVALID_HANDLE_VALUE; }
+
+		auto& name() { return m_name; }
+		auto& id() { return m_id; }
+		auto& handle() { return m_handle; }
+	};
+
+	extern std::vector<process> process_list;
+
+	bool fetch_processes();
+};
-- 
cgit v1.2.3