windows/QuickLook
1
0
Fork 0
mirror of https://github.com/QL-Win/QuickLook.git synced 2025-09-11 17:59:17 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Improve PE parsing with extended buffer size

Browse Source
This commit is contained in:
ema
2025-05-23 03:03:36 +08:00
parent 06e7dbaae7
commit 00ef7d1e01
1 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
QuickLook.Plugin/QuickLook.Plugin.PEViewer/PEInfoPanel.xaml.cs
View File

@@ -83,13 +83,33 @@ public partial class PEInfoPanel : UserControl
try
{
using var stream = new FileStream(path, FileMode.Open, FileAccess.Read);
using var binaryReader = new BinaryReader(stream);
var byteArray = binaryReader.ReadBytes(1024); // Fast reading
var peImage = PEImage.FromBinary(byteArray);
var machine = peImage.CoffHeader.Machine;
int maxAttempts = 3;
int bufferSize = 1024;
arch = machine.ToImageMachineName();
for (int attempt = 0; attempt < maxAttempts; attempt++)
{
try
{
using var stream = new FileStream(path, FileMode.Open, FileAccess.Read);
using var binaryReader = new BinaryReader(stream);
var byteArray = binaryReader.ReadBytes(bufferSize);
var peImage = PEImage.FromBinary(byteArray);
var machine = peImage.CoffHeader.Machine;
arch = machine.ToImageMachineName();
break; // Successfully parsed, jumped out of the loop
}
catch (Exception e) when (e.Message == "Section headers incomplete.")
{
// Extended buffer size
bufferSize *= 2;
}
catch
{
// Non-Section headers errors will not be retryed
break;
}
}
}
catch
{