net/batman-adv/multicast.c

   1 /* Copyright (C) 2014-2015 B.A.T.M.A.N. contributors:
   2  *
   3  * Linus Lüssing
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of version 2 of the GNU General Public
   7  * License as published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #include "multicast.h"
  19 #include "main.h"
  20
  21 #include <linux/atomic.h>
  22 #include <linux/byteorder/generic.h>
  23 #include <linux/errno.h>
  24 #include <linux/etherdevice.h>
  25 #include <linux/fs.h>
  26 #include <linux/if_ether.h>
  27 #include <linux/in6.h>
  28 #include <linux/in.h>
  29 #include <linux/ip.h>
  30 #include <linux/ipv6.h>
  31 #include <linux/list.h>
  32 #include <linux/netdevice.h>
  33 #include <linux/rculist.h>
  34 #include <linux/rcupdate.h>
  35 #include <linux/skbuff.h>
  36 #include <linux/slab.h>
  37 #include <linux/spinlock.h>
  38 #include <linux/stddef.h>
  39 #include <linux/string.h>
  40 #include <linux/types.h>
  41 #include <net/addrconf.h>
  42 #include <net/ipv6.h>
  43
  44 #include "packet.h"
  45 #include "translation-table.h"
  46
  47 /**
  48  * batadv_mcast_mla_softif_get - get softif multicast listeners
  49  * @dev: the device to collect multicast addresses from
  50  * @mcast_list: a list to put found addresses into
  51  *
  52  * Collect multicast addresses of the local multicast listeners
  53  * on the given soft interface, dev, in the given mcast_list.
  54  *
  55  * Returns -ENOMEM on memory allocation error or the number of
  56  * items added to the mcast_list otherwise.
  57  */
  58 static int batadv_mcast_mla_softif_get(struct net_device *dev,
  59                                        struct hlist_head *mcast_list)
  60 {
  61         struct netdev_hw_addr *mc_list_entry;
  62         struct batadv_hw_addr *new;
  63         int ret = 0;
  64
  65         netif_addr_lock_bh(dev);
  66         netdev_for_each_mc_addr(mc_list_entry, dev) {
  67                 new = kmalloc(sizeof(*new), GFP_ATOMIC);
  68                 if (!new) {
  69                         ret = -ENOMEM;
  70                         break;
  71                 }
  72
  73                 ether_addr_copy(new->addr, mc_list_entry->addr);
  74                 hlist_add_head(&new->list, mcast_list);
  75                 ret++;
  76         }
  77         netif_addr_unlock_bh(dev);
  78
  79         return ret;
  80 }
  81
  82 /**
  83  * batadv_mcast_mla_is_duplicate - check whether an address is in a list
  84  * @mcast_addr: the multicast address to check
  85  * @mcast_list: the list with multicast addresses to search in
  86  *
  87  * Returns true if the given address is already in the given list.
  88  * Otherwise returns false.
  89  */
  90 static bool batadv_mcast_mla_is_duplicate(uint8_t *mcast_addr,
  91                                           struct hlist_head *mcast_list)
  92 {
  93         struct batadv_hw_addr *mcast_entry;
  94
  95         hlist_for_each_entry(mcast_entry, mcast_list, list)
  96                 if (batadv_compare_eth(mcast_entry->addr, mcast_addr))
  97                         return true;
  98
  99         return false;
 100 }
 101
 102 /**
 103  * batadv_mcast_mla_list_free - free a list of multicast addresses
 104  * @mcast_list: the list to free
 105  *
 106  * Removes and frees all items in the given mcast_list.
 107  */
 108 static void batadv_mcast_mla_list_free(struct hlist_head *mcast_list)
 109 {
 110         struct batadv_hw_addr *mcast_entry;
 111         struct hlist_node *tmp;
 112
 113         hlist_for_each_entry_safe(mcast_entry, tmp, mcast_list, list) {
 114                 hlist_del(&mcast_entry->list);
 115                 kfree(mcast_entry);
 116         }
 117 }
 118
 119 /**
 120  * batadv_mcast_mla_tt_retract - clean up multicast listener announcements
 121  * @bat_priv: the bat priv with all the soft interface information
 122  * @mcast_list: a list of addresses which should _not_ be removed
 123  *
 124  * Retracts the announcement of any multicast listener from the
 125  * translation table except the ones listed in the given mcast_list.
 126  *
 127  * If mcast_list is NULL then all are retracted.
 128  */
 129 static void batadv_mcast_mla_tt_retract(struct batadv_priv *bat_priv,
 130                                         struct hlist_head *mcast_list)
 131 {
 132         struct batadv_hw_addr *mcast_entry;
 133         struct hlist_node *tmp;
 134
 135         hlist_for_each_entry_safe(mcast_entry, tmp, &bat_priv->mcast.mla_list,
 136                                   list) {
 137                 if (mcast_list &&
 138                     batadv_mcast_mla_is_duplicate(mcast_entry->addr,
 139                                                   mcast_list))
 140                         continue;
 141
 142                 batadv_tt_local_remove(bat_priv, mcast_entry->addr,
 143                                        BATADV_NO_FLAGS,
 144                                        "mcast TT outdated", false);
 145
 146                 hlist_del(&mcast_entry->list);
 147                 kfree(mcast_entry);
 148         }
 149 }
 150
 151 /**
 152  * batadv_mcast_mla_tt_add - add multicast listener announcements
 153  * @bat_priv: the bat priv with all the soft interface information
 154  * @mcast_list: a list of addresses which are going to get added
 155  *
 156  * Adds multicast listener announcements from the given mcast_list to the
 157  * translation table if they have not been added yet.
 158  */
 159 static void batadv_mcast_mla_tt_add(struct batadv_priv *bat_priv,
 160                                     struct hlist_head *mcast_list)
 161 {
 162         struct batadv_hw_addr *mcast_entry;
 163         struct hlist_node *tmp;
 164
 165         if (!mcast_list)
 166                 return;
 167
 168         hlist_for_each_entry_safe(mcast_entry, tmp, mcast_list, list) {
 169                 if (batadv_mcast_mla_is_duplicate(mcast_entry->addr,
 170                                                   &bat_priv->mcast.mla_list))
 171                         continue;
 172
 173                 if (!batadv_tt_local_add(bat_priv->soft_iface,
 174                                          mcast_entry->addr, BATADV_NO_FLAGS,
 175                                          BATADV_NULL_IFINDEX, BATADV_NO_MARK))
 176                         continue;
 177
 178                 hlist_del(&mcast_entry->list);
 179                 hlist_add_head(&mcast_entry->list, &bat_priv->mcast.mla_list);
 180         }
 181 }
 182
 183 /**
 184  * batadv_mcast_has_bridge - check whether the soft-iface is bridged
 185  * @bat_priv: the bat priv with all the soft interface information
 186  *
 187  * Checks whether there is a bridge on top of our soft interface. Returns
 188  * true if so, false otherwise.
 189  */
 190 static bool batadv_mcast_has_bridge(struct batadv_priv *bat_priv)
 191 {
 192         struct net_device *upper = bat_priv->soft_iface;
 193
 194         rcu_read_lock();
 195         do {
 196                 upper = netdev_master_upper_dev_get_rcu(upper);
 197         } while (upper && !(upper->priv_flags & IFF_EBRIDGE));
 198         rcu_read_unlock();
 199
 200         return upper;
 201 }
 202
 203 /**
 204  * batadv_mcast_mla_tvlv_update - update multicast tvlv
 205  * @bat_priv: the bat priv with all the soft interface information
 206  *
 207  * Updates the own multicast tvlv with our current multicast related settings,
 208  * capabilities and inabilities.
 209  *
 210  * Returns true if the tvlv container is registered afterwards. Otherwise
 211  * returns false.
 212  */
 213 static bool batadv_mcast_mla_tvlv_update(struct batadv_priv *bat_priv)
 214 {
 215         struct batadv_tvlv_mcast_data mcast_data;
 216
 217         mcast_data.flags = BATADV_NO_FLAGS;
 218         memset(mcast_data.reserved, 0, sizeof(mcast_data.reserved));
 219
 220         /* Avoid attaching MLAs, if there is a bridge on top of our soft
 221          * interface, we don't support that yet (TODO)
 222          */
 223         if (batadv_mcast_has_bridge(bat_priv)) {
 224                 if (bat_priv->mcast.enabled) {
 225                         batadv_tvlv_container_unregister(bat_priv,
 226                                                          BATADV_TVLV_MCAST, 1);
 227                         bat_priv->mcast.enabled = false;
 228                 }
 229
 230                 return false;
 231         }
 232
 233         if (!bat_priv->mcast.enabled ||
 234             mcast_data.flags != bat_priv->mcast.flags) {
 235                 batadv_tvlv_container_register(bat_priv, BATADV_TVLV_MCAST, 1,
 236                                                &mcast_data, sizeof(mcast_data));
 237                 bat_priv->mcast.flags = mcast_data.flags;
 238                 bat_priv->mcast.enabled = true;
 239         }
 240
 241         return true;
 242 }
 243
 244 /**
 245  * batadv_mcast_mla_update - update the own MLAs
 246  * @bat_priv: the bat priv with all the soft interface information
 247  *
 248  * Updates the own multicast listener announcements in the translation
 249  * table as well as the own, announced multicast tvlv container.
 250  */
 251 void batadv_mcast_mla_update(struct batadv_priv *bat_priv)
 252 {
 253         struct net_device *soft_iface = bat_priv->soft_iface;
 254         struct hlist_head mcast_list = HLIST_HEAD_INIT;
 255         int ret;
 256
 257         if (!batadv_mcast_mla_tvlv_update(bat_priv))
 258                 goto update;
 259
 260         ret = batadv_mcast_mla_softif_get(soft_iface, &mcast_list);
 261         if (ret < 0)
 262                 goto out;
 263
 264 update:
 265         batadv_mcast_mla_tt_retract(bat_priv, &mcast_list);
 266         batadv_mcast_mla_tt_add(bat_priv, &mcast_list);
 267
 268 out:
 269         batadv_mcast_mla_list_free(&mcast_list);
 270 }
 271
 272 /**
 273  * batadv_mcast_forw_mode_check_ipv4 - check for optimized forwarding potential
 274  * @bat_priv: the bat priv with all the soft interface information
 275  * @skb: the IPv4 packet to check
 276  * @is_unsnoopable: stores whether the destination is snoopable
 277  *
 278  * Checks whether the given IPv4 packet has the potential to be forwarded with a
 279  * mode more optimal than classic flooding.
 280  *
 281  * If so then returns 0. Otherwise -EINVAL is returned or -ENOMEM in case of
 282  * memory allocation failure.
 283  */
 284 static int batadv_mcast_forw_mode_check_ipv4(struct batadv_priv *bat_priv,
 285                                              struct sk_buff *skb,
 286                                              bool *is_unsnoopable)
 287 {
 288         struct iphdr *iphdr;
 289
 290         /* We might fail due to out-of-memory -> drop it */
 291         if (!pskb_may_pull(skb, sizeof(struct ethhdr) + sizeof(*iphdr)))
 292                 return -ENOMEM;
 293
 294         iphdr = ip_hdr(skb);
 295
 296         /* TODO: Implement Multicast Router Discovery (RFC4286),
 297          * then allow scope > link local, too
 298          */
 299         if (!ipv4_is_local_multicast(iphdr->daddr))
 300                 return -EINVAL;
 301
 302         /* link-local multicast listeners behind a bridge are
 303          * not snoopable (see RFC4541, section 2.1.2.2)
 304          */
 305         *is_unsnoopable = true;
 306
 307         return 0;
 308 }
 309
 310 /**
 311  * batadv_mcast_forw_mode_check_ipv6 - check for optimized forwarding potential
 312  * @bat_priv: the bat priv with all the soft interface information
 313  * @skb: the IPv6 packet to check
 314  * @is_unsnoopable: stores whether the destination is snoopable
 315  *
 316  * Checks whether the given IPv6 packet has the potential to be forwarded with a
 317  * mode more optimal than classic flooding.
 318  *
 319  * If so then returns 0. Otherwise -EINVAL is returned or -ENOMEM if we are out
 320  * of memory.
 321  */
 322 static int batadv_mcast_forw_mode_check_ipv6(struct batadv_priv *bat_priv,
 323                                              struct sk_buff *skb,
 324                                              bool *is_unsnoopable)
 325 {
 326         struct ipv6hdr *ip6hdr;
 327
 328         /* We might fail due to out-of-memory -> drop it */
 329         if (!pskb_may_pull(skb, sizeof(struct ethhdr) + sizeof(*ip6hdr)))
 330                 return -ENOMEM;
 331
 332         ip6hdr = ipv6_hdr(skb);
 333
 334         /* TODO: Implement Multicast Router Discovery (RFC4286),
 335          * then allow scope > link local, too
 336          */
 337         if (IPV6_ADDR_MC_SCOPE(&ip6hdr->daddr) != IPV6_ADDR_SCOPE_LINKLOCAL)
 338                 return -EINVAL;
 339
 340         /* link-local-all-nodes multicast listeners behind a bridge are
 341          * not snoopable (see RFC4541, section 3, paragraph 3)
 342          */
 343         if (ipv6_addr_is_ll_all_nodes(&ip6hdr->daddr))
 344                 *is_unsnoopable = true;
 345
 346         return 0;
 347 }
 348
 349 /**
 350  * batadv_mcast_forw_mode_check - check for optimized forwarding potential
 351  * @bat_priv: the bat priv with all the soft interface information
 352  * @skb: the multicast frame to check
 353  * @is_unsnoopable: stores whether the destination is snoopable
 354  *
 355  * Checks whether the given multicast ethernet frame has the potential to be
 356  * forwarded with a mode more optimal than classic flooding.
 357  *
 358  * If so then returns 0. Otherwise -EINVAL is returned or -ENOMEM if we are out
 359  * of memory.
 360  */
 361 static int batadv_mcast_forw_mode_check(struct batadv_priv *bat_priv,
 362                                         struct sk_buff *skb,
 363                                         bool *is_unsnoopable)
 364 {
 365         struct ethhdr *ethhdr = eth_hdr(skb);
 366
 367         if (!atomic_read(&bat_priv->multicast_mode))
 368                 return -EINVAL;
 369
 370         if (atomic_read(&bat_priv->mcast.num_disabled))
 371                 return -EINVAL;
 372
 373         switch (ntohs(ethhdr->h_proto)) {
 374         case ETH_P_IP:
 375                 return batadv_mcast_forw_mode_check_ipv4(bat_priv, skb,
 376                                                          is_unsnoopable);
 377         case ETH_P_IPV6:
 378                 return batadv_mcast_forw_mode_check_ipv6(bat_priv, skb,
 379                                                          is_unsnoopable);
 380         default:
 381                 return -EINVAL;
 382         }
 383 }
 384
 385 /**
 386  * batadv_mcast_want_all_ip_count - count nodes with unspecific mcast interest
 387  * @bat_priv: the bat priv with all the soft interface information
 388  * @ethhdr: ethernet header of a packet
 389  *
 390  * Returns the number of nodes which want all IPv4 multicast traffic if the
 391  * given ethhdr is from an IPv4 packet or the number of nodes which want all
 392  * IPv6 traffic if it matches an IPv6 packet.
 393  */
 394 static int batadv_mcast_forw_want_all_ip_count(struct batadv_priv *bat_priv,
 395                                                struct ethhdr *ethhdr)
 396 {
 397         switch (ntohs(ethhdr->h_proto)) {
 398         case ETH_P_IP:
 399                 return atomic_read(&bat_priv->mcast.num_want_all_ipv4);
 400         case ETH_P_IPV6:
 401                 return atomic_read(&bat_priv->mcast.num_want_all_ipv6);
 402         default:
 403                 /* we shouldn't be here... */
 404                 return 0;
 405         }
 406 }
 407
 408 /**
 409  * batadv_mcast_forw_tt_node_get - get a multicast tt node
 410  * @bat_priv: the bat priv with all the soft interface information
 411  * @ethhdr: the ether header containing the multicast destination
 412  *
 413  * Returns an orig_node matching the multicast address provided by ethhdr
 414  * via a translation table lookup. This increases the returned nodes refcount.
 415  */
 416 static struct batadv_orig_node *
 417 batadv_mcast_forw_tt_node_get(struct batadv_priv *bat_priv,
 418                               struct ethhdr *ethhdr)
 419 {
 420         return batadv_transtable_search(bat_priv, ethhdr->h_source,
 421                                         ethhdr->h_dest, BATADV_NO_FLAGS);
 422 }
 423
 424 /**
 425  * batadv_mcast_want_forw_ipv4_node_get - get a node with an ipv4 flag
 426  * @bat_priv: the bat priv with all the soft interface information
 427  *
 428  * Returns an orig_node which has the BATADV_MCAST_WANT_ALL_IPV4 flag set and
 429  * increases its refcount.
 430  */
 431 static struct batadv_orig_node *
 432 batadv_mcast_forw_ipv4_node_get(struct batadv_priv *bat_priv)
 433 {
 434         struct batadv_orig_node *tmp_orig_node, *orig_node = NULL;
 435
 436         rcu_read_lock();
 437         hlist_for_each_entry_rcu(tmp_orig_node,
 438                                  &bat_priv->mcast.want_all_ipv4_list,
 439                                  mcast_want_all_ipv4_node) {
 440                 if (!atomic_inc_not_zero(&tmp_orig_node->refcount))
 441                         continue;
 442
 443                 orig_node = tmp_orig_node;
 444                 break;
 445         }
 446         rcu_read_unlock();
 447
 448         return orig_node;
 449 }
 450
 451 /**
 452  * batadv_mcast_want_forw_ipv6_node_get - get a node with an ipv6 flag
 453  * @bat_priv: the bat priv with all the soft interface information
 454  *
 455  * Returns an orig_node which has the BATADV_MCAST_WANT_ALL_IPV6 flag set
 456  * and increases its refcount.
 457  */
 458 static struct batadv_orig_node *
 459 batadv_mcast_forw_ipv6_node_get(struct batadv_priv *bat_priv)
 460 {
 461         struct batadv_orig_node *tmp_orig_node, *orig_node = NULL;
 462
 463         rcu_read_lock();
 464         hlist_for_each_entry_rcu(tmp_orig_node,
 465                                  &bat_priv->mcast.want_all_ipv6_list,
 466                                  mcast_want_all_ipv6_node) {
 467                 if (!atomic_inc_not_zero(&tmp_orig_node->refcount))
 468                         continue;
 469
 470                 orig_node = tmp_orig_node;
 471                 break;
 472         }
 473         rcu_read_unlock();
 474
 475         return orig_node;
 476 }
 477
 478 /**
 479  * batadv_mcast_want_forw_ip_node_get - get a node with an ipv4/ipv6 flag
 480  * @bat_priv: the bat priv with all the soft interface information
 481  * @ethhdr: an ethernet header to determine the protocol family from
 482  *
 483  * Returns an orig_node which has the BATADV_MCAST_WANT_ALL_IPV4 or
 484  * BATADV_MCAST_WANT_ALL_IPV6 flag, depending on the provided ethhdr, set and
 485  * increases its refcount.
 486  */
 487 static struct batadv_orig_node *
 488 batadv_mcast_forw_ip_node_get(struct batadv_priv *bat_priv,
 489                               struct ethhdr *ethhdr)
 490 {
 491         switch (ntohs(ethhdr->h_proto)) {
 492         case ETH_P_IP:
 493                 return batadv_mcast_forw_ipv4_node_get(bat_priv);
 494         case ETH_P_IPV6:
 495                 return batadv_mcast_forw_ipv6_node_get(bat_priv);
 496         default:
 497                 /* we shouldn't be here... */
 498                 return NULL;
 499         }
 500 }
 501
 502 /**
 503  * batadv_mcast_want_forw_unsnoop_node_get - get a node with an unsnoopable flag
 504  * @bat_priv: the bat priv with all the soft interface information
 505  *
 506  * Returns an orig_node which has the BATADV_MCAST_WANT_ALL_UNSNOOPABLES flag
 507  * set and increases its refcount.
 508  */
 509 static struct batadv_orig_node *
 510 batadv_mcast_forw_unsnoop_node_get(struct batadv_priv *bat_priv)
 511 {
 512         struct batadv_orig_node *tmp_orig_node, *orig_node = NULL;
 513
 514         rcu_read_lock();
 515         hlist_for_each_entry_rcu(tmp_orig_node,
 516                                  &bat_priv->mcast.want_all_unsnoopables_list,
 517                                  mcast_want_all_unsnoopables_node) {
 518                 if (!atomic_inc_not_zero(&tmp_orig_node->refcount))
 519                         continue;
 520
 521                 orig_node = tmp_orig_node;
 522                 break;
 523         }
 524         rcu_read_unlock();
 525
 526         return orig_node;
 527 }
 528
 529 /**
 530  * batadv_mcast_forw_mode - check on how to forward a multicast packet
 531  * @bat_priv: the bat priv with all the soft interface information
 532  * @skb: The multicast packet to check
 533  * @orig: an originator to be set to forward the skb to
 534  *
 535  * Returns the forwarding mode as enum batadv_forw_mode and in case of
 536  * BATADV_FORW_SINGLE set the orig to the single originator the skb
 537  * should be forwarded to.
 538  */
 539 enum batadv_forw_mode
 540 batadv_mcast_forw_mode(struct batadv_priv *bat_priv, struct sk_buff *skb,
 541                        struct batadv_orig_node **orig)
 542 {
 543         int ret, tt_count, ip_count, unsnoop_count, total_count;
 544         bool is_unsnoopable = false;
 545         struct ethhdr *ethhdr;
 546
 547         ret = batadv_mcast_forw_mode_check(bat_priv, skb, &is_unsnoopable);
 548         if (ret == -ENOMEM)
 549                 return BATADV_FORW_NONE;
 550         else if (ret < 0)
 551                 return BATADV_FORW_ALL;
 552
 553         ethhdr = eth_hdr(skb);
 554
 555         tt_count = batadv_tt_global_hash_count(bat_priv, ethhdr->h_dest,
 556                                                BATADV_NO_FLAGS);
 557         ip_count = batadv_mcast_forw_want_all_ip_count(bat_priv, ethhdr);
 558         unsnoop_count = !is_unsnoopable ? 0 :
 559                         atomic_read(&bat_priv->mcast.num_want_all_unsnoopables);
 560
 561         total_count = tt_count + ip_count + unsnoop_count;
 562
 563         switch (total_count) {
 564         case 1:
 565                 if (tt_count)
 566                         *orig = batadv_mcast_forw_tt_node_get(bat_priv, ethhdr);
 567                 else if (ip_count)
 568                         *orig = batadv_mcast_forw_ip_node_get(bat_priv, ethhdr);
 569                 else if (unsnoop_count)
 570                         *orig = batadv_mcast_forw_unsnoop_node_get(bat_priv);
 571
 572                 if (*orig)
 573                         return BATADV_FORW_SINGLE;
 574
 575                 /* fall through */
 576         case 0:
 577                 return BATADV_FORW_NONE;
 578         default:
 579                 return BATADV_FORW_ALL;
 580         }
 581 }
 582
 583 /**
 584  * batadv_mcast_want_unsnoop_update - update unsnoop counter and list
 585  * @bat_priv: the bat priv with all the soft interface information
 586  * @orig: the orig_node which multicast state might have changed of
 587  * @mcast_flags: flags indicating the new multicast state
 588  *
 589  * If the BATADV_MCAST_WANT_ALL_UNSNOOPABLES flag of this originator,
 590  * orig, has toggled then this method updates counter and list accordingly.
 591  */
 592 static void batadv_mcast_want_unsnoop_update(struct batadv_priv *bat_priv,
 593                                              struct batadv_orig_node *orig,
 594                                              uint8_t mcast_flags)
 595 {
 596         /* switched from flag unset to set */
 597         if (mcast_flags & BATADV_MCAST_WANT_ALL_UNSNOOPABLES &&
 598             !(orig->mcast_flags & BATADV_MCAST_WANT_ALL_UNSNOOPABLES)) {
 599                 atomic_inc(&bat_priv->mcast.num_want_all_unsnoopables);
 600
 601                 spin_lock_bh(&bat_priv->mcast.want_lists_lock);
 602                 hlist_add_head_rcu(&orig->mcast_want_all_unsnoopables_node,
 603                                    &bat_priv->mcast.want_all_unsnoopables_list);
 604                 spin_unlock_bh(&bat_priv->mcast.want_lists_lock);
 605         /* switched from flag set to unset */
 606         } else if (!(mcast_flags & BATADV_MCAST_WANT_ALL_UNSNOOPABLES) &&
 607                    orig->mcast_flags & BATADV_MCAST_WANT_ALL_UNSNOOPABLES) {
 608                 atomic_dec(&bat_priv->mcast.num_want_all_unsnoopables);
 609
 610                 spin_lock_bh(&bat_priv->mcast.want_lists_lock);
 611                 hlist_del_rcu(&orig->mcast_want_all_unsnoopables_node);
 612                 spin_unlock_bh(&bat_priv->mcast.want_lists_lock);
 613         }
 614 }
 615
 616 /**
 617  * batadv_mcast_want_ipv4_update - update want-all-ipv4 counter and list
 618  * @bat_priv: the bat priv with all the soft interface information
 619  * @orig: the orig_node which multicast state might have changed of
 620  * @mcast_flags: flags indicating the new multicast state
 621  *
 622  * If the BATADV_MCAST_WANT_ALL_IPV4 flag of this originator, orig, has
 623  * toggled then this method updates counter and list accordingly.
 624  */
 625 static void batadv_mcast_want_ipv4_update(struct batadv_priv *bat_priv,
 626                                           struct batadv_orig_node *orig,
 627                                           uint8_t mcast_flags)
 628 {
 629         /* switched from flag unset to set */
 630         if (mcast_flags & BATADV_MCAST_WANT_ALL_IPV4 &&
 631             !(orig->mcast_flags & BATADV_MCAST_WANT_ALL_IPV4)) {
 632                 atomic_inc(&bat_priv->mcast.num_want_all_ipv4);
 633
 634                 spin_lock_bh(&bat_priv->mcast.want_lists_lock);
 635                 hlist_add_head_rcu(&orig->mcast_want_all_ipv4_node,
 636                                    &bat_priv->mcast.want_all_ipv4_list);
 637                 spin_unlock_bh(&bat_priv->mcast.want_lists_lock);
 638         /* switched from flag set to unset */
 639         } else if (!(mcast_flags & BATADV_MCAST_WANT_ALL_IPV4) &&
 640                    orig->mcast_flags & BATADV_MCAST_WANT_ALL_IPV4) {
 641                 atomic_dec(&bat_priv->mcast.num_want_all_ipv4);
 642
 643                 spin_lock_bh(&bat_priv->mcast.want_lists_lock);
 644                 hlist_del_rcu(&orig->mcast_want_all_ipv4_node);
 645                 spin_unlock_bh(&bat_priv->mcast.want_lists_lock);
 646         }
 647 }
 648
 649 /**
 650  * batadv_mcast_want_ipv6_update - update want-all-ipv6 counter and list
 651  * @bat_priv: the bat priv with all the soft interface information
 652  * @orig: the orig_node which multicast state might have changed of
 653  * @mcast_flags: flags indicating the new multicast state
 654  *
 655  * If the BATADV_MCAST_WANT_ALL_IPV6 flag of this originator, orig, has
 656  * toggled then this method updates counter and list accordingly.
 657  */
 658 static void batadv_mcast_want_ipv6_update(struct batadv_priv *bat_priv,
 659                                           struct batadv_orig_node *orig,
 660                                           uint8_t mcast_flags)
 661 {
 662         /* switched from flag unset to set */
 663         if (mcast_flags & BATADV_MCAST_WANT_ALL_IPV6 &&
 664             !(orig->mcast_flags & BATADV_MCAST_WANT_ALL_IPV6)) {
 665                 atomic_inc(&bat_priv->mcast.num_want_all_ipv6);
 666
 667                 spin_lock_bh(&bat_priv->mcast.want_lists_lock);
 668                 hlist_add_head_rcu(&orig->mcast_want_all_ipv6_node,
 669                                    &bat_priv->mcast.want_all_ipv6_list);
 670                 spin_unlock_bh(&bat_priv->mcast.want_lists_lock);
 671         /* switched from flag set to unset */
 672         } else if (!(mcast_flags & BATADV_MCAST_WANT_ALL_IPV6) &&
 673                    orig->mcast_flags & BATADV_MCAST_WANT_ALL_IPV6) {
 674                 atomic_dec(&bat_priv->mcast.num_want_all_ipv6);
 675
 676                 spin_lock_bh(&bat_priv->mcast.want_lists_lock);
 677                 hlist_del_rcu(&orig->mcast_want_all_ipv6_node);
 678                 spin_unlock_bh(&bat_priv->mcast.want_lists_lock);
 679         }
 680 }
 681
 682 /**
 683  * batadv_mcast_tvlv_ogm_handler_v1 - process incoming multicast tvlv container
 684  * @bat_priv: the bat priv with all the soft interface information
 685  * @orig: the orig_node of the ogm
 686  * @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags)
 687  * @tvlv_value: tvlv buffer containing the multicast data
 688  * @tvlv_value_len: tvlv buffer length
 689  */
 690 static void batadv_mcast_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
 691                                              struct batadv_orig_node *orig,
 692                                              uint8_t flags,
 693                                              void *tvlv_value,
 694                                              uint16_t tvlv_value_len)
 695 {
 696         bool orig_mcast_enabled = !(flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
 697         uint8_t mcast_flags = BATADV_NO_FLAGS;
 698         bool orig_initialized;
 699
 700         orig_initialized = orig->capa_initialized & BATADV_ORIG_CAPA_HAS_MCAST;
 701
 702         /* If mcast support is turned on decrease the disabled mcast node
 703          * counter only if we had increased it for this node before. If this
 704          * is a completely new orig_node no need to decrease the counter.
 705          */
 706         if (orig_mcast_enabled &&
 707             !(orig->capabilities & BATADV_ORIG_CAPA_HAS_MCAST)) {
 708                 if (orig_initialized)
 709                         atomic_dec(&bat_priv->mcast.num_disabled);
 710                 orig->capabilities |= BATADV_ORIG_CAPA_HAS_MCAST;
 711         /* If mcast support is being switched off or if this is an initial
 712          * OGM without mcast support then increase the disabled mcast
 713          * node counter.
 714          */
 715         } else if (!orig_mcast_enabled &&
 716                    (orig->capabilities & BATADV_ORIG_CAPA_HAS_MCAST ||
 717                     !orig_initialized)) {
 718                 atomic_inc(&bat_priv->mcast.num_disabled);
 719                 orig->capabilities &= ~BATADV_ORIG_CAPA_HAS_MCAST;
 720         }
 721
 722         orig->capa_initialized |= BATADV_ORIG_CAPA_HAS_MCAST;
 723
 724         if (orig_mcast_enabled && tvlv_value &&
 725             (tvlv_value_len >= sizeof(mcast_flags)))
 726                 mcast_flags = *(uint8_t *)tvlv_value;
 727
 728         batadv_mcast_want_unsnoop_update(bat_priv, orig, mcast_flags);
 729         batadv_mcast_want_ipv4_update(bat_priv, orig, mcast_flags);
 730         batadv_mcast_want_ipv6_update(bat_priv, orig, mcast_flags);
 731
 732         orig->mcast_flags = mcast_flags;
 733 }
 734
 735 /**
 736  * batadv_mcast_init - initialize the multicast optimizations structures
 737  * @bat_priv: the bat priv with all the soft interface information
 738  */
 739 void batadv_mcast_init(struct batadv_priv *bat_priv)
 740 {
 741         batadv_tvlv_handler_register(bat_priv, batadv_mcast_tvlv_ogm_handler_v1,
 742                                      NULL, BATADV_TVLV_MCAST, 1,
 743                                      BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
 744 }
 745
 746 /**
 747  * batadv_mcast_free - free the multicast optimizations structures
 748  * @bat_priv: the bat priv with all the soft interface information
 749  */
 750 void batadv_mcast_free(struct batadv_priv *bat_priv)
 751 {
 752         batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_MCAST, 1);
 753         batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_MCAST, 1);
 754
 755         batadv_mcast_mla_tt_retract(bat_priv, NULL);
 756 }
 757
 758 /**
 759  * batadv_mcast_purge_orig - reset originator global mcast state modifications
 760  * @orig: the originator which is going to get purged
 761  */
 762 void batadv_mcast_purge_orig(struct batadv_orig_node *orig)
 763 {
 764         struct batadv_priv *bat_priv = orig->bat_priv;
 765
 766         if (!(orig->capabilities & BATADV_ORIG_CAPA_HAS_MCAST) &&
 767             orig->capa_initialized & BATADV_ORIG_CAPA_HAS_MCAST)
 768                 atomic_dec(&bat_priv->mcast.num_disabled);
 769
 770         batadv_mcast_want_unsnoop_update(bat_priv, orig, BATADV_NO_FLAGS);
 771         batadv_mcast_want_ipv4_update(bat_priv, orig, BATADV_NO_FLAGS);
 772         batadv_mcast_want_ipv6_update(bat_priv, orig, BATADV_NO_FLAGS);
 773 }