public STLReader(string fileName)
{
reader = BinaryReaderExtensions.FromFile(fileName);
MPQHeader mHdr = reader.ReadStruct <MPQHeader>();
StlHeader sHdr = reader.ReadStruct <StlHeader>();
//StlEntry sEntry = reader.ReadStruct<StlEntry>();
RecordsCount = sHdr.entriesSize / 0x50;
m_rows = new byte[RecordsCount][];
for (int i = 0; i < RecordsCount; ++i)
{
m_rows[i] = reader.ReadBytes(0x50);
}
//StringTable = new Dictionary<int, string>();
//if (reader.BaseStream.Position != reader.BaseStream.Length)
//{
// while (reader.BaseStream.Position != reader.BaseStream.Length)
// {
// if (reader.PeekChar() == 0)
// {
// reader.BaseStream.Position++;
// continue;
// }
// int offset = (int)reader.BaseStream.Position;
// StringTable[offset] = reader.ReadStringNull();
// }
//}
}
public DB5SparseReader(string fileName)
{
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(String.Format("File {0} is corrupted!", fileName));
}
if (reader.ReadUInt32() != DB5FmtSig)
{
throw new InvalidDataException(String.Format("File {0} isn't valid DB5 sparse file!", fileName));
}
RecordsCount = reader.ReadInt32();
FieldsCount = reader.ReadInt32();
RecordSize = reader.ReadInt32();
int offsetsPos = reader.ReadInt32();
uint tableHash = reader.ReadUInt32();
uint build = reader.ReadUInt32();
int MinId = reader.ReadInt32();
int MaxId = reader.ReadInt32();
int locale = reader.ReadInt32();
int CopyTableSize = reader.ReadInt32();
int metaFlags = reader.ReadInt32();
FieldOffsets = new short[FieldsCount];
FieldBitSizes = new short[FieldsCount];
for (var i = 0; i < FieldsCount; ++i)
{
FieldBitSizes[i] = reader.ReadInt16();
FieldOffsets[i] = reader.ReadInt16();
}
reader.BaseStream.Position = offsetsPos;
var numOffsets = MaxId - MinId + 1;
for (var i = 0; i < numOffsets; ++i)
{
var offset = reader.ReadInt32();
var length = reader.ReadUInt16();
if (offset != 0 && length != 0)
{
var rowData = reader.ReadBytesAt(offset, length);
var rowDataWithId = new byte[length + 4];
Array.Copy(BitConverter.GetBytes(MinId + i), rowDataWithId, 4);
Array.Copy(rowData, 0, rowDataWithId, 4, rowData.Length);
RowData.Add(MinId + i, rowDataWithId);
}
}
RecordsCount = RowData.Count;
}
}
public DB2Reader(string fileName)
{
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(String.Format("File {0} is corrupted!", fileName));
}
if (reader.ReadUInt32() != DB2FmtSig)
{
throw new InvalidDataException(String.Format("File {0} isn't valid DBC file!", fileName));
}
RecordsCount = reader.ReadInt32();
FieldsCount = reader.ReadInt32();
RecordSize = reader.ReadInt32();
StringTableSize = reader.ReadInt32();
// WDB2 specific fields
uint tableHash = reader.ReadUInt32(); // new field in WDB2
uint build = reader.ReadUInt32(); // new field in WDB2
uint unk1 = reader.ReadUInt32(); // new field in WDB2
if (build > 12880) // new extended header
{
int MinId = reader.ReadInt32(); // new field in WDB2
int MaxId = reader.ReadInt32(); // new field in WDB2
int locale = reader.ReadInt32(); // new field in WDB2
int unk5 = reader.ReadInt32(); // new field in WDB2
if (MaxId != 0)
{
var diff = MaxId - MinId + 1; // blizzard is weird people...
reader.ReadBytes(diff * 4); // an index for rows
reader.ReadBytes(diff * 2); // a memory allocation bank
}
}
m_rows = new byte[RecordsCount][];
for (int i = 0; i < RecordsCount; i++)
{
m_rows[i] = reader.ReadBytes(RecordSize);
}
int stringTableStart = (int)reader.BaseStream.Position;
StringTable = new Dictionary <int, string>();
while (reader.BaseStream.Position != reader.BaseStream.Length)
{
int index = (int)reader.BaseStream.Position - stringTableStart;
StringTable[index] = reader.ReadStringNull();
}
}
}
public ADBReader(string fileName)
{
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(String.Format("File {0} is corrupted!", fileName));
}
var signature = reader.ReadUInt32();
if (signature != ADBFmtSig)
{
throw new InvalidDataException(String.Format("File {0} isn't valid DBC file!", fileName));
}
RecordsCount = reader.ReadInt32();
FieldsCount = reader.ReadInt32(); // not fields count in WCH2
RecordSize = reader.ReadInt32();
StringTableSize = reader.ReadInt32();
// WCH2 specific fields
uint tableHash = reader.ReadUInt32(); // new field in WCH2
uint build = reader.ReadUInt32(); // new field in WCH2
int unk1 = reader.ReadInt32(); // Unix time in WCH2
int unk2 = reader.ReadInt32(); // new field in WCH2
int unk3 = reader.ReadInt32(); // new field in WCH2 (index table?)
int locale = reader.ReadInt32(); // new field in WCH2
int unk5 = reader.ReadInt32(); // new field in WCH2
if (unk3 != 0)
{
reader.ReadBytes(unk3 * 4 - HeaderSize); // an index for rows
reader.ReadBytes(unk3 * 2 - HeaderSize * 2); // a memory allocation bank
}
m_rows = new byte[RecordsCount][];
for (int i = 0; i < RecordsCount; i++)
{
m_rows[i] = reader.ReadBytes(RecordSize);
}
int stringTableStart = (int)reader.BaseStream.Position;
StringTable = new Dictionary <int, string>();
while (reader.BaseStream.Position != reader.BaseStream.Length)
{
int index = (int)reader.BaseStream.Position - stringTableStart;
StringTable[index] = reader.ReadStringNull();
}
}
}
public WDBReader(string fileName)
{
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(String.Format("File {0} is corrupted!", fileName));
}
var signature = reader.ReadUInt32();
if (!WDBSigs.Contains(signature))
{
throw new InvalidDataException(String.Format("File {0} isn't valid WDB file!", fileName));
}
uint build = reader.ReadUInt32();
uint locale = reader.ReadUInt32();
var unk1 = reader.ReadInt32();
var unk2 = reader.ReadInt32();
var version = reader.ReadInt32();
m_rows = new Dictionary <int, byte[]>();
while (reader.BaseStream.Position != reader.BaseStream.Length)
{
var entry = reader.ReadInt32();
var size = reader.ReadInt32();
if (entry == 0 && size == 0 && reader.BaseStream.Position == reader.BaseStream.Length)
{
break;
}
var row = new byte[0]
.Concat(BitConverter.GetBytes(entry))
.Concat(reader.ReadBytes(size))
.ToArray();
m_rows.Add(entry, row);
}
RecordsCount = m_rows.Count;
}
}
public DBCReader(string fileName)
{
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(String.Format("File {0} is corrupted!", fileName));
}
if (reader.ReadUInt32() != DBCFmtSig)
{
throw new InvalidDataException(String.Format("File {0} isn't valid DBC file!", fileName));
}
RecordsCount = reader.ReadInt32();
FieldsCount = reader.ReadInt32();
RecordSize = reader.ReadInt32();
StringTableSize = reader.ReadInt32();
m_rows = new byte[RecordsCount][];
for (int i = 0; i < RecordsCount; i++)
{
m_rows[i] = reader.ReadBytes(RecordSize);
}
int stringTableStart = (int)reader.BaseStream.Position;
StringTable = new Dictionary <int, string>();
while (reader.BaseStream.Position != reader.BaseStream.Length)
{
int index = (int)reader.BaseStream.Position - stringTableStart;
StringTable[index] = reader.ReadStringNull();
}
reader.BaseStream.Position = stringTableStart;
Strings = reader.ReadBytes(StringTableSize);
}
}
public DB4Reader(string fileName)
{
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(String.Format("File {0} is corrupted!", fileName));
}
if (reader.ReadUInt32() != DB4FmtSig)
{
throw new InvalidDataException(String.Format("File {0} isn't valid DB4 file!", fileName));
}
RecordsCount = reader.ReadInt32();
FieldsCount = reader.ReadInt32();
RecordSize = reader.ReadInt32();
StringTableSize = reader.ReadInt32();
uint tableHash = reader.ReadUInt32();
uint build = reader.ReadUInt32();
uint unk1 = reader.ReadUInt32();
int MinId = reader.ReadInt32();
int MaxId = reader.ReadInt32();
int locale = reader.ReadInt32();
int CopyTableSize = reader.ReadInt32();
int metaFlags = reader.ReadInt32();
int stringTableStart = HeaderSize + RecordsCount * RecordSize;
int stringTableEnd = stringTableStart + StringTableSize;
// Index table
int[] m_indexes = null;
HasSeparateIndexColumn = stringTableEnd + CopyTableSize < reader.BaseStream.Length;
if (HasSeparateIndexColumn)
{
reader.BaseStream.Position = stringTableEnd;
m_indexes = new int[RecordsCount];
for (int i = 0; i < RecordsCount; i++)
{
m_indexes[i] = reader.ReadInt32();
}
}
// Records table
reader.BaseStream.Position = HeaderSize;
for (int i = 0; i < RecordsCount; i++)
{
byte[] recordBytes = reader.ReadBytes(RecordSize);
if (HasSeparateIndexColumn)
{
byte[] newRecordBytes = new byte[RecordSize + 4];
Array.Copy(BitConverter.GetBytes(m_indexes[i]), newRecordBytes, 4);
Array.Copy(recordBytes, 0, newRecordBytes, 4, recordBytes.Length);
Lookup.Add(m_indexes[i], newRecordBytes);
}
else
{
Lookup.Add(BitConverter.ToInt32(recordBytes, 0), recordBytes);
}
}
// Strings table
reader.BaseStream.Position = stringTableStart;
StringTable = new Dictionary <int, string>();
while (reader.BaseStream.Position != stringTableEnd)
{
int index = (int)reader.BaseStream.Position - stringTableStart;
StringTable[index] = reader.ReadStringNull();
}
// Copy index table
long copyTablePos = stringTableEnd + (HasSeparateIndexColumn ? 4 * RecordsCount : 0);
if (copyTablePos != reader.BaseStream.Length && CopyTableSize != 0)
{
reader.BaseStream.Position = copyTablePos;
while (reader.BaseStream.Position < reader.BaseStream.Length)
{
try
{
int id = reader.ReadInt32();
int idcopy = reader.ReadInt32();
byte[] copyRow = Lookup[idcopy];
byte[] newRow = new byte[copyRow.Length];
Array.Copy(copyRow, newRow, newRow.Length);
Array.Copy(BitConverter.GetBytes(id), newRow, 4);
Lookup.Add(id, newRow);
RecordsCount++;
}
catch (KeyNotFoundException)
{
}
}
}
}
}
public static IWowClientDBReader GetReader(string file, XmlElement definition)
{
IWowClientDBReader reader;
var ext = Path.GetExtension(file).ToUpperInvariant();
if (ext == ".DBC")
{
reader = new DBCReader(file);
}
else if (ext == ".DB2")
{
using (var binaryReader = BinaryReaderExtensions.FromFile(file))
{
switch (binaryReader.ReadUInt32())
{
case DB2Reader.DB2FmtSig:
{
reader = new DB2Reader(file);
break;
}
case DB3Reader.DB3FmtSig:
{
reader = new DB2Reader(file);
break;
}
case DB4Reader.DB4FmtSig:
{
try
{
reader = new DB4Reader(file);
}
catch
{
reader = new DB4SparseReader(file);
}
break;
}
case DB5Reader.DB5FmtSig:
{
try
{
reader = new DB5Reader(file, definition);
}
catch
{
reader = new DB5SparseReader(file);
}
break;
}
default:
throw new InvalidDataException(string.Format("Unknown file type {0}", ext));
}
}
}
else if (ext == ".ADB")
{
reader = new ADBReader(file);
}
else if (ext == ".WDB")
{
reader = new WDBReader(file);
}
else if (ext == ".STL")
{
reader = new STLReader(file);
}
else
{
throw new InvalidDataException(String.Format("Unknown file type {0}", ext));
}
return(reader);
}
public DBCReaderGeneric(string fileName)
{
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(String.Format("File {0} is corrupted!", fileName));
}
if (reader.ReadUInt32() != DBCFmtSig)
{
throw new InvalidDataException(String.Format("File {0} isn't valid DBC file!", fileName));
}
RecordsCount = reader.ReadInt32();
FieldsCount = reader.ReadInt32();
RecordSize = reader.ReadInt32();
StringTableSize = reader.ReadInt32();
long pos = reader.BaseStream.Position;
long stringTableStart = reader.BaseStream.Position + RecordsCount * RecordSize;
reader.BaseStream.Position = stringTableStart;
Dictionary <int, string> StringTable = new Dictionary <int, string>();
while (reader.BaseStream.Position != reader.BaseStream.Length)
{
int index = (int)(reader.BaseStream.Position - stringTableStart);
StringTable[index] = reader.ReadStringNull();
}
reader.BaseStream.Position = pos;
m_rows = new T[RecordsCount];
var props = typeof(T).GetProperties();
for (int i = 0; i < RecordsCount; i++)
{
T row = new T();
long rowStart = reader.BaseStream.Position;
for (int j = 0; j < props.Length; j++)
{
var prop = props[j];
switch (Type.GetTypeCode(prop.PropertyType))
{
case TypeCode.Int32:
prop.SetValue(row, reader.ReadInt32());
break;
case TypeCode.UInt32:
prop.SetValue(row, reader.ReadUInt32());
break;
case TypeCode.Single:
prop.SetValue(row, reader.ReadSingle());
break;
case TypeCode.String:
prop.SetValue(row, StringTable[reader.ReadInt32()]);
break;
default:
throw new Exception("Unsupported field type " + Type.GetTypeCode(prop.PropertyType));
}
}
if (reader.BaseStream.Position - rowStart != RecordSize)
{
// struct bigger than record size
if (reader.BaseStream.Position - rowStart > RecordSize)
{
throw new Exception("Incorrect DBC struct!");
}
// struct smaller than record size (incomplete)
reader.BaseStream.Position += RecordSize - (reader.BaseStream.Position - rowStart);
}
m_rows[i] = row;
}
}
}
public T Read <T>()
{
return(BinaryReaderExtensions.Read <T>(this));
}
public DB5Reader(string fileName, XmlElement definition)
{
var fields = definition.GetElementsByTagName("field");
var db5index = 0;
XmlNodeList indexes = definition.GetElementsByTagName("index");
if (indexes.Count == 1)
{
for (var j = 0; j < fields.Count; ++j)
{
if (fields[j].Attributes["name"].Value == indexes[0]["primary"].InnerText)
{
db5index = j;
break;
}
}
}
using (var reader = BinaryReaderExtensions.FromFile(fileName))
{
if (reader.BaseStream.Length < HeaderSize)
{
throw new InvalidDataException(string.Format("File {0} is corrupted!", fileName));
}
if (reader.ReadUInt32() != DB5FmtSig)
{
throw new InvalidDataException(string.Format("File {0} isn't valid DB5 file!", fileName));
}
RecordsCount = reader.ReadInt32();
FieldsCount = reader.ReadInt32();
RecordSize = reader.ReadInt32();
StringTableSize = reader.ReadInt32();
uint tableHash = reader.ReadUInt32();
uint build = reader.ReadUInt32();
int MinId = reader.ReadInt32();
int MaxId = reader.ReadInt32();
int locale = reader.ReadInt32();
int CopyTableSize = reader.ReadInt32();
int metaFlags = reader.ReadInt32();
FieldOffsets = new short[FieldsCount];
FieldBitSizes = new short[FieldsCount];
for (var i = 0; i < FieldsCount; ++i)
{
FieldBitSizes[i] = reader.ReadInt16();
FieldOffsets[i] = reader.ReadInt16();
}
var headerEnd = reader.BaseStream.Position;
var stringTableStart = headerEnd + RecordsCount * RecordSize;
var stringTableEnd = stringTableStart + StringTableSize;
// Index table
int[] m_indexes = null;
HasSeparateIndexColumn = stringTableEnd + CopyTableSize < reader.BaseStream.Length;
if (HasSeparateIndexColumn)
{
reader.BaseStream.Position = stringTableEnd;
m_indexes = new int[RecordsCount];
for (int i = 0; i < RecordsCount; i++)
{
m_indexes[i] = reader.ReadInt32();
}
}
// Records table
reader.BaseStream.Position = headerEnd;
for (int i = 0; i < RecordsCount; i++)
{
byte[] recordBytes = reader.ReadBytes(RecordSize);
if (HasSeparateIndexColumn)
{
byte[] newRecordBytes = new byte[RecordSize + 4];
Array.Copy(BitConverter.GetBytes(m_indexes[i]), newRecordBytes, 4);
Array.Copy(recordBytes, 0, newRecordBytes, 4, recordBytes.Length);
Lookup.Add(m_indexes[i], newRecordBytes);
}
else
{
var idBytes = new byte[4];
Array.Copy(recordBytes, FieldOffsets[db5index], idBytes, 0, 4 - FieldBitSizes[db5index] / 8);
Lookup.Add(BitConverter.ToInt32(idBytes, 0), recordBytes);
}
}
// Strings table
reader.BaseStream.Position = stringTableStart;
StringTable = new Dictionary <int, string>();
while (reader.BaseStream.Position != stringTableEnd)
{
int index = (int)(reader.BaseStream.Position - stringTableStart);
StringTable[index] = reader.ReadStringNull();
}
// Copy index table
long copyTablePos = stringTableEnd + (HasSeparateIndexColumn ? 4 * RecordsCount : 0);
if (copyTablePos != reader.BaseStream.Length && CopyTableSize != 0)
{
reader.BaseStream.Position = copyTablePos;
while (reader.BaseStream.Position < reader.BaseStream.Length)
{
try
{
int id = reader.ReadInt32();
int idcopy = reader.ReadInt32();
byte[] copyRow = Lookup[idcopy];
byte[] newRow = new byte[copyRow.Length];
Array.Copy(copyRow, newRow, newRow.Length);
Array.Copy(BitConverter.GetBytes(id), newRow, 4);
Lookup.Add(id, newRow);
RecordsCount++;
}
catch (KeyNotFoundException)
{
}
}
}
}
}
public override IEditableVariable Copy(ICR2WCopyAction context)
{
return(BinaryReaderExtensions.CopyViaBuffer(this, base.Copy(context)));
}