From: Wu Fengguang Introduce a set of functions to scan radix tree for hole/data item. - radix_tree_scan_hole(root, index, max_scan) - radix_tree_scan_hole_backward(root, index, max_scan) - radix_tree_scan_data_backward(root, index, max_scan) The implementation is dumb and obviously correct. It will help to debug the smart ones to be introduced in future. Signed-off-by: Wu Fengguang Signed-off-by: Andrew Morton --- include/linux/radix-tree.h | 6 ++ lib/radix-tree.c | 93 +++++++++++++++++++++++++++++++++++ 2 files changed, 99 insertions(+) diff -puN include/linux/radix-tree.h~radixtree-introduce-scan-hole-data-functions include/linux/radix-tree.h --- a/include/linux/radix-tree.h~radixtree-introduce-scan-hole-data-functions +++ a/include/linux/radix-tree.h @@ -155,6 +155,12 @@ void *radix_tree_delete(struct radix_tre unsigned int radix_tree_gang_lookup(struct radix_tree_root *root, void **results, unsigned long first_index, unsigned int max_items); +unsigned long radix_tree_scan_hole(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan); +unsigned long radix_tree_scan_hole_backward(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan); +unsigned long radix_tree_scan_data_backward(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan); int radix_tree_preload(gfp_t gfp_mask); void radix_tree_init(void); void *radix_tree_tag_set(struct radix_tree_root *root, diff -puN lib/radix-tree.c~radixtree-introduce-scan-hole-data-functions lib/radix-tree.c --- a/lib/radix-tree.c~radixtree-introduce-scan-hole-data-functions +++ a/lib/radix-tree.c @@ -598,6 +598,99 @@ int radix_tree_tag_get(struct radix_tree EXPORT_SYMBOL(radix_tree_tag_get); #endif +static unsigned long +radix_tree_scan_hole_dumb(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan) +{ + unsigned long i; + + for (i = 0; i < max_scan; i++) + if (!radix_tree_lookup(root, index) || ++index == 0) + break; + + return index; +} + +static unsigned long +radix_tree_scan_hole_backward_dumb(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan) +{ + unsigned long i; + + for (i = 0; i < max_scan; i++) + if (!radix_tree_lookup(root, index) || --index == ULONG_MAX) + break; + + return index; +} + +static unsigned long +radix_tree_scan_data_backward_dumb(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan) +{ + unsigned long i; + + for (i = 0; i < max_scan; i++) + if (radix_tree_lookup(root, index) || --index == ULONG_MAX) + break; + + return index; +} + +/** + * radix_tree_scan_hole - scan for hole + * @root: radix tree root + * @index: index key + * @max_scan: advice on max items to scan (it may scan a little more) + * + * Scan forward from @index for a hole/empty item, stop when + * - hit hole + * - wrap-around to index 0 + * - @max_scan or more items scanned + */ +unsigned long radix_tree_scan_hole(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan) +{ + return radix_tree_scan_hole_dumb(root, index, max_scan); +} +EXPORT_SYMBOL(radix_tree_scan_hole); + +/** + * radix_tree_scan_hole_backward - scan backward for hole + * @root: radix tree root + * @index: index key + * @max_scan: advice on max items to scan (it may scan a little more) + * + * Scan backward from @index for a hole/empty item, stop when + * - hit hole + * - wrap-around to index ULONG_MAX + * - @max_scan or more items scanned + */ +unsigned long radix_tree_scan_hole_backward(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan) +{ + return radix_tree_scan_hole_backward_dumb(root, index, max_scan); +} +EXPORT_SYMBOL(radix_tree_scan_hole_backward); + +/** + * radix_tree_scan_data_backward - scan backward for data + * @root: radix tree root + * @index: index key + * @max_scan: advice on max items to scan (it may scan a little more) + * + * Scan backward from @index for a data item, stop when + * - hit data + * - wrap-around to index ULONG_MAX + * - @max_scan or more items scanned + */ +unsigned long radix_tree_scan_data_backward(struct radix_tree_root *root, + unsigned long index, unsigned long max_scan) +{ + return radix_tree_scan_data_backward_dumb(root, index, max_scan); +} +EXPORT_SYMBOL(radix_tree_scan_data_backward); + static unsigned int __lookup(struct radix_tree_node *slot, void **results, unsigned long index, unsigned int max_items, unsigned long *next_index) _