diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/common/data_writer.c | 2 | ||||
| -rw-r--r-- | lib/sqfs/block_processor/common.c | 31 |
2 files changed, 20 insertions, 13 deletions
diff --git a/lib/common/data_writer.c b/lib/common/data_writer.c index 8ebd868..0b4dc07 100644 --- a/lib/common/data_writer.c +++ b/lib/common/data_writer.c @@ -22,7 +22,7 @@ int write_data_from_file(const char *filename, sqfs_block_processor_t *data, return -1; } - sqfs_inode_get_file_size(inode, &filesz); + filesz = file->get_size(file); for (offset = 0; offset < filesz; offset += diff) { if (filesz - offset > sizeof(buffer)) { 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); |