Loader: Clean up the NCCH AppLoader.

This commit is contained in:
Emmanuel Gil Peyrot 2015-01-07 01:01:19 +00:00
parent 2c24e539a2
commit 08aaa33500

View file

@ -22,7 +22,7 @@ static const int kBlockSize = 0x200; ///< Size of ExeFS blocks (in bytes)
* @param size Size of compressed buffer * @param size Size of compressed buffer
* @return Size of decompressed buffer * @return Size of decompressed buffer
*/ */
static u32 LZSS_GetDecompressedSize(u8* buffer, u32 size) { static u32 LZSS_GetDecompressedSize(const u8* buffer, u32 size) {
u32 offset_size = *(u32*)(buffer + size - 4); u32 offset_size = *(u32*)(buffer + size - 4);
return offset_size + size; return offset_size + size;
} }
@ -35,9 +35,9 @@ static u32 LZSS_GetDecompressedSize(u8* buffer, u32 size) {
* @param decompressed_size Size of decompressed buffer * @param decompressed_size Size of decompressed buffer
* @return True on success, otherwise false * @return True on success, otherwise false
*/ */
static bool LZSS_Decompress(u8* compressed, u32 compressed_size, u8* decompressed, u32 decompressed_size) { static bool LZSS_Decompress(const u8* compressed, u32 compressed_size, u8* decompressed, u32 decompressed_size) {
u8* footer = compressed + compressed_size - 8; const u8* footer = compressed + compressed_size - 8;
u32 buffer_top_and_bottom = *(u32*)footer; u32 buffer_top_and_bottom = *reinterpret_cast<const u32*>(footer);
u32 out = decompressed_size; u32 out = decompressed_size;
u32 index = compressed_size - ((buffer_top_and_bottom >> 24) & 0xFF); u32 index = compressed_size - ((buffer_top_and_bottom >> 24) & 0xFF);
u32 stop_index = compressed_size - (buffer_top_and_bottom & 0xFFFFFF); u32 stop_index = compressed_size - (buffer_top_and_bottom & 0xFFFFFF);
@ -48,7 +48,7 @@ static bool LZSS_Decompress(u8* compressed, u32 compressed_size, u8* decompresse
while (index > stop_index) { while (index > stop_index) {
u8 control = compressed[--index]; u8 control = compressed[--index];
for(u32 i = 0; i < 8; i++) { for (unsigned i = 0; i < 8; i++) {
if (index <= stop_index) if (index <= stop_index)
break; break;
if (index <= 0) if (index <= 0)
@ -58,9 +58,8 @@ static bool LZSS_Decompress(u8* compressed, u32 compressed_size, u8* decompresse
if (control & 0x80) { if (control & 0x80) {
// Check if compression is out of bounds // Check if compression is out of bounds
if(index < 2) { if (index < 2)
return false; return false;
}
index -= 2; index -= 2;
u32 segment_offset = compressed[index] | (compressed[index + 1] << 8); u32 segment_offset = compressed[index] | (compressed[index + 1] << 8);
@ -69,23 +68,21 @@ static bool LZSS_Decompress(u8* compressed, u32 compressed_size, u8* decompresse
segment_offset += 2; segment_offset += 2;
// Check if compression is out of bounds // Check if compression is out of bounds
if(out < segment_size) { if (out < segment_size)
return false; return false;
}
for(u32 j = 0; j < segment_size; j++) { for (unsigned j = 0; j < segment_size; j++) {
// Check if compression is out of bounds // Check if compression is out of bounds
if(out + segment_offset >= decompressed_size) { if (out + segment_offset >= decompressed_size)
return false; return false;
}
u8 data = decompressed[out + segment_offset]; u8 data = decompressed[out + segment_offset];
decompressed[--out] = data; decompressed[--out] = data;
} }
} else { } else {
// Check if compression is out of bounds // Check if compression is out of bounds
if(out < 1) { if (out < 1)
return false; return false;
}
decompressed[--out] = compressed[--index]; decompressed[--out] = compressed[--index];
} }
control <<= 1; control <<= 1;
@ -126,46 +123,44 @@ ResultStatus AppLoader_NCCH::LoadExec() const {
} }
ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& buffer) const { ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& buffer) const {
// Iterate through the ExeFs archive until we find the .code file...
if (!file->IsOpen()) if (!file->IsOpen())
return ResultStatus::Error; return ResultStatus::Error;
LOG_DEBUG(Loader, "%d sections:", kMaxSections); LOG_DEBUG(Loader, "%d sections:", kMaxSections);
for (int i = 0; i < kMaxSections; i++) { // Iterate through the ExeFs archive until we find the .code file...
// Load the specified section... for (unsigned section_number = 0; section_number < kMaxSections; section_number++) {
if (strcmp((const char*)exefs_header.section[i].name, name) == 0) { const auto& section = exefs_header.section[section_number];
LOG_DEBUG(Loader, "%d - offset: 0x%08X, size: 0x%08X, name: %s", i,
exefs_header.section[i].offset, exefs_header.section[i].size,
exefs_header.section[i].name);
s64 section_offset = (exefs_header.section[i].offset + exefs_offset + // Load the specified section...
sizeof(ExeFs_Header)+ncch_offset); if (strcmp(section.name, name) == 0) {
LOG_DEBUG(Loader, "%d - offset: 0x%08X, size: 0x%08X, name: %s", section_number,
section.offset, section.size, section.name);
s64 section_offset = (section.offset + exefs_offset + sizeof(ExeFs_Header) + ncch_offset);
file->Seek(section_offset, SEEK_SET); file->Seek(section_offset, SEEK_SET);
// Section is compressed... if (is_compressed) {
if (i == 0 && is_compressed) { // Section is compressed, read compressed .code section...
// Read compressed .code section...
std::unique_ptr<u8[]> temp_buffer; std::unique_ptr<u8[]> temp_buffer;
try { try {
temp_buffer.reset(new u8[exefs_header.section[i].size]); temp_buffer.reset(new u8[section.size]);
} catch (std::bad_alloc&) { } catch (std::bad_alloc&) {
return ResultStatus::ErrorMemoryAllocationFailed; return ResultStatus::ErrorMemoryAllocationFailed;
} }
file->ReadBytes(&temp_buffer[0], exefs_header.section[i].size);
if (file->ReadBytes(&temp_buffer[0], section.size) != section.size)
return ResultStatus::Error;
// Decompress .code section... // Decompress .code section...
u32 decompressed_size = LZSS_GetDecompressedSize(&temp_buffer[0], u32 decompressed_size = LZSS_GetDecompressedSize(&temp_buffer[0], section.size);
exefs_header.section[i].size);
buffer.resize(decompressed_size); buffer.resize(decompressed_size);
if (!LZSS_Decompress(&temp_buffer[0], exefs_header.section[i].size, &buffer[0], if (!LZSS_Decompress(&temp_buffer[0], section.size, &buffer[0], decompressed_size))
decompressed_size)) {
return ResultStatus::ErrorInvalidFormat; return ResultStatus::ErrorInvalidFormat;
} } else {
// Section is uncompressed... // Section is uncompressed...
} buffer.resize(section.size);
else { if (file->ReadBytes(&buffer[0], section.size) != section.size)
buffer.resize(exefs_header.section[i].size); return ResultStatus::Error;
file->ReadBytes(&buffer[0], exefs_header.section[i].size);
} }
return ResultStatus::Success; return ResultStatus::Success;
} }
@ -183,7 +178,8 @@ ResultStatus AppLoader_NCCH::Load() {
// Reset read pointer in case this file has been read before. // Reset read pointer in case this file has been read before.
file->Seek(0, SEEK_SET); file->Seek(0, SEEK_SET);
file->ReadBytes(&ncch_header, sizeof(NCCH_Header)); if (file->ReadBytes(&ncch_header, sizeof(NCCH_Header)) != sizeof(NCCH_Header))
return ResultStatus::Error;
// Skip NCSD header and load first NCCH (NCSD is just a container of NCCH files)... // Skip NCSD header and load first NCCH (NCSD is just a container of NCCH files)...
if (MakeMagic('N', 'C', 'S', 'D') == ncch_header.magic) { if (MakeMagic('N', 'C', 'S', 'D') == ncch_header.magic) {
@ -199,7 +195,8 @@ ResultStatus AppLoader_NCCH::Load() {
// Read ExHeader... // Read ExHeader...
file->ReadBytes(&exheader_header, sizeof(ExHeader_Header)); if (file->ReadBytes(&exheader_header, sizeof(ExHeader_Header)) != sizeof(ExHeader_Header))
return ResultStatus::Error;
is_compressed = (exheader_header.codeset_info.flags.flag & 1) == 1; is_compressed = (exheader_header.codeset_info.flags.flag & 1) == 1;
entry_point = exheader_header.codeset_info.text.address; entry_point = exheader_header.codeset_info.text.address;
@ -217,13 +214,12 @@ ResultStatus AppLoader_NCCH::Load() {
LOG_DEBUG(Loader, "ExeFS size: 0x%08X", exefs_size); LOG_DEBUG(Loader, "ExeFS size: 0x%08X", exefs_size);
file->Seek(exefs_offset + ncch_offset, SEEK_SET); file->Seek(exefs_offset + ncch_offset, SEEK_SET);
file->ReadBytes(&exefs_header, sizeof(ExeFs_Header)); if (file->ReadBytes(&exefs_header, sizeof(ExeFs_Header)) != sizeof(ExeFs_Header))
return ResultStatus::Error;
LoadExec(); // Load the executable into memory for booting
is_loaded = true; // Set state to loaded is_loaded = true; // Set state to loaded
return ResultStatus::Success; return LoadExec(); // Load the executable into memory for booting
} }
ResultStatus AppLoader_NCCH::ReadCode(std::vector<u8>& buffer) const { ResultStatus AppLoader_NCCH::ReadCode(std::vector<u8>& buffer) const {
@ -257,7 +253,8 @@ ResultStatus AppLoader_NCCH::ReadRomFS(std::vector<u8>& buffer) const {
buffer.resize(romfs_size); buffer.resize(romfs_size);
file->Seek(romfs_offset, SEEK_SET); file->Seek(romfs_offset, SEEK_SET);
file->ReadBytes(&buffer[0], romfs_size); if (file->ReadBytes(&buffer[0], romfs_size) != romfs_size)
return ResultStatus::Error;
return ResultStatus::Success; return ResultStatus::Success;
} }