SabreTools/BinaryObjectScanner
1
0
Fork 0
mirror of https://github.com/SabreTools/BinaryObjectScanner.git synced 2026-04-25 15:50:18 +00:00
Code Issues 295 Packages Projects Releases 20 Wiki Activity

Pre-scan progress indicator

Browse Source
This commit is contained in:
Matt Nadareski
2020-10-28 13:06:45 -07:00
parent d8d6fac67e
commit cd5f9561bf
1 changed files with 37 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
BurnOutSharp/ProtectionFind.cs
View File

@@ -33,6 +33,16 @@ namespace BurnOutSharp
/// </summary>
private static IProgress<FileProtection> FileProgress = null;
/// <summary>
/// Current file index
/// </summary>
private static int Index = 0;
/// <summary>
/// Current file total
/// </summary>
private static int Total = 0;
/// <summary>
/// Scan a path to find any known copy protection(s)
/// </summary>
@@ -55,11 +65,17 @@ namespace BurnOutSharp
// If we have a file
if (File.Exists(path))
{
// Set total
Total = 1;
// Try using just the file first to get protection info
string fileProtection = ScanPath(path, false);
if (!string.IsNullOrWhiteSpace(fileProtection))
protections[path] = fileProtection;
// Checkpoint
FileProgress?.Report(new FileProtection(path, 1, "Checking file"));
// Now check to see if the file contains any additional information
string contentProtection = ScanContent(path, includePosition)?.Replace("" + (char)0x00, "");
if (!string.IsNullOrWhiteSpace(contentProtection))
@@ -71,7 +87,8 @@ namespace BurnOutSharp
}
// Checkpoint
FileProgress?.Report(new FileProtection(path, 1, contentProtection));
protections.TryGetValue(path, out string fullProtection);
FileProgress?.Report(new FileProtection(path, 1, fullProtection ?? string.Empty));
}
// If we have a directory
else if (Directory.Exists(path))
@@ -79,6 +96,9 @@ namespace BurnOutSharp
// Get the lists of files to be used
var files = Directory.EnumerateFiles(path, "*", SearchOption.AllDirectories);
// Set total
Total = files.Count();
// Try using just the path first to get protection info
string pathProtection = ScanPath(path, true);
if (!string.IsNullOrWhiteSpace(pathProtection))
@@ -87,9 +107,15 @@ namespace BurnOutSharp
// Loop through all files and scan their contents
for (int i = 0; i < files.Count(); i++)
{
// Set index
Index = i;
// Get the current file
string file = files.ElementAt(i);
// Checkpoint
FileProgress?.Report(new FileProtection(file, i / (float)files.Count(), "Checking file"));
// Try using just the file first to get protection info
string fileProtection = ScanPath(file, false);
if (!string.IsNullOrWhiteSpace(fileProtection))
@@ -106,7 +132,8 @@ namespace BurnOutSharp
}
// Checkpoint
FileProgress?.Report(new FileProtection(file, i / (float)files.Count(), contentProtection));
protections.TryGetValue(file, out string fullProtection);
FileProgress?.Report(new FileProtection(file, i / (float)files.Count(), fullProtection ?? string.Empty));
}
}
@@ -394,6 +421,10 @@ namespace BurnOutSharp
// Files can be protected in multiple ways
List<string> protections = new List<string>();
// Cache current values
int tempIndex = Index;
int tempTotal = Total;
// 7-Zip archive
if (SevenZip.ShouldScan(magic))
protections.AddRange(SevenZip.Scan(stream, includePosition));
@@ -450,6 +481,10 @@ namespace BurnOutSharp
if (XZ.ShouldScan(magic))
protections.AddRange(XZ.Scan(stream, includePosition));
// Reset values
Index = tempIndex;
Total = tempTotal;
// Return blank if nothing found, or comma-separated list of protections
if (protections.Count() == 0)
return string.Empty;