|
|
- using Harmony;
- using System.Collections.Generic;
- using System.Reflection;
-
- namespace IPA.Loader
- {
- internal static class HarmonyProtector
- {
- private static HarmonyInstance instance;
- private static Assembly selfAssem;
- private static Assembly harmonyAssem;
-
- public static void Protect(HarmonyInstance inst = null)
- {
- selfAssem = Assembly.GetExecutingAssembly();
- harmonyAssem = typeof(HarmonyInstance).Assembly;
-
- if (inst == null)
- {
- if (instance == null)
- instance = HarmonyInstance.Create("BSIPA Safeguard");
-
- inst = instance;
- }
-
- var target = typeof(PatchProcessor).GetMethod("Patch");
- var patch = typeof(HarmonyProtector).GetMethod(nameof(PatchProcessor_Patch_Prefix));
-
- inst.Patch(target, new HarmonyMethod(patch));
- }
-
- private static void PatchProcessor_Patch_Prefix(PatchProcessor __instance, ref List<MethodBase> ___originals)
- {
- for (int i = 0; i < ___originals.Count; i++)
- {
- var mi = ___originals[i];
- var asm = mi.DeclaringType.Assembly;
-
- if (asm.Equals(selfAssem) || asm.Equals(harmonyAssem))
- ___originals.RemoveAt(i--);
- }
- }
- }
- }
|