[PATCH v8 34/39] btrfs: handle __add_reloc_root failures in btrfs_recover_relocation

[Josef Bacik]

We can already handle errors appropriately from this function, deal with
an error coming from __add_reloc_root appropriately.

Reviewed-by: Qu Wenruo <w...@suse.com>
Signed-off-by: Josef Bacik <jo...@toxicpanda.com>
---
 fs/btrfs/relocation.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/fs/btrfs/relocation.c b/fs/btrfs/relocation.c
index 43037cb302fd..2fc04f96294e 100644
--- a/fs/btrfs/relocation.c
+++ b/fs/btrfs/relocation.c
@@ -4054,7 +4054,12 @@ int btrfs_recover_relocation(struct btrfs_root *root)
                }
 
                err = __add_reloc_root(reloc_root);
-               BUG_ON(err < 0); /* -ENOMEM or logic error */
+               if (err) {
+                       list_add_tail(&reloc_root->root_list, &reloc_roots);
+                       btrfs_put_root(fs_root);
+                       btrfs_end_transaction(trans);
+                       goto out_unset;
+               }
                fs_root->reloc_root = btrfs_grab_root(reloc_root);
                btrfs_put_root(fs_root);
        }
@@ -4269,7 +4274,10 @@ int btrfs_reloc_post_snapshot(struct btrfs_trans_handle 
*trans,
                return PTR_ERR(reloc_root);
 
        ret = __add_reloc_root(reloc_root);
-       BUG_ON(ret < 0);
+       if (ret) {
+               btrfs_put_root(reloc_root);
+               return ret;
+       }
        new_root->reloc_root = btrfs_grab_root(reloc_root);
 
        if (rc->create_reloc_tree)
-- 
2.26.2