diff options
Diffstat (limited to 'MemDriverLib/DLLHelper.h')
| -rw-r--r-- | MemDriverLib/DLLHelper.h | 11 |
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; }; |