Skip to content

Commit

Permalink
Fix #3372: Fix loading a DLL that contains byte sequences matching ZI…
Browse files Browse the repository at this point in the history
…P central directory.
  • Loading branch information
siegfriedpammer committed Jan 16, 2025
1 parent 15d8ed1 commit e94eb91
Show file tree
Hide file tree
Showing 3 changed files with 53 additions and 7 deletions.
3 changes: 2 additions & 1 deletion ICSharpCode.ILSpyX/FileLoaders/FileLoaderRegistry.cs
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,8 @@ public FileLoaderRegistry()
Register(new XamarinCompressedFileLoader());
Register(new WebCilFileLoader());
Register(new MetadataFileLoader());
Register(new BundleFileLoader());
Register(new BundleFileLoader()); // bundles are PE files with a special signature, prefer over normal PE files
Register(new PEFileLoader()); // prefer PE format over archives, because ZIP has no fixed header
Register(new ArchiveFileLoader());
}
}
Expand Down
50 changes: 50 additions & 0 deletions ICSharpCode.ILSpyX/FileLoaders/PEFileLoader.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
// Copyright (c) 2024 Siegfried Pammer
//
// 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.

using System.IO;
using System.Reflection.Metadata;
using System.Reflection.PortableExecutable;
using System.Threading.Tasks;

using ICSharpCode.Decompiler.Metadata;

namespace ICSharpCode.ILSpyX.FileLoaders
{
public sealed class PEFileLoader : IFileLoader
{
public async Task<LoadResult?> Load(string fileName, Stream stream, FileLoadContext context)
{
if (stream.Length < 2 || stream.ReadByte() != 'M' || stream.ReadByte() != 'Z')
{
return null;
}

return await LoadPEFile(fileName, stream, context).ConfigureAwait(false);
}

public static Task<LoadResult> LoadPEFile(string fileName, Stream stream, FileLoadContext context)
{
MetadataReaderOptions options = context.ApplyWinRTProjections
? MetadataReaderOptions.ApplyWindowsRuntimeProjections
: MetadataReaderOptions.None;
stream.Position = 0;
PEFile module = new PEFile(fileName, stream, PEStreamOptions.PrefetchEntireImage | PEStreamOptions.LeaveOpen, metadataOptions: options);
return Task.FromResult(new LoadResult { MetadataFile = module });
}
}
}
7 changes: 1 addition & 6 deletions ICSharpCode.ILSpyX/LoadedAssembly.cs
Original file line number Diff line number Diff line change
Expand Up @@ -358,12 +358,7 @@ async Task<LoadResult> LoadAsync(Task<Stream?>? streamTask)
stream.Position = 0;
try
{
MetadataReaderOptions options = applyWinRTProjections
? MetadataReaderOptions.ApplyWindowsRuntimeProjections
: MetadataReaderOptions.None;

PEFile module = new PEFile(fileName, stream, PEStreamOptions.PrefetchEntireImage, metadataOptions: options);
result = new LoadResult { MetadataFile = module };
result = await PEFileLoader.LoadPEFile(fileName, stream, settings).ConfigureAwait(false);
}
catch (Exception ex)
{
Expand Down

0 comments on commit e94eb91

Please sign in to comment.