diff --git a/Unity Studio/Unity Classes/Shader.cs b/Unity Studio/Unity Classes/Shader.cs index 2abece4..c811298 100644 --- a/Unity Studio/Unity Classes/Shader.cs +++ b/Unity Studio/Unity Classes/Shader.cs @@ -1,4 +1,9 @@ -using System.Text; +using System; +using System.Collections.Generic; +using System.IO; +using System.Linq; +using System.Text; +using Lz4; namespace Unity_Studio { @@ -25,19 +30,39 @@ namespace Unity_Studio if (readSwitch) { - if (sourceFile.version[0] == 5 && sourceFile.version[1] >= 5 || sourceFile.version[0] > 5) + if (sourceFile.version[0] == 5 && sourceFile.version[1] >= 5 || sourceFile.version[0] > 5)//5.5.0 and up { - string str; - if ((str = preloadData.ViewStruct()) != null) + a_Stream.Position = preloadData.Offset; + ClassStruct classStructure; + //TODO 不依赖Bundle进行读取 + if (sourceFile.ClassStructures.TryGetValue(preloadData.Type1, out classStructure)) { - m_Script = Encoding.UTF8.GetBytes(str); + var member = classStructure.members; + m_Script = ReadSerializedShader(member, a_Stream); } else + { m_Script = Encoding.UTF8.GetBytes("Serialized Shader can't be read"); + } } - else + else { m_Script = a_Stream.ReadBytes(a_Stream.ReadInt32()); + if (sourceFile.version[0] == 5 && sourceFile.version[1] >= 3) //5.3 - 5.4 + { + a_Stream.AlignStream(4); + a_Stream.ReadAlignedString(a_Stream.ReadInt32());//m_PathName + var decompressedSize = a_Stream.ReadUInt32(); + var m_SubProgramBlob = a_Stream.ReadBytes(a_Stream.ReadInt32()); + var decompressedBytes = new byte[decompressedSize]; + using (var mstream = new MemoryStream(m_SubProgramBlob)) + { + var decoder = new Lz4DecoderStream(mstream); + decoder.Read(decompressedBytes, 0, (int)decompressedSize); + decoder.Dispose(); + } + m_Script = m_Script.Concat(decompressedBytes.ToArray()).ToArray(); + } } } else @@ -47,5 +72,122 @@ namespace Unity_Studio preloadData.SubItems.AddRange(new[] { preloadData.TypeString, preloadData.Size.ToString() }); } } + + private static byte[] ReadSerializedShader(List members, EndianStream a_Stream) + { + var offsets = new List(); + var compressedLengths = new List(); + var decompressedLengths = new List(); + for (int i = 0; i < members.Count; i++) + { + var member = members[i]; + var level = member.Level; + var varTypeStr = member.Type; + if (member.Name == "offsets") + { + var offsets_size = a_Stream.ReadInt32(); + for (int j = 0; j < offsets_size; j++) + { + offsets.Add(a_Stream.ReadUInt32()); + } + var compressedLengths_size = a_Stream.ReadInt32(); + for (int j = 0; j < compressedLengths_size; j++) + { + compressedLengths.Add(a_Stream.ReadUInt32()); + } + var decompressedLengths_size = a_Stream.ReadInt32(); + for (int j = 0; j < decompressedLengths_size; j++) + { + decompressedLengths.Add(a_Stream.ReadUInt32()); + } + var compressedBlob = a_Stream.ReadBytes(a_Stream.ReadInt32()); + var decompressedStream = new MemoryStream(); + for (int j = 0; j < offsets.Count; j++) + { + var compressedBytes = new byte[compressedLengths[j]]; + Array.Copy(compressedBlob, offsets[j], compressedBytes, 0, compressedLengths[j]); + var decompressedBytes = new byte[decompressedLengths[j]]; + using (var mstream = new MemoryStream(compressedBytes)) + { + var decoder = new Lz4DecoderStream(mstream); + decoder.Read(decompressedBytes, 0, (int)decompressedLengths[j]); + decoder.Dispose(); + } + decompressedStream.Write(decompressedBytes, 0, decompressedBytes.Length); + } + var decompressedBlob = decompressedStream.ToArray(); + return decompressedBlob; + } + var align = (member.Flag & 0x4000) != 0; + if (varTypeStr == "SInt8")//sbyte + { + a_Stream.ReadSByte(); + } + else if (varTypeStr == "UInt8")//byte + { + a_Stream.ReadByte(); + } + else if (varTypeStr == "short" || varTypeStr == "SInt16")//Int16 + { + a_Stream.ReadInt16(); + } + else if (varTypeStr == "UInt16" || varTypeStr == "unsigned short")//UInt16 + { + a_Stream.ReadUInt16(); + } + else if (varTypeStr == "int" || varTypeStr == "SInt32")//Int32 + { + a_Stream.ReadInt32(); + } + else if (varTypeStr == "UInt32" || varTypeStr == "unsigned int")//UInt32 + { + a_Stream.ReadUInt32(); + } + else if (varTypeStr == "long long" || varTypeStr == "SInt64")//Int64 + { + a_Stream.ReadInt64(); + } + else if (varTypeStr == "UInt64" || varTypeStr == "unsigned long long")//UInt64 + { + a_Stream.ReadUInt64(); + } + else if (varTypeStr == "float")//float + { + a_Stream.ReadSingle(); + } + else if (varTypeStr == "double")//double + { + a_Stream.ReadDouble(); + } + else if (varTypeStr == "bool")//bool + { + a_Stream.ReadBoolean(); + } + else if (varTypeStr == "string")//string + { + a_Stream.ReadAlignedString(a_Stream.ReadInt32()); + i += 3;//skip + } + else if (varTypeStr == "Array")//Array + { + if ((members[i - 1].Flag & 0x4000) != 0) + align = true; + var size = a_Stream.ReadInt32(); + var array = AssetPreloadData.ReadArray(members, level, i); + for (int j = 0; j < size; j++) + { + ReadSerializedShader(array, a_Stream); + } + i += array.Count + 1;//skip + } + else + { + align = false; + } + if (align) + a_Stream.AlignStream(4); + } + return null; + } } } diff --git a/Unity Studio/Unity Studio Classes/AssetPreloadData.cs b/Unity Studio/Unity Studio Classes/AssetPreloadData.cs index 9775dae..e207b12 100644 --- a/Unity Studio/Unity Studio Classes/AssetPreloadData.cs +++ b/Unity Studio/Unity Studio Classes/AssetPreloadData.cs @@ -127,7 +127,7 @@ namespace Unity_Studio } } - private static List ReadArray(List members, int level, int index) + public static List ReadArray(List members, int level, int index) { var member2 = new List(); for (int i = index + 2; i < members.Count; i++)//skip int size