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