Skip to content

Commit e94eb91

Browse files
Fix #3372: Fix loading a DLL that contains byte sequences matching ZIP central directory.
1 parent 15d8ed1 commit e94eb91

File tree

3 files changed

+53
-7
lines changed

3 files changed

+53
-7
lines changed

ICSharpCode.ILSpyX/FileLoaders/FileLoaderRegistry.cs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ public FileLoaderRegistry()
4242
Register(new XamarinCompressedFileLoader());
4343
Register(new WebCilFileLoader());
4444
Register(new MetadataFileLoader());
45-
Register(new BundleFileLoader());
45+
Register(new BundleFileLoader()); // bundles are PE files with a special signature, prefer over normal PE files
46+
Register(new PEFileLoader()); // prefer PE format over archives, because ZIP has no fixed header
4647
Register(new ArchiveFileLoader());
4748
}
4849
}
Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
// Copyright (c) 2024 Siegfried Pammer
2+
//
3+
// Permission is hereby granted, free of charge, to any person obtaining a copy of this
4+
// software and associated documentation files (the "Software"), to deal in the Software
5+
// without restriction, including without limitation the rights to use, copy, modify, merge,
6+
// publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons
7+
// to whom the Software is furnished to do so, subject to the following conditions:
8+
//
9+
// The above copyright notice and this permission notice shall be included in all copies or
10+
// substantial portions of the Software.
11+
//
12+
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
13+
// INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
14+
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
15+
// FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
16+
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
17+
// DEALINGS IN THE SOFTWARE.
18+
19+
using System.IO;
20+
using System.Reflection.Metadata;
21+
using System.Reflection.PortableExecutable;
22+
using System.Threading.Tasks;
23+
24+
using ICSharpCode.Decompiler.Metadata;
25+
26+
namespace ICSharpCode.ILSpyX.FileLoaders
27+
{
28+
public sealed class PEFileLoader : IFileLoader
29+
{
30+
public async Task<LoadResult?> Load(string fileName, Stream stream, FileLoadContext context)
31+
{
32+
if (stream.Length < 2 || stream.ReadByte() != 'M' || stream.ReadByte() != 'Z')
33+
{
34+
return null;
35+
}
36+
37+
return await LoadPEFile(fileName, stream, context).ConfigureAwait(false);
38+
}
39+
40+
public static Task<LoadResult> LoadPEFile(string fileName, Stream stream, FileLoadContext context)
41+
{
42+
MetadataReaderOptions options = context.ApplyWinRTProjections
43+
? MetadataReaderOptions.ApplyWindowsRuntimeProjections
44+
: MetadataReaderOptions.None;
45+
stream.Position = 0;
46+
PEFile module = new PEFile(fileName, stream, PEStreamOptions.PrefetchEntireImage | PEStreamOptions.LeaveOpen, metadataOptions: options);
47+
return Task.FromResult(new LoadResult { MetadataFile = module });
48+
}
49+
}
50+
}

ICSharpCode.ILSpyX/LoadedAssembly.cs

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -358,12 +358,7 @@ async Task<LoadResult> LoadAsync(Task<Stream?>? streamTask)
358358
stream.Position = 0;
359359
try
360360
{
361-
MetadataReaderOptions options = applyWinRTProjections
362-
? MetadataReaderOptions.ApplyWindowsRuntimeProjections
363-
: MetadataReaderOptions.None;
364-
365-
PEFile module = new PEFile(fileName, stream, PEStreamOptions.PrefetchEntireImage, metadataOptions: options);
366-
result = new LoadResult { MetadataFile = module };
361+
result = await PEFileLoader.LoadPEFile(fileName, stream, settings).ConfigureAwait(false);
367362
}
368363
catch (Exception ex)
369364
{

0 commit comments

Comments
 (0)