1eb0570c7SDamien Le Moal // SPDX-License-Identifier: GPL-2.0-only 2eb0570c7SDamien Le Moal /* 3eb0570c7SDamien Le Moal * Copyright (c) 2025, Christoph Hellwig. 4eb0570c7SDamien Le Moal * Copyright (c) 2025, Western Digital Corporation or its affiliates. 5eb0570c7SDamien Le Moal * 6eb0570c7SDamien Le Moal * Zoned Loop Device driver - exports a zoned block device using one file per 7eb0570c7SDamien Le Moal * zone as backing storage. 8eb0570c7SDamien Le Moal */ 9eb0570c7SDamien Le Moal #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 10eb0570c7SDamien Le Moal 11eb0570c7SDamien Le Moal #include <linux/module.h> 12eb0570c7SDamien Le Moal #include <linux/blk-mq.h> 13eb0570c7SDamien Le Moal #include <linux/blkzoned.h> 14eb0570c7SDamien Le Moal #include <linux/pagemap.h> 15eb0570c7SDamien Le Moal #include <linux/miscdevice.h> 16eb0570c7SDamien Le Moal #include <linux/falloc.h> 17eb0570c7SDamien Le Moal #include <linux/mutex.h> 18eb0570c7SDamien Le Moal #include <linux/parser.h> 19eb0570c7SDamien Le Moal #include <linux/seq_file.h> 20eb0570c7SDamien Le Moal 21eb0570c7SDamien Le Moal /* 22eb0570c7SDamien Le Moal * Options for adding (and removing) a device. 23eb0570c7SDamien Le Moal */ 24eb0570c7SDamien Le Moal enum { 25eb0570c7SDamien Le Moal ZLOOP_OPT_ERR = 0, 26eb0570c7SDamien Le Moal ZLOOP_OPT_ID = (1 << 0), 27eb0570c7SDamien Le Moal ZLOOP_OPT_CAPACITY = (1 << 1), 28eb0570c7SDamien Le Moal ZLOOP_OPT_ZONE_SIZE = (1 << 2), 29eb0570c7SDamien Le Moal ZLOOP_OPT_ZONE_CAPACITY = (1 << 3), 30eb0570c7SDamien Le Moal ZLOOP_OPT_NR_CONV_ZONES = (1 << 4), 31eb0570c7SDamien Le Moal ZLOOP_OPT_BASE_DIR = (1 << 5), 32eb0570c7SDamien Le Moal ZLOOP_OPT_NR_QUEUES = (1 << 6), 33eb0570c7SDamien Le Moal ZLOOP_OPT_QUEUE_DEPTH = (1 << 7), 34eb0570c7SDamien Le Moal ZLOOP_OPT_BUFFERED_IO = (1 << 8), 359236c5fdSDamien Le Moal ZLOOP_OPT_ZONE_APPEND = (1 << 9), 36fcc6eaa3SDamien Le Moal ZLOOP_OPT_ORDERED_ZONE_APPEND = (1 << 10), 37eb0570c7SDamien Le Moal }; 38eb0570c7SDamien Le Moal 39eb0570c7SDamien Le Moal static const match_table_t zloop_opt_tokens = { 40eb0570c7SDamien Le Moal { ZLOOP_OPT_ID, "id=%d" }, 41eb0570c7SDamien Le Moal { ZLOOP_OPT_CAPACITY, "capacity_mb=%u" }, 42eb0570c7SDamien Le Moal { ZLOOP_OPT_ZONE_SIZE, "zone_size_mb=%u" }, 43eb0570c7SDamien Le Moal { ZLOOP_OPT_ZONE_CAPACITY, "zone_capacity_mb=%u" }, 44eb0570c7SDamien Le Moal { ZLOOP_OPT_NR_CONV_ZONES, "conv_zones=%u" }, 45eb0570c7SDamien Le Moal { ZLOOP_OPT_BASE_DIR, "base_dir=%s" }, 46eb0570c7SDamien Le Moal { ZLOOP_OPT_NR_QUEUES, "nr_queues=%u" }, 47eb0570c7SDamien Le Moal { ZLOOP_OPT_QUEUE_DEPTH, "queue_depth=%u" }, 48eb0570c7SDamien Le Moal { ZLOOP_OPT_BUFFERED_IO, "buffered_io" }, 499236c5fdSDamien Le Moal { ZLOOP_OPT_ZONE_APPEND, "zone_append=%u" }, 50fcc6eaa3SDamien Le Moal { ZLOOP_OPT_ORDERED_ZONE_APPEND, "ordered_zone_append" }, 51eb0570c7SDamien Le Moal { ZLOOP_OPT_ERR, NULL } 52eb0570c7SDamien Le Moal }; 53eb0570c7SDamien Le Moal 54eb0570c7SDamien Le Moal /* Default values for the "add" operation. */ 55eb0570c7SDamien Le Moal #define ZLOOP_DEF_ID -1 56eb0570c7SDamien Le Moal #define ZLOOP_DEF_ZONE_SIZE ((256ULL * SZ_1M) >> SECTOR_SHIFT) 57eb0570c7SDamien Le Moal #define ZLOOP_DEF_NR_ZONES 64 58eb0570c7SDamien Le Moal #define ZLOOP_DEF_NR_CONV_ZONES 8 59eb0570c7SDamien Le Moal #define ZLOOP_DEF_BASE_DIR "/var/local/zloop" 60eb0570c7SDamien Le Moal #define ZLOOP_DEF_NR_QUEUES 1 61eb0570c7SDamien Le Moal #define ZLOOP_DEF_QUEUE_DEPTH 128 62eb0570c7SDamien Le Moal #define ZLOOP_DEF_BUFFERED_IO false 639236c5fdSDamien Le Moal #define ZLOOP_DEF_ZONE_APPEND true 64fcc6eaa3SDamien Le Moal #define ZLOOP_DEF_ORDERED_ZONE_APPEND false 65eb0570c7SDamien Le Moal 66eb0570c7SDamien Le Moal /* Arbitrary limit on the zone size (16GB). */ 67eb0570c7SDamien Le Moal #define ZLOOP_MAX_ZONE_SIZE_MB 16384 68eb0570c7SDamien Le Moal 69eb0570c7SDamien Le Moal struct zloop_options { 70eb0570c7SDamien Le Moal unsigned int mask; 71eb0570c7SDamien Le Moal int id; 72eb0570c7SDamien Le Moal sector_t capacity; 73eb0570c7SDamien Le Moal sector_t zone_size; 74eb0570c7SDamien Le Moal sector_t zone_capacity; 75eb0570c7SDamien Le Moal unsigned int nr_conv_zones; 76eb0570c7SDamien Le Moal char *base_dir; 77eb0570c7SDamien Le Moal unsigned int nr_queues; 78eb0570c7SDamien Le Moal unsigned int queue_depth; 79eb0570c7SDamien Le Moal bool buffered_io; 809236c5fdSDamien Le Moal bool zone_append; 81fcc6eaa3SDamien Le Moal bool ordered_zone_append; 82eb0570c7SDamien Le Moal }; 83eb0570c7SDamien Le Moal 84eb0570c7SDamien Le Moal /* 85eb0570c7SDamien Le Moal * Device states. 86eb0570c7SDamien Le Moal */ 87eb0570c7SDamien Le Moal enum { 88eb0570c7SDamien Le Moal Zlo_creating = 0, 89eb0570c7SDamien Le Moal Zlo_live, 90eb0570c7SDamien Le Moal Zlo_deleting, 91eb0570c7SDamien Le Moal }; 92eb0570c7SDamien Le Moal 93eb0570c7SDamien Le Moal enum zloop_zone_flags { 94eb0570c7SDamien Le Moal ZLOOP_ZONE_CONV = 0, 95eb0570c7SDamien Le Moal ZLOOP_ZONE_SEQ_ERROR, 96eb0570c7SDamien Le Moal }; 97eb0570c7SDamien Le Moal 98eb0570c7SDamien Le Moal struct zloop_zone { 99eb0570c7SDamien Le Moal struct file *file; 100eb0570c7SDamien Le Moal 101eb0570c7SDamien Le Moal unsigned long flags; 102eb0570c7SDamien Le Moal struct mutex lock; 103fcc6eaa3SDamien Le Moal spinlock_t wp_lock; 104eb0570c7SDamien Le Moal enum blk_zone_cond cond; 105eb0570c7SDamien Le Moal sector_t start; 106eb0570c7SDamien Le Moal sector_t wp; 107eb0570c7SDamien Le Moal 108eb0570c7SDamien Le Moal gfp_t old_gfp_mask; 109eb0570c7SDamien Le Moal }; 110eb0570c7SDamien Le Moal 111eb0570c7SDamien Le Moal struct zloop_device { 112eb0570c7SDamien Le Moal unsigned int id; 113eb0570c7SDamien Le Moal unsigned int state; 114eb0570c7SDamien Le Moal 115eb0570c7SDamien Le Moal struct blk_mq_tag_set tag_set; 116eb0570c7SDamien Le Moal struct gendisk *disk; 117eb0570c7SDamien Le Moal 118eb0570c7SDamien Le Moal struct workqueue_struct *workqueue; 119eb0570c7SDamien Le Moal bool buffered_io; 1209236c5fdSDamien Le Moal bool zone_append; 121fcc6eaa3SDamien Le Moal bool ordered_zone_append; 122eb0570c7SDamien Le Moal 123eb0570c7SDamien Le Moal const char *base_dir; 124eb0570c7SDamien Le Moal struct file *data_dir; 125eb0570c7SDamien Le Moal 126eb0570c7SDamien Le Moal unsigned int zone_shift; 127eb0570c7SDamien Le Moal sector_t zone_size; 128eb0570c7SDamien Le Moal sector_t zone_capacity; 129eb0570c7SDamien Le Moal unsigned int nr_zones; 130eb0570c7SDamien Le Moal unsigned int nr_conv_zones; 131eb0570c7SDamien Le Moal unsigned int block_size; 132eb0570c7SDamien Le Moal 133eb0570c7SDamien Le Moal struct zloop_zone zones[] __counted_by(nr_zones); 134eb0570c7SDamien Le Moal }; 135eb0570c7SDamien Le Moal 136eb0570c7SDamien Le Moal struct zloop_cmd { 137eb0570c7SDamien Le Moal struct work_struct work; 138eb0570c7SDamien Le Moal atomic_t ref; 139eb0570c7SDamien Le Moal sector_t sector; 140eb0570c7SDamien Le Moal sector_t nr_sectors; 141eb0570c7SDamien Le Moal long ret; 142eb0570c7SDamien Le Moal struct kiocb iocb; 143eb0570c7SDamien Le Moal struct bio_vec *bvec; 144eb0570c7SDamien Le Moal }; 145eb0570c7SDamien Le Moal 146eb0570c7SDamien Le Moal static DEFINE_IDR(zloop_index_idr); 147eb0570c7SDamien Le Moal static DEFINE_MUTEX(zloop_ctl_mutex); 148eb0570c7SDamien Le Moal 149eb0570c7SDamien Le Moal static unsigned int rq_zone_no(struct request *rq) 150eb0570c7SDamien Le Moal { 151eb0570c7SDamien Le Moal struct zloop_device *zlo = rq->q->queuedata; 152eb0570c7SDamien Le Moal 153eb0570c7SDamien Le Moal return blk_rq_pos(rq) >> zlo->zone_shift; 154eb0570c7SDamien Le Moal } 155eb0570c7SDamien Le Moal 156eb0570c7SDamien Le Moal static int zloop_update_seq_zone(struct zloop_device *zlo, unsigned int zone_no) 157eb0570c7SDamien Le Moal { 158eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 159eb0570c7SDamien Le Moal struct kstat stat; 160eb0570c7SDamien Le Moal sector_t file_sectors; 161fcc6eaa3SDamien Le Moal unsigned long flags; 162eb0570c7SDamien Le Moal int ret; 163eb0570c7SDamien Le Moal 164eb0570c7SDamien Le Moal lockdep_assert_held(&zone->lock); 165eb0570c7SDamien Le Moal 166eb0570c7SDamien Le Moal ret = vfs_getattr(&zone->file->f_path, &stat, STATX_SIZE, 0); 167eb0570c7SDamien Le Moal if (ret < 0) { 168eb0570c7SDamien Le Moal pr_err("Failed to get zone %u file stat (err=%d)\n", 169eb0570c7SDamien Le Moal zone_no, ret); 170eb0570c7SDamien Le Moal set_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags); 171eb0570c7SDamien Le Moal return ret; 172eb0570c7SDamien Le Moal } 173eb0570c7SDamien Le Moal 174eb0570c7SDamien Le Moal file_sectors = stat.size >> SECTOR_SHIFT; 175eb0570c7SDamien Le Moal if (file_sectors > zlo->zone_capacity) { 176eb0570c7SDamien Le Moal pr_err("Zone %u file too large (%llu sectors > %llu)\n", 177eb0570c7SDamien Le Moal zone_no, file_sectors, zlo->zone_capacity); 178eb0570c7SDamien Le Moal return -EINVAL; 179eb0570c7SDamien Le Moal } 180eb0570c7SDamien Le Moal 181eb0570c7SDamien Le Moal if (file_sectors & ((zlo->block_size >> SECTOR_SHIFT) - 1)) { 182eb0570c7SDamien Le Moal pr_err("Zone %u file size not aligned to block size %u\n", 183eb0570c7SDamien Le Moal zone_no, zlo->block_size); 184eb0570c7SDamien Le Moal return -EINVAL; 185eb0570c7SDamien Le Moal } 186eb0570c7SDamien Le Moal 187fcc6eaa3SDamien Le Moal spin_lock_irqsave(&zone->wp_lock, flags); 188eb0570c7SDamien Le Moal if (!file_sectors) { 189eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_EMPTY; 190eb0570c7SDamien Le Moal zone->wp = zone->start; 191eb0570c7SDamien Le Moal } else if (file_sectors == zlo->zone_capacity) { 192eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_FULL; 193866d6574SDamien Le Moal zone->wp = ULLONG_MAX; 194eb0570c7SDamien Le Moal } else { 195eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_CLOSED; 196eb0570c7SDamien Le Moal zone->wp = zone->start + file_sectors; 197eb0570c7SDamien Le Moal } 198fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 199eb0570c7SDamien Le Moal 200eb0570c7SDamien Le Moal return 0; 201eb0570c7SDamien Le Moal } 202eb0570c7SDamien Le Moal 203eb0570c7SDamien Le Moal static int zloop_open_zone(struct zloop_device *zlo, unsigned int zone_no) 204eb0570c7SDamien Le Moal { 205eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 206eb0570c7SDamien Le Moal int ret = 0; 207eb0570c7SDamien Le Moal 208eb0570c7SDamien Le Moal if (test_bit(ZLOOP_ZONE_CONV, &zone->flags)) 209eb0570c7SDamien Le Moal return -EIO; 210eb0570c7SDamien Le Moal 211eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 212eb0570c7SDamien Le Moal 213eb0570c7SDamien Le Moal if (test_and_clear_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags)) { 214eb0570c7SDamien Le Moal ret = zloop_update_seq_zone(zlo, zone_no); 215eb0570c7SDamien Le Moal if (ret) 216eb0570c7SDamien Le Moal goto unlock; 217eb0570c7SDamien Le Moal } 218eb0570c7SDamien Le Moal 219eb0570c7SDamien Le Moal switch (zone->cond) { 220eb0570c7SDamien Le Moal case BLK_ZONE_COND_EXP_OPEN: 221eb0570c7SDamien Le Moal break; 222eb0570c7SDamien Le Moal case BLK_ZONE_COND_EMPTY: 223eb0570c7SDamien Le Moal case BLK_ZONE_COND_CLOSED: 224eb0570c7SDamien Le Moal case BLK_ZONE_COND_IMP_OPEN: 225eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_EXP_OPEN; 226eb0570c7SDamien Le Moal break; 227eb0570c7SDamien Le Moal case BLK_ZONE_COND_FULL: 228eb0570c7SDamien Le Moal default: 229eb0570c7SDamien Le Moal ret = -EIO; 230eb0570c7SDamien Le Moal break; 231eb0570c7SDamien Le Moal } 232eb0570c7SDamien Le Moal 233eb0570c7SDamien Le Moal unlock: 234eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 235eb0570c7SDamien Le Moal 236eb0570c7SDamien Le Moal return ret; 237eb0570c7SDamien Le Moal } 238eb0570c7SDamien Le Moal 239eb0570c7SDamien Le Moal static int zloop_close_zone(struct zloop_device *zlo, unsigned int zone_no) 240eb0570c7SDamien Le Moal { 241eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 242fcc6eaa3SDamien Le Moal unsigned long flags; 243eb0570c7SDamien Le Moal int ret = 0; 244eb0570c7SDamien Le Moal 245eb0570c7SDamien Le Moal if (test_bit(ZLOOP_ZONE_CONV, &zone->flags)) 246eb0570c7SDamien Le Moal return -EIO; 247eb0570c7SDamien Le Moal 248eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 249eb0570c7SDamien Le Moal 250eb0570c7SDamien Le Moal if (test_and_clear_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags)) { 251eb0570c7SDamien Le Moal ret = zloop_update_seq_zone(zlo, zone_no); 252eb0570c7SDamien Le Moal if (ret) 253eb0570c7SDamien Le Moal goto unlock; 254eb0570c7SDamien Le Moal } 255eb0570c7SDamien Le Moal 256eb0570c7SDamien Le Moal switch (zone->cond) { 257eb0570c7SDamien Le Moal case BLK_ZONE_COND_CLOSED: 258eb0570c7SDamien Le Moal break; 259eb0570c7SDamien Le Moal case BLK_ZONE_COND_IMP_OPEN: 260eb0570c7SDamien Le Moal case BLK_ZONE_COND_EXP_OPEN: 261fcc6eaa3SDamien Le Moal spin_lock_irqsave(&zone->wp_lock, flags); 262eb0570c7SDamien Le Moal if (zone->wp == zone->start) 263eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_EMPTY; 264eb0570c7SDamien Le Moal else 265eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_CLOSED; 266fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 267eb0570c7SDamien Le Moal break; 268eb0570c7SDamien Le Moal case BLK_ZONE_COND_EMPTY: 269eb0570c7SDamien Le Moal case BLK_ZONE_COND_FULL: 270eb0570c7SDamien Le Moal default: 271eb0570c7SDamien Le Moal ret = -EIO; 272eb0570c7SDamien Le Moal break; 273eb0570c7SDamien Le Moal } 274eb0570c7SDamien Le Moal 275eb0570c7SDamien Le Moal unlock: 276eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 277eb0570c7SDamien Le Moal 278eb0570c7SDamien Le Moal return ret; 279eb0570c7SDamien Le Moal } 280eb0570c7SDamien Le Moal 281eb0570c7SDamien Le Moal static int zloop_reset_zone(struct zloop_device *zlo, unsigned int zone_no) 282eb0570c7SDamien Le Moal { 283eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 284fcc6eaa3SDamien Le Moal unsigned long flags; 285eb0570c7SDamien Le Moal int ret = 0; 286eb0570c7SDamien Le Moal 287eb0570c7SDamien Le Moal if (test_bit(ZLOOP_ZONE_CONV, &zone->flags)) 288eb0570c7SDamien Le Moal return -EIO; 289eb0570c7SDamien Le Moal 290eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 291eb0570c7SDamien Le Moal 292eb0570c7SDamien Le Moal if (!test_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags) && 293eb0570c7SDamien Le Moal zone->cond == BLK_ZONE_COND_EMPTY) 294eb0570c7SDamien Le Moal goto unlock; 295eb0570c7SDamien Le Moal 296eb0570c7SDamien Le Moal if (vfs_truncate(&zone->file->f_path, 0)) { 297eb0570c7SDamien Le Moal set_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags); 298eb0570c7SDamien Le Moal ret = -EIO; 299eb0570c7SDamien Le Moal goto unlock; 300eb0570c7SDamien Le Moal } 301eb0570c7SDamien Le Moal 302fcc6eaa3SDamien Le Moal spin_lock_irqsave(&zone->wp_lock, flags); 303eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_EMPTY; 304eb0570c7SDamien Le Moal zone->wp = zone->start; 305eb0570c7SDamien Le Moal clear_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags); 306fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 307eb0570c7SDamien Le Moal 308eb0570c7SDamien Le Moal unlock: 309eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 310eb0570c7SDamien Le Moal 311eb0570c7SDamien Le Moal return ret; 312eb0570c7SDamien Le Moal } 313eb0570c7SDamien Le Moal 314eb0570c7SDamien Le Moal static int zloop_reset_all_zones(struct zloop_device *zlo) 315eb0570c7SDamien Le Moal { 316eb0570c7SDamien Le Moal unsigned int i; 317eb0570c7SDamien Le Moal int ret; 318eb0570c7SDamien Le Moal 319eb0570c7SDamien Le Moal for (i = zlo->nr_conv_zones; i < zlo->nr_zones; i++) { 320eb0570c7SDamien Le Moal ret = zloop_reset_zone(zlo, i); 321eb0570c7SDamien Le Moal if (ret) 322eb0570c7SDamien Le Moal return ret; 323eb0570c7SDamien Le Moal } 324eb0570c7SDamien Le Moal 325eb0570c7SDamien Le Moal return 0; 326eb0570c7SDamien Le Moal } 327eb0570c7SDamien Le Moal 328eb0570c7SDamien Le Moal static int zloop_finish_zone(struct zloop_device *zlo, unsigned int zone_no) 329eb0570c7SDamien Le Moal { 330eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 331fcc6eaa3SDamien Le Moal unsigned long flags; 332eb0570c7SDamien Le Moal int ret = 0; 333eb0570c7SDamien Le Moal 334eb0570c7SDamien Le Moal if (test_bit(ZLOOP_ZONE_CONV, &zone->flags)) 335eb0570c7SDamien Le Moal return -EIO; 336eb0570c7SDamien Le Moal 337eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 338eb0570c7SDamien Le Moal 339eb0570c7SDamien Le Moal if (!test_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags) && 340eb0570c7SDamien Le Moal zone->cond == BLK_ZONE_COND_FULL) 341eb0570c7SDamien Le Moal goto unlock; 342eb0570c7SDamien Le Moal 343eb0570c7SDamien Le Moal if (vfs_truncate(&zone->file->f_path, zlo->zone_size << SECTOR_SHIFT)) { 344eb0570c7SDamien Le Moal set_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags); 345eb0570c7SDamien Le Moal ret = -EIO; 346eb0570c7SDamien Le Moal goto unlock; 347eb0570c7SDamien Le Moal } 348eb0570c7SDamien Le Moal 349fcc6eaa3SDamien Le Moal spin_lock_irqsave(&zone->wp_lock, flags); 350eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_FULL; 351866d6574SDamien Le Moal zone->wp = ULLONG_MAX; 352eb0570c7SDamien Le Moal clear_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags); 353fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 354eb0570c7SDamien Le Moal 355eb0570c7SDamien Le Moal unlock: 356eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 357eb0570c7SDamien Le Moal 358eb0570c7SDamien Le Moal return ret; 359eb0570c7SDamien Le Moal } 360eb0570c7SDamien Le Moal 361eb0570c7SDamien Le Moal static void zloop_put_cmd(struct zloop_cmd *cmd) 362eb0570c7SDamien Le Moal { 363eb0570c7SDamien Le Moal struct request *rq = blk_mq_rq_from_pdu(cmd); 364eb0570c7SDamien Le Moal 365eb0570c7SDamien Le Moal if (!atomic_dec_and_test(&cmd->ref)) 366eb0570c7SDamien Le Moal return; 367eb0570c7SDamien Le Moal kfree(cmd->bvec); 368eb0570c7SDamien Le Moal cmd->bvec = NULL; 369eb0570c7SDamien Le Moal if (likely(!blk_should_fake_timeout(rq->q))) 370eb0570c7SDamien Le Moal blk_mq_complete_request(rq); 371eb0570c7SDamien Le Moal } 372eb0570c7SDamien Le Moal 373eb0570c7SDamien Le Moal static void zloop_rw_complete(struct kiocb *iocb, long ret) 374eb0570c7SDamien Le Moal { 375eb0570c7SDamien Le Moal struct zloop_cmd *cmd = container_of(iocb, struct zloop_cmd, iocb); 376eb0570c7SDamien Le Moal 377eb0570c7SDamien Le Moal cmd->ret = ret; 378eb0570c7SDamien Le Moal zloop_put_cmd(cmd); 379eb0570c7SDamien Le Moal } 380eb0570c7SDamien Le Moal 381eb0570c7SDamien Le Moal static void zloop_rw(struct zloop_cmd *cmd) 382eb0570c7SDamien Le Moal { 383eb0570c7SDamien Le Moal struct request *rq = blk_mq_rq_from_pdu(cmd); 384eb0570c7SDamien Le Moal struct zloop_device *zlo = rq->q->queuedata; 385eb0570c7SDamien Le Moal unsigned int zone_no = rq_zone_no(rq); 386eb0570c7SDamien Le Moal sector_t sector = blk_rq_pos(rq); 387eb0570c7SDamien Le Moal sector_t nr_sectors = blk_rq_sectors(rq); 388eb0570c7SDamien Le Moal bool is_append = req_op(rq) == REQ_OP_ZONE_APPEND; 389eb0570c7SDamien Le Moal bool is_write = req_op(rq) == REQ_OP_WRITE || is_append; 390eb0570c7SDamien Le Moal int rw = is_write ? ITER_SOURCE : ITER_DEST; 391eb0570c7SDamien Le Moal struct req_iterator rq_iter; 392eb0570c7SDamien Le Moal struct zloop_zone *zone; 393eb0570c7SDamien Le Moal struct iov_iter iter; 394eb0570c7SDamien Le Moal struct bio_vec tmp; 395fcc6eaa3SDamien Le Moal unsigned long flags; 396eb0570c7SDamien Le Moal sector_t zone_end; 397eb0570c7SDamien Le Moal int nr_bvec = 0; 398eb0570c7SDamien Le Moal int ret; 399eb0570c7SDamien Le Moal 400eb0570c7SDamien Le Moal atomic_set(&cmd->ref, 2); 401eb0570c7SDamien Le Moal cmd->sector = sector; 402eb0570c7SDamien Le Moal cmd->nr_sectors = nr_sectors; 403eb0570c7SDamien Le Moal cmd->ret = 0; 404eb0570c7SDamien Le Moal 4059236c5fdSDamien Le Moal if (WARN_ON_ONCE(is_append && !zlo->zone_append)) { 4069236c5fdSDamien Le Moal ret = -EIO; 4079236c5fdSDamien Le Moal goto out; 4089236c5fdSDamien Le Moal } 4099236c5fdSDamien Le Moal 410eb0570c7SDamien Le Moal /* We should never get an I/O beyond the device capacity. */ 411eb0570c7SDamien Le Moal if (WARN_ON_ONCE(zone_no >= zlo->nr_zones)) { 412eb0570c7SDamien Le Moal ret = -EIO; 413eb0570c7SDamien Le Moal goto out; 414eb0570c7SDamien Le Moal } 415eb0570c7SDamien Le Moal zone = &zlo->zones[zone_no]; 416eb0570c7SDamien Le Moal zone_end = zone->start + zlo->zone_capacity; 417eb0570c7SDamien Le Moal 418eb0570c7SDamien Le Moal /* 419eb0570c7SDamien Le Moal * The block layer should never send requests that are not fully 420eb0570c7SDamien Le Moal * contained within the zone. 421eb0570c7SDamien Le Moal */ 422eb0570c7SDamien Le Moal if (WARN_ON_ONCE(sector + nr_sectors > zone->start + zlo->zone_size)) { 423eb0570c7SDamien Le Moal ret = -EIO; 424eb0570c7SDamien Le Moal goto out; 425eb0570c7SDamien Le Moal } 426eb0570c7SDamien Le Moal 427eb0570c7SDamien Le Moal if (test_and_clear_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags)) { 428eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 429eb0570c7SDamien Le Moal ret = zloop_update_seq_zone(zlo, zone_no); 430eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 431eb0570c7SDamien Le Moal if (ret) 432eb0570c7SDamien Le Moal goto out; 433eb0570c7SDamien Le Moal } 434eb0570c7SDamien Le Moal 435eb0570c7SDamien Le Moal if (!test_bit(ZLOOP_ZONE_CONV, &zone->flags) && is_write) { 436eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 437eb0570c7SDamien Le Moal 438fcc6eaa3SDamien Le Moal spin_lock_irqsave(&zone->wp_lock, flags); 439fcc6eaa3SDamien Le Moal 440e3a96ca9SDamien Le Moal /* 441e3a96ca9SDamien Le Moal * Zone append operations always go at the current write 442e3a96ca9SDamien Le Moal * pointer, but regular write operations must already be 443e3a96ca9SDamien Le Moal * aligned to the write pointer when submitted. 444e3a96ca9SDamien Le Moal */ 445eb0570c7SDamien Le Moal if (is_append) { 446fcc6eaa3SDamien Le Moal /* 447fcc6eaa3SDamien Le Moal * If ordered zone append is in use, we already checked 448fcc6eaa3SDamien Le Moal * and set the target sector in zloop_queue_rq(). 449fcc6eaa3SDamien Le Moal */ 450fcc6eaa3SDamien Le Moal if (!zlo->ordered_zone_append) { 451*a9637ab9SDamien Le Moal if (zone->cond == BLK_ZONE_COND_FULL || 452*a9637ab9SDamien Le Moal zone->wp + nr_sectors > zone_end) { 453fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, 454fcc6eaa3SDamien Le Moal flags); 455cf28f6f9SDamien Le Moal ret = -EIO; 456cf28f6f9SDamien Le Moal goto unlock; 457cf28f6f9SDamien Le Moal } 458eb0570c7SDamien Le Moal sector = zone->wp; 459fcc6eaa3SDamien Le Moal } 460eb0570c7SDamien Le Moal cmd->sector = sector; 461e3a96ca9SDamien Le Moal } else if (sector != zone->wp) { 462fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 463eb0570c7SDamien Le Moal pr_err("Zone %u: unaligned write: sect %llu, wp %llu\n", 464eb0570c7SDamien Le Moal zone_no, sector, zone->wp); 465eb0570c7SDamien Le Moal ret = -EIO; 466eb0570c7SDamien Le Moal goto unlock; 467eb0570c7SDamien Le Moal } 468eb0570c7SDamien Le Moal 469eb0570c7SDamien Le Moal /* Implicitly open the target zone. */ 470eb0570c7SDamien Le Moal if (zone->cond == BLK_ZONE_COND_CLOSED || 471eb0570c7SDamien Le Moal zone->cond == BLK_ZONE_COND_EMPTY) 472eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_IMP_OPEN; 473eb0570c7SDamien Le Moal 474eb0570c7SDamien Le Moal /* 475fcc6eaa3SDamien Le Moal * Advance the write pointer, unless ordered zone append is in 476fcc6eaa3SDamien Le Moal * use. If the write fails, the write pointer position will be 477fcc6eaa3SDamien Le Moal * corrected when the next I/O starts execution. 478eb0570c7SDamien Le Moal */ 479fcc6eaa3SDamien Le Moal if (!is_append || !zlo->ordered_zone_append) { 480eb0570c7SDamien Le Moal zone->wp += nr_sectors; 481866d6574SDamien Le Moal if (zone->wp == zone_end) { 482eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_FULL; 483866d6574SDamien Le Moal zone->wp = ULLONG_MAX; 484866d6574SDamien Le Moal } 485eb0570c7SDamien Le Moal } 486eb0570c7SDamien Le Moal 487fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 488fcc6eaa3SDamien Le Moal } 489fcc6eaa3SDamien Le Moal 490eb0570c7SDamien Le Moal rq_for_each_bvec(tmp, rq, rq_iter) 491eb0570c7SDamien Le Moal nr_bvec++; 492eb0570c7SDamien Le Moal 493eb0570c7SDamien Le Moal if (rq->bio != rq->biotail) { 494eb0570c7SDamien Le Moal struct bio_vec *bvec; 495eb0570c7SDamien Le Moal 496eb0570c7SDamien Le Moal cmd->bvec = kmalloc_array(nr_bvec, sizeof(*cmd->bvec), GFP_NOIO); 497eb0570c7SDamien Le Moal if (!cmd->bvec) { 498eb0570c7SDamien Le Moal ret = -EIO; 499eb0570c7SDamien Le Moal goto unlock; 500eb0570c7SDamien Le Moal } 501eb0570c7SDamien Le Moal 502eb0570c7SDamien Le Moal /* 503eb0570c7SDamien Le Moal * The bios of the request may be started from the middle of 504eb0570c7SDamien Le Moal * the 'bvec' because of bio splitting, so we can't directly 505eb0570c7SDamien Le Moal * copy bio->bi_iov_vec to new bvec. The rq_for_each_bvec 506eb0570c7SDamien Le Moal * API will take care of all details for us. 507eb0570c7SDamien Le Moal */ 508eb0570c7SDamien Le Moal bvec = cmd->bvec; 509eb0570c7SDamien Le Moal rq_for_each_bvec(tmp, rq, rq_iter) { 510eb0570c7SDamien Le Moal *bvec = tmp; 511eb0570c7SDamien Le Moal bvec++; 512eb0570c7SDamien Le Moal } 513eb0570c7SDamien Le Moal iov_iter_bvec(&iter, rw, cmd->bvec, nr_bvec, blk_rq_bytes(rq)); 514eb0570c7SDamien Le Moal } else { 515eb0570c7SDamien Le Moal /* 516eb0570c7SDamien Le Moal * Same here, this bio may be started from the middle of the 517eb0570c7SDamien Le Moal * 'bvec' because of bio splitting, so offset from the bvec 518eb0570c7SDamien Le Moal * must be passed to iov iterator 519eb0570c7SDamien Le Moal */ 520eb0570c7SDamien Le Moal iov_iter_bvec(&iter, rw, 521eb0570c7SDamien Le Moal __bvec_iter_bvec(rq->bio->bi_io_vec, rq->bio->bi_iter), 522eb0570c7SDamien Le Moal nr_bvec, blk_rq_bytes(rq)); 523eb0570c7SDamien Le Moal iter.iov_offset = rq->bio->bi_iter.bi_bvec_done; 524eb0570c7SDamien Le Moal } 525eb0570c7SDamien Le Moal 526eb0570c7SDamien Le Moal cmd->iocb.ki_pos = (sector - zone->start) << SECTOR_SHIFT; 527eb0570c7SDamien Le Moal cmd->iocb.ki_filp = zone->file; 528eb0570c7SDamien Le Moal cmd->iocb.ki_complete = zloop_rw_complete; 529eb0570c7SDamien Le Moal if (!zlo->buffered_io) 530eb0570c7SDamien Le Moal cmd->iocb.ki_flags = IOCB_DIRECT; 531eb0570c7SDamien Le Moal cmd->iocb.ki_ioprio = IOPRIO_PRIO_VALUE(IOPRIO_CLASS_NONE, 0); 532eb0570c7SDamien Le Moal 533eb0570c7SDamien Le Moal if (rw == ITER_SOURCE) 534eb0570c7SDamien Le Moal ret = zone->file->f_op->write_iter(&cmd->iocb, &iter); 535eb0570c7SDamien Le Moal else 536eb0570c7SDamien Le Moal ret = zone->file->f_op->read_iter(&cmd->iocb, &iter); 537eb0570c7SDamien Le Moal unlock: 538eb0570c7SDamien Le Moal if (!test_bit(ZLOOP_ZONE_CONV, &zone->flags) && is_write) 539eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 540eb0570c7SDamien Le Moal out: 541eb0570c7SDamien Le Moal if (ret != -EIOCBQUEUED) 542eb0570c7SDamien Le Moal zloop_rw_complete(&cmd->iocb, ret); 543eb0570c7SDamien Le Moal zloop_put_cmd(cmd); 544eb0570c7SDamien Le Moal } 545eb0570c7SDamien Le Moal 546eb0570c7SDamien Le Moal static void zloop_handle_cmd(struct zloop_cmd *cmd) 547eb0570c7SDamien Le Moal { 548eb0570c7SDamien Le Moal struct request *rq = blk_mq_rq_from_pdu(cmd); 549eb0570c7SDamien Le Moal struct zloop_device *zlo = rq->q->queuedata; 550eb0570c7SDamien Le Moal 551eb0570c7SDamien Le Moal switch (req_op(rq)) { 552eb0570c7SDamien Le Moal case REQ_OP_READ: 553eb0570c7SDamien Le Moal case REQ_OP_WRITE: 554eb0570c7SDamien Le Moal case REQ_OP_ZONE_APPEND: 555eb0570c7SDamien Le Moal /* 556eb0570c7SDamien Le Moal * zloop_rw() always executes asynchronously or completes 557eb0570c7SDamien Le Moal * directly. 558eb0570c7SDamien Le Moal */ 559eb0570c7SDamien Le Moal zloop_rw(cmd); 560eb0570c7SDamien Le Moal return; 561eb0570c7SDamien Le Moal case REQ_OP_FLUSH: 562eb0570c7SDamien Le Moal /* 563eb0570c7SDamien Le Moal * Sync the entire FS containing the zone files instead of 564eb0570c7SDamien Le Moal * walking all files 565eb0570c7SDamien Le Moal */ 566eb0570c7SDamien Le Moal cmd->ret = sync_filesystem(file_inode(zlo->data_dir)->i_sb); 567eb0570c7SDamien Le Moal break; 568eb0570c7SDamien Le Moal case REQ_OP_ZONE_RESET: 569eb0570c7SDamien Le Moal cmd->ret = zloop_reset_zone(zlo, rq_zone_no(rq)); 570eb0570c7SDamien Le Moal break; 571eb0570c7SDamien Le Moal case REQ_OP_ZONE_RESET_ALL: 572eb0570c7SDamien Le Moal cmd->ret = zloop_reset_all_zones(zlo); 573eb0570c7SDamien Le Moal break; 574eb0570c7SDamien Le Moal case REQ_OP_ZONE_FINISH: 575eb0570c7SDamien Le Moal cmd->ret = zloop_finish_zone(zlo, rq_zone_no(rq)); 576eb0570c7SDamien Le Moal break; 577eb0570c7SDamien Le Moal case REQ_OP_ZONE_OPEN: 578eb0570c7SDamien Le Moal cmd->ret = zloop_open_zone(zlo, rq_zone_no(rq)); 579eb0570c7SDamien Le Moal break; 580eb0570c7SDamien Le Moal case REQ_OP_ZONE_CLOSE: 581eb0570c7SDamien Le Moal cmd->ret = zloop_close_zone(zlo, rq_zone_no(rq)); 582eb0570c7SDamien Le Moal break; 583eb0570c7SDamien Le Moal default: 584eb0570c7SDamien Le Moal WARN_ON_ONCE(1); 585eb0570c7SDamien Le Moal pr_err("Unsupported operation %d\n", req_op(rq)); 586eb0570c7SDamien Le Moal cmd->ret = -EOPNOTSUPP; 587eb0570c7SDamien Le Moal break; 588eb0570c7SDamien Le Moal } 589eb0570c7SDamien Le Moal 590eb0570c7SDamien Le Moal blk_mq_complete_request(rq); 591eb0570c7SDamien Le Moal } 592eb0570c7SDamien Le Moal 593eb0570c7SDamien Le Moal static void zloop_cmd_workfn(struct work_struct *work) 594eb0570c7SDamien Le Moal { 595eb0570c7SDamien Le Moal struct zloop_cmd *cmd = container_of(work, struct zloop_cmd, work); 596eb0570c7SDamien Le Moal int orig_flags = current->flags; 597eb0570c7SDamien Le Moal 598eb0570c7SDamien Le Moal current->flags |= PF_LOCAL_THROTTLE | PF_MEMALLOC_NOIO; 599eb0570c7SDamien Le Moal zloop_handle_cmd(cmd); 600eb0570c7SDamien Le Moal current->flags = orig_flags; 601eb0570c7SDamien Le Moal } 602eb0570c7SDamien Le Moal 603eb0570c7SDamien Le Moal static void zloop_complete_rq(struct request *rq) 604eb0570c7SDamien Le Moal { 605eb0570c7SDamien Le Moal struct zloop_cmd *cmd = blk_mq_rq_to_pdu(rq); 606eb0570c7SDamien Le Moal struct zloop_device *zlo = rq->q->queuedata; 607eb0570c7SDamien Le Moal unsigned int zone_no = cmd->sector >> zlo->zone_shift; 608eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 609eb0570c7SDamien Le Moal blk_status_t sts = BLK_STS_OK; 610eb0570c7SDamien Le Moal 611eb0570c7SDamien Le Moal switch (req_op(rq)) { 612eb0570c7SDamien Le Moal case REQ_OP_READ: 613eb0570c7SDamien Le Moal if (cmd->ret < 0) 614eb0570c7SDamien Le Moal pr_err("Zone %u: failed read sector %llu, %llu sectors\n", 615eb0570c7SDamien Le Moal zone_no, cmd->sector, cmd->nr_sectors); 616eb0570c7SDamien Le Moal 617eb0570c7SDamien Le Moal if (cmd->ret >= 0 && cmd->ret != blk_rq_bytes(rq)) { 618eb0570c7SDamien Le Moal /* short read */ 619eb0570c7SDamien Le Moal struct bio *bio; 620eb0570c7SDamien Le Moal 621eb0570c7SDamien Le Moal __rq_for_each_bio(bio, rq) 622eb0570c7SDamien Le Moal zero_fill_bio(bio); 623eb0570c7SDamien Le Moal } 624eb0570c7SDamien Le Moal break; 625eb0570c7SDamien Le Moal case REQ_OP_WRITE: 626eb0570c7SDamien Le Moal case REQ_OP_ZONE_APPEND: 627eb0570c7SDamien Le Moal if (cmd->ret < 0) 628eb0570c7SDamien Le Moal pr_err("Zone %u: failed %swrite sector %llu, %llu sectors\n", 629eb0570c7SDamien Le Moal zone_no, 630eb0570c7SDamien Le Moal req_op(rq) == REQ_OP_WRITE ? "" : "append ", 631eb0570c7SDamien Le Moal cmd->sector, cmd->nr_sectors); 632eb0570c7SDamien Le Moal 633eb0570c7SDamien Le Moal if (cmd->ret >= 0 && cmd->ret != blk_rq_bytes(rq)) { 634eb0570c7SDamien Le Moal pr_err("Zone %u: partial write %ld/%u B\n", 635eb0570c7SDamien Le Moal zone_no, cmd->ret, blk_rq_bytes(rq)); 636eb0570c7SDamien Le Moal cmd->ret = -EIO; 637eb0570c7SDamien Le Moal } 638eb0570c7SDamien Le Moal 639eb0570c7SDamien Le Moal if (cmd->ret < 0 && !test_bit(ZLOOP_ZONE_CONV, &zone->flags)) { 640eb0570c7SDamien Le Moal /* 641eb0570c7SDamien Le Moal * A write to a sequential zone file failed: mark the 642eb0570c7SDamien Le Moal * zone as having an error. This will be corrected and 643eb0570c7SDamien Le Moal * cleared when the next IO is submitted. 644eb0570c7SDamien Le Moal */ 645eb0570c7SDamien Le Moal set_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags); 646eb0570c7SDamien Le Moal break; 647eb0570c7SDamien Le Moal } 648eb0570c7SDamien Le Moal if (req_op(rq) == REQ_OP_ZONE_APPEND) 649eb0570c7SDamien Le Moal rq->__sector = cmd->sector; 650eb0570c7SDamien Le Moal 651eb0570c7SDamien Le Moal break; 652eb0570c7SDamien Le Moal default: 653eb0570c7SDamien Le Moal break; 654eb0570c7SDamien Le Moal } 655eb0570c7SDamien Le Moal 656eb0570c7SDamien Le Moal if (cmd->ret < 0) 657eb0570c7SDamien Le Moal sts = errno_to_blk_status(cmd->ret); 658eb0570c7SDamien Le Moal blk_mq_end_request(rq, sts); 659eb0570c7SDamien Le Moal } 660eb0570c7SDamien Le Moal 661fcc6eaa3SDamien Le Moal static bool zloop_set_zone_append_sector(struct request *rq) 662fcc6eaa3SDamien Le Moal { 663fcc6eaa3SDamien Le Moal struct zloop_device *zlo = rq->q->queuedata; 664fcc6eaa3SDamien Le Moal unsigned int zone_no = rq_zone_no(rq); 665fcc6eaa3SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 666fcc6eaa3SDamien Le Moal sector_t zone_end = zone->start + zlo->zone_capacity; 667fcc6eaa3SDamien Le Moal sector_t nr_sectors = blk_rq_sectors(rq); 668fcc6eaa3SDamien Le Moal unsigned long flags; 669fcc6eaa3SDamien Le Moal 670fcc6eaa3SDamien Le Moal spin_lock_irqsave(&zone->wp_lock, flags); 671fcc6eaa3SDamien Le Moal 672fcc6eaa3SDamien Le Moal if (zone->cond == BLK_ZONE_COND_FULL || 673fcc6eaa3SDamien Le Moal zone->wp + nr_sectors > zone_end) { 674fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 675fcc6eaa3SDamien Le Moal return false; 676fcc6eaa3SDamien Le Moal } 677fcc6eaa3SDamien Le Moal 678fcc6eaa3SDamien Le Moal rq->__sector = zone->wp; 679fcc6eaa3SDamien Le Moal zone->wp += blk_rq_sectors(rq); 680fcc6eaa3SDamien Le Moal if (zone->wp >= zone_end) { 681fcc6eaa3SDamien Le Moal zone->cond = BLK_ZONE_COND_FULL; 682fcc6eaa3SDamien Le Moal zone->wp = ULLONG_MAX; 683fcc6eaa3SDamien Le Moal } 684fcc6eaa3SDamien Le Moal 685fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 686fcc6eaa3SDamien Le Moal 687fcc6eaa3SDamien Le Moal return true; 688fcc6eaa3SDamien Le Moal } 689fcc6eaa3SDamien Le Moal 690eb0570c7SDamien Le Moal static blk_status_t zloop_queue_rq(struct blk_mq_hw_ctx *hctx, 691eb0570c7SDamien Le Moal const struct blk_mq_queue_data *bd) 692eb0570c7SDamien Le Moal { 693eb0570c7SDamien Le Moal struct request *rq = bd->rq; 694eb0570c7SDamien Le Moal struct zloop_cmd *cmd = blk_mq_rq_to_pdu(rq); 695eb0570c7SDamien Le Moal struct zloop_device *zlo = rq->q->queuedata; 696eb0570c7SDamien Le Moal 697eb0570c7SDamien Le Moal if (zlo->state == Zlo_deleting) 698eb0570c7SDamien Le Moal return BLK_STS_IOERR; 699eb0570c7SDamien Le Moal 700fcc6eaa3SDamien Le Moal /* 701fcc6eaa3SDamien Le Moal * If we need to strongly order zone append operations, set the request 702fcc6eaa3SDamien Le Moal * sector to the zone write pointer location now instead of when the 703fcc6eaa3SDamien Le Moal * command work runs. 704fcc6eaa3SDamien Le Moal */ 705fcc6eaa3SDamien Le Moal if (zlo->ordered_zone_append && req_op(rq) == REQ_OP_ZONE_APPEND) { 706fcc6eaa3SDamien Le Moal if (!zloop_set_zone_append_sector(rq)) 707fcc6eaa3SDamien Le Moal return BLK_STS_IOERR; 708fcc6eaa3SDamien Le Moal } 709fcc6eaa3SDamien Le Moal 710eb0570c7SDamien Le Moal blk_mq_start_request(rq); 711eb0570c7SDamien Le Moal 712eb0570c7SDamien Le Moal INIT_WORK(&cmd->work, zloop_cmd_workfn); 713eb0570c7SDamien Le Moal queue_work(zlo->workqueue, &cmd->work); 714eb0570c7SDamien Le Moal 715eb0570c7SDamien Le Moal return BLK_STS_OK; 716eb0570c7SDamien Le Moal } 717eb0570c7SDamien Le Moal 718eb0570c7SDamien Le Moal static const struct blk_mq_ops zloop_mq_ops = { 719eb0570c7SDamien Le Moal .queue_rq = zloop_queue_rq, 720eb0570c7SDamien Le Moal .complete = zloop_complete_rq, 721eb0570c7SDamien Le Moal }; 722eb0570c7SDamien Le Moal 723eb0570c7SDamien Le Moal static int zloop_open(struct gendisk *disk, blk_mode_t mode) 724eb0570c7SDamien Le Moal { 725eb0570c7SDamien Le Moal struct zloop_device *zlo = disk->private_data; 726eb0570c7SDamien Le Moal int ret; 727eb0570c7SDamien Le Moal 728eb0570c7SDamien Le Moal ret = mutex_lock_killable(&zloop_ctl_mutex); 729eb0570c7SDamien Le Moal if (ret) 730eb0570c7SDamien Le Moal return ret; 731eb0570c7SDamien Le Moal 732eb0570c7SDamien Le Moal if (zlo->state != Zlo_live) 733eb0570c7SDamien Le Moal ret = -ENXIO; 734eb0570c7SDamien Le Moal mutex_unlock(&zloop_ctl_mutex); 735eb0570c7SDamien Le Moal return ret; 736eb0570c7SDamien Le Moal } 737eb0570c7SDamien Le Moal 738eb0570c7SDamien Le Moal static int zloop_report_zones(struct gendisk *disk, sector_t sector, 739fdb9aed8SDamien Le Moal unsigned int nr_zones, struct blk_report_zones_args *args) 740eb0570c7SDamien Le Moal { 741eb0570c7SDamien Le Moal struct zloop_device *zlo = disk->private_data; 742eb0570c7SDamien Le Moal struct blk_zone blkz = {}; 743eb0570c7SDamien Le Moal unsigned int first, i; 744fcc6eaa3SDamien Le Moal unsigned long flags; 745eb0570c7SDamien Le Moal int ret; 746eb0570c7SDamien Le Moal 747eb0570c7SDamien Le Moal first = disk_zone_no(disk, sector); 748eb0570c7SDamien Le Moal if (first >= zlo->nr_zones) 749eb0570c7SDamien Le Moal return 0; 750eb0570c7SDamien Le Moal nr_zones = min(nr_zones, zlo->nr_zones - first); 751eb0570c7SDamien Le Moal 752eb0570c7SDamien Le Moal for (i = 0; i < nr_zones; i++) { 753eb0570c7SDamien Le Moal unsigned int zone_no = first + i; 754eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 755eb0570c7SDamien Le Moal 756eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 757eb0570c7SDamien Le Moal 758eb0570c7SDamien Le Moal if (test_and_clear_bit(ZLOOP_ZONE_SEQ_ERROR, &zone->flags)) { 759eb0570c7SDamien Le Moal ret = zloop_update_seq_zone(zlo, zone_no); 760eb0570c7SDamien Le Moal if (ret) { 761eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 762eb0570c7SDamien Le Moal return ret; 763eb0570c7SDamien Le Moal } 764eb0570c7SDamien Le Moal } 765eb0570c7SDamien Le Moal 766eb0570c7SDamien Le Moal blkz.start = zone->start; 767eb0570c7SDamien Le Moal blkz.len = zlo->zone_size; 768fcc6eaa3SDamien Le Moal spin_lock_irqsave(&zone->wp_lock, flags); 769eb0570c7SDamien Le Moal blkz.wp = zone->wp; 770fcc6eaa3SDamien Le Moal spin_unlock_irqrestore(&zone->wp_lock, flags); 771eb0570c7SDamien Le Moal blkz.cond = zone->cond; 772eb0570c7SDamien Le Moal if (test_bit(ZLOOP_ZONE_CONV, &zone->flags)) { 773eb0570c7SDamien Le Moal blkz.type = BLK_ZONE_TYPE_CONVENTIONAL; 774eb0570c7SDamien Le Moal blkz.capacity = zlo->zone_size; 775eb0570c7SDamien Le Moal } else { 776eb0570c7SDamien Le Moal blkz.type = BLK_ZONE_TYPE_SEQWRITE_REQ; 777eb0570c7SDamien Le Moal blkz.capacity = zlo->zone_capacity; 778eb0570c7SDamien Le Moal } 779eb0570c7SDamien Le Moal 780eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 781eb0570c7SDamien Le Moal 782fdb9aed8SDamien Le Moal ret = disk_report_zone(disk, &blkz, i, args); 783eb0570c7SDamien Le Moal if (ret) 784eb0570c7SDamien Le Moal return ret; 785eb0570c7SDamien Le Moal } 786eb0570c7SDamien Le Moal 787eb0570c7SDamien Le Moal return nr_zones; 788eb0570c7SDamien Le Moal } 789eb0570c7SDamien Le Moal 790eb0570c7SDamien Le Moal static void zloop_free_disk(struct gendisk *disk) 791eb0570c7SDamien Le Moal { 792eb0570c7SDamien Le Moal struct zloop_device *zlo = disk->private_data; 793eb0570c7SDamien Le Moal unsigned int i; 794eb0570c7SDamien Le Moal 79576576185SShin'ichiro Kawasaki blk_mq_free_tag_set(&zlo->tag_set); 79676576185SShin'ichiro Kawasaki 797eb0570c7SDamien Le Moal for (i = 0; i < zlo->nr_zones; i++) { 798eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[i]; 799eb0570c7SDamien Le Moal 800eb0570c7SDamien Le Moal mapping_set_gfp_mask(zone->file->f_mapping, 801eb0570c7SDamien Le Moal zone->old_gfp_mask); 802eb0570c7SDamien Le Moal fput(zone->file); 803eb0570c7SDamien Le Moal } 804eb0570c7SDamien Le Moal 805eb0570c7SDamien Le Moal fput(zlo->data_dir); 806eb0570c7SDamien Le Moal destroy_workqueue(zlo->workqueue); 807eb0570c7SDamien Le Moal kfree(zlo->base_dir); 808eb0570c7SDamien Le Moal kvfree(zlo); 809eb0570c7SDamien Le Moal } 810eb0570c7SDamien Le Moal 811eb0570c7SDamien Le Moal static const struct block_device_operations zloop_fops = { 812eb0570c7SDamien Le Moal .owner = THIS_MODULE, 813eb0570c7SDamien Le Moal .open = zloop_open, 814eb0570c7SDamien Le Moal .report_zones = zloop_report_zones, 815eb0570c7SDamien Le Moal .free_disk = zloop_free_disk, 816eb0570c7SDamien Le Moal }; 817eb0570c7SDamien Le Moal 818eb0570c7SDamien Le Moal __printf(3, 4) 819eb0570c7SDamien Le Moal static struct file *zloop_filp_open_fmt(int oflags, umode_t mode, 820eb0570c7SDamien Le Moal const char *fmt, ...) 821eb0570c7SDamien Le Moal { 822eb0570c7SDamien Le Moal struct file *file; 823eb0570c7SDamien Le Moal va_list ap; 824eb0570c7SDamien Le Moal char *p; 825eb0570c7SDamien Le Moal 826eb0570c7SDamien Le Moal va_start(ap, fmt); 827eb0570c7SDamien Le Moal p = kvasprintf(GFP_KERNEL, fmt, ap); 828eb0570c7SDamien Le Moal va_end(ap); 829eb0570c7SDamien Le Moal 830eb0570c7SDamien Le Moal if (!p) 831eb0570c7SDamien Le Moal return ERR_PTR(-ENOMEM); 832eb0570c7SDamien Le Moal file = filp_open(p, oflags, mode); 833eb0570c7SDamien Le Moal kfree(p); 834eb0570c7SDamien Le Moal return file; 835eb0570c7SDamien Le Moal } 836eb0570c7SDamien Le Moal 837eb0570c7SDamien Le Moal static int zloop_get_block_size(struct zloop_device *zlo, 838eb0570c7SDamien Le Moal struct zloop_zone *zone) 839eb0570c7SDamien Le Moal { 840eb0570c7SDamien Le Moal struct block_device *sb_bdev = zone->file->f_mapping->host->i_sb->s_bdev; 841eb0570c7SDamien Le Moal struct kstat st; 842eb0570c7SDamien Le Moal 843eb0570c7SDamien Le Moal /* 844eb0570c7SDamien Le Moal * If the FS block size is lower than or equal to 4K, use that as the 845eb0570c7SDamien Le Moal * device block size. Otherwise, fallback to the FS direct IO alignment 846eb0570c7SDamien Le Moal * constraint if that is provided, and to the FS underlying device 847eb0570c7SDamien Le Moal * physical block size if the direct IO alignment is unknown. 848eb0570c7SDamien Le Moal */ 849eb0570c7SDamien Le Moal if (file_inode(zone->file)->i_sb->s_blocksize <= SZ_4K) 850eb0570c7SDamien Le Moal zlo->block_size = file_inode(zone->file)->i_sb->s_blocksize; 851eb0570c7SDamien Le Moal else if (!vfs_getattr(&zone->file->f_path, &st, STATX_DIOALIGN, 0) && 852eb0570c7SDamien Le Moal (st.result_mask & STATX_DIOALIGN)) 853eb0570c7SDamien Le Moal zlo->block_size = st.dio_offset_align; 854eb0570c7SDamien Le Moal else if (sb_bdev) 855eb0570c7SDamien Le Moal zlo->block_size = bdev_physical_block_size(sb_bdev); 856eb0570c7SDamien Le Moal else 857eb0570c7SDamien Le Moal zlo->block_size = SECTOR_SIZE; 858eb0570c7SDamien Le Moal 859eb0570c7SDamien Le Moal if (zlo->zone_capacity & ((zlo->block_size >> SECTOR_SHIFT) - 1)) { 860eb0570c7SDamien Le Moal pr_err("Zone capacity is not aligned to block size %u\n", 861eb0570c7SDamien Le Moal zlo->block_size); 862eb0570c7SDamien Le Moal return -EINVAL; 863eb0570c7SDamien Le Moal } 864eb0570c7SDamien Le Moal 865eb0570c7SDamien Le Moal return 0; 866eb0570c7SDamien Le Moal } 867eb0570c7SDamien Le Moal 868eb0570c7SDamien Le Moal static int zloop_init_zone(struct zloop_device *zlo, struct zloop_options *opts, 869eb0570c7SDamien Le Moal unsigned int zone_no, bool restore) 870eb0570c7SDamien Le Moal { 871eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[zone_no]; 872eb0570c7SDamien Le Moal int oflags = O_RDWR; 873eb0570c7SDamien Le Moal struct kstat stat; 874eb0570c7SDamien Le Moal sector_t file_sectors; 875eb0570c7SDamien Le Moal int ret; 876eb0570c7SDamien Le Moal 877eb0570c7SDamien Le Moal mutex_init(&zone->lock); 878fcc6eaa3SDamien Le Moal spin_lock_init(&zone->wp_lock); 879eb0570c7SDamien Le Moal zone->start = (sector_t)zone_no << zlo->zone_shift; 880eb0570c7SDamien Le Moal 881eb0570c7SDamien Le Moal if (!restore) 882eb0570c7SDamien Le Moal oflags |= O_CREAT; 883eb0570c7SDamien Le Moal 884eb0570c7SDamien Le Moal if (!opts->buffered_io) 885eb0570c7SDamien Le Moal oflags |= O_DIRECT; 886eb0570c7SDamien Le Moal 887eb0570c7SDamien Le Moal if (zone_no < zlo->nr_conv_zones) { 888eb0570c7SDamien Le Moal /* Conventional zone file. */ 889eb0570c7SDamien Le Moal set_bit(ZLOOP_ZONE_CONV, &zone->flags); 890eb0570c7SDamien Le Moal zone->cond = BLK_ZONE_COND_NOT_WP; 891eb0570c7SDamien Le Moal zone->wp = U64_MAX; 892eb0570c7SDamien Le Moal 893eb0570c7SDamien Le Moal zone->file = zloop_filp_open_fmt(oflags, 0600, "%s/%u/cnv-%06u", 894eb0570c7SDamien Le Moal zlo->base_dir, zlo->id, zone_no); 895eb0570c7SDamien Le Moal if (IS_ERR(zone->file)) { 896eb0570c7SDamien Le Moal pr_err("Failed to open zone %u file %s/%u/cnv-%06u (err=%ld)", 897eb0570c7SDamien Le Moal zone_no, zlo->base_dir, zlo->id, zone_no, 898eb0570c7SDamien Le Moal PTR_ERR(zone->file)); 899eb0570c7SDamien Le Moal return PTR_ERR(zone->file); 900eb0570c7SDamien Le Moal } 901eb0570c7SDamien Le Moal 902eb0570c7SDamien Le Moal if (!zlo->block_size) { 903eb0570c7SDamien Le Moal ret = zloop_get_block_size(zlo, zone); 904eb0570c7SDamien Le Moal if (ret) 905eb0570c7SDamien Le Moal return ret; 906eb0570c7SDamien Le Moal } 907eb0570c7SDamien Le Moal 908eb0570c7SDamien Le Moal ret = vfs_getattr(&zone->file->f_path, &stat, STATX_SIZE, 0); 909eb0570c7SDamien Le Moal if (ret < 0) { 910eb0570c7SDamien Le Moal pr_err("Failed to get zone %u file stat\n", zone_no); 911eb0570c7SDamien Le Moal return ret; 912eb0570c7SDamien Le Moal } 913eb0570c7SDamien Le Moal file_sectors = stat.size >> SECTOR_SHIFT; 914eb0570c7SDamien Le Moal 915eb0570c7SDamien Le Moal if (restore && file_sectors != zlo->zone_size) { 916eb0570c7SDamien Le Moal pr_err("Invalid conventional zone %u file size (%llu sectors != %llu)\n", 917eb0570c7SDamien Le Moal zone_no, file_sectors, zlo->zone_capacity); 918eb0570c7SDamien Le Moal return ret; 919eb0570c7SDamien Le Moal } 920eb0570c7SDamien Le Moal 921eb0570c7SDamien Le Moal ret = vfs_truncate(&zone->file->f_path, 922eb0570c7SDamien Le Moal zlo->zone_size << SECTOR_SHIFT); 923eb0570c7SDamien Le Moal if (ret < 0) { 924eb0570c7SDamien Le Moal pr_err("Failed to truncate zone %u file (err=%d)\n", 925eb0570c7SDamien Le Moal zone_no, ret); 926eb0570c7SDamien Le Moal return ret; 927eb0570c7SDamien Le Moal } 928eb0570c7SDamien Le Moal 929eb0570c7SDamien Le Moal return 0; 930eb0570c7SDamien Le Moal } 931eb0570c7SDamien Le Moal 932eb0570c7SDamien Le Moal /* Sequential zone file. */ 933eb0570c7SDamien Le Moal zone->file = zloop_filp_open_fmt(oflags, 0600, "%s/%u/seq-%06u", 934eb0570c7SDamien Le Moal zlo->base_dir, zlo->id, zone_no); 935eb0570c7SDamien Le Moal if (IS_ERR(zone->file)) { 936eb0570c7SDamien Le Moal pr_err("Failed to open zone %u file %s/%u/seq-%06u (err=%ld)", 937eb0570c7SDamien Le Moal zone_no, zlo->base_dir, zlo->id, zone_no, 938eb0570c7SDamien Le Moal PTR_ERR(zone->file)); 939eb0570c7SDamien Le Moal return PTR_ERR(zone->file); 940eb0570c7SDamien Le Moal } 941eb0570c7SDamien Le Moal 942eb0570c7SDamien Le Moal if (!zlo->block_size) { 943eb0570c7SDamien Le Moal ret = zloop_get_block_size(zlo, zone); 944eb0570c7SDamien Le Moal if (ret) 945eb0570c7SDamien Le Moal return ret; 946eb0570c7SDamien Le Moal } 947eb0570c7SDamien Le Moal 948eb0570c7SDamien Le Moal zloop_get_block_size(zlo, zone); 949eb0570c7SDamien Le Moal 950eb0570c7SDamien Le Moal mutex_lock(&zone->lock); 951eb0570c7SDamien Le Moal ret = zloop_update_seq_zone(zlo, zone_no); 952eb0570c7SDamien Le Moal mutex_unlock(&zone->lock); 953eb0570c7SDamien Le Moal 954eb0570c7SDamien Le Moal return ret; 955eb0570c7SDamien Le Moal } 956eb0570c7SDamien Le Moal 957eb0570c7SDamien Le Moal static bool zloop_dev_exists(struct zloop_device *zlo) 958eb0570c7SDamien Le Moal { 959eb0570c7SDamien Le Moal struct file *cnv, *seq; 960eb0570c7SDamien Le Moal bool exists; 961eb0570c7SDamien Le Moal 962eb0570c7SDamien Le Moal cnv = zloop_filp_open_fmt(O_RDONLY, 0600, "%s/%u/cnv-%06u", 963eb0570c7SDamien Le Moal zlo->base_dir, zlo->id, 0); 964eb0570c7SDamien Le Moal seq = zloop_filp_open_fmt(O_RDONLY, 0600, "%s/%u/seq-%06u", 965eb0570c7SDamien Le Moal zlo->base_dir, zlo->id, 0); 966eb0570c7SDamien Le Moal exists = !IS_ERR(cnv) || !IS_ERR(seq); 967eb0570c7SDamien Le Moal 968eb0570c7SDamien Le Moal if (!IS_ERR(cnv)) 969eb0570c7SDamien Le Moal fput(cnv); 970eb0570c7SDamien Le Moal if (!IS_ERR(seq)) 971eb0570c7SDamien Le Moal fput(seq); 972eb0570c7SDamien Le Moal 973eb0570c7SDamien Le Moal return exists; 974eb0570c7SDamien Le Moal } 975eb0570c7SDamien Le Moal 976eb0570c7SDamien Le Moal static int zloop_ctl_add(struct zloop_options *opts) 977eb0570c7SDamien Le Moal { 978eb0570c7SDamien Le Moal struct queue_limits lim = { 979eb0570c7SDamien Le Moal .max_hw_sectors = SZ_1M >> SECTOR_SHIFT, 980eb0570c7SDamien Le Moal .chunk_sectors = opts->zone_size, 981eb0570c7SDamien Le Moal .features = BLK_FEAT_ZONED, 982eb0570c7SDamien Le Moal }; 983eb0570c7SDamien Le Moal unsigned int nr_zones, i, j; 984eb0570c7SDamien Le Moal struct zloop_device *zlo; 985eb0570c7SDamien Le Moal int ret = -EINVAL; 986eb0570c7SDamien Le Moal bool restore; 987eb0570c7SDamien Le Moal 988eb0570c7SDamien Le Moal __module_get(THIS_MODULE); 989eb0570c7SDamien Le Moal 990eb0570c7SDamien Le Moal nr_zones = opts->capacity >> ilog2(opts->zone_size); 991eb0570c7SDamien Le Moal if (opts->nr_conv_zones >= nr_zones) { 992eb0570c7SDamien Le Moal pr_err("Invalid number of conventional zones %u\n", 993eb0570c7SDamien Le Moal opts->nr_conv_zones); 994eb0570c7SDamien Le Moal goto out; 995eb0570c7SDamien Le Moal } 996eb0570c7SDamien Le Moal 997eb0570c7SDamien Le Moal zlo = kvzalloc(struct_size(zlo, zones, nr_zones), GFP_KERNEL); 998eb0570c7SDamien Le Moal if (!zlo) { 999eb0570c7SDamien Le Moal ret = -ENOMEM; 1000eb0570c7SDamien Le Moal goto out; 1001eb0570c7SDamien Le Moal } 1002eb0570c7SDamien Le Moal zlo->state = Zlo_creating; 1003eb0570c7SDamien Le Moal 1004eb0570c7SDamien Le Moal ret = mutex_lock_killable(&zloop_ctl_mutex); 1005eb0570c7SDamien Le Moal if (ret) 1006eb0570c7SDamien Le Moal goto out_free_dev; 1007eb0570c7SDamien Le Moal 1008eb0570c7SDamien Le Moal /* Allocate id, if @opts->id >= 0, we're requesting that specific id */ 1009eb0570c7SDamien Le Moal if (opts->id >= 0) { 1010eb0570c7SDamien Le Moal ret = idr_alloc(&zloop_index_idr, zlo, 1011eb0570c7SDamien Le Moal opts->id, opts->id + 1, GFP_KERNEL); 1012eb0570c7SDamien Le Moal if (ret == -ENOSPC) 1013eb0570c7SDamien Le Moal ret = -EEXIST; 1014eb0570c7SDamien Le Moal } else { 1015eb0570c7SDamien Le Moal ret = idr_alloc(&zloop_index_idr, zlo, 0, 0, GFP_KERNEL); 1016eb0570c7SDamien Le Moal } 1017eb0570c7SDamien Le Moal mutex_unlock(&zloop_ctl_mutex); 1018eb0570c7SDamien Le Moal if (ret < 0) 1019eb0570c7SDamien Le Moal goto out_free_dev; 1020eb0570c7SDamien Le Moal 1021eb0570c7SDamien Le Moal zlo->id = ret; 1022eb0570c7SDamien Le Moal zlo->zone_shift = ilog2(opts->zone_size); 1023eb0570c7SDamien Le Moal zlo->zone_size = opts->zone_size; 1024eb0570c7SDamien Le Moal if (opts->zone_capacity) 1025eb0570c7SDamien Le Moal zlo->zone_capacity = opts->zone_capacity; 1026eb0570c7SDamien Le Moal else 1027eb0570c7SDamien Le Moal zlo->zone_capacity = zlo->zone_size; 1028eb0570c7SDamien Le Moal zlo->nr_zones = nr_zones; 1029eb0570c7SDamien Le Moal zlo->nr_conv_zones = opts->nr_conv_zones; 1030eb0570c7SDamien Le Moal zlo->buffered_io = opts->buffered_io; 10319236c5fdSDamien Le Moal zlo->zone_append = opts->zone_append; 1032fcc6eaa3SDamien Le Moal if (zlo->zone_append) 1033fcc6eaa3SDamien Le Moal zlo->ordered_zone_append = opts->ordered_zone_append; 1034eb0570c7SDamien Le Moal 1035eb0570c7SDamien Le Moal zlo->workqueue = alloc_workqueue("zloop%d", WQ_UNBOUND | WQ_FREEZABLE, 1036eb0570c7SDamien Le Moal opts->nr_queues * opts->queue_depth, zlo->id); 1037eb0570c7SDamien Le Moal if (!zlo->workqueue) { 1038eb0570c7SDamien Le Moal ret = -ENOMEM; 1039eb0570c7SDamien Le Moal goto out_free_idr; 1040eb0570c7SDamien Le Moal } 1041eb0570c7SDamien Le Moal 1042eb0570c7SDamien Le Moal if (opts->base_dir) 1043eb0570c7SDamien Le Moal zlo->base_dir = kstrdup(opts->base_dir, GFP_KERNEL); 1044eb0570c7SDamien Le Moal else 1045eb0570c7SDamien Le Moal zlo->base_dir = kstrdup(ZLOOP_DEF_BASE_DIR, GFP_KERNEL); 1046eb0570c7SDamien Le Moal if (!zlo->base_dir) { 1047eb0570c7SDamien Le Moal ret = -ENOMEM; 1048eb0570c7SDamien Le Moal goto out_destroy_workqueue; 1049eb0570c7SDamien Le Moal } 1050eb0570c7SDamien Le Moal 1051eb0570c7SDamien Le Moal zlo->data_dir = zloop_filp_open_fmt(O_RDONLY | O_DIRECTORY, 0, "%s/%u", 1052eb0570c7SDamien Le Moal zlo->base_dir, zlo->id); 1053eb0570c7SDamien Le Moal if (IS_ERR(zlo->data_dir)) { 1054eb0570c7SDamien Le Moal ret = PTR_ERR(zlo->data_dir); 1055eb0570c7SDamien Le Moal pr_warn("Failed to open directory %s/%u (err=%d)\n", 1056eb0570c7SDamien Le Moal zlo->base_dir, zlo->id, ret); 1057eb0570c7SDamien Le Moal goto out_free_base_dir; 1058eb0570c7SDamien Le Moal } 1059eb0570c7SDamien Le Moal 1060eb0570c7SDamien Le Moal /* 1061eb0570c7SDamien Le Moal * If we already have zone files, we are restoring a device created by a 1062eb0570c7SDamien Le Moal * previous add operation. In this case, zloop_init_zone() will check 1063eb0570c7SDamien Le Moal * that the zone files are consistent with the zone configuration given. 1064eb0570c7SDamien Le Moal */ 1065eb0570c7SDamien Le Moal restore = zloop_dev_exists(zlo); 1066eb0570c7SDamien Le Moal for (i = 0; i < nr_zones; i++) { 1067eb0570c7SDamien Le Moal ret = zloop_init_zone(zlo, opts, i, restore); 1068eb0570c7SDamien Le Moal if (ret) 1069eb0570c7SDamien Le Moal goto out_close_files; 1070eb0570c7SDamien Le Moal } 1071eb0570c7SDamien Le Moal 1072eb0570c7SDamien Le Moal lim.physical_block_size = zlo->block_size; 1073eb0570c7SDamien Le Moal lim.logical_block_size = zlo->block_size; 10749236c5fdSDamien Le Moal if (zlo->zone_append) 10759236c5fdSDamien Le Moal lim.max_hw_zone_append_sectors = lim.max_hw_sectors; 1076eb0570c7SDamien Le Moal 1077eb0570c7SDamien Le Moal zlo->tag_set.ops = &zloop_mq_ops; 1078eb0570c7SDamien Le Moal zlo->tag_set.nr_hw_queues = opts->nr_queues; 1079eb0570c7SDamien Le Moal zlo->tag_set.queue_depth = opts->queue_depth; 1080eb0570c7SDamien Le Moal zlo->tag_set.numa_node = NUMA_NO_NODE; 1081eb0570c7SDamien Le Moal zlo->tag_set.cmd_size = sizeof(struct zloop_cmd); 1082eb0570c7SDamien Le Moal zlo->tag_set.driver_data = zlo; 1083eb0570c7SDamien Le Moal 1084eb0570c7SDamien Le Moal ret = blk_mq_alloc_tag_set(&zlo->tag_set); 1085eb0570c7SDamien Le Moal if (ret) { 1086eb0570c7SDamien Le Moal pr_err("blk_mq_alloc_tag_set failed (err=%d)\n", ret); 1087eb0570c7SDamien Le Moal goto out_close_files; 1088eb0570c7SDamien Le Moal } 1089eb0570c7SDamien Le Moal 1090eb0570c7SDamien Le Moal zlo->disk = blk_mq_alloc_disk(&zlo->tag_set, &lim, zlo); 1091eb0570c7SDamien Le Moal if (IS_ERR(zlo->disk)) { 1092eb0570c7SDamien Le Moal pr_err("blk_mq_alloc_disk failed (err=%d)\n", ret); 1093eb0570c7SDamien Le Moal ret = PTR_ERR(zlo->disk); 1094eb0570c7SDamien Le Moal goto out_cleanup_tags; 1095eb0570c7SDamien Le Moal } 1096eb0570c7SDamien Le Moal zlo->disk->flags = GENHD_FL_NO_PART; 1097eb0570c7SDamien Le Moal zlo->disk->fops = &zloop_fops; 1098eb0570c7SDamien Le Moal zlo->disk->private_data = zlo; 1099eb0570c7SDamien Le Moal sprintf(zlo->disk->disk_name, "zloop%d", zlo->id); 1100eb0570c7SDamien Le Moal set_capacity(zlo->disk, (u64)lim.chunk_sectors * zlo->nr_zones); 1101eb0570c7SDamien Le Moal 1102eb0570c7SDamien Le Moal ret = blk_revalidate_disk_zones(zlo->disk); 1103eb0570c7SDamien Le Moal if (ret) 1104eb0570c7SDamien Le Moal goto out_cleanup_disk; 1105eb0570c7SDamien Le Moal 1106eb0570c7SDamien Le Moal ret = add_disk(zlo->disk); 1107eb0570c7SDamien Le Moal if (ret) { 1108eb0570c7SDamien Le Moal pr_err("add_disk failed (err=%d)\n", ret); 1109eb0570c7SDamien Le Moal goto out_cleanup_disk; 1110eb0570c7SDamien Le Moal } 1111eb0570c7SDamien Le Moal 1112eb0570c7SDamien Le Moal mutex_lock(&zloop_ctl_mutex); 1113eb0570c7SDamien Le Moal zlo->state = Zlo_live; 1114eb0570c7SDamien Le Moal mutex_unlock(&zloop_ctl_mutex); 1115eb0570c7SDamien Le Moal 11169236c5fdSDamien Le Moal pr_info("zloop: device %d, %u zones of %llu MiB, %u B block size\n", 1117eb0570c7SDamien Le Moal zlo->id, zlo->nr_zones, 1118eb0570c7SDamien Le Moal ((sector_t)zlo->zone_size << SECTOR_SHIFT) >> 20, 1119eb0570c7SDamien Le Moal zlo->block_size); 1120fcc6eaa3SDamien Le Moal pr_info("zloop%d: using %s%s zone append\n", 11219236c5fdSDamien Le Moal zlo->id, 1122fcc6eaa3SDamien Le Moal zlo->ordered_zone_append ? "ordered " : "", 11239236c5fdSDamien Le Moal zlo->zone_append ? "native" : "emulated"); 1124eb0570c7SDamien Le Moal 1125eb0570c7SDamien Le Moal return 0; 1126eb0570c7SDamien Le Moal 1127eb0570c7SDamien Le Moal out_cleanup_disk: 1128eb0570c7SDamien Le Moal put_disk(zlo->disk); 1129eb0570c7SDamien Le Moal out_cleanup_tags: 1130eb0570c7SDamien Le Moal blk_mq_free_tag_set(&zlo->tag_set); 1131eb0570c7SDamien Le Moal out_close_files: 1132eb0570c7SDamien Le Moal for (j = 0; j < i; j++) { 1133eb0570c7SDamien Le Moal struct zloop_zone *zone = &zlo->zones[j]; 1134eb0570c7SDamien Le Moal 1135eb0570c7SDamien Le Moal if (!IS_ERR_OR_NULL(zone->file)) 1136eb0570c7SDamien Le Moal fput(zone->file); 1137eb0570c7SDamien Le Moal } 1138eb0570c7SDamien Le Moal fput(zlo->data_dir); 1139eb0570c7SDamien Le Moal out_free_base_dir: 1140eb0570c7SDamien Le Moal kfree(zlo->base_dir); 1141eb0570c7SDamien Le Moal out_destroy_workqueue: 1142eb0570c7SDamien Le Moal destroy_workqueue(zlo->workqueue); 1143eb0570c7SDamien Le Moal out_free_idr: 1144eb0570c7SDamien Le Moal mutex_lock(&zloop_ctl_mutex); 1145eb0570c7SDamien Le Moal idr_remove(&zloop_index_idr, zlo->id); 1146eb0570c7SDamien Le Moal mutex_unlock(&zloop_ctl_mutex); 1147eb0570c7SDamien Le Moal out_free_dev: 1148eb0570c7SDamien Le Moal kvfree(zlo); 1149eb0570c7SDamien Le Moal out: 1150eb0570c7SDamien Le Moal module_put(THIS_MODULE); 1151eb0570c7SDamien Le Moal if (ret == -ENOENT) 1152eb0570c7SDamien Le Moal ret = -EINVAL; 1153eb0570c7SDamien Le Moal return ret; 1154eb0570c7SDamien Le Moal } 1155eb0570c7SDamien Le Moal 1156eb0570c7SDamien Le Moal static int zloop_ctl_remove(struct zloop_options *opts) 1157eb0570c7SDamien Le Moal { 1158eb0570c7SDamien Le Moal struct zloop_device *zlo; 1159eb0570c7SDamien Le Moal int ret; 1160eb0570c7SDamien Le Moal 1161eb0570c7SDamien Le Moal if (!(opts->mask & ZLOOP_OPT_ID)) { 1162eb0570c7SDamien Le Moal pr_err("No ID specified\n"); 1163eb0570c7SDamien Le Moal return -EINVAL; 1164eb0570c7SDamien Le Moal } 1165eb0570c7SDamien Le Moal 1166eb0570c7SDamien Le Moal ret = mutex_lock_killable(&zloop_ctl_mutex); 1167eb0570c7SDamien Le Moal if (ret) 1168eb0570c7SDamien Le Moal return ret; 1169eb0570c7SDamien Le Moal 1170eb0570c7SDamien Le Moal zlo = idr_find(&zloop_index_idr, opts->id); 1171eb0570c7SDamien Le Moal if (!zlo || zlo->state == Zlo_creating) { 1172eb0570c7SDamien Le Moal ret = -ENODEV; 1173eb0570c7SDamien Le Moal } else if (zlo->state == Zlo_deleting) { 1174eb0570c7SDamien Le Moal ret = -EINVAL; 1175eb0570c7SDamien Le Moal } else { 1176eb0570c7SDamien Le Moal idr_remove(&zloop_index_idr, zlo->id); 1177eb0570c7SDamien Le Moal zlo->state = Zlo_deleting; 1178eb0570c7SDamien Le Moal } 1179eb0570c7SDamien Le Moal 1180eb0570c7SDamien Le Moal mutex_unlock(&zloop_ctl_mutex); 1181eb0570c7SDamien Le Moal if (ret) 1182eb0570c7SDamien Le Moal return ret; 1183eb0570c7SDamien Le Moal 1184eb0570c7SDamien Le Moal del_gendisk(zlo->disk); 1185eb0570c7SDamien Le Moal put_disk(zlo->disk); 1186eb0570c7SDamien Le Moal 1187eb0570c7SDamien Le Moal pr_info("Removed device %d\n", opts->id); 1188eb0570c7SDamien Le Moal 1189eb0570c7SDamien Le Moal module_put(THIS_MODULE); 1190eb0570c7SDamien Le Moal 1191eb0570c7SDamien Le Moal return 0; 1192eb0570c7SDamien Le Moal } 1193eb0570c7SDamien Le Moal 1194eb0570c7SDamien Le Moal static int zloop_parse_options(struct zloop_options *opts, const char *buf) 1195eb0570c7SDamien Le Moal { 1196eb0570c7SDamien Le Moal substring_t args[MAX_OPT_ARGS]; 1197eb0570c7SDamien Le Moal char *options, *o, *p; 1198eb0570c7SDamien Le Moal unsigned int token; 1199eb0570c7SDamien Le Moal int ret = 0; 1200eb0570c7SDamien Le Moal 1201eb0570c7SDamien Le Moal /* Set defaults. */ 1202eb0570c7SDamien Le Moal opts->mask = 0; 1203eb0570c7SDamien Le Moal opts->id = ZLOOP_DEF_ID; 1204eb0570c7SDamien Le Moal opts->capacity = ZLOOP_DEF_ZONE_SIZE * ZLOOP_DEF_NR_ZONES; 1205eb0570c7SDamien Le Moal opts->zone_size = ZLOOP_DEF_ZONE_SIZE; 1206eb0570c7SDamien Le Moal opts->nr_conv_zones = ZLOOP_DEF_NR_CONV_ZONES; 1207eb0570c7SDamien Le Moal opts->nr_queues = ZLOOP_DEF_NR_QUEUES; 1208eb0570c7SDamien Le Moal opts->queue_depth = ZLOOP_DEF_QUEUE_DEPTH; 1209eb0570c7SDamien Le Moal opts->buffered_io = ZLOOP_DEF_BUFFERED_IO; 12109236c5fdSDamien Le Moal opts->zone_append = ZLOOP_DEF_ZONE_APPEND; 1211fcc6eaa3SDamien Le Moal opts->ordered_zone_append = ZLOOP_DEF_ORDERED_ZONE_APPEND; 1212eb0570c7SDamien Le Moal 1213eb0570c7SDamien Le Moal if (!buf) 1214eb0570c7SDamien Le Moal return 0; 1215eb0570c7SDamien Le Moal 1216eb0570c7SDamien Le Moal /* Skip leading spaces before the options. */ 1217eb0570c7SDamien Le Moal while (isspace(*buf)) 1218eb0570c7SDamien Le Moal buf++; 1219eb0570c7SDamien Le Moal 1220eb0570c7SDamien Le Moal options = o = kstrdup(buf, GFP_KERNEL); 1221eb0570c7SDamien Le Moal if (!options) 1222eb0570c7SDamien Le Moal return -ENOMEM; 1223eb0570c7SDamien Le Moal 1224eb0570c7SDamien Le Moal /* Parse the options, doing only some light invalid value checks. */ 1225eb0570c7SDamien Le Moal while ((p = strsep(&o, ",\n")) != NULL) { 1226eb0570c7SDamien Le Moal if (!*p) 1227eb0570c7SDamien Le Moal continue; 1228eb0570c7SDamien Le Moal 1229eb0570c7SDamien Le Moal token = match_token(p, zloop_opt_tokens, args); 1230eb0570c7SDamien Le Moal opts->mask |= token; 1231eb0570c7SDamien Le Moal switch (token) { 1232eb0570c7SDamien Le Moal case ZLOOP_OPT_ID: 1233eb0570c7SDamien Le Moal if (match_int(args, &opts->id)) { 1234eb0570c7SDamien Le Moal ret = -EINVAL; 1235eb0570c7SDamien Le Moal goto out; 1236eb0570c7SDamien Le Moal } 1237eb0570c7SDamien Le Moal break; 1238eb0570c7SDamien Le Moal case ZLOOP_OPT_CAPACITY: 1239eb0570c7SDamien Le Moal if (match_uint(args, &token)) { 1240eb0570c7SDamien Le Moal ret = -EINVAL; 1241eb0570c7SDamien Le Moal goto out; 1242eb0570c7SDamien Le Moal } 1243eb0570c7SDamien Le Moal if (!token) { 1244eb0570c7SDamien Le Moal pr_err("Invalid capacity\n"); 1245eb0570c7SDamien Le Moal ret = -EINVAL; 1246eb0570c7SDamien Le Moal goto out; 1247eb0570c7SDamien Le Moal } 1248eb0570c7SDamien Le Moal opts->capacity = 1249eb0570c7SDamien Le Moal ((sector_t)token * SZ_1M) >> SECTOR_SHIFT; 1250eb0570c7SDamien Le Moal break; 1251eb0570c7SDamien Le Moal case ZLOOP_OPT_ZONE_SIZE: 1252eb0570c7SDamien Le Moal if (match_uint(args, &token)) { 1253eb0570c7SDamien Le Moal ret = -EINVAL; 1254eb0570c7SDamien Le Moal goto out; 1255eb0570c7SDamien Le Moal } 1256eb0570c7SDamien Le Moal if (!token || token > ZLOOP_MAX_ZONE_SIZE_MB || 1257eb0570c7SDamien Le Moal !is_power_of_2(token)) { 1258eb0570c7SDamien Le Moal pr_err("Invalid zone size %u\n", token); 1259eb0570c7SDamien Le Moal ret = -EINVAL; 1260eb0570c7SDamien Le Moal goto out; 1261eb0570c7SDamien Le Moal } 1262eb0570c7SDamien Le Moal opts->zone_size = 1263eb0570c7SDamien Le Moal ((sector_t)token * SZ_1M) >> SECTOR_SHIFT; 1264eb0570c7SDamien Le Moal break; 1265eb0570c7SDamien Le Moal case ZLOOP_OPT_ZONE_CAPACITY: 1266eb0570c7SDamien Le Moal if (match_uint(args, &token)) { 1267eb0570c7SDamien Le Moal ret = -EINVAL; 1268eb0570c7SDamien Le Moal goto out; 1269eb0570c7SDamien Le Moal } 1270eb0570c7SDamien Le Moal if (!token) { 1271eb0570c7SDamien Le Moal pr_err("Invalid zone capacity\n"); 1272eb0570c7SDamien Le Moal ret = -EINVAL; 1273eb0570c7SDamien Le Moal goto out; 1274eb0570c7SDamien Le Moal } 1275eb0570c7SDamien Le Moal opts->zone_capacity = 1276eb0570c7SDamien Le Moal ((sector_t)token * SZ_1M) >> SECTOR_SHIFT; 1277eb0570c7SDamien Le Moal break; 1278eb0570c7SDamien Le Moal case ZLOOP_OPT_NR_CONV_ZONES: 1279eb0570c7SDamien Le Moal if (match_uint(args, &token)) { 1280eb0570c7SDamien Le Moal ret = -EINVAL; 1281eb0570c7SDamien Le Moal goto out; 1282eb0570c7SDamien Le Moal } 1283eb0570c7SDamien Le Moal opts->nr_conv_zones = token; 1284eb0570c7SDamien Le Moal break; 1285eb0570c7SDamien Le Moal case ZLOOP_OPT_BASE_DIR: 1286eb0570c7SDamien Le Moal p = match_strdup(args); 1287eb0570c7SDamien Le Moal if (!p) { 1288eb0570c7SDamien Le Moal ret = -ENOMEM; 1289eb0570c7SDamien Le Moal goto out; 1290eb0570c7SDamien Le Moal } 1291eb0570c7SDamien Le Moal kfree(opts->base_dir); 1292eb0570c7SDamien Le Moal opts->base_dir = p; 1293eb0570c7SDamien Le Moal break; 1294eb0570c7SDamien Le Moal case ZLOOP_OPT_NR_QUEUES: 1295eb0570c7SDamien Le Moal if (match_uint(args, &token)) { 1296eb0570c7SDamien Le Moal ret = -EINVAL; 1297eb0570c7SDamien Le Moal goto out; 1298eb0570c7SDamien Le Moal } 1299eb0570c7SDamien Le Moal if (!token) { 1300eb0570c7SDamien Le Moal pr_err("Invalid number of queues\n"); 1301eb0570c7SDamien Le Moal ret = -EINVAL; 1302eb0570c7SDamien Le Moal goto out; 1303eb0570c7SDamien Le Moal } 1304eb0570c7SDamien Le Moal opts->nr_queues = min(token, num_online_cpus()); 1305eb0570c7SDamien Le Moal break; 1306eb0570c7SDamien Le Moal case ZLOOP_OPT_QUEUE_DEPTH: 1307eb0570c7SDamien Le Moal if (match_uint(args, &token)) { 1308eb0570c7SDamien Le Moal ret = -EINVAL; 1309eb0570c7SDamien Le Moal goto out; 1310eb0570c7SDamien Le Moal } 1311eb0570c7SDamien Le Moal if (!token) { 1312eb0570c7SDamien Le Moal pr_err("Invalid queue depth\n"); 1313eb0570c7SDamien Le Moal ret = -EINVAL; 1314eb0570c7SDamien Le Moal goto out; 1315eb0570c7SDamien Le Moal } 1316eb0570c7SDamien Le Moal opts->queue_depth = token; 1317eb0570c7SDamien Le Moal break; 1318eb0570c7SDamien Le Moal case ZLOOP_OPT_BUFFERED_IO: 1319eb0570c7SDamien Le Moal opts->buffered_io = true; 1320eb0570c7SDamien Le Moal break; 13219236c5fdSDamien Le Moal case ZLOOP_OPT_ZONE_APPEND: 13229236c5fdSDamien Le Moal if (match_uint(args, &token)) { 13239236c5fdSDamien Le Moal ret = -EINVAL; 13249236c5fdSDamien Le Moal goto out; 13259236c5fdSDamien Le Moal } 13269236c5fdSDamien Le Moal if (token != 0 && token != 1) { 13279236c5fdSDamien Le Moal pr_err("Invalid zone_append value\n"); 13289236c5fdSDamien Le Moal ret = -EINVAL; 13299236c5fdSDamien Le Moal goto out; 13309236c5fdSDamien Le Moal } 13319236c5fdSDamien Le Moal opts->zone_append = token; 13329236c5fdSDamien Le Moal break; 1333fcc6eaa3SDamien Le Moal case ZLOOP_OPT_ORDERED_ZONE_APPEND: 1334fcc6eaa3SDamien Le Moal opts->ordered_zone_append = true; 1335fcc6eaa3SDamien Le Moal break; 1336eb0570c7SDamien Le Moal case ZLOOP_OPT_ERR: 1337eb0570c7SDamien Le Moal default: 1338eb0570c7SDamien Le Moal pr_warn("unknown parameter or missing value '%s'\n", p); 1339eb0570c7SDamien Le Moal ret = -EINVAL; 1340eb0570c7SDamien Le Moal goto out; 1341eb0570c7SDamien Le Moal } 1342eb0570c7SDamien Le Moal } 1343eb0570c7SDamien Le Moal 1344eb0570c7SDamien Le Moal ret = -EINVAL; 1345eb0570c7SDamien Le Moal if (opts->capacity <= opts->zone_size) { 1346eb0570c7SDamien Le Moal pr_err("Invalid capacity\n"); 1347eb0570c7SDamien Le Moal goto out; 1348eb0570c7SDamien Le Moal } 1349eb0570c7SDamien Le Moal 1350eb0570c7SDamien Le Moal if (opts->zone_capacity > opts->zone_size) { 1351eb0570c7SDamien Le Moal pr_err("Invalid zone capacity\n"); 1352eb0570c7SDamien Le Moal goto out; 1353eb0570c7SDamien Le Moal } 1354eb0570c7SDamien Le Moal 1355eb0570c7SDamien Le Moal ret = 0; 1356eb0570c7SDamien Le Moal out: 1357eb0570c7SDamien Le Moal kfree(options); 1358eb0570c7SDamien Le Moal return ret; 1359eb0570c7SDamien Le Moal } 1360eb0570c7SDamien Le Moal 1361eb0570c7SDamien Le Moal enum { 1362eb0570c7SDamien Le Moal ZLOOP_CTL_ADD, 1363eb0570c7SDamien Le Moal ZLOOP_CTL_REMOVE, 1364eb0570c7SDamien Le Moal }; 1365eb0570c7SDamien Le Moal 1366eb0570c7SDamien Le Moal static struct zloop_ctl_op { 1367eb0570c7SDamien Le Moal int code; 1368eb0570c7SDamien Le Moal const char *name; 1369eb0570c7SDamien Le Moal } zloop_ctl_ops[] = { 1370eb0570c7SDamien Le Moal { ZLOOP_CTL_ADD, "add" }, 1371eb0570c7SDamien Le Moal { ZLOOP_CTL_REMOVE, "remove" }, 1372eb0570c7SDamien Le Moal { -1, NULL }, 1373eb0570c7SDamien Le Moal }; 1374eb0570c7SDamien Le Moal 1375eb0570c7SDamien Le Moal static ssize_t zloop_ctl_write(struct file *file, const char __user *ubuf, 1376eb0570c7SDamien Le Moal size_t count, loff_t *pos) 1377eb0570c7SDamien Le Moal { 1378eb0570c7SDamien Le Moal struct zloop_options opts = { }; 1379eb0570c7SDamien Le Moal struct zloop_ctl_op *op; 1380eb0570c7SDamien Le Moal const char *buf, *opts_buf; 1381eb0570c7SDamien Le Moal int i, ret; 1382eb0570c7SDamien Le Moal 1383eb0570c7SDamien Le Moal if (count > PAGE_SIZE) 1384eb0570c7SDamien Le Moal return -ENOMEM; 1385eb0570c7SDamien Le Moal 1386eb0570c7SDamien Le Moal buf = memdup_user_nul(ubuf, count); 1387eb0570c7SDamien Le Moal if (IS_ERR(buf)) 1388eb0570c7SDamien Le Moal return PTR_ERR(buf); 1389eb0570c7SDamien Le Moal 1390eb0570c7SDamien Le Moal for (i = 0; i < ARRAY_SIZE(zloop_ctl_ops); i++) { 1391eb0570c7SDamien Le Moal op = &zloop_ctl_ops[i]; 1392eb0570c7SDamien Le Moal if (!op->name) { 1393eb0570c7SDamien Le Moal pr_err("Invalid operation\n"); 1394eb0570c7SDamien Le Moal ret = -EINVAL; 1395eb0570c7SDamien Le Moal goto out; 1396eb0570c7SDamien Le Moal } 1397eb0570c7SDamien Le Moal if (!strncmp(buf, op->name, strlen(op->name))) 1398eb0570c7SDamien Le Moal break; 1399eb0570c7SDamien Le Moal } 1400eb0570c7SDamien Le Moal 1401eb0570c7SDamien Le Moal if (count <= strlen(op->name)) 1402eb0570c7SDamien Le Moal opts_buf = NULL; 1403eb0570c7SDamien Le Moal else 1404eb0570c7SDamien Le Moal opts_buf = buf + strlen(op->name); 1405eb0570c7SDamien Le Moal 1406eb0570c7SDamien Le Moal ret = zloop_parse_options(&opts, opts_buf); 1407eb0570c7SDamien Le Moal if (ret) { 1408eb0570c7SDamien Le Moal pr_err("Failed to parse options\n"); 1409eb0570c7SDamien Le Moal goto out; 1410eb0570c7SDamien Le Moal } 1411eb0570c7SDamien Le Moal 1412eb0570c7SDamien Le Moal switch (op->code) { 1413eb0570c7SDamien Le Moal case ZLOOP_CTL_ADD: 1414eb0570c7SDamien Le Moal ret = zloop_ctl_add(&opts); 1415eb0570c7SDamien Le Moal break; 1416eb0570c7SDamien Le Moal case ZLOOP_CTL_REMOVE: 1417eb0570c7SDamien Le Moal ret = zloop_ctl_remove(&opts); 1418eb0570c7SDamien Le Moal break; 1419eb0570c7SDamien Le Moal default: 1420eb0570c7SDamien Le Moal pr_err("Invalid operation\n"); 1421eb0570c7SDamien Le Moal ret = -EINVAL; 1422eb0570c7SDamien Le Moal goto out; 1423eb0570c7SDamien Le Moal } 1424eb0570c7SDamien Le Moal 1425eb0570c7SDamien Le Moal out: 1426eb0570c7SDamien Le Moal kfree(opts.base_dir); 1427eb0570c7SDamien Le Moal kfree(buf); 1428eb0570c7SDamien Le Moal return ret ? ret : count; 1429eb0570c7SDamien Le Moal } 1430eb0570c7SDamien Le Moal 1431eb0570c7SDamien Le Moal static int zloop_ctl_show(struct seq_file *seq_file, void *private) 1432eb0570c7SDamien Le Moal { 1433eb0570c7SDamien Le Moal const struct match_token *tok; 1434eb0570c7SDamien Le Moal int i; 1435eb0570c7SDamien Le Moal 1436eb0570c7SDamien Le Moal /* Add operation */ 1437eb0570c7SDamien Le Moal seq_printf(seq_file, "%s ", zloop_ctl_ops[0].name); 1438eb0570c7SDamien Le Moal for (i = 0; i < ARRAY_SIZE(zloop_opt_tokens); i++) { 1439eb0570c7SDamien Le Moal tok = &zloop_opt_tokens[i]; 1440eb0570c7SDamien Le Moal if (!tok->pattern) 1441eb0570c7SDamien Le Moal break; 1442eb0570c7SDamien Le Moal if (i) 1443eb0570c7SDamien Le Moal seq_putc(seq_file, ','); 1444eb0570c7SDamien Le Moal seq_puts(seq_file, tok->pattern); 1445eb0570c7SDamien Le Moal } 1446eb0570c7SDamien Le Moal seq_putc(seq_file, '\n'); 1447eb0570c7SDamien Le Moal 1448eb0570c7SDamien Le Moal /* Remove operation */ 1449eb0570c7SDamien Le Moal seq_puts(seq_file, zloop_ctl_ops[1].name); 1450eb0570c7SDamien Le Moal seq_puts(seq_file, " id=%d\n"); 1451eb0570c7SDamien Le Moal 1452eb0570c7SDamien Le Moal return 0; 1453eb0570c7SDamien Le Moal } 1454eb0570c7SDamien Le Moal 1455eb0570c7SDamien Le Moal static int zloop_ctl_open(struct inode *inode, struct file *file) 1456eb0570c7SDamien Le Moal { 1457eb0570c7SDamien Le Moal file->private_data = NULL; 1458eb0570c7SDamien Le Moal return single_open(file, zloop_ctl_show, NULL); 1459eb0570c7SDamien Le Moal } 1460eb0570c7SDamien Le Moal 1461eb0570c7SDamien Le Moal static int zloop_ctl_release(struct inode *inode, struct file *file) 1462eb0570c7SDamien Le Moal { 1463eb0570c7SDamien Le Moal return single_release(inode, file); 1464eb0570c7SDamien Le Moal } 1465eb0570c7SDamien Le Moal 1466eb0570c7SDamien Le Moal static const struct file_operations zloop_ctl_fops = { 1467eb0570c7SDamien Le Moal .owner = THIS_MODULE, 1468eb0570c7SDamien Le Moal .open = zloop_ctl_open, 1469eb0570c7SDamien Le Moal .release = zloop_ctl_release, 1470eb0570c7SDamien Le Moal .write = zloop_ctl_write, 1471eb0570c7SDamien Le Moal .read = seq_read, 1472eb0570c7SDamien Le Moal }; 1473eb0570c7SDamien Le Moal 1474eb0570c7SDamien Le Moal static struct miscdevice zloop_misc = { 1475eb0570c7SDamien Le Moal .minor = MISC_DYNAMIC_MINOR, 1476eb0570c7SDamien Le Moal .name = "zloop-control", 1477eb0570c7SDamien Le Moal .fops = &zloop_ctl_fops, 1478eb0570c7SDamien Le Moal }; 1479eb0570c7SDamien Le Moal 1480eb0570c7SDamien Le Moal static int __init zloop_init(void) 1481eb0570c7SDamien Le Moal { 1482eb0570c7SDamien Le Moal int ret; 1483eb0570c7SDamien Le Moal 1484eb0570c7SDamien Le Moal ret = misc_register(&zloop_misc); 1485eb0570c7SDamien Le Moal if (ret) { 1486eb0570c7SDamien Le Moal pr_err("Failed to register misc device: %d\n", ret); 1487eb0570c7SDamien Le Moal return ret; 1488eb0570c7SDamien Le Moal } 1489eb0570c7SDamien Le Moal pr_info("Module loaded\n"); 1490eb0570c7SDamien Le Moal 1491eb0570c7SDamien Le Moal return 0; 1492eb0570c7SDamien Le Moal } 1493eb0570c7SDamien Le Moal 1494eb0570c7SDamien Le Moal static void __exit zloop_exit(void) 1495eb0570c7SDamien Le Moal { 1496eb0570c7SDamien Le Moal misc_deregister(&zloop_misc); 1497eb0570c7SDamien Le Moal idr_destroy(&zloop_index_idr); 1498eb0570c7SDamien Le Moal } 1499eb0570c7SDamien Le Moal 1500eb0570c7SDamien Le Moal module_init(zloop_init); 1501eb0570c7SDamien Le Moal module_exit(zloop_exit); 1502eb0570c7SDamien Le Moal 1503eb0570c7SDamien Le Moal MODULE_DESCRIPTION("Zoned loopback device"); 1504eb0570c7SDamien Le Moal MODULE_LICENSE("GPL"); 1505