hw/vhost.c

   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
  13 #include <sys/ioctl.h>
  14 #include "vhost.h"
  15 #include "hw/hw.h"
  16 #include "range.h"
  17 #include <linux/vhost.h>
  18
  19 static void vhost_dev_sync_region(struct vhost_dev *dev,
  20                                   uint64_t mfirst, uint64_t mlast,
  21                                   uint64_t rfirst, uint64_t rlast)
  22 {
  23     uint64_t start = MAX(mfirst, rfirst);
  24     uint64_t end = MIN(mlast, rlast);
  25     vhost_log_chunk_t *from = dev->log + start / VHOST_LOG_CHUNK;
  26     vhost_log_chunk_t *to = dev->log + end / VHOST_LOG_CHUNK + 1;
  27     uint64_t addr = (start / VHOST_LOG_CHUNK) * VHOST_LOG_CHUNK;
  28
  29     assert(end / VHOST_LOG_CHUNK < dev->log_size);
  30     assert(start / VHOST_LOG_CHUNK < dev->log_size);
  31     if (end < start) {
  32         return;
  33     }
  34     for (;from < to; ++from) {
  35         vhost_log_chunk_t log;
  36         int bit;
  37         /* We first check with non-atomic: much cheaper,
  38          * and we expect non-dirty to be the common case. */
  39         if (!*from) {
  40             addr += VHOST_LOG_CHUNK;
  41             continue;
  42         }
  43         /* Data must be read atomically. We don't really
  44          * need the barrier semantics of __sync
  45          * builtins, but it's easier to use them than
  46          * roll our own. */
  47         log = __sync_fetch_and_and(from, 0);
  48         while ((bit = sizeof(log) > sizeof(int) ?
  49                 ffsll(log) : ffs(log))) {
  50             ram_addr_t ram_addr;
  51             bit -= 1;
  52             ram_addr = cpu_get_physical_page_desc(addr + bit * VHOST_LOG_PAGE);
  53             cpu_physical_memory_set_dirty(ram_addr);
  54             log &= ~(0x1ull << bit);
  55         }
  56         addr += VHOST_LOG_CHUNK;
  57     }
  58 }
  59
  60 static int vhost_client_sync_dirty_bitmap(CPUPhysMemoryClient *client,
  61                                           target_phys_addr_t start_addr,
  62                                           target_phys_addr_t end_addr)
  63 {
  64     struct vhost_dev *dev = container_of(client, struct vhost_dev, client);
  65     int i;
  66     if (!dev->log_enabled || !dev->started) {
  67         return 0;
  68     }
  69     for (i = 0; i < dev->mem->nregions; ++i) {
  70         struct vhost_memory_region *reg = dev->mem->regions + i;
  71         vhost_dev_sync_region(dev, start_addr, end_addr,
  72                               reg->guest_phys_addr,
  73                               range_get_last(reg->guest_phys_addr,
  74                                              reg->memory_size));
  75     }
  76     for (i = 0; i < dev->nvqs; ++i) {
  77         struct vhost_virtqueue *vq = dev->vqs + i;
  78         vhost_dev_sync_region(dev, start_addr, end_addr, vq->used_phys,
  79                               range_get_last(vq->used_phys, vq->used_size));
  80     }
  81     return 0;
  82 }
  83
  84 /* Assign/unassign. Keep an unsorted array of non-overlapping
  85  * memory regions in dev->mem. */
  86 static void vhost_dev_unassign_memory(struct vhost_dev *dev,
  87                                       uint64_t start_addr,
  88                                       uint64_t size)
  89 {
  90     int from, to, n = dev->mem->nregions;
  91     /* Track overlapping/split regions for sanity checking. */
  92     int overlap_start = 0, overlap_end = 0, overlap_middle = 0, split = 0;
  93
  94     for (from = 0, to = 0; from < n; ++from, ++to) {
  95         struct vhost_memory_region *reg = dev->mem->regions + to;
  96         uint64_t reglast;
  97         uint64_t memlast;
  98         uint64_t change;
  99
 100         /* clone old region */
 101         if (to != from) {
 102             memcpy(reg, dev->mem->regions + from, sizeof *reg);
 103         }
 104
 105         /* No overlap is simple */
 106         if (!ranges_overlap(reg->guest_phys_addr, reg->memory_size,
 107                             start_addr, size)) {
 108             continue;
 109         }
 110
 111         /* Split only happens if supplied region
 112          * is in the middle of an existing one. Thus it can not
 113          * overlap with any other existing region. */
 114         assert(!split);
 115
 116         reglast = range_get_last(reg->guest_phys_addr, reg->memory_size);
 117         memlast = range_get_last(start_addr, size);
 118
 119         /* Remove whole region */
 120         if (start_addr <= reg->guest_phys_addr && memlast >= reglast) {
 121             --dev->mem->nregions;
 122             --to;
 123             ++overlap_middle;
 124             continue;
 125         }
 126
 127         /* Shrink region */
 128         if (memlast >= reglast) {
 129             reg->memory_size = start_addr - reg->guest_phys_addr;
 130             assert(reg->memory_size);
 131             assert(!overlap_end);
 132             ++overlap_end;
 133             continue;
 134         }
 135
 136         /* Shift region */
 137         if (start_addr <= reg->guest_phys_addr) {
 138             change = memlast + 1 - reg->guest_phys_addr;
 139             reg->memory_size -= change;
 140             reg->guest_phys_addr += change;
 141             reg->userspace_addr += change;
 142             assert(reg->memory_size);
 143             assert(!overlap_start);
 144             ++overlap_start;
 145             continue;
 146         }
 147
 148         /* This only happens if supplied region
 149          * is in the middle of an existing one. Thus it can not
 150          * overlap with any other existing region. */
 151         assert(!overlap_start);
 152         assert(!overlap_end);
 153         assert(!overlap_middle);
 154         /* Split region: shrink first part, shift second part. */
 155         memcpy(dev->mem->regions + n, reg, sizeof *reg);
 156         reg->memory_size = start_addr - reg->guest_phys_addr;
 157         assert(reg->memory_size);
 158         change = memlast + 1 - reg->guest_phys_addr;
 159         reg = dev->mem->regions + n;
 160         reg->memory_size -= change;
 161         assert(reg->memory_size);
 162         reg->guest_phys_addr += change;
 163         reg->userspace_addr += change;
 164         /* Never add more than 1 region */
 165         assert(dev->mem->nregions == n);
 166         ++dev->mem->nregions;
 167         ++split;
 168     }
 169 }
 170
 171 /* Called after unassign, so no regions overlap the given range. */
 172 static void vhost_dev_assign_memory(struct vhost_dev *dev,
 173                                     uint64_t start_addr,
 174                                     uint64_t size,
 175                                     uint64_t uaddr)
 176 {
 177     int from, to;
 178     struct vhost_memory_region *merged = NULL;
 179     for (from = 0, to = 0; from < dev->mem->nregions; ++from, ++to) {
 180         struct vhost_memory_region *reg = dev->mem->regions + to;
 181         uint64_t prlast, urlast;
 182         uint64_t pmlast, umlast;
 183         uint64_t s, e, u;
 184
 185         /* clone old region */
 186         if (to != from) {
 187             memcpy(reg, dev->mem->regions + from, sizeof *reg);
 188         }
 189         prlast = range_get_last(reg->guest_phys_addr, reg->memory_size);
 190         pmlast = range_get_last(start_addr, size);
 191         urlast = range_get_last(reg->userspace_addr, reg->memory_size);
 192         umlast = range_get_last(uaddr, size);
 193
 194         /* check for overlapping regions: should never happen. */
 195         assert(prlast < start_addr || pmlast < reg->guest_phys_addr);
 196         /* Not an adjacent or overlapping region - do not merge. */
 197         if ((prlast + 1 != start_addr || urlast + 1 != uaddr) &&
 198             (pmlast + 1 != reg->guest_phys_addr ||
 199              umlast + 1 != reg->userspace_addr)) {
 200             continue;
 201         }
 202
 203         if (merged) {
 204             --to;
 205             assert(to >= 0);
 206         } else {
 207             merged = reg;
 208         }
 209         u = MIN(uaddr, reg->userspace_addr);
 210         s = MIN(start_addr, reg->guest_phys_addr);
 211         e = MAX(pmlast, prlast);
 212         uaddr = merged->userspace_addr = u;
 213         start_addr = merged->guest_phys_addr = s;
 214         size = merged->memory_size = e - s + 1;
 215         assert(merged->memory_size);
 216     }
 217
 218     if (!merged) {
 219         struct vhost_memory_region *reg = dev->mem->regions + to;
 220         memset(reg, 0, sizeof *reg);
 221         reg->memory_size = size;
 222         assert(reg->memory_size);
 223         reg->guest_phys_addr = start_addr;
 224         reg->userspace_addr = uaddr;
 225         ++to;
 226     }
 227     assert(to <= dev->mem->nregions + 1);
 228     dev->mem->nregions = to;
 229 }
 230
 231 static uint64_t vhost_get_log_size(struct vhost_dev *dev)
 232 {
 233     uint64_t log_size = 0;
 234     int i;
 235     for (i = 0; i < dev->mem->nregions; ++i) {
 236         struct vhost_memory_region *reg = dev->mem->regions + i;
 237         uint64_t last = range_get_last(reg->guest_phys_addr,
 238                                        reg->memory_size);
 239         log_size = MAX(log_size, last / VHOST_LOG_CHUNK + 1);
 240     }
 241     for (i = 0; i < dev->nvqs; ++i) {
 242         struct vhost_virtqueue *vq = dev->vqs + i;
 243         uint64_t last = vq->used_phys + vq->used_size - 1;
 244         log_size = MAX(log_size, last / VHOST_LOG_CHUNK + 1);
 245     }
 246     return log_size;
 247 }
 248
 249 static inline void vhost_dev_log_resize(struct vhost_dev* dev, uint64_t size)
 250 {
 251     vhost_log_chunk_t *log;
 252     uint64_t log_base;
 253     int r;
 254     if (size) {
 255         log = g_malloc0(size * sizeof *log);
 256     } else {
 257         log = NULL;
 258     }
 259     log_base = (uint64_t)(unsigned long)log;
 260     r = ioctl(dev->control, VHOST_SET_LOG_BASE, &log_base);
 261     assert(r >= 0);
 262     vhost_client_sync_dirty_bitmap(&dev->client, 0,
 263                                    (target_phys_addr_t)~0x0ull);
 264     if (dev->log) {
 265         g_free(dev->log);
 266     }
 267     dev->log = log;
 268     dev->log_size = size;
 269 }
 270
 271 static int vhost_verify_ring_mappings(struct vhost_dev *dev,
 272                                       uint64_t start_addr,
 273                                       uint64_t size)
 274 {
 275     int i;
 276     for (i = 0; i < dev->nvqs; ++i) {
 277         struct vhost_virtqueue *vq = dev->vqs + i;
 278         target_phys_addr_t l;
 279         void *p;
 280
 281         if (!ranges_overlap(start_addr, size, vq->ring_phys, vq->ring_size)) {
 282             continue;
 283         }
 284         l = vq->ring_size;
 285         p = cpu_physical_memory_map(vq->ring_phys, &l, 1);
 286         if (!p || l != vq->ring_size) {
 287             fprintf(stderr, "Unable to map ring buffer for ring %d\n", i);
 288             return -ENOMEM;
 289         }
 290         if (p != vq->ring) {
 291             fprintf(stderr, "Ring buffer relocated for ring %d\n", i);
 292             return -EBUSY;
 293         }
 294         cpu_physical_memory_unmap(p, l, 0, 0);
 295     }
 296     return 0;
 297 }
 298
 299 static struct vhost_memory_region *vhost_dev_find_reg(struct vhost_dev *dev,
 300                                                       uint64_t start_addr,
 301                                                       uint64_t size)
 302 {
 303     int i, n = dev->mem->nregions;
 304     for (i = 0; i < n; ++i) {
 305         struct vhost_memory_region *reg = dev->mem->regions + i;
 306         if (ranges_overlap(reg->guest_phys_addr, reg->memory_size,
 307                            start_addr, size)) {
 308             return reg;
 309         }
 310     }
 311     return NULL;
 312 }
 313
 314 static bool vhost_dev_cmp_memory(struct vhost_dev *dev,
 315                                  uint64_t start_addr,
 316                                  uint64_t size,
 317                                  uint64_t uaddr)
 318 {
 319     struct vhost_memory_region *reg = vhost_dev_find_reg(dev, start_addr, size);
 320     uint64_t reglast;
 321     uint64_t memlast;
 322
 323     if (!reg) {
 324         return true;
 325     }
 326
 327     reglast = range_get_last(reg->guest_phys_addr, reg->memory_size);
 328     memlast = range_get_last(start_addr, size);
 329
 330     /* Need to extend region? */
 331     if (start_addr < reg->guest_phys_addr || memlast > reglast) {
 332         return true;
 333     }
 334     /* userspace_addr changed? */
 335     return uaddr != reg->userspace_addr + start_addr - reg->guest_phys_addr;
 336 }
 337
 338 static void vhost_client_set_memory(CPUPhysMemoryClient *client,
 339                                     target_phys_addr_t start_addr,
 340                                     ram_addr_t size,
 341                                     ram_addr_t phys_offset,
 342                                     bool log_dirty)
 343 {
 344     struct vhost_dev *dev = container_of(client, struct vhost_dev, client);
 345     ram_addr_t flags = phys_offset & ~TARGET_PAGE_MASK;
 346     int s = offsetof(struct vhost_memory, regions) +
 347         (dev->mem->nregions + 1) * sizeof dev->mem->regions[0];
 348     uint64_t log_size;
 349     int r;
 350
 351     dev->mem = g_realloc(dev->mem, s);
 352
 353     if (log_dirty) {
 354         flags = IO_MEM_UNASSIGNED;
 355     }
 356
 357     assert(size);
 358
 359     /* Optimize no-change case. At least cirrus_vga does this a lot at this time. */
 360     if (flags == IO_MEM_RAM) {
 361         if (!vhost_dev_cmp_memory(dev, start_addr, size,
 362                                   (uintptr_t)qemu_get_ram_ptr(phys_offset))) {
 363             /* Region exists with same address. Nothing to do. */
 364             return;
 365         }
 366     } else {
 367         if (!vhost_dev_find_reg(dev, start_addr, size)) {
 368             /* Removing region that we don't access. Nothing to do. */
 369             return;
 370         }
 371     }
 372
 373     vhost_dev_unassign_memory(dev, start_addr, size);
 374     if (flags == IO_MEM_RAM) {
 375         /* Add given mapping, merging adjacent regions if any */
 376         vhost_dev_assign_memory(dev, start_addr, size,
 377                                 (uintptr_t)qemu_get_ram_ptr(phys_offset));
 378     } else {
 379         /* Remove old mapping for this memory, if any. */
 380         vhost_dev_unassign_memory(dev, start_addr, size);
 381     }
 382
 383     if (!dev->started) {
 384         return;
 385     }
 386
 387     if (dev->started) {
 388         r = vhost_verify_ring_mappings(dev, start_addr, size);
 389         assert(r >= 0);
 390     }
 391
 392     if (!dev->log_enabled) {
 393         r = ioctl(dev->control, VHOST_SET_MEM_TABLE, dev->mem);
 394         assert(r >= 0);
 395         return;
 396     }
 397     log_size = vhost_get_log_size(dev);
 398     /* We allocate an extra 4K bytes to log,
 399      * to reduce the * number of reallocations. */
 400 #define VHOST_LOG_BUFFER (0x1000 / sizeof *dev->log)
 401     /* To log more, must increase log size before table update. */
 402     if (dev->log_size < log_size) {
 403         vhost_dev_log_resize(dev, log_size + VHOST_LOG_BUFFER);
 404     }
 405     r = ioctl(dev->control, VHOST_SET_MEM_TABLE, dev->mem);
 406     assert(r >= 0);
 407     /* To log less, can only decrease log size after table update. */
 408     if (dev->log_size > log_size + VHOST_LOG_BUFFER) {
 409         vhost_dev_log_resize(dev, log_size);
 410     }
 411 }
 412
 413 static int vhost_virtqueue_set_addr(struct vhost_dev *dev,
 414                                     struct vhost_virtqueue *vq,
 415                                     unsigned idx, bool enable_log)
 416 {
 417     struct vhost_vring_addr addr = {
 418         .index = idx,
 419         .desc_user_addr = (uint64_t)(unsigned long)vq->desc,
 420         .avail_user_addr = (uint64_t)(unsigned long)vq->avail,
 421         .used_user_addr = (uint64_t)(unsigned long)vq->used,
 422         .log_guest_addr = vq->used_phys,
 423         .flags = enable_log ? (1 << VHOST_VRING_F_LOG) : 0,
 424     };
 425     int r = ioctl(dev->control, VHOST_SET_VRING_ADDR, &addr);
 426     if (r < 0) {
 427         return -errno;
 428     }
 429     return 0;
 430 }
 431
 432 static int vhost_dev_set_features(struct vhost_dev *dev, bool enable_log)
 433 {
 434     uint64_t features = dev->acked_features;
 435     int r;
 436     if (enable_log) {
 437         features |= 0x1 << VHOST_F_LOG_ALL;
 438     }
 439     r = ioctl(dev->control, VHOST_SET_FEATURES, &features);
 440     return r < 0 ? -errno : 0;
 441 }
 442
 443 static int vhost_dev_set_log(struct vhost_dev *dev, bool enable_log)
 444 {
 445     int r, t, i;
 446     r = vhost_dev_set_features(dev, enable_log);
 447     if (r < 0) {
 448         goto err_features;
 449     }
 450     for (i = 0; i < dev->nvqs; ++i) {
 451         r = vhost_virtqueue_set_addr(dev, dev->vqs + i, i,
 452                                      enable_log);
 453         if (r < 0) {
 454             goto err_vq;
 455         }
 456     }
 457     return 0;
 458 err_vq:
 459     for (; i >= 0; --i) {
 460         t = vhost_virtqueue_set_addr(dev, dev->vqs + i, i,
 461                                      dev->log_enabled);
 462         assert(t >= 0);
 463     }
 464     t = vhost_dev_set_features(dev, dev->log_enabled);
 465     assert(t >= 0);
 466 err_features:
 467     return r;
 468 }
 469
 470 static int vhost_client_migration_log(CPUPhysMemoryClient *client,
 471                                       int enable)
 472 {
 473     struct vhost_dev *dev = container_of(client, struct vhost_dev, client);
 474     int r;
 475     if (!!enable == dev->log_enabled) {
 476         return 0;
 477     }
 478     if (!dev->started) {
 479         dev->log_enabled = enable;
 480         return 0;
 481     }
 482     if (!enable) {
 483         r = vhost_dev_set_log(dev, false);
 484         if (r < 0) {
 485             return r;
 486         }
 487         if (dev->log) {
 488             g_free(dev->log);
 489         }
 490         dev->log = NULL;
 491         dev->log_size = 0;
 492     } else {
 493         vhost_dev_log_resize(dev, vhost_get_log_size(dev));
 494         r = vhost_dev_set_log(dev, true);
 495         if (r < 0) {
 496             return r;
 497         }
 498     }
 499     dev->log_enabled = enable;
 500     return 0;
 501 }
 502
 503 static int vhost_virtqueue_init(struct vhost_dev *dev,
 504                                 struct VirtIODevice *vdev,
 505                                 struct vhost_virtqueue *vq,
 506                                 unsigned idx)
 507 {
 508     target_phys_addr_t s, l, a;
 509     int r;
 510     struct vhost_vring_file file = {
 511         .index = idx,
 512     };
 513     struct vhost_vring_state state = {
 514         .index = idx,
 515     };
 516     struct VirtQueue *vvq = virtio_get_queue(vdev, idx);
 517
 518     if (!vdev->binding->set_host_notifier) {
 519         fprintf(stderr, "binding does not support host notifiers\n");
 520         return -ENOSYS;
 521     }
 522
 523     vq->num = state.num = virtio_queue_get_num(vdev, idx);
 524     r = ioctl(dev->control, VHOST_SET_VRING_NUM, &state);
 525     if (r) {
 526         return -errno;
 527     }
 528
 529     state.num = virtio_queue_get_last_avail_idx(vdev, idx);
 530     r = ioctl(dev->control, VHOST_SET_VRING_BASE, &state);
 531     if (r) {
 532         return -errno;
 533     }
 534
 535     s = l = virtio_queue_get_desc_size(vdev, idx);
 536     a = virtio_queue_get_desc_addr(vdev, idx);
 537     vq->desc = cpu_physical_memory_map(a, &l, 0);
 538     if (!vq->desc || l != s) {
 539         r = -ENOMEM;
 540         goto fail_alloc_desc;
 541     }
 542     s = l = virtio_queue_get_avail_size(vdev, idx);
 543     a = virtio_queue_get_avail_addr(vdev, idx);
 544     vq->avail = cpu_physical_memory_map(a, &l, 0);
 545     if (!vq->avail || l != s) {
 546         r = -ENOMEM;
 547         goto fail_alloc_avail;
 548     }
 549     vq->used_size = s = l = virtio_queue_get_used_size(vdev, idx);
 550     vq->used_phys = a = virtio_queue_get_used_addr(vdev, idx);
 551     vq->used = cpu_physical_memory_map(a, &l, 1);
 552     if (!vq->used || l != s) {
 553         r = -ENOMEM;
 554         goto fail_alloc_used;
 555     }
 556
 557     vq->ring_size = s = l = virtio_queue_get_ring_size(vdev, idx);
 558     vq->ring_phys = a = virtio_queue_get_ring_addr(vdev, idx);
 559     vq->ring = cpu_physical_memory_map(a, &l, 1);
 560     if (!vq->ring || l != s) {
 561         r = -ENOMEM;
 562         goto fail_alloc_ring;
 563     }
 564
 565     r = vhost_virtqueue_set_addr(dev, vq, idx, dev->log_enabled);
 566     if (r < 0) {
 567         r = -errno;
 568         goto fail_alloc;
 569     }
 570     r = vdev->binding->set_host_notifier(vdev->binding_opaque, idx, true);
 571     if (r < 0) {
 572         fprintf(stderr, "Error binding host notifier: %d\n", -r);
 573         goto fail_host_notifier;
 574     }
 575
 576     file.fd = event_notifier_get_fd(virtio_queue_get_host_notifier(vvq));
 577     r = ioctl(dev->control, VHOST_SET_VRING_KICK, &file);
 578     if (r) {
 579         r = -errno;
 580         goto fail_kick;
 581     }
 582
 583     file.fd = event_notifier_get_fd(virtio_queue_get_guest_notifier(vvq));
 584     r = ioctl(dev->control, VHOST_SET_VRING_CALL, &file);
 585     if (r) {
 586         r = -errno;
 587         goto fail_call;
 588     }
 589
 590     return 0;
 591
 592 fail_call:
 593 fail_kick:
 594     vdev->binding->set_host_notifier(vdev->binding_opaque, idx, false);
 595 fail_host_notifier:
 596 fail_alloc:
 597     cpu_physical_memory_unmap(vq->ring, virtio_queue_get_ring_size(vdev, idx),
 598                               0, 0);
 599 fail_alloc_ring:
 600     cpu_physical_memory_unmap(vq->used, virtio_queue_get_used_size(vdev, idx),
 601                               0, 0);
 602 fail_alloc_used:
 603     cpu_physical_memory_unmap(vq->avail, virtio_queue_get_avail_size(vdev, idx),
 604                               0, 0);
 605 fail_alloc_avail:
 606     cpu_physical_memory_unmap(vq->desc, virtio_queue_get_desc_size(vdev, idx),
 607                               0, 0);
 608 fail_alloc_desc:
 609     return r;
 610 }
 611
 612 static void vhost_virtqueue_cleanup(struct vhost_dev *dev,
 613                                     struct VirtIODevice *vdev,
 614                                     struct vhost_virtqueue *vq,
 615                                     unsigned idx)
 616 {
 617     struct vhost_vring_state state = {
 618         .index = idx,
 619     };
 620     int r;
 621     r = vdev->binding->set_host_notifier(vdev->binding_opaque, idx, false);
 622     if (r < 0) {
 623         fprintf(stderr, "vhost VQ %d host cleanup failed: %d\n", idx, r);
 624         fflush(stderr);
 625     }
 626     assert (r >= 0);
 627     r = ioctl(dev->control, VHOST_GET_VRING_BASE, &state);
 628     if (r < 0) {
 629         fprintf(stderr, "vhost VQ %d ring restore failed: %d\n", idx, r);
 630         fflush(stderr);
 631     }
 632     virtio_queue_set_last_avail_idx(vdev, idx, state.num);
 633     assert (r >= 0);
 634     cpu_physical_memory_unmap(vq->ring, virtio_queue_get_ring_size(vdev, idx),
 635                               0, virtio_queue_get_ring_size(vdev, idx));
 636     cpu_physical_memory_unmap(vq->used, virtio_queue_get_used_size(vdev, idx),
 637                               1, virtio_queue_get_used_size(vdev, idx));
 638     cpu_physical_memory_unmap(vq->avail, virtio_queue_get_avail_size(vdev, idx),
 639                               0, virtio_queue_get_avail_size(vdev, idx));
 640     cpu_physical_memory_unmap(vq->desc, virtio_queue_get_desc_size(vdev, idx),
 641                               0, virtio_queue_get_desc_size(vdev, idx));
 642 }
 643
 644 int vhost_dev_init(struct vhost_dev *hdev, int devfd, bool force)
 645 {
 646     uint64_t features;
 647     int r;
 648     if (devfd >= 0) {
 649         hdev->control = devfd;
 650     } else {
 651         hdev->control = open("/dev/vhost-net", O_RDWR);
 652         if (hdev->control < 0) {
 653             return -errno;
 654         }
 655     }
 656     r = ioctl(hdev->control, VHOST_SET_OWNER, NULL);
 657     if (r < 0) {
 658         goto fail;
 659     }
 660
 661     r = ioctl(hdev->control, VHOST_GET_FEATURES, &features);
 662     if (r < 0) {
 663         goto fail;
 664     }
 665     hdev->features = features;
 666
 667     hdev->client.set_memory = vhost_client_set_memory;
 668     hdev->client.sync_dirty_bitmap = vhost_client_sync_dirty_bitmap;
 669     hdev->client.migration_log = vhost_client_migration_log;
 670     hdev->client.log_start = NULL;
 671     hdev->client.log_stop = NULL;
 672     hdev->mem = g_malloc0(offsetof(struct vhost_memory, regions));
 673     hdev->log = NULL;
 674     hdev->log_size = 0;
 675     hdev->log_enabled = false;
 676     hdev->started = false;
 677     cpu_register_phys_memory_client(&hdev->client);
 678     hdev->force = force;
 679     return 0;
 680 fail:
 681     r = -errno;
 682     close(hdev->control);
 683     return r;
 684 }
 685
 686 void vhost_dev_cleanup(struct vhost_dev *hdev)
 687 {
 688     cpu_unregister_phys_memory_client(&hdev->client);
 689     g_free(hdev->mem);
 690     close(hdev->control);
 691 }
 692
 693 bool vhost_dev_query(struct vhost_dev *hdev, VirtIODevice *vdev)
 694 {
 695     return !vdev->binding->query_guest_notifiers ||
 696         vdev->binding->query_guest_notifiers(vdev->binding_opaque) ||
 697         hdev->force;
 698 }
 699
 700 int vhost_dev_start(struct vhost_dev *hdev, VirtIODevice *vdev)
 701 {
 702     int i, r;
 703     if (!vdev->binding->set_guest_notifiers) {
 704         fprintf(stderr, "binding does not support guest notifiers\n");
 705         r = -ENOSYS;
 706         goto fail;
 707     }
 708
 709     r = vdev->binding->set_guest_notifiers(vdev->binding_opaque, true);
 710     if (r < 0) {
 711         fprintf(stderr, "Error binding guest notifier: %d\n", -r);
 712         goto fail_notifiers;
 713     }
 714
 715     r = vhost_dev_set_features(hdev, hdev->log_enabled);
 716     if (r < 0) {
 717         goto fail_features;
 718     }
 719     r = ioctl(hdev->control, VHOST_SET_MEM_TABLE, hdev->mem);
 720     if (r < 0) {
 721         r = -errno;
 722         goto fail_mem;
 723     }
 724     for (i = 0; i < hdev->nvqs; ++i) {
 725         r = vhost_virtqueue_init(hdev,
 726                                  vdev,
 727                                  hdev->vqs + i,
 728                                  i);
 729         if (r < 0) {
 730             goto fail_vq;
 731         }
 732     }
 733
 734     if (hdev->log_enabled) {
 735         hdev->log_size = vhost_get_log_size(hdev);
 736         hdev->log = hdev->log_size ?
 737             g_malloc0(hdev->log_size * sizeof *hdev->log) : NULL;
 738         r = ioctl(hdev->control, VHOST_SET_LOG_BASE,
 739                   (uint64_t)(unsigned long)hdev->log);
 740         if (r < 0) {
 741             r = -errno;
 742             goto fail_log;
 743         }
 744     }
 745
 746     hdev->started = true;
 747
 748     return 0;
 749 fail_log:
 750 fail_vq:
 751     while (--i >= 0) {
 752         vhost_virtqueue_cleanup(hdev,
 753                                 vdev,
 754                                 hdev->vqs + i,
 755                                 i);
 756     }
 757 fail_mem:
 758 fail_features:
 759     vdev->binding->set_guest_notifiers(vdev->binding_opaque, false);
 760 fail_notifiers:
 761 fail:
 762     return r;
 763 }
 764
 765 void vhost_dev_stop(struct vhost_dev *hdev, VirtIODevice *vdev)
 766 {
 767     int i, r;
 768
 769     for (i = 0; i < hdev->nvqs; ++i) {
 770         vhost_virtqueue_cleanup(hdev,
 771                                 vdev,
 772                                 hdev->vqs + i,
 773                                 i);
 774     }
 775     vhost_client_sync_dirty_bitmap(&hdev->client, 0,
 776                                    (target_phys_addr_t)~0x0ull);
 777     r = vdev->binding->set_guest_notifiers(vdev->binding_opaque, false);
 778     if (r < 0) {
 779         fprintf(stderr, "vhost guest notifier cleanup failed: %d\n", r);
 780         fflush(stderr);
 781     }
 782     assert (r >= 0);
 783
 784     hdev->started = false;
 785     g_free(hdev->log);
 786     hdev->log = NULL;
 787     hdev->log_size = 0;
 788 }