Resolve file loading issues originating from Runtime store libraries

src/OpenTelemetry.AutoInstrumentation.Loader/Loader.Net.cs

@@ -13,6 +13,8 @@ internal partial class Loader
 {
     internal static System.Runtime.Loader.AssemblyLoadContext DependencyLoadContext { get; } = new ManagedProfilerAssemblyLoadContext();
 
+    internal static string[]? StoreFiles { get; } = GetStoreFiles();
+
     private static string ResolveManagedProfilerDirectory()
     {
         string tracerFrameworkDirectory = "net";
@@ -21,6 +23,29 @@ private static string ResolveManagedProfilerDirectory()
         return Path.Combine(tracerHomeDirectory, tracerFrameworkDirectory);
     }
 
+    private static string[]? GetStoreFiles()
+    {
+        try
+        {
+            var storeDirectory = Environment.GetEnvironmentVariable("DOTNET_SHARED_STORE");
+            if (storeDirectory == null || !Directory.Exists(storeDirectory))
+            {
+                return null;
+            }
+
+            var architecture = Environment.Is64BitProcess ? "x64" : "x86";
+            var targetFramework = $"net{Environment.Version.Major}.{Environment.Version.Minor}";
+            var finalPath = Path.Combine(storeDirectory, architecture, targetFramework);
+
+            var storeFiles = Directory.GetFiles(finalPath, "Microsoft.Extensions*.dll", SearchOption.AllDirectories);
+            return storeFiles;
+        }
+        catch
+        {
+            return null;
+        }
+    }
+
     private static Assembly? AssemblyResolve_ManagedProfilerDependencies(object? sender, ResolveEventArgs args)
     {
         var assemblyName = new AssemblyName(args.Name);
@@ -56,8 +81,16 @@ private static string ResolveManagedProfilerDirectory()
             Logger.Debug("Loading {0} with DependencyLoadContext.LoadFromAssemblyPath", path);
             return DependencyLoadContext.LoadFromAssemblyPath(path); // Load unresolved framework and third-party dependencies into a custom Assembly Load Context
         }
+        else
+        {
+            var entry = StoreFiles?.FirstOrDefault(e => e.EndsWith($"{assemblyName.Name}.dll"));
+            if (entry != null)
+            {
+                return DependencyLoadContext.LoadFromAssemblyPath(entry);
+            }
 
-        return null;
+            return null;
+        }
     }
 }
 #endif