| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 1 | /* | 
 | 2 |  * Virtio PCI driver | 
 | 3 |  * | 
 | 4 |  * This module allows virtio devices to be used over a virtual PCI device. | 
 | 5 |  * This can be used with QEMU based VMMs like KVM or Xen. | 
 | 6 |  * | 
 | 7 |  * Copyright IBM Corp. 2007 | 
 | 8 |  * | 
 | 9 |  * Authors: | 
 | 10 |  *  Anthony Liguori  <aliguori@us.ibm.com> | 
 | 11 |  * | 
 | 12 |  * This work is licensed under the terms of the GNU GPL, version 2 or later. | 
 | 13 |  * See the COPYING file in the top-level directory. | 
 | 14 |  * | 
 | 15 |  */ | 
 | 16 |  | 
 | 17 | #include <linux/module.h> | 
 | 18 | #include <linux/list.h> | 
 | 19 | #include <linux/pci.h> | 
| Tejun Heo | 5a0e3ad | 2010-03-24 17:04:11 +0900 | [diff] [blame] | 20 | #include <linux/slab.h> | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 21 | #include <linux/interrupt.h> | 
 | 22 | #include <linux/virtio.h> | 
 | 23 | #include <linux/virtio_config.h> | 
 | 24 | #include <linux/virtio_ring.h> | 
 | 25 | #include <linux/virtio_pci.h> | 
 | 26 | #include <linux/highmem.h> | 
 | 27 | #include <linux/spinlock.h> | 
 | 28 |  | 
 | 29 | MODULE_AUTHOR("Anthony Liguori <aliguori@us.ibm.com>"); | 
 | 30 | MODULE_DESCRIPTION("virtio-pci"); | 
 | 31 | MODULE_LICENSE("GPL"); | 
 | 32 | MODULE_VERSION("1"); | 
 | 33 |  | 
 | 34 | /* Our device structure */ | 
 | 35 | struct virtio_pci_device | 
 | 36 | { | 
 | 37 | 	struct virtio_device vdev; | 
 | 38 | 	struct pci_dev *pci_dev; | 
 | 39 |  | 
 | 40 | 	/* the IO mapping for the PCI config space */ | 
| Al Viro | 9796835 | 2008-03-29 03:09:48 +0000 | [diff] [blame] | 41 | 	void __iomem *ioaddr; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 42 |  | 
 | 43 | 	/* a list of queues so we can dispatch IRQs */ | 
 | 44 | 	spinlock_t lock; | 
 | 45 | 	struct list_head virtqueues; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 46 |  | 
 | 47 | 	/* MSI-X support */ | 
 | 48 | 	int msix_enabled; | 
 | 49 | 	int intx_enabled; | 
 | 50 | 	struct msix_entry *msix_entries; | 
 | 51 | 	/* Name strings for interrupts. This size should be enough, | 
 | 52 | 	 * and I'm too lazy to allocate each name separately. */ | 
 | 53 | 	char (*msix_names)[256]; | 
 | 54 | 	/* Number of available vectors */ | 
 | 55 | 	unsigned msix_vectors; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 56 | 	/* Vectors allocated, excluding per-vq vectors if any */ | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 57 | 	unsigned msix_used_vectors; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 58 | 	/* Whether we have vector per vq */ | 
 | 59 | 	bool per_vq_vectors; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 60 | }; | 
 | 61 |  | 
 | 62 | /* Constants for MSI-X */ | 
 | 63 | /* Use first vector for configuration changes, second and the rest for | 
 | 64 |  * virtqueues Thus, we need at least 2 vectors for MSI. */ | 
 | 65 | enum { | 
 | 66 | 	VP_MSIX_CONFIG_VECTOR = 0, | 
 | 67 | 	VP_MSIX_VQ_VECTOR = 1, | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 68 | }; | 
 | 69 |  | 
 | 70 | struct virtio_pci_vq_info | 
 | 71 | { | 
 | 72 | 	/* the actual virtqueue */ | 
 | 73 | 	struct virtqueue *vq; | 
 | 74 |  | 
 | 75 | 	/* the number of entries in the queue */ | 
 | 76 | 	int num; | 
 | 77 |  | 
 | 78 | 	/* the index of the queue */ | 
 | 79 | 	int queue_index; | 
 | 80 |  | 
 | 81 | 	/* the virtual address of the ring queue */ | 
 | 82 | 	void *queue; | 
 | 83 |  | 
 | 84 | 	/* the list node for the virtqueues list */ | 
 | 85 | 	struct list_head node; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 86 |  | 
 | 87 | 	/* MSI-X vector (or none) */ | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 88 | 	unsigned msix_vector; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 89 | }; | 
 | 90 |  | 
 | 91 | /* Qumranet donated their vendor ID for devices 0x1000 thru 0x10FF. */ | 
 | 92 | static struct pci_device_id virtio_pci_id_table[] = { | 
 | 93 | 	{ 0x1af4, PCI_ANY_ID, PCI_ANY_ID, PCI_ANY_ID, 0, 0, 0 }, | 
 | 94 | 	{ 0 }, | 
 | 95 | }; | 
 | 96 |  | 
 | 97 | MODULE_DEVICE_TABLE(pci, virtio_pci_id_table); | 
 | 98 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 99 | /* Convert a generic virtio device to our structure */ | 
 | 100 | static struct virtio_pci_device *to_vp_device(struct virtio_device *vdev) | 
 | 101 | { | 
 | 102 | 	return container_of(vdev, struct virtio_pci_device, vdev); | 
 | 103 | } | 
 | 104 |  | 
| Rusty Russell | c45a681 | 2008-05-02 21:50:50 -0500 | [diff] [blame] | 105 | /* virtio config->get_features() implementation */ | 
 | 106 | static u32 vp_get_features(struct virtio_device *vdev) | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 107 | { | 
 | 108 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 109 |  | 
| Rusty Russell | c45a681 | 2008-05-02 21:50:50 -0500 | [diff] [blame] | 110 | 	/* When someone needs more than 32 feature bits, we'll need to | 
 | 111 | 	 * steal a bit to indicate that the rest are somewhere else. */ | 
 | 112 | 	return ioread32(vp_dev->ioaddr + VIRTIO_PCI_HOST_FEATURES); | 
 | 113 | } | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 114 |  | 
| Rusty Russell | c624896 | 2008-07-25 12:06:07 -0500 | [diff] [blame] | 115 | /* virtio config->finalize_features() implementation */ | 
 | 116 | static void vp_finalize_features(struct virtio_device *vdev) | 
| Rusty Russell | c45a681 | 2008-05-02 21:50:50 -0500 | [diff] [blame] | 117 | { | 
 | 118 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 119 |  | 
| Rusty Russell | e34f872 | 2008-07-25 12:06:13 -0500 | [diff] [blame] | 120 | 	/* Give virtio_ring a chance to accept features. */ | 
 | 121 | 	vring_transport_features(vdev); | 
 | 122 |  | 
| Rusty Russell | c624896 | 2008-07-25 12:06:07 -0500 | [diff] [blame] | 123 | 	/* We only support 32 feature bits. */ | 
 | 124 | 	BUILD_BUG_ON(ARRAY_SIZE(vdev->features) != 1); | 
 | 125 | 	iowrite32(vdev->features[0], vp_dev->ioaddr+VIRTIO_PCI_GUEST_FEATURES); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 126 | } | 
 | 127 |  | 
 | 128 | /* virtio config->get() implementation */ | 
 | 129 | static void vp_get(struct virtio_device *vdev, unsigned offset, | 
 | 130 | 		   void *buf, unsigned len) | 
 | 131 | { | 
 | 132 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 133 | 	void __iomem *ioaddr = vp_dev->ioaddr + | 
 | 134 | 				VIRTIO_PCI_CONFIG(vp_dev) + offset; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 135 | 	u8 *ptr = buf; | 
 | 136 | 	int i; | 
 | 137 |  | 
 | 138 | 	for (i = 0; i < len; i++) | 
 | 139 | 		ptr[i] = ioread8(ioaddr + i); | 
 | 140 | } | 
 | 141 |  | 
 | 142 | /* the config->set() implementation.  it's symmetric to the config->get() | 
 | 143 |  * implementation */ | 
 | 144 | static void vp_set(struct virtio_device *vdev, unsigned offset, | 
 | 145 | 		   const void *buf, unsigned len) | 
 | 146 | { | 
 | 147 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 148 | 	void __iomem *ioaddr = vp_dev->ioaddr + | 
 | 149 | 				VIRTIO_PCI_CONFIG(vp_dev) + offset; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 150 | 	const u8 *ptr = buf; | 
 | 151 | 	int i; | 
 | 152 |  | 
 | 153 | 	for (i = 0; i < len; i++) | 
 | 154 | 		iowrite8(ptr[i], ioaddr + i); | 
 | 155 | } | 
 | 156 |  | 
 | 157 | /* config->{get,set}_status() implementations */ | 
 | 158 | static u8 vp_get_status(struct virtio_device *vdev) | 
 | 159 | { | 
 | 160 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 161 | 	return ioread8(vp_dev->ioaddr + VIRTIO_PCI_STATUS); | 
 | 162 | } | 
 | 163 |  | 
 | 164 | static void vp_set_status(struct virtio_device *vdev, u8 status) | 
 | 165 | { | 
 | 166 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 167 | 	/* We should never be setting status to 0. */ | 
 | 168 | 	BUG_ON(status == 0); | 
| Harvey Harrison | 597d56e | 2008-03-31 17:53:55 -0700 | [diff] [blame] | 169 | 	iowrite8(status, vp_dev->ioaddr + VIRTIO_PCI_STATUS); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 170 | } | 
 | 171 |  | 
 | 172 | static void vp_reset(struct virtio_device *vdev) | 
 | 173 | { | 
 | 174 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 175 | 	/* 0 status means a reset. */ | 
| Harvey Harrison | 597d56e | 2008-03-31 17:53:55 -0700 | [diff] [blame] | 176 | 	iowrite8(0, vp_dev->ioaddr + VIRTIO_PCI_STATUS); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 177 | } | 
 | 178 |  | 
 | 179 | /* the notify function used when creating a virt queue */ | 
 | 180 | static void vp_notify(struct virtqueue *vq) | 
 | 181 | { | 
 | 182 | 	struct virtio_pci_device *vp_dev = to_vp_device(vq->vdev); | 
 | 183 | 	struct virtio_pci_vq_info *info = vq->priv; | 
 | 184 |  | 
 | 185 | 	/* we write the queue's selector into the notification register to | 
 | 186 | 	 * signal the other end */ | 
 | 187 | 	iowrite16(info->queue_index, vp_dev->ioaddr + VIRTIO_PCI_QUEUE_NOTIFY); | 
 | 188 | } | 
 | 189 |  | 
| Michael S. Tsirkin | 77cf5246 | 2009-05-14 13:55:31 +0300 | [diff] [blame] | 190 | /* Handle a configuration change: Tell driver if it wants to know. */ | 
 | 191 | static irqreturn_t vp_config_changed(int irq, void *opaque) | 
 | 192 | { | 
 | 193 | 	struct virtio_pci_device *vp_dev = opaque; | 
 | 194 | 	struct virtio_driver *drv; | 
 | 195 | 	drv = container_of(vp_dev->vdev.dev.driver, | 
 | 196 | 			   struct virtio_driver, driver); | 
 | 197 |  | 
 | 198 | 	if (drv && drv->config_changed) | 
 | 199 | 		drv->config_changed(&vp_dev->vdev); | 
 | 200 | 	return IRQ_HANDLED; | 
 | 201 | } | 
 | 202 |  | 
 | 203 | /* Notify all virtqueues on an interrupt. */ | 
 | 204 | static irqreturn_t vp_vring_interrupt(int irq, void *opaque) | 
 | 205 | { | 
 | 206 | 	struct virtio_pci_device *vp_dev = opaque; | 
 | 207 | 	struct virtio_pci_vq_info *info; | 
 | 208 | 	irqreturn_t ret = IRQ_NONE; | 
 | 209 | 	unsigned long flags; | 
 | 210 |  | 
 | 211 | 	spin_lock_irqsave(&vp_dev->lock, flags); | 
 | 212 | 	list_for_each_entry(info, &vp_dev->virtqueues, node) { | 
 | 213 | 		if (vring_interrupt(irq, info->vq) == IRQ_HANDLED) | 
 | 214 | 			ret = IRQ_HANDLED; | 
 | 215 | 	} | 
 | 216 | 	spin_unlock_irqrestore(&vp_dev->lock, flags); | 
 | 217 |  | 
 | 218 | 	return ret; | 
 | 219 | } | 
 | 220 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 221 | /* A small wrapper to also acknowledge the interrupt when it's handled. | 
 | 222 |  * I really need an EIO hook for the vring so I can ack the interrupt once we | 
 | 223 |  * know that we'll be handling the IRQ but before we invoke the callback since | 
 | 224 |  * the callback may notify the host which results in the host attempting to | 
 | 225 |  * raise an interrupt that we would then mask once we acknowledged the | 
 | 226 |  * interrupt. */ | 
 | 227 | static irqreturn_t vp_interrupt(int irq, void *opaque) | 
 | 228 | { | 
 | 229 | 	struct virtio_pci_device *vp_dev = opaque; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 230 | 	u8 isr; | 
 | 231 |  | 
 | 232 | 	/* reading the ISR has the effect of also clearing it so it's very | 
 | 233 | 	 * important to save off the value. */ | 
 | 234 | 	isr = ioread8(vp_dev->ioaddr + VIRTIO_PCI_ISR); | 
 | 235 |  | 
 | 236 | 	/* It's definitely not us if the ISR was not high */ | 
 | 237 | 	if (!isr) | 
 | 238 | 		return IRQ_NONE; | 
 | 239 |  | 
 | 240 | 	/* Configuration change?  Tell driver if it wants to know. */ | 
| Michael S. Tsirkin | 77cf5246 | 2009-05-14 13:55:31 +0300 | [diff] [blame] | 241 | 	if (isr & VIRTIO_PCI_ISR_CONFIG) | 
 | 242 | 		vp_config_changed(irq, opaque); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 243 |  | 
| Michael S. Tsirkin | 77cf5246 | 2009-05-14 13:55:31 +0300 | [diff] [blame] | 244 | 	return vp_vring_interrupt(irq, opaque); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 245 | } | 
 | 246 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 247 | static void vp_free_vectors(struct virtio_device *vdev) | 
 | 248 | { | 
 | 249 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 250 | 	int i; | 
 | 251 |  | 
 | 252 | 	if (vp_dev->intx_enabled) { | 
 | 253 | 		free_irq(vp_dev->pci_dev->irq, vp_dev); | 
 | 254 | 		vp_dev->intx_enabled = 0; | 
 | 255 | 	} | 
 | 256 |  | 
 | 257 | 	for (i = 0; i < vp_dev->msix_used_vectors; ++i) | 
 | 258 | 		free_irq(vp_dev->msix_entries[i].vector, vp_dev); | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 259 |  | 
 | 260 | 	if (vp_dev->msix_enabled) { | 
 | 261 | 		/* Disable the vector used for configuration */ | 
 | 262 | 		iowrite16(VIRTIO_MSI_NO_VECTOR, | 
 | 263 | 			  vp_dev->ioaddr + VIRTIO_MSI_CONFIG_VECTOR); | 
 | 264 | 		/* Flush the write out to device */ | 
 | 265 | 		ioread16(vp_dev->ioaddr + VIRTIO_MSI_CONFIG_VECTOR); | 
 | 266 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 267 | 		pci_disable_msix(vp_dev->pci_dev); | 
| Michael S. Tsirkin | ff52c3f | 2009-07-23 14:57:37 +0300 | [diff] [blame] | 268 | 		vp_dev->msix_enabled = 0; | 
 | 269 | 		vp_dev->msix_vectors = 0; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 270 | 	} | 
| Michael S. Tsirkin | ff52c3f | 2009-07-23 14:57:37 +0300 | [diff] [blame] | 271 |  | 
 | 272 | 	vp_dev->msix_used_vectors = 0; | 
 | 273 | 	kfree(vp_dev->msix_names); | 
 | 274 | 	vp_dev->msix_names = NULL; | 
 | 275 | 	kfree(vp_dev->msix_entries); | 
 | 276 | 	vp_dev->msix_entries = NULL; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 277 | } | 
 | 278 |  | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 279 | static int vp_request_msix_vectors(struct virtio_device *vdev, int nvectors, | 
 | 280 | 				   bool per_vq_vectors) | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 281 | { | 
 | 282 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 283 | 	const char *name = dev_name(&vp_dev->vdev.dev); | 
 | 284 | 	unsigned i, v; | 
 | 285 | 	int err = -ENOMEM; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 286 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 287 | 	vp_dev->msix_entries = kmalloc(nvectors * sizeof *vp_dev->msix_entries, | 
 | 288 | 				       GFP_KERNEL); | 
 | 289 | 	if (!vp_dev->msix_entries) | 
| Michael S. Tsirkin | ff52c3f | 2009-07-23 14:57:37 +0300 | [diff] [blame] | 290 | 		goto error; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 291 | 	vp_dev->msix_names = kmalloc(nvectors * sizeof *vp_dev->msix_names, | 
 | 292 | 				     GFP_KERNEL); | 
 | 293 | 	if (!vp_dev->msix_names) | 
| Michael S. Tsirkin | ff52c3f | 2009-07-23 14:57:37 +0300 | [diff] [blame] | 294 | 		goto error; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 295 |  | 
 | 296 | 	for (i = 0; i < nvectors; ++i) | 
 | 297 | 		vp_dev->msix_entries[i].entry = i; | 
 | 298 |  | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 299 | 	/* pci_enable_msix returns positive if we can't get this many. */ | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 300 | 	err = pci_enable_msix(vp_dev->pci_dev, vp_dev->msix_entries, nvectors); | 
 | 301 | 	if (err > 0) | 
 | 302 | 		err = -ENOSPC; | 
 | 303 | 	if (err) | 
 | 304 | 		goto error; | 
 | 305 | 	vp_dev->msix_vectors = nvectors; | 
 | 306 | 	vp_dev->msix_enabled = 1; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 307 |  | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 308 | 	/* Set the vector used for configuration */ | 
 | 309 | 	v = vp_dev->msix_used_vectors; | 
 | 310 | 	snprintf(vp_dev->msix_names[v], sizeof *vp_dev->msix_names, | 
 | 311 | 		 "%s-config", name); | 
 | 312 | 	err = request_irq(vp_dev->msix_entries[v].vector, | 
 | 313 | 			  vp_config_changed, 0, vp_dev->msix_names[v], | 
 | 314 | 			  vp_dev); | 
 | 315 | 	if (err) | 
 | 316 | 		goto error; | 
 | 317 | 	++vp_dev->msix_used_vectors; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 318 |  | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 319 | 	iowrite16(v, vp_dev->ioaddr + VIRTIO_MSI_CONFIG_VECTOR); | 
 | 320 | 	/* Verify we had enough resources to assign the vector */ | 
 | 321 | 	v = ioread16(vp_dev->ioaddr + VIRTIO_MSI_CONFIG_VECTOR); | 
 | 322 | 	if (v == VIRTIO_MSI_NO_VECTOR) { | 
 | 323 | 		err = -EBUSY; | 
 | 324 | 		goto error; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 325 | 	} | 
 | 326 |  | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 327 | 	if (!per_vq_vectors) { | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 328 | 		/* Shared vector for all VQs */ | 
 | 329 | 		v = vp_dev->msix_used_vectors; | 
 | 330 | 		snprintf(vp_dev->msix_names[v], sizeof *vp_dev->msix_names, | 
 | 331 | 			 "%s-virtqueues", name); | 
 | 332 | 		err = request_irq(vp_dev->msix_entries[v].vector, | 
 | 333 | 				  vp_vring_interrupt, 0, vp_dev->msix_names[v], | 
 | 334 | 				  vp_dev); | 
 | 335 | 		if (err) | 
| Michael S. Tsirkin | ff52c3f | 2009-07-23 14:57:37 +0300 | [diff] [blame] | 336 | 			goto error; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 337 | 		++vp_dev->msix_used_vectors; | 
 | 338 | 	} | 
 | 339 | 	return 0; | 
| Michael S. Tsirkin | ff52c3f | 2009-07-23 14:57:37 +0300 | [diff] [blame] | 340 | error: | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 341 | 	vp_free_vectors(vdev); | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 342 | 	return err; | 
 | 343 | } | 
 | 344 |  | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 345 | static int vp_request_intx(struct virtio_device *vdev) | 
 | 346 | { | 
 | 347 | 	int err; | 
 | 348 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 349 |  | 
 | 350 | 	err = request_irq(vp_dev->pci_dev->irq, vp_interrupt, | 
 | 351 | 			  IRQF_SHARED, dev_name(&vdev->dev), vp_dev); | 
 | 352 | 	if (!err) | 
 | 353 | 		vp_dev->intx_enabled = 1; | 
 | 354 | 	return err; | 
 | 355 | } | 
 | 356 |  | 
 | 357 | static struct virtqueue *setup_vq(struct virtio_device *vdev, unsigned index, | 
 | 358 | 				  void (*callback)(struct virtqueue *vq), | 
 | 359 | 				  const char *name, | 
 | 360 | 				  u16 msix_vec) | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 361 | { | 
 | 362 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
 | 363 | 	struct virtio_pci_vq_info *info; | 
 | 364 | 	struct virtqueue *vq; | 
| Hollis Blanchard | 13b1eb3 | 2008-12-02 16:24:40 -0600 | [diff] [blame] | 365 | 	unsigned long flags, size; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 366 | 	u16 num; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 367 | 	int err; | 
 | 368 |  | 
 | 369 | 	/* Select the queue we're interested in */ | 
 | 370 | 	iowrite16(index, vp_dev->ioaddr + VIRTIO_PCI_QUEUE_SEL); | 
 | 371 |  | 
 | 372 | 	/* Check if queue is either not available or already active. */ | 
 | 373 | 	num = ioread16(vp_dev->ioaddr + VIRTIO_PCI_QUEUE_NUM); | 
 | 374 | 	if (!num || ioread32(vp_dev->ioaddr + VIRTIO_PCI_QUEUE_PFN)) | 
 | 375 | 		return ERR_PTR(-ENOENT); | 
 | 376 |  | 
 | 377 | 	/* allocate and fill out our structure the represents an active | 
 | 378 | 	 * queue */ | 
 | 379 | 	info = kmalloc(sizeof(struct virtio_pci_vq_info), GFP_KERNEL); | 
 | 380 | 	if (!info) | 
 | 381 | 		return ERR_PTR(-ENOMEM); | 
 | 382 |  | 
 | 383 | 	info->queue_index = index; | 
 | 384 | 	info->num = num; | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 385 | 	info->msix_vector = msix_vec; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 386 |  | 
| Rusty Russell | 498af14 | 2008-12-30 09:25:57 -0600 | [diff] [blame] | 387 | 	size = PAGE_ALIGN(vring_size(num, VIRTIO_PCI_VRING_ALIGN)); | 
| Hollis Blanchard | 13b1eb3 | 2008-12-02 16:24:40 -0600 | [diff] [blame] | 388 | 	info->queue = alloc_pages_exact(size, GFP_KERNEL|__GFP_ZERO); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 389 | 	if (info->queue == NULL) { | 
 | 390 | 		err = -ENOMEM; | 
 | 391 | 		goto out_info; | 
 | 392 | 	} | 
 | 393 |  | 
 | 394 | 	/* activate the queue */ | 
| Rusty Russell | 480daab | 2008-12-30 09:25:56 -0600 | [diff] [blame] | 395 | 	iowrite32(virt_to_phys(info->queue) >> VIRTIO_PCI_QUEUE_ADDR_SHIFT, | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 396 | 		  vp_dev->ioaddr + VIRTIO_PCI_QUEUE_PFN); | 
 | 397 |  | 
 | 398 | 	/* create the vring */ | 
| Rusty Russell | 87c7d57 | 2008-12-30 09:26:03 -0600 | [diff] [blame] | 399 | 	vq = vring_new_virtqueue(info->num, VIRTIO_PCI_VRING_ALIGN, | 
| Rusty Russell | 9499f5e | 2009-06-12 22:16:35 -0600 | [diff] [blame] | 400 | 				 vdev, info->queue, vp_notify, callback, name); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 401 | 	if (!vq) { | 
 | 402 | 		err = -ENOMEM; | 
 | 403 | 		goto out_activate_queue; | 
 | 404 | 	} | 
 | 405 |  | 
 | 406 | 	vq->priv = info; | 
 | 407 | 	info->vq = vq; | 
 | 408 |  | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 409 | 	if (msix_vec != VIRTIO_MSI_NO_VECTOR) { | 
 | 410 | 		iowrite16(msix_vec, vp_dev->ioaddr + VIRTIO_MSI_QUEUE_VECTOR); | 
 | 411 | 		msix_vec = ioread16(vp_dev->ioaddr + VIRTIO_MSI_QUEUE_VECTOR); | 
 | 412 | 		if (msix_vec == VIRTIO_MSI_NO_VECTOR) { | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 413 | 			err = -EBUSY; | 
 | 414 | 			goto out_assign; | 
 | 415 | 		} | 
 | 416 | 	} | 
 | 417 |  | 
| Anthony Liguori | 27ebe30 | 2008-03-02 16:37:48 -0600 | [diff] [blame] | 418 | 	spin_lock_irqsave(&vp_dev->lock, flags); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 419 | 	list_add(&info->node, &vp_dev->virtqueues); | 
| Anthony Liguori | 27ebe30 | 2008-03-02 16:37:48 -0600 | [diff] [blame] | 420 | 	spin_unlock_irqrestore(&vp_dev->lock, flags); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 421 |  | 
 | 422 | 	return vq; | 
 | 423 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 424 | out_assign: | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 425 | 	vring_del_virtqueue(vq); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 426 | out_activate_queue: | 
 | 427 | 	iowrite32(0, vp_dev->ioaddr + VIRTIO_PCI_QUEUE_PFN); | 
| Hollis Blanchard | 13b1eb3 | 2008-12-02 16:24:40 -0600 | [diff] [blame] | 428 | 	free_pages_exact(info->queue, size); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 429 | out_info: | 
 | 430 | 	kfree(info); | 
 | 431 | 	return ERR_PTR(err); | 
 | 432 | } | 
 | 433 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 434 | static void vp_del_vq(struct virtqueue *vq) | 
 | 435 | { | 
 | 436 | 	struct virtio_pci_device *vp_dev = to_vp_device(vq->vdev); | 
 | 437 | 	struct virtio_pci_vq_info *info = vq->priv; | 
| Michael S. Tsirkin | f6c8250 | 2009-07-26 15:48:01 +0300 | [diff] [blame] | 438 | 	unsigned long flags, size; | 
 | 439 |  | 
 | 440 | 	spin_lock_irqsave(&vp_dev->lock, flags); | 
 | 441 | 	list_del(&info->node); | 
 | 442 | 	spin_unlock_irqrestore(&vp_dev->lock, flags); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 443 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 444 | 	iowrite16(info->queue_index, vp_dev->ioaddr + VIRTIO_PCI_QUEUE_SEL); | 
 | 445 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 446 | 	if (vp_dev->msix_enabled) { | 
 | 447 | 		iowrite16(VIRTIO_MSI_NO_VECTOR, | 
 | 448 | 			  vp_dev->ioaddr + VIRTIO_MSI_QUEUE_VECTOR); | 
 | 449 | 		/* Flush the write out to device */ | 
 | 450 | 		ioread8(vp_dev->ioaddr + VIRTIO_PCI_ISR); | 
 | 451 | 	} | 
 | 452 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 453 | 	vring_del_virtqueue(vq); | 
 | 454 |  | 
 | 455 | 	/* Select and deactivate the queue */ | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 456 | 	iowrite32(0, vp_dev->ioaddr + VIRTIO_PCI_QUEUE_PFN); | 
 | 457 |  | 
| Rusty Russell | 498af14 | 2008-12-30 09:25:57 -0600 | [diff] [blame] | 458 | 	size = PAGE_ALIGN(vring_size(info->num, VIRTIO_PCI_VRING_ALIGN)); | 
| Hollis Blanchard | 13b1eb3 | 2008-12-02 16:24:40 -0600 | [diff] [blame] | 459 | 	free_pages_exact(info->queue, size); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 460 | 	kfree(info); | 
 | 461 | } | 
 | 462 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 463 | /* the config->del_vqs() implementation */ | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 464 | static void vp_del_vqs(struct virtio_device *vdev) | 
 | 465 | { | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 466 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 467 | 	struct virtqueue *vq, *n; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 468 | 	struct virtio_pci_vq_info *info; | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 469 |  | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 470 | 	list_for_each_entry_safe(vq, n, &vdev->vqs, list) { | 
 | 471 | 		info = vq->priv; | 
| Michael S. Tsirkin | 3119815 | 2010-02-25 19:08:55 +0200 | [diff] [blame] | 472 | 		if (vp_dev->per_vq_vectors && | 
 | 473 | 			info->msix_vector != VIRTIO_MSI_NO_VECTOR) | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 474 | 			free_irq(vp_dev->msix_entries[info->msix_vector].vector, | 
 | 475 | 				 vq); | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 476 | 		vp_del_vq(vq); | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 477 | 	} | 
 | 478 | 	vp_dev->per_vq_vectors = false; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 479 |  | 
 | 480 | 	vp_free_vectors(vdev); | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 481 | } | 
 | 482 |  | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 483 | static int vp_try_to_find_vqs(struct virtio_device *vdev, unsigned nvqs, | 
 | 484 | 			      struct virtqueue *vqs[], | 
 | 485 | 			      vq_callback_t *callbacks[], | 
 | 486 | 			      const char *names[], | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 487 | 			      bool use_msix, | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 488 | 			      bool per_vq_vectors) | 
 | 489 | { | 
 | 490 | 	struct virtio_pci_device *vp_dev = to_vp_device(vdev); | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 491 | 	u16 msix_vec; | 
 | 492 | 	int i, err, nvectors, allocated_vectors; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 493 |  | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 494 | 	if (!use_msix) { | 
 | 495 | 		/* Old style: one normal interrupt for change and all vqs. */ | 
 | 496 | 		err = vp_request_intx(vdev); | 
 | 497 | 		if (err) | 
 | 498 | 			goto error_request; | 
 | 499 | 	} else { | 
 | 500 | 		if (per_vq_vectors) { | 
 | 501 | 			/* Best option: one for change interrupt, one per vq. */ | 
 | 502 | 			nvectors = 1; | 
 | 503 | 			for (i = 0; i < nvqs; ++i) | 
 | 504 | 				if (callbacks[i]) | 
 | 505 | 					++nvectors; | 
 | 506 | 		} else { | 
 | 507 | 			/* Second best: one for change, shared for all vqs. */ | 
 | 508 | 			nvectors = 2; | 
 | 509 | 		} | 
 | 510 |  | 
 | 511 | 		err = vp_request_msix_vectors(vdev, nvectors, per_vq_vectors); | 
 | 512 | 		if (err) | 
 | 513 | 			goto error_request; | 
 | 514 | 	} | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 515 |  | 
 | 516 | 	vp_dev->per_vq_vectors = per_vq_vectors; | 
 | 517 | 	allocated_vectors = vp_dev->msix_used_vectors; | 
 | 518 | 	for (i = 0; i < nvqs; ++i) { | 
 | 519 | 		if (!callbacks[i] || !vp_dev->msix_enabled) | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 520 | 			msix_vec = VIRTIO_MSI_NO_VECTOR; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 521 | 		else if (vp_dev->per_vq_vectors) | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 522 | 			msix_vec = allocated_vectors++; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 523 | 		else | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 524 | 			msix_vec = VP_MSIX_VQ_VECTOR; | 
 | 525 | 		vqs[i] = setup_vq(vdev, i, callbacks[i], names[i], msix_vec); | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 526 | 		if (IS_ERR(vqs[i])) { | 
 | 527 | 			err = PTR_ERR(vqs[i]); | 
 | 528 | 			goto error_find; | 
 | 529 | 		} | 
| Michael S. Tsirkin | 0b22bd0 | 2009-10-22 15:06:06 +0200 | [diff] [blame] | 530 |  | 
 | 531 | 		if (!vp_dev->per_vq_vectors || msix_vec == VIRTIO_MSI_NO_VECTOR) | 
 | 532 | 			continue; | 
 | 533 |  | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 534 | 		/* allocate per-vq irq if available and necessary */ | 
| Michael S. Tsirkin | 0b22bd0 | 2009-10-22 15:06:06 +0200 | [diff] [blame] | 535 | 		snprintf(vp_dev->msix_names[msix_vec], | 
 | 536 | 			 sizeof *vp_dev->msix_names, | 
 | 537 | 			 "%s-%s", | 
 | 538 | 			 dev_name(&vp_dev->vdev.dev), names[i]); | 
 | 539 | 		err = request_irq(vp_dev->msix_entries[msix_vec].vector, | 
 | 540 | 				  vring_interrupt, 0, | 
 | 541 | 				  vp_dev->msix_names[msix_vec], | 
 | 542 | 				  vqs[i]); | 
 | 543 | 		if (err) { | 
 | 544 | 			vp_del_vq(vqs[i]); | 
 | 545 | 			goto error_find; | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 546 | 		} | 
 | 547 | 	} | 
 | 548 | 	return 0; | 
 | 549 |  | 
 | 550 | error_find: | 
 | 551 | 	vp_del_vqs(vdev); | 
 | 552 |  | 
 | 553 | error_request: | 
 | 554 | 	return err; | 
 | 555 | } | 
 | 556 |  | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 557 | /* the config->find_vqs() implementation */ | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 558 | static int vp_find_vqs(struct virtio_device *vdev, unsigned nvqs, | 
 | 559 | 		       struct virtqueue *vqs[], | 
 | 560 | 		       vq_callback_t *callbacks[], | 
 | 561 | 		       const char *names[]) | 
 | 562 | { | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 563 | 	int err; | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 564 |  | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 565 | 	/* Try MSI-X with one vector per queue. */ | 
 | 566 | 	err = vp_try_to_find_vqs(vdev, nvqs, vqs, callbacks, names, true, true); | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 567 | 	if (!err) | 
 | 568 | 		return 0; | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 569 | 	/* Fallback: MSI-X with one vector for config, one shared for queues. */ | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 570 | 	err = vp_try_to_find_vqs(vdev, nvqs, vqs, callbacks, names, | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 571 | 				 true, false); | 
| Michael S. Tsirkin | e969fed | 2009-07-26 15:48:08 +0300 | [diff] [blame] | 572 | 	if (!err) | 
 | 573 | 		return 0; | 
 | 574 | 	/* Finally fall back to regular interrupts. */ | 
| Rusty Russell | f68d240 | 2009-09-23 22:26:29 -0600 | [diff] [blame] | 575 | 	return vp_try_to_find_vqs(vdev, nvqs, vqs, callbacks, names, | 
 | 576 | 				  false, false); | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 577 | } | 
 | 578 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 579 | static struct virtio_config_ops virtio_pci_config_ops = { | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 580 | 	.get		= vp_get, | 
 | 581 | 	.set		= vp_set, | 
 | 582 | 	.get_status	= vp_get_status, | 
 | 583 | 	.set_status	= vp_set_status, | 
 | 584 | 	.reset		= vp_reset, | 
| Michael S. Tsirkin | d2a7ddd | 2009-06-12 22:16:36 -0600 | [diff] [blame] | 585 | 	.find_vqs	= vp_find_vqs, | 
 | 586 | 	.del_vqs	= vp_del_vqs, | 
| Rusty Russell | c45a681 | 2008-05-02 21:50:50 -0500 | [diff] [blame] | 587 | 	.get_features	= vp_get_features, | 
| Rusty Russell | c624896 | 2008-07-25 12:06:07 -0500 | [diff] [blame] | 588 | 	.finalize_features = vp_finalize_features, | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 589 | }; | 
 | 590 |  | 
| Mark McLoughlin | 29f9f12 | 2008-12-10 17:45:34 +0000 | [diff] [blame] | 591 | static void virtio_pci_release_dev(struct device *_d) | 
 | 592 | { | 
| Amit Shah | 31a3ddd | 2011-03-14 17:45:02 +0530 | [diff] [blame] | 593 | 	struct virtio_device *dev = container_of(_d, struct virtio_device, | 
 | 594 | 						 dev); | 
| Mark McLoughlin | 29f9f12 | 2008-12-10 17:45:34 +0000 | [diff] [blame] | 595 | 	struct virtio_pci_device *vp_dev = to_vp_device(dev); | 
| Mark McLoughlin | 29f9f12 | 2008-12-10 17:45:34 +0000 | [diff] [blame] | 596 |  | 
| Mark McLoughlin | 29f9f12 | 2008-12-10 17:45:34 +0000 | [diff] [blame] | 597 | 	kfree(vp_dev); | 
 | 598 | } | 
 | 599 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 600 | /* the PCI probing function */ | 
 | 601 | static int __devinit virtio_pci_probe(struct pci_dev *pci_dev, | 
 | 602 | 				      const struct pci_device_id *id) | 
 | 603 | { | 
 | 604 | 	struct virtio_pci_device *vp_dev; | 
 | 605 | 	int err; | 
 | 606 |  | 
 | 607 | 	/* We only own devices >= 0x1000 and <= 0x103f: leave the rest. */ | 
 | 608 | 	if (pci_dev->device < 0x1000 || pci_dev->device > 0x103f) | 
 | 609 | 		return -ENODEV; | 
 | 610 |  | 
| Anthony Liguori | 55a7c06 | 2008-01-28 09:59:59 -0600 | [diff] [blame] | 611 | 	if (pci_dev->revision != VIRTIO_PCI_ABI_VERSION) { | 
 | 612 | 		printk(KERN_ERR "virtio_pci: expected ABI version %d, got %d\n", | 
 | 613 | 		       VIRTIO_PCI_ABI_VERSION, pci_dev->revision); | 
 | 614 | 		return -ENODEV; | 
 | 615 | 	} | 
 | 616 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 617 | 	/* allocate our structure and fill it out */ | 
 | 618 | 	vp_dev = kzalloc(sizeof(struct virtio_pci_device), GFP_KERNEL); | 
 | 619 | 	if (vp_dev == NULL) | 
 | 620 | 		return -ENOMEM; | 
 | 621 |  | 
| Milton Miller | 8b3bb3e | 2011-01-07 02:55:06 -0600 | [diff] [blame] | 622 | 	vp_dev->vdev.dev.parent = &pci_dev->dev; | 
| Mark McLoughlin | 29f9f12 | 2008-12-10 17:45:34 +0000 | [diff] [blame] | 623 | 	vp_dev->vdev.dev.release = virtio_pci_release_dev; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 624 | 	vp_dev->vdev.config = &virtio_pci_config_ops; | 
 | 625 | 	vp_dev->pci_dev = pci_dev; | 
 | 626 | 	INIT_LIST_HEAD(&vp_dev->virtqueues); | 
 | 627 | 	spin_lock_init(&vp_dev->lock); | 
 | 628 |  | 
| Michael S. Tsirkin | b03214d | 2010-06-23 22:49:06 -0600 | [diff] [blame] | 629 | 	/* Disable MSI/MSIX to bring device to a known good state. */ | 
 | 630 | 	pci_msi_off(pci_dev); | 
 | 631 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 632 | 	/* enable the device */ | 
 | 633 | 	err = pci_enable_device(pci_dev); | 
 | 634 | 	if (err) | 
 | 635 | 		goto out; | 
 | 636 |  | 
 | 637 | 	err = pci_request_regions(pci_dev, "virtio-pci"); | 
 | 638 | 	if (err) | 
 | 639 | 		goto out_enable_device; | 
 | 640 |  | 
 | 641 | 	vp_dev->ioaddr = pci_iomap(pci_dev, 0, 0); | 
 | 642 | 	if (vp_dev->ioaddr == NULL) | 
 | 643 | 		goto out_req_regions; | 
 | 644 |  | 
 | 645 | 	pci_set_drvdata(pci_dev, vp_dev); | 
| Michael S. Tsirkin | bc505f3 | 2009-11-29 17:52:00 +0200 | [diff] [blame] | 646 | 	pci_set_master(pci_dev); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 647 |  | 
 | 648 | 	/* we use the subsystem vendor/device id as the virtio vendor/device | 
 | 649 | 	 * id.  this allows us to use the same PCI vendor/device id for all | 
 | 650 | 	 * virtio devices and to identify the particular virtio driver by | 
| Thomas Weber | 8839316 | 2010-03-16 11:47:56 +0100 | [diff] [blame] | 651 | 	 * the subsystem ids */ | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 652 | 	vp_dev->vdev.id.vendor = pci_dev->subsystem_vendor; | 
 | 653 | 	vp_dev->vdev.id.device = pci_dev->subsystem_device; | 
 | 654 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 655 | 	/* finally register the virtio device */ | 
 | 656 | 	err = register_virtio_device(&vp_dev->vdev); | 
 | 657 | 	if (err) | 
| Michael S. Tsirkin | 82af8ce | 2009-05-14 13:55:41 +0300 | [diff] [blame] | 658 | 		goto out_set_drvdata; | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 659 |  | 
 | 660 | 	return 0; | 
 | 661 |  | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 662 | out_set_drvdata: | 
 | 663 | 	pci_set_drvdata(pci_dev, NULL); | 
 | 664 | 	pci_iounmap(pci_dev, vp_dev->ioaddr); | 
 | 665 | out_req_regions: | 
 | 666 | 	pci_release_regions(pci_dev); | 
 | 667 | out_enable_device: | 
 | 668 | 	pci_disable_device(pci_dev); | 
 | 669 | out: | 
 | 670 | 	kfree(vp_dev); | 
 | 671 | 	return err; | 
 | 672 | } | 
 | 673 |  | 
 | 674 | static void __devexit virtio_pci_remove(struct pci_dev *pci_dev) | 
 | 675 | { | 
 | 676 | 	struct virtio_pci_device *vp_dev = pci_get_drvdata(pci_dev); | 
 | 677 |  | 
| Anthony Liguori | bd6c269 | 2008-03-19 20:35:04 -0500 | [diff] [blame] | 678 | 	unregister_virtio_device(&vp_dev->vdev); | 
| Amit Shah | 31a3ddd | 2011-03-14 17:45:02 +0530 | [diff] [blame] | 679 |  | 
 | 680 | 	vp_del_vqs(&vp_dev->vdev); | 
 | 681 | 	pci_set_drvdata(pci_dev, NULL); | 
 | 682 | 	pci_iounmap(pci_dev, vp_dev->ioaddr); | 
 | 683 | 	pci_release_regions(pci_dev); | 
 | 684 | 	pci_disable_device(pci_dev); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 685 | } | 
 | 686 |  | 
 | 687 | #ifdef CONFIG_PM | 
 | 688 | static int virtio_pci_suspend(struct pci_dev *pci_dev, pm_message_t state) | 
 | 689 | { | 
 | 690 | 	pci_save_state(pci_dev); | 
 | 691 | 	pci_set_power_state(pci_dev, PCI_D3hot); | 
 | 692 | 	return 0; | 
 | 693 | } | 
 | 694 |  | 
 | 695 | static int virtio_pci_resume(struct pci_dev *pci_dev) | 
 | 696 | { | 
 | 697 | 	pci_restore_state(pci_dev); | 
 | 698 | 	pci_set_power_state(pci_dev, PCI_D0); | 
 | 699 | 	return 0; | 
 | 700 | } | 
 | 701 | #endif | 
 | 702 |  | 
 | 703 | static struct pci_driver virtio_pci_driver = { | 
 | 704 | 	.name		= "virtio-pci", | 
 | 705 | 	.id_table	= virtio_pci_id_table, | 
 | 706 | 	.probe		= virtio_pci_probe, | 
| Jamie Lokier | 1f08b83 | 2010-01-08 22:01:43 +0000 | [diff] [blame] | 707 | 	.remove		= __devexit_p(virtio_pci_remove), | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 708 | #ifdef CONFIG_PM | 
 | 709 | 	.suspend	= virtio_pci_suspend, | 
 | 710 | 	.resume		= virtio_pci_resume, | 
 | 711 | #endif | 
 | 712 | }; | 
 | 713 |  | 
 | 714 | static int __init virtio_pci_init(void) | 
 | 715 | { | 
| Milton Miller | 8b3bb3e | 2011-01-07 02:55:06 -0600 | [diff] [blame] | 716 | 	return pci_register_driver(&virtio_pci_driver); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 717 | } | 
 | 718 |  | 
 | 719 | module_init(virtio_pci_init); | 
 | 720 |  | 
 | 721 | static void __exit virtio_pci_exit(void) | 
 | 722 | { | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 723 | 	pci_unregister_driver(&virtio_pci_driver); | 
| Anthony Liguori | 3343660 | 2007-11-12 21:30:26 -0600 | [diff] [blame] | 724 | } | 
 | 725 |  | 
 | 726 | module_exit(virtio_pci_exit); |