Statistics
| Branch: | Revision:

root / hw / vfio_pci.c @ a1bc20df

History | View | Annotate | Download (57.3 kB)

1 65501a74 Alex Williamson
/*
2 65501a74 Alex Williamson
 * vfio based device assignment support
3 65501a74 Alex Williamson
 *
4 65501a74 Alex Williamson
 * Copyright Red Hat, Inc. 2012
5 65501a74 Alex Williamson
 *
6 65501a74 Alex Williamson
 * Authors:
7 65501a74 Alex Williamson
 *  Alex Williamson <alex.williamson@redhat.com>
8 65501a74 Alex Williamson
 *
9 65501a74 Alex Williamson
 * This work is licensed under the terms of the GNU GPL, version 2.  See
10 65501a74 Alex Williamson
 * the COPYING file in the top-level directory.
11 65501a74 Alex Williamson
 *
12 65501a74 Alex Williamson
 * Based on qemu-kvm device-assignment:
13 65501a74 Alex Williamson
 *  Adapted for KVM by Qumranet.
14 65501a74 Alex Williamson
 *  Copyright (c) 2007, Neocleus, Alex Novik (alex@neocleus.com)
15 65501a74 Alex Williamson
 *  Copyright (c) 2007, Neocleus, Guy Zana (guy@neocleus.com)
16 65501a74 Alex Williamson
 *  Copyright (C) 2008, Qumranet, Amit Shah (amit.shah@qumranet.com)
17 65501a74 Alex Williamson
 *  Copyright (C) 2008, Red Hat, Amit Shah (amit.shah@redhat.com)
18 65501a74 Alex Williamson
 *  Copyright (C) 2008, IBM, Muli Ben-Yehuda (muli@il.ibm.com)
19 65501a74 Alex Williamson
 */
20 65501a74 Alex Williamson
21 65501a74 Alex Williamson
#include <dirent.h>
22 65501a74 Alex Williamson
#include <unistd.h>
23 65501a74 Alex Williamson
#include <sys/ioctl.h>
24 65501a74 Alex Williamson
#include <sys/mman.h>
25 65501a74 Alex Williamson
#include <sys/stat.h>
26 65501a74 Alex Williamson
#include <sys/types.h>
27 65501a74 Alex Williamson
#include <linux/vfio.h>
28 65501a74 Alex Williamson
29 65501a74 Alex Williamson
#include "config.h"
30 65501a74 Alex Williamson
#include "event_notifier.h"
31 65501a74 Alex Williamson
#include "exec-memory.h"
32 65501a74 Alex Williamson
#include "kvm.h"
33 65501a74 Alex Williamson
#include "memory.h"
34 65501a74 Alex Williamson
#include "msi.h"
35 65501a74 Alex Williamson
#include "msix.h"
36 5c97e5eb Alex Williamson
#include "pci.h"
37 5c97e5eb Alex Williamson
#include "qemu-common.h"
38 65501a74 Alex Williamson
#include "qemu-error.h"
39 5c97e5eb Alex Williamson
#include "qemu-queue.h"
40 65501a74 Alex Williamson
#include "range.h"
41 65501a74 Alex Williamson
42 65501a74 Alex Williamson
/* #define DEBUG_VFIO */
43 65501a74 Alex Williamson
#ifdef DEBUG_VFIO
44 65501a74 Alex Williamson
#define DPRINTF(fmt, ...) \
45 65501a74 Alex Williamson
    do { fprintf(stderr, "vfio: " fmt, ## __VA_ARGS__); } while (0)
46 65501a74 Alex Williamson
#else
47 65501a74 Alex Williamson
#define DPRINTF(fmt, ...) \
48 65501a74 Alex Williamson
    do { } while (0)
49 65501a74 Alex Williamson
#endif
50 65501a74 Alex Williamson
51 5c97e5eb Alex Williamson
typedef struct VFIOBAR {
52 5c97e5eb Alex Williamson
    off_t fd_offset; /* offset of BAR within device fd */
53 5c97e5eb Alex Williamson
    int fd; /* device fd, allows us to pass VFIOBAR as opaque data */
54 5c97e5eb Alex Williamson
    MemoryRegion mem; /* slow, read/write access */
55 5c97e5eb Alex Williamson
    MemoryRegion mmap_mem; /* direct mapped access */
56 5c97e5eb Alex Williamson
    void *mmap;
57 5c97e5eb Alex Williamson
    size_t size;
58 5c97e5eb Alex Williamson
    uint32_t flags; /* VFIO region flags (rd/wr/mmap) */
59 5c97e5eb Alex Williamson
    uint8_t nr; /* cache the BAR number for debug */
60 5c97e5eb Alex Williamson
} VFIOBAR;
61 5c97e5eb Alex Williamson
62 5c97e5eb Alex Williamson
typedef struct VFIOINTx {
63 5c97e5eb Alex Williamson
    bool pending; /* interrupt pending */
64 5c97e5eb Alex Williamson
    bool kvm_accel; /* set when QEMU bypass through KVM enabled */
65 5c97e5eb Alex Williamson
    uint8_t pin; /* which pin to pull for qemu_set_irq */
66 5c97e5eb Alex Williamson
    EventNotifier interrupt; /* eventfd triggered on interrupt */
67 5c97e5eb Alex Williamson
    EventNotifier unmask; /* eventfd for unmask on QEMU bypass */
68 5c97e5eb Alex Williamson
    PCIINTxRoute route; /* routing info for QEMU bypass */
69 5c97e5eb Alex Williamson
    uint32_t mmap_timeout; /* delay to re-enable mmaps after interrupt */
70 5c97e5eb Alex Williamson
    QEMUTimer *mmap_timer; /* enable mmaps after periods w/o interrupts */
71 5c97e5eb Alex Williamson
} VFIOINTx;
72 5c97e5eb Alex Williamson
73 5c97e5eb Alex Williamson
struct VFIODevice;
74 5c97e5eb Alex Williamson
75 5c97e5eb Alex Williamson
typedef struct VFIOMSIVector {
76 5c97e5eb Alex Williamson
    EventNotifier interrupt; /* eventfd triggered on interrupt */
77 5c97e5eb Alex Williamson
    struct VFIODevice *vdev; /* back pointer to device */
78 5c97e5eb Alex Williamson
    int virq; /* KVM irqchip route for QEMU bypass */
79 5c97e5eb Alex Williamson
    bool use;
80 5c97e5eb Alex Williamson
} VFIOMSIVector;
81 5c97e5eb Alex Williamson
82 5c97e5eb Alex Williamson
enum {
83 5c97e5eb Alex Williamson
    VFIO_INT_NONE = 0,
84 5c97e5eb Alex Williamson
    VFIO_INT_INTx = 1,
85 5c97e5eb Alex Williamson
    VFIO_INT_MSI  = 2,
86 5c97e5eb Alex Williamson
    VFIO_INT_MSIX = 3,
87 5c97e5eb Alex Williamson
};
88 5c97e5eb Alex Williamson
89 5c97e5eb Alex Williamson
struct VFIOGroup;
90 5c97e5eb Alex Williamson
91 5c97e5eb Alex Williamson
typedef struct VFIOContainer {
92 5c97e5eb Alex Williamson
    int fd; /* /dev/vfio/vfio, empowered by the attached groups */
93 5c97e5eb Alex Williamson
    struct {
94 5c97e5eb Alex Williamson
        /* enable abstraction to support various iommu backends */
95 5c97e5eb Alex Williamson
        union {
96 5c97e5eb Alex Williamson
            MemoryListener listener; /* Used by type1 iommu */
97 5c97e5eb Alex Williamson
        };
98 5c97e5eb Alex Williamson
        void (*release)(struct VFIOContainer *);
99 5c97e5eb Alex Williamson
    } iommu_data;
100 5c97e5eb Alex Williamson
    QLIST_HEAD(, VFIOGroup) group_list;
101 5c97e5eb Alex Williamson
    QLIST_ENTRY(VFIOContainer) next;
102 5c97e5eb Alex Williamson
} VFIOContainer;
103 5c97e5eb Alex Williamson
104 5c97e5eb Alex Williamson
/* Cache of MSI-X setup plus extra mmap and memory region for split BAR map */
105 5c97e5eb Alex Williamson
typedef struct VFIOMSIXInfo {
106 5c97e5eb Alex Williamson
    uint8_t table_bar;
107 5c97e5eb Alex Williamson
    uint8_t pba_bar;
108 5c97e5eb Alex Williamson
    uint16_t entries;
109 5c97e5eb Alex Williamson
    uint32_t table_offset;
110 5c97e5eb Alex Williamson
    uint32_t pba_offset;
111 5c97e5eb Alex Williamson
    MemoryRegion mmap_mem;
112 5c97e5eb Alex Williamson
    void *mmap;
113 5c97e5eb Alex Williamson
} VFIOMSIXInfo;
114 5c97e5eb Alex Williamson
115 5c97e5eb Alex Williamson
typedef struct VFIODevice {
116 5c97e5eb Alex Williamson
    PCIDevice pdev;
117 5c97e5eb Alex Williamson
    int fd;
118 5c97e5eb Alex Williamson
    VFIOINTx intx;
119 5c97e5eb Alex Williamson
    unsigned int config_size;
120 5c97e5eb Alex Williamson
    off_t config_offset; /* Offset of config space region within device fd */
121 5c97e5eb Alex Williamson
    unsigned int rom_size;
122 5c97e5eb Alex Williamson
    off_t rom_offset; /* Offset of ROM region within device fd */
123 5c97e5eb Alex Williamson
    int msi_cap_size;
124 5c97e5eb Alex Williamson
    VFIOMSIVector *msi_vectors;
125 5c97e5eb Alex Williamson
    VFIOMSIXInfo *msix;
126 5c97e5eb Alex Williamson
    int nr_vectors; /* Number of MSI/MSIX vectors currently in use */
127 5c97e5eb Alex Williamson
    int interrupt; /* Current interrupt type */
128 5c97e5eb Alex Williamson
    VFIOBAR bars[PCI_NUM_REGIONS - 1]; /* No ROM */
129 5c97e5eb Alex Williamson
    PCIHostDeviceAddress host;
130 5c97e5eb Alex Williamson
    QLIST_ENTRY(VFIODevice) next;
131 5c97e5eb Alex Williamson
    struct VFIOGroup *group;
132 5c97e5eb Alex Williamson
    bool reset_works;
133 5c97e5eb Alex Williamson
} VFIODevice;
134 5c97e5eb Alex Williamson
135 5c97e5eb Alex Williamson
typedef struct VFIOGroup {
136 5c97e5eb Alex Williamson
    int fd;
137 5c97e5eb Alex Williamson
    int groupid;
138 5c97e5eb Alex Williamson
    VFIOContainer *container;
139 5c97e5eb Alex Williamson
    QLIST_HEAD(, VFIODevice) device_list;
140 5c97e5eb Alex Williamson
    QLIST_ENTRY(VFIOGroup) next;
141 5c97e5eb Alex Williamson
    QLIST_ENTRY(VFIOGroup) container_next;
142 5c97e5eb Alex Williamson
} VFIOGroup;
143 5c97e5eb Alex Williamson
144 65501a74 Alex Williamson
#define MSIX_CAP_LENGTH 12
145 65501a74 Alex Williamson
146 65501a74 Alex Williamson
static QLIST_HEAD(, VFIOContainer)
147 65501a74 Alex Williamson
    container_list = QLIST_HEAD_INITIALIZER(container_list);
148 65501a74 Alex Williamson
149 65501a74 Alex Williamson
static QLIST_HEAD(, VFIOGroup)
150 65501a74 Alex Williamson
    group_list = QLIST_HEAD_INITIALIZER(group_list);
151 65501a74 Alex Williamson
152 65501a74 Alex Williamson
static void vfio_disable_interrupts(VFIODevice *vdev);
153 65501a74 Alex Williamson
static uint32_t vfio_pci_read_config(PCIDevice *pdev, uint32_t addr, int len);
154 65501a74 Alex Williamson
static void vfio_mmap_set_enabled(VFIODevice *vdev, bool enabled);
155 65501a74 Alex Williamson
156 65501a74 Alex Williamson
/*
157 65501a74 Alex Williamson
 * Common VFIO interrupt disable
158 65501a74 Alex Williamson
 */
159 65501a74 Alex Williamson
static void vfio_disable_irqindex(VFIODevice *vdev, int index)
160 65501a74 Alex Williamson
{
161 65501a74 Alex Williamson
    struct vfio_irq_set irq_set = {
162 65501a74 Alex Williamson
        .argsz = sizeof(irq_set),
163 65501a74 Alex Williamson
        .flags = VFIO_IRQ_SET_DATA_NONE | VFIO_IRQ_SET_ACTION_TRIGGER,
164 65501a74 Alex Williamson
        .index = index,
165 65501a74 Alex Williamson
        .start = 0,
166 65501a74 Alex Williamson
        .count = 0,
167 65501a74 Alex Williamson
    };
168 65501a74 Alex Williamson
169 65501a74 Alex Williamson
    ioctl(vdev->fd, VFIO_DEVICE_SET_IRQS, &irq_set);
170 65501a74 Alex Williamson
}
171 65501a74 Alex Williamson
172 65501a74 Alex Williamson
/*
173 65501a74 Alex Williamson
 * INTx
174 65501a74 Alex Williamson
 */
175 65501a74 Alex Williamson
static void vfio_unmask_intx(VFIODevice *vdev)
176 65501a74 Alex Williamson
{
177 65501a74 Alex Williamson
    struct vfio_irq_set irq_set = {
178 65501a74 Alex Williamson
        .argsz = sizeof(irq_set),
179 65501a74 Alex Williamson
        .flags = VFIO_IRQ_SET_DATA_NONE | VFIO_IRQ_SET_ACTION_UNMASK,
180 65501a74 Alex Williamson
        .index = VFIO_PCI_INTX_IRQ_INDEX,
181 65501a74 Alex Williamson
        .start = 0,
182 65501a74 Alex Williamson
        .count = 1,
183 65501a74 Alex Williamson
    };
184 65501a74 Alex Williamson
185 65501a74 Alex Williamson
    ioctl(vdev->fd, VFIO_DEVICE_SET_IRQS, &irq_set);
186 65501a74 Alex Williamson
}
187 65501a74 Alex Williamson
188 ea486926 Alex Williamson
/*
189 ea486926 Alex Williamson
 * Disabling BAR mmaping can be slow, but toggling it around INTx can
190 ea486926 Alex Williamson
 * also be a huge overhead.  We try to get the best of both worlds by
191 ea486926 Alex Williamson
 * waiting until an interrupt to disable mmaps (subsequent transitions
192 ea486926 Alex Williamson
 * to the same state are effectively no overhead).  If the interrupt has
193 ea486926 Alex Williamson
 * been serviced and the time gap is long enough, we re-enable mmaps for
194 ea486926 Alex Williamson
 * performance.  This works well for things like graphics cards, which
195 ea486926 Alex Williamson
 * may not use their interrupt at all and are penalized to an unusable
196 ea486926 Alex Williamson
 * level by read/write BAR traps.  Other devices, like NICs, have more
197 ea486926 Alex Williamson
 * regular interrupts and see much better latency by staying in non-mmap
198 ea486926 Alex Williamson
 * mode.  We therefore set the default mmap_timeout such that a ping
199 ea486926 Alex Williamson
 * is just enough to keep the mmap disabled.  Users can experiment with
200 ea486926 Alex Williamson
 * other options with the x-intx-mmap-timeout-ms parameter (a value of
201 ea486926 Alex Williamson
 * zero disables the timer).
202 ea486926 Alex Williamson
 */
203 ea486926 Alex Williamson
static void vfio_intx_mmap_enable(void *opaque)
204 ea486926 Alex Williamson
{
205 ea486926 Alex Williamson
    VFIODevice *vdev = opaque;
206 ea486926 Alex Williamson
207 ea486926 Alex Williamson
    if (vdev->intx.pending) {
208 ea486926 Alex Williamson
        qemu_mod_timer(vdev->intx.mmap_timer,
209 ea486926 Alex Williamson
                       qemu_get_clock_ms(vm_clock) + vdev->intx.mmap_timeout);
210 ea486926 Alex Williamson
        return;
211 ea486926 Alex Williamson
    }
212 ea486926 Alex Williamson
213 ea486926 Alex Williamson
    vfio_mmap_set_enabled(vdev, true);
214 ea486926 Alex Williamson
}
215 ea486926 Alex Williamson
216 65501a74 Alex Williamson
static void vfio_intx_interrupt(void *opaque)
217 65501a74 Alex Williamson
{
218 65501a74 Alex Williamson
    VFIODevice *vdev = opaque;
219 65501a74 Alex Williamson
220 65501a74 Alex Williamson
    if (!event_notifier_test_and_clear(&vdev->intx.interrupt)) {
221 65501a74 Alex Williamson
        return;
222 65501a74 Alex Williamson
    }
223 65501a74 Alex Williamson
224 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x) Pin %c\n", __func__, vdev->host.domain,
225 65501a74 Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function,
226 65501a74 Alex Williamson
            'A' + vdev->intx.pin);
227 65501a74 Alex Williamson
228 65501a74 Alex Williamson
    vdev->intx.pending = true;
229 65501a74 Alex Williamson
    qemu_set_irq(vdev->pdev.irq[vdev->intx.pin], 1);
230 ea486926 Alex Williamson
    vfio_mmap_set_enabled(vdev, false);
231 ea486926 Alex Williamson
    if (vdev->intx.mmap_timeout) {
232 ea486926 Alex Williamson
        qemu_mod_timer(vdev->intx.mmap_timer,
233 ea486926 Alex Williamson
                       qemu_get_clock_ms(vm_clock) + vdev->intx.mmap_timeout);
234 ea486926 Alex Williamson
    }
235 65501a74 Alex Williamson
}
236 65501a74 Alex Williamson
237 65501a74 Alex Williamson
static void vfio_eoi(VFIODevice *vdev)
238 65501a74 Alex Williamson
{
239 65501a74 Alex Williamson
    if (!vdev->intx.pending) {
240 65501a74 Alex Williamson
        return;
241 65501a74 Alex Williamson
    }
242 65501a74 Alex Williamson
243 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x) EOI\n", __func__, vdev->host.domain,
244 65501a74 Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
245 65501a74 Alex Williamson
246 65501a74 Alex Williamson
    vdev->intx.pending = false;
247 65501a74 Alex Williamson
    qemu_set_irq(vdev->pdev.irq[vdev->intx.pin], 0);
248 65501a74 Alex Williamson
    vfio_unmask_intx(vdev);
249 65501a74 Alex Williamson
}
250 65501a74 Alex Williamson
251 65501a74 Alex Williamson
static int vfio_enable_intx(VFIODevice *vdev)
252 65501a74 Alex Williamson
{
253 65501a74 Alex Williamson
    uint8_t pin = vfio_pci_read_config(&vdev->pdev, PCI_INTERRUPT_PIN, 1);
254 1a403133 Alex Williamson
    int ret, argsz;
255 1a403133 Alex Williamson
    struct vfio_irq_set *irq_set;
256 1a403133 Alex Williamson
    int32_t *pfd;
257 65501a74 Alex Williamson
258 ea486926 Alex Williamson
    if (!pin) {
259 65501a74 Alex Williamson
        return 0;
260 65501a74 Alex Williamson
    }
261 65501a74 Alex Williamson
262 65501a74 Alex Williamson
    vfio_disable_interrupts(vdev);
263 65501a74 Alex Williamson
264 65501a74 Alex Williamson
    vdev->intx.pin = pin - 1; /* Pin A (1) -> irq[0] */
265 65501a74 Alex Williamson
    ret = event_notifier_init(&vdev->intx.interrupt, 0);
266 65501a74 Alex Williamson
    if (ret) {
267 65501a74 Alex Williamson
        error_report("vfio: Error: event_notifier_init failed\n");
268 65501a74 Alex Williamson
        return ret;
269 65501a74 Alex Williamson
    }
270 65501a74 Alex Williamson
271 1a403133 Alex Williamson
    argsz = sizeof(*irq_set) + sizeof(*pfd);
272 1a403133 Alex Williamson
273 1a403133 Alex Williamson
    irq_set = g_malloc0(argsz);
274 1a403133 Alex Williamson
    irq_set->argsz = argsz;
275 1a403133 Alex Williamson
    irq_set->flags = VFIO_IRQ_SET_DATA_EVENTFD | VFIO_IRQ_SET_ACTION_TRIGGER;
276 1a403133 Alex Williamson
    irq_set->index = VFIO_PCI_INTX_IRQ_INDEX;
277 1a403133 Alex Williamson
    irq_set->start = 0;
278 1a403133 Alex Williamson
    irq_set->count = 1;
279 1a403133 Alex Williamson
    pfd = (int32_t *)&irq_set->data;
280 1a403133 Alex Williamson
281 1a403133 Alex Williamson
    *pfd = event_notifier_get_fd(&vdev->intx.interrupt);
282 1a403133 Alex Williamson
    qemu_set_fd_handler(*pfd, vfio_intx_interrupt, NULL, vdev);
283 65501a74 Alex Williamson
284 1a403133 Alex Williamson
    ret = ioctl(vdev->fd, VFIO_DEVICE_SET_IRQS, irq_set);
285 1a403133 Alex Williamson
    g_free(irq_set);
286 1a403133 Alex Williamson
    if (ret) {
287 65501a74 Alex Williamson
        error_report("vfio: Error: Failed to setup INTx fd: %m\n");
288 1a403133 Alex Williamson
        qemu_set_fd_handler(*pfd, NULL, NULL, vdev);
289 ce59af2d Alex Williamson
        event_notifier_cleanup(&vdev->intx.interrupt);
290 65501a74 Alex Williamson
        return -errno;
291 65501a74 Alex Williamson
    }
292 65501a74 Alex Williamson
293 65501a74 Alex Williamson
    vdev->interrupt = VFIO_INT_INTx;
294 65501a74 Alex Williamson
295 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x)\n", __func__, vdev->host.domain,
296 65501a74 Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
297 65501a74 Alex Williamson
298 65501a74 Alex Williamson
    return 0;
299 65501a74 Alex Williamson
}
300 65501a74 Alex Williamson
301 65501a74 Alex Williamson
static void vfio_disable_intx(VFIODevice *vdev)
302 65501a74 Alex Williamson
{
303 65501a74 Alex Williamson
    int fd;
304 65501a74 Alex Williamson
305 ea486926 Alex Williamson
    qemu_del_timer(vdev->intx.mmap_timer);
306 65501a74 Alex Williamson
    vfio_disable_irqindex(vdev, VFIO_PCI_INTX_IRQ_INDEX);
307 65501a74 Alex Williamson
    vdev->intx.pending = false;
308 65501a74 Alex Williamson
    qemu_set_irq(vdev->pdev.irq[vdev->intx.pin], 0);
309 65501a74 Alex Williamson
    vfio_mmap_set_enabled(vdev, true);
310 65501a74 Alex Williamson
311 65501a74 Alex Williamson
    fd = event_notifier_get_fd(&vdev->intx.interrupt);
312 65501a74 Alex Williamson
    qemu_set_fd_handler(fd, NULL, NULL, vdev);
313 65501a74 Alex Williamson
    event_notifier_cleanup(&vdev->intx.interrupt);
314 65501a74 Alex Williamson
315 65501a74 Alex Williamson
    vdev->interrupt = VFIO_INT_NONE;
316 65501a74 Alex Williamson
317 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x)\n", __func__, vdev->host.domain,
318 65501a74 Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
319 65501a74 Alex Williamson
}
320 65501a74 Alex Williamson
321 65501a74 Alex Williamson
/*
322 65501a74 Alex Williamson
 * MSI/X
323 65501a74 Alex Williamson
 */
324 65501a74 Alex Williamson
static void vfio_msi_interrupt(void *opaque)
325 65501a74 Alex Williamson
{
326 65501a74 Alex Williamson
    VFIOMSIVector *vector = opaque;
327 65501a74 Alex Williamson
    VFIODevice *vdev = vector->vdev;
328 65501a74 Alex Williamson
    int nr = vector - vdev->msi_vectors;
329 65501a74 Alex Williamson
330 65501a74 Alex Williamson
    if (!event_notifier_test_and_clear(&vector->interrupt)) {
331 65501a74 Alex Williamson
        return;
332 65501a74 Alex Williamson
    }
333 65501a74 Alex Williamson
334 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x) vector %d\n", __func__,
335 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
336 65501a74 Alex Williamson
            vdev->host.function, nr);
337 65501a74 Alex Williamson
338 65501a74 Alex Williamson
    if (vdev->interrupt == VFIO_INT_MSIX) {
339 65501a74 Alex Williamson
        msix_notify(&vdev->pdev, nr);
340 65501a74 Alex Williamson
    } else if (vdev->interrupt == VFIO_INT_MSI) {
341 65501a74 Alex Williamson
        msi_notify(&vdev->pdev, nr);
342 65501a74 Alex Williamson
    } else {
343 65501a74 Alex Williamson
        error_report("vfio: MSI interrupt receieved, but not enabled?\n");
344 65501a74 Alex Williamson
    }
345 65501a74 Alex Williamson
}
346 65501a74 Alex Williamson
347 65501a74 Alex Williamson
static int vfio_enable_vectors(VFIODevice *vdev, bool msix)
348 65501a74 Alex Williamson
{
349 65501a74 Alex Williamson
    struct vfio_irq_set *irq_set;
350 65501a74 Alex Williamson
    int ret = 0, i, argsz;
351 65501a74 Alex Williamson
    int32_t *fds;
352 65501a74 Alex Williamson
353 65501a74 Alex Williamson
    argsz = sizeof(*irq_set) + (vdev->nr_vectors * sizeof(*fds));
354 65501a74 Alex Williamson
355 65501a74 Alex Williamson
    irq_set = g_malloc0(argsz);
356 65501a74 Alex Williamson
    irq_set->argsz = argsz;
357 65501a74 Alex Williamson
    irq_set->flags = VFIO_IRQ_SET_DATA_EVENTFD | VFIO_IRQ_SET_ACTION_TRIGGER;
358 65501a74 Alex Williamson
    irq_set->index = msix ? VFIO_PCI_MSIX_IRQ_INDEX : VFIO_PCI_MSI_IRQ_INDEX;
359 65501a74 Alex Williamson
    irq_set->start = 0;
360 65501a74 Alex Williamson
    irq_set->count = vdev->nr_vectors;
361 65501a74 Alex Williamson
    fds = (int32_t *)&irq_set->data;
362 65501a74 Alex Williamson
363 65501a74 Alex Williamson
    for (i = 0; i < vdev->nr_vectors; i++) {
364 65501a74 Alex Williamson
        if (!vdev->msi_vectors[i].use) {
365 65501a74 Alex Williamson
            fds[i] = -1;
366 65501a74 Alex Williamson
            continue;
367 65501a74 Alex Williamson
        }
368 65501a74 Alex Williamson
369 65501a74 Alex Williamson
        fds[i] = event_notifier_get_fd(&vdev->msi_vectors[i].interrupt);
370 65501a74 Alex Williamson
    }
371 65501a74 Alex Williamson
372 65501a74 Alex Williamson
    ret = ioctl(vdev->fd, VFIO_DEVICE_SET_IRQS, irq_set);
373 65501a74 Alex Williamson
374 65501a74 Alex Williamson
    g_free(irq_set);
375 65501a74 Alex Williamson
376 65501a74 Alex Williamson
    return ret;
377 65501a74 Alex Williamson
}
378 65501a74 Alex Williamson
379 65501a74 Alex Williamson
static int vfio_msix_vector_use(PCIDevice *pdev,
380 65501a74 Alex Williamson
                                unsigned int nr, MSIMessage msg)
381 65501a74 Alex Williamson
{
382 65501a74 Alex Williamson
    VFIODevice *vdev = DO_UPCAST(VFIODevice, pdev, pdev);
383 65501a74 Alex Williamson
    VFIOMSIVector *vector;
384 65501a74 Alex Williamson
    int ret;
385 65501a74 Alex Williamson
386 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x) vector %d used\n", __func__,
387 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
388 65501a74 Alex Williamson
            vdev->host.function, nr);
389 65501a74 Alex Williamson
390 65501a74 Alex Williamson
    vector = &vdev->msi_vectors[nr];
391 65501a74 Alex Williamson
    vector->vdev = vdev;
392 65501a74 Alex Williamson
    vector->use = true;
393 65501a74 Alex Williamson
394 65501a74 Alex Williamson
    msix_vector_use(pdev, nr);
395 65501a74 Alex Williamson
396 65501a74 Alex Williamson
    if (event_notifier_init(&vector->interrupt, 0)) {
397 65501a74 Alex Williamson
        error_report("vfio: Error: event_notifier_init failed\n");
398 65501a74 Alex Williamson
    }
399 65501a74 Alex Williamson
400 65501a74 Alex Williamson
    /*
401 65501a74 Alex Williamson
     * Attempt to enable route through KVM irqchip,
402 65501a74 Alex Williamson
     * default to userspace handling if unavailable.
403 65501a74 Alex Williamson
     */
404 65501a74 Alex Williamson
    vector->virq = kvm_irqchip_add_msi_route(kvm_state, msg);
405 65501a74 Alex Williamson
    if (vector->virq < 0 ||
406 65501a74 Alex Williamson
        kvm_irqchip_add_irqfd_notifier(kvm_state, &vector->interrupt,
407 65501a74 Alex Williamson
                                       vector->virq) < 0) {
408 65501a74 Alex Williamson
        if (vector->virq >= 0) {
409 65501a74 Alex Williamson
            kvm_irqchip_release_virq(kvm_state, vector->virq);
410 65501a74 Alex Williamson
            vector->virq = -1;
411 65501a74 Alex Williamson
        }
412 65501a74 Alex Williamson
        qemu_set_fd_handler(event_notifier_get_fd(&vector->interrupt),
413 65501a74 Alex Williamson
                            vfio_msi_interrupt, NULL, vector);
414 65501a74 Alex Williamson
    }
415 65501a74 Alex Williamson
416 65501a74 Alex Williamson
    /*
417 65501a74 Alex Williamson
     * We don't want to have the host allocate all possible MSI vectors
418 65501a74 Alex Williamson
     * for a device if they're not in use, so we shutdown and incrementally
419 65501a74 Alex Williamson
     * increase them as needed.
420 65501a74 Alex Williamson
     */
421 65501a74 Alex Williamson
    if (vdev->nr_vectors < nr + 1) {
422 65501a74 Alex Williamson
        vfio_disable_irqindex(vdev, VFIO_PCI_MSIX_IRQ_INDEX);
423 65501a74 Alex Williamson
        vdev->nr_vectors = nr + 1;
424 65501a74 Alex Williamson
        ret = vfio_enable_vectors(vdev, true);
425 65501a74 Alex Williamson
        if (ret) {
426 65501a74 Alex Williamson
            error_report("vfio: failed to enable vectors, %d\n", ret);
427 65501a74 Alex Williamson
        }
428 65501a74 Alex Williamson
    } else {
429 1a403133 Alex Williamson
        int argsz;
430 1a403133 Alex Williamson
        struct vfio_irq_set *irq_set;
431 1a403133 Alex Williamson
        int32_t *pfd;
432 1a403133 Alex Williamson
433 1a403133 Alex Williamson
        argsz = sizeof(*irq_set) + sizeof(*pfd);
434 1a403133 Alex Williamson
435 1a403133 Alex Williamson
        irq_set = g_malloc0(argsz);
436 1a403133 Alex Williamson
        irq_set->argsz = argsz;
437 1a403133 Alex Williamson
        irq_set->flags = VFIO_IRQ_SET_DATA_EVENTFD |
438 1a403133 Alex Williamson
                         VFIO_IRQ_SET_ACTION_TRIGGER;
439 1a403133 Alex Williamson
        irq_set->index = VFIO_PCI_MSIX_IRQ_INDEX;
440 1a403133 Alex Williamson
        irq_set->start = nr;
441 1a403133 Alex Williamson
        irq_set->count = 1;
442 1a403133 Alex Williamson
        pfd = (int32_t *)&irq_set->data;
443 1a403133 Alex Williamson
444 1a403133 Alex Williamson
        *pfd = event_notifier_get_fd(&vector->interrupt);
445 1a403133 Alex Williamson
446 1a403133 Alex Williamson
        ret = ioctl(vdev->fd, VFIO_DEVICE_SET_IRQS, irq_set);
447 1a403133 Alex Williamson
        g_free(irq_set);
448 65501a74 Alex Williamson
        if (ret) {
449 65501a74 Alex Williamson
            error_report("vfio: failed to modify vector, %d\n", ret);
450 65501a74 Alex Williamson
        }
451 65501a74 Alex Williamson
    }
452 65501a74 Alex Williamson
453 65501a74 Alex Williamson
    return 0;
454 65501a74 Alex Williamson
}
455 65501a74 Alex Williamson
456 65501a74 Alex Williamson
static void vfio_msix_vector_release(PCIDevice *pdev, unsigned int nr)
457 65501a74 Alex Williamson
{
458 65501a74 Alex Williamson
    VFIODevice *vdev = DO_UPCAST(VFIODevice, pdev, pdev);
459 65501a74 Alex Williamson
    VFIOMSIVector *vector = &vdev->msi_vectors[nr];
460 1a403133 Alex Williamson
    int argsz;
461 1a403133 Alex Williamson
    struct vfio_irq_set *irq_set;
462 1a403133 Alex Williamson
    int32_t *pfd;
463 65501a74 Alex Williamson
464 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x) vector %d released\n", __func__,
465 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
466 65501a74 Alex Williamson
            vdev->host.function, nr);
467 65501a74 Alex Williamson
468 65501a74 Alex Williamson
    /*
469 65501a74 Alex Williamson
     * XXX What's the right thing to do here?  This turns off the interrupt
470 65501a74 Alex Williamson
     * completely, but do we really just want to switch the interrupt to
471 65501a74 Alex Williamson
     * bouncing through userspace and let msix.c drop it?  Not sure.
472 65501a74 Alex Williamson
     */
473 65501a74 Alex Williamson
    msix_vector_unuse(pdev, nr);
474 1a403133 Alex Williamson
475 1a403133 Alex Williamson
    argsz = sizeof(*irq_set) + sizeof(*pfd);
476 1a403133 Alex Williamson
477 1a403133 Alex Williamson
    irq_set = g_malloc0(argsz);
478 1a403133 Alex Williamson
    irq_set->argsz = argsz;
479 1a403133 Alex Williamson
    irq_set->flags = VFIO_IRQ_SET_DATA_EVENTFD |
480 1a403133 Alex Williamson
                     VFIO_IRQ_SET_ACTION_TRIGGER;
481 1a403133 Alex Williamson
    irq_set->index = VFIO_PCI_MSIX_IRQ_INDEX;
482 1a403133 Alex Williamson
    irq_set->start = nr;
483 1a403133 Alex Williamson
    irq_set->count = 1;
484 1a403133 Alex Williamson
    pfd = (int32_t *)&irq_set->data;
485 1a403133 Alex Williamson
486 1a403133 Alex Williamson
    *pfd = -1;
487 1a403133 Alex Williamson
488 1a403133 Alex Williamson
    ioctl(vdev->fd, VFIO_DEVICE_SET_IRQS, irq_set);
489 1a403133 Alex Williamson
490 1a403133 Alex Williamson
    g_free(irq_set);
491 65501a74 Alex Williamson
492 65501a74 Alex Williamson
    if (vector->virq < 0) {
493 65501a74 Alex Williamson
        qemu_set_fd_handler(event_notifier_get_fd(&vector->interrupt),
494 65501a74 Alex Williamson
                            NULL, NULL, NULL);
495 65501a74 Alex Williamson
    } else {
496 65501a74 Alex Williamson
        kvm_irqchip_remove_irqfd_notifier(kvm_state, &vector->interrupt,
497 65501a74 Alex Williamson
                                          vector->virq);
498 65501a74 Alex Williamson
        kvm_irqchip_release_virq(kvm_state, vector->virq);
499 65501a74 Alex Williamson
        vector->virq = -1;
500 65501a74 Alex Williamson
    }
501 65501a74 Alex Williamson
502 65501a74 Alex Williamson
    event_notifier_cleanup(&vector->interrupt);
503 65501a74 Alex Williamson
    vector->use = false;
504 65501a74 Alex Williamson
}
505 65501a74 Alex Williamson
506 65501a74 Alex Williamson
/* TODO This should move to msi.c */
507 65501a74 Alex Williamson
static MSIMessage msi_get_msg(PCIDevice *pdev, unsigned int vector)
508 65501a74 Alex Williamson
{
509 65501a74 Alex Williamson
    uint16_t flags = pci_get_word(pdev->config + pdev->msi_cap + PCI_MSI_FLAGS);
510 65501a74 Alex Williamson
    bool msi64bit = flags & PCI_MSI_FLAGS_64BIT;
511 65501a74 Alex Williamson
    MSIMessage msg;
512 65501a74 Alex Williamson
513 65501a74 Alex Williamson
    if (msi64bit) {
514 65501a74 Alex Williamson
        msg.address = pci_get_quad(pdev->config +
515 65501a74 Alex Williamson
                                   pdev->msi_cap + PCI_MSI_ADDRESS_LO);
516 65501a74 Alex Williamson
    } else {
517 65501a74 Alex Williamson
        msg.address = pci_get_long(pdev->config +
518 65501a74 Alex Williamson
                                   pdev->msi_cap + PCI_MSI_ADDRESS_LO);
519 65501a74 Alex Williamson
    }
520 65501a74 Alex Williamson
521 65501a74 Alex Williamson
    msg.data = pci_get_word(pdev->config + pdev->msi_cap +
522 65501a74 Alex Williamson
                            (msi64bit ? PCI_MSI_DATA_64 : PCI_MSI_DATA_32));
523 65501a74 Alex Williamson
    msg.data += vector;
524 65501a74 Alex Williamson
525 65501a74 Alex Williamson
    return msg;
526 65501a74 Alex Williamson
}
527 65501a74 Alex Williamson
528 fd704adc Alex Williamson
static void vfio_enable_msix(VFIODevice *vdev)
529 fd704adc Alex Williamson
{
530 fd704adc Alex Williamson
    vfio_disable_interrupts(vdev);
531 fd704adc Alex Williamson
532 fd704adc Alex Williamson
    vdev->msi_vectors = g_malloc0(vdev->msix->entries * sizeof(VFIOMSIVector));
533 fd704adc Alex Williamson
534 fd704adc Alex Williamson
    vdev->interrupt = VFIO_INT_MSIX;
535 fd704adc Alex Williamson
536 fd704adc Alex Williamson
    if (msix_set_vector_notifiers(&vdev->pdev, vfio_msix_vector_use,
537 fd704adc Alex Williamson
                                  vfio_msix_vector_release)) {
538 fd704adc Alex Williamson
        error_report("vfio: msix_set_vector_notifiers failed\n");
539 fd704adc Alex Williamson
    }
540 fd704adc Alex Williamson
541 fd704adc Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x)\n", __func__, vdev->host.domain,
542 fd704adc Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
543 fd704adc Alex Williamson
}
544 fd704adc Alex Williamson
545 65501a74 Alex Williamson
static void vfio_enable_msi(VFIODevice *vdev)
546 65501a74 Alex Williamson
{
547 65501a74 Alex Williamson
    int ret, i;
548 65501a74 Alex Williamson
549 65501a74 Alex Williamson
    vfio_disable_interrupts(vdev);
550 65501a74 Alex Williamson
551 65501a74 Alex Williamson
    vdev->nr_vectors = msi_nr_vectors_allocated(&vdev->pdev);
552 65501a74 Alex Williamson
retry:
553 65501a74 Alex Williamson
    vdev->msi_vectors = g_malloc0(vdev->nr_vectors * sizeof(VFIOMSIVector));
554 65501a74 Alex Williamson
555 65501a74 Alex Williamson
    for (i = 0; i < vdev->nr_vectors; i++) {
556 65501a74 Alex Williamson
        MSIMessage msg;
557 65501a74 Alex Williamson
        VFIOMSIVector *vector = &vdev->msi_vectors[i];
558 65501a74 Alex Williamson
559 65501a74 Alex Williamson
        vector->vdev = vdev;
560 65501a74 Alex Williamson
        vector->use = true;
561 65501a74 Alex Williamson
562 65501a74 Alex Williamson
        if (event_notifier_init(&vector->interrupt, 0)) {
563 65501a74 Alex Williamson
            error_report("vfio: Error: event_notifier_init failed\n");
564 65501a74 Alex Williamson
        }
565 65501a74 Alex Williamson
566 65501a74 Alex Williamson
        msg = msi_get_msg(&vdev->pdev, i);
567 65501a74 Alex Williamson
568 65501a74 Alex Williamson
        /*
569 65501a74 Alex Williamson
         * Attempt to enable route through KVM irqchip,
570 65501a74 Alex Williamson
         * default to userspace handling if unavailable.
571 65501a74 Alex Williamson
         */
572 65501a74 Alex Williamson
        vector->virq = kvm_irqchip_add_msi_route(kvm_state, msg);
573 65501a74 Alex Williamson
        if (vector->virq < 0 ||
574 65501a74 Alex Williamson
            kvm_irqchip_add_irqfd_notifier(kvm_state, &vector->interrupt,
575 65501a74 Alex Williamson
                                           vector->virq) < 0) {
576 65501a74 Alex Williamson
            qemu_set_fd_handler(event_notifier_get_fd(&vector->interrupt),
577 65501a74 Alex Williamson
                                vfio_msi_interrupt, NULL, vector);
578 65501a74 Alex Williamson
        }
579 65501a74 Alex Williamson
    }
580 65501a74 Alex Williamson
581 65501a74 Alex Williamson
    ret = vfio_enable_vectors(vdev, false);
582 65501a74 Alex Williamson
    if (ret) {
583 65501a74 Alex Williamson
        if (ret < 0) {
584 65501a74 Alex Williamson
            error_report("vfio: Error: Failed to setup MSI fds: %m\n");
585 65501a74 Alex Williamson
        } else if (ret != vdev->nr_vectors) {
586 65501a74 Alex Williamson
            error_report("vfio: Error: Failed to enable %d "
587 65501a74 Alex Williamson
                         "MSI vectors, retry with %d\n", vdev->nr_vectors, ret);
588 65501a74 Alex Williamson
        }
589 65501a74 Alex Williamson
590 65501a74 Alex Williamson
        for (i = 0; i < vdev->nr_vectors; i++) {
591 65501a74 Alex Williamson
            VFIOMSIVector *vector = &vdev->msi_vectors[i];
592 65501a74 Alex Williamson
            if (vector->virq >= 0) {
593 65501a74 Alex Williamson
                kvm_irqchip_remove_irqfd_notifier(kvm_state, &vector->interrupt,
594 65501a74 Alex Williamson
                                                  vector->virq);
595 65501a74 Alex Williamson
                kvm_irqchip_release_virq(kvm_state, vector->virq);
596 65501a74 Alex Williamson
                vector->virq = -1;
597 65501a74 Alex Williamson
            } else {
598 65501a74 Alex Williamson
                qemu_set_fd_handler(event_notifier_get_fd(&vector->interrupt),
599 65501a74 Alex Williamson
                                    NULL, NULL, NULL);
600 65501a74 Alex Williamson
            }
601 65501a74 Alex Williamson
            event_notifier_cleanup(&vector->interrupt);
602 65501a74 Alex Williamson
        }
603 65501a74 Alex Williamson
604 65501a74 Alex Williamson
        g_free(vdev->msi_vectors);
605 65501a74 Alex Williamson
606 65501a74 Alex Williamson
        if (ret > 0 && ret != vdev->nr_vectors) {
607 65501a74 Alex Williamson
            vdev->nr_vectors = ret;
608 65501a74 Alex Williamson
            goto retry;
609 65501a74 Alex Williamson
        }
610 65501a74 Alex Williamson
        vdev->nr_vectors = 0;
611 65501a74 Alex Williamson
612 65501a74 Alex Williamson
        return;
613 65501a74 Alex Williamson
    }
614 65501a74 Alex Williamson
615 fd704adc Alex Williamson
    vdev->interrupt = VFIO_INT_MSI;
616 fd704adc Alex Williamson
617 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x) Enabled %d MSI vectors\n", __func__,
618 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
619 65501a74 Alex Williamson
            vdev->host.function, vdev->nr_vectors);
620 65501a74 Alex Williamson
}
621 65501a74 Alex Williamson
622 fd704adc Alex Williamson
static void vfio_disable_msi_common(VFIODevice *vdev)
623 fd704adc Alex Williamson
{
624 fd704adc Alex Williamson
    g_free(vdev->msi_vectors);
625 fd704adc Alex Williamson
    vdev->msi_vectors = NULL;
626 fd704adc Alex Williamson
    vdev->nr_vectors = 0;
627 fd704adc Alex Williamson
    vdev->interrupt = VFIO_INT_NONE;
628 fd704adc Alex Williamson
629 fd704adc Alex Williamson
    vfio_enable_intx(vdev);
630 fd704adc Alex Williamson
}
631 fd704adc Alex Williamson
632 fd704adc Alex Williamson
static void vfio_disable_msix(VFIODevice *vdev)
633 fd704adc Alex Williamson
{
634 fd704adc Alex Williamson
    msix_unset_vector_notifiers(&vdev->pdev);
635 fd704adc Alex Williamson
636 fd704adc Alex Williamson
    if (vdev->nr_vectors) {
637 fd704adc Alex Williamson
        vfio_disable_irqindex(vdev, VFIO_PCI_MSIX_IRQ_INDEX);
638 fd704adc Alex Williamson
    }
639 fd704adc Alex Williamson
640 fd704adc Alex Williamson
    vfio_disable_msi_common(vdev);
641 fd704adc Alex Williamson
642 a011b10e Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x)\n", __func__, vdev->host.domain,
643 a011b10e Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
644 fd704adc Alex Williamson
}
645 fd704adc Alex Williamson
646 fd704adc Alex Williamson
static void vfio_disable_msi(VFIODevice *vdev)
647 65501a74 Alex Williamson
{
648 65501a74 Alex Williamson
    int i;
649 65501a74 Alex Williamson
650 fd704adc Alex Williamson
    vfio_disable_irqindex(vdev, VFIO_PCI_MSI_IRQ_INDEX);
651 65501a74 Alex Williamson
652 65501a74 Alex Williamson
    for (i = 0; i < vdev->nr_vectors; i++) {
653 65501a74 Alex Williamson
        VFIOMSIVector *vector = &vdev->msi_vectors[i];
654 65501a74 Alex Williamson
655 65501a74 Alex Williamson
        if (!vector->use) {
656 65501a74 Alex Williamson
            continue;
657 65501a74 Alex Williamson
        }
658 65501a74 Alex Williamson
659 65501a74 Alex Williamson
        if (vector->virq >= 0) {
660 65501a74 Alex Williamson
            kvm_irqchip_remove_irqfd_notifier(kvm_state,
661 65501a74 Alex Williamson
                                              &vector->interrupt, vector->virq);
662 65501a74 Alex Williamson
            kvm_irqchip_release_virq(kvm_state, vector->virq);
663 65501a74 Alex Williamson
            vector->virq = -1;
664 65501a74 Alex Williamson
        } else {
665 65501a74 Alex Williamson
            qemu_set_fd_handler(event_notifier_get_fd(&vector->interrupt),
666 65501a74 Alex Williamson
                                NULL, NULL, NULL);
667 65501a74 Alex Williamson
        }
668 65501a74 Alex Williamson
669 65501a74 Alex Williamson
        event_notifier_cleanup(&vector->interrupt);
670 65501a74 Alex Williamson
    }
671 65501a74 Alex Williamson
672 fd704adc Alex Williamson
    vfio_disable_msi_common(vdev);
673 65501a74 Alex Williamson
674 fd704adc Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x)\n", __func__, vdev->host.domain,
675 fd704adc Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
676 65501a74 Alex Williamson
}
677 65501a74 Alex Williamson
678 65501a74 Alex Williamson
/*
679 65501a74 Alex Williamson
 * IO Port/MMIO - Beware of the endians, VFIO is always little endian
680 65501a74 Alex Williamson
 */
681 a8170e5e Avi Kivity
static void vfio_bar_write(void *opaque, hwaddr addr,
682 65501a74 Alex Williamson
                           uint64_t data, unsigned size)
683 65501a74 Alex Williamson
{
684 65501a74 Alex Williamson
    VFIOBAR *bar = opaque;
685 65501a74 Alex Williamson
    union {
686 65501a74 Alex Williamson
        uint8_t byte;
687 65501a74 Alex Williamson
        uint16_t word;
688 65501a74 Alex Williamson
        uint32_t dword;
689 65501a74 Alex Williamson
        uint64_t qword;
690 65501a74 Alex Williamson
    } buf;
691 65501a74 Alex Williamson
692 65501a74 Alex Williamson
    switch (size) {
693 65501a74 Alex Williamson
    case 1:
694 65501a74 Alex Williamson
        buf.byte = data;
695 65501a74 Alex Williamson
        break;
696 65501a74 Alex Williamson
    case 2:
697 65501a74 Alex Williamson
        buf.word = cpu_to_le16(data);
698 65501a74 Alex Williamson
        break;
699 65501a74 Alex Williamson
    case 4:
700 65501a74 Alex Williamson
        buf.dword = cpu_to_le32(data);
701 65501a74 Alex Williamson
        break;
702 65501a74 Alex Williamson
    default:
703 65501a74 Alex Williamson
        hw_error("vfio: unsupported write size, %d bytes\n", size);
704 65501a74 Alex Williamson
        break;
705 65501a74 Alex Williamson
    }
706 65501a74 Alex Williamson
707 65501a74 Alex Williamson
    if (pwrite(bar->fd, &buf, size, bar->fd_offset + addr) != size) {
708 a8170e5e Avi Kivity
        error_report("%s(,0x%"HWADDR_PRIx", 0x%"PRIx64", %d) failed: %m\n",
709 65501a74 Alex Williamson
                     __func__, addr, data, size);
710 65501a74 Alex Williamson
    }
711 65501a74 Alex Williamson
712 a8170e5e Avi Kivity
    DPRINTF("%s(BAR%d+0x%"HWADDR_PRIx", 0x%"PRIx64", %d)\n",
713 65501a74 Alex Williamson
            __func__, bar->nr, addr, data, size);
714 65501a74 Alex Williamson
715 65501a74 Alex Williamson
    /*
716 65501a74 Alex Williamson
     * A read or write to a BAR always signals an INTx EOI.  This will
717 65501a74 Alex Williamson
     * do nothing if not pending (including not in INTx mode).  We assume
718 65501a74 Alex Williamson
     * that a BAR access is in response to an interrupt and that BAR
719 65501a74 Alex Williamson
     * accesses will service the interrupt.  Unfortunately, we don't know
720 65501a74 Alex Williamson
     * which access will service the interrupt, so we're potentially
721 65501a74 Alex Williamson
     * getting quite a few host interrupts per guest interrupt.
722 65501a74 Alex Williamson
     */
723 3a4f2816 Jan Kiszka
    vfio_eoi(container_of(bar, VFIODevice, bars[bar->nr]));
724 65501a74 Alex Williamson
}
725 65501a74 Alex Williamson
726 65501a74 Alex Williamson
static uint64_t vfio_bar_read(void *opaque,
727 a8170e5e Avi Kivity
                              hwaddr addr, unsigned size)
728 65501a74 Alex Williamson
{
729 65501a74 Alex Williamson
    VFIOBAR *bar = opaque;
730 65501a74 Alex Williamson
    union {
731 65501a74 Alex Williamson
        uint8_t byte;
732 65501a74 Alex Williamson
        uint16_t word;
733 65501a74 Alex Williamson
        uint32_t dword;
734 65501a74 Alex Williamson
        uint64_t qword;
735 65501a74 Alex Williamson
    } buf;
736 65501a74 Alex Williamson
    uint64_t data = 0;
737 65501a74 Alex Williamson
738 65501a74 Alex Williamson
    if (pread(bar->fd, &buf, size, bar->fd_offset + addr) != size) {
739 a8170e5e Avi Kivity
        error_report("%s(,0x%"HWADDR_PRIx", %d) failed: %m\n",
740 65501a74 Alex Williamson
                     __func__, addr, size);
741 65501a74 Alex Williamson
        return (uint64_t)-1;
742 65501a74 Alex Williamson
    }
743 65501a74 Alex Williamson
744 65501a74 Alex Williamson
    switch (size) {
745 65501a74 Alex Williamson
    case 1:
746 65501a74 Alex Williamson
        data = buf.byte;
747 65501a74 Alex Williamson
        break;
748 65501a74 Alex Williamson
    case 2:
749 65501a74 Alex Williamson
        data = le16_to_cpu(buf.word);
750 65501a74 Alex Williamson
        break;
751 65501a74 Alex Williamson
    case 4:
752 65501a74 Alex Williamson
        data = le32_to_cpu(buf.dword);
753 65501a74 Alex Williamson
        break;
754 65501a74 Alex Williamson
    default:
755 65501a74 Alex Williamson
        hw_error("vfio: unsupported read size, %d bytes\n", size);
756 65501a74 Alex Williamson
        break;
757 65501a74 Alex Williamson
    }
758 65501a74 Alex Williamson
759 a8170e5e Avi Kivity
    DPRINTF("%s(BAR%d+0x%"HWADDR_PRIx", %d) = 0x%"PRIx64"\n",
760 65501a74 Alex Williamson
            __func__, bar->nr, addr, size, data);
761 65501a74 Alex Williamson
762 65501a74 Alex Williamson
    /* Same as write above */
763 3a4f2816 Jan Kiszka
    vfio_eoi(container_of(bar, VFIODevice, bars[bar->nr]));
764 65501a74 Alex Williamson
765 65501a74 Alex Williamson
    return data;
766 65501a74 Alex Williamson
}
767 65501a74 Alex Williamson
768 65501a74 Alex Williamson
static const MemoryRegionOps vfio_bar_ops = {
769 65501a74 Alex Williamson
    .read = vfio_bar_read,
770 65501a74 Alex Williamson
    .write = vfio_bar_write,
771 65501a74 Alex Williamson
    .endianness = DEVICE_LITTLE_ENDIAN,
772 65501a74 Alex Williamson
};
773 65501a74 Alex Williamson
774 65501a74 Alex Williamson
/*
775 65501a74 Alex Williamson
 * PCI config space
776 65501a74 Alex Williamson
 */
777 65501a74 Alex Williamson
static uint32_t vfio_pci_read_config(PCIDevice *pdev, uint32_t addr, int len)
778 65501a74 Alex Williamson
{
779 65501a74 Alex Williamson
    VFIODevice *vdev = DO_UPCAST(VFIODevice, pdev, pdev);
780 65501a74 Alex Williamson
    uint32_t val = 0;
781 65501a74 Alex Williamson
782 65501a74 Alex Williamson
    /*
783 65501a74 Alex Williamson
     * We only need QEMU PCI config support for the ROM BAR, the MSI and MSIX
784 65501a74 Alex Williamson
     * capabilities, and the multifunction bit below.  We let VFIO handle
785 65501a74 Alex Williamson
     * virtualizing everything else.  Performance is not a concern here.
786 65501a74 Alex Williamson
     */
787 65501a74 Alex Williamson
    if (ranges_overlap(addr, len, PCI_ROM_ADDRESS, 4) ||
788 65501a74 Alex Williamson
        (pdev->cap_present & QEMU_PCI_CAP_MSIX &&
789 65501a74 Alex Williamson
         ranges_overlap(addr, len, pdev->msix_cap, MSIX_CAP_LENGTH)) ||
790 65501a74 Alex Williamson
        (pdev->cap_present & QEMU_PCI_CAP_MSI &&
791 65501a74 Alex Williamson
         ranges_overlap(addr, len, pdev->msi_cap, vdev->msi_cap_size))) {
792 65501a74 Alex Williamson
793 65501a74 Alex Williamson
        val = pci_default_read_config(pdev, addr, len);
794 65501a74 Alex Williamson
    } else {
795 65501a74 Alex Williamson
        if (pread(vdev->fd, &val, len, vdev->config_offset + addr) != len) {
796 65501a74 Alex Williamson
            error_report("%s(%04x:%02x:%02x.%x, 0x%x, 0x%x) failed: %m\n",
797 65501a74 Alex Williamson
                         __func__, vdev->host.domain, vdev->host.bus,
798 65501a74 Alex Williamson
                         vdev->host.slot, vdev->host.function, addr, len);
799 65501a74 Alex Williamson
            return -errno;
800 65501a74 Alex Williamson
        }
801 65501a74 Alex Williamson
        val = le32_to_cpu(val);
802 65501a74 Alex Williamson
    }
803 65501a74 Alex Williamson
804 65501a74 Alex Williamson
    /* Multifunction bit is virualized in QEMU */
805 65501a74 Alex Williamson
    if (unlikely(ranges_overlap(addr, len, PCI_HEADER_TYPE, 1))) {
806 65501a74 Alex Williamson
        uint32_t mask = PCI_HEADER_TYPE_MULTI_FUNCTION;
807 65501a74 Alex Williamson
808 65501a74 Alex Williamson
        if (len == 4) {
809 65501a74 Alex Williamson
            mask <<= 16;
810 65501a74 Alex Williamson
        }
811 65501a74 Alex Williamson
812 65501a74 Alex Williamson
        if (pdev->cap_present & QEMU_PCI_CAP_MULTIFUNCTION) {
813 65501a74 Alex Williamson
            val |= mask;
814 65501a74 Alex Williamson
        } else {
815 65501a74 Alex Williamson
            val &= ~mask;
816 65501a74 Alex Williamson
        }
817 65501a74 Alex Williamson
    }
818 65501a74 Alex Williamson
819 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x, @0x%x, len=0x%x) %x\n", __func__,
820 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
821 65501a74 Alex Williamson
            vdev->host.function, addr, len, val);
822 65501a74 Alex Williamson
823 65501a74 Alex Williamson
    return val;
824 65501a74 Alex Williamson
}
825 65501a74 Alex Williamson
826 65501a74 Alex Williamson
static void vfio_pci_write_config(PCIDevice *pdev, uint32_t addr,
827 65501a74 Alex Williamson
                                  uint32_t val, int len)
828 65501a74 Alex Williamson
{
829 65501a74 Alex Williamson
    VFIODevice *vdev = DO_UPCAST(VFIODevice, pdev, pdev);
830 65501a74 Alex Williamson
    uint32_t val_le = cpu_to_le32(val);
831 65501a74 Alex Williamson
832 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x, @0x%x, 0x%x, len=0x%x)\n", __func__,
833 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
834 65501a74 Alex Williamson
            vdev->host.function, addr, val, len);
835 65501a74 Alex Williamson
836 65501a74 Alex Williamson
    /* Write everything to VFIO, let it filter out what we can't write */
837 65501a74 Alex Williamson
    if (pwrite(vdev->fd, &val_le, len, vdev->config_offset + addr) != len) {
838 65501a74 Alex Williamson
        error_report("%s(%04x:%02x:%02x.%x, 0x%x, 0x%x, 0x%x) failed: %m\n",
839 65501a74 Alex Williamson
                     __func__, vdev->host.domain, vdev->host.bus,
840 65501a74 Alex Williamson
                     vdev->host.slot, vdev->host.function, addr, val, len);
841 65501a74 Alex Williamson
    }
842 65501a74 Alex Williamson
843 65501a74 Alex Williamson
    /* Write standard header bits to emulation */
844 65501a74 Alex Williamson
    if (addr < PCI_CONFIG_HEADER_SIZE) {
845 65501a74 Alex Williamson
        pci_default_write_config(pdev, addr, val, len);
846 65501a74 Alex Williamson
        return;
847 65501a74 Alex Williamson
    }
848 65501a74 Alex Williamson
849 65501a74 Alex Williamson
    /* MSI/MSI-X Enabling/Disabling */
850 65501a74 Alex Williamson
    if (pdev->cap_present & QEMU_PCI_CAP_MSI &&
851 65501a74 Alex Williamson
        ranges_overlap(addr, len, pdev->msi_cap, vdev->msi_cap_size)) {
852 65501a74 Alex Williamson
        int is_enabled, was_enabled = msi_enabled(pdev);
853 65501a74 Alex Williamson
854 65501a74 Alex Williamson
        pci_default_write_config(pdev, addr, val, len);
855 65501a74 Alex Williamson
856 65501a74 Alex Williamson
        is_enabled = msi_enabled(pdev);
857 65501a74 Alex Williamson
858 65501a74 Alex Williamson
        if (!was_enabled && is_enabled) {
859 65501a74 Alex Williamson
            vfio_enable_msi(vdev);
860 65501a74 Alex Williamson
        } else if (was_enabled && !is_enabled) {
861 fd704adc Alex Williamson
            vfio_disable_msi(vdev);
862 65501a74 Alex Williamson
        }
863 65501a74 Alex Williamson
    }
864 65501a74 Alex Williamson
865 65501a74 Alex Williamson
    if (pdev->cap_present & QEMU_PCI_CAP_MSIX &&
866 65501a74 Alex Williamson
        ranges_overlap(addr, len, pdev->msix_cap, MSIX_CAP_LENGTH)) {
867 65501a74 Alex Williamson
        int is_enabled, was_enabled = msix_enabled(pdev);
868 65501a74 Alex Williamson
869 65501a74 Alex Williamson
        pci_default_write_config(pdev, addr, val, len);
870 65501a74 Alex Williamson
871 65501a74 Alex Williamson
        is_enabled = msix_enabled(pdev);
872 65501a74 Alex Williamson
873 65501a74 Alex Williamson
        if (!was_enabled && is_enabled) {
874 fd704adc Alex Williamson
            vfio_enable_msix(vdev);
875 65501a74 Alex Williamson
        } else if (was_enabled && !is_enabled) {
876 fd704adc Alex Williamson
            vfio_disable_msix(vdev);
877 65501a74 Alex Williamson
        }
878 65501a74 Alex Williamson
    }
879 65501a74 Alex Williamson
}
880 65501a74 Alex Williamson
881 65501a74 Alex Williamson
/*
882 65501a74 Alex Williamson
 * DMA - Mapping and unmapping for the "type1" IOMMU interface used on x86
883 65501a74 Alex Williamson
 */
884 af6bc27e Alex Williamson
static int vfio_dma_unmap(VFIOContainer *container,
885 a8170e5e Avi Kivity
                          hwaddr iova, ram_addr_t size)
886 af6bc27e Alex Williamson
{
887 af6bc27e Alex Williamson
    struct vfio_iommu_type1_dma_unmap unmap = {
888 af6bc27e Alex Williamson
        .argsz = sizeof(unmap),
889 af6bc27e Alex Williamson
        .flags = 0,
890 af6bc27e Alex Williamson
        .iova = iova,
891 af6bc27e Alex Williamson
        .size = size,
892 af6bc27e Alex Williamson
    };
893 af6bc27e Alex Williamson
894 af6bc27e Alex Williamson
    if (ioctl(container->fd, VFIO_IOMMU_UNMAP_DMA, &unmap)) {
895 af6bc27e Alex Williamson
        DPRINTF("VFIO_UNMAP_DMA: %d\n", -errno);
896 af6bc27e Alex Williamson
        return -errno;
897 af6bc27e Alex Williamson
    }
898 af6bc27e Alex Williamson
899 af6bc27e Alex Williamson
    return 0;
900 af6bc27e Alex Williamson
}
901 af6bc27e Alex Williamson
902 a8170e5e Avi Kivity
static int vfio_dma_map(VFIOContainer *container, hwaddr iova,
903 65501a74 Alex Williamson
                        ram_addr_t size, void *vaddr, bool readonly)
904 65501a74 Alex Williamson
{
905 65501a74 Alex Williamson
    struct vfio_iommu_type1_dma_map map = {
906 65501a74 Alex Williamson
        .argsz = sizeof(map),
907 65501a74 Alex Williamson
        .flags = VFIO_DMA_MAP_FLAG_READ,
908 5976cdd5 Alex Williamson
        .vaddr = (__u64)(uintptr_t)vaddr,
909 65501a74 Alex Williamson
        .iova = iova,
910 65501a74 Alex Williamson
        .size = size,
911 65501a74 Alex Williamson
    };
912 65501a74 Alex Williamson
913 65501a74 Alex Williamson
    if (!readonly) {
914 65501a74 Alex Williamson
        map.flags |= VFIO_DMA_MAP_FLAG_WRITE;
915 65501a74 Alex Williamson
    }
916 65501a74 Alex Williamson
917 12af1344 Alex Williamson
    /*
918 12af1344 Alex Williamson
     * Try the mapping, if it fails with EBUSY, unmap the region and try
919 12af1344 Alex Williamson
     * again.  This shouldn't be necessary, but we sometimes see it in
920 12af1344 Alex Williamson
     * the the VGA ROM space.
921 12af1344 Alex Williamson
     */
922 12af1344 Alex Williamson
    if (ioctl(container->fd, VFIO_IOMMU_MAP_DMA, &map) == 0 ||
923 12af1344 Alex Williamson
        (errno == EBUSY && vfio_dma_unmap(container, iova, size) == 0 &&
924 12af1344 Alex Williamson
         ioctl(container->fd, VFIO_IOMMU_MAP_DMA, &map) == 0)) {
925 12af1344 Alex Williamson
        return 0;
926 65501a74 Alex Williamson
    }
927 65501a74 Alex Williamson
928 12af1344 Alex Williamson
    DPRINTF("VFIO_MAP_DMA: %d\n", -errno);
929 12af1344 Alex Williamson
    return -errno;
930 65501a74 Alex Williamson
}
931 65501a74 Alex Williamson
932 65501a74 Alex Williamson
static bool vfio_listener_skipped_section(MemoryRegionSection *section)
933 65501a74 Alex Williamson
{
934 65501a74 Alex Williamson
    return !memory_region_is_ram(section->mr);
935 65501a74 Alex Williamson
}
936 65501a74 Alex Williamson
937 65501a74 Alex Williamson
static void vfio_listener_region_add(MemoryListener *listener,
938 65501a74 Alex Williamson
                                     MemoryRegionSection *section)
939 65501a74 Alex Williamson
{
940 65501a74 Alex Williamson
    VFIOContainer *container = container_of(listener, VFIOContainer,
941 65501a74 Alex Williamson
                                            iommu_data.listener);
942 a8170e5e Avi Kivity
    hwaddr iova, end;
943 65501a74 Alex Williamson
    void *vaddr;
944 65501a74 Alex Williamson
    int ret;
945 65501a74 Alex Williamson
946 65501a74 Alex Williamson
    if (vfio_listener_skipped_section(section)) {
947 a8170e5e Avi Kivity
        DPRINTF("vfio: SKIPPING region_add %"HWADDR_PRIx" - %"PRIx64"\n",
948 65501a74 Alex Williamson
                section->offset_within_address_space,
949 65501a74 Alex Williamson
                section->offset_within_address_space + section->size - 1);
950 65501a74 Alex Williamson
        return;
951 65501a74 Alex Williamson
    }
952 65501a74 Alex Williamson
953 65501a74 Alex Williamson
    if (unlikely((section->offset_within_address_space & ~TARGET_PAGE_MASK) !=
954 65501a74 Alex Williamson
                 (section->offset_within_region & ~TARGET_PAGE_MASK))) {
955 65501a74 Alex Williamson
        error_report("%s received unaligned region\n", __func__);
956 65501a74 Alex Williamson
        return;
957 65501a74 Alex Williamson
    }
958 65501a74 Alex Williamson
959 65501a74 Alex Williamson
    iova = TARGET_PAGE_ALIGN(section->offset_within_address_space);
960 65501a74 Alex Williamson
    end = (section->offset_within_address_space + section->size) &
961 65501a74 Alex Williamson
          TARGET_PAGE_MASK;
962 65501a74 Alex Williamson
963 65501a74 Alex Williamson
    if (iova >= end) {
964 65501a74 Alex Williamson
        return;
965 65501a74 Alex Williamson
    }
966 65501a74 Alex Williamson
967 65501a74 Alex Williamson
    vaddr = memory_region_get_ram_ptr(section->mr) +
968 65501a74 Alex Williamson
            section->offset_within_region +
969 65501a74 Alex Williamson
            (iova - section->offset_within_address_space);
970 65501a74 Alex Williamson
971 a8170e5e Avi Kivity
    DPRINTF("vfio: region_add %"HWADDR_PRIx" - %"HWADDR_PRIx" [%p]\n",
972 65501a74 Alex Williamson
            iova, end - 1, vaddr);
973 65501a74 Alex Williamson
974 65501a74 Alex Williamson
    ret = vfio_dma_map(container, iova, end - iova, vaddr, section->readonly);
975 65501a74 Alex Williamson
    if (ret) {
976 a8170e5e Avi Kivity
        error_report("vfio_dma_map(%p, 0x%"HWADDR_PRIx", "
977 a8170e5e Avi Kivity
                     "0x%"HWADDR_PRIx", %p) = %d (%m)\n",
978 65501a74 Alex Williamson
                     container, iova, end - iova, vaddr, ret);
979 65501a74 Alex Williamson
    }
980 65501a74 Alex Williamson
}
981 65501a74 Alex Williamson
982 65501a74 Alex Williamson
static void vfio_listener_region_del(MemoryListener *listener,
983 65501a74 Alex Williamson
                                     MemoryRegionSection *section)
984 65501a74 Alex Williamson
{
985 65501a74 Alex Williamson
    VFIOContainer *container = container_of(listener, VFIOContainer,
986 65501a74 Alex Williamson
                                            iommu_data.listener);
987 a8170e5e Avi Kivity
    hwaddr iova, end;
988 65501a74 Alex Williamson
    int ret;
989 65501a74 Alex Williamson
990 65501a74 Alex Williamson
    if (vfio_listener_skipped_section(section)) {
991 a8170e5e Avi Kivity
        DPRINTF("vfio: SKIPPING region_del %"HWADDR_PRIx" - %"PRIx64"\n",
992 65501a74 Alex Williamson
                section->offset_within_address_space,
993 65501a74 Alex Williamson
                section->offset_within_address_space + section->size - 1);
994 65501a74 Alex Williamson
        return;
995 65501a74 Alex Williamson
    }
996 65501a74 Alex Williamson
997 65501a74 Alex Williamson
    if (unlikely((section->offset_within_address_space & ~TARGET_PAGE_MASK) !=
998 65501a74 Alex Williamson
                 (section->offset_within_region & ~TARGET_PAGE_MASK))) {
999 65501a74 Alex Williamson
        error_report("%s received unaligned region\n", __func__);
1000 65501a74 Alex Williamson
        return;
1001 65501a74 Alex Williamson
    }
1002 65501a74 Alex Williamson
1003 65501a74 Alex Williamson
    iova = TARGET_PAGE_ALIGN(section->offset_within_address_space);
1004 65501a74 Alex Williamson
    end = (section->offset_within_address_space + section->size) &
1005 65501a74 Alex Williamson
          TARGET_PAGE_MASK;
1006 65501a74 Alex Williamson
1007 65501a74 Alex Williamson
    if (iova >= end) {
1008 65501a74 Alex Williamson
        return;
1009 65501a74 Alex Williamson
    }
1010 65501a74 Alex Williamson
1011 a8170e5e Avi Kivity
    DPRINTF("vfio: region_del %"HWADDR_PRIx" - %"HWADDR_PRIx"\n",
1012 65501a74 Alex Williamson
            iova, end - 1);
1013 65501a74 Alex Williamson
1014 65501a74 Alex Williamson
    ret = vfio_dma_unmap(container, iova, end - iova);
1015 65501a74 Alex Williamson
    if (ret) {
1016 a8170e5e Avi Kivity
        error_report("vfio_dma_unmap(%p, 0x%"HWADDR_PRIx", "
1017 a8170e5e Avi Kivity
                     "0x%"HWADDR_PRIx") = %d (%m)\n",
1018 65501a74 Alex Williamson
                     container, iova, end - iova, ret);
1019 65501a74 Alex Williamson
    }
1020 65501a74 Alex Williamson
}
1021 65501a74 Alex Williamson
1022 65501a74 Alex Williamson
static MemoryListener vfio_memory_listener = {
1023 65501a74 Alex Williamson
    .region_add = vfio_listener_region_add,
1024 65501a74 Alex Williamson
    .region_del = vfio_listener_region_del,
1025 65501a74 Alex Williamson
};
1026 65501a74 Alex Williamson
1027 65501a74 Alex Williamson
static void vfio_listener_release(VFIOContainer *container)
1028 65501a74 Alex Williamson
{
1029 65501a74 Alex Williamson
    memory_listener_unregister(&container->iommu_data.listener);
1030 65501a74 Alex Williamson
}
1031 65501a74 Alex Williamson
1032 65501a74 Alex Williamson
/*
1033 65501a74 Alex Williamson
 * Interrupt setup
1034 65501a74 Alex Williamson
 */
1035 65501a74 Alex Williamson
static void vfio_disable_interrupts(VFIODevice *vdev)
1036 65501a74 Alex Williamson
{
1037 65501a74 Alex Williamson
    switch (vdev->interrupt) {
1038 65501a74 Alex Williamson
    case VFIO_INT_INTx:
1039 65501a74 Alex Williamson
        vfio_disable_intx(vdev);
1040 65501a74 Alex Williamson
        break;
1041 65501a74 Alex Williamson
    case VFIO_INT_MSI:
1042 fd704adc Alex Williamson
        vfio_disable_msi(vdev);
1043 65501a74 Alex Williamson
        break;
1044 65501a74 Alex Williamson
    case VFIO_INT_MSIX:
1045 fd704adc Alex Williamson
        vfio_disable_msix(vdev);
1046 65501a74 Alex Williamson
        break;
1047 65501a74 Alex Williamson
    }
1048 65501a74 Alex Williamson
}
1049 65501a74 Alex Williamson
1050 65501a74 Alex Williamson
static int vfio_setup_msi(VFIODevice *vdev, int pos)
1051 65501a74 Alex Williamson
{
1052 65501a74 Alex Williamson
    uint16_t ctrl;
1053 65501a74 Alex Williamson
    bool msi_64bit, msi_maskbit;
1054 65501a74 Alex Williamson
    int ret, entries;
1055 65501a74 Alex Williamson
1056 65501a74 Alex Williamson
    if (pread(vdev->fd, &ctrl, sizeof(ctrl),
1057 65501a74 Alex Williamson
              vdev->config_offset + pos + PCI_CAP_FLAGS) != sizeof(ctrl)) {
1058 65501a74 Alex Williamson
        return -errno;
1059 65501a74 Alex Williamson
    }
1060 65501a74 Alex Williamson
    ctrl = le16_to_cpu(ctrl);
1061 65501a74 Alex Williamson
1062 65501a74 Alex Williamson
    msi_64bit = !!(ctrl & PCI_MSI_FLAGS_64BIT);
1063 65501a74 Alex Williamson
    msi_maskbit = !!(ctrl & PCI_MSI_FLAGS_MASKBIT);
1064 65501a74 Alex Williamson
    entries = 1 << ((ctrl & PCI_MSI_FLAGS_QMASK) >> 1);
1065 65501a74 Alex Williamson
1066 65501a74 Alex Williamson
    DPRINTF("%04x:%02x:%02x.%x PCI MSI CAP @0x%x\n", vdev->host.domain,
1067 65501a74 Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function, pos);
1068 65501a74 Alex Williamson
1069 65501a74 Alex Williamson
    ret = msi_init(&vdev->pdev, pos, entries, msi_64bit, msi_maskbit);
1070 65501a74 Alex Williamson
    if (ret < 0) {
1071 e43b9a5a Alex Williamson
        if (ret == -ENOTSUP) {
1072 e43b9a5a Alex Williamson
            return 0;
1073 e43b9a5a Alex Williamson
        }
1074 65501a74 Alex Williamson
        error_report("vfio: msi_init failed\n");
1075 65501a74 Alex Williamson
        return ret;
1076 65501a74 Alex Williamson
    }
1077 65501a74 Alex Williamson
    vdev->msi_cap_size = 0xa + (msi_maskbit ? 0xa : 0) + (msi_64bit ? 0x4 : 0);
1078 65501a74 Alex Williamson
1079 65501a74 Alex Williamson
    return 0;
1080 65501a74 Alex Williamson
}
1081 65501a74 Alex Williamson
1082 65501a74 Alex Williamson
/*
1083 65501a74 Alex Williamson
 * We don't have any control over how pci_add_capability() inserts
1084 65501a74 Alex Williamson
 * capabilities into the chain.  In order to setup MSI-X we need a
1085 65501a74 Alex Williamson
 * MemoryRegion for the BAR.  In order to setup the BAR and not
1086 65501a74 Alex Williamson
 * attempt to mmap the MSI-X table area, which VFIO won't allow, we
1087 65501a74 Alex Williamson
 * need to first look for where the MSI-X table lives.  So we
1088 65501a74 Alex Williamson
 * unfortunately split MSI-X setup across two functions.
1089 65501a74 Alex Williamson
 */
1090 65501a74 Alex Williamson
static int vfio_early_setup_msix(VFIODevice *vdev)
1091 65501a74 Alex Williamson
{
1092 65501a74 Alex Williamson
    uint8_t pos;
1093 65501a74 Alex Williamson
    uint16_t ctrl;
1094 65501a74 Alex Williamson
    uint32_t table, pba;
1095 65501a74 Alex Williamson
1096 65501a74 Alex Williamson
    pos = pci_find_capability(&vdev->pdev, PCI_CAP_ID_MSIX);
1097 65501a74 Alex Williamson
    if (!pos) {
1098 65501a74 Alex Williamson
        return 0;
1099 65501a74 Alex Williamson
    }
1100 65501a74 Alex Williamson
1101 65501a74 Alex Williamson
    if (pread(vdev->fd, &ctrl, sizeof(ctrl),
1102 65501a74 Alex Williamson
              vdev->config_offset + pos + PCI_CAP_FLAGS) != sizeof(ctrl)) {
1103 65501a74 Alex Williamson
        return -errno;
1104 65501a74 Alex Williamson
    }
1105 65501a74 Alex Williamson
1106 65501a74 Alex Williamson
    if (pread(vdev->fd, &table, sizeof(table),
1107 65501a74 Alex Williamson
              vdev->config_offset + pos + PCI_MSIX_TABLE) != sizeof(table)) {
1108 65501a74 Alex Williamson
        return -errno;
1109 65501a74 Alex Williamson
    }
1110 65501a74 Alex Williamson
1111 65501a74 Alex Williamson
    if (pread(vdev->fd, &pba, sizeof(pba),
1112 65501a74 Alex Williamson
              vdev->config_offset + pos + PCI_MSIX_PBA) != sizeof(pba)) {
1113 65501a74 Alex Williamson
        return -errno;
1114 65501a74 Alex Williamson
    }
1115 65501a74 Alex Williamson
1116 65501a74 Alex Williamson
    ctrl = le16_to_cpu(ctrl);
1117 65501a74 Alex Williamson
    table = le32_to_cpu(table);
1118 65501a74 Alex Williamson
    pba = le32_to_cpu(pba);
1119 65501a74 Alex Williamson
1120 65501a74 Alex Williamson
    vdev->msix = g_malloc0(sizeof(*(vdev->msix)));
1121 65501a74 Alex Williamson
    vdev->msix->table_bar = table & PCI_MSIX_FLAGS_BIRMASK;
1122 65501a74 Alex Williamson
    vdev->msix->table_offset = table & ~PCI_MSIX_FLAGS_BIRMASK;
1123 65501a74 Alex Williamson
    vdev->msix->pba_bar = pba & PCI_MSIX_FLAGS_BIRMASK;
1124 65501a74 Alex Williamson
    vdev->msix->pba_offset = pba & ~PCI_MSIX_FLAGS_BIRMASK;
1125 65501a74 Alex Williamson
    vdev->msix->entries = (ctrl & PCI_MSIX_FLAGS_QSIZE) + 1;
1126 65501a74 Alex Williamson
1127 65501a74 Alex Williamson
    DPRINTF("%04x:%02x:%02x.%x "
1128 65501a74 Alex Williamson
            "PCI MSI-X CAP @0x%x, BAR %d, offset 0x%x, entries %d\n",
1129 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
1130 65501a74 Alex Williamson
            vdev->host.function, pos, vdev->msix->table_bar,
1131 65501a74 Alex Williamson
            vdev->msix->table_offset, vdev->msix->entries);
1132 65501a74 Alex Williamson
1133 65501a74 Alex Williamson
    return 0;
1134 65501a74 Alex Williamson
}
1135 65501a74 Alex Williamson
1136 65501a74 Alex Williamson
static int vfio_setup_msix(VFIODevice *vdev, int pos)
1137 65501a74 Alex Williamson
{
1138 65501a74 Alex Williamson
    int ret;
1139 65501a74 Alex Williamson
1140 65501a74 Alex Williamson
    ret = msix_init(&vdev->pdev, vdev->msix->entries,
1141 65501a74 Alex Williamson
                    &vdev->bars[vdev->msix->table_bar].mem,
1142 65501a74 Alex Williamson
                    vdev->msix->table_bar, vdev->msix->table_offset,
1143 65501a74 Alex Williamson
                    &vdev->bars[vdev->msix->pba_bar].mem,
1144 65501a74 Alex Williamson
                    vdev->msix->pba_bar, vdev->msix->pba_offset, pos);
1145 65501a74 Alex Williamson
    if (ret < 0) {
1146 e43b9a5a Alex Williamson
        if (ret == -ENOTSUP) {
1147 e43b9a5a Alex Williamson
            return 0;
1148 e43b9a5a Alex Williamson
        }
1149 65501a74 Alex Williamson
        error_report("vfio: msix_init failed\n");
1150 65501a74 Alex Williamson
        return ret;
1151 65501a74 Alex Williamson
    }
1152 65501a74 Alex Williamson
1153 65501a74 Alex Williamson
    return 0;
1154 65501a74 Alex Williamson
}
1155 65501a74 Alex Williamson
1156 65501a74 Alex Williamson
static void vfio_teardown_msi(VFIODevice *vdev)
1157 65501a74 Alex Williamson
{
1158 65501a74 Alex Williamson
    msi_uninit(&vdev->pdev);
1159 65501a74 Alex Williamson
1160 65501a74 Alex Williamson
    if (vdev->msix) {
1161 65501a74 Alex Williamson
        msix_uninit(&vdev->pdev, &vdev->bars[vdev->msix->table_bar].mem,
1162 65501a74 Alex Williamson
                    &vdev->bars[vdev->msix->pba_bar].mem);
1163 65501a74 Alex Williamson
    }
1164 65501a74 Alex Williamson
}
1165 65501a74 Alex Williamson
1166 65501a74 Alex Williamson
/*
1167 65501a74 Alex Williamson
 * Resource setup
1168 65501a74 Alex Williamson
 */
1169 65501a74 Alex Williamson
static void vfio_mmap_set_enabled(VFIODevice *vdev, bool enabled)
1170 65501a74 Alex Williamson
{
1171 65501a74 Alex Williamson
    int i;
1172 65501a74 Alex Williamson
1173 65501a74 Alex Williamson
    for (i = 0; i < PCI_ROM_SLOT; i++) {
1174 65501a74 Alex Williamson
        VFIOBAR *bar = &vdev->bars[i];
1175 65501a74 Alex Williamson
1176 65501a74 Alex Williamson
        if (!bar->size) {
1177 65501a74 Alex Williamson
            continue;
1178 65501a74 Alex Williamson
        }
1179 65501a74 Alex Williamson
1180 65501a74 Alex Williamson
        memory_region_set_enabled(&bar->mmap_mem, enabled);
1181 65501a74 Alex Williamson
        if (vdev->msix && vdev->msix->table_bar == i) {
1182 65501a74 Alex Williamson
            memory_region_set_enabled(&vdev->msix->mmap_mem, enabled);
1183 65501a74 Alex Williamson
        }
1184 65501a74 Alex Williamson
    }
1185 65501a74 Alex Williamson
}
1186 65501a74 Alex Williamson
1187 65501a74 Alex Williamson
static void vfio_unmap_bar(VFIODevice *vdev, int nr)
1188 65501a74 Alex Williamson
{
1189 65501a74 Alex Williamson
    VFIOBAR *bar = &vdev->bars[nr];
1190 65501a74 Alex Williamson
1191 65501a74 Alex Williamson
    if (!bar->size) {
1192 65501a74 Alex Williamson
        return;
1193 65501a74 Alex Williamson
    }
1194 65501a74 Alex Williamson
1195 65501a74 Alex Williamson
    memory_region_del_subregion(&bar->mem, &bar->mmap_mem);
1196 65501a74 Alex Williamson
    munmap(bar->mmap, memory_region_size(&bar->mmap_mem));
1197 65501a74 Alex Williamson
1198 65501a74 Alex Williamson
    if (vdev->msix && vdev->msix->table_bar == nr) {
1199 65501a74 Alex Williamson
        memory_region_del_subregion(&bar->mem, &vdev->msix->mmap_mem);
1200 65501a74 Alex Williamson
        munmap(vdev->msix->mmap, memory_region_size(&vdev->msix->mmap_mem));
1201 65501a74 Alex Williamson
    }
1202 65501a74 Alex Williamson
1203 65501a74 Alex Williamson
    memory_region_destroy(&bar->mem);
1204 65501a74 Alex Williamson
}
1205 65501a74 Alex Williamson
1206 65501a74 Alex Williamson
static int vfio_mmap_bar(VFIOBAR *bar, MemoryRegion *mem, MemoryRegion *submem,
1207 65501a74 Alex Williamson
                         void **map, size_t size, off_t offset,
1208 65501a74 Alex Williamson
                         const char *name)
1209 65501a74 Alex Williamson
{
1210 65501a74 Alex Williamson
    int ret = 0;
1211 65501a74 Alex Williamson
1212 65501a74 Alex Williamson
    if (size && bar->flags & VFIO_REGION_INFO_FLAG_MMAP) {
1213 65501a74 Alex Williamson
        int prot = 0;
1214 65501a74 Alex Williamson
1215 65501a74 Alex Williamson
        if (bar->flags & VFIO_REGION_INFO_FLAG_READ) {
1216 65501a74 Alex Williamson
            prot |= PROT_READ;
1217 65501a74 Alex Williamson
        }
1218 65501a74 Alex Williamson
1219 65501a74 Alex Williamson
        if (bar->flags & VFIO_REGION_INFO_FLAG_WRITE) {
1220 65501a74 Alex Williamson
            prot |= PROT_WRITE;
1221 65501a74 Alex Williamson
        }
1222 65501a74 Alex Williamson
1223 65501a74 Alex Williamson
        *map = mmap(NULL, size, prot, MAP_SHARED,
1224 65501a74 Alex Williamson
                    bar->fd, bar->fd_offset + offset);
1225 65501a74 Alex Williamson
        if (*map == MAP_FAILED) {
1226 65501a74 Alex Williamson
            *map = NULL;
1227 65501a74 Alex Williamson
            ret = -errno;
1228 65501a74 Alex Williamson
            goto empty_region;
1229 65501a74 Alex Williamson
        }
1230 65501a74 Alex Williamson
1231 65501a74 Alex Williamson
        memory_region_init_ram_ptr(submem, name, size, *map);
1232 65501a74 Alex Williamson
    } else {
1233 65501a74 Alex Williamson
empty_region:
1234 65501a74 Alex Williamson
        /* Create a zero sized sub-region to make cleanup easy. */
1235 65501a74 Alex Williamson
        memory_region_init(submem, name, 0);
1236 65501a74 Alex Williamson
    }
1237 65501a74 Alex Williamson
1238 65501a74 Alex Williamson
    memory_region_add_subregion(mem, offset, submem);
1239 65501a74 Alex Williamson
1240 65501a74 Alex Williamson
    return ret;
1241 65501a74 Alex Williamson
}
1242 65501a74 Alex Williamson
1243 65501a74 Alex Williamson
static void vfio_map_bar(VFIODevice *vdev, int nr)
1244 65501a74 Alex Williamson
{
1245 65501a74 Alex Williamson
    VFIOBAR *bar = &vdev->bars[nr];
1246 65501a74 Alex Williamson
    unsigned size = bar->size;
1247 65501a74 Alex Williamson
    char name[64];
1248 65501a74 Alex Williamson
    uint32_t pci_bar;
1249 65501a74 Alex Williamson
    uint8_t type;
1250 65501a74 Alex Williamson
    int ret;
1251 65501a74 Alex Williamson
1252 65501a74 Alex Williamson
    /* Skip both unimplemented BARs and the upper half of 64bit BARS. */
1253 65501a74 Alex Williamson
    if (!size) {
1254 65501a74 Alex Williamson
        return;
1255 65501a74 Alex Williamson
    }
1256 65501a74 Alex Williamson
1257 65501a74 Alex Williamson
    snprintf(name, sizeof(name), "VFIO %04x:%02x:%02x.%x BAR %d",
1258 65501a74 Alex Williamson
             vdev->host.domain, vdev->host.bus, vdev->host.slot,
1259 65501a74 Alex Williamson
             vdev->host.function, nr);
1260 65501a74 Alex Williamson
1261 65501a74 Alex Williamson
    /* Determine what type of BAR this is for registration */
1262 65501a74 Alex Williamson
    ret = pread(vdev->fd, &pci_bar, sizeof(pci_bar),
1263 65501a74 Alex Williamson
                vdev->config_offset + PCI_BASE_ADDRESS_0 + (4 * nr));
1264 65501a74 Alex Williamson
    if (ret != sizeof(pci_bar)) {
1265 65501a74 Alex Williamson
        error_report("vfio: Failed to read BAR %d (%m)\n", nr);
1266 65501a74 Alex Williamson
        return;
1267 65501a74 Alex Williamson
    }
1268 65501a74 Alex Williamson
1269 65501a74 Alex Williamson
    pci_bar = le32_to_cpu(pci_bar);
1270 65501a74 Alex Williamson
    type = pci_bar & (pci_bar & PCI_BASE_ADDRESS_SPACE_IO ?
1271 65501a74 Alex Williamson
           ~PCI_BASE_ADDRESS_IO_MASK : ~PCI_BASE_ADDRESS_MEM_MASK);
1272 65501a74 Alex Williamson
1273 65501a74 Alex Williamson
    /* A "slow" read/write mapping underlies all BARs */
1274 65501a74 Alex Williamson
    memory_region_init_io(&bar->mem, &vfio_bar_ops, bar, name, size);
1275 65501a74 Alex Williamson
    pci_register_bar(&vdev->pdev, nr, type, &bar->mem);
1276 65501a74 Alex Williamson
1277 65501a74 Alex Williamson
    /*
1278 65501a74 Alex Williamson
     * We can't mmap areas overlapping the MSIX vector table, so we
1279 65501a74 Alex Williamson
     * potentially insert a direct-mapped subregion before and after it.
1280 65501a74 Alex Williamson
     */
1281 65501a74 Alex Williamson
    if (vdev->msix && vdev->msix->table_bar == nr) {
1282 65501a74 Alex Williamson
        size = vdev->msix->table_offset & TARGET_PAGE_MASK;
1283 65501a74 Alex Williamson
    }
1284 65501a74 Alex Williamson
1285 65501a74 Alex Williamson
    strncat(name, " mmap", sizeof(name) - strlen(name) - 1);
1286 65501a74 Alex Williamson
    if (vfio_mmap_bar(bar, &bar->mem,
1287 65501a74 Alex Williamson
                      &bar->mmap_mem, &bar->mmap, size, 0, name)) {
1288 65501a74 Alex Williamson
        error_report("%s unsupported. Performance may be slow\n", name);
1289 65501a74 Alex Williamson
    }
1290 65501a74 Alex Williamson
1291 65501a74 Alex Williamson
    if (vdev->msix && vdev->msix->table_bar == nr) {
1292 65501a74 Alex Williamson
        unsigned start;
1293 65501a74 Alex Williamson
1294 65501a74 Alex Williamson
        start = TARGET_PAGE_ALIGN(vdev->msix->table_offset +
1295 65501a74 Alex Williamson
                                  (vdev->msix->entries * PCI_MSIX_ENTRY_SIZE));
1296 65501a74 Alex Williamson
1297 65501a74 Alex Williamson
        size = start < bar->size ? bar->size - start : 0;
1298 65501a74 Alex Williamson
        strncat(name, " msix-hi", sizeof(name) - strlen(name) - 1);
1299 65501a74 Alex Williamson
        /* VFIOMSIXInfo contains another MemoryRegion for this mapping */
1300 65501a74 Alex Williamson
        if (vfio_mmap_bar(bar, &bar->mem, &vdev->msix->mmap_mem,
1301 65501a74 Alex Williamson
                          &vdev->msix->mmap, size, start, name)) {
1302 65501a74 Alex Williamson
            error_report("%s unsupported. Performance may be slow\n", name);
1303 65501a74 Alex Williamson
        }
1304 65501a74 Alex Williamson
    }
1305 65501a74 Alex Williamson
}
1306 65501a74 Alex Williamson
1307 65501a74 Alex Williamson
static void vfio_map_bars(VFIODevice *vdev)
1308 65501a74 Alex Williamson
{
1309 65501a74 Alex Williamson
    int i;
1310 65501a74 Alex Williamson
1311 65501a74 Alex Williamson
    for (i = 0; i < PCI_ROM_SLOT; i++) {
1312 65501a74 Alex Williamson
        vfio_map_bar(vdev, i);
1313 65501a74 Alex Williamson
    }
1314 65501a74 Alex Williamson
}
1315 65501a74 Alex Williamson
1316 65501a74 Alex Williamson
static void vfio_unmap_bars(VFIODevice *vdev)
1317 65501a74 Alex Williamson
{
1318 65501a74 Alex Williamson
    int i;
1319 65501a74 Alex Williamson
1320 65501a74 Alex Williamson
    for (i = 0; i < PCI_ROM_SLOT; i++) {
1321 65501a74 Alex Williamson
        vfio_unmap_bar(vdev, i);
1322 65501a74 Alex Williamson
    }
1323 65501a74 Alex Williamson
}
1324 65501a74 Alex Williamson
1325 65501a74 Alex Williamson
/*
1326 65501a74 Alex Williamson
 * General setup
1327 65501a74 Alex Williamson
 */
1328 65501a74 Alex Williamson
static uint8_t vfio_std_cap_max_size(PCIDevice *pdev, uint8_t pos)
1329 65501a74 Alex Williamson
{
1330 65501a74 Alex Williamson
    uint8_t tmp, next = 0xff;
1331 65501a74 Alex Williamson
1332 65501a74 Alex Williamson
    for (tmp = pdev->config[PCI_CAPABILITY_LIST]; tmp;
1333 65501a74 Alex Williamson
         tmp = pdev->config[tmp + 1]) {
1334 65501a74 Alex Williamson
        if (tmp > pos && tmp < next) {
1335 65501a74 Alex Williamson
            next = tmp;
1336 65501a74 Alex Williamson
        }
1337 65501a74 Alex Williamson
    }
1338 65501a74 Alex Williamson
1339 65501a74 Alex Williamson
    return next - pos;
1340 65501a74 Alex Williamson
}
1341 65501a74 Alex Williamson
1342 65501a74 Alex Williamson
static int vfio_add_std_cap(VFIODevice *vdev, uint8_t pos)
1343 65501a74 Alex Williamson
{
1344 65501a74 Alex Williamson
    PCIDevice *pdev = &vdev->pdev;
1345 65501a74 Alex Williamson
    uint8_t cap_id, next, size;
1346 65501a74 Alex Williamson
    int ret;
1347 65501a74 Alex Williamson
1348 65501a74 Alex Williamson
    cap_id = pdev->config[pos];
1349 65501a74 Alex Williamson
    next = pdev->config[pos + 1];
1350 65501a74 Alex Williamson
1351 65501a74 Alex Williamson
    /*
1352 65501a74 Alex Williamson
     * If it becomes important to configure capabilities to their actual
1353 65501a74 Alex Williamson
     * size, use this as the default when it's something we don't recognize.
1354 65501a74 Alex Williamson
     * Since QEMU doesn't actually handle many of the config accesses,
1355 65501a74 Alex Williamson
     * exact size doesn't seem worthwhile.
1356 65501a74 Alex Williamson
     */
1357 65501a74 Alex Williamson
    size = vfio_std_cap_max_size(pdev, pos);
1358 65501a74 Alex Williamson
1359 65501a74 Alex Williamson
    /*
1360 65501a74 Alex Williamson
     * pci_add_capability always inserts the new capability at the head
1361 65501a74 Alex Williamson
     * of the chain.  Therefore to end up with a chain that matches the
1362 65501a74 Alex Williamson
     * physical device, we insert from the end by making this recursive.
1363 65501a74 Alex Williamson
     * This is also why we pre-caclulate size above as cached config space
1364 65501a74 Alex Williamson
     * will be changed as we unwind the stack.
1365 65501a74 Alex Williamson
     */
1366 65501a74 Alex Williamson
    if (next) {
1367 65501a74 Alex Williamson
        ret = vfio_add_std_cap(vdev, next);
1368 65501a74 Alex Williamson
        if (ret) {
1369 65501a74 Alex Williamson
            return ret;
1370 65501a74 Alex Williamson
        }
1371 65501a74 Alex Williamson
    } else {
1372 65501a74 Alex Williamson
        pdev->config[PCI_CAPABILITY_LIST] = 0; /* Begin the rebuild */
1373 65501a74 Alex Williamson
    }
1374 65501a74 Alex Williamson
1375 65501a74 Alex Williamson
    switch (cap_id) {
1376 65501a74 Alex Williamson
    case PCI_CAP_ID_MSI:
1377 65501a74 Alex Williamson
        ret = vfio_setup_msi(vdev, pos);
1378 65501a74 Alex Williamson
        break;
1379 65501a74 Alex Williamson
    case PCI_CAP_ID_MSIX:
1380 65501a74 Alex Williamson
        ret = vfio_setup_msix(vdev, pos);
1381 65501a74 Alex Williamson
        break;
1382 65501a74 Alex Williamson
    default:
1383 65501a74 Alex Williamson
        ret = pci_add_capability(pdev, cap_id, pos, size);
1384 65501a74 Alex Williamson
        break;
1385 65501a74 Alex Williamson
    }
1386 65501a74 Alex Williamson
1387 65501a74 Alex Williamson
    if (ret < 0) {
1388 65501a74 Alex Williamson
        error_report("vfio: %04x:%02x:%02x.%x Error adding PCI capability "
1389 65501a74 Alex Williamson
                     "0x%x[0x%x]@0x%x: %d\n", vdev->host.domain,
1390 65501a74 Alex Williamson
                     vdev->host.bus, vdev->host.slot, vdev->host.function,
1391 65501a74 Alex Williamson
                     cap_id, size, pos, ret);
1392 65501a74 Alex Williamson
        return ret;
1393 65501a74 Alex Williamson
    }
1394 65501a74 Alex Williamson
1395 65501a74 Alex Williamson
    return 0;
1396 65501a74 Alex Williamson
}
1397 65501a74 Alex Williamson
1398 65501a74 Alex Williamson
static int vfio_add_capabilities(VFIODevice *vdev)
1399 65501a74 Alex Williamson
{
1400 65501a74 Alex Williamson
    PCIDevice *pdev = &vdev->pdev;
1401 65501a74 Alex Williamson
1402 65501a74 Alex Williamson
    if (!(pdev->config[PCI_STATUS] & PCI_STATUS_CAP_LIST) ||
1403 65501a74 Alex Williamson
        !pdev->config[PCI_CAPABILITY_LIST]) {
1404 65501a74 Alex Williamson
        return 0; /* Nothing to add */
1405 65501a74 Alex Williamson
    }
1406 65501a74 Alex Williamson
1407 65501a74 Alex Williamson
    return vfio_add_std_cap(vdev, pdev->config[PCI_CAPABILITY_LIST]);
1408 65501a74 Alex Williamson
}
1409 65501a74 Alex Williamson
1410 65501a74 Alex Williamson
static int vfio_load_rom(VFIODevice *vdev)
1411 65501a74 Alex Williamson
{
1412 65501a74 Alex Williamson
    uint64_t size = vdev->rom_size;
1413 65501a74 Alex Williamson
    char name[32];
1414 65501a74 Alex Williamson
    off_t off = 0, voff = vdev->rom_offset;
1415 65501a74 Alex Williamson
    ssize_t bytes;
1416 65501a74 Alex Williamson
    void *ptr;
1417 65501a74 Alex Williamson
1418 65501a74 Alex Williamson
    /* If loading ROM from file, pci handles it */
1419 65501a74 Alex Williamson
    if (vdev->pdev.romfile || !vdev->pdev.rom_bar || !size) {
1420 65501a74 Alex Williamson
        return 0;
1421 65501a74 Alex Williamson
    }
1422 65501a74 Alex Williamson
1423 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x)\n", __func__, vdev->host.domain,
1424 65501a74 Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
1425 65501a74 Alex Williamson
1426 65501a74 Alex Williamson
    snprintf(name, sizeof(name), "vfio[%04x:%02x:%02x.%x].rom",
1427 65501a74 Alex Williamson
             vdev->host.domain, vdev->host.bus, vdev->host.slot,
1428 65501a74 Alex Williamson
             vdev->host.function);
1429 65501a74 Alex Williamson
    memory_region_init_ram(&vdev->pdev.rom, name, size);
1430 65501a74 Alex Williamson
    ptr = memory_region_get_ram_ptr(&vdev->pdev.rom);
1431 65501a74 Alex Williamson
    memset(ptr, 0xff, size);
1432 65501a74 Alex Williamson
1433 65501a74 Alex Williamson
    while (size) {
1434 65501a74 Alex Williamson
        bytes = pread(vdev->fd, ptr + off, size, voff + off);
1435 65501a74 Alex Williamson
        if (bytes == 0) {
1436 65501a74 Alex Williamson
            break; /* expect that we could get back less than the ROM BAR */
1437 65501a74 Alex Williamson
        } else if (bytes > 0) {
1438 65501a74 Alex Williamson
            off += bytes;
1439 65501a74 Alex Williamson
            size -= bytes;
1440 65501a74 Alex Williamson
        } else {
1441 65501a74 Alex Williamson
            if (errno == EINTR || errno == EAGAIN) {
1442 65501a74 Alex Williamson
                continue;
1443 65501a74 Alex Williamson
            }
1444 65501a74 Alex Williamson
            error_report("vfio: Error reading device ROM: %m\n");
1445 65501a74 Alex Williamson
            memory_region_destroy(&vdev->pdev.rom);
1446 65501a74 Alex Williamson
            return -errno;
1447 65501a74 Alex Williamson
        }
1448 65501a74 Alex Williamson
    }
1449 65501a74 Alex Williamson
1450 65501a74 Alex Williamson
    pci_register_bar(&vdev->pdev, PCI_ROM_SLOT, 0, &vdev->pdev.rom);
1451 65501a74 Alex Williamson
    vdev->pdev.has_rom = true;
1452 65501a74 Alex Williamson
    return 0;
1453 65501a74 Alex Williamson
}
1454 65501a74 Alex Williamson
1455 65501a74 Alex Williamson
static int vfio_connect_container(VFIOGroup *group)
1456 65501a74 Alex Williamson
{
1457 65501a74 Alex Williamson
    VFIOContainer *container;
1458 65501a74 Alex Williamson
    int ret, fd;
1459 65501a74 Alex Williamson
1460 65501a74 Alex Williamson
    if (group->container) {
1461 65501a74 Alex Williamson
        return 0;
1462 65501a74 Alex Williamson
    }
1463 65501a74 Alex Williamson
1464 65501a74 Alex Williamson
    QLIST_FOREACH(container, &container_list, next) {
1465 65501a74 Alex Williamson
        if (!ioctl(group->fd, VFIO_GROUP_SET_CONTAINER, &container->fd)) {
1466 65501a74 Alex Williamson
            group->container = container;
1467 65501a74 Alex Williamson
            QLIST_INSERT_HEAD(&container->group_list, group, container_next);
1468 65501a74 Alex Williamson
            return 0;
1469 65501a74 Alex Williamson
        }
1470 65501a74 Alex Williamson
    }
1471 65501a74 Alex Williamson
1472 65501a74 Alex Williamson
    fd = qemu_open("/dev/vfio/vfio", O_RDWR);
1473 65501a74 Alex Williamson
    if (fd < 0) {
1474 65501a74 Alex Williamson
        error_report("vfio: failed to open /dev/vfio/vfio: %m\n");
1475 65501a74 Alex Williamson
        return -errno;
1476 65501a74 Alex Williamson
    }
1477 65501a74 Alex Williamson
1478 65501a74 Alex Williamson
    ret = ioctl(fd, VFIO_GET_API_VERSION);
1479 65501a74 Alex Williamson
    if (ret != VFIO_API_VERSION) {
1480 65501a74 Alex Williamson
        error_report("vfio: supported vfio version: %d, "
1481 65501a74 Alex Williamson
                     "reported version: %d\n", VFIO_API_VERSION, ret);
1482 65501a74 Alex Williamson
        close(fd);
1483 65501a74 Alex Williamson
        return -EINVAL;
1484 65501a74 Alex Williamson
    }
1485 65501a74 Alex Williamson
1486 65501a74 Alex Williamson
    container = g_malloc0(sizeof(*container));
1487 65501a74 Alex Williamson
    container->fd = fd;
1488 65501a74 Alex Williamson
1489 65501a74 Alex Williamson
    if (ioctl(fd, VFIO_CHECK_EXTENSION, VFIO_TYPE1_IOMMU)) {
1490 65501a74 Alex Williamson
        ret = ioctl(group->fd, VFIO_GROUP_SET_CONTAINER, &fd);
1491 65501a74 Alex Williamson
        if (ret) {
1492 65501a74 Alex Williamson
            error_report("vfio: failed to set group container: %m\n");
1493 65501a74 Alex Williamson
            g_free(container);
1494 65501a74 Alex Williamson
            close(fd);
1495 65501a74 Alex Williamson
            return -errno;
1496 65501a74 Alex Williamson
        }
1497 65501a74 Alex Williamson
1498 65501a74 Alex Williamson
        ret = ioctl(fd, VFIO_SET_IOMMU, VFIO_TYPE1_IOMMU);
1499 65501a74 Alex Williamson
        if (ret) {
1500 65501a74 Alex Williamson
            error_report("vfio: failed to set iommu for container: %m\n");
1501 65501a74 Alex Williamson
            g_free(container);
1502 65501a74 Alex Williamson
            close(fd);
1503 65501a74 Alex Williamson
            return -errno;
1504 65501a74 Alex Williamson
        }
1505 65501a74 Alex Williamson
1506 65501a74 Alex Williamson
        container->iommu_data.listener = vfio_memory_listener;
1507 65501a74 Alex Williamson
        container->iommu_data.release = vfio_listener_release;
1508 65501a74 Alex Williamson
1509 f6790af6 Avi Kivity
        memory_listener_register(&container->iommu_data.listener, &address_space_memory);
1510 65501a74 Alex Williamson
    } else {
1511 65501a74 Alex Williamson
        error_report("vfio: No available IOMMU models\n");
1512 65501a74 Alex Williamson
        g_free(container);
1513 65501a74 Alex Williamson
        close(fd);
1514 65501a74 Alex Williamson
        return -EINVAL;
1515 65501a74 Alex Williamson
    }
1516 65501a74 Alex Williamson
1517 65501a74 Alex Williamson
    QLIST_INIT(&container->group_list);
1518 65501a74 Alex Williamson
    QLIST_INSERT_HEAD(&container_list, container, next);
1519 65501a74 Alex Williamson
1520 65501a74 Alex Williamson
    group->container = container;
1521 65501a74 Alex Williamson
    QLIST_INSERT_HEAD(&container->group_list, group, container_next);
1522 65501a74 Alex Williamson
1523 65501a74 Alex Williamson
    return 0;
1524 65501a74 Alex Williamson
}
1525 65501a74 Alex Williamson
1526 65501a74 Alex Williamson
static void vfio_disconnect_container(VFIOGroup *group)
1527 65501a74 Alex Williamson
{
1528 65501a74 Alex Williamson
    VFIOContainer *container = group->container;
1529 65501a74 Alex Williamson
1530 65501a74 Alex Williamson
    if (ioctl(group->fd, VFIO_GROUP_UNSET_CONTAINER, &container->fd)) {
1531 65501a74 Alex Williamson
        error_report("vfio: error disconnecting group %d from container\n",
1532 65501a74 Alex Williamson
                     group->groupid);
1533 65501a74 Alex Williamson
    }
1534 65501a74 Alex Williamson
1535 65501a74 Alex Williamson
    QLIST_REMOVE(group, container_next);
1536 65501a74 Alex Williamson
    group->container = NULL;
1537 65501a74 Alex Williamson
1538 65501a74 Alex Williamson
    if (QLIST_EMPTY(&container->group_list)) {
1539 65501a74 Alex Williamson
        if (container->iommu_data.release) {
1540 65501a74 Alex Williamson
            container->iommu_data.release(container);
1541 65501a74 Alex Williamson
        }
1542 65501a74 Alex Williamson
        QLIST_REMOVE(container, next);
1543 65501a74 Alex Williamson
        DPRINTF("vfio_disconnect_container: close container->fd\n");
1544 65501a74 Alex Williamson
        close(container->fd);
1545 65501a74 Alex Williamson
        g_free(container);
1546 65501a74 Alex Williamson
    }
1547 65501a74 Alex Williamson
}
1548 65501a74 Alex Williamson
1549 65501a74 Alex Williamson
static VFIOGroup *vfio_get_group(int groupid)
1550 65501a74 Alex Williamson
{
1551 65501a74 Alex Williamson
    VFIOGroup *group;
1552 65501a74 Alex Williamson
    char path[32];
1553 65501a74 Alex Williamson
    struct vfio_group_status status = { .argsz = sizeof(status) };
1554 65501a74 Alex Williamson
1555 65501a74 Alex Williamson
    QLIST_FOREACH(group, &group_list, next) {
1556 65501a74 Alex Williamson
        if (group->groupid == groupid) {
1557 65501a74 Alex Williamson
            return group;
1558 65501a74 Alex Williamson
        }
1559 65501a74 Alex Williamson
    }
1560 65501a74 Alex Williamson
1561 65501a74 Alex Williamson
    group = g_malloc0(sizeof(*group));
1562 65501a74 Alex Williamson
1563 65501a74 Alex Williamson
    snprintf(path, sizeof(path), "/dev/vfio/%d", groupid);
1564 65501a74 Alex Williamson
    group->fd = qemu_open(path, O_RDWR);
1565 65501a74 Alex Williamson
    if (group->fd < 0) {
1566 65501a74 Alex Williamson
        error_report("vfio: error opening %s: %m\n", path);
1567 65501a74 Alex Williamson
        g_free(group);
1568 65501a74 Alex Williamson
        return NULL;
1569 65501a74 Alex Williamson
    }
1570 65501a74 Alex Williamson
1571 65501a74 Alex Williamson
    if (ioctl(group->fd, VFIO_GROUP_GET_STATUS, &status)) {
1572 65501a74 Alex Williamson
        error_report("vfio: error getting group status: %m\n");
1573 65501a74 Alex Williamson
        close(group->fd);
1574 65501a74 Alex Williamson
        g_free(group);
1575 65501a74 Alex Williamson
        return NULL;
1576 65501a74 Alex Williamson
    }
1577 65501a74 Alex Williamson
1578 65501a74 Alex Williamson
    if (!(status.flags & VFIO_GROUP_FLAGS_VIABLE)) {
1579 65501a74 Alex Williamson
        error_report("vfio: error, group %d is not viable, please ensure "
1580 65501a74 Alex Williamson
                     "all devices within the iommu_group are bound to their "
1581 65501a74 Alex Williamson
                     "vfio bus driver.\n", groupid);
1582 65501a74 Alex Williamson
        close(group->fd);
1583 65501a74 Alex Williamson
        g_free(group);
1584 65501a74 Alex Williamson
        return NULL;
1585 65501a74 Alex Williamson
    }
1586 65501a74 Alex Williamson
1587 65501a74 Alex Williamson
    group->groupid = groupid;
1588 65501a74 Alex Williamson
    QLIST_INIT(&group->device_list);
1589 65501a74 Alex Williamson
1590 65501a74 Alex Williamson
    if (vfio_connect_container(group)) {
1591 65501a74 Alex Williamson
        error_report("vfio: failed to setup container for group %d\n", groupid);
1592 65501a74 Alex Williamson
        close(group->fd);
1593 65501a74 Alex Williamson
        g_free(group);
1594 65501a74 Alex Williamson
        return NULL;
1595 65501a74 Alex Williamson
    }
1596 65501a74 Alex Williamson
1597 65501a74 Alex Williamson
    QLIST_INSERT_HEAD(&group_list, group, next);
1598 65501a74 Alex Williamson
1599 65501a74 Alex Williamson
    return group;
1600 65501a74 Alex Williamson
}
1601 65501a74 Alex Williamson
1602 65501a74 Alex Williamson
static void vfio_put_group(VFIOGroup *group)
1603 65501a74 Alex Williamson
{
1604 65501a74 Alex Williamson
    if (!QLIST_EMPTY(&group->device_list)) {
1605 65501a74 Alex Williamson
        return;
1606 65501a74 Alex Williamson
    }
1607 65501a74 Alex Williamson
1608 65501a74 Alex Williamson
    vfio_disconnect_container(group);
1609 65501a74 Alex Williamson
    QLIST_REMOVE(group, next);
1610 65501a74 Alex Williamson
    DPRINTF("vfio_put_group: close group->fd\n");
1611 65501a74 Alex Williamson
    close(group->fd);
1612 65501a74 Alex Williamson
    g_free(group);
1613 65501a74 Alex Williamson
}
1614 65501a74 Alex Williamson
1615 65501a74 Alex Williamson
static int vfio_get_device(VFIOGroup *group, const char *name, VFIODevice *vdev)
1616 65501a74 Alex Williamson
{
1617 65501a74 Alex Williamson
    struct vfio_device_info dev_info = { .argsz = sizeof(dev_info) };
1618 65501a74 Alex Williamson
    struct vfio_region_info reg_info = { .argsz = sizeof(reg_info) };
1619 65501a74 Alex Williamson
    int ret, i;
1620 65501a74 Alex Williamson
1621 65501a74 Alex Williamson
    ret = ioctl(group->fd, VFIO_GROUP_GET_DEVICE_FD, name);
1622 65501a74 Alex Williamson
    if (ret < 0) {
1623 65501a74 Alex Williamson
        error_report("vfio: error getting device %s from group %d: %m\n",
1624 65501a74 Alex Williamson
                     name, group->groupid);
1625 65501a74 Alex Williamson
        error_report("Verify all devices in group %d are bound to vfio-pci "
1626 65501a74 Alex Williamson
                     "or pci-stub and not already in use\n", group->groupid);
1627 65501a74 Alex Williamson
        return ret;
1628 65501a74 Alex Williamson
    }
1629 65501a74 Alex Williamson
1630 65501a74 Alex Williamson
    vdev->fd = ret;
1631 65501a74 Alex Williamson
    vdev->group = group;
1632 65501a74 Alex Williamson
    QLIST_INSERT_HEAD(&group->device_list, vdev, next);
1633 65501a74 Alex Williamson
1634 65501a74 Alex Williamson
    /* Sanity check device */
1635 65501a74 Alex Williamson
    ret = ioctl(vdev->fd, VFIO_DEVICE_GET_INFO, &dev_info);
1636 65501a74 Alex Williamson
    if (ret) {
1637 65501a74 Alex Williamson
        error_report("vfio: error getting device info: %m\n");
1638 65501a74 Alex Williamson
        goto error;
1639 65501a74 Alex Williamson
    }
1640 65501a74 Alex Williamson
1641 65501a74 Alex Williamson
    DPRINTF("Device %s flags: %u, regions: %u, irgs: %u\n", name,
1642 65501a74 Alex Williamson
            dev_info.flags, dev_info.num_regions, dev_info.num_irqs);
1643 65501a74 Alex Williamson
1644 65501a74 Alex Williamson
    if (!(dev_info.flags & VFIO_DEVICE_FLAGS_PCI)) {
1645 65501a74 Alex Williamson
        error_report("vfio: Um, this isn't a PCI device\n");
1646 65501a74 Alex Williamson
        goto error;
1647 65501a74 Alex Williamson
    }
1648 65501a74 Alex Williamson
1649 65501a74 Alex Williamson
    vdev->reset_works = !!(dev_info.flags & VFIO_DEVICE_FLAGS_RESET);
1650 65501a74 Alex Williamson
    if (!vdev->reset_works) {
1651 65501a74 Alex Williamson
        error_report("Warning, device %s does not support reset\n", name);
1652 65501a74 Alex Williamson
    }
1653 65501a74 Alex Williamson
1654 65501a74 Alex Williamson
    if (dev_info.num_regions != VFIO_PCI_NUM_REGIONS) {
1655 65501a74 Alex Williamson
        error_report("vfio: unexpected number of io regions %u\n",
1656 65501a74 Alex Williamson
                     dev_info.num_regions);
1657 65501a74 Alex Williamson
        goto error;
1658 65501a74 Alex Williamson
    }
1659 65501a74 Alex Williamson
1660 65501a74 Alex Williamson
    if (dev_info.num_irqs != VFIO_PCI_NUM_IRQS) {
1661 65501a74 Alex Williamson
        error_report("vfio: unexpected number of irqs %u\n", dev_info.num_irqs);
1662 65501a74 Alex Williamson
        goto error;
1663 65501a74 Alex Williamson
    }
1664 65501a74 Alex Williamson
1665 65501a74 Alex Williamson
    for (i = VFIO_PCI_BAR0_REGION_INDEX; i < VFIO_PCI_ROM_REGION_INDEX; i++) {
1666 65501a74 Alex Williamson
        reg_info.index = i;
1667 65501a74 Alex Williamson
1668 65501a74 Alex Williamson
        ret = ioctl(vdev->fd, VFIO_DEVICE_GET_REGION_INFO, &reg_info);
1669 65501a74 Alex Williamson
        if (ret) {
1670 65501a74 Alex Williamson
            error_report("vfio: Error getting region %d info: %m\n", i);
1671 65501a74 Alex Williamson
            goto error;
1672 65501a74 Alex Williamson
        }
1673 65501a74 Alex Williamson
1674 65501a74 Alex Williamson
        DPRINTF("Device %s region %d:\n", name, i);
1675 65501a74 Alex Williamson
        DPRINTF("  size: 0x%lx, offset: 0x%lx, flags: 0x%lx\n",
1676 65501a74 Alex Williamson
                (unsigned long)reg_info.size, (unsigned long)reg_info.offset,
1677 65501a74 Alex Williamson
                (unsigned long)reg_info.flags);
1678 65501a74 Alex Williamson
1679 65501a74 Alex Williamson
        vdev->bars[i].flags = reg_info.flags;
1680 65501a74 Alex Williamson
        vdev->bars[i].size = reg_info.size;
1681 65501a74 Alex Williamson
        vdev->bars[i].fd_offset = reg_info.offset;
1682 65501a74 Alex Williamson
        vdev->bars[i].fd = vdev->fd;
1683 65501a74 Alex Williamson
        vdev->bars[i].nr = i;
1684 65501a74 Alex Williamson
    }
1685 65501a74 Alex Williamson
1686 65501a74 Alex Williamson
    reg_info.index = VFIO_PCI_ROM_REGION_INDEX;
1687 65501a74 Alex Williamson
1688 65501a74 Alex Williamson
    ret = ioctl(vdev->fd, VFIO_DEVICE_GET_REGION_INFO, &reg_info);
1689 65501a74 Alex Williamson
    if (ret) {
1690 65501a74 Alex Williamson
        error_report("vfio: Error getting ROM info: %m\n");
1691 65501a74 Alex Williamson
        goto error;
1692 65501a74 Alex Williamson
    }
1693 65501a74 Alex Williamson
1694 65501a74 Alex Williamson
    DPRINTF("Device %s ROM:\n", name);
1695 65501a74 Alex Williamson
    DPRINTF("  size: 0x%lx, offset: 0x%lx, flags: 0x%lx\n",
1696 65501a74 Alex Williamson
            (unsigned long)reg_info.size, (unsigned long)reg_info.offset,
1697 65501a74 Alex Williamson
            (unsigned long)reg_info.flags);
1698 65501a74 Alex Williamson
1699 65501a74 Alex Williamson
    vdev->rom_size = reg_info.size;
1700 65501a74 Alex Williamson
    vdev->rom_offset = reg_info.offset;
1701 65501a74 Alex Williamson
1702 65501a74 Alex Williamson
    reg_info.index = VFIO_PCI_CONFIG_REGION_INDEX;
1703 65501a74 Alex Williamson
1704 65501a74 Alex Williamson
    ret = ioctl(vdev->fd, VFIO_DEVICE_GET_REGION_INFO, &reg_info);
1705 65501a74 Alex Williamson
    if (ret) {
1706 65501a74 Alex Williamson
        error_report("vfio: Error getting config info: %m\n");
1707 65501a74 Alex Williamson
        goto error;
1708 65501a74 Alex Williamson
    }
1709 65501a74 Alex Williamson
1710 65501a74 Alex Williamson
    DPRINTF("Device %s config:\n", name);
1711 65501a74 Alex Williamson
    DPRINTF("  size: 0x%lx, offset: 0x%lx, flags: 0x%lx\n",
1712 65501a74 Alex Williamson
            (unsigned long)reg_info.size, (unsigned long)reg_info.offset,
1713 65501a74 Alex Williamson
            (unsigned long)reg_info.flags);
1714 65501a74 Alex Williamson
1715 65501a74 Alex Williamson
    vdev->config_size = reg_info.size;
1716 65501a74 Alex Williamson
    vdev->config_offset = reg_info.offset;
1717 65501a74 Alex Williamson
1718 65501a74 Alex Williamson
error:
1719 65501a74 Alex Williamson
    if (ret) {
1720 65501a74 Alex Williamson
        QLIST_REMOVE(vdev, next);
1721 65501a74 Alex Williamson
        vdev->group = NULL;
1722 65501a74 Alex Williamson
        close(vdev->fd);
1723 65501a74 Alex Williamson
    }
1724 65501a74 Alex Williamson
    return ret;
1725 65501a74 Alex Williamson
}
1726 65501a74 Alex Williamson
1727 65501a74 Alex Williamson
static void vfio_put_device(VFIODevice *vdev)
1728 65501a74 Alex Williamson
{
1729 65501a74 Alex Williamson
    QLIST_REMOVE(vdev, next);
1730 65501a74 Alex Williamson
    vdev->group = NULL;
1731 65501a74 Alex Williamson
    DPRINTF("vfio_put_device: close vdev->fd\n");
1732 65501a74 Alex Williamson
    close(vdev->fd);
1733 65501a74 Alex Williamson
    if (vdev->msix) {
1734 65501a74 Alex Williamson
        g_free(vdev->msix);
1735 65501a74 Alex Williamson
        vdev->msix = NULL;
1736 65501a74 Alex Williamson
    }
1737 65501a74 Alex Williamson
}
1738 65501a74 Alex Williamson
1739 65501a74 Alex Williamson
static int vfio_initfn(PCIDevice *pdev)
1740 65501a74 Alex Williamson
{
1741 65501a74 Alex Williamson
    VFIODevice *pvdev, *vdev = DO_UPCAST(VFIODevice, pdev, pdev);
1742 65501a74 Alex Williamson
    VFIOGroup *group;
1743 65501a74 Alex Williamson
    char path[PATH_MAX], iommu_group_path[PATH_MAX], *group_name;
1744 65501a74 Alex Williamson
    ssize_t len;
1745 65501a74 Alex Williamson
    struct stat st;
1746 65501a74 Alex Williamson
    int groupid;
1747 65501a74 Alex Williamson
    int ret;
1748 65501a74 Alex Williamson
1749 65501a74 Alex Williamson
    /* Check that the host device exists */
1750 65501a74 Alex Williamson
    snprintf(path, sizeof(path),
1751 65501a74 Alex Williamson
             "/sys/bus/pci/devices/%04x:%02x:%02x.%01x/",
1752 65501a74 Alex Williamson
             vdev->host.domain, vdev->host.bus, vdev->host.slot,
1753 65501a74 Alex Williamson
             vdev->host.function);
1754 65501a74 Alex Williamson
    if (stat(path, &st) < 0) {
1755 65501a74 Alex Williamson
        error_report("vfio: error: no such host device: %s\n", path);
1756 65501a74 Alex Williamson
        return -errno;
1757 65501a74 Alex Williamson
    }
1758 65501a74 Alex Williamson
1759 65501a74 Alex Williamson
    strncat(path, "iommu_group", sizeof(path) - strlen(path) - 1);
1760 65501a74 Alex Williamson
1761 65501a74 Alex Williamson
    len = readlink(path, iommu_group_path, PATH_MAX);
1762 65501a74 Alex Williamson
    if (len <= 0) {
1763 65501a74 Alex Williamson
        error_report("vfio: error no iommu_group for device\n");
1764 65501a74 Alex Williamson
        return -errno;
1765 65501a74 Alex Williamson
    }
1766 65501a74 Alex Williamson
1767 65501a74 Alex Williamson
    iommu_group_path[len] = 0;
1768 65501a74 Alex Williamson
    group_name = basename(iommu_group_path);
1769 65501a74 Alex Williamson
1770 65501a74 Alex Williamson
    if (sscanf(group_name, "%d", &groupid) != 1) {
1771 65501a74 Alex Williamson
        error_report("vfio: error reading %s: %m\n", path);
1772 65501a74 Alex Williamson
        return -errno;
1773 65501a74 Alex Williamson
    }
1774 65501a74 Alex Williamson
1775 65501a74 Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x) group %d\n", __func__, vdev->host.domain,
1776 65501a74 Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function, groupid);
1777 65501a74 Alex Williamson
1778 65501a74 Alex Williamson
    group = vfio_get_group(groupid);
1779 65501a74 Alex Williamson
    if (!group) {
1780 65501a74 Alex Williamson
        error_report("vfio: failed to get group %d\n", groupid);
1781 65501a74 Alex Williamson
        return -ENOENT;
1782 65501a74 Alex Williamson
    }
1783 65501a74 Alex Williamson
1784 65501a74 Alex Williamson
    snprintf(path, sizeof(path), "%04x:%02x:%02x.%01x",
1785 65501a74 Alex Williamson
            vdev->host.domain, vdev->host.bus, vdev->host.slot,
1786 65501a74 Alex Williamson
            vdev->host.function);
1787 65501a74 Alex Williamson
1788 65501a74 Alex Williamson
    QLIST_FOREACH(pvdev, &group->device_list, next) {
1789 65501a74 Alex Williamson
        if (pvdev->host.domain == vdev->host.domain &&
1790 65501a74 Alex Williamson
            pvdev->host.bus == vdev->host.bus &&
1791 65501a74 Alex Williamson
            pvdev->host.slot == vdev->host.slot &&
1792 65501a74 Alex Williamson
            pvdev->host.function == vdev->host.function) {
1793 65501a74 Alex Williamson
1794 65501a74 Alex Williamson
            error_report("vfio: error: device %s is already attached\n", path);
1795 65501a74 Alex Williamson
            vfio_put_group(group);
1796 65501a74 Alex Williamson
            return -EBUSY;
1797 65501a74 Alex Williamson
        }
1798 65501a74 Alex Williamson
    }
1799 65501a74 Alex Williamson
1800 65501a74 Alex Williamson
    ret = vfio_get_device(group, path, vdev);
1801 65501a74 Alex Williamson
    if (ret) {
1802 65501a74 Alex Williamson
        error_report("vfio: failed to get device %s\n", path);
1803 65501a74 Alex Williamson
        vfio_put_group(group);
1804 65501a74 Alex Williamson
        return ret;
1805 65501a74 Alex Williamson
    }
1806 65501a74 Alex Williamson
1807 65501a74 Alex Williamson
    /* Get a copy of config space */
1808 65501a74 Alex Williamson
    ret = pread(vdev->fd, vdev->pdev.config,
1809 65501a74 Alex Williamson
                MIN(pci_config_size(&vdev->pdev), vdev->config_size),
1810 65501a74 Alex Williamson
                vdev->config_offset);
1811 65501a74 Alex Williamson
    if (ret < (int)MIN(pci_config_size(&vdev->pdev), vdev->config_size)) {
1812 65501a74 Alex Williamson
        ret = ret < 0 ? -errno : -EFAULT;
1813 65501a74 Alex Williamson
        error_report("vfio: Failed to read device config space\n");
1814 65501a74 Alex Williamson
        goto out_put;
1815 65501a74 Alex Williamson
    }
1816 65501a74 Alex Williamson
1817 65501a74 Alex Williamson
    /*
1818 65501a74 Alex Williamson
     * Clear host resource mapping info.  If we choose not to register a
1819 65501a74 Alex Williamson
     * BAR, such as might be the case with the option ROM, we can get
1820 65501a74 Alex Williamson
     * confusing, unwritable, residual addresses from the host here.
1821 65501a74 Alex Williamson
     */
1822 65501a74 Alex Williamson
    memset(&vdev->pdev.config[PCI_BASE_ADDRESS_0], 0, 24);
1823 65501a74 Alex Williamson
    memset(&vdev->pdev.config[PCI_ROM_ADDRESS], 0, 4);
1824 65501a74 Alex Williamson
1825 65501a74 Alex Williamson
    vfio_load_rom(vdev);
1826 65501a74 Alex Williamson
1827 65501a74 Alex Williamson
    ret = vfio_early_setup_msix(vdev);
1828 65501a74 Alex Williamson
    if (ret) {
1829 65501a74 Alex Williamson
        goto out_put;
1830 65501a74 Alex Williamson
    }
1831 65501a74 Alex Williamson
1832 65501a74 Alex Williamson
    vfio_map_bars(vdev);
1833 65501a74 Alex Williamson
1834 65501a74 Alex Williamson
    ret = vfio_add_capabilities(vdev);
1835 65501a74 Alex Williamson
    if (ret) {
1836 65501a74 Alex Williamson
        goto out_teardown;
1837 65501a74 Alex Williamson
    }
1838 65501a74 Alex Williamson
1839 65501a74 Alex Williamson
    if (vfio_pci_read_config(&vdev->pdev, PCI_INTERRUPT_PIN, 1)) {
1840 ea486926 Alex Williamson
        vdev->intx.mmap_timer = qemu_new_timer_ms(vm_clock,
1841 ea486926 Alex Williamson
                                                  vfio_intx_mmap_enable, vdev);
1842 65501a74 Alex Williamson
        ret = vfio_enable_intx(vdev);
1843 65501a74 Alex Williamson
        if (ret) {
1844 65501a74 Alex Williamson
            goto out_teardown;
1845 65501a74 Alex Williamson
        }
1846 65501a74 Alex Williamson
    }
1847 65501a74 Alex Williamson
1848 65501a74 Alex Williamson
    return 0;
1849 65501a74 Alex Williamson
1850 65501a74 Alex Williamson
out_teardown:
1851 65501a74 Alex Williamson
    pci_device_set_intx_routing_notifier(&vdev->pdev, NULL);
1852 65501a74 Alex Williamson
    vfio_teardown_msi(vdev);
1853 65501a74 Alex Williamson
    vfio_unmap_bars(vdev);
1854 65501a74 Alex Williamson
out_put:
1855 65501a74 Alex Williamson
    vfio_put_device(vdev);
1856 65501a74 Alex Williamson
    vfio_put_group(group);
1857 65501a74 Alex Williamson
    return ret;
1858 65501a74 Alex Williamson
}
1859 65501a74 Alex Williamson
1860 65501a74 Alex Williamson
static void vfio_exitfn(PCIDevice *pdev)
1861 65501a74 Alex Williamson
{
1862 65501a74 Alex Williamson
    VFIODevice *vdev = DO_UPCAST(VFIODevice, pdev, pdev);
1863 65501a74 Alex Williamson
    VFIOGroup *group = vdev->group;
1864 65501a74 Alex Williamson
1865 65501a74 Alex Williamson
    pci_device_set_intx_routing_notifier(&vdev->pdev, NULL);
1866 65501a74 Alex Williamson
    vfio_disable_interrupts(vdev);
1867 ea486926 Alex Williamson
    if (vdev->intx.mmap_timer) {
1868 ea486926 Alex Williamson
        qemu_free_timer(vdev->intx.mmap_timer);
1869 ea486926 Alex Williamson
    }
1870 65501a74 Alex Williamson
    vfio_teardown_msi(vdev);
1871 65501a74 Alex Williamson
    vfio_unmap_bars(vdev);
1872 65501a74 Alex Williamson
    vfio_put_device(vdev);
1873 65501a74 Alex Williamson
    vfio_put_group(group);
1874 65501a74 Alex Williamson
}
1875 65501a74 Alex Williamson
1876 65501a74 Alex Williamson
static void vfio_pci_reset(DeviceState *dev)
1877 65501a74 Alex Williamson
{
1878 65501a74 Alex Williamson
    PCIDevice *pdev = DO_UPCAST(PCIDevice, qdev, dev);
1879 65501a74 Alex Williamson
    VFIODevice *vdev = DO_UPCAST(VFIODevice, pdev, pdev);
1880 5834a83f Alex Williamson
    uint16_t cmd;
1881 65501a74 Alex Williamson
1882 5834a83f Alex Williamson
    DPRINTF("%s(%04x:%02x:%02x.%x)\n", __func__, vdev->host.domain,
1883 5834a83f Alex Williamson
            vdev->host.bus, vdev->host.slot, vdev->host.function);
1884 5834a83f Alex Williamson
1885 5834a83f Alex Williamson
    vfio_disable_interrupts(vdev);
1886 65501a74 Alex Williamson
1887 5834a83f Alex Williamson
    /*
1888 5834a83f Alex Williamson
     * Stop any ongoing DMA by disconecting I/O, MMIO, and bus master.
1889 5834a83f Alex Williamson
     * Also put INTx Disable in known state.
1890 5834a83f Alex Williamson
     */
1891 5834a83f Alex Williamson
    cmd = vfio_pci_read_config(pdev, PCI_COMMAND, 2);
1892 5834a83f Alex Williamson
    cmd &= ~(PCI_COMMAND_IO | PCI_COMMAND_MEMORY | PCI_COMMAND_MASTER |
1893 5834a83f Alex Williamson
             PCI_COMMAND_INTX_DISABLE);
1894 5834a83f Alex Williamson
    vfio_pci_write_config(pdev, PCI_COMMAND, cmd, 2);
1895 5834a83f Alex Williamson
1896 5834a83f Alex Williamson
    if (vdev->reset_works) {
1897 5834a83f Alex Williamson
        if (ioctl(vdev->fd, VFIO_DEVICE_RESET)) {
1898 5834a83f Alex Williamson
            error_report("vfio: Error unable to reset physical device "
1899 5834a83f Alex Williamson
                         "(%04x:%02x:%02x.%x): %m\n", vdev->host.domain,
1900 5834a83f Alex Williamson
                         vdev->host.bus, vdev->host.slot, vdev->host.function);
1901 5834a83f Alex Williamson
        }
1902 65501a74 Alex Williamson
    }
1903 5834a83f Alex Williamson
1904 5834a83f Alex Williamson
    vfio_enable_intx(vdev);
1905 65501a74 Alex Williamson
}
1906 65501a74 Alex Williamson
1907 65501a74 Alex Williamson
static Property vfio_pci_dev_properties[] = {
1908 65501a74 Alex Williamson
    DEFINE_PROP_PCI_HOST_DEVADDR("host", VFIODevice, host),
1909 ea486926 Alex Williamson
    DEFINE_PROP_UINT32("x-intx-mmap-timeout-ms", VFIODevice,
1910 ea486926 Alex Williamson
                       intx.mmap_timeout, 1100),
1911 65501a74 Alex Williamson
    /*
1912 65501a74 Alex Williamson
     * TODO - support passed fds... is this necessary?
1913 65501a74 Alex Williamson
     * DEFINE_PROP_STRING("vfiofd", VFIODevice, vfiofd_name),
1914 65501a74 Alex Williamson
     * DEFINE_PROP_STRING("vfiogroupfd, VFIODevice, vfiogroupfd_name),
1915 65501a74 Alex Williamson
     */
1916 65501a74 Alex Williamson
    DEFINE_PROP_END_OF_LIST(),
1917 65501a74 Alex Williamson
};
1918 65501a74 Alex Williamson
1919 d9f0e638 Alex Williamson
static const VMStateDescription vfio_pci_vmstate = {
1920 d9f0e638 Alex Williamson
    .name = "vfio-pci",
1921 d9f0e638 Alex Williamson
    .unmigratable = 1,
1922 d9f0e638 Alex Williamson
};
1923 65501a74 Alex Williamson
1924 65501a74 Alex Williamson
static void vfio_pci_dev_class_init(ObjectClass *klass, void *data)
1925 65501a74 Alex Williamson
{
1926 65501a74 Alex Williamson
    DeviceClass *dc = DEVICE_CLASS(klass);
1927 65501a74 Alex Williamson
    PCIDeviceClass *pdc = PCI_DEVICE_CLASS(klass);
1928 65501a74 Alex Williamson
1929 65501a74 Alex Williamson
    dc->reset = vfio_pci_reset;
1930 65501a74 Alex Williamson
    dc->props = vfio_pci_dev_properties;
1931 d9f0e638 Alex Williamson
    dc->vmsd = &vfio_pci_vmstate;
1932 d9f0e638 Alex Williamson
    dc->desc = "VFIO-based PCI device assignment";
1933 65501a74 Alex Williamson
    pdc->init = vfio_initfn;
1934 65501a74 Alex Williamson
    pdc->exit = vfio_exitfn;
1935 65501a74 Alex Williamson
    pdc->config_read = vfio_pci_read_config;
1936 65501a74 Alex Williamson
    pdc->config_write = vfio_pci_write_config;
1937 65501a74 Alex Williamson
}
1938 65501a74 Alex Williamson
1939 65501a74 Alex Williamson
static const TypeInfo vfio_pci_dev_info = {
1940 65501a74 Alex Williamson
    .name = "vfio-pci",
1941 65501a74 Alex Williamson
    .parent = TYPE_PCI_DEVICE,
1942 65501a74 Alex Williamson
    .instance_size = sizeof(VFIODevice),
1943 65501a74 Alex Williamson
    .class_init = vfio_pci_dev_class_init,
1944 65501a74 Alex Williamson
};
1945 65501a74 Alex Williamson
1946 65501a74 Alex Williamson
static void register_vfio_pci_dev_type(void)
1947 65501a74 Alex Williamson
{
1948 65501a74 Alex Williamson
    type_register_static(&vfio_pci_dev_info);
1949 65501a74 Alex Williamson
}
1950 65501a74 Alex Williamson
1951 65501a74 Alex Williamson
type_init(register_vfio_pci_dev_type)