[PATCH 05/12] btrfs-progs: load and check zone information

[Naohiro Aota]

This patch checks if a device added to btrfs is a zoned block device. If it
is, load zones information and the zone size for the device.

For a btrfs volume composed of multiple zoned block devices, all devices
must have the same zone size.

Signed-off-by: Naohiro Aota <naohiro.a...@wdc.com>
---
 utils.c   | 10 ++++++++++
 volumes.c | 18 ++++++++++++++++++
 volumes.h |  3 +++
 3 files changed, 31 insertions(+)

diff --git a/utils.c b/utils.c
index d50304b1be80..a26fe7a5743c 100644
--- a/utils.c
+++ b/utils.c
@@ -250,6 +250,16 @@ int btrfs_add_to_fsid(struct btrfs_trans_handle *trans,
                goto out;
        }
 
+       ret = btrfs_get_zone_info(fd, path, fs_info->fs_devices->hmzoned,
+                                 &device->zinfo);
+       if (ret)
+               goto out;
+       if (device->zinfo.zone_size != fs_info->fs_devices->zone_size) {
+               error("Device zone size differ\n");
+               ret = -EINVAL;
+               goto out;
+       }
+
        disk_super = (struct btrfs_super_block *)buf;
        dev_item = &disk_super->dev_item;
 
diff --git a/volumes.c b/volumes.c
index 3a91b43b378b..f6d1b1e9dc7f 100644
--- a/volumes.c
+++ b/volumes.c
@@ -168,6 +168,8 @@ static int device_list_add(const char *path,
        u64 found_transid = btrfs_super_generation(disk_super);
        bool metadata_uuid = (btrfs_super_incompat_flags(disk_super) &
                BTRFS_FEATURE_INCOMPAT_METADATA_UUID);
+       int hmzoned = btrfs_super_incompat_flags(disk_super) &
+                       BTRFS_FEATURE_INCOMPAT_HMZONED;
 
        if (metadata_uuid)
                fs_devices = find_fsid(disk_super->fsid,
@@ -257,6 +259,8 @@ static int device_list_add(const char *path,
        if (fs_devices->lowest_devid > devid) {
                fs_devices->lowest_devid = devid;
        }
+       if (hmzoned)
+               fs_devices->hmzoned = 1;
        *fs_devices_ret = fs_devices;
        return 0;
 }
@@ -327,6 +331,8 @@ int btrfs_open_devices(struct btrfs_fs_devices *fs_devices, 
int flags)
        struct btrfs_device *device;
        int ret;
 
+       fs_devices->zone_size = 0;
+
        list_for_each_entry(device, &fs_devices->devices, dev_list) {
                if (!device->name) {
                        printk("no name for device %llu, skip it now\n", 
device->devid);
@@ -350,6 +356,18 @@ int btrfs_open_devices(struct btrfs_fs_devices 
*fs_devices, int flags)
                device->fd = fd;
                if (flags & O_RDWR)
                        device->writeable = 1;
+
+               ret = btrfs_get_zone_info(fd, device->name, fs_devices->hmzoned,
+                                         &device->zinfo);
+               if (ret != 0)
+                       goto fail;
+               if (!fs_devices->zone_size) {
+                       fs_devices->zone_size = device->zinfo.zone_size;
+               } else if (device->zinfo.zone_size != fs_devices->zone_size) {
+                       fprintf(stderr, "Device zone size differ\n");
+                       ret = -EINVAL;
+                       goto fail;
+               }
        }
        return 0;
 fail:
diff --git a/volumes.h b/volumes.h
index c9262ceaea93..6ec83fe43cfe 100644
--- a/volumes.h
+++ b/volumes.h
@@ -115,6 +115,9 @@ struct btrfs_fs_devices {
 
        int seeding;
        struct btrfs_fs_devices *seed;
+
+       u64 zone_size;
+       unsigned int hmzoned:1;
 };
 
 struct btrfs_bio_stripe {
-- 
2.21.0