From 550c9d27d1438c30a657016a0281fe8e7a133e53 Mon Sep 17 00:00:00 2001 From: Yuanhan Liu Date: Fri, 29 Apr 2016 11:59:47 +0800 Subject: [PATCH] vhost: set/reset device flags internally It does not make sense to ask the application to set/unset the flag VIRTIO_DEV_RUNNING (that used internal only) at new_device()/ destroy_device() callback. Instead, it should be set after new_device() succeeds and reset before destroy_device() is invoked inside vhost lib. This patch fixes it. Signed-off-by: Yuanhan Liu Tested-by: Rich Lane Acked-by: Rich Lane --- drivers/net/vhost/rte_eth_vhost.c | 2 -- examples/vhost/main.c | 3 --- lib/librte_vhost/vhost_user/virtio-net-user.c | 11 ++++++---- lib/librte_vhost/virtio-net.c | 21 ++++++++++++------- 4 files changed, 21 insertions(+), 16 deletions(-) diff --git a/drivers/net/vhost/rte_eth_vhost.c b/drivers/net/vhost/rte_eth_vhost.c index 310cbefcf2..63538c17e2 100644 --- a/drivers/net/vhost/rte_eth_vhost.c +++ b/drivers/net/vhost/rte_eth_vhost.c @@ -278,7 +278,6 @@ new_device(struct virtio_net *dev) for (i = 0; i < dev->virt_qp_nb * VIRTIO_QNUM; i++) rte_vhost_enable_guest_notification(dev, i, 0); - dev->flags |= VIRTIO_DEV_RUNNING; dev->priv = eth_dev; eth_dev->data->dev_link.link_status = ETH_LINK_UP; @@ -341,7 +340,6 @@ destroy_device(volatile struct virtio_net *dev) eth_dev->data->dev_link.link_status = ETH_LINK_DOWN; dev->priv = NULL; - dev->flags &= ~VIRTIO_DEV_RUNNING; for (i = 0; i < eth_dev->data->nb_rx_queues; i++) { vq = eth_dev->data->rx_queues[i]; diff --git a/examples/vhost/main.c b/examples/vhost/main.c index 671a7ebda8..1a0fe49ce0 100644 --- a/examples/vhost/main.c +++ b/examples/vhost/main.c @@ -1180,8 +1180,6 @@ destroy_device (volatile struct virtio_net *dev) struct vhost_dev *vdev; int lcore; - dev->flags &= ~VIRTIO_DEV_RUNNING; - vdev = (struct vhost_dev *)dev->priv; /*set the remove flag. */ vdev->remove = 1; @@ -1261,7 +1259,6 @@ new_device (struct virtio_net *dev) /* Disable notifications. */ rte_vhost_enable_guest_notification(dev, VIRTIO_RXQ, 0); rte_vhost_enable_guest_notification(dev, VIRTIO_TXQ, 0); - dev->flags |= VIRTIO_DEV_RUNNING; RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Device has been added to data core %d\n", dev->device_fh, vdev->coreid); diff --git a/lib/librte_vhost/vhost_user/virtio-net-user.c b/lib/librte_vhost/vhost_user/virtio-net-user.c index f5248bc427..e775e45210 100644 --- a/lib/librte_vhost/vhost_user/virtio-net-user.c +++ b/lib/librte_vhost/vhost_user/virtio-net-user.c @@ -115,8 +115,10 @@ user_set_mem_table(struct vhost_device_ctx ctx, struct VhostUserMsg *pmsg) return -1; /* Remove from the data plane. */ - if (dev->flags & VIRTIO_DEV_RUNNING) + if (dev->flags & VIRTIO_DEV_RUNNING) { + dev->flags &= ~VIRTIO_DEV_RUNNING; notify_ops->destroy_device(dev); + } if (dev->mem) { free_mem_region(dev); @@ -286,9 +288,10 @@ user_set_vring_kick(struct vhost_device_ctx ctx, struct VhostUserMsg *pmsg) "vring kick idx:%d file:%d\n", file.index, file.fd); vhost_set_vring_kick(ctx, &file); - if (virtio_is_ready(dev) && - !(dev->flags & VIRTIO_DEV_RUNNING)) - notify_ops->new_device(dev); + if (virtio_is_ready(dev) && !(dev->flags & VIRTIO_DEV_RUNNING)) { + if (notify_ops->new_device(dev) == 0) + dev->flags |= VIRTIO_DEV_RUNNING; + } } /* diff --git a/lib/librte_vhost/virtio-net.c b/lib/librte_vhost/virtio-net.c index 1a6259bd04..c45ed1c5e3 100644 --- a/lib/librte_vhost/virtio-net.c +++ b/lib/librte_vhost/virtio-net.c @@ -296,8 +296,10 @@ vhost_destroy_device(struct vhost_device_ctx ctx) if (dev == NULL) return; - if (dev->flags & VIRTIO_DEV_RUNNING) + if (dev->flags & VIRTIO_DEV_RUNNING) { + dev->flags &= ~VIRTIO_DEV_RUNNING; notify_ops->destroy_device(dev); + } cleanup_device(dev, 1); free_device(dev); @@ -353,8 +355,10 @@ vhost_reset_owner(struct vhost_device_ctx ctx) if (dev == NULL) return -1; - if (dev->flags & VIRTIO_DEV_RUNNING) + if (dev->flags & VIRTIO_DEV_RUNNING) { + dev->flags &= ~VIRTIO_DEV_RUNNING; notify_ops->destroy_device(dev); + } cleanup_device(dev, 0); reset_device(dev); @@ -719,12 +723,15 @@ vhost_set_backend(struct vhost_device_ctx ctx, struct vhost_vring_file *file) if (!(dev->flags & VIRTIO_DEV_RUNNING)) { if (dev->virtqueue[VIRTIO_TXQ]->backend != VIRTIO_DEV_STOPPED && dev->virtqueue[VIRTIO_RXQ]->backend != VIRTIO_DEV_STOPPED) { - return notify_ops->new_device(dev); + if (notify_ops->new_device(dev) < 0) + return -1; + dev->flags |= VIRTIO_DEV_RUNNING; } - /* Otherwise we remove it. */ - } else - if (file->fd == VIRTIO_DEV_STOPPED) - notify_ops->destroy_device(dev); + } else if (file->fd == VIRTIO_DEV_STOPPED) { + dev->flags &= ~VIRTIO_DEV_RUNNING; + notify_ops->destroy_device(dev); + } + return 0; } -- 2.20.1