From 2c3f2c0f63478b5d158ae72e255627a4a39f00b1 Mon Sep 17 00:00:00 2001
From: Toni Uhlig <matzeton@googlemail.com>
Date: Sun, 22 Sep 2019 12:46:04 +0200
Subject: DLLHelper: added Has* functions to check availability of some data
 dirs, implemented copy PE header and sections function, fixed invalid VAlloc
 request size

---
 include/DLLHelper.h | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'include/DLLHelper.h')

diff --git a/include/DLLHelper.h b/include/DLLHelper.h
index 764b16f..5631626 100644
--- a/include/DLLHelper.h
+++ b/include/DLLHelper.h
@@ -3,6 +3,9 @@
 #include <string>
 #include <Windows.h>
 
+
+typedef void(*LibEntry_FN)(void);
+
 class DLLHelper
 {
 public:
@@ -12,8 +15,17 @@ public:
 	bool Init(HANDLE targetPID, const char * fullDllPath);
 	bool VerifyHeader();
 	bool InitTargetMemory();
+	bool HasImports() {
+		return m_NTHeader &&
+			m_NTHeader->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size;
+	}
 	bool FixImports();
+	bool HasRelocs() {
+		return m_NTHeader &&
+			m_NTHeader->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].Size;
+	}
 	bool FixRelocs();
+	bool CopyHeaderAndSections();
 
 private:
 	HANDLE m_TargetPID = 0;
-- 
cgit v1.2.3