Fix assembly issue (#14850)

Author: msJinLei

NuGet.Config

@@ -4,7 +4,7 @@
     <clear/>
     <!-- Do not add any additional feeds if new packages are needed they need to come from nuget.org or our azure-sdk-for-net DevOps feed -->
     <add key="local-feed" value="tools/LocalFeed" />
-    <add key="myget-azure-powershell" value="https://www.myget.org/F/azure-powershell/api/v3/index.json" />
+    <add key="azure-powershell" value="https://pkgs.dev.azure.com/azure-sdk/public/_packaging/azure-powershell/nuget/v3/index.json" />
     <add key="nuget.org" value="https://api.nuget.org/v3/index.json" />
   </packageSources>
   <disabledPackageSources>

src/Accounts/Accounts/Accounts.csproj

@@ -23,8 +23,12 @@
     <PreLoadAssemblies Include="$(RepoSrc)lib\System.ServiceModel.Primitives.dll" />
     <PreLoadAssemblies Include="$(RepoSrc)lib\Microsoft.Identity.Client\NetFx\Microsoft.Identity.Client.dll" />
     <PreLoadAssemblies Include="$(RepoSrc)lib\Microsoft.Identity.Client.Extensions.Msal\NetFx\Microsoft.Identity.Client.Extensions.Msal.dll" />
+	<PreLoadAssemblies Include="$(RepoSrc)lib\NetCorePreloadAssemblies\Azure.Core.dll" />
+	<PreLoadAssemblies Include="$(RepoSrc)lib\NetCorePreloadAssemblies\Microsoft.Bcl.AsyncInterfaces.dll" />
     <NetCoreAssemblies Include="$(RepoSrc)lib\Microsoft.Identity.Client\NetCore\Microsoft.Identity.Client.dll" />
     <NetCoreAssemblies Include="$(RepoSrc)lib\Microsoft.Identity.Client.Extensions.Msal\NetCore\Microsoft.Identity.Client.Extensions.Msal.dll" />
+	<NetCoreAssemblies Include="$(RepoSrc)lib\NetFxPreloadAssemblies\Azure.Core.dll" />
+	<NetCoreAssemblies Include="$(RepoSrc)lib\NetFxPreloadAssemblies\Microsoft.Bcl.AsyncInterfaces.dll" />
     <StorageDependencies Include="$(RepoSrc)lib\WindowsAzure.Storage\9.3.0\Microsoft.WindowsAzure.Storage.dll" />
     <StorageDependencies Include="$(RepoSrc)lib\WindowsAzure.Storage\9.3.0\Microsoft.WindowsAzure.Storage.DataMovement.dll" />
   </ItemGroup>

src/lib/NetCorePreloadAssemblies/Azure.Core.dll

@@ -17,11 +17,32 @@ param(
     [System.String]$BuildConfig
 )
 
+function Get-PreloadAssemblies{
+    param(
+        [Parameter(Mandatory=$True)]
+        [string] $ModuleFolder
+    )
+
+    $preloadAssemblies = @()
+    if($PSEdition -eq 'Core') {
+        $preloadFolderName = "NetCoreAssemblies"
+    } else {
+        $preloadFolderName = "PreloadAssemblies"
+    }
+    $preloadFolder = [System.IO.Path]::Combine($ModuleFolder, $preloadFolderName)
+    if(Test-Path $preloadFolder){
+        $preloadAssemblies = (Get-ChildItem $preloadFolder -Filter "*.dll").Name | ForEach-Object { $_ -replace ".dll", ""}
+    }
+    $preloadAssemblies
+}
+
 $ProjectPaths = @( "$PSScriptRoot\..\artifacts\$BuildConfig" )
 $DependencyMapPath = "$PSScriptRoot\..\artifacts\StaticAnalysisResults\DependencyMap.csv"
 
 $DependencyMap = Import-Csv -Path $DependencyMapPath
 
+
+.($PSScriptRoot + "\PreloadToolDll.ps1")
 $ModuleManifestFiles = $ProjectPaths | ForEach-Object { Get-ChildItem -Path $_ -Filter "*.psd1" -Recurse | Where-Object { $_.FullName -like "*$($BuildConfig)*" -and `
             $_.FullName -notlike "*Netcore*" -and `
             $_.FullName -notlike "*dll-Help.psd1*" -and `
@@ -38,6 +59,7 @@ foreach ($ModuleManifest in $ModuleManifestFiles) {
         $LoadedAssemblies += $ModuleMetadata.RequiredAssemblies
     }
 
+    $LoadedAssemblies += Get-PreloadAssemblies $ModuleManifest.Directory
     $LoadedAssemblies += $ModuleMetadata.NestedModules
 
     if ($ModuleMetadata.RequiredModules) {
@@ -58,9 +80,11 @@ foreach ($ModuleManifest in $ModuleManifestFiles) {
                 }
                 $LoadedAssemblies += $ModuleMetadata.NestedModules
             }
+            $LoadedAssemblies += Get-PreloadAssemblies $RequiredModuleManifest.Directory
         }
     }
 
+    $LoadedAssemblies = $LoadedAssemblies | Where-Object { $_ }
     $LoadedAssemblies = $LoadedAssemblies | ForEach-Object { $_.Replace(".dll", "") }
 
     $Found = @()

src/lib/NetCorePreloadAssemblies/Microsoft.Bcl.AsyncInterfaces.dll

@@ -0,0 +1,12 @@
+Function Load-Dll {
+    Param(
+        [string]
+        $DllPath
+    )
+    if (([System.AppDomain]::CurrentDomain.GetAssemblies() | Where-Object { $_.Location -eq $DllPath}).Count -eq 0)
+    {
+        $null = [System.Reflection.Assembly]::LoadFrom($DllPath)
+    }
+}
+
+Load-Dll -DllPath ([System.IO.Path]::Combine($PSScriptRoot, '..', 'artifacts', 'StaticAnalysis', 'Tools.Common.dll'))