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

[CHDFile, FileTools] Hook up CHDFile; create helper class

Browse Source
This commit is contained in:
Matt Nadareski
2017-10-31 02:09:59 -07:00
parent 9c46dd6f01
commit 3ddad2505e
2 changed files with 210 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files

241
SabreTools.Library/External/CHDFile.cs vendored
View File

@@ -65,131 +65,34 @@ namespace SabreTools.Library.External
/// ---------------------------------------------- /// ----------------------------------------------
/// </remrks> /// </remrks>
public class CHDFile public class CHDFile
{
/// <summary>
/// Information regarding the CHD, mostly unused
/// </summary>
private class CHD
{ {
// Core parameters from the header // Core parameters from the header
private ulong m_signature; // signature public ulong m_signature; // signature
private uint m_headersize; // size of the header public uint m_headersize; // size of the header
private uint m_version; // version of the header public uint m_version; // version of the header
private ulong m_logicalbytes; // logical size of the raw CHD data in bytes public ulong m_logicalbytes; // logical size of the raw CHD data in bytes
private ulong m_mapoffset; // offset of map public ulong m_mapoffset; // offset of map
private ulong m_metaoffset; // offset to first metadata bit public ulong m_metaoffset; // offset to first metadata bit
private uint m_hunkbytes; // size of each raw hunk in bytes public uint m_hunkbytes; // size of each raw hunk in bytes
private ulong m_hunkcount; // number of hunks represented private ulong m_hunkcount; // number of hunks represented
private uint m_unitbytes; // size of each unit in bytes public uint m_unitbytes; // size of each unit in bytes
private ulong m_unitcount; // number of units represented public ulong m_unitcount; // number of units represented
private CHDCodecType[] m_compression = new CHDCodecType[4]; // array of compression types used public CHDCodecType[] m_compression = new CHDCodecType[4]; // array of compression types used
// map information // map information
uint m_mapentrybytes; // length of each entry in a map public uint m_mapentrybytes; // length of each entry in a map
/// <summary>
/// Get internal metadata from a CHD
/// </summary>
/// <param name="filename">Filename of possible CHD</param>
/// <returns>A Disk object with internal SHA-1 on success, null on error, empty Disk otherwise</returns>
/// <remarks>
/// Original code had a "writable" param. This is not required for metadata checking
/// </remarks>
public DatItem GetCHDInfo(string filename)
{
FileStream fs = FileTools.TryOpenRead(filename);
DatItem datItem = GetCHDInfo(fs);
fs.Dispose();
return datItem;
}
/// <summary>
/// Get internal metadata from a CHD
/// </summary>
/// <param name="fs">Stream of possible CHD</param>
/// <returns>A Disk object with internal SHA-1 on success, null on error, empty Disk otherwise</returns>
/// <remarks>
/// Original code had a "writable" param. This is not required for metadata checking
/// </remarks>
public DatItem GetCHDInfo(Stream fs)
{
// Create a blank Disk to populate and return
Disk datItem = new Disk();
// Get a binary reader to make life easier
BinaryReader br = new BinaryReader(fs);
// Read and verify the CHD signature
m_signature = br.ReadUInt64();
if (m_signature != Constants.CHDSignature)
{
// throw CHDERR_INVALID_FILE;
return null;
}
// Get the header size and version
m_headersize = br.ReadUInt32();
m_version = br.ReadUInt32();
// Create a placeholder for the extracted SHA-1
byte[] sha1 = new byte[20];
// If we have a CHD v3 file, parse it accordingly
if (m_headersize == Constants.CHD_V3_HEADER_SIZE && m_version == 3)
{
sha1 = ParseCHDv3Header(br);
}
// If we have a CHD v4 file, parse it accordingly
else if (m_headersize == Constants.CHD_V4_HEADER_SIZE && m_version == 4)
{
sha1 = ParseCHDv4Header(br);
}
// If we have a CHD v5 file, parse it accordingly
else if (m_headersize == Constants.CHD_V5_HEADER_SIZE && m_version == 5)
{
sha1 = ParseCHDv5Header(br);
}
// If we don't have a valid combination, return null
else
{
// throw CHDERR_UNSUPPORTED_VERSION;
// throw CHDERR_INVALID_FILE;
return null;
}
// Set the SHA-1 of the Disk to return
datItem.SHA1 = BitConverter.ToString(sha1).Replace("-", string.Empty).ToLowerInvariant();
return datItem;
}
/// <summary>
/// Get if file is a valid CHD
/// </summary>
/// <param name="filename">Filename of possible CHD</param>
/// <returns>True if a the file is a valid CHD, false otherwise</returns>
public bool IsValidCHD(string filename)
{
DatItem datItem = GetCHDInfo(filename);
return datItem != null
&& datItem.Type == ItemType.Disk
&& ((Disk)datItem).SHA1 != null;
}
/// <summary>
/// Get if stream is a valid CHD
/// </summary>
/// <param name="fs">Stream of possible CHD</param>
/// <returns>True if a the file is a valid CHD, false otherwise</returns>
public bool IsValidCHD(Stream fs)
{
DatItem datItem = GetCHDInfo(fs);
return datItem != null
&& datItem.Type == ItemType.Disk
&& ((Disk)datItem).SHA1 != null;
}
/// <summary> /// <summary>
/// Parse a CHD v3 header /// Parse a CHD v3 header
/// </summary> /// </summary>
/// <param name="br">Binary reader representing the input stream</param> /// <param name="br">Binary reader representing the input stream</param>
/// <returns>The extracted SHA-1 on success, null otherwise</returns> /// <returns>The extracted SHA-1 on success, null otherwise</returns>
private byte[] ParseCHDv3Header(BinaryReader br) public byte[] ParseCHDv3Header(BinaryReader br)
{ {
// Set the blank SHA-1 hash // Set the blank SHA-1 hash
byte[] sha1 = new byte[20]; byte[] sha1 = new byte[20];
@@ -235,7 +138,7 @@ namespace SabreTools.Library.External
/// </summary> /// </summary>
/// <param name="br">Binary reader representing the input stream</param> /// <param name="br">Binary reader representing the input stream</param>
/// <returns>The extracted SHA-1 on success, null otherwise</returns> /// <returns>The extracted SHA-1 on success, null otherwise</returns>
private byte[] ParseCHDv4Header(BinaryReader br) public byte[] ParseCHDv4Header(BinaryReader br)
{ {
// Set the blank SHA-1 hash // Set the blank SHA-1 hash
byte[] sha1 = new byte[20]; byte[] sha1 = new byte[20];
@@ -280,7 +183,7 @@ namespace SabreTools.Library.External
/// </summary> /// </summary>
/// <param name="br">Binary reader representing the input stream</param> /// <param name="br">Binary reader representing the input stream</param>
/// <returns>The extracted SHA-1 on success, null otherwise</returns> /// <returns>The extracted SHA-1 on success, null otherwise</returns>
private byte[] ParseCHDv5Header(BinaryReader br) public byte[] ParseCHDv5Header(BinaryReader br)
{ {
// Set the blank SHA-1 hash // Set the blank SHA-1 hash
byte[] sha1 = new byte[20]; byte[] sha1 = new byte[20];
@@ -309,4 +212,110 @@ namespace SabreTools.Library.External
return sha1; return sha1;
} }
} }
/// <summary>
/// Get internal metadata from a CHD
/// </summary>
/// <param name="filename">Filename of possible CHD</param>
/// <returns>A Disk object with internal SHA-1 on success, null on error, empty Disk otherwise</returns>
/// <remarks>
/// Original code had a "writable" param. This is not required for metadata checking
/// </remarks>
public static DatItem GetCHDInfo(string filename)
{
FileStream fs = FileTools.TryOpenRead(filename);
DatItem datItem = GetCHDInfo(fs);
fs.Dispose();
return datItem;
}
/// <summary>
/// Get internal metadata from a CHD
/// </summary>
/// <param name="fs">Stream of possible CHD</param>
/// <returns>A Disk object with internal SHA-1 on success, null on error, empty Disk otherwise</returns>
/// <remarks>
/// Original code had a "writable" param. This is not required for metadata checking
/// </remarks>
public static DatItem GetCHDInfo(Stream fs)
{
// Create a blank Disk to populate and return
Disk datItem = new Disk();
// Get a CHD object to store the data
CHD chd = new CHD();
// Get a binary reader to make life easier
BinaryReader br = new BinaryReader(fs);
// Read and verify the CHD signature
chd.m_signature = br.ReadUInt64();
if (chd.m_signature != Constants.CHDSignature)
{
// throw CHDERR_INVALID_FILE;
return null;
}
// Get the header size and version
chd.m_headersize = br.ReadUInt32();
chd.m_version = br.ReadUInt32();
// Create a placeholder for the extracted SHA-1
byte[] sha1 = new byte[20];
// If we have a CHD v3 file, parse it accordingly
if (chd.m_headersize == Constants.CHD_V3_HEADER_SIZE && chd.m_version == 3)
{
sha1 = chd.ParseCHDv3Header(br);
}
// If we have a CHD v4 file, parse it accordingly
else if (chd.m_headersize == Constants.CHD_V4_HEADER_SIZE && chd.m_version == 4)
{
sha1 = chd.ParseCHDv4Header(br);
}
// If we have a CHD v5 file, parse it accordingly
else if (chd.m_headersize == Constants.CHD_V5_HEADER_SIZE && chd.m_version == 5)
{
sha1 = chd.ParseCHDv5Header(br);
}
// If we don't have a valid combination, return null
else
{
// throw CHDERR_UNSUPPORTED_VERSION;
// throw CHDERR_INVALID_FILE;
return null;
}
// Set the SHA-1 of the Disk to return
datItem.SHA1 = BitConverter.ToString(sha1).Replace("-", string.Empty).ToLowerInvariant();
return datItem;
}
/// <summary>
/// Get if file is a valid CHD
/// </summary>
/// <param name="filename">Filename of possible CHD</param>
/// <returns>True if a the file is a valid CHD, false otherwise</returns>
public static bool IsValidCHD(string filename)
{
DatItem datItem = GetCHDInfo(filename);
return datItem != null
&& datItem.Type == ItemType.Disk
&& ((Disk)datItem).SHA1 != null;
}
/// <summary>
/// Get if stream is a valid CHD
/// </summary>
/// <param name="fs">Stream of possible CHD</param>
/// <returns>True if a the file is a valid CHD, false otherwise</returns>
public static bool IsValidCHD(Stream fs)
{
DatItem datItem = GetCHDInfo(fs);
return datItem != null
&& datItem.Type == ItemType.Disk
&& ((Disk)datItem).SHA1 != null;
}
}
} }

49
SabreTools.Library/Tools/FileTools.cs
View File

@@ -734,8 +734,55 @@ namespace SabreTools.Library.Tools
/// <param name="keepReadOpen">True if the underlying read stream should be kept open, false otherwise</param> /// <param name="keepReadOpen">True if the underlying read stream should be kept open, false otherwise</param>
/// <returns>Populated DatItem object if success, empty one on error</returns> /// <returns>Populated DatItem object if success, empty one on error</returns>
public static DatItem GetStreamInfo(Stream input, long size, Hash omitFromScan = 0x0, public static DatItem GetStreamInfo(Stream input, long size, Hash omitFromScan = 0x0,
long offset = 0, bool keepReadOpen = false) long offset = 0, bool keepReadOpen = false, bool ignorechd = true)
{ {
// We first check to see if it's a CHD
if (ignorechd == false && CHDFile.IsValidCHD(input))
{
// Seek to the starting position, if one is set
try
{
if (offset < 0)
{
input.Seek(offset, SeekOrigin.End);
}
else if (offset > 0)
{
input.Seek(offset, SeekOrigin.Begin);
}
}
catch (NotSupportedException)
{
Globals.Logger.Verbose("Stream does not support seeking. Stream position not changed");
}
catch (NotImplementedException)
{
Globals.Logger.Warning("Stream does not support seeking. Stream position not changed");
}
// Get the Disk from the information
DatItem disk = CHDFile.GetCHDInfo(input);
// Seek to the beginning of the stream if possible
try
{
input.Seek(0, SeekOrigin.Begin);
}
catch (NotSupportedException)
{
Globals.Logger.Verbose("Stream does not support seeking. Stream position not changed");
}
catch (NotImplementedException)
{
Globals.Logger.Verbose("Stream does not support seeking. Stream position not changed");
}
if (!keepReadOpen)
{
input.Dispose();
}
}
Rom rom = new Rom Rom rom = new Rom
{ {
Type = ItemType.Rom, Type = ItemType.Rom,