Properly call btrfs_search_slot while shrinking

The shrinking code used btrfs_next_leaf to find the next item, but
this does not cow the blocks it touches.  This fix calls search_slot after
finding the next item to do appropriate cow and balancing.

Signed-off-by: Chris Mason <chris.mason@oracle.com>

1 files changed, 9 insertions, 1 deletions

diff --git a/fs/btrfs/extent-tree.c b/fs/btrfs/extent-tree.c
index 426a0222e12f..c390fd847a27 100644
--- a/fs/btrfs/extent-tree.c
+++ b/fs/btrfs/extent-tree.c
@@ -2434,7 +2434,15 @@ bg_next:
                                 break;
                         }
                         leaf = path->nodes[0];
-                        nritems = btrfs_header_nritems(leaf);
+                        btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
+
+                        /*
+                         * btrfs_next_leaf doesn't cow buffers, we have to
+                         * do the search again
+                         */
+                        memcpy(&key, &found_key, sizeof(key));
+                        btrfs_release_path(root, path);
+                        continue;
                 }
 
                 btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);