Public-Mirror/GARbro-mirror
1
0
Fork 0
mirror of https://github.com/crskycode/GARbro.git synced 2024-12-28 13:54:15 +08:00
Code Issues Projects Releases Wiki Activity

(Unity): decompress LZMA blocks.

Browse Source
This commit is contained in:
morkt 2018-10-03 17:55:33 +04:00
parent dddc29a141
commit d6f00029fd
1 changed files with 30 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
ArcFormats/Unity/BundleStream.cs
View File

@ -28,6 +28,7 @@ using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using GameRes.Compression;
using LZMA = SevenZip.Compression.LZMA;
namespace GameRes.Formats.Unity
{
@ -97,20 +98,45 @@ namespace GameRes.Formats.Unity
}
}
byte[] PrepareBuffer (uint length)
{
if (null == m_buffer || length > m_buffer.Length)
m_buffer = new byte[length];
return m_buffer;
}
void ReadCompressedSegment (BundleSegment segment)
{
m_input.Position = segment.Offset;
int method = segment.Compression & 0x3F;
if (1 == method)
{
m_buffer_len = LzmaDecompressBlock (segment.PackedSize, segment.UnpackedSize);
return;
}
if (null == m_packed || segment.PackedSize > m_packed.Length)
m_packed = new byte[segment.PackedSize];
int packed_size = m_input.Read (m_packed, 0, (int)segment.PackedSize);
if (null == m_buffer || segment.UnpackedSize > m_buffer.Length)
m_buffer = new byte[segment.UnpackedSize];
if (3 == segment.Compression)
m_buffer_len = Lz4Compressor.DecompressBlock (m_packed, packed_size, m_buffer, (int)segment.UnpackedSize);
var output = PrepareBuffer (segment.UnpackedSize);
if (3 == method)
m_buffer_len = Lz4Compressor.DecompressBlock (m_packed, packed_size, output, (int)segment.UnpackedSize);
else
throw new NotImplementedException ("Not supported Unity asset bundle compression.");
}
int LzmaDecompressBlock (uint packed_size, uint unpacked_size)
{
var decoder = new LZMA.Decoder();
var props = m_input.ReadBytes (5);
decoder.SetDecoderProperties (props);
var buffer = PrepareBuffer (unpacked_size);
using (var output = new MemoryStream (buffer))
{
decoder.Code (m_input, output, packed_size-5, unpacked_size, null);
return (int)output.Length;
}
}
int ReadFromSegment (BundleSegment segment, byte[] buffer, int offset, int count)
{
Debug.Assert (m_position >= segment.UnpackedOffset && m_position <= segment.UnpackedOffset + segment.UnpackedSize);