@@ -376,6 +376,19 @@ vduse_device_create(const char *path)
uint64_t features = VDUSE_NET_SUPPORTED_FEATURES;
struct vduse_dev_config *dev_config = NULL;
const char *name = path + strlen("/dev/vduse/");
+ char reconnect_dev[PATH_MAX];
+ struct vhost_reconnect_data *log = NULL;
+ struct vduse_reconnect_mmap_info mmap_info;
+ bool reconnect = false;
+
+ ret = snprintf(reconnect_dev, sizeof(reconnect_dev), "%s/%s", "/dev/vduse", name);
+ if (access(reconnect_dev, F_OK) == 0) {
+ reconnect = true;
+ VHOST_LOG_CONFIG(name, INFO, "Device already exists, reconnecting...\n");
+ } else {
+ reconnect = false;
+ VHOST_LOG_CONFIG(name, ERR, "device %s not exists, creating...\n", reconnect_dev);
+ }
/* If first device, create events dispatcher thread */
if (vduse_events_thread == false) {
@@ -407,16 +420,8 @@ vduse_device_create(const char *path)
}
if (ioctl(control_fd, VDUSE_SET_API_VERSION, &ver)) {
- VHOST_LOG_CONFIG(name, ERR, "Failed to set API version: %" PRIu64 ": %s\n",
- ver, strerror(errno));
- ret = -1;
- goto out_ctrl_close;
- }
-
- dev_config = malloc(offsetof(struct vduse_dev_config, config) +
- sizeof(vnet_config));
- if (!dev_config) {
- VHOST_LOG_CONFIG(name, ERR, "Failed to allocate VDUSE config\n");
+ VHOST_LOG_CONFIG(name, ERR, "Failed to set API version: %" PRIu64 ": %s\n", ver,
+ strerror(errno));
ret = -1;
goto out_ctrl_close;
}
@@ -424,7 +429,7 @@ vduse_device_create(const char *path)
ret = rte_vhost_driver_get_queue_num(path, &max_queue_pairs);
if (ret < 0) {
VHOST_LOG_CONFIG(name, ERR, "Failed to get max queue pairs\n");
- goto out_free;
+ goto out_ctrl_close;
}
VHOST_LOG_CONFIG(path, INFO, "VDUSE max queue pairs: %u\n", max_queue_pairs);
@@ -435,23 +440,34 @@ vduse_device_create(const char *path)
else
total_queues += 1; /* Includes ctrl queue */
- vnet_config.max_virtqueue_pairs = max_queue_pairs;
- memset(dev_config, 0, sizeof(struct vduse_dev_config));
-
- strncpy(dev_config->name, name, VDUSE_NAME_MAX - 1);
- dev_config->device_id = VIRTIO_ID_NET;
- dev_config->vendor_id = 0;
- dev_config->features = features;
- dev_config->vq_num = total_queues;
- dev_config->vq_align = sysconf(_SC_PAGE_SIZE);
- dev_config->config_size = sizeof(struct virtio_net_config);
- memcpy(dev_config->config, &vnet_config, sizeof(vnet_config));
+ if (reconnect != true) {
+ dev_config =
+ malloc(offsetof(struct vduse_dev_config, config) + sizeof(vnet_config));
+ if (!dev_config) {
+ VHOST_LOG_CONFIG(name, ERR, "Failed to allocate VDUSE config\n");
+ ret = -1;
+ goto out_ctrl_close;
+ }
- ret = ioctl(control_fd, VDUSE_CREATE_DEV, dev_config);
- if (ret < 0) {
- VHOST_LOG_CONFIG(name, ERR, "Failed to create VDUSE device: %s\n",
- strerror(errno));
- goto out_free;
+ vnet_config.max_virtqueue_pairs = max_queue_pairs;
+ memset(dev_config, 0, sizeof(struct vduse_dev_config));
+
+ strncpy(dev_config->name, name, VDUSE_NAME_MAX - 1);
+ dev_config->device_id = VIRTIO_ID_NET;
+ dev_config->vendor_id = 0;
+ dev_config->features = features;
+ dev_config->vq_num = total_queues;
+ dev_config->vq_align = sysconf(_SC_PAGE_SIZE);
+ dev_config->config_size = sizeof(struct virtio_net_config);
+ memcpy(dev_config->config, &vnet_config, sizeof(vnet_config));
+
+ ret = ioctl(control_fd, VDUSE_CREATE_DEV, dev_config);
+ free(dev_config);
+ if (ret < 0) {
+ VHOST_LOG_CONFIG(name, ERR, "Failed to create VDUSE device: %s\n",
+ strerror(errno));
+ goto out_ctrl_close;
+ }
}
dev_fd = open(path, O_RDWR);
@@ -485,10 +501,43 @@ vduse_device_create(const char *path)
strncpy(dev->ifname, path, IF_NAME_SZ - 1);
dev->vduse_ctrl_fd = control_fd;
dev->vduse_dev_fd = dev_fd;
+
+ ret = ioctl(dev_fd, VDUSE_GET_RECONNECT_INFO, &mmap_info);
+ if (ret < 0) {
+ VHOST_LOG_CONFIG(name, ERR, "Failed to get reconnect info VDUSE device: %s\n",
+ strerror(errno));
+ goto out_dev_close;
+ }
+ dev->mmap_info.size = mmap_info.size;
+ dev->mmap_info.max_index = mmap_info.max_index;
+ log = (struct vhost_reconnect_data *)mmap(NULL, mmap_info.size, PROT_READ | PROT_WRITE,
+ MAP_SHARED, dev->vduse_dev_fd, 0);
+ if (log == MAP_FAILED) {
+ VHOST_LOG_CONFIG(name, ERR, "Failed to mapping vduse reconnect data\n");
+ goto out_dev_close;
+ }
+
+ dev->log = log;
+
+ if (reconnect == true) {
+ dev->status = dev->log->status;
+ log->version = VHOST_VDUSE_API_VERSION;
+ log->reconnect_time += 1;
+ log->nr_vrings = total_queues;
+ }
+
vhost_setup_virtio_net(dev->vid, true, true, true, true);
+ if (total_queues > mmap_info.max_index - 1) {
+ VHOST_LOG_CONFIG(name, ERR, "The max vring number %d lager then %d\n", total_queues,
+ mmap_info.max_index - 1);
+ goto out_dev_close;
+ }
+
for (i = 0; i < total_queues; i++) {
struct vduse_vq_config vq_cfg = { 0 };
+ struct vhost_reconnect_vring *log_vq;
+ struct vhost_virtqueue *vq;
ret = alloc_vring_queue(dev, i);
if (ret) {
@@ -496,6 +545,22 @@ vduse_device_create(const char *path)
goto out_dev_destroy;
}
+ log_vq = (struct vhost_reconnect_vring *)mmap(NULL, mmap_info.size,
+ PROT_READ | PROT_WRITE, MAP_SHARED,
+ dev->vduse_dev_fd,
+ (i + 1) * mmap_info.size);
+ if (log_vq == MAP_FAILED) {
+ VHOST_LOG_CONFIG(name, ERR, "Failed to mapping vring %d reconnect data\n",
+ i);
+
+ goto out_dev_destroy;
+ }
+
+ vq = dev->virtqueue[i];
+ vq->log = log_vq;
+ if (reconnect)
+ continue;
+
vq_cfg.index = i;
vq_cfg.max_size = 1024;
@@ -516,7 +581,8 @@ vduse_device_create(const char *path)
}
fdset_pipe_notify(&vduse.fdset);
- free(dev_config);
+ if (reconnect && dev->status & VIRTIO_DEVICE_STATUS_DRIVER_OK)
+ vduse_device_start(dev, true);
return 0;
@@ -526,8 +592,6 @@ vduse_device_create(const char *path)
if (dev_fd >= 0)
close(dev_fd);
ioctl(control_fd, VDUSE_DESTROY_DEV, name);
-out_free:
- free(dev_config);
out_ctrl_close:
close(control_fd);
@@ -553,7 +617,29 @@ vduse_device_destroy(const char *path)
if (vid == RTE_MAX_VHOST_DEVICE)
return -1;
-
+ if (dev->log) {
+ for (uint32_t i = 0; i < dev->log->nr_vrings; i++) {
+ struct vhost_virtqueue *vq;
+
+ vq = dev->virtqueue[i];
+ if (vq->log) {
+ ret = munmap(vq->log, dev->mmap_info.size);
+ if (ret) {
+ VHOST_LOG_CONFIG(name, ERR,
+ "Failed to unmap device %s vq %d: %s\n",
+ path, i, strerror(errno));
+ ret = -1;
+ }
+ }
+ }
+ ret = munmap(dev->log, dev->mmap_info.size);
+ if (ret) {
+ VHOST_LOG_CONFIG(name, ERR, "Failed to unmap device %s dev status: %s\n",
+ path, strerror(errno));
+ ret = -1;
+ }
+ }
+ dev->log = NULL;
if (dev->cvq && dev->cvq->kickfd >= 0) {
fdset_del(&vduse.fdset, dev->cvq->kickfd);
fdset_pipe_notify(&vduse.fdset);
@@ -19,6 +19,7 @@
#include <rte_ether.h>
#include <rte_malloc.h>
#include <rte_dmadev.h>
+#include <linux/vduse.h>
#include "rte_vhost.h"
#include "vdpa_driver.h"
@@ -344,6 +345,8 @@ struct vhost_virtqueue {
struct vhost_vring_addr ring_addrs;
struct virtqueue_stats stats;
+
+ struct vhost_reconnect_vring *log;
} __rte_cache_aligned;
/* Virtio device status as per Virtio specification */
@@ -537,6 +540,9 @@ struct virtio_net {
struct rte_vhost_user_extern_ops extern_ops;
struct vhost_backend_ops *backend_ops;
+
+ struct vhost_reconnect_data *log;
+ struct vduse_reconnect_mmap_info mmap_info;
} __rte_cache_aligned;
static inline void
@@ -582,6 +588,10 @@ vq_inc_last_avail_packed(struct vhost_virtqueue *vq, uint16_t num)
vq->avail_wrap_counter ^= 1;
vq->last_avail_idx -= vq->size;
}
+ if (vq->log) {
+ vq->log->last_avail_idx = vq->last_avail_idx;
+ vq->log->avail_wrap_counter = vq->avail_wrap_counter;
+ }
}
void __vhost_log_cache_write(struct virtio_net *dev,