virtio_net: Add a set_rx_mode interface

Make use of the RX_MODE control virtqueue class to enable the
set_rx_mode netdev interface.  This allows us to selectively
enable/disable promiscuous and allmulti mode so we don't see
packets we don't want.  For now, we automatically enable these
as needed if additional unicast or multicast addresses are
requested.

Signed-off-by: Alex Williamson <alex.williamson@hp.com>
Acked-by: Rusty Russell <rusty@rustcorp.com.au>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 33 insertions, 1 deletions

diff --git a/drivers/net/virtio_net.c b/drivers/net/virtio_net.c
index 67bb583b7fc9..1abea9dc6f0f 100644
--- a/drivers/net/virtio_net.c
+++ b/drivers/net/virtio_net.c
@@ -37,7 +37,7 @@ module_param(gso, bool, 0444);
 #define MAX_PACKET_LEN (ETH_HLEN + VLAN_HLEN + ETH_DATA_LEN)
 #define GOOD_COPY_LEN   128
 
-#define VIRTNET_SEND_COMMAND_SG_MAX    0
+#define VIRTNET_SEND_COMMAND_SG_MAX    1
 
 struct virtnet_info
 {
@@ -658,6 +658,36 @@ static int virtnet_set_tx_csum(struct net_device *dev, u32 data)
         return ethtool_op_set_tx_hw_csum(dev, data);
 }
 
+static void virtnet_set_rx_mode(struct net_device *dev)
+{
+        struct virtnet_info *vi = netdev_priv(dev);
+        struct scatterlist sg;
+        u8 promisc, allmulti;
+
+        /* We can't dynamicaly set ndo_set_rx_mode, so return gracefully */
+        if (!virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_RX))
+                return;
+
+        promisc = ((dev->flags & IFF_PROMISC) != 0 || dev->uc_count > 0);
+        allmulti = ((dev->flags & IFF_ALLMULTI) != 0 || dev->mc_count > 0);
+
+        sg_set_buf(&sg, &promisc, sizeof(promisc));
+
+        if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_RX,
+                                  VIRTIO_NET_CTRL_RX_PROMISC,
+                                  &sg, 1, 0))
+                dev_warn(&dev->dev, "Failed to %sable promisc mode.\n",
+                         promisc ? "en" : "dis");
+
+        sg_set_buf(&sg, &allmulti, sizeof(allmulti));
+
+        if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_RX,
+                                  VIRTIO_NET_CTRL_RX_ALLMULTI,
+                                  &sg, 1, 0))
+                dev_warn(&dev->dev, "Failed to %sable allmulti mode.\n",
+                         allmulti ? "en" : "dis");
+}
+
 static struct ethtool_ops virtnet_ethtool_ops = {
         .set_tx_csum = virtnet_set_tx_csum,
         .set_sg = ethtool_op_set_sg,
@@ -682,6 +712,7 @@ static const struct net_device_ops virtnet_netdev = {
         .ndo_start_xmit      = start_xmit,
         .ndo_validate_addr   = eth_validate_addr,
         .ndo_set_mac_address = eth_mac_addr,
+        .ndo_set_rx_mode     = virtnet_set_rx_mode,
         .ndo_change_mtu      = virtnet_change_mtu,
 #ifdef CONFIG_NET_POLL_CONTROLLER
         .ndo_poll_controller = virtnet_netpoll,
@@ -897,6 +928,7 @@ static unsigned int features[] = {
         VIRTIO_NET_F_HOST_ECN, VIRTIO_NET_F_GUEST_TSO4, VIRTIO_NET_F_GUEST_TSO6,
         VIRTIO_NET_F_GUEST_ECN, /* We don't yet handle UFO input. */
         VIRTIO_NET_F_MRG_RXBUF, VIRTIO_NET_F_STATUS, VIRTIO_NET_F_CTRL_VQ,
+        VIRTIO_NET_F_CTRL_RX,
         VIRTIO_F_NOTIFY_ON_EMPTY,
 };