[dpdk-dev] [RFC 10/19] vhost-user: handle IOTLB update and invalidate requests

[Maxime Coquelin]

Vhost-user device IOTLB protocol extension introduces
VHOST_USER_IOTLB message type. The associated payload is the
vhost_iotlb_msg struct defined in Kernel, which in this was can
be either an IOTLB update or invalidate message.

On IOTLB update, the virtqueues get notified of a new entry.

Signed-off-by: Maxime Coquelin <maxime.coque...@redhat.com>
---
 lib/librte_vhost/vhost_user.c | 45 +++++++++++++++++++++++++++++++++++++++++++
 lib/librte_vhost/vhost_user.h |  1 +
 2 files changed, 46 insertions(+)

diff --git a/lib/librte_vhost/vhost_user.c b/lib/librte_vhost/vhost_user.c
index 6b41fea..225f993 100644
--- a/lib/librte_vhost/vhost_user.c
+++ b/lib/librte_vhost/vhost_user.c
@@ -48,6 +48,7 @@
 #include <rte_malloc.h>
 #include <rte_log.h>
 
+#include "iotlb.h"
 #include "vhost.h"
 #include "vhost_user.h"
 
@@ -77,6 +78,7 @@ static const char *vhost_message_str[VHOST_USER_MAX] = {
        [VHOST_USER_SEND_RARP]  = "VHOST_USER_SEND_RARP",
        [VHOST_USER_NET_SET_MTU]  = "VHOST_USER_NET_SET_MTU",
        [VHOST_USER_SET_SLAVE_REQ_FD]  = "VHOST_USER_SET_SLAVE_REQ_FD",
+       [VHOST_USER_IOTLB_MSG]  = "VHOST_USER_IOTLB_MSG",
 };
 
 static uint64_t
@@ -198,6 +200,7 @@ vhost_user_set_features(struct virtio_net *dev, uint64_t 
features)
        } else {
                dev->vhost_hlen = sizeof(struct virtio_net_hdr);
        }
+
        LOG_DEBUG(VHOST_CONFIG,
                "(%d) mergeable RX buffers %s, virtio 1 %s\n",
                dev->vid,
@@ -877,6 +880,44 @@ vhost_user_set_req_fd(struct virtio_net *dev, struct 
VhostUserMsg *msg)
        return 0;
 }
 
+static int
+vhost_user_iotlb_msg(struct virtio_net *dev, struct VhostUserMsg *msg)
+{
+       struct vhost_iotlb_msg *imsg = &msg->payload.iotlb;
+       uint16_t i;
+       uint64_t vva;
+
+       switch (imsg->type) {
+       case VHOST_IOTLB_UPDATE:
+               vva = qva_to_vva(dev, imsg->uaddr);
+               if (!vva)
+                       return -1;
+
+               for (i = 0; i < dev->nr_vring; i++) {
+                       struct vhost_virtqueue *vq = dev->virtqueue[i];
+
+                       vhost_user_iotlb_insert(vq, imsg->iova, vva,
+                                       imsg->size, imsg->perm);
+               }
+               notify_iotlb_event(dev);
+
+               break;
+       case VHOST_IOTLB_INVALIDATE:
+               for (i = 0; i < dev->nr_vring; i++) {
+                       struct vhost_virtqueue *vq = dev->virtqueue[i];
+
+                       vhost_user_iotlb_remove(vq, imsg->iova, imsg->size);
+               }
+               break;
+       default:
+               RTE_LOG(ERR, VHOST_CONFIG, "Invalid IOTLB message type (%d)\n",
+                               imsg->type);
+               return -1;
+       }
+
+       return 0;
+}
+
 /* return bytes# of read on success or negative val on failure. */
 static int
 read_vhost_message(int sockfd, struct VhostUserMsg *msg)
@@ -1110,6 +1151,10 @@ vhost_user_msg_handler(int vid, int fd)
                ret = vhost_user_set_req_fd(dev, &msg);
                break;
 
+       case VHOST_USER_IOTLB_MSG:
+               ret = vhost_user_iotlb_msg(dev, &msg);
+               break;
+
        default:
                ret = -1;
                break;
diff --git a/lib/librte_vhost/vhost_user.h b/lib/librte_vhost/vhost_user.h
index 0b2aff1..46c6ff9 100644
--- a/lib/librte_vhost/vhost_user.h
+++ b/lib/librte_vhost/vhost_user.h
@@ -80,6 +80,7 @@ typedef enum VhostUserRequest {
        VHOST_USER_SEND_RARP = 19,
        VHOST_USER_NET_SET_MTU = 20,
        VHOST_USER_SET_SLAVE_REQ_FD = 21,
+       VHOST_USER_IOTLB_MSG = 22,
        VHOST_USER_MAX
 } VhostUserRequest;
 
-- 
2.9.4