private static ReturnCode FixDir(RvFile dir, bool lastSelected, List <RvFile> fileProcessQueue, ref int totalFixed, ref int reportedFixed, Stopwatch cacheSaveTimer)
{
//Debug.WriteLine(dir.FullName);
bool thisSelected = lastSelected;
if (dir.Tree != null)
{
thisSelected = dir.Tree.Checked == RvTreeRow.TreeSelect.Selected;
}
List <RvFile> lstToProcess = new List <RvFile>();
for (int j = 0; j < dir.ChildCount; j++)
{
lstToProcess.Add(dir.Child(j));
}
foreach (RvFile child in lstToProcess)
{
ReturnCode returnCode = FixBase(child, thisSelected, fileProcessQueue, ref totalFixed, ref reportedFixed, cacheSaveTimer);
if (returnCode != ReturnCode.Good)
{
return(returnCode);
}
while (fileProcessQueue.Any())
{
returnCode = FixBase(fileProcessQueue[0], true, fileProcessQueue, ref totalFixed, ref reportedFixed, cacheSaveTimer);
if (returnCode != ReturnCode.Good)
{
return(returnCode);
}
fileProcessQueue.RemoveAt(0);
if (Report.CancellationPending())
{
break;
}
}
if (totalFixed != reportedFixed)
{
Report.ReportProgress(new bgwProgress(totalFixed));
reportedFixed = totalFixed;
}
if (Report.CancellationPending())
{
break;
}
}
// here we check to see if the directory we just scanned should be deleted
FixFileUtils.CheckDeleteFile(dir);
return(ReturnCode.Good);
}
public static void PerformFixes(ThreadWorker thWrk)
{
try
{
Stopwatch cacheSaveTimer = new Stopwatch();
cacheSaveTimer.Reset();
if (Settings.rvSettings.CacheSaveTimerEnabled)
{
cacheSaveTimer.Start();
}
if (!Report.Set(thWrk))
{
return;
}
Report.ReportProgress(new bgwText("Fixing Files"));
int totalFixes = 0;
int totalFixed = 0;
int reportedFixed = 0;
for (int i = 0; i < DB.DirRoot.ChildCount; i++)
{
RvFile tdir = DB.DirRoot.Child(i);
totalFixes += CountFixDir(tdir, tdir.Tree.Checked == RvTreeRow.TreeSelect.Selected);
}
Report.ReportProgress(new bgwSetRange(totalFixes));
List <RvFile> fileProcessQueue = new List <RvFile>();
for (int i = 0; i < DB.DirRoot.ChildCount; i++)
{
RvFile tdir = DB.DirRoot.Child(i);
ReturnCode returnCode = FixDir(tdir, tdir.Tree.Checked == RvTreeRow.TreeSelect.Selected, fileProcessQueue, ref totalFixed, ref reportedFixed, cacheSaveTimer);
if (returnCode != ReturnCode.Good && returnCode != ReturnCode.Cancel)
{
RepairStatus.ReportStatusReset(DB.DirRoot);
break;
}
if (Report.CancellationPending())
{
break;
}
}
Report.ReportProgress(new bgwText("Updating Cache"));
DB.Write();
Report.ReportProgress(new bgwText("Complete"));
Report.Set(null);
}
catch (Exception exc)
{
ReportError.UnhandledExceptionHandler(exc);
Report.ReportProgress(new bgwText("Updating Cache"));
DB.Write();
Report.ReportProgress(new bgwText("Complete"));
Report.Set(null);
}
}
public static ReturnCode FixZip(RvFile fixZip, List <RvFile> fileProcessQueue, ref int totalFixed, out string errorMessage)
{
errorMessage = "";
//Check for error status
if (fixZip.DirStatus.HasUnknown())
{
return(ReturnCode.FindFixes); // Error
}
bool needsTrrntzipped = fixZip.ZipStatus != ZipStatus.TrrntZip && fixZip.GotStatus == GotStatus.Got && fixZip.DatStatus == DatStatus.InDatCollect && Settings.rvSettings.ConvertToTrrntzip;
// zip files should not internally need a case fix, this code was added due to another error.
// that while scanning matched no-case matches in a zip file.
bool needsCaseFixed = false;
for (int iRom = 0; iRom < fixZip.ChildCount; iRom++)
{
needsCaseFixed |= !string.IsNullOrEmpty(fixZip.Child(iRom).FileName);
}
// file corrupt and not in tosort
// if file cannot be fully fixed copy to corrupt
// process zipfile
if (fixZip.GotStatus == GotStatus.Corrupt && fixZip.DatStatus != DatStatus.InToSort && !fixZip.DirStatus.HasFixable())
{
ReturnCode moveReturnCode = FixAZipFunctions.MoveZipToCorrupt(fixZip, out errorMessage);
if (moveReturnCode != ReturnCode.Good)
{
return(moveReturnCode);
}
}
// has fixable
// process zipfile
else if (fixZip.DirStatus.HasFixable())
{
// do nothing here but continue on to process zip.
}
// need trrntzipped
// process zipfile
else if (needsTrrntzipped)
{
// rv7Zip format is not finalized yet so do not use
if (!Settings.rvSettings.ConvertToRV7Z && (fixZip.FileType == FileType.SevenZip))
//if (fixZip.FileType == FileType.SevenZip)
{
needsTrrntzipped = false;
}
// do nothing here but continue on to process zip.
}
else if (needsCaseFixed)
{
// do nothing here but continue on to process zip.
}
// got empty zip that should be deleted
// process zipfile
else if (fixZip.GotStatus == GotStatus.Got && fixZip.GotStatus != GotStatus.Corrupt && !fixZip.DirStatus.HasAnyFiles())
{
// do nothing here but continue on to process zip.
}
// else
// skip this zipfile
else
{
// nothing can be done to return
return(ReturnCode.Good);
}
if (!fixZip.DirStatus.HasFixable() && !needsTrrntzipped && !needsCaseFixed)
{
return(ReturnCode.Good);
}
string fixZipFullName = fixZip.TreeFullName;
if (!fixZip.DirStatus.HasFixable() && needsTrrntzipped)
{
Report.ReportProgress(new bgwShowFix(Path.GetDirectoryName(fixZipFullName), Path.GetFileName(fixZipFullName), "", 0, "TrrntZipping", "", "", ""));
}
FixFileUtils.CheckCreateDirectories(fixZip.Parent);
string filename = fixZip.FullName;
if (fixZip.GotStatus == GotStatus.NotGot)
{
if (File.Exists(filename))
{
errorMessage = "Unexpected file found in directory. Rescan needed.\n" + filename;
return(ReturnCode.RescanNeeded);
}
}
ReportError.LogOut("");
ReportError.LogOut(fixZipFullName + " : " + fixZip.RepStatus);
ReportError.LogOut("------------------------------------------------------------");
Debug.WriteLine(fixZipFullName + " : " + fixZip.RepStatus);
ReportError.LogOut("Zip File Status Before Fix:");
int deleteCount = 0;
int moveToSortCount = 0;
int notThereCount = 0;
for (int intLoop = 0; intLoop < fixZip.ChildCount; intLoop++)
{
var fixZipFile = fixZip.Child(intLoop);
ReportError.LogOut(fixZipFile);
if (fixZipFile.RepStatus == RepStatus.MoveToSort || fixZipFile.RepStatus == RepStatus.MoveToCorrupt)
{
moveToSortCount++;
}
if (fixZipFile.RepStatus == RepStatus.Delete)
{
deleteCount++;
}
if (fixZipFile.RepStatus == RepStatus.Missing || fixZip.RepStatus == RepStatus.NotCollected)
{
notThereCount++;
}
}
ReportError.LogOut("");
ReturnCode returnCode = ReturnCode.Good;
RepStatus fileRepStatus = RepStatus.UnSet;
ICompress tempFixZip = null;
ICompress toSortCorruptOut = null;
ICompress toSortZipOut = null;
try
{
RvFile toSortGame = null;
RvFile toSortCorruptGame = null;
Dictionary <string, RvFile> filesUsedForFix = new Dictionary <string, RvFile>();
List <RvFile> fixZipTemp = new List <RvFile>();
FileType fixFileType = fixZip.FileType;
for (int iRom = 0; iRom < fixZip.ChildCount; iRom++)
{
RvFile fixZippedFile = new RvFile(DBTypeGet.FileFromDir(fixFileType));
fixZip.Child(iRom).CopyTo(fixZippedFile);
fixZipTemp.Add(fixZippedFile);
ReportError.LogOut(fixZippedFile.RepStatus + " : " + fixZip.Child(iRom).FullName);
fileRepStatus = fixZippedFile.RepStatus;
switch (fixZippedFile.RepStatus)
{
#region Nothing to copy
// any file we do not have or do not want in the destination zip
case RepStatus.Missing:
case RepStatus.NotCollected:
case RepStatus.Rename:
case RepStatus.Delete:
if (!
(
// got the file in the original zip but will be deleting it
fixZippedFile.DatStatus == DatStatus.NotInDat && fixZippedFile.GotStatus == GotStatus.Got ||
fixZippedFile.DatStatus == DatStatus.NotInDat && fixZippedFile.GotStatus == GotStatus.Corrupt ||
fixZippedFile.DatStatus == DatStatus.InDatMerged && fixZippedFile.GotStatus == GotStatus.Got ||
fixZippedFile.DatStatus == DatStatus.InToSort && fixZippedFile.GotStatus == GotStatus.Got ||
fixZippedFile.DatStatus == DatStatus.InToSort && fixZippedFile.GotStatus == GotStatus.Corrupt ||
// do not have this file and cannot fix it here
fixZippedFile.DatStatus == DatStatus.InDatCollect && fixZippedFile.GotStatus == GotStatus.NotGot ||
fixZippedFile.DatStatus == DatStatus.InDatBad && fixZippedFile.GotStatus == GotStatus.NotGot ||
fixZippedFile.DatStatus == DatStatus.InDatMerged && fixZippedFile.GotStatus == GotStatus.NotGot
)
)
{
ReportError.SendAndShow("Error in Fix Rom Status " + fixZippedFile.RepStatus + " : " +
fixZippedFile.DatStatus + " : " + fixZippedFile.GotStatus);
}
if (fixZippedFile.RepStatus == RepStatus.Delete)
{
if (Settings.rvSettings.DetailedFixReporting)
{
Report.ReportProgress(new bgwShowFix(Path.GetDirectoryName(fixZipFullName), Path.GetFileName(fixZipFullName), fixZippedFile.Name, fixZippedFile.Size, "Delete", "", "", ""));
}
returnCode = FixFileUtils.DoubleCheckDelete(fixZip.Child(iRom), out errorMessage);
if (returnCode != ReturnCode.Good)
{
CloseZipFile(ref tempFixZip);
CloseToSortGame(toSortGame, ref toSortZipOut);
CloseToSortCorruptGame(toSortGame, ref toSortZipOut);
return(returnCode);
}
}
fixZippedFile.GotStatus = GotStatus.NotGot;
break;
#endregion
// any files we are just moving from the original zip to the destination zip
case RepStatus.Correct:
case RepStatus.InToSort:
case RepStatus.NeededForFix:
case RepStatus.Corrupt:
{
returnCode = FixAZipFunctions.CorrectZipFile(fixZip, fixZippedFile, ref tempFixZip, iRom, fileProcessQueue, out errorMessage);
if (returnCode != ReturnCode.Good)
{
CloseZipFile(ref tempFixZip);
CloseToSortGame(toSortGame, ref toSortZipOut);
CloseToSortCorruptGame(toSortGame, ref toSortZipOut);
return(returnCode);
}
break;
}
case RepStatus.CanBeFixed:
case RepStatus.CorruptCanBeFixed:
{
returnCode = FixAZipFunctions.CanBeFixed(fixZip, fixZippedFile, ref tempFixZip, filesUsedForFix, ref totalFixed, out errorMessage);
if (returnCode != ReturnCode.Good)
{
CloseZipFile(ref tempFixZip);
CloseToSortGame(toSortGame, ref toSortZipOut);
CloseToSortCorruptGame(toSortGame, ref toSortZipOut);
return(returnCode);
}
break;
}
case RepStatus.MoveToSort:
{
returnCode = FixAZipFunctions.MovetoSort(fixZip, fixZippedFile, ref toSortGame, ref toSortZipOut, iRom);
if (returnCode != ReturnCode.Good)
{
CloseZipFile(ref tempFixZip);
CloseToSortGame(toSortGame, ref toSortZipOut);
CloseToSortCorruptGame(toSortGame, ref toSortZipOut);
return(returnCode);
}
break;
}
case RepStatus.MoveToCorrupt:
returnCode = FixAZipFunctions.MoveToCorrupt(fixZip, fixZippedFile, ref toSortCorruptGame, ref toSortCorruptOut, iRom);
if (returnCode != ReturnCode.Good)
{
CloseZipFile(ref tempFixZip);
CloseToSortGame(toSortGame, ref toSortZipOut);
CloseToSortCorruptGame(toSortGame, ref toSortZipOut);
return(returnCode);
}
break;
default:
ReportError.UnhandledExceptionHandler(
"Unknown file status found " + fixZippedFile.RepStatus + " while fixing file " +
fixZip.Name + " Dat Status = " + fixZippedFile.DatStatus + " GotStatus " +
fixZippedFile.GotStatus);
break;
}
if (Report.CancellationPending())
{
tempFixZip?.ZipFileCloseFailed();
toSortZipOut?.ZipFileCloseFailed();
toSortCorruptOut?.ZipFileCloseFailed();
tempFixZip = null;
toSortZipOut = null;
toSortCorruptOut = null;
errorMessage = "Cancel";
return(ReturnCode.Cancel);
}
}
//if ToSort Zip Made then close the zip and add this new zip to the Database
CloseToSortGame(toSortGame, ref toSortZipOut);
//if Corrupt Zip Made then close the zip and add this new zip to the Database
CloseToSortCorruptGame(toSortCorruptGame, ref toSortCorruptOut);
#region Process original Zip
if (File.Exists(filename))
{
if (!File.SetAttributes(filename, FileAttributes.Normal))
{
int error = Error.GetLastError();
Report.ReportProgress(new bgwShowError(filename,
"Error Setting File Attributes to Normal. Deleting Original Fix File. Code " + error));
}
try
{
File.Delete(filename);
}
catch (Exception e)
{
errorMessage = "Error While trying to delete file " + filename + ". " + e.Message;
if (tempFixZip != null && tempFixZip.ZipOpen != ZipOpenType.Closed)
{
tempFixZip.ZipFileClose();
tempFixZip = null;
}
return(ReturnCode.RescanNeeded);
}
}
#endregion
bool checkDelete = false;
#region process the temp Zip rename it to the original Zip
if (tempFixZip != null && tempFixZip.ZipOpen != ZipOpenType.Closed)
{
string tempFilename = tempFixZip.ZipFilename;
tempFixZip.ZipFileClose();
if (tempFixZip.LocalFilesCount() > 0)
{
// now rename the temp fix file to the correct filename
File.Move(tempFilename, filename);
FileInfo nFile = new FileInfo(filename);
RvFile tmpZip = new RvFile(FileType.Zip)
{
Name = Path.GetFileName(filename),
FileModTimeStamp = nFile.LastWriteTime
};
tmpZip.SetStatus(fixZip.DatStatus, GotStatus.Got);
fixZip.FileAdd(tmpZip, false);
fixZip.ZipStatus = tempFixZip.ZipStatus;
}
else
{
File.Delete(tempFilename);
checkDelete = true;
}
tempFixZip = null;
}
else
{
checkDelete = true;
}
#endregion
#region Now put the New Game Status information into the Database.
int intLoopFix = 0;
foreach (RvFile tmpZip in fixZipTemp)
{
tmpZip.CopyTo(fixZip.Child(intLoopFix));
if (fixZip.Child(intLoopFix).RepStatus == RepStatus.Deleted)
{
if (fixZip.Child(intLoopFix).FileRemove() == EFile.Delete)
{
fixZip.ChildRemove(intLoopFix);
continue;
}
}
intLoopFix++;
}
#endregion
List <RvFile> usedFiles = filesUsedForFix.Values.ToList();
FixFileUtils.CheckFilesUsedForFix(usedFiles, fileProcessQueue, false);
if (checkDelete)
{
FixFileUtils.CheckDeleteFile(fixZip);
}
ReportError.LogOut("");
ReportError.LogOut("Zip File Status After Fix:");
for (int intLoop = 0; intLoop < fixZip.ChildCount; intLoop++)
{
ReportError.LogOut(fixZip.Child(intLoop));
}
ReportError.LogOut("");
return(ReturnCode.Good);
}
catch (ZipFileException ex)
{
tempFixZip?.ZipFileCloseFailed();
toSortZipOut?.ZipFileCloseFailed();
toSortCorruptOut?.ZipFileCloseFailed();
tempFixZip = null;
toSortZipOut = null;
toSortCorruptOut = null;
errorMessage = ex.Message;
return(ex.returnCode);
}
catch (Exception ex)
{
tempFixZip?.ZipFileCloseFailed();
toSortZipOut?.ZipFileCloseFailed();
toSortCorruptOut?.ZipFileCloseFailed();
tempFixZip = null;
toSortZipOut = null;
toSortCorruptOut = null;
errorMessage = ex.Message;
return(ReturnCode.LogicError);
}
finally
{
if (tempFixZip != null)
{
ReportError.UnhandledExceptionHandler($"{tempFixZip.ZipFilename} tempZipOut was left open, ZipFile= {fixZipFullName} , fileRepStatus= {fileRepStatus} , returnCode= {returnCode}");
}
if (toSortZipOut != null)
{
ReportError.UnhandledExceptionHandler($"{toSortZipOut.ZipFilename} toSortZipOut was left open");
}
if (toSortCorruptOut != null)
{
ReportError.UnhandledExceptionHandler($"{toSortCorruptOut.ZipFilename} toSortCorruptOut was left open");
}
}
}