ToxicMushroom/Dalamud
1
0
Fork 0
mirror of https://github.com/goatcorp/Dalamud.git synced 2025-12-12 18:27:23 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

another fix

Browse source
This commit is contained in:
Soreepeong 2023-03-10 22:53:48 +09:00
parent 9ee8ad67b4
commit f16628beb0
1 changed files with 25 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

44
Dalamud/Hooking/Internal/FunctionPointerVariableHook.cs
View file

@ -12,7 +12,7 @@ namespace Dalamud.Hooking.Internal;
/// Manages a hook with MinHook.
/// </summary>
/// <typeparam name="T">Delegate type to represents a function prototype. This must be the same prototype as original function do.</typeparam>
internal unsafe class FunctionPointerVariableHook<T> : Hook<T>
internal class FunctionPointerVariableHook<T> : Hook<T>
where T : Delegate
{
private readonly nint pfnDetour;
@ -21,13 +21,13 @@ internal unsafe class FunctionPointerVariableHook<T> : Hook<T>
// ReSharper disable once NotAccessedField.Local
private readonly T detourDelegate;
private readonly byte* pfnThunk;
private readonly nint* ppfnThunkJumpTarget;
private readonly nint pfnThunk;
private readonly nint ppfnThunkJumpTarget;
private readonly nint pfnOriginal;
private readonly T originalDelegate;
private bool enabled = false;
private bool enabled;
/// <summary>
/// Initializes a new instance of the <see cref="FunctionPointerVariableHook{T}"/> class.
@ -55,20 +55,26 @@ internal unsafe class FunctionPointerVariableHook<T> : Hook<T>
this.detourDelegate = detour;
this.pfnDetour = Marshal.GetFunctionPointerForDelegate(detour);
this.pfnThunk = (byte*)NativeFunctions.HeapAlloc(HookManager.NoFreeExecutableHeap, 0, 12);
if (this.pfnThunk == null)
unsafe
{
throw new OutOfMemoryException("Failed to allocate memory for import hooks.");
var pfnThunkBytes = (byte*)NativeFunctions.HeapAlloc(HookManager.NoFreeExecutableHeap, 0, 12);
if (pfnThunkBytes == null)
{
throw new OutOfMemoryException("Failed to allocate memory for import hooks.");
}
// movabs rax, imm
pfnThunkBytes[0] = 0x48;
pfnThunkBytes[1] = 0xB8;
// jmp rax
pfnThunkBytes[10] = 0xFF;
pfnThunkBytes[11] = 0xE0;
this.pfnThunk = (nint)pfnThunkBytes;
}
// movabs rax, imm
this.pfnThunk[0] = 0x48;
this.pfnThunk[1] = 0xB8;
this.ppfnThunkJumpTarget = (nint*)&this.pfnThunk[2];
// jmp rax
this.pfnThunk[10] = 0xFF;
this.pfnThunk[11] = 0xE0;
this.ppfnThunkJumpTarget = this.pfnThunk + 2;
if (!NativeFunctions.VirtualProtect(
this.Address,
@ -81,8 +87,8 @@ internal unsafe class FunctionPointerVariableHook<T> : Hook<T>
this.pfnOriginal = Marshal.ReadIntPtr(this.Address);
this.originalDelegate = Marshal.GetDelegateForFunctionPointer<T>(this.pfnOriginal);
*this.ppfnThunkJumpTarget = this.pfnOriginal;
Marshal.WriteIntPtr(this.Address, this.pfnDetour);
Marshal.WriteIntPtr(this.ppfnThunkJumpTarget, this.pfnOriginal);
Marshal.WriteIntPtr(this.Address, this.pfnThunk);
// This really should not fail, but then even if it does, whatever.
NativeFunctions.VirtualProtect(this.Address, (UIntPtr)Marshal.SizeOf<IntPtr>(), oldProtect, out _);
@ -145,7 +151,7 @@ internal unsafe class FunctionPointerVariableHook<T> : Hook<T>
lock (HookManager.HookEnableSyncRoot)
{
*this.ppfnThunkJumpTarget = this.pfnDetour;
Marshal.WriteIntPtr(this.ppfnThunkJumpTarget, this.pfnDetour);
this.enabled = true;
}
}
@ -162,7 +168,7 @@ internal unsafe class FunctionPointerVariableHook<T> : Hook<T>
lock (HookManager.HookEnableSyncRoot)
{
*this.ppfnThunkJumpTarget = this.pfnOriginal;
Marshal.WriteIntPtr(this.ppfnThunkJumpTarget, this.pfnOriginal);
this.enabled = false;
}
}