Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net
[linux-block.git] / net / batman-adv / multicast.c
index d70640135e3ab2a925c8eab23ef918f6e81c016a..de3a055f7dd8af8b8a4678bb2bc4768ed0c51acf 100644 (file)
@@ -1,5 +1,5 @@
 // SPDX-License-Identifier: GPL-2.0
-/* Copyright (C) 2014-2017  B.A.T.M.A.N. contributors:
+/* Copyright (C) 2014-2018  B.A.T.M.A.N. contributors:
  *
  * Linus Lüssing
  *
@@ -40,6 +40,7 @@
 #include <linux/list.h>
 #include <linux/lockdep.h>
 #include <linux/netdevice.h>
+#include <linux/netlink.h>
 #include <linux/printk.h>
 #include <linux/rculist.h>
 #include <linux/rcupdate.h>
 #include <linux/types.h>
 #include <linux/workqueue.h>
 #include <net/addrconf.h>
+#include <net/genetlink.h>
 #include <net/if_inet6.h>
 #include <net/ip.h>
 #include <net/ipv6.h>
+#include <net/netlink.h>
+#include <net/sock.h>
 #include <uapi/linux/batadv_packet.h>
+#include <uapi/linux/batman_adv.h>
 
 #include "hard-interface.h"
 #include "hash.h"
 #include "log.h"
+#include "netlink.h"
+#include "soft-interface.h"
 #include "translation-table.h"
 #include "tvlv.h"
 
@@ -101,8 +108,37 @@ static struct net_device *batadv_mcast_get_bridge(struct net_device *soft_iface)
        return upper;
 }
 
+/**
+ * batadv_mcast_addr_is_ipv4() - check if multicast MAC is IPv4
+ * @addr: the MAC address to check
+ *
+ * Return: True, if MAC address is one reserved for IPv4 multicast, false
+ * otherwise.
+ */
+static bool batadv_mcast_addr_is_ipv4(const u8 *addr)
+{
+       static const u8 prefix[] = {0x01, 0x00, 0x5E};
+
+       return memcmp(prefix, addr, sizeof(prefix)) == 0;
+}
+
+/**
+ * batadv_mcast_addr_is_ipv6() - check if multicast MAC is IPv6
+ * @addr: the MAC address to check
+ *
+ * Return: True, if MAC address is one reserved for IPv6 multicast, false
+ * otherwise.
+ */
+static bool batadv_mcast_addr_is_ipv6(const u8 *addr)
+{
+       static const u8 prefix[] = {0x33, 0x33};
+
+       return memcmp(prefix, addr, sizeof(prefix)) == 0;
+}
+
 /**
  * batadv_mcast_mla_softif_get() - get softif multicast listeners
+ * @bat_priv: the bat priv with all the soft interface information
  * @dev: the device to collect multicast addresses from
  * @mcast_list: a list to put found addresses into
  *
@@ -119,9 +155,12 @@ static struct net_device *batadv_mcast_get_bridge(struct net_device *soft_iface)
  * Return: -ENOMEM on memory allocation error or the number of
  * items added to the mcast_list otherwise.
  */
-static int batadv_mcast_mla_softif_get(struct net_device *dev,
+static int batadv_mcast_mla_softif_get(struct batadv_priv *bat_priv,
+                                      struct net_device *dev,
                                       struct hlist_head *mcast_list)
 {
+       bool all_ipv4 = bat_priv->mcast.flags & BATADV_MCAST_WANT_ALL_IPV4;
+       bool all_ipv6 = bat_priv->mcast.flags & BATADV_MCAST_WANT_ALL_IPV6;
        struct net_device *bridge = batadv_mcast_get_bridge(dev);
        struct netdev_hw_addr *mc_list_entry;
        struct batadv_hw_addr *new;
@@ -129,6 +168,12 @@ static int batadv_mcast_mla_softif_get(struct net_device *dev,
 
        netif_addr_lock_bh(bridge ? bridge : dev);
        netdev_for_each_mc_addr(mc_list_entry, bridge ? bridge : dev) {
+               if (all_ipv4 && batadv_mcast_addr_is_ipv4(mc_list_entry->addr))
+                       continue;
+
+               if (all_ipv6 && batadv_mcast_addr_is_ipv6(mc_list_entry->addr))
+                       continue;
+
                new = kmalloc(sizeof(*new), GFP_ATOMIC);
                if (!new) {
                        ret = -ENOMEM;
@@ -193,6 +238,7 @@ static void batadv_mcast_mla_br_addr_cpy(char *dst, const struct br_ip *src)
 
 /**
  * batadv_mcast_mla_bridge_get() - get bridged-in multicast listeners
+ * @bat_priv: the bat priv with all the soft interface information
  * @dev: a bridge slave whose bridge to collect multicast addresses from
  * @mcast_list: a list to put found addresses into
  *
@@ -204,10 +250,13 @@ static void batadv_mcast_mla_br_addr_cpy(char *dst, const struct br_ip *src)
  * Return: -ENOMEM on memory allocation error or the number of
  * items added to the mcast_list otherwise.
  */
-static int batadv_mcast_mla_bridge_get(struct net_device *dev,
+static int batadv_mcast_mla_bridge_get(struct batadv_priv *bat_priv,
+                                      struct net_device *dev,
                                       struct hlist_head *mcast_list)
 {
        struct list_head bridge_mcast_list = LIST_HEAD_INIT(bridge_mcast_list);
+       bool all_ipv4 = bat_priv->mcast.flags & BATADV_MCAST_WANT_ALL_IPV4;
+       bool all_ipv6 = bat_priv->mcast.flags & BATADV_MCAST_WANT_ALL_IPV6;
        struct br_ip_list *br_ip_entry, *tmp;
        struct batadv_hw_addr *new;
        u8 mcast_addr[ETH_ALEN];
@@ -221,6 +270,12 @@ static int batadv_mcast_mla_bridge_get(struct net_device *dev,
                goto out;
 
        list_for_each_entry(br_ip_entry, &bridge_mcast_list, list) {
+               if (all_ipv4 && br_ip_entry->addr.proto == htons(ETH_P_IP))
+                       continue;
+
+               if (all_ipv6 && br_ip_entry->addr.proto == htons(ETH_P_IPV6))
+                       continue;
+
                batadv_mcast_mla_br_addr_cpy(mcast_addr, &br_ip_entry->addr);
                if (batadv_mcast_mla_is_duplicate(mcast_addr, mcast_list))
                        continue;
@@ -568,11 +623,11 @@ static void __batadv_mcast_mla_update(struct batadv_priv *bat_priv)
        if (!batadv_mcast_mla_tvlv_update(bat_priv))
                goto update;
 
-       ret = batadv_mcast_mla_softif_get(soft_iface, &mcast_list);
+       ret = batadv_mcast_mla_softif_get(bat_priv, soft_iface, &mcast_list);
        if (ret < 0)
                goto out;
 
-       ret = batadv_mcast_mla_bridge_get(soft_iface, &mcast_list);
+       ret = batadv_mcast_mla_bridge_get(bat_priv, soft_iface, &mcast_list);
        if (ret < 0)
                goto out;
 
@@ -1285,6 +1340,236 @@ int batadv_mcast_flags_seq_print_text(struct seq_file *seq, void *offset)
 }
 #endif
 
+/**
+ * batadv_mcast_mesh_info_put() - put multicast info into a netlink message
+ * @msg: buffer for the message
+ * @bat_priv: the bat priv with all the soft interface information
+ *
+ * Return: 0 or error code.
+ */
+int batadv_mcast_mesh_info_put(struct sk_buff *msg,
+                              struct batadv_priv *bat_priv)
+{
+       u32 flags = bat_priv->mcast.flags;
+       u32 flags_priv = BATADV_NO_FLAGS;
+
+       if (bat_priv->mcast.bridged) {
+               flags_priv |= BATADV_MCAST_FLAGS_BRIDGED;
+
+               if (bat_priv->mcast.querier_ipv4.exists)
+                       flags_priv |= BATADV_MCAST_FLAGS_QUERIER_IPV4_EXISTS;
+               if (bat_priv->mcast.querier_ipv6.exists)
+                       flags_priv |= BATADV_MCAST_FLAGS_QUERIER_IPV6_EXISTS;
+               if (bat_priv->mcast.querier_ipv4.shadowing)
+                       flags_priv |= BATADV_MCAST_FLAGS_QUERIER_IPV4_SHADOWING;
+               if (bat_priv->mcast.querier_ipv6.shadowing)
+                       flags_priv |= BATADV_MCAST_FLAGS_QUERIER_IPV6_SHADOWING;
+       }
+
+       if (nla_put_u32(msg, BATADV_ATTR_MCAST_FLAGS, flags) ||
+           nla_put_u32(msg, BATADV_ATTR_MCAST_FLAGS_PRIV, flags_priv))
+               return -EMSGSIZE;
+
+       return 0;
+}
+
+/**
+ * batadv_mcast_flags_dump_entry() - dump one entry of the multicast flags table
+ *  to a netlink socket
+ * @msg: buffer for the message
+ * @portid: netlink port
+ * @seq: Sequence number of netlink message
+ * @orig_node: originator to dump the multicast flags of
+ *
+ * Return: 0 or error code.
+ */
+static int
+batadv_mcast_flags_dump_entry(struct sk_buff *msg, u32 portid, u32 seq,
+                             struct batadv_orig_node *orig_node)
+{
+       void *hdr;
+
+       hdr = genlmsg_put(msg, portid, seq, &batadv_netlink_family,
+                         NLM_F_MULTI, BATADV_CMD_GET_MCAST_FLAGS);
+       if (!hdr)
+               return -ENOBUFS;
+
+       if (nla_put(msg, BATADV_ATTR_ORIG_ADDRESS, ETH_ALEN,
+                   orig_node->orig)) {
+               genlmsg_cancel(msg, hdr);
+               return -EMSGSIZE;
+       }
+
+       if (test_bit(BATADV_ORIG_CAPA_HAS_MCAST,
+                    &orig_node->capabilities)) {
+               if (nla_put_u32(msg, BATADV_ATTR_MCAST_FLAGS,
+                               orig_node->mcast_flags)) {
+                       genlmsg_cancel(msg, hdr);
+                       return -EMSGSIZE;
+               }
+       }
+
+       genlmsg_end(msg, hdr);
+       return 0;
+}
+
+/**
+ * batadv_mcast_flags_dump_bucket() - dump one bucket of the multicast flags
+ *  table to a netlink socket
+ * @msg: buffer for the message
+ * @portid: netlink port
+ * @seq: Sequence number of netlink message
+ * @head: bucket to dump
+ * @idx_skip: How many entries to skip
+ *
+ * Return: 0 or error code.
+ */
+static int
+batadv_mcast_flags_dump_bucket(struct sk_buff *msg, u32 portid, u32 seq,
+                              struct hlist_head *head, long *idx_skip)
+{
+       struct batadv_orig_node *orig_node;
+       long idx = 0;
+
+       rcu_read_lock();
+       hlist_for_each_entry_rcu(orig_node, head, hash_entry) {
+               if (!test_bit(BATADV_ORIG_CAPA_HAS_MCAST,
+                             &orig_node->capa_initialized))
+                       continue;
+
+               if (idx < *idx_skip)
+                       goto skip;
+
+               if (batadv_mcast_flags_dump_entry(msg, portid, seq,
+                                                 orig_node)) {
+                       rcu_read_unlock();
+                       *idx_skip = idx;
+
+                       return -EMSGSIZE;
+               }
+
+skip:
+               idx++;
+       }
+       rcu_read_unlock();
+
+       return 0;
+}
+
+/**
+ * __batadv_mcast_flags_dump() - dump multicast flags table to a netlink socket
+ * @msg: buffer for the message
+ * @portid: netlink port
+ * @seq: Sequence number of netlink message
+ * @bat_priv: the bat priv with all the soft interface information
+ * @bucket: current bucket to dump
+ * @idx: index in current bucket to the next entry to dump
+ *
+ * Return: 0 or error code.
+ */
+static int
+__batadv_mcast_flags_dump(struct sk_buff *msg, u32 portid, u32 seq,
+                         struct batadv_priv *bat_priv, long *bucket, long *idx)
+{
+       struct batadv_hashtable *hash = bat_priv->orig_hash;
+       long bucket_tmp = *bucket;
+       struct hlist_head *head;
+       long idx_tmp = *idx;
+
+       while (bucket_tmp < hash->size) {
+               head = &hash->table[bucket_tmp];
+
+               if (batadv_mcast_flags_dump_bucket(msg, portid, seq, head,
+                                                  &idx_tmp))
+                       break;
+
+               bucket_tmp++;
+               idx_tmp = 0;
+       }
+
+       *bucket = bucket_tmp;
+       *idx = idx_tmp;
+
+       return msg->len;
+}
+
+/**
+ * batadv_mcast_netlink_get_primary() - get primary interface from netlink
+ *  callback
+ * @cb: netlink callback structure
+ * @primary_if: the primary interface pointer to return the result in
+ *
+ * Return: 0 or error code.
+ */
+static int
+batadv_mcast_netlink_get_primary(struct netlink_callback *cb,
+                                struct batadv_hard_iface **primary_if)
+{
+       struct batadv_hard_iface *hard_iface = NULL;
+       struct net *net = sock_net(cb->skb->sk);
+       struct net_device *soft_iface;
+       struct batadv_priv *bat_priv;
+       int ifindex;
+       int ret = 0;
+
+       ifindex = batadv_netlink_get_ifindex(cb->nlh, BATADV_ATTR_MESH_IFINDEX);
+       if (!ifindex)
+               return -EINVAL;
+
+       soft_iface = dev_get_by_index(net, ifindex);
+       if (!soft_iface || !batadv_softif_is_valid(soft_iface)) {
+               ret = -ENODEV;
+               goto out;
+       }
+
+       bat_priv = netdev_priv(soft_iface);
+
+       hard_iface = batadv_primary_if_get_selected(bat_priv);
+       if (!hard_iface || hard_iface->if_status != BATADV_IF_ACTIVE) {
+               ret = -ENOENT;
+               goto out;
+       }
+
+out:
+       if (soft_iface)
+               dev_put(soft_iface);
+
+       if (!ret && primary_if)
+               *primary_if = hard_iface;
+       else
+               batadv_hardif_put(hard_iface);
+
+       return ret;
+}
+
+/**
+ * batadv_mcast_flags_dump() - dump multicast flags table to a netlink socket
+ * @msg: buffer for the message
+ * @cb: callback structure containing arguments
+ *
+ * Return: message length.
+ */
+int batadv_mcast_flags_dump(struct sk_buff *msg, struct netlink_callback *cb)
+{
+       struct batadv_hard_iface *primary_if = NULL;
+       int portid = NETLINK_CB(cb->skb).portid;
+       struct batadv_priv *bat_priv;
+       long *bucket = &cb->args[0];
+       long *idx = &cb->args[1];
+       int ret;
+
+       ret = batadv_mcast_netlink_get_primary(cb, &primary_if);
+       if (ret)
+               return ret;
+
+       bat_priv = netdev_priv(primary_if->soft_iface);
+       ret = __batadv_mcast_flags_dump(msg, portid, cb->nlh->nlmsg_seq,
+                                       bat_priv, bucket, idx);
+
+       batadv_hardif_put(primary_if);
+       return ret;
+}
+
 /**
  * batadv_mcast_free() - free the multicast optimizations structures
  * @bat_priv: the bat priv with all the soft interface information