mirrors/nstool
1
0
Fork 0
mirror of https://github.com/jakcron/nstool.git synced 2024-10-18 13:31:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CompressedRomFs] Support romfs footer being compressed in multiple chunks.

Browse source
This commit is contained in:
jakcron 2020-08-16 14:11:30 +08:00
parent 9088f285d8
commit 82b1424abb
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/RomfsProcess.cpp
View file

@ -273,14 +273,18 @@ void RomfsProcess::resolveRomfs()
// initial and final entries // initial and final entries
nn::hac::sCompressionEntry entry[2]; nn::hac::sCompressionEntry entry[2];
// read final compression entry
(*mFile)->read((byte_t*)&entry[1], physical_size - sizeof(nn::hac::sCompressionEntry), sizeof(nn::hac::sCompressionEntry)); (*mFile)->read((byte_t*)&entry[1], physical_size - sizeof(nn::hac::sCompressionEntry), sizeof(nn::hac::sCompressionEntry));
// the final compression entry should be for the romfs footer, for which the logical offset is detailed in the romfs header // the final compression entry should be for the (final part, in the case of metadata > 0x10000) romfs footer, for which the logical offset is detailed in the romfs header
// the compression is always enabled for non-header compression entries // the compression is always enabled for non-header compression entries
if (entry[1].virtual_offset.get() != mHdr.sections[nn::hac::romfs::DIR_HASHMAP_TABLE].offset.get() || \ uint64_t romfs_metadata_begin_offset = mHdr.sections[nn::hac::romfs::DIR_HASHMAP_TABLE].offset.get();
uint64_t romfs_metadata_end_offset = mHdr.sections[nn::hac::romfs::FILE_NODE_TABLE].offset.get() + mHdr.sections[nn::hac::romfs::FILE_NODE_TABLE].size.get();
if ((entry[1].virtual_offset.get() >= romfs_metadata_begin_offset && entry[1].virtual_offset.get() < romfs_metadata_end_offset) == false || \
entry[1].compression_type != (byte_t)nn::hac::compression::CompressionType::Lz4) entry[1].compression_type != (byte_t)nn::hac::compression::CompressionType::Lz4)
{ {
throw fnd::Exception(kModuleName, "RomFs appears corrupted (bad final compression entry)"); throw fnd::Exception(kModuleName, "RomFs appears corrupted (bad final compression entry virtual offset/compression type)");
} }
// the first compression entry follows the physical placement of the final data chunk (specified in the final compression entry) // the first compression entry follows the physical placement of the final data chunk (specified in the final compression entry)
@ -289,10 +293,10 @@ void RomfsProcess::resolveRomfs()
// quick check to make sure the offset at least before the last entry offset // quick check to make sure the offset at least before the last entry offset
if (first_entry_offset >= (physical_size - sizeof(nn::hac::sCompressionEntry))) if (first_entry_offset >= (physical_size - sizeof(nn::hac::sCompressionEntry)))
{ {
throw fnd::Exception(kModuleName, "RomFs appears corrupted (bad final compression entry)"); throw fnd::Exception(kModuleName, "RomFs appears corrupted (bad final compression entry physical offset/size)");
} }
// read the first compression entry // read first compression entry
(*mFile)->read((byte_t*)&entry[0], first_entry_offset, sizeof(nn::hac::sCompressionEntry)); (*mFile)->read((byte_t*)&entry[0], first_entry_offset, sizeof(nn::hac::sCompressionEntry));
// validate first compression entry // validate first compression entry