diff options
author | David Oberhollenzer <david.oberhollenzer@sigma-star.at> | 2021-03-05 19:01:22 +0100 |
---|---|---|
committer | David Oberhollenzer <david.oberhollenzer@sigma-star.at> | 2021-03-06 22:08:36 +0100 |
commit | 8e2b5d633d1e33ec7bc478bf97b2f1e94776b925 (patch) | |
tree | b04748a5041252e1dc7d03dada3b729042b67b3d /lib/sqfs/xattr/xattr_writer_record.c | |
parent | 919d1e85f2cc17059f72db48c3bc38e0b524f6c0 (diff) |
Store xattr writer block description in a red-black tree
By storing the blocks in a tree, the de-duplication can lookup
existing blocks in logartihmic instead of linear time.
The linked list is still maintained, because we need to iterate
over the blocks in creation order during serialization.
Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>
Diffstat (limited to 'lib/sqfs/xattr/xattr_writer_record.c')
-rw-r--r-- | lib/sqfs/xattr/xattr_writer_record.c | 60 |
1 files changed, 23 insertions, 37 deletions
diff --git a/lib/sqfs/xattr/xattr_writer_record.c b/lib/sqfs/xattr/xattr_writer_record.c index aa87246..510ca17 100644 --- a/lib/sqfs/xattr/xattr_writer_record.c +++ b/lib/sqfs/xattr/xattr_writer_record.c @@ -114,59 +114,45 @@ int sqfs_xattr_writer_add(sqfs_xattr_writer_t *xwr, const char *key, int sqfs_xattr_writer_end(sqfs_xattr_writer_t *xwr, sqfs_u32 *out) { - kv_block_desc_t *blk, *blk_prev; + kv_block_desc_t blk; + rbtree_node_t *n; sqfs_u32 index; - size_t count; int ret; - count = xwr->num_pairs - xwr->kv_start; - if (count == 0) { + memset(&blk, 0, sizeof(blk)); + blk.start = xwr->kv_start; + blk.count = xwr->num_pairs - xwr->kv_start; + + if (blk.count == 0) { *out = 0xFFFFFFFF; return 0; } - qsort(xwr->kv_pairs + xwr->kv_start, count, + qsort(xwr->kv_pairs + blk.start, blk.count, sizeof(xwr->kv_pairs[0]), compare_u64); - blk_prev = NULL; - blk = xwr->kv_blocks; - index = 0; - - while (blk != NULL) { - if (blk->count == count) { - ret = memcmp(xwr->kv_pairs + blk->start, - xwr->kv_pairs + xwr->kv_start, - sizeof(xwr->kv_pairs[0]) * count); - - if (ret == 0) - break; - } + n = rbtree_lookup(&xwr->kv_block_tree, &blk); - if (index == 0xFFFFFFFF) - return SQFS_ERROR_OVERFLOW; - - ++index; - blk_prev = blk; - blk = blk->next; - } - - if (blk != NULL) { + if (n != NULL) { + index = *((sqfs_u32 *)rbtree_node_value(n)); xwr->num_pairs = xwr->kv_start; } else { - blk = calloc(1, sizeof(*blk)); - if (blk == NULL) - return SQFS_ERROR_ALLOC; + index = xwr->num_blocks; + + ret = rbtree_insert(&xwr->kv_block_tree, &blk, &index); + if (ret != 0) + return ret; - blk->start = xwr->kv_start; - blk->count = count; + xwr->num_blocks += 1; + n = rbtree_lookup(&xwr->kv_block_tree, &blk); - if (blk_prev == NULL) { - xwr->kv_blocks = blk; + if (xwr->kv_block_last == NULL) { + xwr->kv_block_first = rbtree_node_key(n); + xwr->kv_block_last = xwr->kv_block_first; } else { - blk_prev->next = blk; + xwr->kv_block_last->next = rbtree_node_key(n); + xwr->kv_block_last = xwr->kv_block_last->next; } - - xwr->num_blocks += 1; } *out = index; |