]> Git Repo - qemu.git/blob - hw/virtio/virtio.c
Merge remote-tracking branch 'jasowang/tags/net-pull-request' into staging
[qemu.git] / hw / virtio / virtio.c
1 /*
2  * Virtio Support
3  *
4  * Copyright IBM, Corp. 2007
5  *
6  * Authors:
7  *  Anthony Liguori   <[email protected]>
8  *
9  * This work is licensed under the terms of the GNU GPL, version 2.  See
10  * the COPYING file in the top-level directory.
11  *
12  */
13
14 #include "qemu/osdep.h"
15 #include "qapi/error.h"
16 #include "qemu-common.h"
17 #include "cpu.h"
18 #include "trace.h"
19 #include "exec/address-spaces.h"
20 #include "qemu/error-report.h"
21 #include "hw/virtio/virtio.h"
22 #include "qemu/atomic.h"
23 #include "hw/virtio/virtio-bus.h"
24 #include "hw/virtio/virtio-access.h"
25 #include "sysemu/dma.h"
26
27 /*
28  * The alignment to use between consumer and producer parts of vring.
29  * x86 pagesize again. This is the default, used by transports like PCI
30  * which don't provide a means for the guest to tell the host the alignment.
31  */
32 #define VIRTIO_PCI_VRING_ALIGN         4096
33
34 typedef struct VRingDesc
35 {
36     uint64_t addr;
37     uint32_t len;
38     uint16_t flags;
39     uint16_t next;
40 } VRingDesc;
41
42 typedef struct VRingAvail
43 {
44     uint16_t flags;
45     uint16_t idx;
46     uint16_t ring[0];
47 } VRingAvail;
48
49 typedef struct VRingUsedElem
50 {
51     uint32_t id;
52     uint32_t len;
53 } VRingUsedElem;
54
55 typedef struct VRingUsed
56 {
57     uint16_t flags;
58     uint16_t idx;
59     VRingUsedElem ring[0];
60 } VRingUsed;
61
62 typedef struct VRingMemoryRegionCaches {
63     struct rcu_head rcu;
64     MemoryRegionCache desc;
65     MemoryRegionCache avail;
66     MemoryRegionCache used;
67 } VRingMemoryRegionCaches;
68
69 typedef struct VRing
70 {
71     unsigned int num;
72     unsigned int num_default;
73     unsigned int align;
74     hwaddr desc;
75     hwaddr avail;
76     hwaddr used;
77     VRingMemoryRegionCaches *caches;
78 } VRing;
79
80 struct VirtQueue
81 {
82     VRing vring;
83
84     /* Next head to pop */
85     uint16_t last_avail_idx;
86
87     /* Last avail_idx read from VQ. */
88     uint16_t shadow_avail_idx;
89
90     uint16_t used_idx;
91
92     /* Last used index value we have signalled on */
93     uint16_t signalled_used;
94
95     /* Last used index value we have signalled on */
96     bool signalled_used_valid;
97
98     /* Notification enabled? */
99     bool notification;
100
101     uint16_t queue_index;
102
103     unsigned int inuse;
104
105     uint16_t vector;
106     VirtIOHandleOutput handle_output;
107     VirtIOHandleAIOOutput handle_aio_output;
108     VirtIODevice *vdev;
109     EventNotifier guest_notifier;
110     EventNotifier host_notifier;
111     QLIST_ENTRY(VirtQueue) node;
112 };
113
114 static void virtio_free_region_cache(VRingMemoryRegionCaches *caches)
115 {
116     if (!caches) {
117         return;
118     }
119
120     address_space_cache_destroy(&caches->desc);
121     address_space_cache_destroy(&caches->avail);
122     address_space_cache_destroy(&caches->used);
123     g_free(caches);
124 }
125
126 static void virtio_init_region_cache(VirtIODevice *vdev, int n)
127 {
128     VirtQueue *vq = &vdev->vq[n];
129     VRingMemoryRegionCaches *old = vq->vring.caches;
130     VRingMemoryRegionCaches *new;
131     hwaddr addr, size;
132     int event_size;
133     int64_t len;
134
135     event_size = virtio_vdev_has_feature(vq->vdev, VIRTIO_RING_F_EVENT_IDX) ? 2 : 0;
136
137     addr = vq->vring.desc;
138     if (!addr) {
139         return;
140     }
141     new = g_new0(VRingMemoryRegionCaches, 1);
142     size = virtio_queue_get_desc_size(vdev, n);
143     len = address_space_cache_init(&new->desc, vdev->dma_as,
144                                    addr, size, false);
145     if (len < size) {
146         virtio_error(vdev, "Cannot map desc");
147         goto err_desc;
148     }
149
150     size = virtio_queue_get_used_size(vdev, n) + event_size;
151     len = address_space_cache_init(&new->used, vdev->dma_as,
152                                    vq->vring.used, size, true);
153     if (len < size) {
154         virtio_error(vdev, "Cannot map used");
155         goto err_used;
156     }
157
158     size = virtio_queue_get_avail_size(vdev, n) + event_size;
159     len = address_space_cache_init(&new->avail, vdev->dma_as,
160                                    vq->vring.avail, size, false);
161     if (len < size) {
162         virtio_error(vdev, "Cannot map avail");
163         goto err_avail;
164     }
165
166     atomic_rcu_set(&vq->vring.caches, new);
167     if (old) {
168         call_rcu(old, virtio_free_region_cache, rcu);
169     }
170     return;
171
172 err_avail:
173     address_space_cache_destroy(&new->used);
174 err_used:
175     address_space_cache_destroy(&new->desc);
176 err_desc:
177     g_free(new);
178 }
179
180 /* virt queue functions */
181 void virtio_queue_update_rings(VirtIODevice *vdev, int n)
182 {
183     VRing *vring = &vdev->vq[n].vring;
184
185     if (!vring->desc) {
186         /* not yet setup -> nothing to do */
187         return;
188     }
189     vring->avail = vring->desc + vring->num * sizeof(VRingDesc);
190     vring->used = vring_align(vring->avail +
191                               offsetof(VRingAvail, ring[vring->num]),
192                               vring->align);
193     virtio_init_region_cache(vdev, n);
194 }
195
196 /* Called within rcu_read_lock().  */
197 static void vring_desc_read(VirtIODevice *vdev, VRingDesc *desc,
198                             MemoryRegionCache *cache, int i)
199 {
200     address_space_read_cached(cache, i * sizeof(VRingDesc),
201                               desc, sizeof(VRingDesc));
202     virtio_tswap64s(vdev, &desc->addr);
203     virtio_tswap32s(vdev, &desc->len);
204     virtio_tswap16s(vdev, &desc->flags);
205     virtio_tswap16s(vdev, &desc->next);
206 }
207
208 static VRingMemoryRegionCaches *vring_get_region_caches(struct VirtQueue *vq)
209 {
210     VRingMemoryRegionCaches *caches = atomic_rcu_read(&vq->vring.caches);
211     assert(caches != NULL);
212     return caches;
213 }
214 /* Called within rcu_read_lock().  */
215 static inline uint16_t vring_avail_flags(VirtQueue *vq)
216 {
217     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
218     hwaddr pa = offsetof(VRingAvail, flags);
219     return virtio_lduw_phys_cached(vq->vdev, &caches->avail, pa);
220 }
221
222 /* Called within rcu_read_lock().  */
223 static inline uint16_t vring_avail_idx(VirtQueue *vq)
224 {
225     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
226     hwaddr pa = offsetof(VRingAvail, idx);
227     vq->shadow_avail_idx = virtio_lduw_phys_cached(vq->vdev, &caches->avail, pa);
228     return vq->shadow_avail_idx;
229 }
230
231 /* Called within rcu_read_lock().  */
232 static inline uint16_t vring_avail_ring(VirtQueue *vq, int i)
233 {
234     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
235     hwaddr pa = offsetof(VRingAvail, ring[i]);
236     return virtio_lduw_phys_cached(vq->vdev, &caches->avail, pa);
237 }
238
239 /* Called within rcu_read_lock().  */
240 static inline uint16_t vring_get_used_event(VirtQueue *vq)
241 {
242     return vring_avail_ring(vq, vq->vring.num);
243 }
244
245 /* Called within rcu_read_lock().  */
246 static inline void vring_used_write(VirtQueue *vq, VRingUsedElem *uelem,
247                                     int i)
248 {
249     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
250     hwaddr pa = offsetof(VRingUsed, ring[i]);
251     virtio_tswap32s(vq->vdev, &uelem->id);
252     virtio_tswap32s(vq->vdev, &uelem->len);
253     address_space_write_cached(&caches->used, pa, uelem, sizeof(VRingUsedElem));
254     address_space_cache_invalidate(&caches->used, pa, sizeof(VRingUsedElem));
255 }
256
257 /* Called within rcu_read_lock().  */
258 static uint16_t vring_used_idx(VirtQueue *vq)
259 {
260     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
261     hwaddr pa = offsetof(VRingUsed, idx);
262     return virtio_lduw_phys_cached(vq->vdev, &caches->used, pa);
263 }
264
265 /* Called within rcu_read_lock().  */
266 static inline void vring_used_idx_set(VirtQueue *vq, uint16_t val)
267 {
268     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
269     hwaddr pa = offsetof(VRingUsed, idx);
270     virtio_stw_phys_cached(vq->vdev, &caches->used, pa, val);
271     address_space_cache_invalidate(&caches->used, pa, sizeof(val));
272     vq->used_idx = val;
273 }
274
275 /* Called within rcu_read_lock().  */
276 static inline void vring_used_flags_set_bit(VirtQueue *vq, int mask)
277 {
278     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
279     VirtIODevice *vdev = vq->vdev;
280     hwaddr pa = offsetof(VRingUsed, flags);
281     uint16_t flags = virtio_lduw_phys_cached(vq->vdev, &caches->used, pa);
282
283     virtio_stw_phys_cached(vdev, &caches->used, pa, flags | mask);
284     address_space_cache_invalidate(&caches->used, pa, sizeof(flags));
285 }
286
287 /* Called within rcu_read_lock().  */
288 static inline void vring_used_flags_unset_bit(VirtQueue *vq, int mask)
289 {
290     VRingMemoryRegionCaches *caches = vring_get_region_caches(vq);
291     VirtIODevice *vdev = vq->vdev;
292     hwaddr pa = offsetof(VRingUsed, flags);
293     uint16_t flags = virtio_lduw_phys_cached(vq->vdev, &caches->used, pa);
294
295     virtio_stw_phys_cached(vdev, &caches->used, pa, flags & ~mask);
296     address_space_cache_invalidate(&caches->used, pa, sizeof(flags));
297 }
298
299 /* Called within rcu_read_lock().  */
300 static inline void vring_set_avail_event(VirtQueue *vq, uint16_t val)
301 {
302     VRingMemoryRegionCaches *caches;
303     hwaddr pa;
304     if (!vq->notification) {
305         return;
306     }
307
308     caches = vring_get_region_caches(vq);
309     pa = offsetof(VRingUsed, ring[vq->vring.num]);
310     virtio_stw_phys_cached(vq->vdev, &caches->used, pa, val);
311     address_space_cache_invalidate(&caches->used, pa, sizeof(val));
312 }
313
314 void virtio_queue_set_notification(VirtQueue *vq, int enable)
315 {
316     vq->notification = enable;
317
318     if (!vq->vring.desc) {
319         return;
320     }
321
322     rcu_read_lock();
323     if (virtio_vdev_has_feature(vq->vdev, VIRTIO_RING_F_EVENT_IDX)) {
324         vring_set_avail_event(vq, vring_avail_idx(vq));
325     } else if (enable) {
326         vring_used_flags_unset_bit(vq, VRING_USED_F_NO_NOTIFY);
327     } else {
328         vring_used_flags_set_bit(vq, VRING_USED_F_NO_NOTIFY);
329     }
330     if (enable) {
331         /* Expose avail event/used flags before caller checks the avail idx. */
332         smp_mb();
333     }
334     rcu_read_unlock();
335 }
336
337 int virtio_queue_ready(VirtQueue *vq)
338 {
339     return vq->vring.avail != 0;
340 }
341
342 /* Fetch avail_idx from VQ memory only when we really need to know if
343  * guest has added some buffers.
344  * Called within rcu_read_lock().  */
345 static int virtio_queue_empty_rcu(VirtQueue *vq)
346 {
347     if (unlikely(!vq->vring.avail)) {
348         return 1;
349     }
350
351     if (vq->shadow_avail_idx != vq->last_avail_idx) {
352         return 0;
353     }
354
355     return vring_avail_idx(vq) == vq->last_avail_idx;
356 }
357
358 int virtio_queue_empty(VirtQueue *vq)
359 {
360     bool empty;
361
362     if (unlikely(!vq->vring.avail)) {
363         return 1;
364     }
365
366     if (vq->shadow_avail_idx != vq->last_avail_idx) {
367         return 0;
368     }
369
370     rcu_read_lock();
371     empty = vring_avail_idx(vq) == vq->last_avail_idx;
372     rcu_read_unlock();
373     return empty;
374 }
375
376 static void virtqueue_unmap_sg(VirtQueue *vq, const VirtQueueElement *elem,
377                                unsigned int len)
378 {
379     AddressSpace *dma_as = vq->vdev->dma_as;
380     unsigned int offset;
381     int i;
382
383     offset = 0;
384     for (i = 0; i < elem->in_num; i++) {
385         size_t size = MIN(len - offset, elem->in_sg[i].iov_len);
386
387         dma_memory_unmap(dma_as, elem->in_sg[i].iov_base,
388                          elem->in_sg[i].iov_len,
389                          DMA_DIRECTION_FROM_DEVICE, size);
390
391         offset += size;
392     }
393
394     for (i = 0; i < elem->out_num; i++)
395         dma_memory_unmap(dma_as, elem->out_sg[i].iov_base,
396                          elem->out_sg[i].iov_len,
397                          DMA_DIRECTION_TO_DEVICE,
398                          elem->out_sg[i].iov_len);
399 }
400
401 /* virtqueue_detach_element:
402  * @vq: The #VirtQueue
403  * @elem: The #VirtQueueElement
404  * @len: number of bytes written
405  *
406  * Detach the element from the virtqueue.  This function is suitable for device
407  * reset or other situations where a #VirtQueueElement is simply freed and will
408  * not be pushed or discarded.
409  */
410 void virtqueue_detach_element(VirtQueue *vq, const VirtQueueElement *elem,
411                               unsigned int len)
412 {
413     vq->inuse--;
414     virtqueue_unmap_sg(vq, elem, len);
415 }
416
417 /* virtqueue_unpop:
418  * @vq: The #VirtQueue
419  * @elem: The #VirtQueueElement
420  * @len: number of bytes written
421  *
422  * Pretend the most recent element wasn't popped from the virtqueue.  The next
423  * call to virtqueue_pop() will refetch the element.
424  */
425 void virtqueue_unpop(VirtQueue *vq, const VirtQueueElement *elem,
426                      unsigned int len)
427 {
428     vq->last_avail_idx--;
429     virtqueue_detach_element(vq, elem, len);
430 }
431
432 /* virtqueue_rewind:
433  * @vq: The #VirtQueue
434  * @num: Number of elements to push back
435  *
436  * Pretend that elements weren't popped from the virtqueue.  The next
437  * virtqueue_pop() will refetch the oldest element.
438  *
439  * Use virtqueue_unpop() instead if you have a VirtQueueElement.
440  *
441  * Returns: true on success, false if @num is greater than the number of in use
442  * elements.
443  */
444 bool virtqueue_rewind(VirtQueue *vq, unsigned int num)
445 {
446     if (num > vq->inuse) {
447         return false;
448     }
449     vq->last_avail_idx -= num;
450     vq->inuse -= num;
451     return true;
452 }
453
454 /* Called within rcu_read_lock().  */
455 void virtqueue_fill(VirtQueue *vq, const VirtQueueElement *elem,
456                     unsigned int len, unsigned int idx)
457 {
458     VRingUsedElem uelem;
459
460     trace_virtqueue_fill(vq, elem, len, idx);
461
462     virtqueue_unmap_sg(vq, elem, len);
463
464     if (unlikely(vq->vdev->broken)) {
465         return;
466     }
467
468     if (unlikely(!vq->vring.used)) {
469         return;
470     }
471
472     idx = (idx + vq->used_idx) % vq->vring.num;
473
474     uelem.id = elem->index;
475     uelem.len = len;
476     vring_used_write(vq, &uelem, idx);
477 }
478
479 /* Called within rcu_read_lock().  */
480 void virtqueue_flush(VirtQueue *vq, unsigned int count)
481 {
482     uint16_t old, new;
483
484     if (unlikely(vq->vdev->broken)) {
485         vq->inuse -= count;
486         return;
487     }
488
489     if (unlikely(!vq->vring.used)) {
490         return;
491     }
492
493     /* Make sure buffer is written before we update index. */
494     smp_wmb();
495     trace_virtqueue_flush(vq, count);
496     old = vq->used_idx;
497     new = old + count;
498     vring_used_idx_set(vq, new);
499     vq->inuse -= count;
500     if (unlikely((int16_t)(new - vq->signalled_used) < (uint16_t)(new - old)))
501         vq->signalled_used_valid = false;
502 }
503
504 void virtqueue_push(VirtQueue *vq, const VirtQueueElement *elem,
505                     unsigned int len)
506 {
507     rcu_read_lock();
508     virtqueue_fill(vq, elem, len, 0);
509     virtqueue_flush(vq, 1);
510     rcu_read_unlock();
511 }
512
513 /* Called within rcu_read_lock().  */
514 static int virtqueue_num_heads(VirtQueue *vq, unsigned int idx)
515 {
516     uint16_t num_heads = vring_avail_idx(vq) - idx;
517
518     /* Check it isn't doing very strange things with descriptor numbers. */
519     if (num_heads > vq->vring.num) {
520         virtio_error(vq->vdev, "Guest moved used index from %u to %u",
521                      idx, vq->shadow_avail_idx);
522         return -EINVAL;
523     }
524     /* On success, callers read a descriptor at vq->last_avail_idx.
525      * Make sure descriptor read does not bypass avail index read. */
526     if (num_heads) {
527         smp_rmb();
528     }
529
530     return num_heads;
531 }
532
533 /* Called within rcu_read_lock().  */
534 static bool virtqueue_get_head(VirtQueue *vq, unsigned int idx,
535                                unsigned int *head)
536 {
537     /* Grab the next descriptor number they're advertising, and increment
538      * the index we've seen. */
539     *head = vring_avail_ring(vq, idx % vq->vring.num);
540
541     /* If their number is silly, that's a fatal mistake. */
542     if (*head >= vq->vring.num) {
543         virtio_error(vq->vdev, "Guest says index %u is available", *head);
544         return false;
545     }
546
547     return true;
548 }
549
550 enum {
551     VIRTQUEUE_READ_DESC_ERROR = -1,
552     VIRTQUEUE_READ_DESC_DONE = 0,   /* end of chain */
553     VIRTQUEUE_READ_DESC_MORE = 1,   /* more buffers in chain */
554 };
555
556 static int virtqueue_read_next_desc(VirtIODevice *vdev, VRingDesc *desc,
557                                     MemoryRegionCache *desc_cache, unsigned int max,
558                                     unsigned int *next)
559 {
560     /* If this descriptor says it doesn't chain, we're done. */
561     if (!(desc->flags & VRING_DESC_F_NEXT)) {
562         return VIRTQUEUE_READ_DESC_DONE;
563     }
564
565     /* Check they're not leading us off end of descriptors. */
566     *next = desc->next;
567     /* Make sure compiler knows to grab that: we don't want it changing! */
568     smp_wmb();
569
570     if (*next >= max) {
571         virtio_error(vdev, "Desc next is %u", *next);
572         return VIRTQUEUE_READ_DESC_ERROR;
573     }
574
575     vring_desc_read(vdev, desc, desc_cache, *next);
576     return VIRTQUEUE_READ_DESC_MORE;
577 }
578
579 void virtqueue_get_avail_bytes(VirtQueue *vq, unsigned int *in_bytes,
580                                unsigned int *out_bytes,
581                                unsigned max_in_bytes, unsigned max_out_bytes)
582 {
583     VirtIODevice *vdev = vq->vdev;
584     unsigned int max, idx;
585     unsigned int total_bufs, in_total, out_total;
586     VRingMemoryRegionCaches *caches;
587     MemoryRegionCache indirect_desc_cache = MEMORY_REGION_CACHE_INVALID;
588     int64_t len = 0;
589     int rc;
590
591     if (unlikely(!vq->vring.desc)) {
592         if (in_bytes) {
593             *in_bytes = 0;
594         }
595         if (out_bytes) {
596             *out_bytes = 0;
597         }
598         return;
599     }
600
601     rcu_read_lock();
602     idx = vq->last_avail_idx;
603     total_bufs = in_total = out_total = 0;
604
605     max = vq->vring.num;
606     caches = vring_get_region_caches(vq);
607     if (caches->desc.len < max * sizeof(VRingDesc)) {
608         virtio_error(vdev, "Cannot map descriptor ring");
609         goto err;
610     }
611
612     while ((rc = virtqueue_num_heads(vq, idx)) > 0) {
613         MemoryRegionCache *desc_cache = &caches->desc;
614         unsigned int num_bufs;
615         VRingDesc desc;
616         unsigned int i;
617
618         num_bufs = total_bufs;
619
620         if (!virtqueue_get_head(vq, idx++, &i)) {
621             goto err;
622         }
623
624         vring_desc_read(vdev, &desc, desc_cache, i);
625
626         if (desc.flags & VRING_DESC_F_INDIRECT) {
627             if (desc.len % sizeof(VRingDesc)) {
628                 virtio_error(vdev, "Invalid size for indirect buffer table");
629                 goto err;
630             }
631
632             /* If we've got too many, that implies a descriptor loop. */
633             if (num_bufs >= max) {
634                 virtio_error(vdev, "Looped descriptor");
635                 goto err;
636             }
637
638             /* loop over the indirect descriptor table */
639             len = address_space_cache_init(&indirect_desc_cache,
640                                            vdev->dma_as,
641                                            desc.addr, desc.len, false);
642             desc_cache = &indirect_desc_cache;
643             if (len < desc.len) {
644                 virtio_error(vdev, "Cannot map indirect buffer");
645                 goto err;
646             }
647
648             max = desc.len / sizeof(VRingDesc);
649             num_bufs = i = 0;
650             vring_desc_read(vdev, &desc, desc_cache, i);
651         }
652
653         do {
654             /* If we've got too many, that implies a descriptor loop. */
655             if (++num_bufs > max) {
656                 virtio_error(vdev, "Looped descriptor");
657                 goto err;
658             }
659
660             if (desc.flags & VRING_DESC_F_WRITE) {
661                 in_total += desc.len;
662             } else {
663                 out_total += desc.len;
664             }
665             if (in_total >= max_in_bytes && out_total >= max_out_bytes) {
666                 goto done;
667             }
668
669             rc = virtqueue_read_next_desc(vdev, &desc, desc_cache, max, &i);
670         } while (rc == VIRTQUEUE_READ_DESC_MORE);
671
672         if (rc == VIRTQUEUE_READ_DESC_ERROR) {
673             goto err;
674         }
675
676         if (desc_cache == &indirect_desc_cache) {
677             address_space_cache_destroy(&indirect_desc_cache);
678             total_bufs++;
679         } else {
680             total_bufs = num_bufs;
681         }
682     }
683
684     if (rc < 0) {
685         goto err;
686     }
687
688 done:
689     address_space_cache_destroy(&indirect_desc_cache);
690     if (in_bytes) {
691         *in_bytes = in_total;
692     }
693     if (out_bytes) {
694         *out_bytes = out_total;
695     }
696     rcu_read_unlock();
697     return;
698
699 err:
700     in_total = out_total = 0;
701     goto done;
702 }
703
704 int virtqueue_avail_bytes(VirtQueue *vq, unsigned int in_bytes,
705                           unsigned int out_bytes)
706 {
707     unsigned int in_total, out_total;
708
709     virtqueue_get_avail_bytes(vq, &in_total, &out_total, in_bytes, out_bytes);
710     return in_bytes <= in_total && out_bytes <= out_total;
711 }
712
713 static bool virtqueue_map_desc(VirtIODevice *vdev, unsigned int *p_num_sg,
714                                hwaddr *addr, struct iovec *iov,
715                                unsigned int max_num_sg, bool is_write,
716                                hwaddr pa, size_t sz)
717 {
718     bool ok = false;
719     unsigned num_sg = *p_num_sg;
720     assert(num_sg <= max_num_sg);
721
722     if (!sz) {
723         virtio_error(vdev, "virtio: zero sized buffers are not allowed");
724         goto out;
725     }
726
727     while (sz) {
728         hwaddr len = sz;
729
730         if (num_sg == max_num_sg) {
731             virtio_error(vdev, "virtio: too many write descriptors in "
732                                "indirect table");
733             goto out;
734         }
735
736         iov[num_sg].iov_base = dma_memory_map(vdev->dma_as, pa, &len,
737                                               is_write ?
738                                               DMA_DIRECTION_FROM_DEVICE :
739                                               DMA_DIRECTION_TO_DEVICE);
740         if (!iov[num_sg].iov_base) {
741             virtio_error(vdev, "virtio: bogus descriptor or out of resources");
742             goto out;
743         }
744
745         iov[num_sg].iov_len = len;
746         addr[num_sg] = pa;
747
748         sz -= len;
749         pa += len;
750         num_sg++;
751     }
752     ok = true;
753
754 out:
755     *p_num_sg = num_sg;
756     return ok;
757 }
758
759 /* Only used by error code paths before we have a VirtQueueElement (therefore
760  * virtqueue_unmap_sg() can't be used).  Assumes buffers weren't written to
761  * yet.
762  */
763 static void virtqueue_undo_map_desc(unsigned int out_num, unsigned int in_num,
764                                     struct iovec *iov)
765 {
766     unsigned int i;
767
768     for (i = 0; i < out_num + in_num; i++) {
769         int is_write = i >= out_num;
770
771         cpu_physical_memory_unmap(iov->iov_base, iov->iov_len, is_write, 0);
772         iov++;
773     }
774 }
775
776 static void virtqueue_map_iovec(VirtIODevice *vdev, struct iovec *sg,
777                                 hwaddr *addr, unsigned int *num_sg,
778                                 int is_write)
779 {
780     unsigned int i;
781     hwaddr len;
782
783     for (i = 0; i < *num_sg; i++) {
784         len = sg[i].iov_len;
785         sg[i].iov_base = dma_memory_map(vdev->dma_as,
786                                         addr[i], &len, is_write ?
787                                         DMA_DIRECTION_FROM_DEVICE :
788                                         DMA_DIRECTION_TO_DEVICE);
789         if (!sg[i].iov_base) {
790             error_report("virtio: error trying to map MMIO memory");
791             exit(1);
792         }
793         if (len != sg[i].iov_len) {
794             error_report("virtio: unexpected memory split");
795             exit(1);
796         }
797     }
798 }
799
800 void virtqueue_map(VirtIODevice *vdev, VirtQueueElement *elem)
801 {
802     virtqueue_map_iovec(vdev, elem->in_sg, elem->in_addr, &elem->in_num, 1);
803     virtqueue_map_iovec(vdev, elem->out_sg, elem->out_addr, &elem->out_num, 0);
804 }
805
806 static void *virtqueue_alloc_element(size_t sz, unsigned out_num, unsigned in_num)
807 {
808     VirtQueueElement *elem;
809     size_t in_addr_ofs = QEMU_ALIGN_UP(sz, __alignof__(elem->in_addr[0]));
810     size_t out_addr_ofs = in_addr_ofs + in_num * sizeof(elem->in_addr[0]);
811     size_t out_addr_end = out_addr_ofs + out_num * sizeof(elem->out_addr[0]);
812     size_t in_sg_ofs = QEMU_ALIGN_UP(out_addr_end, __alignof__(elem->in_sg[0]));
813     size_t out_sg_ofs = in_sg_ofs + in_num * sizeof(elem->in_sg[0]);
814     size_t out_sg_end = out_sg_ofs + out_num * sizeof(elem->out_sg[0]);
815
816     assert(sz >= sizeof(VirtQueueElement));
817     elem = g_malloc(out_sg_end);
818     elem->out_num = out_num;
819     elem->in_num = in_num;
820     elem->in_addr = (void *)elem + in_addr_ofs;
821     elem->out_addr = (void *)elem + out_addr_ofs;
822     elem->in_sg = (void *)elem + in_sg_ofs;
823     elem->out_sg = (void *)elem + out_sg_ofs;
824     return elem;
825 }
826
827 void *virtqueue_pop(VirtQueue *vq, size_t sz)
828 {
829     unsigned int i, head, max;
830     VRingMemoryRegionCaches *caches;
831     MemoryRegionCache indirect_desc_cache = MEMORY_REGION_CACHE_INVALID;
832     MemoryRegionCache *desc_cache;
833     int64_t len;
834     VirtIODevice *vdev = vq->vdev;
835     VirtQueueElement *elem = NULL;
836     unsigned out_num, in_num;
837     hwaddr addr[VIRTQUEUE_MAX_SIZE];
838     struct iovec iov[VIRTQUEUE_MAX_SIZE];
839     VRingDesc desc;
840     int rc;
841
842     if (unlikely(vdev->broken)) {
843         return NULL;
844     }
845     rcu_read_lock();
846     if (virtio_queue_empty_rcu(vq)) {
847         goto done;
848     }
849     /* Needed after virtio_queue_empty(), see comment in
850      * virtqueue_num_heads(). */
851     smp_rmb();
852
853     /* When we start there are none of either input nor output. */
854     out_num = in_num = 0;
855
856     max = vq->vring.num;
857
858     if (vq->inuse >= vq->vring.num) {
859         virtio_error(vdev, "Virtqueue size exceeded");
860         goto done;
861     }
862
863     if (!virtqueue_get_head(vq, vq->last_avail_idx++, &head)) {
864         goto done;
865     }
866
867     if (virtio_vdev_has_feature(vdev, VIRTIO_RING_F_EVENT_IDX)) {
868         vring_set_avail_event(vq, vq->last_avail_idx);
869     }
870
871     i = head;
872
873     caches = vring_get_region_caches(vq);
874     if (caches->desc.len < max * sizeof(VRingDesc)) {
875         virtio_error(vdev, "Cannot map descriptor ring");
876         goto done;
877     }
878
879     desc_cache = &caches->desc;
880     vring_desc_read(vdev, &desc, desc_cache, i);
881     if (desc.flags & VRING_DESC_F_INDIRECT) {
882         if (desc.len % sizeof(VRingDesc)) {
883             virtio_error(vdev, "Invalid size for indirect buffer table");
884             goto done;
885         }
886
887         /* loop over the indirect descriptor table */
888         len = address_space_cache_init(&indirect_desc_cache, vdev->dma_as,
889                                        desc.addr, desc.len, false);
890         desc_cache = &indirect_desc_cache;
891         if (len < desc.len) {
892             virtio_error(vdev, "Cannot map indirect buffer");
893             goto done;
894         }
895
896         max = desc.len / sizeof(VRingDesc);
897         i = 0;
898         vring_desc_read(vdev, &desc, desc_cache, i);
899     }
900
901     /* Collect all the descriptors */
902     do {
903         bool map_ok;
904
905         if (desc.flags & VRING_DESC_F_WRITE) {
906             map_ok = virtqueue_map_desc(vdev, &in_num, addr + out_num,
907                                         iov + out_num,
908                                         VIRTQUEUE_MAX_SIZE - out_num, true,
909                                         desc.addr, desc.len);
910         } else {
911             if (in_num) {
912                 virtio_error(vdev, "Incorrect order for descriptors");
913                 goto err_undo_map;
914             }
915             map_ok = virtqueue_map_desc(vdev, &out_num, addr, iov,
916                                         VIRTQUEUE_MAX_SIZE, false,
917                                         desc.addr, desc.len);
918         }
919         if (!map_ok) {
920             goto err_undo_map;
921         }
922
923         /* If we've got too many, that implies a descriptor loop. */
924         if ((in_num + out_num) > max) {
925             virtio_error(vdev, "Looped descriptor");
926             goto err_undo_map;
927         }
928
929         rc = virtqueue_read_next_desc(vdev, &desc, desc_cache, max, &i);
930     } while (rc == VIRTQUEUE_READ_DESC_MORE);
931
932     if (rc == VIRTQUEUE_READ_DESC_ERROR) {
933         goto err_undo_map;
934     }
935
936     /* Now copy what we have collected and mapped */
937     elem = virtqueue_alloc_element(sz, out_num, in_num);
938     elem->index = head;
939     for (i = 0; i < out_num; i++) {
940         elem->out_addr[i] = addr[i];
941         elem->out_sg[i] = iov[i];
942     }
943     for (i = 0; i < in_num; i++) {
944         elem->in_addr[i] = addr[out_num + i];
945         elem->in_sg[i] = iov[out_num + i];
946     }
947
948     vq->inuse++;
949
950     trace_virtqueue_pop(vq, elem, elem->in_num, elem->out_num);
951 done:
952     address_space_cache_destroy(&indirect_desc_cache);
953     rcu_read_unlock();
954
955     return elem;
956
957 err_undo_map:
958     virtqueue_undo_map_desc(out_num, in_num, iov);
959     goto done;
960 }
961
962 /* virtqueue_drop_all:
963  * @vq: The #VirtQueue
964  * Drops all queued buffers and indicates them to the guest
965  * as if they are done. Useful when buffers can not be
966  * processed but must be returned to the guest.
967  */
968 unsigned int virtqueue_drop_all(VirtQueue *vq)
969 {
970     unsigned int dropped = 0;
971     VirtQueueElement elem = {};
972     VirtIODevice *vdev = vq->vdev;
973     bool fEventIdx = virtio_vdev_has_feature(vdev, VIRTIO_RING_F_EVENT_IDX);
974
975     if (unlikely(vdev->broken)) {
976         return 0;
977     }
978
979     while (!virtio_queue_empty(vq) && vq->inuse < vq->vring.num) {
980         /* works similar to virtqueue_pop but does not map buffers
981         * and does not allocate any memory */
982         smp_rmb();
983         if (!virtqueue_get_head(vq, vq->last_avail_idx, &elem.index)) {
984             break;
985         }
986         vq->inuse++;
987         vq->last_avail_idx++;
988         if (fEventIdx) {
989             vring_set_avail_event(vq, vq->last_avail_idx);
990         }
991         /* immediately push the element, nothing to unmap
992          * as both in_num and out_num are set to 0 */
993         virtqueue_push(vq, &elem, 0);
994         dropped++;
995     }
996
997     return dropped;
998 }
999
1000 /* Reading and writing a structure directly to QEMUFile is *awful*, but
1001  * it is what QEMU has always done by mistake.  We can change it sooner
1002  * or later by bumping the version number of the affected vm states.
1003  * In the meanwhile, since the in-memory layout of VirtQueueElement
1004  * has changed, we need to marshal to and from the layout that was
1005  * used before the change.
1006  */
1007 typedef struct VirtQueueElementOld {
1008     unsigned int index;
1009     unsigned int out_num;
1010     unsigned int in_num;
1011     hwaddr in_addr[VIRTQUEUE_MAX_SIZE];
1012     hwaddr out_addr[VIRTQUEUE_MAX_SIZE];
1013     struct iovec in_sg[VIRTQUEUE_MAX_SIZE];
1014     struct iovec out_sg[VIRTQUEUE_MAX_SIZE];
1015 } VirtQueueElementOld;
1016
1017 void *qemu_get_virtqueue_element(VirtIODevice *vdev, QEMUFile *f, size_t sz)
1018 {
1019     VirtQueueElement *elem;
1020     VirtQueueElementOld data;
1021     int i;
1022
1023     qemu_get_buffer(f, (uint8_t *)&data, sizeof(VirtQueueElementOld));
1024
1025     /* TODO: teach all callers that this can fail, and return failure instead
1026      * of asserting here.
1027      * When we do, we might be able to re-enable NDEBUG below.
1028      */
1029 #ifdef NDEBUG
1030 #error building with NDEBUG is not supported
1031 #endif
1032     assert(ARRAY_SIZE(data.in_addr) >= data.in_num);
1033     assert(ARRAY_SIZE(data.out_addr) >= data.out_num);
1034
1035     elem = virtqueue_alloc_element(sz, data.out_num, data.in_num);
1036     elem->index = data.index;
1037
1038     for (i = 0; i < elem->in_num; i++) {
1039         elem->in_addr[i] = data.in_addr[i];
1040     }
1041
1042     for (i = 0; i < elem->out_num; i++) {
1043         elem->out_addr[i] = data.out_addr[i];
1044     }
1045
1046     for (i = 0; i < elem->in_num; i++) {
1047         /* Base is overwritten by virtqueue_map.  */
1048         elem->in_sg[i].iov_base = 0;
1049         elem->in_sg[i].iov_len = data.in_sg[i].iov_len;
1050     }
1051
1052     for (i = 0; i < elem->out_num; i++) {
1053         /* Base is overwritten by virtqueue_map.  */
1054         elem->out_sg[i].iov_base = 0;
1055         elem->out_sg[i].iov_len = data.out_sg[i].iov_len;
1056     }
1057
1058     virtqueue_map(vdev, elem);
1059     return elem;
1060 }
1061
1062 void qemu_put_virtqueue_element(QEMUFile *f, VirtQueueElement *elem)
1063 {
1064     VirtQueueElementOld data;
1065     int i;
1066
1067     memset(&data, 0, sizeof(data));
1068     data.index = elem->index;
1069     data.in_num = elem->in_num;
1070     data.out_num = elem->out_num;
1071
1072     for (i = 0; i < elem->in_num; i++) {
1073         data.in_addr[i] = elem->in_addr[i];
1074     }
1075
1076     for (i = 0; i < elem->out_num; i++) {
1077         data.out_addr[i] = elem->out_addr[i];
1078     }
1079
1080     for (i = 0; i < elem->in_num; i++) {
1081         /* Base is overwritten by virtqueue_map when loading.  Do not
1082          * save it, as it would leak the QEMU address space layout.  */
1083         data.in_sg[i].iov_len = elem->in_sg[i].iov_len;
1084     }
1085
1086     for (i = 0; i < elem->out_num; i++) {
1087         /* Do not save iov_base as above.  */
1088         data.out_sg[i].iov_len = elem->out_sg[i].iov_len;
1089     }
1090     qemu_put_buffer(f, (uint8_t *)&data, sizeof(VirtQueueElementOld));
1091 }
1092
1093 /* virtio device */
1094 static void virtio_notify_vector(VirtIODevice *vdev, uint16_t vector)
1095 {
1096     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
1097     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
1098
1099     if (unlikely(vdev->broken)) {
1100         return;
1101     }
1102
1103     if (k->notify) {
1104         k->notify(qbus->parent, vector);
1105     }
1106 }
1107
1108 void virtio_update_irq(VirtIODevice *vdev)
1109 {
1110     virtio_notify_vector(vdev, VIRTIO_NO_VECTOR);
1111 }
1112
1113 static int virtio_validate_features(VirtIODevice *vdev)
1114 {
1115     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1116
1117     if (virtio_host_has_feature(vdev, VIRTIO_F_IOMMU_PLATFORM) &&
1118         !virtio_vdev_has_feature(vdev, VIRTIO_F_IOMMU_PLATFORM)) {
1119         return -EFAULT;
1120     }
1121
1122     if (k->validate_features) {
1123         return k->validate_features(vdev);
1124     } else {
1125         return 0;
1126     }
1127 }
1128
1129 int virtio_set_status(VirtIODevice *vdev, uint8_t val)
1130 {
1131     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1132     trace_virtio_set_status(vdev, val);
1133
1134     if (virtio_vdev_has_feature(vdev, VIRTIO_F_VERSION_1)) {
1135         if (!(vdev->status & VIRTIO_CONFIG_S_FEATURES_OK) &&
1136             val & VIRTIO_CONFIG_S_FEATURES_OK) {
1137             int ret = virtio_validate_features(vdev);
1138
1139             if (ret) {
1140                 return ret;
1141             }
1142         }
1143     }
1144     if (k->set_status) {
1145         k->set_status(vdev, val);
1146     }
1147     vdev->status = val;
1148     return 0;
1149 }
1150
1151 bool target_words_bigendian(void);
1152 static enum virtio_device_endian virtio_default_endian(void)
1153 {
1154     if (target_words_bigendian()) {
1155         return VIRTIO_DEVICE_ENDIAN_BIG;
1156     } else {
1157         return VIRTIO_DEVICE_ENDIAN_LITTLE;
1158     }
1159 }
1160
1161 static enum virtio_device_endian virtio_current_cpu_endian(void)
1162 {
1163     CPUClass *cc = CPU_GET_CLASS(current_cpu);
1164
1165     if (cc->virtio_is_big_endian(current_cpu)) {
1166         return VIRTIO_DEVICE_ENDIAN_BIG;
1167     } else {
1168         return VIRTIO_DEVICE_ENDIAN_LITTLE;
1169     }
1170 }
1171
1172 static void virtio_virtqueue_reset_region_cache(struct VirtQueue *vq)
1173 {
1174     VRingMemoryRegionCaches *caches;
1175
1176     caches = atomic_read(&vq->vring.caches);
1177     atomic_rcu_set(&vq->vring.caches, NULL);
1178     if (caches) {
1179         call_rcu(caches, virtio_free_region_cache, rcu);
1180     }
1181 }
1182
1183 void virtio_reset(void *opaque)
1184 {
1185     VirtIODevice *vdev = opaque;
1186     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1187     int i;
1188
1189     virtio_set_status(vdev, 0);
1190     if (current_cpu) {
1191         /* Guest initiated reset */
1192         vdev->device_endian = virtio_current_cpu_endian();
1193     } else {
1194         /* System reset */
1195         vdev->device_endian = virtio_default_endian();
1196     }
1197
1198     if (k->reset) {
1199         k->reset(vdev);
1200     }
1201
1202     vdev->broken = false;
1203     vdev->guest_features = 0;
1204     vdev->queue_sel = 0;
1205     vdev->status = 0;
1206     atomic_set(&vdev->isr, 0);
1207     vdev->config_vector = VIRTIO_NO_VECTOR;
1208     virtio_notify_vector(vdev, vdev->config_vector);
1209
1210     for(i = 0; i < VIRTIO_QUEUE_MAX; i++) {
1211         vdev->vq[i].vring.desc = 0;
1212         vdev->vq[i].vring.avail = 0;
1213         vdev->vq[i].vring.used = 0;
1214         vdev->vq[i].last_avail_idx = 0;
1215         vdev->vq[i].shadow_avail_idx = 0;
1216         vdev->vq[i].used_idx = 0;
1217         virtio_queue_set_vector(vdev, i, VIRTIO_NO_VECTOR);
1218         vdev->vq[i].signalled_used = 0;
1219         vdev->vq[i].signalled_used_valid = false;
1220         vdev->vq[i].notification = true;
1221         vdev->vq[i].vring.num = vdev->vq[i].vring.num_default;
1222         vdev->vq[i].inuse = 0;
1223         virtio_virtqueue_reset_region_cache(&vdev->vq[i]);
1224     }
1225 }
1226
1227 uint32_t virtio_config_readb(VirtIODevice *vdev, uint32_t addr)
1228 {
1229     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1230     uint8_t val;
1231
1232     if (addr + sizeof(val) > vdev->config_len) {
1233         return (uint32_t)-1;
1234     }
1235
1236     k->get_config(vdev, vdev->config);
1237
1238     val = ldub_p(vdev->config + addr);
1239     return val;
1240 }
1241
1242 uint32_t virtio_config_readw(VirtIODevice *vdev, uint32_t addr)
1243 {
1244     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1245     uint16_t val;
1246
1247     if (addr + sizeof(val) > vdev->config_len) {
1248         return (uint32_t)-1;
1249     }
1250
1251     k->get_config(vdev, vdev->config);
1252
1253     val = lduw_p(vdev->config + addr);
1254     return val;
1255 }
1256
1257 uint32_t virtio_config_readl(VirtIODevice *vdev, uint32_t addr)
1258 {
1259     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1260     uint32_t val;
1261
1262     if (addr + sizeof(val) > vdev->config_len) {
1263         return (uint32_t)-1;
1264     }
1265
1266     k->get_config(vdev, vdev->config);
1267
1268     val = ldl_p(vdev->config + addr);
1269     return val;
1270 }
1271
1272 void virtio_config_writeb(VirtIODevice *vdev, uint32_t addr, uint32_t data)
1273 {
1274     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1275     uint8_t val = data;
1276
1277     if (addr + sizeof(val) > vdev->config_len) {
1278         return;
1279     }
1280
1281     stb_p(vdev->config + addr, val);
1282
1283     if (k->set_config) {
1284         k->set_config(vdev, vdev->config);
1285     }
1286 }
1287
1288 void virtio_config_writew(VirtIODevice *vdev, uint32_t addr, uint32_t data)
1289 {
1290     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1291     uint16_t val = data;
1292
1293     if (addr + sizeof(val) > vdev->config_len) {
1294         return;
1295     }
1296
1297     stw_p(vdev->config + addr, val);
1298
1299     if (k->set_config) {
1300         k->set_config(vdev, vdev->config);
1301     }
1302 }
1303
1304 void virtio_config_writel(VirtIODevice *vdev, uint32_t addr, uint32_t data)
1305 {
1306     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1307     uint32_t val = data;
1308
1309     if (addr + sizeof(val) > vdev->config_len) {
1310         return;
1311     }
1312
1313     stl_p(vdev->config + addr, val);
1314
1315     if (k->set_config) {
1316         k->set_config(vdev, vdev->config);
1317     }
1318 }
1319
1320 uint32_t virtio_config_modern_readb(VirtIODevice *vdev, uint32_t addr)
1321 {
1322     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1323     uint8_t val;
1324
1325     if (addr + sizeof(val) > vdev->config_len) {
1326         return (uint32_t)-1;
1327     }
1328
1329     k->get_config(vdev, vdev->config);
1330
1331     val = ldub_p(vdev->config + addr);
1332     return val;
1333 }
1334
1335 uint32_t virtio_config_modern_readw(VirtIODevice *vdev, uint32_t addr)
1336 {
1337     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1338     uint16_t val;
1339
1340     if (addr + sizeof(val) > vdev->config_len) {
1341         return (uint32_t)-1;
1342     }
1343
1344     k->get_config(vdev, vdev->config);
1345
1346     val = lduw_le_p(vdev->config + addr);
1347     return val;
1348 }
1349
1350 uint32_t virtio_config_modern_readl(VirtIODevice *vdev, uint32_t addr)
1351 {
1352     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1353     uint32_t val;
1354
1355     if (addr + sizeof(val) > vdev->config_len) {
1356         return (uint32_t)-1;
1357     }
1358
1359     k->get_config(vdev, vdev->config);
1360
1361     val = ldl_le_p(vdev->config + addr);
1362     return val;
1363 }
1364
1365 void virtio_config_modern_writeb(VirtIODevice *vdev,
1366                                  uint32_t addr, uint32_t data)
1367 {
1368     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1369     uint8_t val = data;
1370
1371     if (addr + sizeof(val) > vdev->config_len) {
1372         return;
1373     }
1374
1375     stb_p(vdev->config + addr, val);
1376
1377     if (k->set_config) {
1378         k->set_config(vdev, vdev->config);
1379     }
1380 }
1381
1382 void virtio_config_modern_writew(VirtIODevice *vdev,
1383                                  uint32_t addr, uint32_t data)
1384 {
1385     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1386     uint16_t val = data;
1387
1388     if (addr + sizeof(val) > vdev->config_len) {
1389         return;
1390     }
1391
1392     stw_le_p(vdev->config + addr, val);
1393
1394     if (k->set_config) {
1395         k->set_config(vdev, vdev->config);
1396     }
1397 }
1398
1399 void virtio_config_modern_writel(VirtIODevice *vdev,
1400                                  uint32_t addr, uint32_t data)
1401 {
1402     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1403     uint32_t val = data;
1404
1405     if (addr + sizeof(val) > vdev->config_len) {
1406         return;
1407     }
1408
1409     stl_le_p(vdev->config + addr, val);
1410
1411     if (k->set_config) {
1412         k->set_config(vdev, vdev->config);
1413     }
1414 }
1415
1416 void virtio_queue_set_addr(VirtIODevice *vdev, int n, hwaddr addr)
1417 {
1418     vdev->vq[n].vring.desc = addr;
1419     virtio_queue_update_rings(vdev, n);
1420 }
1421
1422 hwaddr virtio_queue_get_addr(VirtIODevice *vdev, int n)
1423 {
1424     return vdev->vq[n].vring.desc;
1425 }
1426
1427 void virtio_queue_set_rings(VirtIODevice *vdev, int n, hwaddr desc,
1428                             hwaddr avail, hwaddr used)
1429 {
1430     vdev->vq[n].vring.desc = desc;
1431     vdev->vq[n].vring.avail = avail;
1432     vdev->vq[n].vring.used = used;
1433     virtio_init_region_cache(vdev, n);
1434 }
1435
1436 void virtio_queue_set_num(VirtIODevice *vdev, int n, int num)
1437 {
1438     /* Don't allow guest to flip queue between existent and
1439      * nonexistent states, or to set it to an invalid size.
1440      */
1441     if (!!num != !!vdev->vq[n].vring.num ||
1442         num > VIRTQUEUE_MAX_SIZE ||
1443         num < 0) {
1444         return;
1445     }
1446     vdev->vq[n].vring.num = num;
1447 }
1448
1449 VirtQueue *virtio_vector_first_queue(VirtIODevice *vdev, uint16_t vector)
1450 {
1451     return QLIST_FIRST(&vdev->vector_queues[vector]);
1452 }
1453
1454 VirtQueue *virtio_vector_next_queue(VirtQueue *vq)
1455 {
1456     return QLIST_NEXT(vq, node);
1457 }
1458
1459 int virtio_queue_get_num(VirtIODevice *vdev, int n)
1460 {
1461     return vdev->vq[n].vring.num;
1462 }
1463
1464 int virtio_queue_get_max_num(VirtIODevice *vdev, int n)
1465 {
1466     return vdev->vq[n].vring.num_default;
1467 }
1468
1469 int virtio_get_num_queues(VirtIODevice *vdev)
1470 {
1471     int i;
1472
1473     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
1474         if (!virtio_queue_get_num(vdev, i)) {
1475             break;
1476         }
1477     }
1478
1479     return i;
1480 }
1481
1482 void virtio_queue_set_align(VirtIODevice *vdev, int n, int align)
1483 {
1484     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
1485     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
1486
1487     /* virtio-1 compliant devices cannot change the alignment */
1488     if (virtio_vdev_has_feature(vdev, VIRTIO_F_VERSION_1)) {
1489         error_report("tried to modify queue alignment for virtio-1 device");
1490         return;
1491     }
1492     /* Check that the transport told us it was going to do this
1493      * (so a buggy transport will immediately assert rather than
1494      * silently failing to migrate this state)
1495      */
1496     assert(k->has_variable_vring_alignment);
1497
1498     vdev->vq[n].vring.align = align;
1499     virtio_queue_update_rings(vdev, n);
1500 }
1501
1502 static bool virtio_queue_notify_aio_vq(VirtQueue *vq)
1503 {
1504     if (vq->vring.desc && vq->handle_aio_output) {
1505         VirtIODevice *vdev = vq->vdev;
1506
1507         trace_virtio_queue_notify(vdev, vq - vdev->vq, vq);
1508         return vq->handle_aio_output(vdev, vq);
1509     }
1510
1511     return false;
1512 }
1513
1514 static void virtio_queue_notify_vq(VirtQueue *vq)
1515 {
1516     if (vq->vring.desc && vq->handle_output) {
1517         VirtIODevice *vdev = vq->vdev;
1518
1519         if (unlikely(vdev->broken)) {
1520             return;
1521         }
1522
1523         trace_virtio_queue_notify(vdev, vq - vdev->vq, vq);
1524         vq->handle_output(vdev, vq);
1525     }
1526 }
1527
1528 void virtio_queue_notify(VirtIODevice *vdev, int n)
1529 {
1530     VirtQueue *vq = &vdev->vq[n];
1531
1532     if (unlikely(!vq->vring.desc || vdev->broken)) {
1533         return;
1534     }
1535
1536     trace_virtio_queue_notify(vdev, vq - vdev->vq, vq);
1537     if (vq->handle_aio_output) {
1538         event_notifier_set(&vq->host_notifier);
1539     } else if (vq->handle_output) {
1540         vq->handle_output(vdev, vq);
1541     }
1542 }
1543
1544 uint16_t virtio_queue_vector(VirtIODevice *vdev, int n)
1545 {
1546     return n < VIRTIO_QUEUE_MAX ? vdev->vq[n].vector :
1547         VIRTIO_NO_VECTOR;
1548 }
1549
1550 void virtio_queue_set_vector(VirtIODevice *vdev, int n, uint16_t vector)
1551 {
1552     VirtQueue *vq = &vdev->vq[n];
1553
1554     if (n < VIRTIO_QUEUE_MAX) {
1555         if (vdev->vector_queues &&
1556             vdev->vq[n].vector != VIRTIO_NO_VECTOR) {
1557             QLIST_REMOVE(vq, node);
1558         }
1559         vdev->vq[n].vector = vector;
1560         if (vdev->vector_queues &&
1561             vector != VIRTIO_NO_VECTOR) {
1562             QLIST_INSERT_HEAD(&vdev->vector_queues[vector], vq, node);
1563         }
1564     }
1565 }
1566
1567 VirtQueue *virtio_add_queue(VirtIODevice *vdev, int queue_size,
1568                             VirtIOHandleOutput handle_output)
1569 {
1570     int i;
1571
1572     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
1573         if (vdev->vq[i].vring.num == 0)
1574             break;
1575     }
1576
1577     if (i == VIRTIO_QUEUE_MAX || queue_size > VIRTQUEUE_MAX_SIZE)
1578         abort();
1579
1580     vdev->vq[i].vring.num = queue_size;
1581     vdev->vq[i].vring.num_default = queue_size;
1582     vdev->vq[i].vring.align = VIRTIO_PCI_VRING_ALIGN;
1583     vdev->vq[i].handle_output = handle_output;
1584     vdev->vq[i].handle_aio_output = NULL;
1585
1586     return &vdev->vq[i];
1587 }
1588
1589 void virtio_del_queue(VirtIODevice *vdev, int n)
1590 {
1591     if (n < 0 || n >= VIRTIO_QUEUE_MAX) {
1592         abort();
1593     }
1594
1595     vdev->vq[n].vring.num = 0;
1596     vdev->vq[n].vring.num_default = 0;
1597 }
1598
1599 static void virtio_set_isr(VirtIODevice *vdev, int value)
1600 {
1601     uint8_t old = atomic_read(&vdev->isr);
1602
1603     /* Do not write ISR if it does not change, so that its cacheline remains
1604      * shared in the common case where the guest does not read it.
1605      */
1606     if ((old & value) != value) {
1607         atomic_or(&vdev->isr, value);
1608     }
1609 }
1610
1611 /* Called within rcu_read_lock().  */
1612 static bool virtio_should_notify(VirtIODevice *vdev, VirtQueue *vq)
1613 {
1614     uint16_t old, new;
1615     bool v;
1616     /* We need to expose used array entries before checking used event. */
1617     smp_mb();
1618     /* Always notify when queue is empty (when feature acknowledge) */
1619     if (virtio_vdev_has_feature(vdev, VIRTIO_F_NOTIFY_ON_EMPTY) &&
1620         !vq->inuse && virtio_queue_empty(vq)) {
1621         return true;
1622     }
1623
1624     if (!virtio_vdev_has_feature(vdev, VIRTIO_RING_F_EVENT_IDX)) {
1625         return !(vring_avail_flags(vq) & VRING_AVAIL_F_NO_INTERRUPT);
1626     }
1627
1628     v = vq->signalled_used_valid;
1629     vq->signalled_used_valid = true;
1630     old = vq->signalled_used;
1631     new = vq->signalled_used = vq->used_idx;
1632     return !v || vring_need_event(vring_get_used_event(vq), new, old);
1633 }
1634
1635 void virtio_notify_irqfd(VirtIODevice *vdev, VirtQueue *vq)
1636 {
1637     bool should_notify;
1638     rcu_read_lock();
1639     should_notify = virtio_should_notify(vdev, vq);
1640     rcu_read_unlock();
1641
1642     if (!should_notify) {
1643         return;
1644     }
1645
1646     trace_virtio_notify_irqfd(vdev, vq);
1647
1648     /*
1649      * virtio spec 1.0 says ISR bit 0 should be ignored with MSI, but
1650      * windows drivers included in virtio-win 1.8.0 (circa 2015) are
1651      * incorrectly polling this bit during crashdump and hibernation
1652      * in MSI mode, causing a hang if this bit is never updated.
1653      * Recent releases of Windows do not really shut down, but rather
1654      * log out and hibernate to make the next startup faster.  Hence,
1655      * this manifested as a more serious hang during shutdown with
1656      *
1657      * Next driver release from 2016 fixed this problem, so working around it
1658      * is not a must, but it's easy to do so let's do it here.
1659      *
1660      * Note: it's safe to update ISR from any thread as it was switched
1661      * to an atomic operation.
1662      */
1663     virtio_set_isr(vq->vdev, 0x1);
1664     event_notifier_set(&vq->guest_notifier);
1665 }
1666
1667 static void virtio_irq(VirtQueue *vq)
1668 {
1669     virtio_set_isr(vq->vdev, 0x1);
1670     virtio_notify_vector(vq->vdev, vq->vector);
1671 }
1672
1673 void virtio_notify(VirtIODevice *vdev, VirtQueue *vq)
1674 {
1675     bool should_notify;
1676     rcu_read_lock();
1677     should_notify = virtio_should_notify(vdev, vq);
1678     rcu_read_unlock();
1679
1680     if (!should_notify) {
1681         return;
1682     }
1683
1684     trace_virtio_notify(vdev, vq);
1685     virtio_irq(vq);
1686 }
1687
1688 void virtio_notify_config(VirtIODevice *vdev)
1689 {
1690     if (!(vdev->status & VIRTIO_CONFIG_S_DRIVER_OK))
1691         return;
1692
1693     virtio_set_isr(vdev, 0x3);
1694     vdev->generation++;
1695     virtio_notify_vector(vdev, vdev->config_vector);
1696 }
1697
1698 static bool virtio_device_endian_needed(void *opaque)
1699 {
1700     VirtIODevice *vdev = opaque;
1701
1702     assert(vdev->device_endian != VIRTIO_DEVICE_ENDIAN_UNKNOWN);
1703     if (!virtio_vdev_has_feature(vdev, VIRTIO_F_VERSION_1)) {
1704         return vdev->device_endian != virtio_default_endian();
1705     }
1706     /* Devices conforming to VIRTIO 1.0 or later are always LE. */
1707     return vdev->device_endian != VIRTIO_DEVICE_ENDIAN_LITTLE;
1708 }
1709
1710 static bool virtio_64bit_features_needed(void *opaque)
1711 {
1712     VirtIODevice *vdev = opaque;
1713
1714     return (vdev->host_features >> 32) != 0;
1715 }
1716
1717 static bool virtio_virtqueue_needed(void *opaque)
1718 {
1719     VirtIODevice *vdev = opaque;
1720
1721     return virtio_host_has_feature(vdev, VIRTIO_F_VERSION_1);
1722 }
1723
1724 static bool virtio_ringsize_needed(void *opaque)
1725 {
1726     VirtIODevice *vdev = opaque;
1727     int i;
1728
1729     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
1730         if (vdev->vq[i].vring.num != vdev->vq[i].vring.num_default) {
1731             return true;
1732         }
1733     }
1734     return false;
1735 }
1736
1737 static bool virtio_extra_state_needed(void *opaque)
1738 {
1739     VirtIODevice *vdev = opaque;
1740     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
1741     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
1742
1743     return k->has_extra_state &&
1744         k->has_extra_state(qbus->parent);
1745 }
1746
1747 static bool virtio_broken_needed(void *opaque)
1748 {
1749     VirtIODevice *vdev = opaque;
1750
1751     return vdev->broken;
1752 }
1753
1754 static const VMStateDescription vmstate_virtqueue = {
1755     .name = "virtqueue_state",
1756     .version_id = 1,
1757     .minimum_version_id = 1,
1758     .fields = (VMStateField[]) {
1759         VMSTATE_UINT64(vring.avail, struct VirtQueue),
1760         VMSTATE_UINT64(vring.used, struct VirtQueue),
1761         VMSTATE_END_OF_LIST()
1762     }
1763 };
1764
1765 static const VMStateDescription vmstate_virtio_virtqueues = {
1766     .name = "virtio/virtqueues",
1767     .version_id = 1,
1768     .minimum_version_id = 1,
1769     .needed = &virtio_virtqueue_needed,
1770     .fields = (VMStateField[]) {
1771         VMSTATE_STRUCT_VARRAY_POINTER_KNOWN(vq, struct VirtIODevice,
1772                       VIRTIO_QUEUE_MAX, 0, vmstate_virtqueue, VirtQueue),
1773         VMSTATE_END_OF_LIST()
1774     }
1775 };
1776
1777 static const VMStateDescription vmstate_ringsize = {
1778     .name = "ringsize_state",
1779     .version_id = 1,
1780     .minimum_version_id = 1,
1781     .fields = (VMStateField[]) {
1782         VMSTATE_UINT32(vring.num_default, struct VirtQueue),
1783         VMSTATE_END_OF_LIST()
1784     }
1785 };
1786
1787 static const VMStateDescription vmstate_virtio_ringsize = {
1788     .name = "virtio/ringsize",
1789     .version_id = 1,
1790     .minimum_version_id = 1,
1791     .needed = &virtio_ringsize_needed,
1792     .fields = (VMStateField[]) {
1793         VMSTATE_STRUCT_VARRAY_POINTER_KNOWN(vq, struct VirtIODevice,
1794                       VIRTIO_QUEUE_MAX, 0, vmstate_ringsize, VirtQueue),
1795         VMSTATE_END_OF_LIST()
1796     }
1797 };
1798
1799 static int get_extra_state(QEMUFile *f, void *pv, size_t size,
1800                            VMStateField *field)
1801 {
1802     VirtIODevice *vdev = pv;
1803     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
1804     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
1805
1806     if (!k->load_extra_state) {
1807         return -1;
1808     } else {
1809         return k->load_extra_state(qbus->parent, f);
1810     }
1811 }
1812
1813 static int put_extra_state(QEMUFile *f, void *pv, size_t size,
1814                            VMStateField *field, QJSON *vmdesc)
1815 {
1816     VirtIODevice *vdev = pv;
1817     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
1818     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
1819
1820     k->save_extra_state(qbus->parent, f);
1821     return 0;
1822 }
1823
1824 static const VMStateInfo vmstate_info_extra_state = {
1825     .name = "virtqueue_extra_state",
1826     .get = get_extra_state,
1827     .put = put_extra_state,
1828 };
1829
1830 static const VMStateDescription vmstate_virtio_extra_state = {
1831     .name = "virtio/extra_state",
1832     .version_id = 1,
1833     .minimum_version_id = 1,
1834     .needed = &virtio_extra_state_needed,
1835     .fields = (VMStateField[]) {
1836         {
1837             .name         = "extra_state",
1838             .version_id   = 0,
1839             .field_exists = NULL,
1840             .size         = 0,
1841             .info         = &vmstate_info_extra_state,
1842             .flags        = VMS_SINGLE,
1843             .offset       = 0,
1844         },
1845         VMSTATE_END_OF_LIST()
1846     }
1847 };
1848
1849 static const VMStateDescription vmstate_virtio_device_endian = {
1850     .name = "virtio/device_endian",
1851     .version_id = 1,
1852     .minimum_version_id = 1,
1853     .needed = &virtio_device_endian_needed,
1854     .fields = (VMStateField[]) {
1855         VMSTATE_UINT8(device_endian, VirtIODevice),
1856         VMSTATE_END_OF_LIST()
1857     }
1858 };
1859
1860 static const VMStateDescription vmstate_virtio_64bit_features = {
1861     .name = "virtio/64bit_features",
1862     .version_id = 1,
1863     .minimum_version_id = 1,
1864     .needed = &virtio_64bit_features_needed,
1865     .fields = (VMStateField[]) {
1866         VMSTATE_UINT64(guest_features, VirtIODevice),
1867         VMSTATE_END_OF_LIST()
1868     }
1869 };
1870
1871 static const VMStateDescription vmstate_virtio_broken = {
1872     .name = "virtio/broken",
1873     .version_id = 1,
1874     .minimum_version_id = 1,
1875     .needed = &virtio_broken_needed,
1876     .fields = (VMStateField[]) {
1877         VMSTATE_BOOL(broken, VirtIODevice),
1878         VMSTATE_END_OF_LIST()
1879     }
1880 };
1881
1882 static const VMStateDescription vmstate_virtio = {
1883     .name = "virtio",
1884     .version_id = 1,
1885     .minimum_version_id = 1,
1886     .minimum_version_id_old = 1,
1887     .fields = (VMStateField[]) {
1888         VMSTATE_END_OF_LIST()
1889     },
1890     .subsections = (const VMStateDescription*[]) {
1891         &vmstate_virtio_device_endian,
1892         &vmstate_virtio_64bit_features,
1893         &vmstate_virtio_virtqueues,
1894         &vmstate_virtio_ringsize,
1895         &vmstate_virtio_broken,
1896         &vmstate_virtio_extra_state,
1897         NULL
1898     }
1899 };
1900
1901 void virtio_save(VirtIODevice *vdev, QEMUFile *f)
1902 {
1903     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
1904     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
1905     VirtioDeviceClass *vdc = VIRTIO_DEVICE_GET_CLASS(vdev);
1906     uint32_t guest_features_lo = (vdev->guest_features & 0xffffffff);
1907     int i;
1908
1909     if (k->save_config) {
1910         k->save_config(qbus->parent, f);
1911     }
1912
1913     qemu_put_8s(f, &vdev->status);
1914     qemu_put_8s(f, &vdev->isr);
1915     qemu_put_be16s(f, &vdev->queue_sel);
1916     qemu_put_be32s(f, &guest_features_lo);
1917     qemu_put_be32(f, vdev->config_len);
1918     qemu_put_buffer(f, vdev->config, vdev->config_len);
1919
1920     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
1921         if (vdev->vq[i].vring.num == 0)
1922             break;
1923     }
1924
1925     qemu_put_be32(f, i);
1926
1927     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
1928         if (vdev->vq[i].vring.num == 0)
1929             break;
1930
1931         qemu_put_be32(f, vdev->vq[i].vring.num);
1932         if (k->has_variable_vring_alignment) {
1933             qemu_put_be32(f, vdev->vq[i].vring.align);
1934         }
1935         /*
1936          * Save desc now, the rest of the ring addresses are saved in
1937          * subsections for VIRTIO-1 devices.
1938          */
1939         qemu_put_be64(f, vdev->vq[i].vring.desc);
1940         qemu_put_be16s(f, &vdev->vq[i].last_avail_idx);
1941         if (k->save_queue) {
1942             k->save_queue(qbus->parent, i, f);
1943         }
1944     }
1945
1946     if (vdc->save != NULL) {
1947         vdc->save(vdev, f);
1948     }
1949
1950     if (vdc->vmsd) {
1951         vmstate_save_state(f, vdc->vmsd, vdev, NULL);
1952     }
1953
1954     /* Subsections */
1955     vmstate_save_state(f, &vmstate_virtio, vdev, NULL);
1956 }
1957
1958 /* A wrapper for use as a VMState .put function */
1959 static int virtio_device_put(QEMUFile *f, void *opaque, size_t size,
1960                               VMStateField *field, QJSON *vmdesc)
1961 {
1962     virtio_save(VIRTIO_DEVICE(opaque), f);
1963
1964     return 0;
1965 }
1966
1967 /* A wrapper for use as a VMState .get function */
1968 static int virtio_device_get(QEMUFile *f, void *opaque, size_t size,
1969                              VMStateField *field)
1970 {
1971     VirtIODevice *vdev = VIRTIO_DEVICE(opaque);
1972     DeviceClass *dc = DEVICE_CLASS(VIRTIO_DEVICE_GET_CLASS(vdev));
1973
1974     return virtio_load(vdev, f, dc->vmsd->version_id);
1975 }
1976
1977 const VMStateInfo  virtio_vmstate_info = {
1978     .name = "virtio",
1979     .get = virtio_device_get,
1980     .put = virtio_device_put,
1981 };
1982
1983 static int virtio_set_features_nocheck(VirtIODevice *vdev, uint64_t val)
1984 {
1985     VirtioDeviceClass *k = VIRTIO_DEVICE_GET_CLASS(vdev);
1986     bool bad = (val & ~(vdev->host_features)) != 0;
1987
1988     val &= vdev->host_features;
1989     if (k->set_features) {
1990         k->set_features(vdev, val);
1991     }
1992     vdev->guest_features = val;
1993     return bad ? -1 : 0;
1994 }
1995
1996 int virtio_set_features(VirtIODevice *vdev, uint64_t val)
1997 {
1998    /*
1999      * The driver must not attempt to set features after feature negotiation
2000      * has finished.
2001      */
2002     if (vdev->status & VIRTIO_CONFIG_S_FEATURES_OK) {
2003         return -EINVAL;
2004     }
2005     return virtio_set_features_nocheck(vdev, val);
2006 }
2007
2008 int virtio_load(VirtIODevice *vdev, QEMUFile *f, int version_id)
2009 {
2010     int i, ret;
2011     int32_t config_len;
2012     uint32_t num;
2013     uint32_t features;
2014     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2015     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
2016     VirtioDeviceClass *vdc = VIRTIO_DEVICE_GET_CLASS(vdev);
2017
2018     /*
2019      * We poison the endianness to ensure it does not get used before
2020      * subsections have been loaded.
2021      */
2022     vdev->device_endian = VIRTIO_DEVICE_ENDIAN_UNKNOWN;
2023
2024     if (k->load_config) {
2025         ret = k->load_config(qbus->parent, f);
2026         if (ret)
2027             return ret;
2028     }
2029
2030     qemu_get_8s(f, &vdev->status);
2031     qemu_get_8s(f, &vdev->isr);
2032     qemu_get_be16s(f, &vdev->queue_sel);
2033     if (vdev->queue_sel >= VIRTIO_QUEUE_MAX) {
2034         return -1;
2035     }
2036     qemu_get_be32s(f, &features);
2037
2038     /*
2039      * Temporarily set guest_features low bits - needed by
2040      * virtio net load code testing for VIRTIO_NET_F_CTRL_GUEST_OFFLOADS
2041      * VIRTIO_NET_F_GUEST_ANNOUNCE and VIRTIO_NET_F_CTRL_VQ.
2042      *
2043      * Note: devices should always test host features in future - don't create
2044      * new dependencies like this.
2045      */
2046     vdev->guest_features = features;
2047
2048     config_len = qemu_get_be32(f);
2049
2050     /*
2051      * There are cases where the incoming config can be bigger or smaller
2052      * than what we have; so load what we have space for, and skip
2053      * any excess that's in the stream.
2054      */
2055     qemu_get_buffer(f, vdev->config, MIN(config_len, vdev->config_len));
2056
2057     while (config_len > vdev->config_len) {
2058         qemu_get_byte(f);
2059         config_len--;
2060     }
2061
2062     num = qemu_get_be32(f);
2063
2064     if (num > VIRTIO_QUEUE_MAX) {
2065         error_report("Invalid number of virtqueues: 0x%x", num);
2066         return -1;
2067     }
2068
2069     for (i = 0; i < num; i++) {
2070         vdev->vq[i].vring.num = qemu_get_be32(f);
2071         if (k->has_variable_vring_alignment) {
2072             vdev->vq[i].vring.align = qemu_get_be32(f);
2073         }
2074         vdev->vq[i].vring.desc = qemu_get_be64(f);
2075         qemu_get_be16s(f, &vdev->vq[i].last_avail_idx);
2076         vdev->vq[i].signalled_used_valid = false;
2077         vdev->vq[i].notification = true;
2078
2079         if (!vdev->vq[i].vring.desc && vdev->vq[i].last_avail_idx) {
2080             error_report("VQ %d address 0x0 "
2081                          "inconsistent with Host index 0x%x",
2082                          i, vdev->vq[i].last_avail_idx);
2083             return -1;
2084         }
2085         if (k->load_queue) {
2086             ret = k->load_queue(qbus->parent, i, f);
2087             if (ret)
2088                 return ret;
2089         }
2090     }
2091
2092     virtio_notify_vector(vdev, VIRTIO_NO_VECTOR);
2093
2094     if (vdc->load != NULL) {
2095         ret = vdc->load(vdev, f, version_id);
2096         if (ret) {
2097             return ret;
2098         }
2099     }
2100
2101     if (vdc->vmsd) {
2102         ret = vmstate_load_state(f, vdc->vmsd, vdev, version_id);
2103         if (ret) {
2104             return ret;
2105         }
2106     }
2107
2108     /* Subsections */
2109     ret = vmstate_load_state(f, &vmstate_virtio, vdev, 1);
2110     if (ret) {
2111         return ret;
2112     }
2113
2114     if (vdev->device_endian == VIRTIO_DEVICE_ENDIAN_UNKNOWN) {
2115         vdev->device_endian = virtio_default_endian();
2116     }
2117
2118     if (virtio_64bit_features_needed(vdev)) {
2119         /*
2120          * Subsection load filled vdev->guest_features.  Run them
2121          * through virtio_set_features to sanity-check them against
2122          * host_features.
2123          */
2124         uint64_t features64 = vdev->guest_features;
2125         if (virtio_set_features_nocheck(vdev, features64) < 0) {
2126             error_report("Features 0x%" PRIx64 " unsupported. "
2127                          "Allowed features: 0x%" PRIx64,
2128                          features64, vdev->host_features);
2129             return -1;
2130         }
2131     } else {
2132         if (virtio_set_features_nocheck(vdev, features) < 0) {
2133             error_report("Features 0x%x unsupported. "
2134                          "Allowed features: 0x%" PRIx64,
2135                          features, vdev->host_features);
2136             return -1;
2137         }
2138     }
2139
2140     rcu_read_lock();
2141     for (i = 0; i < num; i++) {
2142         if (vdev->vq[i].vring.desc) {
2143             uint16_t nheads;
2144
2145             /*
2146              * VIRTIO-1 devices migrate desc, used, and avail ring addresses so
2147              * only the region cache needs to be set up.  Legacy devices need
2148              * to calculate used and avail ring addresses based on the desc
2149              * address.
2150              */
2151             if (virtio_vdev_has_feature(vdev, VIRTIO_F_VERSION_1)) {
2152                 virtio_init_region_cache(vdev, i);
2153             } else {
2154                 virtio_queue_update_rings(vdev, i);
2155             }
2156
2157             nheads = vring_avail_idx(&vdev->vq[i]) - vdev->vq[i].last_avail_idx;
2158             /* Check it isn't doing strange things with descriptor numbers. */
2159             if (nheads > vdev->vq[i].vring.num) {
2160                 error_report("VQ %d size 0x%x Guest index 0x%x "
2161                              "inconsistent with Host index 0x%x: delta 0x%x",
2162                              i, vdev->vq[i].vring.num,
2163                              vring_avail_idx(&vdev->vq[i]),
2164                              vdev->vq[i].last_avail_idx, nheads);
2165                 return -1;
2166             }
2167             vdev->vq[i].used_idx = vring_used_idx(&vdev->vq[i]);
2168             vdev->vq[i].shadow_avail_idx = vring_avail_idx(&vdev->vq[i]);
2169
2170             /*
2171              * Some devices migrate VirtQueueElements that have been popped
2172              * from the avail ring but not yet returned to the used ring.
2173              * Since max ring size < UINT16_MAX it's safe to use modulo
2174              * UINT16_MAX + 1 subtraction.
2175              */
2176             vdev->vq[i].inuse = (uint16_t)(vdev->vq[i].last_avail_idx -
2177                                 vdev->vq[i].used_idx);
2178             if (vdev->vq[i].inuse > vdev->vq[i].vring.num) {
2179                 error_report("VQ %d size 0x%x < last_avail_idx 0x%x - "
2180                              "used_idx 0x%x",
2181                              i, vdev->vq[i].vring.num,
2182                              vdev->vq[i].last_avail_idx,
2183                              vdev->vq[i].used_idx);
2184                 return -1;
2185             }
2186         }
2187     }
2188     rcu_read_unlock();
2189
2190     return 0;
2191 }
2192
2193 void virtio_cleanup(VirtIODevice *vdev)
2194 {
2195     qemu_del_vm_change_state_handler(vdev->vmstate);
2196 }
2197
2198 static void virtio_vmstate_change(void *opaque, int running, RunState state)
2199 {
2200     VirtIODevice *vdev = opaque;
2201     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2202     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
2203     bool backend_run = running && (vdev->status & VIRTIO_CONFIG_S_DRIVER_OK);
2204     vdev->vm_running = running;
2205
2206     if (backend_run) {
2207         virtio_set_status(vdev, vdev->status);
2208     }
2209
2210     if (k->vmstate_change) {
2211         k->vmstate_change(qbus->parent, backend_run);
2212     }
2213
2214     if (!backend_run) {
2215         virtio_set_status(vdev, vdev->status);
2216     }
2217 }
2218
2219 void virtio_instance_init_common(Object *proxy_obj, void *data,
2220                                  size_t vdev_size, const char *vdev_name)
2221 {
2222     DeviceState *vdev = data;
2223
2224     object_initialize(vdev, vdev_size, vdev_name);
2225     object_property_add_child(proxy_obj, "virtio-backend", OBJECT(vdev), NULL);
2226     object_unref(OBJECT(vdev));
2227     qdev_alias_all_properties(vdev, proxy_obj);
2228 }
2229
2230 void virtio_init(VirtIODevice *vdev, const char *name,
2231                  uint16_t device_id, size_t config_size)
2232 {
2233     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2234     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(qbus);
2235     int i;
2236     int nvectors = k->query_nvectors ? k->query_nvectors(qbus->parent) : 0;
2237
2238     if (nvectors) {
2239         vdev->vector_queues =
2240             g_malloc0(sizeof(*vdev->vector_queues) * nvectors);
2241     }
2242
2243     vdev->device_id = device_id;
2244     vdev->status = 0;
2245     atomic_set(&vdev->isr, 0);
2246     vdev->queue_sel = 0;
2247     vdev->config_vector = VIRTIO_NO_VECTOR;
2248     vdev->vq = g_malloc0(sizeof(VirtQueue) * VIRTIO_QUEUE_MAX);
2249     vdev->vm_running = runstate_is_running();
2250     vdev->broken = false;
2251     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
2252         vdev->vq[i].vector = VIRTIO_NO_VECTOR;
2253         vdev->vq[i].vdev = vdev;
2254         vdev->vq[i].queue_index = i;
2255     }
2256
2257     vdev->name = name;
2258     vdev->config_len = config_size;
2259     if (vdev->config_len) {
2260         vdev->config = g_malloc0(config_size);
2261     } else {
2262         vdev->config = NULL;
2263     }
2264     vdev->vmstate = qemu_add_vm_change_state_handler(virtio_vmstate_change,
2265                                                      vdev);
2266     vdev->device_endian = virtio_default_endian();
2267     vdev->use_guest_notifier_mask = true;
2268 }
2269
2270 hwaddr virtio_queue_get_desc_addr(VirtIODevice *vdev, int n)
2271 {
2272     return vdev->vq[n].vring.desc;
2273 }
2274
2275 hwaddr virtio_queue_get_avail_addr(VirtIODevice *vdev, int n)
2276 {
2277     return vdev->vq[n].vring.avail;
2278 }
2279
2280 hwaddr virtio_queue_get_used_addr(VirtIODevice *vdev, int n)
2281 {
2282     return vdev->vq[n].vring.used;
2283 }
2284
2285 hwaddr virtio_queue_get_desc_size(VirtIODevice *vdev, int n)
2286 {
2287     return sizeof(VRingDesc) * vdev->vq[n].vring.num;
2288 }
2289
2290 hwaddr virtio_queue_get_avail_size(VirtIODevice *vdev, int n)
2291 {
2292     return offsetof(VRingAvail, ring) +
2293         sizeof(uint16_t) * vdev->vq[n].vring.num;
2294 }
2295
2296 hwaddr virtio_queue_get_used_size(VirtIODevice *vdev, int n)
2297 {
2298     return offsetof(VRingUsed, ring) +
2299         sizeof(VRingUsedElem) * vdev->vq[n].vring.num;
2300 }
2301
2302 uint16_t virtio_queue_get_last_avail_idx(VirtIODevice *vdev, int n)
2303 {
2304     return vdev->vq[n].last_avail_idx;
2305 }
2306
2307 void virtio_queue_set_last_avail_idx(VirtIODevice *vdev, int n, uint16_t idx)
2308 {
2309     vdev->vq[n].last_avail_idx = idx;
2310     vdev->vq[n].shadow_avail_idx = idx;
2311 }
2312
2313 void virtio_queue_update_used_idx(VirtIODevice *vdev, int n)
2314 {
2315     rcu_read_lock();
2316     if (vdev->vq[n].vring.desc) {
2317         vdev->vq[n].used_idx = vring_used_idx(&vdev->vq[n]);
2318     }
2319     rcu_read_unlock();
2320 }
2321
2322 void virtio_queue_invalidate_signalled_used(VirtIODevice *vdev, int n)
2323 {
2324     vdev->vq[n].signalled_used_valid = false;
2325 }
2326
2327 VirtQueue *virtio_get_queue(VirtIODevice *vdev, int n)
2328 {
2329     return vdev->vq + n;
2330 }
2331
2332 uint16_t virtio_get_queue_index(VirtQueue *vq)
2333 {
2334     return vq->queue_index;
2335 }
2336
2337 static void virtio_queue_guest_notifier_read(EventNotifier *n)
2338 {
2339     VirtQueue *vq = container_of(n, VirtQueue, guest_notifier);
2340     if (event_notifier_test_and_clear(n)) {
2341         virtio_irq(vq);
2342     }
2343 }
2344
2345 void virtio_queue_set_guest_notifier_fd_handler(VirtQueue *vq, bool assign,
2346                                                 bool with_irqfd)
2347 {
2348     if (assign && !with_irqfd) {
2349         event_notifier_set_handler(&vq->guest_notifier,
2350                                    virtio_queue_guest_notifier_read);
2351     } else {
2352         event_notifier_set_handler(&vq->guest_notifier, NULL);
2353     }
2354     if (!assign) {
2355         /* Test and clear notifier before closing it,
2356          * in case poll callback didn't have time to run. */
2357         virtio_queue_guest_notifier_read(&vq->guest_notifier);
2358     }
2359 }
2360
2361 EventNotifier *virtio_queue_get_guest_notifier(VirtQueue *vq)
2362 {
2363     return &vq->guest_notifier;
2364 }
2365
2366 static void virtio_queue_host_notifier_aio_read(EventNotifier *n)
2367 {
2368     VirtQueue *vq = container_of(n, VirtQueue, host_notifier);
2369     if (event_notifier_test_and_clear(n)) {
2370         virtio_queue_notify_aio_vq(vq);
2371     }
2372 }
2373
2374 static void virtio_queue_host_notifier_aio_poll_begin(EventNotifier *n)
2375 {
2376     VirtQueue *vq = container_of(n, VirtQueue, host_notifier);
2377
2378     virtio_queue_set_notification(vq, 0);
2379 }
2380
2381 static bool virtio_queue_host_notifier_aio_poll(void *opaque)
2382 {
2383     EventNotifier *n = opaque;
2384     VirtQueue *vq = container_of(n, VirtQueue, host_notifier);
2385     bool progress;
2386
2387     if (!vq->vring.desc || virtio_queue_empty(vq)) {
2388         return false;
2389     }
2390
2391     progress = virtio_queue_notify_aio_vq(vq);
2392
2393     /* In case the handler function re-enabled notifications */
2394     virtio_queue_set_notification(vq, 0);
2395     return progress;
2396 }
2397
2398 static void virtio_queue_host_notifier_aio_poll_end(EventNotifier *n)
2399 {
2400     VirtQueue *vq = container_of(n, VirtQueue, host_notifier);
2401
2402     /* Caller polls once more after this to catch requests that race with us */
2403     virtio_queue_set_notification(vq, 1);
2404 }
2405
2406 void virtio_queue_aio_set_host_notifier_handler(VirtQueue *vq, AioContext *ctx,
2407                                                 VirtIOHandleAIOOutput handle_output)
2408 {
2409     if (handle_output) {
2410         vq->handle_aio_output = handle_output;
2411         aio_set_event_notifier(ctx, &vq->host_notifier, true,
2412                                virtio_queue_host_notifier_aio_read,
2413                                virtio_queue_host_notifier_aio_poll);
2414         aio_set_event_notifier_poll(ctx, &vq->host_notifier,
2415                                     virtio_queue_host_notifier_aio_poll_begin,
2416                                     virtio_queue_host_notifier_aio_poll_end);
2417     } else {
2418         aio_set_event_notifier(ctx, &vq->host_notifier, true, NULL, NULL);
2419         /* Test and clear notifier before after disabling event,
2420          * in case poll callback didn't have time to run. */
2421         virtio_queue_host_notifier_aio_read(&vq->host_notifier);
2422         vq->handle_aio_output = NULL;
2423     }
2424 }
2425
2426 void virtio_queue_host_notifier_read(EventNotifier *n)
2427 {
2428     VirtQueue *vq = container_of(n, VirtQueue, host_notifier);
2429     if (event_notifier_test_and_clear(n)) {
2430         virtio_queue_notify_vq(vq);
2431     }
2432 }
2433
2434 EventNotifier *virtio_queue_get_host_notifier(VirtQueue *vq)
2435 {
2436     return &vq->host_notifier;
2437 }
2438
2439 void virtio_device_set_child_bus_name(VirtIODevice *vdev, char *bus_name)
2440 {
2441     g_free(vdev->bus_name);
2442     vdev->bus_name = g_strdup(bus_name);
2443 }
2444
2445 void GCC_FMT_ATTR(2, 3) virtio_error(VirtIODevice *vdev, const char *fmt, ...)
2446 {
2447     va_list ap;
2448
2449     va_start(ap, fmt);
2450     error_vreport(fmt, ap);
2451     va_end(ap);
2452
2453     if (virtio_vdev_has_feature(vdev, VIRTIO_F_VERSION_1)) {
2454         virtio_set_status(vdev, vdev->status | VIRTIO_CONFIG_S_NEEDS_RESET);
2455         virtio_notify_config(vdev);
2456     }
2457
2458     vdev->broken = true;
2459 }
2460
2461 static void virtio_memory_listener_commit(MemoryListener *listener)
2462 {
2463     VirtIODevice *vdev = container_of(listener, VirtIODevice, listener);
2464     int i;
2465
2466     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
2467         if (vdev->vq[i].vring.num == 0) {
2468             break;
2469         }
2470         virtio_init_region_cache(vdev, i);
2471     }
2472 }
2473
2474 static void virtio_device_realize(DeviceState *dev, Error **errp)
2475 {
2476     VirtIODevice *vdev = VIRTIO_DEVICE(dev);
2477     VirtioDeviceClass *vdc = VIRTIO_DEVICE_GET_CLASS(dev);
2478     Error *err = NULL;
2479
2480     /* Devices should either use vmsd or the load/save methods */
2481     assert(!vdc->vmsd || !vdc->load);
2482
2483     if (vdc->realize != NULL) {
2484         vdc->realize(dev, &err);
2485         if (err != NULL) {
2486             error_propagate(errp, err);
2487             return;
2488         }
2489     }
2490
2491     virtio_bus_device_plugged(vdev, &err);
2492     if (err != NULL) {
2493         error_propagate(errp, err);
2494         return;
2495     }
2496
2497     vdev->listener.commit = virtio_memory_listener_commit;
2498     memory_listener_register(&vdev->listener, vdev->dma_as);
2499 }
2500
2501 static void virtio_device_unrealize(DeviceState *dev, Error **errp)
2502 {
2503     VirtIODevice *vdev = VIRTIO_DEVICE(dev);
2504     VirtioDeviceClass *vdc = VIRTIO_DEVICE_GET_CLASS(dev);
2505     Error *err = NULL;
2506
2507     virtio_bus_device_unplugged(vdev);
2508
2509     if (vdc->unrealize != NULL) {
2510         vdc->unrealize(dev, &err);
2511         if (err != NULL) {
2512             error_propagate(errp, err);
2513             return;
2514         }
2515     }
2516
2517     g_free(vdev->bus_name);
2518     vdev->bus_name = NULL;
2519 }
2520
2521 static void virtio_device_free_virtqueues(VirtIODevice *vdev)
2522 {
2523     int i;
2524     if (!vdev->vq) {
2525         return;
2526     }
2527
2528     for (i = 0; i < VIRTIO_QUEUE_MAX; i++) {
2529         if (vdev->vq[i].vring.num == 0) {
2530             break;
2531         }
2532         virtio_virtqueue_reset_region_cache(&vdev->vq[i]);
2533     }
2534     g_free(vdev->vq);
2535 }
2536
2537 static void virtio_device_instance_finalize(Object *obj)
2538 {
2539     VirtIODevice *vdev = VIRTIO_DEVICE(obj);
2540
2541     memory_listener_unregister(&vdev->listener);
2542     virtio_device_free_virtqueues(vdev);
2543
2544     g_free(vdev->config);
2545     g_free(vdev->vector_queues);
2546 }
2547
2548 static Property virtio_properties[] = {
2549     DEFINE_VIRTIO_COMMON_FEATURES(VirtIODevice, host_features),
2550     DEFINE_PROP_END_OF_LIST(),
2551 };
2552
2553 static int virtio_device_start_ioeventfd_impl(VirtIODevice *vdev)
2554 {
2555     VirtioBusState *qbus = VIRTIO_BUS(qdev_get_parent_bus(DEVICE(vdev)));
2556     int n, r, err;
2557
2558     for (n = 0; n < VIRTIO_QUEUE_MAX; n++) {
2559         VirtQueue *vq = &vdev->vq[n];
2560         if (!virtio_queue_get_num(vdev, n)) {
2561             continue;
2562         }
2563         r = virtio_bus_set_host_notifier(qbus, n, true);
2564         if (r < 0) {
2565             err = r;
2566             goto assign_error;
2567         }
2568         event_notifier_set_handler(&vq->host_notifier,
2569                                    virtio_queue_host_notifier_read);
2570     }
2571
2572     for (n = 0; n < VIRTIO_QUEUE_MAX; n++) {
2573         /* Kick right away to begin processing requests already in vring */
2574         VirtQueue *vq = &vdev->vq[n];
2575         if (!vq->vring.num) {
2576             continue;
2577         }
2578         event_notifier_set(&vq->host_notifier);
2579     }
2580     return 0;
2581
2582 assign_error:
2583     while (--n >= 0) {
2584         VirtQueue *vq = &vdev->vq[n];
2585         if (!virtio_queue_get_num(vdev, n)) {
2586             continue;
2587         }
2588
2589         event_notifier_set_handler(&vq->host_notifier, NULL);
2590         r = virtio_bus_set_host_notifier(qbus, n, false);
2591         assert(r >= 0);
2592     }
2593     return err;
2594 }
2595
2596 int virtio_device_start_ioeventfd(VirtIODevice *vdev)
2597 {
2598     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2599     VirtioBusState *vbus = VIRTIO_BUS(qbus);
2600
2601     return virtio_bus_start_ioeventfd(vbus);
2602 }
2603
2604 static void virtio_device_stop_ioeventfd_impl(VirtIODevice *vdev)
2605 {
2606     VirtioBusState *qbus = VIRTIO_BUS(qdev_get_parent_bus(DEVICE(vdev)));
2607     int n, r;
2608
2609     for (n = 0; n < VIRTIO_QUEUE_MAX; n++) {
2610         VirtQueue *vq = &vdev->vq[n];
2611
2612         if (!virtio_queue_get_num(vdev, n)) {
2613             continue;
2614         }
2615         event_notifier_set_handler(&vq->host_notifier, NULL);
2616         r = virtio_bus_set_host_notifier(qbus, n, false);
2617         assert(r >= 0);
2618     }
2619 }
2620
2621 void virtio_device_stop_ioeventfd(VirtIODevice *vdev)
2622 {
2623     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2624     VirtioBusState *vbus = VIRTIO_BUS(qbus);
2625
2626     virtio_bus_stop_ioeventfd(vbus);
2627 }
2628
2629 int virtio_device_grab_ioeventfd(VirtIODevice *vdev)
2630 {
2631     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2632     VirtioBusState *vbus = VIRTIO_BUS(qbus);
2633
2634     return virtio_bus_grab_ioeventfd(vbus);
2635 }
2636
2637 void virtio_device_release_ioeventfd(VirtIODevice *vdev)
2638 {
2639     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2640     VirtioBusState *vbus = VIRTIO_BUS(qbus);
2641
2642     virtio_bus_release_ioeventfd(vbus);
2643 }
2644
2645 static void virtio_device_class_init(ObjectClass *klass, void *data)
2646 {
2647     /* Set the default value here. */
2648     VirtioDeviceClass *vdc = VIRTIO_DEVICE_CLASS(klass);
2649     DeviceClass *dc = DEVICE_CLASS(klass);
2650
2651     dc->realize = virtio_device_realize;
2652     dc->unrealize = virtio_device_unrealize;
2653     dc->bus_type = TYPE_VIRTIO_BUS;
2654     dc->props = virtio_properties;
2655     vdc->start_ioeventfd = virtio_device_start_ioeventfd_impl;
2656     vdc->stop_ioeventfd = virtio_device_stop_ioeventfd_impl;
2657
2658     vdc->legacy_features |= VIRTIO_LEGACY_FEATURES;
2659 }
2660
2661 bool virtio_device_ioeventfd_enabled(VirtIODevice *vdev)
2662 {
2663     BusState *qbus = qdev_get_parent_bus(DEVICE(vdev));
2664     VirtioBusState *vbus = VIRTIO_BUS(qbus);
2665
2666     return virtio_bus_ioeventfd_enabled(vbus);
2667 }
2668
2669 static const TypeInfo virtio_device_info = {
2670     .name = TYPE_VIRTIO_DEVICE,
2671     .parent = TYPE_DEVICE,
2672     .instance_size = sizeof(VirtIODevice),
2673     .class_init = virtio_device_class_init,
2674     .instance_finalize = virtio_device_instance_finalize,
2675     .abstract = true,
2676     .class_size = sizeof(VirtioDeviceClass),
2677 };
2678
2679 static void virtio_register_types(void)
2680 {
2681     type_register_static(&virtio_device_info);
2682 }
2683
2684 type_init(virtio_register_types)
This page took 0.171124 seconds and 4 git commands to generate.