NINTENSHIT/nxdumptool
1
0
Fork 0
mirror of https://github.com/DarkMatterCore/nxdumptool.git synced 2025-01-25 02:33:13 -03:00
Code Issues Projects Releases Packages Wiki Activity

Verify decompressed .rodata segment hash.

Browse source
This commit is contained in:
Pablo Curiel 2020-10-11 13:23:58 -04:00
parent e3e6552c4e
commit f45d1a21b5
1 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
source/nso.c
View file

@ -197,7 +197,8 @@ static bool nsoGetModuleName(NsoContext *nso_ctx)
static u8 *nsoGetRodataSegment(NsoContext *nso_ctx) static u8 *nsoGetRodataSegment(NsoContext *nso_ctx)
{ {
bool compressed = (nso_ctx->nso_header.flags & NsoFlags_RoCompress); int lz4_res = 0;
bool compressed = (nso_ctx->nso_header.flags & NsoFlags_RoCompress), verify = (nso_ctx->nso_header.flags & NsoFlags_RoHash);
u8 *rodata_buf = NULL; u8 *rodata_buf = NULL;
u64 rodata_buf_size = (compressed ? LZ4_DECOMPRESS_INPLACE_BUFFER_SIZE(nso_ctx->nso_header.rodata_segment_header.size) : nso_ctx->nso_header.rodata_segment_header.size); u64 rodata_buf_size = (compressed ? LZ4_DECOMPRESS_INPLACE_BUFFER_SIZE(nso_ctx->nso_header.rodata_segment_header.size) : nso_ctx->nso_header.rodata_segment_header.size);
@ -205,7 +206,8 @@ static u8 *nsoGetRodataSegment(NsoContext *nso_ctx)
u8 *rodata_read_ptr = (compressed ? (rodata_buf + (rodata_buf_size - nso_ctx->nso_header.rodata_file_size)) : rodata_buf); u8 *rodata_read_ptr = (compressed ? (rodata_buf + (rodata_buf_size - nso_ctx->nso_header.rodata_file_size)) : rodata_buf);
u64 rodata_read_size = (compressed ? nso_ctx->nso_header.rodata_file_size : nso_ctx->nso_header.rodata_segment_header.size); u64 rodata_read_size = (compressed ? nso_ctx->nso_header.rodata_file_size : nso_ctx->nso_header.rodata_segment_header.size);
int lz4_res = 0; u8 rodata_hash[SHA256_HASH_SIZE] = {0};
bool success = false; bool success = false;
/* Allocate memory for the .rodata buffer. */ /* Allocate memory for the .rodata buffer. */
@ -233,6 +235,17 @@ static u8 *nsoGetRodataSegment(NsoContext *nso_ctx)
} }
} }
if (verify)
{
/* Verify .rodata segment hash. */
sha256CalculateHash(rodata_hash, rodata_buf, nso_ctx->nso_header.rodata_segment_header.size);
if (memcmp(rodata_hash, nso_ctx->nso_header.rodata_segment_hash, SHA256_HASH_SIZE) != 0)
{
LOGFILE(".rodata segment checksum mismatch for NRO \"%s\"!", nso_ctx->nso_filename);
goto end;
}
}
success = true; success = true;
end: end: