separate code into library

misc
2018-11-19 06:48:06 +08:00
parent b93866d196
commit 8ea998b81f
89 changed files with 2492 additions and 2273 deletions
@@ -0,0 +1,92 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.IO.Compression;
+using System.Linq;
+using System.Text;
+using Org.Brotli.Dec;
+
+namespace AssetStudio
+{
+    public class WebFile
+    {
+        public static byte[] gzipMagic = { 0x1f, 0x8b };
+        public static byte[] brotliMagic = { 0x62, 0x72, 0x6F, 0x74, 0x6C, 0x69 };
+        public List<StreamFile> fileList = new List<StreamFile>();
+
+        private class WebData
+        {
+            public int dataOffset;
+            public int dataLength;
+            public string path;
+        }
+
+        public WebFile(EndianBinaryReader reader)
+        {
+            var magic = reader.ReadBytes(2);
+            reader.Position = 0;
+            if (gzipMagic.SequenceEqual(magic))
+            {
+                var stream = new MemoryStream();
+                using (var gs = new GZipStream(reader.BaseStream, CompressionMode.Decompress))
+                {
+                    gs.CopyTo(stream);
+                }
+                stream.Position = 0;
+                using (var binaryReader = new BinaryReader(stream))
+                {
+                    ReadWebData(binaryReader);
+                }
+            }
+            else
+            {
+                reader.Position = 0x20;
+                magic = reader.ReadBytes(6);
+                reader.Position = 0;
+                if (brotliMagic.SequenceEqual(magic))
+                {
+                    var brotliStream = new BrotliInputStream(reader.BaseStream);
+                    var stream = new MemoryStream();
+                    brotliStream.CopyTo(stream);
+                    stream.Position = 0;
+                    using (var binaryReader = new BinaryReader(stream))
+                    {
+                        ReadWebData(binaryReader);
+                    }
+                }
+                else
+                {
+                    reader.endian = EndianType.LittleEndian;
+                    ReadWebData(reader);
+                }
+            }
+        }
+
+        private void ReadWebData(BinaryReader reader)
+        {
+            var signature = reader.ReadStringToNull();
+            if (signature != "UnityWebData1.0")
+                return;
+            var headLength = reader.ReadInt32();
+            var dataList = new List<WebData>();
+            while (reader.BaseStream.Position < headLength)
+            {
+                var data = new WebData();
+                data.dataOffset = reader.ReadInt32();
+                data.dataLength = reader.ReadInt32();
+                var pathLength = reader.ReadInt32();
+                data.path = Encoding.UTF8.GetString(reader.ReadBytes(pathLength));
+                dataList.Add(data);
+            }
+
+            foreach (var data in dataList)
+            {
+                var file = new StreamFile();
+                file.fileName = Path.GetFileName(data.path);
+                reader.BaseStream.Position = data.dataOffset;
+                file.stream = new MemoryStream(reader.ReadBytes(data.dataLength));
+                fileList.Add(file);
+            }
+        }
+    }
+}