|
| 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 | +} |
0 commit comments