public void BuildAndApplyPatches(object sender, DoWorkEventArgs args) { BackgroundWorker worker = sender as BackgroundWorker; PatchIsoArgs patchArgs = args.Argument as PatchIsoArgs; if (patchArgs == null) { throw new Exception("Incorrect args passed to BuildAndApplyPatches"); } if (patchArgs.Patcher == null) { throw new ArgumentNullException("Patcher", "Patcher cannot be null"); } using (Stream stream = File.Open(patchArgs.Filename, FileMode.Open, FileAccess.ReadWrite)) { if (stream == null) { throw new Exception("Could not open ISO file"); } IList <ISerializableFile> files = new List <ISerializableFile>(Files.Count); Files.FindAll(f => f is ISerializableFile).ForEach(f => files.Add((ISerializableFile)f)); List <ISerializableFile> dteFiles = new List <ISerializableFile>(); List <ISerializableFile> nonDteFiles = new List <ISerializableFile>(); List <PatchedByteArray> patches = new List <PatchedByteArray>(); foreach (ISerializableFile file in files) { worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Starting, Task = ProgressForm.Task.IsDteNeeded }); if (file.IsDteNeeded()) { dteFiles.Add(file); } else { nonDteFiles.Add(file); } worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Done, Task = ProgressForm.Task.IsDteNeeded }); if (worker.CancellationPending) { args.Cancel = true; return; } } if (dteFiles.Count > 0) { Set <string> pairs = DTE.GetDteGroups(this.Filetype); if (worker.CancellationPending) { args.Cancel = true; return; } Set <KeyValuePair <string, byte> > currentPairs = new Set <KeyValuePair <string, byte> >((x, y) => x.Key.Equals(y.Key) && (x.Value == y.Value) ? 0 : -1); var filePreferredPairs = new Dictionary <ISerializableFile, Set <KeyValuePair <string, byte> > >(dteFiles.Count); dteFiles.Sort((x, y) => (y.ToCDByteArray().Length - y.Layout.Size).CompareTo(x.ToCDByteArray().Length - x.Layout.Size)); Stack <byte> dteBytes = DTE.GetAllowedDteBytes(); if (worker.CancellationPending) { args.Cancel = true; return; } foreach (var dte in dteFiles) { worker.ReportProgress(0, new ProgressForm.FileProgress { File = dte, State = ProgressForm.TaskState.Starting, Task = ProgressForm.Task.CalculateDte }); filePreferredPairs[dte] = dte.GetPreferredDTEPairs(pairs, currentPairs, dteBytes); currentPairs.AddRange(filePreferredPairs[dte]); if (filePreferredPairs[dte] == null) { throw new DTE.DteException(dte); } worker.ReportProgress(0, new ProgressForm.FileProgress { File = dte, State = ProgressForm.TaskState.Done, Task = ProgressForm.Task.CalculateDte }); if (worker.CancellationPending) { args.Cancel = true; return; } } foreach (var file in dteFiles) { worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Starting, Task = ProgressForm.Task.GeneratePatch }); var currentFileEncoding = PatcherLib.Utilities.Utilities.DictionaryFromKVPs(filePreferredPairs[file]); patches.AddRange(file.GetDtePatches(currentFileEncoding)); worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Done, Task = ProgressForm.Task.GeneratePatch }); if (worker.CancellationPending) { args.Cancel = true; return; } } patches.AddRange(DTE.GenerateDtePatches(this.Filetype, currentPairs)); } foreach (var file in nonDteFiles) { worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Starting, Task = ProgressForm.Task.GeneratePatch }); patches.AddRange(file.GetNonDtePatches()); worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Done, Task = ProgressForm.Task.GeneratePatch }); if (worker.CancellationPending) { args.Cancel = true; return; } } if (worker.CancellationPending) { args.Cancel = true; return; } worker.ReportProgress(0, new ProgressForm.FileProgress { File = null, State = ProgressForm.TaskState.Starting, Task = ProgressForm.Task.ApplyingPatches }); patchArgs.Patcher(stream, patches); worker.ReportProgress(0, new ProgressForm.FileProgress { File = null, State = ProgressForm.TaskState.Done, Task = ProgressForm.Task.ApplyingPatches }); } }
private IList <PatchedByteArray> DoDteCrap(IList <ISerializableFile> dteFiles, BackgroundWorker worker, DoWorkEventArgs args) { List <PatchedByteArray> patches = new List <PatchedByteArray>(); if (worker.CancellationPending) { args.Cancel = true; return(null); } dteFiles.Sort((x, y) => (y.ToCDByteArray().Length - y.Layout.Size).CompareTo(x.ToCDByteArray().Length - x.Layout.Size)); if (worker.CancellationPending) { args.Cancel = true; return(null); } IDictionary <ISerializableFile, Set <KeyValuePair <string, byte> > > filePreferredPairs = null; Set <KeyValuePair <string, byte> > currentPairs = null; DteResult result = DteResult.Empty; if (dteFiles.Count > 0) { int tries = dteFiles.Count; //DteResult result = DoDteForFiles( dteFiles, worker, args, out filePreferredPairs, out currentPairs ); do { result = DoDteForFiles(dteFiles, worker, args, out filePreferredPairs, out currentPairs); switch (result.ResultCode) { case DteResult.Result.Cancelled: args.Cancel = true; return(null); case DteResult.Result.Fail: var failedFile = result.FailedFile; if (dteFiles[0] == failedFile) { // Failed on the first file... this is hopeless throw new FFTPatcher.TextEditor.DTE.DteException(failedFile); } // Bump the failed file to the top of the list dteFiles.Remove(failedFile); dteFiles.Insert(0, failedFile); break; case DteResult.Result.Success: // do nothing break; } } while (result.ResultCode != DteResult.Result.Success && --tries >= 0); } switch (result.ResultCode) { case DteResult.Result.Fail: throw new FFTPatcher.TextEditor.DTE.DteException(dteFiles[0]); case DteResult.Result.Cancelled: args.Cancel = true; return(null); } foreach (var file in dteFiles) { worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Starting, Task = ProgressForm.Task.GeneratePatch }); var currentFileEncoding = PatcherLib.Utilities.Utilities.DictionaryFromKVPs(filePreferredPairs[file]); patches.AddRange(file.GetDtePatches(currentFileEncoding)); worker.ReportProgress(0, new ProgressForm.FileProgress { File = file, State = ProgressForm.TaskState.Done, Task = ProgressForm.Task.GeneratePatch }); if (worker.CancellationPending) { args.Cancel = true; return(null); } } patches.AddRange(DTE.GenerateDtePatches(this.Filetype, currentPairs)); return(patches.AsReadOnly()); }