maple_tree: set write store type in mas_store()

Set the correct mas->store_type in mas_store() so that the subsequent
changes to mas_wr_store_entry() can use the store type to complete the
write.

Signed-off-by: Sidhartha Kumar <sidhartha.kumar@oracle.com>

diff --git a/lib/maple_tree.c b/lib/maple_tree.c
index 6ffc93bdf16189cc6aa3d8cea63a1cf3a42c8b86..d6d683d4bc545ab6ce7391ef318a655b6a8b7ff0 100644 (file)
--- a/lib/maple_tree.c
+++ b/lib/maple_tree.c
@@ -5540,7 +5540,8 @@ void *mas_store(struct ma_state *mas, void *entry)
         * want to examine what happens if a single store operation was to
         * overwrite multiple entries within a self-balancing B-Tree.
         */
-       mas_wr_store_setup(&wr_mas);
+       mas_wr_prealloc_setup(&wr_mas);
+       mas_wr_store_type(&wr_mas);
        mas_wr_store_entry(&wr_mas);
        return wr_mas.content;
 }