• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Bridge multicast support.
3  *
4  * Copyright (c) 2010 Herbert Xu <herbert@gondor.apana.org.au>
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU General Public License as published by the Free
8  * Software Foundation; either version 2 of the License, or (at your option)
9  * any later version.
10  *
11  */
12 
13 #include <linux/err.h>
14 #include <linux/export.h>
15 #include <linux/if_ether.h>
16 #include <linux/igmp.h>
17 #include <linux/jhash.h>
18 #include <linux/kernel.h>
19 #include <linux/log2.h>
20 #include <linux/netdevice.h>
21 #include <linux/netfilter_bridge.h>
22 #include <linux/random.h>
23 #include <linux/rculist.h>
24 #include <linux/skbuff.h>
25 #include <linux/slab.h>
26 #include <linux/timer.h>
27 #include <linux/inetdevice.h>
28 #include <linux/mroute.h>
29 #include <net/ip.h>
30 #include <net/switchdev.h>
31 #if IS_ENABLED(CONFIG_IPV6)
32 #include <net/ipv6.h>
33 #include <net/mld.h>
34 #include <net/ip6_checksum.h>
35 #include <net/addrconf.h>
36 #endif
37 
38 #include "br_private.h"
39 
40 static void br_multicast_start_querier(struct net_bridge *br,
41 				       struct bridge_mcast_own_query *query);
42 static void br_multicast_add_router(struct net_bridge *br,
43 				    struct net_bridge_port *port);
44 static void br_ip4_multicast_leave_group(struct net_bridge *br,
45 					 struct net_bridge_port *port,
46 					 __be32 group,
47 					 __u16 vid,
48 					 const unsigned char *src);
49 
50 static void __del_port_router(struct net_bridge_port *p);
51 #if IS_ENABLED(CONFIG_IPV6)
52 static void br_ip6_multicast_leave_group(struct net_bridge *br,
53 					 struct net_bridge_port *port,
54 					 const struct in6_addr *group,
55 					 __u16 vid, const unsigned char *src);
56 #endif
57 unsigned int br_mdb_rehash_seq;
58 
br_ip_equal(const struct br_ip * a,const struct br_ip * b)59 static inline int br_ip_equal(const struct br_ip *a, const struct br_ip *b)
60 {
61 	if (a->proto != b->proto)
62 		return 0;
63 	if (a->vid != b->vid)
64 		return 0;
65 	switch (a->proto) {
66 	case htons(ETH_P_IP):
67 		return a->u.ip4 == b->u.ip4;
68 #if IS_ENABLED(CONFIG_IPV6)
69 	case htons(ETH_P_IPV6):
70 		return ipv6_addr_equal(&a->u.ip6, &b->u.ip6);
71 #endif
72 	}
73 	return 0;
74 }
75 
__br_ip4_hash(struct net_bridge_mdb_htable * mdb,__be32 ip,__u16 vid)76 static inline int __br_ip4_hash(struct net_bridge_mdb_htable *mdb, __be32 ip,
77 				__u16 vid)
78 {
79 	return jhash_2words((__force u32)ip, vid, mdb->secret) & (mdb->max - 1);
80 }
81 
82 #if IS_ENABLED(CONFIG_IPV6)
__br_ip6_hash(struct net_bridge_mdb_htable * mdb,const struct in6_addr * ip,__u16 vid)83 static inline int __br_ip6_hash(struct net_bridge_mdb_htable *mdb,
84 				const struct in6_addr *ip,
85 				__u16 vid)
86 {
87 	return jhash_2words(ipv6_addr_hash(ip), vid,
88 			    mdb->secret) & (mdb->max - 1);
89 }
90 #endif
91 
br_ip_hash(struct net_bridge_mdb_htable * mdb,struct br_ip * ip)92 static inline int br_ip_hash(struct net_bridge_mdb_htable *mdb,
93 			     struct br_ip *ip)
94 {
95 	switch (ip->proto) {
96 	case htons(ETH_P_IP):
97 		return __br_ip4_hash(mdb, ip->u.ip4, ip->vid);
98 #if IS_ENABLED(CONFIG_IPV6)
99 	case htons(ETH_P_IPV6):
100 		return __br_ip6_hash(mdb, &ip->u.ip6, ip->vid);
101 #endif
102 	}
103 	return 0;
104 }
105 
__br_mdb_ip_get(struct net_bridge_mdb_htable * mdb,struct br_ip * dst,int hash)106 static struct net_bridge_mdb_entry *__br_mdb_ip_get(
107 	struct net_bridge_mdb_htable *mdb, struct br_ip *dst, int hash)
108 {
109 	struct net_bridge_mdb_entry *mp;
110 
111 	hlist_for_each_entry_rcu(mp, &mdb->mhash[hash], hlist[mdb->ver]) {
112 		if (br_ip_equal(&mp->addr, dst))
113 			return mp;
114 	}
115 
116 	return NULL;
117 }
118 
br_mdb_ip_get(struct net_bridge_mdb_htable * mdb,struct br_ip * dst)119 struct net_bridge_mdb_entry *br_mdb_ip_get(struct net_bridge_mdb_htable *mdb,
120 					   struct br_ip *dst)
121 {
122 	if (!mdb)
123 		return NULL;
124 
125 	return __br_mdb_ip_get(mdb, dst, br_ip_hash(mdb, dst));
126 }
127 
br_mdb_ip4_get(struct net_bridge_mdb_htable * mdb,__be32 dst,__u16 vid)128 static struct net_bridge_mdb_entry *br_mdb_ip4_get(
129 	struct net_bridge_mdb_htable *mdb, __be32 dst, __u16 vid)
130 {
131 	struct br_ip br_dst;
132 
133 	br_dst.u.ip4 = dst;
134 	br_dst.proto = htons(ETH_P_IP);
135 	br_dst.vid = vid;
136 
137 	return br_mdb_ip_get(mdb, &br_dst);
138 }
139 
140 #if IS_ENABLED(CONFIG_IPV6)
br_mdb_ip6_get(struct net_bridge_mdb_htable * mdb,const struct in6_addr * dst,__u16 vid)141 static struct net_bridge_mdb_entry *br_mdb_ip6_get(
142 	struct net_bridge_mdb_htable *mdb, const struct in6_addr *dst,
143 	__u16 vid)
144 {
145 	struct br_ip br_dst;
146 
147 	br_dst.u.ip6 = *dst;
148 	br_dst.proto = htons(ETH_P_IPV6);
149 	br_dst.vid = vid;
150 
151 	return br_mdb_ip_get(mdb, &br_dst);
152 }
153 #endif
154 
br_mdb_get(struct net_bridge * br,struct sk_buff * skb,u16 vid)155 struct net_bridge_mdb_entry *br_mdb_get(struct net_bridge *br,
156 					struct sk_buff *skb, u16 vid)
157 {
158 	struct net_bridge_mdb_htable *mdb = rcu_dereference(br->mdb);
159 	struct br_ip ip;
160 
161 	if (br->multicast_disabled)
162 		return NULL;
163 
164 	if (BR_INPUT_SKB_CB(skb)->igmp)
165 		return NULL;
166 
167 	ip.proto = skb->protocol;
168 	ip.vid = vid;
169 
170 	switch (skb->protocol) {
171 	case htons(ETH_P_IP):
172 		ip.u.ip4 = ip_hdr(skb)->daddr;
173 		break;
174 #if IS_ENABLED(CONFIG_IPV6)
175 	case htons(ETH_P_IPV6):
176 		ip.u.ip6 = ipv6_hdr(skb)->daddr;
177 		break;
178 #endif
179 	default:
180 		return NULL;
181 	}
182 
183 	return br_mdb_ip_get(mdb, &ip);
184 }
185 
br_mdb_free(struct rcu_head * head)186 static void br_mdb_free(struct rcu_head *head)
187 {
188 	struct net_bridge_mdb_htable *mdb =
189 		container_of(head, struct net_bridge_mdb_htable, rcu);
190 	struct net_bridge_mdb_htable *old = mdb->old;
191 
192 	mdb->old = NULL;
193 	kfree(old->mhash);
194 	kfree(old);
195 }
196 
br_mdb_copy(struct net_bridge_mdb_htable * new,struct net_bridge_mdb_htable * old,int elasticity)197 static int br_mdb_copy(struct net_bridge_mdb_htable *new,
198 		       struct net_bridge_mdb_htable *old,
199 		       int elasticity)
200 {
201 	struct net_bridge_mdb_entry *mp;
202 	int maxlen;
203 	int len;
204 	int i;
205 
206 	for (i = 0; i < old->max; i++)
207 		hlist_for_each_entry(mp, &old->mhash[i], hlist[old->ver])
208 			hlist_add_head(&mp->hlist[new->ver],
209 				       &new->mhash[br_ip_hash(new, &mp->addr)]);
210 
211 	if (!elasticity)
212 		return 0;
213 
214 	maxlen = 0;
215 	for (i = 0; i < new->max; i++) {
216 		len = 0;
217 		hlist_for_each_entry(mp, &new->mhash[i], hlist[new->ver])
218 			len++;
219 		if (len > maxlen)
220 			maxlen = len;
221 	}
222 
223 	return maxlen > elasticity ? -EINVAL : 0;
224 }
225 
br_multicast_free_pg(struct rcu_head * head)226 void br_multicast_free_pg(struct rcu_head *head)
227 {
228 	struct net_bridge_port_group *p =
229 		container_of(head, struct net_bridge_port_group, rcu);
230 
231 	kfree(p);
232 }
233 
br_multicast_free_group(struct rcu_head * head)234 static void br_multicast_free_group(struct rcu_head *head)
235 {
236 	struct net_bridge_mdb_entry *mp =
237 		container_of(head, struct net_bridge_mdb_entry, rcu);
238 
239 	kfree(mp);
240 }
241 
br_multicast_group_expired(unsigned long data)242 static void br_multicast_group_expired(unsigned long data)
243 {
244 	struct net_bridge_mdb_entry *mp = (void *)data;
245 	struct net_bridge *br = mp->br;
246 	struct net_bridge_mdb_htable *mdb;
247 
248 	spin_lock(&br->multicast_lock);
249 	if (!netif_running(br->dev) || timer_pending(&mp->timer))
250 		goto out;
251 
252 	mp->mglist = false;
253 
254 	if (mp->ports)
255 		goto out;
256 
257 	mdb = mlock_dereference(br->mdb, br);
258 
259 	hlist_del_rcu(&mp->hlist[mdb->ver]);
260 	mdb->size--;
261 
262 	call_rcu_bh(&mp->rcu, br_multicast_free_group);
263 
264 out:
265 	spin_unlock(&br->multicast_lock);
266 }
267 
br_multicast_del_pg(struct net_bridge * br,struct net_bridge_port_group * pg)268 static void br_multicast_del_pg(struct net_bridge *br,
269 				struct net_bridge_port_group *pg)
270 {
271 	struct net_bridge_mdb_htable *mdb;
272 	struct net_bridge_mdb_entry *mp;
273 	struct net_bridge_port_group *p;
274 	struct net_bridge_port_group __rcu **pp;
275 
276 	mdb = mlock_dereference(br->mdb, br);
277 
278 	mp = br_mdb_ip_get(mdb, &pg->addr);
279 	if (WARN_ON(!mp))
280 		return;
281 
282 	for (pp = &mp->ports;
283 	     (p = mlock_dereference(*pp, br)) != NULL;
284 	     pp = &p->next) {
285 		if (p != pg)
286 			continue;
287 
288 		rcu_assign_pointer(*pp, p->next);
289 		hlist_del_init(&p->mglist);
290 		del_timer(&p->timer);
291 		br_mdb_notify(br->dev, p->port, &pg->addr, RTM_DELMDB,
292 			      p->flags);
293 		call_rcu_bh(&p->rcu, br_multicast_free_pg);
294 
295 		if (!mp->ports && !mp->mglist &&
296 		    netif_running(br->dev))
297 			mod_timer(&mp->timer, jiffies);
298 
299 		return;
300 	}
301 
302 	WARN_ON(1);
303 }
304 
br_multicast_port_group_expired(unsigned long data)305 static void br_multicast_port_group_expired(unsigned long data)
306 {
307 	struct net_bridge_port_group *pg = (void *)data;
308 	struct net_bridge *br = pg->port->br;
309 
310 	spin_lock(&br->multicast_lock);
311 	if (!netif_running(br->dev) || timer_pending(&pg->timer) ||
312 	    hlist_unhashed(&pg->mglist) || pg->flags & MDB_PG_FLAGS_PERMANENT)
313 		goto out;
314 
315 	br_multicast_del_pg(br, pg);
316 
317 out:
318 	spin_unlock(&br->multicast_lock);
319 }
320 
br_mdb_rehash(struct net_bridge_mdb_htable __rcu ** mdbp,int max,int elasticity)321 static int br_mdb_rehash(struct net_bridge_mdb_htable __rcu **mdbp, int max,
322 			 int elasticity)
323 {
324 	struct net_bridge_mdb_htable *old = rcu_dereference_protected(*mdbp, 1);
325 	struct net_bridge_mdb_htable *mdb;
326 	int err;
327 
328 	mdb = kmalloc(sizeof(*mdb), GFP_ATOMIC);
329 	if (!mdb)
330 		return -ENOMEM;
331 
332 	mdb->max = max;
333 	mdb->old = old;
334 
335 	mdb->mhash = kzalloc(max * sizeof(*mdb->mhash), GFP_ATOMIC);
336 	if (!mdb->mhash) {
337 		kfree(mdb);
338 		return -ENOMEM;
339 	}
340 
341 	mdb->size = old ? old->size : 0;
342 	mdb->ver = old ? old->ver ^ 1 : 0;
343 
344 	if (!old || elasticity)
345 		get_random_bytes(&mdb->secret, sizeof(mdb->secret));
346 	else
347 		mdb->secret = old->secret;
348 
349 	if (!old)
350 		goto out;
351 
352 	err = br_mdb_copy(mdb, old, elasticity);
353 	if (err) {
354 		kfree(mdb->mhash);
355 		kfree(mdb);
356 		return err;
357 	}
358 
359 	br_mdb_rehash_seq++;
360 	call_rcu_bh(&mdb->rcu, br_mdb_free);
361 
362 out:
363 	rcu_assign_pointer(*mdbp, mdb);
364 
365 	return 0;
366 }
367 
br_ip4_multicast_alloc_query(struct net_bridge * br,__be32 group,u8 * igmp_type)368 static struct sk_buff *br_ip4_multicast_alloc_query(struct net_bridge *br,
369 						    __be32 group,
370 						    u8 *igmp_type)
371 {
372 	struct igmpv3_query *ihv3;
373 	size_t igmp_hdr_size;
374 	struct sk_buff *skb;
375 	struct igmphdr *ih;
376 	struct ethhdr *eth;
377 	struct iphdr *iph;
378 
379 	igmp_hdr_size = sizeof(*ih);
380 	if (br->multicast_igmp_version == 3)
381 		igmp_hdr_size = sizeof(*ihv3);
382 	skb = netdev_alloc_skb_ip_align(br->dev, sizeof(*eth) + sizeof(*iph) +
383 						 igmp_hdr_size + 4);
384 	if (!skb)
385 		goto out;
386 
387 	skb->protocol = htons(ETH_P_IP);
388 
389 	skb_reset_mac_header(skb);
390 	eth = eth_hdr(skb);
391 
392 	ether_addr_copy(eth->h_source, br->dev->dev_addr);
393 	eth->h_dest[0] = 1;
394 	eth->h_dest[1] = 0;
395 	eth->h_dest[2] = 0x5e;
396 	eth->h_dest[3] = 0;
397 	eth->h_dest[4] = 0;
398 	eth->h_dest[5] = 1;
399 	eth->h_proto = htons(ETH_P_IP);
400 	skb_put(skb, sizeof(*eth));
401 
402 	skb_set_network_header(skb, skb->len);
403 	iph = ip_hdr(skb);
404 
405 	iph->version = 4;
406 	iph->ihl = 6;
407 	iph->tos = 0xc0;
408 	iph->tot_len = htons(sizeof(*iph) + igmp_hdr_size + 4);
409 	iph->id = 0;
410 	iph->frag_off = htons(IP_DF);
411 	iph->ttl = 1;
412 	iph->protocol = IPPROTO_IGMP;
413 	iph->saddr = br->multicast_query_use_ifaddr ?
414 		     inet_select_addr(br->dev, 0, RT_SCOPE_LINK) : 0;
415 	iph->daddr = htonl(INADDR_ALLHOSTS_GROUP);
416 	((u8 *)&iph[1])[0] = IPOPT_RA;
417 	((u8 *)&iph[1])[1] = 4;
418 	((u8 *)&iph[1])[2] = 0;
419 	((u8 *)&iph[1])[3] = 0;
420 	ip_send_check(iph);
421 	skb_put(skb, 24);
422 
423 	skb_set_transport_header(skb, skb->len);
424 	*igmp_type = IGMP_HOST_MEMBERSHIP_QUERY;
425 
426 	switch (br->multicast_igmp_version) {
427 	case 2:
428 		ih = igmp_hdr(skb);
429 		ih->type = IGMP_HOST_MEMBERSHIP_QUERY;
430 		ih->code = (group ? br->multicast_last_member_interval :
431 				    br->multicast_query_response_interval) /
432 			   (HZ / IGMP_TIMER_SCALE);
433 		ih->group = group;
434 		ih->csum = 0;
435 		ih->csum = ip_compute_csum((void *)ih, sizeof(*ih));
436 		break;
437 	case 3:
438 		ihv3 = igmpv3_query_hdr(skb);
439 		ihv3->type = IGMP_HOST_MEMBERSHIP_QUERY;
440 		ihv3->code = (group ? br->multicast_last_member_interval :
441 				      br->multicast_query_response_interval) /
442 			     (HZ / IGMP_TIMER_SCALE);
443 		ihv3->group = group;
444 		ihv3->qqic = br->multicast_query_interval / HZ;
445 		ihv3->nsrcs = 0;
446 		ihv3->resv = 0;
447 		ihv3->suppress = 0;
448 		ihv3->qrv = 2;
449 		ihv3->csum = 0;
450 		ihv3->csum = ip_compute_csum((void *)ihv3, sizeof(*ihv3));
451 		break;
452 	}
453 
454 	skb_put(skb, igmp_hdr_size);
455 	__skb_pull(skb, sizeof(*eth));
456 
457 out:
458 	return skb;
459 }
460 
461 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_alloc_query(struct net_bridge * br,const struct in6_addr * grp,u8 * igmp_type)462 static struct sk_buff *br_ip6_multicast_alloc_query(struct net_bridge *br,
463 						    const struct in6_addr *grp,
464 						    u8 *igmp_type)
465 {
466 	struct mld2_query *mld2q;
467 	unsigned long interval;
468 	struct ipv6hdr *ip6h;
469 	struct mld_msg *mldq;
470 	size_t mld_hdr_size;
471 	struct sk_buff *skb;
472 	struct ethhdr *eth;
473 	u8 *hopopt;
474 
475 	mld_hdr_size = sizeof(*mldq);
476 	if (br->multicast_mld_version == 2)
477 		mld_hdr_size = sizeof(*mld2q);
478 	skb = netdev_alloc_skb_ip_align(br->dev, sizeof(*eth) + sizeof(*ip6h) +
479 						 8 + mld_hdr_size);
480 	if (!skb)
481 		goto out;
482 
483 	skb->protocol = htons(ETH_P_IPV6);
484 
485 	/* Ethernet header */
486 	skb_reset_mac_header(skb);
487 	eth = eth_hdr(skb);
488 
489 	ether_addr_copy(eth->h_source, br->dev->dev_addr);
490 	eth->h_proto = htons(ETH_P_IPV6);
491 	skb_put(skb, sizeof(*eth));
492 
493 	/* IPv6 header + HbH option */
494 	skb_set_network_header(skb, skb->len);
495 	ip6h = ipv6_hdr(skb);
496 
497 	*(__force __be32 *)ip6h = htonl(0x60000000);
498 	ip6h->payload_len = htons(8 + mld_hdr_size);
499 	ip6h->nexthdr = IPPROTO_HOPOPTS;
500 	ip6h->hop_limit = 1;
501 	ipv6_addr_set(&ip6h->daddr, htonl(0xff020000), 0, 0, htonl(1));
502 	if (ipv6_dev_get_saddr(dev_net(br->dev), br->dev, &ip6h->daddr, 0,
503 			       &ip6h->saddr)) {
504 		kfree_skb(skb);
505 		br->has_ipv6_addr = 0;
506 		return NULL;
507 	}
508 
509 	br->has_ipv6_addr = 1;
510 	ipv6_eth_mc_map(&ip6h->daddr, eth->h_dest);
511 
512 	hopopt = (u8 *)(ip6h + 1);
513 	hopopt[0] = IPPROTO_ICMPV6;		/* next hdr */
514 	hopopt[1] = 0;				/* length of HbH */
515 	hopopt[2] = IPV6_TLV_ROUTERALERT;	/* Router Alert */
516 	hopopt[3] = 2;				/* Length of RA Option */
517 	hopopt[4] = 0;				/* Type = 0x0000 (MLD) */
518 	hopopt[5] = 0;
519 	hopopt[6] = IPV6_TLV_PAD1;		/* Pad1 */
520 	hopopt[7] = IPV6_TLV_PAD1;		/* Pad1 */
521 
522 	skb_put(skb, sizeof(*ip6h) + 8);
523 
524 	/* ICMPv6 */
525 	skb_set_transport_header(skb, skb->len);
526 	interval = ipv6_addr_any(grp) ?
527 			br->multicast_query_response_interval :
528 			br->multicast_last_member_interval;
529 	*igmp_type = ICMPV6_MGM_QUERY;
530 	switch (br->multicast_mld_version) {
531 	case 1:
532 		mldq = (struct mld_msg *)icmp6_hdr(skb);
533 		mldq->mld_type = ICMPV6_MGM_QUERY;
534 		mldq->mld_code = 0;
535 		mldq->mld_cksum = 0;
536 		mldq->mld_maxdelay = htons((u16)jiffies_to_msecs(interval));
537 		mldq->mld_reserved = 0;
538 		mldq->mld_mca = *grp;
539 		mldq->mld_cksum = csum_ipv6_magic(&ip6h->saddr, &ip6h->daddr,
540 						  sizeof(*mldq), IPPROTO_ICMPV6,
541 						  csum_partial(mldq,
542 							       sizeof(*mldq),
543 							       0));
544 		break;
545 	case 2:
546 		mld2q = (struct mld2_query *)icmp6_hdr(skb);
547 		mld2q->mld2q_mrc = htons((u16)jiffies_to_msecs(interval));
548 		mld2q->mld2q_type = ICMPV6_MGM_QUERY;
549 		mld2q->mld2q_code = 0;
550 		mld2q->mld2q_cksum = 0;
551 		mld2q->mld2q_resv1 = 0;
552 		mld2q->mld2q_resv2 = 0;
553 		mld2q->mld2q_suppress = 0;
554 		mld2q->mld2q_qrv = 2;
555 		mld2q->mld2q_nsrcs = 0;
556 		mld2q->mld2q_qqic = br->multicast_query_interval / HZ;
557 		mld2q->mld2q_mca = *grp;
558 		mld2q->mld2q_cksum = csum_ipv6_magic(&ip6h->saddr, &ip6h->daddr,
559 						     sizeof(*mld2q),
560 						     IPPROTO_ICMPV6,
561 						     csum_partial(mld2q,
562 								  sizeof(*mld2q),
563 								  0));
564 		break;
565 	}
566 	skb_put(skb, mld_hdr_size);
567 
568 	__skb_pull(skb, sizeof(*eth));
569 
570 out:
571 	return skb;
572 }
573 #endif
574 
br_multicast_alloc_query(struct net_bridge * br,struct br_ip * addr,u8 * igmp_type)575 static struct sk_buff *br_multicast_alloc_query(struct net_bridge *br,
576 						struct br_ip *addr,
577 						u8 *igmp_type)
578 {
579 	switch (addr->proto) {
580 	case htons(ETH_P_IP):
581 		return br_ip4_multicast_alloc_query(br, addr->u.ip4, igmp_type);
582 #if IS_ENABLED(CONFIG_IPV6)
583 	case htons(ETH_P_IPV6):
584 		return br_ip6_multicast_alloc_query(br, &addr->u.ip6,
585 						    igmp_type);
586 #endif
587 	}
588 	return NULL;
589 }
590 
br_multicast_get_group(struct net_bridge * br,struct net_bridge_port * port,struct br_ip * group,int hash)591 static struct net_bridge_mdb_entry *br_multicast_get_group(
592 	struct net_bridge *br, struct net_bridge_port *port,
593 	struct br_ip *group, int hash)
594 {
595 	struct net_bridge_mdb_htable *mdb;
596 	struct net_bridge_mdb_entry *mp;
597 	unsigned int count = 0;
598 	unsigned int max;
599 	int elasticity;
600 	int err;
601 
602 	mdb = rcu_dereference_protected(br->mdb, 1);
603 	hlist_for_each_entry(mp, &mdb->mhash[hash], hlist[mdb->ver]) {
604 		count++;
605 		if (unlikely(br_ip_equal(group, &mp->addr)))
606 			return mp;
607 	}
608 
609 	elasticity = 0;
610 	max = mdb->max;
611 
612 	if (unlikely(count > br->hash_elasticity && count)) {
613 		if (net_ratelimit())
614 			br_info(br, "Multicast hash table "
615 				"chain limit reached: %s\n",
616 				port ? port->dev->name : br->dev->name);
617 
618 		elasticity = br->hash_elasticity;
619 	}
620 
621 	if (mdb->size >= max) {
622 		max *= 2;
623 		if (unlikely(max > br->hash_max)) {
624 			br_warn(br, "Multicast hash table maximum of %d "
625 				"reached, disabling snooping: %s\n",
626 				br->hash_max,
627 				port ? port->dev->name : br->dev->name);
628 			err = -E2BIG;
629 disable:
630 			br->multicast_disabled = 1;
631 			goto err;
632 		}
633 	}
634 
635 	if (max > mdb->max || elasticity) {
636 		if (mdb->old) {
637 			if (net_ratelimit())
638 				br_info(br, "Multicast hash table "
639 					"on fire: %s\n",
640 					port ? port->dev->name : br->dev->name);
641 			err = -EEXIST;
642 			goto err;
643 		}
644 
645 		err = br_mdb_rehash(&br->mdb, max, elasticity);
646 		if (err) {
647 			br_warn(br, "Cannot rehash multicast "
648 				"hash table, disabling snooping: %s, %d, %d\n",
649 				port ? port->dev->name : br->dev->name,
650 				mdb->size, err);
651 			goto disable;
652 		}
653 
654 		err = -EAGAIN;
655 		goto err;
656 	}
657 
658 	return NULL;
659 
660 err:
661 	mp = ERR_PTR(err);
662 	return mp;
663 }
664 
br_multicast_new_group(struct net_bridge * br,struct net_bridge_port * p,struct br_ip * group)665 struct net_bridge_mdb_entry *br_multicast_new_group(struct net_bridge *br,
666 						    struct net_bridge_port *p,
667 						    struct br_ip *group)
668 {
669 	struct net_bridge_mdb_htable *mdb;
670 	struct net_bridge_mdb_entry *mp;
671 	int hash;
672 	int err;
673 
674 	mdb = rcu_dereference_protected(br->mdb, 1);
675 	if (!mdb) {
676 		err = br_mdb_rehash(&br->mdb, BR_HASH_SIZE, 0);
677 		if (err)
678 			return ERR_PTR(err);
679 		goto rehash;
680 	}
681 
682 	hash = br_ip_hash(mdb, group);
683 	mp = br_multicast_get_group(br, p, group, hash);
684 	switch (PTR_ERR(mp)) {
685 	case 0:
686 		break;
687 
688 	case -EAGAIN:
689 rehash:
690 		mdb = rcu_dereference_protected(br->mdb, 1);
691 		hash = br_ip_hash(mdb, group);
692 		break;
693 
694 	default:
695 		goto out;
696 	}
697 
698 	mp = kzalloc(sizeof(*mp), GFP_ATOMIC);
699 	if (unlikely(!mp))
700 		return ERR_PTR(-ENOMEM);
701 
702 	mp->br = br;
703 	mp->addr = *group;
704 	setup_timer(&mp->timer, br_multicast_group_expired,
705 		    (unsigned long)mp);
706 
707 	hlist_add_head_rcu(&mp->hlist[mdb->ver], &mdb->mhash[hash]);
708 	mdb->size++;
709 
710 out:
711 	return mp;
712 }
713 
br_multicast_new_port_group(struct net_bridge_port * port,struct br_ip * group,struct net_bridge_port_group __rcu * next,unsigned char flags,const unsigned char * src)714 struct net_bridge_port_group *br_multicast_new_port_group(
715 			struct net_bridge_port *port,
716 			struct br_ip *group,
717 			struct net_bridge_port_group __rcu *next,
718 			unsigned char flags,
719 			const unsigned char *src)
720 {
721 	struct net_bridge_port_group *p;
722 
723 	p = kzalloc(sizeof(*p), GFP_ATOMIC);
724 	if (unlikely(!p))
725 		return NULL;
726 
727 	p->addr = *group;
728 	p->port = port;
729 	p->flags = flags;
730 	rcu_assign_pointer(p->next, next);
731 	hlist_add_head(&p->mglist, &port->mglist);
732 	setup_timer(&p->timer, br_multicast_port_group_expired,
733 		    (unsigned long)p);
734 
735 	if (src)
736 		memcpy(p->eth_addr, src, ETH_ALEN);
737 	else
738 		memset(p->eth_addr, 0xff, ETH_ALEN);
739 
740 	return p;
741 }
742 
br_port_group_equal(struct net_bridge_port_group * p,struct net_bridge_port * port,const unsigned char * src)743 static bool br_port_group_equal(struct net_bridge_port_group *p,
744 				struct net_bridge_port *port,
745 				const unsigned char *src)
746 {
747 	if (p->port != port)
748 		return false;
749 
750 	if (!(port->flags & BR_MULTICAST_TO_UNICAST))
751 		return true;
752 
753 	return ether_addr_equal(src, p->eth_addr);
754 }
755 
br_multicast_add_group(struct net_bridge * br,struct net_bridge_port * port,struct br_ip * group,const unsigned char * src)756 static int br_multicast_add_group(struct net_bridge *br,
757 				  struct net_bridge_port *port,
758 				  struct br_ip *group,
759 				  const unsigned char *src)
760 {
761 	struct net_bridge_port_group __rcu **pp;
762 	struct net_bridge_port_group *p;
763 	struct net_bridge_mdb_entry *mp;
764 	unsigned long now = jiffies;
765 	int err;
766 
767 	spin_lock(&br->multicast_lock);
768 	if (!netif_running(br->dev) ||
769 	    (port && port->state == BR_STATE_DISABLED))
770 		goto out;
771 
772 	mp = br_multicast_new_group(br, port, group);
773 	err = PTR_ERR(mp);
774 	if (IS_ERR(mp))
775 		goto err;
776 
777 	if (!port) {
778 		mp->mglist = true;
779 		mod_timer(&mp->timer, now + br->multicast_membership_interval);
780 		goto out;
781 	}
782 
783 	for (pp = &mp->ports;
784 	     (p = mlock_dereference(*pp, br)) != NULL;
785 	     pp = &p->next) {
786 		if (br_port_group_equal(p, port, src))
787 			goto found;
788 		if ((unsigned long)p->port < (unsigned long)port)
789 			break;
790 	}
791 
792 	p = br_multicast_new_port_group(port, group, *pp, 0, src);
793 	if (unlikely(!p))
794 		goto err;
795 	rcu_assign_pointer(*pp, p);
796 	br_mdb_notify(br->dev, port, group, RTM_NEWMDB, 0);
797 
798 found:
799 	mod_timer(&p->timer, now + br->multicast_membership_interval);
800 out:
801 	err = 0;
802 
803 err:
804 	spin_unlock(&br->multicast_lock);
805 	return err;
806 }
807 
br_ip4_multicast_add_group(struct net_bridge * br,struct net_bridge_port * port,__be32 group,__u16 vid,const unsigned char * src)808 static int br_ip4_multicast_add_group(struct net_bridge *br,
809 				      struct net_bridge_port *port,
810 				      __be32 group,
811 				      __u16 vid,
812 				      const unsigned char *src)
813 {
814 	struct br_ip br_group;
815 
816 	if (ipv4_is_local_multicast(group))
817 		return 0;
818 
819 	br_group.u.ip4 = group;
820 	br_group.proto = htons(ETH_P_IP);
821 	br_group.vid = vid;
822 
823 	return br_multicast_add_group(br, port, &br_group, src);
824 }
825 
826 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_add_group(struct net_bridge * br,struct net_bridge_port * port,const struct in6_addr * group,__u16 vid,const unsigned char * src)827 static int br_ip6_multicast_add_group(struct net_bridge *br,
828 				      struct net_bridge_port *port,
829 				      const struct in6_addr *group,
830 				      __u16 vid,
831 				      const unsigned char *src)
832 {
833 	struct br_ip br_group;
834 
835 	if (ipv6_addr_is_ll_all_nodes(group))
836 		return 0;
837 
838 	br_group.u.ip6 = *group;
839 	br_group.proto = htons(ETH_P_IPV6);
840 	br_group.vid = vid;
841 
842 	return br_multicast_add_group(br, port, &br_group, src);
843 }
844 #endif
845 
br_multicast_router_expired(unsigned long data)846 static void br_multicast_router_expired(unsigned long data)
847 {
848 	struct net_bridge_port *port = (void *)data;
849 	struct net_bridge *br = port->br;
850 
851 	spin_lock(&br->multicast_lock);
852 	if (port->multicast_router == MDB_RTR_TYPE_DISABLED ||
853 	    port->multicast_router == MDB_RTR_TYPE_PERM ||
854 	    timer_pending(&port->multicast_router_timer))
855 		goto out;
856 
857 	__del_port_router(port);
858 out:
859 	spin_unlock(&br->multicast_lock);
860 }
861 
br_multicast_local_router_expired(unsigned long data)862 static void br_multicast_local_router_expired(unsigned long data)
863 {
864 }
865 
br_multicast_querier_expired(struct net_bridge * br,struct bridge_mcast_own_query * query)866 static void br_multicast_querier_expired(struct net_bridge *br,
867 					 struct bridge_mcast_own_query *query)
868 {
869 	spin_lock(&br->multicast_lock);
870 	if (!netif_running(br->dev) || br->multicast_disabled)
871 		goto out;
872 
873 	br_multicast_start_querier(br, query);
874 
875 out:
876 	spin_unlock(&br->multicast_lock);
877 }
878 
br_ip4_multicast_querier_expired(unsigned long data)879 static void br_ip4_multicast_querier_expired(unsigned long data)
880 {
881 	struct net_bridge *br = (void *)data;
882 
883 	br_multicast_querier_expired(br, &br->ip4_own_query);
884 }
885 
886 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_querier_expired(unsigned long data)887 static void br_ip6_multicast_querier_expired(unsigned long data)
888 {
889 	struct net_bridge *br = (void *)data;
890 
891 	br_multicast_querier_expired(br, &br->ip6_own_query);
892 }
893 #endif
894 
br_multicast_select_own_querier(struct net_bridge * br,struct br_ip * ip,struct sk_buff * skb)895 static void br_multicast_select_own_querier(struct net_bridge *br,
896 					    struct br_ip *ip,
897 					    struct sk_buff *skb)
898 {
899 	if (ip->proto == htons(ETH_P_IP))
900 		br->ip4_querier.addr.u.ip4 = ip_hdr(skb)->saddr;
901 #if IS_ENABLED(CONFIG_IPV6)
902 	else
903 		br->ip6_querier.addr.u.ip6 = ipv6_hdr(skb)->saddr;
904 #endif
905 }
906 
__br_multicast_send_query(struct net_bridge * br,struct net_bridge_port * port,struct br_ip * ip)907 static void __br_multicast_send_query(struct net_bridge *br,
908 				      struct net_bridge_port *port,
909 				      struct br_ip *ip)
910 {
911 	struct sk_buff *skb;
912 	u8 igmp_type;
913 
914 	skb = br_multicast_alloc_query(br, ip, &igmp_type);
915 	if (!skb)
916 		return;
917 
918 	if (port) {
919 		skb->dev = port->dev;
920 		br_multicast_count(br, port, skb, igmp_type,
921 				   BR_MCAST_DIR_TX);
922 		NF_HOOK(NFPROTO_BRIDGE, NF_BR_LOCAL_OUT,
923 			dev_net(port->dev), NULL, skb, NULL, skb->dev,
924 			br_dev_queue_push_xmit);
925 	} else {
926 		br_multicast_select_own_querier(br, ip, skb);
927 		br_multicast_count(br, port, skb, igmp_type,
928 				   BR_MCAST_DIR_RX);
929 		netif_rx(skb);
930 	}
931 }
932 
br_multicast_send_query(struct net_bridge * br,struct net_bridge_port * port,struct bridge_mcast_own_query * own_query)933 static void br_multicast_send_query(struct net_bridge *br,
934 				    struct net_bridge_port *port,
935 				    struct bridge_mcast_own_query *own_query)
936 {
937 	struct bridge_mcast_other_query *other_query = NULL;
938 	struct br_ip br_group;
939 	unsigned long time;
940 
941 	if (!netif_running(br->dev) || br->multicast_disabled ||
942 	    !br->multicast_querier)
943 		return;
944 
945 	memset(&br_group.u, 0, sizeof(br_group.u));
946 
947 	if (port ? (own_query == &port->ip4_own_query) :
948 		   (own_query == &br->ip4_own_query)) {
949 		other_query = &br->ip4_other_query;
950 		br_group.proto = htons(ETH_P_IP);
951 #if IS_ENABLED(CONFIG_IPV6)
952 	} else {
953 		other_query = &br->ip6_other_query;
954 		br_group.proto = htons(ETH_P_IPV6);
955 #endif
956 	}
957 
958 	if (!other_query || timer_pending(&other_query->timer))
959 		return;
960 
961 	__br_multicast_send_query(br, port, &br_group);
962 
963 	time = jiffies;
964 	time += own_query->startup_sent < br->multicast_startup_query_count ?
965 		br->multicast_startup_query_interval :
966 		br->multicast_query_interval;
967 	mod_timer(&own_query->timer, time);
968 }
969 
970 static void
br_multicast_port_query_expired(struct net_bridge_port * port,struct bridge_mcast_own_query * query)971 br_multicast_port_query_expired(struct net_bridge_port *port,
972 				struct bridge_mcast_own_query *query)
973 {
974 	struct net_bridge *br = port->br;
975 
976 	spin_lock(&br->multicast_lock);
977 	if (port->state == BR_STATE_DISABLED ||
978 	    port->state == BR_STATE_BLOCKING)
979 		goto out;
980 
981 	if (query->startup_sent < br->multicast_startup_query_count)
982 		query->startup_sent++;
983 
984 	br_multicast_send_query(port->br, port, query);
985 
986 out:
987 	spin_unlock(&br->multicast_lock);
988 }
989 
br_ip4_multicast_port_query_expired(unsigned long data)990 static void br_ip4_multicast_port_query_expired(unsigned long data)
991 {
992 	struct net_bridge_port *port = (void *)data;
993 
994 	br_multicast_port_query_expired(port, &port->ip4_own_query);
995 }
996 
997 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_port_query_expired(unsigned long data)998 static void br_ip6_multicast_port_query_expired(unsigned long data)
999 {
1000 	struct net_bridge_port *port = (void *)data;
1001 
1002 	br_multicast_port_query_expired(port, &port->ip6_own_query);
1003 }
1004 #endif
1005 
br_mc_disabled_update(struct net_device * dev,bool value)1006 static void br_mc_disabled_update(struct net_device *dev, bool value)
1007 {
1008 	struct switchdev_attr attr = {
1009 		.orig_dev = dev,
1010 		.id = SWITCHDEV_ATTR_ID_BRIDGE_MC_DISABLED,
1011 		.flags = SWITCHDEV_F_DEFER,
1012 		.u.mc_disabled = value,
1013 	};
1014 
1015 	switchdev_port_attr_set(dev, &attr);
1016 }
1017 
br_multicast_add_port(struct net_bridge_port * port)1018 int br_multicast_add_port(struct net_bridge_port *port)
1019 {
1020 	port->multicast_router = MDB_RTR_TYPE_TEMP_QUERY;
1021 
1022 	setup_timer(&port->multicast_router_timer, br_multicast_router_expired,
1023 		    (unsigned long)port);
1024 	setup_timer(&port->ip4_own_query.timer,
1025 		    br_ip4_multicast_port_query_expired, (unsigned long)port);
1026 #if IS_ENABLED(CONFIG_IPV6)
1027 	setup_timer(&port->ip6_own_query.timer,
1028 		    br_ip6_multicast_port_query_expired, (unsigned long)port);
1029 #endif
1030 	br_mc_disabled_update(port->dev, port->br->multicast_disabled);
1031 
1032 	port->mcast_stats = netdev_alloc_pcpu_stats(struct bridge_mcast_stats);
1033 	if (!port->mcast_stats)
1034 		return -ENOMEM;
1035 
1036 	return 0;
1037 }
1038 
br_multicast_del_port(struct net_bridge_port * port)1039 void br_multicast_del_port(struct net_bridge_port *port)
1040 {
1041 	struct net_bridge *br = port->br;
1042 	struct net_bridge_port_group *pg;
1043 	struct hlist_node *n;
1044 
1045 	/* Take care of the remaining groups, only perm ones should be left */
1046 	spin_lock_bh(&br->multicast_lock);
1047 	hlist_for_each_entry_safe(pg, n, &port->mglist, mglist)
1048 		br_multicast_del_pg(br, pg);
1049 	spin_unlock_bh(&br->multicast_lock);
1050 	del_timer_sync(&port->multicast_router_timer);
1051 	free_percpu(port->mcast_stats);
1052 }
1053 
br_multicast_enable(struct bridge_mcast_own_query * query)1054 static void br_multicast_enable(struct bridge_mcast_own_query *query)
1055 {
1056 	query->startup_sent = 0;
1057 
1058 	if (try_to_del_timer_sync(&query->timer) >= 0 ||
1059 	    del_timer(&query->timer))
1060 		mod_timer(&query->timer, jiffies);
1061 }
1062 
__br_multicast_enable_port(struct net_bridge_port * port)1063 static void __br_multicast_enable_port(struct net_bridge_port *port)
1064 {
1065 	struct net_bridge *br = port->br;
1066 
1067 	if (br->multicast_disabled || !netif_running(br->dev))
1068 		return;
1069 
1070 	br_multicast_enable(&port->ip4_own_query);
1071 #if IS_ENABLED(CONFIG_IPV6)
1072 	br_multicast_enable(&port->ip6_own_query);
1073 #endif
1074 	if (port->multicast_router == MDB_RTR_TYPE_PERM &&
1075 	    hlist_unhashed(&port->rlist))
1076 		br_multicast_add_router(br, port);
1077 }
1078 
br_multicast_enable_port(struct net_bridge_port * port)1079 void br_multicast_enable_port(struct net_bridge_port *port)
1080 {
1081 	struct net_bridge *br = port->br;
1082 
1083 	spin_lock(&br->multicast_lock);
1084 	__br_multicast_enable_port(port);
1085 	spin_unlock(&br->multicast_lock);
1086 }
1087 
br_multicast_disable_port(struct net_bridge_port * port)1088 void br_multicast_disable_port(struct net_bridge_port *port)
1089 {
1090 	struct net_bridge *br = port->br;
1091 	struct net_bridge_port_group *pg;
1092 	struct hlist_node *n;
1093 
1094 	spin_lock(&br->multicast_lock);
1095 	hlist_for_each_entry_safe(pg, n, &port->mglist, mglist)
1096 		if (!(pg->flags & MDB_PG_FLAGS_PERMANENT))
1097 			br_multicast_del_pg(br, pg);
1098 
1099 	__del_port_router(port);
1100 
1101 	del_timer(&port->multicast_router_timer);
1102 	del_timer(&port->ip4_own_query.timer);
1103 #if IS_ENABLED(CONFIG_IPV6)
1104 	del_timer(&port->ip6_own_query.timer);
1105 #endif
1106 	spin_unlock(&br->multicast_lock);
1107 }
1108 
br_ip4_multicast_igmp3_report(struct net_bridge * br,struct net_bridge_port * port,struct sk_buff * skb,u16 vid)1109 static int br_ip4_multicast_igmp3_report(struct net_bridge *br,
1110 					 struct net_bridge_port *port,
1111 					 struct sk_buff *skb,
1112 					 u16 vid)
1113 {
1114 	const unsigned char *src;
1115 	struct igmpv3_report *ih;
1116 	struct igmpv3_grec *grec;
1117 	int i;
1118 	int len;
1119 	int num;
1120 	int type;
1121 	int err = 0;
1122 	__be32 group;
1123 	u16 nsrcs;
1124 
1125 	ih = igmpv3_report_hdr(skb);
1126 	num = ntohs(ih->ngrec);
1127 	len = skb_transport_offset(skb) + sizeof(*ih);
1128 
1129 	for (i = 0; i < num; i++) {
1130 		len += sizeof(*grec);
1131 		if (!pskb_may_pull(skb, len))
1132 			return -EINVAL;
1133 
1134 		grec = (void *)(skb->data + len - sizeof(*grec));
1135 		group = grec->grec_mca;
1136 		type = grec->grec_type;
1137 		nsrcs = ntohs(grec->grec_nsrcs);
1138 
1139 		len += nsrcs * 4;
1140 		if (!pskb_may_pull(skb, len))
1141 			return -EINVAL;
1142 
1143 		/* We treat this as an IGMPv2 report for now. */
1144 		switch (type) {
1145 		case IGMPV3_MODE_IS_INCLUDE:
1146 		case IGMPV3_MODE_IS_EXCLUDE:
1147 		case IGMPV3_CHANGE_TO_INCLUDE:
1148 		case IGMPV3_CHANGE_TO_EXCLUDE:
1149 		case IGMPV3_ALLOW_NEW_SOURCES:
1150 		case IGMPV3_BLOCK_OLD_SOURCES:
1151 			break;
1152 
1153 		default:
1154 			continue;
1155 		}
1156 
1157 		src = eth_hdr(skb)->h_source;
1158 		if ((type == IGMPV3_CHANGE_TO_INCLUDE ||
1159 		     type == IGMPV3_MODE_IS_INCLUDE) &&
1160 		    nsrcs == 0) {
1161 			br_ip4_multicast_leave_group(br, port, group, vid, src);
1162 		} else {
1163 			err = br_ip4_multicast_add_group(br, port, group, vid,
1164 							 src);
1165 			if (err)
1166 				break;
1167 		}
1168 	}
1169 
1170 	return err;
1171 }
1172 
1173 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_mld2_report(struct net_bridge * br,struct net_bridge_port * port,struct sk_buff * skb,u16 vid)1174 static int br_ip6_multicast_mld2_report(struct net_bridge *br,
1175 					struct net_bridge_port *port,
1176 					struct sk_buff *skb,
1177 					u16 vid)
1178 {
1179 	const unsigned char *src;
1180 	struct icmp6hdr *icmp6h;
1181 	struct mld2_grec *grec;
1182 	int i;
1183 	int len;
1184 	int num;
1185 	int err = 0;
1186 
1187 	if (!pskb_may_pull(skb, sizeof(*icmp6h)))
1188 		return -EINVAL;
1189 
1190 	icmp6h = icmp6_hdr(skb);
1191 	num = ntohs(icmp6h->icmp6_dataun.un_data16[1]);
1192 	len = skb_transport_offset(skb) + sizeof(*icmp6h);
1193 
1194 	for (i = 0; i < num; i++) {
1195 		__be16 *_nsrcs, __nsrcs;
1196 		u16 nsrcs;
1197 
1198 		_nsrcs = skb_header_pointer(skb,
1199 					    len + offsetof(struct mld2_grec,
1200 							   grec_nsrcs),
1201 					    sizeof(__nsrcs), &__nsrcs);
1202 		if (!_nsrcs)
1203 			return -EINVAL;
1204 
1205 		nsrcs = ntohs(*_nsrcs);
1206 
1207 		if (!pskb_may_pull(skb,
1208 				   len + sizeof(*grec) +
1209 				   sizeof(struct in6_addr) * nsrcs))
1210 			return -EINVAL;
1211 
1212 		grec = (struct mld2_grec *)(skb->data + len);
1213 		len += sizeof(*grec) +
1214 		       sizeof(struct in6_addr) * nsrcs;
1215 
1216 		/* We treat these as MLDv1 reports for now. */
1217 		switch (grec->grec_type) {
1218 		case MLD2_MODE_IS_INCLUDE:
1219 		case MLD2_MODE_IS_EXCLUDE:
1220 		case MLD2_CHANGE_TO_INCLUDE:
1221 		case MLD2_CHANGE_TO_EXCLUDE:
1222 		case MLD2_ALLOW_NEW_SOURCES:
1223 		case MLD2_BLOCK_OLD_SOURCES:
1224 			break;
1225 
1226 		default:
1227 			continue;
1228 		}
1229 
1230 		src = eth_hdr(skb)->h_source;
1231 		if ((grec->grec_type == MLD2_CHANGE_TO_INCLUDE ||
1232 		     grec->grec_type == MLD2_MODE_IS_INCLUDE) &&
1233 		    nsrcs == 0) {
1234 			br_ip6_multicast_leave_group(br, port, &grec->grec_mca,
1235 						     vid, src);
1236 		} else {
1237 			err = br_ip6_multicast_add_group(br, port,
1238 							 &grec->grec_mca, vid,
1239 							 src);
1240 			if (err)
1241 				break;
1242 		}
1243 	}
1244 
1245 	return err;
1246 }
1247 #endif
1248 
br_ip4_multicast_select_querier(struct net_bridge * br,struct net_bridge_port * port,__be32 saddr)1249 static bool br_ip4_multicast_select_querier(struct net_bridge *br,
1250 					    struct net_bridge_port *port,
1251 					    __be32 saddr)
1252 {
1253 	if (!timer_pending(&br->ip4_own_query.timer) &&
1254 	    !timer_pending(&br->ip4_other_query.timer))
1255 		goto update;
1256 
1257 	if (!br->ip4_querier.addr.u.ip4)
1258 		goto update;
1259 
1260 	if (ntohl(saddr) <= ntohl(br->ip4_querier.addr.u.ip4))
1261 		goto update;
1262 
1263 	return false;
1264 
1265 update:
1266 	br->ip4_querier.addr.u.ip4 = saddr;
1267 
1268 	/* update protected by general multicast_lock by caller */
1269 	rcu_assign_pointer(br->ip4_querier.port, port);
1270 
1271 	return true;
1272 }
1273 
1274 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_select_querier(struct net_bridge * br,struct net_bridge_port * port,struct in6_addr * saddr)1275 static bool br_ip6_multicast_select_querier(struct net_bridge *br,
1276 					    struct net_bridge_port *port,
1277 					    struct in6_addr *saddr)
1278 {
1279 	if (!timer_pending(&br->ip6_own_query.timer) &&
1280 	    !timer_pending(&br->ip6_other_query.timer))
1281 		goto update;
1282 
1283 	if (ipv6_addr_cmp(saddr, &br->ip6_querier.addr.u.ip6) <= 0)
1284 		goto update;
1285 
1286 	return false;
1287 
1288 update:
1289 	br->ip6_querier.addr.u.ip6 = *saddr;
1290 
1291 	/* update protected by general multicast_lock by caller */
1292 	rcu_assign_pointer(br->ip6_querier.port, port);
1293 
1294 	return true;
1295 }
1296 #endif
1297 
br_multicast_select_querier(struct net_bridge * br,struct net_bridge_port * port,struct br_ip * saddr)1298 static bool br_multicast_select_querier(struct net_bridge *br,
1299 					struct net_bridge_port *port,
1300 					struct br_ip *saddr)
1301 {
1302 	switch (saddr->proto) {
1303 	case htons(ETH_P_IP):
1304 		return br_ip4_multicast_select_querier(br, port, saddr->u.ip4);
1305 #if IS_ENABLED(CONFIG_IPV6)
1306 	case htons(ETH_P_IPV6):
1307 		return br_ip6_multicast_select_querier(br, port, &saddr->u.ip6);
1308 #endif
1309 	}
1310 
1311 	return false;
1312 }
1313 
1314 static void
br_multicast_update_query_timer(struct net_bridge * br,struct bridge_mcast_other_query * query,unsigned long max_delay)1315 br_multicast_update_query_timer(struct net_bridge *br,
1316 				struct bridge_mcast_other_query *query,
1317 				unsigned long max_delay)
1318 {
1319 	if (!timer_pending(&query->timer))
1320 		query->delay_time = jiffies + max_delay;
1321 
1322 	mod_timer(&query->timer, jiffies + br->multicast_querier_interval);
1323 }
1324 
br_port_mc_router_state_change(struct net_bridge_port * p,bool is_mc_router)1325 static void br_port_mc_router_state_change(struct net_bridge_port *p,
1326 					   bool is_mc_router)
1327 {
1328 	struct switchdev_attr attr = {
1329 		.orig_dev = p->dev,
1330 		.id = SWITCHDEV_ATTR_ID_PORT_MROUTER,
1331 		.flags = SWITCHDEV_F_DEFER,
1332 		.u.mrouter = is_mc_router,
1333 	};
1334 
1335 	switchdev_port_attr_set(p->dev, &attr);
1336 }
1337 
1338 /*
1339  * Add port to router_list
1340  *  list is maintained ordered by pointer value
1341  *  and locked by br->multicast_lock and RCU
1342  */
br_multicast_add_router(struct net_bridge * br,struct net_bridge_port * port)1343 static void br_multicast_add_router(struct net_bridge *br,
1344 				    struct net_bridge_port *port)
1345 {
1346 	struct net_bridge_port *p;
1347 	struct hlist_node *slot = NULL;
1348 
1349 	if (!hlist_unhashed(&port->rlist))
1350 		return;
1351 
1352 	hlist_for_each_entry(p, &br->router_list, rlist) {
1353 		if ((unsigned long) port >= (unsigned long) p)
1354 			break;
1355 		slot = &p->rlist;
1356 	}
1357 
1358 	if (slot)
1359 		hlist_add_behind_rcu(&port->rlist, slot);
1360 	else
1361 		hlist_add_head_rcu(&port->rlist, &br->router_list);
1362 	br_rtr_notify(br->dev, port, RTM_NEWMDB);
1363 	br_port_mc_router_state_change(port, true);
1364 }
1365 
br_multicast_mark_router(struct net_bridge * br,struct net_bridge_port * port)1366 static void br_multicast_mark_router(struct net_bridge *br,
1367 				     struct net_bridge_port *port)
1368 {
1369 	unsigned long now = jiffies;
1370 
1371 	if (!port) {
1372 		if (br->multicast_router == MDB_RTR_TYPE_TEMP_QUERY)
1373 			mod_timer(&br->multicast_router_timer,
1374 				  now + br->multicast_querier_interval);
1375 		return;
1376 	}
1377 
1378 	if (port->multicast_router == MDB_RTR_TYPE_DISABLED ||
1379 	    port->multicast_router == MDB_RTR_TYPE_PERM)
1380 		return;
1381 
1382 	br_multicast_add_router(br, port);
1383 
1384 	mod_timer(&port->multicast_router_timer,
1385 		  now + br->multicast_querier_interval);
1386 }
1387 
br_multicast_query_received(struct net_bridge * br,struct net_bridge_port * port,struct bridge_mcast_other_query * query,struct br_ip * saddr,unsigned long max_delay)1388 static void br_multicast_query_received(struct net_bridge *br,
1389 					struct net_bridge_port *port,
1390 					struct bridge_mcast_other_query *query,
1391 					struct br_ip *saddr,
1392 					unsigned long max_delay)
1393 {
1394 	if (!br_multicast_select_querier(br, port, saddr))
1395 		return;
1396 
1397 	br_multicast_update_query_timer(br, query, max_delay);
1398 	br_multicast_mark_router(br, port);
1399 }
1400 
br_ip4_multicast_query(struct net_bridge * br,struct net_bridge_port * port,struct sk_buff * skb,u16 vid)1401 static int br_ip4_multicast_query(struct net_bridge *br,
1402 				  struct net_bridge_port *port,
1403 				  struct sk_buff *skb,
1404 				  u16 vid)
1405 {
1406 	const struct iphdr *iph = ip_hdr(skb);
1407 	struct igmphdr *ih = igmp_hdr(skb);
1408 	struct net_bridge_mdb_entry *mp;
1409 	struct igmpv3_query *ih3;
1410 	struct net_bridge_port_group *p;
1411 	struct net_bridge_port_group __rcu **pp;
1412 	struct br_ip saddr;
1413 	unsigned long max_delay;
1414 	unsigned long now = jiffies;
1415 	unsigned int offset = skb_transport_offset(skb);
1416 	__be32 group;
1417 	int err = 0;
1418 
1419 	spin_lock(&br->multicast_lock);
1420 	if (!netif_running(br->dev) ||
1421 	    (port && port->state == BR_STATE_DISABLED))
1422 		goto out;
1423 
1424 	group = ih->group;
1425 
1426 	if (skb->len == offset + sizeof(*ih)) {
1427 		max_delay = ih->code * (HZ / IGMP_TIMER_SCALE);
1428 
1429 		if (!max_delay) {
1430 			max_delay = 10 * HZ;
1431 			group = 0;
1432 		}
1433 	} else if (skb->len >= offset + sizeof(*ih3)) {
1434 		ih3 = igmpv3_query_hdr(skb);
1435 		if (ih3->nsrcs)
1436 			goto out;
1437 
1438 		max_delay = ih3->code ?
1439 			    IGMPV3_MRC(ih3->code) * (HZ / IGMP_TIMER_SCALE) : 1;
1440 	} else {
1441 		goto out;
1442 	}
1443 
1444 	if (!group) {
1445 		saddr.proto = htons(ETH_P_IP);
1446 		saddr.u.ip4 = iph->saddr;
1447 
1448 		br_multicast_query_received(br, port, &br->ip4_other_query,
1449 					    &saddr, max_delay);
1450 		goto out;
1451 	}
1452 
1453 	mp = br_mdb_ip4_get(mlock_dereference(br->mdb, br), group, vid);
1454 	if (!mp)
1455 		goto out;
1456 
1457 	max_delay *= br->multicast_last_member_count;
1458 
1459 	if (mp->mglist &&
1460 	    (timer_pending(&mp->timer) ?
1461 	     time_after(mp->timer.expires, now + max_delay) :
1462 	     try_to_del_timer_sync(&mp->timer) >= 0))
1463 		mod_timer(&mp->timer, now + max_delay);
1464 
1465 	for (pp = &mp->ports;
1466 	     (p = mlock_dereference(*pp, br)) != NULL;
1467 	     pp = &p->next) {
1468 		if (timer_pending(&p->timer) ?
1469 		    time_after(p->timer.expires, now + max_delay) :
1470 		    try_to_del_timer_sync(&p->timer) >= 0)
1471 			mod_timer(&p->timer, now + max_delay);
1472 	}
1473 
1474 out:
1475 	spin_unlock(&br->multicast_lock);
1476 	return err;
1477 }
1478 
1479 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_query(struct net_bridge * br,struct net_bridge_port * port,struct sk_buff * skb,u16 vid)1480 static int br_ip6_multicast_query(struct net_bridge *br,
1481 				  struct net_bridge_port *port,
1482 				  struct sk_buff *skb,
1483 				  u16 vid)
1484 {
1485 	struct mld_msg *mld;
1486 	struct net_bridge_mdb_entry *mp;
1487 	struct mld2_query *mld2q;
1488 	struct net_bridge_port_group *p;
1489 	struct net_bridge_port_group __rcu **pp;
1490 	struct br_ip saddr;
1491 	unsigned long max_delay;
1492 	unsigned long now = jiffies;
1493 	unsigned int offset = skb_transport_offset(skb);
1494 	const struct in6_addr *group = NULL;
1495 	bool is_general_query;
1496 	int err = 0;
1497 
1498 	spin_lock(&br->multicast_lock);
1499 	if (!netif_running(br->dev) ||
1500 	    (port && port->state == BR_STATE_DISABLED))
1501 		goto out;
1502 
1503 	if (skb->len == offset + sizeof(*mld)) {
1504 		if (!pskb_may_pull(skb, offset + sizeof(*mld))) {
1505 			err = -EINVAL;
1506 			goto out;
1507 		}
1508 		mld = (struct mld_msg *) icmp6_hdr(skb);
1509 		max_delay = msecs_to_jiffies(ntohs(mld->mld_maxdelay));
1510 		if (max_delay)
1511 			group = &mld->mld_mca;
1512 	} else {
1513 		if (!pskb_may_pull(skb, offset + sizeof(*mld2q))) {
1514 			err = -EINVAL;
1515 			goto out;
1516 		}
1517 		mld2q = (struct mld2_query *)icmp6_hdr(skb);
1518 		if (!mld2q->mld2q_nsrcs)
1519 			group = &mld2q->mld2q_mca;
1520 
1521 		max_delay = max(msecs_to_jiffies(mldv2_mrc(mld2q)), 1UL);
1522 	}
1523 
1524 	is_general_query = group && ipv6_addr_any(group);
1525 
1526 	if (is_general_query) {
1527 		saddr.proto = htons(ETH_P_IPV6);
1528 		saddr.u.ip6 = ipv6_hdr(skb)->saddr;
1529 
1530 		br_multicast_query_received(br, port, &br->ip6_other_query,
1531 					    &saddr, max_delay);
1532 		goto out;
1533 	} else if (!group) {
1534 		goto out;
1535 	}
1536 
1537 	mp = br_mdb_ip6_get(mlock_dereference(br->mdb, br), group, vid);
1538 	if (!mp)
1539 		goto out;
1540 
1541 	max_delay *= br->multicast_last_member_count;
1542 	if (mp->mglist &&
1543 	    (timer_pending(&mp->timer) ?
1544 	     time_after(mp->timer.expires, now + max_delay) :
1545 	     try_to_del_timer_sync(&mp->timer) >= 0))
1546 		mod_timer(&mp->timer, now + max_delay);
1547 
1548 	for (pp = &mp->ports;
1549 	     (p = mlock_dereference(*pp, br)) != NULL;
1550 	     pp = &p->next) {
1551 		if (timer_pending(&p->timer) ?
1552 		    time_after(p->timer.expires, now + max_delay) :
1553 		    try_to_del_timer_sync(&p->timer) >= 0)
1554 			mod_timer(&p->timer, now + max_delay);
1555 	}
1556 
1557 out:
1558 	spin_unlock(&br->multicast_lock);
1559 	return err;
1560 }
1561 #endif
1562 
1563 static void
br_multicast_leave_group(struct net_bridge * br,struct net_bridge_port * port,struct br_ip * group,struct bridge_mcast_other_query * other_query,struct bridge_mcast_own_query * own_query,const unsigned char * src)1564 br_multicast_leave_group(struct net_bridge *br,
1565 			 struct net_bridge_port *port,
1566 			 struct br_ip *group,
1567 			 struct bridge_mcast_other_query *other_query,
1568 			 struct bridge_mcast_own_query *own_query,
1569 			 const unsigned char *src)
1570 {
1571 	struct net_bridge_mdb_htable *mdb;
1572 	struct net_bridge_mdb_entry *mp;
1573 	struct net_bridge_port_group *p;
1574 	unsigned long now;
1575 	unsigned long time;
1576 
1577 	spin_lock(&br->multicast_lock);
1578 	if (!netif_running(br->dev) ||
1579 	    (port && port->state == BR_STATE_DISABLED))
1580 		goto out;
1581 
1582 	mdb = mlock_dereference(br->mdb, br);
1583 	mp = br_mdb_ip_get(mdb, group);
1584 	if (!mp)
1585 		goto out;
1586 
1587 	if (port && (port->flags & BR_MULTICAST_FAST_LEAVE)) {
1588 		struct net_bridge_port_group __rcu **pp;
1589 
1590 		for (pp = &mp->ports;
1591 		     (p = mlock_dereference(*pp, br)) != NULL;
1592 		     pp = &p->next) {
1593 			if (!br_port_group_equal(p, port, src))
1594 				continue;
1595 
1596 			if (p->flags & MDB_PG_FLAGS_PERMANENT)
1597 				break;
1598 
1599 			rcu_assign_pointer(*pp, p->next);
1600 			hlist_del_init(&p->mglist);
1601 			del_timer(&p->timer);
1602 			call_rcu_bh(&p->rcu, br_multicast_free_pg);
1603 			br_mdb_notify(br->dev, port, group, RTM_DELMDB,
1604 				      p->flags);
1605 
1606 			if (!mp->ports && !mp->mglist &&
1607 			    netif_running(br->dev))
1608 				mod_timer(&mp->timer, jiffies);
1609 		}
1610 		goto out;
1611 	}
1612 
1613 	if (timer_pending(&other_query->timer))
1614 		goto out;
1615 
1616 	if (br->multicast_querier) {
1617 		__br_multicast_send_query(br, port, &mp->addr);
1618 
1619 		time = jiffies + br->multicast_last_member_count *
1620 				 br->multicast_last_member_interval;
1621 
1622 		mod_timer(&own_query->timer, time);
1623 
1624 		for (p = mlock_dereference(mp->ports, br);
1625 		     p != NULL;
1626 		     p = mlock_dereference(p->next, br)) {
1627 			if (!br_port_group_equal(p, port, src))
1628 				continue;
1629 
1630 			if (!hlist_unhashed(&p->mglist) &&
1631 			    (timer_pending(&p->timer) ?
1632 			     time_after(p->timer.expires, time) :
1633 			     try_to_del_timer_sync(&p->timer) >= 0)) {
1634 				mod_timer(&p->timer, time);
1635 			}
1636 
1637 			break;
1638 		}
1639 	}
1640 
1641 	now = jiffies;
1642 	time = now + br->multicast_last_member_count *
1643 		     br->multicast_last_member_interval;
1644 
1645 	if (!port) {
1646 		if (mp->mglist &&
1647 		    (timer_pending(&mp->timer) ?
1648 		     time_after(mp->timer.expires, time) :
1649 		     try_to_del_timer_sync(&mp->timer) >= 0)) {
1650 			mod_timer(&mp->timer, time);
1651 		}
1652 
1653 		goto out;
1654 	}
1655 
1656 	for (p = mlock_dereference(mp->ports, br);
1657 	     p != NULL;
1658 	     p = mlock_dereference(p->next, br)) {
1659 		if (p->port != port)
1660 			continue;
1661 
1662 		if (!hlist_unhashed(&p->mglist) &&
1663 		    (timer_pending(&p->timer) ?
1664 		     time_after(p->timer.expires, time) :
1665 		     try_to_del_timer_sync(&p->timer) >= 0)) {
1666 			mod_timer(&p->timer, time);
1667 		}
1668 
1669 		break;
1670 	}
1671 out:
1672 	spin_unlock(&br->multicast_lock);
1673 }
1674 
br_ip4_multicast_leave_group(struct net_bridge * br,struct net_bridge_port * port,__be32 group,__u16 vid,const unsigned char * src)1675 static void br_ip4_multicast_leave_group(struct net_bridge *br,
1676 					 struct net_bridge_port *port,
1677 					 __be32 group,
1678 					 __u16 vid,
1679 					 const unsigned char *src)
1680 {
1681 	struct br_ip br_group;
1682 	struct bridge_mcast_own_query *own_query;
1683 
1684 	if (ipv4_is_local_multicast(group))
1685 		return;
1686 
1687 	own_query = port ? &port->ip4_own_query : &br->ip4_own_query;
1688 
1689 	br_group.u.ip4 = group;
1690 	br_group.proto = htons(ETH_P_IP);
1691 	br_group.vid = vid;
1692 
1693 	br_multicast_leave_group(br, port, &br_group, &br->ip4_other_query,
1694 				 own_query, src);
1695 }
1696 
1697 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_leave_group(struct net_bridge * br,struct net_bridge_port * port,const struct in6_addr * group,__u16 vid,const unsigned char * src)1698 static void br_ip6_multicast_leave_group(struct net_bridge *br,
1699 					 struct net_bridge_port *port,
1700 					 const struct in6_addr *group,
1701 					 __u16 vid,
1702 					 const unsigned char *src)
1703 {
1704 	struct br_ip br_group;
1705 	struct bridge_mcast_own_query *own_query;
1706 
1707 	if (ipv6_addr_is_ll_all_nodes(group))
1708 		return;
1709 
1710 	own_query = port ? &port->ip6_own_query : &br->ip6_own_query;
1711 
1712 	br_group.u.ip6 = *group;
1713 	br_group.proto = htons(ETH_P_IPV6);
1714 	br_group.vid = vid;
1715 
1716 	br_multicast_leave_group(br, port, &br_group, &br->ip6_other_query,
1717 				 own_query, src);
1718 }
1719 #endif
1720 
br_multicast_err_count(const struct net_bridge * br,const struct net_bridge_port * p,__be16 proto)1721 static void br_multicast_err_count(const struct net_bridge *br,
1722 				   const struct net_bridge_port *p,
1723 				   __be16 proto)
1724 {
1725 	struct bridge_mcast_stats __percpu *stats;
1726 	struct bridge_mcast_stats *pstats;
1727 
1728 	if (!br->multicast_stats_enabled)
1729 		return;
1730 
1731 	if (p)
1732 		stats = p->mcast_stats;
1733 	else
1734 		stats = br->mcast_stats;
1735 	if (WARN_ON(!stats))
1736 		return;
1737 
1738 	pstats = this_cpu_ptr(stats);
1739 
1740 	u64_stats_update_begin(&pstats->syncp);
1741 	switch (proto) {
1742 	case htons(ETH_P_IP):
1743 		pstats->mstats.igmp_parse_errors++;
1744 		break;
1745 #if IS_ENABLED(CONFIG_IPV6)
1746 	case htons(ETH_P_IPV6):
1747 		pstats->mstats.mld_parse_errors++;
1748 		break;
1749 #endif
1750 	}
1751 	u64_stats_update_end(&pstats->syncp);
1752 }
1753 
br_multicast_pim(struct net_bridge * br,struct net_bridge_port * port,const struct sk_buff * skb)1754 static void br_multicast_pim(struct net_bridge *br,
1755 			     struct net_bridge_port *port,
1756 			     const struct sk_buff *skb)
1757 {
1758 	unsigned int offset = skb_transport_offset(skb);
1759 	struct pimhdr *pimhdr, _pimhdr;
1760 
1761 	pimhdr = skb_header_pointer(skb, offset, sizeof(_pimhdr), &_pimhdr);
1762 	if (!pimhdr || pim_hdr_version(pimhdr) != PIM_VERSION ||
1763 	    pim_hdr_type(pimhdr) != PIM_TYPE_HELLO)
1764 		return;
1765 
1766 	br_multicast_mark_router(br, port);
1767 }
1768 
br_multicast_ipv4_rcv(struct net_bridge * br,struct net_bridge_port * port,struct sk_buff * skb,u16 vid)1769 static int br_multicast_ipv4_rcv(struct net_bridge *br,
1770 				 struct net_bridge_port *port,
1771 				 struct sk_buff *skb,
1772 				 u16 vid)
1773 {
1774 	struct sk_buff *skb_trimmed = NULL;
1775 	const unsigned char *src;
1776 	struct igmphdr *ih;
1777 	int err;
1778 
1779 	err = ip_mc_check_igmp(skb, &skb_trimmed);
1780 
1781 	if (err == -ENOMSG) {
1782 		if (!ipv4_is_local_multicast(ip_hdr(skb)->daddr)) {
1783 			BR_INPUT_SKB_CB(skb)->mrouters_only = 1;
1784 		} else if (pim_ipv4_all_pim_routers(ip_hdr(skb)->daddr)) {
1785 			if (ip_hdr(skb)->protocol == IPPROTO_PIM)
1786 				br_multicast_pim(br, port, skb);
1787 		}
1788 		return 0;
1789 	} else if (err < 0) {
1790 		br_multicast_err_count(br, port, skb->protocol);
1791 		return err;
1792 	}
1793 
1794 	ih = igmp_hdr(skb);
1795 	src = eth_hdr(skb)->h_source;
1796 	BR_INPUT_SKB_CB(skb)->igmp = ih->type;
1797 
1798 	switch (ih->type) {
1799 	case IGMP_HOST_MEMBERSHIP_REPORT:
1800 	case IGMPV2_HOST_MEMBERSHIP_REPORT:
1801 		BR_INPUT_SKB_CB(skb)->mrouters_only = 1;
1802 		err = br_ip4_multicast_add_group(br, port, ih->group, vid, src);
1803 		break;
1804 	case IGMPV3_HOST_MEMBERSHIP_REPORT:
1805 		err = br_ip4_multicast_igmp3_report(br, port, skb_trimmed, vid);
1806 		break;
1807 	case IGMP_HOST_MEMBERSHIP_QUERY:
1808 		err = br_ip4_multicast_query(br, port, skb_trimmed, vid);
1809 		break;
1810 	case IGMP_HOST_LEAVE_MESSAGE:
1811 		br_ip4_multicast_leave_group(br, port, ih->group, vid, src);
1812 		break;
1813 	}
1814 
1815 	if (skb_trimmed && skb_trimmed != skb)
1816 		kfree_skb(skb_trimmed);
1817 
1818 	br_multicast_count(br, port, skb, BR_INPUT_SKB_CB(skb)->igmp,
1819 			   BR_MCAST_DIR_RX);
1820 
1821 	return err;
1822 }
1823 
1824 #if IS_ENABLED(CONFIG_IPV6)
br_multicast_ipv6_rcv(struct net_bridge * br,struct net_bridge_port * port,struct sk_buff * skb,u16 vid)1825 static int br_multicast_ipv6_rcv(struct net_bridge *br,
1826 				 struct net_bridge_port *port,
1827 				 struct sk_buff *skb,
1828 				 u16 vid)
1829 {
1830 	struct sk_buff *skb_trimmed = NULL;
1831 	const unsigned char *src;
1832 	struct mld_msg *mld;
1833 	int err;
1834 
1835 	err = ipv6_mc_check_mld(skb, &skb_trimmed);
1836 
1837 	if (err == -ENOMSG) {
1838 		if (!ipv6_addr_is_ll_all_nodes(&ipv6_hdr(skb)->daddr))
1839 			BR_INPUT_SKB_CB(skb)->mrouters_only = 1;
1840 		return 0;
1841 	} else if (err < 0) {
1842 		br_multicast_err_count(br, port, skb->protocol);
1843 		return err;
1844 	}
1845 
1846 	mld = (struct mld_msg *)skb_transport_header(skb);
1847 	BR_INPUT_SKB_CB(skb)->igmp = mld->mld_type;
1848 
1849 	switch (mld->mld_type) {
1850 	case ICMPV6_MGM_REPORT:
1851 		src = eth_hdr(skb)->h_source;
1852 		BR_INPUT_SKB_CB(skb)->mrouters_only = 1;
1853 		err = br_ip6_multicast_add_group(br, port, &mld->mld_mca, vid,
1854 						 src);
1855 		break;
1856 	case ICMPV6_MLD2_REPORT:
1857 		err = br_ip6_multicast_mld2_report(br, port, skb_trimmed, vid);
1858 		break;
1859 	case ICMPV6_MGM_QUERY:
1860 		err = br_ip6_multicast_query(br, port, skb_trimmed, vid);
1861 		break;
1862 	case ICMPV6_MGM_REDUCTION:
1863 		src = eth_hdr(skb)->h_source;
1864 		br_ip6_multicast_leave_group(br, port, &mld->mld_mca, vid, src);
1865 		break;
1866 	}
1867 
1868 	if (skb_trimmed && skb_trimmed != skb)
1869 		kfree_skb(skb_trimmed);
1870 
1871 	br_multicast_count(br, port, skb, BR_INPUT_SKB_CB(skb)->igmp,
1872 			   BR_MCAST_DIR_RX);
1873 
1874 	return err;
1875 }
1876 #endif
1877 
br_multicast_rcv(struct net_bridge * br,struct net_bridge_port * port,struct sk_buff * skb,u16 vid)1878 int br_multicast_rcv(struct net_bridge *br, struct net_bridge_port *port,
1879 		     struct sk_buff *skb, u16 vid)
1880 {
1881 	int ret = 0;
1882 
1883 	BR_INPUT_SKB_CB(skb)->igmp = 0;
1884 	BR_INPUT_SKB_CB(skb)->mrouters_only = 0;
1885 
1886 	if (br->multicast_disabled)
1887 		return 0;
1888 
1889 	switch (skb->protocol) {
1890 	case htons(ETH_P_IP):
1891 		ret = br_multicast_ipv4_rcv(br, port, skb, vid);
1892 		break;
1893 #if IS_ENABLED(CONFIG_IPV6)
1894 	case htons(ETH_P_IPV6):
1895 		ret = br_multicast_ipv6_rcv(br, port, skb, vid);
1896 		break;
1897 #endif
1898 	}
1899 
1900 	return ret;
1901 }
1902 
br_multicast_query_expired(struct net_bridge * br,struct bridge_mcast_own_query * query,struct bridge_mcast_querier * querier)1903 static void br_multicast_query_expired(struct net_bridge *br,
1904 				       struct bridge_mcast_own_query *query,
1905 				       struct bridge_mcast_querier *querier)
1906 {
1907 	spin_lock(&br->multicast_lock);
1908 	if (query->startup_sent < br->multicast_startup_query_count)
1909 		query->startup_sent++;
1910 
1911 	RCU_INIT_POINTER(querier->port, NULL);
1912 	br_multicast_send_query(br, NULL, query);
1913 	spin_unlock(&br->multicast_lock);
1914 }
1915 
br_ip4_multicast_query_expired(unsigned long data)1916 static void br_ip4_multicast_query_expired(unsigned long data)
1917 {
1918 	struct net_bridge *br = (void *)data;
1919 
1920 	br_multicast_query_expired(br, &br->ip4_own_query, &br->ip4_querier);
1921 }
1922 
1923 #if IS_ENABLED(CONFIG_IPV6)
br_ip6_multicast_query_expired(unsigned long data)1924 static void br_ip6_multicast_query_expired(unsigned long data)
1925 {
1926 	struct net_bridge *br = (void *)data;
1927 
1928 	br_multicast_query_expired(br, &br->ip6_own_query, &br->ip6_querier);
1929 }
1930 #endif
1931 
br_multicast_init(struct net_bridge * br)1932 void br_multicast_init(struct net_bridge *br)
1933 {
1934 	br->hash_elasticity = 4;
1935 	br->hash_max = 512;
1936 
1937 	br->multicast_router = MDB_RTR_TYPE_TEMP_QUERY;
1938 	br->multicast_querier = 0;
1939 	br->multicast_query_use_ifaddr = 0;
1940 	br->multicast_last_member_count = 2;
1941 	br->multicast_startup_query_count = 2;
1942 
1943 	br->multicast_last_member_interval = HZ;
1944 	br->multicast_query_response_interval = 10 * HZ;
1945 	br->multicast_startup_query_interval = 125 * HZ / 4;
1946 	br->multicast_query_interval = 125 * HZ;
1947 	br->multicast_querier_interval = 255 * HZ;
1948 	br->multicast_membership_interval = 260 * HZ;
1949 
1950 	br->ip4_other_query.delay_time = 0;
1951 	br->ip4_querier.port = NULL;
1952 	br->multicast_igmp_version = 2;
1953 #if IS_ENABLED(CONFIG_IPV6)
1954 	br->multicast_mld_version = 1;
1955 	br->ip6_other_query.delay_time = 0;
1956 	br->ip6_querier.port = NULL;
1957 #endif
1958 	br->has_ipv6_addr = 1;
1959 
1960 	spin_lock_init(&br->multicast_lock);
1961 	setup_timer(&br->multicast_router_timer,
1962 		    br_multicast_local_router_expired, 0);
1963 	setup_timer(&br->ip4_other_query.timer,
1964 		    br_ip4_multicast_querier_expired, (unsigned long)br);
1965 	setup_timer(&br->ip4_own_query.timer, br_ip4_multicast_query_expired,
1966 		    (unsigned long)br);
1967 #if IS_ENABLED(CONFIG_IPV6)
1968 	setup_timer(&br->ip6_other_query.timer,
1969 		    br_ip6_multicast_querier_expired, (unsigned long)br);
1970 	setup_timer(&br->ip6_own_query.timer, br_ip6_multicast_query_expired,
1971 		    (unsigned long)br);
1972 #endif
1973 }
1974 
__br_multicast_open(struct net_bridge * br,struct bridge_mcast_own_query * query)1975 static void __br_multicast_open(struct net_bridge *br,
1976 				struct bridge_mcast_own_query *query)
1977 {
1978 	query->startup_sent = 0;
1979 
1980 	if (br->multicast_disabled)
1981 		return;
1982 
1983 	mod_timer(&query->timer, jiffies);
1984 }
1985 
br_multicast_open(struct net_bridge * br)1986 void br_multicast_open(struct net_bridge *br)
1987 {
1988 	__br_multicast_open(br, &br->ip4_own_query);
1989 #if IS_ENABLED(CONFIG_IPV6)
1990 	__br_multicast_open(br, &br->ip6_own_query);
1991 #endif
1992 }
1993 
br_multicast_stop(struct net_bridge * br)1994 void br_multicast_stop(struct net_bridge *br)
1995 {
1996 	del_timer_sync(&br->multicast_router_timer);
1997 	del_timer_sync(&br->ip4_other_query.timer);
1998 	del_timer_sync(&br->ip4_own_query.timer);
1999 #if IS_ENABLED(CONFIG_IPV6)
2000 	del_timer_sync(&br->ip6_other_query.timer);
2001 	del_timer_sync(&br->ip6_own_query.timer);
2002 #endif
2003 }
2004 
br_multicast_dev_del(struct net_bridge * br)2005 void br_multicast_dev_del(struct net_bridge *br)
2006 {
2007 	struct net_bridge_mdb_htable *mdb;
2008 	struct net_bridge_mdb_entry *mp;
2009 	struct hlist_node *n;
2010 	u32 ver;
2011 	int i;
2012 
2013 	spin_lock_bh(&br->multicast_lock);
2014 	mdb = mlock_dereference(br->mdb, br);
2015 	if (!mdb)
2016 		goto out;
2017 
2018 	br->mdb = NULL;
2019 
2020 	ver = mdb->ver;
2021 	for (i = 0; i < mdb->max; i++) {
2022 		hlist_for_each_entry_safe(mp, n, &mdb->mhash[i],
2023 					  hlist[ver]) {
2024 			del_timer(&mp->timer);
2025 			call_rcu_bh(&mp->rcu, br_multicast_free_group);
2026 		}
2027 	}
2028 
2029 	if (mdb->old) {
2030 		spin_unlock_bh(&br->multicast_lock);
2031 		rcu_barrier_bh();
2032 		spin_lock_bh(&br->multicast_lock);
2033 		WARN_ON(mdb->old);
2034 	}
2035 
2036 	mdb->old = mdb;
2037 	call_rcu_bh(&mdb->rcu, br_mdb_free);
2038 
2039 out:
2040 	spin_unlock_bh(&br->multicast_lock);
2041 }
2042 
br_multicast_set_router(struct net_bridge * br,unsigned long val)2043 int br_multicast_set_router(struct net_bridge *br, unsigned long val)
2044 {
2045 	int err = -EINVAL;
2046 
2047 	spin_lock_bh(&br->multicast_lock);
2048 
2049 	switch (val) {
2050 	case MDB_RTR_TYPE_DISABLED:
2051 	case MDB_RTR_TYPE_PERM:
2052 		del_timer(&br->multicast_router_timer);
2053 		/* fall through */
2054 	case MDB_RTR_TYPE_TEMP_QUERY:
2055 		br->multicast_router = val;
2056 		err = 0;
2057 		break;
2058 	}
2059 
2060 	spin_unlock_bh(&br->multicast_lock);
2061 
2062 	return err;
2063 }
2064 
__del_port_router(struct net_bridge_port * p)2065 static void __del_port_router(struct net_bridge_port *p)
2066 {
2067 	if (hlist_unhashed(&p->rlist))
2068 		return;
2069 	hlist_del_init_rcu(&p->rlist);
2070 	br_rtr_notify(p->br->dev, p, RTM_DELMDB);
2071 	br_port_mc_router_state_change(p, false);
2072 
2073 	/* don't allow timer refresh */
2074 	if (p->multicast_router == MDB_RTR_TYPE_TEMP)
2075 		p->multicast_router = MDB_RTR_TYPE_TEMP_QUERY;
2076 }
2077 
br_multicast_set_port_router(struct net_bridge_port * p,unsigned long val)2078 int br_multicast_set_port_router(struct net_bridge_port *p, unsigned long val)
2079 {
2080 	struct net_bridge *br = p->br;
2081 	unsigned long now = jiffies;
2082 	int err = -EINVAL;
2083 
2084 	spin_lock(&br->multicast_lock);
2085 	if (p->multicast_router == val) {
2086 		/* Refresh the temp router port timer */
2087 		if (p->multicast_router == MDB_RTR_TYPE_TEMP)
2088 			mod_timer(&p->multicast_router_timer,
2089 				  now + br->multicast_querier_interval);
2090 		err = 0;
2091 		goto unlock;
2092 	}
2093 	switch (val) {
2094 	case MDB_RTR_TYPE_DISABLED:
2095 		p->multicast_router = MDB_RTR_TYPE_DISABLED;
2096 		__del_port_router(p);
2097 		del_timer(&p->multicast_router_timer);
2098 		break;
2099 	case MDB_RTR_TYPE_TEMP_QUERY:
2100 		p->multicast_router = MDB_RTR_TYPE_TEMP_QUERY;
2101 		__del_port_router(p);
2102 		break;
2103 	case MDB_RTR_TYPE_PERM:
2104 		p->multicast_router = MDB_RTR_TYPE_PERM;
2105 		del_timer(&p->multicast_router_timer);
2106 		br_multicast_add_router(br, p);
2107 		break;
2108 	case MDB_RTR_TYPE_TEMP:
2109 		p->multicast_router = MDB_RTR_TYPE_TEMP;
2110 		br_multicast_mark_router(br, p);
2111 		break;
2112 	default:
2113 		goto unlock;
2114 	}
2115 	err = 0;
2116 unlock:
2117 	spin_unlock(&br->multicast_lock);
2118 
2119 	return err;
2120 }
2121 
br_multicast_start_querier(struct net_bridge * br,struct bridge_mcast_own_query * query)2122 static void br_multicast_start_querier(struct net_bridge *br,
2123 				       struct bridge_mcast_own_query *query)
2124 {
2125 	struct net_bridge_port *port;
2126 
2127 	__br_multicast_open(br, query);
2128 
2129 	rcu_read_lock();
2130 	list_for_each_entry_rcu(port, &br->port_list, list) {
2131 		if (port->state == BR_STATE_DISABLED ||
2132 		    port->state == BR_STATE_BLOCKING)
2133 			continue;
2134 
2135 		if (query == &br->ip4_own_query)
2136 			br_multicast_enable(&port->ip4_own_query);
2137 #if IS_ENABLED(CONFIG_IPV6)
2138 		else
2139 			br_multicast_enable(&port->ip6_own_query);
2140 #endif
2141 	}
2142 	rcu_read_unlock();
2143 }
2144 
br_multicast_toggle(struct net_bridge * br,unsigned long val)2145 int br_multicast_toggle(struct net_bridge *br, unsigned long val)
2146 {
2147 	struct net_bridge_mdb_htable *mdb;
2148 	struct net_bridge_port *port;
2149 	int err = 0;
2150 
2151 	spin_lock_bh(&br->multicast_lock);
2152 	if (br->multicast_disabled == !val)
2153 		goto unlock;
2154 
2155 	br_mc_disabled_update(br->dev, !val);
2156 	br->multicast_disabled = !val;
2157 	if (br->multicast_disabled)
2158 		goto unlock;
2159 
2160 	if (!netif_running(br->dev))
2161 		goto unlock;
2162 
2163 	mdb = mlock_dereference(br->mdb, br);
2164 	if (mdb) {
2165 		if (mdb->old) {
2166 			err = -EEXIST;
2167 rollback:
2168 			br->multicast_disabled = !!val;
2169 			goto unlock;
2170 		}
2171 
2172 		err = br_mdb_rehash(&br->mdb, mdb->max,
2173 				    br->hash_elasticity);
2174 		if (err)
2175 			goto rollback;
2176 	}
2177 
2178 	br_multicast_open(br);
2179 	list_for_each_entry(port, &br->port_list, list)
2180 		__br_multicast_enable_port(port);
2181 
2182 unlock:
2183 	spin_unlock_bh(&br->multicast_lock);
2184 
2185 	return err;
2186 }
2187 
br_multicast_enabled(const struct net_device * dev)2188 bool br_multicast_enabled(const struct net_device *dev)
2189 {
2190 	struct net_bridge *br = netdev_priv(dev);
2191 
2192 	return !br->multicast_disabled;
2193 }
2194 EXPORT_SYMBOL_GPL(br_multicast_enabled);
2195 
br_multicast_set_querier(struct net_bridge * br,unsigned long val)2196 int br_multicast_set_querier(struct net_bridge *br, unsigned long val)
2197 {
2198 	unsigned long max_delay;
2199 
2200 	val = !!val;
2201 
2202 	spin_lock_bh(&br->multicast_lock);
2203 	if (br->multicast_querier == val)
2204 		goto unlock;
2205 
2206 	br->multicast_querier = val;
2207 	if (!val)
2208 		goto unlock;
2209 
2210 	max_delay = br->multicast_query_response_interval;
2211 
2212 	if (!timer_pending(&br->ip4_other_query.timer))
2213 		br->ip4_other_query.delay_time = jiffies + max_delay;
2214 
2215 	br_multicast_start_querier(br, &br->ip4_own_query);
2216 
2217 #if IS_ENABLED(CONFIG_IPV6)
2218 	if (!timer_pending(&br->ip6_other_query.timer))
2219 		br->ip6_other_query.delay_time = jiffies + max_delay;
2220 
2221 	br_multicast_start_querier(br, &br->ip6_own_query);
2222 #endif
2223 
2224 unlock:
2225 	spin_unlock_bh(&br->multicast_lock);
2226 
2227 	return 0;
2228 }
2229 
br_multicast_set_hash_max(struct net_bridge * br,unsigned long val)2230 int br_multicast_set_hash_max(struct net_bridge *br, unsigned long val)
2231 {
2232 	int err = -EINVAL;
2233 	u32 old;
2234 	struct net_bridge_mdb_htable *mdb;
2235 
2236 	spin_lock_bh(&br->multicast_lock);
2237 	if (!is_power_of_2(val))
2238 		goto unlock;
2239 
2240 	mdb = mlock_dereference(br->mdb, br);
2241 	if (mdb && val < mdb->size)
2242 		goto unlock;
2243 
2244 	err = 0;
2245 
2246 	old = br->hash_max;
2247 	br->hash_max = val;
2248 
2249 	if (mdb) {
2250 		if (mdb->old) {
2251 			err = -EEXIST;
2252 rollback:
2253 			br->hash_max = old;
2254 			goto unlock;
2255 		}
2256 
2257 		err = br_mdb_rehash(&br->mdb, br->hash_max,
2258 				    br->hash_elasticity);
2259 		if (err)
2260 			goto rollback;
2261 	}
2262 
2263 unlock:
2264 	spin_unlock_bh(&br->multicast_lock);
2265 
2266 	return err;
2267 }
2268 
br_multicast_set_igmp_version(struct net_bridge * br,unsigned long val)2269 int br_multicast_set_igmp_version(struct net_bridge *br, unsigned long val)
2270 {
2271 	/* Currently we support only version 2 and 3 */
2272 	switch (val) {
2273 	case 2:
2274 	case 3:
2275 		break;
2276 	default:
2277 		return -EINVAL;
2278 	}
2279 
2280 	spin_lock_bh(&br->multicast_lock);
2281 	br->multicast_igmp_version = val;
2282 	spin_unlock_bh(&br->multicast_lock);
2283 
2284 	return 0;
2285 }
2286 
2287 #if IS_ENABLED(CONFIG_IPV6)
br_multicast_set_mld_version(struct net_bridge * br,unsigned long val)2288 int br_multicast_set_mld_version(struct net_bridge *br, unsigned long val)
2289 {
2290 	/* Currently we support version 1 and 2 */
2291 	switch (val) {
2292 	case 1:
2293 	case 2:
2294 		break;
2295 	default:
2296 		return -EINVAL;
2297 	}
2298 
2299 	spin_lock_bh(&br->multicast_lock);
2300 	br->multicast_mld_version = val;
2301 	spin_unlock_bh(&br->multicast_lock);
2302 
2303 	return 0;
2304 }
2305 #endif
2306 
2307 /**
2308  * br_multicast_list_adjacent - Returns snooped multicast addresses
2309  * @dev:	The bridge port adjacent to which to retrieve addresses
2310  * @br_ip_list:	The list to store found, snooped multicast IP addresses in
2311  *
2312  * Creates a list of IP addresses (struct br_ip_list) sensed by the multicast
2313  * snooping feature on all bridge ports of dev's bridge device, excluding
2314  * the addresses from dev itself.
2315  *
2316  * Returns the number of items added to br_ip_list.
2317  *
2318  * Notes:
2319  * - br_ip_list needs to be initialized by caller
2320  * - br_ip_list might contain duplicates in the end
2321  *   (needs to be taken care of by caller)
2322  * - br_ip_list needs to be freed by caller
2323  */
br_multicast_list_adjacent(struct net_device * dev,struct list_head * br_ip_list)2324 int br_multicast_list_adjacent(struct net_device *dev,
2325 			       struct list_head *br_ip_list)
2326 {
2327 	struct net_bridge *br;
2328 	struct net_bridge_port *port;
2329 	struct net_bridge_port_group *group;
2330 	struct br_ip_list *entry;
2331 	int count = 0;
2332 
2333 	rcu_read_lock();
2334 	if (!br_ip_list || !br_port_exists(dev))
2335 		goto unlock;
2336 
2337 	port = br_port_get_rcu(dev);
2338 	if (!port || !port->br)
2339 		goto unlock;
2340 
2341 	br = port->br;
2342 
2343 	list_for_each_entry_rcu(port, &br->port_list, list) {
2344 		if (!port->dev || port->dev == dev)
2345 			continue;
2346 
2347 		hlist_for_each_entry_rcu(group, &port->mglist, mglist) {
2348 			entry = kmalloc(sizeof(*entry), GFP_ATOMIC);
2349 			if (!entry)
2350 				goto unlock;
2351 
2352 			entry->addr = group->addr;
2353 			list_add(&entry->list, br_ip_list);
2354 			count++;
2355 		}
2356 	}
2357 
2358 unlock:
2359 	rcu_read_unlock();
2360 	return count;
2361 }
2362 EXPORT_SYMBOL_GPL(br_multicast_list_adjacent);
2363 
2364 /**
2365  * br_multicast_has_querier_anywhere - Checks for a querier on a bridge
2366  * @dev: The bridge port providing the bridge on which to check for a querier
2367  * @proto: The protocol family to check for: IGMP -> ETH_P_IP, MLD -> ETH_P_IPV6
2368  *
2369  * Checks whether the given interface has a bridge on top and if so returns
2370  * true if a valid querier exists anywhere on the bridged link layer.
2371  * Otherwise returns false.
2372  */
br_multicast_has_querier_anywhere(struct net_device * dev,int proto)2373 bool br_multicast_has_querier_anywhere(struct net_device *dev, int proto)
2374 {
2375 	struct net_bridge *br;
2376 	struct net_bridge_port *port;
2377 	struct ethhdr eth;
2378 	bool ret = false;
2379 
2380 	rcu_read_lock();
2381 	if (!br_port_exists(dev))
2382 		goto unlock;
2383 
2384 	port = br_port_get_rcu(dev);
2385 	if (!port || !port->br)
2386 		goto unlock;
2387 
2388 	br = port->br;
2389 
2390 	memset(&eth, 0, sizeof(eth));
2391 	eth.h_proto = htons(proto);
2392 
2393 	ret = br_multicast_querier_exists(br, &eth);
2394 
2395 unlock:
2396 	rcu_read_unlock();
2397 	return ret;
2398 }
2399 EXPORT_SYMBOL_GPL(br_multicast_has_querier_anywhere);
2400 
2401 /**
2402  * br_multicast_has_querier_adjacent - Checks for a querier behind a bridge port
2403  * @dev: The bridge port adjacent to which to check for a querier
2404  * @proto: The protocol family to check for: IGMP -> ETH_P_IP, MLD -> ETH_P_IPV6
2405  *
2406  * Checks whether the given interface has a bridge on top and if so returns
2407  * true if a selected querier is behind one of the other ports of this
2408  * bridge. Otherwise returns false.
2409  */
br_multicast_has_querier_adjacent(struct net_device * dev,int proto)2410 bool br_multicast_has_querier_adjacent(struct net_device *dev, int proto)
2411 {
2412 	struct net_bridge *br;
2413 	struct net_bridge_port *port;
2414 	bool ret = false;
2415 
2416 	rcu_read_lock();
2417 	if (!br_port_exists(dev))
2418 		goto unlock;
2419 
2420 	port = br_port_get_rcu(dev);
2421 	if (!port || !port->br)
2422 		goto unlock;
2423 
2424 	br = port->br;
2425 
2426 	switch (proto) {
2427 	case ETH_P_IP:
2428 		if (!timer_pending(&br->ip4_other_query.timer) ||
2429 		    rcu_dereference(br->ip4_querier.port) == port)
2430 			goto unlock;
2431 		break;
2432 #if IS_ENABLED(CONFIG_IPV6)
2433 	case ETH_P_IPV6:
2434 		if (!timer_pending(&br->ip6_other_query.timer) ||
2435 		    rcu_dereference(br->ip6_querier.port) == port)
2436 			goto unlock;
2437 		break;
2438 #endif
2439 	default:
2440 		goto unlock;
2441 	}
2442 
2443 	ret = true;
2444 unlock:
2445 	rcu_read_unlock();
2446 	return ret;
2447 }
2448 EXPORT_SYMBOL_GPL(br_multicast_has_querier_adjacent);
2449 
br_mcast_stats_add(struct bridge_mcast_stats __percpu * stats,const struct sk_buff * skb,u8 type,u8 dir)2450 static void br_mcast_stats_add(struct bridge_mcast_stats __percpu *stats,
2451 			       const struct sk_buff *skb, u8 type, u8 dir)
2452 {
2453 	struct bridge_mcast_stats *pstats = this_cpu_ptr(stats);
2454 	__be16 proto = skb->protocol;
2455 	unsigned int t_len;
2456 
2457 	u64_stats_update_begin(&pstats->syncp);
2458 	switch (proto) {
2459 	case htons(ETH_P_IP):
2460 		t_len = ntohs(ip_hdr(skb)->tot_len) - ip_hdrlen(skb);
2461 		switch (type) {
2462 		case IGMP_HOST_MEMBERSHIP_REPORT:
2463 			pstats->mstats.igmp_v1reports[dir]++;
2464 			break;
2465 		case IGMPV2_HOST_MEMBERSHIP_REPORT:
2466 			pstats->mstats.igmp_v2reports[dir]++;
2467 			break;
2468 		case IGMPV3_HOST_MEMBERSHIP_REPORT:
2469 			pstats->mstats.igmp_v3reports[dir]++;
2470 			break;
2471 		case IGMP_HOST_MEMBERSHIP_QUERY:
2472 			if (t_len != sizeof(struct igmphdr)) {
2473 				pstats->mstats.igmp_v3queries[dir]++;
2474 			} else {
2475 				unsigned int offset = skb_transport_offset(skb);
2476 				struct igmphdr *ih, _ihdr;
2477 
2478 				ih = skb_header_pointer(skb, offset,
2479 							sizeof(_ihdr), &_ihdr);
2480 				if (!ih)
2481 					break;
2482 				if (!ih->code)
2483 					pstats->mstats.igmp_v1queries[dir]++;
2484 				else
2485 					pstats->mstats.igmp_v2queries[dir]++;
2486 			}
2487 			break;
2488 		case IGMP_HOST_LEAVE_MESSAGE:
2489 			pstats->mstats.igmp_leaves[dir]++;
2490 			break;
2491 		}
2492 		break;
2493 #if IS_ENABLED(CONFIG_IPV6)
2494 	case htons(ETH_P_IPV6):
2495 		t_len = ntohs(ipv6_hdr(skb)->payload_len) +
2496 			sizeof(struct ipv6hdr);
2497 		t_len -= skb_network_header_len(skb);
2498 		switch (type) {
2499 		case ICMPV6_MGM_REPORT:
2500 			pstats->mstats.mld_v1reports[dir]++;
2501 			break;
2502 		case ICMPV6_MLD2_REPORT:
2503 			pstats->mstats.mld_v2reports[dir]++;
2504 			break;
2505 		case ICMPV6_MGM_QUERY:
2506 			if (t_len != sizeof(struct mld_msg))
2507 				pstats->mstats.mld_v2queries[dir]++;
2508 			else
2509 				pstats->mstats.mld_v1queries[dir]++;
2510 			break;
2511 		case ICMPV6_MGM_REDUCTION:
2512 			pstats->mstats.mld_leaves[dir]++;
2513 			break;
2514 		}
2515 		break;
2516 #endif /* CONFIG_IPV6 */
2517 	}
2518 	u64_stats_update_end(&pstats->syncp);
2519 }
2520 
br_multicast_count(struct net_bridge * br,const struct net_bridge_port * p,const struct sk_buff * skb,u8 type,u8 dir)2521 void br_multicast_count(struct net_bridge *br, const struct net_bridge_port *p,
2522 			const struct sk_buff *skb, u8 type, u8 dir)
2523 {
2524 	struct bridge_mcast_stats __percpu *stats;
2525 
2526 	/* if multicast_disabled is true then igmp type can't be set */
2527 	if (!type || !br->multicast_stats_enabled)
2528 		return;
2529 
2530 	if (p)
2531 		stats = p->mcast_stats;
2532 	else
2533 		stats = br->mcast_stats;
2534 	if (WARN_ON(!stats))
2535 		return;
2536 
2537 	br_mcast_stats_add(stats, skb, type, dir);
2538 }
2539 
br_multicast_init_stats(struct net_bridge * br)2540 int br_multicast_init_stats(struct net_bridge *br)
2541 {
2542 	br->mcast_stats = netdev_alloc_pcpu_stats(struct bridge_mcast_stats);
2543 	if (!br->mcast_stats)
2544 		return -ENOMEM;
2545 
2546 	return 0;
2547 }
2548 
br_multicast_uninit_stats(struct net_bridge * br)2549 void br_multicast_uninit_stats(struct net_bridge *br)
2550 {
2551 	free_percpu(br->mcast_stats);
2552 }
2553 
mcast_stats_add_dir(u64 * dst,u64 * src)2554 static void mcast_stats_add_dir(u64 *dst, u64 *src)
2555 {
2556 	dst[BR_MCAST_DIR_RX] += src[BR_MCAST_DIR_RX];
2557 	dst[BR_MCAST_DIR_TX] += src[BR_MCAST_DIR_TX];
2558 }
2559 
br_multicast_get_stats(const struct net_bridge * br,const struct net_bridge_port * p,struct br_mcast_stats * dest)2560 void br_multicast_get_stats(const struct net_bridge *br,
2561 			    const struct net_bridge_port *p,
2562 			    struct br_mcast_stats *dest)
2563 {
2564 	struct bridge_mcast_stats __percpu *stats;
2565 	struct br_mcast_stats tdst;
2566 	int i;
2567 
2568 	memset(dest, 0, sizeof(*dest));
2569 	if (p)
2570 		stats = p->mcast_stats;
2571 	else
2572 		stats = br->mcast_stats;
2573 	if (WARN_ON(!stats))
2574 		return;
2575 
2576 	memset(&tdst, 0, sizeof(tdst));
2577 	for_each_possible_cpu(i) {
2578 		struct bridge_mcast_stats *cpu_stats = per_cpu_ptr(stats, i);
2579 		struct br_mcast_stats temp;
2580 		unsigned int start;
2581 
2582 		do {
2583 			start = u64_stats_fetch_begin_irq(&cpu_stats->syncp);
2584 			memcpy(&temp, &cpu_stats->mstats, sizeof(temp));
2585 		} while (u64_stats_fetch_retry_irq(&cpu_stats->syncp, start));
2586 
2587 		mcast_stats_add_dir(tdst.igmp_v1queries, temp.igmp_v1queries);
2588 		mcast_stats_add_dir(tdst.igmp_v2queries, temp.igmp_v2queries);
2589 		mcast_stats_add_dir(tdst.igmp_v3queries, temp.igmp_v3queries);
2590 		mcast_stats_add_dir(tdst.igmp_leaves, temp.igmp_leaves);
2591 		mcast_stats_add_dir(tdst.igmp_v1reports, temp.igmp_v1reports);
2592 		mcast_stats_add_dir(tdst.igmp_v2reports, temp.igmp_v2reports);
2593 		mcast_stats_add_dir(tdst.igmp_v3reports, temp.igmp_v3reports);
2594 		tdst.igmp_parse_errors += temp.igmp_parse_errors;
2595 
2596 		mcast_stats_add_dir(tdst.mld_v1queries, temp.mld_v1queries);
2597 		mcast_stats_add_dir(tdst.mld_v2queries, temp.mld_v2queries);
2598 		mcast_stats_add_dir(tdst.mld_leaves, temp.mld_leaves);
2599 		mcast_stats_add_dir(tdst.mld_v1reports, temp.mld_v1reports);
2600 		mcast_stats_add_dir(tdst.mld_v2reports, temp.mld_v2reports);
2601 		tdst.mld_parse_errors += temp.mld_parse_errors;
2602 	}
2603 	memcpy(dest, &tdst, sizeof(*dest));
2604 }
2605