xref: /qemu/hw/misc/ivshmem.c (revision 20daa90a)
1 /*
2  * Inter-VM Shared Memory PCI device.
3  *
4  * Author:
5  *      Cam Macdonell <cam@cs.ualberta.ca>
6  *
7  * Based On: cirrus_vga.c
8  *          Copyright (c) 2004 Fabrice Bellard
9  *          Copyright (c) 2004 Makoto Suzuki (suzu)
10  *
11  *      and rtl8139.c
12  *          Copyright (c) 2006 Igor Kovalenko
13  *
14  * This code is licensed under the GNU GPL v2.
15  *
16  * Contributions after 2012-01-13 are licensed under the terms of the
17  * GNU GPL, version 2 or (at your option) any later version.
18  */
19 #include "qemu/osdep.h"
20 #include "qapi/error.h"
21 #include "qemu/cutils.h"
22 #include "hw/hw.h"
23 #include "hw/i386/pc.h"
24 #include "hw/pci/pci.h"
25 #include "hw/pci/msi.h"
26 #include "hw/pci/msix.h"
27 #include "sysemu/kvm.h"
28 #include "migration/migration.h"
29 #include "qemu/error-report.h"
30 #include "qemu/event_notifier.h"
31 #include "qom/object_interfaces.h"
32 #include "sysemu/char.h"
33 #include "sysemu/hostmem.h"
34 #include "sysemu/qtest.h"
35 #include "qapi/visitor.h"
36 
37 #include "hw/misc/ivshmem.h"
38 
39 #define PCI_VENDOR_ID_IVSHMEM   PCI_VENDOR_ID_REDHAT_QUMRANET
40 #define PCI_DEVICE_ID_IVSHMEM   0x1110
41 
42 #define IVSHMEM_MAX_PEERS UINT16_MAX
43 #define IVSHMEM_IOEVENTFD   0
44 #define IVSHMEM_MSI     1
45 
46 #define IVSHMEM_REG_BAR_SIZE 0x100
47 
48 #define IVSHMEM_DEBUG 0
49 #define IVSHMEM_DPRINTF(fmt, ...)                       \
50     do {                                                \
51         if (IVSHMEM_DEBUG) {                            \
52             printf("IVSHMEM: " fmt, ## __VA_ARGS__);    \
53         }                                               \
54     } while (0)
55 
56 #define TYPE_IVSHMEM_COMMON "ivshmem-common"
57 #define IVSHMEM_COMMON(obj) \
58     OBJECT_CHECK(IVShmemState, (obj), TYPE_IVSHMEM_COMMON)
59 
60 #define TYPE_IVSHMEM_PLAIN "ivshmem-plain"
61 #define IVSHMEM_PLAIN(obj) \
62     OBJECT_CHECK(IVShmemState, (obj), TYPE_IVSHMEM_PLAIN)
63 
64 #define TYPE_IVSHMEM_DOORBELL "ivshmem-doorbell"
65 #define IVSHMEM_DOORBELL(obj) \
66     OBJECT_CHECK(IVShmemState, (obj), TYPE_IVSHMEM_DOORBELL)
67 
68 #define TYPE_IVSHMEM "ivshmem"
69 #define IVSHMEM(obj) \
70     OBJECT_CHECK(IVShmemState, (obj), TYPE_IVSHMEM)
71 
72 typedef struct Peer {
73     int nb_eventfds;
74     EventNotifier *eventfds;
75 } Peer;
76 
77 typedef struct MSIVector {
78     PCIDevice *pdev;
79     int virq;
80 } MSIVector;
81 
82 typedef struct IVShmemState {
83     /*< private >*/
84     PCIDevice parent_obj;
85     /*< public >*/
86 
87     uint32_t features;
88 
89     /* exactly one of these two may be set */
90     HostMemoryBackend *hostmem; /* with interrupts */
91     CharBackend server_chr; /* without interrupts */
92 
93     /* registers */
94     uint32_t intrmask;
95     uint32_t intrstatus;
96     int vm_id;
97 
98     /* BARs */
99     MemoryRegion ivshmem_mmio;  /* BAR 0 (registers) */
100     MemoryRegion *ivshmem_bar2; /* BAR 2 (shared memory) */
101     MemoryRegion server_bar2;   /* used with server_chr */
102 
103     /* interrupt support */
104     Peer *peers;
105     int nb_peers;               /* space in @peers[] */
106     uint32_t vectors;
107     MSIVector *msi_vectors;
108     uint64_t msg_buf;           /* buffer for receiving server messages */
109     int msg_buffered_bytes;     /* #bytes in @msg_buf */
110 
111     /* migration stuff */
112     OnOffAuto master;
113     Error *migration_blocker;
114 
115     /* legacy cruft */
116     char *role;
117     char *shmobj;
118     char *sizearg;
119     size_t legacy_size;
120     uint32_t not_legacy_32bit;
121 } IVShmemState;
122 
123 /* registers for the Inter-VM shared memory device */
124 enum ivshmem_registers {
125     INTRMASK = 0,
126     INTRSTATUS = 4,
127     IVPOSITION = 8,
128     DOORBELL = 12,
129 };
130 
131 static inline uint32_t ivshmem_has_feature(IVShmemState *ivs,
132                                                     unsigned int feature) {
133     return (ivs->features & (1 << feature));
134 }
135 
136 static inline bool ivshmem_is_master(IVShmemState *s)
137 {
138     assert(s->master != ON_OFF_AUTO_AUTO);
139     return s->master == ON_OFF_AUTO_ON;
140 }
141 
142 static void ivshmem_update_irq(IVShmemState *s)
143 {
144     PCIDevice *d = PCI_DEVICE(s);
145     uint32_t isr = s->intrstatus & s->intrmask;
146 
147     /*
148      * Do nothing unless the device actually uses INTx.  Here's how
149      * the device variants signal interrupts, what they put in PCI
150      * config space:
151      * Device variant    Interrupt  Interrupt Pin  MSI-X cap.
152      * ivshmem-plain         none            0         no
153      * ivshmem-doorbell     MSI-X            1        yes(1)
154      * ivshmem,msi=off       INTx            1         no
155      * ivshmem,msi=on       MSI-X            1(2)     yes(1)
156      * (1) if guest enabled MSI-X
157      * (2) the device lies
158      * Leads to the condition for doing nothing:
159      */
160     if (ivshmem_has_feature(s, IVSHMEM_MSI)
161         || !d->config[PCI_INTERRUPT_PIN]) {
162         return;
163     }
164 
165     /* don't print ISR resets */
166     if (isr) {
167         IVSHMEM_DPRINTF("Set IRQ to %d (%04x %04x)\n",
168                         isr ? 1 : 0, s->intrstatus, s->intrmask);
169     }
170 
171     pci_set_irq(d, isr != 0);
172 }
173 
174 static void ivshmem_IntrMask_write(IVShmemState *s, uint32_t val)
175 {
176     IVSHMEM_DPRINTF("IntrMask write(w) val = 0x%04x\n", val);
177 
178     s->intrmask = val;
179     ivshmem_update_irq(s);
180 }
181 
182 static uint32_t ivshmem_IntrMask_read(IVShmemState *s)
183 {
184     uint32_t ret = s->intrmask;
185 
186     IVSHMEM_DPRINTF("intrmask read(w) val = 0x%04x\n", ret);
187     return ret;
188 }
189 
190 static void ivshmem_IntrStatus_write(IVShmemState *s, uint32_t val)
191 {
192     IVSHMEM_DPRINTF("IntrStatus write(w) val = 0x%04x\n", val);
193 
194     s->intrstatus = val;
195     ivshmem_update_irq(s);
196 }
197 
198 static uint32_t ivshmem_IntrStatus_read(IVShmemState *s)
199 {
200     uint32_t ret = s->intrstatus;
201 
202     /* reading ISR clears all interrupts */
203     s->intrstatus = 0;
204     ivshmem_update_irq(s);
205     return ret;
206 }
207 
208 static void ivshmem_io_write(void *opaque, hwaddr addr,
209                              uint64_t val, unsigned size)
210 {
211     IVShmemState *s = opaque;
212 
213     uint16_t dest = val >> 16;
214     uint16_t vector = val & 0xff;
215 
216     addr &= 0xfc;
217 
218     IVSHMEM_DPRINTF("writing to addr " TARGET_FMT_plx "\n", addr);
219     switch (addr)
220     {
221         case INTRMASK:
222             ivshmem_IntrMask_write(s, val);
223             break;
224 
225         case INTRSTATUS:
226             ivshmem_IntrStatus_write(s, val);
227             break;
228 
229         case DOORBELL:
230             /* check that dest VM ID is reasonable */
231             if (dest >= s->nb_peers) {
232                 IVSHMEM_DPRINTF("Invalid destination VM ID (%d)\n", dest);
233                 break;
234             }
235 
236             /* check doorbell range */
237             if (vector < s->peers[dest].nb_eventfds) {
238                 IVSHMEM_DPRINTF("Notifying VM %d on vector %d\n", dest, vector);
239                 event_notifier_set(&s->peers[dest].eventfds[vector]);
240             } else {
241                 IVSHMEM_DPRINTF("Invalid destination vector %d on VM %d\n",
242                                 vector, dest);
243             }
244             break;
245         default:
246             IVSHMEM_DPRINTF("Unhandled write " TARGET_FMT_plx "\n", addr);
247     }
248 }
249 
250 static uint64_t ivshmem_io_read(void *opaque, hwaddr addr,
251                                 unsigned size)
252 {
253 
254     IVShmemState *s = opaque;
255     uint32_t ret;
256 
257     switch (addr)
258     {
259         case INTRMASK:
260             ret = ivshmem_IntrMask_read(s);
261             break;
262 
263         case INTRSTATUS:
264             ret = ivshmem_IntrStatus_read(s);
265             break;
266 
267         case IVPOSITION:
268             ret = s->vm_id;
269             break;
270 
271         default:
272             IVSHMEM_DPRINTF("why are we reading " TARGET_FMT_plx "\n", addr);
273             ret = 0;
274     }
275 
276     return ret;
277 }
278 
279 static const MemoryRegionOps ivshmem_mmio_ops = {
280     .read = ivshmem_io_read,
281     .write = ivshmem_io_write,
282     .endianness = DEVICE_NATIVE_ENDIAN,
283     .impl = {
284         .min_access_size = 4,
285         .max_access_size = 4,
286     },
287 };
288 
289 static void ivshmem_vector_notify(void *opaque)
290 {
291     MSIVector *entry = opaque;
292     PCIDevice *pdev = entry->pdev;
293     IVShmemState *s = IVSHMEM_COMMON(pdev);
294     int vector = entry - s->msi_vectors;
295     EventNotifier *n = &s->peers[s->vm_id].eventfds[vector];
296 
297     if (!event_notifier_test_and_clear(n)) {
298         return;
299     }
300 
301     IVSHMEM_DPRINTF("interrupt on vector %p %d\n", pdev, vector);
302     if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
303         if (msix_enabled(pdev)) {
304             msix_notify(pdev, vector);
305         }
306     } else {
307         ivshmem_IntrStatus_write(s, 1);
308     }
309 }
310 
311 static int ivshmem_vector_unmask(PCIDevice *dev, unsigned vector,
312                                  MSIMessage msg)
313 {
314     IVShmemState *s = IVSHMEM_COMMON(dev);
315     EventNotifier *n = &s->peers[s->vm_id].eventfds[vector];
316     MSIVector *v = &s->msi_vectors[vector];
317     int ret;
318 
319     IVSHMEM_DPRINTF("vector unmask %p %d\n", dev, vector);
320 
321     ret = kvm_irqchip_update_msi_route(kvm_state, v->virq, msg, dev);
322     if (ret < 0) {
323         return ret;
324     }
325     kvm_irqchip_commit_routes(kvm_state);
326 
327     return kvm_irqchip_add_irqfd_notifier_gsi(kvm_state, n, NULL, v->virq);
328 }
329 
330 static void ivshmem_vector_mask(PCIDevice *dev, unsigned vector)
331 {
332     IVShmemState *s = IVSHMEM_COMMON(dev);
333     EventNotifier *n = &s->peers[s->vm_id].eventfds[vector];
334     int ret;
335 
336     IVSHMEM_DPRINTF("vector mask %p %d\n", dev, vector);
337 
338     ret = kvm_irqchip_remove_irqfd_notifier_gsi(kvm_state, n,
339                                                 s->msi_vectors[vector].virq);
340     if (ret != 0) {
341         error_report("remove_irqfd_notifier_gsi failed");
342     }
343 }
344 
345 static void ivshmem_vector_poll(PCIDevice *dev,
346                                 unsigned int vector_start,
347                                 unsigned int vector_end)
348 {
349     IVShmemState *s = IVSHMEM_COMMON(dev);
350     unsigned int vector;
351 
352     IVSHMEM_DPRINTF("vector poll %p %d-%d\n", dev, vector_start, vector_end);
353 
354     vector_end = MIN(vector_end, s->vectors);
355 
356     for (vector = vector_start; vector < vector_end; vector++) {
357         EventNotifier *notifier = &s->peers[s->vm_id].eventfds[vector];
358 
359         if (!msix_is_masked(dev, vector)) {
360             continue;
361         }
362 
363         if (event_notifier_test_and_clear(notifier)) {
364             msix_set_pending(dev, vector);
365         }
366     }
367 }
368 
369 static void watch_vector_notifier(IVShmemState *s, EventNotifier *n,
370                                  int vector)
371 {
372     int eventfd = event_notifier_get_fd(n);
373 
374     assert(!s->msi_vectors[vector].pdev);
375     s->msi_vectors[vector].pdev = PCI_DEVICE(s);
376 
377     qemu_set_fd_handler(eventfd, ivshmem_vector_notify,
378                         NULL, &s->msi_vectors[vector]);
379 }
380 
381 static void ivshmem_add_eventfd(IVShmemState *s, int posn, int i)
382 {
383     memory_region_add_eventfd(&s->ivshmem_mmio,
384                               DOORBELL,
385                               4,
386                               true,
387                               (posn << 16) | i,
388                               &s->peers[posn].eventfds[i]);
389 }
390 
391 static void ivshmem_del_eventfd(IVShmemState *s, int posn, int i)
392 {
393     memory_region_del_eventfd(&s->ivshmem_mmio,
394                               DOORBELL,
395                               4,
396                               true,
397                               (posn << 16) | i,
398                               &s->peers[posn].eventfds[i]);
399 }
400 
401 static void close_peer_eventfds(IVShmemState *s, int posn)
402 {
403     int i, n;
404 
405     assert(posn >= 0 && posn < s->nb_peers);
406     n = s->peers[posn].nb_eventfds;
407 
408     if (ivshmem_has_feature(s, IVSHMEM_IOEVENTFD)) {
409         memory_region_transaction_begin();
410         for (i = 0; i < n; i++) {
411             ivshmem_del_eventfd(s, posn, i);
412         }
413         memory_region_transaction_commit();
414     }
415 
416     for (i = 0; i < n; i++) {
417         event_notifier_cleanup(&s->peers[posn].eventfds[i]);
418     }
419 
420     g_free(s->peers[posn].eventfds);
421     s->peers[posn].nb_eventfds = 0;
422 }
423 
424 static void resize_peers(IVShmemState *s, int nb_peers)
425 {
426     int old_nb_peers = s->nb_peers;
427     int i;
428 
429     assert(nb_peers > old_nb_peers);
430     IVSHMEM_DPRINTF("bumping storage to %d peers\n", nb_peers);
431 
432     s->peers = g_realloc(s->peers, nb_peers * sizeof(Peer));
433     s->nb_peers = nb_peers;
434 
435     for (i = old_nb_peers; i < nb_peers; i++) {
436         s->peers[i].eventfds = g_new0(EventNotifier, s->vectors);
437         s->peers[i].nb_eventfds = 0;
438     }
439 }
440 
441 static void ivshmem_add_kvm_msi_virq(IVShmemState *s, int vector,
442                                      Error **errp)
443 {
444     PCIDevice *pdev = PCI_DEVICE(s);
445     int ret;
446 
447     IVSHMEM_DPRINTF("ivshmem_add_kvm_msi_virq vector:%d\n", vector);
448     assert(!s->msi_vectors[vector].pdev);
449 
450     ret = kvm_irqchip_add_msi_route(kvm_state, vector, pdev);
451     if (ret < 0) {
452         error_setg(errp, "kvm_irqchip_add_msi_route failed");
453         return;
454     }
455 
456     s->msi_vectors[vector].virq = ret;
457     s->msi_vectors[vector].pdev = pdev;
458 }
459 
460 static void setup_interrupt(IVShmemState *s, int vector, Error **errp)
461 {
462     EventNotifier *n = &s->peers[s->vm_id].eventfds[vector];
463     bool with_irqfd = kvm_msi_via_irqfd_enabled() &&
464         ivshmem_has_feature(s, IVSHMEM_MSI);
465     PCIDevice *pdev = PCI_DEVICE(s);
466     Error *err = NULL;
467 
468     IVSHMEM_DPRINTF("setting up interrupt for vector: %d\n", vector);
469 
470     if (!with_irqfd) {
471         IVSHMEM_DPRINTF("with eventfd\n");
472         watch_vector_notifier(s, n, vector);
473     } else if (msix_enabled(pdev)) {
474         IVSHMEM_DPRINTF("with irqfd\n");
475         ivshmem_add_kvm_msi_virq(s, vector, &err);
476         if (err) {
477             error_propagate(errp, err);
478             return;
479         }
480 
481         if (!msix_is_masked(pdev, vector)) {
482             kvm_irqchip_add_irqfd_notifier_gsi(kvm_state, n, NULL,
483                                                s->msi_vectors[vector].virq);
484             /* TODO handle error */
485         }
486     } else {
487         /* it will be delayed until msix is enabled, in write_config */
488         IVSHMEM_DPRINTF("with irqfd, delayed until msix enabled\n");
489     }
490 }
491 
492 static void process_msg_shmem(IVShmemState *s, int fd, Error **errp)
493 {
494     struct stat buf;
495     size_t size;
496     void *ptr;
497 
498     if (s->ivshmem_bar2) {
499         error_setg(errp, "server sent unexpected shared memory message");
500         close(fd);
501         return;
502     }
503 
504     if (fstat(fd, &buf) < 0) {
505         error_setg_errno(errp, errno,
506             "can't determine size of shared memory sent by server");
507         close(fd);
508         return;
509     }
510 
511     size = buf.st_size;
512 
513     /* Legacy cruft */
514     if (s->legacy_size != SIZE_MAX) {
515         if (size < s->legacy_size) {
516             error_setg(errp, "server sent only %zd bytes of shared memory",
517                        (size_t)buf.st_size);
518             close(fd);
519             return;
520         }
521         size = s->legacy_size;
522     }
523 
524     /* mmap the region and map into the BAR2 */
525     ptr = mmap(0, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
526     if (ptr == MAP_FAILED) {
527         error_setg_errno(errp, errno, "Failed to mmap shared memory");
528         close(fd);
529         return;
530     }
531     memory_region_init_ram_ptr(&s->server_bar2, OBJECT(s),
532                                "ivshmem.bar2", size, ptr);
533     memory_region_set_fd(&s->server_bar2, fd);
534     s->ivshmem_bar2 = &s->server_bar2;
535 }
536 
537 static void process_msg_disconnect(IVShmemState *s, uint16_t posn,
538                                    Error **errp)
539 {
540     IVSHMEM_DPRINTF("posn %d has gone away\n", posn);
541     if (posn >= s->nb_peers || posn == s->vm_id) {
542         error_setg(errp, "invalid peer %d", posn);
543         return;
544     }
545     close_peer_eventfds(s, posn);
546 }
547 
548 static void process_msg_connect(IVShmemState *s, uint16_t posn, int fd,
549                                 Error **errp)
550 {
551     Peer *peer = &s->peers[posn];
552     int vector;
553 
554     /*
555      * The N-th connect message for this peer comes with the file
556      * descriptor for vector N-1.  Count messages to find the vector.
557      */
558     if (peer->nb_eventfds >= s->vectors) {
559         error_setg(errp, "Too many eventfd received, device has %d vectors",
560                    s->vectors);
561         close(fd);
562         return;
563     }
564     vector = peer->nb_eventfds++;
565 
566     IVSHMEM_DPRINTF("eventfds[%d][%d] = %d\n", posn, vector, fd);
567     event_notifier_init_fd(&peer->eventfds[vector], fd);
568     fcntl_setfl(fd, O_NONBLOCK); /* msix/irqfd poll non block */
569 
570     if (posn == s->vm_id) {
571         setup_interrupt(s, vector, errp);
572         /* TODO do we need to handle the error? */
573     }
574 
575     if (ivshmem_has_feature(s, IVSHMEM_IOEVENTFD)) {
576         ivshmem_add_eventfd(s, posn, vector);
577     }
578 }
579 
580 static void process_msg(IVShmemState *s, int64_t msg, int fd, Error **errp)
581 {
582     IVSHMEM_DPRINTF("posn is %" PRId64 ", fd is %d\n", msg, fd);
583 
584     if (msg < -1 || msg > IVSHMEM_MAX_PEERS) {
585         error_setg(errp, "server sent invalid message %" PRId64, msg);
586         close(fd);
587         return;
588     }
589 
590     if (msg == -1) {
591         process_msg_shmem(s, fd, errp);
592         return;
593     }
594 
595     if (msg >= s->nb_peers) {
596         resize_peers(s, msg + 1);
597     }
598 
599     if (fd >= 0) {
600         process_msg_connect(s, msg, fd, errp);
601     } else {
602         process_msg_disconnect(s, msg, errp);
603     }
604 }
605 
606 static int ivshmem_can_receive(void *opaque)
607 {
608     IVShmemState *s = opaque;
609 
610     assert(s->msg_buffered_bytes < sizeof(s->msg_buf));
611     return sizeof(s->msg_buf) - s->msg_buffered_bytes;
612 }
613 
614 static void ivshmem_read(void *opaque, const uint8_t *buf, int size)
615 {
616     IVShmemState *s = opaque;
617     Error *err = NULL;
618     int fd;
619     int64_t msg;
620 
621     assert(size >= 0 && s->msg_buffered_bytes + size <= sizeof(s->msg_buf));
622     memcpy((unsigned char *)&s->msg_buf + s->msg_buffered_bytes, buf, size);
623     s->msg_buffered_bytes += size;
624     if (s->msg_buffered_bytes < sizeof(s->msg_buf)) {
625         return;
626     }
627     msg = le64_to_cpu(s->msg_buf);
628     s->msg_buffered_bytes = 0;
629 
630     fd = qemu_chr_fe_get_msgfd(&s->server_chr);
631 
632     process_msg(s, msg, fd, &err);
633     if (err) {
634         error_report_err(err);
635     }
636 }
637 
638 static int64_t ivshmem_recv_msg(IVShmemState *s, int *pfd, Error **errp)
639 {
640     int64_t msg;
641     int n, ret;
642 
643     n = 0;
644     do {
645         ret = qemu_chr_fe_read_all(&s->server_chr, (uint8_t *)&msg + n,
646                                    sizeof(msg) - n);
647         if (ret < 0 && ret != -EINTR) {
648             error_setg_errno(errp, -ret, "read from server failed");
649             return INT64_MIN;
650         }
651         n += ret;
652     } while (n < sizeof(msg));
653 
654     *pfd = qemu_chr_fe_get_msgfd(&s->server_chr);
655     return msg;
656 }
657 
658 static void ivshmem_recv_setup(IVShmemState *s, Error **errp)
659 {
660     Error *err = NULL;
661     int64_t msg;
662     int fd;
663 
664     msg = ivshmem_recv_msg(s, &fd, &err);
665     if (err) {
666         error_propagate(errp, err);
667         return;
668     }
669     if (msg != IVSHMEM_PROTOCOL_VERSION) {
670         error_setg(errp, "server sent version %" PRId64 ", expecting %d",
671                    msg, IVSHMEM_PROTOCOL_VERSION);
672         return;
673     }
674     if (fd != -1) {
675         error_setg(errp, "server sent invalid version message");
676         return;
677     }
678 
679     /*
680      * ivshmem-server sends the remaining initial messages in a fixed
681      * order, but the device has always accepted them in any order.
682      * Stay as compatible as practical, just in case people use
683      * servers that behave differently.
684      */
685 
686     /*
687      * ivshmem_device_spec.txt has always required the ID message
688      * right here, and ivshmem-server has always complied.  However,
689      * older versions of the device accepted it out of order, but
690      * broke when an interrupt setup message arrived before it.
691      */
692     msg = ivshmem_recv_msg(s, &fd, &err);
693     if (err) {
694         error_propagate(errp, err);
695         return;
696     }
697     if (fd != -1 || msg < 0 || msg > IVSHMEM_MAX_PEERS) {
698         error_setg(errp, "server sent invalid ID message");
699         return;
700     }
701     s->vm_id = msg;
702 
703     /*
704      * Receive more messages until we got shared memory.
705      */
706     do {
707         msg = ivshmem_recv_msg(s, &fd, &err);
708         if (err) {
709             error_propagate(errp, err);
710             return;
711         }
712         process_msg(s, msg, fd, &err);
713         if (err) {
714             error_propagate(errp, err);
715             return;
716         }
717     } while (msg != -1);
718 
719     /*
720      * This function must either map the shared memory or fail.  The
721      * loop above ensures that: it terminates normally only after it
722      * successfully processed the server's shared memory message.
723      * Assert that actually mapped the shared memory:
724      */
725     assert(s->ivshmem_bar2);
726 }
727 
728 /* Select the MSI-X vectors used by device.
729  * ivshmem maps events to vectors statically, so
730  * we just enable all vectors on init and after reset. */
731 static void ivshmem_msix_vector_use(IVShmemState *s)
732 {
733     PCIDevice *d = PCI_DEVICE(s);
734     int i;
735 
736     for (i = 0; i < s->vectors; i++) {
737         msix_vector_use(d, i);
738     }
739 }
740 
741 static void ivshmem_reset(DeviceState *d)
742 {
743     IVShmemState *s = IVSHMEM_COMMON(d);
744 
745     s->intrstatus = 0;
746     s->intrmask = 0;
747     if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
748         ivshmem_msix_vector_use(s);
749     }
750 }
751 
752 static int ivshmem_setup_interrupts(IVShmemState *s)
753 {
754     /* allocate QEMU callback data for receiving interrupts */
755     s->msi_vectors = g_malloc0(s->vectors * sizeof(MSIVector));
756 
757     if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
758         if (msix_init_exclusive_bar(PCI_DEVICE(s), s->vectors, 1)) {
759             return -1;
760         }
761 
762         IVSHMEM_DPRINTF("msix initialized (%d vectors)\n", s->vectors);
763         ivshmem_msix_vector_use(s);
764     }
765 
766     return 0;
767 }
768 
769 static void ivshmem_enable_irqfd(IVShmemState *s)
770 {
771     PCIDevice *pdev = PCI_DEVICE(s);
772     int i;
773 
774     for (i = 0; i < s->peers[s->vm_id].nb_eventfds; i++) {
775         Error *err = NULL;
776 
777         ivshmem_add_kvm_msi_virq(s, i, &err);
778         if (err) {
779             error_report_err(err);
780             /* TODO do we need to handle the error? */
781         }
782     }
783 
784     if (msix_set_vector_notifiers(pdev,
785                                   ivshmem_vector_unmask,
786                                   ivshmem_vector_mask,
787                                   ivshmem_vector_poll)) {
788         error_report("ivshmem: msix_set_vector_notifiers failed");
789     }
790 }
791 
792 static void ivshmem_remove_kvm_msi_virq(IVShmemState *s, int vector)
793 {
794     IVSHMEM_DPRINTF("ivshmem_remove_kvm_msi_virq vector:%d\n", vector);
795 
796     if (s->msi_vectors[vector].pdev == NULL) {
797         return;
798     }
799 
800     /* it was cleaned when masked in the frontend. */
801     kvm_irqchip_release_virq(kvm_state, s->msi_vectors[vector].virq);
802 
803     s->msi_vectors[vector].pdev = NULL;
804 }
805 
806 static void ivshmem_disable_irqfd(IVShmemState *s)
807 {
808     PCIDevice *pdev = PCI_DEVICE(s);
809     int i;
810 
811     for (i = 0; i < s->peers[s->vm_id].nb_eventfds; i++) {
812         ivshmem_remove_kvm_msi_virq(s, i);
813     }
814 
815     msix_unset_vector_notifiers(pdev);
816 }
817 
818 static void ivshmem_write_config(PCIDevice *pdev, uint32_t address,
819                                  uint32_t val, int len)
820 {
821     IVShmemState *s = IVSHMEM_COMMON(pdev);
822     int is_enabled, was_enabled = msix_enabled(pdev);
823 
824     pci_default_write_config(pdev, address, val, len);
825     is_enabled = msix_enabled(pdev);
826 
827     if (kvm_msi_via_irqfd_enabled()) {
828         if (!was_enabled && is_enabled) {
829             ivshmem_enable_irqfd(s);
830         } else if (was_enabled && !is_enabled) {
831             ivshmem_disable_irqfd(s);
832         }
833     }
834 }
835 
836 static void ivshmem_common_realize(PCIDevice *dev, Error **errp)
837 {
838     IVShmemState *s = IVSHMEM_COMMON(dev);
839     Error *err = NULL;
840     uint8_t *pci_conf;
841     uint8_t attr = PCI_BASE_ADDRESS_SPACE_MEMORY |
842         PCI_BASE_ADDRESS_MEM_PREFETCH;
843     Error *local_err = NULL;
844 
845     /* IRQFD requires MSI */
846     if (ivshmem_has_feature(s, IVSHMEM_IOEVENTFD) &&
847         !ivshmem_has_feature(s, IVSHMEM_MSI)) {
848         error_setg(errp, "ioeventfd/irqfd requires MSI");
849         return;
850     }
851 
852     pci_conf = dev->config;
853     pci_conf[PCI_COMMAND] = PCI_COMMAND_IO | PCI_COMMAND_MEMORY;
854 
855     memory_region_init_io(&s->ivshmem_mmio, OBJECT(s), &ivshmem_mmio_ops, s,
856                           "ivshmem-mmio", IVSHMEM_REG_BAR_SIZE);
857 
858     /* region for registers*/
859     pci_register_bar(dev, 0, PCI_BASE_ADDRESS_SPACE_MEMORY,
860                      &s->ivshmem_mmio);
861 
862     if (s->not_legacy_32bit) {
863         attr |= PCI_BASE_ADDRESS_MEM_TYPE_64;
864     }
865 
866     if (s->hostmem != NULL) {
867         IVSHMEM_DPRINTF("using hostmem\n");
868 
869         s->ivshmem_bar2 = host_memory_backend_get_memory(s->hostmem,
870                                                          &error_abort);
871     } else {
872         CharDriverState *chr = qemu_chr_fe_get_driver(&s->server_chr);
873         assert(chr);
874 
875         IVSHMEM_DPRINTF("using shared memory server (socket = %s)\n",
876                         chr->filename);
877 
878         /* we allocate enough space for 16 peers and grow as needed */
879         resize_peers(s, 16);
880 
881         /*
882          * Receive setup messages from server synchronously.
883          * Older versions did it asynchronously, but that creates a
884          * number of entertaining race conditions.
885          */
886         ivshmem_recv_setup(s, &err);
887         if (err) {
888             error_propagate(errp, err);
889             return;
890         }
891 
892         if (s->master == ON_OFF_AUTO_ON && s->vm_id != 0) {
893             error_setg(errp,
894                        "master must connect to the server before any peers");
895             return;
896         }
897 
898         qemu_chr_fe_set_handlers(&s->server_chr, ivshmem_can_receive,
899                                  ivshmem_read, NULL, s, NULL, true);
900 
901         if (ivshmem_setup_interrupts(s) < 0) {
902             error_setg(errp, "failed to initialize interrupts");
903             return;
904         }
905     }
906 
907     if (s->master == ON_OFF_AUTO_AUTO) {
908         s->master = s->vm_id == 0 ? ON_OFF_AUTO_ON : ON_OFF_AUTO_OFF;
909     }
910 
911     if (!ivshmem_is_master(s)) {
912         error_setg(&s->migration_blocker,
913                    "Migration is disabled when using feature 'peer mode' in device 'ivshmem'");
914         migrate_add_blocker(s->migration_blocker, &local_err);
915         if (local_err) {
916             error_propagate(errp, local_err);
917             error_free(s->migration_blocker);
918             return;
919         }
920     }
921 
922     vmstate_register_ram(s->ivshmem_bar2, DEVICE(s));
923     pci_register_bar(PCI_DEVICE(s), 2, attr, s->ivshmem_bar2);
924 }
925 
926 static void ivshmem_exit(PCIDevice *dev)
927 {
928     IVShmemState *s = IVSHMEM_COMMON(dev);
929     int i;
930 
931     if (s->migration_blocker) {
932         migrate_del_blocker(s->migration_blocker);
933         error_free(s->migration_blocker);
934     }
935 
936     if (memory_region_is_mapped(s->ivshmem_bar2)) {
937         if (!s->hostmem) {
938             void *addr = memory_region_get_ram_ptr(s->ivshmem_bar2);
939             int fd;
940 
941             if (munmap(addr, memory_region_size(s->ivshmem_bar2) == -1)) {
942                 error_report("Failed to munmap shared memory %s",
943                              strerror(errno));
944             }
945 
946             fd = memory_region_get_fd(s->ivshmem_bar2);
947             close(fd);
948         }
949 
950         vmstate_unregister_ram(s->ivshmem_bar2, DEVICE(dev));
951     }
952 
953     if (s->peers) {
954         for (i = 0; i < s->nb_peers; i++) {
955             close_peer_eventfds(s, i);
956         }
957         g_free(s->peers);
958     }
959 
960     if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
961         msix_uninit_exclusive_bar(dev);
962     }
963 
964     g_free(s->msi_vectors);
965 }
966 
967 static int ivshmem_pre_load(void *opaque)
968 {
969     IVShmemState *s = opaque;
970 
971     if (!ivshmem_is_master(s)) {
972         error_report("'peer' devices are not migratable");
973         return -EINVAL;
974     }
975 
976     return 0;
977 }
978 
979 static int ivshmem_post_load(void *opaque, int version_id)
980 {
981     IVShmemState *s = opaque;
982 
983     if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
984         ivshmem_msix_vector_use(s);
985     }
986     return 0;
987 }
988 
989 static void ivshmem_common_class_init(ObjectClass *klass, void *data)
990 {
991     DeviceClass *dc = DEVICE_CLASS(klass);
992     PCIDeviceClass *k = PCI_DEVICE_CLASS(klass);
993 
994     k->realize = ivshmem_common_realize;
995     k->exit = ivshmem_exit;
996     k->config_write = ivshmem_write_config;
997     k->vendor_id = PCI_VENDOR_ID_IVSHMEM;
998     k->device_id = PCI_DEVICE_ID_IVSHMEM;
999     k->class_id = PCI_CLASS_MEMORY_RAM;
1000     k->revision = 1;
1001     dc->reset = ivshmem_reset;
1002     set_bit(DEVICE_CATEGORY_MISC, dc->categories);
1003     dc->desc = "Inter-VM shared memory";
1004 }
1005 
1006 static const TypeInfo ivshmem_common_info = {
1007     .name          = TYPE_IVSHMEM_COMMON,
1008     .parent        = TYPE_PCI_DEVICE,
1009     .instance_size = sizeof(IVShmemState),
1010     .abstract      = true,
1011     .class_init    = ivshmem_common_class_init,
1012 };
1013 
1014 static void ivshmem_check_memdev_is_busy(Object *obj, const char *name,
1015                                          Object *val, Error **errp)
1016 {
1017     if (host_memory_backend_is_mapped(MEMORY_BACKEND(val))) {
1018         char *path = object_get_canonical_path_component(val);
1019         error_setg(errp, "can't use already busy memdev: %s", path);
1020         g_free(path);
1021     } else {
1022         qdev_prop_allow_set_link_before_realize(obj, name, val, errp);
1023     }
1024 }
1025 
1026 static const VMStateDescription ivshmem_plain_vmsd = {
1027     .name = TYPE_IVSHMEM_PLAIN,
1028     .version_id = 0,
1029     .minimum_version_id = 0,
1030     .pre_load = ivshmem_pre_load,
1031     .post_load = ivshmem_post_load,
1032     .fields = (VMStateField[]) {
1033         VMSTATE_PCI_DEVICE(parent_obj, IVShmemState),
1034         VMSTATE_UINT32(intrstatus, IVShmemState),
1035         VMSTATE_UINT32(intrmask, IVShmemState),
1036         VMSTATE_END_OF_LIST()
1037     },
1038 };
1039 
1040 static Property ivshmem_plain_properties[] = {
1041     DEFINE_PROP_ON_OFF_AUTO("master", IVShmemState, master, ON_OFF_AUTO_OFF),
1042     DEFINE_PROP_END_OF_LIST(),
1043 };
1044 
1045 static void ivshmem_plain_init(Object *obj)
1046 {
1047     IVShmemState *s = IVSHMEM_PLAIN(obj);
1048 
1049     object_property_add_link(obj, "memdev", TYPE_MEMORY_BACKEND,
1050                              (Object **)&s->hostmem,
1051                              ivshmem_check_memdev_is_busy,
1052                              OBJ_PROP_LINK_UNREF_ON_RELEASE,
1053                              &error_abort);
1054     s->not_legacy_32bit = 1;
1055 }
1056 
1057 static void ivshmem_plain_realize(PCIDevice *dev, Error **errp)
1058 {
1059     IVShmemState *s = IVSHMEM_COMMON(dev);
1060 
1061     if (!s->hostmem) {
1062         error_setg(errp, "You must specify a 'memdev'");
1063         return;
1064     }
1065 
1066     ivshmem_common_realize(dev, errp);
1067     host_memory_backend_set_mapped(s->hostmem, true);
1068 }
1069 
1070 static void ivshmem_plain_exit(PCIDevice *pci_dev)
1071 {
1072     IVShmemState *s = IVSHMEM_COMMON(pci_dev);
1073 
1074     host_memory_backend_set_mapped(s->hostmem, false);
1075 }
1076 
1077 static void ivshmem_plain_class_init(ObjectClass *klass, void *data)
1078 {
1079     DeviceClass *dc = DEVICE_CLASS(klass);
1080     PCIDeviceClass *k = PCI_DEVICE_CLASS(klass);
1081 
1082     k->realize = ivshmem_plain_realize;
1083     k->exit = ivshmem_plain_exit;
1084     dc->props = ivshmem_plain_properties;
1085     dc->vmsd = &ivshmem_plain_vmsd;
1086 }
1087 
1088 static const TypeInfo ivshmem_plain_info = {
1089     .name          = TYPE_IVSHMEM_PLAIN,
1090     .parent        = TYPE_IVSHMEM_COMMON,
1091     .instance_size = sizeof(IVShmemState),
1092     .instance_init = ivshmem_plain_init,
1093     .class_init    = ivshmem_plain_class_init,
1094 };
1095 
1096 static const VMStateDescription ivshmem_doorbell_vmsd = {
1097     .name = TYPE_IVSHMEM_DOORBELL,
1098     .version_id = 0,
1099     .minimum_version_id = 0,
1100     .pre_load = ivshmem_pre_load,
1101     .post_load = ivshmem_post_load,
1102     .fields = (VMStateField[]) {
1103         VMSTATE_PCI_DEVICE(parent_obj, IVShmemState),
1104         VMSTATE_MSIX(parent_obj, IVShmemState),
1105         VMSTATE_UINT32(intrstatus, IVShmemState),
1106         VMSTATE_UINT32(intrmask, IVShmemState),
1107         VMSTATE_END_OF_LIST()
1108     },
1109 };
1110 
1111 static Property ivshmem_doorbell_properties[] = {
1112     DEFINE_PROP_CHR("chardev", IVShmemState, server_chr),
1113     DEFINE_PROP_UINT32("vectors", IVShmemState, vectors, 1),
1114     DEFINE_PROP_BIT("ioeventfd", IVShmemState, features, IVSHMEM_IOEVENTFD,
1115                     true),
1116     DEFINE_PROP_ON_OFF_AUTO("master", IVShmemState, master, ON_OFF_AUTO_OFF),
1117     DEFINE_PROP_END_OF_LIST(),
1118 };
1119 
1120 static void ivshmem_doorbell_init(Object *obj)
1121 {
1122     IVShmemState *s = IVSHMEM_DOORBELL(obj);
1123 
1124     s->features |= (1 << IVSHMEM_MSI);
1125     s->legacy_size = SIZE_MAX;  /* whatever the server sends */
1126     s->not_legacy_32bit = 1;
1127 }
1128 
1129 static void ivshmem_doorbell_realize(PCIDevice *dev, Error **errp)
1130 {
1131     IVShmemState *s = IVSHMEM_COMMON(dev);
1132 
1133     if (!qemu_chr_fe_get_driver(&s->server_chr)) {
1134         error_setg(errp, "You must specify a 'chardev'");
1135         return;
1136     }
1137 
1138     ivshmem_common_realize(dev, errp);
1139 }
1140 
1141 static void ivshmem_doorbell_class_init(ObjectClass *klass, void *data)
1142 {
1143     DeviceClass *dc = DEVICE_CLASS(klass);
1144     PCIDeviceClass *k = PCI_DEVICE_CLASS(klass);
1145 
1146     k->realize = ivshmem_doorbell_realize;
1147     dc->props = ivshmem_doorbell_properties;
1148     dc->vmsd = &ivshmem_doorbell_vmsd;
1149 }
1150 
1151 static const TypeInfo ivshmem_doorbell_info = {
1152     .name          = TYPE_IVSHMEM_DOORBELL,
1153     .parent        = TYPE_IVSHMEM_COMMON,
1154     .instance_size = sizeof(IVShmemState),
1155     .instance_init = ivshmem_doorbell_init,
1156     .class_init    = ivshmem_doorbell_class_init,
1157 };
1158 
1159 static int ivshmem_load_old(QEMUFile *f, void *opaque, int version_id)
1160 {
1161     IVShmemState *s = opaque;
1162     PCIDevice *pdev = PCI_DEVICE(s);
1163     int ret;
1164 
1165     IVSHMEM_DPRINTF("ivshmem_load_old\n");
1166 
1167     if (version_id != 0) {
1168         return -EINVAL;
1169     }
1170 
1171     ret = ivshmem_pre_load(s);
1172     if (ret) {
1173         return ret;
1174     }
1175 
1176     ret = pci_device_load(pdev, f);
1177     if (ret) {
1178         return ret;
1179     }
1180 
1181     if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
1182         msix_load(pdev, f);
1183         ivshmem_msix_vector_use(s);
1184     } else {
1185         s->intrstatus = qemu_get_be32(f);
1186         s->intrmask = qemu_get_be32(f);
1187     }
1188 
1189     return 0;
1190 }
1191 
1192 static bool test_msix(void *opaque, int version_id)
1193 {
1194     IVShmemState *s = opaque;
1195 
1196     return ivshmem_has_feature(s, IVSHMEM_MSI);
1197 }
1198 
1199 static bool test_no_msix(void *opaque, int version_id)
1200 {
1201     return !test_msix(opaque, version_id);
1202 }
1203 
1204 static const VMStateDescription ivshmem_vmsd = {
1205     .name = "ivshmem",
1206     .version_id = 1,
1207     .minimum_version_id = 1,
1208     .pre_load = ivshmem_pre_load,
1209     .post_load = ivshmem_post_load,
1210     .fields = (VMStateField[]) {
1211         VMSTATE_PCI_DEVICE(parent_obj, IVShmemState),
1212 
1213         VMSTATE_MSIX_TEST(parent_obj, IVShmemState, test_msix),
1214         VMSTATE_UINT32_TEST(intrstatus, IVShmemState, test_no_msix),
1215         VMSTATE_UINT32_TEST(intrmask, IVShmemState, test_no_msix),
1216 
1217         VMSTATE_END_OF_LIST()
1218     },
1219     .load_state_old = ivshmem_load_old,
1220     .minimum_version_id_old = 0
1221 };
1222 
1223 static Property ivshmem_properties[] = {
1224     DEFINE_PROP_CHR("chardev", IVShmemState, server_chr),
1225     DEFINE_PROP_STRING("size", IVShmemState, sizearg),
1226     DEFINE_PROP_UINT32("vectors", IVShmemState, vectors, 1),
1227     DEFINE_PROP_BIT("ioeventfd", IVShmemState, features, IVSHMEM_IOEVENTFD,
1228                     false),
1229     DEFINE_PROP_BIT("msi", IVShmemState, features, IVSHMEM_MSI, true),
1230     DEFINE_PROP_STRING("shm", IVShmemState, shmobj),
1231     DEFINE_PROP_STRING("role", IVShmemState, role),
1232     DEFINE_PROP_UINT32("use64", IVShmemState, not_legacy_32bit, 1),
1233     DEFINE_PROP_END_OF_LIST(),
1234 };
1235 
1236 static void desugar_shm(IVShmemState *s)
1237 {
1238     Object *obj;
1239     char *path;
1240 
1241     obj = object_new("memory-backend-file");
1242     path = g_strdup_printf("/dev/shm/%s", s->shmobj);
1243     object_property_set_str(obj, path, "mem-path", &error_abort);
1244     g_free(path);
1245     object_property_set_int(obj, s->legacy_size, "size", &error_abort);
1246     object_property_set_bool(obj, true, "share", &error_abort);
1247     object_property_add_child(OBJECT(s), "internal-shm-backend", obj,
1248                               &error_abort);
1249     user_creatable_complete(obj, &error_abort);
1250     s->hostmem = MEMORY_BACKEND(obj);
1251 }
1252 
1253 static void ivshmem_realize(PCIDevice *dev, Error **errp)
1254 {
1255     IVShmemState *s = IVSHMEM_COMMON(dev);
1256 
1257     if (!qtest_enabled()) {
1258         error_report("ivshmem is deprecated, please use ivshmem-plain"
1259                      " or ivshmem-doorbell instead");
1260     }
1261 
1262     if (!!qemu_chr_fe_get_driver(&s->server_chr) + !!s->shmobj != 1) {
1263         error_setg(errp, "You must specify either 'shm' or 'chardev'");
1264         return;
1265     }
1266 
1267     if (s->sizearg == NULL) {
1268         s->legacy_size = 4 << 20; /* 4 MB default */
1269     } else {
1270         char *end;
1271         int64_t size = qemu_strtosz(s->sizearg, &end);
1272         if (size < 0 || (size_t)size != size || *end != '\0'
1273             || !is_power_of_2(size)) {
1274             error_setg(errp, "Invalid size %s", s->sizearg);
1275             return;
1276         }
1277         s->legacy_size = size;
1278     }
1279 
1280     /* check that role is reasonable */
1281     if (s->role) {
1282         if (strncmp(s->role, "peer", 5) == 0) {
1283             s->master = ON_OFF_AUTO_OFF;
1284         } else if (strncmp(s->role, "master", 7) == 0) {
1285             s->master = ON_OFF_AUTO_ON;
1286         } else {
1287             error_setg(errp, "'role' must be 'peer' or 'master'");
1288             return;
1289         }
1290     } else {
1291         s->master = ON_OFF_AUTO_AUTO;
1292     }
1293 
1294     if (s->shmobj) {
1295         desugar_shm(s);
1296     }
1297 
1298     /*
1299      * Note: we don't use INTx with IVSHMEM_MSI at all, so this is a
1300      * bald-faced lie then.  But it's a backwards compatible lie.
1301      */
1302     pci_config_set_interrupt_pin(dev->config, 1);
1303 
1304     ivshmem_common_realize(dev, errp);
1305 }
1306 
1307 static void ivshmem_class_init(ObjectClass *klass, void *data)
1308 {
1309     DeviceClass *dc = DEVICE_CLASS(klass);
1310     PCIDeviceClass *k = PCI_DEVICE_CLASS(klass);
1311 
1312     k->realize = ivshmem_realize;
1313     k->revision = 0;
1314     dc->desc = "Inter-VM shared memory (legacy)";
1315     dc->props = ivshmem_properties;
1316     dc->vmsd = &ivshmem_vmsd;
1317 }
1318 
1319 static const TypeInfo ivshmem_info = {
1320     .name          = TYPE_IVSHMEM,
1321     .parent        = TYPE_IVSHMEM_COMMON,
1322     .instance_size = sizeof(IVShmemState),
1323     .class_init    = ivshmem_class_init,
1324 };
1325 
1326 static void ivshmem_register_types(void)
1327 {
1328     type_register_static(&ivshmem_common_info);
1329     type_register_static(&ivshmem_plain_info);
1330     type_register_static(&ivshmem_doorbell_info);
1331     type_register_static(&ivshmem_info);
1332 }
1333 
1334 type_init(ivshmem_register_types)
1335