[PATCH v12 27/41] btrfs: enable zone append writing for direct IO

[Naohiro Aota]

Likewise to buffered IO, enable zone append writing for direct IO when its
used on a zoned block device.

Reviewed-by: Josef Bacik <jo...@toxicpanda.com>
Signed-off-by: Naohiro Aota <naohiro.a...@wdc.com>
---
 fs/btrfs/inode.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/fs/btrfs/inode.c b/fs/btrfs/inode.c
index 6b5f273a0d83..4f0915346c9d 100644
--- a/fs/btrfs/inode.c
+++ b/fs/btrfs/inode.c
@@ -7708,6 +7708,9 @@ static int btrfs_dio_iomap_begin(struct inode *inode, 
loff_t start,
        iomap->bdev = fs_info->fs_devices->latest_bdev;
        iomap->length = len;
 
+       if (write && btrfs_use_zone_append(BTRFS_I(inode), em))
+               iomap->flags |= IOMAP_F_ZONE_APPEND;
+
        free_extent_map(em);
 
        return 0;
@@ -7936,6 +7939,8 @@ static void btrfs_end_dio_bio(struct bio *bio)
        if (err)
                dip->dio_bio->bi_status = err;
 
+       btrfs_record_physical_zoned(dip->inode, dip->logical_offset, bio);
+
        bio_put(bio);
        btrfs_dio_private_put(dip);
 }
@@ -8088,6 +8093,18 @@ static blk_qc_t btrfs_submit_direct(struct inode *inode, 
struct iomap *iomap,
                bio->bi_end_io = btrfs_end_dio_bio;
                btrfs_io_bio(bio)->logical = file_offset;
 
+               WARN_ON_ONCE(write && btrfs_is_zoned(fs_info) &&
+                            fs_info->max_zone_append_size &&
+                            bio_op(bio) != REQ_OP_ZONE_APPEND);
+
+               if (bio_op(bio) == REQ_OP_ZONE_APPEND) {
+                       ret = extract_ordered_extent(inode, bio, file_offset);
+                       if (ret) {
+                               bio_put(bio);
+                               goto out_err;
+                       }
+               }
+
                ASSERT(submit_len >= clone_len);
                submit_len -= clone_len;
 
-- 
2.27.0