增加base::hook::patch

Development/Editor/Core/YDWEBase/base/hook/detail/memory_protect.cpp

@@ -2,19 +2,15 @@
 
 namespace base { namespace hook { namespace detail {
 
-	memory_protect::memory_protect(uintptr_t address)
+	memory_protect::memory_protect(uintptr_t address, size_t size)
 		: success_(false)
+		, access_(0)
+		, address_(address)
+		, size_(size)
 	{
-		if (0 == ::VirtualQuery((LPCVOID)address, &mbi_, sizeof(mbi_)))
-			return ;
-
-
-		if (!::VirtualProtect(mbi_.BaseAddress, mbi_.RegionSize, PAGE_WRITECOPY, &access_))
+		if (!::VirtualProtect((LPVOID)address_, size_, PAGE_READWRITE, &access_))
 		{
-			if (!::VirtualProtect(mbi_.BaseAddress, mbi_.RegionSize, PAGE_READWRITE, &access_))
-			{
-				return ;
-			}
+			return;
 		}
 
 		success_ = true;
@@ -24,8 +20,8 @@ namespace base { namespace hook { namespace detail {
 	{
 		if (success())
 		{
-			DWORD newAccess;
-			::VirtualProtect(mbi_.BaseAddress, mbi_.RegionSize, access_, &newAccess);
+			DWORD new_access;
+			::VirtualProtect((LPVOID)address_, size_, access_, &new_access);
 		}
 	}
 

Development/Editor/Core/YDWEBase/base/hook/detail/memory_protect.h

@@ -3,19 +3,18 @@
 #include <base/config.h>
 #include <Windows.h>
 
-namespace base { 
-namespace hook { namespace detail {
+namespace base { namespace hook { namespace detail {
 	class memory_protect
 	{
 	public:
-		memory_protect(uintptr_t address);
+		memory_protect(uintptr_t address, size_t size = sizeof(uintptr_t));
 		~memory_protect();
 		bool success() const;
 
 	private:
-		bool  success_;
-		MEMORY_BASIC_INFORMATION mbi_;
-		DWORD access_;
+		bool      success_;
+		DWORD     access_;
+		uintptr_t address_;
+		size_t    size_;
 	};
-}}
-}
+}}}

Development/Editor/Core/YDWEBase/base/hook/detail/replace_pointer.cpp

@@ -20,6 +20,8 @@ namespace base { namespace hook { namespace detail {
 			{
 				return 0;
 			}
+
+			::FlushInstructionCache(::GetCurrentProcess(), (LPVOID)address, sizeof(uintptr_t));
 		}
 
 		return old_value;

Development/Editor/Core/YDWEBase/base/hook/patch.cpp

@@ -0,0 +1,21 @@
+#include <base/hook/patch.h>
+#include <base/hook/detail/memory_protect.h>
+#include <Windows.h>
+
+namespace base { namespace hook {
+
+	bool patch(uintptr_t address, const char* value, size_t value_size)
+	{
+		detail::memory_protect protect_(address, value_size);
+		if (protect_.success())
+		{
+			DWORD written;
+			if (::WriteProcessMemory(::GetCurrentProcess(), (LPVOID)address, value, value_size, &written))
+			{
+				::FlushInstructionCache(::GetCurrentProcess(), (LPVOID)address, value_size);
+				return true;
+			}
+		}
+		return false;
+	}
+}}

Development/Editor/Core/YDWEBase/base/hook/patch.h

@@ -0,0 +1,23 @@
+#pragma once
+
+#include <base/config.h>
+#include <cassert>
+#include <cstdint>
+
+namespace base { namespace hook {
+	_BASE_API bool patch(uintptr_t address, const char* value, size_t value_size);
+
+
+	template <size_t n>
+	bool patch(uintptr_t address, const char(&value)[n])
+	{
+		assert(n > 1);
+		return patch(address, value, n - 1);
+	}
+
+	template <size_t n>
+	bool patch(uintptr_t address, char(&value)[n])
+	{
+		return patch(address, value, std::string::traits_type::length(value));
+	}
+}}