test_maple_tree: add test for mas_spanning_rebalance() on insufficient data

Add a test to the maple tree test suite for the spanning rebalance
insufficient node issue does not go undetected again.

Link: https://lkml.kernel.org/r/20221219161922.2708732-3-Liam.Howlett@oracle.com
Fixes: 54a611b60590 ("Maple Tree: add new data structure")
Signed-off-by: Liam R. Howlett <Liam.Howlett@oracle.com>
Cc: Andrei Vagin <avagin@gmail.com>
Cc: Mike Rapoport <rppt@kernel.org>
Cc: Muhammad Usama Anjum <usama.anjum@collabora.com>
Cc: <stable@vger.kernel.org>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>

diff --git a/lib/test_maple_tree.c b/lib/test_maple_tree.c
index f425f169ef08910149b6f845d770f73118dbef30..497fc93ccf9ecf550c0904c03bf20d707022a088 100644 (file)
--- a/lib/test_maple_tree.c
+++ b/lib/test_maple_tree.c
@@ -2498,6 +2498,25 @@ static noinline void check_dup(struct maple_tree *mt)
        }
 }
 
+static noinline void check_bnode_min_spanning(struct maple_tree *mt)
+{
+       int i = 50;
+       MA_STATE(mas, mt, 0, 0);
+
+       mt_set_non_kernel(9999);
+       mas_lock(&mas);
+       do {
+               mas_set_range(&mas, i*10, i*10+9);
+               mas_store(&mas, check_bnode_min_spanning);
+       } while (i--);
+
+       mas_set_range(&mas, 240, 509);
+       mas_store(&mas, NULL);
+       mas_unlock(&mas);
+       mas_destroy(&mas);
+       mt_set_non_kernel(0);
+}
+
 static DEFINE_MTREE(tree);
 static int maple_tree_seed(void)
 {
@@ -2742,6 +2761,10 @@ static int maple_tree_seed(void)
        check_dup(&tree);
        mtree_destroy(&tree);
 
+       mt_init_flags(&tree, MT_FLAGS_ALLOC_RANGE);
+       check_bnode_min_spanning(&tree);
+       mtree_destroy(&tree);
+
 #if defined(BENCH)
 skip:
 #endif