[PATCH 5/5] Btrfs: extent_write_locked_range() should attach inode->i_wb

[Tejun Heo]

From: Chris Mason <c...@fb.com>

extent_write_locked_range() is used when we're falling back to buffered
IO from inside of compression.  It allocates its own wbc and should
associate it with the inode's i_wb to make sure the IO goes down from
the correct cgroup.

Signed-off-by: Chris Mason <c...@fb.com>
Reviewed-by: Josef Bacik <jo...@toxicpanda.com>
---
 fs/btrfs/extent_io.c | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/fs/btrfs/extent_io.c b/fs/btrfs/extent_io.c
index 3f3942618e92..5606a38b64ff 100644
--- a/fs/btrfs/extent_io.c
+++ b/fs/btrfs/extent_io.c
@@ -4178,6 +4178,7 @@ int extent_write_locked_range(struct inode *inode, u64 
start, u64 end,
                .no_cgroup_owner = 1,
        };
 
+       wbc_attach_fdatawrite_inode(&wbc_writepages, inode);
        while (start <= end) {
                page = find_get_page(mapping, start >> PAGE_SHIFT);
                if (clear_page_dirty_for_io(page))
@@ -4192,11 +4193,12 @@ int extent_write_locked_range(struct inode *inode, u64 
start, u64 end,
        }
 
        ASSERT(ret <= 0);
-       if (ret < 0) {
+       if (ret == 0)
+               ret = flush_write_bio(&epd);
+       else
                end_write_bio(&epd, ret);
-               return ret;
-       }
-       ret = flush_write_bio(&epd);
+
+       wbc_detach_inode(&wbc_writepages);
        return ret;
 }
 
-- 
2.17.1