diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/sqfs/block_processor/common.c | 11 | ||||
| -rw-r--r-- | lib/sqfs/block_processor/fileapi.c | 8 | ||||
| -rw-r--r-- | lib/sqfs/block_processor/internal.h | 2 | ||||
| -rw-r--r-- | lib/sqfs/block_writer.c | 21 |
4 files changed, 42 insertions, 0 deletions
diff --git a/lib/sqfs/block_processor/common.c b/lib/sqfs/block_processor/common.c index fe9773e..bedc923 100644 --- a/lib/sqfs/block_processor/common.c +++ b/lib/sqfs/block_processor/common.c @@ -29,6 +29,9 @@ int block_processor_init(sqfs_block_processor_t *proc, size_t max_block_size, proc->cmp = cmp; proc->frag_tbl = tbl; proc->wr = wr; + + memset(&proc->stats, 0, sizeof(proc->stats)); + proc->stats.size = sizeof(proc->stats); return 0; } @@ -149,6 +152,7 @@ int process_completed_fragment(sqfs_block_processor_t *proc, sqfs_block_t *frag, proc->frag_block->index = index; proc->frag_block->flags = SQFS_BLK_FRAGMENT_BLOCK; + proc->stats.frag_block_count += 1; } err = sqfs_frag_table_add_tail_end(proc->frag_tbl, @@ -170,6 +174,7 @@ int process_completed_fragment(sqfs_block_processor_t *proc, sqfs_block_t *frag, proc->frag_block->flags |= (frag->flags & SQFS_BLK_DONT_COMPRESS); proc->frag_block->size += frag->size; + proc->stats.actual_frag_count += 1; return 0; fail: free(*blk_out); @@ -184,3 +189,9 @@ out_duplicate: } return 0; } + +const sqfs_block_processor_stats_t +*sqfs_block_processor_get_stats(const sqfs_block_processor_t *proc) +{ + return &proc->stats; +} diff --git a/lib/sqfs/block_processor/fileapi.c b/lib/sqfs/block_processor/fileapi.c index 9e59c1d..dee1d5e 100644 --- a/lib/sqfs/block_processor/fileapi.c +++ b/lib/sqfs/block_processor/fileapi.c @@ -69,15 +69,21 @@ static int flush_block(sqfs_block_processor_t *proc, sqfs_block_t *block) proc->inode->num_file_blocks += 1; proc->inode->extra[block->index] = 0; free(block); + + proc->stats.sparse_block_count += 1; return 0; } if (block->size < proc->max_block_size && !(block->flags & SQFS_BLK_DONT_FRAGMENT)) { block->flags |= SQFS_BLK_IS_FRAGMENT; + + proc->stats.total_frag_count += 1; } else { proc->inode->num_file_blocks += 1; proc->blk_flags &= ~SQFS_BLK_FIRST_BLOCK; + + proc->stats.data_block_count += 1; } return enqueue_block(proc, block); @@ -121,6 +127,8 @@ int sqfs_block_processor_append(sqfs_block_processor_t *proc, const void *data, size -= diff; proc->blk_current->size += diff; data = (const char *)data + diff; + + proc->stats.input_bytes_read += diff; } if (proc->blk_current != NULL && diff --git a/lib/sqfs/block_processor/internal.h b/lib/sqfs/block_processor/internal.h index 09e879b..dcf8a09 100644 --- a/lib/sqfs/block_processor/internal.h +++ b/lib/sqfs/block_processor/internal.h @@ -71,6 +71,8 @@ struct sqfs_block_processor_t { const sqfs_block_hooks_t *hooks; void *user_ptr; + sqfs_block_processor_stats_t stats; + /* file API */ sqfs_inode_generic_t *inode; sqfs_block_t *blk_current; diff --git a/lib/sqfs/block_writer.c b/lib/sqfs/block_writer.c index c955532..94bd044 100644 --- a/lib/sqfs/block_writer.c +++ b/lib/sqfs/block_writer.c @@ -33,9 +33,13 @@ struct sqfs_block_writer_t { blk_info_t *blocks; size_t devblksz; + sqfs_block_writer_stats_t stats; + const sqfs_block_hooks_t *hooks; void *user_ptr; + sqfs_u64 data_area_start; + sqfs_u64 start; size_t file_start; }; @@ -129,6 +133,8 @@ sqfs_block_writer_t *sqfs_block_writer_create(sqfs_file_t *file, wr->file = file; wr->devblksz = devblksz; wr->max_blocks = INIT_BLOCK_COUNT; + wr->stats.size = sizeof(wr->stats); + wr->data_area_start = wr->file->get_size(wr->file); wr->blocks = alloc_array(sizeof(wr->blocks[0]), wr->max_blocks); if (wr->blocks == NULL) { @@ -192,6 +198,12 @@ int sqfs_block_writer_write(sqfs_block_writer_t *wr, sqfs_block_t *block, block->data, block->size); if (err) return err; + + wr->stats.bytes_submitted += block->size; + wr->stats.blocks_submitted += 1; + wr->stats.blocks_written = wr->num_blocks; + wr->stats.bytes_written = offset + block->size - + wr->data_area_start; } if (wr->hooks != NULL && wr->hooks->post_block_write != NULL) @@ -229,7 +241,16 @@ int sqfs_block_writer_write(sqfs_block_writer_t *wr, sqfs_block_t *block, err = wr->file->truncate(wr->file, wr->start); if (err) return err; + + wr->stats.blocks_written = wr->num_blocks; + wr->stats.bytes_written = wr->start - wr->data_area_start; } return 0; } + +const sqfs_block_writer_stats_t +*sqfs_block_writer_get_stats(const sqfs_block_writer_t *wr) +{ + return &wr->stats; +} |