[dpdk-dev] [PATCH 5/7] examples/vhost: handle broadcast packet

[Yuanhan Liu]

Every time I do a VM2VM iperf test with vhost example, I have to set
the arp table manually, as vhost-switch just ignores the broadcast
packet, leaving the ARP request not served.

Here we do a transmit a broadcast packet (such as ARP request) to
every vhost device, as well as the physical port, to fix above
arp table issue.

Cc: "Xu, Qian Q" <qian.q.xu at intel.com>
Signed-off-by: Yuanhan Liu <yuanhan.liu at linux.intel.com>
---
 examples/vhost/main.c | 39 +++++++++++++++++++++++++++++----------
 1 file changed, 29 insertions(+), 10 deletions(-)

diff --git a/examples/vhost/main.c b/examples/vhost/main.c
index 6cc34b7..16ba216 100644
--- a/examples/vhost/main.c
+++ b/examples/vhost/main.c
@@ -833,6 +833,21 @@ unlink_vmdq(struct vhost_dev *vdev)
 	}
 }
 
+static inline void __attribute__((always_inline))
+virtio_xmit(struct virtio_net *dst_dev, struct virtio_net *src_dev,
+	    struct rte_mbuf *m)
+{
+	uint16_t ret;
+
+	ret = rte_vhost_enqueue_burst(dst_dev, VIRTIO_RXQ, &m, 1);
+	if (enable_stats) {
+		rte_atomic64_inc(&dev_statistics[dst_dev->device_fh].rx_total_atomic);
+		rte_atomic64_add(&dev_statistics[dst_dev->device_fh].rx_atomic, ret);
+		dev_statistics[src_dev->device_fh].tx_total++;
+		dev_statistics[src_dev->device_fh].tx += ret;
+	}
+}
+
 /*
  * Check if the packet destination MAC address is for a local device. If so then put
  * the packet on that devices RX queue. If not then return.
@@ -841,7 +856,6 @@ static inline int __attribute__((always_inline))
 virtio_tx_local(struct vhost_dev *vdev, struct rte_mbuf *m)
 {
 	struct ether_hdr *pkt_hdr;
-	uint64_t ret = 0;
 	struct virtio_net *dev = vdev->dev;
 	struct virtio_net *tdev; /* destination virito device */
 	struct vhost_dev *vdev2;
@@ -872,15 +886,7 @@ virtio_tx_local(struct vhost_dev *vdev, struct rte_mbuf *m)
 		return 0;
 	}
 
-	/* send the packet to the local virtio device */
-	ret = rte_vhost_enqueue_burst(tdev, VIRTIO_RXQ, &m, 1);
-	if (enable_stats) {
-		rte_atomic64_inc(&dev_statistics[fh].rx_total_atomic);
-		rte_atomic64_add(&dev_statistics[fh].rx_atomic, ret);
-		dev_statistics[vdev->dev->device_fh].tx_total++;
-		dev_statistics[vdev->dev->device_fh].tx += ret;
-	}
-
+	virtio_xmit(tdev, vdev->dev, m);
 	return 0;
 }
 
@@ -964,6 +970,17 @@ virtio_tx_route(struct vhost_dev *vdev, struct rte_mbuf *m, uint16_t vlan_tag)
 	struct virtio_net *dev = vdev->dev;
 	struct ether_hdr *nh;
 
+
+	nh = rte_pktmbuf_mtod(m, struct ether_hdr *);
+	if (unlikely(is_broadcast_ether_addr(&nh->d_addr))) {
+		struct vhost_dev *vdev2;
+
+		TAILQ_FOREACH(vdev2, &vhost_dev_list, next) {
+			virtio_xmit(vdev2->dev, vdev->dev, m);
+		}
+		goto queue2nic;
+	}
+
 	/*check if destination is local VM*/
 	if ((vm2vm_mode == VM2VM_SOFTWARE) && (virtio_tx_local(vdev, m) == 0)) {
 		rte_pktmbuf_free(m);
@@ -980,6 +997,8 @@ virtio_tx_route(struct vhost_dev *vdev, struct rte_mbuf *m, uint16_t vlan_tag)
 	RTE_LOG(DEBUG, VHOST_DATA, "(%" PRIu64 ") TX: "
 		"MAC address is external\n", dev->device_fh);
 
+queue2nic:
+
 	/*Add packet to the port tx queue*/
 	tx_q = &lcore_tx_queue[lcore_id];
 	len = tx_q->len;
-- 
1.9.0