diff --git a/IPA.Injector/Injector.cs b/IPA.Injector/Injector.cs index e55a9513..cbbc1c38 100644 --- a/IPA.Injector/Injector.cs +++ b/IPA.Injector/Injector.cs @@ -109,8 +109,7 @@ namespace IPA.Injector { if (loadingDone) return; loadingDone = true; - AppDomain.CurrentDomain.AssemblyResolve += LibLoader.AssemblyLibLoader; - LibLoader.SetupAssemblyFilenames(true); + LibLoader.Configure(); } private static void InstallBootstrapPatch() diff --git a/IPA.Loader/Loader/LibLoader.cs b/IPA.Loader/Loader/LibLoader.cs index febff3df..c235a945 100644 --- a/IPA.Loader/Loader/LibLoader.cs +++ b/IPA.Loader/Loader/LibLoader.cs @@ -1,7 +1,11 @@ using System; using System.Collections.Generic; +using System.ComponentModel; +using System.Diagnostics.CodeAnalysis; using System.IO; using System.Reflection; +using System.Runtime.InteropServices; +using System.Linq; using IPA.Logging; using Mono.Cecil; @@ -34,6 +38,13 @@ namespace IPA.Loader internal static string NativeLibraryPath => Path.Combine(LibraryPath, "Native"); internal static Dictionary FilenameLocations; + internal static void Configure() + { + AppDomain.CurrentDomain.AssemblyResolve -= AssemblyLibLoader; + AppDomain.CurrentDomain.AssemblyResolve += AssemblyLibLoader; + SetupAssemblyFilenames(true); + } + internal static void SetupAssemblyFilenames(bool force = false) { if (FilenameLocations == null || force) @@ -44,6 +55,36 @@ namespace IPA.Loader if (FilenameLocations.ContainsKey(fn.Name)) Log(Logger.Level.Critical, $"Multiple instances of {fn.Name} exist in Libs! Ignoring {fn.FullName}"); else FilenameLocations.Add(fn.Name, fn.FullName); + + + if (!SetDefaultDllDirectories(LoadLibraryFlags.LOAD_LIBRARY_SEARCH_USER_DIRS | LoadLibraryFlags.LOAD_LIBRARY_SEARCH_SYSTEM32 + | LoadLibraryFlags.LOAD_LIBRARY_SEARCH_DEFAULT_DIRS | LoadLibraryFlags.LOAD_LIBRARY_SEARCH_APPLICATION_DIR)) + { + var err = new Win32Exception(); + Log(Logger.Level.Critical, $"Error configuring DLL search path"); + Log(Logger.Level.Critical, err); + return; + } + + void AddDir(string path) + { + var retPtr = AddDllDirectory(path); + if (retPtr == IntPtr.Zero) + { + var err = new Win32Exception(); + Log(Logger.Level.Warning, $"Could not add DLL directory"); + Log(Logger.Level.Warning, err); + } + } + + if (Directory.Exists(NativeLibraryPath)) + { + AddDir(NativeLibraryPath); + TraverseTree(NativeLibraryPath, dir => + { // this is a terrible hack for iterating directories + AddDir(dir); return true; + }).All(f => true); // force it to iterate all + } } } @@ -92,8 +133,17 @@ namespace IPA.Loader if (((byte)lvl & (byte)StandardLogger.PrintFilter) != 0) Console.WriteLine($"[{lvl}] {message}"); } + internal static void Log(Logger.Level lvl, Exception message) + { // multiple proxy methods to delay loading of assemblies until it's done + if (Logger.LogCreated) + AssemblyLibLoaderCallLogger(lvl, message); + else + if (((byte)lvl & (byte)StandardLogger.PrintFilter) != 0) + Console.WriteLine($"[{lvl}] {message}"); + } private static void AssemblyLibLoaderCallLogger(Logger.Level lvl, string message) => Logger.libLoader.Log(lvl, message); + private static void AssemblyLibLoaderCallLogger(Logger.Level lvl, Exception message) => Logger.libLoader.Log(lvl, message); // https://docs.microsoft.com/en-us/dotnet/csharp/programming-guide/file-system/how-to-iterate-through-a-directory-tree private static IEnumerable TraverseTree(string root, Func dirValidator = null) @@ -115,13 +165,9 @@ namespace IPA.Loader subDirs = Directory.GetDirectories(currentDir); } catch (UnauthorizedAccessException) - { - continue; - } + { continue; } catch (DirectoryNotFoundException) - { - continue; - } + { continue; } string[] files; try @@ -129,13 +175,9 @@ namespace IPA.Loader files = Directory.GetFiles(currentDir); } catch (UnauthorizedAccessException) - { - continue; - } + { continue; } catch (DirectoryNotFoundException) - { - continue; - } + { continue; } foreach (string str in subDirs) if (dirValidator(str)) dirs.Push(str); @@ -148,15 +190,29 @@ namespace IPA.Loader nextValue = new FileInfo(file); } catch (FileNotFoundException) - { - continue; - } + { continue; } yield return nextValue; } } } - + [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)] + private static extern IntPtr AddDllDirectory(string lpPathName); + + [Flags] + [SuppressMessage("ReSharper", "InconsistentNaming")] + [SuppressMessage("ReSharper", "UnusedMember.Local")] + private enum LoadLibraryFlags : uint + { + None = 0, + LOAD_LIBRARY_SEARCH_APPLICATION_DIR = 0x00000200, + LOAD_LIBRARY_SEARCH_DEFAULT_DIRS = 0x00001000, + LOAD_LIBRARY_SEARCH_SYSTEM32 = 0x00000800, + LOAD_LIBRARY_SEARCH_USER_DIRS = 0x00000400, + } + + [DllImport("kernel32.dll", SetLastError = true)] + private static extern bool SetDefaultDllDirectories(LoadLibraryFlags dwFlags); } }