xref: /openbmc/linux/drivers/block/null_blk/zoned.c (revision ee1cd5048959de496cd005c50b137212a5b62062)
1eebf34a8SDamien Le Moal // SPDX-License-Identifier: GPL-2.0
2eebf34a8SDamien Le Moal #include <linux/vmalloc.h>
3eebf34a8SDamien Le Moal #include <linux/bitmap.h>
4eebf34a8SDamien Le Moal #include "null_blk.h"
5eebf34a8SDamien Le Moal 
6eebf34a8SDamien Le Moal #define CREATE_TRACE_POINTS
7eebf34a8SDamien Le Moal #include "trace.h"
8eebf34a8SDamien Le Moal 
9db060f54SDamien Le Moal #undef pr_fmt
10db060f54SDamien Le Moal #define pr_fmt(fmt)	"null_blk: " fmt
11db060f54SDamien Le Moal 
mb_to_sects(unsigned long mb)12cd92cdb9SDamien Le Moal static inline sector_t mb_to_sects(unsigned long mb)
13cd92cdb9SDamien Le Moal {
14cd92cdb9SDamien Le Moal 	return ((sector_t)mb * SZ_1M) >> SECTOR_SHIFT;
15cd92cdb9SDamien Le Moal }
16eebf34a8SDamien Le Moal 
null_zone_no(struct nullb_device * dev,sector_t sect)17eebf34a8SDamien Le Moal static inline unsigned int null_zone_no(struct nullb_device *dev, sector_t sect)
18eebf34a8SDamien Le Moal {
19eebf34a8SDamien Le Moal 	return sect >> ilog2(dev->zone_size_sects);
20eebf34a8SDamien Le Moal }
21eebf34a8SDamien Le Moal 
null_lock_zone_res(struct nullb_device * dev)22eebf34a8SDamien Le Moal static inline void null_lock_zone_res(struct nullb_device *dev)
23eebf34a8SDamien Le Moal {
24eebf34a8SDamien Le Moal 	if (dev->need_zone_res_mgmt)
25eebf34a8SDamien Le Moal 		spin_lock_irq(&dev->zone_res_lock);
26eebf34a8SDamien Le Moal }
27eebf34a8SDamien Le Moal 
null_unlock_zone_res(struct nullb_device * dev)28eebf34a8SDamien Le Moal static inline void null_unlock_zone_res(struct nullb_device *dev)
29eebf34a8SDamien Le Moal {
30eebf34a8SDamien Le Moal 	if (dev->need_zone_res_mgmt)
31eebf34a8SDamien Le Moal 		spin_unlock_irq(&dev->zone_res_lock);
32eebf34a8SDamien Le Moal }
33eebf34a8SDamien Le Moal 
null_init_zone_lock(struct nullb_device * dev,struct nullb_zone * zone)34eebf34a8SDamien Le Moal static inline void null_init_zone_lock(struct nullb_device *dev,
35eebf34a8SDamien Le Moal 				       struct nullb_zone *zone)
36eebf34a8SDamien Le Moal {
37eebf34a8SDamien Le Moal 	if (!dev->memory_backed)
38eebf34a8SDamien Le Moal 		spin_lock_init(&zone->spinlock);
39eebf34a8SDamien Le Moal 	else
40eebf34a8SDamien Le Moal 		mutex_init(&zone->mutex);
41eebf34a8SDamien Le Moal }
42eebf34a8SDamien Le Moal 
null_lock_zone(struct nullb_device * dev,struct nullb_zone * zone)43eebf34a8SDamien Le Moal static inline void null_lock_zone(struct nullb_device *dev,
44eebf34a8SDamien Le Moal 				  struct nullb_zone *zone)
45eebf34a8SDamien Le Moal {
46eebf34a8SDamien Le Moal 	if (!dev->memory_backed)
47eebf34a8SDamien Le Moal 		spin_lock_irq(&zone->spinlock);
48eebf34a8SDamien Le Moal 	else
49eebf34a8SDamien Le Moal 		mutex_lock(&zone->mutex);
50eebf34a8SDamien Le Moal }
51eebf34a8SDamien Le Moal 
null_unlock_zone(struct nullb_device * dev,struct nullb_zone * zone)52eebf34a8SDamien Le Moal static inline void null_unlock_zone(struct nullb_device *dev,
53eebf34a8SDamien Le Moal 				    struct nullb_zone *zone)
54eebf34a8SDamien Le Moal {
55eebf34a8SDamien Le Moal 	if (!dev->memory_backed)
56eebf34a8SDamien Le Moal 		spin_unlock_irq(&zone->spinlock);
57eebf34a8SDamien Le Moal 	else
58eebf34a8SDamien Le Moal 		mutex_unlock(&zone->mutex);
59eebf34a8SDamien Le Moal }
60eebf34a8SDamien Le Moal 
null_init_zoned_dev(struct nullb_device * dev,struct request_queue * q)61eebf34a8SDamien Le Moal int null_init_zoned_dev(struct nullb_device *dev, struct request_queue *q)
62eebf34a8SDamien Le Moal {
63eebf34a8SDamien Le Moal 	sector_t dev_capacity_sects, zone_capacity_sects;
64eebf34a8SDamien Le Moal 	struct nullb_zone *zone;
65eebf34a8SDamien Le Moal 	sector_t sector = 0;
66eebf34a8SDamien Le Moal 	unsigned int i;
67eebf34a8SDamien Le Moal 
68eebf34a8SDamien Le Moal 	if (!is_power_of_2(dev->zone_size)) {
69eebf34a8SDamien Le Moal 		pr_err("zone_size must be power-of-two\n");
70eebf34a8SDamien Le Moal 		return -EINVAL;
71eebf34a8SDamien Le Moal 	}
72eebf34a8SDamien Le Moal 	if (dev->zone_size > dev->size) {
73eebf34a8SDamien Le Moal 		pr_err("Zone size larger than device capacity\n");
74eebf34a8SDamien Le Moal 		return -EINVAL;
75eebf34a8SDamien Le Moal 	}
76eebf34a8SDamien Le Moal 
77eebf34a8SDamien Le Moal 	if (!dev->zone_capacity)
78eebf34a8SDamien Le Moal 		dev->zone_capacity = dev->zone_size;
79eebf34a8SDamien Le Moal 
80eebf34a8SDamien Le Moal 	if (dev->zone_capacity > dev->zone_size) {
81db060f54SDamien Le Moal 		pr_err("zone capacity (%lu MB) larger than zone size (%lu MB)\n",
82eebf34a8SDamien Le Moal 		       dev->zone_capacity, dev->zone_size);
83eebf34a8SDamien Le Moal 		return -EINVAL;
84eebf34a8SDamien Le Moal 	}
85eebf34a8SDamien Le Moal 
86*9ac895a8SDamien Le Moal 	/*
87*9ac895a8SDamien Le Moal 	 * If a smaller zone capacity was requested, do not allow a smaller last
88*9ac895a8SDamien Le Moal 	 * zone at the same time as such zone configuration does not correspond
89*9ac895a8SDamien Le Moal 	 * to any real zoned device.
90*9ac895a8SDamien Le Moal 	 */
91*9ac895a8SDamien Le Moal 	if (dev->zone_capacity != dev->zone_size &&
92*9ac895a8SDamien Le Moal 	    dev->size & (dev->zone_size - 1)) {
93*9ac895a8SDamien Le Moal 		pr_err("A smaller last zone is not allowed with zone capacity smaller than zone size.\n");
94*9ac895a8SDamien Le Moal 		return -EINVAL;
95*9ac895a8SDamien Le Moal 	}
96*9ac895a8SDamien Le Moal 
97cd92cdb9SDamien Le Moal 	zone_capacity_sects = mb_to_sects(dev->zone_capacity);
98cd92cdb9SDamien Le Moal 	dev_capacity_sects = mb_to_sects(dev->size);
99cd92cdb9SDamien Le Moal 	dev->zone_size_sects = mb_to_sects(dev->zone_size);
100cd92cdb9SDamien Le Moal 	dev->nr_zones = round_up(dev_capacity_sects, dev->zone_size_sects)
101cd92cdb9SDamien Le Moal 		>> ilog2(dev->zone_size_sects);
102eebf34a8SDamien Le Moal 
103eebf34a8SDamien Le Moal 	dev->zones = kvmalloc_array(dev->nr_zones, sizeof(struct nullb_zone),
104eebf34a8SDamien Le Moal 				    GFP_KERNEL | __GFP_ZERO);
105eebf34a8SDamien Le Moal 	if (!dev->zones)
106eebf34a8SDamien Le Moal 		return -ENOMEM;
107eebf34a8SDamien Le Moal 
108eebf34a8SDamien Le Moal 	spin_lock_init(&dev->zone_res_lock);
109eebf34a8SDamien Le Moal 
110eebf34a8SDamien Le Moal 	if (dev->zone_nr_conv >= dev->nr_zones) {
111eebf34a8SDamien Le Moal 		dev->zone_nr_conv = dev->nr_zones - 1;
112eebf34a8SDamien Le Moal 		pr_info("changed the number of conventional zones to %u",
113eebf34a8SDamien Le Moal 			dev->zone_nr_conv);
114eebf34a8SDamien Le Moal 	}
115eebf34a8SDamien Le Moal 
116eebf34a8SDamien Le Moal 	/* Max active zones has to be < nbr of seq zones in order to be enforceable */
117eebf34a8SDamien Le Moal 	if (dev->zone_max_active >= dev->nr_zones - dev->zone_nr_conv) {
118eebf34a8SDamien Le Moal 		dev->zone_max_active = 0;
119eebf34a8SDamien Le Moal 		pr_info("zone_max_active limit disabled, limit >= zone count\n");
120eebf34a8SDamien Le Moal 	}
121eebf34a8SDamien Le Moal 
122eebf34a8SDamien Le Moal 	/* Max open zones has to be <= max active zones */
123eebf34a8SDamien Le Moal 	if (dev->zone_max_active && dev->zone_max_open > dev->zone_max_active) {
124eebf34a8SDamien Le Moal 		dev->zone_max_open = dev->zone_max_active;
125eebf34a8SDamien Le Moal 		pr_info("changed the maximum number of open zones to %u\n",
1262c581ca0SDamien Le Moal 			dev->zone_max_open);
127eebf34a8SDamien Le Moal 	} else if (dev->zone_max_open >= dev->nr_zones - dev->zone_nr_conv) {
128eebf34a8SDamien Le Moal 		dev->zone_max_open = 0;
129eebf34a8SDamien Le Moal 		pr_info("zone_max_open limit disabled, limit >= zone count\n");
130eebf34a8SDamien Le Moal 	}
131eebf34a8SDamien Le Moal 	dev->need_zone_res_mgmt = dev->zone_max_active || dev->zone_max_open;
132eebf34a8SDamien Le Moal 	dev->imp_close_zone_no = dev->zone_nr_conv;
133eebf34a8SDamien Le Moal 
134eebf34a8SDamien Le Moal 	for (i = 0; i <  dev->zone_nr_conv; i++) {
135eebf34a8SDamien Le Moal 		zone = &dev->zones[i];
136eebf34a8SDamien Le Moal 
137eebf34a8SDamien Le Moal 		null_init_zone_lock(dev, zone);
138eebf34a8SDamien Le Moal 		zone->start = sector;
139eebf34a8SDamien Le Moal 		zone->len = dev->zone_size_sects;
140eebf34a8SDamien Le Moal 		zone->capacity = zone->len;
141eebf34a8SDamien Le Moal 		zone->wp = zone->start + zone->len;
142eebf34a8SDamien Le Moal 		zone->type = BLK_ZONE_TYPE_CONVENTIONAL;
143eebf34a8SDamien Le Moal 		zone->cond = BLK_ZONE_COND_NOT_WP;
144eebf34a8SDamien Le Moal 
145eebf34a8SDamien Le Moal 		sector += dev->zone_size_sects;
146eebf34a8SDamien Le Moal 	}
147eebf34a8SDamien Le Moal 
148eebf34a8SDamien Le Moal 	for (i = dev->zone_nr_conv; i < dev->nr_zones; i++) {
149eebf34a8SDamien Le Moal 		zone = &dev->zones[i];
150eebf34a8SDamien Le Moal 
151eebf34a8SDamien Le Moal 		null_init_zone_lock(dev, zone);
152eebf34a8SDamien Le Moal 		zone->start = zone->wp = sector;
153eebf34a8SDamien Le Moal 		if (zone->start + dev->zone_size_sects > dev_capacity_sects)
154eebf34a8SDamien Le Moal 			zone->len = dev_capacity_sects - zone->start;
155eebf34a8SDamien Le Moal 		else
156eebf34a8SDamien Le Moal 			zone->len = dev->zone_size_sects;
157eebf34a8SDamien Le Moal 		zone->capacity =
158eebf34a8SDamien Le Moal 			min_t(sector_t, zone->len, zone_capacity_sects);
159eebf34a8SDamien Le Moal 		zone->type = BLK_ZONE_TYPE_SEQWRITE_REQ;
160eebf34a8SDamien Le Moal 		zone->cond = BLK_ZONE_COND_EMPTY;
161eebf34a8SDamien Le Moal 
162eebf34a8SDamien Le Moal 		sector += dev->zone_size_sects;
163eebf34a8SDamien Le Moal 	}
164eebf34a8SDamien Le Moal 
165eebf34a8SDamien Le Moal 	return 0;
166eebf34a8SDamien Le Moal }
167eebf34a8SDamien Le Moal 
null_register_zoned_dev(struct nullb * nullb)168eebf34a8SDamien Le Moal int null_register_zoned_dev(struct nullb *nullb)
169eebf34a8SDamien Le Moal {
170eebf34a8SDamien Le Moal 	struct nullb_device *dev = nullb->dev;
171eebf34a8SDamien Le Moal 	struct request_queue *q = nullb->q;
172eebf34a8SDamien Le Moal 
1736b2bd274SChristoph Hellwig 	disk_set_zoned(nullb->disk, BLK_ZONED_HM);
1745752dc78SDamien Le Moal 	blk_queue_flag_set(QUEUE_FLAG_ZONE_RESETALL, q);
1755752dc78SDamien Le Moal 	blk_queue_required_elevator_features(q, ELEVATOR_F_ZBD_SEQ_WRITE);
176eebf34a8SDamien Le Moal 	blk_queue_chunk_sectors(q, dev->zone_size_sects);
177d86e716aSChristoph Hellwig 	nullb->disk->nr_zones = bdev_nr_zones(nullb->disk->part0);
178eebf34a8SDamien Le Moal 	blk_queue_max_zone_append_sectors(q, dev->zone_size_sects);
179982977dfSChristoph Hellwig 	disk_set_max_open_zones(nullb->disk, dev->zone_max_open);
180982977dfSChristoph Hellwig 	disk_set_max_active_zones(nullb->disk, dev->zone_max_active);
181eebf34a8SDamien Le Moal 
182a442b899SDamien Le Moal 	if (queue_is_mq(q))
183a442b899SDamien Le Moal 		return blk_revalidate_disk_zones(nullb->disk, NULL);
184a442b899SDamien Le Moal 
185eebf34a8SDamien Le Moal 	return 0;
186eebf34a8SDamien Le Moal }
187eebf34a8SDamien Le Moal 
null_free_zoned_dev(struct nullb_device * dev)188eebf34a8SDamien Le Moal void null_free_zoned_dev(struct nullb_device *dev)
189eebf34a8SDamien Le Moal {
190eebf34a8SDamien Le Moal 	kvfree(dev->zones);
19172ce11ddSLv Yunlong 	dev->zones = NULL;
192eebf34a8SDamien Le Moal }
193eebf34a8SDamien Le Moal 
null_report_zones(struct gendisk * disk,sector_t sector,unsigned int nr_zones,report_zones_cb cb,void * data)194eebf34a8SDamien Le Moal int null_report_zones(struct gendisk *disk, sector_t sector,
195eebf34a8SDamien Le Moal 		unsigned int nr_zones, report_zones_cb cb, void *data)
196eebf34a8SDamien Le Moal {
197eebf34a8SDamien Le Moal 	struct nullb *nullb = disk->private_data;
198eebf34a8SDamien Le Moal 	struct nullb_device *dev = nullb->dev;
199eebf34a8SDamien Le Moal 	unsigned int first_zone, i;
200eebf34a8SDamien Le Moal 	struct nullb_zone *zone;
201eebf34a8SDamien Le Moal 	struct blk_zone blkz;
202eebf34a8SDamien Le Moal 	int error;
203eebf34a8SDamien Le Moal 
204eebf34a8SDamien Le Moal 	first_zone = null_zone_no(dev, sector);
205eebf34a8SDamien Le Moal 	if (first_zone >= dev->nr_zones)
206eebf34a8SDamien Le Moal 		return 0;
207eebf34a8SDamien Le Moal 
208eebf34a8SDamien Le Moal 	nr_zones = min(nr_zones, dev->nr_zones - first_zone);
209eebf34a8SDamien Le Moal 	trace_nullb_report_zones(nullb, nr_zones);
210eebf34a8SDamien Le Moal 
211eebf34a8SDamien Le Moal 	memset(&blkz, 0, sizeof(struct blk_zone));
212eebf34a8SDamien Le Moal 	zone = &dev->zones[first_zone];
213eebf34a8SDamien Le Moal 	for (i = 0; i < nr_zones; i++, zone++) {
214eebf34a8SDamien Le Moal 		/*
215eebf34a8SDamien Le Moal 		 * Stacked DM target drivers will remap the zone information by
216eebf34a8SDamien Le Moal 		 * modifying the zone information passed to the report callback.
217eebf34a8SDamien Le Moal 		 * So use a local copy to avoid corruption of the device zone
218eebf34a8SDamien Le Moal 		 * array.
219eebf34a8SDamien Le Moal 		 */
220eebf34a8SDamien Le Moal 		null_lock_zone(dev, zone);
221eebf34a8SDamien Le Moal 		blkz.start = zone->start;
222eebf34a8SDamien Le Moal 		blkz.len = zone->len;
223eebf34a8SDamien Le Moal 		blkz.wp = zone->wp;
224eebf34a8SDamien Le Moal 		blkz.type = zone->type;
225eebf34a8SDamien Le Moal 		blkz.cond = zone->cond;
226eebf34a8SDamien Le Moal 		blkz.capacity = zone->capacity;
227eebf34a8SDamien Le Moal 		null_unlock_zone(dev, zone);
228eebf34a8SDamien Le Moal 
229eebf34a8SDamien Le Moal 		error = cb(&blkz, i, data);
230eebf34a8SDamien Le Moal 		if (error)
231eebf34a8SDamien Le Moal 			return error;
232eebf34a8SDamien Le Moal 	}
233eebf34a8SDamien Le Moal 
234eebf34a8SDamien Le Moal 	return nr_zones;
235eebf34a8SDamien Le Moal }
236eebf34a8SDamien Le Moal 
237eebf34a8SDamien Le Moal /*
238eebf34a8SDamien Le Moal  * This is called in the case of memory backing from null_process_cmd()
239eebf34a8SDamien Le Moal  * with the target zone already locked.
240eebf34a8SDamien Le Moal  */
null_zone_valid_read_len(struct nullb * nullb,sector_t sector,unsigned int len)241eebf34a8SDamien Le Moal size_t null_zone_valid_read_len(struct nullb *nullb,
242eebf34a8SDamien Le Moal 				sector_t sector, unsigned int len)
243eebf34a8SDamien Le Moal {
244eebf34a8SDamien Le Moal 	struct nullb_device *dev = nullb->dev;
245eebf34a8SDamien Le Moal 	struct nullb_zone *zone = &dev->zones[null_zone_no(dev, sector)];
246eebf34a8SDamien Le Moal 	unsigned int nr_sectors = len >> SECTOR_SHIFT;
247eebf34a8SDamien Le Moal 
248eebf34a8SDamien Le Moal 	/* Read must be below the write pointer position */
249eebf34a8SDamien Le Moal 	if (zone->type == BLK_ZONE_TYPE_CONVENTIONAL ||
250eebf34a8SDamien Le Moal 	    sector + nr_sectors <= zone->wp)
251eebf34a8SDamien Le Moal 		return len;
252eebf34a8SDamien Le Moal 
253eebf34a8SDamien Le Moal 	if (sector > zone->wp)
254eebf34a8SDamien Le Moal 		return 0;
255eebf34a8SDamien Le Moal 
256eebf34a8SDamien Le Moal 	return (zone->wp - sector) << SECTOR_SHIFT;
257eebf34a8SDamien Le Moal }
258eebf34a8SDamien Le Moal 
__null_close_zone(struct nullb_device * dev,struct nullb_zone * zone)259eebf34a8SDamien Le Moal static blk_status_t __null_close_zone(struct nullb_device *dev,
260eebf34a8SDamien Le Moal 				      struct nullb_zone *zone)
261eebf34a8SDamien Le Moal {
262eebf34a8SDamien Le Moal 	switch (zone->cond) {
263eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_CLOSED:
264eebf34a8SDamien Le Moal 		/* close operation on closed is not an error */
265eebf34a8SDamien Le Moal 		return BLK_STS_OK;
266eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_IMP_OPEN:
267eebf34a8SDamien Le Moal 		dev->nr_zones_imp_open--;
268eebf34a8SDamien Le Moal 		break;
269eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EXP_OPEN:
270eebf34a8SDamien Le Moal 		dev->nr_zones_exp_open--;
271eebf34a8SDamien Le Moal 		break;
272eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EMPTY:
273eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_FULL:
274eebf34a8SDamien Le Moal 	default:
275eebf34a8SDamien Le Moal 		return BLK_STS_IOERR;
276eebf34a8SDamien Le Moal 	}
277eebf34a8SDamien Le Moal 
278eebf34a8SDamien Le Moal 	if (zone->wp == zone->start) {
279eebf34a8SDamien Le Moal 		zone->cond = BLK_ZONE_COND_EMPTY;
280eebf34a8SDamien Le Moal 	} else {
281eebf34a8SDamien Le Moal 		zone->cond = BLK_ZONE_COND_CLOSED;
282eebf34a8SDamien Le Moal 		dev->nr_zones_closed++;
283eebf34a8SDamien Le Moal 	}
284eebf34a8SDamien Le Moal 
285eebf34a8SDamien Le Moal 	return BLK_STS_OK;
286eebf34a8SDamien Le Moal }
287eebf34a8SDamien Le Moal 
null_close_imp_open_zone(struct nullb_device * dev)288eebf34a8SDamien Le Moal static void null_close_imp_open_zone(struct nullb_device *dev)
289eebf34a8SDamien Le Moal {
290eebf34a8SDamien Le Moal 	struct nullb_zone *zone;
291eebf34a8SDamien Le Moal 	unsigned int zno, i;
292eebf34a8SDamien Le Moal 
293eebf34a8SDamien Le Moal 	zno = dev->imp_close_zone_no;
294eebf34a8SDamien Le Moal 	if (zno >= dev->nr_zones)
295eebf34a8SDamien Le Moal 		zno = dev->zone_nr_conv;
296eebf34a8SDamien Le Moal 
297eebf34a8SDamien Le Moal 	for (i = dev->zone_nr_conv; i < dev->nr_zones; i++) {
298eebf34a8SDamien Le Moal 		zone = &dev->zones[zno];
299eebf34a8SDamien Le Moal 		zno++;
300eebf34a8SDamien Le Moal 		if (zno >= dev->nr_zones)
301eebf34a8SDamien Le Moal 			zno = dev->zone_nr_conv;
302eebf34a8SDamien Le Moal 
303eebf34a8SDamien Le Moal 		if (zone->cond == BLK_ZONE_COND_IMP_OPEN) {
304eebf34a8SDamien Le Moal 			__null_close_zone(dev, zone);
305eebf34a8SDamien Le Moal 			dev->imp_close_zone_no = zno;
306eebf34a8SDamien Le Moal 			return;
307eebf34a8SDamien Le Moal 		}
308eebf34a8SDamien Le Moal 	}
309eebf34a8SDamien Le Moal }
310eebf34a8SDamien Le Moal 
null_check_active(struct nullb_device * dev)311eebf34a8SDamien Le Moal static blk_status_t null_check_active(struct nullb_device *dev)
312eebf34a8SDamien Le Moal {
313eebf34a8SDamien Le Moal 	if (!dev->zone_max_active)
314eebf34a8SDamien Le Moal 		return BLK_STS_OK;
315eebf34a8SDamien Le Moal 
316eebf34a8SDamien Le Moal 	if (dev->nr_zones_exp_open + dev->nr_zones_imp_open +
317eebf34a8SDamien Le Moal 			dev->nr_zones_closed < dev->zone_max_active)
318eebf34a8SDamien Le Moal 		return BLK_STS_OK;
319eebf34a8SDamien Le Moal 
320eebf34a8SDamien Le Moal 	return BLK_STS_ZONE_ACTIVE_RESOURCE;
321eebf34a8SDamien Le Moal }
322eebf34a8SDamien Le Moal 
null_check_open(struct nullb_device * dev)323eebf34a8SDamien Le Moal static blk_status_t null_check_open(struct nullb_device *dev)
324eebf34a8SDamien Le Moal {
325eebf34a8SDamien Le Moal 	if (!dev->zone_max_open)
326eebf34a8SDamien Le Moal 		return BLK_STS_OK;
327eebf34a8SDamien Le Moal 
328eebf34a8SDamien Le Moal 	if (dev->nr_zones_exp_open + dev->nr_zones_imp_open < dev->zone_max_open)
329eebf34a8SDamien Le Moal 		return BLK_STS_OK;
330eebf34a8SDamien Le Moal 
331eebf34a8SDamien Le Moal 	if (dev->nr_zones_imp_open) {
332eebf34a8SDamien Le Moal 		if (null_check_active(dev) == BLK_STS_OK) {
333eebf34a8SDamien Le Moal 			null_close_imp_open_zone(dev);
334eebf34a8SDamien Le Moal 			return BLK_STS_OK;
335eebf34a8SDamien Le Moal 		}
336eebf34a8SDamien Le Moal 	}
337eebf34a8SDamien Le Moal 
338eebf34a8SDamien Le Moal 	return BLK_STS_ZONE_OPEN_RESOURCE;
339eebf34a8SDamien Le Moal }
340eebf34a8SDamien Le Moal 
341eebf34a8SDamien Le Moal /*
342eebf34a8SDamien Le Moal  * This function matches the manage open zone resources function in the ZBC standard,
343eebf34a8SDamien Le Moal  * with the addition of max active zones support (added in the ZNS standard).
344eebf34a8SDamien Le Moal  *
345eebf34a8SDamien Le Moal  * The function determines if a zone can transition to implicit open or explicit open,
346eebf34a8SDamien Le Moal  * while maintaining the max open zone (and max active zone) limit(s). It may close an
347eebf34a8SDamien Le Moal  * implicit open zone in order to make additional zone resources available.
348eebf34a8SDamien Le Moal  *
349eebf34a8SDamien Le Moal  * ZBC states that an implicit open zone shall be closed only if there is not
350eebf34a8SDamien Le Moal  * room within the open limit. However, with the addition of an active limit,
351eebf34a8SDamien Le Moal  * it is not certain that closing an implicit open zone will allow a new zone
352eebf34a8SDamien Le Moal  * to be opened, since we might already be at the active limit capacity.
353eebf34a8SDamien Le Moal  */
null_check_zone_resources(struct nullb_device * dev,struct nullb_zone * zone)354eebf34a8SDamien Le Moal static blk_status_t null_check_zone_resources(struct nullb_device *dev,
355eebf34a8SDamien Le Moal 					      struct nullb_zone *zone)
356eebf34a8SDamien Le Moal {
357eebf34a8SDamien Le Moal 	blk_status_t ret;
358eebf34a8SDamien Le Moal 
359eebf34a8SDamien Le Moal 	switch (zone->cond) {
360eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EMPTY:
361eebf34a8SDamien Le Moal 		ret = null_check_active(dev);
362eebf34a8SDamien Le Moal 		if (ret != BLK_STS_OK)
363eebf34a8SDamien Le Moal 			return ret;
364eebf34a8SDamien Le Moal 		fallthrough;
365eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_CLOSED:
366eebf34a8SDamien Le Moal 		return null_check_open(dev);
367eebf34a8SDamien Le Moal 	default:
368eebf34a8SDamien Le Moal 		/* Should never be called for other states */
369eebf34a8SDamien Le Moal 		WARN_ON(1);
370eebf34a8SDamien Le Moal 		return BLK_STS_IOERR;
371eebf34a8SDamien Le Moal 	}
372eebf34a8SDamien Le Moal }
373eebf34a8SDamien Le Moal 
null_zone_write(struct nullb_cmd * cmd,sector_t sector,unsigned int nr_sectors,bool append)374eebf34a8SDamien Le Moal static blk_status_t null_zone_write(struct nullb_cmd *cmd, sector_t sector,
375eebf34a8SDamien Le Moal 				    unsigned int nr_sectors, bool append)
376eebf34a8SDamien Le Moal {
377eebf34a8SDamien Le Moal 	struct nullb_device *dev = cmd->nq->dev;
378eebf34a8SDamien Le Moal 	unsigned int zno = null_zone_no(dev, sector);
379eebf34a8SDamien Le Moal 	struct nullb_zone *zone = &dev->zones[zno];
380eebf34a8SDamien Le Moal 	blk_status_t ret;
381eebf34a8SDamien Le Moal 
382eebf34a8SDamien Le Moal 	trace_nullb_zone_op(cmd, zno, zone->cond);
383eebf34a8SDamien Le Moal 
384eebf34a8SDamien Le Moal 	if (zone->type == BLK_ZONE_TYPE_CONVENTIONAL) {
385eebf34a8SDamien Le Moal 		if (append)
386eebf34a8SDamien Le Moal 			return BLK_STS_IOERR;
387eebf34a8SDamien Le Moal 		return null_process_cmd(cmd, REQ_OP_WRITE, sector, nr_sectors);
388eebf34a8SDamien Le Moal 	}
389eebf34a8SDamien Le Moal 
390eebf34a8SDamien Le Moal 	null_lock_zone(dev, zone);
391eebf34a8SDamien Le Moal 
392d3a57388SShin'ichiro Kawasaki 	if (zone->cond == BLK_ZONE_COND_FULL ||
393d3a57388SShin'ichiro Kawasaki 	    zone->cond == BLK_ZONE_COND_READONLY ||
394d3a57388SShin'ichiro Kawasaki 	    zone->cond == BLK_ZONE_COND_OFFLINE) {
395d3a57388SShin'ichiro Kawasaki 		/* Cannot write to the zone */
396eebf34a8SDamien Le Moal 		ret = BLK_STS_IOERR;
397eebf34a8SDamien Le Moal 		goto unlock;
398eebf34a8SDamien Le Moal 	}
399eebf34a8SDamien Le Moal 
400eebf34a8SDamien Le Moal 	/*
401eebf34a8SDamien Le Moal 	 * Regular writes must be at the write pointer position.
402eebf34a8SDamien Le Moal 	 * Zone append writes are automatically issued at the write
403eebf34a8SDamien Le Moal 	 * pointer and the position returned using the request or BIO
404eebf34a8SDamien Le Moal 	 * sector.
405eebf34a8SDamien Le Moal 	 */
406eebf34a8SDamien Le Moal 	if (append) {
407eebf34a8SDamien Le Moal 		sector = zone->wp;
408aacae8c4SDamien Le Moal 		if (dev->queue_mode == NULL_Q_MQ)
409eebf34a8SDamien Le Moal 			cmd->rq->__sector = sector;
410aacae8c4SDamien Le Moal 		else
411aacae8c4SDamien Le Moal 			cmd->bio->bi_iter.bi_sector = sector;
412eebf34a8SDamien Le Moal 	} else if (sector != zone->wp) {
413eebf34a8SDamien Le Moal 		ret = BLK_STS_IOERR;
414eebf34a8SDamien Le Moal 		goto unlock;
415eebf34a8SDamien Le Moal 	}
416eebf34a8SDamien Le Moal 
417eebf34a8SDamien Le Moal 	if (zone->wp + nr_sectors > zone->start + zone->capacity) {
418eebf34a8SDamien Le Moal 		ret = BLK_STS_IOERR;
419eebf34a8SDamien Le Moal 		goto unlock;
420eebf34a8SDamien Le Moal 	}
421eebf34a8SDamien Le Moal 
422eebf34a8SDamien Le Moal 	if (zone->cond == BLK_ZONE_COND_CLOSED ||
423eebf34a8SDamien Le Moal 	    zone->cond == BLK_ZONE_COND_EMPTY) {
424eebf34a8SDamien Le Moal 		null_lock_zone_res(dev);
425eebf34a8SDamien Le Moal 
426eebf34a8SDamien Le Moal 		ret = null_check_zone_resources(dev, zone);
427eebf34a8SDamien Le Moal 		if (ret != BLK_STS_OK) {
428eebf34a8SDamien Le Moal 			null_unlock_zone_res(dev);
429eebf34a8SDamien Le Moal 			goto unlock;
430eebf34a8SDamien Le Moal 		}
431eebf34a8SDamien Le Moal 		if (zone->cond == BLK_ZONE_COND_CLOSED) {
432eebf34a8SDamien Le Moal 			dev->nr_zones_closed--;
433eebf34a8SDamien Le Moal 			dev->nr_zones_imp_open++;
434eebf34a8SDamien Le Moal 		} else if (zone->cond == BLK_ZONE_COND_EMPTY) {
435eebf34a8SDamien Le Moal 			dev->nr_zones_imp_open++;
436eebf34a8SDamien Le Moal 		}
437eebf34a8SDamien Le Moal 
438eebf34a8SDamien Le Moal 		if (zone->cond != BLK_ZONE_COND_EXP_OPEN)
439eebf34a8SDamien Le Moal 			zone->cond = BLK_ZONE_COND_IMP_OPEN;
440eebf34a8SDamien Le Moal 
441eebf34a8SDamien Le Moal 		null_unlock_zone_res(dev);
442eebf34a8SDamien Le Moal 	}
443eebf34a8SDamien Le Moal 
444eebf34a8SDamien Le Moal 	ret = null_process_cmd(cmd, REQ_OP_WRITE, sector, nr_sectors);
445eebf34a8SDamien Le Moal 	if (ret != BLK_STS_OK)
446eebf34a8SDamien Le Moal 		goto unlock;
447eebf34a8SDamien Le Moal 
448eebf34a8SDamien Le Moal 	zone->wp += nr_sectors;
449eebf34a8SDamien Le Moal 	if (zone->wp == zone->start + zone->capacity) {
450eebf34a8SDamien Le Moal 		null_lock_zone_res(dev);
451eebf34a8SDamien Le Moal 		if (zone->cond == BLK_ZONE_COND_EXP_OPEN)
452eebf34a8SDamien Le Moal 			dev->nr_zones_exp_open--;
453eebf34a8SDamien Le Moal 		else if (zone->cond == BLK_ZONE_COND_IMP_OPEN)
454eebf34a8SDamien Le Moal 			dev->nr_zones_imp_open--;
455eebf34a8SDamien Le Moal 		zone->cond = BLK_ZONE_COND_FULL;
456eebf34a8SDamien Le Moal 		null_unlock_zone_res(dev);
457eebf34a8SDamien Le Moal 	}
458eebf34a8SDamien Le Moal 
459eebf34a8SDamien Le Moal 	ret = BLK_STS_OK;
460eebf34a8SDamien Le Moal 
461eebf34a8SDamien Le Moal unlock:
462eebf34a8SDamien Le Moal 	null_unlock_zone(dev, zone);
463eebf34a8SDamien Le Moal 
464eebf34a8SDamien Le Moal 	return ret;
465eebf34a8SDamien Le Moal }
466eebf34a8SDamien Le Moal 
null_open_zone(struct nullb_device * dev,struct nullb_zone * zone)467eebf34a8SDamien Le Moal static blk_status_t null_open_zone(struct nullb_device *dev,
468eebf34a8SDamien Le Moal 				   struct nullb_zone *zone)
469eebf34a8SDamien Le Moal {
470eebf34a8SDamien Le Moal 	blk_status_t ret = BLK_STS_OK;
471eebf34a8SDamien Le Moal 
472eebf34a8SDamien Le Moal 	if (zone->type == BLK_ZONE_TYPE_CONVENTIONAL)
473eebf34a8SDamien Le Moal 		return BLK_STS_IOERR;
474eebf34a8SDamien Le Moal 
475eebf34a8SDamien Le Moal 	null_lock_zone_res(dev);
476eebf34a8SDamien Le Moal 
477eebf34a8SDamien Le Moal 	switch (zone->cond) {
478eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EXP_OPEN:
479eebf34a8SDamien Le Moal 		/* open operation on exp open is not an error */
480eebf34a8SDamien Le Moal 		goto unlock;
481eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EMPTY:
482eebf34a8SDamien Le Moal 		ret = null_check_zone_resources(dev, zone);
483eebf34a8SDamien Le Moal 		if (ret != BLK_STS_OK)
484eebf34a8SDamien Le Moal 			goto unlock;
485eebf34a8SDamien Le Moal 		break;
486eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_IMP_OPEN:
487eebf34a8SDamien Le Moal 		dev->nr_zones_imp_open--;
488eebf34a8SDamien Le Moal 		break;
489eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_CLOSED:
490eebf34a8SDamien Le Moal 		ret = null_check_zone_resources(dev, zone);
491eebf34a8SDamien Le Moal 		if (ret != BLK_STS_OK)
492eebf34a8SDamien Le Moal 			goto unlock;
493eebf34a8SDamien Le Moal 		dev->nr_zones_closed--;
494eebf34a8SDamien Le Moal 		break;
495eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_FULL:
496eebf34a8SDamien Le Moal 	default:
497eebf34a8SDamien Le Moal 		ret = BLK_STS_IOERR;
498eebf34a8SDamien Le Moal 		goto unlock;
499eebf34a8SDamien Le Moal 	}
500eebf34a8SDamien Le Moal 
501eebf34a8SDamien Le Moal 	zone->cond = BLK_ZONE_COND_EXP_OPEN;
502eebf34a8SDamien Le Moal 	dev->nr_zones_exp_open++;
503eebf34a8SDamien Le Moal 
504eebf34a8SDamien Le Moal unlock:
505eebf34a8SDamien Le Moal 	null_unlock_zone_res(dev);
506eebf34a8SDamien Le Moal 
507eebf34a8SDamien Le Moal 	return ret;
508eebf34a8SDamien Le Moal }
509eebf34a8SDamien Le Moal 
null_close_zone(struct nullb_device * dev,struct nullb_zone * zone)510eebf34a8SDamien Le Moal static blk_status_t null_close_zone(struct nullb_device *dev,
511eebf34a8SDamien Le Moal 				    struct nullb_zone *zone)
512eebf34a8SDamien Le Moal {
513eebf34a8SDamien Le Moal 	blk_status_t ret;
514eebf34a8SDamien Le Moal 
515eebf34a8SDamien Le Moal 	if (zone->type == BLK_ZONE_TYPE_CONVENTIONAL)
516eebf34a8SDamien Le Moal 		return BLK_STS_IOERR;
517eebf34a8SDamien Le Moal 
518eebf34a8SDamien Le Moal 	null_lock_zone_res(dev);
519eebf34a8SDamien Le Moal 	ret = __null_close_zone(dev, zone);
520eebf34a8SDamien Le Moal 	null_unlock_zone_res(dev);
521eebf34a8SDamien Le Moal 
522eebf34a8SDamien Le Moal 	return ret;
523eebf34a8SDamien Le Moal }
524eebf34a8SDamien Le Moal 
null_finish_zone(struct nullb_device * dev,struct nullb_zone * zone)525eebf34a8SDamien Le Moal static blk_status_t null_finish_zone(struct nullb_device *dev,
526eebf34a8SDamien Le Moal 				     struct nullb_zone *zone)
527eebf34a8SDamien Le Moal {
528eebf34a8SDamien Le Moal 	blk_status_t ret = BLK_STS_OK;
529eebf34a8SDamien Le Moal 
530eebf34a8SDamien Le Moal 	if (zone->type == BLK_ZONE_TYPE_CONVENTIONAL)
531eebf34a8SDamien Le Moal 		return BLK_STS_IOERR;
532eebf34a8SDamien Le Moal 
533eebf34a8SDamien Le Moal 	null_lock_zone_res(dev);
534eebf34a8SDamien Le Moal 
535eebf34a8SDamien Le Moal 	switch (zone->cond) {
536eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_FULL:
537eebf34a8SDamien Le Moal 		/* finish operation on full is not an error */
538eebf34a8SDamien Le Moal 		goto unlock;
539eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EMPTY:
540eebf34a8SDamien Le Moal 		ret = null_check_zone_resources(dev, zone);
541eebf34a8SDamien Le Moal 		if (ret != BLK_STS_OK)
542eebf34a8SDamien Le Moal 			goto unlock;
543eebf34a8SDamien Le Moal 		break;
544eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_IMP_OPEN:
545eebf34a8SDamien Le Moal 		dev->nr_zones_imp_open--;
546eebf34a8SDamien Le Moal 		break;
547eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EXP_OPEN:
548eebf34a8SDamien Le Moal 		dev->nr_zones_exp_open--;
549eebf34a8SDamien Le Moal 		break;
550eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_CLOSED:
551eebf34a8SDamien Le Moal 		ret = null_check_zone_resources(dev, zone);
552eebf34a8SDamien Le Moal 		if (ret != BLK_STS_OK)
553eebf34a8SDamien Le Moal 			goto unlock;
554eebf34a8SDamien Le Moal 		dev->nr_zones_closed--;
555eebf34a8SDamien Le Moal 		break;
556eebf34a8SDamien Le Moal 	default:
557eebf34a8SDamien Le Moal 		ret = BLK_STS_IOERR;
558eebf34a8SDamien Le Moal 		goto unlock;
559eebf34a8SDamien Le Moal 	}
560eebf34a8SDamien Le Moal 
561eebf34a8SDamien Le Moal 	zone->cond = BLK_ZONE_COND_FULL;
562eebf34a8SDamien Le Moal 	zone->wp = zone->start + zone->len;
563eebf34a8SDamien Le Moal 
564eebf34a8SDamien Le Moal unlock:
565eebf34a8SDamien Le Moal 	null_unlock_zone_res(dev);
566eebf34a8SDamien Le Moal 
567eebf34a8SDamien Le Moal 	return ret;
568eebf34a8SDamien Le Moal }
569eebf34a8SDamien Le Moal 
null_reset_zone(struct nullb_device * dev,struct nullb_zone * zone)570eebf34a8SDamien Le Moal static blk_status_t null_reset_zone(struct nullb_device *dev,
571eebf34a8SDamien Le Moal 				    struct nullb_zone *zone)
572eebf34a8SDamien Le Moal {
573eebf34a8SDamien Le Moal 	if (zone->type == BLK_ZONE_TYPE_CONVENTIONAL)
574eebf34a8SDamien Le Moal 		return BLK_STS_IOERR;
575eebf34a8SDamien Le Moal 
576eebf34a8SDamien Le Moal 	null_lock_zone_res(dev);
577eebf34a8SDamien Le Moal 
578eebf34a8SDamien Le Moal 	switch (zone->cond) {
579eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EMPTY:
580eebf34a8SDamien Le Moal 		/* reset operation on empty is not an error */
581eebf34a8SDamien Le Moal 		null_unlock_zone_res(dev);
582eebf34a8SDamien Le Moal 		return BLK_STS_OK;
583eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_IMP_OPEN:
584eebf34a8SDamien Le Moal 		dev->nr_zones_imp_open--;
585eebf34a8SDamien Le Moal 		break;
586eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_EXP_OPEN:
587eebf34a8SDamien Le Moal 		dev->nr_zones_exp_open--;
588eebf34a8SDamien Le Moal 		break;
589eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_CLOSED:
590eebf34a8SDamien Le Moal 		dev->nr_zones_closed--;
591eebf34a8SDamien Le Moal 		break;
592eebf34a8SDamien Le Moal 	case BLK_ZONE_COND_FULL:
593eebf34a8SDamien Le Moal 		break;
594eebf34a8SDamien Le Moal 	default:
595eebf34a8SDamien Le Moal 		null_unlock_zone_res(dev);
596eebf34a8SDamien Le Moal 		return BLK_STS_IOERR;
597eebf34a8SDamien Le Moal 	}
598eebf34a8SDamien Le Moal 
599eebf34a8SDamien Le Moal 	zone->cond = BLK_ZONE_COND_EMPTY;
600eebf34a8SDamien Le Moal 	zone->wp = zone->start;
601eebf34a8SDamien Le Moal 
602eebf34a8SDamien Le Moal 	null_unlock_zone_res(dev);
603eebf34a8SDamien Le Moal 
604eebf34a8SDamien Le Moal 	if (dev->memory_backed)
605eebf34a8SDamien Le Moal 		return null_handle_discard(dev, zone->start, zone->len);
606eebf34a8SDamien Le Moal 
607eebf34a8SDamien Le Moal 	return BLK_STS_OK;
608eebf34a8SDamien Le Moal }
609eebf34a8SDamien Le Moal 
null_zone_mgmt(struct nullb_cmd * cmd,enum req_op op,sector_t sector)610ff07a02eSBart Van Assche static blk_status_t null_zone_mgmt(struct nullb_cmd *cmd, enum req_op op,
611eebf34a8SDamien Le Moal 				   sector_t sector)
612eebf34a8SDamien Le Moal {
613eebf34a8SDamien Le Moal 	struct nullb_device *dev = cmd->nq->dev;
614eebf34a8SDamien Le Moal 	unsigned int zone_no;
615eebf34a8SDamien Le Moal 	struct nullb_zone *zone;
616eebf34a8SDamien Le Moal 	blk_status_t ret;
617eebf34a8SDamien Le Moal 	size_t i;
618eebf34a8SDamien Le Moal 
619eebf34a8SDamien Le Moal 	if (op == REQ_OP_ZONE_RESET_ALL) {
620eebf34a8SDamien Le Moal 		for (i = dev->zone_nr_conv; i < dev->nr_zones; i++) {
621eebf34a8SDamien Le Moal 			zone = &dev->zones[i];
622eebf34a8SDamien Le Moal 			null_lock_zone(dev, zone);
623d3a57388SShin'ichiro Kawasaki 			if (zone->cond != BLK_ZONE_COND_EMPTY &&
624d3a57388SShin'ichiro Kawasaki 			    zone->cond != BLK_ZONE_COND_READONLY &&
625d3a57388SShin'ichiro Kawasaki 			    zone->cond != BLK_ZONE_COND_OFFLINE) {
626eebf34a8SDamien Le Moal 				null_reset_zone(dev, zone);
627eebf34a8SDamien Le Moal 				trace_nullb_zone_op(cmd, i, zone->cond);
628eebf34a8SDamien Le Moal 			}
629eebf34a8SDamien Le Moal 			null_unlock_zone(dev, zone);
630eebf34a8SDamien Le Moal 		}
631eebf34a8SDamien Le Moal 		return BLK_STS_OK;
632eebf34a8SDamien Le Moal 	}
633eebf34a8SDamien Le Moal 
634eebf34a8SDamien Le Moal 	zone_no = null_zone_no(dev, sector);
635eebf34a8SDamien Le Moal 	zone = &dev->zones[zone_no];
636eebf34a8SDamien Le Moal 
637eebf34a8SDamien Le Moal 	null_lock_zone(dev, zone);
638eebf34a8SDamien Le Moal 
639d3a57388SShin'ichiro Kawasaki 	if (zone->cond == BLK_ZONE_COND_READONLY ||
640d3a57388SShin'ichiro Kawasaki 	    zone->cond == BLK_ZONE_COND_OFFLINE) {
641d3a57388SShin'ichiro Kawasaki 		ret = BLK_STS_IOERR;
642d3a57388SShin'ichiro Kawasaki 		goto unlock;
643d3a57388SShin'ichiro Kawasaki 	}
644d3a57388SShin'ichiro Kawasaki 
645eebf34a8SDamien Le Moal 	switch (op) {
646eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_RESET:
647eebf34a8SDamien Le Moal 		ret = null_reset_zone(dev, zone);
648eebf34a8SDamien Le Moal 		break;
649eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_OPEN:
650eebf34a8SDamien Le Moal 		ret = null_open_zone(dev, zone);
651eebf34a8SDamien Le Moal 		break;
652eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_CLOSE:
653eebf34a8SDamien Le Moal 		ret = null_close_zone(dev, zone);
654eebf34a8SDamien Le Moal 		break;
655eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_FINISH:
656eebf34a8SDamien Le Moal 		ret = null_finish_zone(dev, zone);
657eebf34a8SDamien Le Moal 		break;
658eebf34a8SDamien Le Moal 	default:
659eebf34a8SDamien Le Moal 		ret = BLK_STS_NOTSUPP;
660eebf34a8SDamien Le Moal 		break;
661eebf34a8SDamien Le Moal 	}
662eebf34a8SDamien Le Moal 
663eebf34a8SDamien Le Moal 	if (ret == BLK_STS_OK)
664eebf34a8SDamien Le Moal 		trace_nullb_zone_op(cmd, zone_no, zone->cond);
665eebf34a8SDamien Le Moal 
666d3a57388SShin'ichiro Kawasaki unlock:
667eebf34a8SDamien Le Moal 	null_unlock_zone(dev, zone);
668eebf34a8SDamien Le Moal 
669eebf34a8SDamien Le Moal 	return ret;
670eebf34a8SDamien Le Moal }
671eebf34a8SDamien Le Moal 
null_process_zoned_cmd(struct nullb_cmd * cmd,enum req_op op,sector_t sector,sector_t nr_sectors)672ff07a02eSBart Van Assche blk_status_t null_process_zoned_cmd(struct nullb_cmd *cmd, enum req_op op,
673eebf34a8SDamien Le Moal 				    sector_t sector, sector_t nr_sectors)
674eebf34a8SDamien Le Moal {
675eebf34a8SDamien Le Moal 	struct nullb_device *dev;
676eebf34a8SDamien Le Moal 	struct nullb_zone *zone;
677eebf34a8SDamien Le Moal 	blk_status_t sts;
678eebf34a8SDamien Le Moal 
679eebf34a8SDamien Le Moal 	switch (op) {
680eebf34a8SDamien Le Moal 	case REQ_OP_WRITE:
681eebf34a8SDamien Le Moal 		return null_zone_write(cmd, sector, nr_sectors, false);
682eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_APPEND:
683eebf34a8SDamien Le Moal 		return null_zone_write(cmd, sector, nr_sectors, true);
684eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_RESET:
685eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_RESET_ALL:
686eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_OPEN:
687eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_CLOSE:
688eebf34a8SDamien Le Moal 	case REQ_OP_ZONE_FINISH:
689eebf34a8SDamien Le Moal 		return null_zone_mgmt(cmd, op, sector);
690eebf34a8SDamien Le Moal 	default:
691eebf34a8SDamien Le Moal 		dev = cmd->nq->dev;
692eebf34a8SDamien Le Moal 		zone = &dev->zones[null_zone_no(dev, sector)];
693d3a57388SShin'ichiro Kawasaki 		if (zone->cond == BLK_ZONE_COND_OFFLINE)
694d3a57388SShin'ichiro Kawasaki 			return BLK_STS_IOERR;
695eebf34a8SDamien Le Moal 
696eebf34a8SDamien Le Moal 		null_lock_zone(dev, zone);
697eebf34a8SDamien Le Moal 		sts = null_process_cmd(cmd, op, sector, nr_sectors);
698eebf34a8SDamien Le Moal 		null_unlock_zone(dev, zone);
699eebf34a8SDamien Le Moal 		return sts;
700eebf34a8SDamien Le Moal 	}
701eebf34a8SDamien Le Moal }
702d3a57388SShin'ichiro Kawasaki 
703d3a57388SShin'ichiro Kawasaki /*
704d3a57388SShin'ichiro Kawasaki  * Set a zone in the read-only or offline condition.
705d3a57388SShin'ichiro Kawasaki  */
null_set_zone_cond(struct nullb_device * dev,struct nullb_zone * zone,enum blk_zone_cond cond)706d3a57388SShin'ichiro Kawasaki static void null_set_zone_cond(struct nullb_device *dev,
707d3a57388SShin'ichiro Kawasaki 			       struct nullb_zone *zone, enum blk_zone_cond cond)
708d3a57388SShin'ichiro Kawasaki {
709d3a57388SShin'ichiro Kawasaki 	if (WARN_ON_ONCE(cond != BLK_ZONE_COND_READONLY &&
710d3a57388SShin'ichiro Kawasaki 			 cond != BLK_ZONE_COND_OFFLINE))
711d3a57388SShin'ichiro Kawasaki 		return;
712d3a57388SShin'ichiro Kawasaki 
713d3a57388SShin'ichiro Kawasaki 	null_lock_zone(dev, zone);
714d3a57388SShin'ichiro Kawasaki 
715d3a57388SShin'ichiro Kawasaki 	/*
716d3a57388SShin'ichiro Kawasaki 	 * If the read-only condition is requested again to zones already in
717d3a57388SShin'ichiro Kawasaki 	 * read-only condition, restore back normal empty condition. Do the same
718d3a57388SShin'ichiro Kawasaki 	 * if the offline condition is requested for offline zones. Otherwise,
719d3a57388SShin'ichiro Kawasaki 	 * set the specified zone condition to the zones. Finish the zones
720d3a57388SShin'ichiro Kawasaki 	 * beforehand to free up zone resources.
721d3a57388SShin'ichiro Kawasaki 	 */
722d3a57388SShin'ichiro Kawasaki 	if (zone->cond == cond) {
723d3a57388SShin'ichiro Kawasaki 		zone->cond = BLK_ZONE_COND_EMPTY;
724d3a57388SShin'ichiro Kawasaki 		zone->wp = zone->start;
725d3a57388SShin'ichiro Kawasaki 		if (dev->memory_backed)
726d3a57388SShin'ichiro Kawasaki 			null_handle_discard(dev, zone->start, zone->len);
727d3a57388SShin'ichiro Kawasaki 	} else {
728d3a57388SShin'ichiro Kawasaki 		if (zone->cond != BLK_ZONE_COND_READONLY &&
729d3a57388SShin'ichiro Kawasaki 		    zone->cond != BLK_ZONE_COND_OFFLINE)
730d3a57388SShin'ichiro Kawasaki 			null_finish_zone(dev, zone);
731d3a57388SShin'ichiro Kawasaki 		zone->cond = cond;
732d3a57388SShin'ichiro Kawasaki 		zone->wp = (sector_t)-1;
733d3a57388SShin'ichiro Kawasaki 	}
734d3a57388SShin'ichiro Kawasaki 
735d3a57388SShin'ichiro Kawasaki 	null_unlock_zone(dev, zone);
736d3a57388SShin'ichiro Kawasaki }
737d3a57388SShin'ichiro Kawasaki 
738d3a57388SShin'ichiro Kawasaki /*
739d3a57388SShin'ichiro Kawasaki  * Identify a zone from the sector written to configfs file. Then set zone
740d3a57388SShin'ichiro Kawasaki  * condition to the zone.
741d3a57388SShin'ichiro Kawasaki  */
zone_cond_store(struct nullb_device * dev,const char * page,size_t count,enum blk_zone_cond cond)742d3a57388SShin'ichiro Kawasaki ssize_t zone_cond_store(struct nullb_device *dev, const char *page,
743d3a57388SShin'ichiro Kawasaki 			size_t count, enum blk_zone_cond cond)
744d3a57388SShin'ichiro Kawasaki {
745d3a57388SShin'ichiro Kawasaki 	unsigned long long sector;
746d3a57388SShin'ichiro Kawasaki 	unsigned int zone_no;
747d3a57388SShin'ichiro Kawasaki 	int ret;
748d3a57388SShin'ichiro Kawasaki 
749d3a57388SShin'ichiro Kawasaki 	if (!dev->zoned) {
750d3a57388SShin'ichiro Kawasaki 		pr_err("null_blk device is not zoned\n");
751d3a57388SShin'ichiro Kawasaki 		return -EINVAL;
752d3a57388SShin'ichiro Kawasaki 	}
753d3a57388SShin'ichiro Kawasaki 
754d3a57388SShin'ichiro Kawasaki 	if (!dev->zones) {
755d3a57388SShin'ichiro Kawasaki 		pr_err("null_blk device is not yet powered\n");
756d3a57388SShin'ichiro Kawasaki 		return -EINVAL;
757d3a57388SShin'ichiro Kawasaki 	}
758d3a57388SShin'ichiro Kawasaki 
759d3a57388SShin'ichiro Kawasaki 	ret = kstrtoull(page, 0, &sector);
760d3a57388SShin'ichiro Kawasaki 	if (ret < 0)
761d3a57388SShin'ichiro Kawasaki 		return ret;
762d3a57388SShin'ichiro Kawasaki 
763d3a57388SShin'ichiro Kawasaki 	zone_no = null_zone_no(dev, sector);
764d3a57388SShin'ichiro Kawasaki 	if (zone_no >= dev->nr_zones) {
765d3a57388SShin'ichiro Kawasaki 		pr_err("Sector out of range\n");
766d3a57388SShin'ichiro Kawasaki 		return -EINVAL;
767d3a57388SShin'ichiro Kawasaki 	}
768d3a57388SShin'ichiro Kawasaki 
769d3a57388SShin'ichiro Kawasaki 	if (dev->zones[zone_no].type == BLK_ZONE_TYPE_CONVENTIONAL) {
770d3a57388SShin'ichiro Kawasaki 		pr_err("Can not change condition of conventional zones\n");
771d3a57388SShin'ichiro Kawasaki 		return -EINVAL;
772d3a57388SShin'ichiro Kawasaki 	}
773d3a57388SShin'ichiro Kawasaki 
774d3a57388SShin'ichiro Kawasaki 	null_set_zone_cond(dev, &dev->zones[zone_no], cond);
775d3a57388SShin'ichiro Kawasaki 
776d3a57388SShin'ichiro Kawasaki 	return count;
777d3a57388SShin'ichiro Kawasaki }
778