]> Git Repo - qemu.git/blob - hw/misc/ivshmem.c
ivshmem: Fix fd leak on error
[qemu.git] / hw / misc / ivshmem.c
1 /*
2  * Inter-VM Shared Memory PCI device.
3  *
4  * Author:
5  *      Cam Macdonell <[email protected]>
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 "hw/hw.h"
20 #include "hw/i386/pc.h"
21 #include "hw/pci/pci.h"
22 #include "hw/pci/msix.h"
23 #include "sysemu/kvm.h"
24 #include "migration/migration.h"
25 #include "qapi/qmp/qerror.h"
26 #include "qemu/event_notifier.h"
27 #include "qemu/fifo8.h"
28 #include "sysemu/char.h"
29
30 #include <sys/mman.h>
31 #include <sys/types.h>
32 #include <limits.h>
33
34 #define PCI_VENDOR_ID_IVSHMEM   PCI_VENDOR_ID_REDHAT_QUMRANET
35 #define PCI_DEVICE_ID_IVSHMEM   0x1110
36
37 #define IVSHMEM_IOEVENTFD   0
38 #define IVSHMEM_MSI     1
39
40 #define IVSHMEM_PEER    0
41 #define IVSHMEM_MASTER  1
42
43 #define IVSHMEM_REG_BAR_SIZE 0x100
44
45 //#define DEBUG_IVSHMEM
46 #ifdef DEBUG_IVSHMEM
47 #define IVSHMEM_DPRINTF(fmt, ...)        \
48     do {printf("IVSHMEM: " fmt, ## __VA_ARGS__); } while (0)
49 #else
50 #define IVSHMEM_DPRINTF(fmt, ...)
51 #endif
52
53 #define TYPE_IVSHMEM "ivshmem"
54 #define IVSHMEM(obj) \
55     OBJECT_CHECK(IVShmemState, (obj), TYPE_IVSHMEM)
56
57 typedef struct Peer {
58     int nb_eventfds;
59     EventNotifier *eventfds;
60 } Peer;
61
62 typedef struct EventfdEntry {
63     PCIDevice *pdev;
64     int vector;
65 } EventfdEntry;
66
67 typedef struct IVShmemState {
68     /*< private >*/
69     PCIDevice parent_obj;
70     /*< public >*/
71
72     uint32_t intrmask;
73     uint32_t intrstatus;
74     uint32_t doorbell;
75
76     CharDriverState **eventfd_chr;
77     CharDriverState *server_chr;
78     Fifo8 incoming_fifo;
79     MemoryRegion ivshmem_mmio;
80
81     /* We might need to register the BAR before we actually have the memory.
82      * So prepare a container MemoryRegion for the BAR immediately and
83      * add a subregion when we have the memory.
84      */
85     MemoryRegion bar;
86     MemoryRegion ivshmem;
87     uint64_t ivshmem_size; /* size of shared memory region */
88     uint32_t ivshmem_attr;
89     uint32_t ivshmem_64bit;
90     int shm_fd; /* shared memory file descriptor */
91
92     Peer *peers;
93     int nb_peers; /* how many guests we have space for */
94     int max_peer; /* maximum numbered peer */
95
96     int vm_id;
97     uint32_t vectors;
98     uint32_t features;
99     EventfdEntry *eventfd_table;
100
101     Error *migration_blocker;
102
103     char * shmobj;
104     char * sizearg;
105     char * role;
106     int role_val;   /* scalar to avoid multiple string comparisons */
107 } IVShmemState;
108
109 /* registers for the Inter-VM shared memory device */
110 enum ivshmem_registers {
111     INTRMASK = 0,
112     INTRSTATUS = 4,
113     IVPOSITION = 8,
114     DOORBELL = 12,
115 };
116
117 static inline uint32_t ivshmem_has_feature(IVShmemState *ivs,
118                                                     unsigned int feature) {
119     return (ivs->features & (1 << feature));
120 }
121
122 static inline bool is_power_of_two(uint64_t x) {
123     return (x & (x - 1)) == 0;
124 }
125
126 /* accessing registers - based on rtl8139 */
127 static void ivshmem_update_irq(IVShmemState *s, int val)
128 {
129     PCIDevice *d = PCI_DEVICE(s);
130     int isr;
131     isr = (s->intrstatus & s->intrmask) & 0xffffffff;
132
133     /* don't print ISR resets */
134     if (isr) {
135         IVSHMEM_DPRINTF("Set IRQ to %d (%04x %04x)\n",
136            isr ? 1 : 0, s->intrstatus, s->intrmask);
137     }
138
139     pci_set_irq(d, (isr != 0));
140 }
141
142 static void ivshmem_IntrMask_write(IVShmemState *s, uint32_t val)
143 {
144     IVSHMEM_DPRINTF("IntrMask write(w) val = 0x%04x\n", val);
145
146     s->intrmask = val;
147
148     ivshmem_update_irq(s, val);
149 }
150
151 static uint32_t ivshmem_IntrMask_read(IVShmemState *s)
152 {
153     uint32_t ret = s->intrmask;
154
155     IVSHMEM_DPRINTF("intrmask read(w) val = 0x%04x\n", ret);
156
157     return ret;
158 }
159
160 static void ivshmem_IntrStatus_write(IVShmemState *s, uint32_t val)
161 {
162     IVSHMEM_DPRINTF("IntrStatus write(w) val = 0x%04x\n", val);
163
164     s->intrstatus = val;
165
166     ivshmem_update_irq(s, val);
167 }
168
169 static uint32_t ivshmem_IntrStatus_read(IVShmemState *s)
170 {
171     uint32_t ret = s->intrstatus;
172
173     /* reading ISR clears all interrupts */
174     s->intrstatus = 0;
175
176     ivshmem_update_irq(s, 0);
177
178     return ret;
179 }
180
181 static void ivshmem_io_write(void *opaque, hwaddr addr,
182                              uint64_t val, unsigned size)
183 {
184     IVShmemState *s = opaque;
185
186     uint16_t dest = val >> 16;
187     uint16_t vector = val & 0xff;
188
189     addr &= 0xfc;
190
191     IVSHMEM_DPRINTF("writing to addr " TARGET_FMT_plx "\n", addr);
192     switch (addr)
193     {
194         case INTRMASK:
195             ivshmem_IntrMask_write(s, val);
196             break;
197
198         case INTRSTATUS:
199             ivshmem_IntrStatus_write(s, val);
200             break;
201
202         case DOORBELL:
203             /* check that dest VM ID is reasonable */
204             if (dest > s->max_peer) {
205                 IVSHMEM_DPRINTF("Invalid destination VM ID (%d)\n", dest);
206                 break;
207             }
208
209             /* check doorbell range */
210             if (vector < s->peers[dest].nb_eventfds) {
211                 IVSHMEM_DPRINTF("Notifying VM %d on vector %d\n", dest, vector);
212                 event_notifier_set(&s->peers[dest].eventfds[vector]);
213             }
214             break;
215         default:
216             IVSHMEM_DPRINTF("Invalid VM Doorbell VM %d\n", dest);
217     }
218 }
219
220 static uint64_t ivshmem_io_read(void *opaque, hwaddr addr,
221                                 unsigned size)
222 {
223
224     IVShmemState *s = opaque;
225     uint32_t ret;
226
227     switch (addr)
228     {
229         case INTRMASK:
230             ret = ivshmem_IntrMask_read(s);
231             break;
232
233         case INTRSTATUS:
234             ret = ivshmem_IntrStatus_read(s);
235             break;
236
237         case IVPOSITION:
238             /* return my VM ID if the memory is mapped */
239             if (s->shm_fd > 0) {
240                 ret = s->vm_id;
241             } else {
242                 ret = -1;
243             }
244             break;
245
246         default:
247             IVSHMEM_DPRINTF("why are we reading " TARGET_FMT_plx "\n", addr);
248             ret = 0;
249     }
250
251     return ret;
252 }
253
254 static const MemoryRegionOps ivshmem_mmio_ops = {
255     .read = ivshmem_io_read,
256     .write = ivshmem_io_write,
257     .endianness = DEVICE_NATIVE_ENDIAN,
258     .impl = {
259         .min_access_size = 4,
260         .max_access_size = 4,
261     },
262 };
263
264 static void ivshmem_receive(void *opaque, const uint8_t *buf, int size)
265 {
266     IVShmemState *s = opaque;
267
268     ivshmem_IntrStatus_write(s, *buf);
269
270     IVSHMEM_DPRINTF("ivshmem_receive 0x%02x\n", *buf);
271 }
272
273 static int ivshmem_can_receive(void * opaque)
274 {
275     return 8;
276 }
277
278 static void ivshmem_event(void *opaque, int event)
279 {
280     IVSHMEM_DPRINTF("ivshmem_event %d\n", event);
281 }
282
283 static void fake_irqfd(void *opaque, const uint8_t *buf, int size) {
284
285     EventfdEntry *entry = opaque;
286     PCIDevice *pdev = entry->pdev;
287
288     IVSHMEM_DPRINTF("interrupt on vector %p %d\n", pdev, entry->vector);
289     msix_notify(pdev, entry->vector);
290 }
291
292 static CharDriverState* create_eventfd_chr_device(void * opaque, EventNotifier *n,
293                                                   int vector)
294 {
295     /* create a event character device based on the passed eventfd */
296     IVShmemState *s = opaque;
297     CharDriverState * chr;
298     int eventfd = event_notifier_get_fd(n);
299
300     chr = qemu_chr_open_eventfd(eventfd);
301
302     if (chr == NULL) {
303         fprintf(stderr, "creating eventfd for eventfd %d failed\n", eventfd);
304         exit(-1);
305     }
306     qemu_chr_fe_claim_no_fail(chr);
307
308     /* if MSI is supported we need multiple interrupts */
309     if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
310         s->eventfd_table[vector].pdev = PCI_DEVICE(s);
311         s->eventfd_table[vector].vector = vector;
312
313         qemu_chr_add_handlers(chr, ivshmem_can_receive, fake_irqfd,
314                       ivshmem_event, &s->eventfd_table[vector]);
315     } else {
316         qemu_chr_add_handlers(chr, ivshmem_can_receive, ivshmem_receive,
317                       ivshmem_event, s);
318     }
319
320     return chr;
321
322 }
323
324 static int check_shm_size(IVShmemState *s, int fd) {
325     /* check that the guest isn't going to try and map more memory than the
326      * the object has allocated return -1 to indicate error */
327
328     struct stat buf;
329
330     if (fstat(fd, &buf) < 0) {
331         fprintf(stderr, "ivshmem: exiting: fstat on fd %d failed: %s\n",
332                 fd, strerror(errno));
333         return -1;
334     }
335
336     if (s->ivshmem_size > buf.st_size) {
337         fprintf(stderr,
338                 "IVSHMEM ERROR: Requested memory size greater"
339                 " than shared object size (%" PRIu64 " > %" PRIu64")\n",
340                 s->ivshmem_size, (uint64_t)buf.st_size);
341         return -1;
342     } else {
343         return 0;
344     }
345 }
346
347 /* create the shared memory BAR when we are not using the server, so we can
348  * create the BAR and map the memory immediately */
349 static void create_shared_memory_BAR(IVShmemState *s, int fd) {
350
351     void * ptr;
352
353     s->shm_fd = fd;
354
355     ptr = mmap(0, s->ivshmem_size, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0);
356
357     memory_region_init_ram_ptr(&s->ivshmem, OBJECT(s), "ivshmem.bar2",
358                                s->ivshmem_size, ptr);
359     vmstate_register_ram(&s->ivshmem, DEVICE(s));
360     memory_region_add_subregion(&s->bar, 0, &s->ivshmem);
361
362     /* region for shared memory */
363     pci_register_bar(PCI_DEVICE(s), 2, s->ivshmem_attr, &s->bar);
364 }
365
366 static void ivshmem_add_eventfd(IVShmemState *s, int posn, int i)
367 {
368     memory_region_add_eventfd(&s->ivshmem_mmio,
369                               DOORBELL,
370                               4,
371                               true,
372                               (posn << 16) | i,
373                               &s->peers[posn].eventfds[i]);
374 }
375
376 static void ivshmem_del_eventfd(IVShmemState *s, int posn, int i)
377 {
378     memory_region_del_eventfd(&s->ivshmem_mmio,
379                               DOORBELL,
380                               4,
381                               true,
382                               (posn << 16) | i,
383                               &s->peers[posn].eventfds[i]);
384 }
385
386 static void close_guest_eventfds(IVShmemState *s, int posn)
387 {
388     int i, guest_curr_max;
389
390     if (!ivshmem_has_feature(s, IVSHMEM_IOEVENTFD)) {
391         return;
392     }
393     if (posn < 0 || posn >= s->nb_peers) {
394         return;
395     }
396
397     guest_curr_max = s->peers[posn].nb_eventfds;
398
399     memory_region_transaction_begin();
400     for (i = 0; i < guest_curr_max; i++) {
401         ivshmem_del_eventfd(s, posn, i);
402     }
403     memory_region_transaction_commit();
404     for (i = 0; i < guest_curr_max; i++) {
405         event_notifier_cleanup(&s->peers[posn].eventfds[i]);
406     }
407
408     g_free(s->peers[posn].eventfds);
409     s->peers[posn].nb_eventfds = 0;
410 }
411
412 /* this function increase the dynamic storage need to store data about other
413  * guests */
414 static int increase_dynamic_storage(IVShmemState *s, int new_min_size)
415 {
416
417     int j, old_nb_alloc;
418
419     /* check for integer overflow */
420     if (new_min_size >= INT_MAX / sizeof(Peer) - 1 || new_min_size <= 0) {
421         return -1;
422     }
423
424     old_nb_alloc = s->nb_peers;
425
426     if (new_min_size >= s->nb_peers) {
427         /* +1 because #new_min_size is used as last array index */
428         s->nb_peers = new_min_size + 1;
429     } else {
430         return 0;
431     }
432
433     IVSHMEM_DPRINTF("bumping storage to %d guests\n", s->nb_peers);
434     s->peers = g_realloc(s->peers, s->nb_peers * sizeof(Peer));
435
436     /* zero out new pointers */
437     for (j = old_nb_alloc; j < s->nb_peers; j++) {
438         s->peers[j].eventfds = NULL;
439         s->peers[j].nb_eventfds = 0;
440     }
441
442     return 0;
443 }
444
445 static void ivshmem_read(void *opaque, const uint8_t *buf, int size)
446 {
447     IVShmemState *s = opaque;
448     int incoming_fd, tmp_fd;
449     int guest_max_eventfd;
450     long incoming_posn;
451
452     if (fifo8_is_empty(&s->incoming_fifo) && size == sizeof(incoming_posn)) {
453         memcpy(&incoming_posn, buf, size);
454     } else {
455         const uint8_t *p;
456         uint32_t num;
457
458         IVSHMEM_DPRINTF("short read of %d bytes\n", size);
459         num = MAX(size, sizeof(long) - fifo8_num_used(&s->incoming_fifo));
460         fifo8_push_all(&s->incoming_fifo, buf, num);
461         if (fifo8_num_used(&s->incoming_fifo) < sizeof(incoming_posn)) {
462             return;
463         }
464         size -= num;
465         buf += num;
466         p = fifo8_pop_buf(&s->incoming_fifo, sizeof(incoming_posn), &num);
467         g_assert(num == sizeof(incoming_posn));
468         memcpy(&incoming_posn, p, sizeof(incoming_posn));
469         if (size > 0) {
470             fifo8_push_all(&s->incoming_fifo, buf, size);
471         }
472     }
473
474     if (incoming_posn < -1) {
475         IVSHMEM_DPRINTF("invalid incoming_posn %ld\n", incoming_posn);
476         return;
477     }
478
479     /* pick off s->server_chr->msgfd and store it, posn should accompany msg */
480     tmp_fd = qemu_chr_fe_get_msgfd(s->server_chr);
481     IVSHMEM_DPRINTF("posn is %ld, fd is %d\n", incoming_posn, tmp_fd);
482
483     /* make sure we have enough space for this guest */
484     if (incoming_posn >= s->nb_peers) {
485         if (increase_dynamic_storage(s, incoming_posn) < 0) {
486             error_report("increase_dynamic_storage() failed");
487             if (tmp_fd != -1) {
488                 close(tmp_fd);
489             }
490             return;
491         }
492     }
493
494     if (tmp_fd == -1) {
495         /* if posn is positive and unseen before then this is our posn*/
496         if ((incoming_posn >= 0) &&
497                             (s->peers[incoming_posn].eventfds == NULL)) {
498             /* receive our posn */
499             s->vm_id = incoming_posn;
500             return;
501         } else {
502             /* otherwise an fd == -1 means an existing guest has gone away */
503             IVSHMEM_DPRINTF("posn %ld has gone away\n", incoming_posn);
504             close_guest_eventfds(s, incoming_posn);
505             return;
506         }
507     }
508
509     /* because of the implementation of get_msgfd, we need a dup */
510     incoming_fd = dup(tmp_fd);
511
512     if (incoming_fd == -1) {
513         fprintf(stderr, "could not allocate file descriptor %s\n",
514                                                             strerror(errno));
515         close(tmp_fd);
516         return;
517     }
518
519     /* if the position is -1, then it's shared memory region fd */
520     if (incoming_posn == -1) {
521
522         void * map_ptr;
523
524         s->max_peer = 0;
525
526         if (check_shm_size(s, incoming_fd) == -1) {
527             exit(-1);
528         }
529
530         /* mmap the region and map into the BAR2 */
531         map_ptr = mmap(0, s->ivshmem_size, PROT_READ|PROT_WRITE, MAP_SHARED,
532                                                             incoming_fd, 0);
533         memory_region_init_ram_ptr(&s->ivshmem, OBJECT(s),
534                                    "ivshmem.bar2", s->ivshmem_size, map_ptr);
535         vmstate_register_ram(&s->ivshmem, DEVICE(s));
536
537         IVSHMEM_DPRINTF("guest h/w addr = %p, size = %" PRIu64 "\n",
538                          map_ptr, s->ivshmem_size);
539
540         memory_region_add_subregion(&s->bar, 0, &s->ivshmem);
541
542         /* only store the fd if it is successfully mapped */
543         s->shm_fd = incoming_fd;
544
545         return;
546     }
547
548     /* each guest has an array of eventfds, and we keep track of how many
549      * guests for each VM */
550     guest_max_eventfd = s->peers[incoming_posn].nb_eventfds;
551
552     if (guest_max_eventfd == 0) {
553         /* one eventfd per MSI vector */
554         s->peers[incoming_posn].eventfds = g_new(EventNotifier, s->vectors);
555     }
556
557     /* this is an eventfd for a particular guest VM */
558     IVSHMEM_DPRINTF("eventfds[%ld][%d] = %d\n", incoming_posn,
559                                             guest_max_eventfd, incoming_fd);
560     event_notifier_init_fd(&s->peers[incoming_posn].eventfds[guest_max_eventfd],
561                            incoming_fd);
562
563     /* increment count for particular guest */
564     s->peers[incoming_posn].nb_eventfds++;
565
566     /* keep track of the maximum VM ID */
567     if (incoming_posn > s->max_peer) {
568         s->max_peer = incoming_posn;
569     }
570
571     if (incoming_posn == s->vm_id) {
572         s->eventfd_chr[guest_max_eventfd] = create_eventfd_chr_device(s,
573                    &s->peers[s->vm_id].eventfds[guest_max_eventfd],
574                    guest_max_eventfd);
575     }
576
577     if (ivshmem_has_feature(s, IVSHMEM_IOEVENTFD)) {
578         ivshmem_add_eventfd(s, incoming_posn, guest_max_eventfd);
579     }
580 }
581
582 /* Select the MSI-X vectors used by device.
583  * ivshmem maps events to vectors statically, so
584  * we just enable all vectors on init and after reset. */
585 static void ivshmem_use_msix(IVShmemState * s)
586 {
587     PCIDevice *d = PCI_DEVICE(s);
588     int i;
589
590     if (!msix_present(d)) {
591         return;
592     }
593
594     for (i = 0; i < s->vectors; i++) {
595         msix_vector_use(d, i);
596     }
597 }
598
599 static void ivshmem_reset(DeviceState *d)
600 {
601     IVShmemState *s = IVSHMEM(d);
602
603     s->intrstatus = 0;
604     ivshmem_use_msix(s);
605 }
606
607 static uint64_t ivshmem_get_size(IVShmemState * s) {
608
609     uint64_t value;
610     char *ptr;
611
612     value = strtoull(s->sizearg, &ptr, 10);
613     switch (*ptr) {
614         case 0: case 'M': case 'm':
615             value <<= 20;
616             break;
617         case 'G': case 'g':
618             value <<= 30;
619             break;
620         default:
621             fprintf(stderr, "qemu: invalid ram size: %s\n", s->sizearg);
622             exit(1);
623     }
624
625     /* BARs must be a power of 2 */
626     if (!is_power_of_two(value)) {
627         fprintf(stderr, "ivshmem: size must be power of 2\n");
628         exit(1);
629     }
630
631     return value;
632 }
633
634 static void ivshmem_setup_msi(IVShmemState * s)
635 {
636     if (msix_init_exclusive_bar(PCI_DEVICE(s), s->vectors, 1)) {
637         IVSHMEM_DPRINTF("msix initialization failed\n");
638         exit(1);
639     }
640
641     IVSHMEM_DPRINTF("msix initialized (%d vectors)\n", s->vectors);
642
643     /* allocate QEMU char devices for receiving interrupts */
644     s->eventfd_table = g_malloc0(s->vectors * sizeof(EventfdEntry));
645
646     ivshmem_use_msix(s);
647 }
648
649 static void ivshmem_save(QEMUFile* f, void *opaque)
650 {
651     IVShmemState *proxy = opaque;
652     PCIDevice *pci_dev = PCI_DEVICE(proxy);
653
654     IVSHMEM_DPRINTF("ivshmem_save\n");
655     pci_device_save(pci_dev, f);
656
657     if (ivshmem_has_feature(proxy, IVSHMEM_MSI)) {
658         msix_save(pci_dev, f);
659     } else {
660         qemu_put_be32(f, proxy->intrstatus);
661         qemu_put_be32(f, proxy->intrmask);
662     }
663
664 }
665
666 static int ivshmem_load(QEMUFile* f, void *opaque, int version_id)
667 {
668     IVSHMEM_DPRINTF("ivshmem_load\n");
669
670     IVShmemState *proxy = opaque;
671     PCIDevice *pci_dev = PCI_DEVICE(proxy);
672     int ret;
673
674     if (version_id > 0) {
675         return -EINVAL;
676     }
677
678     if (proxy->role_val == IVSHMEM_PEER) {
679         fprintf(stderr, "ivshmem: 'peer' devices are not migratable\n");
680         return -EINVAL;
681     }
682
683     ret = pci_device_load(pci_dev, f);
684     if (ret) {
685         return ret;
686     }
687
688     if (ivshmem_has_feature(proxy, IVSHMEM_MSI)) {
689         msix_load(pci_dev, f);
690         ivshmem_use_msix(proxy);
691     } else {
692         proxy->intrstatus = qemu_get_be32(f);
693         proxy->intrmask = qemu_get_be32(f);
694     }
695
696     return 0;
697 }
698
699 static void ivshmem_write_config(PCIDevice *pci_dev, uint32_t address,
700                                  uint32_t val, int len)
701 {
702     pci_default_write_config(pci_dev, address, val, len);
703     msix_write_config(pci_dev, address, val, len);
704 }
705
706 static int pci_ivshmem_init(PCIDevice *dev)
707 {
708     IVShmemState *s = IVSHMEM(dev);
709     uint8_t *pci_conf;
710
711     if (s->sizearg == NULL)
712         s->ivshmem_size = 4 << 20; /* 4 MB default */
713     else {
714         s->ivshmem_size = ivshmem_get_size(s);
715     }
716
717     fifo8_create(&s->incoming_fifo, sizeof(long));
718
719     register_savevm(DEVICE(dev), "ivshmem", 0, 0, ivshmem_save, ivshmem_load,
720                                                                         dev);
721
722     /* IRQFD requires MSI */
723     if (ivshmem_has_feature(s, IVSHMEM_IOEVENTFD) &&
724         !ivshmem_has_feature(s, IVSHMEM_MSI)) {
725         fprintf(stderr, "ivshmem: ioeventfd/irqfd requires MSI\n");
726         exit(1);
727     }
728
729     /* check that role is reasonable */
730     if (s->role) {
731         if (strncmp(s->role, "peer", 5) == 0) {
732             s->role_val = IVSHMEM_PEER;
733         } else if (strncmp(s->role, "master", 7) == 0) {
734             s->role_val = IVSHMEM_MASTER;
735         } else {
736             fprintf(stderr, "ivshmem: 'role' must be 'peer' or 'master'\n");
737             exit(1);
738         }
739     } else {
740         s->role_val = IVSHMEM_MASTER; /* default */
741     }
742
743     if (s->role_val == IVSHMEM_PEER) {
744         error_setg(&s->migration_blocker,
745                    "Migration is disabled when using feature 'peer mode' in device 'ivshmem'");
746         migrate_add_blocker(s->migration_blocker);
747     }
748
749     pci_conf = dev->config;
750     pci_conf[PCI_COMMAND] = PCI_COMMAND_IO | PCI_COMMAND_MEMORY;
751
752     pci_config_set_interrupt_pin(pci_conf, 1);
753
754     s->shm_fd = 0;
755
756     memory_region_init_io(&s->ivshmem_mmio, OBJECT(s), &ivshmem_mmio_ops, s,
757                           "ivshmem-mmio", IVSHMEM_REG_BAR_SIZE);
758
759     /* region for registers*/
760     pci_register_bar(dev, 0, PCI_BASE_ADDRESS_SPACE_MEMORY,
761                      &s->ivshmem_mmio);
762
763     memory_region_init(&s->bar, OBJECT(s), "ivshmem-bar2-container", s->ivshmem_size);
764     s->ivshmem_attr = PCI_BASE_ADDRESS_SPACE_MEMORY |
765         PCI_BASE_ADDRESS_MEM_PREFETCH;
766     if (s->ivshmem_64bit) {
767         s->ivshmem_attr |= PCI_BASE_ADDRESS_MEM_TYPE_64;
768     }
769
770     if ((s->server_chr != NULL) &&
771                         (strncmp(s->server_chr->filename, "unix:", 5) == 0)) {
772         /* if we get a UNIX socket as the parameter we will talk
773          * to the ivshmem server to receive the memory region */
774
775         if (s->shmobj != NULL) {
776             fprintf(stderr, "WARNING: do not specify both 'chardev' "
777                                                 "and 'shm' with ivshmem\n");
778         }
779
780         IVSHMEM_DPRINTF("using shared memory server (socket = %s)\n",
781                                                     s->server_chr->filename);
782
783         if (ivshmem_has_feature(s, IVSHMEM_MSI)) {
784             ivshmem_setup_msi(s);
785         }
786
787         /* we allocate enough space for 16 guests and grow as needed */
788         s->nb_peers = 16;
789         s->vm_id = -1;
790
791         /* allocate/initialize space for interrupt handling */
792         s->peers = g_malloc0(s->nb_peers * sizeof(Peer));
793
794         pci_register_bar(dev, 2, s->ivshmem_attr, &s->bar);
795
796         s->eventfd_chr = g_malloc0(s->vectors * sizeof(CharDriverState *));
797
798         qemu_chr_add_handlers(s->server_chr, ivshmem_can_receive, ivshmem_read,
799                      ivshmem_event, s);
800     } else {
801         /* just map the file immediately, we're not using a server */
802         int fd;
803
804         if (s->shmobj == NULL) {
805             fprintf(stderr, "Must specify 'chardev' or 'shm' to ivshmem\n");
806             exit(1);
807         }
808
809         IVSHMEM_DPRINTF("using shm_open (shm object = %s)\n", s->shmobj);
810
811         /* try opening with O_EXCL and if it succeeds zero the memory
812          * by truncating to 0 */
813         if ((fd = shm_open(s->shmobj, O_CREAT|O_RDWR|O_EXCL,
814                         S_IRWXU|S_IRWXG|S_IRWXO)) > 0) {
815            /* truncate file to length PCI device's memory */
816             if (ftruncate(fd, s->ivshmem_size) != 0) {
817                 fprintf(stderr, "ivshmem: could not truncate shared file\n");
818             }
819
820         } else if ((fd = shm_open(s->shmobj, O_CREAT|O_RDWR,
821                         S_IRWXU|S_IRWXG|S_IRWXO)) < 0) {
822             fprintf(stderr, "ivshmem: could not open shared file\n");
823             exit(-1);
824
825         }
826
827         if (check_shm_size(s, fd) == -1) {
828             exit(-1);
829         }
830
831         create_shared_memory_BAR(s, fd);
832
833     }
834
835     dev->config_write = ivshmem_write_config;
836
837     return 0;
838 }
839
840 static void pci_ivshmem_uninit(PCIDevice *dev)
841 {
842     IVShmemState *s = IVSHMEM(dev);
843
844     if (s->migration_blocker) {
845         migrate_del_blocker(s->migration_blocker);
846         error_free(s->migration_blocker);
847     }
848
849     memory_region_del_subregion(&s->bar, &s->ivshmem);
850     vmstate_unregister_ram(&s->ivshmem, DEVICE(dev));
851     unregister_savevm(DEVICE(dev), "ivshmem", s);
852     fifo8_destroy(&s->incoming_fifo);
853 }
854
855 static Property ivshmem_properties[] = {
856     DEFINE_PROP_CHR("chardev", IVShmemState, server_chr),
857     DEFINE_PROP_STRING("size", IVShmemState, sizearg),
858     DEFINE_PROP_UINT32("vectors", IVShmemState, vectors, 1),
859     DEFINE_PROP_BIT("ioeventfd", IVShmemState, features, IVSHMEM_IOEVENTFD, false),
860     DEFINE_PROP_BIT("msi", IVShmemState, features, IVSHMEM_MSI, true),
861     DEFINE_PROP_STRING("shm", IVShmemState, shmobj),
862     DEFINE_PROP_STRING("role", IVShmemState, role),
863     DEFINE_PROP_UINT32("use64", IVShmemState, ivshmem_64bit, 1),
864     DEFINE_PROP_END_OF_LIST(),
865 };
866
867 static void ivshmem_class_init(ObjectClass *klass, void *data)
868 {
869     DeviceClass *dc = DEVICE_CLASS(klass);
870     PCIDeviceClass *k = PCI_DEVICE_CLASS(klass);
871
872     k->init = pci_ivshmem_init;
873     k->exit = pci_ivshmem_uninit;
874     k->vendor_id = PCI_VENDOR_ID_IVSHMEM;
875     k->device_id = PCI_DEVICE_ID_IVSHMEM;
876     k->class_id = PCI_CLASS_MEMORY_RAM;
877     dc->reset = ivshmem_reset;
878     dc->props = ivshmem_properties;
879     set_bit(DEVICE_CATEGORY_MISC, dc->categories);
880 }
881
882 static const TypeInfo ivshmem_info = {
883     .name          = TYPE_IVSHMEM,
884     .parent        = TYPE_PCI_DEVICE,
885     .instance_size = sizeof(IVShmemState),
886     .class_init    = ivshmem_class_init,
887 };
888
889 static void ivshmem_register_types(void)
890 {
891     type_register_static(&ivshmem_info);
892 }
893
894 type_init(ivshmem_register_types)
This page took 0.078184 seconds and 4 git commands to generate.