Commit | Line | Data |
---|---|---|
3343660d | 1 | /* |
a90fdce9 | 2 | * Virtio PCI driver - common functionality for all device versions |
3343660d AL |
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 | |
a90fdce9 | 8 | * Copyright Red Hat, Inc. 2014 |
3343660d AL |
9 | * |
10 | * Authors: | |
11 | * Anthony Liguori <aliguori@us.ibm.com> | |
a90fdce9 MT |
12 | * Rusty Russell <rusty@rustcorp.com.au> |
13 | * Michael S. Tsirkin <mst@redhat.com> | |
3343660d AL |
14 | * |
15 | * This work is licensed under the terms of the GNU GPL, version 2 or later. | |
16 | * See the COPYING file in the top-level directory. | |
17 | * | |
18 | */ | |
19 | ||
5f4c9760 | 20 | #include "virtio_pci_common.h" |
3343660d | 21 | |
ac399d8f MT |
22 | static bool force_legacy = false; |
23 | ||
24 | #if IS_ENABLED(CONFIG_VIRTIO_PCI_LEGACY) | |
25 | module_param(force_legacy, bool, 0444); | |
26 | MODULE_PARM_DESC(force_legacy, | |
27 | "Force legacy mode for transitional virtio 1 devices"); | |
28 | #endif | |
29 | ||
e6af578c | 30 | /* wait for pending irq handlers */ |
38eb4a29 | 31 | void vp_synchronize_vectors(struct virtio_device *vdev) |
e6af578c MT |
32 | { |
33 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
34 | int i; | |
35 | ||
36 | if (vp_dev->intx_enabled) | |
37 | synchronize_irq(vp_dev->pci_dev->irq); | |
38 | ||
39 | for (i = 0; i < vp_dev->msix_vectors; ++i) | |
40 | synchronize_irq(vp_dev->msix_entries[i].vector); | |
41 | } | |
42 | ||
3343660d | 43 | /* the notify function used when creating a virt queue */ |
38eb4a29 | 44 | bool vp_notify(struct virtqueue *vq) |
3343660d | 45 | { |
3343660d AL |
46 | /* we write the queue's selector into the notification register to |
47 | * signal the other end */ | |
f30eaf4a | 48 | iowrite16(vq->index, (void __iomem *)vq->priv); |
46f9c2b9 | 49 | return true; |
3343660d AL |
50 | } |
51 | ||
77cf5246 MT |
52 | /* Handle a configuration change: Tell driver if it wants to know. */ |
53 | static irqreturn_t vp_config_changed(int irq, void *opaque) | |
54 | { | |
55 | struct virtio_pci_device *vp_dev = opaque; | |
77cf5246 | 56 | |
016c98c6 | 57 | virtio_config_changed(&vp_dev->vdev); |
77cf5246 MT |
58 | return IRQ_HANDLED; |
59 | } | |
60 | ||
61 | /* Notify all virtqueues on an interrupt. */ | |
62 | static irqreturn_t vp_vring_interrupt(int irq, void *opaque) | |
63 | { | |
64 | struct virtio_pci_device *vp_dev = opaque; | |
65 | struct virtio_pci_vq_info *info; | |
66 | irqreturn_t ret = IRQ_NONE; | |
67 | unsigned long flags; | |
68 | ||
69 | spin_lock_irqsave(&vp_dev->lock, flags); | |
70 | list_for_each_entry(info, &vp_dev->virtqueues, node) { | |
71 | if (vring_interrupt(irq, info->vq) == IRQ_HANDLED) | |
72 | ret = IRQ_HANDLED; | |
73 | } | |
74 | spin_unlock_irqrestore(&vp_dev->lock, flags); | |
75 | ||
76 | return ret; | |
77 | } | |
78 | ||
3343660d AL |
79 | /* A small wrapper to also acknowledge the interrupt when it's handled. |
80 | * I really need an EIO hook for the vring so I can ack the interrupt once we | |
81 | * know that we'll be handling the IRQ but before we invoke the callback since | |
82 | * the callback may notify the host which results in the host attempting to | |
83 | * raise an interrupt that we would then mask once we acknowledged the | |
84 | * interrupt. */ | |
85 | static irqreturn_t vp_interrupt(int irq, void *opaque) | |
86 | { | |
87 | struct virtio_pci_device *vp_dev = opaque; | |
3343660d AL |
88 | u8 isr; |
89 | ||
90 | /* reading the ISR has the effect of also clearing it so it's very | |
91 | * important to save off the value. */ | |
af535722 | 92 | isr = ioread8(vp_dev->isr); |
3343660d AL |
93 | |
94 | /* It's definitely not us if the ISR was not high */ | |
95 | if (!isr) | |
96 | return IRQ_NONE; | |
97 | ||
98 | /* Configuration change? Tell driver if it wants to know. */ | |
77cf5246 MT |
99 | if (isr & VIRTIO_PCI_ISR_CONFIG) |
100 | vp_config_changed(irq, opaque); | |
3343660d | 101 | |
77cf5246 | 102 | return vp_vring_interrupt(irq, opaque); |
3343660d AL |
103 | } |
104 | ||
82af8ce8 MT |
105 | static void vp_free_vectors(struct virtio_device *vdev) |
106 | { | |
107 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
108 | int i; | |
109 | ||
110 | if (vp_dev->intx_enabled) { | |
111 | free_irq(vp_dev->pci_dev->irq, vp_dev); | |
112 | vp_dev->intx_enabled = 0; | |
113 | } | |
114 | ||
115 | for (i = 0; i < vp_dev->msix_used_vectors; ++i) | |
116 | free_irq(vp_dev->msix_entries[i].vector, vp_dev); | |
82af8ce8 | 117 | |
75a0a52b JW |
118 | for (i = 0; i < vp_dev->msix_vectors; i++) |
119 | if (vp_dev->msix_affinity_masks[i]) | |
120 | free_cpumask_var(vp_dev->msix_affinity_masks[i]); | |
121 | ||
82af8ce8 MT |
122 | if (vp_dev->msix_enabled) { |
123 | /* Disable the vector used for configuration */ | |
6f8f23d6 | 124 | vp_dev->config_vector(vp_dev, VIRTIO_MSI_NO_VECTOR); |
82af8ce8 | 125 | |
82af8ce8 | 126 | pci_disable_msix(vp_dev->pci_dev); |
ff52c3fc | 127 | vp_dev->msix_enabled = 0; |
82af8ce8 | 128 | } |
ff52c3fc | 129 | |
f11335db | 130 | vp_dev->msix_vectors = 0; |
ff52c3fc MT |
131 | vp_dev->msix_used_vectors = 0; |
132 | kfree(vp_dev->msix_names); | |
133 | vp_dev->msix_names = NULL; | |
134 | kfree(vp_dev->msix_entries); | |
135 | vp_dev->msix_entries = NULL; | |
75a0a52b JW |
136 | kfree(vp_dev->msix_affinity_masks); |
137 | vp_dev->msix_affinity_masks = NULL; | |
82af8ce8 MT |
138 | } |
139 | ||
f68d2408 RR |
140 | static int vp_request_msix_vectors(struct virtio_device *vdev, int nvectors, |
141 | bool per_vq_vectors) | |
82af8ce8 MT |
142 | { |
143 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
144 | const char *name = dev_name(&vp_dev->vdev.dev); | |
145 | unsigned i, v; | |
146 | int err = -ENOMEM; | |
e969fed5 | 147 | |
f11335db AV |
148 | vp_dev->msix_vectors = nvectors; |
149 | ||
82af8ce8 MT |
150 | vp_dev->msix_entries = kmalloc(nvectors * sizeof *vp_dev->msix_entries, |
151 | GFP_KERNEL); | |
152 | if (!vp_dev->msix_entries) | |
ff52c3fc | 153 | goto error; |
82af8ce8 MT |
154 | vp_dev->msix_names = kmalloc(nvectors * sizeof *vp_dev->msix_names, |
155 | GFP_KERNEL); | |
156 | if (!vp_dev->msix_names) | |
ff52c3fc | 157 | goto error; |
75a0a52b JW |
158 | vp_dev->msix_affinity_masks |
159 | = kzalloc(nvectors * sizeof *vp_dev->msix_affinity_masks, | |
160 | GFP_KERNEL); | |
161 | if (!vp_dev->msix_affinity_masks) | |
162 | goto error; | |
163 | for (i = 0; i < nvectors; ++i) | |
164 | if (!alloc_cpumask_var(&vp_dev->msix_affinity_masks[i], | |
165 | GFP_KERNEL)) | |
166 | goto error; | |
82af8ce8 MT |
167 | |
168 | for (i = 0; i < nvectors; ++i) | |
169 | vp_dev->msix_entries[i].entry = i; | |
170 | ||
5e37f670 AG |
171 | err = pci_enable_msix_exact(vp_dev->pci_dev, |
172 | vp_dev->msix_entries, nvectors); | |
e969fed5 MT |
173 | if (err) |
174 | goto error; | |
e969fed5 MT |
175 | vp_dev->msix_enabled = 1; |
176 | ||
177 | /* Set the vector used for configuration */ | |
178 | v = vp_dev->msix_used_vectors; | |
179 | snprintf(vp_dev->msix_names[v], sizeof *vp_dev->msix_names, | |
180 | "%s-config", name); | |
181 | err = request_irq(vp_dev->msix_entries[v].vector, | |
182 | vp_config_changed, 0, vp_dev->msix_names[v], | |
183 | vp_dev); | |
184 | if (err) | |
185 | goto error; | |
186 | ++vp_dev->msix_used_vectors; | |
82af8ce8 | 187 | |
6f8f23d6 | 188 | v = vp_dev->config_vector(vp_dev, v); |
e969fed5 | 189 | /* Verify we had enough resources to assign the vector */ |
e969fed5 MT |
190 | if (v == VIRTIO_MSI_NO_VECTOR) { |
191 | err = -EBUSY; | |
192 | goto error; | |
82af8ce8 MT |
193 | } |
194 | ||
e969fed5 | 195 | if (!per_vq_vectors) { |
82af8ce8 MT |
196 | /* Shared vector for all VQs */ |
197 | v = vp_dev->msix_used_vectors; | |
198 | snprintf(vp_dev->msix_names[v], sizeof *vp_dev->msix_names, | |
199 | "%s-virtqueues", name); | |
200 | err = request_irq(vp_dev->msix_entries[v].vector, | |
201 | vp_vring_interrupt, 0, vp_dev->msix_names[v], | |
202 | vp_dev); | |
203 | if (err) | |
ff52c3fc | 204 | goto error; |
82af8ce8 MT |
205 | ++vp_dev->msix_used_vectors; |
206 | } | |
207 | return 0; | |
ff52c3fc | 208 | error: |
82af8ce8 | 209 | vp_free_vectors(vdev); |
82af8ce8 MT |
210 | return err; |
211 | } | |
212 | ||
f68d2408 RR |
213 | static int vp_request_intx(struct virtio_device *vdev) |
214 | { | |
215 | int err; | |
216 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
217 | ||
218 | err = request_irq(vp_dev->pci_dev->irq, vp_interrupt, | |
219 | IRQF_SHARED, dev_name(&vdev->dev), vp_dev); | |
220 | if (!err) | |
221 | vp_dev->intx_enabled = 1; | |
222 | return err; | |
223 | } | |
224 | ||
b09f00bb MT |
225 | static struct virtqueue *vp_setup_vq(struct virtio_device *vdev, unsigned index, |
226 | void (*callback)(struct virtqueue *vq), | |
227 | const char *name, | |
228 | u16 msix_vec) | |
229 | { | |
230 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
231 | struct virtio_pci_vq_info *info = kmalloc(sizeof *info, GFP_KERNEL); | |
232 | struct virtqueue *vq; | |
233 | unsigned long flags; | |
234 | ||
235 | /* fill out our structure that represents an active queue */ | |
236 | if (!info) | |
237 | return ERR_PTR(-ENOMEM); | |
238 | ||
239 | vq = vp_dev->setup_vq(vp_dev, info, index, callback, name, msix_vec); | |
240 | if (IS_ERR(vq)) | |
241 | goto out_info; | |
242 | ||
243 | info->vq = vq; | |
005b20a8 KK |
244 | if (callback) { |
245 | spin_lock_irqsave(&vp_dev->lock, flags); | |
246 | list_add(&info->node, &vp_dev->virtqueues); | |
247 | spin_unlock_irqrestore(&vp_dev->lock, flags); | |
248 | } else { | |
249 | INIT_LIST_HEAD(&info->node); | |
250 | } | |
3343660d | 251 | |
3ec7a77b | 252 | vp_dev->vqs[index] = info; |
3343660d AL |
253 | return vq; |
254 | ||
3343660d AL |
255 | out_info: |
256 | kfree(info); | |
b09f00bb | 257 | return vq; |
3343660d AL |
258 | } |
259 | ||
5386cef2 MT |
260 | static void vp_del_vq(struct virtqueue *vq) |
261 | { | |
262 | struct virtio_pci_device *vp_dev = to_vp_device(vq->vdev); | |
263 | struct virtio_pci_vq_info *info = vp_dev->vqs[vq->index]; | |
264 | unsigned long flags; | |
265 | ||
266 | spin_lock_irqsave(&vp_dev->lock, flags); | |
267 | list_del(&info->node); | |
268 | spin_unlock_irqrestore(&vp_dev->lock, flags); | |
269 | ||
270 | vp_dev->del_vq(info); | |
3343660d AL |
271 | kfree(info); |
272 | } | |
273 | ||
82af8ce8 | 274 | /* the config->del_vqs() implementation */ |
38eb4a29 | 275 | void vp_del_vqs(struct virtio_device *vdev) |
d2a7ddda | 276 | { |
e969fed5 | 277 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); |
d2a7ddda | 278 | struct virtqueue *vq, *n; |
e969fed5 | 279 | struct virtio_pci_vq_info *info; |
d2a7ddda | 280 | |
e969fed5 | 281 | list_for_each_entry_safe(vq, n, &vdev->vqs, list) { |
3ec7a77b | 282 | info = vp_dev->vqs[vq->index]; |
31198159 MT |
283 | if (vp_dev->per_vq_vectors && |
284 | info->msix_vector != VIRTIO_MSI_NO_VECTOR) | |
f68d2408 RR |
285 | free_irq(vp_dev->msix_entries[info->msix_vector].vector, |
286 | vq); | |
d2a7ddda | 287 | vp_del_vq(vq); |
e969fed5 MT |
288 | } |
289 | vp_dev->per_vq_vectors = false; | |
82af8ce8 MT |
290 | |
291 | vp_free_vectors(vdev); | |
3ec7a77b | 292 | kfree(vp_dev->vqs); |
80e9541f | 293 | vp_dev->vqs = NULL; |
d2a7ddda MT |
294 | } |
295 | ||
e969fed5 MT |
296 | static int vp_try_to_find_vqs(struct virtio_device *vdev, unsigned nvqs, |
297 | struct virtqueue *vqs[], | |
298 | vq_callback_t *callbacks[], | |
f7ad26ff | 299 | const char * const names[], |
f68d2408 | 300 | bool use_msix, |
e969fed5 | 301 | bool per_vq_vectors) |
d2a7ddda | 302 | { |
e969fed5 | 303 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); |
f68d2408 RR |
304 | u16 msix_vec; |
305 | int i, err, nvectors, allocated_vectors; | |
82af8ce8 | 306 | |
3ec7a77b MT |
307 | vp_dev->vqs = kmalloc(nvqs * sizeof *vp_dev->vqs, GFP_KERNEL); |
308 | if (!vp_dev->vqs) | |
309 | return -ENOMEM; | |
310 | ||
f68d2408 RR |
311 | if (!use_msix) { |
312 | /* Old style: one normal interrupt for change and all vqs. */ | |
313 | err = vp_request_intx(vdev); | |
314 | if (err) | |
3ec7a77b | 315 | goto error_find; |
f68d2408 RR |
316 | } else { |
317 | if (per_vq_vectors) { | |
318 | /* Best option: one for change interrupt, one per vq. */ | |
319 | nvectors = 1; | |
320 | for (i = 0; i < nvqs; ++i) | |
321 | if (callbacks[i]) | |
322 | ++nvectors; | |
323 | } else { | |
324 | /* Second best: one for change, shared for all vqs. */ | |
325 | nvectors = 2; | |
326 | } | |
327 | ||
328 | err = vp_request_msix_vectors(vdev, nvectors, per_vq_vectors); | |
329 | if (err) | |
3ec7a77b | 330 | goto error_find; |
f68d2408 | 331 | } |
d2a7ddda | 332 | |
e969fed5 MT |
333 | vp_dev->per_vq_vectors = per_vq_vectors; |
334 | allocated_vectors = vp_dev->msix_used_vectors; | |
d2a7ddda | 335 | for (i = 0; i < nvqs; ++i) { |
6457f126 MT |
336 | if (!names[i]) { |
337 | vqs[i] = NULL; | |
338 | continue; | |
339 | } else if (!callbacks[i] || !vp_dev->msix_enabled) | |
f68d2408 | 340 | msix_vec = VIRTIO_MSI_NO_VECTOR; |
e969fed5 | 341 | else if (vp_dev->per_vq_vectors) |
f68d2408 | 342 | msix_vec = allocated_vectors++; |
e969fed5 | 343 | else |
f68d2408 | 344 | msix_vec = VP_MSIX_VQ_VECTOR; |
b09f00bb | 345 | vqs[i] = vp_setup_vq(vdev, i, callbacks[i], names[i], msix_vec); |
e969fed5 MT |
346 | if (IS_ERR(vqs[i])) { |
347 | err = PTR_ERR(vqs[i]); | |
82af8ce8 | 348 | goto error_find; |
e969fed5 | 349 | } |
0b22bd0b MT |
350 | |
351 | if (!vp_dev->per_vq_vectors || msix_vec == VIRTIO_MSI_NO_VECTOR) | |
352 | continue; | |
353 | ||
e969fed5 | 354 | /* allocate per-vq irq if available and necessary */ |
0b22bd0b MT |
355 | snprintf(vp_dev->msix_names[msix_vec], |
356 | sizeof *vp_dev->msix_names, | |
357 | "%s-%s", | |
358 | dev_name(&vp_dev->vdev.dev), names[i]); | |
359 | err = request_irq(vp_dev->msix_entries[msix_vec].vector, | |
360 | vring_interrupt, 0, | |
361 | vp_dev->msix_names[msix_vec], | |
362 | vqs[i]); | |
363 | if (err) { | |
364 | vp_del_vq(vqs[i]); | |
365 | goto error_find; | |
e969fed5 | 366 | } |
d2a7ddda MT |
367 | } |
368 | return 0; | |
369 | ||
82af8ce8 | 370 | error_find: |
d2a7ddda | 371 | vp_del_vqs(vdev); |
e969fed5 MT |
372 | return err; |
373 | } | |
374 | ||
375 | /* the config->find_vqs() implementation */ | |
38eb4a29 MT |
376 | int vp_find_vqs(struct virtio_device *vdev, unsigned nvqs, |
377 | struct virtqueue *vqs[], | |
378 | vq_callback_t *callbacks[], | |
f7ad26ff | 379 | const char * const names[]) |
e969fed5 | 380 | { |
f68d2408 | 381 | int err; |
e969fed5 | 382 | |
f68d2408 RR |
383 | /* Try MSI-X with one vector per queue. */ |
384 | err = vp_try_to_find_vqs(vdev, nvqs, vqs, callbacks, names, true, true); | |
e969fed5 MT |
385 | if (!err) |
386 | return 0; | |
f68d2408 | 387 | /* Fallback: MSI-X with one vector for config, one shared for queues. */ |
e969fed5 | 388 | err = vp_try_to_find_vqs(vdev, nvqs, vqs, callbacks, names, |
f68d2408 | 389 | true, false); |
e969fed5 MT |
390 | if (!err) |
391 | return 0; | |
392 | /* Finally fall back to regular interrupts. */ | |
f68d2408 RR |
393 | return vp_try_to_find_vqs(vdev, nvqs, vqs, callbacks, names, |
394 | false, false); | |
d2a7ddda MT |
395 | } |
396 | ||
38eb4a29 | 397 | const char *vp_bus_name(struct virtio_device *vdev) |
66846048 RJ |
398 | { |
399 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
400 | ||
401 | return pci_name(vp_dev->pci_dev); | |
402 | } | |
403 | ||
75a0a52b JW |
404 | /* Setup the affinity for a virtqueue: |
405 | * - force the affinity for per vq vector | |
406 | * - OR over all affinities for shared MSI | |
407 | * - ignore the affinity request if we're using INTX | |
408 | */ | |
38eb4a29 | 409 | int vp_set_vq_affinity(struct virtqueue *vq, int cpu) |
75a0a52b JW |
410 | { |
411 | struct virtio_device *vdev = vq->vdev; | |
412 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
3ec7a77b | 413 | struct virtio_pci_vq_info *info = vp_dev->vqs[vq->index]; |
75a0a52b JW |
414 | struct cpumask *mask; |
415 | unsigned int irq; | |
416 | ||
417 | if (!vq->callback) | |
418 | return -EINVAL; | |
419 | ||
420 | if (vp_dev->msix_enabled) { | |
421 | mask = vp_dev->msix_affinity_masks[info->msix_vector]; | |
422 | irq = vp_dev->msix_entries[info->msix_vector].vector; | |
423 | if (cpu == -1) | |
424 | irq_set_affinity_hint(irq, NULL); | |
425 | else { | |
210d150e | 426 | cpumask_clear(mask); |
75a0a52b JW |
427 | cpumask_set_cpu(cpu, mask); |
428 | irq_set_affinity_hint(irq, mask); | |
429 | } | |
430 | } | |
431 | return 0; | |
432 | } | |
433 | ||
9e266ece | 434 | #ifdef CONFIG_PM_SLEEP |
f0fe6f11 AS |
435 | static int virtio_pci_freeze(struct device *dev) |
436 | { | |
437 | struct pci_dev *pci_dev = to_pci_dev(dev); | |
438 | struct virtio_pci_device *vp_dev = pci_get_drvdata(pci_dev); | |
f0fe6f11 AS |
439 | int ret; |
440 | ||
c6716bae | 441 | ret = virtio_device_freeze(&vp_dev->vdev); |
f0fe6f11 AS |
442 | |
443 | if (!ret) | |
444 | pci_disable_device(pci_dev); | |
445 | return ret; | |
446 | } | |
447 | ||
0517fdd1 | 448 | static int virtio_pci_restore(struct device *dev) |
f0fe6f11 AS |
449 | { |
450 | struct pci_dev *pci_dev = to_pci_dev(dev); | |
451 | struct virtio_pci_device *vp_dev = pci_get_drvdata(pci_dev); | |
452 | int ret; | |
453 | ||
454 | ret = pci_enable_device(pci_dev); | |
455 | if (ret) | |
456 | return ret; | |
0517fdd1 | 457 | |
f0fe6f11 | 458 | pci_set_master(pci_dev); |
c6716bae | 459 | return virtio_device_restore(&vp_dev->vdev); |
f0fe6f11 AS |
460 | } |
461 | ||
9a4253db | 462 | static const struct dev_pm_ops virtio_pci_pm_ops = { |
f878d0be | 463 | SET_SYSTEM_SLEEP_PM_OPS(virtio_pci_freeze, virtio_pci_restore) |
d0775363 | 464 | }; |
3343660d | 465 | #endif |
9a4253db MT |
466 | |
467 | ||
468 | /* Qumranet donated their vendor ID for devices 0x1000 thru 0x10FF. */ | |
469 | static const struct pci_device_id virtio_pci_id_table[] = { | |
caf02abf | 470 | { PCI_DEVICE(PCI_VENDOR_ID_REDHAT_QUMRANET, PCI_ANY_ID) }, |
9a4253db MT |
471 | { 0 } |
472 | }; | |
473 | ||
474 | MODULE_DEVICE_TABLE(pci, virtio_pci_id_table); | |
475 | ||
ff31d2e2 MT |
476 | static void virtio_pci_release_dev(struct device *_d) |
477 | { | |
478 | struct virtio_device *vdev = dev_to_virtio(_d); | |
479 | struct virtio_pci_device *vp_dev = to_vp_device(vdev); | |
480 | ||
481 | /* As struct device is a kobject, it's not safe to | |
482 | * free the memory (including the reference counter itself) | |
483 | * until it's release callback. */ | |
484 | kfree(vp_dev); | |
485 | } | |
486 | ||
9a4253db MT |
487 | static int virtio_pci_probe(struct pci_dev *pci_dev, |
488 | const struct pci_device_id *id) | |
489 | { | |
ff31d2e2 MT |
490 | struct virtio_pci_device *vp_dev; |
491 | int rc; | |
492 | ||
493 | /* allocate our structure and fill it out */ | |
494 | vp_dev = kzalloc(sizeof(struct virtio_pci_device), GFP_KERNEL); | |
495 | if (!vp_dev) | |
496 | return -ENOMEM; | |
497 | ||
498 | pci_set_drvdata(pci_dev, vp_dev); | |
499 | vp_dev->vdev.dev.parent = &pci_dev->dev; | |
500 | vp_dev->vdev.dev.release = virtio_pci_release_dev; | |
501 | vp_dev->pci_dev = pci_dev; | |
502 | INIT_LIST_HEAD(&vp_dev->virtqueues); | |
503 | spin_lock_init(&vp_dev->lock); | |
504 | ||
ff31d2e2 MT |
505 | /* enable the device */ |
506 | rc = pci_enable_device(pci_dev); | |
507 | if (rc) | |
508 | goto err_enable_device; | |
509 | ||
ac399d8f | 510 | if (force_legacy) { |
1fcf0512 | 511 | rc = virtio_pci_legacy_probe(vp_dev); |
ac399d8f MT |
512 | /* Also try modern mode if we can't map BAR0 (no IO space). */ |
513 | if (rc == -ENODEV || rc == -ENOMEM) | |
514 | rc = virtio_pci_modern_probe(vp_dev); | |
515 | if (rc) | |
516 | goto err_probe; | |
517 | } else { | |
518 | rc = virtio_pci_modern_probe(vp_dev); | |
519 | if (rc == -ENODEV) | |
520 | rc = virtio_pci_legacy_probe(vp_dev); | |
521 | if (rc) | |
522 | goto err_probe; | |
523 | } | |
ff31d2e2 MT |
524 | |
525 | pci_set_master(pci_dev); | |
526 | ||
527 | rc = register_virtio_device(&vp_dev->vdev); | |
528 | if (rc) | |
529 | goto err_register; | |
530 | ||
531 | return 0; | |
532 | ||
533 | err_register: | |
1fcf0512 MT |
534 | if (vp_dev->ioaddr) |
535 | virtio_pci_legacy_remove(vp_dev); | |
536 | else | |
537 | virtio_pci_modern_remove(vp_dev); | |
ff31d2e2 | 538 | err_probe: |
ff31d2e2 MT |
539 | pci_disable_device(pci_dev); |
540 | err_enable_device: | |
541 | kfree(vp_dev); | |
542 | return rc; | |
9a4253db MT |
543 | } |
544 | ||
545 | static void virtio_pci_remove(struct pci_dev *pci_dev) | |
546 | { | |
ff31d2e2 | 547 | struct virtio_pci_device *vp_dev = pci_get_drvdata(pci_dev); |
2989be09 | 548 | struct device *dev = get_device(&vp_dev->vdev.dev); |
ff31d2e2 MT |
549 | |
550 | unregister_virtio_device(&vp_dev->vdev); | |
551 | ||
1fcf0512 MT |
552 | if (vp_dev->ioaddr) |
553 | virtio_pci_legacy_remove(vp_dev); | |
554 | else | |
555 | virtio_pci_modern_remove(vp_dev); | |
ff31d2e2 | 556 | |
ff31d2e2 | 557 | pci_disable_device(pci_dev); |
2989be09 | 558 | put_device(dev); |
9a4253db MT |
559 | } |
560 | ||
561 | static struct pci_driver virtio_pci_driver = { | |
562 | .name = "virtio-pci", | |
563 | .id_table = virtio_pci_id_table, | |
564 | .probe = virtio_pci_probe, | |
565 | .remove = virtio_pci_remove, | |
566 | #ifdef CONFIG_PM_SLEEP | |
567 | .driver.pm = &virtio_pci_pm_ops, | |
568 | #endif | |
569 | }; | |
570 | ||
571 | module_pci_driver(virtio_pci_driver); | |
5ff16110 HX |
572 | |
573 | MODULE_AUTHOR("Anthony Liguori <aliguori@us.ibm.com>"); | |
574 | MODULE_DESCRIPTION("virtio-pci"); | |
575 | MODULE_LICENSE("GPL"); | |
576 | MODULE_VERSION("1"); |