1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*  Xenbus code for blkif backend
3     Copyright (C) 2005 Rusty Russell <rusty@rustcorp.com.au>
4     Copyright (C) 2005 XenSource Ltd
5 
6 
7 */
8 
9 #define pr_fmt(fmt) "xen-blkback: " fmt
10 
11 #include <stdarg.h>
12 #include <linux/module.h>
13 #include <linux/kthread.h>
14 #include <xen/events.h>
15 #include <xen/grant_table.h>
16 #include "common.h"
17 
18 /* On the XenBus the max length of 'ring-ref%u'. */
19 #define RINGREF_NAME_LEN (20)
20 
21 struct backend_info {
22 	struct xenbus_device	*dev;
23 	struct xen_blkif	*blkif;
24 	struct xenbus_watch	backend_watch;
25 	unsigned		major;
26 	unsigned		minor;
27 	char			*mode;
28 };
29 
30 static struct kmem_cache *xen_blkif_cachep;
31 static void connect(struct backend_info *);
32 static int connect_ring(struct backend_info *);
33 static void backend_changed(struct xenbus_watch *, const char *,
34 			    const char *);
35 static void xen_blkif_free(struct xen_blkif *blkif);
36 static void xen_vbd_free(struct xen_vbd *vbd);
37 
xen_blkbk_xenbus(struct backend_info * be)38 struct xenbus_device *xen_blkbk_xenbus(struct backend_info *be)
39 {
40 	return be->dev;
41 }
42 
43 /*
44  * The last request could free the device from softirq context and
45  * xen_blkif_free() can sleep.
46  */
xen_blkif_deferred_free(struct work_struct * work)47 static void xen_blkif_deferred_free(struct work_struct *work)
48 {
49 	struct xen_blkif *blkif;
50 
51 	blkif = container_of(work, struct xen_blkif, free_work);
52 	xen_blkif_free(blkif);
53 }
54 
blkback_name(struct xen_blkif * blkif,char * buf)55 static int blkback_name(struct xen_blkif *blkif, char *buf)
56 {
57 	char *devpath, *devname;
58 	struct xenbus_device *dev = blkif->be->dev;
59 
60 	devpath = xenbus_read(XBT_NIL, dev->nodename, "dev", NULL);
61 	if (IS_ERR(devpath))
62 		return PTR_ERR(devpath);
63 
64 	devname = strstr(devpath, "/dev/");
65 	if (devname != NULL)
66 		devname += strlen("/dev/");
67 	else
68 		devname  = devpath;
69 
70 	snprintf(buf, TASK_COMM_LEN, "%d.%s", blkif->domid, devname);
71 	kfree(devpath);
72 
73 	return 0;
74 }
75 
xen_update_blkif_status(struct xen_blkif * blkif)76 static void xen_update_blkif_status(struct xen_blkif *blkif)
77 {
78 	int err;
79 	char name[TASK_COMM_LEN];
80 	struct xen_blkif_ring *ring;
81 	int i;
82 
83 	/* Not ready to connect? */
84 	if (!blkif->rings || !blkif->rings[0].irq || !blkif->vbd.bdev)
85 		return;
86 
87 	/* Already connected? */
88 	if (blkif->be->dev->state == XenbusStateConnected)
89 		return;
90 
91 	/* Attempt to connect: exit if we fail to. */
92 	connect(blkif->be);
93 	if (blkif->be->dev->state != XenbusStateConnected)
94 		return;
95 
96 	err = blkback_name(blkif, name);
97 	if (err) {
98 		xenbus_dev_error(blkif->be->dev, err, "get blkback dev name");
99 		return;
100 	}
101 
102 	err = filemap_write_and_wait(blkif->vbd.bdev->bd_inode->i_mapping);
103 	if (err) {
104 		xenbus_dev_error(blkif->be->dev, err, "block flush");
105 		return;
106 	}
107 	invalidate_inode_pages2(blkif->vbd.bdev->bd_inode->i_mapping);
108 
109 	for (i = 0; i < blkif->nr_rings; i++) {
110 		ring = &blkif->rings[i];
111 		ring->xenblkd = kthread_run(xen_blkif_schedule, ring, "%s-%d", name, i);
112 		if (IS_ERR(ring->xenblkd)) {
113 			err = PTR_ERR(ring->xenblkd);
114 			ring->xenblkd = NULL;
115 			xenbus_dev_fatal(blkif->be->dev, err,
116 					"start %s-%d xenblkd", name, i);
117 			goto out;
118 		}
119 	}
120 	return;
121 
122 out:
123 	while (--i >= 0) {
124 		ring = &blkif->rings[i];
125 		kthread_stop(ring->xenblkd);
126 	}
127 	return;
128 }
129 
xen_blkif_alloc_rings(struct xen_blkif * blkif)130 static int xen_blkif_alloc_rings(struct xen_blkif *blkif)
131 {
132 	unsigned int r;
133 
134 	blkif->rings = kcalloc(blkif->nr_rings, sizeof(struct xen_blkif_ring),
135 			       GFP_KERNEL);
136 	if (!blkif->rings)
137 		return -ENOMEM;
138 
139 	for (r = 0; r < blkif->nr_rings; r++) {
140 		struct xen_blkif_ring *ring = &blkif->rings[r];
141 
142 		spin_lock_init(&ring->blk_ring_lock);
143 		init_waitqueue_head(&ring->wq);
144 		INIT_LIST_HEAD(&ring->pending_free);
145 		INIT_LIST_HEAD(&ring->persistent_purge_list);
146 		INIT_WORK(&ring->persistent_purge_work, xen_blkbk_unmap_purged_grants);
147 		spin_lock_init(&ring->free_pages_lock);
148 		INIT_LIST_HEAD(&ring->free_pages);
149 
150 		spin_lock_init(&ring->pending_free_lock);
151 		init_waitqueue_head(&ring->pending_free_wq);
152 		init_waitqueue_head(&ring->shutdown_wq);
153 		ring->blkif = blkif;
154 		ring->st_print = jiffies;
155 		ring->active = true;
156 	}
157 
158 	return 0;
159 }
160 
xen_blkif_alloc(domid_t domid)161 static struct xen_blkif *xen_blkif_alloc(domid_t domid)
162 {
163 	struct xen_blkif *blkif;
164 
165 	BUILD_BUG_ON(MAX_INDIRECT_PAGES > BLKIF_MAX_INDIRECT_PAGES_PER_REQUEST);
166 
167 	blkif = kmem_cache_zalloc(xen_blkif_cachep, GFP_KERNEL);
168 	if (!blkif)
169 		return ERR_PTR(-ENOMEM);
170 
171 	blkif->domid = domid;
172 	atomic_set(&blkif->refcnt, 1);
173 	init_completion(&blkif->drain_complete);
174 	INIT_WORK(&blkif->free_work, xen_blkif_deferred_free);
175 
176 	return blkif;
177 }
178 
xen_blkif_map(struct xen_blkif_ring * ring,grant_ref_t * gref,unsigned int nr_grefs,unsigned int evtchn)179 static int xen_blkif_map(struct xen_blkif_ring *ring, grant_ref_t *gref,
180 			 unsigned int nr_grefs, unsigned int evtchn)
181 {
182 	int err;
183 	struct xen_blkif *blkif = ring->blkif;
184 
185 	/* Already connected through? */
186 	if (ring->irq)
187 		return 0;
188 
189 	err = xenbus_map_ring_valloc(blkif->be->dev, gref, nr_grefs,
190 				     &ring->blk_ring);
191 	if (err < 0)
192 		return err;
193 
194 	switch (blkif->blk_protocol) {
195 	case BLKIF_PROTOCOL_NATIVE:
196 	{
197 		struct blkif_sring *sring;
198 		sring = (struct blkif_sring *)ring->blk_ring;
199 		BACK_RING_INIT(&ring->blk_rings.native, sring,
200 			       XEN_PAGE_SIZE * nr_grefs);
201 		break;
202 	}
203 	case BLKIF_PROTOCOL_X86_32:
204 	{
205 		struct blkif_x86_32_sring *sring_x86_32;
206 		sring_x86_32 = (struct blkif_x86_32_sring *)ring->blk_ring;
207 		BACK_RING_INIT(&ring->blk_rings.x86_32, sring_x86_32,
208 			       XEN_PAGE_SIZE * nr_grefs);
209 		break;
210 	}
211 	case BLKIF_PROTOCOL_X86_64:
212 	{
213 		struct blkif_x86_64_sring *sring_x86_64;
214 		sring_x86_64 = (struct blkif_x86_64_sring *)ring->blk_ring;
215 		BACK_RING_INIT(&ring->blk_rings.x86_64, sring_x86_64,
216 			       XEN_PAGE_SIZE * nr_grefs);
217 		break;
218 	}
219 	default:
220 		BUG();
221 	}
222 
223 	err = bind_interdomain_evtchn_to_irqhandler(blkif->domid, evtchn,
224 						    xen_blkif_be_int, 0,
225 						    "blkif-backend", ring);
226 	if (err < 0) {
227 		xenbus_unmap_ring_vfree(blkif->be->dev, ring->blk_ring);
228 		ring->blk_rings.common.sring = NULL;
229 		return err;
230 	}
231 	ring->irq = err;
232 
233 	return 0;
234 }
235 
xen_blkif_disconnect(struct xen_blkif * blkif)236 static int xen_blkif_disconnect(struct xen_blkif *blkif)
237 {
238 	struct pending_req *req, *n;
239 	unsigned int j, r;
240 	bool busy = false;
241 
242 	for (r = 0; r < blkif->nr_rings; r++) {
243 		struct xen_blkif_ring *ring = &blkif->rings[r];
244 		unsigned int i = 0;
245 
246 		if (!ring->active)
247 			continue;
248 
249 		if (ring->xenblkd) {
250 			kthread_stop(ring->xenblkd);
251 			wake_up(&ring->shutdown_wq);
252 		}
253 
254 		/* The above kthread_stop() guarantees that at this point we
255 		 * don't have any discard_io or other_io requests. So, checking
256 		 * for inflight IO is enough.
257 		 */
258 		if (atomic_read(&ring->inflight) > 0) {
259 			busy = true;
260 			continue;
261 		}
262 
263 		if (ring->irq) {
264 			unbind_from_irqhandler(ring->irq, ring);
265 			ring->irq = 0;
266 		}
267 
268 		if (ring->blk_rings.common.sring) {
269 			xenbus_unmap_ring_vfree(blkif->be->dev, ring->blk_ring);
270 			ring->blk_rings.common.sring = NULL;
271 		}
272 
273 		/* Remove all persistent grants and the cache of ballooned pages. */
274 		xen_blkbk_free_caches(ring);
275 
276 		/* Check that there is no request in use */
277 		list_for_each_entry_safe(req, n, &ring->pending_free, free_list) {
278 			list_del(&req->free_list);
279 
280 			for (j = 0; j < MAX_INDIRECT_SEGMENTS; j++)
281 				kfree(req->segments[j]);
282 
283 			for (j = 0; j < MAX_INDIRECT_PAGES; j++)
284 				kfree(req->indirect_pages[j]);
285 
286 			kfree(req);
287 			i++;
288 		}
289 
290 		BUG_ON(atomic_read(&ring->persistent_gnt_in_use) != 0);
291 		BUG_ON(!list_empty(&ring->persistent_purge_list));
292 		BUG_ON(!RB_EMPTY_ROOT(&ring->persistent_gnts));
293 		BUG_ON(!list_empty(&ring->free_pages));
294 		BUG_ON(ring->free_pages_num != 0);
295 		BUG_ON(ring->persistent_gnt_c != 0);
296 		WARN_ON(i != (XEN_BLKIF_REQS_PER_PAGE * blkif->nr_ring_pages));
297 		ring->active = false;
298 	}
299 	if (busy)
300 		return -EBUSY;
301 
302 	blkif->nr_ring_pages = 0;
303 	/*
304 	 * blkif->rings was allocated in connect_ring, so we should free it in
305 	 * here.
306 	 */
307 	kfree(blkif->rings);
308 	blkif->rings = NULL;
309 	blkif->nr_rings = 0;
310 
311 	return 0;
312 }
313 
xen_blkif_free(struct xen_blkif * blkif)314 static void xen_blkif_free(struct xen_blkif *blkif)
315 {
316 	WARN_ON(xen_blkif_disconnect(blkif));
317 	xen_vbd_free(&blkif->vbd);
318 	kfree(blkif->be->mode);
319 	kfree(blkif->be);
320 
321 	/* Make sure everything is drained before shutting down */
322 	kmem_cache_free(xen_blkif_cachep, blkif);
323 }
324 
xen_blkif_interface_init(void)325 int __init xen_blkif_interface_init(void)
326 {
327 	xen_blkif_cachep = kmem_cache_create("blkif_cache",
328 					     sizeof(struct xen_blkif),
329 					     0, 0, NULL);
330 	if (!xen_blkif_cachep)
331 		return -ENOMEM;
332 
333 	return 0;
334 }
335 
336 /*
337  *  sysfs interface for VBD I/O requests
338  */
339 
340 #define VBD_SHOW_ALLRING(name, format)					\
341 	static ssize_t show_##name(struct device *_dev,			\
342 				   struct device_attribute *attr,	\
343 				   char *buf)				\
344 	{								\
345 		struct xenbus_device *dev = to_xenbus_device(_dev);	\
346 		struct backend_info *be = dev_get_drvdata(&dev->dev);	\
347 		struct xen_blkif *blkif = be->blkif;			\
348 		unsigned int i;						\
349 		unsigned long long result = 0;				\
350 									\
351 		if (!blkif->rings)				\
352 			goto out;					\
353 									\
354 		for (i = 0; i < blkif->nr_rings; i++) {		\
355 			struct xen_blkif_ring *ring = &blkif->rings[i];	\
356 									\
357 			result += ring->st_##name;			\
358 		}							\
359 									\
360 out:									\
361 		return sprintf(buf, format, result);			\
362 	}								\
363 	static DEVICE_ATTR(name, 0444, show_##name, NULL)
364 
365 VBD_SHOW_ALLRING(oo_req,  "%llu\n");
366 VBD_SHOW_ALLRING(rd_req,  "%llu\n");
367 VBD_SHOW_ALLRING(wr_req,  "%llu\n");
368 VBD_SHOW_ALLRING(f_req,  "%llu\n");
369 VBD_SHOW_ALLRING(ds_req,  "%llu\n");
370 VBD_SHOW_ALLRING(rd_sect, "%llu\n");
371 VBD_SHOW_ALLRING(wr_sect, "%llu\n");
372 
373 static struct attribute *xen_vbdstat_attrs[] = {
374 	&dev_attr_oo_req.attr,
375 	&dev_attr_rd_req.attr,
376 	&dev_attr_wr_req.attr,
377 	&dev_attr_f_req.attr,
378 	&dev_attr_ds_req.attr,
379 	&dev_attr_rd_sect.attr,
380 	&dev_attr_wr_sect.attr,
381 	NULL
382 };
383 
384 static const struct attribute_group xen_vbdstat_group = {
385 	.name = "statistics",
386 	.attrs = xen_vbdstat_attrs,
387 };
388 
389 #define VBD_SHOW(name, format, args...)					\
390 	static ssize_t show_##name(struct device *_dev,			\
391 				   struct device_attribute *attr,	\
392 				   char *buf)				\
393 	{								\
394 		struct xenbus_device *dev = to_xenbus_device(_dev);	\
395 		struct backend_info *be = dev_get_drvdata(&dev->dev);	\
396 									\
397 		return sprintf(buf, format, ##args);			\
398 	}								\
399 	static DEVICE_ATTR(name, 0444, show_##name, NULL)
400 
401 VBD_SHOW(physical_device, "%x:%x\n", be->major, be->minor);
402 VBD_SHOW(mode, "%s\n", be->mode);
403 
xenvbd_sysfs_addif(struct xenbus_device * dev)404 static int xenvbd_sysfs_addif(struct xenbus_device *dev)
405 {
406 	int error;
407 
408 	error = device_create_file(&dev->dev, &dev_attr_physical_device);
409 	if (error)
410 		goto fail1;
411 
412 	error = device_create_file(&dev->dev, &dev_attr_mode);
413 	if (error)
414 		goto fail2;
415 
416 	error = sysfs_create_group(&dev->dev.kobj, &xen_vbdstat_group);
417 	if (error)
418 		goto fail3;
419 
420 	return 0;
421 
422 fail3:	sysfs_remove_group(&dev->dev.kobj, &xen_vbdstat_group);
423 fail2:	device_remove_file(&dev->dev, &dev_attr_mode);
424 fail1:	device_remove_file(&dev->dev, &dev_attr_physical_device);
425 	return error;
426 }
427 
xenvbd_sysfs_delif(struct xenbus_device * dev)428 static void xenvbd_sysfs_delif(struct xenbus_device *dev)
429 {
430 	sysfs_remove_group(&dev->dev.kobj, &xen_vbdstat_group);
431 	device_remove_file(&dev->dev, &dev_attr_mode);
432 	device_remove_file(&dev->dev, &dev_attr_physical_device);
433 }
434 
435 
xen_vbd_free(struct xen_vbd * vbd)436 static void xen_vbd_free(struct xen_vbd *vbd)
437 {
438 	if (vbd->bdev)
439 		blkdev_put(vbd->bdev, vbd->readonly ? FMODE_READ : FMODE_WRITE);
440 	vbd->bdev = NULL;
441 }
442 
xen_vbd_create(struct xen_blkif * blkif,blkif_vdev_t handle,unsigned major,unsigned minor,int readonly,int cdrom)443 static int xen_vbd_create(struct xen_blkif *blkif, blkif_vdev_t handle,
444 			  unsigned major, unsigned minor, int readonly,
445 			  int cdrom)
446 {
447 	struct xen_vbd *vbd;
448 	struct block_device *bdev;
449 	struct request_queue *q;
450 
451 	vbd = &blkif->vbd;
452 	vbd->handle   = handle;
453 	vbd->readonly = readonly;
454 	vbd->type     = 0;
455 
456 	vbd->pdevice  = MKDEV(major, minor);
457 
458 	bdev = blkdev_get_by_dev(vbd->pdevice, vbd->readonly ?
459 				 FMODE_READ : FMODE_WRITE, NULL);
460 
461 	if (IS_ERR(bdev)) {
462 		pr_warn("xen_vbd_create: device %08x could not be opened\n",
463 			vbd->pdevice);
464 		return -ENOENT;
465 	}
466 
467 	vbd->bdev = bdev;
468 	if (vbd->bdev->bd_disk == NULL) {
469 		pr_warn("xen_vbd_create: device %08x doesn't exist\n",
470 			vbd->pdevice);
471 		xen_vbd_free(vbd);
472 		return -ENOENT;
473 	}
474 	vbd->size = vbd_sz(vbd);
475 
476 	if (vbd->bdev->bd_disk->flags & GENHD_FL_CD || cdrom)
477 		vbd->type |= VDISK_CDROM;
478 	if (vbd->bdev->bd_disk->flags & GENHD_FL_REMOVABLE)
479 		vbd->type |= VDISK_REMOVABLE;
480 
481 	q = bdev_get_queue(bdev);
482 	if (q && test_bit(QUEUE_FLAG_WC, &q->queue_flags))
483 		vbd->flush_support = true;
484 
485 	if (q && blk_queue_secure_erase(q))
486 		vbd->discard_secure = true;
487 
488 	pr_debug("Successful creation of handle=%04x (dom=%u)\n",
489 		handle, blkif->domid);
490 	return 0;
491 }
xen_blkbk_remove(struct xenbus_device * dev)492 static int xen_blkbk_remove(struct xenbus_device *dev)
493 {
494 	struct backend_info *be = dev_get_drvdata(&dev->dev);
495 
496 	pr_debug("%s %p %d\n", __func__, dev, dev->otherend_id);
497 
498 	if (be->major || be->minor)
499 		xenvbd_sysfs_delif(dev);
500 
501 	if (be->backend_watch.node) {
502 		unregister_xenbus_watch(&be->backend_watch);
503 		kfree(be->backend_watch.node);
504 		be->backend_watch.node = NULL;
505 	}
506 
507 	dev_set_drvdata(&dev->dev, NULL);
508 
509 	if (be->blkif) {
510 		xen_blkif_disconnect(be->blkif);
511 
512 		/* Put the reference we set in xen_blkif_alloc(). */
513 		xen_blkif_put(be->blkif);
514 	}
515 
516 	return 0;
517 }
518 
xen_blkbk_flush_diskcache(struct xenbus_transaction xbt,struct backend_info * be,int state)519 int xen_blkbk_flush_diskcache(struct xenbus_transaction xbt,
520 			      struct backend_info *be, int state)
521 {
522 	struct xenbus_device *dev = be->dev;
523 	int err;
524 
525 	err = xenbus_printf(xbt, dev->nodename, "feature-flush-cache",
526 			    "%d", state);
527 	if (err)
528 		dev_warn(&dev->dev, "writing feature-flush-cache (%d)", err);
529 
530 	return err;
531 }
532 
xen_blkbk_discard(struct xenbus_transaction xbt,struct backend_info * be)533 static void xen_blkbk_discard(struct xenbus_transaction xbt, struct backend_info *be)
534 {
535 	struct xenbus_device *dev = be->dev;
536 	struct xen_blkif *blkif = be->blkif;
537 	int err;
538 	int state = 0;
539 	struct block_device *bdev = be->blkif->vbd.bdev;
540 	struct request_queue *q = bdev_get_queue(bdev);
541 
542 	if (!xenbus_read_unsigned(dev->nodename, "discard-enable", 1))
543 		return;
544 
545 	if (blk_queue_discard(q)) {
546 		err = xenbus_printf(xbt, dev->nodename,
547 			"discard-granularity", "%u",
548 			q->limits.discard_granularity);
549 		if (err) {
550 			dev_warn(&dev->dev, "writing discard-granularity (%d)", err);
551 			return;
552 		}
553 		err = xenbus_printf(xbt, dev->nodename,
554 			"discard-alignment", "%u",
555 			q->limits.discard_alignment);
556 		if (err) {
557 			dev_warn(&dev->dev, "writing discard-alignment (%d)", err);
558 			return;
559 		}
560 		state = 1;
561 		/* Optional. */
562 		err = xenbus_printf(xbt, dev->nodename,
563 				    "discard-secure", "%d",
564 				    blkif->vbd.discard_secure);
565 		if (err) {
566 			dev_warn(&dev->dev, "writing discard-secure (%d)", err);
567 			return;
568 		}
569 	}
570 	err = xenbus_printf(xbt, dev->nodename, "feature-discard",
571 			    "%d", state);
572 	if (err)
573 		dev_warn(&dev->dev, "writing feature-discard (%d)", err);
574 }
xen_blkbk_barrier(struct xenbus_transaction xbt,struct backend_info * be,int state)575 int xen_blkbk_barrier(struct xenbus_transaction xbt,
576 		      struct backend_info *be, int state)
577 {
578 	struct xenbus_device *dev = be->dev;
579 	int err;
580 
581 	err = xenbus_printf(xbt, dev->nodename, "feature-barrier",
582 			    "%d", state);
583 	if (err)
584 		dev_warn(&dev->dev, "writing feature-barrier (%d)", err);
585 
586 	return err;
587 }
588 
589 /*
590  * Entry point to this code when a new device is created.  Allocate the basic
591  * structures, and watch the store waiting for the hotplug scripts to tell us
592  * the device's physical major and minor numbers.  Switch to InitWait.
593  */
xen_blkbk_probe(struct xenbus_device * dev,const struct xenbus_device_id * id)594 static int xen_blkbk_probe(struct xenbus_device *dev,
595 			   const struct xenbus_device_id *id)
596 {
597 	int err;
598 	struct backend_info *be = kzalloc(sizeof(struct backend_info),
599 					  GFP_KERNEL);
600 
601 	/* match the pr_debug in xen_blkbk_remove */
602 	pr_debug("%s %p %d\n", __func__, dev, dev->otherend_id);
603 
604 	if (!be) {
605 		xenbus_dev_fatal(dev, -ENOMEM,
606 				 "allocating backend structure");
607 		return -ENOMEM;
608 	}
609 	be->dev = dev;
610 	dev_set_drvdata(&dev->dev, be);
611 
612 	be->blkif = xen_blkif_alloc(dev->otherend_id);
613 	if (IS_ERR(be->blkif)) {
614 		err = PTR_ERR(be->blkif);
615 		be->blkif = NULL;
616 		xenbus_dev_fatal(dev, err, "creating block interface");
617 		goto fail;
618 	}
619 
620 	err = xenbus_printf(XBT_NIL, dev->nodename,
621 			    "feature-max-indirect-segments", "%u",
622 			    MAX_INDIRECT_SEGMENTS);
623 	if (err)
624 		dev_warn(&dev->dev,
625 			 "writing %s/feature-max-indirect-segments (%d)",
626 			 dev->nodename, err);
627 
628 	/* Multi-queue: advertise how many queues are supported by us.*/
629 	err = xenbus_printf(XBT_NIL, dev->nodename,
630 			    "multi-queue-max-queues", "%u", xenblk_max_queues);
631 	if (err)
632 		pr_warn("Error writing multi-queue-max-queues\n");
633 
634 	/* setup back pointer */
635 	be->blkif->be = be;
636 
637 	err = xenbus_watch_pathfmt(dev, &be->backend_watch, backend_changed,
638 				   "%s/%s", dev->nodename, "physical-device");
639 	if (err)
640 		goto fail;
641 
642 	err = xenbus_printf(XBT_NIL, dev->nodename, "max-ring-page-order", "%u",
643 			    xen_blkif_max_ring_order);
644 	if (err)
645 		pr_warn("%s write out 'max-ring-page-order' failed\n", __func__);
646 
647 	err = xenbus_switch_state(dev, XenbusStateInitWait);
648 	if (err)
649 		goto fail;
650 
651 	return 0;
652 
653 fail:
654 	pr_warn("%s failed\n", __func__);
655 	xen_blkbk_remove(dev);
656 	return err;
657 }
658 
659 
660 /*
661  * Callback received when the hotplug scripts have placed the physical-device
662  * node.  Read it and the mode node, and create a vbd.  If the frontend is
663  * ready, connect.
664  */
backend_changed(struct xenbus_watch * watch,const char * path,const char * token)665 static void backend_changed(struct xenbus_watch *watch,
666 			    const char *path, const char *token)
667 {
668 	int err;
669 	unsigned major;
670 	unsigned minor;
671 	struct backend_info *be
672 		= container_of(watch, struct backend_info, backend_watch);
673 	struct xenbus_device *dev = be->dev;
674 	int cdrom = 0;
675 	unsigned long handle;
676 	char *device_type;
677 
678 	pr_debug("%s %p %d\n", __func__, dev, dev->otherend_id);
679 
680 	err = xenbus_scanf(XBT_NIL, dev->nodename, "physical-device", "%x:%x",
681 			   &major, &minor);
682 	if (XENBUS_EXIST_ERR(err)) {
683 		/*
684 		 * Since this watch will fire once immediately after it is
685 		 * registered, we expect this.  Ignore it, and wait for the
686 		 * hotplug scripts.
687 		 */
688 		return;
689 	}
690 	if (err != 2) {
691 		xenbus_dev_fatal(dev, err, "reading physical-device");
692 		return;
693 	}
694 
695 	if (be->major | be->minor) {
696 		if (be->major != major || be->minor != minor)
697 			pr_warn("changing physical device (from %x:%x to %x:%x) not supported.\n",
698 				be->major, be->minor, major, minor);
699 		return;
700 	}
701 
702 	be->mode = xenbus_read(XBT_NIL, dev->nodename, "mode", NULL);
703 	if (IS_ERR(be->mode)) {
704 		err = PTR_ERR(be->mode);
705 		be->mode = NULL;
706 		xenbus_dev_fatal(dev, err, "reading mode");
707 		return;
708 	}
709 
710 	device_type = xenbus_read(XBT_NIL, dev->otherend, "device-type", NULL);
711 	if (!IS_ERR(device_type)) {
712 		cdrom = strcmp(device_type, "cdrom") == 0;
713 		kfree(device_type);
714 	}
715 
716 	/* Front end dir is a number, which is used as the handle. */
717 	err = kstrtoul(strrchr(dev->otherend, '/') + 1, 0, &handle);
718 	if (err) {
719 		kfree(be->mode);
720 		be->mode = NULL;
721 		return;
722 	}
723 
724 	be->major = major;
725 	be->minor = minor;
726 
727 	err = xen_vbd_create(be->blkif, handle, major, minor,
728 			     !strchr(be->mode, 'w'), cdrom);
729 
730 	if (err)
731 		xenbus_dev_fatal(dev, err, "creating vbd structure");
732 	else {
733 		err = xenvbd_sysfs_addif(dev);
734 		if (err) {
735 			xen_vbd_free(&be->blkif->vbd);
736 			xenbus_dev_fatal(dev, err, "creating sysfs entries");
737 		}
738 	}
739 
740 	if (err) {
741 		kfree(be->mode);
742 		be->mode = NULL;
743 		be->major = 0;
744 		be->minor = 0;
745 	} else {
746 		/* We're potentially connected now */
747 		xen_update_blkif_status(be->blkif);
748 	}
749 }
750 
751 
752 /*
753  * Callback received when the frontend's state changes.
754  */
frontend_changed(struct xenbus_device * dev,enum xenbus_state frontend_state)755 static void frontend_changed(struct xenbus_device *dev,
756 			     enum xenbus_state frontend_state)
757 {
758 	struct backend_info *be = dev_get_drvdata(&dev->dev);
759 	int err;
760 
761 	pr_debug("%s %p %s\n", __func__, dev, xenbus_strstate(frontend_state));
762 
763 	switch (frontend_state) {
764 	case XenbusStateInitialising:
765 		if (dev->state == XenbusStateClosed) {
766 			pr_info("%s: prepare for reconnect\n", dev->nodename);
767 			xenbus_switch_state(dev, XenbusStateInitWait);
768 		}
769 		break;
770 
771 	case XenbusStateInitialised:
772 	case XenbusStateConnected:
773 		/*
774 		 * Ensure we connect even when two watches fire in
775 		 * close succession and we miss the intermediate value
776 		 * of frontend_state.
777 		 */
778 		if (dev->state == XenbusStateConnected)
779 			break;
780 
781 		/*
782 		 * Enforce precondition before potential leak point.
783 		 * xen_blkif_disconnect() is idempotent.
784 		 */
785 		err = xen_blkif_disconnect(be->blkif);
786 		if (err) {
787 			xenbus_dev_fatal(dev, err, "pending I/O");
788 			break;
789 		}
790 
791 		err = connect_ring(be);
792 		if (err) {
793 			/*
794 			 * Clean up so that memory resources can be used by
795 			 * other devices. connect_ring reported already error.
796 			 */
797 			xen_blkif_disconnect(be->blkif);
798 			break;
799 		}
800 		xen_update_blkif_status(be->blkif);
801 		break;
802 
803 	case XenbusStateClosing:
804 		xenbus_switch_state(dev, XenbusStateClosing);
805 		break;
806 
807 	case XenbusStateClosed:
808 		xen_blkif_disconnect(be->blkif);
809 		xenbus_switch_state(dev, XenbusStateClosed);
810 		if (xenbus_dev_is_online(dev))
811 			break;
812 		/* fall through */
813 		/* if not online */
814 	case XenbusStateUnknown:
815 		/* implies xen_blkif_disconnect() via xen_blkbk_remove() */
816 		device_unregister(&dev->dev);
817 		break;
818 
819 	default:
820 		xenbus_dev_fatal(dev, -EINVAL, "saw state %d at frontend",
821 				 frontend_state);
822 		break;
823 	}
824 }
825 
826 
827 /* ** Connection ** */
828 
829 
830 /*
831  * Write the physical details regarding the block device to the store, and
832  * switch to Connected state.
833  */
connect(struct backend_info * be)834 static void connect(struct backend_info *be)
835 {
836 	struct xenbus_transaction xbt;
837 	int err;
838 	struct xenbus_device *dev = be->dev;
839 
840 	pr_debug("%s %s\n", __func__, dev->otherend);
841 
842 	/* Supply the information about the device the frontend needs */
843 again:
844 	err = xenbus_transaction_start(&xbt);
845 	if (err) {
846 		xenbus_dev_fatal(dev, err, "starting transaction");
847 		return;
848 	}
849 
850 	/* If we can't advertise it is OK. */
851 	xen_blkbk_flush_diskcache(xbt, be, be->blkif->vbd.flush_support);
852 
853 	xen_blkbk_discard(xbt, be);
854 
855 	xen_blkbk_barrier(xbt, be, be->blkif->vbd.flush_support);
856 
857 	err = xenbus_printf(xbt, dev->nodename, "feature-persistent", "%u", 1);
858 	if (err) {
859 		xenbus_dev_fatal(dev, err, "writing %s/feature-persistent",
860 				 dev->nodename);
861 		goto abort;
862 	}
863 
864 	err = xenbus_printf(xbt, dev->nodename, "sectors", "%llu",
865 			    (unsigned long long)vbd_sz(&be->blkif->vbd));
866 	if (err) {
867 		xenbus_dev_fatal(dev, err, "writing %s/sectors",
868 				 dev->nodename);
869 		goto abort;
870 	}
871 
872 	/* FIXME: use a typename instead */
873 	err = xenbus_printf(xbt, dev->nodename, "info", "%u",
874 			    be->blkif->vbd.type |
875 			    (be->blkif->vbd.readonly ? VDISK_READONLY : 0));
876 	if (err) {
877 		xenbus_dev_fatal(dev, err, "writing %s/info",
878 				 dev->nodename);
879 		goto abort;
880 	}
881 	err = xenbus_printf(xbt, dev->nodename, "sector-size", "%lu",
882 			    (unsigned long)
883 			    bdev_logical_block_size(be->blkif->vbd.bdev));
884 	if (err) {
885 		xenbus_dev_fatal(dev, err, "writing %s/sector-size",
886 				 dev->nodename);
887 		goto abort;
888 	}
889 	err = xenbus_printf(xbt, dev->nodename, "physical-sector-size", "%u",
890 			    bdev_physical_block_size(be->blkif->vbd.bdev));
891 	if (err)
892 		xenbus_dev_error(dev, err, "writing %s/physical-sector-size",
893 				 dev->nodename);
894 
895 	err = xenbus_transaction_end(xbt, 0);
896 	if (err == -EAGAIN)
897 		goto again;
898 	if (err)
899 		xenbus_dev_fatal(dev, err, "ending transaction");
900 
901 	err = xenbus_switch_state(dev, XenbusStateConnected);
902 	if (err)
903 		xenbus_dev_fatal(dev, err, "%s: switching to Connected state",
904 				 dev->nodename);
905 
906 	return;
907  abort:
908 	xenbus_transaction_end(xbt, 1);
909 }
910 
911 /*
912  * Each ring may have multi pages, depends on "ring-page-order".
913  */
read_per_ring_refs(struct xen_blkif_ring * ring,const char * dir)914 static int read_per_ring_refs(struct xen_blkif_ring *ring, const char *dir)
915 {
916 	unsigned int ring_ref[XENBUS_MAX_RING_GRANTS];
917 	struct pending_req *req, *n;
918 	int err, i, j;
919 	struct xen_blkif *blkif = ring->blkif;
920 	struct xenbus_device *dev = blkif->be->dev;
921 	unsigned int nr_grefs, evtchn;
922 
923 	err = xenbus_scanf(XBT_NIL, dir, "event-channel", "%u",
924 			  &evtchn);
925 	if (err != 1) {
926 		err = -EINVAL;
927 		xenbus_dev_fatal(dev, err, "reading %s/event-channel", dir);
928 		return err;
929 	}
930 
931 	nr_grefs = blkif->nr_ring_pages;
932 
933 	if (unlikely(!nr_grefs)) {
934 		WARN_ON(true);
935 		return -EINVAL;
936 	}
937 
938 	for (i = 0; i < nr_grefs; i++) {
939 		char ring_ref_name[RINGREF_NAME_LEN];
940 
941 		snprintf(ring_ref_name, RINGREF_NAME_LEN, "ring-ref%u", i);
942 		err = xenbus_scanf(XBT_NIL, dir, ring_ref_name,
943 				   "%u", &ring_ref[i]);
944 
945 		if (err != 1) {
946 			if (nr_grefs == 1)
947 				break;
948 
949 			err = -EINVAL;
950 			xenbus_dev_fatal(dev, err, "reading %s/%s",
951 					 dir, ring_ref_name);
952 			return err;
953 		}
954 	}
955 
956 	if (err != 1) {
957 		WARN_ON(nr_grefs != 1);
958 
959 		err = xenbus_scanf(XBT_NIL, dir, "ring-ref", "%u",
960 				   &ring_ref[0]);
961 		if (err != 1) {
962 			err = -EINVAL;
963 			xenbus_dev_fatal(dev, err, "reading %s/ring-ref", dir);
964 			return err;
965 		}
966 	}
967 
968 	err = -ENOMEM;
969 	for (i = 0; i < nr_grefs * XEN_BLKIF_REQS_PER_PAGE; i++) {
970 		req = kzalloc(sizeof(*req), GFP_KERNEL);
971 		if (!req)
972 			goto fail;
973 		list_add_tail(&req->free_list, &ring->pending_free);
974 		for (j = 0; j < MAX_INDIRECT_SEGMENTS; j++) {
975 			req->segments[j] = kzalloc(sizeof(*req->segments[0]), GFP_KERNEL);
976 			if (!req->segments[j])
977 				goto fail;
978 		}
979 		for (j = 0; j < MAX_INDIRECT_PAGES; j++) {
980 			req->indirect_pages[j] = kzalloc(sizeof(*req->indirect_pages[0]),
981 							 GFP_KERNEL);
982 			if (!req->indirect_pages[j])
983 				goto fail;
984 		}
985 	}
986 
987 	/* Map the shared frame, irq etc. */
988 	err = xen_blkif_map(ring, ring_ref, nr_grefs, evtchn);
989 	if (err) {
990 		xenbus_dev_fatal(dev, err, "mapping ring-ref port %u", evtchn);
991 		goto fail;
992 	}
993 
994 	return 0;
995 
996 fail:
997 	list_for_each_entry_safe(req, n, &ring->pending_free, free_list) {
998 		list_del(&req->free_list);
999 		for (j = 0; j < MAX_INDIRECT_SEGMENTS; j++) {
1000 			if (!req->segments[j])
1001 				break;
1002 			kfree(req->segments[j]);
1003 		}
1004 		for (j = 0; j < MAX_INDIRECT_PAGES; j++) {
1005 			if (!req->indirect_pages[j])
1006 				break;
1007 			kfree(req->indirect_pages[j]);
1008 		}
1009 		kfree(req);
1010 	}
1011 	return err;
1012 }
1013 
connect_ring(struct backend_info * be)1014 static int connect_ring(struct backend_info *be)
1015 {
1016 	struct xenbus_device *dev = be->dev;
1017 	struct xen_blkif *blkif = be->blkif;
1018 	unsigned int pers_grants;
1019 	char protocol[64] = "";
1020 	int err, i;
1021 	char *xspath;
1022 	size_t xspathsize;
1023 	const size_t xenstore_path_ext_size = 11; /* sufficient for "/queue-NNN" */
1024 	unsigned int requested_num_queues = 0;
1025 	unsigned int ring_page_order;
1026 
1027 	pr_debug("%s %s\n", __func__, dev->otherend);
1028 
1029 	blkif->blk_protocol = BLKIF_PROTOCOL_DEFAULT;
1030 	err = xenbus_scanf(XBT_NIL, dev->otherend, "protocol",
1031 			   "%63s", protocol);
1032 	if (err <= 0)
1033 		strcpy(protocol, "unspecified, assuming default");
1034 	else if (0 == strcmp(protocol, XEN_IO_PROTO_ABI_NATIVE))
1035 		blkif->blk_protocol = BLKIF_PROTOCOL_NATIVE;
1036 	else if (0 == strcmp(protocol, XEN_IO_PROTO_ABI_X86_32))
1037 		blkif->blk_protocol = BLKIF_PROTOCOL_X86_32;
1038 	else if (0 == strcmp(protocol, XEN_IO_PROTO_ABI_X86_64))
1039 		blkif->blk_protocol = BLKIF_PROTOCOL_X86_64;
1040 	else {
1041 		xenbus_dev_fatal(dev, err, "unknown fe protocol %s", protocol);
1042 		return -ENOSYS;
1043 	}
1044 	pers_grants = xenbus_read_unsigned(dev->otherend, "feature-persistent",
1045 					   0);
1046 	blkif->vbd.feature_gnt_persistent = pers_grants;
1047 	blkif->vbd.overflow_max_grants = 0;
1048 
1049 	/*
1050 	 * Read the number of hardware queues from frontend.
1051 	 */
1052 	requested_num_queues = xenbus_read_unsigned(dev->otherend,
1053 						    "multi-queue-num-queues",
1054 						    1);
1055 	if (requested_num_queues > xenblk_max_queues
1056 	    || requested_num_queues == 0) {
1057 		/* Buggy or malicious guest. */
1058 		xenbus_dev_fatal(dev, err,
1059 				"guest requested %u queues, exceeding the maximum of %u.",
1060 				requested_num_queues, xenblk_max_queues);
1061 		return -ENOSYS;
1062 	}
1063 	blkif->nr_rings = requested_num_queues;
1064 	if (xen_blkif_alloc_rings(blkif))
1065 		return -ENOMEM;
1066 
1067 	pr_info("%s: using %d queues, protocol %d (%s) %s\n", dev->nodename,
1068 		 blkif->nr_rings, blkif->blk_protocol, protocol,
1069 		 pers_grants ? "persistent grants" : "");
1070 
1071 	ring_page_order = xenbus_read_unsigned(dev->otherend,
1072 					       "ring-page-order", 0);
1073 
1074 	if (ring_page_order > xen_blkif_max_ring_order) {
1075 		err = -EINVAL;
1076 		xenbus_dev_fatal(dev, err,
1077 				 "requested ring page order %d exceed max:%d",
1078 				 ring_page_order,
1079 				 xen_blkif_max_ring_order);
1080 		return err;
1081 	}
1082 
1083 	blkif->nr_ring_pages = 1 << ring_page_order;
1084 
1085 	if (blkif->nr_rings == 1)
1086 		return read_per_ring_refs(&blkif->rings[0], dev->otherend);
1087 	else {
1088 		xspathsize = strlen(dev->otherend) + xenstore_path_ext_size;
1089 		xspath = kmalloc(xspathsize, GFP_KERNEL);
1090 		if (!xspath) {
1091 			xenbus_dev_fatal(dev, -ENOMEM, "reading ring references");
1092 			return -ENOMEM;
1093 		}
1094 
1095 		for (i = 0; i < blkif->nr_rings; i++) {
1096 			memset(xspath, 0, xspathsize);
1097 			snprintf(xspath, xspathsize, "%s/queue-%u", dev->otherend, i);
1098 			err = read_per_ring_refs(&blkif->rings[i], xspath);
1099 			if (err) {
1100 				kfree(xspath);
1101 				return err;
1102 			}
1103 		}
1104 		kfree(xspath);
1105 	}
1106 	return 0;
1107 }
1108 
1109 static const struct xenbus_device_id xen_blkbk_ids[] = {
1110 	{ "vbd" },
1111 	{ "" }
1112 };
1113 
1114 static struct xenbus_driver xen_blkbk_driver = {
1115 	.ids  = xen_blkbk_ids,
1116 	.probe = xen_blkbk_probe,
1117 	.remove = xen_blkbk_remove,
1118 	.otherend_changed = frontend_changed
1119 };
1120 
xen_blkif_xenbus_init(void)1121 int xen_blkif_xenbus_init(void)
1122 {
1123 	return xenbus_register_backend(&xen_blkbk_driver);
1124 }
1125