windows/QuickLook
1
0
Fork 0
mirror of https://github.com/QL-Win/QuickLook.git synced 2025-09-12 18:39:45 +00:00
Code Issues Actions 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

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

@@ -81,15 +81,35 @@ public partial class PEInfoPanel : UserControl
var size = new FileInfo(path).Length; var size = new FileInfo(path).Length;
var arch = default(string); var arch = default(string);
try
{
int maxAttempts = 3;
int bufferSize = 1024;
for (int attempt = 0; attempt < maxAttempts; attempt++)
{
try try
{ {
using var stream = new FileStream(path, FileMode.Open, FileAccess.Read); using var stream = new FileStream(path, FileMode.Open, FileAccess.Read);
using var binaryReader = new BinaryReader(stream); using var binaryReader = new BinaryReader(stream);
var byteArray = binaryReader.ReadBytes(1024); // Fast reading var byteArray = binaryReader.ReadBytes(bufferSize);
var peImage = PEImage.FromBinary(byteArray); var peImage = PEImage.FromBinary(byteArray);
var machine = peImage.CoffHeader.Machine; var machine = peImage.CoffHeader.Machine;
arch = machine.ToImageMachineName(); 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 catch
{ {