From aecc6dc5a9971eb4f1309f0754b0f213347c4d11 Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Sat, 22 Feb 2020 23:37:18 +0100 Subject: Move inode size accounting completely to the block processor Signed-off-by: David Oberhollenzer --- lib/sqfs/block_processor/common.c | 31 +++++++++++++++++++------------ 1 file changed, 19 insertions(+), 12 deletions(-) (limited to 'lib/sqfs') diff --git a/lib/sqfs/block_processor/common.c b/lib/sqfs/block_processor/common.c index ab5d1bc..d855b51 100644 --- a/lib/sqfs/block_processor/common.c +++ b/lib/sqfs/block_processor/common.c @@ -7,6 +7,15 @@ #define SQFS_BUILDING_DLL #include "internal.h" +static void set_block_size(sqfs_inode_generic_t *inode, + sqfs_u32 index, sqfs_u32 size) +{ + inode->extra[index] = size; + + if (index >= inode->num_file_blocks) + inode->num_file_blocks = index + 1; +} + int process_completed_block(sqfs_block_processor_t *proc, sqfs_block_t *blk) { sqfs_u64 location; @@ -21,10 +30,8 @@ int process_completed_block(sqfs_block_processor_t *proc, sqfs_block_t *blk) if (blk->flags & SQFS_BLK_IS_SPARSE) { sqfs_inode_make_extended(blk->inode); blk->inode->data.file_ext.sparse += blk->size; - blk->inode->extra[blk->index] = 0; - if (blk->index >= blk->inode->num_file_blocks) - blk->inode->num_file_blocks = blk->index + 1; + set_block_size(blk->inode, blk->index, 0); proc->stats.sparse_block_count += 1; } else if (blk->size != 0) { @@ -38,11 +45,7 @@ int process_completed_block(sqfs_block_processor_t *proc, sqfs_block_t *blk) if (err) return err; } else { - blk->inode->extra[blk->index] = size; - - if (blk->index >= blk->inode->num_file_blocks) - blk->inode->num_file_blocks = blk->index + 1; - + set_block_size(blk->inode, blk->index, size); proc->stats.data_block_count += 1; } } @@ -98,11 +101,8 @@ int process_completed_fragment(sqfs_block_processor_t *proc, sqfs_block_t *frag, if (frag->flags & SQFS_BLK_IS_SPARSE) { sqfs_inode_make_extended(frag->inode); + set_block_size(frag->inode, frag->index, 0); frag->inode->data.file_ext.sparse += frag->size; - frag->inode->extra[frag->index] = 0; - - if (frag->index >= frag->inode->num_file_blocks) - frag->inode->num_file_blocks = frag->index + 1; proc->stats.sparse_block_count += 1; return 0; @@ -206,6 +206,9 @@ int sqfs_block_processor_begin_file(sqfs_block_processor_t *proc, if (flags & ~SQFS_BLK_USER_SETTABLE_FLAGS) return SQFS_ERROR_UNSUPPORTED; + sqfs_inode_set_file_size(inode, 0); + sqfs_inode_set_frag_location(inode, 0xFFFFFFFF, 0xFFFFFFFF); + proc->inode = inode; proc->blk_flags = flags | SQFS_BLK_FIRST_BLOCK; proc->blk_index = 0; @@ -216,9 +219,13 @@ int sqfs_block_processor_append(sqfs_block_processor_t *proc, const void *data, size_t size) { sqfs_block_t *new; + sqfs_u64 filesize; size_t diff; int err; + sqfs_inode_get_file_size(proc->inode, &filesize); + sqfs_inode_set_file_size(proc->inode, filesize + size); + while (size > 0) { if (proc->blk_current == NULL) { new = alloc_flex(sizeof(*new), 1, proc->max_block_size); -- cgit v1.2.3