This patch makes mtty device migration capable. Purpose od this code is
to test migration interface. Only stop-and-copy phase is implemented.
Postcopy migration is not supported.
Actual data for mtty device migration is very less. Appended dummy data to
migration data stream, default 100 Mbytes. Added sysfs file
'dummy_data_size_MB' to get dummy data size from user which can be used
to check performance of based of data size. During resuming dummy data is
read and discarded.
Signed-off-by: Kirti Wankhede <kwankhede@xxxxxxxxxx>
---
samples/vfio-mdev/mtty.c | 602 ++++++++++++++++++++++++++++++++++++++++++++---
1 file changed, 574 insertions(+), 28 deletions(-)
diff --git a/samples/vfio-mdev/mtty.c b/samples/vfio-mdev/mtty.c
index bf666cce5bb7..f9194234fc6a 100644
--- a/samples/vfio-mdev/mtty.c
+++ b/samples/vfio-mdev/mtty.c
@@ -44,9 +44,23 @@
#define MTTY_STRING_LEN 16
-#define MTTY_CONFIG_SPACE_SIZE 0xff
-#define MTTY_IO_BAR_SIZE 0x8
-#define MTTY_MMIO_BAR_SIZE 0x100000
+#define MTTY_CONFIG_SPACE_SIZE 0xff
+#define MTTY_IO_BAR_SIZE 0x8
+#define MTTY_MMIO_BAR_SIZE 0x100000
+#define MTTY_MIGRATION_REGION_SIZE 0x1000000 // 16M
+
+#define MTTY_MIGRATION_REGION_INDEX VFIO_PCI_NUM_REGIONS
+#define MTTY_REGIONS_MAX (MTTY_MIGRATION_REGION_INDEX + 1)
+
+/* Data section start from page aligned offset */
+#define MTTY_MIGRATION_REGION_DATA_OFFSET (0x1000)
+
+/* First page is used for struct vfio_device_migration_info */
+#define MTTY_MIGRATION_REGION_SIZE_MMAP \
+ (MTTY_MIGRATION_REGION_SIZE - MTTY_MIGRATION_REGION_DATA_OFFSET)
+
+#define MIGRATION_INFO_OFFSET(MEMBER) \
+ offsetof(struct vfio_device_migration_info, MEMBER)
#define STORE_LE16(addr, val) (*(u16 *)addr = val)
#define STORE_LE32(addr, val) (*(u32 *)addr = val)
@@ -129,6 +143,28 @@ struct serial_port {
u8 intr_trigger_level; /* interrupt trigger level */
};
+/* Migration packet */
+#define PACKET_ID (u16)(0xfeedbaba)
+
+#define PACKET_FLAGS_ACTUAL_DATA (1 << 0)
+#define PACKET_FLAGS_DUMMY_DATA (1 << 1)
+
+#define PACKET_DATA_SIZE_MAX (8 * 1024 * 1024)
+
+struct packet {
+ u16 id;
+ u16 flags;
+ u32 data_size;
+ u8 data[];
+};
+
+enum {
+ PACKET_STATE_NONE = 0,
+ PACKET_STATE_PREPARED,
+ PACKET_STATE_COPIED,
+ PACKET_STATE_LAST,
+};
+
/* State of each mdev device */
struct mdev_state {
int irq_fd;
@@ -138,22 +174,37 @@ struct mdev_state {
u8 *vconfig;
struct mutex ops_lock;
struct mdev_device *mdev;
- struct mdev_region_info region_info[VFIO_PCI_NUM_REGIONS];
- u32 bar_mask[VFIO_PCI_NUM_REGIONS];
+ struct mdev_region_info region_info[MTTY_REGIONS_MAX];
+ u32 bar_mask[MTTY_REGIONS_MAX];
struct list_head next;
struct serial_port s[2];
struct mutex rxtx_lock;
struct vfio_device_info dev_info;
- int nr_ports;
+ u32 nr_ports;
/* List of pinned gpfns, gpfn as index and content is translated hpfn */
unsigned long *gpfn_to_hpfn;
struct notifier_block nb;
+
+ u32 device_state;
+ u64 saved_size;
+ void *mig_region_base;
+ bool is_actual_data_sent;
+ struct packet *pkt;
+ u32 packet_state;
+ u64 dummy_data_size;
};
static struct mutex mdev_list_lock;
static struct list_head mdev_devices_list;
+/*
+ * Default dummy data size set to 100 MB. To change value of dummy data size at
+ * runtime but before migration write size in MB to sysfs file
+ * dummy_data_size_MB
+ */
+static unsigned long user_dummy_data_size = (100 * 1024 * 1024);
+
static const struct file_operations vd_fops = {
.owner = THIS_MODULE,
};
@@ -639,6 +690,288 @@ static void mdev_read_base(struct mdev_state *mdev_state)
}
}
+static int save_setup(struct mdev_state *mdev_state)
+{
+ mdev_state->is_actual_data_sent = false;
+
+ memset(mdev_state->pkt, 0, sizeof(struct packet) +
+ PACKET_DATA_SIZE_MAX);
+
+ return 0;
+}
+
+static int set_device_state(struct mdev_state *mdev_state, u32 device_state)
+{
+ int ret = 0;
+
+ if (mdev_state->device_state == device_state)
+ return 0;
+
+ if (device_state & VFIO_DEVICE_STATE_RUNNING) {
+#if defined(DEBUG)
+ if (device_state & VFIO_DEVICE_STATE_SAVING) {
+ pr_info("%s: %s Pre-copy\n", __func__,
+ dev_name(mdev_dev(mdev_state->mdev)));
+ } else
+ pr_info("%s: %s Running\n", __func__,
+ dev_name(mdev_dev(mdev_state->mdev)));
+#endif
+ } else {
+ if (device_state & VFIO_DEVICE_STATE_SAVING) {
+#if defined(DEBUG)
+ pr_info("%s: %s Stop-n-copy\n", __func__,
+ dev_name(mdev_dev(mdev_state->mdev)));
+#endif
+ ret = save_setup(mdev_state);
+
+ } else if (device_state & VFIO_DEVICE_STATE_RESUMING) {
+#if defined(DEBUG)
+ pr_info("%s: %s Resuming\n", __func__,
+ dev_name(mdev_dev(mdev_state->mdev)));
+ } else {
+ pr_info("%s: %s Stopped\n", __func__,
+ dev_name(mdev_dev(mdev_state->mdev)));
+#endif
+ }
+ }
+
+ mdev_state->device_state = device_state;
+
+ return ret;
+}
+
+static u32 get_device_state(struct mdev_state *mdev_state)
+{
+ return mdev_state->device_state;
+}
+
+static void write_to_packet(struct packet *pkt, u8 *data, size_t size)
+{
+ if ((pkt->data_size + size) > PACKET_DATA_SIZE_MAX) {
+ pr_err("%s: packet data overflow\n", __func__);
+ return;
+ }
+ memcpy((void *)&pkt->data[pkt->data_size], (void *)data, size);
+ pkt->data_size += size;
+}
+
+static void read_from_packet(struct packet *pkt, u8 *data,
+ int index, size_t size)
+{
+ if ((index + size) > PACKET_DATA_SIZE_MAX) {
+ pr_err("%s: packet data overflow\n", __func__);
+ return;
+ }
+
+ memcpy((void *)data, (void *)&pkt->data[index], size);
+}
+
+static int save_device_data(struct mdev_state *mdev_state, u64 *pending)
+{
+ /* Save device data only during stop-and-copy phase */
+ if (mdev_state->device_state != VFIO_DEVICE_STATE_SAVING) {
+ *pending = 0;
+ return 0;
+ }
+
+ if (mdev_state->packet_state == PACKET_STATE_PREPARED) {
+ *pending = sizeof(struct packet) + mdev_state->pkt->data_size;
+ return 0;
+ }
+
+ if (!mdev_state->is_actual_data_sent) {
+
+ /* create actual data packet */
+ write_to_packet(mdev_state->pkt, (u8 *)&mdev_state->nr_ports,
+ sizeof(mdev_state->nr_ports));
+ write_to_packet(mdev_state->pkt, (u8 *)&mdev_state->s,
+ sizeof(struct serial_port) * 2);
+
+ write_to_packet(mdev_state->pkt, mdev_state->vconfig,
+ MTTY_CONFIG_SPACE_SIZE);
+
+ write_to_packet(mdev_state->pkt, (u8 *)mdev_state->gpfn_to_hpfn,
+ sizeof(unsigned long) * MAX_GPFN_COUNT);
+
+ mdev_state->pkt->id = PACKET_ID;
+ mdev_state->pkt->flags = PACKET_FLAGS_ACTUAL_DATA;
+
+ mdev_state->is_actual_data_sent = true;
+ } else {
+ /* create dummy data packet */
+ if (mdev_state->dummy_data_size > user_dummy_data_size) {
+ *pending = 0;
+ mdev_state->packet_state = PACKET_STATE_NONE;
+ return 0;
+ }
+
+ memset(mdev_state->pkt->data, 0xa5, PACKET_DATA_SIZE_MAX);
+
+ mdev_state->pkt->id = PACKET_ID;
+ mdev_state->pkt->flags = PACKET_FLAGS_DUMMY_DATA;
+ mdev_state->pkt->data_size = PACKET_DATA_SIZE_MAX;
+ mdev_state->dummy_data_size += PACKET_DATA_SIZE_MAX;
+ }
+
+ *pending = sizeof(struct packet) + mdev_state->pkt->data_size;
+ mdev_state->packet_state = PACKET_STATE_PREPARED;
+ mdev_state->saved_size = 0;
+
+ return 0;
+}
+
+static int copy_device_data(struct mdev_state *mdev_state)
+{
+ u64 size;
+
+ if (!mdev_state->pkt || !mdev_state->mig_region_base)
+ return -EINVAL;
+
+ if (mdev_state->packet_state == PACKET_STATE_COPIED)
+ return 0;
+
+ if (!mdev_state->pkt->data_size)
+ return 0;
+
+ size = sizeof(struct packet) + mdev_state->pkt->data_size;
+
+ memcpy(mdev_state->mig_region_base, mdev_state->pkt, size);
+
