-
Notifications
You must be signed in to change notification settings - Fork 108
Expand file tree
/
Copy pathModuleInitAndCleanup.cs
More file actions
124 lines (106 loc) · 5.26 KB
/
Copy pathModuleInitAndCleanup.cs
File metadata and controls
124 lines (106 loc) · 5.26 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
using System;
using System.Collections.Generic;
using System.Management.Automation;
using System.IO;
using System.Reflection;
namespace Microsoft.PowerShell.PSResourceGet.Cmdlets
{
public class UnsafeAssemblyHandler : IModuleAssemblyInitializer, IModuleAssemblyCleanup
{
private static readonly Assembly s_self;
private static readonly string s_dependencyFolder;
private static readonly HashSet<string> s_dependencies;
private static readonly AssemblyLoadContextProxy s_proxy;
private static readonly HashSet<string> NetFrameworkLoadFromPath = new HashSet<string>(StringComparer.OrdinalIgnoreCase)
{
"System.Runtime.CompilerServices.Unsafe",
"System.Memory",
"System.Diagnostics.DiagnosticSource",
"System.Text.Json",
"System.Security.Cryptography.ProtectedData"
};
static UnsafeAssemblyHandler()
{
s_self = Assembly.GetExecutingAssembly();
s_dependencyFolder = Path.Combine(Path.GetDirectoryName(s_self.Location), "dependencies");
s_dependencies = new HashSet<string>(StringComparer.Ordinal);
s_proxy = AssemblyLoadContextProxy.CreateLoadContext("psresourceget-load-context");
foreach (string filePath in Directory.EnumerateFiles(s_dependencyFolder, "*.dll"))
{
s_dependencies.Add(AssemblyName.GetAssemblyName(filePath).FullName);
}
}
public void OnImport()
{
AppDomain.CurrentDomain.AssemblyResolve += HandleAssemblyResolve;
}
public void OnRemove(PSModuleInfo psModuleInfo)
{
AppDomain.CurrentDomain.AssemblyResolve -= HandleAssemblyResolve;
}
private static bool IsAssemblyMatching(AssemblyName assemblyName, Assembly requestingAssembly)
{
// The requesting assembly is always available in .NET, but could be null in .NET Framework.
// - When the requesting assembly is available, we check whether the loading request came from this
// module, so as to make sure we only act on the request from this module.
// - When the requesting assembly is not available, we just have to depend on the assembly name only.
// Sometimes the requesting assembly can be one of the NuGet.* dlls, in such case, we check if the
// requesting assembly is part of our pre-defined dependencies list.
return requestingAssembly is not null
? (requestingAssembly == s_self || (s_dependencies.Contains(requestingAssembly.FullName))) && s_dependencies.Contains(assemblyName.FullName)
: s_dependencies.Contains(assemblyName.FullName);
}
private static Assembly HandleAssemblyResolve(object sender, ResolveEventArgs args)
{
var requiredAssembly = new AssemblyName(args.Name);
string requiredAssemblyName = requiredAssembly.Name;
// If on .NET framework load specific assemblies from dependency folder
if (s_proxy is null
&& NetFrameworkLoadFromPath.Contains(requiredAssemblyName))
{
var netFxDepDllPath = Path.Combine(s_dependencyFolder, $"{requiredAssemblyName}.dll");
return Assembly.LoadFrom(netFxDepDllPath);
}
if (IsAssemblyMatching(requiredAssembly, args.RequestingAssembly))
{
string possibleAssembly = Path.Combine(s_dependencyFolder, $"{requiredAssembly.Name}.dll");
if (File.Exists(possibleAssembly))
{
// - In .NET, load the assembly into the custom assembly load context.
// - In .NET Framework, assembly conflict is not a problem, so we load the assembly
// by 'Assembly.LoadFrom', the same as what powershell.exe would do.
return s_proxy is not null
? s_proxy.LoadFromAssemblyPath(possibleAssembly)
: Assembly.LoadFrom(possibleAssembly);
}
}
return null;
}
}
internal class AssemblyLoadContextProxy
{
private readonly object _customContext;
private readonly MethodInfo _loadFromAssemblyPath;
private AssemblyLoadContextProxy(Type alc, string loadContextName)
{
var ctor = alc.GetConstructor(new[] { typeof(string), typeof(bool) });
_loadFromAssemblyPath = alc.GetMethod("LoadFromAssemblyPath", new[] { typeof(string) });
_customContext = ctor.Invoke(new object[] { loadContextName, false });
}
internal Assembly LoadFromAssemblyPath(string assemblyPath)
{
return (Assembly) _loadFromAssemblyPath.Invoke(_customContext, new[] { assemblyPath });
}
internal static AssemblyLoadContextProxy CreateLoadContext(string name)
{
if (string.IsNullOrEmpty(name))
{
throw new ArgumentNullException(nameof(name));
}
var alc = typeof(object).Assembly.GetType("System.Runtime.Loader.AssemblyLoadContext");
return alc is not null
? new AssemblyLoadContextProxy(alc, name)
: null;
}
}
}