-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Add handler in case the existing mmhook is corrupted * Code maid * add precise hashing config and impl Co-authored-by: xiaoxiao921 <xiaoxiao921@hotmail.fr>
- Loading branch information
1 parent
dd8b082
commit e488c5b
Showing
1 changed file
with
154 additions
and
121 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,122 +1,155 @@ | ||
| using BepInEx.Configuration; | ||
| using Mono.Cecil; | ||
| using MonoMod; | ||
| using MonoMod.RuntimeDetour.HookGen; | ||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.IO; | ||
|
|
||
| namespace BepInEx.MonoMod.HookGenPatcher | ||
| { | ||
| public static class HookGenPatcher | ||
| { | ||
| internal static Logging.ManualLogSource Logger = Logging.Logger.CreateLogSource("HookGenPatcher"); | ||
|
|
||
| private const string CONFIG_FILE_NAME = "HookGenPatcher.cfg"; | ||
|
|
||
| private static readonly ConfigFile Config = new ConfigFile(Path.Combine(Paths.ConfigPath, CONFIG_FILE_NAME), true); | ||
|
|
||
| private const char EntrySeparator = ','; | ||
|
|
||
| private static readonly ConfigEntry<string> AssemblyNamesToHookGenPatch = Config.Bind("General", "MMHOOKAssemblyNames", | ||
| "Assembly-CSharp.dll", $"Assembly names to make mmhooks for, separate entries with : {EntrySeparator} "); | ||
|
|
||
| public static IEnumerable<string> TargetDLLs { get; } = new string[] { }; | ||
|
|
||
| /** | ||
| * Code largely based on https://github.com/MonoMod/MonoMod/blob/master/MonoMod.RuntimeDetour.HookGen/Program.cs | ||
| */ | ||
|
|
||
| public static void Initialize() | ||
| { | ||
| var assemblyNames = AssemblyNamesToHookGenPatch.Value.Split(EntrySeparator); | ||
|
|
||
| var mmhookFolder = Path.Combine(Paths.PluginPath, "MMHOOK"); | ||
|
|
||
| foreach (var customAssemblyName in assemblyNames) | ||
| { | ||
| var mmhookFileName = "MMHOOK_" + customAssemblyName; | ||
|
|
||
| string pathIn = Path.Combine(Paths.ManagedPath, customAssemblyName); | ||
| string pathOut = Path.Combine(mmhookFolder, mmhookFileName); | ||
| bool shouldCreateDirectory = true; | ||
|
|
||
| foreach (string mmhookFile in Directory.EnumerateFiles(Paths.PluginPath, mmhookFileName, SearchOption.AllDirectories)) | ||
| { | ||
| if (Path.GetFileName(mmhookFile).Equals(mmhookFileName)) | ||
| { | ||
| pathOut = mmhookFile; | ||
| Logger.LogInfo("Previous MMHOOK location found. Using that location to save instead."); | ||
| shouldCreateDirectory = false; | ||
| break; | ||
| } | ||
| } | ||
|
|
||
| if (shouldCreateDirectory) | ||
| { | ||
| Directory.CreateDirectory(mmhookFolder); | ||
| } | ||
| var size = new FileInfo(pathIn).Length; | ||
|
|
||
| if (File.Exists(pathOut)) | ||
| { | ||
| try | ||
| { | ||
| using (var oldMM = AssemblyDefinition.ReadAssembly(pathOut)) | ||
| { | ||
| bool hash = oldMM.MainModule.GetType("BepHookGen.hash" + size) != null; | ||
| if (hash) | ||
| { | ||
| Logger.LogInfo("Already ran for this version, reusing that file."); | ||
| continue; | ||
| } | ||
| } | ||
| } | ||
| catch (BadImageFormatException) | ||
| { | ||
| Logger.LogWarning($"Failed to read {Path.GetFileName(pathOut)}, probably corrupted, remaking one."); | ||
| } | ||
| } | ||
|
|
||
| Environment.SetEnvironmentVariable("MONOMOD_HOOKGEN_PRIVATE", "1"); | ||
| Environment.SetEnvironmentVariable("MONOMOD_DEPENDENCY_MISSING_THROW", "0"); | ||
|
|
||
| using (MonoModder mm = new MonoModder() | ||
| { | ||
| InputPath = pathIn, | ||
| OutputPath = pathOut, | ||
| ReadingMode = ReadingMode.Deferred | ||
| }) | ||
| { | ||
| (mm.AssemblyResolver as BaseAssemblyResolver)?.AddSearchDirectory(Paths.BepInExAssemblyDirectory); | ||
|
|
||
| mm.Read(); | ||
|
|
||
| mm.MapDependencies(); | ||
|
|
||
| if (File.Exists(pathOut)) | ||
| { | ||
| Logger.LogDebug($"Clearing {pathOut}"); | ||
| File.Delete(pathOut); | ||
| } | ||
|
|
||
| Logger.LogInfo("Starting HookGenerator"); | ||
| HookGenerator gen = new HookGenerator(mm, Path.GetFileName(pathOut)); | ||
|
|
||
| using (ModuleDefinition mOut = gen.OutputModule) | ||
| { | ||
| gen.Generate(); | ||
| mOut.Types.Add(new TypeDefinition("BepHookGen", "hash" + size, TypeAttributes.AutoClass)); | ||
| mOut.Write(pathOut); | ||
| } | ||
|
|
||
| Logger.LogInfo("Done."); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| public static void Patch(AssemblyDefinition _) | ||
| { | ||
| } | ||
| } | ||
| using BepInEx.Configuration; | ||
| using Mono.Cecil; | ||
| using MonoMod; | ||
| using MonoMod.RuntimeDetour.HookGen; | ||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.IO; | ||
| using System.Security.Cryptography; | ||
|
|
||
| namespace BepInEx.MonoMod.HookGenPatcher | ||
| { | ||
| public static class HookGenPatcher | ||
| { | ||
| internal static Logging.ManualLogSource Logger = Logging.Logger.CreateLogSource("HookGenPatcher"); | ||
|
|
||
| private const string CONFIG_FILE_NAME = "HookGenPatcher.cfg"; | ||
|
|
||
| private static readonly ConfigFile Config = new ConfigFile(Path.Combine(Paths.ConfigPath, CONFIG_FILE_NAME), true); | ||
|
|
||
| private const char EntrySeparator = ','; | ||
|
|
||
| private static readonly ConfigEntry<string> AssemblyNamesToHookGenPatch = Config.Bind("General", "MMHOOKAssemblyNames", | ||
| "Assembly-CSharp.dll", $"Assembly names to make mmhooks for, separate entries with : {EntrySeparator} "); | ||
|
|
||
| private static readonly ConfigEntry<bool> preciseHash = Config.Bind<bool>("General", "Preciser filehashing", false, "Hash file using contents instead of size. Minor perfomance impact."); | ||
| private static bool skipHashing => !preciseHash.Value; | ||
|
|
||
| public static IEnumerable<string> TargetDLLs { get; } = new string[] { }; | ||
|
|
||
| /** | ||
| * Code largely based on https://github.com/MonoMod/MonoMod/blob/master/MonoMod.RuntimeDetour.HookGen/Program.cs | ||
| */ | ||
|
|
||
| public static void Initialize() | ||
| { | ||
| var assemblyNames = AssemblyNamesToHookGenPatch.Value.Split(EntrySeparator); | ||
|
|
||
| var mmhookFolder = Path.Combine(Paths.PluginPath, "MMHOOK"); | ||
|
|
||
| foreach (var customAssemblyName in assemblyNames) | ||
| { | ||
| var mmhookFileName = "MMHOOK_" + customAssemblyName; | ||
|
|
||
| string pathIn = Path.Combine(Paths.ManagedPath, customAssemblyName); | ||
| string pathOut = Path.Combine(mmhookFolder, mmhookFileName); | ||
| bool shouldCreateDirectory = true; | ||
|
|
||
| foreach (string mmhookFile in Directory.EnumerateFiles(Paths.PluginPath, mmhookFileName, SearchOption.AllDirectories)) | ||
| { | ||
| if (Path.GetFileName(mmhookFile).Equals(mmhookFileName)) | ||
| { | ||
| pathOut = mmhookFile; | ||
| Logger.LogInfo("Previous MMHOOK location found. Using that location to save instead."); | ||
| shouldCreateDirectory = false; | ||
| break; | ||
| } | ||
| } | ||
|
|
||
| if (shouldCreateDirectory) | ||
| { | ||
| Directory.CreateDirectory(mmhookFolder); | ||
| } | ||
|
|
||
| var fileInfo = new FileInfo(pathIn); | ||
| var size = fileInfo.Length; | ||
| long hash = 0; | ||
|
|
||
| if (File.Exists(pathOut)) | ||
| { | ||
| try | ||
| { | ||
| using (var oldMM = AssemblyDefinition.ReadAssembly(pathOut)) | ||
| { | ||
| bool mmSizeHash = oldMM.MainModule.GetType("BepHookGen.size" + size) != null; | ||
| if (mmSizeHash) | ||
| { | ||
| if (skipHashing) | ||
| { | ||
| Logger.LogInfo("Already ran for this version, reusing that file."); | ||
| continue; | ||
| } | ||
| hash = fileInfo.makeHash(); | ||
| bool mmContentHash = oldMM.MainModule.GetType("BepHookGen.content" + hash) != null; | ||
| if (mmContentHash) | ||
| { | ||
| Logger.LogInfo("Already ran for this version, reusing that file."); | ||
| continue; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| catch (BadImageFormatException) | ||
| { | ||
| Logger.LogWarning($"Failed to read {Path.GetFileName(pathOut)}, probably corrupted, remaking one."); | ||
| } | ||
| } | ||
|
|
||
| Environment.SetEnvironmentVariable("MONOMOD_HOOKGEN_PRIVATE", "1"); | ||
| Environment.SetEnvironmentVariable("MONOMOD_DEPENDENCY_MISSING_THROW", "0"); | ||
|
|
||
| using (MonoModder mm = new MonoModder() | ||
| { | ||
| InputPath = pathIn, | ||
| OutputPath = pathOut, | ||
| ReadingMode = ReadingMode.Deferred | ||
| }) | ||
| { | ||
| (mm.AssemblyResolver as BaseAssemblyResolver)?.AddSearchDirectory(Paths.BepInExAssemblyDirectory); | ||
|
|
||
| mm.Read(); | ||
|
|
||
| mm.MapDependencies(); | ||
|
|
||
| if (File.Exists(pathOut)) | ||
| { | ||
| Logger.LogDebug($"Clearing {pathOut}"); | ||
| File.Delete(pathOut); | ||
| } | ||
|
|
||
| Logger.LogInfo("Starting HookGenerator"); | ||
| HookGenerator gen = new HookGenerator(mm, Path.GetFileName(pathOut)); | ||
|
|
||
| using (ModuleDefinition mOut = gen.OutputModule) | ||
| { | ||
| gen.Generate(); | ||
| mOut.Types.Add(new TypeDefinition("BepHookGen", "size" + size, TypeAttributes.AutoClass)); | ||
| if (!skipHashing) | ||
| { | ||
| mOut.Types.Add(new TypeDefinition("BepHookGen", "content" + (hash == 0 ? fileInfo.makeHash() : hash), TypeAttributes.AutoClass)); | ||
| } | ||
| mOut.Write(pathOut); | ||
| } | ||
|
|
||
| Logger.LogInfo("Done."); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| public static void Patch(AssemblyDefinition _) | ||
| { | ||
| } | ||
|
|
||
| private static long makeHash(this FileInfo fileInfo) | ||
| { | ||
| var fileStream = fileInfo.OpenRead(); | ||
| byte[] hashbuffer = null; | ||
| using (MD5 md5 = new MD5CryptoServiceProvider()) | ||
| { | ||
| hashbuffer = md5.ComputeHash(fileStream); | ||
| } | ||
| long hash = BitConverter.ToInt64(hashbuffer, 0); | ||
| return hash != 0 ? hash : 1; | ||
| } | ||
| } | ||
| } |