claunia/SabreTools
1
0
Fork 0
mirror of https://github.com/claunia/SabreTools.git synced 2025-12-16 19:14:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ArchiveTools] Create GetExtendedArchiveFileInfo

Browse Source
This commit is contained in:
Matt Nadareski
2017-03-16 01:02:55 -07:00
parent b30320b017
commit e06226b142
1 changed files with 181 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

181
SabreTools.Helper/Tools/ArchiveTools.cs
View File

@@ -571,6 +571,187 @@ namespace SabreTools.Helper.Tools
return roms;
}
/// <summary>
/// Generate a list of RomData objects from the header values in an archive
/// </summary>
/// <param name="input">Input file to get data from</param>
/// <returns>List of RomData objects representing the found data</returns>
/// <remarks>TODO: All instances of Hash.DeepHashes should be made into 0x0 eventually</remarks>
public static List<Rom> GetExtendedArchiveFileInfo(string input, Hash omitFromScan = Hash.DeepHashes)
{
List<Rom> found = new List<Rom>();
// First get the archive type
ArchiveType? at = GetCurrentArchiveType(input);
// If we got back null, then it's not an archive, so we we return
if (at == null)
{
return null;
}
try
{
Globals.Logger.Verbose("Found archive of type: " + at);
switch (at)
{
// 7-zip
case ArchiveType.SevenZip:
SevenZipArchive sza = SevenZipArchive.Open(FileTools.TryOpenRead(input));
foreach (SevenZipArchiveEntry entry in sza.Entries)
{
// Create and populate the entry stream
MemoryStream entryStream = new MemoryStream();
entry.WriteTo(entryStream);
// Get and add the extended Rom information
Rom sevenZipEntryRom = FileTools.GetStreamInfo(entryStream, entryStream.Length, omitFromScan: omitFromScan);
sevenZipEntryRom.Name = entry.Key;
sevenZipEntryRom.Machine = new Machine()
{
Name = Path.GetFileNameWithoutExtension(input),
Description = Path.GetFileNameWithoutExtension(input),
};
sevenZipEntryRom.Date = entry.LastModifiedTime?.ToString("yyyy/MM/dd hh:mm:ss");
found.Add(sevenZipEntryRom);
}
// Dispose of the archive
sza.Dispose();
break;
// GZip
case ArchiveType.GZip:
// Create and populate the entry stream
MemoryStream outstream = new MemoryStream();
GZipStream gzstream = new GZipStream(FileTools.TryOpenRead(input), Ionic.Zlib.CompressionMode.Decompress);
gzstream.CopyTo(outstream);
// Get and add the extended Rom information
Rom gzipEntryRom = FileTools.GetStreamInfo(outstream, outstream.Length, omitFromScan: omitFromScan);
gzipEntryRom.Name = gzstream.FileName;
gzipEntryRom.Machine = new Machine()
{
Name = Path.GetFileNameWithoutExtension(input),
Description = Path.GetFileNameWithoutExtension(input),
};
gzipEntryRom.Date = gzstream.LastModified?.ToString("yyyy/MM/dd hh:mm:ss");
found.Add(gzipEntryRom);
// Dispose of the archive
gzstream.Dispose();
break;
// RAR
case ArchiveType.Rar:
RarArchive ra = RarArchive.Open(input);
foreach (RarArchiveEntry entry in ra.Entries)
{
// Create and populate the entry stream
MemoryStream entryStream = new MemoryStream();
entry.WriteTo(entryStream);
// Get and add the extended Rom information
Rom rarEntryRom = FileTools.GetStreamInfo(entryStream, entryStream.Length, omitFromScan: omitFromScan);
rarEntryRom.Name = entry.Key;
rarEntryRom.Machine = new Machine()
{
Name = Path.GetFileNameWithoutExtension(input),
Description = Path.GetFileNameWithoutExtension(input),
};
rarEntryRom.Date = entry.LastModifiedTime?.ToString("yyyy/MM/dd hh:mm:ss");
found.Add(rarEntryRom);
}
// Dispose of the archive
ra.Dispose();
break;
// TAR
case ArchiveType.Tar:
// Extract all files to the temp directory
TarArchive ta = TarArchive.Open(input);
foreach (TarArchiveEntry entry in ta.Entries)
{
// Create and populate the entry stream
MemoryStream entryStream = new MemoryStream();
entry.WriteTo(entryStream);
// Get and add the extended Rom information
Rom tarEntryName = FileTools.GetStreamInfo(entryStream, entryStream.Length, omitFromScan: omitFromScan);
tarEntryName.Name = entry.Key;
tarEntryName.Machine = new Machine()
{
Name = Path.GetFileNameWithoutExtension(input),
Description = Path.GetFileNameWithoutExtension(input),
};
tarEntryName.Date = entry.LastModifiedTime?.ToString("yyyy/MM/dd hh:mm:ss");
found.Add(tarEntryName);
}
// Dispose of the archive
ta.Dispose();
break;
// Zip
case ArchiveType.Zip:
ZipFile zf = new ZipFile();
ZipReturn zr = zf.Open(input, new FileInfo(input).LastWriteTime.Ticks, true);
if (zr != ZipReturn.ZipGood)
{
throw new Exception(ZipFile.ZipErrorMessageText(zr));
}
for (int i = 0; i < zf.EntriesCount && zr == ZipReturn.ZipGood; i++)
{
// Open the read stream
zr = zf.OpenReadStream(i, false, out Stream readStream, out ulong streamsize, out SabreTools.Helper.Data.CompressionMethod cm, out uint lastMod);
// If the entry ends with a directory separator, continue to the next item, if any
if (zf.Entries[i].FileName.EndsWith(Path.DirectorySeparatorChar.ToString())
|| zf.Entries[i].FileName.EndsWith(Path.AltDirectorySeparatorChar.ToString())
|| zf.Entries[i].FileName.EndsWith(Path.PathSeparator.ToString()))
{
continue;
}
MemoryStream entryStream = new MemoryStream();
byte[] ibuffer = new byte[_bufferSize];
int ilen;
while ((ilen = readStream.Read(ibuffer, 0, _bufferSize)) > 0)
{
entryStream.Write(ibuffer, 0, ilen);
entryStream.Flush();
}
zr = zf.CloseReadStream();
// Get and add the extended Rom information
Rom rarEntryRom = FileTools.GetStreamInfo(entryStream, entryStream.Length, omitFromScan: omitFromScan);
rarEntryRom.Name = zf.Entries[i].FileName;
rarEntryRom.Machine = new Machine()
{
Name = Path.GetFileNameWithoutExtension(input),
Description = Path.GetFileNameWithoutExtension(input),
};
rarEntryRom.Date = zf.Entries[i].LastMod.ToString("yyyy/MM/dd hh:mm:ss"); // TODO: Fix this from ticks
found.Add(rarEntryRom);
}
// Dispose of the archive
zf.Close();
break;
}
}
catch (Exception)
{
// Don't log file open errors
}
return found;
}
/// <summary>
/// Retrieve file information for a single torrent GZ file
/// </summary>