diff options
| author | Toni Uhlig <matzeton@googlemail.com> | 2019-09-23 21:23:20 +0200 |
|---|---|---|
| committer | Toni Uhlig <matzeton@googlemail.com> | 2019-09-23 21:23:20 +0200 |
| commit | 673cec06fab76718ade9b3763b8d43daddeaeeda (patch) | |
| tree | 3c905d31ed0a48159aea6536ae7d35fefe69fba5 /include | |
| parent | b4766e2d77ed4973122914b3aa2b60723973837f (diff) | |
added function to verify PE header values and removed inlined functions from DLLHelper
Diffstat (limited to 'include')
| -rw-r--r-- | include/DLLHelper.h | 17 |
1 files changed, 3 insertions, 14 deletions
diff --git a/include/DLLHelper.h b/include/DLLHelper.h index fc17711..7fdff37 100644 --- a/include/DLLHelper.h +++ b/include/DLLHelper.h @@ -6,18 +6,8 @@ typedef void(*LibEntry_FN)(void); -static inline bool LoadAndTestLibraryEntry(const char * const fullDllPath) -{ - HMODULE TestDLLModule = LoadLibraryA(fullDllPath); - LibEntry_FN LibEntryProc = (LibEntry_FN)GetProcAddress(TestDLLModule, "LibEntry"); - if (LibEntryProc) { - LibEntryProc(); - return true; - } - else { - return false; - } -} +static inline bool LoadAndTestLibraryEntry(const char * const fullDllPath); +bool VerifyPeHeader(UINT8 const * const buf, SIZE_T siz, IMAGE_NT_HEADERS ** const return_NTHeader); class DLLHelper { @@ -25,7 +15,7 @@ public: DLLHelper(); ~DLLHelper(); - bool Init(HANDLE targetPID, const char * fullDllPath); + bool Init(HANDLE targetPID, const char * const fullDllPath); bool VerifyHeader(); bool InitTargetMemory(); bool HasImports() { @@ -54,7 +44,6 @@ private: std::string m_DLLPath; DWORD m_DLLSize = 0; UINT8 *m_DLLPtr = nullptr; - IMAGE_DOS_HEADER *m_DOSHeader = nullptr; IMAGE_NT_HEADERS *m_NTHeader = nullptr; PVOID m_TargetBaseAddress = nullptr; }; |