mirror of https://github.com/icsharpcode/ILSpy.git
Siegfried Pammer
3 years ago
4 changed files with 174 additions and 99 deletions
@ -0,0 +1,138 @@
@@ -0,0 +1,138 @@
|
||||
// Copyright (c) 2011 AlphaSierraPapa for the SharpDevelop Team
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy of this
|
||||
// software and associated documentation files (the "Software"), to deal in the Software
|
||||
// without restriction, including without limitation the rights to use, copy, modify, merge,
|
||||
// publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons
|
||||
// to whom the Software is furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in all copies or
|
||||
// substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
|
||||
// INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
|
||||
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
|
||||
// FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
||||
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||||
// DEALINGS IN THE SOFTWARE.
|
||||
|
||||
#nullable enable |
||||
|
||||
using System; |
||||
using System.Collections.Generic; |
||||
using System.Collections.Immutable; |
||||
using System.Threading.Tasks; |
||||
|
||||
using ICSharpCode.Decompiler.Metadata; |
||||
using ICSharpCode.Decompiler.Util; |
||||
|
||||
namespace ICSharpCode.ILSpy |
||||
{ |
||||
class AssemblyListSnapshot |
||||
{ |
||||
readonly ImmutableArray<LoadedAssembly> assemblies; |
||||
Dictionary<string, PEFile>? asmLookupByFullName; |
||||
Dictionary<string, PEFile>? asmLookupByShortName; |
||||
|
||||
public ImmutableArray<LoadedAssembly> Assemblies => assemblies; |
||||
|
||||
public AssemblyListSnapshot(ImmutableArray<LoadedAssembly> assemblies) |
||||
{ |
||||
this.assemblies = assemblies; |
||||
} |
||||
|
||||
public async Task<PEFile?> TryGetModuleAsync(IAssemblyReference reference, string tfm) |
||||
{ |
||||
bool isWinRT = reference.IsWindowsRuntime; |
||||
string key = tfm + ";" + (isWinRT ? reference.Name : reference.FullName); |
||||
var lookup = LazyInit.VolatileRead(ref isWinRT ? ref asmLookupByShortName : ref asmLookupByFullName); |
||||
if (lookup == null) |
||||
{ |
||||
lookup = await CreateLoadedAssemblyLookupAsync(shortNames: isWinRT).ConfigureAwait(false); |
||||
lookup = LazyInit.GetOrSet(ref isWinRT ? ref asmLookupByShortName : ref asmLookupByFullName, lookup); |
||||
} |
||||
if (lookup.TryGetValue(key, out PEFile module)) |
||||
return module; |
||||
return null; |
||||
} |
||||
|
||||
private async Task<Dictionary<string, PEFile>> CreateLoadedAssemblyLookupAsync(bool shortNames) |
||||
{ |
||||
var result = new Dictionary<string, PEFile>(StringComparer.OrdinalIgnoreCase); |
||||
foreach (LoadedAssembly loaded in assemblies) |
||||
{ |
||||
try |
||||
{ |
||||
var module = await loaded.GetPEFileOrNullAsync().ConfigureAwait(false); |
||||
if (module == null) |
||||
continue; |
||||
var reader = module.Metadata; |
||||
if (reader == null || !reader.IsAssembly) |
||||
continue; |
||||
string tfm = await loaded.GetTargetFrameworkIdAsync().ConfigureAwait(false); |
||||
string key = tfm + ";" |
||||
+ (shortNames ? module.Name : module.FullName); |
||||
if (!result.ContainsKey(key)) |
||||
{ |
||||
result.Add(key, module); |
||||
} |
||||
} |
||||
catch (BadImageFormatException) |
||||
{ |
||||
continue; |
||||
} |
||||
} |
||||
return result; |
||||
} |
||||
|
||||
/// <summary>
|
||||
/// Gets all loaded assemblies recursively, including assemblies found in bundles or packages.
|
||||
/// </summary>
|
||||
public async Task<IList<LoadedAssembly>> GetAllAssembliesAsync() |
||||
{ |
||||
var results = new List<LoadedAssembly>(assemblies.Length); |
||||
|
||||
foreach (var asm in assemblies) |
||||
{ |
||||
LoadedAssembly.LoadResult result; |
||||
try |
||||
{ |
||||
result = await asm.GetLoadResultAsync().ConfigureAwait(false); |
||||
} |
||||
catch |
||||
{ |
||||
results.Add(asm); |
||||
continue; |
||||
} |
||||
if (result.Package != null) |
||||
{ |
||||
AddDescendants(result.Package.RootFolder); |
||||
} |
||||
else if (result.PEFile != null) |
||||
{ |
||||
results.Add(asm); |
||||
} |
||||
} |
||||
|
||||
void AddDescendants(PackageFolder folder) |
||||
{ |
||||
foreach (var subFolder in folder.Folders) |
||||
{ |
||||
AddDescendants(subFolder); |
||||
} |
||||
|
||||
foreach (var entry in folder.Entries) |
||||
{ |
||||
if (!entry.Name.EndsWith(".dll", StringComparison.OrdinalIgnoreCase)) |
||||
continue; |
||||
var asm = folder.ResolveFileName(entry.Name); |
||||
if (asm == null) |
||||
continue; |
||||
results.Add(asm); |
||||
} |
||||
} |
||||
|
||||
return results; |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue