private string getFileNameForChecksum(string pFileName, string pHashKey)
{
string checksumKey;
string newFileName;
string ret;
using (FileStream fs = File.OpenRead(pFileName))
{
checksumKey = String.Format("{0}_{1}_{2}", ChecksumUtil.GetCrc32OfFullFile(fs), ChecksumUtil.GetMd5OfFullFile(fs),
ChecksumUtil.GetSha1OfFullFile(fs));
}
if (this.fileNameHash[pHashKey].ContainsKey(checksumKey))
{
ret = this.fileNameHash[pHashKey][checksumKey];
}
else
{
newFileName = this.fileNameHash[pHashKey].Count.ToString("X8");
this.fileNameHash[pHashKey].Add(checksumKey, newFileName);
ret = newFileName;
}
return(ret);
}
protected CriAfs2Archive InitializeExternalAwbArchive()
{
CriAfs2Archive afs2 = null;
string awbDirectory;
string awbMask;
string acbBaseFileName;
string[] awbFiles;
byte[] awbHashStored;
byte[] awbHashCalculated;
awbDirectory = Path.GetDirectoryName(this.SourceFile);
// try format 1
acbBaseFileName = Path.GetFileNameWithoutExtension(this.SourceFile);
awbMask = String.Format(AWB_FORMAT1, acbBaseFileName);
awbFiles = Directory.GetFiles(awbDirectory, awbMask, SearchOption.TopDirectoryOnly);
if (awbFiles.Length < 1)
{
// try format 2
awbMask = String.Format(AWB_FORMAT2, acbBaseFileName);
awbFiles = Directory.GetFiles(awbDirectory, awbMask, SearchOption.TopDirectoryOnly);
}
if (awbFiles.Length < 1)
{
// try format 3
awbMask = String.Format(AWB_FORMAT3, acbBaseFileName);
awbFiles = Directory.GetFiles(awbDirectory, awbMask, SearchOption.TopDirectoryOnly);
}
// file not found
if (awbFiles.Length < 1)
{
throw new FileNotFoundException(String.Format("Cannot find AWB file. Please verify corresponding AWB file is named '{0}', '{1}', or '{2}'.",
String.Format(AWB_FORMAT1, acbBaseFileName), String.Format(AWB_FORMAT2, acbBaseFileName), String.Format(AWB_FORMAT3, acbBaseFileName)));
}
if (awbFiles.Length > 1)
{
throw new FileNotFoundException(String.Format("More than one matching AWB file for this ACB. Please verify only one AWB file is named '{1}' or '{2}'.",
String.Format(AWB_FORMAT1, acbBaseFileName), String.Format(AWB_FORMAT2, acbBaseFileName)));
}
// initialize AFS2 file
using (FileStream fs = File.Open(awbFiles[0], FileMode.Open, FileAccess.Read, FileShare.Read))
{
awbHashStored = this.StreamAwbHash;
if (awbHashStored.Length == 0x10) // MD5, hash in newer format unknown
{
// validate MD5 checksum
awbHashCalculated = ByteConversion.GetBytesFromHexString(ChecksumUtil.GetMd5OfFullFile(fs));
if (!ParseFile.CompareSegment(awbHashCalculated, 0, awbHashStored))
{
throw new FormatException(String.Format("AWB file, <{0}>, did not match checksum inside ACB file.", Path.GetFileName(fs.Name)));
}
}
afs2 = new CriAfs2Archive(fs, 0);
}
return(afs2);
}
protected override void DoTaskForFile(string pPath,
IVgmtWorkerStruct pExamineChecksumGeneratorStruct, DoWorkEventArgs e)
{
ExamineChecksumGeneratorStruct examineChecksumGeneratorStruct =
(ExamineChecksumGeneratorStruct)pExamineChecksumGeneratorStruct;
string crc32;
string md5;
string sha1;
string vgmtCrc32 = "Not implemented for this format.";
string vgmtMd5 = "Not implemented for this format.";
string vgmtSha1 = "Not implemented for this format.";
string checksumKey;
Type formatType = null;
IFormat vgmData = null;
using (FileStream fs = File.OpenRead(pPath))
{
crc32 = ChecksumUtil.GetCrc32OfFullFile(fs);
md5 = ChecksumUtil.GetMd5OfFullFile(fs);
sha1 = ChecksumUtil.GetSha1OfFullFile(fs);
if (examineChecksumGeneratorStruct.CheckForDuplicates)
{
checksumKey = String.Format("{0}/{1}/{2}", crc32, md5, sha1);
this.addChecksumToHash(checksumKey, pPath, true);
}
if (examineChecksumGeneratorStruct.DoVgmtChecksums)
{
formatType = FormatUtil.getObjectType(fs);
if (formatType != null)
{
vgmData = (IFormat)Activator.CreateInstance(formatType);
vgmData.Initialize(fs, pPath);
}
}
}
if (vgmData != null)
{
Crc32 crc32Generator = new Crc32();
MD5CryptoServiceProvider md5Hash = new MD5CryptoServiceProvider();
FileStream md5FileStream = new FileStream(Path.GetTempFileName(), FileMode.Create, FileAccess.Write);
CryptoStream md5CryptoStream = new CryptoStream(md5FileStream, md5Hash, CryptoStreamMode.Write);
SHA1CryptoServiceProvider sha1Hash = new SHA1CryptoServiceProvider();
FileStream sha1FileStream = new FileStream(Path.GetTempFileName(), FileMode.Create, FileAccess.Write);
CryptoStream sha1CryptoStream = new CryptoStream(sha1FileStream, sha1Hash, CryptoStreamMode.Write);
vgmData.GetDatFileChecksums(ref crc32Generator, ref md5CryptoStream,
ref sha1CryptoStream);
md5CryptoStream.FlushFinalBlock();
sha1CryptoStream.FlushFinalBlock();
vgmtCrc32 = crc32Generator.Value.ToString("X8");
vgmtMd5 = ParseFile.ByteArrayToString(md5Hash.Hash);
vgmtSha1 = ParseFile.ByteArrayToString(sha1Hash.Hash);
if (examineChecksumGeneratorStruct.CheckForDuplicates)
{
checksumKey = String.Format("{0}/{1}/{2}", vgmtCrc32, vgmtMd5, vgmtSha1);
this.addChecksumToHash(checksumKey, pPath, false);
}
md5FileStream.Close();
md5FileStream.Dispose();
sha1FileStream.Close();
sha1FileStream.Dispose();
md5CryptoStream.Close();
md5CryptoStream.Dispose();
sha1CryptoStream.Close();
sha1CryptoStream.Dispose();
}
this.outputBuffer.AppendFormat("<{0}>{1}", pPath, Environment.NewLine);
this.outputBuffer.AppendFormat("CRC32: {0}{1}", crc32, Environment.NewLine);
this.outputBuffer.AppendFormat("MD5: {0}{1}", md5, Environment.NewLine);
this.outputBuffer.AppendFormat("SHA1: {0}{1}", sha1, Environment.NewLine);
if (examineChecksumGeneratorStruct.DoVgmtChecksums)
{
this.outputBuffer.AppendFormat("CRC32 (VGMT): {0}{1}", vgmtCrc32, Environment.NewLine);
this.outputBuffer.AppendFormat("MD5 (VGMT): {0}{1}", vgmtMd5, Environment.NewLine);
this.outputBuffer.AppendFormat("SHA1 (VGMT): {0}{1}", vgmtSha1, Environment.NewLine);
}
this.outputBuffer.AppendLine();
}
