If any of the reference count manipulation stuff fails in replace_path
we need to abort the transaction, as we've modified the blocks already.
We can simply break at this point and everything will be cleaned up.

Reviewed-by: Qu Wenruo <[email protected]>
Signed-off-by: Josef Bacik <[email protected]>
---
 fs/btrfs/relocation.c | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/fs/btrfs/relocation.c b/fs/btrfs/relocation.c
index 6e8d89e4733a..1a0e07507796 100644
--- a/fs/btrfs/relocation.c
+++ b/fs/btrfs/relocation.c
@@ -1363,27 +1363,39 @@ int replace_path(struct btrfs_trans_handle *trans, 
struct reloc_control *rc,
                ref.skip_qgroup = true;
                btrfs_init_tree_ref(&ref, level - 1, src->root_key.objectid);
                ret = btrfs_inc_extent_ref(trans, &ref);
-               BUG_ON(ret);
+               if (ret) {
+                       btrfs_abort_transaction(trans, ret);
+                       break;
+               }
                btrfs_init_generic_ref(&ref, BTRFS_ADD_DELAYED_REF, new_bytenr,
                                       blocksize, 0);
                ref.skip_qgroup = true;
                btrfs_init_tree_ref(&ref, level - 1, dest->root_key.objectid);
                ret = btrfs_inc_extent_ref(trans, &ref);
-               BUG_ON(ret);
+               if (ret) {
+                       btrfs_abort_transaction(trans, ret);
+                       break;
+               }
 
                btrfs_init_generic_ref(&ref, BTRFS_DROP_DELAYED_REF, new_bytenr,
                                       blocksize, path->nodes[level]->start);
                btrfs_init_tree_ref(&ref, level - 1, src->root_key.objectid);
                ref.skip_qgroup = true;
                ret = btrfs_free_extent(trans, &ref);
-               BUG_ON(ret);
+               if (ret) {
+                       btrfs_abort_transaction(trans, ret);
+                       break;
+               }
 
                btrfs_init_generic_ref(&ref, BTRFS_DROP_DELAYED_REF, old_bytenr,
                                       blocksize, 0);
                btrfs_init_tree_ref(&ref, level - 1, dest->root_key.objectid);
                ref.skip_qgroup = true;
                ret = btrfs_free_extent(trans, &ref);
-               BUG_ON(ret);
+               if (ret) {
+                       btrfs_abort_transaction(trans, ret);
+                       break;
+               }
 
                btrfs_unlock_up_safe(path, 0);
 
-- 
2.26.2

Reply via email to