mirror of
https://github.com/KillzXGaming/Switch-Toolbox
synced 2024-11-10 07:04:36 +00:00
First pass for TRPAK archive files (#544)
first seen in Pokemon Scarlet/Violet
This commit is contained in:
parent
8fc71d3c32
commit
d55f738297
4 changed files with 292 additions and 1 deletions
|
@ -0,0 +1,116 @@
|
|||
// <auto-generated>
|
||||
// automatically generated by the FlatBuffers compiler, do not modify
|
||||
// </auto-generated>
|
||||
|
||||
using global::System;
|
||||
using global::FlatBuffers;
|
||||
|
||||
namespace FlatBuffers.TRPAK
|
||||
{
|
||||
public struct TRPAK : IFlatbufferObject
|
||||
{
|
||||
private Table __p;
|
||||
public ByteBuffer ByteBuffer { get { return __p.bb; } }
|
||||
public static TRPAK GetRootAsTRPAK(ByteBuffer _bb) { return GetRootAsTRPAK(_bb, new TRPAK()); }
|
||||
public static TRPAK GetRootAsTRPAK(ByteBuffer _bb, TRPAK obj) { return (obj.__assign(_bb.GetInt(_bb.Position) + _bb.Position, _bb)); }
|
||||
public void __init(int _i, ByteBuffer _bb) { __p.bb_pos = _i; __p.bb = _bb; }
|
||||
public TRPAK __assign(int _i, ByteBuffer _bb) { __init(_i, _bb); return this; }
|
||||
|
||||
public ulong Hashes(int j) { int o = __p.__offset(4); return o != 0 ? __p.bb.GetUlong(__p.__vector(o) + j * 8) : (ulong)0; }
|
||||
public int HashesLength { get { int o = __p.__offset(4); return o != 0 ? __p.__vector_len(o) : 0; } }
|
||||
#if ENABLE_SPAN_T
|
||||
public Span<byte> GetHashesBytes() { return __p.__vector_as_span(4); }
|
||||
#else
|
||||
public ArraySegment<byte>? GetHashesBytes() { return __p.__vector_as_arraysegment(4); }
|
||||
#endif
|
||||
public ulong[] GetHashesArray() { return __p.__vector_as_array<ulong>(4); }
|
||||
public File? Files(int j) { int o = __p.__offset(6); return o != 0 ? (File?)(new File()).__assign(__p.__indirect(__p.__vector(o) + j * 4), __p.bb) : null; }
|
||||
public int FilesLength { get { int o = __p.__offset(6); return o != 0 ? __p.__vector_len(o) : 0; } }
|
||||
|
||||
public static Offset<TRPAK> CreateTRPAK(FlatBufferBuilder builder,
|
||||
VectorOffset hashesOffset = default(VectorOffset),
|
||||
VectorOffset filesOffset = default(VectorOffset))
|
||||
{
|
||||
builder.StartObject(2);
|
||||
TRPAK.AddFiles(builder, filesOffset);
|
||||
TRPAK.AddHashes(builder, hashesOffset);
|
||||
return TRPAK.EndTRPAK(builder);
|
||||
}
|
||||
|
||||
public static void StartTRPAK(FlatBufferBuilder builder) { builder.StartObject(2); }
|
||||
public static void AddHashes(FlatBufferBuilder builder, VectorOffset hashesOffset) { builder.AddOffset(0, hashesOffset.Value, 0); }
|
||||
public static VectorOffset CreateHashesVector(FlatBufferBuilder builder, ulong[] data) { builder.StartVector(8, data.Length, 8); for (int i = data.Length - 1; i >= 0; i--) builder.AddUlong(data[i]); return builder.EndVector(); }
|
||||
public static VectorOffset CreateHashesVectorBlock(FlatBufferBuilder builder, ulong[] data) { builder.StartVector(8, data.Length, 8); builder.Add(data); return builder.EndVector(); }
|
||||
public static void StartHashesVector(FlatBufferBuilder builder, int numElems) { builder.StartVector(8, numElems, 8); }
|
||||
public static void AddFiles(FlatBufferBuilder builder, VectorOffset filesOffset) { builder.AddOffset(1, filesOffset.Value, 0); }
|
||||
public static VectorOffset CreateFilesVector(FlatBufferBuilder builder, Offset<File>[] data) { builder.StartVector(4, data.Length, 4); for (int i = data.Length - 1; i >= 0; i--) builder.AddOffset(data[i].Value); return builder.EndVector(); }
|
||||
public static VectorOffset CreateFilesVectorBlock(FlatBufferBuilder builder, Offset<File>[] data) { builder.StartVector(4, data.Length, 4); builder.Add(data); return builder.EndVector(); }
|
||||
public static void StartFilesVector(FlatBufferBuilder builder, int numElems) { builder.StartVector(4, numElems, 4); }
|
||||
public static Offset<TRPAK> EndTRPAK(FlatBufferBuilder builder)
|
||||
{
|
||||
int o = builder.EndObject();
|
||||
return new Offset<TRPAK>(o);
|
||||
}
|
||||
public static void FinishTRPAKBuffer(FlatBufferBuilder builder, Offset<TRPAK> offset) { builder.Finish(offset.Value); }
|
||||
public static void FinishSizePrefixedTRPAKBuffer(FlatBufferBuilder builder, Offset<TRPAK> offset) { builder.FinishSizePrefixed(offset.Value); }
|
||||
};
|
||||
public struct File : IFlatbufferObject
|
||||
{
|
||||
private Table __p;
|
||||
public ByteBuffer ByteBuffer { get { return __p.bb; } }
|
||||
public static File GetRootAsFile(ByteBuffer _bb) { return GetRootAsFile(_bb, new File()); }
|
||||
public static File GetRootAsFile(ByteBuffer _bb, File obj) { return (obj.__assign(_bb.GetInt(_bb.Position) + _bb.Position, _bb)); }
|
||||
public void __init(int _i, ByteBuffer _bb) { __p.bb_pos = _i; __p.bb = _bb; }
|
||||
public File __assign(int _i, ByteBuffer _bb) { __init(_i, _bb); return this; }
|
||||
|
||||
public byte Unused { get { int o = __p.__offset(4); return o != 0 ? __p.bb.Get(o + __p.bb_pos) : (byte)0; } }
|
||||
public Compression CompressionType { get { int o = __p.__offset(6); return o != 0 ? (Compression)__p.bb.Get(o + __p.bb_pos) : Compression.NONE; } }
|
||||
public byte Unk1 { get { int o = __p.__offset(8); return o != 0 ? __p.bb.Get(o + __p.bb_pos) : (byte)0; } }
|
||||
public ulong DecompressedSize { get { int o = __p.__offset(10); return o != 0 ? __p.bb.GetUlong(o + __p.bb_pos) : (ulong)0; } }
|
||||
public byte Data(int j) { int o = __p.__offset(12); return o != 0 ? __p.bb.Get(__p.__vector(o) + j * 1) : (byte)0; }
|
||||
public int DataLength { get { int o = __p.__offset(12); return o != 0 ? __p.__vector_len(o) : 0; } }
|
||||
#if ENABLE_SPAN_T
|
||||
public Span<byte> GetDataBytes() { return __p.__vector_as_span(12); }
|
||||
#else
|
||||
public ArraySegment<byte>? GetDataBytes() { return __p.__vector_as_arraysegment(12); }
|
||||
#endif
|
||||
public byte[] GetDataArray() { return __p.__vector_as_array<byte>(12); }
|
||||
|
||||
public static Offset<File> CreateFile(FlatBufferBuilder builder,
|
||||
byte unused = 0,
|
||||
Compression compression_type = Compression.NONE,
|
||||
byte unk1 = 0,
|
||||
ulong decompressed_size = 0,
|
||||
VectorOffset dataOffset = default(VectorOffset))
|
||||
{
|
||||
builder.StartObject(5);
|
||||
File.AddDecompressedSize(builder, decompressed_size);
|
||||
File.AddData(builder, dataOffset);
|
||||
File.AddUnk1(builder, unk1);
|
||||
File.AddCompressionType(builder, compression_type);
|
||||
File.AddUnused(builder, unused);
|
||||
return File.EndFile(builder);
|
||||
}
|
||||
|
||||
public static void StartFile(FlatBufferBuilder builder) { builder.StartObject(5); }
|
||||
public static void AddUnused(FlatBufferBuilder builder, byte unused) { builder.AddByte(0, unused, 0); }
|
||||
public static void AddCompressionType(FlatBufferBuilder builder, Compression compressionType) { builder.AddByte(1, (byte)compressionType, 255); }
|
||||
public static void AddUnk1(FlatBufferBuilder builder, byte unk1) { builder.AddByte(2, unk1, 0); }
|
||||
public static void AddDecompressedSize(FlatBufferBuilder builder, ulong decompressedSize) { builder.AddUlong(3, decompressedSize, 0); }
|
||||
public static void AddData(FlatBufferBuilder builder, VectorOffset dataOffset) { builder.AddOffset(4, dataOffset.Value, 0); }
|
||||
public static VectorOffset CreateDataVector(FlatBufferBuilder builder, byte[] data) { builder.StartVector(1, data.Length, 1); for (int i = data.Length - 1; i >= 0; i--) builder.AddByte(data[i]); return builder.EndVector(); }
|
||||
public static VectorOffset CreateDataVectorBlock(FlatBufferBuilder builder, byte[] data) { builder.StartVector(1, data.Length, 1); builder.Add(data); return builder.EndVector(); }
|
||||
public static void StartDataVector(FlatBufferBuilder builder, int numElems) { builder.StartVector(1, numElems, 1); }
|
||||
public static Offset<File> EndFile(FlatBufferBuilder builder)
|
||||
{
|
||||
int o = builder.EndObject();
|
||||
return new Offset<File>(o);
|
||||
}
|
||||
};
|
||||
public enum Compression : byte
|
||||
{
|
||||
OODLE = 3,
|
||||
NONE = 255,
|
||||
};
|
||||
}
|
||||
|
172
File_Format_Library/FileFormats/Archives/TRPAK/TRPAK.cs
Normal file
172
File_Format_Library/FileFormats/Archives/TRPAK/TRPAK.cs
Normal file
|
@ -0,0 +1,172 @@
|
|||
using FirstPlugin.FileFormats.Hashes;
|
||||
using FlatBuffers.TRPAK;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Windows.Forms;
|
||||
using Toolbox.Library;
|
||||
using Toolbox.Library.IO;
|
||||
|
||||
namespace FirstPlugin
|
||||
{
|
||||
internal class TRPAK : TreeNodeFile, IArchiveFile, IFileFormat
|
||||
{
|
||||
public static bool shownOodleError;
|
||||
|
||||
public bool CanAddFiles => false;
|
||||
public bool CanDeleteFiles => false;
|
||||
public bool CanRenameFiles => false;
|
||||
public bool CanReplaceFiles => false;
|
||||
public bool CanSave { get; set; } = false;
|
||||
public string[] Description { get; set; } = new string[] { "tr Package" };
|
||||
public string[] Extension { get; set; } = new string[] { "*.trpak" };
|
||||
public string FileName { get; set; }
|
||||
public string FilePath { get; set; }
|
||||
public IEnumerable<ArchiveFileInfo> Files { get; set; }
|
||||
public FileType FileType { get; set; } = FileType.Archive;
|
||||
public IFileInfo IFileInfo { get; set; }
|
||||
|
||||
public Type[] Types
|
||||
{
|
||||
get
|
||||
{
|
||||
List<Type> types = new List<Type>();
|
||||
return types.ToArray();
|
||||
}
|
||||
}
|
||||
|
||||
public bool AddFile(ArchiveFileInfo archiveFileInfo)
|
||||
{
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public void ClearFiles()
|
||||
{
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public bool DeleteFile(ArchiveFileInfo archiveFileInfo)
|
||||
{
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public bool Identify(Stream stream)
|
||||
{
|
||||
return Utils.GetExtension(FileName) == ".trpak";
|
||||
}
|
||||
|
||||
public void Load(Stream stream)
|
||||
{
|
||||
GFPAKHashCache.EnsureHashCache();
|
||||
Files = new List<ArchiveFileInfo>();
|
||||
FlatBuffers.TRPAK.TRPAK trpak = FlatBuffers.TRPAK.TRPAK.GetRootAsTRPAK(new FlatBuffers.ByteBuffer(stream.ToBytes()));
|
||||
if (trpak.FilesLength != trpak.HashesLength)
|
||||
{
|
||||
throw new Exception("not the same amount of Hashes and File Entries in Trpak Container");
|
||||
}
|
||||
for (int i = 0; i < trpak.FilesLength; i++)
|
||||
{
|
||||
FlatBuffers.TRPAK.File? file = trpak.Files(i);
|
||||
ulong hash = trpak.Hashes(i);
|
||||
if (file.HasValue)
|
||||
{
|
||||
ArchiveFileInfo AFI = new ArchiveFileInfo();
|
||||
byte[] FileData = file.Value.GetDataArray();
|
||||
if (file.Value.CompressionType == Compression.OODLE)
|
||||
{
|
||||
if (!shownOodleError && !System.IO.File.Exists($"{Runtime.ExecutableDir}\\oo2core_6_win64.dll"))
|
||||
{
|
||||
MessageBox.Show("'oo2core_6_win64.dll' not found in the executable folder! User must provide their own copy!");
|
||||
shownOodleError = true;
|
||||
}
|
||||
byte[] FileDatadecompressed = Toolbox.Library.Compression.Oodle.Decompress(FileData, (long)file.Value.DecompressedSize);
|
||||
FileData = FileDatadecompressed;
|
||||
}
|
||||
AFI.FileData = FileData;
|
||||
AFI.Name = GetName(hash, FileData);
|
||||
AFI.FileName = AFI.Name;
|
||||
((List<ArchiveFileInfo>)Files).Add(AFI);
|
||||
}
|
||||
}
|
||||
GFPAKHashCache.WriteCache();
|
||||
}
|
||||
|
||||
public void Save(Stream stream)
|
||||
{
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public void Unload()
|
||||
{
|
||||
foreach (var file in Files)
|
||||
{
|
||||
if (file.FileFormat != null)
|
||||
file.FileFormat.Unload();
|
||||
|
||||
file.FileData = null;
|
||||
}
|
||||
|
||||
((List<ArchiveFileInfo>)Files).Clear();
|
||||
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
|
||||
private string FindMatch(byte[] f, string FileName)
|
||||
{
|
||||
foreach (IFileFormat fileFormat in FileManager.GetFileFormats())
|
||||
{
|
||||
fileFormat.FileName = FileName;
|
||||
|
||||
if (fileFormat.Identify(new MemoryStream(f)))
|
||||
{
|
||||
return fileFormat.Extension[0].Replace("*", "");
|
||||
}
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
//For BNTX, BNSH, etc
|
||||
private string GetBinaryHeaderName(byte[] Data)
|
||||
{
|
||||
using (var reader = new FileReader(Data))
|
||||
{
|
||||
reader.Seek(0x10, SeekOrigin.Begin);
|
||||
uint NameOffset = reader.ReadUInt32();
|
||||
|
||||
reader.Seek(NameOffset, SeekOrigin.Begin);
|
||||
return reader.ReadString(Syroot.BinaryData.BinaryStringFormat.ZeroTerminated);
|
||||
}
|
||||
}
|
||||
|
||||
private string GetName(ulong fileHash, byte[] Data)
|
||||
{
|
||||
string fileHashName = GFPAKHashCache.GetHashName(fileHash) ?? "";
|
||||
string ext = FindMatch(Data, fileHashName);
|
||||
if (ext == ".bntx" || ext == ".bfres" || ext == ".bnsh" || ext == ".bfsha")
|
||||
{
|
||||
string fileName = GetBinaryHeaderName(Data);
|
||||
//Check for matches for shaders
|
||||
if (ext == ".bnsh")
|
||||
{
|
||||
if (FNV64A1.Calculate($"{fileName}.bnsh_fsh") == fileHash)
|
||||
fileName = $"{fileName}.bnsh_fsh";
|
||||
else if (FNV64A1.Calculate($"{fileName}.bnsh_vsh") == fileHash)
|
||||
fileName = $"{fileName}.bnsh_vsh";
|
||||
}
|
||||
else
|
||||
fileName = $"{fileName}{ext}";
|
||||
|
||||
return $"{fileName}";
|
||||
}
|
||||
else
|
||||
{
|
||||
if (fileHashName != "")
|
||||
{
|
||||
return $"{fileHashName}{ext}";
|
||||
}
|
||||
else
|
||||
return $"{fileHash.ToString("X")}{ext}";
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
|
@ -226,6 +226,8 @@
|
|||
<Compile Include="FileFormats\Archives\QuickAccess\QuickAccessFile.cs" />
|
||||
<Compile Include="FileFormats\Archives\QuickAccess\QuickAccessFolder.cs" />
|
||||
<Compile Include="FileFormats\Archives\RARC\RARC_Parser.cs" />
|
||||
<Compile Include="FileFormats\Archives\TRPAK\Flatbuffers\TRPAK.cs" />
|
||||
<Compile Include="FileFormats\Archives\TRPAK\TRPAK.cs" />
|
||||
<Compile Include="FileFormats\Archives\WTA.cs" />
|
||||
<Compile Include="FileFormats\Audio\BARS\BarsFile.cs" />
|
||||
<Compile Include="FileFormats\BCH\BCH.cs" />
|
||||
|
|
|
@ -460,7 +460,8 @@ namespace FirstPlugin
|
|||
Formats.Add(typeof(MTXT));
|
||||
Formats.Add(typeof(NKN));
|
||||
Formats.Add(typeof(MetroidDreadLibrary.BSMAT));
|
||||
|
||||
Formats.Add(typeof(TRPAK));
|
||||
|
||||
//Formats.Add(typeof(XLINK_FILE));
|
||||
|
||||
// Formats.Add(typeof(MPBIN));
|
||||
|
|
Loading…
Reference in a new issue