123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180 |
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Reflection;
- using BepInEx.Logging;
- using Mono.Cecil;
- namespace BepInEx.Preloader.Patcher
- {
-
-
-
-
- internal delegate void AssemblyPatcherDelegate(ref AssemblyDefinition assembly);
-
-
-
-
- internal static class AssemblyPatcher
- {
- private static readonly List<PatcherPlugin> patchers = new List<PatcherPlugin>();
-
-
-
- private static bool DumpingEnabled =>
- Utility.SafeParseBool(Config.GetEntry("dump-assemblies", "false", "Preloader"));
-
-
-
-
- public static void AddPatcher(PatcherPlugin patcher)
- {
- patchers.Add(patcher);
- }
-
-
-
-
-
- public static void AddPatchersFromDirectory(string directory,
- Func<Assembly, List<PatcherPlugin>> patcherLocator)
- {
- if (!Directory.Exists(directory))
- return;
- var sortedPatchers = new SortedDictionary<string, PatcherPlugin>();
- foreach (string assemblyPath in Directory.GetFiles(directory, "*.dll"))
- try
- {
- var assembly = Assembly.LoadFrom(assemblyPath);
- foreach (var patcher in patcherLocator(assembly))
- sortedPatchers.Add(patcher.Name, patcher);
- }
- catch (BadImageFormatException)
- {
- }
- catch (ReflectionTypeLoadException)
- {
- }
- foreach (KeyValuePair<string, PatcherPlugin> patcher in sortedPatchers)
- AddPatcher(patcher.Value);
- }
- private static void InitializePatchers()
- {
- foreach (var assemblyPatcher in patchers)
- assemblyPatcher.Initializer?.Invoke();
- }
- private static void FinalizePatching()
- {
- foreach (var assemblyPatcher in patchers)
- assemblyPatcher.Finalizer?.Invoke();
- }
-
-
-
- public static void DisposePatchers()
- {
- patchers.Clear();
- }
-
-
-
-
- public static void PatchAndLoad(string directory)
- {
-
- var assemblies = new Dictionary<string, AssemblyDefinition>();
- foreach (string assemblyPath in Directory.GetFiles(directory, "*.dll"))
- {
- var assembly = AssemblyDefinition.ReadAssembly(assemblyPath);
-
-
-
-
- if (assembly.Name.Name == "System"
- || assembly.Name.Name == "mscorlib"
- )
- {
- assembly.Dispose();
- continue;
- }
- if (UnityPatches.AssemblyLocations.ContainsKey(assembly.FullName))
- {
- Logger.Log(LogLevel.Warning,
- $"Tried to load duplicate assembly {Path.GetFileName(assemblyPath)} from Managed folder! Skipping...");
- continue;
- }
- assemblies.Add(Path.GetFileName(assemblyPath), assembly);
- UnityPatches.AssemblyLocations.Add(assembly.FullName, Path.GetFullPath(assemblyPath));
- }
-
- InitializePatchers();
-
- var patchedAssemblies = new HashSet<string>();
- foreach (var assemblyPatcher in patchers)
- foreach (string targetDll in assemblyPatcher.TargetDLLs)
- if (assemblies.TryGetValue(targetDll, out var assembly))
- {
- assemblyPatcher.Patcher?.Invoke(ref assembly);
- assemblies[targetDll] = assembly;
- patchedAssemblies.Add(targetDll);
- }
-
- foreach (KeyValuePair<string, AssemblyDefinition> kv in assemblies)
- {
- string filename = kv.Key;
- var assembly = kv.Value;
- if (DumpingEnabled && patchedAssemblies.Contains(filename))
- using (var mem = new MemoryStream())
- {
- string dirPath = Path.Combine(Paths.PluginPath, "DumpedAssemblies");
- if (!Directory.Exists(dirPath))
- Directory.CreateDirectory(dirPath);
- assembly.Write(mem);
- File.WriteAllBytes(Path.Combine(dirPath, filename), mem.ToArray());
- }
- Load(assembly);
- assembly.Dispose();
- }
-
- FinalizePatching();
- }
-
-
-
-
- public static void Load(AssemblyDefinition assembly)
- {
- using (var assemblyStream = new MemoryStream())
- {
- assembly.Write(assemblyStream);
- Assembly.Load(assemblyStream.ToArray());
- }
- }
- }
- }
|