public CASCHandlerBase(CASCConfig config, BackgroundWorkerEx worker)
{
Config = config;
Logger.WriteLine("CASCHandlerBase: loading CDN indices...");
using (var _ = new PerfCounter("CDNIndexHandler.Initialize()"))
{
CDNIndex = CDNIndexHandler.Initialize(config, worker);
}
Logger.WriteLine("CASCHandlerBase: loaded {0} CDN indexes", CDNIndex.Count);
if (!config.OnlineMode)
{
CDNIndexHandler.Cache.Enabled = false;
Logger.WriteLine("CASCHandlerBase: loading local indices...");
using (var _ = new PerfCounter("LocalIndexHandler.Initialize()"))
{
LocalIndex = LocalIndexHandler.Initialize(config, worker);
}
Logger.WriteLine("CASCHandlerBase: loaded {0} local indexes", LocalIndex.Count);
}
}
private CASCHandlerLite(CASCConfig config, LocaleFlags locale, BackgroundWorkerEx worker) : base(config, worker)
{
if (config.GameType != CASCGameType.WoW)
throw new Exception("Unsupported game " + config.BuildUID);
Logger.WriteLine("CASCHandlerLite: loading encoding data...");
EncodingHandler EncodingHandler;
using (var _ = new PerfCounter("new EncodingHandler()"))
{
using (var fs = OpenEncodingFile(this))
EncodingHandler = new EncodingHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} encoding data", EncodingHandler.Count);
Logger.WriteLine("CASCHandlerLite: loading root data...");
RootHandlerBase RootHandler;
using (var _ = new PerfCounter("new RootHandler()"))
{
using (var fs = OpenRootFile(EncodingHandler, this))
RootHandler = new WowRootHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} root data", RootHandler.Count);
RootHandler.SetFlags(locale, ContentFlags.None, false);
RootEntry rootEntry;
foreach (var entry in RootHandler.GetAllEntries())
{
rootEntry = entry.Value;
if ((rootEntry.Block.LocaleFlags == locale || (rootEntry.Block.LocaleFlags & locale) != LocaleFlags.None) && (rootEntry.Block.ContentFlags & ContentFlags.LowViolence) == ContentFlags.None)
{
var enc = EncodingHandler.GetEntry(rootEntry.MD5);
if (enc != null)
{
if (!HashToKey.ContainsKey(entry.Key))
{
HashToKey.Add(entry.Key, enc.Key);
FileDataIdToHash.Add(rootEntry.FileDataId, entry.Key);
}
}
}
}
RootHandler.Clear();
EncodingHandler.Clear();
RootHandler = null;
EncodingHandler = null;
GC.Collect();
Logger.WriteLine("CASCHandlerLite: loaded {0} files", HashToKey.Count);
}
private static CASCHandlerLite Open(LocaleFlags locale, BackgroundWorkerEx worker, CASCConfig config)
{
using (var _ = new PerfCounter("new CASCHandlerLite()"))
{
return(new CASCHandlerLite(config, locale, worker));
}
}
private bool LoadPreHashedListFile(string pathbin, string pathtext, BackgroundWorkerEx worker = null)
{
using (var _ = new PerfCounter("WowRootHandler::LoadPreHashedListFile()"))
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
if (!File.Exists(pathbin))
{
return(false);
}
var timebin = File.GetLastWriteTime(pathbin);
var timetext = File.GetLastWriteTime(pathtext);
if (timebin != timetext) // text has been modified, recreate crehashed file
{
return(false);
}
Logger.WriteLine("WowRootHandler: loading file names...");
using (var fs = new FileStream(pathbin, FileMode.Open))
using (var br = new BinaryReader(fs))
{
int numFolders = br.ReadInt32();
for (int i = 0; i < numFolders; i++)
{
string dirName = br.ReadString();
Logger.WriteLine(dirName);
int numFiles = br.ReadInt32();
for (int j = 0; j < numFiles; j++)
{
ulong fileHash = br.ReadUInt64();
string fileName = br.ReadString();
string fileNameFull = dirName != String.Empty ? dirName + "\\" + fileName : fileName;
// skip invalid names
if (!RootData.ContainsKey(fileHash))
{
Logger.WriteLine("Invalid file name: {0}", fileNameFull);
continue;
}
CASCFile.Files[fileHash] = new CASCFile(fileHash, fileNameFull);
}
worker?.ReportProgress((int)(fs.Position / (float)fs.Length * 100));
}
Logger.WriteLine("WowRootHandler: loaded {0} valid file names", CASCFile.Files.Count);
}
}
return(true);
}
private static CASCHandler Open(BackgroundWorkerEx worker, CASCConfig config)
{
using (var _ = new PerfCounter("new CASCHandler()"))
{
return(new CASCHandler(config, worker));
}
}
public static LocalIndexHandler Initialize(CASCConfig config, BackgroundWorkerEx worker)
{
var handler = new LocalIndexHandler();
var idxFiles = GetIdxFiles(config);
if (idxFiles.Count == 0)
{
throw new FileNotFoundException("idx files missing!");
}
worker?.ReportProgress(0, "Loading \"local indexes\"...");
int idxIndex = 0;
foreach (var idx in idxFiles)
{
handler.ParseIndex(idx);
worker?.ReportProgress((int)(++idxIndex / (float)idxFiles.Count * 100));
}
Logger.WriteLine("LocalIndexHandler: loaded {0} indexes", handler.Count);
return(handler);
}
public CASCHandlerBase(CASCConfig config, BackgroundWorkerEx worker)
{
Config = config;
Logger.WriteLine("CASCHandlerBase: loading CDN indices...");
using (var _ = new PerfCounter("CDNIndexHandler.Initialize()"))
{
CDNIndex = CDNIndexHandler.Initialize(config, worker);
}
Logger.WriteLine("CASCHandlerBase: loaded {0} CDN indexes", CDNIndex.Count);
if (!config.OnlineMode)
{
CDNIndexHandler.Cache.Enabled = false;
Logger.WriteLine("CASCHandlerBase: loading local indices...");
using (var _ = new PerfCounter("LocalIndexHandler.Initialize()"))
{
LocalIndex = LocalIndexHandler.Initialize(config, worker);
}
Logger.WriteLine("CASCHandlerBase: loaded {0} local indexes", LocalIndex.Count);
}
}
public HSRootHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"root\"...");
// Hearthstone root file happened to be game executable! Just ignore it.
worker?.ReportProgress(100);
}
public AgentRootHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"root\"...");
string hash = Encoding.ASCII.GetString(stream.ReadBytes((int)stream.BaseStream.Length)); // what is this for?
worker?.ReportProgress(100);
}
public HSRootHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"root\"...");
// Hearthstone root file happened to be game executable! Just ignore it.
worker?.ReportProgress(100);
}
public AgentRootHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"root\"...");
string hash = Encoding.ASCII.GetString(stream.ReadBytes((int)stream.BaseStream.Length)); // what is this for?
worker?.ReportProgress(100);
}
public void Clear()
{
CDNIndexData.Clear();
CDNIndexData = null;
config = null;
worker = null;
downloader = null;
}
public OwRootHandler(BinaryReader stream, BackgroundWorkerEx worker, CASCHandler casc)
{
worker?.ReportProgress(0, "Loading \"root\"...");
string str = Encoding.ASCII.GetString(stream.ReadBytes((int)stream.BaseStream.Length));
string[] array = str.Split(new[] { '\r', '\n' }, StringSplitOptions.RemoveEmptyEntries);
List <string> APMNames = new List <string>();
for (int i = 1; i < array.Length; i++)
{
string[] filedata = array[i].Split('|');
string name = filedata[4];
if (Path.GetExtension(name) == ".apm" && name.Contains("RDEV"))
{
APMNames.Add(Path.GetFileNameWithoutExtension(name));
if (!name.Contains("L" + LanguageScan))
{
continue;
}
// add apm file for dev purposes
ulong apmNameHash = Hasher.ComputeHash(name);
MD5Hash apmMD5 = filedata[0].ToByteArray().ToMD5();
_rootData[apmNameHash] = new OWRootEntry()
{
baseEntry = new RootEntry()
{
MD5 = apmMD5, LocaleFlags = LocaleFlags.All, ContentFlags = ContentFlags.None
}
};
CASCFile.FileNames[apmNameHash] = name;
EncodingEntry apmEnc;
if (!casc.Encoding.GetEntry(apmMD5, out apmEnc))
{
continue;
}
using (Stream apmStream = casc.OpenFile(apmEnc.Key))
{
apmFiles.Add(new APMFile(name, apmStream, casc));
}
}
worker?.ReportProgress((int)(i / (array.Length / 100f)));
}
APMList = APMNames.ToArray();
APMNames.Clear();
}
public DownloadHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"download\"...");
stream.Skip(2); // DL
byte b1 = stream.ReadByte();
byte b2 = stream.ReadByte();
byte b3 = stream.ReadByte();
int numFiles = stream.ReadInt32BE();
short numTags = stream.ReadInt16BE();
int numMaskBytes = (numFiles + 7) / 8;
for (int i = 0; i < numFiles; i++)
{
MD5Hash key = stream.Read <MD5Hash>();
//byte[] unk = stream.ReadBytes(0xA);
stream.Skip(0xA);
//var entry = new DownloadEntry() { Index = i, Unk = unk };
var entry = new DownloadEntry()
{
Index = i
};
DownloadData.Add(key, entry);
worker?.ReportProgress((int)((i + 1) / (float)numFiles * 100));
}
for (int i = 0; i < numTags; i++)
{
DownloadTag tag = new DownloadTag();
string name = stream.ReadCString();
tag.Type = stream.ReadInt16BE();
byte[] bits = stream.ReadBytes(numMaskBytes);
for (int j = 0; j < numMaskBytes; j++)
{
bits[j] = (byte)((bits[j] * 0x0202020202 & 0x010884422010) % 1023);
}
tag.Bits = new BitArray(bits);
Tags.Add(name, tag);
}
}
public InstallHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"install\"...");
stream.ReadBytes(2); // IN
byte b1 = stream.ReadByte();
byte b2 = stream.ReadByte();
short numTags = stream.ReadInt16BE();
int numFiles = stream.ReadInt32BE();
int numMaskBytes = (numFiles + 7) / 8;
List <InstallTag> Tags = new List <InstallTag>();
for (int i = 0; i < numTags; i++)
{
InstallTag tag = new InstallTag()
{
Name = stream.ReadCString(),
Type = stream.ReadInt16BE()
};
byte[] bits = stream.ReadBytes(numMaskBytes);
for (int j = 0; j < numMaskBytes; j++)
{
bits[j] = (byte)((bits[j] * 0x0202020202 & 0x010884422010) % 1023);
}
tag.Bits = new BitArray(bits);
Tags.Add(tag);
}
for (int i = 0; i < numFiles; i++)
{
InstallEntry entry = new InstallEntry()
{
Name = stream.ReadCString(),
MD5 = stream.Read <MD5Hash>(),
Size = stream.ReadInt32BE()
};
InstallData.Add(entry);
entry.Tags = Tags.FindAll(tag => tag.Bits[i]);
worker?.ReportProgress((int)((i + 1) / (float)numFiles * 100));
}
}
public DownloadHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"download\"...");
stream.Skip(2); // DL
byte b1 = stream.ReadByte();
byte b2 = stream.ReadByte();
byte b3 = stream.ReadByte();
int numFiles = stream.ReadInt32BE();
short numTags = stream.ReadInt16BE();
int numMaskBytes = (numFiles + 7) / 8;
for (int i = 0; i < numFiles; i++)
{
MD5Hash key = stream.Read<MD5Hash>();
//byte[] unk = stream.ReadBytes(0xA);
stream.Skip(0xA);
//var entry = new DownloadEntry() { Index = i, Unk = unk };
var entry = new DownloadEntry() { Index = i };
DownloadData.Add(key, entry);
worker?.ReportProgress((int)((i + 1) / (float)numFiles * 100));
}
for (int i = 0; i < numTags; i++)
{
DownloadTag tag = new DownloadTag();
string name = stream.ReadCString();
tag.Type = stream.ReadInt16BE();
byte[] bits = stream.ReadBytes(numMaskBytes);
for (int j = 0; j < numMaskBytes; j++)
bits[j] = (byte)((bits[j] * 0x0202020202 & 0x010884422010) % 1023);
tag.Bits = new BitArray(bits);
Tags.Add(name, tag);
}
}
public InstallHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"install\"...");
stream.ReadBytes(2); // IN
byte b1 = stream.ReadByte();
byte b2 = stream.ReadByte();
short numTags = stream.ReadInt16BE();
int numFiles = stream.ReadInt32BE();
int numMaskBytes = (numFiles + 7) / 8;
List<InstallTag> Tags = new List<InstallTag>();
for (int i = 0; i < numTags; i++)
{
InstallTag tag = new InstallTag();
tag.Name = stream.ReadCString();
tag.Type = stream.ReadInt16BE();
byte[] bits = stream.ReadBytes(numMaskBytes);
for (int j = 0; j < numMaskBytes; j++)
bits[j] = (byte)((bits[j] * 0x0202020202 & 0x010884422010) % 1023);
tag.Bits = new BitArray(bits);
Tags.Add(tag);
}
for (int i = 0; i < numFiles; i++)
{
InstallEntry entry = new InstallEntry();
entry.Name = stream.ReadCString();
entry.MD5 = stream.ReadBytes(16);
entry.Size = stream.ReadInt32BE();
InstallData.Add(entry);
entry.Tags = Tags.FindAll(tag => tag.Bits[i]);
worker?.ReportProgress((int)((i + 1) / (float)numFiles * 100));
}
}
public S1RootHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"root\"...");
using (StreamReader sr = new StreamReader(stream.BaseStream))
{
string line;
while ((line = sr.ReadLine()) != null)
{
string[] tokens = line.Split('|');
string file;
LocaleFlags locale = LocaleFlags.All;
if (tokens[0].IndexOf(':') != -1)
{
string[] tokens2 = tokens[0].Split(':');
file = tokens2[0];
locale = (LocaleFlags)Enum.Parse(typeof(LocaleFlags), tokens2[1]);
}
else
{
file = tokens[0];
}
ulong fileHash = Hasher.ComputeHash(file);
RootData[fileHash] = new RootEntry()
{
LocaleFlags = locale,
ContentFlags = ContentFlags.None,
MD5 = tokens[1].ToByteArray().ToMD5()
};
CASCFile.FileNames[fileHash] = file;
}
}
worker?.ReportProgress(100);
}
public static CDNIndexHandler Initialize(CASCConfig config, BackgroundWorkerEx worker)
{
var handler = new CDNIndexHandler(config, worker);
worker?.ReportProgress(0, "Loading \"CDN indexes\"...");
for (int i = 0; i < config.Archives.Count; i++)
{
string archive = config.Archives[i];
if (config.OnlineMode)
handler.DownloadIndexFile(archive, i);
else
handler.OpenIndexFile(archive, i);
worker?.ReportProgress((int)((i + 1) / (float)config.Archives.Count * 100));
}
return handler;
}
public static void InitCasc(BackgroundWorkerEx worker = null, string basedir = null, string program = "wowt"){
CASCConfig.LoadFlags &= ~(LoadFlags.Download | LoadFlags.Install);
if (basedir == null)
{
CASCConfig config = CASCConfig.LoadOnlineStorageConfig(program, "us", true);
Console.WriteLine("Initializing CASC from web with program " + program + " and build " + config.BuildName);
cascHandler = CASCHandler.OpenStorage(config, worker);
}
else
{
Console.WriteLine("Initializing CASC from local disk with basedir " + basedir);
cascHandler = CASCHandler.OpenLocalStorage(basedir, worker);
}
cascHandler.Root.SetFlags(LocaleFlags.enUS, ContentFlags.None, false);
fIsCASCInit = true;
}
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
Logger.WriteLine("D3RootHandler: loading file names...");
int numFiles = D3RootData.Sum(p => p.Value.Count);
int i = 0;
foreach (var kv in D3RootData)
{
foreach (var e in kv.Value)
{
AddFile(kv.Key, e);
worker?.ReportProgress((int)(++i / (float)numFiles * 100));
}
}
Logger.WriteLine("D3RootHandler: loaded {0} file names", i);
}
public static CDNIndexHandler Initialize(CASCConfig config, BackgroundWorkerEx worker)
{
var handler = new CDNIndexHandler(config, worker);
worker?.ReportProgress(0, "Loading \"CDN indexes\"...");
for (int i = 0; i < config.Archives.Count; i++)
{
string archive = config.Archives[i];
if (config.OnlineMode)
{
handler.DownloadIndexFile(archive, i);
}
else
{
handler.OpenIndexFile(archive, i);
}
worker?.ReportProgress((int)((i + 1) / (float)config.Archives.Count * 100));
}
return(handler);
}
public static LocalIndexHandler Initialize(CASCConfig config, BackgroundWorkerEx worker)
{
var handler = new LocalIndexHandler();
var idxFiles = GetIdxFiles(config);
if (idxFiles.Count == 0)
throw new FileNotFoundException("idx files missing!");
worker?.ReportProgress(0, "Loading \"local indexes\"...");
int idxIndex = 0;
foreach (var idx in idxFiles)
{
handler.ParseIndex(idx);
worker?.ReportProgress((int)(++idxIndex / (float)idxFiles.Count * 100));
}
Logger.WriteLine("LocalIndexHandler: loaded {0} indexes", handler.Count);
return handler;
}
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
Logger.WriteLine("D3RootHandler: loading file names...");
int numFiles = D3RootData.Sum(p => p.Value.Count);
int i = 0;
foreach (var kv in D3RootData)
{
foreach (var e in kv.Value)
{
AddFile(kv.Key, e);
worker?.ReportProgress((int)(++i / (float)numFiles * 100));
}
}
Logger.WriteLine("D3RootHandler: loaded {0} file names", i);
}
public static CASCHandler OpenLocalStorage(string basePath, BackgroundWorkerEx worker = null)
{
CASCConfig config = CASCConfig.LoadLocalStorageConfig(basePath);
return(Open(worker, config));
}
public WowRootHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"root\"...");
while (stream.BaseStream.Position < stream.BaseStream.Length)
{
int count = stream.ReadInt32();
ContentFlags contentFlags = (ContentFlags)stream.ReadUInt32();
LocaleFlags localeFlags = (LocaleFlags)stream.ReadUInt32();
if (localeFlags == LocaleFlags.None)
{
throw new Exception("block.LocaleFlags == LocaleFlags.None");
}
if (contentFlags != ContentFlags.None && (contentFlags & (ContentFlags.F00000008 | ContentFlags.F00000010 | ContentFlags.LowViolence | ContentFlags.NoCompression | ContentFlags.F20000000)) == 0)
{
throw new Exception("block.ContentFlags != ContentFlags.None");
}
RootEntry[] entries = new RootEntry[count];
int[] filedataIds = new int[count];
int fileDataIndex = 0;
for (var i = 0; i < count; ++i)
{
entries[i].LocaleFlags = localeFlags;
entries[i].ContentFlags = contentFlags;
filedataIds[i] = fileDataIndex + stream.ReadInt32();
fileDataIndex = filedataIds[i] + 1;
}
//Console.WriteLine("Block: {0} {1} (size {2})", block.ContentFlags, block.LocaleFlags, count);
for (var i = 0; i < count; ++i)
{
entries[i].MD5 = stream.Read <MD5Hash>();
ulong hash = stream.ReadUInt64();
RootData.Add(hash, entries[i]);
//Console.WriteLine("File: {0:X8} {1:X16} {2}", entries[i].FileDataId, hash, entries[i].MD5.ToHexString());
int fileDataId = filedataIds[i];
if (FileDataStore.TryGetValue(fileDataId, out ulong hash2))
{
if (hash2 == hash)
{
// duplicate, skipping
continue;
}
else
{
Logger.WriteLine("ERROR: got miltiple hashes for filedataid {0}", fileDataId);
continue;
}
}
FileDataStore.Add(fileDataId, hash);
FileDataStoreReverse.Add(hash, fileDataId);
}
worker?.ReportProgress((int)(stream.BaseStream.Position / (float)stream.BaseStream.Length * 100));
}
}
public EncodingHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"encoding\"...");
stream.Skip(2); // EN
byte b1 = stream.ReadByte();
byte checksumSizeA = stream.ReadByte();
byte checksumSizeB = stream.ReadByte();
ushort flagsA = stream.ReadUInt16();
ushort flagsB = stream.ReadUInt16();
int numEntriesA = stream.ReadInt32BE();
int numEntriesB = stream.ReadInt32BE();
byte b4 = stream.ReadByte();
int stringBlockSize = stream.ReadInt32BE();
stream.Skip(stringBlockSize);
//string[] strings = Encoding.ASCII.GetString(stream.ReadBytes(stringBlockSize)).Split(new[] { '\0' }, StringSplitOptions.RemoveEmptyEntries);
stream.Skip(numEntriesA * 32);
//for (int i = 0; i < numEntriesA; ++i)
//{
// byte[] firstHash = stream.ReadBytes(16);
// byte[] blockHash = stream.ReadBytes(16);
//}
long chunkStart = stream.BaseStream.Position;
for (int i = 0; i < numEntriesA; ++i)
{
ushort keysCount;
while ((keysCount = stream.ReadUInt16()) != 0)
{
int fileSize = stream.ReadInt32BE();
byte[] md5 = stream.ReadBytes(16);
EncodingEntry entry = new EncodingEntry();
entry.Size = fileSize;
// how do we handle multiple keys?
for (int ki = 0; ki < keysCount; ++ki)
{
byte[] key = stream.ReadBytes(16);
// use first key for now
if (ki == 0)
{
entry.Key = key;
}
else
{
Logger.WriteLine("Multiple encoding keys for MD5 {0}: {1}", md5.ToHexString(), key.ToHexString());
}
}
//Encodings[md5] = entry;
EncodingData.Add(md5, entry);
}
// each chunk is 4096 bytes, and zero padding at the end
long remaining = CHUNK_SIZE - ((stream.BaseStream.Position - chunkStart) % CHUNK_SIZE);
if (remaining > 0)
{
stream.BaseStream.Position += remaining;
}
worker?.ReportProgress((int)((i + 1) / (float)numEntriesA * 100));
}
stream.Skip(numEntriesB * 32);
//for (int i = 0; i < numEntriesB; ++i)
//{
// byte[] firstKey = stream.ReadBytes(16);
// byte[] blockHash = stream.ReadBytes(16);
//}
long chunkStart2 = stream.BaseStream.Position;
for (int i = 0; i < numEntriesB; ++i)
{
byte[] key = stream.ReadBytes(16);
int stringIndex = stream.ReadInt32BE();
byte unk1 = stream.ReadByte();
int fileSize = stream.ReadInt32BE();
// each chunk is 4096 bytes, and zero padding at the end
long remaining = CHUNK_SIZE - ((stream.BaseStream.Position - chunkStart2) % CHUNK_SIZE);
if (remaining > 0)
{
stream.BaseStream.Position += remaining;
}
}
// string block till the end of file
}
public static CASCHandler OpenLocalStorage(string basePath, BackgroundWorkerEx worker = null)
{
CASCConfig config = CASCConfig.LoadLocalStorageConfig(basePath);
return Open(worker, config);
}
public OWRootHandler(BinaryReader stream, BackgroundWorkerEx worker, CASCHandler casc)
{
worker?.ReportProgress(0, "Loading \"root\"...");
string str = Encoding.ASCII.GetString(stream.ReadBytes((int)stream.BaseStream.Length));
string[] array = str.Split(new[] { '\r', '\n' }, StringSplitOptions.RemoveEmptyEntries);
// need to figure out what to do with those apm files
for (int i = 1; i < array.Length; i++)
{
string[] filedata = array[i].Split('|');
if (Path.GetExtension(filedata[2]) == ".apm")
{
// add apm file for dev purposes
ulong fileHash1 = Hasher.ComputeHash(filedata[2]);
RootData[fileHash1] = new RootEntry() { MD5 = filedata[0].ToByteArray(), Block = RootBlock.Empty };
CASCFile.FileNames[fileHash1] = filedata[2];
// add files listed in apm file
byte[] md5 = filedata[0].ToByteArray();
EncodingEntry enc = casc.Encoding.GetEntry(md5);
using (BinaryReader s = new BinaryReader(casc.OpenFile(enc.Key)))
{
if (s != null)
{
// still need to figure out complete apm structure
// at start of file there's a lot of data that is same in all apm files
s.BaseStream.Position = 0xC;
uint count = s.ReadUInt32();
s.BaseStream.Position = 0x894;
// size of each entry seems to be 0x48 bytes (0x2C bytes unk data; int size; ulong unk; byte[16] md5)
for (int j = 0; j < count; j++)
{
s.BaseStream.Position += 0x2C; // skip unknown
int size = s.ReadInt32(); // size (matches size in encoding file)
s.BaseStream.Position += 8; // skip unknown
byte[] md5_2 = s.ReadBytes(16);
EncodingEntry enc2 = casc.Encoding.GetEntry(md5_2);
if (enc2 == null)
{
throw new Exception("enc2 == null");
}
string fakeName = Path.GetFileNameWithoutExtension(filedata[2]) + "/" + md5_2.ToHexString();
ulong fileHash = Hasher.ComputeHash(fakeName);
RootData[fileHash] = new RootEntry() { MD5 = md5_2, Block = RootBlock.Empty };
CASCFile.FileNames[fileHash] = fakeName;
}
}
}
}
}
int current = 0;
Func<string, LocaleFlags> tag2locale = (s) =>
{
LocaleFlags locale;
if (Enum.TryParse(s, out locale))
return locale;
return LocaleFlags.All;
};
foreach (var entry in casc.Encoding.Entries)
{
DownloadEntry dl = casc.Download.GetEntry(entry.Value.Key);
if (dl != null)
{
string fakeName = "unknown" + "/" + entry.Key[0].ToString("X2") + "/" + entry.Key.ToHexString();
var locales = dl.Tags.Where(tag => tag.Value.Type == 4).Select(tag => tag2locale(tag.Key));
LocaleFlags locale = LocaleFlags.None;
foreach (var loc in locales)
locale |= loc;
ulong fileHash = Hasher.ComputeHash(fakeName);
RootData.Add(fileHash, new RootEntry() { MD5 = entry.Key, Block = new RootBlock() { LocaleFlags = locale } });
CASCFile.FileNames[fileHash] = fakeName;
}
worker?.ReportProgress((int)(++current / (float)casc.Encoding.Count * 100));
}
}
static void Main(string[] args)
{
if (args.Length != 4)
{
Console.WriteLine("Invalid arguments count!");
Console.WriteLine("Usage: CASCConsole <pattern> <destination> <localeFlags> <contentFlags>");
return;
}
Console.WriteLine("Settings:");
Console.WriteLine(" WowPath: {0}", Settings.Default.StoragePath);
Console.WriteLine(" OnlineMode: {0}", Settings.Default.OnlineMode);
Console.WriteLine("Loading...");
BackgroundWorkerEx bgLoader = new BackgroundWorkerEx();
bgLoader.ProgressChanged += BgLoader_ProgressChanged;
CASCConfig config = Settings.Default.OnlineMode
? CASCConfig.LoadOnlineStorageConfig(Settings.Default.Product, "us")
: CASCConfig.LoadLocalStorageConfig(Settings.Default.StoragePath);
CASCHandler cascHandler = CASCHandler.OpenStorage(config, bgLoader);
string pattern = args[0];
string dest = args[1];
LocaleFlags locale = (LocaleFlags)Enum.Parse(typeof(LocaleFlags), args[2]);
ContentFlags content = (ContentFlags)Enum.Parse(typeof(ContentFlags), args[3]);
cascHandler.Root.LoadListFile(Path.Combine(Environment.CurrentDirectory, "listfile.txt"), bgLoader);
CASCFolder root = cascHandler.Root.SetFlags(locale, content);
Console.WriteLine("Loaded.");
Console.WriteLine("Extract params:", pattern, dest, locale);
Console.WriteLine(" Pattern: {0}", pattern);
Console.WriteLine(" Destination: {0}", dest);
Console.WriteLine(" LocaleFlags: {0}", locale);
Console.WriteLine(" ContentFlags: {0}", content);
Wildcard wildcard = new Wildcard(pattern, true, RegexOptions.IgnoreCase);
foreach (var file in root.GetFiles())
{
if (wildcard.IsMatch(file.FullName))
{
Console.Write("Extracting '{0}'...", file.FullName);
try
{
cascHandler.SaveFileTo(file.FullName, dest);
Console.WriteLine(" Ok!");
}
catch (Exception exc)
{
Console.WriteLine(" Error!");
Logger.WriteLine(exc.Message);
}
}
}
Console.WriteLine("Extracted.");
}
private CASCHandler(CASCConfig config, BackgroundWorkerEx worker) : base(config, worker)
{
Logger.WriteLine("CASCHandler: loading encoding data...");
using (var _ = new PerfCounter("new EncodingHandler()"))
{
using (var fs = OpenEncodingFile(this))
EncodingHandler = new EncodingHandler(fs, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} encoding data", EncodingHandler.Count);
if ((CASCConfig.LoadFlags & LoadFlags.Download) != 0)
{
Logger.WriteLine("CASCHandler: loading download data...");
using (var _ = new PerfCounter("new DownloadHandler()"))
{
using (var fs = OpenDownloadFile(EncodingHandler, this))
DownloadHandler = new DownloadHandler(fs, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} download data", EncodingHandler.Count);
}
Logger.WriteLine("CASCHandler: loading root data...");
using (var _ = new PerfCounter("new RootHandler()"))
{
using (var fs = OpenRootFile(EncodingHandler, this))
{
if (config.GameType == CASCGameType.S2 || config.GameType == CASCGameType.HotS)
RootHandler = new MNDXRootHandler(fs, worker);
else if (config.GameType == CASCGameType.D3)
RootHandler = new D3RootHandler(fs, worker, this);
else if (config.GameType == CASCGameType.WoW)
RootHandler = new WowRootHandler(fs, worker);
else if (config.GameType == CASCGameType.Agent || config.GameType == CASCGameType.Bna || config.GameType == CASCGameType.Client)
RootHandler = new AgentRootHandler(fs, worker);
else if (config.GameType == CASCGameType.Hearthstone)
RootHandler = new HSRootHandler(fs, worker);
else if (config.GameType == CASCGameType.Overwatch)
RootHandler = new OwRootHandler(fs, worker, this);
else
throw new Exception("Unsupported game " + config.BuildUID);
}
}
Logger.WriteLine("CASCHandler: loaded {0} root data", RootHandler.Count);
if ((CASCConfig.LoadFlags & LoadFlags.Install) != 0)
{
Logger.WriteLine("CASCHandler: loading install data...");
using (var _ = new PerfCounter("new InstallHandler()"))
{
using (var fs = OpenInstallFile(EncodingHandler, this))
InstallHandler = new InstallHandler(fs, worker);
InstallHandler.Print();
}
Logger.WriteLine("CASCHandler: loaded {0} install data", InstallHandler.Count);
}
}
public SyncDownloader(BackgroundWorkerEx progressReporter)
{
this.progressReporter = progressReporter;
}
public D3RootHandler(BinaryReader stream, BackgroundWorkerEx worker, CASCHandler casc)
{
worker?.ReportProgress(0, "Loading \"root\"...");
byte b1 = stream.ReadByte();
byte b2 = stream.ReadByte();
byte b3 = stream.ReadByte();
byte b4 = stream.ReadByte();
int count = stream.ReadInt32();
for (int j = 0; j < count; j++)
{
byte[] md5 = stream.ReadBytes(16);
string name = stream.ReadCString();
var entries = new List <D3RootEntry>();
D3RootData[name] = entries;
EncodingEntry enc = casc.Encoding.GetEntry(md5);
using (BinaryReader s = new BinaryReader(casc.OpenFile(enc.Key)))
{
if (s != null)
{
uint magic = s.ReadUInt32();
int nEntries0 = s.ReadInt32();
for (int i = 0; i < nEntries0; i++)
{
entries.Add(D3RootEntry.Read(0, s));
}
int nEntries1 = s.ReadInt32();
for (int i = 0; i < nEntries1; i++)
{
entries.Add(D3RootEntry.Read(1, s));
}
int nNamedEntries = s.ReadInt32();
for (int i = 0; i < nNamedEntries; i++)
{
entries.Add(D3RootEntry.Read(2, s));
}
}
}
worker?.ReportProgress((int)((j + 1) / (float)(count + 2) * 100));
}
// Parse CoreTOC.dat
var coreTocEntry = D3RootData["Base"].Find(e => e.Name == "CoreTOC.dat");
EncodingEntry enc1 = casc.Encoding.GetEntry(coreTocEntry.MD5);
using (var file = casc.OpenFile(enc1.Key))
tocParser = new CoreTOCParser(file);
worker?.ReportProgress((int)((count + 1) / (float)(count + 2) * 100));
// Parse Packages.dat
var pkgEntry = D3RootData["Base"].Find(e => e.Name == "Data_D3\\PC\\Misc\\Packages.dat");
EncodingEntry enc2 = casc.Encoding.GetEntry(pkgEntry.MD5);
using (var file = casc.OpenFile(enc2.Key))
pkgParser = new PackagesParser(file);
worker?.ReportProgress(100);
}
public abstract void LoadListFile(string path, BackgroundWorkerEx worker = null);
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
Logger.WriteLine("OWRootHandler: loading file names...");
float pkgOnePct = apmFiles.Sum(a => a.Packages.Length) / 100f;
int pkgCount = 0;
foreach (var apm in apmFiles)
{
for (int i = 0; i < apm.Packages.Length; i++)
{
APMPackage package = apm.Packages[i];
MD5Hash pkgIndexMD5 = package.indexContentKey;
string apmName = Path.GetFileNameWithoutExtension(apm.Name);
string pkgName = string.Format("{0}/package_{1:X4}_{2:X16}", apmName, i, package.packageKey);
string fakeName = string.Format("{0}_index", pkgName);
ulong fileHash = Hasher.ComputeHash(fakeName);
Logger.WriteLine("Adding package: {0:X16} {1}", fileHash, package.indexContentKey.ToHexString());
if (_rootData.ContainsKey(fileHash))
{
if (!_rootData[fileHash].baseEntry.MD5.EqualsTo(package.indexContentKey))
Logger.WriteLine("Weird duplicate package: {0:X16} {1}", fileHash, package.indexContentKey.ToHexString());
else
Logger.WriteLine("Duplicate package: {0:X16} {1}", fileHash, package.indexContentKey.ToHexString());
continue;
}
_rootData[fileHash] = new OWRootEntry()
{
baseEntry = new RootEntry() { MD5 = pkgIndexMD5, LocaleFlags = LocaleFlags.All, ContentFlags = ContentFlags.None }
};
CASCFile.FileNames[fileHash] = fakeName;
PackageIndex pkgIndex = apm.Indexes[i];
fakeName = string.Format("{0}_bundle_{1:X16}", pkgName, pkgIndex.bundleKey);
fileHash = Hasher.ComputeHash(fakeName);
Logger.WriteLine("Adding bundle: {0:X16} {1}", fileHash, pkgIndex.bundleContentKey.ToHexString());
if (_rootData.ContainsKey(fileHash))
{
if (!_rootData[fileHash].baseEntry.MD5.EqualsTo(pkgIndex.bundleContentKey))
Logger.WriteLine("Weird duplicate bundle: {0:X16} {1}", fileHash, pkgIndex.bundleContentKey.ToHexString());
else
Logger.WriteLine("Duplicate bundle: {0:X16} {1}", fileHash, pkgIndex.bundleContentKey.ToHexString());
continue;
}
_rootData[fileHash] = new OWRootEntry()
{
baseEntry = new RootEntry() { MD5 = pkgIndex.bundleContentKey, LocaleFlags = LocaleFlags.All, ContentFlags = ContentFlags.None },
pkgIndex = pkgIndex
};
CASCFile.FileNames[fileHash] = fakeName;
PackageIndexRecord[] records = apm.Records[i];
for (int k = 0; k < records.Length; k++)
{
fakeName = string.Format("files/{0:X3}/{1:X12}.{0:X3}", keyToTypeID(records[k].Key), records[k].Key & 0xFFFFFFFFFFFF);
fileHash = Hasher.ComputeHash(fakeName);
//Logger.WriteLine("Adding package record: key {0:X16} hash {1} flags {2:X8}", fileHash, records[k].contentKey.ToHexString(), records[k].flags);
if (_rootData.ContainsKey(fileHash))
{
if (!_rootData[fileHash].baseEntry.MD5.EqualsTo(records[k].ContentKey))
Logger.WriteLine("Weird duplicate package record: {0:X16} {1}", fileHash, records[k].ContentKey.ToHexString());
//else
// Logger.WriteLine("Duplicate package record: {0:X16} {1}", fileHash, records[k].contentKey.ToHexString());
continue;
}
_rootData[fileHash] = new OWRootEntry()
{
baseEntry = new RootEntry() { MD5 = records[k].ContentKey, LocaleFlags = LocaleFlags.All, ContentFlags = (ContentFlags)records[k].Flags },
pkgIndex = pkgIndex,
pkgIndexRec = records[k]
};
CASCFile.FileNames[fileHash] = fakeName;
}
worker?.ReportProgress((int)(++pkgCount / pkgOnePct));
}
}
Logger.WriteLine("OWRootHandler: loaded {0} file names", _rootData.Count);
}
public OwRootHandler(BinaryReader stream, BackgroundWorkerEx worker, CASCHandler casc)
{
worker?.ReportProgress(0, "Loading \"root\"...");
string str = Encoding.ASCII.GetString(stream.ReadBytes((int)stream.BaseStream.Length));
string[] array = str.Split(new[] { '\r', '\n' }, StringSplitOptions.RemoveEmptyEntries);
List<string> APMNames = new List<string>();
for (int i = 1; i < array.Length; i++)
{
string[] filedata = array[i].Split('|');
string name = filedata[4];
if (Path.GetExtension(name) == ".apm" && name.Contains("RDEV"))
{
APMNames.Add(Path.GetFileNameWithoutExtension(name));
if(!name.Contains("L"+LanguageScan)) {
continue;
}
// add apm file for dev purposes
ulong apmNameHash = Hasher.ComputeHash(name);
MD5Hash apmMD5 = filedata[0].ToByteArray().ToMD5();
_rootData[apmNameHash] = new OWRootEntry()
{
baseEntry = new RootEntry() { MD5 = apmMD5, LocaleFlags = LocaleFlags.All, ContentFlags = ContentFlags.None }
};
CASCFile.FileNames[apmNameHash] = name;
EncodingEntry apmEnc;
if (!casc.Encoding.GetEntry(apmMD5, out apmEnc))
continue;
using (Stream apmStream = casc.OpenFile(apmEnc.Key))
{
apmFiles.Add(new APMFile(name, apmStream, casc));
}
}
worker?.ReportProgress((int)(i / (array.Length / 100f)));
}
APMList = APMNames.ToArray();
APMNames.Clear();
}
private CDNIndexHandler(CASCConfig cascConfig, BackgroundWorkerEx worker)
{
config = cascConfig;
this.worker = worker;
downloader = new SyncDownloader(worker);
}
public SyncDownloader(BackgroundWorkerEx progressReporter)
{
this.progressReporter = progressReporter;
}
private static CASCHandler Open(BackgroundWorkerEx worker, CASCConfig config)
{
using (var _ = new PerfCounter("new CASCHandler()"))
{
return new CASCHandler(config, worker);
}
}
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
}
public static CASCHandler OpenStorage(CASCConfig config, BackgroundWorkerEx worker = null) => Open(worker, config);
public OWRootHandler(BinaryReader stream, BackgroundWorkerEx worker, CASCHandler casc)
{
worker?.ReportProgress(0, "Loading \"root\"...");
string str = Encoding.ASCII.GetString(stream.ReadBytes((int)stream.BaseStream.Length));
string[] array = str.Split(new[] { '\r', '\n' }, StringSplitOptions.RemoveEmptyEntries);
// need to figure out what to do with those apm files
for (int i = 1; i < array.Length; i++)
{
string[] filedata = array[i].Split('|');
if (Path.GetExtension(filedata[2]) == ".apm")
{
// add apm file for dev purposes
ulong fileHash1 = Hasher.ComputeHash(filedata[2]);
RootData[fileHash1] = new RootEntry()
{
MD5 = filedata[0].ToByteArray(), Block = RootBlock.Empty
};
CASCFile.FileNames[fileHash1] = filedata[2];
// add files listed in apm file
byte[] md5 = filedata[0].ToByteArray();
EncodingEntry enc = casc.Encoding.GetEntry(md5);
using (BinaryReader s = new BinaryReader(casc.OpenFile(enc.Key)))
{
if (s != null)
{
// still need to figure out complete apm structure
// at start of file there's a lot of data that is same in all apm files
s.BaseStream.Position = 0xC;
uint count = s.ReadUInt32();
s.BaseStream.Position = 0x894;
// size of each entry seems to be 0x48 bytes (0x2C bytes unk data; int size; ulong unk; byte[16] md5)
for (int j = 0; j < count; j++)
{
s.BaseStream.Position += 0x2C; // skip unknown
int size = s.ReadInt32(); // size (matches size in encoding file)
s.BaseStream.Position += 8; // skip unknown
byte[] md5_2 = s.ReadBytes(16);
EncodingEntry enc2 = casc.Encoding.GetEntry(md5_2);
if (enc2 == null)
{
throw new Exception("enc2 == null");
}
string fakeName = Path.GetFileNameWithoutExtension(filedata[2]) + "/" + md5_2.ToHexString();
ulong fileHash = Hasher.ComputeHash(fakeName);
RootData[fileHash] = new RootEntry()
{
MD5 = md5_2, Block = RootBlock.Empty
};
CASCFile.FileNames[fileHash] = fakeName;
}
}
}
}
}
int current = 0;
Func <string, LocaleFlags> tag2locale = (s) =>
{
LocaleFlags locale;
if (Enum.TryParse(s, out locale))
{
return(locale);
}
return(LocaleFlags.All);
};
foreach (var entry in casc.Encoding.Entries)
{
DownloadEntry dl = casc.Download.GetEntry(entry.Value.Key);
if (dl != null)
{
string fakeName = "unknown" + "/" + entry.Key[0].ToString("X2") + "/" + entry.Key.ToHexString();
var locales = dl.Tags.Where(tag => tag.Value.Type == 4).Select(tag => tag2locale(tag.Key));
LocaleFlags locale = LocaleFlags.None;
foreach (var loc in locales)
{
locale |= loc;
}
ulong fileHash = Hasher.ComputeHash(fakeName);
RootData.Add(fileHash, new RootEntry()
{
MD5 = entry.Key, Block = new RootBlock()
{
LocaleFlags = locale
}
});
CASCFile.FileNames[fileHash] = fakeName;
}
worker?.ReportProgress((int)(++current / (float)casc.Encoding.Count * 100));
}
}
public static CASCHandler OpenOnlineStorage(string product, string region = "us", BackgroundWorkerEx worker = null)
{
CASCConfig config = CASCConfig.LoadOnlineStorageConfig(product, region);
return Open(worker, config);
}
private CASCHandlerLite(CASCConfig config, LocaleFlags locale, BackgroundWorkerEx worker) : base(config, worker)
{
if (config.GameType != CASCGameType.WoW)
{
throw new Exception("Unsupported game " + config.BuildUID);
}
Logger.WriteLine("CASCHandlerLite: loading encoding data...");
EncodingHandler EncodingHandler;
using (var _ = new PerfCounter("new EncodingHandler()"))
{
using (var fs = OpenEncodingFile(this))
EncodingHandler = new EncodingHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} encoding data", EncodingHandler.Count);
Logger.WriteLine("CASCHandlerLite: loading root data...");
WowRootHandler RootHandler;
using (var _ = new PerfCounter("new RootHandler()"))
{
using (var fs = OpenRootFile(EncodingHandler, this))
RootHandler = new WowRootHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} root data", RootHandler.Count);
RootHandler.SetFlags(locale, ContentFlags.None, false);
CDNIndexData = new Dictionary <MD5Hash, IndexEntry>(comparer);
if (LocalIndex != null)
{
LocalIndexData = new Dictionary <MD5Hash, IndexEntry>(comparer);
}
RootEntry rootEntry;
foreach (var entry in RootHandler.GetAllEntries())
{
rootEntry = entry.Value;
if ((rootEntry.LocaleFlags == locale || (rootEntry.LocaleFlags & locale) != LocaleFlags.None) && (rootEntry.ContentFlags & ContentFlags.LowViolence) == ContentFlags.None)
{
if (EncodingHandler.GetEntry(rootEntry.MD5, out EncodingEntry enc))
{
if (!HashToKey.ContainsKey(entry.Key))
{
HashToKey.Add(entry.Key, enc.Key);
FileDataIdToHash.Add(RootHandler.GetFileDataIdByHash(entry.Key), entry.Key);
if (LocalIndex != null)
{
IndexEntry iLocal = LocalIndex.GetIndexInfo(enc.Key);
if (iLocal != null && !LocalIndexData.ContainsKey(enc.Key))
{
LocalIndexData.Add(enc.Key, iLocal);
}
}
IndexEntry iCDN = CDNIndex.GetIndexInfo(enc.Key);
if (iCDN != null && !CDNIndexData.ContainsKey(enc.Key))
{
CDNIndexData.Add(enc.Key, iCDN);
}
}
}
}
}
CDNIndex.Clear();
//CDNIndex = null;
LocalIndex?.Clear();
LocalIndex = null;
RootHandler.Clear();
RootHandler = null;
EncodingHandler.Clear();
EncodingHandler = null;
GC.Collect();
Logger.WriteLine("CASCHandlerLite: loaded {0} files", HashToKey.Count);
}
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
if (LoadPreHashedListFile("listfile.bin", path, worker))
{
return;
}
using (var _ = new PerfCounter("WowRootHandler::LoadListFile()"))
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
if (!File.Exists(path))
{
Logger.WriteLine("WowRootHandler: list file missing!");
return;
}
Logger.WriteLine("WowRootHandler: loading file names...");
Dictionary <string, Dictionary <ulong, string> > dirData = new Dictionary <string, Dictionary <ulong, string> >(StringComparer.OrdinalIgnoreCase)
{
[""] = new Dictionary <ulong, string>()
};
using (var fs = new FileStream("listfile.bin", FileMode.Create))
using (var bw = new BinaryWriter(fs))
using (var fs2 = File.Open(path, FileMode.Open))
using (var sr = new StreamReader(fs2))
{
string file;
while ((file = sr.ReadLine()) != null)
{
ulong fileHash = Hasher.ComputeHash(file);
// skip invalid names
if (!RootData.ContainsKey(fileHash))
{
Logger.WriteLine("Invalid file name: {0}", file);
continue;
}
CASCFile.Files[fileHash] = new CASCFile(fileHash, file);
int dirSepIndex = file.LastIndexOf('\\');
if (dirSepIndex >= 0)
{
string key = file.Substring(0, dirSepIndex);
if (!dirData.ContainsKey(key))
{
dirData[key] = new Dictionary <ulong, string>();
}
dirData[key][fileHash] = file.Substring(dirSepIndex + 1);
}
else
{
dirData[""][fileHash] = file;
}
worker?.ReportProgress((int)(sr.BaseStream.Position / (float)sr.BaseStream.Length * 100));
}
bw.Write(dirData.Count); // count of dirs
foreach (var dir in dirData)
{
bw.Write(dir.Key); // dir name
Logger.WriteLine(dir.Key);
bw.Write(dirData[dir.Key].Count); // count of files in dir
foreach (var fh in dirData[dir.Key])
{
bw.Write(fh.Key); // file name hash
bw.Write(fh.Value); // file name (without dir name)
}
}
Logger.WriteLine("WowRootHandler: loaded {0} valid file names", CASCFile.Files.Count);
}
File.SetLastWriteTime("listfile.bin", File.GetLastWriteTime(path));
}
}
public static CASCHandlerLite OpenStorage(LocaleFlags locale, CASCConfig config, BackgroundWorkerEx worker = null)
{
return(Open(locale, worker, config));
}
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
if (LoadPreHashedListFile("listfile.bin", path, worker))
return;
using (var _ = new PerfCounter("WowRootHandler::LoadListFile()"))
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
if (!File.Exists(path))
{
Logger.WriteLine("WowRootHandler: list file missing!");
return;
}
Logger.WriteLine("WowRootHandler: loading file names...");
Dictionary<string, Dictionary<ulong, string>> dirData = new Dictionary<string, Dictionary<ulong, string>>(StringComparer.OrdinalIgnoreCase);
dirData[""] = new Dictionary<ulong, string>();
using (var fs = new FileStream("listfile.bin", FileMode.Create))
using (var bw = new BinaryWriter(fs))
using (var fs2 = File.Open(path, FileMode.Open))
using (var sr = new StreamReader(fs2))
{
string file;
while ((file = sr.ReadLine()) != null)
{
ulong fileHash = Hasher.ComputeHash(file);
// skip invalid names
if (!RootData.ContainsKey(fileHash))
{
Logger.WriteLine("Invalid file name: {0}", file);
continue;
}
CASCFile.FileNames[fileHash] = file;
int dirSepIndex = file.LastIndexOf('\\');
if (dirSepIndex >= 0)
{
string key = file.Substring(0, dirSepIndex);
if (!dirData.ContainsKey(key))
{
dirData[key] = new Dictionary<ulong, string>();
}
dirData[key][fileHash] = file.Substring(dirSepIndex + 1);
}
else
dirData[""][fileHash] = file;
worker?.ReportProgress((int)(sr.BaseStream.Position / (float)sr.BaseStream.Length * 100));
}
bw.Write(dirData.Count); // count of dirs
foreach (var dir in dirData)
{
bw.Write(dir.Key); // dir name
Logger.WriteLine(dir.Key);
bw.Write(dirData[dir.Key].Count); // count of files in dir
foreach (var fh in dirData[dir.Key])
{
bw.Write(fh.Key); // file name hash
bw.Write(fh.Value); // file name (without dir name)
}
}
Logger.WriteLine("WowRootHandler: loaded {0} valid file names", CASCFile.FileNames.Count);
}
File.SetLastWriteTime("listfile.bin", File.GetLastWriteTime(path));
}
}
private CASCHandlerLite(CASCConfig config, LocaleFlags locale, BackgroundWorkerEx worker) : base(config, worker)
{
if (config.GameType != CASCGameType.WoW)
{
throw new Exception("Unsupported game " + config.BuildUID);
}
Logger.WriteLine("CASCHandlerLite: loading encoding data...");
EncodingHandler EncodingHandler;
using (var _ = new PerfCounter("new EncodingHandler()"))
{
using (var fs = OpenEncodingFile(this))
EncodingHandler = new EncodingHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} encoding data", EncodingHandler.Count);
Logger.WriteLine("CASCHandlerLite: loading root data...");
RootHandlerBase RootHandler;
using (var _ = new PerfCounter("new RootHandler()"))
{
using (var fs = OpenRootFile(EncodingHandler, this))
RootHandler = new WowRootHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} root data", RootHandler.Count);
RootHandler.SetFlags(locale, ContentFlags.None, false);
RootEntry rootEntry;
foreach (var entry in RootHandler.GetAllEntries())
{
rootEntry = entry.Value;
if ((rootEntry.Block.LocaleFlags == locale || (rootEntry.Block.LocaleFlags & locale) != LocaleFlags.None) && (rootEntry.Block.ContentFlags & ContentFlags.LowViolence) == ContentFlags.None)
{
var enc = EncodingHandler.GetEntry(rootEntry.MD5);
if (enc != null)
{
if (!HashToKey.ContainsKey(entry.Key))
{
HashToKey.Add(entry.Key, enc.Key);
FileDataIdToHash.Add(rootEntry.FileDataId, entry.Key);
}
}
}
}
RootHandler.Clear();
EncodingHandler.Clear();
RootHandler = null;
EncodingHandler = null;
GC.Collect();
Logger.WriteLine("CASCHandlerLite: loaded {0} files", HashToKey.Count);
}
public D3RootHandler(BinaryReader stream, BackgroundWorkerEx worker, CASCHandler casc)
{
worker?.ReportProgress(0, "Loading \"root\"...");
byte b1 = stream.ReadByte();
byte b2 = stream.ReadByte();
byte b3 = stream.ReadByte();
byte b4 = stream.ReadByte();
int count = stream.ReadInt32();
for (int j = 0; j < count; j++)
{
MD5Hash md5 = stream.Read<MD5Hash>();
string name = stream.ReadCString();
var entries = new List<D3RootEntry>();
D3RootData[name] = entries;
EncodingEntry enc;
if (!casc.Encoding.GetEntry(md5, out enc))
continue;
using (BinaryReader s = new BinaryReader(casc.OpenFile(enc.Key)))
{
uint magic = s.ReadUInt32();
int nEntries0 = s.ReadInt32();
for (int i = 0; i < nEntries0; i++)
{
entries.Add(D3RootEntry.Read(0, s));
}
int nEntries1 = s.ReadInt32();
for (int i = 0; i < nEntries1; i++)
{
entries.Add(D3RootEntry.Read(1, s));
}
int nNamedEntries = s.ReadInt32();
for (int i = 0; i < nNamedEntries; i++)
{
entries.Add(D3RootEntry.Read(2, s));
}
}
worker?.ReportProgress((int)((j + 1) / (float)(count + 2) * 100));
}
// Parse CoreTOC.dat
var coreTocEntry = D3RootData["Base"].Find(e => e.Name == "CoreTOC.dat");
EncodingEntry enc1;
casc.Encoding.GetEntry(coreTocEntry.MD5, out enc1);
using (var file = casc.OpenFile(enc1.Key))
tocParser = new CoreTOCParser(file);
worker?.ReportProgress((int)((count + 1) / (float)(count + 2) * 100));
// Parse Packages.dat
var pkgEntry = D3RootData["Base"].Find(e => e.Name == "Data_D3\\PC\\Misc\\Packages.dat");
EncodingEntry enc2;
casc.Encoding.GetEntry(pkgEntry.MD5, out enc2);
using (var file = casc.OpenFile(enc2.Key))
pkgParser = new PackagesParser(file);
worker?.ReportProgress(100);
}
public static CASCHandler OpenStorage(CASCConfig config, BackgroundWorkerEx worker = null) => Open(worker, config);
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
Logger.WriteLine("OWRootHandler: loading file names...");
float pkgOnePct = apmFiles.Sum(a => a.Packages.Length) / 100f;
int pkgCount = 0;
foreach (var apm in apmFiles)
{
for (int i = 0; i < apm.Packages.Length; i++)
{
APMPackage package = apm.Packages[i];
MD5Hash pkgIndexMD5 = package.indexContentKey;
string apmName = Path.GetFileNameWithoutExtension(apm.Name);
string pkgName = string.Format("{0}/package_{1:X4}_{2:X16}", apmName, i, package.packageKey);
string fakeName = string.Format("{0}_index", pkgName);
ulong fileHash = Hasher.ComputeHash(fakeName);
Logger.WriteLine("Adding package: {0:X16} {1}", fileHash, package.indexContentKey.ToHexString());
if (_rootData.ContainsKey(fileHash))
{
if (!_rootData[fileHash].baseEntry.MD5.EqualsTo(package.indexContentKey))
{
Logger.WriteLine("Weird duplicate package: {0:X16} {1}", fileHash, package.indexContentKey.ToHexString());
}
else
{
Logger.WriteLine("Duplicate package: {0:X16} {1}", fileHash, package.indexContentKey.ToHexString());
}
continue;
}
_rootData[fileHash] = new OWRootEntry()
{
baseEntry = new RootEntry()
{
MD5 = pkgIndexMD5, LocaleFlags = LocaleFlags.All, ContentFlags = ContentFlags.None
}
};
CASCFile.Files[fileHash] = new CASCFile(fileHash, fakeName);
PackageIndex pkgIndex = apm.Indexes[i];
fakeName = string.Format("{0}_bundle_{1:X16}", pkgName, pkgIndex.bundleKey);
fileHash = Hasher.ComputeHash(fakeName);
Logger.WriteLine("Adding bundle: {0:X16} {1}", fileHash, pkgIndex.bundleContentKey.ToHexString());
if (_rootData.ContainsKey(fileHash))
{
if (!_rootData[fileHash].baseEntry.MD5.EqualsTo(pkgIndex.bundleContentKey))
{
Logger.WriteLine("Weird duplicate bundle: {0:X16} {1}", fileHash, pkgIndex.bundleContentKey.ToHexString());
}
else
{
Logger.WriteLine("Duplicate bundle: {0:X16} {1}", fileHash, pkgIndex.bundleContentKey.ToHexString());
}
continue;
}
_rootData[fileHash] = new OWRootEntry()
{
baseEntry = new RootEntry()
{
MD5 = pkgIndex.bundleContentKey, LocaleFlags = LocaleFlags.All, ContentFlags = ContentFlags.None
},
pkgIndex = pkgIndex
};
CASCFile.Files[fileHash] = new CASCFile(fileHash, fakeName);
PackageIndexRecord[] records = apm.Records[i];
for (int k = 0; k < records.Length; k++)
{
fakeName = string.Format("files/{0:X3}/{1:X12}.{0:X3}", KeyToTypeID(records[k].Key), records[k].Key & 0xFFFFFFFFFFFF);
fileHash = Hasher.ComputeHash(fakeName);
//Logger.WriteLine("Adding package record: key {0:X16} hash {1} flags {2:X8}", fileHash, records[k].contentKey.ToHexString(), records[k].flags);
if (_rootData.ContainsKey(fileHash))
{
if (!_rootData[fileHash].baseEntry.MD5.EqualsTo(records[k].ContentKey))
{
Logger.WriteLine("Weird duplicate package record: {0:X16} {1}", fileHash, records[k].ContentKey.ToHexString());
}
//else
// Logger.WriteLine("Duplicate package record: {0:X16} {1}", fileHash, records[k].contentKey.ToHexString());
continue;
}
_rootData[fileHash] = new OWRootEntry()
{
baseEntry = new RootEntry()
{
MD5 = records[k].ContentKey, LocaleFlags = LocaleFlags.All, ContentFlags = (ContentFlags)records[k].Flags
},
pkgIndex = pkgIndex,
pkgIndexRec = records[k]
};
CASCFile.Files[fileHash] = new CASCFile(fileHash, fakeName);
}
worker?.ReportProgress((int)(++pkgCount / pkgOnePct));
}
}
Logger.WriteLine("OWRootHandler: loaded {0} file names", _rootData.Count);
}
public void Clear()
{
CDNIndexData.Clear();
CDNIndexData = null;
config = null;
worker = null;
downloader = null;
}
public static CASCHandler OpenOnlineStorage(string product, string region = "us", BackgroundWorkerEx worker = null)
{
CASCConfig config = CASCConfig.LoadOnlineStorageConfig(product, region);
return(Open(worker, config));
}
private CASCHandler(CASCConfig config, BackgroundWorkerEx worker)
{
Config = config;
Logger.WriteLine("CASCHandler: loading CDN indices...");
using (var _ = new PerfCounter("CDNIndexHandler.Initialize()"))
{
CDNIndex = CDNIndexHandler.Initialize(config, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} CDN indexes", CDNIndex.Count);
if (!config.OnlineMode)
{
CDNIndexHandler.Cache.Enabled = false;
Logger.WriteLine("CASCHandler: loading local indices...");
using (var _ = new PerfCounter("LocalIndexHandler.Initialize()"))
{
LocalIndex = LocalIndexHandler.Initialize(config, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} local indexes", LocalIndex.Count);
}
Logger.WriteLine("CASCHandler: loading encoding data...");
using (var _ = new PerfCounter("new EncodingHandler()"))
{
using (var fs = OpenEncodingFile())
EncodingHandler = new EncodingHandler(fs, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} encoding data", EncodingHandler.Count);
if ((CASCConfig.LoadFlags & LoadFlags.Download) != 0)
{
Logger.WriteLine("CASCHandler: loading download data...");
using (var _ = new PerfCounter("new DownloadHandler()"))
{
using (var fs = OpenDownloadFile())
DownloadHandler = new DownloadHandler(fs, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} download data", EncodingHandler.Count);
}
Logger.WriteLine("CASCHandler: loading root data...");
using (var _ = new PerfCounter("new RootHandler()"))
{
using (var fs = OpenRootFile())
{
if (config.GameType == CASCGameType.S2 || config.GameType == CASCGameType.HotS)
RootHandler = new MNDXRootHandler(fs, worker);
else if (config.GameType == CASCGameType.D3)
RootHandler = new D3RootHandler(fs, worker, this);
else if (config.GameType == CASCGameType.WoW)
RootHandler = new WowRootHandler(fs, worker);
else if (config.GameType == CASCGameType.Agent)
RootHandler = new AgentRootHandler(fs, worker);
else if (config.GameType == CASCGameType.Hearthstone)
RootHandler = new HSRootHandler(fs, worker);
else if (config.GameType == CASCGameType.Overwatch)
RootHandler = new OWRootHandler(fs, worker, this);
else
throw new Exception("Unsupported game " + config.BuildUID);
}
}
Logger.WriteLine("CASCHandler: loaded {0} root data", RootHandler.Count);
if ((CASCConfig.LoadFlags & LoadFlags.Install) != 0)
{
Logger.WriteLine("CASCHandler: loading install data...");
using (var _ = new PerfCounter("new InstallHandler()"))
{
using (var fs = OpenInstallFile())
InstallHandler = new InstallHandler(fs, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} install data", InstallHandler.Count);
}
}
private CASCHandler(CASCConfig config, BackgroundWorkerEx worker) : base(config, worker)
{
Logger.WriteLine("CASCHandler: loading encoding data...");
using (var _ = new PerfCounter("new EncodingHandler()"))
{
using (var fs = OpenEncodingFile(this))
EncodingHandler = new EncodingHandler(fs, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} encoding data", EncodingHandler.Count);
if ((CASCConfig.LoadFlags & LoadFlags.Download) != 0)
{
Logger.WriteLine("CASCHandler: loading download data...");
using (var _ = new PerfCounter("new DownloadHandler()"))
{
using (var fs = OpenDownloadFile(EncodingHandler, this))
DownloadHandler = new DownloadHandler(fs, worker);
}
Logger.WriteLine("CASCHandler: loaded {0} download data", EncodingHandler.Count);
}
Logger.WriteLine("CASCHandler: loading root data...");
using (var _ = new PerfCounter("new RootHandler()"))
{
using (var fs = OpenRootFile(EncodingHandler, this))
{
if (config.GameType == CASCGameType.S2 || config.GameType == CASCGameType.HotS)
{
RootHandler = new MNDXRootHandler(fs, worker);
}
else if (config.GameType == CASCGameType.D3)
{
RootHandler = new D3RootHandler(fs, worker, this);
}
else if (config.GameType == CASCGameType.WoW)
{
RootHandler = new WowRootHandler(fs, worker);
}
else if (config.GameType == CASCGameType.Agent || config.GameType == CASCGameType.Bna || config.GameType == CASCGameType.Client)
{
RootHandler = new AgentRootHandler(fs, worker);
}
else if (config.GameType == CASCGameType.Hearthstone)
{
RootHandler = new HSRootHandler(fs, worker);
}
else if (config.GameType == CASCGameType.Overwatch)
{
RootHandler = new OwRootHandler(fs, worker, this);
}
else
{
throw new Exception("Unsupported game " + config.BuildUID);
}
}
}
Logger.WriteLine("CASCHandler: loaded {0} root data", RootHandler.Count);
if ((CASCConfig.LoadFlags & LoadFlags.Install) != 0)
{
Logger.WriteLine("CASCHandler: loading install data...");
using (var _ = new PerfCounter("new InstallHandler()"))
{
using (var fs = OpenInstallFile(EncodingHandler, this))
InstallHandler = new InstallHandler(fs, worker);
InstallHandler.Print();
}
Logger.WriteLine("CASCHandler: loaded {0} install data", InstallHandler.Count);
}
}
private CASCHandlerLite(CASCConfig config, LocaleFlags locale, BackgroundWorkerEx worker) : base(config, worker)
{
if (config.GameType != CASCGameType.WoW)
throw new Exception("Unsupported game " + config.BuildUID);
Logger.WriteLine("CASCHandlerLite: loading encoding data...");
EncodingHandler EncodingHandler;
using (var _ = new PerfCounter("new EncodingHandler()"))
{
using (var fs = OpenEncodingFile(this))
EncodingHandler = new EncodingHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} encoding data", EncodingHandler.Count);
Logger.WriteLine("CASCHandlerLite: loading root data...");
WowRootHandler RootHandler;
using (var _ = new PerfCounter("new RootHandler()"))
{
using (var fs = OpenRootFile(EncodingHandler, this))
RootHandler = new WowRootHandler(fs, worker);
}
Logger.WriteLine("CASCHandlerLite: loaded {0} root data", RootHandler.Count);
RootHandler.SetFlags(locale, ContentFlags.None, false);
CDNIndexData = new Dictionary<MD5Hash, IndexEntry>(comparer);
if (LocalIndex != null)
LocalIndexData = new Dictionary<MD5Hash, IndexEntry>(comparer);
RootEntry rootEntry;
foreach (var entry in RootHandler.GetAllEntries())
{
rootEntry = entry.Value;
if ((rootEntry.LocaleFlags == locale || (rootEntry.LocaleFlags & locale) != LocaleFlags.None) && (rootEntry.ContentFlags & ContentFlags.LowViolence) == ContentFlags.None)
{
EncodingEntry enc;
if (EncodingHandler.GetEntry(rootEntry.MD5, out enc))
{
if (!HashToKey.ContainsKey(entry.Key))
{
HashToKey.Add(entry.Key, enc.Key);
FileDataIdToHash.Add(RootHandler.GetFileDataIdByHash(entry.Key), entry.Key);
if (LocalIndex != null)
{
IndexEntry iLocal = LocalIndex.GetIndexInfo(enc.Key);
if (iLocal != null && !LocalIndexData.ContainsKey(enc.Key))
LocalIndexData.Add(enc.Key, iLocal);
}
IndexEntry iCDN = CDNIndex.GetIndexInfo(enc.Key);
if (iCDN != null && !CDNIndexData.ContainsKey(enc.Key))
CDNIndexData.Add(enc.Key, iCDN);
}
}
}
}
CDNIndex.Clear();
//CDNIndex = null;
LocalIndex?.Clear();
LocalIndex = null;
RootHandler.Clear();
RootHandler = null;
EncodingHandler.Clear();
EncodingHandler = null;
GC.Collect();
Logger.WriteLine("CASCHandlerLite: loaded {0} files", HashToKey.Count);
}
private bool LoadPreHashedListFile(string pathbin, string pathtext, BackgroundWorkerEx worker = null)
{
using (var _ = new PerfCounter("WowRootHandler::LoadPreHashedListFile()"))
{
worker?.ReportProgress(0, "Loading \"listfile\"...");
if (!File.Exists(pathbin))
return false;
var timebin = File.GetLastWriteTime(pathbin);
var timetext = File.GetLastWriteTime(pathtext);
if (timebin != timetext) // text has been modified, recreate crehashed file
return false;
Logger.WriteLine("WowRootHandler: loading file names...");
using (var fs = new FileStream(pathbin, FileMode.Open))
using (var br = new BinaryReader(fs))
{
int numFolders = br.ReadInt32();
for (int i = 0; i < numFolders; i++)
{
string dirName = br.ReadString();
Logger.WriteLine(dirName);
int numFiles = br.ReadInt32();
for (int j = 0; j < numFiles; j++)
{
ulong fileHash = br.ReadUInt64();
string fileName = br.ReadString();
string fileNameFull = dirName != String.Empty ? dirName + "\\" + fileName : fileName;
// skip invalid names
if (!RootData.ContainsKey(fileHash))
{
Logger.WriteLine("Invalid file name: {0}", fileNameFull);
continue;
}
CASCFile.FileNames[fileHash] = fileNameFull;
}
worker?.ReportProgress((int)(fs.Position / (float)fs.Length * 100));
}
Logger.WriteLine("WowRootHandler: loaded {0} valid file names", CASCFile.FileNames.Count);
}
}
return true;
}
public override void LoadListFile(string path, BackgroundWorkerEx worker = null)
{
}
public WowRootHandler(BinaryReader stream, BackgroundWorkerEx worker)
{
worker?.ReportProgress(0, "Loading \"root\"...");
while (stream.BaseStream.Position < stream.BaseStream.Length)
{
int count = stream.ReadInt32();
ContentFlags contentFlags = (ContentFlags)stream.ReadUInt32();
LocaleFlags localeFlags = (LocaleFlags)stream.ReadUInt32();
if (localeFlags == LocaleFlags.None)
throw new Exception("block.LocaleFlags == LocaleFlags.None");
if (contentFlags != ContentFlags.None && (contentFlags & (ContentFlags.LowViolence | ContentFlags.NoCompression | ContentFlags.F20000000)) == 0)
throw new Exception("block.ContentFlags != ContentFlags.None");
RootEntry[] entries = new RootEntry[count];
int[] filedataIds = new int[count];
int fileDataIndex = 0;
for (var i = 0; i < count; ++i)
{
entries[i].LocaleFlags = localeFlags;
entries[i].ContentFlags = contentFlags;
filedataIds[i] = fileDataIndex + stream.ReadInt32();
fileDataIndex = filedataIds[i] + 1;
}
//Console.WriteLine("Block: {0} {1} (size {2})", block.ContentFlags, block.LocaleFlags, count);
for (var i = 0; i < count; ++i)
{
entries[i].MD5 = stream.Read<MD5Hash>();
ulong hash = stream.ReadUInt64();
RootData.Add(hash, entries[i]);
//Console.WriteLine("File: {0:X8} {1:X16} {2}", entries[i].FileDataId, hash, entries[i].MD5.ToHexString());
ulong hash2;
int fileDataId = filedataIds[i];
if (FileDataStore.TryGetValue(fileDataId, out hash2))
{
if (hash2 == hash)
{
// duplicate, skipping
continue;
}
else
{
Logger.WriteLine("ERROR: got miltiple hashes for filedataid {0}", fileDataId);
continue;
}
}
FileDataStore.Add(fileDataId, hash);
FileDataStoreReverse.Add(hash, fileDataId);
}
worker?.ReportProgress((int)(stream.BaseStream.Position / (float)stream.BaseStream.Length * 100));
}
}
private static CASCHandlerLite Open(LocaleFlags locale, BackgroundWorkerEx worker, CASCConfig config)
{
using (var _ = new PerfCounter("new CASCHandlerLite()"))
{
return new CASCHandlerLite(config, locale, worker);
}
}