tyler/Libation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change assembly loadig

Browse Source
This commit is contained in:
Michael Bucari-Tovo 2022-08-18 10:29:30 -06:00
parent 7ffdf45164
commit 8af4c71101
1 changed files with 32 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
Source/LibationFileManager/InteropFactory.cs
View File

@ -32,21 +32,27 @@ namespace LibationFileManager
: Configuration.IsMacOs ? a => Path.GetFileName(a).StartsWithInsensitive("mac") || Path.GetFileName(a).StartsWithInsensitive("osx") : Configuration.IsMacOs ? a => Path.GetFileName(a).StartsWithInsensitive("mac") || Path.GetFileName(a).StartsWithInsensitive("osx")
: _ => false; : _ => false;
private const string CONFIG_APP_ENDING = "ConfigApp.exe"; private const string CONFIG_APP_ENDING = "ConfigApp.dll";
private static List<ProcessModule> ModuleList { get; } = new(); private static List<ProcessModule> ModuleList { get; } = new();
static InteropFactory() static InteropFactory()
{ {
// searches file names for potential matches; doesn't run anything // searches file names for potential matches; doesn't run anything
var configApp = getOSConfigApp(); var configApp = getOSConfigApp();
#if DEBUG
// nothing to load // nothing to load
if (configApp is null) if (configApp is null)
{
Serilog.Log.Logger.Error($"Unable to locate *{CONFIG_APP_ENDING}");
return; return;
}
// runs the exe and gets the exe's loaded modules // runs the exe and gets the exe's loaded modules
ModuleList = LoadModuleList(Path.GetFileNameWithoutExtension(configApp)) ModuleList = LoadModuleList(Path.GetFileNameWithoutExtension(configApp))
.OrderBy(x => x.ModuleName) .OrderBy(x => x.ModuleName)
.ToList(); .ToList();
#endif
AppDomain.CurrentDomain.AssemblyResolve += CurrentDomain_AssemblyResolve; AppDomain.CurrentDomain.AssemblyResolve += CurrentDomain_AssemblyResolve;
@ -61,18 +67,13 @@ namespace LibationFileManager
var here = Path.GetDirectoryName(Environment.ProcessPath); var here = Path.GetDirectoryName(Environment.ProcessPath);
// find '*ConfigApp.exe' files // find '*ConfigApp.exe' files
var exes = var appName =
Directory.EnumerateFiles(here, $"*{CONFIG_APP_ENDING}", SearchOption.TopDirectoryOnly) Directory.EnumerateFiles(here, $"*{CONFIG_APP_ENDING}*", SearchOption.TopDirectoryOnly)
// sanity check. shouldn't ever be true // sanity check. shouldn't ever be true
.Except(new[] { Environment.ProcessPath }) .Except(new[] { Environment.ProcessPath })
.Where(exe => .FirstOrDefault(exe => MatchesOS(exe));
// has a corresponding dll
File.Exists(Path.ChangeExtension(exe, "dll")) return appName;
&& MatchesOS(exe)
)
.ToList();
var exeName = exes.FirstOrDefault();
return exeName;
} }
private static List<ProcessModule> LoadModuleList(string exeName) private static List<ProcessModule> LoadModuleList(string exeName)
@ -115,12 +116,28 @@ namespace LibationFileManager
private static Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args) private static Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args)
{ {
// e.g. "System.Windows.Forms, Version=6.0.2.0, Culture=neutral, PublicKeyToken=b77a5c561934e089" // e.g. "System.Windows.Forms, Version=6.0.2.0, Culture=neutral, PublicKeyToken=b77a5c561934e089"
var asmName = args.Name.Split(',')[0]; var asmName = args.Name.Split(',')[0] + ".dll";
#if DEBUG
// `First` instead of `FirstOrDefault`. If it's not present we're going to fail anyway. May as well be here // `First` instead of `FirstOrDefault`. If it's not present we're going to fail anyway. May as well be here
var module = ModuleList.First(m => m.ModuleName.StartsWith(asmName)); var modulePath = ModuleList.SingleOrDefault(m => m.ModuleName.EqualsInsensitive(asmName)).FileName;
#else
var here = Path.GetDirectoryName(Environment.ProcessPath);
return Assembly.LoadFrom(module.FileName); // find '*ConfigApp.dll' files
var modulePath =
Directory.EnumerateFiles(here, asmName, SearchOption.TopDirectoryOnly)
.SingleOrDefault();
#endif
if (modulePath is null)
{
Serilog.Log.Logger.Error($"Unable to load module {args.Name}");
return null;
}
return Assembly.LoadFrom(modulePath);
} }
#endregion #endregion