Files
WoWDBDefs/code/C#/DBDefsDumper/Program.cs
2024-12-19 17:16:50 -07:00

1170 lines
50 KiB
C#

using System;
using System.Collections.Generic;
using System.IO;
using System.IO.MemoryMappedFiles;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Text.Json;
using System.Text.Json.Serialization;
namespace DBDefsDumper
{
class Program
{
static void Main(string[] args)
{
if (args.Length < 2)
{
throw new ArgumentException("Not enough arguments! Required: file, outdir, (build in x.x.x.build format), (pattern name to always use)");
}
if (!File.Exists(args[0]))
{
throw new FileNotFoundException("File not found!");
}
var file = MemoryMappedFile.CreateFromFile(args[0], FileMode.Open);
var maps = EXEParsing.GenerateMap(file.CreateViewAccessor());
ulong translate(ulong search) => maps.Select(map => map.translateMemoryToFile(search)).FirstOrDefault(r => r != 0x0);
using (var bin = new BinaryReader(file.CreateViewStream()))
{
var chunkSize = 1024;
var build = "";
var patternOverride = "";
if (args.Length >= 3)
{
build = args[2];
if (args.Length >= 4)
{
patternOverride = args[3];
}
}
// Find version
var buildPattern = new byte?[] { 0x42, 0x75, 0x69, 0x6C, 0x64, 0x20, null, null, null, null, null, 0x20, 0x28, null, null, null, null, null, 0x29, 0x20, 0x28 };
var buildPatternLength = buildPattern.Length;
if (build == "")
{
Console.WriteLine("No build given, scanning executable for \"Build ????? (?????) (\"");
while (true)
{
if ((bin.BaseStream.Length - bin.BaseStream.Position) < chunkSize)
{
break;
}
var posInStack = Search(bin.ReadBytes(chunkSize), buildPattern);
if (posInStack != chunkSize)
{
var matchPos = bin.BaseStream.Position - chunkSize + posInStack;
bin.BaseStream.Position = matchPos;
bin.ReadBytes(6);
var buildNumber = new string(bin.ReadChars(5));
bin.ReadBytes(2);
var patch = new string(bin.ReadChars(5));
build = patch + "." + buildNumber;
}
else
{
bin.BaseStream.Position = bin.BaseStream.Position - buildPatternLength;
}
}
}
if (build == "")
{
Console.WriteLine("No build found, rescanning executable for \"Build ????? (??????) (\"");
bin.BaseStream.Position = 0;
buildPattern = new byte?[] { 0x42, 0x75, 0x69, 0x6C, 0x64, 0x20, null, null, null, null, null, 0x20, 0x28, null, null, null, null, null, null, 0x29, 0x20, 0x28 };
buildPatternLength = buildPattern.Length;
while (true)
{
if ((bin.BaseStream.Length - bin.BaseStream.Position) < chunkSize)
{
break;
}
var posInStack = Search(bin.ReadBytes(chunkSize), buildPattern);
if (posInStack != chunkSize)
{
var matchPos = bin.BaseStream.Position - chunkSize + posInStack;
bin.BaseStream.Position = matchPos;
bin.ReadBytes(6);
var buildNumber = new string(bin.ReadChars(5));
bin.ReadBytes(2);
var patch = new string(bin.ReadChars(6));
build = patch + "." + buildNumber;
}
else
{
bin.BaseStream.Position = bin.BaseStream.Position - buildPatternLength;
}
}
}
if (build == "")
{
// Retry with backup pattern (crash log output)
Console.WriteLine("No build found, rescanning executable for \"0x00<Version> \"");
bin.BaseStream.Position = 0;
buildPattern = new byte?[] { 0x00, 0x3C, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6F, 0x6E, 0x3E, 0x20 }; // <Version>
buildPatternLength = buildPattern.Length;
while (true)
{
if ((bin.BaseStream.Length - bin.BaseStream.Position) < chunkSize)
{
break;
}
var posInStack = Search(bin.ReadBytes(chunkSize), buildPattern);
if (posInStack != chunkSize)
{
var matchPos = bin.BaseStream.Position - chunkSize + posInStack;
bin.BaseStream.Position = matchPos;
bin.ReadBytes(11);
var sb = new StringBuilder();
var buildPart = bin.ReadChar();
var sanity = 0;
while (buildPart != 0x0D)
{
sb.Append(buildPart);
buildPart = bin.ReadChar();
sanity++;
if (sanity > 20)
{
sb.Clear();
break;
}
}
// Filter out second pattern match of "<Version> %s%s" by checking if build is formatted correctly
if (sb.ToString().Split('.').Length == 4)
build = sb.ToString();
}
else
{
bin.BaseStream.Position = bin.BaseStream.Position - buildPatternLength;
}
}
}
if (build == "")
{
// Retry with RenderService pattern..
bin.BaseStream.Position = 0;
Console.WriteLine("No build found, rescanning executable for \"RenderService ?????0x00\"");
buildPattern = new byte?[] { 0x52, 0x65, 0x6e, 0x64, 0x65, 0x72, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x20, null, null, null, null, null, 0x00 }; // RenderService ?????0x00
buildPatternLength = buildPattern.Length;
while (true)
{
if ((bin.BaseStream.Length - bin.BaseStream.Position) < chunkSize)
{
break;
}
var posInStack = Search(bin.ReadBytes(chunkSize), buildPattern);
if (posInStack != chunkSize)
{
var matchPos = bin.BaseStream.Position - chunkSize + posInStack;
bin.BaseStream.Position = matchPos;
bin.ReadBytes(14);
build = new string(bin.ReadChars(5));
if (args.Length == 3)
{
build = args[2];
}
else
{
var binaryName = Path.GetFileNameWithoutExtension(args[0]);
if (binaryName.StartsWith("WOW-"))
{
Console.WriteLine("Expansion, major and minor version not found in binary, checking filename..");
var splitBinaryName = binaryName.Split('-');
if (splitBinaryName[1].Contains("patch"))
{
var patchName = splitBinaryName[1].Split('_')[0].Split("patch")[1];
if (patchName.Split('.').Length == 3)
{
build = patchName + "." + build; ;
}
else
{
Console.WriteLine("Expansion, major and minor version not found in binary. Please enter it in this format X.X.X: ");
build = Console.ReadLine() + "." + build;
}
}
else
{
Console.WriteLine("Expansion, major and minor version not found in binary. Please enter it in this format X.X.X: ");
build = Console.ReadLine() + "." + build;
}
}
else
{
Console.WriteLine("Expansion, major and minor version not found in binary. Please enter it in this format X.X.X: ");
build = Console.ReadLine() + "." + build;
}
}
}
else
{
bin.BaseStream.Position = bin.BaseStream.Position - buildPatternLength;
}
}
}
if (build == "")
{
Console.WriteLine("Build was not found! Please enter a build in this format: X.X.X.XXXXX");
build = Console.ReadLine();
}
if (build == "8.0.1.26321")
{
Console.WriteLine("Build 8.0.1.26321 has incorrect DBMeta, skipping..");
return;
}
// Reset position for DBMeta reading
bin.BaseStream.Position = 0;
// Extract DBMeta
var metas = new Dictionary<string, DBMeta>();
// Manual list of names to ignore as these are known bad pattern matches
var badNames = new List<string>() { "rippleDetail", "RAIDrippleDetail", "pbeWithMD5AndDES-CBC" };
var patternBuilder = new PatternBuilder();
Pattern usedPattern = null;
foreach (var pattern in patternBuilder.patterns)
{
if (patternOverride == "")
{
if (!pattern.allows(build))
{
Console.WriteLine("Skipping " + pattern.name + " as it does not list " + build + " as compatible!");
continue;
}
}
else
{
if (patternOverride != pattern.name)
continue;
}
usedPattern = pattern;
var patternBytes = ParsePattern(pattern.cur_pattern).ToArray();
var patternLength = patternBytes.Length;
while (true)
{
if ((bin.BaseStream.Length - bin.BaseStream.Position) < chunkSize)
{
break;
}
var posInStack = Search(bin.ReadBytes(chunkSize), patternBytes);
if (posInStack != chunkSize)
{
var matchPos = bin.BaseStream.Position - chunkSize + posInStack;
Console.WriteLine("Pattern " + pattern.name + " matched at " + matchPos);
if (pattern.offsets.ContainsKey(Name.FDID))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.FDID];
var fdid = bin.ReadUInt32();
if (fdid < 53183 || fdid > (6363366 * 4) /*4x max at 2024-11-07*/)
{
Console.WriteLine("Invalid filedataid " + fdid + ", skipping match..");
continue;
}
}
if (pattern.offsets.ContainsKey(Name.RECORD_SIZE))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.RECORD_SIZE];
if (bin.ReadUInt32() == 0)
{
Console.WriteLine("Record size is 0, skipping match..");
continue;
}
}
if (pattern.offsets.ContainsKey(Name.DB_NAME))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.DB_NAME];
if (bin.ReadUInt32() < 10)
{
Console.WriteLine("Name offset is invalid, skipping match..");
continue;
}
bin.BaseStream.Position = matchPos + pattern.offsets[Name.DB_NAME];
var targetOffset = (long)translate(bin.ReadUInt64());
if (targetOffset > bin.BaseStream.Length)
{
Console.WriteLine("Name offset is out of range of file, skipping match..");
continue;
}
}
if (pattern.offsets.ContainsKey(Name.CONVERT_STRINGREFS))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.CONVERT_STRINGREFS];
var targetOffset = (long)translate(bin.ReadUInt64());
if (targetOffset > bin.BaseStream.Length)
{
Console.WriteLine("CONVERT_STRINGREFS offset is out of range of file, skipping match..");
continue;
}
}
if (pattern.offsets.ContainsKey(Name.DB_FILENAME))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.DB_FILENAME];
if (bin.ReadUInt32() < 10)
{
Console.WriteLine("Name offset is invalid, skipping match..");
continue;
}
bin.BaseStream.Position = matchPos + pattern.offsets[Name.DB_FILENAME];
var targetOffset = (long)translate(bin.ReadUInt64());
if (targetOffset > bin.BaseStream.Length)
{
Console.WriteLine("Name offset is out of range of file, skipping match..");
continue;
}
}
if (pattern.offsets.ContainsKey(Name.NUM_FIELD_IN_FILE))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.NUM_FIELD_IN_FILE];
if (bin.ReadUInt32() > 5000)
{
Console.WriteLine("Num fields in file is over 5000, skipping match..");
continue;
}
}
if (pattern.offsets.ContainsKey(Name.FIELD_TYPES_IN_FILE) && pattern.offsets.ContainsKey(Name.FIELD_SIZES_IN_FILE))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.FIELD_TYPES_IN_FILE];
var fieldTypesInFile = bin.ReadInt64();
bin.BaseStream.Position = matchPos + pattern.offsets[Name.FIELD_SIZES_IN_FILE];
var fieldSizesInFileOffs = bin.ReadInt64();
if (fieldTypesInFile == fieldSizesInFileOffs)
{
Console.WriteLine("Field types in file offset == field sizes in file offset, skipping match..");
continue;
}
}
if (pattern.offsets.ContainsKey(Name.DB_CACHE_FILENAME))
{
bin.BaseStream.Position = matchPos + pattern.offsets[Name.DB_CACHE_FILENAME];
bin.BaseStream.Position = (long)translate((ulong)bin.ReadInt64());
var adbname = bin.ReadCString();
bin.BaseStream.Position = matchPos + pattern.offsets[Name.DB_CACHE_FILENAME];
if (!adbname.EndsWith("adb"))
{
Console.WriteLine("ADB filename does not end in adb, skipping match..");
continue;
}
}
bin.BaseStream.Position = matchPos;
var meta = ReadMeta(bin, pattern);
if (pattern.offsets.ContainsKey(Name.DB_NAME))
{
bin.BaseStream.Position = (long)translate((ulong)meta.nameOffset);
var filename = bin.ReadCString();
if (filename.Contains("DBFilesClient"))
{
filename = filename.Substring(filename.IndexOf("\\") + 1);
}
if (badNames.Contains(Path.GetFileNameWithoutExtension(filename)))
{
Console.WriteLine("Skipping known bad match (" + filename + ")");
}
else
{
metas.TryAdd(Path.GetFileNameWithoutExtension(filename), meta);
}
}
else if (pattern.offsets.ContainsKey(Name.DB_FILENAME))
{
bin.BaseStream.Position = (long)translate((ulong)meta.dbFilenameOffs);
var name = bin.ReadCString();
if (name.Contains("DBFilesClient"))
{
name = name.Substring(name.IndexOf("\\") + 1);
}
if (badNames.Contains(Path.GetFileNameWithoutExtension(name)))
{
Console.WriteLine("Skipping known bad match (" + name + ")");
}
else
{
metas.TryAdd(Path.GetFileNameWithoutExtension(name), meta);
}
}
bin.BaseStream.Position = matchPos + patternLength;
}
else
{
bin.BaseStream.Position = bin.BaseStream.Position - patternLength;
}
}
bin.BaseStream.Position = 0;
}
var outputDirectory = Path.Combine(args[1], build);
if (!Directory.Exists(outputDirectory))
{
Directory.CreateDirectory(outputDirectory);
}
// Process DBMetas
foreach (var meta in metas)
{
if ((long)translate((ulong)meta.Value.field_sizes_offs) > bin.BaseStream.Length)
{
Console.WriteLine("Skipping reading of " + meta.Key + " because field sizes offset (" + (long)translate((ulong)meta.Value.field_sizes_offs) + ") is outside of file range (" + bin.BaseStream.Length + ")!");
continue;
}
//if (meta.Value.table_hash != 0 && usedPattern.offsets.ContainsKey(Name.TABLE_HASH))
//{
// if (MakeTableHash(meta.Key) != meta.Value.table_hash)
// {
// Console.WriteLine("TableHash mismatch, skipping match..");
// continue;
// }
// else
// {
// Console.WriteLine("TableHash for " + meta.Key + "ok!");
// }
//}
var writer = new StreamWriter(Path.Combine(outputDirectory, meta.Key + ".dbd"));
writer.WriteLine("COLUMNS");
Console.Write("Writing " + meta.Key + ".dbd..");
var fieldCount = 0;
if (meta.Value.num_fields == 0 && meta.Value.num_fields_in_file != 0)
{
fieldCount = meta.Value.num_fields_in_file;
}
else
{
fieldCount = meta.Value.num_fields;
}
// \todo For some reason in 7.2.0, these are dynamically initialized, so unavailable. Luckily we don't use them anyway.
/// var field_offsets = ReadFieldArray(bin, fieldCount, (long)translate((ulong)meta.Value.field_offsets_offs)); // TODO: Field offsets is currently unused, use to verify sizes
var field_sizes = ReadFieldArray(bin, fieldCount, (long)translate((ulong)meta.Value.field_sizes_offs));
var field_types = ReadFieldArray(bin, fieldCount, (long)translate((ulong)meta.Value.field_types_offs));
var field_flags = new List<int>();
if ((build.StartsWith("7.3.0.") || build.StartsWith("7.3.2.")) && (long)translate((ulong)meta.Value.field_flags_offs) > bin.BaseStream.Length)
{
for (var fc = 0; fc < fieldCount; fc++)
{
field_flags.Add(0);
}
}
else
{
field_flags = ReadFieldArray(bin, fieldCount, (long)translate((ulong)meta.Value.field_flags_offs));
}
var field_sizes_in_file = ReadFieldArray(bin, fieldCount, (long)translate((ulong)meta.Value.field_sizes_in_file_offs));
var field_types_in_file = ReadFieldArray(bin, fieldCount, (long)translate((ulong)meta.Value.field_types_in_file_offs));
var field_flags_in_file = ReadFieldArray(bin, fieldCount, (long)translate((ulong)meta.Value.field_flags_in_file_offs));
var field_names_in_file = ReadFieldOffsetArray(bin, fieldCount, (long)translate((ulong)meta.Value.namesInFileOffs));
if (meta.Value.id_column == -1)
{
writer.WriteLine("int ID");
}
var columnNames = new List<string>();
var columnTypeFlags = new List<Tuple<int, int>>();
for (var i = 0; i < meta.Value.num_fields_in_file; i++)
{
if (field_flags_in_file.Count == 0)
{
columnTypeFlags.Add(new Tuple<int, int>(field_types_in_file[i], 0));
}
else
{
columnTypeFlags.Add(new Tuple<int, int>(field_types_in_file[i], field_flags_in_file[i]));
}
}
if (meta.Value.num_fields != 0 && (meta.Value.num_fields_in_file != meta.Value.num_fields))
{
if (meta.Value.num_fields_in_file > field_flags.Count())
{
columnTypeFlags.Add(new Tuple<int, int>(field_types[meta.Value.num_fields_in_file], 0));
}
else
{
columnTypeFlags.Add(new Tuple<int, int>(field_types[meta.Value.num_fields_in_file], field_flags[meta.Value.num_fields_in_file]));
}
}
for (var i = 0; i < columnTypeFlags.Count; i++)
{
if (field_names_in_file.Count > 0)
{
bin.BaseStream.Position = (long)translate(field_names_in_file[i]);
columnNames.Add(CleanRealName(bin.ReadCString()));
}
else
{
columnNames.Add(GenerateName(i, build));
}
var t = TypeToT(columnTypeFlags[i].Item1, (FieldFlags)columnTypeFlags[i].Item2);
if (field_names_in_file.Count > 0)
{
if (t.Item1 == "locstring")
{
writer.WriteLine(t.Item1 + " " + columnNames[i] + "_lang");
}
else
{
if (t.Item1 == "uint")
{
writer.WriteLine("int " + columnNames[i]);
}
else
{
writer.WriteLine(t.Item1 + " " + columnNames[i]);
}
}
}
else
{
if (t.Item1 == "locstring")
{
writer.WriteLine(t.Item1 + " " + columnNames[i] + "_lang?");
}
else
{
if (t.Item1 == "uint")
{
writer.WriteLine("int " + columnNames[i] + "?");
}
else
{
writer.WriteLine(t.Item1 + " " + columnNames[i] + "?");
}
}
}
}
writer.WriteLine();
if (meta.Value.layout_hash != 0)
{
writer.WriteLine("LAYOUT " + meta.Value.layout_hash.ToString("X8").ToUpper());
}
writer.WriteLine("BUILD " + build);
if (meta.Value.sparseTable == 1)
{
writer.WriteLine("COMMENT table is sparse");
}
if (meta.Value.id_column == -1)
{
writer.WriteLine("$noninline,id$ID<32>");
}
for (var i = 0; i < meta.Value.num_fields_in_file; i++)
{
var typeFlags = ("int", 32);
if (field_flags_in_file.Count == 0)
{
typeFlags = TypeToT(field_types_in_file[i], 0);
}
else
{
typeFlags = TypeToT(field_types_in_file[i], (FieldFlags)field_flags_in_file[i]);
}
if (meta.Value.id_column == i)
{
writer.Write("$id$");
}
if (meta.Value.has_relation != 0)
{
if (meta.Value.relation_col == i)
{
writer.Write("$relation$");
if (meta.Value.relation_col_in_file != i)
{
throw new Exception("No relation_col_in_file but there is relation_col send help!");
}
}
}
writer.Write(columnNames[i]);
if (typeFlags.Item1 == "locstring")
{
writer.Write("_lang");
}
if (typeFlags.Item2 > 0)
{
if (typeFlags.Item1 == "uint")
{
writer.Write("<u" + typeFlags.Item2 + ">");
}
else
{
writer.Write("<" + typeFlags.Item2 + ">");
}
}
if (field_sizes_in_file[i] != 1)
{
// 6.0.1 has sizes in bytes
if (build.StartsWith("6."))
{
var supposedSize = 0;
if ((typeFlags.Item1 == "uint" || typeFlags.Item1 == "int") && typeFlags.Item2 != 32)
{
supposedSize = typeFlags.Item2 / 8;
}
else
{
supposedSize = 4;
}
var fixedSize = field_sizes_in_file[i] / supposedSize;
if (fixedSize > 1)
{
writer.Write("[" + fixedSize + "]");
}
}
else
{
writer.Write("[" + field_sizes_in_file[i] + "]");
}
}
writer.WriteLine();
}
if (meta.Value.num_fields != 0 && (meta.Value.num_fields_in_file != meta.Value.num_fields))
{
var i = meta.Value.num_fields_in_file;
var typeFlags = TypeToT(columnTypeFlags[i].Item1, (FieldFlags)columnTypeFlags[i].Item2);
writer.Write("$noninline,relation$" + columnNames[i]);
if (typeFlags.Item1 == "locstring")
{
writer.Write("_lang");
}
if (typeFlags.Item2 > 0)
{
if (typeFlags.Item1 == "uint")
{
writer.Write("<u" + typeFlags.Item2 + ">");
}
else if (typeFlags.Item1 == "int")
{
writer.Write("<" + typeFlags.Item2 + ">");
}
}
if (field_sizes[i] != 1)
{
writer.Write("[" + field_sizes[i] + "]");
}
}
writer.Flush();
writer.Close();
Console.Write("..done!\n");
}
if (usedPattern != null && metas.Count > 0)
{
Console.Write("Writing manifest..");
var entries = new List<ManifestEntry>();
foreach (var meta in metas)
{
var entry = new ManifestEntry();
entry.tableName = meta.Key;
if (usedPattern.offsets.ContainsKey(Name.TABLE_HASH))
{
entry.tableHash = meta.Value.table_hash.ToString("X8").ToUpper();
}
else
{
entry.tableHash = MakeTableHash(entry.tableName).ToString("X8").ToUpper();
}
if (usedPattern.offsets.ContainsKey(Name.FDID))
{
entry.db2FileDataID = meta.Value.fileDataID;
}
entries.Add(entry);
}
File.WriteAllText(build + ".json", JsonSerializer.Serialize(entries.OrderBy(x => x.tableName).ToArray(), new JsonSerializerOptions() { WriteIndented = true }));
Console.Write("..done!\n");
}
}
Environment.Exit(0);
}
public static uint MakeTableHash(string tableName)
{
tableName = tableName.ToUpper();
uint[] s_hashtable = new uint[] {
0x486E26EE, 0xDCAA16B3, 0xE1918EEF, 0x202DAFDB,
0x341C7DC7, 0x1C365303, 0x40EF2D37, 0x65FD5E49,
0xD6057177, 0x904ECE93, 0x1C38024F, 0x98FD323B,
0xE3061AE7, 0xA39B0FA1, 0x9797F25F, 0xE4444563,
};
uint v = 0x7fed7fed;
uint x = 0xeeeeeeee;
for (int i = 0; i < tableName.Length; i++)
{
byte c = (byte)tableName[i];
v += x;
v ^= s_hashtable[(c >> 4) & 0xf] - s_hashtable[c & 0xf];
x = x * 33 + v + c + 3;
}
return v;
}
public struct ManifestEntry
{
public string tableName { get; set; }
public string tableHash { get; set; }
[JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingDefault)]
public int dbcFileDataID { get; set; }
[JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingDefault)]
public int db2FileDataID { get; set; }
}
private static List<int> ReadFieldArray(BinaryReader bin, int fieldCount, long offset)
{
var returnList = new List<int>();
if (offset != 0)
{
bin.BaseStream.Position = offset;
for (var i = 0; i < fieldCount; i++)
{
returnList.Add(bin.ReadInt32());
}
}
return returnList;
}
private static List<ulong> ReadFieldOffsetArray(BinaryReader bin, int fieldCount, long offset)
{
var returnList = new List<ulong>();
if (offset != 0)
{
bin.BaseStream.Position = offset;
for (var i = 0; i < fieldCount; i++)
{
returnList.Add(bin.ReadUInt64());
}
}
return returnList;
}
private static string GenerateName(int fieldIndex, string build)
{
// This function should generate a name that is the same between dumps of the same build
return "Field_" + build.Replace(".", "_") + "_" + fieldIndex.ToString().PadLeft(3, '0');
}
private static string CleanRealName(string name)
{
if (name.Length > 2 && name.Substring(0, 2) == "m_")
{
return name.Remove(0, 2);
}
else
{
return name;
}
}
private static DBMeta ReadMeta(BinaryReader bin, Pattern pattern)
{
var matchPos = bin.BaseStream.Position;
var meta = new DBMeta();
foreach (var offset in pattern.offsets)
{
bin.BaseStream.Position = matchPos + offset.Value;
switch (offset.Key)
{
case Name.DB_NAME:
meta.nameOffset = bin.ReadInt64();
break;
case Name.NUM_FIELD_IN_FILE:
meta.num_fields_in_file = bin.ReadInt32();
break;
case Name.RECORD_SIZE:
meta.record_size = bin.ReadInt32();
break;
case Name.NUM_FIELD:
meta.num_fields = bin.ReadInt32();
break;
case Name.ID_COLUMN:
meta.id_column = bin.ReadInt32();
break;
case Name.SPARSE_TABLE:
meta.sparseTable = bin.ReadByte();
break;
case Name.FIELD_OFFSETS:
meta.field_offsets_offs = bin.ReadInt64();
break;
case Name.FIELD_SIZES:
meta.field_sizes_offs = bin.ReadInt64();
break;
case Name.FIELD_TYPES:
meta.field_types_offs = bin.ReadInt64();
break;
case Name.FIELD_FLAGS:
meta.field_flags_offs = bin.ReadInt64();
break;
case Name.FIELD_SIZES_IN_FILE:
meta.field_sizes_in_file_offs = bin.ReadInt64();
break;
case Name.FIELD_TYPES_IN_FILE:
meta.field_types_in_file_offs = bin.ReadInt64();
break;
case Name.FIELD_FLAGS_IN_FILE:
meta.field_flags_in_file_offs = bin.ReadInt64();
break;
case Name.FLAGS_58_21:
meta.flags_58_2_1 = bin.ReadByte();
break;
case Name.TABLE_HASH:
meta.table_hash = bin.ReadInt32();
break;
case Name.LAYOUT_HASH:
meta.layout_hash = bin.ReadInt32();
break;
case Name.FLAGS_68_421:
meta.flags_68_4_2_1 = bin.ReadByte();
break;
case Name.FIELD_NUM_IDX_INT:
meta.nbUniqueIdxByInt = bin.ReadInt32();
break;
case Name.FIELD_NUM_IDX_STRING:
meta.nbUniqueIdxByString = bin.ReadInt32();
break;
case Name.FIELD_IDX_INT:
meta.uniqueIdxByInt = bin.ReadInt32();
break;
case Name.FIELD_IDX_STRING:
meta.uniqueIdxByString = bin.ReadInt32();
break;
case Name.HAS_RELATION:
meta.has_relation = bin.ReadByte();
break;
case Name.FIELD_RELATION:
meta.relation_col = bin.ReadInt32();
break;
case Name.FIELD_RELATION_IN_FILE:
meta.relation_col_in_file = bin.ReadInt32();
break;
case Name.SORT_FUNC:
meta.sortFunctionOffs = bin.ReadInt64();
break;
case Name.UNKC0:
meta.bool_C0 = bin.ReadByte();
break;
case Name.FDID:
meta.fileDataID = bin.ReadInt32();
break;
case Name.DB_CACHE_FILENAME:
meta.cacheNameOffset = bin.ReadInt64();
break;
case Name.CONVERT_STRINGREFS:
meta.string_ref_offs = bin.ReadInt64();
break;
case Name.DB_FILENAME:
meta.dbFilenameOffs = bin.ReadInt64();
break;
case Name.FIELD_ENCRYPTED:
meta.field_encrypted = bin.ReadInt64();
break;
case Name.SQL_QUERY:
meta.sql_query = bin.ReadInt64();
break;
case Name.SIBLING_TABLE_HASH:
meta.siblingTableHash = bin.ReadInt32();
break;
case Name.FIELD_NAMES:
bin.ReadInt64();
break;
case Name.FIELD_NAMES_IN_FILE:
meta.namesInFileOffs = bin.ReadInt64();
break;
case Name.DB_NAME_DUPLICATE:
bin.ReadInt64();
break;
case Name.UNK_BOOL_601_x24:
case Name.UNK_BOOL_601dbc_x38:
case Name.UNK_BOOL_601dbc_x39:
case Name.UNK_BOOL_601dbc_x3a:
case Name.UNK_BOOL_601dbc_x3b:
case Name.UNK_FLAGS_601_x48_421:
case Name.UNK_BOOL_11DB2_x1C:
case Name.UNK_BOOL_11DB2_x1D:
bin.ReadByte();
break;
default:
throw new Exception("Unknown name: " + offset.Key);
}
}
return meta;
}
public enum FieldFlags : int
{
f_maybe_fk = 1,
f_maybe_compressed = 2, // according to simca
f_unsigned = 4,
f_localized = 8,
};
public static (string, int) TypeToT(int type, FieldFlags flag)
{
switch (type)
{
case 0:
switch (flag)
{
case 0 | 0 | 0 | 0:
case 0 | 0 | 0 | FieldFlags.f_maybe_fk:
case 0 | 0 | FieldFlags.f_maybe_compressed | 0:
case 0 | 0 | FieldFlags.f_maybe_compressed | FieldFlags.f_maybe_fk:
return ("int", 32);
case 0 | FieldFlags.f_unsigned | 0 | 0:
case 0 | FieldFlags.f_unsigned | 0 | FieldFlags.f_maybe_fk:
case 0 | FieldFlags.f_unsigned | FieldFlags.f_maybe_compressed | 0:
case 0 | FieldFlags.f_unsigned | FieldFlags.f_maybe_compressed | FieldFlags.f_maybe_fk:
return ("uint", 32);
default:
throw new Exception("Unknown flag combination! type=" + type + ", flag=" + flag);
}
case 1:
switch (flag)
{
case 0 | 0 | 0 | 0:
case 0 | 0 | FieldFlags.f_maybe_compressed | FieldFlags.f_maybe_fk:
return ("int", 64);
case 0 | FieldFlags.f_unsigned | 0 | 0:
case 0 | FieldFlags.f_unsigned | 0 | FieldFlags.f_maybe_fk:
return ("uint", 64);
default:
throw new Exception("Unknown flag combination! type=" + type + ", flag=" + flag);
}
case 2:
switch (flag)
{
case 0 | 0 | 0 | 0:
return ("string", 0);
case FieldFlags.f_localized | 0 | 0 | 0:
return ("locstring", 0);
default:
throw new Exception("Unknown flag combination! type=" + type + ", flag=" + flag);
}
case 3:
switch (flag)
{
case 0 | 0 | 0 | 0:
case 0 | 0 | FieldFlags.f_maybe_compressed | 0:
return ("float", 0);
default:
throw new Exception("Unknown flag combination! type=" + type + ", flag=" + flag);
}
case 4:
switch (flag)
{
case 0 | 0 | 0 | 0:
case 0 | 0 | FieldFlags.f_maybe_compressed | 0:
case 0 | 0 | FieldFlags.f_maybe_compressed | FieldFlags.f_maybe_fk:
return ("int", 8);
case 0 | FieldFlags.f_unsigned | 0 | 0:
case 0 | FieldFlags.f_unsigned | FieldFlags.f_maybe_compressed | 0:
case 0 | FieldFlags.f_unsigned | FieldFlags.f_maybe_compressed | FieldFlags.f_maybe_fk:
return ("uint", 8);
default:
throw new Exception("Unknown flag combination! type=" + type + ", flag=" + flag);
}
case 5:
switch (flag)
{
case 0 | 0 | 0 | 0:
case 0 | 0 | 0 | FieldFlags.f_maybe_fk:
case 0 | 0 | FieldFlags.f_maybe_compressed | 0:
case 0 | 0 | FieldFlags.f_maybe_compressed | FieldFlags.f_maybe_fk:
return ("int", 16);
case 0 | FieldFlags.f_unsigned | 0 | 0:
case 0 | FieldFlags.f_unsigned | 0 | FieldFlags.f_maybe_fk:
case 0 | FieldFlags.f_unsigned | FieldFlags.f_maybe_compressed | 0:
case 0 | FieldFlags.f_unsigned | FieldFlags.f_maybe_compressed | FieldFlags.f_maybe_fk:
return ("uint", 16);
default:
throw new Exception("Unknown flag combination! type=" + type + ", flag=" + flag);
}
default:
throw new Exception("Ran into unknown field type: " + type);
}
}
public static int Search(byte[] haystack, byte?[] needle)
{
int first = 0;
for (; ; ++first)
{
int it = first;
for (int s_it = 0; ; ++it, ++s_it)
{
if (s_it == needle.Length)
{
return first;
}
if (it == haystack.Length)
{
return haystack.Length;
}
if (needle[s_it].HasValue && haystack[it] != needle[s_it].Value)
{
break;
}
}
}
}
public static List<byte?> ParsePattern(string pattern)
{
// Parse pattern
var explodedPattern = pattern.Split(' ');
var patternList = new List<byte?>();
foreach (var part in explodedPattern)
{
if (part == string.Empty)
{
continue;
}
if (part != "?")
{
patternList.Add(Convert.ToByte(part));
}
else
{
patternList.Add(null);
}
}
return patternList;
}
}
#region BinaryReaderExtensions
static class BinaryReaderExtensios
{
/// <summary> Reads the NULL terminated string from
/// the current stream and advances the current position of the stream by string length + 1.
/// <seealso cref="BinaryReader.ReadString"/>
/// </summary>
public static string ReadCString(this BinaryReader reader)
{
return reader.ReadCString(Encoding.UTF8);
}
/// <summary> Reads the NULL terminated string from
/// the current stream and advances the current position of the stream by string length + 1.
/// <seealso cref="BinaryReader.ReadString"/>
/// </summary>
public static string ReadCString(this BinaryReader reader, Encoding encoding)
{
var bytes = new List<byte>();
byte b;
while ((b = reader.ReadByte()) != 0)
bytes.Add(b);
return encoding.GetString(bytes.ToArray());
}
public static T Read<T>(this BinaryReader bin)
{
var bytes = bin.ReadBytes(Marshal.SizeOf(typeof(T)));
var handle = GCHandle.Alloc(bytes, GCHandleType.Pinned);
T ret = (T)Marshal.PtrToStructure(handle.AddrOfPinnedObject(), typeof(T));
handle.Free();
return ret;
}
}
#endregion
}