1 files changed, 7 insertions, 0 deletions

diff --git a/fs/btrfs/extent-tree.c b/fs/btrfs/extent-tree.c
index e2f16b68fcbf..51b5e2da708c 100644
--- a/fs/btrfs/extent-tree.c
+++ b/fs/btrfs/extent-tree.c
@@ -3142,7 +3142,11 @@ static noinline int check_delayed_ref(struct btrfs_root *root,
         struct rb_node *node;
         int ret = 0;
 
+        spin_lock(&root->fs_info->trans_lock);
         cur_trans = root->fs_info->running_transaction;
+        if (cur_trans)
+                refcount_inc(&cur_trans->use_count);
+        spin_unlock(&root->fs_info->trans_lock);
         if (!cur_trans)
                 return 0;
 
@@ -3151,6 +3155,7 @@ static noinline int check_delayed_ref(struct btrfs_root *root,
         head = btrfs_find_delayed_ref_head(delayed_refs, bytenr);
         if (!head) {
                 spin_unlock(&delayed_refs->lock);
+                btrfs_put_transaction(cur_trans);
                 return 0;
         }
 
@@ -3167,6 +3172,7 @@ static noinline int check_delayed_ref(struct btrfs_root *root,
                 mutex_lock(&head->mutex);
                 mutex_unlock(&head->mutex);
                 btrfs_put_delayed_ref_head(head);
+                btrfs_put_transaction(cur_trans);
                 return -EAGAIN;
         }
         spin_unlock(&delayed_refs->lock);
@@ -3199,6 +3205,7 @@ static noinline int check_delayed_ref(struct btrfs_root *root,
         }
         spin_unlock(&head->lock);
         mutex_unlock(&head->mutex);
+        btrfs_put_transaction(cur_trans);
         return ret;
 }