/// <summary>
/// Primitive patching. Inserts a jump to 'target' at 'site'. Works even if both methods'
/// callers have already been compiled.
/// </summary>
/// <param name="site"></param>
/// <param name="target"></param>
private static RedirectCallsState PatchJumpTo(IntPtr site, IntPtr target)
{
RedirectCallsState state = new RedirectCallsState();
// R11 is volatile.
unsafe
{
byte* sitePtr = (byte*)site.ToPointer();
state.a = *sitePtr;
state.b = *(sitePtr + 1);
state.c = *(sitePtr + 10);
state.d = *(sitePtr + 11);
state.e = *(sitePtr + 12);
state.f = *((ulong*) (sitePtr + 2));
*sitePtr = 0x49; // mov r11, target
*(sitePtr + 1) = 0xBB;
*((ulong*)(sitePtr + 2)) = (ulong)target.ToInt64();
*(sitePtr + 10) = 0x41; // jmp r11
*(sitePtr + 11) = 0xFF;
*(sitePtr + 12) = 0xE3;
}
return state;
}
public static void Deploy()
{
if (!deployed)
{
try
{
MethodInfo patch = typeof(PatchSeagulls).GetMethod("Patch", BindingFlags.Instance | BindingFlags.NonPublic);
state1 = RedirectionHelper.RedirectCalls(typeof(ParkAI).GetMethod("CountAnimals", BindingFlags.Instance | BindingFlags.NonPublic), patch);
state2 = RedirectionHelper.RedirectCalls(typeof(LandfillSiteAI).GetMethod("CountAnimals", BindingFlags.Instance | BindingFlags.NonPublic), patch);
state3 = RedirectionHelper.RedirectCalls(typeof(HarborAI).GetMethod("CountAnimals", BindingFlags.Instance | BindingFlags.NonPublic), patch);
state4 = RedirectionHelper.RedirectCalls(typeof(CargoHarborAI).GetMethod("CountAnimals", BindingFlags.Instance | BindingFlags.NonPublic), patch);
state5 = RedirectionHelper.RedirectCalls(typeof(ParkAI).GetMethod("TargetAnimals", BindingFlags.Instance | BindingFlags.NonPublic), patch);
deployed = true;
}
catch (Exception e)
{
Util.DebugPrint("Deploy:");
UnityEngine.Debug.LogException(e);
}
}
}
public static void RevertRedirect(MethodInfo m, RedirectCallsState state)
{
var fptr1 = m.MethodHandle.GetFunctionPointer();
RevertJumpTo(fptr1, state);
}
private static void RevertJumpTo(IntPtr site, RedirectCallsState state)
{
unsafe
{
byte* sitePtr = (byte*)site.ToPointer();
*sitePtr = state.a; // mov r11, target
*(sitePtr + 1) = state.b;
*((ulong*)(sitePtr + 2)) = state.f;
*(sitePtr + 10) = state.c; // jmp r11
*(sitePtr + 11) = state.d;
*(sitePtr + 12) = state.e;
}
}
