add new memory lib

1 files changed, 87 insertions, 0 deletions

diff --git a/NorthstarDLL/memory.h b/NorthstarDLL/memory.h
new file mode 100644
index 00000000..7ed4ed59
--- /dev/null
+++ b/NorthstarDLL/memory.h
@@ -0,0 +1,87 @@
+#pragma	once
+
+class MemoryAddress
+{
+  public:
+	uintptr_t m_nAddress;
+
+  public:
+	MemoryAddress();
+	MemoryAddress(const uintptr_t nAddress);
+	MemoryAddress(const void* pAddress);
+
+	//operators 
+	operator uintptr_t() const;
+	operator void*() const;
+	operator bool() const;
+
+	bool operator==(const MemoryAddress& other) const;
+	bool operator!=(const MemoryAddress& other) const;
+	bool operator==(const uintptr_t& addr) const;
+	bool operator!=(const uintptr_t& addr) const;
+
+	MemoryAddress operator+(const MemoryAddress& other) const;
+	MemoryAddress operator-(const MemoryAddress& other) const;
+	MemoryAddress operator+(const uintptr_t& other) const;
+	MemoryAddress operator-(const uintptr_t& other) const;
+	MemoryAddress operator*() const;
+
+	template <typename T> T As() { return reinterpret_cast<T>(m_nAddress); }
+
+	// traversal
+	MemoryAddress Offset(const uintptr_t nOffset) const;
+	MemoryAddress Deref(const int nNumDerefs = 1) const;
+
+	// patching
+	void Patch(const uint8_t* pBytes, const size_t nSize);
+	void Patch(const std::initializer_list<uint8_t> bytes);
+	void Patch(const char* pBytes);
+	void NOP(const size_t nSize);
+
+	bool IsMemoryReadable(const size_t nSize);
+};
+
+// based on https://github.com/Mauler125/r5sdk/blob/master/r5dev/public/include/module.h
+class CModule : public MemoryAddress
+{
+  public:
+	struct ModuleSections_t
+	{
+		ModuleSections_t(void) = default;
+		ModuleSections_t(const std::string& svSectionName, uintptr_t pSectionBase, size_t nSectionSize)
+			: m_svSectionName(svSectionName), m_pSectionBase(pSectionBase), m_nSectionSize(nSectionSize)
+		{
+		}
+
+		bool IsSectionValid(void) const
+		{
+			return m_nSectionSize != 0;
+		}
+
+		std::string m_svSectionName; // Name of section.
+		uintptr_t m_pSectionBase {}; // Start address of section.
+		size_t m_nSectionSize {}; // Size of section.
+	};
+
+	ModuleSections_t m_ExecutableCode;
+	ModuleSections_t m_ExceptionTable;
+	ModuleSections_t m_RunTimeData;
+	ModuleSections_t m_ReadOnlyData;
+
+  private:
+	std::string m_svModuleName;
+	uintptr_t m_pModuleBase {};
+	DWORD m_nModuleSize {};
+	IMAGE_NT_HEADERS64* m_pNTHeaders = nullptr;
+	IMAGE_DOS_HEADER* m_pDOSHeader = nullptr;
+	std::vector<ModuleSections_t> m_vModuleSections;
+
+  public:
+	CModule() = delete; // no default, we need a module name
+	CModule(const HMODULE pModule);
+	CModule(const char* pModuleName);
+
+	MemoryAddress GetExport(const char* pExportName);
+	MemoryAddress FindPattern(const uint8_t* pPattern, const char* pMask);
+	MemoryAddress FindPattern(const char* pPattern);
+};
\ No newline at end of file