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 } }