Skip to content

Commit

Permalink
radix-tree: tidy up __radix_tree_create()
Browse files Browse the repository at this point in the history
1. Rename the existing variable 'slot' to 'child'.
2. Introduce a new variable called 'slot' which is the address of the
   slot we're dealing with.  This lets us simplify the tree insertion,
   and removes the recalculation of 'slot' at the end of the function.
3. Using 'slot' in the sibling pointer insertion part makes the code
   more readable.

Signed-off-by: Matthew Wilcox <willy@linux.intel.com>
Cc: Konstantin Khlebnikov <koct9i@gmail.com>
Cc: Kirill Shutemov <kirill.shutemov@linux.intel.com>
Cc: Jan Kara <jack@suse.com>
Cc: Neil Brown <neilb@suse.de>
Cc: Ross Zwisler <ross.zwisler@linux.intel.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
  • Loading branch information
Matthew Wilcox authored and Linus Torvalds committed May 21, 2016
1 parent a8e4da2 commit 89148aa
Showing 1 changed file with 23 additions and 25 deletions.
48 changes: 23 additions & 25 deletions lib/radix-tree.c
Original file line number Diff line number Diff line change
Expand Up @@ -499,64 +499,62 @@ int __radix_tree_create(struct radix_tree_root *root, unsigned long index,
unsigned order, struct radix_tree_node **nodep,
void ***slotp)
{
struct radix_tree_node *node = NULL, *slot;
struct radix_tree_node *node = NULL, *child;
void **slot = (void **)&root->rnode;
unsigned long maxindex;
unsigned int shift, offset;
unsigned int shift, offset = 0;
unsigned long max = index | ((1UL << order) - 1);

shift = radix_tree_load_root(root, &slot, &maxindex);
shift = radix_tree_load_root(root, &child, &maxindex);

/* Make sure the tree is high enough. */
if (max > maxindex) {
int error = radix_tree_extend(root, max, shift);
if (error < 0)
return error;
shift = error;
slot = root->rnode;
child = root->rnode;
if (order == shift)
shift += RADIX_TREE_MAP_SHIFT;
}

offset = 0; /* uninitialised var warning */
while (shift > order) {
shift -= RADIX_TREE_MAP_SHIFT;
if (slot == NULL) {
if (child == NULL) {
/* Have to add a child node. */
slot = radix_tree_node_alloc(root);
if (!slot)
child = radix_tree_node_alloc(root);
if (!child)
return -ENOMEM;
slot->shift = shift;
slot->offset = offset;
slot->parent = node;
if (node) {
rcu_assign_pointer(node->slots[offset],
node_to_entry(slot));
child->shift = shift;
child->offset = offset;
child->parent = node;
rcu_assign_pointer(*slot, node_to_entry(child));
if (node)
node->count++;
} else
rcu_assign_pointer(root->rnode,
node_to_entry(slot));
} else if (!radix_tree_is_internal_node(slot))
} else if (!radix_tree_is_internal_node(child))
break;

/* Go a level down */
node = entry_to_node(slot);
node = entry_to_node(child);
offset = (index >> shift) & RADIX_TREE_MAP_MASK;
offset = radix_tree_descend(node, &slot, offset);
offset = radix_tree_descend(node, &child, offset);
slot = &node->slots[offset];
}

#ifdef CONFIG_RADIX_TREE_MULTIORDER
/* Insert pointers to the canonical entry */
if (order > shift) {
int i, n = 1 << (order - shift);
unsigned i, n = 1 << (order - shift);
offset = offset & ~(n - 1);
slot = node_to_entry(&node->slots[offset]);
slot = &node->slots[offset];
child = node_to_entry(slot);
for (i = 0; i < n; i++) {
if (node->slots[offset + i])
if (slot[i])
return -EEXIST;
}

for (i = 1; i < n; i++) {
rcu_assign_pointer(node->slots[offset + i], slot);
rcu_assign_pointer(slot[i], child);
node->count++;
}
}
Expand All @@ -565,7 +563,7 @@ int __radix_tree_create(struct radix_tree_root *root, unsigned long index,
if (nodep)
*nodep = node;
if (slotp)
*slotp = node ? node->slots + offset : (void **)&root->rnode;
*slotp = slot;
return 0;
}

Expand Down

0 comments on commit 89148aa

Please sign in to comment.