added InitTargetMemory for allocating a memory block in the target process (cause we need the base address ASAP)

1 files changed, 7 insertions, 4 deletions

diff --git a/MemDriverLib/DLLHelper.h b/MemDriverLib/DLLHelper.h
index 0a6176f..3384ec6 100644
--- a/MemDriverLib/DLLHelper.h
+++ b/MemDriverLib/DLLHelper.h
@@ -8,14 +8,17 @@ public:
 	DLLHelper();
 	~DLLHelper();
 
-	bool Init(std::string& fullDllPath);
+	bool Init(HANDLE targetPID, std::string& fullDllPath);
 	bool VerifyHeader();
+	bool InitTargetMemory();
 
 private:
+	HANDLE m_TargetPID = 0;
 	std::string m_DLLPath;
-	DWORD m_DLLSize;
+	DWORD m_DLLSize = 0;
 	UINT8 *m_DLLPtr = nullptr;
-	IMAGE_DOS_HEADER *m_DOSHeader;
-	IMAGE_NT_HEADERS *m_NTHeader;
+	IMAGE_DOS_HEADER *m_DOSHeader = nullptr;
+	IMAGE_NT_HEADERS *m_NTHeader = nullptr;
+	PVOID m_TargetBaseAddress = nullptr;
 };