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