aboutsummaryrefslogtreecommitdiff
path: root/include/DLLHelper.h
diff options
context:
space:
mode:
authorToni Uhlig <matzeton@googlemail.com>2019-09-22 21:16:54 +0200
committerToni Uhlig <matzeton@googlemail.com>2019-09-22 21:16:54 +0200
commit7dfd4eaf40541f30a8089fd35893524238944148 (patch)
tree08a3deaad89ea5c8b7384256da6dd3e34f093063 /include/DLLHelper.h
parent19b860efd3b00a164979234f85d8884a3e5e6cf1 (diff)
fixed bug during import table function resolving, added public getter function for EntryPoint and BaseAddress
Diffstat (limited to 'include/DLLHelper.h')
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;
Powered by cgit, Themed by Ted Yin.