fergalmoran/DnsServer
1
0
Fork 0
mirror of https://github.com/fergalmoran/DnsServer.git synced 2025-12-22 09:29:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

DnsApplicationAssemblyLoadContext: updated implementation to load managed reference DLLs too using temp files to allow update/uninstall of app at runtime on Windows.

Browse Source
This commit is contained in:
Shreyas Zare
2025-01-26 17:11:44 +05:30
parent 02355affb6
commit 764fa4935e
1 changed files with 28 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
DnsServerCore/Dns/Applications/DnsApplicationAssemblyLoadContext.cs
View File

@@ -37,7 +37,7 @@ namespace DnsServerCore.Dns.Applications
readonly AssemblyDependencyResolver _dependencyResolver;
readonly Dictionary<string, IntPtr> _loadedUnmanagedDlls = new Dictionary<string, IntPtr>();
readonly List<string> _unmanagedDllTempPaths = new List<string>();
readonly List<string> _dllTempPaths = new List<string>();
#endregion
@@ -50,11 +50,11 @@ namespace DnsServerCore.Dns.Applications
Unloading += delegate (AssemblyLoadContext obj)
{
foreach (string unmanagedDllTempPath in _unmanagedDllTempPaths)
foreach (string dllTempPath in _dllTempPaths)
{
try
{
File.Delete(unmanagedDllTempPath);
File.Delete(dllTempPath);
}
catch
{ }
@@ -116,8 +116,13 @@ namespace DnsServerCore.Dns.Applications
{
string resolvedPath = _dependencyResolver.ResolveAssemblyToPath(assemblyName);
if (!string.IsNullOrEmpty(resolvedPath) && File.Exists(resolvedPath))
{
if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
return LoadFromAssemblyPath(GetTempDllFile(resolvedPath));
else
return LoadFromAssemblyPath(resolvedPath);
}
}
foreach (Assembly loadedAssembly in Default.Assemblies)
{
@@ -199,26 +204,9 @@ namespace DnsServerCore.Dns.Applications
//load the unmanaged DLL
if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
{
//copy unmanaged dll into temp file for loading to allow uninstalling/updating app at runtime.
string tempPath = Path.GetTempFileName();
using (FileStream srcFile = new FileStream(unmanagedDllPath, FileMode.Open, FileAccess.Read))
{
using (FileStream dstFile = new FileStream(tempPath, FileMode.Create, FileAccess.Write))
{
srcFile.CopyTo(dstFile);
}
}
_unmanagedDllTempPaths.Add(tempPath);
value = LoadUnmanagedDllFromPath(tempPath);
}
value = LoadUnmanagedDllFromPath(GetTempDllFile(unmanagedDllPath));
else
{
value = LoadUnmanagedDllFromPath(unmanagedDllPath);
}
_loadedUnmanagedDlls.Add(unmanagedDllPath.ToLowerInvariant(), value);
}
@@ -231,6 +219,24 @@ namespace DnsServerCore.Dns.Applications
#region private
private string GetTempDllFile(string dllFile)
{
//copy dll into temp file for loading to allow uninstalling/updating app at runtime.
string tempPath = Path.GetTempFileName();
using (FileStream srcFile = new FileStream(dllFile, FileMode.Open, FileAccess.Read))
{
using (FileStream dstFile = new FileStream(tempPath, FileMode.Create, FileAccess.Write))
{
srcFile.CopyTo(dstFile);
}
}
_dllTempPaths.Add(tempPath);
return tempPath;
}
private string FindUnmanagedDllPath(string unmanagedDllName, string runtime, string[] prefixes, string[] extensions)
{
foreach (string prefix in prefixes)