aboutsummaryrefslogtreecommitdiff
path: root/NorthstarDLL/sigscanning.cpp
diff options
context:
space:
mode:
authorBobTheBob <32057864+BobTheBob9@users.noreply.github.com>2022-08-24 00:32:31 +0100
committerGeckoEidechse <gecko.eidechse+git@pm.me>2022-08-25 21:32:55 +0200
commit46bed8d09f32360377350d71816a76abdfbf2c7c (patch)
tree7feedae4910dbfe0e53d26ea1e00c4616d2c1850 /NorthstarDLL/sigscanning.cpp
parenta97f1f07bb5aa31629cd87023e09669eb4bf1fe9 (diff)
downloadNorthstarLauncher-46bed8d09f32360377350d71816a76abdfbf2c7c.tar.gz
NorthstarLauncher-46bed8d09f32360377350d71816a76abdfbf2c7c.zip
Adjust folder structure (#242)v1.9.5-rc1v1.9.5
* Adjust folder structure * change launcher directory name
Diffstat (limited to 'NorthstarDLL/sigscanning.cpp')
-rw-r--r--NorthstarDLL/sigscanning.cpp41
1 files changed, 41 insertions, 0 deletions
diff --git a/NorthstarDLL/sigscanning.cpp b/NorthstarDLL/sigscanning.cpp
new file mode 100644
index 00000000..761f9f08
--- /dev/null
+++ b/NorthstarDLL/sigscanning.cpp
@@ -0,0 +1,41 @@
+#include "pch.h"
+#include "sigscanning.h"
+#include <map>
+
+// note: sigscanning is only really intended to be used for resolving stuff like shared function definitions
+// we mostly use raw function addresses for stuff
+
+size_t GetModuleLength(HMODULE moduleHandle)
+{
+ // based on sigscn code from ttf2sdk, which is in turn based on CSigScan from https://wiki.alliedmods.net/Signature_Scanning
+ MEMORY_BASIC_INFORMATION mem;
+ VirtualQuery(moduleHandle, &mem, sizeof(mem));
+
+ IMAGE_DOS_HEADER* dos = (IMAGE_DOS_HEADER*)mem.AllocationBase;
+ IMAGE_NT_HEADERS* pe = (IMAGE_NT_HEADERS*)((unsigned char*)dos + dos->e_lfanew);
+
+ return pe->OptionalHeader.SizeOfImage;
+}
+
+void* FindSignature(std::string dllName, const char* sig, const char* mask)
+{
+ HMODULE module = GetModuleHandleA(dllName.c_str());
+
+ unsigned char* dllAddress = (unsigned char*)module;
+ unsigned char* dllEnd = dllAddress + GetModuleLength(module);
+
+ size_t sigLength = strlen(mask);
+
+ for (auto i = dllAddress; i < dllEnd - sigLength + 1; i++)
+ {
+ int j = 0;
+ for (; j < sigLength; j++)
+ if (mask[j] != '?' && sig[j] != i[j])
+ break;
+
+ if (j == sigLength) // loop finished of its own accord
+ return i;
+ }
+
+ return nullptr;
+}