net/mlx5: fix unlimited parsing of switch info

[dpdk.git] / drivers / net / virtio / virtio_user / vhost_user.c

diff --git a/drivers/net/virtio/virtio_user/vhost_user.c b/drivers/net/virtio/virtio_user/vhost_user.c
index e330675..ad48baf 100644 (file)
--- a/drivers/net/virtio/virtio_user/vhost_user.c
+++ b/drivers/net/virtio/virtio_user/vhost_user.c
@@ -32,6 +32,7 @@ struct vhost_user_msg {
 
 #define VHOST_USER_VERSION_MASK     0x3
 #define VHOST_USER_REPLY_MASK       (0x1 << 2)
+#define VHOST_USER_NEED_REPLY_MASK  (0x1 << 3)
        uint32_t flags;
        uint32_t size; /* the following payload size */
        union {
@@ -253,6 +254,7 @@ vhost_user_sock(struct virtio_user_dev *dev,
        struct vhost_user_msg msg;
        struct vhost_vring_file *file = 0;
        int need_reply = 0;
+       int has_reply_ack = 0;
        int fds[VHOST_MEMORY_MAX_NREGIONS];
        int fd_num = 0;
        int len;
@@ -265,6 +267,9 @@ vhost_user_sock(struct virtio_user_dev *dev,
        if (dev->is_server && vhostfd < 0)
                return -1;
 
+       if (dev->protocol_features & (1ULL << VHOST_USER_PROTOCOL_F_REPLY_ACK))
+               has_reply_ack = 1;
+
        msg.request = req;
        msg.flags = VHOST_USER_VERSION;
        msg.size = 0;
@@ -293,6 +298,9 @@ vhost_user_sock(struct virtio_user_dev *dev,
                msg.size = sizeof(m.payload.memory.nregions);
                msg.size += sizeof(m.payload.memory.padding);
                msg.size += fd_num * sizeof(struct vhost_memory_region);
+
+               if (has_reply_ack)
+                       msg.flags |= VHOST_USER_NEED_REPLY_MASK;
                break;
 
        case VHOST_USER_SET_LOG_FD:
@@ -341,7 +349,7 @@ vhost_user_sock(struct virtio_user_dev *dev,
                return -1;
        }
 
-       if (need_reply) {
+       if (need_reply || msg.flags & VHOST_USER_NEED_REPLY_MASK) {
                if (vhost_user_read(vhostfd, &msg) < 0) {
                        PMD_DRV_LOG(ERR, "Received msg failed: %s",
                                    strerror(errno));
@@ -371,8 +379,18 @@ vhost_user_sock(struct virtio_user_dev *dev,
                               sizeof(struct vhost_vring_state));
                        break;
                default:
-                       PMD_DRV_LOG(ERR, "Received unexpected msg type");
-                       return -1;
+                       /* Reply-ack handling */
+                       if (msg.size != sizeof(m.payload.u64)) {
+                               PMD_DRV_LOG(ERR, "Received bad msg size");
+                               return -1;
+                       }
+
+                       if (msg.payload.u64 != 0) {
+                               PMD_DRV_LOG(ERR, "Slave replied NACK");
+                               return -1;
+                       }
+
+                       break;
                }
        }