First rough stab at WoW-specific TVFS support

Author: Marlamin

TACTSharp/Build.cs

@@ -1,5 +1,6 @@
 using System.Text.Json;
 using System.Text.Json.Nodes;
+using TACTSharp.Interfaces;
 
 namespace TACTSharp
 {
@@ -10,7 +11,7 @@ public class BuildInstance
         public string? ProductConfig { get; private set; }
 
         public EncodingInstance? Encoding { get; private set; }
-        public RootInstance? Root { get; private set; }
+        public IRootInstance? Root { get; private set; }
         public InstallInstance? Install { get; private set; }
         public IndexInstance? GroupIndex { get; private set; }
         public IndexInstance? FileIndex { get; private set; }
@@ -189,17 +190,64 @@ public void Load()
             }
 
             timer.Restart();
-            if (!BuildConfig.Values.TryGetValue("root", out var rootKey))
-                throw new Exception("No root key found in build config");
 
-            var rootEncodingKeys = Encoding.FindContentKey(Convert.FromHexString(rootKey[0]));
-            if (!rootEncodingKeys)
-                throw new Exception("Root key not found in encoding");
+            var useTVFS = Settings.ManifestType == AssetManifestType.TVFS;
+            if (!BuildConfig.Values.TryGetValue("root", out var rootKey) || rootKey[0] == "00000000000000000000000000000000")
+            {
+                if (Settings.LogLevel <= TSLogLevel.Info)
+                    Console.WriteLine("No root key found in build config, falling back to TVFS.");
+
+                useTVFS = true;
+            }
+
+            if (!useTVFS)
+            {
+                var rootEncodingKeys = Encoding.FindContentKey(Convert.FromHexString(rootKey[0]));
+                if (!rootEncodingKeys)
+                    throw new Exception("Root key not found in encoding");
+
+                Root = new RootInstance(cdn.GetDecodedFilePath("data", Convert.ToHexStringLower(rootEncodingKeys[0]), 0, rootEncodingKeys.DecodedFileSize), Settings);
+
+                if (Settings.LogLevel <= TSLogLevel.Info)
+                    Console.WriteLine("Root loaded in " + Math.Ceiling(timer.Elapsed.TotalMilliseconds) + "ms");
+            }
+            else
+            {
+                var vfsKeys = new Dictionary<uint, (string cKey, string eKey)>();
+                var vfsSizes = new Dictionary<uint, (uint decodedSize, uint encodedSize)>();
+
+                foreach (var bcKey in BuildConfig.Values)
+                {
+                    if (bcKey.Key.StartsWith("vfs") && !bcKey.Key.StartsWith("vfs-root")) // vfs-1 always seems to be the same as vfs-root
+                    {
+                        if (bcKey.Key.EndsWith("-size"))
+                        {
+                            if (uint.TryParse(bcKey.Key.Replace("-size", "").Replace("vfs-", ""), out var vfsIndex))
+                                vfsSizes.Add(vfsIndex, (uint.Parse(bcKey.Value[0]), uint.Parse(bcKey.Value[1])));
+                            else
+                                if (Settings.LogLevel <= TSLogLevel.Warn)
+                                    Console.WriteLine("Warning: Failed to parse buildconfig VFS size " + bcKey.Key + ", skipping..");
+                        }
+                        else
+                        {
+                            if (uint.TryParse(bcKey.Key.Replace("vfs-", ""), out var vfsIndex))
+                                vfsKeys.Add(vfsIndex, (bcKey.Value[0], bcKey.Value[1]));
+                            else
+                                if (Settings.LogLevel <= TSLogLevel.Warn)
+                                    Console.WriteLine("Warning: Failed to parse buildconfig VFS key " + bcKey.Key + ", skipping..");
+                        }
+                    }
+                }
+
+                // TODO: Not great to feed all this stuff into it
+                Root = new TVFSInstance(vfsKeys, vfsSizes, CDNConfig, GroupIndex, FileIndex, cdn, Settings);
 
-            Root = new RootInstance(cdn.GetDecodedFilePath("data", Convert.ToHexStringLower(rootEncodingKeys[0]), 0, rootEncodingKeys.DecodedFileSize), Settings);
+                if (Settings.LogLevel <= TSLogLevel.Info)
+                    Console.WriteLine("TVFS loaded in " + Math.Ceiling(timer.Elapsed.TotalMilliseconds) + "ms");
+            }
+          
             timer.Stop();
-            if (Settings.LogLevel <= TSLogLevel.Info)
-                Console.WriteLine("Root loaded in " + Math.Ceiling(timer.Elapsed.TotalMilliseconds) + "ms");
+         
 
             timer.Restart();
             if (!BuildConfig.Values.TryGetValue("install", out var installKey))

TACTSharp/Interfaces/IRootInstance.cs

@@ -0,0 +1,14 @@
+using static TACTSharp.RootInstance;
+
+namespace TACTSharp.Interfaces
+{
+    public interface IRootInstance
+    {
+        public List<RootEntry> GetEntriesByFDID(uint fileDataID);
+        public List<RootEntry> GetEntriesByLookup(ulong lookup);
+        public uint[] GetAvailableFDIDs();
+        public ulong[] GetAvailableLookups();
+        public bool FileExists(ulong lookup);
+        public bool FileExists(uint fileDataID);
+    }
+}

TACTSharp/RootInstance.cs

@@ -3,11 +3,12 @@
 using System.IO.MemoryMappedFiles;
 using System.Runtime.CompilerServices;
 using System.Runtime.InteropServices;
+using TACTSharp.Interfaces;
 
 namespace TACTSharp
 {
     // THIS IS MOSTLY YOINKED FROM BUILDBACKUP, REMAKE AT SOME POINT
-    public class RootInstance
+    public class RootInstance : IRootInstance
     {
         private readonly MemoryMappedFile rootFile;
         private readonly MemoryMappedViewAccessor accessor;
@@ -340,16 +341,6 @@ unsafe public RootInstance(string path, Settings Settings)
                 blockCount++;
             }
 
-            //if(newRoot)
-            //{
-            //    Console.WriteLine("Read " + entriesFDID.Count + "/" + totalFiles + " total files from root");
-            //    Console.WriteLine("Read " + namedCount + "/" + namedFiles + " named files from root");
-            //}
-            //else
-            //{
-            //    Console.WriteLine("Read " + entriesFDID.Count + " files from root");
-            //}
-
             mmapViewHandle.ReleasePointer();
         }
     }

TACTSharp/Settings.cs

@@ -6,6 +6,7 @@ public class Settings
         public string Product = "wow";
         public RootInstance.LocaleFlags Locale = RootInstance.LocaleFlags.enUS;
         public RootInstance.LoadMode RootMode = RootInstance.LoadMode.Normal;
+        public AssetManifestType ManifestType = AssetManifestType.Root;
         public string? BaseDir;
         public string? BuildConfig;
         public string? CDNConfig;
@@ -20,6 +21,12 @@ public class Settings
         public static TSLogLevel LogLevel = TSLogLevel.Info;
     }
 
+    public enum AssetManifestType
+    {
+        Root,
+        TVFS
+    }
+
     public enum TSLogLevel
     {
         Debug,