You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
apphost-extract/src/apphost-extract/apphost-extract-v2/Models/General/IApphostFile.cs

57 lines
1.7 KiB

using System;
using System.Collections.Generic;
using System.IO;
using System.Reflection.PortableExecutable;
using System.Text;
using System.Threading.Tasks;
namespace apphost_extract_v2.General
{
public abstract class IApphostFile
{
internal FileStream FileStream;
public PEHeaders PEHeader;
public AppHostFileHeader Header;
public IApphostFile(FileStream fs, PEHeaders peheader)
{
FileStream = fs;
PEHeader = peheader;
}
public void ExtractAll(string outputDir)
{
Directory.CreateDirectory(outputDir);
//foreach(var fileEntry in Header.Manifest.FileEntries)
Parallel.ForEach(Header.Manifest.FileEntries, fileEntry =>
{
try
{
var bytes = FileStream.ReadBuffer(fileEntry.Offset, fileEntry.Size);
var name = fileEntry.Name;
if (FileChecker.IsKnownFile(bytes))
{
Log.Info($"Extracting {name} --> Known file", ConsoleColor.Green);
}
else
{
Log.Info($"Extracting {name} --> Unknown file", ConsoleColor.Yellow);
name = name.Insert(0, "_");
}
var filePath = Path.Combine(outputDir, name);
File.WriteAllBytes(filePath, bytes);
}
catch (Exception ex)
{
Log.Error($"Could not extract {fileEntry.Name}: {ex.Message}");
}
});
Console.WriteLine();
}
public abstract void Close();
}
}