[PATCH v8 24/39] btrfs: handle btrfs_update_reloc_root failure in prepare_to_merge

[Josef Bacik]

btrfs_update_reloc_root will will return errors in the future, so handle
an error properly in prepare_to_merge.

Signed-off-by: Josef Bacik <jo...@toxicpanda.com>
---
 fs/btrfs/relocation.c | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/fs/btrfs/relocation.c b/fs/btrfs/relocation.c
index e6f9304d6a29..b9cc99d4b950 100644
--- a/fs/btrfs/relocation.c
+++ b/fs/btrfs/relocation.c
@@ -1868,10 +1868,21 @@ int prepare_to_merge(struct reloc_control *rc, int err)
                 */
                if (!err)
                        btrfs_set_root_refs(&reloc_root->root_item, 1);
-               btrfs_update_reloc_root(trans, root);
+               ret = btrfs_update_reloc_root(trans, root);
 
+               /*
+                * Even if we have an error we need this reloc root back on our
+                * list so we can clean up properly.
+                */
                list_add(&reloc_root->root_list, &reloc_roots);
                btrfs_put_root(root);
+
+               if (ret) {
+                       btrfs_abort_transaction(trans, ret);
+                       if (!err)
+                               err = ret;
+                       break;
+               }
        }
 
        list_splice(&reloc_roots, &rc->reloc_roots);
-- 
2.26.2