fixed bug during import table function resolving, added public getter function for EntryPoint and BaseAddress

1 files changed, 9 insertions, 0 deletions

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;