xref: /openbmc/qemu/hw/virtio/vhost.c (revision 354908ce)
1 /*
2  * vhost support
3  *
4  * Copyright Red Hat, Inc. 2010
5  *
6  * Authors:
7  *  Michael S. Tsirkin <mst@redhat.com>
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  * Contributions after 2012-01-13 are licensed under the terms of the
13  * GNU GPL, version 2 or (at your option) any later version.
14  */
15 
16 #include "qemu/osdep.h"
17 #include "qapi/error.h"
18 #include "hw/virtio/vhost.h"
19 #include "qemu/atomic.h"
20 #include "qemu/range.h"
21 #include "qemu/error-report.h"
22 #include "qemu/memfd.h"
23 #include "standard-headers/linux/vhost_types.h"
24 #include "exec/address-spaces.h"
25 #include "hw/virtio/virtio-bus.h"
26 #include "hw/virtio/virtio-access.h"
27 #include "migration/blocker.h"
28 #include "migration/qemu-file-types.h"
29 #include "sysemu/dma.h"
30 #include "sysemu/tcg.h"
31 #include "trace.h"
32 
33 /* enabled until disconnected backend stabilizes */
34 #define _VHOST_DEBUG 1
35 
36 #ifdef _VHOST_DEBUG
37 #define VHOST_OPS_DEBUG(fmt, ...) \
38     do { error_report(fmt ": %s (%d)", ## __VA_ARGS__, \
39                       strerror(errno), errno); } while (0)
40 #else
41 #define VHOST_OPS_DEBUG(fmt, ...) \
42     do { } while (0)
43 #endif
44 
45 static struct vhost_log *vhost_log;
46 static struct vhost_log *vhost_log_shm;
47 
48 static unsigned int used_memslots;
49 static QLIST_HEAD(, vhost_dev) vhost_devices =
50     QLIST_HEAD_INITIALIZER(vhost_devices);
51 
52 bool vhost_has_free_slot(void)
53 {
54     unsigned int slots_limit = ~0U;
55     struct vhost_dev *hdev;
56 
57     QLIST_FOREACH(hdev, &vhost_devices, entry) {
58         unsigned int r = hdev->vhost_ops->vhost_backend_memslots_limit(hdev);
59         slots_limit = MIN(slots_limit, r);
60     }
61     return slots_limit > used_memslots;
62 }
63 
64 static void vhost_dev_sync_region(struct vhost_dev *dev,
65                                   MemoryRegionSection *section,
66                                   uint64_t mfirst, uint64_t mlast,
67                                   uint64_t rfirst, uint64_t rlast)
68 {
69     vhost_log_chunk_t *log = dev->log->log;
70 
71     uint64_t start = MAX(mfirst, rfirst);
72     uint64_t end = MIN(mlast, rlast);
73     vhost_log_chunk_t *from = log + start / VHOST_LOG_CHUNK;
74     vhost_log_chunk_t *to = log + end / VHOST_LOG_CHUNK + 1;
75     uint64_t addr = QEMU_ALIGN_DOWN(start, VHOST_LOG_CHUNK);
76 
77     if (end < start) {
78         return;
79     }
80     assert(end / VHOST_LOG_CHUNK < dev->log_size);
81     assert(start / VHOST_LOG_CHUNK < dev->log_size);
82 
83     for (;from < to; ++from) {
84         vhost_log_chunk_t log;
85         /* We first check with non-atomic: much cheaper,
86          * and we expect non-dirty to be the common case. */
87         if (!*from) {
88             addr += VHOST_LOG_CHUNK;
89             continue;
90         }
91         /* Data must be read atomically. We don't really need barrier semantics
92          * but it's easier to use atomic_* than roll our own. */
93         log = atomic_xchg(from, 0);
94         while (log) {
95             int bit = ctzl(log);
96             hwaddr page_addr;
97             hwaddr section_offset;
98             hwaddr mr_offset;
99             page_addr = addr + bit * VHOST_LOG_PAGE;
100             section_offset = page_addr - section->offset_within_address_space;
101             mr_offset = section_offset + section->offset_within_region;
102             memory_region_set_dirty(section->mr, mr_offset, VHOST_LOG_PAGE);
103             log &= ~(0x1ull << bit);
104         }
105         addr += VHOST_LOG_CHUNK;
106     }
107 }
108 
109 static int vhost_sync_dirty_bitmap(struct vhost_dev *dev,
110                                    MemoryRegionSection *section,
111                                    hwaddr first,
112                                    hwaddr last)
113 {
114     int i;
115     hwaddr start_addr;
116     hwaddr end_addr;
117 
118     if (!dev->log_enabled || !dev->started) {
119         return 0;
120     }
121     start_addr = section->offset_within_address_space;
122     end_addr = range_get_last(start_addr, int128_get64(section->size));
123     start_addr = MAX(first, start_addr);
124     end_addr = MIN(last, end_addr);
125 
126     for (i = 0; i < dev->mem->nregions; ++i) {
127         struct vhost_memory_region *reg = dev->mem->regions + i;
128         vhost_dev_sync_region(dev, section, start_addr, end_addr,
129                               reg->guest_phys_addr,
130                               range_get_last(reg->guest_phys_addr,
131                                              reg->memory_size));
132     }
133     for (i = 0; i < dev->nvqs; ++i) {
134         struct vhost_virtqueue *vq = dev->vqs + i;
135 
136         if (!vq->used_phys && !vq->used_size) {
137             continue;
138         }
139 
140         vhost_dev_sync_region(dev, section, start_addr, end_addr, vq->used_phys,
141                               range_get_last(vq->used_phys, vq->used_size));
142     }
143     return 0;
144 }
145 
146 static void vhost_log_sync(MemoryListener *listener,
147                           MemoryRegionSection *section)
148 {
149     struct vhost_dev *dev = container_of(listener, struct vhost_dev,
150                                          memory_listener);
151     vhost_sync_dirty_bitmap(dev, section, 0x0, ~0x0ULL);
152 }
153 
154 static void vhost_log_sync_range(struct vhost_dev *dev,
155                                  hwaddr first, hwaddr last)
156 {
157     int i;
158     /* FIXME: this is N^2 in number of sections */
159     for (i = 0; i < dev->n_mem_sections; ++i) {
160         MemoryRegionSection *section = &dev->mem_sections[i];
161         vhost_sync_dirty_bitmap(dev, section, first, last);
162     }
163 }
164 
165 static uint64_t vhost_get_log_size(struct vhost_dev *dev)
166 {
167     uint64_t log_size = 0;
168     int i;
169     for (i = 0; i < dev->mem->nregions; ++i) {
170         struct vhost_memory_region *reg = dev->mem->regions + i;
171         uint64_t last = range_get_last(reg->guest_phys_addr,
172                                        reg->memory_size);
173         log_size = MAX(log_size, last / VHOST_LOG_CHUNK + 1);
174     }
175     for (i = 0; i < dev->nvqs; ++i) {
176         struct vhost_virtqueue *vq = dev->vqs + i;
177 
178         if (!vq->used_phys && !vq->used_size) {
179             continue;
180         }
181 
182         uint64_t last = vq->used_phys + vq->used_size - 1;
183         log_size = MAX(log_size, last / VHOST_LOG_CHUNK + 1);
184     }
185     return log_size;
186 }
187 
188 static struct vhost_log *vhost_log_alloc(uint64_t size, bool share)
189 {
190     Error *err = NULL;
191     struct vhost_log *log;
192     uint64_t logsize = size * sizeof(*(log->log));
193     int fd = -1;
194 
195     log = g_new0(struct vhost_log, 1);
196     if (share) {
197         log->log = qemu_memfd_alloc("vhost-log", logsize,
198                                     F_SEAL_GROW | F_SEAL_SHRINK | F_SEAL_SEAL,
199                                     &fd, &err);
200         if (err) {
201             error_report_err(err);
202             g_free(log);
203             return NULL;
204         }
205         memset(log->log, 0, logsize);
206     } else {
207         log->log = g_malloc0(logsize);
208     }
209 
210     log->size = size;
211     log->refcnt = 1;
212     log->fd = fd;
213 
214     return log;
215 }
216 
217 static struct vhost_log *vhost_log_get(uint64_t size, bool share)
218 {
219     struct vhost_log *log = share ? vhost_log_shm : vhost_log;
220 
221     if (!log || log->size != size) {
222         log = vhost_log_alloc(size, share);
223         if (share) {
224             vhost_log_shm = log;
225         } else {
226             vhost_log = log;
227         }
228     } else {
229         ++log->refcnt;
230     }
231 
232     return log;
233 }
234 
235 static void vhost_log_put(struct vhost_dev *dev, bool sync)
236 {
237     struct vhost_log *log = dev->log;
238 
239     if (!log) {
240         return;
241     }
242 
243     --log->refcnt;
244     if (log->refcnt == 0) {
245         /* Sync only the range covered by the old log */
246         if (dev->log_size && sync) {
247             vhost_log_sync_range(dev, 0, dev->log_size * VHOST_LOG_CHUNK - 1);
248         }
249 
250         if (vhost_log == log) {
251             g_free(log->log);
252             vhost_log = NULL;
253         } else if (vhost_log_shm == log) {
254             qemu_memfd_free(log->log, log->size * sizeof(*(log->log)),
255                             log->fd);
256             vhost_log_shm = NULL;
257         }
258 
259         g_free(log);
260     }
261 
262     dev->log = NULL;
263     dev->log_size = 0;
264 }
265 
266 static bool vhost_dev_log_is_shared(struct vhost_dev *dev)
267 {
268     return dev->vhost_ops->vhost_requires_shm_log &&
269            dev->vhost_ops->vhost_requires_shm_log(dev);
270 }
271 
272 static inline void vhost_dev_log_resize(struct vhost_dev *dev, uint64_t size)
273 {
274     struct vhost_log *log = vhost_log_get(size, vhost_dev_log_is_shared(dev));
275     uint64_t log_base = (uintptr_t)log->log;
276     int r;
277 
278     /* inform backend of log switching, this must be done before
279        releasing the current log, to ensure no logging is lost */
280     r = dev->vhost_ops->vhost_set_log_base(dev, log_base, log);
281     if (r < 0) {
282         VHOST_OPS_DEBUG("vhost_set_log_base failed");
283     }
284 
285     vhost_log_put(dev, true);
286     dev->log = log;
287     dev->log_size = size;
288 }
289 
290 static int vhost_dev_has_iommu(struct vhost_dev *dev)
291 {
292     VirtIODevice *vdev = dev->vdev;
293 
294     /*
295      * For vhost, VIRTIO_F_IOMMU_PLATFORM means the backend support
296      * incremental memory mapping API via IOTLB API. For platform that
297      * does not have IOMMU, there's no need to enable this feature
298      * which may cause unnecessary IOTLB miss/update trnasactions.
299      */
300     return vdev->dma_as != &address_space_memory &&
301            virtio_host_has_feature(vdev, VIRTIO_F_IOMMU_PLATFORM);
302 }
303 
304 static void *vhost_memory_map(struct vhost_dev *dev, hwaddr addr,
305                               hwaddr *plen, bool is_write)
306 {
307     if (!vhost_dev_has_iommu(dev)) {
308         return cpu_physical_memory_map(addr, plen, is_write);
309     } else {
310         return (void *)(uintptr_t)addr;
311     }
312 }
313 
314 static void vhost_memory_unmap(struct vhost_dev *dev, void *buffer,
315                                hwaddr len, int is_write,
316                                hwaddr access_len)
317 {
318     if (!vhost_dev_has_iommu(dev)) {
319         cpu_physical_memory_unmap(buffer, len, is_write, access_len);
320     }
321 }
322 
323 static int vhost_verify_ring_part_mapping(void *ring_hva,
324                                           uint64_t ring_gpa,
325                                           uint64_t ring_size,
326                                           void *reg_hva,
327                                           uint64_t reg_gpa,
328                                           uint64_t reg_size)
329 {
330     uint64_t hva_ring_offset;
331     uint64_t ring_last = range_get_last(ring_gpa, ring_size);
332     uint64_t reg_last = range_get_last(reg_gpa, reg_size);
333 
334     if (ring_last < reg_gpa || ring_gpa > reg_last) {
335         return 0;
336     }
337     /* check that whole ring's is mapped */
338     if (ring_last > reg_last) {
339         return -ENOMEM;
340     }
341     /* check that ring's MemoryRegion wasn't replaced */
342     hva_ring_offset = ring_gpa - reg_gpa;
343     if (ring_hva != reg_hva + hva_ring_offset) {
344         return -EBUSY;
345     }
346 
347     return 0;
348 }
349 
350 static int vhost_verify_ring_mappings(struct vhost_dev *dev,
351                                       void *reg_hva,
352                                       uint64_t reg_gpa,
353                                       uint64_t reg_size)
354 {
355     int i, j;
356     int r = 0;
357     const char *part_name[] = {
358         "descriptor table",
359         "available ring",
360         "used ring"
361     };
362 
363     if (vhost_dev_has_iommu(dev)) {
364         return 0;
365     }
366 
367     for (i = 0; i < dev->nvqs; ++i) {
368         struct vhost_virtqueue *vq = dev->vqs + i;
369 
370         if (vq->desc_phys == 0) {
371             continue;
372         }
373 
374         j = 0;
375         r = vhost_verify_ring_part_mapping(
376                 vq->desc, vq->desc_phys, vq->desc_size,
377                 reg_hva, reg_gpa, reg_size);
378         if (r) {
379             break;
380         }
381 
382         j++;
383         r = vhost_verify_ring_part_mapping(
384                 vq->avail, vq->avail_phys, vq->avail_size,
385                 reg_hva, reg_gpa, reg_size);
386         if (r) {
387             break;
388         }
389 
390         j++;
391         r = vhost_verify_ring_part_mapping(
392                 vq->used, vq->used_phys, vq->used_size,
393                 reg_hva, reg_gpa, reg_size);
394         if (r) {
395             break;
396         }
397     }
398 
399     if (r == -ENOMEM) {
400         error_report("Unable to map %s for ring %d", part_name[j], i);
401     } else if (r == -EBUSY) {
402         error_report("%s relocated for ring %d", part_name[j], i);
403     }
404     return r;
405 }
406 
407 /*
408  * vhost_section: identify sections needed for vhost access
409  *
410  * We only care about RAM sections here (where virtqueue and guest
411  * internals accessed by virtio might live). If we find one we still
412  * allow the backend to potentially filter it out of our list.
413  */
414 static bool vhost_section(struct vhost_dev *dev, MemoryRegionSection *section)
415 {
416     MemoryRegion *mr = section->mr;
417 
418     if (memory_region_is_ram(mr) && !memory_region_is_rom(mr)) {
419         uint8_t dirty_mask = memory_region_get_dirty_log_mask(mr);
420         uint8_t handled_dirty;
421 
422         /*
423          * Kernel based vhost doesn't handle any block which is doing
424          * dirty-tracking other than migration for which it has
425          * specific logging support. However for TCG the kernel never
426          * gets involved anyway so we can also ignore it's
427          * self-modiying code detection flags. However a vhost-user
428          * client could still confuse a TCG guest if it re-writes
429          * executable memory that has already been translated.
430          */
431         handled_dirty = (1 << DIRTY_MEMORY_MIGRATION) |
432             (1 << DIRTY_MEMORY_CODE);
433 
434         if (dirty_mask & ~handled_dirty) {
435             trace_vhost_reject_section(mr->name, 1);
436             return false;
437         }
438 
439         if (dev->vhost_ops->vhost_backend_mem_section_filter &&
440             !dev->vhost_ops->vhost_backend_mem_section_filter(dev, section)) {
441             trace_vhost_reject_section(mr->name, 2);
442             return false;
443         }
444 
445         trace_vhost_section(mr->name);
446         return true;
447     } else {
448         trace_vhost_reject_section(mr->name, 3);
449         return false;
450     }
451 }
452 
453 static void vhost_begin(MemoryListener *listener)
454 {
455     struct vhost_dev *dev = container_of(listener, struct vhost_dev,
456                                          memory_listener);
457     dev->tmp_sections = NULL;
458     dev->n_tmp_sections = 0;
459 }
460 
461 static void vhost_commit(MemoryListener *listener)
462 {
463     struct vhost_dev *dev = container_of(listener, struct vhost_dev,
464                                          memory_listener);
465     MemoryRegionSection *old_sections;
466     int n_old_sections;
467     uint64_t log_size;
468     size_t regions_size;
469     int r;
470     int i;
471     bool changed = false;
472 
473     /* Note we can be called before the device is started, but then
474      * starting the device calls set_mem_table, so we need to have
475      * built the data structures.
476      */
477     old_sections = dev->mem_sections;
478     n_old_sections = dev->n_mem_sections;
479     dev->mem_sections = dev->tmp_sections;
480     dev->n_mem_sections = dev->n_tmp_sections;
481 
482     if (dev->n_mem_sections != n_old_sections) {
483         changed = true;
484     } else {
485         /* Same size, lets check the contents */
486         for (int i = 0; i < n_old_sections; i++) {
487             if (!MemoryRegionSection_eq(&old_sections[i],
488                                         &dev->mem_sections[i])) {
489                 changed = true;
490                 break;
491             }
492         }
493     }
494 
495     trace_vhost_commit(dev->started, changed);
496     if (!changed) {
497         goto out;
498     }
499 
500     /* Rebuild the regions list from the new sections list */
501     regions_size = offsetof(struct vhost_memory, regions) +
502                        dev->n_mem_sections * sizeof dev->mem->regions[0];
503     dev->mem = g_realloc(dev->mem, regions_size);
504     dev->mem->nregions = dev->n_mem_sections;
505     used_memslots = dev->mem->nregions;
506     for (i = 0; i < dev->n_mem_sections; i++) {
507         struct vhost_memory_region *cur_vmr = dev->mem->regions + i;
508         struct MemoryRegionSection *mrs = dev->mem_sections + i;
509 
510         cur_vmr->guest_phys_addr = mrs->offset_within_address_space;
511         cur_vmr->memory_size     = int128_get64(mrs->size);
512         cur_vmr->userspace_addr  =
513             (uintptr_t)memory_region_get_ram_ptr(mrs->mr) +
514             mrs->offset_within_region;
515         cur_vmr->flags_padding   = 0;
516     }
517 
518     if (!dev->started) {
519         goto out;
520     }
521 
522     for (i = 0; i < dev->mem->nregions; i++) {
523         if (vhost_verify_ring_mappings(dev,
524                        (void *)(uintptr_t)dev->mem->regions[i].userspace_addr,
525                        dev->mem->regions[i].guest_phys_addr,
526                        dev->mem->regions[i].memory_size)) {
527             error_report("Verify ring failure on region %d", i);
528             abort();
529         }
530     }
531 
532     if (!dev->log_enabled) {
533         r = dev->vhost_ops->vhost_set_mem_table(dev, dev->mem);
534         if (r < 0) {
535             VHOST_OPS_DEBUG("vhost_set_mem_table failed");
536         }
537         goto out;
538     }
539     log_size = vhost_get_log_size(dev);
540     /* We allocate an extra 4K bytes to log,
541      * to reduce the * number of reallocations. */
542 #define VHOST_LOG_BUFFER (0x1000 / sizeof *dev->log)
543     /* To log more, must increase log size before table update. */
544     if (dev->log_size < log_size) {
545         vhost_dev_log_resize(dev, log_size + VHOST_LOG_BUFFER);
546     }
547     r = dev->vhost_ops->vhost_set_mem_table(dev, dev->mem);
548     if (r < 0) {
549         VHOST_OPS_DEBUG("vhost_set_mem_table failed");
550     }
551     /* To log less, can only decrease log size after table update. */
552     if (dev->log_size > log_size + VHOST_LOG_BUFFER) {
553         vhost_dev_log_resize(dev, log_size);
554     }
555 
556 out:
557     /* Deref the old list of sections, this must happen _after_ the
558      * vhost_set_mem_table to ensure the client isn't still using the
559      * section we're about to unref.
560      */
561     while (n_old_sections--) {
562         memory_region_unref(old_sections[n_old_sections].mr);
563     }
564     g_free(old_sections);
565     return;
566 }
567 
568 /* Adds the section data to the tmp_section structure.
569  * It relies on the listener calling us in memory address order
570  * and for each region (via the _add and _nop methods) to
571  * join neighbours.
572  */
573 static void vhost_region_add_section(struct vhost_dev *dev,
574                                      MemoryRegionSection *section)
575 {
576     bool need_add = true;
577     uint64_t mrs_size = int128_get64(section->size);
578     uint64_t mrs_gpa = section->offset_within_address_space;
579     uintptr_t mrs_host = (uintptr_t)memory_region_get_ram_ptr(section->mr) +
580                          section->offset_within_region;
581     RAMBlock *mrs_rb = section->mr->ram_block;
582 
583     trace_vhost_region_add_section(section->mr->name, mrs_gpa, mrs_size,
584                                    mrs_host);
585 
586     if (dev->vhost_ops->backend_type == VHOST_BACKEND_TYPE_USER) {
587         /* Round the section to it's page size */
588         /* First align the start down to a page boundary */
589         size_t mrs_page = qemu_ram_pagesize(mrs_rb);
590         uint64_t alignage = mrs_host & (mrs_page - 1);
591         if (alignage) {
592             mrs_host -= alignage;
593             mrs_size += alignage;
594             mrs_gpa  -= alignage;
595         }
596         /* Now align the size up to a page boundary */
597         alignage = mrs_size & (mrs_page - 1);
598         if (alignage) {
599             mrs_size += mrs_page - alignage;
600         }
601         trace_vhost_region_add_section_aligned(section->mr->name, mrs_gpa,
602                                                mrs_size, mrs_host);
603     }
604 
605     if (dev->n_tmp_sections) {
606         /* Since we already have at least one section, lets see if
607          * this extends it; since we're scanning in order, we only
608          * have to look at the last one, and the FlatView that calls
609          * us shouldn't have overlaps.
610          */
611         MemoryRegionSection *prev_sec = dev->tmp_sections +
612                                                (dev->n_tmp_sections - 1);
613         uint64_t prev_gpa_start = prev_sec->offset_within_address_space;
614         uint64_t prev_size = int128_get64(prev_sec->size);
615         uint64_t prev_gpa_end   = range_get_last(prev_gpa_start, prev_size);
616         uint64_t prev_host_start =
617                         (uintptr_t)memory_region_get_ram_ptr(prev_sec->mr) +
618                         prev_sec->offset_within_region;
619         uint64_t prev_host_end   = range_get_last(prev_host_start, prev_size);
620 
621         if (mrs_gpa <= (prev_gpa_end + 1)) {
622             /* OK, looks like overlapping/intersecting - it's possible that
623              * the rounding to page sizes has made them overlap, but they should
624              * match up in the same RAMBlock if they do.
625              */
626             if (mrs_gpa < prev_gpa_start) {
627                 error_report("%s:Section '%s' rounded to %"PRIx64
628                              " prior to previous '%s' %"PRIx64,
629                              __func__, section->mr->name, mrs_gpa,
630                              prev_sec->mr->name, prev_gpa_start);
631                 /* A way to cleanly fail here would be better */
632                 return;
633             }
634             /* Offset from the start of the previous GPA to this GPA */
635             size_t offset = mrs_gpa - prev_gpa_start;
636 
637             if (prev_host_start + offset == mrs_host &&
638                 section->mr == prev_sec->mr &&
639                 (!dev->vhost_ops->vhost_backend_can_merge ||
640                  dev->vhost_ops->vhost_backend_can_merge(dev,
641                     mrs_host, mrs_size,
642                     prev_host_start, prev_size))) {
643                 uint64_t max_end = MAX(prev_host_end, mrs_host + mrs_size);
644                 need_add = false;
645                 prev_sec->offset_within_address_space =
646                     MIN(prev_gpa_start, mrs_gpa);
647                 prev_sec->offset_within_region =
648                     MIN(prev_host_start, mrs_host) -
649                     (uintptr_t)memory_region_get_ram_ptr(prev_sec->mr);
650                 prev_sec->size = int128_make64(max_end - MIN(prev_host_start,
651                                                mrs_host));
652                 trace_vhost_region_add_section_merge(section->mr->name,
653                                         int128_get64(prev_sec->size),
654                                         prev_sec->offset_within_address_space,
655                                         prev_sec->offset_within_region);
656             } else {
657                 /* adjoining regions are fine, but overlapping ones with
658                  * different blocks/offsets shouldn't happen
659                  */
660                 if (mrs_gpa != prev_gpa_end + 1) {
661                     error_report("%s: Overlapping but not coherent sections "
662                                  "at %"PRIx64,
663                                  __func__, mrs_gpa);
664                     return;
665                 }
666             }
667         }
668     }
669 
670     if (need_add) {
671         ++dev->n_tmp_sections;
672         dev->tmp_sections = g_renew(MemoryRegionSection, dev->tmp_sections,
673                                     dev->n_tmp_sections);
674         dev->tmp_sections[dev->n_tmp_sections - 1] = *section;
675         /* The flatview isn't stable and we don't use it, making it NULL
676          * means we can memcmp the list.
677          */
678         dev->tmp_sections[dev->n_tmp_sections - 1].fv = NULL;
679         memory_region_ref(section->mr);
680     }
681 }
682 
683 /* Used for both add and nop callbacks */
684 static void vhost_region_addnop(MemoryListener *listener,
685                                 MemoryRegionSection *section)
686 {
687     struct vhost_dev *dev = container_of(listener, struct vhost_dev,
688                                          memory_listener);
689 
690     if (!vhost_section(dev, section)) {
691         return;
692     }
693     vhost_region_add_section(dev, section);
694 }
695 
696 static void vhost_iommu_unmap_notify(IOMMUNotifier *n, IOMMUTLBEntry *iotlb)
697 {
698     struct vhost_iommu *iommu = container_of(n, struct vhost_iommu, n);
699     struct vhost_dev *hdev = iommu->hdev;
700     hwaddr iova = iotlb->iova + iommu->iommu_offset;
701 
702     if (vhost_backend_invalidate_device_iotlb(hdev, iova,
703                                               iotlb->addr_mask + 1)) {
704         error_report("Fail to invalidate device iotlb");
705     }
706 }
707 
708 static void vhost_iommu_region_add(MemoryListener *listener,
709                                    MemoryRegionSection *section)
710 {
711     struct vhost_dev *dev = container_of(listener, struct vhost_dev,
712                                          iommu_listener);
713     struct vhost_iommu *iommu;
714     Int128 end;
715     int iommu_idx, ret;
716     IOMMUMemoryRegion *iommu_mr;
717     Error *err = NULL;
718 
719     if (!memory_region_is_iommu(section->mr)) {
720         return;
721     }
722 
723     iommu_mr = IOMMU_MEMORY_REGION(section->mr);
724 
725     iommu = g_malloc0(sizeof(*iommu));
726     end = int128_add(int128_make64(section->offset_within_region),
727                      section->size);
728     end = int128_sub(end, int128_one());
729     iommu_idx = memory_region_iommu_attrs_to_index(iommu_mr,
730                                                    MEMTXATTRS_UNSPECIFIED);
731     iommu_notifier_init(&iommu->n, vhost_iommu_unmap_notify,
732                         IOMMU_NOTIFIER_UNMAP,
733                         section->offset_within_region,
734                         int128_get64(end),
735                         iommu_idx);
736     iommu->mr = section->mr;
737     iommu->iommu_offset = section->offset_within_address_space -
738                           section->offset_within_region;
739     iommu->hdev = dev;
740     ret = memory_region_register_iommu_notifier(section->mr, &iommu->n, &err);
741     if (ret) {
742         error_report_err(err);
743         exit(1);
744     }
745     QLIST_INSERT_HEAD(&dev->iommu_list, iommu, iommu_next);
746     /* TODO: can replay help performance here? */
747 }
748 
749 static void vhost_iommu_region_del(MemoryListener *listener,
750                                    MemoryRegionSection *section)
751 {
752     struct vhost_dev *dev = container_of(listener, struct vhost_dev,
753                                          iommu_listener);
754     struct vhost_iommu *iommu;
755 
756     if (!memory_region_is_iommu(section->mr)) {
757         return;
758     }
759 
760     QLIST_FOREACH(iommu, &dev->iommu_list, iommu_next) {
761         if (iommu->mr == section->mr &&
762             iommu->n.start == section->offset_within_region) {
763             memory_region_unregister_iommu_notifier(iommu->mr,
764                                                     &iommu->n);
765             QLIST_REMOVE(iommu, iommu_next);
766             g_free(iommu);
767             break;
768         }
769     }
770 }
771 
772 static int vhost_virtqueue_set_addr(struct vhost_dev *dev,
773                                     struct vhost_virtqueue *vq,
774                                     unsigned idx, bool enable_log)
775 {
776     struct vhost_vring_addr addr = {
777         .index = idx,
778         .desc_user_addr = (uint64_t)(unsigned long)vq->desc,
779         .avail_user_addr = (uint64_t)(unsigned long)vq->avail,
780         .used_user_addr = (uint64_t)(unsigned long)vq->used,
781         .log_guest_addr = vq->used_phys,
782         .flags = enable_log ? (1 << VHOST_VRING_F_LOG) : 0,
783     };
784     int r = dev->vhost_ops->vhost_set_vring_addr(dev, &addr);
785     if (r < 0) {
786         VHOST_OPS_DEBUG("vhost_set_vring_addr failed");
787         return -errno;
788     }
789     return 0;
790 }
791 
792 static int vhost_dev_set_features(struct vhost_dev *dev,
793                                   bool enable_log)
794 {
795     uint64_t features = dev->acked_features;
796     int r;
797     if (enable_log) {
798         features |= 0x1ULL << VHOST_F_LOG_ALL;
799     }
800     if (!vhost_dev_has_iommu(dev)) {
801         features &= ~(0x1ULL << VIRTIO_F_IOMMU_PLATFORM);
802     }
803     r = dev->vhost_ops->vhost_set_features(dev, features);
804     if (r < 0) {
805         VHOST_OPS_DEBUG("vhost_set_features failed");
806     }
807     return r < 0 ? -errno : 0;
808 }
809 
810 static int vhost_dev_set_log(struct vhost_dev *dev, bool enable_log)
811 {
812     int r, i, idx;
813     r = vhost_dev_set_features(dev, enable_log);
814     if (r < 0) {
815         goto err_features;
816     }
817     for (i = 0; i < dev->nvqs; ++i) {
818         idx = dev->vhost_ops->vhost_get_vq_index(dev, dev->vq_index + i);
819         r = vhost_virtqueue_set_addr(dev, dev->vqs + i, idx,
820                                      enable_log);
821         if (r < 0) {
822             goto err_vq;
823         }
824     }
825     return 0;
826 err_vq:
827     for (; i >= 0; --i) {
828         idx = dev->vhost_ops->vhost_get_vq_index(dev, dev->vq_index + i);
829         vhost_virtqueue_set_addr(dev, dev->vqs + i, idx,
830                                  dev->log_enabled);
831     }
832     vhost_dev_set_features(dev, dev->log_enabled);
833 err_features:
834     return r;
835 }
836 
837 static int vhost_migration_log(MemoryListener *listener, bool enable)
838 {
839     struct vhost_dev *dev = container_of(listener, struct vhost_dev,
840                                          memory_listener);
841     int r;
842     if (enable == dev->log_enabled) {
843         return 0;
844     }
845     if (!dev->started) {
846         dev->log_enabled = enable;
847         return 0;
848     }
849     if (!enable) {
850         r = vhost_dev_set_log(dev, false);
851         if (r < 0) {
852             return r;
853         }
854         vhost_log_put(dev, false);
855     } else {
856         vhost_dev_log_resize(dev, vhost_get_log_size(dev));
857         r = vhost_dev_set_log(dev, true);
858         if (r < 0) {
859             return r;
860         }
861     }
862     dev->log_enabled = enable;
863     return 0;
864 }
865 
866 static void vhost_log_global_start(MemoryListener *listener)
867 {
868     int r;
869 
870     r = vhost_migration_log(listener, true);
871     if (r < 0) {
872         abort();
873     }
874 }
875 
876 static void vhost_log_global_stop(MemoryListener *listener)
877 {
878     int r;
879 
880     r = vhost_migration_log(listener, false);
881     if (r < 0) {
882         abort();
883     }
884 }
885 
886 static void vhost_log_start(MemoryListener *listener,
887                             MemoryRegionSection *section,
888                             int old, int new)
889 {
890     /* FIXME: implement */
891 }
892 
893 static void vhost_log_stop(MemoryListener *listener,
894                            MemoryRegionSection *section,
895                            int old, int new)
896 {
897     /* FIXME: implement */
898 }
899 
900 /* The vhost driver natively knows how to handle the vrings of non
901  * cross-endian legacy devices and modern devices. Only legacy devices
902  * exposed to a bi-endian guest may require the vhost driver to use a
903  * specific endianness.
904  */
905 static inline bool vhost_needs_vring_endian(VirtIODevice *vdev)
906 {
907     if (virtio_vdev_has_feature(vdev, VIRTIO_F_VERSION_1)) {
908         return false;
909     }
910 #ifdef HOST_WORDS_BIGENDIAN
911     return vdev->device_endian == VIRTIO_DEVICE_ENDIAN_LITTLE;
912 #else
913     return vdev->device_endian == VIRTIO_DEVICE_ENDIAN_BIG;
914 #endif
915 }
916 
917 static int vhost_virtqueue_set_vring_endian_legacy(struct vhost_dev *dev,
918                                                    bool is_big_endian,
919                                                    int vhost_vq_index)
920 {
921     struct vhost_vring_state s = {
922         .index = vhost_vq_index,
923         .num = is_big_endian
924     };
925 
926     if (!dev->vhost_ops->vhost_set_vring_endian(dev, &s)) {
927         return 0;
928     }
929 
930     VHOST_OPS_DEBUG("vhost_set_vring_endian failed");
931     if (errno == ENOTTY) {
932         error_report("vhost does not support cross-endian");
933         return -ENOSYS;
934     }
935 
936     return -errno;
937 }
938 
939 static int vhost_memory_region_lookup(struct vhost_dev *hdev,
940                                       uint64_t gpa, uint64_t *uaddr,
941                                       uint64_t *len)
942 {
943     int i;
944 
945     for (i = 0; i < hdev->mem->nregions; i++) {
946         struct vhost_memory_region *reg = hdev->mem->regions + i;
947 
948         if (gpa >= reg->guest_phys_addr &&
949             reg->guest_phys_addr + reg->memory_size > gpa) {
950             *uaddr = reg->userspace_addr + gpa - reg->guest_phys_addr;
951             *len = reg->guest_phys_addr + reg->memory_size - gpa;
952             return 0;
953         }
954     }
955 
956     return -EFAULT;
957 }
958 
959 int vhost_device_iotlb_miss(struct vhost_dev *dev, uint64_t iova, int write)
960 {
961     IOMMUTLBEntry iotlb;
962     uint64_t uaddr, len;
963     int ret = -EFAULT;
964 
965     RCU_READ_LOCK_GUARD();
966 
967     trace_vhost_iotlb_miss(dev, 1);
968 
969     iotlb = address_space_get_iotlb_entry(dev->vdev->dma_as,
970                                           iova, write,
971                                           MEMTXATTRS_UNSPECIFIED);
972     if (iotlb.target_as != NULL) {
973         ret = vhost_memory_region_lookup(dev, iotlb.translated_addr,
974                                          &uaddr, &len);
975         if (ret) {
976             trace_vhost_iotlb_miss(dev, 3);
977             error_report("Fail to lookup the translated address "
978                          "%"PRIx64, iotlb.translated_addr);
979             goto out;
980         }
981 
982         len = MIN(iotlb.addr_mask + 1, len);
983         iova = iova & ~iotlb.addr_mask;
984 
985         ret = vhost_backend_update_device_iotlb(dev, iova, uaddr,
986                                                 len, iotlb.perm);
987         if (ret) {
988             trace_vhost_iotlb_miss(dev, 4);
989             error_report("Fail to update device iotlb");
990             goto out;
991         }
992     }
993 
994     trace_vhost_iotlb_miss(dev, 2);
995 
996 out:
997     return ret;
998 }
999 
1000 static int vhost_virtqueue_start(struct vhost_dev *dev,
1001                                 struct VirtIODevice *vdev,
1002                                 struct vhost_virtqueue *vq,
1003                                 unsigned idx)
1004 {
1005     BusState *qbus = BUS(qdev_get_parent_bus(DEVICE(vdev)));
1006     VirtioBusState *vbus = VIRTIO_BUS(qbus);
1007     VirtioBusClass *k = VIRTIO_BUS_GET_CLASS(vbus);
1008     hwaddr s, l, a;
1009     int r;
1010     int vhost_vq_index = dev->vhost_ops->vhost_get_vq_index(dev, idx);
1011     struct vhost_vring_file file = {
1012         .index = vhost_vq_index
1013     };
1014     struct vhost_vring_state state = {
1015         .index = vhost_vq_index
1016     };
1017     struct VirtQueue *vvq = virtio_get_queue(vdev, idx);
1018 
1019     a = virtio_queue_get_desc_addr(vdev, idx);
1020     if (a == 0) {
1021         /* Queue might not be ready for start */
1022         return 0;
1023     }
1024 
1025     vq->num = state.num = virtio_queue_get_num(vdev, idx);
1026     r = dev->vhost_ops->vhost_set_vring_num(dev, &state);
1027     if (r) {
1028         VHOST_OPS_DEBUG("vhost_set_vring_num failed");
1029         return -errno;
1030     }
1031 
1032     state.num = virtio_queue_get_last_avail_idx(vdev, idx);
1033     r = dev->vhost_ops->vhost_set_vring_base(dev, &state);
1034     if (r) {
1035         VHOST_OPS_DEBUG("vhost_set_vring_base failed");
1036         return -errno;
1037     }
1038 
1039     if (vhost_needs_vring_endian(vdev)) {
1040         r = vhost_virtqueue_set_vring_endian_legacy(dev,
1041                                                     virtio_is_big_endian(vdev),
1042                                                     vhost_vq_index);
1043         if (r) {
1044             return -errno;
1045         }
1046     }
1047 
1048     vq->desc_size = s = l = virtio_queue_get_desc_size(vdev, idx);
1049     vq->desc_phys = a;
1050     vq->desc = vhost_memory_map(dev, a, &l, false);
1051     if (!vq->desc || l != s) {
1052         r = -ENOMEM;
1053         goto fail_alloc_desc;
1054     }
1055     vq->avail_size = s = l = virtio_queue_get_avail_size(vdev, idx);
1056     vq->avail_phys = a = virtio_queue_get_avail_addr(vdev, idx);
1057     vq->avail = vhost_memory_map(dev, a, &l, false);
1058     if (!vq->avail || l != s) {
1059         r = -ENOMEM;
1060         goto fail_alloc_avail;
1061     }
1062     vq->used_size = s = l = virtio_queue_get_used_size(vdev, idx);
1063     vq->used_phys = a = virtio_queue_get_used_addr(vdev, idx);
1064     vq->used = vhost_memory_map(dev, a, &l, true);
1065     if (!vq->used || l != s) {
1066         r = -ENOMEM;
1067         goto fail_alloc_used;
1068     }
1069 
1070     r = vhost_virtqueue_set_addr(dev, vq, vhost_vq_index, dev->log_enabled);
1071     if (r < 0) {
1072         r = -errno;
1073         goto fail_alloc;
1074     }
1075 
1076     file.fd = event_notifier_get_fd(virtio_queue_get_host_notifier(vvq));
1077     r = dev->vhost_ops->vhost_set_vring_kick(dev, &file);
1078     if (r) {
1079         VHOST_OPS_DEBUG("vhost_set_vring_kick failed");
1080         r = -errno;
1081         goto fail_kick;
1082     }
1083 
1084     /* Clear and discard previous events if any. */
1085     event_notifier_test_and_clear(&vq->masked_notifier);
1086 
1087     /* Init vring in unmasked state, unless guest_notifier_mask
1088      * will do it later.
1089      */
1090     if (!vdev->use_guest_notifier_mask) {
1091         /* TODO: check and handle errors. */
1092         vhost_virtqueue_mask(dev, vdev, idx, false);
1093     }
1094 
1095     if (k->query_guest_notifiers &&
1096         k->query_guest_notifiers(qbus->parent) &&
1097         virtio_queue_vector(vdev, idx) == VIRTIO_NO_VECTOR) {
1098         file.fd = -1;
1099         r = dev->vhost_ops->vhost_set_vring_call(dev, &file);
1100         if (r) {
1101             goto fail_vector;
1102         }
1103     }
1104 
1105     return 0;
1106 
1107 fail_vector:
1108 fail_kick:
1109 fail_alloc:
1110     vhost_memory_unmap(dev, vq->used, virtio_queue_get_used_size(vdev, idx),
1111                        0, 0);
1112 fail_alloc_used:
1113     vhost_memory_unmap(dev, vq->avail, virtio_queue_get_avail_size(vdev, idx),
1114                        0, 0);
1115 fail_alloc_avail:
1116     vhost_memory_unmap(dev, vq->desc, virtio_queue_get_desc_size(vdev, idx),
1117                        0, 0);
1118 fail_alloc_desc:
1119     return r;
1120 }
1121 
1122 static void vhost_virtqueue_stop(struct vhost_dev *dev,
1123                                     struct VirtIODevice *vdev,
1124                                     struct vhost_virtqueue *vq,
1125                                     unsigned idx)
1126 {
1127     int vhost_vq_index = dev->vhost_ops->vhost_get_vq_index(dev, idx);
1128     struct vhost_vring_state state = {
1129         .index = vhost_vq_index,
1130     };
1131     int r;
1132 
1133     if (virtio_queue_get_desc_addr(vdev, idx) == 0) {
1134         /* Don't stop the virtqueue which might have not been started */
1135         return;
1136     }
1137 
1138     r = dev->vhost_ops->vhost_get_vring_base(dev, &state);
1139     if (r < 0) {
1140         VHOST_OPS_DEBUG("vhost VQ %u ring restore failed: %d", idx, r);
1141         /* Connection to the backend is broken, so let's sync internal
1142          * last avail idx to the device used idx.
1143          */
1144         virtio_queue_restore_last_avail_idx(vdev, idx);
1145     } else {
1146         virtio_queue_set_last_avail_idx(vdev, idx, state.num);
1147     }
1148     virtio_queue_invalidate_signalled_used(vdev, idx);
1149     virtio_queue_update_used_idx(vdev, idx);
1150 
1151     /* In the cross-endian case, we need to reset the vring endianness to
1152      * native as legacy devices expect so by default.
1153      */
1154     if (vhost_needs_vring_endian(vdev)) {
1155         vhost_virtqueue_set_vring_endian_legacy(dev,
1156                                                 !virtio_is_big_endian(vdev),
1157                                                 vhost_vq_index);
1158     }
1159 
1160     vhost_memory_unmap(dev, vq->used, virtio_queue_get_used_size(vdev, idx),
1161                        1, virtio_queue_get_used_size(vdev, idx));
1162     vhost_memory_unmap(dev, vq->avail, virtio_queue_get_avail_size(vdev, idx),
1163                        0, virtio_queue_get_avail_size(vdev, idx));
1164     vhost_memory_unmap(dev, vq->desc, virtio_queue_get_desc_size(vdev, idx),
1165                        0, virtio_queue_get_desc_size(vdev, idx));
1166 }
1167 
1168 static void vhost_eventfd_add(MemoryListener *listener,
1169                               MemoryRegionSection *section,
1170                               bool match_data, uint64_t data, EventNotifier *e)
1171 {
1172 }
1173 
1174 static void vhost_eventfd_del(MemoryListener *listener,
1175                               MemoryRegionSection *section,
1176                               bool match_data, uint64_t data, EventNotifier *e)
1177 {
1178 }
1179 
1180 static int vhost_virtqueue_set_busyloop_timeout(struct vhost_dev *dev,
1181                                                 int n, uint32_t timeout)
1182 {
1183     int vhost_vq_index = dev->vhost_ops->vhost_get_vq_index(dev, n);
1184     struct vhost_vring_state state = {
1185         .index = vhost_vq_index,
1186         .num = timeout,
1187     };
1188     int r;
1189 
1190     if (!dev->vhost_ops->vhost_set_vring_busyloop_timeout) {
1191         return -EINVAL;
1192     }
1193 
1194     r = dev->vhost_ops->vhost_set_vring_busyloop_timeout(dev, &state);
1195     if (r) {
1196         VHOST_OPS_DEBUG("vhost_set_vring_busyloop_timeout failed");
1197         return r;
1198     }
1199 
1200     return 0;
1201 }
1202 
1203 static int vhost_virtqueue_init(struct vhost_dev *dev,
1204                                 struct vhost_virtqueue *vq, int n)
1205 {
1206     int vhost_vq_index = dev->vhost_ops->vhost_get_vq_index(dev, n);
1207     struct vhost_vring_file file = {
1208         .index = vhost_vq_index,
1209     };
1210     int r = event_notifier_init(&vq->masked_notifier, 0);
1211     if (r < 0) {
1212         return r;
1213     }
1214 
1215     file.fd = event_notifier_get_fd(&vq->masked_notifier);
1216     r = dev->vhost_ops->vhost_set_vring_call(dev, &file);
1217     if (r) {
1218         VHOST_OPS_DEBUG("vhost_set_vring_call failed");
1219         r = -errno;
1220         goto fail_call;
1221     }
1222 
1223     vq->dev = dev;
1224 
1225     return 0;
1226 fail_call:
1227     event_notifier_cleanup(&vq->masked_notifier);
1228     return r;
1229 }
1230 
1231 static void vhost_virtqueue_cleanup(struct vhost_virtqueue *vq)
1232 {
1233     event_notifier_cleanup(&vq->masked_notifier);
1234 }
1235 
1236 int vhost_dev_init(struct vhost_dev *hdev, void *opaque,
1237                    VhostBackendType backend_type, uint32_t busyloop_timeout)
1238 {
1239     uint64_t features;
1240     int i, r, n_initialized_vqs = 0;
1241     Error *local_err = NULL;
1242 
1243     hdev->vdev = NULL;
1244     hdev->migration_blocker = NULL;
1245 
1246     r = vhost_set_backend_type(hdev, backend_type);
1247     assert(r >= 0);
1248 
1249     r = hdev->vhost_ops->vhost_backend_init(hdev, opaque);
1250     if (r < 0) {
1251         goto fail;
1252     }
1253 
1254     r = hdev->vhost_ops->vhost_set_owner(hdev);
1255     if (r < 0) {
1256         VHOST_OPS_DEBUG("vhost_set_owner failed");
1257         goto fail;
1258     }
1259 
1260     r = hdev->vhost_ops->vhost_get_features(hdev, &features);
1261     if (r < 0) {
1262         VHOST_OPS_DEBUG("vhost_get_features failed");
1263         goto fail;
1264     }
1265 
1266     for (i = 0; i < hdev->nvqs; ++i, ++n_initialized_vqs) {
1267         r = vhost_virtqueue_init(hdev, hdev->vqs + i, hdev->vq_index + i);
1268         if (r < 0) {
1269             goto fail;
1270         }
1271     }
1272 
1273     if (busyloop_timeout) {
1274         for (i = 0; i < hdev->nvqs; ++i) {
1275             r = vhost_virtqueue_set_busyloop_timeout(hdev, hdev->vq_index + i,
1276                                                      busyloop_timeout);
1277             if (r < 0) {
1278                 goto fail_busyloop;
1279             }
1280         }
1281     }
1282 
1283     hdev->features = features;
1284 
1285     hdev->memory_listener = (MemoryListener) {
1286         .begin = vhost_begin,
1287         .commit = vhost_commit,
1288         .region_add = vhost_region_addnop,
1289         .region_nop = vhost_region_addnop,
1290         .log_start = vhost_log_start,
1291         .log_stop = vhost_log_stop,
1292         .log_sync = vhost_log_sync,
1293         .log_global_start = vhost_log_global_start,
1294         .log_global_stop = vhost_log_global_stop,
1295         .eventfd_add = vhost_eventfd_add,
1296         .eventfd_del = vhost_eventfd_del,
1297         .priority = 10
1298     };
1299 
1300     hdev->iommu_listener = (MemoryListener) {
1301         .region_add = vhost_iommu_region_add,
1302         .region_del = vhost_iommu_region_del,
1303     };
1304 
1305     if (hdev->migration_blocker == NULL) {
1306         if (!(hdev->features & (0x1ULL << VHOST_F_LOG_ALL))) {
1307             error_setg(&hdev->migration_blocker,
1308                        "Migration disabled: vhost lacks VHOST_F_LOG_ALL feature.");
1309         } else if (vhost_dev_log_is_shared(hdev) && !qemu_memfd_alloc_check()) {
1310             error_setg(&hdev->migration_blocker,
1311                        "Migration disabled: failed to allocate shared memory");
1312         }
1313     }
1314 
1315     if (hdev->migration_blocker != NULL) {
1316         r = migrate_add_blocker(hdev->migration_blocker, &local_err);
1317         if (local_err) {
1318             error_report_err(local_err);
1319             error_free(hdev->migration_blocker);
1320             goto fail_busyloop;
1321         }
1322     }
1323 
1324     hdev->mem = g_malloc0(offsetof(struct vhost_memory, regions));
1325     hdev->n_mem_sections = 0;
1326     hdev->mem_sections = NULL;
1327     hdev->log = NULL;
1328     hdev->log_size = 0;
1329     hdev->log_enabled = false;
1330     hdev->started = false;
1331     memory_listener_register(&hdev->memory_listener, &address_space_memory);
1332     QLIST_INSERT_HEAD(&vhost_devices, hdev, entry);
1333 
1334     if (used_memslots > hdev->vhost_ops->vhost_backend_memslots_limit(hdev)) {
1335         error_report("vhost backend memory slots limit is less"
1336                 " than current number of present memory slots");
1337         r = -1;
1338         if (busyloop_timeout) {
1339             goto fail_busyloop;
1340         } else {
1341             goto fail;
1342         }
1343     }
1344 
1345     return 0;
1346 
1347 fail_busyloop:
1348     while (--i >= 0) {
1349         vhost_virtqueue_set_busyloop_timeout(hdev, hdev->vq_index + i, 0);
1350     }
1351 fail:
1352     hdev->nvqs = n_initialized_vqs;
1353     vhost_dev_cleanup(hdev);
1354     return r;
1355 }
1356 
1357 void vhost_dev_cleanup(struct vhost_dev *hdev)
1358 {
1359     int i;
1360 
1361     for (i = 0; i < hdev->nvqs; ++i) {
1362         vhost_virtqueue_cleanup(hdev->vqs + i);
1363     }
1364     if (hdev->mem) {
1365         /* those are only safe after successful init */
1366         memory_listener_unregister(&hdev->memory_listener);
1367         QLIST_REMOVE(hdev, entry);
1368     }
1369     if (hdev->migration_blocker) {
1370         migrate_del_blocker(hdev->migration_blocker);
1371         error_free(hdev->migration_blocker);
1372     }
1373     g_free(hdev->mem);
1374     g_free(hdev->mem_sections);
1375     if (hdev->vhost_ops) {
1376         hdev->vhost_ops->vhost_backend_cleanup(hdev);
1377     }
1378     assert(!hdev->log);
1379 
1380     memset(hdev, 0, sizeof(struct vhost_dev));
1381 }
1382 
1383 /* Stop processing guest IO notifications in qemu.
1384  * Start processing them in vhost in kernel.
1385  */
1386 int vhost_dev_enable_notifiers(struct vhost_dev *hdev, VirtIODevice *vdev)
1387 {
1388     BusState *qbus = BUS(qdev_get_parent_bus(DEVICE(vdev)));
1389     int i, r, e;
1390 
1391     /* We will pass the notifiers to the kernel, make sure that QEMU
1392      * doesn't interfere.
1393      */
1394     r = virtio_device_grab_ioeventfd(vdev);
1395     if (r < 0) {
1396         error_report("binding does not support host notifiers");
1397         goto fail;
1398     }
1399 
1400     for (i = 0; i < hdev->nvqs; ++i) {
1401         r = virtio_bus_set_host_notifier(VIRTIO_BUS(qbus), hdev->vq_index + i,
1402                                          true);
1403         if (r < 0) {
1404             error_report("vhost VQ %d notifier binding failed: %d", i, -r);
1405             goto fail_vq;
1406         }
1407     }
1408 
1409     return 0;
1410 fail_vq:
1411     while (--i >= 0) {
1412         e = virtio_bus_set_host_notifier(VIRTIO_BUS(qbus), hdev->vq_index + i,
1413                                          false);
1414         if (e < 0) {
1415             error_report("vhost VQ %d notifier cleanup error: %d", i, -r);
1416         }
1417         assert (e >= 0);
1418         virtio_bus_cleanup_host_notifier(VIRTIO_BUS(qbus), hdev->vq_index + i);
1419     }
1420     virtio_device_release_ioeventfd(vdev);
1421 fail:
1422     return r;
1423 }
1424 
1425 /* Stop processing guest IO notifications in vhost.
1426  * Start processing them in qemu.
1427  * This might actually run the qemu handlers right away,
1428  * so virtio in qemu must be completely setup when this is called.
1429  */
1430 void vhost_dev_disable_notifiers(struct vhost_dev *hdev, VirtIODevice *vdev)
1431 {
1432     BusState *qbus = BUS(qdev_get_parent_bus(DEVICE(vdev)));
1433     int i, r;
1434 
1435     for (i = 0; i < hdev->nvqs; ++i) {
1436         r = virtio_bus_set_host_notifier(VIRTIO_BUS(qbus), hdev->vq_index + i,
1437                                          false);
1438         if (r < 0) {
1439             error_report("vhost VQ %d notifier cleanup failed: %d", i, -r);
1440         }
1441         assert (r >= 0);
1442         virtio_bus_cleanup_host_notifier(VIRTIO_BUS(qbus), hdev->vq_index + i);
1443     }
1444     virtio_device_release_ioeventfd(vdev);
1445 }
1446 
1447 /* Test and clear event pending status.
1448  * Should be called after unmask to avoid losing events.
1449  */
1450 bool vhost_virtqueue_pending(struct vhost_dev *hdev, int n)
1451 {
1452     struct vhost_virtqueue *vq = hdev->vqs + n - hdev->vq_index;
1453     assert(n >= hdev->vq_index && n < hdev->vq_index + hdev->nvqs);
1454     return event_notifier_test_and_clear(&vq->masked_notifier);
1455 }
1456 
1457 /* Mask/unmask events from this vq. */
1458 void vhost_virtqueue_mask(struct vhost_dev *hdev, VirtIODevice *vdev, int n,
1459                          bool mask)
1460 {
1461     struct VirtQueue *vvq = virtio_get_queue(vdev, n);
1462     int r, index = n - hdev->vq_index;
1463     struct vhost_vring_file file;
1464 
1465     /* should only be called after backend is connected */
1466     assert(hdev->vhost_ops);
1467 
1468     if (mask) {
1469         assert(vdev->use_guest_notifier_mask);
1470         file.fd = event_notifier_get_fd(&hdev->vqs[index].masked_notifier);
1471     } else {
1472         file.fd = event_notifier_get_fd(virtio_queue_get_guest_notifier(vvq));
1473     }
1474 
1475     file.index = hdev->vhost_ops->vhost_get_vq_index(hdev, n);
1476     r = hdev->vhost_ops->vhost_set_vring_call(hdev, &file);
1477     if (r < 0) {
1478         VHOST_OPS_DEBUG("vhost_set_vring_call failed");
1479     }
1480 }
1481 
1482 uint64_t vhost_get_features(struct vhost_dev *hdev, const int *feature_bits,
1483                             uint64_t features)
1484 {
1485     const int *bit = feature_bits;
1486     while (*bit != VHOST_INVALID_FEATURE_BIT) {
1487         uint64_t bit_mask = (1ULL << *bit);
1488         if (!(hdev->features & bit_mask)) {
1489             features &= ~bit_mask;
1490         }
1491         bit++;
1492     }
1493     return features;
1494 }
1495 
1496 void vhost_ack_features(struct vhost_dev *hdev, const int *feature_bits,
1497                         uint64_t features)
1498 {
1499     const int *bit = feature_bits;
1500     while (*bit != VHOST_INVALID_FEATURE_BIT) {
1501         uint64_t bit_mask = (1ULL << *bit);
1502         if (features & bit_mask) {
1503             hdev->acked_features |= bit_mask;
1504         }
1505         bit++;
1506     }
1507 }
1508 
1509 int vhost_dev_get_config(struct vhost_dev *hdev, uint8_t *config,
1510                          uint32_t config_len)
1511 {
1512     assert(hdev->vhost_ops);
1513 
1514     if (hdev->vhost_ops->vhost_get_config) {
1515         return hdev->vhost_ops->vhost_get_config(hdev, config, config_len);
1516     }
1517 
1518     return -1;
1519 }
1520 
1521 int vhost_dev_set_config(struct vhost_dev *hdev, const uint8_t *data,
1522                          uint32_t offset, uint32_t size, uint32_t flags)
1523 {
1524     assert(hdev->vhost_ops);
1525 
1526     if (hdev->vhost_ops->vhost_set_config) {
1527         return hdev->vhost_ops->vhost_set_config(hdev, data, offset,
1528                                                  size, flags);
1529     }
1530 
1531     return -1;
1532 }
1533 
1534 void vhost_dev_set_config_notifier(struct vhost_dev *hdev,
1535                                    const VhostDevConfigOps *ops)
1536 {
1537     hdev->config_ops = ops;
1538 }
1539 
1540 void vhost_dev_free_inflight(struct vhost_inflight *inflight)
1541 {
1542     if (inflight && inflight->addr) {
1543         qemu_memfd_free(inflight->addr, inflight->size, inflight->fd);
1544         inflight->addr = NULL;
1545         inflight->fd = -1;
1546     }
1547 }
1548 
1549 static int vhost_dev_resize_inflight(struct vhost_inflight *inflight,
1550                                      uint64_t new_size)
1551 {
1552     Error *err = NULL;
1553     int fd = -1;
1554     void *addr = qemu_memfd_alloc("vhost-inflight", new_size,
1555                                   F_SEAL_GROW | F_SEAL_SHRINK | F_SEAL_SEAL,
1556                                   &fd, &err);
1557 
1558     if (err) {
1559         error_report_err(err);
1560         return -1;
1561     }
1562 
1563     vhost_dev_free_inflight(inflight);
1564     inflight->offset = 0;
1565     inflight->addr = addr;
1566     inflight->fd = fd;
1567     inflight->size = new_size;
1568 
1569     return 0;
1570 }
1571 
1572 void vhost_dev_save_inflight(struct vhost_inflight *inflight, QEMUFile *f)
1573 {
1574     if (inflight->addr) {
1575         qemu_put_be64(f, inflight->size);
1576         qemu_put_be16(f, inflight->queue_size);
1577         qemu_put_buffer(f, inflight->addr, inflight->size);
1578     } else {
1579         qemu_put_be64(f, 0);
1580     }
1581 }
1582 
1583 int vhost_dev_load_inflight(struct vhost_inflight *inflight, QEMUFile *f)
1584 {
1585     uint64_t size;
1586 
1587     size = qemu_get_be64(f);
1588     if (!size) {
1589         return 0;
1590     }
1591 
1592     if (inflight->size != size) {
1593         if (vhost_dev_resize_inflight(inflight, size)) {
1594             return -1;
1595         }
1596     }
1597     inflight->queue_size = qemu_get_be16(f);
1598 
1599     qemu_get_buffer(f, inflight->addr, size);
1600 
1601     return 0;
1602 }
1603 
1604 int vhost_dev_set_inflight(struct vhost_dev *dev,
1605                            struct vhost_inflight *inflight)
1606 {
1607     int r;
1608 
1609     if (dev->vhost_ops->vhost_set_inflight_fd && inflight->addr) {
1610         r = dev->vhost_ops->vhost_set_inflight_fd(dev, inflight);
1611         if (r) {
1612             VHOST_OPS_DEBUG("vhost_set_inflight_fd failed");
1613             return -errno;
1614         }
1615     }
1616 
1617     return 0;
1618 }
1619 
1620 int vhost_dev_get_inflight(struct vhost_dev *dev, uint16_t queue_size,
1621                            struct vhost_inflight *inflight)
1622 {
1623     int r;
1624 
1625     if (dev->vhost_ops->vhost_get_inflight_fd) {
1626         r = dev->vhost_ops->vhost_get_inflight_fd(dev, queue_size, inflight);
1627         if (r) {
1628             VHOST_OPS_DEBUG("vhost_get_inflight_fd failed");
1629             return -errno;
1630         }
1631     }
1632 
1633     return 0;
1634 }
1635 
1636 /* Host notifiers must be enabled at this point. */
1637 int vhost_dev_start(struct vhost_dev *hdev, VirtIODevice *vdev)
1638 {
1639     int i, r;
1640 
1641     /* should only be called after backend is connected */
1642     assert(hdev->vhost_ops);
1643 
1644     hdev->started = true;
1645     hdev->vdev = vdev;
1646 
1647     r = vhost_dev_set_features(hdev, hdev->log_enabled);
1648     if (r < 0) {
1649         goto fail_features;
1650     }
1651 
1652     if (vhost_dev_has_iommu(hdev)) {
1653         memory_listener_register(&hdev->iommu_listener, vdev->dma_as);
1654     }
1655 
1656     r = hdev->vhost_ops->vhost_set_mem_table(hdev, hdev->mem);
1657     if (r < 0) {
1658         VHOST_OPS_DEBUG("vhost_set_mem_table failed");
1659         r = -errno;
1660         goto fail_mem;
1661     }
1662     for (i = 0; i < hdev->nvqs; ++i) {
1663         r = vhost_virtqueue_start(hdev,
1664                                   vdev,
1665                                   hdev->vqs + i,
1666                                   hdev->vq_index + i);
1667         if (r < 0) {
1668             goto fail_vq;
1669         }
1670     }
1671 
1672     if (hdev->log_enabled) {
1673         uint64_t log_base;
1674 
1675         hdev->log_size = vhost_get_log_size(hdev);
1676         hdev->log = vhost_log_get(hdev->log_size,
1677                                   vhost_dev_log_is_shared(hdev));
1678         log_base = (uintptr_t)hdev->log->log;
1679         r = hdev->vhost_ops->vhost_set_log_base(hdev,
1680                                                 hdev->log_size ? log_base : 0,
1681                                                 hdev->log);
1682         if (r < 0) {
1683             VHOST_OPS_DEBUG("vhost_set_log_base failed");
1684             r = -errno;
1685             goto fail_log;
1686         }
1687     }
1688 
1689     if (vhost_dev_has_iommu(hdev)) {
1690         hdev->vhost_ops->vhost_set_iotlb_callback(hdev, true);
1691 
1692         /* Update used ring information for IOTLB to work correctly,
1693          * vhost-kernel code requires for this.*/
1694         for (i = 0; i < hdev->nvqs; ++i) {
1695             struct vhost_virtqueue *vq = hdev->vqs + i;
1696             vhost_device_iotlb_miss(hdev, vq->used_phys, true);
1697         }
1698     }
1699     return 0;
1700 fail_log:
1701     vhost_log_put(hdev, false);
1702 fail_vq:
1703     while (--i >= 0) {
1704         vhost_virtqueue_stop(hdev,
1705                              vdev,
1706                              hdev->vqs + i,
1707                              hdev->vq_index + i);
1708     }
1709 
1710 fail_mem:
1711 fail_features:
1712 
1713     hdev->started = false;
1714     return r;
1715 }
1716 
1717 /* Host notifiers must be enabled at this point. */
1718 void vhost_dev_stop(struct vhost_dev *hdev, VirtIODevice *vdev)
1719 {
1720     int i;
1721 
1722     /* should only be called after backend is connected */
1723     assert(hdev->vhost_ops);
1724 
1725     for (i = 0; i < hdev->nvqs; ++i) {
1726         vhost_virtqueue_stop(hdev,
1727                              vdev,
1728                              hdev->vqs + i,
1729                              hdev->vq_index + i);
1730     }
1731 
1732     if (vhost_dev_has_iommu(hdev)) {
1733         hdev->vhost_ops->vhost_set_iotlb_callback(hdev, false);
1734         memory_listener_unregister(&hdev->iommu_listener);
1735     }
1736     vhost_log_put(hdev, true);
1737     hdev->started = false;
1738     hdev->vdev = NULL;
1739 }
1740 
1741 int vhost_net_set_backend(struct vhost_dev *hdev,
1742                           struct vhost_vring_file *file)
1743 {
1744     if (hdev->vhost_ops->vhost_net_set_backend) {
1745         return hdev->vhost_ops->vhost_net_set_backend(hdev, file);
1746     }
1747 
1748     return -1;
1749 }
1750