--- /dev/null
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * Copyright (C) 2026 Western Digital. All rights reserved.
+ */
+
+#include <linux/cleanup.h>
+#include <linux/sizes.h>
+
+#include "btrfs-tests.h"
+#include "../space-info.h"
+#include "../volumes.h"
+#include "../zoned.h"
+
+#define WP_MISSING_DEV ((u64)-1)
+#define WP_CONVENTIONAL ((u64)-2)
+#define ZONE_SIZE SZ_256M
+
+#define HALF_STRIPE_LEN (BTRFS_STRIPE_LEN >> 1)
+
+struct load_zone_info_test_vector {
+ u64 raid_type;
+ u64 num_stripes;
+ u64 alloc_offsets[8];
+ u64 last_alloc;
+ u64 bg_length;
+ bool degraded;
+
+ int expected_result;
+ u64 expected_alloc_offset;
+
+ const char *description;
+};
+
+struct zone_info {
+ u64 physical;
+ u64 capacity;
+ u64 alloc_offset;
+};
+
+static int test_load_zone_info(struct btrfs_fs_info *fs_info,
+ const struct load_zone_info_test_vector *test)
+{
+ struct btrfs_block_group *bg __free(btrfs_free_dummy_block_group) = NULL;
+ struct btrfs_chunk_map *map __free(btrfs_free_chunk_map) = NULL;
+ struct zone_info AUTO_KFREE(zone_info);
+ unsigned long AUTO_KFREE(active);
+ int ret;
+
+ bg = btrfs_alloc_dummy_block_group(fs_info, test->bg_length);
+ if (!bg) {
+ test_std_err(TEST_ALLOC_BLOCK_GROUP);
+ return -ENOMEM;
+ }
+
+ map = btrfs_alloc_chunk_map(test->num_stripes, GFP_KERNEL);
+ if (!map) {
+ test_std_err(TEST_ALLOC_EXTENT_MAP);
+ return -ENOMEM;
+ }
+
+ zone_info = kcalloc(test->num_stripes, sizeof(*zone_info), GFP_KERNEL);
+ if (!zone_info) {
+ test_err("cannot allocate zone info");
+ return -ENOMEM;
+ }
+
+ active = bitmap_zalloc(test->num_stripes, GFP_KERNEL);
+ if (!zone_info) {
+ test_err("cannot allocate active bitmap");
+ return -ENOMEM;
+ }
+
+ map->type = test->raid_type;
+ map->num_stripes = test->num_stripes;
+ if (test->raid_type == BTRFS_BLOCK_GROUP_RAID10)
+ map->sub_stripes = 2;
+ for (int i = 0; i < test->num_stripes; i++) {
+ zone_info[i].physical = 0;
+ zone_info[i].alloc_offset = test->alloc_offsets[i];
+ zone_info[i].capacity = ZONE_SIZE;
+ if (zone_info[i].alloc_offset && zone_info[i].alloc_offset < ZONE_SIZE)
+ __set_bit(i, active);
+ }
+ if (test->degraded)
+ btrfs_set_opt(fs_info->mount_opt, DEGRADED);
+ else
+ btrfs_clear_opt(fs_info->mount_opt, DEGRADED);
+
+ ret = btrfs_load_block_group_by_raid_type(bg, map, zone_info, active,
+ test->last_alloc);
+
+ if (ret != test->expected_result) {
+ test_err("unexpected return value: ret %d expected %d", ret,
+ test->expected_result);
+ return -EINVAL;
+ }
+
+ if (!ret && bg->alloc_offset != test->expected_alloc_offset) {
+ test_err("unexpected alloc_offset: alloc_offset %llu expected %llu",
+ bg->alloc_offset, test->expected_alloc_offset);
+ return -EINVAL;
+ }
+
+ return 0;
+}
+
+static const struct load_zone_info_test_vector load_zone_info_tests[] = {
+ /* SINGLE */
+ {
+ .description = "SINGLE: load write pointer from sequential zone",
+ .raid_type = 0,
+ .num_stripes = 1,
+ .alloc_offsets = {
+ SZ_1M,
+ },
+ .expected_alloc_offset = SZ_1M,
+ },
+ /*
+ * SINGLE block group on a conventional zone sets last_alloc outside of
+ * btrfs_load_block_group_*(). Do not test that case.
+ */
+
+ /* DUP */
+ /* Normal case */
+ {
+ .description = "DUP: having matching write pointers",
+ .raid_type = BTRFS_BLOCK_GROUP_DUP,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, SZ_1M,
+ },
+ .expected_alloc_offset = SZ_1M,
+ },
+ /*
+ * One sequential zone and one conventional zone, having matching
+ * last_alloc.
+ */
+ {
+ .description = "DUP: seq zone and conv zone, matching last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_DUP,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_1M,
+ .expected_alloc_offset = SZ_1M,
+ },
+ /*
+ * One sequential and one conventional zone, but having smaller
+ * last_alloc than write pointer.
+ */
+ {
+ .description = "DUP: seq zone and conv zone, smaller last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_DUP,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = 0,
+ .expected_alloc_offset = SZ_1M,
+ },
+ /* Error case: having different write pointers. */
+ {
+ .description = "DUP: fail: different write pointers",
+ .raid_type = BTRFS_BLOCK_GROUP_DUP,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, SZ_2M,
+ },
+ .expected_result = -EIO,
+ },
+ /* Error case: partial missing device should not happen on DUP. */
+ {
+ .description = "DUP: fail: missing device",
+ .raid_type = BTRFS_BLOCK_GROUP_DUP,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_MISSING_DEV,
+ },
+ .expected_result = -EIO,
+ },
+ /*
+ * Error case: one sequential and one conventional zone, but having larger
+ * last_alloc than write pointer.
+ */
+ {
+ .description = "DUP: fail: seq zone and conv zone, larger last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_DUP,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_2M,
+ .expected_result = -EIO,
+ },
+
+ /* RAID1 */
+ /* Normal case */
+ {
+ .description = "RAID1: having matching write pointers",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID1,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, SZ_1M,
+ },
+ .expected_alloc_offset = SZ_1M,
+ },
+ /*
+ * One sequential zone and one conventional zone, having matching
+ * last_alloc.
+ */
+ {
+ .description = "RAID1: seq zone and conv zone, matching last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID1,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_1M,
+ .expected_alloc_offset = SZ_1M,
+ },
+ /*
+ * One sequential and one conventional zone, but having smaller
+ * last_alloc than write pointer.
+ */
+ {
+ .description = "RAID1: seq zone and conv zone, smaller last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID1,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = 0,
+ .expected_alloc_offset = SZ_1M,
+ },
+ /* Partial missing device should be recovered on DEGRADED mount */
+ {
+ .description = "RAID1: fail: missing device on DEGRADED",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID1,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_MISSING_DEV,
+ },
+ .degraded = true,
+ .expected_alloc_offset = SZ_1M,
+ },
+ /* Error case: having different write pointers. */
+ {
+ .description = "RAID1: fail: different write pointers",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID1,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, SZ_2M,
+ },
+ .expected_result = -EIO,
+ },
+ /*
+ * Partial missing device is not allowed on non-DEGRADED mount never happen
+ * as it is rejected beforehand.
+ */
+ /*
+ * Error case: one sequential and one conventional zone, but having larger
+ * last_alloc than write pointer.
+ */
+ {
+ .description = "RAID1: fail: seq zone and conv zone, larger last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID1,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_2M,
+ .expected_result = -EIO,
+ },
+
+ /* RAID0 */
+ /* Normal case */
+ {
+ .description = "RAID0: initial partial write",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ HALF_STRIPE_LEN, 0, 0, 0,
+ },
+ .expected_alloc_offset = HALF_STRIPE_LEN,
+ },
+ {
+ .description = "RAID0: while in second stripe",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN + HALF_STRIPE_LEN,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ },
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 5 + HALF_STRIPE_LEN,
+ },
+ {
+ .description = "RAID0: one stripe advanced",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M + BTRFS_STRIPE_LEN, SZ_1M,
+ },
+ .expected_alloc_offset = SZ_2M + BTRFS_STRIPE_LEN,
+ },
+ /* Error case: having different write pointers. */
+ {
+ .description = "RAID0: fail: disordered stripes",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN * 2,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ },
+ .expected_result = -EIO,
+ },
+ {
+ .description = "RAID0: fail: far distance",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 3, BTRFS_STRIPE_LEN,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ },
+ .expected_result = -EIO,
+ },
+ {
+ .description = "RAID0: fail: too many partial write",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ HALF_STRIPE_LEN, HALF_STRIPE_LEN, 0, 0,
+ },
+ .expected_result = -EIO,
+ },
+ /*
+ * Error case: Partial missing device is not allowed even on non-DEGRADED
+ * mount.
+ */
+ {
+ .description = "RAID0: fail: missing device on DEGRADED",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_MISSING_DEV,
+ },
+ .degraded = true,
+ .expected_result = -EIO,
+ },
+
+ /*
+ * One sequential zone and one conventional zone, having matching
+ * last_alloc.
+ */
+ {
+ .description = "RAID0: seq zone and conv zone, partially written stripe",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_2M - SZ_4K,
+ .expected_alloc_offset = SZ_2M - SZ_4K,
+ },
+ {
+ .description = "RAID0: conv zone and seq zone, partially written stripe",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ WP_CONVENTIONAL, SZ_1M,
+ },
+ .last_alloc = SZ_2M + SZ_4K,
+ .expected_alloc_offset = SZ_2M + SZ_4K,
+ },
+ /*
+ * Error case: one sequential and one conventional zone, but having larger
+ * last_alloc than write pointer.
+ */
+ {
+ .description = "RAID0: fail: seq zone and conv zone, larger last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 2,
+ .alloc_offsets = {
+ SZ_1M, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_2M + BTRFS_STRIPE_LEN * 2,
+ .expected_result = -EIO,
+ },
+
+ /* RAID0, 4 stripes with seq zones and conv zones. */
+ {
+ .description = "RAID0: stripes [2, 2, ?, ?] last_alloc = 6",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 6,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 6,
+ },
+ {
+ .description = "RAID0: stripes [2, 2, ?, ?] last_alloc = 7.5",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 7 + HALF_STRIPE_LEN,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 7 + HALF_STRIPE_LEN,
+ },
+ {
+ .description = "RAID0: stripes [3, ?, ?, ?] last_alloc = 1",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 3, WP_CONVENTIONAL,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 9,
+ },
+ {
+ .description = "RAID0: stripes [2, ?, 1, ?] last_alloc = 5",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, WP_CONVENTIONAL,
+ BTRFS_STRIPE_LEN, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 5,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 5,
+ },
+ {
+ .description = "RAID0: fail: stripes [2, ?, 1, ?] last_alloc = 7",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID0,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, WP_CONVENTIONAL,
+ BTRFS_STRIPE_LEN, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 7,
+ .expected_result = -EIO,
+ },
+
+ /* RAID10 */
+ /* Normal case */
+ {
+ .description = "RAID10: initial partial write",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ HALF_STRIPE_LEN, HALF_STRIPE_LEN, 0, 0,
+ },
+ .expected_alloc_offset = HALF_STRIPE_LEN,
+ },
+ {
+ .description = "RAID10: while in second stripe",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ BTRFS_STRIPE_LEN + HALF_STRIPE_LEN,
+ BTRFS_STRIPE_LEN + HALF_STRIPE_LEN,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ },
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 5 + HALF_STRIPE_LEN,
+ },
+ {
+ .description = "RAID10: one stripe advanced",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ SZ_1M + BTRFS_STRIPE_LEN, SZ_1M + BTRFS_STRIPE_LEN,
+ SZ_1M, SZ_1M,
+ },
+ .expected_alloc_offset = SZ_2M + BTRFS_STRIPE_LEN,
+ },
+ {
+ .description = "RAID10: one stripe advanced, with conventional zone",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ SZ_1M + BTRFS_STRIPE_LEN, WP_CONVENTIONAL,
+ WP_CONVENTIONAL, SZ_1M,
+ },
+ .expected_alloc_offset = SZ_2M + BTRFS_STRIPE_LEN,
+ },
+ /* Error case: having different write pointers. */
+ {
+ .description = "RAID10: fail: disordered stripes",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ },
+ .expected_result = -EIO,
+ },
+ {
+ .description = "RAID10: fail: far distance",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 3, BTRFS_STRIPE_LEN * 3,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ },
+ .expected_result = -EIO,
+ },
+ {
+ .description = "RAID10: fail: too many partial write",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ HALF_STRIPE_LEN, HALF_STRIPE_LEN,
+ HALF_STRIPE_LEN, HALF_STRIPE_LEN,
+ 0, 0, 0, 0,
+ },
+ .expected_result = -EIO,
+ },
+ /*
+ * Error case: Partial missing device in RAID0 level is not allowed even on
+ * non-DEGRADED mount.
+ */
+ {
+ .description = "RAID10: fail: missing device on DEGRADED",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ SZ_1M, SZ_1M,
+ WP_MISSING_DEV, WP_MISSING_DEV,
+ },
+ .degraded = true,
+ .expected_result = -EIO,
+ },
+
+ /*
+ * One sequential zone and one conventional zone, having matching
+ * last_alloc.
+ */
+ {
+ .description = "RAID10: seq zone and conv zone, partially written stripe",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ SZ_1M, SZ_1M,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_2M - SZ_4K,
+ .expected_alloc_offset = SZ_2M - SZ_4K,
+ },
+ {
+ .description = "RAID10: conv zone and seq zone, partially written stripe",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ SZ_1M, SZ_1M,
+ },
+ .last_alloc = SZ_2M + SZ_4K,
+ .expected_alloc_offset = SZ_2M + SZ_4K,
+ },
+ /*
+ * Error case: one sequential and one conventional zone, but having larger
+ * last_alloc than write pointer.
+ */
+ {
+ .description = "RAID10: fail: seq zone and conv zone, larger last_alloc",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 4,
+ .alloc_offsets = {
+ SZ_1M, SZ_1M,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = SZ_2M + BTRFS_STRIPE_LEN * 2,
+ .expected_result = -EIO,
+ },
+
+ /* RAID10, 4 stripes with seq zones and conv zones. */
+ {
+ .description = "RAID10: stripes [2, 2, ?, ?] last_alloc = 6",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 6,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 6,
+ },
+ {
+ .description = "RAID10: stripes [2, 2, ?, ?] last_alloc = 7.5",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 7 + HALF_STRIPE_LEN,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 7 + HALF_STRIPE_LEN,
+ },
+ {
+ .description = "RAID10: stripes [3, ?, ?, ?] last_alloc = 1",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 3, BTRFS_STRIPE_LEN * 3,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 9,
+ },
+ {
+ .description = "RAID10: stripes [2, ?, 1, ?] last_alloc = 5",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 5,
+ .expected_alloc_offset = BTRFS_STRIPE_LEN * 5,
+ },
+ {
+ .description = "RAID10: fail: stripes [2, ?, 1, ?] last_alloc = 7",
+ .raid_type = BTRFS_BLOCK_GROUP_RAID10,
+ .num_stripes = 8,
+ .alloc_offsets = {
+ BTRFS_STRIPE_LEN * 2, BTRFS_STRIPE_LEN * 2,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ BTRFS_STRIPE_LEN, BTRFS_STRIPE_LEN,
+ WP_CONVENTIONAL, WP_CONVENTIONAL,
+ },
+ .last_alloc = BTRFS_STRIPE_LEN * 7,
+ .expected_result = -EIO,
+ },
+};
+
+int btrfs_test_zoned(void)
+{
+ struct btrfs_fs_info *fs_info __free(btrfs_free_dummy_fs_info) = NULL;
+ int ret;
+
+ test_msg("running zoned tests (error messages are expected)");
+
+ fs_info = btrfs_alloc_dummy_fs_info(PAGE_SIZE, PAGE_SIZE);
+ if (!fs_info) {
+ test_std_err(TEST_ALLOC_FS_INFO);
+ return -ENOMEM;
+ }
+
+ for (int i = 0; i < ARRAY_SIZE(load_zone_info_tests); i++) {
+ ret = test_load_zone_info(fs_info, &load_zone_info_tests[i]);
+ if (ret) {
+ test_err("test case \"%s\" failed", load_zone_info_tests[i].description);
+ return ret;
+ }
+ }
+
+ return 0;
+}
projects / thirdparty / kernel / linux.git / commitdiff
