From f9bc3c9d1834cb8bd5f872b749b057c33e8b0102 Mon Sep 17 00:00:00 2001
From: BobTheBob <32057864+BobTheBob9@users.noreply.github.com>
Date: Wed, 24 Aug 2022 00:32:31 +0100
Subject: Adjust folder structure (#242)

* Adjust folder structure

* change launcher directory name
---
 NorthstarDLL/NSMem.h | 193 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 193 insertions(+)
 create mode 100644 NorthstarDLL/NSMem.h

(limited to 'NorthstarDLL/NSMem.h')

diff --git a/NorthstarDLL/NSMem.h b/NorthstarDLL/NSMem.h
new file mode 100644
index 00000000..9b1f9103
--- /dev/null
+++ b/NorthstarDLL/NSMem.h
@@ -0,0 +1,193 @@
+#pragma once
+#include "pch.h"
+
+// KittenPopo's memory stuff, made for northstar (because I really can't handle working with northstar's original memory stuff tbh)
+
+#pragma region Pattern Scanning
+namespace NSMem
+{
+	inline std::vector<int> HexBytesToString(const char* str)
+	{
+		std::vector<int> patternNums;
+		int size = strlen(str);
+		for (int i = 0; i < size; i++)
+		{
+			char c = str[i];
+
+			// If this is a space character, ignore it
+			if (c == ' ' || c == '\t')
+				continue;
+
+			if (c == '?')
+			{
+				// Add a wildcard (-1)
+				patternNums.push_back(-1);
+			}
+			else if (i < size - 1)
+			{
+				BYTE result = 0;
+				for (int j = 0; j < 2; j++)
+				{
+					int val = 0;
+					char c = *(str + i + j);
+					if (c >= 'a')
+					{
+						val = c - 'a' + 0xA;
+					}
+					else if (c >= 'A')
+					{
+						val = c - 'A' + 0xA;
+					}
+					else if (isdigit(c))
+					{
+						val = c - '0';
+					}
+					else
+					{
+						assert(false);
+						val = -1;
+					}
+
+					result += (j == 0) ? val * 16 : val;
+				}
+				patternNums.push_back(result);
+			}
+
+			i++;
+		}
+
+		return patternNums;
+	}
+
+	inline void* PatternScan(void* module, const int* pattern, int patternSize, int offset)
+	{
+		if (!module)
+			return NULL;
+
+		auto dosHeader = (PIMAGE_DOS_HEADER)module;
+		auto ntHeaders = (PIMAGE_NT_HEADERS)((BYTE*)module + dosHeader->e_lfanew);
+
+		auto sizeOfImage = ntHeaders->OptionalHeader.SizeOfImage;
+
+		auto scanBytes = (BYTE*)module;
+
+		for (auto i = 0; i < sizeOfImage - patternSize; ++i)
+		{
+			bool found = true;
+			for (auto j = 0; j < patternSize; ++j)
+			{
+				if (scanBytes[i + j] != pattern[j] && pattern[j] != -1)
+				{
+					found = false;
+					break;
+				}
+			}
+
+			if (found)
+			{
+				uintptr_t addressInt = (uintptr_t)(&scanBytes[i]) + offset;
+				return (uint8_t*)addressInt;
+			}
+		}
+
+		return nullptr;
+	}
+
+	inline void* PatternScan(const char* moduleName, const char* pattern, int offset = 0)
+	{
+		std::vector<int> patternNums = HexBytesToString(pattern);
+
+		return PatternScan(GetModuleHandleA(moduleName), &patternNums[0], patternNums.size(), offset);
+	}
+
+	inline void BytePatch(uintptr_t address, const BYTE* vals, int size)
+	{
+		WriteProcessMemory(GetCurrentProcess(), (LPVOID)address, vals, size, NULL);
+	}
+
+	inline void BytePatch(uintptr_t address, std::initializer_list<BYTE> vals)
+	{
+		std::vector<BYTE> bytes = vals;
+		if (!bytes.empty())
+			BytePatch(address, &bytes[0], bytes.size());
+	}
+
+	inline void BytePatch(uintptr_t address, const char* bytesStr)
+	{
+		std::vector<int> byteInts = HexBytesToString(bytesStr);
+		std::vector<BYTE> bytes;
+		for (int v : byteInts)
+			bytes.push_back(v);
+
+		if (!bytes.empty())
+			BytePatch(address, &bytes[0], bytes.size());
+	}
+
+	inline void NOP(uintptr_t address, int size)
+	{
+		BYTE* buf = (BYTE*)malloc(size);
+		memset(buf, 0x90, size);
+		BytePatch(address, buf, size);
+		free(buf);
+	}
+
+	inline bool IsMemoryReadable(void* ptr, size_t size)
+	{
+		static SYSTEM_INFO sysInfo;
+		if (!sysInfo.dwPageSize)
+			GetSystemInfo(&sysInfo); // This should always be 4096 unless ur playing on NES or some shit but whatever
+
+		MEMORY_BASIC_INFORMATION memInfo;
+
+		if (!VirtualQuery(ptr, &memInfo, sizeof(memInfo)))
+			return false;
+
+		if (memInfo.RegionSize < size)
+			return false;
+
+		return (memInfo.State & MEM_COMMIT) && !(memInfo.Protect & PAGE_NOACCESS);
+	}
+} // namespace NSMem
+
+#pragma region KHOOK
+struct KHookPatternInfo
+{
+	const char *moduleName, *pattern;
+	int offset = 0;
+
+	KHookPatternInfo(const char* moduleName, const char* pattern, int offset = 0) : moduleName(moduleName), pattern(pattern), offset(offset)
+	{
+	}
+};
+
+struct KHook
+{
+	KHookPatternInfo targetFunc;
+	void* targetFuncAddr;
+	void* hookFunc;
+	void** original;
+
+	static inline std::vector<KHook*> _allHooks;
+
+	KHook(KHookPatternInfo targetFunc, void* hookFunc, void** original) : targetFunc(targetFunc)
+	{
+		this->hookFunc = hookFunc;
+		this->original = original;
+		_allHooks.push_back(this);
+	}
+
+	bool Setup()
+	{
+		targetFuncAddr = NSMem::PatternScan(targetFunc.moduleName, targetFunc.pattern, targetFunc.offset);
+		if (!targetFuncAddr)
+			return false;
+
+		return (MH_CreateHook(targetFuncAddr, hookFunc, original) == MH_OK) && (MH_EnableHook(targetFuncAddr) == MH_OK);
+	}
+};
+#define KHOOK(name, funcPatternInfo, returnType, convention, args)                                                                         \
+	returnType convention hk##name args;                                                                                                   \
+	auto o##name = (returnType(convention*) args)0;                                                                                        \
+	KHook k##name = KHook(KHookPatternInfo funcPatternInfo, reinterpret_cast<void*>(&hk##name), (void**)&o##name);                         \
+	returnType convention hk##name args
+#pragma endregion
-- 
cgit v1.2.3