static void Main(string[] args)
{
List <string> infiles = new List <string>();
Prefilter filter = Prefilter.None;
Exhaustion exhaustion = Exhaustion.Standard;
for (int i = 0; i < args.Length; ++i)
{
if (args[i] == "--filter")
{
++i;
filter = (Prefilter)Enum.Parse(typeof(Prefilter), args[i], true);
}
else if (args[i] == "--exhaustion")
{
++i;
exhaustion = (Exhaustion)Enum.Parse(typeof(Exhaustion), args[i], true);
}
else
{
infiles.Add(args[i]);
}
}
if (infiles.Count == 0)
{
Console.WriteLine("Usage: FileCompressor [options] files");
Console.WriteLine("Options:");
Console.WriteLine(" --filter");
foreach (object e in Enum.GetValues(typeof(Prefilter)))
{
Console.WriteLine(" " + e.ToString());
}
Console.WriteLine(" --exhaustion");
foreach (object e in Enum.GetValues(typeof(Exhaustion)))
{
Console.WriteLine(" " + e.ToString());
}
}
foreach (string infile in infiles)
{
string outfile = infile + "_" + filter.ToString() + "_" + exhaustion.ToString() + ".bin";
Console.WriteLine("Generating: " + outfile);
Stream ms;
using (var fs = new FileStream(infile, FileMode.Open, FileAccess.Read, FileShare.Read)) {
ms = fs.CopyToMemory();
fs.Close();
}
using (var compressed = Compress(ms, filter, exhaustion)) {
using (FileStream os = new FileStream(outfile, FileMode.Create)) {
compressed.Position = 0;
StreamUtils.CopyStream(compressed, os);
}
}
}
}
/// <summary>
/// Returns the memory regions associated with the current snapshot. If none exist, a query will be done. Will not read any memory.
/// </summary>
/// <returns>The current active snapshot of memory in the target process.</returns>
private static Snapshot GetActiveSnapshotCreateIfNone(DataType dataType)
{
lock (SnapshotManager.AccessLock)
{
if (SnapshotManager.Snapshots.Count == 0 || SnapshotManager.Snapshots.Peek() == null || SnapshotManager.Snapshots.Peek().ElementCount == 0)
{
Snapshot snapshot = Prefilter.GetInstance().GetPrefilteredSnapshot(dataType);
snapshot.Alignment = ScanSettings.Default.Alignment;
return(snapshot);
}
// Return the snapshot
return(SnapshotManager.Snapshots.Peek());
}
}
/// <summary>
/// Returns the memory regions associated with the current snapshot. If none exist, a query will be done. Will not read any memory.
/// </summary>
/// <returns>The current active snapshot of memory in the target process.</returns>
private Snapshot GetActiveSnapshotCreateIfNone()
{
lock (this.AccessLock)
{
if (this.Snapshots.Count == 0 || this.Snapshots.Peek() == null || this.Snapshots.Peek().ElementCount == 0)
{
Snapshot snapshot = Prefilter.GetInstance().GetPrefilteredSnapshot();
snapshot.Alignment = SettingsViewModel.GetInstance().Alignment;
snapshot.ElementDataType = ScanResultsViewModel.GetInstance().ActiveType;
return(snapshot);
}
// Return the snapshot
return(this.Snapshots.Peek());
}
}
/// <summary>
/// Returns the memory regions associated with the current snapshot. If none exist, a query will be done.
/// </summary>
/// <param name="createIfNone">Creates a snapshot if none exists.</param>
/// <returns>The current active snapshot of memory in the target process.</returns>
public Snapshot GetActiveSnapshot(Boolean createIfNone = true)
{
lock (this.AccessLock)
{
// Take a snapshot if there are none, or the current one is empty
if (this.Snapshots.Count == 0 || this.Snapshots.Peek() == null || this.Snapshots.Peek().ElementCount == 0)
{
if (createIfNone)
{
Snapshot snapshot = Prefilter.GetInstance().GetPrefilteredSnapshot();
snapshot.UpdateSettings(ScanResultsViewModel.GetInstance().ActiveType, SettingsViewModel.GetInstance().Alignment);
return(snapshot);
}
else
{
return(null);
}
}
// Return the snapshot
return(this.Snapshots.Peek());
}
}
public static Stream Compress(Stream instream, Prefilter filter, Exhaustion exhaustion)
{
MemoryStream outstream = new MemoryStream((int)instream.Length);
uint crc = CRC.CalculateDigest(instream.CopyToByteArray(), 0, (uint)instream.Length);
if (filter == Prefilter.None)
{
MemoryStream memorystream = new MemoryStream((int)instream.Length);
instream.Position = 0;
StreamUtils.CopyStream(instream, memorystream);
outstream.WriteUInt8(0); // filter ID
outstream.WriteUInt32(crc, EndianUtils.Endianness.LittleEndian);
CompressInternalLzmaSemiOptimized(memorystream, outstream, exhaustion);
return(outstream);
}
if (filter == Prefilter.Delta2LE_0x30Lead)
{
if ((instream.Length % 2) != 0 || instream.Length < 0x30)
{
throw new Exception("unsupported length for " + filter.ToString());
}
MemoryStream filteredstream = new MemoryStream((int)instream.Length);
instream.Position = 0;
StreamUtils.CopyStream(instream, filteredstream, 0x30);
ushort last = 0;
while (instream.Position < instream.Length)
{
ushort curr = instream.ReadUInt16(EndianUtils.Endianness.LittleEndian);
ushort diff = (ushort)(last - curr);
filteredstream.WriteUInt16(diff, EndianUtils.Endianness.LittleEndian);
last = curr;
}
outstream.WriteUInt8(1); // filter ID
outstream.WriteUInt32(crc, EndianUtils.Endianness.LittleEndian);
filteredstream.Position = 0;
CompressInternalLzmaSemiOptimized(filteredstream, outstream, exhaustion);
return(outstream);
}
if (filter == Prefilter.Delta4LE_0x30Lead)
{
if ((instream.Length % 4) != 0 || instream.Length < 0x30)
{
throw new Exception("unsupported length for " + filter.ToString());
}
MemoryStream filteredstream = new MemoryStream((int)instream.Length);
instream.Position = 0;
StreamUtils.CopyStream(instream, filteredstream, 0x30);
uint last = 0;
while (instream.Position < instream.Length)
{
uint curr = instream.ReadUInt32(EndianUtils.Endianness.LittleEndian);
uint diff = (uint)(last - curr);
filteredstream.WriteUInt32(diff, EndianUtils.Endianness.LittleEndian);
last = curr;
}
outstream.WriteUInt8(2); // filter ID
outstream.WriteUInt32(crc, EndianUtils.Endianness.LittleEndian);
filteredstream.Position = 0;
CompressInternalLzmaSemiOptimized(filteredstream, outstream, exhaustion);
return(outstream);
}
if (filter == Prefilter.Delta2LE_Deinterleaved_0x30Lead)
{
if ((instream.Length % 4) != 0 || instream.Length < 0x30)
{
throw new Exception("unsupported length for " + filter.ToString());
}
MemoryStream filteredstream = new MemoryStream((int)instream.Length);
instream.Position = 0;
StreamUtils.CopyStream(instream, filteredstream, 0x30);
long pos = instream.Position;
ushort last = 0;
while (instream.Position < instream.Length)
{
ushort curr = instream.ReadUInt16(EndianUtils.Endianness.LittleEndian);
ushort diff = (ushort)(last - curr);
filteredstream.WriteUInt16(diff, EndianUtils.Endianness.LittleEndian);
last = curr;
instream.ReadUInt16(EndianUtils.Endianness.LittleEndian);
}
instream.Position = pos;
last = 0;
while (instream.Position < instream.Length)
{
instream.ReadUInt16(EndianUtils.Endianness.LittleEndian);
ushort curr = instream.ReadUInt16(EndianUtils.Endianness.LittleEndian);
ushort diff = (ushort)(last - curr);
filteredstream.WriteUInt16(diff, EndianUtils.Endianness.LittleEndian);
last = curr;
}
outstream.WriteUInt8(3); // filter ID
outstream.WriteUInt32(crc, EndianUtils.Endianness.LittleEndian);
filteredstream.Position = 0;
CompressInternalLzmaSemiOptimized(filteredstream, outstream, exhaustion);
return(outstream);
}
throw new Exception("unexpected filter");
}
