On Tue, Feb 19, 2019 at 11:01:45AM +0000, Dr. David Alan Gilbert wrote:
> * Yan Zhao (yan.y.zhao@xxxxxxxxx) wrote:
> > Device config is the default data that every device should have. so
> > device config capability is by default on, no need to set.
> >
> > - Currently two type of resources are saved/loaded for device of device
> > config capability:
> > General PCI config data, and Device config data.
> > They are copies as a whole when precopy is stopped.
> >
> > Migration setup flow:
> > - Setup device state regions, check its device state version and capabilities.
> > Mmap Device Config Region and Dirty Bitmap Region, if available.
> > - If device state regions are failed to get setup, a migration blocker is
> > registered instead.
> > - Added SaveVMHandlers to register device state save/load handlers.
> > - Register VM state change handler to set device's running/stop states.
> > - On migration startup on source machine, set device's state to
> > VFIO_DEVICE_STATE_LOGGING
> >
> > Signed-off-by: Yan Zhao <yan.y.zhao@xxxxxxxxx>
> > Signed-off-by: Yulei Zhang <yulei.zhang@xxxxxxxxx>
> > ---
> > hw/vfio/Makefile.objs | 2 +-
> > hw/vfio/migration.c | 633 ++++++++++++++++++++++++++++++++++++++++++
> > hw/vfio/pci.c | 1 -
> > hw/vfio/pci.h | 25 +-
> > include/hw/vfio/vfio-common.h | 1 +
> > 5 files changed, 659 insertions(+), 3 deletions(-)
> > create mode 100644 hw/vfio/migration.c
> >
> > diff --git a/hw/vfio/Makefile.objs b/hw/vfio/Makefile.objs
> > index 8b3f664..f32ff19 100644
> > --- a/hw/vfio/Makefile.objs
> > +++ b/hw/vfio/Makefile.objs
> > @@ -1,6 +1,6 @@
> > ifeq ($(CONFIG_LINUX), y)
> > obj-$(CONFIG_SOFTMMU) += common.o
> > -obj-$(CONFIG_PCI) += pci.o pci-quirks.o display.o
> > +obj-$(CONFIG_PCI) += pci.o pci-quirks.o display.o migration.o
> > obj-$(CONFIG_VFIO_CCW) += ccw.o
> > obj-$(CONFIG_SOFTMMU) += platform.o
> > obj-$(CONFIG_VFIO_XGMAC) += calxeda-xgmac.o
> > diff --git a/hw/vfio/migration.c b/hw/vfio/migration.c
> > new file mode 100644
> > index 0000000..16d6395
> > --- /dev/null
> > +++ b/hw/vfio/migration.c
> > @@ -0,0 +1,633 @@
> > +#include "qemu/osdep.h"
> > +
> > +#include "hw/vfio/vfio-common.h"
> > +#include "migration/blocker.h"
> > +#include "migration/register.h"
> > +#include "qapi/error.h"
> > +#include "pci.h"
> > +#include "sysemu/kvm.h"
> > +#include "exec/ram_addr.h"
> > +
> > +#define VFIO_SAVE_FLAG_SETUP 0
> > +#define VFIO_SAVE_FLAG_PCI 1
> > +#define VFIO_SAVE_FLAG_DEVCONFIG 2
> > +#define VFIO_SAVE_FLAG_DEVMEMORY 4
> > +#define VFIO_SAVE_FLAG_CONTINUE 8
> > +
> > +static int vfio_device_state_region_setup(VFIOPCIDevice *vdev,
> > + VFIORegion *region, uint32_t subtype, const char *name)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + struct vfio_region_info *info;
> > + int ret;
> > +
> > + ret = vfio_get_dev_region_info(vbasedev, VFIO_REGION_TYPE_DEVICE_STATE,
> > + subtype, &info);
> > + if (ret) {
> > + error_report("Failed to get info of region %s", name);
> > + return ret;
> > + }
> > +
> > + if (vfio_region_setup(OBJECT(vdev), vbasedev,
> > + region, info->index, name)) {
> > + error_report("Failed to setup migrtion region %s", name);
> > + return ret;
> > + }
> > +
> > + if (vfio_region_mmap(region)) {
> > + error_report("Failed to mmap migrtion region %s", name);
> > + }
> > +
> > + return 0;
> > +}
> > +
> > +bool vfio_device_data_cap_system_memory(VFIOPCIDevice *vdev)
> > +{
> > + return !!(vdev->migration->data_caps & VFIO_DEVICE_DATA_CAP_SYSTEM_MEMORY);
> > +}
> > +
> > +bool vfio_device_data_cap_device_memory(VFIOPCIDevice *vdev)
> > +{
> > + return !!(vdev->migration->data_caps & VFIO_DEVICE_DATA_CAP_DEVICE_MEMORY);
> > +}
> > +
> > +static bool vfio_device_state_region_mmaped(VFIORegion *region)
> > +{
> > + bool mmaped = true;
> > + if (region->nr_mmaps != 1 || region->mmaps[0].offset ||
> > + (region->size != region->mmaps[0].size) ||
> > + (region->mmaps[0].mmap == NULL)) {
> > + mmaped = false;
> > + }
> > +
> > + return mmaped;
> > +}
> > +
> > +static int vfio_get_device_config_size(VFIOPCIDevice *vdev)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region_ctl =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > + VFIORegion *region_config =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_CONFIG];
> > + uint64_t len;
> > + int sz;
> > +
> > + sz = sizeof(len);
> > + if (pread(vbasedev->fd, &len, sz,
> > + region_ctl->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, device_config.size))
> > + != sz) {
> > + error_report("vfio: Failed to get length of device config");
> > + return -1;
> > + }
> > + if (len > region_config->size) {
> > + error_report("vfio: Error device config length");
> > + return -1;
> > + }
> > + vdev->migration->devconfig_size = len;
> > +
> > + return 0;
> > +}
> > +
> > +static int vfio_set_device_config_size(VFIOPCIDevice *vdev, uint64_t size)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region_ctl =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > + VFIORegion *region_config =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_CONFIG];
> > + int sz;
> > +
> > + if (size > region_config->size) {
> > + return -1;
> > + }
> > +
> > + sz = sizeof(size);
> > + if (pwrite(vbasedev->fd, &size, sz,
> > + region_ctl->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, device_config.size))
> > + != sz) {
> > + error_report("vfio: Failed to set length of device config");
> > + return -1;
> > + }
> > + vdev->migration->devconfig_size = size;
> > + return 0;
> > +}
> > +
> > +static int vfio_save_data_device_config(VFIOPCIDevice *vdev, QEMUFile *f)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region_ctl =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > + VFIORegion *region_config =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_CONFIG];
> > + void *dest;
> > + uint32_t sz;
> > + uint8_t *buf = NULL;
> > + uint32_t action = VFIO_DEVICE_DATA_ACTION_GET_BUFFER;
> > + uint64_t len = vdev->migration->devconfig_size;
> > +
> > + qemu_put_be64(f, len);
> > +
> > + sz = sizeof(action);
> > + if (pwrite(vbasedev->fd, &action, sz,
> > + region_ctl->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, device_config.action))
> > + != sz) {
> > + error_report("vfio: action failure for device config get buffer");
> > + return -1;
> > + }
> > +
> > + if (!vfio_device_state_region_mmaped(region_config)) {
> > + buf = g_malloc(len);
> > + if (buf == NULL) {
> > + error_report("vfio: Failed to allocate memory for migrate");
> > + return -1;
> > + }
>
> g_malloc never returns NULL; it aborts on failure to allocate.
> So you can either drop the check, or my preference is to use
> g_try_malloc for large/unknown areas, and it can return NULL.
>
ok. got that. I'll use g_try_malloc next time :)
> > + if (pread(vbasedev->fd, buf, len, region_config->fd_offset) != len) {
> > + error_report("vfio: Failed read device config buffer");
> > + return -1;
> > + }
> > + qemu_put_buffer(f, buf, len);
> > + g_free(buf);
> > + } else {
> > + dest = region_config->mmaps[0].mmap;
> > + qemu_put_buffer(f, dest, len);
> > + }
> > + return 0;
> > +}
> > +
> > +static int vfio_load_data_device_config(VFIOPCIDevice *vdev,
> > + QEMUFile *f, uint64_t len)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region_ctl =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > + VFIORegion *region_config =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_CONFIG];
> > + void *dest;
> > + uint32_t sz;
> > + uint8_t *buf = NULL;
> > + uint32_t action = VFIO_DEVICE_DATA_ACTION_SET_BUFFER;
> > +
> > + vfio_set_device_config_size(vdev, len);
> > +
> > + if (!vfio_device_state_region_mmaped(region_config)) {
> > + buf = g_malloc(len);
> > + if (buf == NULL) {
> > + error_report("vfio: Failed to allocate memory for migrate");
> > + return -1;
> > + }
> > + qemu_get_buffer(f, buf, len);
> > + if (pwrite(vbasedev->fd, buf, len,
> > + region_config->fd_offset) != len) {
> > + error_report("vfio: Failed to write devie config buffer");
> > + return -1;
> > + }
> > + g_free(buf);
> > + } else {
> > + dest = region_config->mmaps[0].mmap;
> > + qemu_get_buffer(f, dest, len);
> > + }
> > +
> > + sz = sizeof(action);
> > + if (pwrite(vbasedev->fd, &action, sz,
> > + region_ctl->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, device_config.action))
> > + != sz) {
> > + error_report("vfio: action failure for device config set buffer");
> > + return -1;
> > + }
> > +
> > + return 0;
> > +}
> > +
> > +static int vfio_set_dirty_page_bitmap_chunk(VFIOPCIDevice *vdev,
> > + uint64_t start_addr, uint64_t page_nr)
> > +{
> > +
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region_ctl =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > + VFIORegion *region_bitmap =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_BITMAP];
> > + unsigned long bitmap_size =
> > + BITS_TO_LONGS(page_nr) * sizeof(unsigned long);
> > + uint32_t sz;
> > +
> > + struct {
> > + __u64 start_addr;
> > + __u64 page_nr;
> > + } system_memory;
> > + system_memory.start_addr = start_addr;
> > + system_memory.page_nr = page_nr;
> > + sz = sizeof(system_memory);
> > + if (pwrite(vbasedev->fd, &system_memory, sz,
> > + region_ctl->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, system_memory))
> > + != sz) {
> > + error_report("vfio: Failed to set system memory range for dirty pages");
> > + return -1;
> > + }
> > +
> > + if (!vfio_device_state_region_mmaped(region_bitmap)) {
> > + void *bitmap = g_malloc0(bitmap_size);
> > +
> > + if (pread(vbasedev->fd, bitmap, bitmap_size,
> > + region_bitmap->fd_offset) != bitmap_size) {
> > + error_report("vfio: Failed to read dirty bitmap data");
> > + return -1;
> > + }
> > +
> > + cpu_physical_memory_set_dirty_lebitmap(bitmap, start_addr, page_nr);
> > +
> > + g_free(bitmap);
> > + } else {
> > + cpu_physical_memory_set_dirty_lebitmap(
> > + region_bitmap->mmaps[0].mmap,
> > + start_addr, page_nr);
> > + }
> > + return 0;
> > +}
> > +
> > +int vfio_set_dirty_page_bitmap(VFIOPCIDevice *vdev,
> > + uint64_t start_addr, uint64_t page_nr)
> > +{
> > + VFIORegion *region_bitmap =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_BITMAP];
> > + unsigned long chunk_size = region_bitmap->size;
> > + uint64_t chunk_pg_nr = (chunk_size / sizeof(unsigned long)) *
> > + BITS_PER_LONG;
> > +
> > + uint64_t cnt_left;
> > + int rc = 0;
> > +
> > + cnt_left = page_nr;
> > +
> > + while (cnt_left >= chunk_pg_nr) {
> > + rc = vfio_set_dirty_page_bitmap_chunk(vdev, start_addr, chunk_pg_nr);
> > + if (rc) {
> > + goto exit;
> > + }
> > + cnt_left -= chunk_pg_nr;
> > + start_addr += start_addr;
> > + }
> > + rc = vfio_set_dirty_page_bitmap_chunk(vdev, start_addr, cnt_left);
> > +
> > +exit:
> > + return rc;
> > +}
> > +
> > +static int vfio_set_device_state(VFIOPCIDevice *vdev,
> > + uint32_t dev_state)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > + uint32_t sz = sizeof(dev_state);
> > +
> > + if (!vdev->migration) {
> > + return -1;
> > + }
> > +
> > + if (pwrite(vbasedev->fd, &dev_state, sz,
> > + region->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, device_state))
> > + != sz) {
> > + error_report("vfio: Failed to set device state %d", dev_state);
> > + return -1;
> > + }
> > + vdev->migration->device_state = dev_state;
> > + return 0;
> > +}
> > +
> > +static int vfio_get_device_data_caps(VFIOPCIDevice *vdev)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > +
> > + uint32_t caps;
> > + uint32_t size = sizeof(caps);
> > +
> > + if (pread(vbasedev->fd, &caps, size,
> > + region->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, caps))
> > + != size) {
> > + error_report("%s Failed to read data caps of device states",
> > + vbasedev->name);
> > + return -1;
> > + }
> > + vdev->migration->data_caps = caps;
> > + return 0;
> > +}
> > +
> > +
> > +static int vfio_check_devstate_version(VFIOPCIDevice *vdev)
> > +{
> > + VFIODevice *vbasedev = &vdev->vbasedev;
> > + VFIORegion *region =
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL];
> > +
> > + uint32_t version;
> > + uint32_t size = sizeof(version);
> > +
> > + if (pread(vbasedev->fd, &version, size,
> > + region->fd_offset +
> > + offsetof(struct vfio_device_state_ctl, version))
> > + != size) {
> > + error_report("%s Failed to read version of device state interfaces",
> > + vbasedev->name);
> > + return -1;
> > + }
> > +
> > + if (version != VFIO_DEVICE_STATE_INTERFACE_VERSION) {
> > + error_report("%s migration version mismatch, right version is %d",
> > + vbasedev->name, VFIO_DEVICE_STATE_INTERFACE_VERSION);
> > + return -1;
> > + }
> > +
> > + return 0;
> > +}
> > +
> > +static void vfio_vm_change_state_handler(void *pv, int running, RunState state)
> > +{
> > + VFIOPCIDevice *vdev = pv;
> > + uint32_t dev_state = vdev->migration->device_state;
> > +
> > + if (!running) {
> > + dev_state |= VFIO_DEVICE_STATE_STOP;
> > + } else {
> > + dev_state &= ~VFIO_DEVICE_STATE_STOP;
> > + }
> > +
> > + vfio_set_device_state(vdev, dev_state);
> > +}
> > +
> > +static void vfio_save_live_pending(QEMUFile *f, void *opaque,
> > + uint64_t max_size,
> > + uint64_t *res_precopy_only,
> > + uint64_t *res_compatible,
> > + uint64_t *res_post_copy_only)
> > +{
> > + VFIOPCIDevice *vdev = opaque;
> > +
> > + if (!vfio_device_data_cap_device_memory(vdev)) {
> > + return;
> > + }
> > +
> > + return;
> > +}
> > +
> > +static int vfio_save_iterate(QEMUFile *f, void *opaque)
> > +{
> > + VFIOPCIDevice *vdev = opaque;
> > +
> > + if (!vfio_device_data_cap_device_memory(vdev)) {
> > + return 0;
> > + }
> > +
> > + return 0;
> > +}
> > +
> > +static void vfio_pci_load_config(VFIOPCIDevice *vdev, QEMUFile *f)
> > +{
> > + PCIDevice *pdev = &vdev->pdev;
> > + uint32_t ctl, msi_lo, msi_hi, msi_data, bar_cfg, i;
> > + bool msi_64bit;
> > +
> > + /* retore pci bar configuration */
> > + ctl = pci_default_read_config(pdev, PCI_COMMAND, 2);
> > + vfio_pci_write_config(pdev, PCI_COMMAND,
> > + ctl & (!(PCI_COMMAND_IO | PCI_COMMAND_MEMORY)), 2);
> > + for (i = 0; i < PCI_ROM_SLOT; i++) {
> > + bar_cfg = qemu_get_be32(f);
> > + vfio_pci_write_config(pdev, PCI_BASE_ADDRESS_0 + i * 4, bar_cfg, 4);
> > + }
> > + vfio_pci_write_config(pdev, PCI_COMMAND,
> > + ctl | PCI_COMMAND_IO | PCI_COMMAND_MEMORY, 2);
> > +
> > + /* restore msi configuration */
> > + ctl = pci_default_read_config(pdev, pdev->msi_cap + PCI_MSI_FLAGS, 2);
> > + msi_64bit = !!(ctl & PCI_MSI_FLAGS_64BIT);
> > +
> > + vfio_pci_write_config(&vdev->pdev,
> > + pdev->msi_cap + PCI_MSI_FLAGS,
> > + ctl & (!PCI_MSI_FLAGS_ENABLE), 2);
> > +
> > + msi_lo = qemu_get_be32(f);
> > + vfio_pci_write_config(pdev, pdev->msi_cap + PCI_MSI_ADDRESS_LO, msi_lo, 4);
> > +
> > + if (msi_64bit) {
> > + msi_hi = qemu_get_be32(f);
> > + vfio_pci_write_config(pdev, pdev->msi_cap + PCI_MSI_ADDRESS_HI,
> > + msi_hi, 4);
> > + }
> > + msi_data = qemu_get_be32(f);
> > + vfio_pci_write_config(pdev,
> > + pdev->msi_cap + (msi_64bit ? PCI_MSI_DATA_64 : PCI_MSI_DATA_32),
> > + msi_data, 2);
> > +
> > + vfio_pci_write_config(&vdev->pdev, pdev->msi_cap + PCI_MSI_FLAGS,
> > + ctl | PCI_MSI_FLAGS_ENABLE, 2);
>
> It would probably be best to use a VMStateDescription and the macros
> for this if possible; I bet you'll want to add more fields in the future
> for example.
yes, it is also a good idea to use VMStateDescription for pci general
config data, which are saved one time in stop-and-copy phase.
But it's a little hard to maintain two type of interfaces.
Maybe use the way as what VMStateDescription did, i.e. using field list and
macros?
> Also what happens if the data read from the migration stream is bad or
> doesn't agree with this devices hardware? How does this fail?
>
right, errors in migration stream needs to be checked. I'll add the error
check.
For the hardware incompatiable issue, seems vfio_pci_write_config does not
return error if device driver returns error for this write.
Seems it needs to be addressed in somewhere else like checking
compitibility before lauching migration.
> > +}
> > +
> > +static int vfio_load_state(QEMUFile *f, void *opaque, int version_id)
> > +{
> > + VFIOPCIDevice *vdev = opaque;
> > + int flag;
> > + uint64_t len;
> > + int ret = 0;
> > +
> > + if (version_id != VFIO_DEVICE_STATE_INTERFACE_VERSION) {
> > + return -EINVAL;
> > + }
> > +
> > + do {
> > + flag = qemu_get_byte(f);
> > +
> > + switch (flag & ~VFIO_SAVE_FLAG_CONTINUE) {
> > + case VFIO_SAVE_FLAG_SETUP:
> > + break;
> > + case VFIO_SAVE_FLAG_PCI:
> > + vfio_pci_load_config(vdev, f);
> > + break;
> > + case VFIO_SAVE_FLAG_DEVCONFIG:
> > + len = qemu_get_be64(f);
> > + vfio_load_data_device_config(vdev, f, len);
> > + break;
> > + default:
> > + ret = -EINVAL;
> > + }
> > + } while (flag & VFIO_SAVE_FLAG_CONTINUE);
> > +
> > + return ret;
> > +}
> > +
> > +static void vfio_pci_save_config(VFIOPCIDevice *vdev, QEMUFile *f)
> > +{
> > + PCIDevice *pdev = &vdev->pdev;
> > + uint32_t msi_cfg, msi_lo, msi_hi, msi_data, bar_cfg, i;
> > + bool msi_64bit;
> > +
> > + for (i = 0; i < PCI_ROM_SLOT; i++) {
> > + bar_cfg = pci_default_read_config(pdev, PCI_BASE_ADDRESS_0 + i * 4, 4);
> > + qemu_put_be32(f, bar_cfg);
> > + }
> > +
> > + msi_cfg = pci_default_read_config(pdev, pdev->msi_cap + PCI_MSI_FLAGS, 2);
> > + msi_64bit = !!(msi_cfg & PCI_MSI_FLAGS_64BIT);
> > +
> > + msi_lo = pci_default_read_config(pdev,
> > + pdev->msi_cap + PCI_MSI_ADDRESS_LO, 4);
> > + qemu_put_be32(f, msi_lo);
> > +
> > + if (msi_64bit) {
> > + msi_hi = pci_default_read_config(pdev,
> > + pdev->msi_cap + PCI_MSI_ADDRESS_HI,
> > + 4);
> > + qemu_put_be32(f, msi_hi);
> > + }
> > +
> > + msi_data = pci_default_read_config(pdev,
> > + pdev->msi_cap + (msi_64bit ? PCI_MSI_DATA_64 : PCI_MSI_DATA_32),
> > + 2);
> > + qemu_put_be32(f, msi_data);
> > +
> > +}
> > +
> > +static int vfio_save_complete_precopy(QEMUFile *f, void *opaque)
> > +{
> > + VFIOPCIDevice *vdev = opaque;
> > + int rc = 0;
> > +
> > + qemu_put_byte(f, VFIO_SAVE_FLAG_PCI | VFIO_SAVE_FLAG_CONTINUE);
> > + vfio_pci_save_config(vdev, f);
> > +
> > + qemu_put_byte(f, VFIO_SAVE_FLAG_DEVCONFIG);
> > + rc += vfio_get_device_config_size(vdev);
> > + rc += vfio_save_data_device_config(vdev, f);
> > +
> > + return rc;
> > +}
> > +
> > +static int vfio_save_setup(QEMUFile *f, void *opaque)
> > +{
> > + VFIOPCIDevice *vdev = opaque;
> > + qemu_put_byte(f, VFIO_SAVE_FLAG_SETUP);
> > +
> > + vfio_set_device_state(vdev, VFIO_DEVICE_STATE_RUNNING |
> > + VFIO_DEVICE_STATE_LOGGING);
> > + return 0;
> > +}
> > +
> > +static int vfio_load_setup(QEMUFile *f, void *opaque)
> > +{
> > + return 0;
> > +}
> > +
> > +static void vfio_save_cleanup(void *opaque)
> > +{
> > + VFIOPCIDevice *vdev = opaque;
> > + uint32_t dev_state = vdev->migration->device_state;
> > +
> > + dev_state &= ~VFIO_DEVICE_STATE_LOGGING;
> > +
> > + vfio_set_device_state(vdev, dev_state);
> > +}
> > +
> > +static SaveVMHandlers savevm_vfio_handlers = {
> > + .save_setup = vfio_save_setup,
> > + .save_live_pending = vfio_save_live_pending,
> > + .save_live_iterate = vfio_save_iterate,
> > + .save_live_complete_precopy = vfio_save_complete_precopy,
> > + .save_cleanup = vfio_save_cleanup,
> > + .load_setup = vfio_load_setup,
> > + .load_state = vfio_load_state,
> > +};
> > +
> > +int vfio_migration_init(VFIOPCIDevice *vdev, Error **errp)
> > +{
> > + int ret;
> > + Error *local_err = NULL;
> > + vdev->migration = g_new0(VFIOMigration, 1);
> > +
> > + if (vfio_device_state_region_setup(vdev,
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_CTL],
> > + VFIO_REGION_SUBTYPE_DEVICE_STATE_CTL,
> > + "device-state-ctl")) {
> > + goto error;
> > + }
> > +
> > + if (vfio_check_devstate_version(vdev)) {
> > + goto error;
> > + }
> > +
> > + if (vfio_get_device_data_caps(vdev)) {
> > + goto error;
> > + }
> > +
> > + if (vfio_device_state_region_setup(vdev,
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_CONFIG],
> > + VFIO_REGION_SUBTYPE_DEVICE_STATE_DATA_CONFIG,
> > + "device-state-data-device-config")) {
> > + goto error;
> > + }
> > +
> > + if (vfio_device_data_cap_device_memory(vdev)) {
> > + error_report("No suppport of data cap device memory Yet");
> > + goto error;
> > + }
> > +
> > + if (vfio_device_data_cap_system_memory(vdev) &&
> > + vfio_device_state_region_setup(vdev,
> > + &vdev->migration->region[VFIO_DEVSTATE_REGION_DATA_BITMAP],
> > + VFIO_REGION_SUBTYPE_DEVICE_STATE_DATA_DIRTYBITMAP,
> > + "device-state-data-dirtybitmap")) {
> > + goto error;
> > + }
> > +
> > + vdev->migration->device_state = VFIO_DEVICE_STATE_RUNNING;
> > +
> > + register_savevm_live(NULL, TYPE_VFIO_PCI, -1,
> > + VFIO_DEVICE_STATE_INTERFACE_VERSION,
> > + &savevm_vfio_handlers,
> > + vdev);
> > +
> > + vdev->migration->vm_state =
> > + qemu_add_vm_change_state_handler(vfio_vm_change_state_handler, vdev);
> > +
> > + return 0;
> > +error:
> > + error_setg(&vdev->migration_blocker,
> > + "VFIO device doesn't support migration");
> > + ret = migrate_add_blocker(vdev->migration_blocker, &local_err);
> > + if (local_err) {
> > + error_propagate(errp, local_err);
> > + error_free(vdev->migration_blocker);
> > + }
> > +
> > + g_free(vdev->migration);
> > + vdev->migration = NULL;
> > +
> > + return ret;
> > +}
> > +
> > +void vfio_migration_finalize(VFIOPCIDevice *vdev)
> > +{
> > + if (vdev->migration) {
> > + int i;
> > + qemu_del_vm_change_state_handler(vdev->migration->vm_state);
> > + unregister_savevm(NULL, TYPE_VFIO_PCI, vdev);
> > + for (i = 0; i < VFIO_DEVSTATE_REGION_NUM; i++) {
> > + vfio_region_finalize(&vdev->migration->region[i]);
> > + }
> > + g_free(vdev->migration);
> > + vdev->migration = NULL;
> > + } else if (vdev->migration_blocker) {
> > + migrate_del_blocker(vdev->migration_blocker);
> > + error_free(vdev->migration_blocker);
> > + }
> > +}
> > diff --git a/hw/vfio/pci.c b/hw/vfio/pci.c
> > index c0cb1ec..b8e006b 100644
> > --- a/hw/vfio/pci.c
> > +++ b/hw/vfio/pci.c
> > @@ -37,7 +37,6 @@
> >
> > #define MSIX_CAP_LENGTH 12
> >
> > -#define TYPE_VFIO_PCI "vfio-pci"
> > #define PCI_VFIO(obj) OBJECT_CHECK(VFIOPCIDevice, obj, TYPE_VFIO_PCI)
> >
> > static void vfio_disable_interrupts(VFIOPCIDevice *vdev);
> > diff --git a/hw/vfio/pci.h b/hw/vfio/pci.h
> > index b1ae4c0..4b7b1bb 100644
> > --- a/hw/vfio/pci.h
> > +++ b/hw/vfio/pci.h
> > @@ -19,6 +19,7 @@
> > #include "qemu/event_notifier.h"
> > #include "qemu/queue.h"
> > #include "qemu/timer.h"
> > +#include "sysemu/sysemu.h"
> >
> > #define PCI_ANY_ID (~0)
> >
> > @@ -56,6 +57,21 @@ typedef struct VFIOBAR {
> > QLIST_HEAD(, VFIOQuirk) quirks;
> > } VFIOBAR;
> >
> > +enum {
> > + VFIO_DEVSTATE_REGION_CTL = 0,
> > + VFIO_DEVSTATE_REGION_DATA_CONFIG,
> > + VFIO_DEVSTATE_REGION_DATA_DEVICE_MEMORY,
> > + VFIO_DEVSTATE_REGION_DATA_BITMAP,
> > + VFIO_DEVSTATE_REGION_NUM,
> > +};
> > +typedef struct VFIOMigration {
> > + VFIORegion region[VFIO_DEVSTATE_REGION_NUM];
> > + uint32_t data_caps;
> > + uint32_t device_state;
> > + uint64_t devconfig_size;
> > + VMChangeStateEntry *vm_state;
> > +} VFIOMigration;
> > +
> > typedef struct VFIOVGARegion {
> > MemoryRegion mem;
> > off_t offset;
> > @@ -132,6 +148,8 @@ typedef struct VFIOPCIDevice {
> > VFIOBAR bars[PCI_NUM_REGIONS - 1]; /* No ROM */
> > VFIOVGA *vga; /* 0xa0000, 0x3b0, 0x3c0 */
> > void *igd_opregion;
> > + VFIOMigration *migration;
> > + Error *migration_blocker;
> > PCIHostDeviceAddress host;
> > EventNotifier err_notifier;
> > EventNotifier req_notifier;
> > @@ -198,5 +216,10 @@ int vfio_pci_igd_opregion_init(VFIOPCIDevice *vdev,
> > void vfio_display_reset(VFIOPCIDevice *vdev);
> > int vfio_display_probe(VFIOPCIDevice *vdev, Error **errp);
> > void vfio_display_finalize(VFIOPCIDevice *vdev);
> > -
> > +bool vfio_device_data_cap_system_memory(VFIOPCIDevice *vdev);
> > +bool vfio_device_data_cap_device_memory(VFIOPCIDevice *vdev);
> > +int vfio_set_dirty_page_bitmap(VFIOPCIDevice *vdev,
> > + uint64_t start_addr, uint64_t page_nr);
> > +int vfio_migration_init(VFIOPCIDevice *vdev, Error **errp);
> > +void vfio_migration_finalize(VFIOPCIDevice *vdev);
> > #endif /* HW_VFIO_VFIO_PCI_H */
> > diff --git a/include/hw/vfio/vfio-common.h b/include/hw/vfio/vfio-common.h
> > index 1b434d0..ed43613 100644
> > --- a/include/hw/vfio/vfio-common.h
> > +++ b/include/hw/vfio/vfio-common.h
> > @@ -32,6 +32,7 @@
> > #endif
> >
> > #define VFIO_MSG_PREFIX "vfio %s: "
> > +#define TYPE_VFIO_PCI "vfio-pci"
> >
> > enum {
> > VFIO_DEVICE_TYPE_PCI = 0,
> > --
> > 2.7.4
> >
> --
> Dr. David Alan Gilbert / dgilbert@xxxxxxxxxx / Manchester, UK
> _______________________________________________
> intel-gvt-dev mailing list
> intel-gvt-dev@xxxxxxxxxxxxxxxxxxxxx
> https://lists.freedesktop.org/mailman/listinfo/intel-gvt-dev
