internal static RegDiff ReadFromHive(RegistryHive Input)
{
var Out = new RegDiff();
var Root = Input.Root;
foreach (var name in Root.GetValueNames())
{
Out.Data[String.Empty][name] = new ValueObject(Root.GetValueType(name), Root.GetValue(name));
}
foreach (var sub in Root.GetSubKeyNames())
{
InnerRead(Root.OpenSubKey(sub), Out.Data);
}
return Out;
}
internal static RegDiff ReadFromHive(RegistryHive Input)
{
var Out = new RegDiff();
var Root = Input.Root;
foreach (var name in Root.GetValueNames())
{
Out.Data[String.Empty][name] = new ValueObject(Root.GetValueType(name), Root.GetValue(name));
}
foreach (var sub in Root.GetSubKeyNames())
{
InnerRead(Root.OpenSubKey(sub), Out.Data);
}
return(Out);
}
private static void DiffPart(DiscFileSystem PartA, DiscFileSystem PartB, DiscFileSystem Output, ComparisonStyle Style = ComparisonStyle.DateTimeOnly, CopyQueue WriteQueue = null)
{
if (PartA == null)
{
throw new ArgumentNullException("PartA");
}
if (PartB == null)
{
throw new ArgumentNullException("PartB");
}
if (Output == null)
{
throw new ArgumentNullException("Output");
}
if (PartA is NtfsFileSystem)
{
((NtfsFileSystem)PartA).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem)PartA).NtfsOptions.HideSystemFiles = false;
}
if (PartB is NtfsFileSystem)
{
((NtfsFileSystem)PartB).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem)PartB).NtfsOptions.HideSystemFiles = false;
}
if (Output is NtfsFileSystem)
{
((NtfsFileSystem)Output).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem)Output).NtfsOptions.HideSystemFiles = false;
}
if (WriteQueue == null)
{
WriteQueue = new CopyQueue();
}
var RootA = PartA.Root;
var RootB = PartB.Root;
var OutRoot = Output.Root;
var OutFileRoot = Output.GetDirectoryInfo(RootFiles);
if (!OutFileRoot.Exists)
{
OutFileRoot.Create();
}
CompareTree(RootA, RootB, OutFileRoot, WriteQueue, Style);
WriteQueue.Go();
// Now handle registry files (if any)
ParallelQuery <DiscFileInfo> Ofiles;
lock (OutFileRoot.FileSystem)
Ofiles = OutFileRoot.GetFiles("*.*", SearchOption.AllDirectories).AsParallel();
Ofiles = Ofiles.Where(dfi =>
SystemRegistryFiles.Contains(dfi.FullName, StringComparer.CurrentCultureIgnoreCase));
foreach (var file in Ofiles)
{
var A = PartA.GetFileInfo(file.FullName.Substring(RootFiles.Length + 1));
if (!A.Exists)
{
file.FileSystem.MoveFile(file.FullName, String.Concat(RootSystemRegistry, A.FullName));
continue;
}
//else
MemoryStream SideA = new MemoryStream();
using (var tmp = A.OpenRead()) tmp.CopyTo(SideA);
MemoryStream SideB = new MemoryStream();
using (var tmp = file.OpenRead()) tmp.CopyTo(SideB);
var comp = new RegistryComparison(SideA, SideB, RegistryComparison.Side.B);
comp.DoCompare();
var diff = new RegDiff(comp, RegistryComparison.Side.B);
var outFile = Output.GetFileInfo(Path.Combine(RootSystemRegistry, file.FullName));
if (!outFile.Directory.Exists)
{
outFile.Directory.Create();
}
using (var OUT = outFile.Open(outFile.Exists ? FileMode.Truncate : FileMode.CreateNew, FileAccess.ReadWrite))
diff.WriteToStream(OUT);
file.Delete(); // remove this file from the set of file to copy and overwrite
}
lock (OutFileRoot.FileSystem)
Ofiles = OutFileRoot.GetFiles("*.*", SearchOption.AllDirectories).AsParallel();
Ofiles = Ofiles.Where(dfi => UserRegisrtyFiles.IsMatch(dfi.FullName));
foreach (var file in Ofiles)
{
var match = UserRegisrtyFiles.Match(file.FullName);
var A = PartA.GetFileInfo(file.FullName.Substring(RootFiles.Length + 1));
if (!A.Exists)
{
file.FileSystem.MoveFile(file.FullName,
Path.Combine(RootUserRegistry, match.Groups["user"].Value, A.Name));
continue;
}
//else
MemoryStream SideA = new MemoryStream();
using (var tmp = A.OpenRead()) tmp.CopyTo(SideA);
MemoryStream SideB = new MemoryStream();
using (var tmp = file.OpenRead()) tmp.CopyTo(SideB);
var comp = new RegistryComparison(SideA, SideB, RegistryComparison.Side.B);
comp.DoCompare();
var diff = new RegDiff(comp, RegistryComparison.Side.B);
var outFile =
Output.GetFileInfo(Path.Combine(RootUserRegistry, match.Groups["user"].Value, file.FullName));
if (!outFile.Directory.Exists)
{
outFile.Directory.Create();
}
using (var OUT = outFile.Open(outFile.Exists ? FileMode.Truncate : FileMode.CreateNew, FileAccess.ReadWrite))
diff.WriteToStream(OUT);
file.Delete(); // remove this file from the set of file to copy and overwrite
}
}
private static void ApplyPartDiff(PartitionInfo Base, PartitionInfo Diff)
{
CopyQueue queue = new CopyQueue();
var BFS = DetectFileSystem(Base);
var DFS = DetectFileSystem(Diff);
if (BFS is NtfsFileSystem)
{
((NtfsFileSystem)BFS).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem)BFS).NtfsOptions.HideSystemFiles = false;
}
if (DFS is NtfsFileSystem)
{
((NtfsFileSystem)DFS).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem)DFS).NtfsOptions.HideSystemFiles = false;
}
var DRoot = DFS.Root;
var DFRoots = DRoot.GetDirectories(RootFiles);
if (DFRoots.Any())
{
var DFileRoot = DFRoots.Single();
foreach (var file in DFileRoot.GetFiles("*.*", SearchOption.AllDirectories))
{
var BFile = BFS.GetFileInfo(file.FullName.Substring(RootFiles.Length + 1));
queue.Add(file, BFile);
/*
* Stream OutStream;
* if (BFile.Exists) OutStream = BFile.Open(FileMode.Truncate, FileAccess.ReadWrite);
* else
* {
* if (!BFile.Directory.Exists)
* BFile.Directory.Create();
* OutStream = BFile.Create();
* }
* using (var InStream = file.OpenRead())
* using (OutStream)
* InStream.CopyTo(OutStream);
*/
}
queue.Go();
}
var DsysRegs = DRoot.GetDirectories(RootSystemRegistry);
if (DsysRegs.Any())
{
var DsysReg = DsysRegs.Single();
foreach (var file in DsysReg.GetFiles("*.*", SearchOption.AllDirectories))
{
var BReg = BFS.GetFileInfo(file.FullName.Substring(RootSystemRegistry.Length + 1));
if (!BReg.Exists)
{
queue.Add(file, BReg);
}
else
{
var BHive = new RegistryHive(BReg.Open(FileMode.Open, FileAccess.ReadWrite));
RegDiff.ReadFromStream(file.OpenRead()).ApplyTo(BHive.Root);
}
}
queue.Go();
}
var DuserRegs = DRoot.GetDirectories(RootUserRegistry);
if (DuserRegs.Any())
{
var DuserReg = DuserRegs.Single();
var Bfiles =
BFS.GetFiles(String.Empty, "*.*", SearchOption.AllDirectories)
.Where(str => UserRegisrtyFiles.IsMatch(str))
.ToArray();
foreach (var file in DuserReg.GetFiles("*.*", SearchOption.AllDirectories))
{
var username = DiffUserRegistry.Match(file.FullName).Groups["user"].Value;
var userFile = Bfiles.Where(str => GetUserRegex(username).IsMatch(str)).ToArray();
if (!userFile.Any())
{
continue;
}
var BReg = BFS.GetFileInfo(userFile.Single());
if (!BReg.Exists)
{
continue;
}
var BHive = new RegistryHive(BReg.Open(FileMode.Open, FileAccess.ReadWrite));
RegDiff.ReadFromStream(file.OpenRead()).ApplyTo(BHive.Root);
}
}
}
private static void DiffPart(DiscFileSystem PartA, DiscFileSystem PartB, DiscFileSystem Output, ComparisonStyle Style = ComparisonStyle.DateTimeOnly, CopyQueue WriteQueue = null)
{
if (PartA == null) throw new ArgumentNullException("PartA");
if (PartB == null) throw new ArgumentNullException("PartB");
if (Output == null) throw new ArgumentNullException("Output");
if (PartA is NtfsFileSystem)
{
((NtfsFileSystem) PartA).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem) PartA).NtfsOptions.HideSystemFiles = false;
}
if (PartB is NtfsFileSystem)
{
((NtfsFileSystem) PartB).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem) PartB).NtfsOptions.HideSystemFiles = false;
}
if (Output is NtfsFileSystem)
{
((NtfsFileSystem) Output).NtfsOptions.HideHiddenFiles = false;
((NtfsFileSystem) Output).NtfsOptions.HideSystemFiles = false;
}
if (WriteQueue == null) WriteQueue = new CopyQueue();
var RootA = PartA.Root;
var RootB = PartB.Root;
var OutRoot = Output.Root;
var OutFileRoot = Output.GetDirectoryInfo(RootFiles);
if (!OutFileRoot.Exists) OutFileRoot.Create();
CompareTree(RootA, RootB, OutFileRoot, WriteQueue, Style);
WriteQueue.Go();
// Now handle registry files (if any)
ParallelQuery<DiscFileInfo> Ofiles;
lock(OutFileRoot.FileSystem)
Ofiles = OutFileRoot.GetFiles("*.*", SearchOption.AllDirectories).AsParallel();
Ofiles = Ofiles.Where(dfi =>
SystemRegistryFiles.Contains(dfi.FullName, StringComparer.CurrentCultureIgnoreCase));
foreach (var file in Ofiles)
{
var A = PartA.GetFileInfo(file.FullName.Substring(RootFiles.Length + 1));
if (!A.Exists)
{
file.FileSystem.MoveFile(file.FullName, String.Concat(RootSystemRegistry, A.FullName));
continue;
}
//else
MemoryStream SideA = new MemoryStream();
using (var tmp = A.OpenRead()) tmp.CopyTo(SideA);
MemoryStream SideB = new MemoryStream();
using (var tmp = file.OpenRead()) tmp.CopyTo(SideB);
var comp = new RegistryComparison(SideA, SideB, RegistryComparison.Side.B);
comp.DoCompare();
var diff = new RegDiff(comp, RegistryComparison.Side.B);
var outFile = Output.GetFileInfo(Path.Combine(RootSystemRegistry, file.FullName));
if (!outFile.Directory.Exists)
{
outFile.Directory.Create();
}
using (var OUT = outFile.Open(outFile.Exists ? FileMode.Truncate : FileMode.CreateNew, FileAccess.ReadWrite))
diff.WriteToStream(OUT);
file.Delete(); // remove this file from the set of file to copy and overwrite
}
lock (OutFileRoot.FileSystem)
Ofiles = OutFileRoot.GetFiles("*.*", SearchOption.AllDirectories).AsParallel();
Ofiles = Ofiles.Where(dfi => UserRegisrtyFiles.IsMatch(dfi.FullName));
foreach (var file in Ofiles)
{
var match = UserRegisrtyFiles.Match(file.FullName);
var A = PartA.GetFileInfo(file.FullName.Substring(RootFiles.Length + 1));
if (!A.Exists)
{
file.FileSystem.MoveFile(file.FullName,
Path.Combine(RootUserRegistry, match.Groups["user"].Value, A.Name));
continue;
}
//else
MemoryStream SideA = new MemoryStream();
using (var tmp = A.OpenRead()) tmp.CopyTo(SideA);
MemoryStream SideB = new MemoryStream();
using (var tmp = file.OpenRead()) tmp.CopyTo(SideB);
var comp = new RegistryComparison(SideA, SideB, RegistryComparison.Side.B);
comp.DoCompare();
var diff = new RegDiff(comp, RegistryComparison.Side.B);
var outFile =
Output.GetFileInfo(Path.Combine(RootUserRegistry, match.Groups["user"].Value, file.FullName));
if (!outFile.Directory.Exists)
{
outFile.Directory.Create();
}
using (var OUT = outFile.Open(outFile.Exists ? FileMode.Truncate : FileMode.CreateNew, FileAccess.ReadWrite))
diff.WriteToStream(OUT);
file.Delete(); // remove this file from the set of file to copy and overwrite
}
}
