mirror of
https://github.com/DarkMatterCore/nxdumptool.git
synced 2024-11-08 11:51:48 +00:00
pfs: change name table padding handling
Updates pfsAddEntryInformationToImageContext() to handle name table padding and alignment on its own. As a result, the logic from pfsWriteImageContextHeaderToMemoryBuffer() has been heavily simplified.
This commit is contained in:
parent
9d792279d3
commit
4b57609860
2 changed files with 49 additions and 36 deletions
|
@ -97,6 +97,8 @@ bool pfsGenerateEntryPatch(PartitionFileSystemContext *ctx, PartitionFileSystemE
|
|||
bool pfsAddEntryInformationToImageContext(PartitionFileSystemImageContext *ctx, const char *entry_name, u64 entry_size, u32 *out_entry_idx);
|
||||
|
||||
/// Updates the name from a Partition FS entry in an existing PartitionFileSystemImageContext, using an entry index and the new entry name.
|
||||
/// The new name length must not exceed the previous name length.
|
||||
/// Furthermore, if the new name is shorter than the previous name, the rest of the previous name will be preserved.
|
||||
bool pfsUpdateEntryNameFromImageContext(PartitionFileSystemImageContext *ctx, u32 entry_idx, const char *new_entry_name);
|
||||
|
||||
/// Generates a full Partition FS header from an existing PartitionFileSystemImageContext and writes it to the provided memory buffer.
|
||||
|
|
|
@ -246,24 +246,22 @@ bool pfsGenerateEntryPatch(PartitionFileSystemContext *ctx, PartitionFileSystemE
|
|||
|
||||
bool pfsAddEntryInformationToImageContext(PartitionFileSystemImageContext *ctx, const char *entry_name, u64 entry_size, u32 *out_entry_idx)
|
||||
{
|
||||
if (!ctx || !entry_name || !*entry_name)
|
||||
size_t entry_name_len = 0;
|
||||
|
||||
if (!ctx || !entry_name || !(entry_name_len = strlen(entry_name)))
|
||||
{
|
||||
LOG_MSG_ERROR("Invalid parameters!");
|
||||
return false;
|
||||
}
|
||||
|
||||
PartitionFileSystemHeader *header = &(ctx->header);
|
||||
|
||||
PartitionFileSystemEntry *tmp_pfs_entries = NULL, *cur_pfs_entry = NULL, *prev_pfs_entry = NULL;
|
||||
u64 tmp_pfs_entries_size = ((header->entry_count + 1) * sizeof(PartitionFileSystemEntry));
|
||||
|
||||
char *tmp_name_table = NULL;
|
||||
u32 tmp_name_table_size = (header->name_table_size + strlen(entry_name) + 1);
|
||||
u32 unpadded_name_table_size = 0;
|
||||
|
||||
/* Reallocate Partition FS entries. */
|
||||
if (!(tmp_pfs_entries = realloc(ctx->entries, tmp_pfs_entries_size)))
|
||||
if (!(tmp_pfs_entries = realloc(ctx->entries, (header->entry_count + 1) * sizeof(PartitionFileSystemEntry))))
|
||||
{
|
||||
LOG_MSG_ERROR("Failed to reallocate Partition FS entries!");
|
||||
LOG_MSG_ERROR("Failed to reallocate Partition FS entries! (%u).", header->entry_count + 1);
|
||||
return false;
|
||||
}
|
||||
|
||||
|
@ -278,26 +276,46 @@ bool pfsAddEntryInformationToImageContext(PartitionFileSystemImageContext *ctx,
|
|||
|
||||
cur_pfs_entry->offset = (prev_pfs_entry ? (prev_pfs_entry->offset + prev_pfs_entry->size) : 0);
|
||||
cur_pfs_entry->size = entry_size;
|
||||
cur_pfs_entry->name_offset = header->name_table_size;
|
||||
cur_pfs_entry->name_offset = (prev_pfs_entry ? (prev_pfs_entry->name_offset + strlen(ctx->name_table + prev_pfs_entry->name_offset) + 1) : 0);
|
||||
|
||||
/* Reallocate Partition FS name table. */
|
||||
if (!(tmp_name_table = realloc(ctx->name_table, tmp_name_table_size)))
|
||||
/* Calculate unpadded name table size. Reserve space for a NULL terminator. */
|
||||
unpadded_name_table_size = (cur_pfs_entry->name_offset + (u32)entry_name_len + 1);
|
||||
|
||||
/* Check if the name for this new entry exceeds our current name table size. */
|
||||
if (unpadded_name_table_size >= header->name_table_size)
|
||||
{
|
||||
LOG_MSG_ERROR("Failed to reallocate Partition FS name table!");
|
||||
return false;
|
||||
/* Calculate padded name table size. */
|
||||
char *tmp_name_table = NULL;
|
||||
u32 nameless_header_size = (u32)(sizeof(PartitionFileSystemHeader) + ((header->entry_count + 1) * sizeof(PartitionFileSystemEntry)));
|
||||
u32 padded_name_table_size = (ALIGN_UP(nameless_header_size + unpadded_name_table_size, PFS_HEADER_PADDING_ALIGNMENT) - nameless_header_size);
|
||||
|
||||
/* Add manual padding if the full Partition FS header would already be properly aligned. */
|
||||
if (padded_name_table_size == unpadded_name_table_size) padded_name_table_size += PFS_HEADER_PADDING_ALIGNMENT;
|
||||
|
||||
/* Reallocate Partition FS name table. */
|
||||
if (!(tmp_name_table = realloc(ctx->name_table, padded_name_table_size)))
|
||||
{
|
||||
LOG_MSG_ERROR("Failed to reallocate Partition FS name table! (0x%X).", padded_name_table_size);
|
||||
return false;
|
||||
}
|
||||
|
||||
ctx->name_table = tmp_name_table;
|
||||
tmp_name_table = NULL;
|
||||
|
||||
/* Clear new allocated area. */
|
||||
memset(ctx->name_table + cur_pfs_entry->name_offset, 0, padded_name_table_size - cur_pfs_entry->name_offset);
|
||||
|
||||
/* Update Partition FS name table size. */
|
||||
header->name_table_size = padded_name_table_size;
|
||||
}
|
||||
|
||||
ctx->name_table = tmp_name_table;
|
||||
tmp_name_table = NULL;
|
||||
|
||||
/* Update Partition FS name table. */
|
||||
sprintf(ctx->name_table + header->name_table_size, "%s", entry_name);
|
||||
header->name_table_size = tmp_name_table_size;
|
||||
/* Append new entry name to the Partition FS name table. */
|
||||
memcpy(ctx->name_table + cur_pfs_entry->name_offset, entry_name, entry_name_len);
|
||||
|
||||
/* Update output entry index. */
|
||||
if (out_entry_idx) *out_entry_idx = header->entry_count;
|
||||
|
||||
/* Update Partition FS entry count, name table size and data size. */
|
||||
/* Update Partition FS entry count and data size. */
|
||||
header->entry_count++;
|
||||
ctx->fs_size += entry_size;
|
||||
|
||||
|
@ -306,7 +324,10 @@ bool pfsAddEntryInformationToImageContext(PartitionFileSystemImageContext *ctx,
|
|||
|
||||
bool pfsUpdateEntryNameFromImageContext(PartitionFileSystemImageContext *ctx, u32 entry_idx, const char *new_entry_name)
|
||||
{
|
||||
if (!ctx || !ctx->header.entry_count || !ctx->header.name_table_size || !ctx->entries || !ctx->name_table || entry_idx >= ctx->header.entry_count || !new_entry_name || !*new_entry_name)
|
||||
size_t new_entry_name_len = 0;
|
||||
|
||||
if (!ctx || !ctx->header.entry_count || !ctx->header.name_table_size || !ctx->entries || !ctx->name_table || entry_idx >= ctx->header.entry_count || \
|
||||
!new_entry_name || !(new_entry_name_len = strlen(new_entry_name)))
|
||||
{
|
||||
LOG_MSG_ERROR("Invalid parameters!");
|
||||
return false;
|
||||
|
@ -315,7 +336,6 @@ bool pfsUpdateEntryNameFromImageContext(PartitionFileSystemImageContext *ctx, u3
|
|||
PartitionFileSystemEntry *pfs_entry = &(ctx->entries[entry_idx]);
|
||||
|
||||
char *name_table_entry = (ctx->name_table + pfs_entry->name_offset);
|
||||
size_t new_entry_name_len = strlen(new_entry_name);
|
||||
size_t cur_entry_name_len = strlen(name_table_entry);
|
||||
|
||||
if (new_entry_name_len > cur_entry_name_len)
|
||||
|
@ -324,6 +344,7 @@ bool pfsUpdateEntryNameFromImageContext(PartitionFileSystemImageContext *ctx, u3
|
|||
return false;
|
||||
}
|
||||
|
||||
//if (new_entry_name_len < cur_entry_name_len) memset(name_table_entry + new_entry_name_len, 0, cur_entry_name_len - new_entry_name_len);
|
||||
memcpy(name_table_entry, new_entry_name, new_entry_name_len);
|
||||
|
||||
return true;
|
||||
|
@ -339,29 +360,22 @@ bool pfsWriteImageContextHeaderToMemoryBuffer(PartitionFileSystemImageContext *c
|
|||
|
||||
PartitionFileSystemHeader *header = &(ctx->header);
|
||||
u8 *buf_u8 = (u8*)buf;
|
||||
u64 header_size = 0, padded_header_size = 0, block_offset = 0, block_size = 0;
|
||||
u32 padding_size = 0;
|
||||
u64 header_size = 0, block_offset = 0, block_size = 0;
|
||||
|
||||
/* Calculate header size. */
|
||||
header_size = (sizeof(PartitionFileSystemHeader) + (header->entry_count * sizeof(PartitionFileSystemEntry)) + header->name_table_size);
|
||||
|
||||
/* Calculate padded header size and padding size. */
|
||||
padded_header_size = (IS_ALIGNED(header_size, PFS_HEADER_PADDING_ALIGNMENT) ? (header_size + PFS_HEADER_PADDING_ALIGNMENT) : ALIGN_UP(header_size, PFS_HEADER_PADDING_ALIGNMENT));
|
||||
padding_size = (u32)(padded_header_size - header_size);
|
||||
|
||||
/* Check buffer size. */
|
||||
if (buf_size < padded_header_size)
|
||||
if (buf_size < header_size)
|
||||
{
|
||||
LOG_MSG_ERROR("Not enough space available in input buffer to write full Partition FS header! (got 0x%lX, need 0x%lX).", buf_size, padded_header_size);
|
||||
LOG_MSG_ERROR("Not enough space available in input buffer to write full Partition FS header! (got 0x%lX, need 0x%lX).", buf_size, header_size);
|
||||
return false;
|
||||
}
|
||||
|
||||
/* Write full header. */
|
||||
header->name_table_size += padding_size;
|
||||
block_size = sizeof(PartitionFileSystemHeader);
|
||||
memcpy(buf_u8 + block_offset, header, block_size);
|
||||
block_offset += block_size;
|
||||
header->name_table_size -= padding_size;
|
||||
|
||||
block_size = (header->entry_count * sizeof(PartitionFileSystemEntry));
|
||||
memcpy(buf_u8 + block_offset, ctx->entries, block_size);
|
||||
|
@ -369,12 +383,9 @@ bool pfsWriteImageContextHeaderToMemoryBuffer(PartitionFileSystemImageContext *c
|
|||
|
||||
block_size = header->name_table_size;
|
||||
memcpy(buf_u8 + block_offset, ctx->name_table, block_size);
|
||||
block_offset += block_size;
|
||||
|
||||
memset(buf_u8 + block_offset, 0, padding_size);
|
||||
|
||||
/* Update output header size. */
|
||||
*out_header_size = padded_header_size;
|
||||
*out_header_size = header_size;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue