XArray: add xa_get_order

[linux-2.6-microblaze.git] / lib / test_xarray.c

diff --git a/lib/test_xarray.c b/lib/test_xarray.c
index d4f9792..bdd4d79 100644 (file)
--- a/lib/test_xarray.c
+++ b/lib/test_xarray.c
@@ -1649,6 +1649,26 @@ static noinline void check_account(struct xarray *xa)
 #endif
 }
 
+static noinline void check_get_order(struct xarray *xa)
+{
+       unsigned int max_order = IS_ENABLED(CONFIG_XARRAY_MULTI) ? 20 : 1;
+       unsigned int order;
+       unsigned long i, j;
+
+       for (i = 0; i < 3; i++)
+               XA_BUG_ON(xa, xa_get_order(xa, i) != 0);
+
+       for (order = 0; order < max_order; order++) {
+               for (i = 0; i < 10; i++) {
+                       xa_store_order(xa, i << order, order,
+                                       xa_mk_index(i << order), GFP_KERNEL);
+                       for (j = i << order; j < (i + 1) << order; j++)
+                               XA_BUG_ON(xa, xa_get_order(xa, j) != order);
+                       xa_erase(xa, i << order);
+               }
+       }
+}
+
 static noinline void check_destroy(struct xarray *xa)
 {
        unsigned long index;
@@ -1697,6 +1717,7 @@ static int xarray_checks(void)
        check_reserve(&array);
        check_reserve(&xa0);
        check_multi_store(&array);
+       check_get_order(&array);
        check_xa_alloc();
        check_find(&array);
        check_find_entry(&array);