chizui/YarikStudio
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed games handling

Browse Source
This commit is contained in:
Escartem
2024-05-01 15:32:41 +02:00
parent a4e9e5c434
commit 5c7f402997
1 changed files with 47 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
AssetStudio/OffsetStream.cs
View File

@@ -8,7 +8,6 @@ namespace AssetStudio
{ {
public class OffsetStream : Stream public class OffsetStream : Stream
{ {
private readonly Stream _baseStream; private readonly Stream _baseStream;
private long _offset; private long _offset;
@@ -81,34 +80,59 @@ namespace AssetStudio
else else
{ {
using var reader = new FileReader(path, this, true); using var reader = new FileReader(path, this, true);
var signature = reader.FileType switch
{
FileType.BundleFile => "UnityFS\x00",
FileType.MhyFile => "mhy",
FileType.Blb3File => "Blb\x03",
_ => throw new InvalidOperationException()
};
Logger.Verbose($"Parsed signature: {signature}"); var readSignature = reader.FileType;
var signature = "";
var signatureBytes = Encoding.UTF8.GetBytes(signature); if (readSignature == FileType.BundleFile ||
var buffer = ArrayPool<byte>.Shared.Rent((int)reader.Length); readSignature == FileType.MhyFile ||
while (Remaining > 0) readSignature == FileType.Blb3File)
{ {
var index = 0; switch (reader.FileType)
var absOffset = AbsolutePosition;
var read = Read(buffer);
while (index < read)
{ {
index = buffer.AsSpan(0, read).Search(signatureBytes, index); case FileType.BundleFile:
if (index == -1) break; signature = "UnityFS\x00";
var offset = absOffset + index; break;
Offset = offset; case FileType.MhyFile:
yield return offset; signature = "mhy";
index++; break;
case FileType.Blb3File:
signature = "Blb\x03";
break;
}
Logger.Verbose($"Parsed signature: {signature}");
var signatureBytes = Encoding.UTF8.GetBytes(signature);
var buffer = ArrayPool<byte>.Shared.Rent((int)reader.Length);
while (Remaining > 0)
{
var index = 0;
var absOffset = AbsolutePosition;
var read = Read(buffer);
while (index < read)
{
index = buffer.AsSpan(0, read).Search(signatureBytes, index);
if (index == -1) break;
var offset = absOffset + index;
Offset = offset;
yield return offset;
index++;
}
}
ArrayPool<byte>.Shared.Return(buffer);
} else
{
while (Remaining > 0)
{
Offset = AbsolutePosition;
yield return AbsolutePosition;
if (Offset == AbsolutePosition)
{
break;
}
} }
} }
ArrayPool<byte>.Shared.Return(buffer);
} }
} }
} }