Merge tag 'kvm-4.9-1' of git://git.kernel.org/pub/scm/virt/kvm/kvm
[cascardo/linux.git] / net / batman-adv / bat_iv_ogm.c
index a40cdf2..e2d18d0 100644 (file)
@@ -35,6 +35,7 @@
 #include <linux/list.h>
 #include <linux/lockdep.h>
 #include <linux/netdevice.h>
+#include <linux/netlink.h>
 #include <linux/pkt_sched.h>
 #include <linux/printk.h>
 #include <linux/random.h>
@@ -48,6 +49,9 @@
 #include <linux/string.h>
 #include <linux/types.h>
 #include <linux/workqueue.h>
+#include <net/genetlink.h>
+#include <net/netlink.h>
+#include <uapi/linux/batman_adv.h>
 
 #include "bat_algo.h"
 #include "bitarray.h"
@@ -55,6 +59,7 @@
 #include "hard-interface.h"
 #include "hash.h"
 #include "log.h"
+#include "netlink.h"
 #include "network-coding.h"
 #include "originator.h"
 #include "packet.h"
@@ -319,17 +324,18 @@ batadv_iv_ogm_orig_get(struct batadv_priv *bat_priv, const u8 *addr)
        if (!orig_node->bat_iv.bcast_own_sum)
                goto free_orig_node;
 
+       kref_get(&orig_node->refcount);
        hash_added = batadv_hash_add(bat_priv->orig_hash, batadv_compare_orig,
                                     batadv_choose_orig, orig_node,
                                     &orig_node->hash_entry);
        if (hash_added != 0)
-               goto free_orig_node;
+               goto free_orig_node_hash;
 
        return orig_node;
 
-free_orig_node:
-       /* free twice, as batadv_orig_node_new sets refcount to 2 */
+free_orig_node_hash:
        batadv_orig_node_put(orig_node);
+free_orig_node:
        batadv_orig_node_put(orig_node);
 
        return NULL;
@@ -1849,6 +1855,7 @@ static int batadv_iv_ogm_receive(struct sk_buff *skb,
        return NET_RX_SUCCESS;
 }
 
+#ifdef CONFIG_BATMAN_ADV_DEBUGFS
 /**
  * batadv_iv_ogm_orig_print_neigh - print neighbors for the originator table
  * @orig_node: the orig_node for which the neighbors are printed
@@ -1946,7 +1953,238 @@ next:
        if (batman_count == 0)
                seq_puts(seq, "No batman nodes in range ...\n");
 }
+#endif
+
+/**
+ * batadv_iv_ogm_neigh_get_tq_avg - Get the TQ average for a neighbour on a
+ *  given outgoing interface.
+ * @neigh_node: Neighbour of interest
+ * @if_outgoing: Outgoing interface of interest
+ * @tq_avg: Pointer of where to store the TQ average
+ *
+ * Return: False if no average TQ available, otherwise true.
+ */
+static bool
+batadv_iv_ogm_neigh_get_tq_avg(struct batadv_neigh_node *neigh_node,
+                              struct batadv_hard_iface *if_outgoing,
+                              u8 *tq_avg)
+{
+       struct batadv_neigh_ifinfo *n_ifinfo;
+
+       n_ifinfo = batadv_neigh_ifinfo_get(neigh_node, if_outgoing);
+       if (!n_ifinfo)
+               return false;
+
+       *tq_avg = n_ifinfo->bat_iv.tq_avg;
+       batadv_neigh_ifinfo_put(n_ifinfo);
+
+       return true;
+}
+
+/**
+ * batadv_iv_ogm_orig_dump_subentry - Dump an originator subentry into a
+ *  message
+ * @msg: Netlink message to dump into
+ * @portid: Port making netlink request
+ * @seq: Sequence number of netlink message
+ * @bat_priv: The bat priv with all the soft interface information
+ * @if_outgoing: Limit dump to entries with this outgoing interface
+ * @orig_node: Originator to dump
+ * @neigh_node: Single hops neighbour
+ * @best: Is the best originator
+ *
+ * Return: Error code, or 0 on success
+ */
+static int
+batadv_iv_ogm_orig_dump_subentry(struct sk_buff *msg, u32 portid, u32 seq,
+                                struct batadv_priv *bat_priv,
+                                struct batadv_hard_iface *if_outgoing,
+                                struct batadv_orig_node *orig_node,
+                                struct batadv_neigh_node *neigh_node,
+                                bool best)
+{
+       void *hdr;
+       u8 tq_avg;
+       unsigned int last_seen_msecs;
+
+       last_seen_msecs = jiffies_to_msecs(jiffies - orig_node->last_seen);
+
+       if (!batadv_iv_ogm_neigh_get_tq_avg(neigh_node, if_outgoing, &tq_avg))
+               return 0;
+
+       if (if_outgoing != BATADV_IF_DEFAULT &&
+           if_outgoing != neigh_node->if_incoming)
+               return 0;
+
+       hdr = genlmsg_put(msg, portid, seq, &batadv_netlink_family,
+                         NLM_F_MULTI, BATADV_CMD_GET_ORIGINATORS);
+       if (!hdr)
+               return -ENOBUFS;
+
+       if (nla_put(msg, BATADV_ATTR_ORIG_ADDRESS, ETH_ALEN,
+                   orig_node->orig) ||
+           nla_put(msg, BATADV_ATTR_NEIGH_ADDRESS, ETH_ALEN,
+                   neigh_node->addr) ||
+           nla_put_u32(msg, BATADV_ATTR_HARD_IFINDEX,
+                       neigh_node->if_incoming->net_dev->ifindex) ||
+           nla_put_u8(msg, BATADV_ATTR_TQ, tq_avg) ||
+           nla_put_u32(msg, BATADV_ATTR_LAST_SEEN_MSECS,
+                       last_seen_msecs))
+               goto nla_put_failure;
+
+       if (best && nla_put_flag(msg, BATADV_ATTR_FLAG_BEST))
+               goto nla_put_failure;
+
+       genlmsg_end(msg, hdr);
+       return 0;
+
+ nla_put_failure:
+       genlmsg_cancel(msg, hdr);
+       return -EMSGSIZE;
+}
 
+/**
+ * batadv_iv_ogm_orig_dump_entry - Dump an originator entry into a message
+ * @msg: Netlink message to dump into
+ * @portid: Port making netlink request
+ * @seq: Sequence number of netlink message
+ * @bat_priv: The bat priv with all the soft interface information
+ * @if_outgoing: Limit dump to entries with this outgoing interface
+ * @orig_node: Originator to dump
+ * @sub_s: Number of sub entries to skip
+ *
+ * This function assumes the caller holds rcu_read_lock().
+ *
+ * Return: Error code, or 0 on success
+ */
+static int
+batadv_iv_ogm_orig_dump_entry(struct sk_buff *msg, u32 portid, u32 seq,
+                             struct batadv_priv *bat_priv,
+                             struct batadv_hard_iface *if_outgoing,
+                             struct batadv_orig_node *orig_node, int *sub_s)
+{
+       struct batadv_neigh_node *neigh_node_best;
+       struct batadv_neigh_node *neigh_node;
+       int sub = 0;
+       bool best;
+       u8 tq_avg_best;
+
+       neigh_node_best = batadv_orig_router_get(orig_node, if_outgoing);
+       if (!neigh_node_best)
+               goto out;
+
+       if (!batadv_iv_ogm_neigh_get_tq_avg(neigh_node_best, if_outgoing,
+                                           &tq_avg_best))
+               goto out;
+
+       if (tq_avg_best == 0)
+               goto out;
+
+       hlist_for_each_entry_rcu(neigh_node, &orig_node->neigh_list, list) {
+               if (sub++ < *sub_s)
+                       continue;
+
+               best = (neigh_node == neigh_node_best);
+
+               if (batadv_iv_ogm_orig_dump_subentry(msg, portid, seq,
+                                                    bat_priv, if_outgoing,
+                                                    orig_node, neigh_node,
+                                                    best)) {
+                       batadv_neigh_node_put(neigh_node_best);
+
+                       *sub_s = sub - 1;
+                       return -EMSGSIZE;
+               }
+       }
+
+ out:
+       if (neigh_node_best)
+               batadv_neigh_node_put(neigh_node_best);
+
+       *sub_s = 0;
+       return 0;
+}
+
+/**
+ * batadv_iv_ogm_orig_dump_bucket - Dump an originator bucket into a
+ *  message
+ * @msg: Netlink message to dump into
+ * @portid: Port making netlink request
+ * @seq: Sequence number of netlink message
+ * @bat_priv: The bat priv with all the soft interface information
+ * @if_outgoing: Limit dump to entries with this outgoing interface
+ * @head: Bucket to be dumped
+ * @idx_s: Number of entries to be skipped
+ * @sub: Number of sub entries to be skipped
+ *
+ * Return: Error code, or 0 on success
+ */
+static int
+batadv_iv_ogm_orig_dump_bucket(struct sk_buff *msg, u32 portid, u32 seq,
+                              struct batadv_priv *bat_priv,
+                              struct batadv_hard_iface *if_outgoing,
+                              struct hlist_head *head, int *idx_s, int *sub)
+{
+       struct batadv_orig_node *orig_node;
+       int idx = 0;
+
+       rcu_read_lock();
+       hlist_for_each_entry_rcu(orig_node, head, hash_entry) {
+               if (idx++ < *idx_s)
+                       continue;
+
+               if (batadv_iv_ogm_orig_dump_entry(msg, portid, seq, bat_priv,
+                                                 if_outgoing, orig_node,
+                                                 sub)) {
+                       rcu_read_unlock();
+                       *idx_s = idx - 1;
+                       return -EMSGSIZE;
+               }
+       }
+       rcu_read_unlock();
+
+       *idx_s = 0;
+       *sub = 0;
+       return 0;
+}
+
+/**
+ * batadv_iv_ogm_orig_dump - Dump the originators into a message
+ * @msg: Netlink message to dump into
+ * @cb: Control block containing additional options
+ * @bat_priv: The bat priv with all the soft interface information
+ * @if_outgoing: Limit dump to entries with this outgoing interface
+ */
+static void
+batadv_iv_ogm_orig_dump(struct sk_buff *msg, struct netlink_callback *cb,
+                       struct batadv_priv *bat_priv,
+                       struct batadv_hard_iface *if_outgoing)
+{
+       struct batadv_hashtable *hash = bat_priv->orig_hash;
+       struct hlist_head *head;
+       int bucket = cb->args[0];
+       int idx = cb->args[1];
+       int sub = cb->args[2];
+       int portid = NETLINK_CB(cb->skb).portid;
+
+       while (bucket < hash->size) {
+               head = &hash->table[bucket];
+
+               if (batadv_iv_ogm_orig_dump_bucket(msg, portid,
+                                                  cb->nlh->nlmsg_seq,
+                                                  bat_priv, if_outgoing, head,
+                                                  &idx, &sub))
+                       break;
+
+               bucket++;
+       }
+
+       cb->args[0] = bucket;
+       cb->args[1] = idx;
+       cb->args[2] = sub;
+}
+
+#ifdef CONFIG_BATMAN_ADV_DEBUGFS
 /**
  * batadv_iv_hardif_neigh_print - print a single hop neighbour node
  * @seq: neighbour table seq_file struct
@@ -1997,6 +2235,7 @@ static void batadv_iv_neigh_print(struct batadv_priv *bat_priv,
        if (batman_count == 0)
                seq_puts(seq, "No batman nodes in range ...\n");
 }
+#endif
 
 /**
  * batadv_iv_ogm_neigh_diff - calculate tq difference of two neighbors
@@ -2043,6 +2282,136 @@ out:
        return ret;
 }
 
+/**
+ * batadv_iv_ogm_neigh_dump_neigh - Dump a neighbour into a netlink message
+ * @msg: Netlink message to dump into
+ * @portid: Port making netlink request
+ * @seq: Sequence number of netlink message
+ * @hardif_neigh: Neighbour to be dumped
+ *
+ * Return: Error code, or 0 on success
+ */
+static int
+batadv_iv_ogm_neigh_dump_neigh(struct sk_buff *msg, u32 portid, u32 seq,
+                              struct batadv_hardif_neigh_node *hardif_neigh)
+{
+       void *hdr;
+       unsigned int last_seen_msecs;
+
+       last_seen_msecs = jiffies_to_msecs(jiffies - hardif_neigh->last_seen);
+
+       hdr = genlmsg_put(msg, portid, seq, &batadv_netlink_family,
+                         NLM_F_MULTI, BATADV_CMD_GET_NEIGHBORS);
+       if (!hdr)
+               return -ENOBUFS;
+
+       if (nla_put(msg, BATADV_ATTR_NEIGH_ADDRESS, ETH_ALEN,
+                   hardif_neigh->addr) ||
+           nla_put_u32(msg, BATADV_ATTR_HARD_IFINDEX,
+                       hardif_neigh->if_incoming->net_dev->ifindex) ||
+           nla_put_u32(msg, BATADV_ATTR_LAST_SEEN_MSECS,
+                       last_seen_msecs))
+               goto nla_put_failure;
+
+       genlmsg_end(msg, hdr);
+       return 0;
+
+ nla_put_failure:
+       genlmsg_cancel(msg, hdr);
+       return -EMSGSIZE;
+}
+
+/**
+ * batadv_iv_ogm_neigh_dump_hardif - Dump the neighbours of a hard interface
+ *  into a message
+ * @msg: Netlink message to dump into
+ * @portid: Port making netlink request
+ * @seq: Sequence number of netlink message
+ * @bat_priv: The bat priv with all the soft interface information
+ * @hard_iface: Hard interface to dump the neighbours for
+ * @idx_s: Number of entries to skip
+ *
+ * This function assumes the caller holds rcu_read_lock().
+ *
+ * Return: Error code, or 0 on success
+ */
+static int
+batadv_iv_ogm_neigh_dump_hardif(struct sk_buff *msg, u32 portid, u32 seq,
+                               struct batadv_priv *bat_priv,
+                               struct batadv_hard_iface *hard_iface,
+                               int *idx_s)
+{
+       struct batadv_hardif_neigh_node *hardif_neigh;
+       int idx = 0;
+
+       hlist_for_each_entry_rcu(hardif_neigh,
+                                &hard_iface->neigh_list, list) {
+               if (idx++ < *idx_s)
+                       continue;
+
+               if (batadv_iv_ogm_neigh_dump_neigh(msg, portid, seq,
+                                                  hardif_neigh)) {
+                       *idx_s = idx - 1;
+                       return -EMSGSIZE;
+               }
+       }
+
+       *idx_s = 0;
+       return 0;
+}
+
+/**
+ * batadv_iv_ogm_neigh_dump - Dump the neighbours into a message
+ * @msg: Netlink message to dump into
+ * @cb: Control block containing additional options
+ * @bat_priv: The bat priv with all the soft interface information
+ * @single_hardif: Limit dump to this hard interfaace
+ */
+static void
+batadv_iv_ogm_neigh_dump(struct sk_buff *msg, struct netlink_callback *cb,
+                        struct batadv_priv *bat_priv,
+                        struct batadv_hard_iface *single_hardif)
+{
+       struct batadv_hard_iface *hard_iface;
+       int i_hardif = 0;
+       int i_hardif_s = cb->args[0];
+       int idx = cb->args[1];
+       int portid = NETLINK_CB(cb->skb).portid;
+
+       rcu_read_lock();
+       if (single_hardif) {
+               if (i_hardif_s == 0) {
+                       if (batadv_iv_ogm_neigh_dump_hardif(msg, portid,
+                                                           cb->nlh->nlmsg_seq,
+                                                           bat_priv,
+                                                           single_hardif,
+                                                           &idx) == 0)
+                               i_hardif++;
+               }
+       } else {
+               list_for_each_entry_rcu(hard_iface, &batadv_hardif_list,
+                                       list) {
+                       if (hard_iface->soft_iface != bat_priv->soft_iface)
+                               continue;
+
+                       if (i_hardif++ < i_hardif_s)
+                               continue;
+
+                       if (batadv_iv_ogm_neigh_dump_hardif(msg, portid,
+                                                           cb->nlh->nlmsg_seq,
+                                                           bat_priv,
+                                                           hard_iface, &idx)) {
+                               i_hardif--;
+                               break;
+                       }
+               }
+       }
+       rcu_read_unlock();
+
+       cb->args[0] = i_hardif;
+       cb->args[1] = idx;
+}
+
 /**
  * batadv_iv_ogm_neigh_cmp - compare the metrics of two neighbors
  * @neigh1: the first neighbor object of the comparison
@@ -2253,6 +2622,7 @@ out:
        return ret;
 }
 
+#ifdef CONFIG_BATMAN_ADV_DEBUGFS
 /* fails if orig_node has no router */
 static int batadv_iv_gw_write_buffer_text(struct batadv_priv *bat_priv,
                                          struct seq_file *seq,
@@ -2316,6 +2686,111 @@ static void batadv_iv_gw_print(struct batadv_priv *bat_priv,
        if (gw_count == 0)
                seq_puts(seq, "No gateways in range ...\n");
 }
+#endif
+
+/**
+ * batadv_iv_gw_dump_entry - Dump a gateway into a message
+ * @msg: Netlink message to dump into
+ * @portid: Port making netlink request
+ * @seq: Sequence number of netlink message
+ * @bat_priv: The bat priv with all the soft interface information
+ * @gw_node: Gateway to be dumped
+ *
+ * Return: Error code, or 0 on success
+ */
+static int batadv_iv_gw_dump_entry(struct sk_buff *msg, u32 portid, u32 seq,
+                                  struct batadv_priv *bat_priv,
+                                  struct batadv_gw_node *gw_node)
+{
+       struct batadv_neigh_ifinfo *router_ifinfo = NULL;
+       struct batadv_neigh_node *router;
+       struct batadv_gw_node *curr_gw;
+       int ret = -EINVAL;
+       void *hdr;
+
+       router = batadv_orig_router_get(gw_node->orig_node, BATADV_IF_DEFAULT);
+       if (!router)
+               goto out;
+
+       router_ifinfo = batadv_neigh_ifinfo_get(router, BATADV_IF_DEFAULT);
+       if (!router_ifinfo)
+               goto out;
+
+       curr_gw = batadv_gw_get_selected_gw_node(bat_priv);
+
+       hdr = genlmsg_put(msg, portid, seq, &batadv_netlink_family,
+                         NLM_F_MULTI, BATADV_CMD_GET_GATEWAYS);
+       if (!hdr) {
+               ret = -ENOBUFS;
+               goto out;
+       }
+
+       ret = -EMSGSIZE;
+
+       if (curr_gw == gw_node)
+               if (nla_put_flag(msg, BATADV_ATTR_FLAG_BEST)) {
+                       genlmsg_cancel(msg, hdr);
+                       goto out;
+               }
+
+       if (nla_put(msg, BATADV_ATTR_ORIG_ADDRESS, ETH_ALEN,
+                   gw_node->orig_node->orig) ||
+           nla_put_u8(msg, BATADV_ATTR_TQ, router_ifinfo->bat_iv.tq_avg) ||
+           nla_put(msg, BATADV_ATTR_ROUTER, ETH_ALEN,
+                   router->addr) ||
+           nla_put_string(msg, BATADV_ATTR_HARD_IFNAME,
+                          router->if_incoming->net_dev->name) ||
+           nla_put_u32(msg, BATADV_ATTR_BANDWIDTH_DOWN,
+                       gw_node->bandwidth_down) ||
+           nla_put_u32(msg, BATADV_ATTR_BANDWIDTH_UP,
+                       gw_node->bandwidth_up)) {
+               genlmsg_cancel(msg, hdr);
+               goto out;
+       }
+
+       genlmsg_end(msg, hdr);
+       ret = 0;
+
+out:
+       if (router_ifinfo)
+               batadv_neigh_ifinfo_put(router_ifinfo);
+       if (router)
+               batadv_neigh_node_put(router);
+       return ret;
+}
+
+/**
+ * batadv_iv_gw_dump - Dump gateways into a message
+ * @msg: Netlink message to dump into
+ * @cb: Control block containing additional options
+ * @bat_priv: The bat priv with all the soft interface information
+ */
+static void batadv_iv_gw_dump(struct sk_buff *msg, struct netlink_callback *cb,
+                             struct batadv_priv *bat_priv)
+{
+       int portid = NETLINK_CB(cb->skb).portid;
+       struct batadv_gw_node *gw_node;
+       int idx_skip = cb->args[0];
+       int idx = 0;
+
+       rcu_read_lock();
+       hlist_for_each_entry_rcu(gw_node, &bat_priv->gw.list, list) {
+               if (idx++ < idx_skip)
+                       continue;
+
+               if (batadv_iv_gw_dump_entry(msg, portid, cb->nlh->nlmsg_seq,
+                                           bat_priv, gw_node)) {
+                       idx_skip = idx - 1;
+                       goto unlock;
+               }
+       }
+
+       idx_skip = idx;
+unlock:
+       rcu_read_unlock();
+
+       cb->args[0] = idx_skip;
+}
 
 static struct batadv_algo_ops batadv_batman_iv __read_mostly = {
        .name = "BATMAN_IV",
@@ -2329,10 +2804,16 @@ static struct batadv_algo_ops batadv_batman_iv __read_mostly = {
        .neigh = {
                .cmp = batadv_iv_ogm_neigh_cmp,
                .is_similar_or_better = batadv_iv_ogm_neigh_is_sob,
+#ifdef CONFIG_BATMAN_ADV_DEBUGFS
                .print = batadv_iv_neigh_print,
+#endif
+               .dump = batadv_iv_ogm_neigh_dump,
        },
        .orig = {
+#ifdef CONFIG_BATMAN_ADV_DEBUGFS
                .print = batadv_iv_ogm_orig_print,
+#endif
+               .dump = batadv_iv_ogm_orig_dump,
                .free = batadv_iv_ogm_orig_free,
                .add_if = batadv_iv_ogm_orig_add_if,
                .del_if = batadv_iv_ogm_orig_del_if,
@@ -2340,7 +2821,10 @@ static struct batadv_algo_ops batadv_batman_iv __read_mostly = {
        .gw = {
                .get_best_gw_node = batadv_iv_gw_get_best_gw_node,
                .is_eligible = batadv_iv_gw_is_eligible,
+#ifdef CONFIG_BATMAN_ADV_DEBUGFS
                .print = batadv_iv_gw_print,
+#endif
+               .dump = batadv_iv_gw_dump,
        },
 };