]>
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; | |
d5970055 MT |
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 = qemu_mallocz(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 | qemu_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 | ||
4e789564 MT |
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 | ||
d5970055 MT |
338 | static void vhost_client_set_memory(CPUPhysMemoryClient *client, |
339 | target_phys_addr_t start_addr, | |
340 | ram_addr_t size, | |
0fd542fb MT |
341 | ram_addr_t phys_offset, |
342 | bool log_dirty) | |
d5970055 MT |
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; | |
4e789564 | 350 | |
d5970055 MT |
351 | dev->mem = qemu_realloc(dev->mem, s); |
352 | ||
f5a4e64f MT |
353 | if (log_dirty) { |
354 | flags = IO_MEM_UNASSIGNED; | |
355 | } | |
356 | ||
d5970055 MT |
357 | assert(size); |
358 | ||
4e789564 MT |
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 | ||
d5970055 MT |
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, | |
2b3af999 SW |
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, | |
d5970055 MT |
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 | qemu_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 | ||
d5970055 MT |
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 | } | |
d5970055 MT |
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) { | |
c8852121 | 579 | r = -errno; |
d5970055 MT |
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) { | |
c8852121 | 586 | r = -errno; |
d5970055 MT |
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: | |
d5970055 MT |
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; | |
d5970055 MT |
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 | ||
5430a28f | 644 | int vhost_dev_init(struct vhost_dev *hdev, int devfd, bool force) |
d5970055 MT |
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; | |
e5896b12 AP |
670 | hdev->client.log_start = NULL; |
671 | hdev->client.log_stop = NULL; | |
d5970055 MT |
672 | hdev->mem = qemu_mallocz(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); | |
5430a28f | 678 | hdev->force = force; |
d5970055 MT |
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 | qemu_free(hdev->mem); | |
690 | close(hdev->control); | |
691 | } | |
692 | ||
5430a28f MT |
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 | ||
d5970055 MT |
700 | int vhost_dev_start(struct vhost_dev *hdev, VirtIODevice *vdev) |
701 | { | |
702 | int i, r; | |
54dd9321 MT |
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 | } | |
d5970055 MT |
714 | |
715 | r = vhost_dev_set_features(hdev, hdev->log_enabled); | |
716 | if (r < 0) { | |
54dd9321 | 717 | goto fail_features; |
d5970055 MT |
718 | } |
719 | r = ioctl(hdev->control, VHOST_SET_MEM_TABLE, hdev->mem); | |
720 | if (r < 0) { | |
721 | r = -errno; | |
54dd9321 | 722 | goto fail_mem; |
d5970055 | 723 | } |
d154e0ba MT |
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 | ||
d5970055 MT |
734 | if (hdev->log_enabled) { |
735 | hdev->log_size = vhost_get_log_size(hdev); | |
736 | hdev->log = hdev->log_size ? | |
737 | qemu_mallocz(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; | |
54dd9321 | 742 | goto fail_log; |
d5970055 MT |
743 | } |
744 | } | |
d154e0ba | 745 | |
d5970055 MT |
746 | hdev->started = true; |
747 | ||
748 | return 0; | |
54dd9321 | 749 | fail_log: |
d5970055 MT |
750 | fail_vq: |
751 | while (--i >= 0) { | |
752 | vhost_virtqueue_cleanup(hdev, | |
753 | vdev, | |
754 | hdev->vqs + i, | |
755 | i); | |
756 | } | |
54dd9321 MT |
757 | fail_mem: |
758 | fail_features: | |
759 | vdev->binding->set_guest_notifiers(vdev->binding_opaque, false); | |
760 | fail_notifiers: | |
d5970055 MT |
761 | fail: |
762 | return r; | |
763 | } | |
764 | ||
765 | void vhost_dev_stop(struct vhost_dev *hdev, VirtIODevice *vdev) | |
766 | { | |
54dd9321 MT |
767 | int i, r; |
768 | ||
d5970055 MT |
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); | |
54dd9321 MT |
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 | ||
d5970055 MT |
784 | hdev->started = false; |
785 | qemu_free(hdev->log); | |
c1be973a | 786 | hdev->log = NULL; |
d5970055 MT |
787 | hdev->log_size = 0; |
788 | } |