From 7dfd4eaf40541f30a8089fd35893524238944148 Mon Sep 17 00:00:00 2001
From: Toni Uhlig <matzeton@googlemail.com>
Date: Sun, 22 Sep 2019 21:16:54 +0200
Subject: fixed bug during import table function resolving, added public getter
 function for EntryPoint and BaseAddress

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

(limited to 'include/DLLHelper.h')

diff --git a/include/DLLHelper.h b/include/DLLHelper.h
index cc61572..fc17711 100644
--- a/include/DLLHelper.h
+++ b/include/DLLHelper.h
@@ -39,6 +39,15 @@ public:
 	}
 	bool FixRelocs();
 	bool CopyHeaderAndSections();
+	UINT64 GetEntryPoint() {
+		if (!m_NTHeader) {
+			return 0;
+		}
+		return (UINT64)m_TargetBaseAddress + m_NTHeader->OptionalHeader.AddressOfEntryPoint;
+	}
+	UINT64 GetBaseAddress() {
+		return (UINT64)m_TargetBaseAddress;
+	}
 
 private:
 	HANDLE m_TargetPID = 0;
-- 
cgit v1.2.3