From d344ea3ee86ab95d9ebce7cc06dd939f9734a06a Mon Sep 17 00:00:00 2001 From: Toni Uhlig Date: Thu, 19 Sep 2019 22:16:17 +0200 Subject: added InitTargetMemory for allocating a memory block in the target process (cause we need the base address ASAP) --- MemDriverLib/DLLHelper.h | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'MemDriverLib/DLLHelper.h') 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; }; -- cgit v1.2.3