Create a managed implementation of assembly binder

src/coreclr/System.Private.CoreLib/System.Private.CoreLib.csproj

@@ -215,7 +215,19 @@
     <Compile Include="$(BclSourcesRoot)\System\Runtime\InteropServices\MemoryMarshal.CoreCLR.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Runtime\InteropServices\NativeLibrary.CoreCLR.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Runtime\Intrinsics\X86\X86Base.CoreCLR.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\ApplicationContext.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\Assembly.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyLoadContext.Binder.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyLoadContext.BinderCommon.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyLoadContext.CoreCLR.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyLoadContext.CustomBinder.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyLoadContext.DefaultBinder.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyIdentity.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyName.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\AssemblyVersion.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\BindResult.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\TextualIdentityParser.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Runtime\Loader\Tracing\ResolutionAttemptedOperation.cs" />
     <Compile Include="$(BclSourcesRoot)\System\RuntimeArgumentHandle.cs" />
     <Compile Include="$(BclSourcesRoot)\System\RuntimeHandles.cs" />
     <Compile Include="$(BclSourcesRoot)\System\RuntimeType.ActivatorCache.cs" />
@@ -334,4 +346,4 @@
       <FileWrites Include="@(EventingSourceFile)" />
     </ItemGroup>
   </Target>
-</Project>
+</Project>

src/coreclr/System.Private.CoreLib/src/System/Reflection/MdImport.cs

@@ -312,6 +312,12 @@ internal MetadataImport(RuntimeModule module)
             // since the instance can be replaced during HotReload and EnC scenarios.
             m_metadataImport2 = GetMetadataImport(module);
         }
+
+        internal MetadataImport(IntPtr pIMetaDataAssemblyImport2)
+        {
+            ArgumentNullException.ThrowIfNull(pIMetaDataAssemblyImport2);
+            m_metadataImport2 = pIMetaDataAssemblyImport2;
+        }
         #endregion
 
         [LibraryImport(RuntimeHelpers.QCall, EntryPoint = "MetadataImport_Enum")]
@@ -610,6 +616,38 @@ public bool IsValidToken(int token)
         {
             return IsValidToken(m_metadataImport2, token);
         }
+
+        [MethodImpl(MethodImplOptions.InternalCall)]
+        private static extern int GetAssemblyFromScope(IntPtr scope, out uint tkAssembly);
+        public uint GetAssemblyFromScope()
+        {
+            ThrowBadImageExceptionForHR(GetAssemblyFromScope(m_metadataImport2, out uint tkAssembly));
+            return tkAssembly;
+        }
+
+        [MethodImpl(MethodImplOptions.InternalCall)]
+        private static extern unsafe int GetAssemblyProps(
+            IntPtr scope,
+            uint mda,
+            out byte* ppbPublicKey,
+            out uint pcbPublicKey,
+            out uint pulHashAlgId,
+            void** pszName,
+            void* pMetadata,
+            out uint pdwAsselblyFlags);
+        public unsafe void GetAssemblyProps(
+            uint assemblyToken,
+            out byte* publicKey,
+            out uint publicKeyLength,
+            out uint hashAlgId,
+            out string assemblyName,
+            void* pMetadata,
+            out uint asselblyFlags)
+        {
+            void* _name;
+            ThrowBadImageExceptionForHR(GetAssemblyProps(m_metadataImport2, assemblyToken, out publicKey, out publicKeyLength, out hashAlgId, &_name, pMetadata, out asselblyFlags));
+            assemblyName = new MdUtf8String(_name).ToString();
+        }
         #endregion
 
         private static void ThrowBadImageExceptionForHR(int hr)

src/coreclr/System.Private.CoreLib/src/System/Runtime/Loader/ApplicationContext.cs

@@ -0,0 +1,301 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.IO;
+using System.Threading;
+
+namespace System.Runtime.Loader
+{
+    internal sealed class TPAEntry
+    {
+        public string? ILFileName;
+        public string? NIFileName;
+    }
+
+    internal sealed class FailureCacheKey : IEquatable<FailureCacheKey>
+    {
+        public readonly string SimpleName;
+        public readonly AssemblyVersion AssemblyVersion;
+
+        public FailureCacheKey(string simpleName, AssemblyVersion AssemblyVersion)
+        {
+            SimpleName = simpleName;
+            this.AssemblyVersion = AssemblyVersion;
+        }
+
+        public override bool Equals(object? obj) => obj is FailureCacheKey other && Equals(other);
+
+        public bool Equals(FailureCacheKey? other) => other != null && SimpleName == other.SimpleName && AssemblyVersion.Equals(other.AssemblyVersion);
+
+        public override int GetHashCode() => HashCode.Combine(SimpleName, AssemblyVersion.Major, AssemblyVersion.Minor, AssemblyVersion.Build, AssemblyVersion.Revision);
+    }
+
+    internal sealed class ApplicationContext
+    {
+        private volatile int _version;
+
+        public int Version => _version;
+
+        public Dictionary<BinderAssemblyName, BinderAssembly> ExecutionContext { get; } = new Dictionary<BinderAssemblyName, BinderAssembly>();
+
+        public Dictionary<FailureCacheKey, int> FailureCache { get; } = new Dictionary<FailureCacheKey, int>();
+
+        public object ContextCriticalSection { get; } = new object();
+
+        public List<string> PlatformResourceRoots { get; } = new List<string>();
+
+        public List<string> AppPaths { get; } = new List<string>();
+
+        public Dictionary<string, TPAEntry>? TrustedPlatformAssemblyMap { get; private set; }
+
+        public void IncrementVersion() => Interlocked.Increment(ref _version);
+
+        private static bool GetNextPath(string paths, ref int startPos, out string outPath)
+        {
+            bool wrappedWithQuotes = false;
+
+            // Skip any leading spaces or path separators
+            while (startPos < paths.Length && paths[startPos] is ' ' or PathInternal.PathSeparator)
+                startPos++;
+
+            if (startPos == paths.Length)
+            {
+                // No more paths in the string and we just skipped over some white space
+                outPath = string.Empty;
+                return false;
+            }
+
+            // Support paths being wrapped with quotations
+            while (startPos < paths.Length && paths[startPos] == '\"')
+            {
+                startPos++;
+                wrappedWithQuotes = true;
+            }
+
+            int iEnd = startPos; // Where current path ends
+            int iNext;           // Where next path starts
+
+            static int IndexOfInRange(ReadOnlySpan<char> str, int start, char ch)
+            {
+                int index = str[start..].IndexOf(ch);
+                return index >= 0 ? index + start : index;
+            }
+
+            if (wrappedWithQuotes)
+            {
+                iEnd = IndexOfInRange(paths, iEnd, '\"');
+                if (iEnd != -1)
+                {
+                    // Find where the next path starts - there should be a path separator right after the closing quotation mark
+                    iNext = IndexOfInRange(paths, iEnd, PathInternal.PathSeparator);
+                    if (iNext != -1)
+                    {
+                        iNext++;
+                    }
+                    else
+                    {
+                        iNext = paths.Length;
+                    }
+                }
+                else
+                {
+                    // There was no terminating quotation mark - that's bad
+                    throw new ArgumentException(nameof(paths));
+                }
+            }
+            else if ((iEnd = IndexOfInRange(paths, iEnd, PathInternal.PathSeparator)) != -1)
+            {
+                iNext = iEnd + 1;
+            }
+            else
+            {
+                iNext = iEnd = paths.Length;
+            }
+
+            // Skip any trailing spaces
+            while (paths[iEnd - 1] == ' ')
+            {
+                iEnd--;
+            }
+
+            Debug.Assert(startPos < iEnd);
+
+            outPath = paths[startPos..iEnd];
+            startPos = iNext;
+            return true;
+        }
+
+        private static bool GetNextTPAPath(string paths, ref int startPos, bool dllOnly, out string outPath, out string simpleName, out bool isNativeImage)
+        {
+            isNativeImage = false;
+
+            while (true)
+            {
+                if (!GetNextPath(paths, ref startPos, out outPath))
+                {
+                    simpleName = string.Empty;
+                    return false;
+                }
+
+                if (!Path.IsPathFullyQualified(outPath))
+                {
+                    throw new ArgumentException(nameof(paths));
+                }
+
+                // Find the beginning of the simple name
+                int iSimpleNameStart = outPath.LastIndexOf(PathInternal.DirectorySeparatorChar);
+                if (iSimpleNameStart == -1)
+                {
+                    iSimpleNameStart = 0;
+                }
+                else
+                {
+                    // Advance past the directory separator to the first character of the file name
+                    iSimpleNameStart++;
+                }
+
+                if (iSimpleNameStart == outPath.Length)
+                {
+                    throw new ArgumentException(nameof(paths));
+                }
+
+                if (dllOnly && (outPath.EndsWith(".exe", StringComparison.OrdinalIgnoreCase)
+                    || outPath.EndsWith(".ni.exe", StringComparison.OrdinalIgnoreCase)))
+                {
+                    // Skip exe files when the caller requested only dlls
+                    continue;
+                }
+
+                if (outPath.EndsWith(".ni.dll", StringComparison.OrdinalIgnoreCase)
+                    || outPath.EndsWith(".ni.exe", StringComparison.OrdinalIgnoreCase))
+                {
+                    simpleName = outPath[iSimpleNameStart..^7];
+                    isNativeImage = true;
+                }
+                else if (outPath.EndsWith(".dll", StringComparison.OrdinalIgnoreCase)
+                    || outPath.EndsWith(".exe", StringComparison.OrdinalIgnoreCase))
+                {
+                    simpleName = outPath[iSimpleNameStart..^4];
+                }
+                else
+                {
+                    // Invalid filename
+                    throw new ArgumentException(nameof(paths));
+                }
+
+                return true; ;
+            }
+        }
+
+        public void SetupBindingPaths(string trustedPlatformAssemblies, string platformResourceRoots, string appPaths, bool acquireLock)
+        {
+            if (acquireLock)
+            {
+                lock (ContextCriticalSection)
+                {
+                    Core(trustedPlatformAssemblies, platformResourceRoots, appPaths);
+                }
+            }
+            else
+            {
+                Core(trustedPlatformAssemblies, platformResourceRoots, appPaths);
+            }
+
+            void Core(string trustedPlatformAssemblies, string platformResourceRoots, string appPaths)
+            {
+                if (TrustedPlatformAssemblyMap != null)
+                {
+                    return;
+                }
+
+                //
+                // Parse TrustedPlatformAssemblies
+                //
+
+                TrustedPlatformAssemblyMap = new Dictionary<string, TPAEntry>(StringComparer.OrdinalIgnoreCase);
+                for (int i = 0; i < trustedPlatformAssemblies.Length;)
+                {
+                    if (!GetNextTPAPath(trustedPlatformAssemblies, ref i, dllOnly: false, out string fileName, out string simpleName, out bool isNativeImage))
+                    {
+                        break;
+                    }
+
+                    if (TrustedPlatformAssemblyMap.TryGetValue(simpleName, out TPAEntry? existingEntry))
+                    {
+                        //
+                        // We want to store only the first entry matching a simple name we encounter.
+                        // The exception is if we first store an IL reference and later in the string
+                        // we encounter a native image.  Since we don't touch IL in the presence of
+                        // native images, we replace the IL entry with the NI.
+                        //
+                        if ((existingEntry.ILFileName != null && !isNativeImage) ||
+                            (existingEntry.NIFileName != null && isNativeImage))
+                        {
+                            continue;
+                        }
+                    }
+
+                    if (isNativeImage)
+                    {
+                        existingEntry ??= new TPAEntry();
+                        existingEntry.NIFileName = fileName;
+                    }
+                    else
+                    {
+                        existingEntry ??= new TPAEntry();
+                        existingEntry.ILFileName = fileName;
+                    }
+
+                    TrustedPlatformAssemblyMap[simpleName] = existingEntry;
+                }
+
+                //
+                // Parse PlatformResourceRoots
+                //
+
+                for (int i = 0; i < platformResourceRoots.Length;)
+                {
+                    if (!GetNextPath(platformResourceRoots, ref i, out string pathName))
+                    {
+                        break;
+                    }
+
+                    if (!Path.IsPathFullyQualified(pathName))
+                    {
+                        throw new ArgumentException(nameof(pathName));
+                    }
+
+                    PlatformResourceRoots.Add(pathName);
+                }
+
+                //
+                // Parse AppPaths
+                //
+
+                for (int i = 0; i < appPaths.Length;)
+                {
+                    if (!GetNextPath(appPaths, ref i, out string pathName))
+                    {
+                        break;
+                    }
+
+
+                    if (!Path.IsPathFullyQualified(pathName))
+                    {
+                        throw new ArgumentException(nameof(pathName));
+                    }
+
+                    AppPaths.Add(pathName);
+                }
+            }
+        }
+
+        public void AddToFailureCache(BinderAssemblyName assemblyName, int hresult)
+        {
+            FailureCache.Add(new FailureCacheKey(assemblyName.SimpleName, assemblyName.Version), hresult);
+            IncrementVersion();
+        }
+    }
+}