mirror of
https://github.com/SabreTools/SabreTools.Serialization.git
synced 2026-09-22 23:05:12 +00:00
Initial migration of ExtractionTool
This commit is contained in:
Matt Nadareski
424
ExtractionTool/Program.cs
Normal file
424
ExtractionTool/Program.cs
Normal file
@@ -0,0 +1,424 @@
|
||||
using System;
|
||||
using System.IO;
|
||||
using SabreTools.IO.Extensions;
|
||||
using SabreTools.Serialization.Wrappers;
|
||||
|
||||
namespace ExtractionTool
|
||||
{
|
||||
class Program
|
||||
{
|
||||
static void Main(string[] args)
|
||||
{
|
||||
#if NET462_OR_GREATER || NETCOREAPP
|
||||
// Register the codepages
|
||||
System.Text.Encoding.RegisterProvider(System.Text.CodePagesEncodingProvider.Instance);
|
||||
#endif
|
||||
|
||||
// Get the options from the arguments
|
||||
var options = Options.ParseOptions(args);
|
||||
|
||||
// If we have an invalid state
|
||||
if (options == null)
|
||||
{
|
||||
Options.DisplayHelp();
|
||||
return;
|
||||
}
|
||||
|
||||
// Loop through the input paths
|
||||
foreach (string inputPath in options.InputPaths)
|
||||
{
|
||||
ExtractPath(inputPath, options.OutputPath, options.Debug);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Wrapper to extract data for a single path
|
||||
/// </summary>
|
||||
/// <param name="path">File or directory path</param>
|
||||
/// <param name="outputDirectory">Output directory path</param>
|
||||
/// <param name="includeDebug">Enable including debug information</param>
|
||||
private static void ExtractPath(string path, string outputDirectory, bool includeDebug)
|
||||
{
|
||||
// Normalize by getting the full path
|
||||
path = Path.GetFullPath(path);
|
||||
Console.WriteLine($"Checking possible path: {path}");
|
||||
|
||||
// Check if the file or directory exists
|
||||
if (File.Exists(path))
|
||||
{
|
||||
ExtractFile(path, outputDirectory, includeDebug);
|
||||
}
|
||||
else if (Directory.Exists(path))
|
||||
{
|
||||
foreach (string file in IOExtensions.SafeEnumerateFiles(path, "*", SearchOption.AllDirectories))
|
||||
{
|
||||
ExtractFile(file, outputDirectory, includeDebug);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Console.WriteLine($"{path} does not exist, skipping...");
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Print information for a single file, if possible
|
||||
/// </summary>
|
||||
private static void ExtractFile(string file, string outputDirectory, bool includeDebug)
|
||||
{
|
||||
Console.WriteLine($"Attempting to extract all files from {file}");
|
||||
using Stream stream = File.Open(file, FileMode.Open, FileAccess.Read, FileShare.ReadWrite);
|
||||
|
||||
// Get the extension for certain checks
|
||||
string extension = Path.GetExtension(file).ToLower().TrimStart('.');
|
||||
|
||||
// Get the first 16 bytes for matching
|
||||
byte[] magic = new byte[16];
|
||||
try
|
||||
{
|
||||
int read = stream.Read(magic, 0, 16);
|
||||
stream.Seek(0, SeekOrigin.Begin);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
if (includeDebug) Console.Error.WriteLine(ex);
|
||||
return;
|
||||
}
|
||||
|
||||
// TODO: When extractable wrapper types are exposed to this, use them instead of guessing
|
||||
|
||||
// Get the file type
|
||||
WrapperType ft = WrapperFactory.GetFileType(magic, extension);
|
||||
var wrapper = WrapperFactory.CreateWrapper(ft, stream);
|
||||
|
||||
// Create the output directory
|
||||
Directory.CreateDirectory(outputDirectory);
|
||||
|
||||
// 7-zip
|
||||
if (wrapper is SevenZip sz)
|
||||
{
|
||||
Console.WriteLine("Extracting 7-zip contents");
|
||||
Console.WriteLine();
|
||||
|
||||
#if NET20 || NET35 || NET40 || NET452
|
||||
Console.WriteLine("Extraction is not supported for this framework!");
|
||||
Console.WriteLine();
|
||||
#else
|
||||
sz.Extract(outputDirectory, includeDebug);
|
||||
#endif
|
||||
}
|
||||
|
||||
// BFPK archive
|
||||
else if (wrapper is BFPK bfpk)
|
||||
{
|
||||
Console.WriteLine("Extracting BFPK contents");
|
||||
Console.WriteLine();
|
||||
|
||||
bfpk.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// BSP
|
||||
else if (wrapper is BSP bsp)
|
||||
{
|
||||
Console.WriteLine("Extracting BSP contents");
|
||||
Console.WriteLine();
|
||||
|
||||
bsp.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// bzip2
|
||||
else if (wrapper is BZip2 bzip2)
|
||||
{
|
||||
Console.WriteLine("Extracting bzip2 contents");
|
||||
Console.WriteLine();
|
||||
|
||||
bzip2.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// CFB
|
||||
else if (wrapper is CFB cfb)
|
||||
{
|
||||
// Build the installer information
|
||||
Console.WriteLine("Extracting CFB contents");
|
||||
Console.WriteLine();
|
||||
|
||||
#if NET20 || NET35
|
||||
Console.WriteLine("Extraction is not supported for this framework!");
|
||||
Console.WriteLine();
|
||||
#else
|
||||
cfb.Extract(outputDirectory, includeDebug);
|
||||
#endif
|
||||
}
|
||||
|
||||
// Executable
|
||||
else if (ft == WrapperType.Executable)
|
||||
{
|
||||
Console.WriteLine("Extracting executable contents");
|
||||
Console.WriteLine();
|
||||
|
||||
var exe = WrapperFactory.CreateExecutableWrapper(stream);
|
||||
if (exe == null)
|
||||
return;
|
||||
|
||||
// New Executable
|
||||
if (exe is NewExecutable nex)
|
||||
{
|
||||
// Wise Installer -- Reimplement
|
||||
// var wi = new WiseInstaller();
|
||||
// if (wi.CheckExecutable(file, nex, includeDebug) != null)
|
||||
// wi.Extract(file, nex, outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// Portable Executable
|
||||
else if (exe is PortableExecutable pex)
|
||||
{
|
||||
// CExe, Embedded Archives, Embedded Executables
|
||||
pex.Extract(outputDirectory, includeDebug);
|
||||
|
||||
// 7-zip SFX -- Reimplement
|
||||
// var szsfx = new SevenZipSFX();
|
||||
// if (szsfx.CheckExecutable(file, pex, includeDebug) != null)
|
||||
// szsfx.Extract(file, pex, outputDirectory, includeDebug);
|
||||
|
||||
// WinRAR SFX -- Reimplement
|
||||
// var wrsfx = new WinRARSFX();
|
||||
// if (wrsfx.CheckExecutable(file, pex, includeDebug) != null)
|
||||
// wrsfx.Extract(file, pex, outputDirectory, includeDebug);
|
||||
|
||||
// WinZip SFX -- Reimplement
|
||||
// var wzsfx = new WinZipSFX();
|
||||
// if (wzsfx.CheckExecutable(file, pex, includeDebug) != null)
|
||||
// wzsfx.Extract(file, pex, outputDirectory, includeDebug);
|
||||
|
||||
// Wise Installer -- Reimplement
|
||||
// var wi = new WiseInstaller();
|
||||
// if (wi.CheckExecutable(file, pex, includeDebug) != null)
|
||||
// wi.Extract(file, pex, outputDirectory, includeDebug);
|
||||
}
|
||||
}
|
||||
|
||||
// GCF
|
||||
else if (wrapper is GCF gcf)
|
||||
{
|
||||
Console.WriteLine("Extracting GCF contents");
|
||||
Console.WriteLine();
|
||||
|
||||
gcf.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// gzip
|
||||
else if (wrapper is GZip gzip)
|
||||
{
|
||||
Console.WriteLine("Extracting gzip contents");
|
||||
Console.WriteLine();
|
||||
|
||||
gzip.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// InstallShield Archive V3 (Z)
|
||||
else if (wrapper is InstallShieldArchiveV3 isv3)
|
||||
{
|
||||
Console.WriteLine("Extracting InstallShield Archive V3 contents");
|
||||
Console.WriteLine();
|
||||
|
||||
isv3.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// IS-CAB archive -- Reimplement
|
||||
// else if (wrapper is InstallShieldCabinet iscab)
|
||||
// {
|
||||
// Console.WriteLine("Extracting IS-CAB contents");
|
||||
// Console.WriteLine();
|
||||
|
||||
// iscab.Extract(outputDirectory, includeDebug);
|
||||
// }
|
||||
|
||||
// LZ-compressed file, KWAJ variant
|
||||
else if (wrapper is LZKWAJ kwaj)
|
||||
{
|
||||
Console.WriteLine("Extracting LZ-compressed file, KWAJ variant contents");
|
||||
Console.WriteLine();
|
||||
|
||||
kwaj.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// LZ-compressed file, QBasic variant
|
||||
else if (wrapper is LZQBasic qbasic)
|
||||
{
|
||||
Console.WriteLine("Extracting LZ-compressed file, QBasic variant contents");
|
||||
Console.WriteLine();
|
||||
|
||||
qbasic.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// LZ-compressed file, SZDD variant
|
||||
else if (wrapper is LZSZDD szdd)
|
||||
{
|
||||
Console.WriteLine("Extracting LZ-compressed file, SZDD variant contents");
|
||||
Console.WriteLine();
|
||||
|
||||
szdd.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// Microsoft Cabinet archive
|
||||
else if (wrapper is MicrosoftCabinet mscab)
|
||||
{
|
||||
Console.WriteLine("Extracting MS-CAB contents");
|
||||
Console.WriteLine("WARNING: LZX and Quantum compression schemes are not supported so some files may be skipped!");
|
||||
Console.WriteLine();
|
||||
|
||||
MicrosoftCabinet.ExtractSet(file, outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// MoPaQ (MPQ) archive -- Reimplement
|
||||
// else if (wrapper is MoPaQ mpq)
|
||||
// {
|
||||
// Console.WriteLine("Extracting MoPaQ contents");
|
||||
// Console.WriteLine();
|
||||
|
||||
#if NET20 || NET35 || !(WINX86 || WINX64)
|
||||
// Console.WriteLine("Extraction is not supported for this framework!");
|
||||
// Console.WriteLine();
|
||||
#else
|
||||
// mpq.Extract(stream, file, outputDirectory, includeDebug: true);
|
||||
#endif
|
||||
// }
|
||||
|
||||
// PAK
|
||||
else if (wrapper is PAK pak)
|
||||
{
|
||||
Console.WriteLine("Extracting PAK contents");
|
||||
Console.WriteLine();
|
||||
|
||||
pak.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// PFF
|
||||
else if (wrapper is PFF pff)
|
||||
{
|
||||
Console.WriteLine("Extracting PFF contents");
|
||||
Console.WriteLine();
|
||||
|
||||
pff.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// PKZIP
|
||||
else if (wrapper is PKZIP pkzip)
|
||||
{
|
||||
Console.WriteLine("Extracting PKZIP contents");
|
||||
Console.WriteLine();
|
||||
|
||||
#if NET20 || NET35 || NET40 || NET452
|
||||
Console.WriteLine("Extraction is not supported for this framework!");
|
||||
Console.WriteLine();
|
||||
#else
|
||||
pkzip.Extract(outputDirectory, includeDebug);
|
||||
#endif
|
||||
}
|
||||
|
||||
// Quantum
|
||||
else if (wrapper is Quantum quantum)
|
||||
{
|
||||
Console.WriteLine("Extracting Quantum contents");
|
||||
Console.WriteLine();
|
||||
|
||||
quantum.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// RAR
|
||||
else if (wrapper is RAR rar)
|
||||
{
|
||||
Console.WriteLine("Extracting RAR contents");
|
||||
Console.WriteLine();
|
||||
|
||||
#if NET20 || NET35 || NET40 || NET452
|
||||
Console.WriteLine("Extraction is not supported for this framework!");
|
||||
Console.WriteLine();
|
||||
#else
|
||||
rar.Extract(outputDirectory, includeDebug);
|
||||
#endif
|
||||
}
|
||||
|
||||
// SGA
|
||||
else if (wrapper is SGA sga)
|
||||
{
|
||||
Console.WriteLine("Extracting SGA contents");
|
||||
Console.WriteLine();
|
||||
|
||||
sga.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// Tape Archive
|
||||
else if (wrapper is TapeArchive tar)
|
||||
{
|
||||
Console.WriteLine("Extracting Tape Archive contents");
|
||||
Console.WriteLine();
|
||||
|
||||
#if NET20 || NET35 || NET40 || NET452
|
||||
Console.WriteLine("Extraction is not supported for this framework!");
|
||||
Console.WriteLine();
|
||||
#else
|
||||
tar.Extract(outputDirectory, includeDebug);
|
||||
#endif
|
||||
}
|
||||
|
||||
// VBSP
|
||||
else if (wrapper is VBSP vbsp)
|
||||
{
|
||||
Console.WriteLine("Extracting VBSP contents");
|
||||
Console.WriteLine();
|
||||
|
||||
vbsp.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// VPK
|
||||
else if (wrapper is VPK vpk)
|
||||
{
|
||||
Console.WriteLine("Extracting VPK contents");
|
||||
Console.WriteLine();
|
||||
|
||||
vpk.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// WAD3
|
||||
else if (wrapper is WAD3 wad)
|
||||
{
|
||||
Console.WriteLine("Extracting WAD3 contents");
|
||||
Console.WriteLine();
|
||||
|
||||
wad.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// xz
|
||||
else if (wrapper is XZ xz)
|
||||
{
|
||||
Console.WriteLine("Extracting xz contents");
|
||||
Console.WriteLine();
|
||||
|
||||
#if NET20 || NET35 || NET40 || NET452
|
||||
Console.WriteLine("Extraction is not supported for this framework!");
|
||||
Console.WriteLine();
|
||||
#else
|
||||
xz.Extract(outputDirectory, includeDebug);
|
||||
#endif
|
||||
}
|
||||
|
||||
// XZP
|
||||
else if (wrapper is XZP xzp)
|
||||
{
|
||||
Console.WriteLine("Extracting XZP contents");
|
||||
Console.WriteLine();
|
||||
|
||||
xzp.Extract(outputDirectory, includeDebug);
|
||||
}
|
||||
|
||||
// Everything else
|
||||
else
|
||||
{
|
||||
Console.WriteLine("Not a supported extractable file format, skipping...");
|
||||
Console.WriteLine();
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user