xref: /openbmc/linux/net/ipv6/mcast.c (revision fd5e9fccbd504c5179ab57ff695c610bca8809d6)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *	Multicast support for IPv6
4  *	Linux INET6 implementation
5  *
6  *	Authors:
7  *	Pedro Roque		<roque@di.fc.ul.pt>
8  *
9  *	Based on linux/ipv4/igmp.c and linux/ipv4/ip_sockglue.c
10  */
11 
12 /* Changes:
13  *
14  *	yoshfuji	: fix format of router-alert option
15  *	YOSHIFUJI Hideaki @USAGI:
16  *		Fixed source address for MLD message based on
17  *		<draft-ietf-magma-mld-source-05.txt>.
18  *	YOSHIFUJI Hideaki @USAGI:
19  *		- Ignore Queries for invalid addresses.
20  *		- MLD for link-local addresses.
21  *	David L Stevens <dlstevens@us.ibm.com>:
22  *		- MLDv2 support
23  */
24 
25 #include <linux/module.h>
26 #include <linux/errno.h>
27 #include <linux/types.h>
28 #include <linux/string.h>
29 #include <linux/socket.h>
30 #include <linux/sockios.h>
31 #include <linux/jiffies.h>
32 #include <linux/net.h>
33 #include <linux/in.h>
34 #include <linux/in6.h>
35 #include <linux/netdevice.h>
36 #include <linux/if_arp.h>
37 #include <linux/route.h>
38 #include <linux/init.h>
39 #include <linux/proc_fs.h>
40 #include <linux/seq_file.h>
41 #include <linux/slab.h>
42 #include <linux/pkt_sched.h>
43 #include <net/mld.h>
44 #include <linux/workqueue.h>
45 
46 #include <linux/netfilter.h>
47 #include <linux/netfilter_ipv6.h>
48 
49 #include <net/net_namespace.h>
50 #include <net/sock.h>
51 #include <net/snmp.h>
52 
53 #include <net/ipv6.h>
54 #include <net/protocol.h>
55 #include <net/if_inet6.h>
56 #include <net/ndisc.h>
57 #include <net/addrconf.h>
58 #include <net/ip6_route.h>
59 #include <net/inet_common.h>
60 
61 #include <net/ip6_checksum.h>
62 
63 /* Ensure that we have struct in6_addr aligned on 32bit word. */
64 static int __mld2_query_bugs[] __attribute__((__unused__)) = {
65 	BUILD_BUG_ON_ZERO(offsetof(struct mld2_query, mld2q_srcs) % 4),
66 	BUILD_BUG_ON_ZERO(offsetof(struct mld2_report, mld2r_grec) % 4),
67 	BUILD_BUG_ON_ZERO(offsetof(struct mld2_grec, grec_mca) % 4)
68 };
69 
70 static struct workqueue_struct *mld_wq;
71 static struct in6_addr mld2_all_mcr = MLD2_ALL_MCR_INIT;
72 
73 static void igmp6_join_group(struct ifmcaddr6 *ma);
74 static void igmp6_leave_group(struct ifmcaddr6 *ma);
75 static void mld_mca_work(struct work_struct *work);
76 
77 static void mld_ifc_event(struct inet6_dev *idev);
78 static bool mld_in_v1_mode(const struct inet6_dev *idev);
79 static int sf_setstate(struct ifmcaddr6 *pmc);
80 static void sf_markstate(struct ifmcaddr6 *pmc);
81 static void ip6_mc_clear_src(struct ifmcaddr6 *pmc);
82 static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
83 			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
84 			  int delta);
85 static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
86 			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
87 			  int delta);
88 static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
89 			    struct inet6_dev *idev);
90 static int __ipv6_dev_mc_inc(struct net_device *dev,
91 			     const struct in6_addr *addr, unsigned int mode);
92 
93 #define MLD_QRV_DEFAULT		2
94 /* RFC3810, 9.2. Query Interval */
95 #define MLD_QI_DEFAULT		(125 * HZ)
96 /* RFC3810, 9.3. Query Response Interval */
97 #define MLD_QRI_DEFAULT		(10 * HZ)
98 
99 /* RFC3810, 8.1 Query Version Distinctions */
100 #define MLD_V1_QUERY_LEN	24
101 #define MLD_V2_QUERY_LEN_MIN	28
102 
103 #define IPV6_MLD_MAX_MSF	64
104 
105 int sysctl_mld_max_msf __read_mostly = IPV6_MLD_MAX_MSF;
106 int sysctl_mld_qrv __read_mostly = MLD_QRV_DEFAULT;
107 
108 /*
109  *	socket join on multicast group
110  */
111 #define mc_dereference(e, idev) \
112 	rcu_dereference_protected(e, lockdep_is_held(&(idev)->mc_lock))
113 
114 #define sock_dereference(e, sk) \
115 	rcu_dereference_protected(e, lockdep_sock_is_held(sk))
116 
117 #define for_each_pmc_socklock(np, sk, pmc)			\
118 	for (pmc = sock_dereference((np)->ipv6_mc_list, sk);	\
119 	     pmc;						\
120 	     pmc = sock_dereference(pmc->next, sk))
121 
122 #define for_each_pmc_rcu(np, pmc)				\
123 	for (pmc = rcu_dereference((np)->ipv6_mc_list);		\
124 	     pmc;						\
125 	     pmc = rcu_dereference(pmc->next))
126 
127 #define for_each_psf_mclock(mc, psf)				\
128 	for (psf = mc_dereference((mc)->mca_sources, mc->idev);	\
129 	     psf;						\
130 	     psf = mc_dereference(psf->sf_next, mc->idev))
131 
132 #define for_each_psf_rcu(mc, psf)				\
133 	for (psf = rcu_dereference((mc)->mca_sources);		\
134 	     psf;						\
135 	     psf = rcu_dereference(psf->sf_next))
136 
137 #define for_each_psf_tomb(mc, psf)				\
138 	for (psf = mc_dereference((mc)->mca_tomb, mc->idev);	\
139 	     psf;						\
140 	     psf = mc_dereference(psf->sf_next, mc->idev))
141 
142 #define for_each_mc_mclock(idev, mc)				\
143 	for (mc = mc_dereference((idev)->mc_list, idev);	\
144 	     mc;						\
145 	     mc = mc_dereference(mc->next, idev))
146 
147 #define for_each_mc_rcu(idev, mc)				\
148 	for (mc = rcu_dereference((idev)->mc_list);             \
149 	     mc;                                                \
150 	     mc = rcu_dereference(mc->next))
151 
152 #define for_each_mc_tomb(idev, mc)				\
153 	for (mc = mc_dereference((idev)->mc_tomb, idev);	\
154 	     mc;						\
155 	     mc = mc_dereference(mc->next, idev))
156 
unsolicited_report_interval(struct inet6_dev * idev)157 static int unsolicited_report_interval(struct inet6_dev *idev)
158 {
159 	int iv;
160 
161 	if (mld_in_v1_mode(idev))
162 		iv = idev->cnf.mldv1_unsolicited_report_interval;
163 	else
164 		iv = idev->cnf.mldv2_unsolicited_report_interval;
165 
166 	return iv > 0 ? iv : 1;
167 }
168 
__ipv6_sock_mc_join(struct sock * sk,int ifindex,const struct in6_addr * addr,unsigned int mode)169 static int __ipv6_sock_mc_join(struct sock *sk, int ifindex,
170 			       const struct in6_addr *addr, unsigned int mode)
171 {
172 	struct net_device *dev = NULL;
173 	struct ipv6_mc_socklist *mc_lst;
174 	struct ipv6_pinfo *np = inet6_sk(sk);
175 	struct net *net = sock_net(sk);
176 	int err;
177 
178 	ASSERT_RTNL();
179 
180 	if (!ipv6_addr_is_multicast(addr))
181 		return -EINVAL;
182 
183 	for_each_pmc_socklock(np, sk, mc_lst) {
184 		if ((ifindex == 0 || mc_lst->ifindex == ifindex) &&
185 		    ipv6_addr_equal(&mc_lst->addr, addr))
186 			return -EADDRINUSE;
187 	}
188 
189 	mc_lst = sock_kmalloc(sk, sizeof(struct ipv6_mc_socklist), GFP_KERNEL);
190 
191 	if (!mc_lst)
192 		return -ENOMEM;
193 
194 	mc_lst->next = NULL;
195 	mc_lst->addr = *addr;
196 
197 	if (ifindex == 0) {
198 		struct rt6_info *rt;
199 		rt = rt6_lookup(net, addr, NULL, 0, NULL, 0);
200 		if (rt) {
201 			dev = rt->dst.dev;
202 			ip6_rt_put(rt);
203 		}
204 	} else
205 		dev = __dev_get_by_index(net, ifindex);
206 
207 	if (!dev) {
208 		sock_kfree_s(sk, mc_lst, sizeof(*mc_lst));
209 		return -ENODEV;
210 	}
211 
212 	mc_lst->ifindex = dev->ifindex;
213 	mc_lst->sfmode = mode;
214 	RCU_INIT_POINTER(mc_lst->sflist, NULL);
215 
216 	/*
217 	 *	now add/increase the group membership on the device
218 	 */
219 
220 	err = __ipv6_dev_mc_inc(dev, addr, mode);
221 
222 	if (err) {
223 		sock_kfree_s(sk, mc_lst, sizeof(*mc_lst));
224 		return err;
225 	}
226 
227 	mc_lst->next = np->ipv6_mc_list;
228 	rcu_assign_pointer(np->ipv6_mc_list, mc_lst);
229 
230 	return 0;
231 }
232 
ipv6_sock_mc_join(struct sock * sk,int ifindex,const struct in6_addr * addr)233 int ipv6_sock_mc_join(struct sock *sk, int ifindex, const struct in6_addr *addr)
234 {
235 	return __ipv6_sock_mc_join(sk, ifindex, addr, MCAST_EXCLUDE);
236 }
237 EXPORT_SYMBOL(ipv6_sock_mc_join);
238 
ipv6_sock_mc_join_ssm(struct sock * sk,int ifindex,const struct in6_addr * addr,unsigned int mode)239 int ipv6_sock_mc_join_ssm(struct sock *sk, int ifindex,
240 			  const struct in6_addr *addr, unsigned int mode)
241 {
242 	return __ipv6_sock_mc_join(sk, ifindex, addr, mode);
243 }
244 
245 /*
246  *	socket leave on multicast group
247  */
ipv6_sock_mc_drop(struct sock * sk,int ifindex,const struct in6_addr * addr)248 int ipv6_sock_mc_drop(struct sock *sk, int ifindex, const struct in6_addr *addr)
249 {
250 	struct ipv6_pinfo *np = inet6_sk(sk);
251 	struct ipv6_mc_socklist *mc_lst;
252 	struct ipv6_mc_socklist __rcu **lnk;
253 	struct net *net = sock_net(sk);
254 
255 	ASSERT_RTNL();
256 
257 	if (!ipv6_addr_is_multicast(addr))
258 		return -EINVAL;
259 
260 	for (lnk = &np->ipv6_mc_list;
261 	     (mc_lst = sock_dereference(*lnk, sk)) != NULL;
262 	      lnk = &mc_lst->next) {
263 		if ((ifindex == 0 || mc_lst->ifindex == ifindex) &&
264 		    ipv6_addr_equal(&mc_lst->addr, addr)) {
265 			struct net_device *dev;
266 
267 			*lnk = mc_lst->next;
268 
269 			dev = __dev_get_by_index(net, mc_lst->ifindex);
270 			if (dev) {
271 				struct inet6_dev *idev = __in6_dev_get(dev);
272 
273 				ip6_mc_leave_src(sk, mc_lst, idev);
274 				if (idev)
275 					__ipv6_dev_mc_dec(idev, &mc_lst->addr);
276 			} else {
277 				ip6_mc_leave_src(sk, mc_lst, NULL);
278 			}
279 
280 			atomic_sub(sizeof(*mc_lst), &sk->sk_omem_alloc);
281 			kfree_rcu(mc_lst, rcu);
282 			return 0;
283 		}
284 	}
285 
286 	return -EADDRNOTAVAIL;
287 }
288 EXPORT_SYMBOL(ipv6_sock_mc_drop);
289 
ip6_mc_find_dev_rtnl(struct net * net,const struct in6_addr * group,int ifindex)290 static struct inet6_dev *ip6_mc_find_dev_rtnl(struct net *net,
291 					      const struct in6_addr *group,
292 					      int ifindex)
293 {
294 	struct net_device *dev = NULL;
295 	struct inet6_dev *idev = NULL;
296 
297 	if (ifindex == 0) {
298 		struct rt6_info *rt = rt6_lookup(net, group, NULL, 0, NULL, 0);
299 
300 		if (rt) {
301 			dev = rt->dst.dev;
302 			ip6_rt_put(rt);
303 		}
304 	} else {
305 		dev = __dev_get_by_index(net, ifindex);
306 	}
307 
308 	if (!dev)
309 		return NULL;
310 	idev = __in6_dev_get(dev);
311 	if (!idev)
312 		return NULL;
313 	if (idev->dead)
314 		return NULL;
315 	return idev;
316 }
317 
__ipv6_sock_mc_close(struct sock * sk)318 void __ipv6_sock_mc_close(struct sock *sk)
319 {
320 	struct ipv6_pinfo *np = inet6_sk(sk);
321 	struct ipv6_mc_socklist *mc_lst;
322 	struct net *net = sock_net(sk);
323 
324 	ASSERT_RTNL();
325 
326 	while ((mc_lst = sock_dereference(np->ipv6_mc_list, sk)) != NULL) {
327 		struct net_device *dev;
328 
329 		np->ipv6_mc_list = mc_lst->next;
330 
331 		dev = __dev_get_by_index(net, mc_lst->ifindex);
332 		if (dev) {
333 			struct inet6_dev *idev = __in6_dev_get(dev);
334 
335 			ip6_mc_leave_src(sk, mc_lst, idev);
336 			if (idev)
337 				__ipv6_dev_mc_dec(idev, &mc_lst->addr);
338 		} else {
339 			ip6_mc_leave_src(sk, mc_lst, NULL);
340 		}
341 
342 		atomic_sub(sizeof(*mc_lst), &sk->sk_omem_alloc);
343 		kfree_rcu(mc_lst, rcu);
344 	}
345 }
346 
ipv6_sock_mc_close(struct sock * sk)347 void ipv6_sock_mc_close(struct sock *sk)
348 {
349 	struct ipv6_pinfo *np = inet6_sk(sk);
350 
351 	if (!rcu_access_pointer(np->ipv6_mc_list))
352 		return;
353 
354 	rtnl_lock();
355 	lock_sock(sk);
356 	__ipv6_sock_mc_close(sk);
357 	release_sock(sk);
358 	rtnl_unlock();
359 }
360 
ip6_mc_source(int add,int omode,struct sock * sk,struct group_source_req * pgsr)361 int ip6_mc_source(int add, int omode, struct sock *sk,
362 	struct group_source_req *pgsr)
363 {
364 	struct in6_addr *source, *group;
365 	struct ipv6_mc_socklist *pmc;
366 	struct inet6_dev *idev;
367 	struct ipv6_pinfo *inet6 = inet6_sk(sk);
368 	struct ip6_sf_socklist *psl;
369 	struct net *net = sock_net(sk);
370 	int i, j, rv;
371 	int leavegroup = 0;
372 	int err;
373 
374 	source = &((struct sockaddr_in6 *)&pgsr->gsr_source)->sin6_addr;
375 	group = &((struct sockaddr_in6 *)&pgsr->gsr_group)->sin6_addr;
376 
377 	if (!ipv6_addr_is_multicast(group))
378 		return -EINVAL;
379 
380 	idev = ip6_mc_find_dev_rtnl(net, group, pgsr->gsr_interface);
381 	if (!idev)
382 		return -ENODEV;
383 
384 	err = -EADDRNOTAVAIL;
385 
386 	mutex_lock(&idev->mc_lock);
387 	for_each_pmc_socklock(inet6, sk, pmc) {
388 		if (pgsr->gsr_interface && pmc->ifindex != pgsr->gsr_interface)
389 			continue;
390 		if (ipv6_addr_equal(&pmc->addr, group))
391 			break;
392 	}
393 	if (!pmc) {		/* must have a prior join */
394 		err = -EINVAL;
395 		goto done;
396 	}
397 	/* if a source filter was set, must be the same mode as before */
398 	if (rcu_access_pointer(pmc->sflist)) {
399 		if (pmc->sfmode != omode) {
400 			err = -EINVAL;
401 			goto done;
402 		}
403 	} else if (pmc->sfmode != omode) {
404 		/* allow mode switches for empty-set filters */
405 		ip6_mc_add_src(idev, group, omode, 0, NULL, 0);
406 		ip6_mc_del_src(idev, group, pmc->sfmode, 0, NULL, 0);
407 		pmc->sfmode = omode;
408 	}
409 
410 	psl = sock_dereference(pmc->sflist, sk);
411 	if (!add) {
412 		if (!psl)
413 			goto done;	/* err = -EADDRNOTAVAIL */
414 		rv = !0;
415 		for (i = 0; i < psl->sl_count; i++) {
416 			rv = !ipv6_addr_equal(&psl->sl_addr[i], source);
417 			if (rv == 0)
418 				break;
419 		}
420 		if (rv)		/* source not found */
421 			goto done;	/* err = -EADDRNOTAVAIL */
422 
423 		/* special case - (INCLUDE, empty) == LEAVE_GROUP */
424 		if (psl->sl_count == 1 && omode == MCAST_INCLUDE) {
425 			leavegroup = 1;
426 			goto done;
427 		}
428 
429 		/* update the interface filter */
430 		ip6_mc_del_src(idev, group, omode, 1, source, 1);
431 
432 		for (j = i+1; j < psl->sl_count; j++)
433 			psl->sl_addr[j-1] = psl->sl_addr[j];
434 		psl->sl_count--;
435 		err = 0;
436 		goto done;
437 	}
438 	/* else, add a new source to the filter */
439 
440 	if (psl && psl->sl_count >= sysctl_mld_max_msf) {
441 		err = -ENOBUFS;
442 		goto done;
443 	}
444 	if (!psl || psl->sl_count == psl->sl_max) {
445 		struct ip6_sf_socklist *newpsl;
446 		int count = IP6_SFBLOCK;
447 
448 		if (psl)
449 			count += psl->sl_max;
450 		newpsl = sock_kmalloc(sk, struct_size(newpsl, sl_addr, count),
451 				      GFP_KERNEL);
452 		if (!newpsl) {
453 			err = -ENOBUFS;
454 			goto done;
455 		}
456 		newpsl->sl_max = count;
457 		newpsl->sl_count = count - IP6_SFBLOCK;
458 		if (psl) {
459 			for (i = 0; i < psl->sl_count; i++)
460 				newpsl->sl_addr[i] = psl->sl_addr[i];
461 			atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
462 				   &sk->sk_omem_alloc);
463 		}
464 		rcu_assign_pointer(pmc->sflist, newpsl);
465 		kfree_rcu(psl, rcu);
466 		psl = newpsl;
467 	}
468 	rv = 1;	/* > 0 for insert logic below if sl_count is 0 */
469 	for (i = 0; i < psl->sl_count; i++) {
470 		rv = !ipv6_addr_equal(&psl->sl_addr[i], source);
471 		if (rv == 0) /* There is an error in the address. */
472 			goto done;
473 	}
474 	for (j = psl->sl_count-1; j >= i; j--)
475 		psl->sl_addr[j+1] = psl->sl_addr[j];
476 	psl->sl_addr[i] = *source;
477 	psl->sl_count++;
478 	err = 0;
479 	/* update the interface list */
480 	ip6_mc_add_src(idev, group, omode, 1, source, 1);
481 done:
482 	mutex_unlock(&idev->mc_lock);
483 	if (leavegroup)
484 		err = ipv6_sock_mc_drop(sk, pgsr->gsr_interface, group);
485 	return err;
486 }
487 
ip6_mc_msfilter(struct sock * sk,struct group_filter * gsf,struct sockaddr_storage * list)488 int ip6_mc_msfilter(struct sock *sk, struct group_filter *gsf,
489 		    struct sockaddr_storage *list)
490 {
491 	const struct in6_addr *group;
492 	struct ipv6_mc_socklist *pmc;
493 	struct inet6_dev *idev;
494 	struct ipv6_pinfo *inet6 = inet6_sk(sk);
495 	struct ip6_sf_socklist *newpsl, *psl;
496 	struct net *net = sock_net(sk);
497 	int leavegroup = 0;
498 	int i, err;
499 
500 	group = &((struct sockaddr_in6 *)&gsf->gf_group)->sin6_addr;
501 
502 	if (!ipv6_addr_is_multicast(group))
503 		return -EINVAL;
504 	if (gsf->gf_fmode != MCAST_INCLUDE &&
505 	    gsf->gf_fmode != MCAST_EXCLUDE)
506 		return -EINVAL;
507 
508 	idev = ip6_mc_find_dev_rtnl(net, group, gsf->gf_interface);
509 	if (!idev)
510 		return -ENODEV;
511 
512 	err = 0;
513 
514 	if (gsf->gf_fmode == MCAST_INCLUDE && gsf->gf_numsrc == 0) {
515 		leavegroup = 1;
516 		goto done;
517 	}
518 
519 	for_each_pmc_socklock(inet6, sk, pmc) {
520 		if (pmc->ifindex != gsf->gf_interface)
521 			continue;
522 		if (ipv6_addr_equal(&pmc->addr, group))
523 			break;
524 	}
525 	if (!pmc) {		/* must have a prior join */
526 		err = -EINVAL;
527 		goto done;
528 	}
529 	if (gsf->gf_numsrc) {
530 		newpsl = sock_kmalloc(sk, struct_size(newpsl, sl_addr,
531 						      gsf->gf_numsrc),
532 				      GFP_KERNEL);
533 		if (!newpsl) {
534 			err = -ENOBUFS;
535 			goto done;
536 		}
537 		newpsl->sl_max = newpsl->sl_count = gsf->gf_numsrc;
538 		for (i = 0; i < newpsl->sl_count; ++i, ++list) {
539 			struct sockaddr_in6 *psin6;
540 
541 			psin6 = (struct sockaddr_in6 *)list;
542 			newpsl->sl_addr[i] = psin6->sin6_addr;
543 		}
544 		mutex_lock(&idev->mc_lock);
545 		err = ip6_mc_add_src(idev, group, gsf->gf_fmode,
546 				     newpsl->sl_count, newpsl->sl_addr, 0);
547 		if (err) {
548 			mutex_unlock(&idev->mc_lock);
549 			sock_kfree_s(sk, newpsl, struct_size(newpsl, sl_addr,
550 							     newpsl->sl_max));
551 			goto done;
552 		}
553 		mutex_unlock(&idev->mc_lock);
554 	} else {
555 		newpsl = NULL;
556 		mutex_lock(&idev->mc_lock);
557 		ip6_mc_add_src(idev, group, gsf->gf_fmode, 0, NULL, 0);
558 		mutex_unlock(&idev->mc_lock);
559 	}
560 
561 	mutex_lock(&idev->mc_lock);
562 	psl = sock_dereference(pmc->sflist, sk);
563 	if (psl) {
564 		ip6_mc_del_src(idev, group, pmc->sfmode,
565 			       psl->sl_count, psl->sl_addr, 0);
566 		atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
567 			   &sk->sk_omem_alloc);
568 	} else {
569 		ip6_mc_del_src(idev, group, pmc->sfmode, 0, NULL, 0);
570 	}
571 	rcu_assign_pointer(pmc->sflist, newpsl);
572 	mutex_unlock(&idev->mc_lock);
573 	kfree_rcu(psl, rcu);
574 	pmc->sfmode = gsf->gf_fmode;
575 	err = 0;
576 done:
577 	if (leavegroup)
578 		err = ipv6_sock_mc_drop(sk, gsf->gf_interface, group);
579 	return err;
580 }
581 
ip6_mc_msfget(struct sock * sk,struct group_filter * gsf,sockptr_t optval,size_t ss_offset)582 int ip6_mc_msfget(struct sock *sk, struct group_filter *gsf,
583 		  sockptr_t optval, size_t ss_offset)
584 {
585 	struct ipv6_pinfo *inet6 = inet6_sk(sk);
586 	const struct in6_addr *group;
587 	struct ipv6_mc_socklist *pmc;
588 	struct ip6_sf_socklist *psl;
589 	int i, count, copycount;
590 
591 	group = &((struct sockaddr_in6 *)&gsf->gf_group)->sin6_addr;
592 
593 	if (!ipv6_addr_is_multicast(group))
594 		return -EINVAL;
595 
596 	/* changes to the ipv6_mc_list require the socket lock and
597 	 * rtnl lock. We have the socket lock, so reading the list is safe.
598 	 */
599 
600 	for_each_pmc_socklock(inet6, sk, pmc) {
601 		if (pmc->ifindex != gsf->gf_interface)
602 			continue;
603 		if (ipv6_addr_equal(group, &pmc->addr))
604 			break;
605 	}
606 	if (!pmc)		/* must have a prior join */
607 		return -EADDRNOTAVAIL;
608 
609 	gsf->gf_fmode = pmc->sfmode;
610 	psl = sock_dereference(pmc->sflist, sk);
611 	count = psl ? psl->sl_count : 0;
612 
613 	copycount = count < gsf->gf_numsrc ? count : gsf->gf_numsrc;
614 	gsf->gf_numsrc = count;
615 	for (i = 0; i < copycount; i++) {
616 		struct sockaddr_in6 *psin6;
617 		struct sockaddr_storage ss;
618 
619 		psin6 = (struct sockaddr_in6 *)&ss;
620 		memset(&ss, 0, sizeof(ss));
621 		psin6->sin6_family = AF_INET6;
622 		psin6->sin6_addr = psl->sl_addr[i];
623 		if (copy_to_sockptr_offset(optval, ss_offset, &ss, sizeof(ss)))
624 			return -EFAULT;
625 		ss_offset += sizeof(ss);
626 	}
627 	return 0;
628 }
629 
inet6_mc_check(const struct sock * sk,const struct in6_addr * mc_addr,const struct in6_addr * src_addr)630 bool inet6_mc_check(const struct sock *sk, const struct in6_addr *mc_addr,
631 		    const struct in6_addr *src_addr)
632 {
633 	const struct ipv6_pinfo *np = inet6_sk(sk);
634 	const struct ipv6_mc_socklist *mc;
635 	const struct ip6_sf_socklist *psl;
636 	bool rv = true;
637 
638 	rcu_read_lock();
639 	for_each_pmc_rcu(np, mc) {
640 		if (ipv6_addr_equal(&mc->addr, mc_addr))
641 			break;
642 	}
643 	if (!mc) {
644 		rcu_read_unlock();
645 		return np->mc_all;
646 	}
647 	psl = rcu_dereference(mc->sflist);
648 	if (!psl) {
649 		rv = mc->sfmode == MCAST_EXCLUDE;
650 	} else {
651 		int i;
652 
653 		for (i = 0; i < psl->sl_count; i++) {
654 			if (ipv6_addr_equal(&psl->sl_addr[i], src_addr))
655 				break;
656 		}
657 		if (mc->sfmode == MCAST_INCLUDE && i >= psl->sl_count)
658 			rv = false;
659 		if (mc->sfmode == MCAST_EXCLUDE && i < psl->sl_count)
660 			rv = false;
661 	}
662 	rcu_read_unlock();
663 
664 	return rv;
665 }
666 
667 /* called with mc_lock */
igmp6_group_added(struct ifmcaddr6 * mc)668 static void igmp6_group_added(struct ifmcaddr6 *mc)
669 {
670 	struct net_device *dev = mc->idev->dev;
671 	char buf[MAX_ADDR_LEN];
672 
673 	if (IPV6_ADDR_MC_SCOPE(&mc->mca_addr) <
674 	    IPV6_ADDR_SCOPE_LINKLOCAL)
675 		return;
676 
677 	if (!(mc->mca_flags&MAF_LOADED)) {
678 		mc->mca_flags |= MAF_LOADED;
679 		if (ndisc_mc_map(&mc->mca_addr, buf, dev, 0) == 0)
680 			dev_mc_add(dev, buf);
681 	}
682 
683 	if (!(dev->flags & IFF_UP) || (mc->mca_flags & MAF_NOREPORT))
684 		return;
685 
686 	if (mld_in_v1_mode(mc->idev)) {
687 		igmp6_join_group(mc);
688 		return;
689 	}
690 	/* else v2 */
691 
692 	/* Based on RFC3810 6.1, for newly added INCLUDE SSM, we
693 	 * should not send filter-mode change record as the mode
694 	 * should be from IN() to IN(A).
695 	 */
696 	if (mc->mca_sfmode == MCAST_EXCLUDE)
697 		mc->mca_crcount = mc->idev->mc_qrv;
698 
699 	mld_ifc_event(mc->idev);
700 }
701 
702 /* called with mc_lock */
igmp6_group_dropped(struct ifmcaddr6 * mc)703 static void igmp6_group_dropped(struct ifmcaddr6 *mc)
704 {
705 	struct net_device *dev = mc->idev->dev;
706 	char buf[MAX_ADDR_LEN];
707 
708 	if (IPV6_ADDR_MC_SCOPE(&mc->mca_addr) <
709 	    IPV6_ADDR_SCOPE_LINKLOCAL)
710 		return;
711 
712 	if (mc->mca_flags&MAF_LOADED) {
713 		mc->mca_flags &= ~MAF_LOADED;
714 		if (ndisc_mc_map(&mc->mca_addr, buf, dev, 0) == 0)
715 			dev_mc_del(dev, buf);
716 	}
717 
718 	if (mc->mca_flags & MAF_NOREPORT)
719 		return;
720 
721 	if (!mc->idev->dead)
722 		igmp6_leave_group(mc);
723 
724 	if (cancel_delayed_work(&mc->mca_work))
725 		refcount_dec(&mc->mca_refcnt);
726 }
727 
728 /*
729  * deleted ifmcaddr6 manipulation
730  * called with mc_lock
731  */
mld_add_delrec(struct inet6_dev * idev,struct ifmcaddr6 * im)732 static void mld_add_delrec(struct inet6_dev *idev, struct ifmcaddr6 *im)
733 {
734 	struct ifmcaddr6 *pmc;
735 
736 	/* this is an "ifmcaddr6" for convenience; only the fields below
737 	 * are actually used. In particular, the refcnt and users are not
738 	 * used for management of the delete list. Using the same structure
739 	 * for deleted items allows change reports to use common code with
740 	 * non-deleted or query-response MCA's.
741 	 */
742 	pmc = kzalloc(sizeof(*pmc), GFP_KERNEL);
743 	if (!pmc)
744 		return;
745 
746 	pmc->idev = im->idev;
747 	in6_dev_hold(idev);
748 	pmc->mca_addr = im->mca_addr;
749 	pmc->mca_crcount = idev->mc_qrv;
750 	pmc->mca_sfmode = im->mca_sfmode;
751 	if (pmc->mca_sfmode == MCAST_INCLUDE) {
752 		struct ip6_sf_list *psf;
753 
754 		rcu_assign_pointer(pmc->mca_tomb,
755 				   mc_dereference(im->mca_tomb, idev));
756 		rcu_assign_pointer(pmc->mca_sources,
757 				   mc_dereference(im->mca_sources, idev));
758 		RCU_INIT_POINTER(im->mca_tomb, NULL);
759 		RCU_INIT_POINTER(im->mca_sources, NULL);
760 
761 		for_each_psf_mclock(pmc, psf)
762 			psf->sf_crcount = pmc->mca_crcount;
763 	}
764 
765 	rcu_assign_pointer(pmc->next, idev->mc_tomb);
766 	rcu_assign_pointer(idev->mc_tomb, pmc);
767 }
768 
769 /* called with mc_lock */
mld_del_delrec(struct inet6_dev * idev,struct ifmcaddr6 * im)770 static void mld_del_delrec(struct inet6_dev *idev, struct ifmcaddr6 *im)
771 {
772 	struct ip6_sf_list *psf, *sources, *tomb;
773 	struct in6_addr *pmca = &im->mca_addr;
774 	struct ifmcaddr6 *pmc, *pmc_prev;
775 
776 	pmc_prev = NULL;
777 	for_each_mc_tomb(idev, pmc) {
778 		if (ipv6_addr_equal(&pmc->mca_addr, pmca))
779 			break;
780 		pmc_prev = pmc;
781 	}
782 	if (pmc) {
783 		if (pmc_prev)
784 			rcu_assign_pointer(pmc_prev->next, pmc->next);
785 		else
786 			rcu_assign_pointer(idev->mc_tomb, pmc->next);
787 	}
788 
789 	if (pmc) {
790 		im->idev = pmc->idev;
791 		if (im->mca_sfmode == MCAST_INCLUDE) {
792 			tomb = rcu_replace_pointer(im->mca_tomb,
793 						   mc_dereference(pmc->mca_tomb, pmc->idev),
794 						   lockdep_is_held(&im->idev->mc_lock));
795 			rcu_assign_pointer(pmc->mca_tomb, tomb);
796 
797 			sources = rcu_replace_pointer(im->mca_sources,
798 						      mc_dereference(pmc->mca_sources, pmc->idev),
799 						      lockdep_is_held(&im->idev->mc_lock));
800 			rcu_assign_pointer(pmc->mca_sources, sources);
801 			for_each_psf_mclock(im, psf)
802 				psf->sf_crcount = idev->mc_qrv;
803 		} else {
804 			im->mca_crcount = idev->mc_qrv;
805 		}
806 		in6_dev_put(pmc->idev);
807 		ip6_mc_clear_src(pmc);
808 		kfree_rcu(pmc, rcu);
809 	}
810 }
811 
812 /* called with mc_lock */
mld_clear_delrec(struct inet6_dev * idev)813 static void mld_clear_delrec(struct inet6_dev *idev)
814 {
815 	struct ifmcaddr6 *pmc, *nextpmc;
816 
817 	pmc = mc_dereference(idev->mc_tomb, idev);
818 	RCU_INIT_POINTER(idev->mc_tomb, NULL);
819 
820 	for (; pmc; pmc = nextpmc) {
821 		nextpmc = mc_dereference(pmc->next, idev);
822 		ip6_mc_clear_src(pmc);
823 		in6_dev_put(pmc->idev);
824 		kfree_rcu(pmc, rcu);
825 	}
826 
827 	/* clear dead sources, too */
828 	for_each_mc_mclock(idev, pmc) {
829 		struct ip6_sf_list *psf, *psf_next;
830 
831 		psf = mc_dereference(pmc->mca_tomb, idev);
832 		RCU_INIT_POINTER(pmc->mca_tomb, NULL);
833 		for (; psf; psf = psf_next) {
834 			psf_next = mc_dereference(psf->sf_next, idev);
835 			kfree_rcu(psf, rcu);
836 		}
837 	}
838 }
839 
mld_clear_query(struct inet6_dev * idev)840 static void mld_clear_query(struct inet6_dev *idev)
841 {
842 	struct sk_buff *skb;
843 
844 	spin_lock_bh(&idev->mc_query_lock);
845 	while ((skb = __skb_dequeue(&idev->mc_query_queue)))
846 		kfree_skb(skb);
847 	spin_unlock_bh(&idev->mc_query_lock);
848 }
849 
mld_clear_report(struct inet6_dev * idev)850 static void mld_clear_report(struct inet6_dev *idev)
851 {
852 	struct sk_buff *skb;
853 
854 	spin_lock_bh(&idev->mc_report_lock);
855 	while ((skb = __skb_dequeue(&idev->mc_report_queue)))
856 		kfree_skb(skb);
857 	spin_unlock_bh(&idev->mc_report_lock);
858 }
859 
mca_get(struct ifmcaddr6 * mc)860 static void mca_get(struct ifmcaddr6 *mc)
861 {
862 	refcount_inc(&mc->mca_refcnt);
863 }
864 
ma_put(struct ifmcaddr6 * mc)865 static void ma_put(struct ifmcaddr6 *mc)
866 {
867 	if (refcount_dec_and_test(&mc->mca_refcnt)) {
868 		in6_dev_put(mc->idev);
869 		kfree_rcu(mc, rcu);
870 	}
871 }
872 
873 /* called with mc_lock */
mca_alloc(struct inet6_dev * idev,const struct in6_addr * addr,unsigned int mode)874 static struct ifmcaddr6 *mca_alloc(struct inet6_dev *idev,
875 				   const struct in6_addr *addr,
876 				   unsigned int mode)
877 {
878 	struct ifmcaddr6 *mc;
879 
880 	mc = kzalloc(sizeof(*mc), GFP_KERNEL);
881 	if (!mc)
882 		return NULL;
883 
884 	INIT_DELAYED_WORK(&mc->mca_work, mld_mca_work);
885 
886 	mc->mca_addr = *addr;
887 	mc->idev = idev; /* reference taken by caller */
888 	mc->mca_users = 1;
889 	/* mca_stamp should be updated upon changes */
890 	mc->mca_cstamp = mc->mca_tstamp = jiffies;
891 	refcount_set(&mc->mca_refcnt, 1);
892 
893 	mc->mca_sfmode = mode;
894 	mc->mca_sfcount[mode] = 1;
895 
896 	if (ipv6_addr_is_ll_all_nodes(&mc->mca_addr) ||
897 	    IPV6_ADDR_MC_SCOPE(&mc->mca_addr) < IPV6_ADDR_SCOPE_LINKLOCAL)
898 		mc->mca_flags |= MAF_NOREPORT;
899 
900 	return mc;
901 }
902 
903 /*
904  *	device multicast group inc (add if not found)
905  */
__ipv6_dev_mc_inc(struct net_device * dev,const struct in6_addr * addr,unsigned int mode)906 static int __ipv6_dev_mc_inc(struct net_device *dev,
907 			     const struct in6_addr *addr, unsigned int mode)
908 {
909 	struct ifmcaddr6 *mc;
910 	struct inet6_dev *idev;
911 
912 	ASSERT_RTNL();
913 
914 	/* we need to take a reference on idev */
915 	idev = in6_dev_get(dev);
916 
917 	if (!idev)
918 		return -EINVAL;
919 
920 	if (idev->dead) {
921 		in6_dev_put(idev);
922 		return -ENODEV;
923 	}
924 
925 	mutex_lock(&idev->mc_lock);
926 	for_each_mc_mclock(idev, mc) {
927 		if (ipv6_addr_equal(&mc->mca_addr, addr)) {
928 			mc->mca_users++;
929 			ip6_mc_add_src(idev, &mc->mca_addr, mode, 0, NULL, 0);
930 			mutex_unlock(&idev->mc_lock);
931 			in6_dev_put(idev);
932 			return 0;
933 		}
934 	}
935 
936 	mc = mca_alloc(idev, addr, mode);
937 	if (!mc) {
938 		mutex_unlock(&idev->mc_lock);
939 		in6_dev_put(idev);
940 		return -ENOMEM;
941 	}
942 
943 	rcu_assign_pointer(mc->next, idev->mc_list);
944 	rcu_assign_pointer(idev->mc_list, mc);
945 
946 	mca_get(mc);
947 
948 	mld_del_delrec(idev, mc);
949 	igmp6_group_added(mc);
950 	mutex_unlock(&idev->mc_lock);
951 	ma_put(mc);
952 	return 0;
953 }
954 
ipv6_dev_mc_inc(struct net_device * dev,const struct in6_addr * addr)955 int ipv6_dev_mc_inc(struct net_device *dev, const struct in6_addr *addr)
956 {
957 	return __ipv6_dev_mc_inc(dev, addr, MCAST_EXCLUDE);
958 }
959 EXPORT_SYMBOL(ipv6_dev_mc_inc);
960 
961 /*
962  * device multicast group del
963  */
__ipv6_dev_mc_dec(struct inet6_dev * idev,const struct in6_addr * addr)964 int __ipv6_dev_mc_dec(struct inet6_dev *idev, const struct in6_addr *addr)
965 {
966 	struct ifmcaddr6 *ma, __rcu **map;
967 
968 	ASSERT_RTNL();
969 
970 	mutex_lock(&idev->mc_lock);
971 	for (map = &idev->mc_list;
972 	     (ma = mc_dereference(*map, idev));
973 	     map = &ma->next) {
974 		if (ipv6_addr_equal(&ma->mca_addr, addr)) {
975 			if (--ma->mca_users == 0) {
976 				*map = ma->next;
977 
978 				igmp6_group_dropped(ma);
979 				ip6_mc_clear_src(ma);
980 				mutex_unlock(&idev->mc_lock);
981 
982 				ma_put(ma);
983 				return 0;
984 			}
985 			mutex_unlock(&idev->mc_lock);
986 			return 0;
987 		}
988 	}
989 
990 	mutex_unlock(&idev->mc_lock);
991 	return -ENOENT;
992 }
993 
ipv6_dev_mc_dec(struct net_device * dev,const struct in6_addr * addr)994 int ipv6_dev_mc_dec(struct net_device *dev, const struct in6_addr *addr)
995 {
996 	struct inet6_dev *idev;
997 	int err;
998 
999 	ASSERT_RTNL();
1000 
1001 	idev = __in6_dev_get(dev);
1002 	if (!idev)
1003 		err = -ENODEV;
1004 	else
1005 		err = __ipv6_dev_mc_dec(idev, addr);
1006 
1007 	return err;
1008 }
1009 EXPORT_SYMBOL(ipv6_dev_mc_dec);
1010 
1011 /*
1012  *	check if the interface/address pair is valid
1013  */
ipv6_chk_mcast_addr(struct net_device * dev,const struct in6_addr * group,const struct in6_addr * src_addr)1014 bool ipv6_chk_mcast_addr(struct net_device *dev, const struct in6_addr *group,
1015 			 const struct in6_addr *src_addr)
1016 {
1017 	struct inet6_dev *idev;
1018 	struct ifmcaddr6 *mc;
1019 	bool rv = false;
1020 
1021 	rcu_read_lock();
1022 	idev = __in6_dev_get(dev);
1023 	if (idev) {
1024 		for_each_mc_rcu(idev, mc) {
1025 			if (ipv6_addr_equal(&mc->mca_addr, group))
1026 				break;
1027 		}
1028 		if (mc) {
1029 			if (src_addr && !ipv6_addr_any(src_addr)) {
1030 				struct ip6_sf_list *psf;
1031 
1032 				for_each_psf_rcu(mc, psf) {
1033 					if (ipv6_addr_equal(&psf->sf_addr, src_addr))
1034 						break;
1035 				}
1036 				if (psf)
1037 					rv = psf->sf_count[MCAST_INCLUDE] ||
1038 						psf->sf_count[MCAST_EXCLUDE] !=
1039 						mc->mca_sfcount[MCAST_EXCLUDE];
1040 				else
1041 					rv = mc->mca_sfcount[MCAST_EXCLUDE] != 0;
1042 			} else
1043 				rv = true; /* don't filter unspecified source */
1044 		}
1045 	}
1046 	rcu_read_unlock();
1047 	return rv;
1048 }
1049 
1050 /* called with mc_lock */
mld_gq_start_work(struct inet6_dev * idev)1051 static void mld_gq_start_work(struct inet6_dev *idev)
1052 {
1053 	unsigned long tv = get_random_u32_below(idev->mc_maxdelay);
1054 
1055 	idev->mc_gq_running = 1;
1056 	if (!mod_delayed_work(mld_wq, &idev->mc_gq_work, tv + 2))
1057 		in6_dev_hold(idev);
1058 }
1059 
1060 /* called with mc_lock */
mld_gq_stop_work(struct inet6_dev * idev)1061 static void mld_gq_stop_work(struct inet6_dev *idev)
1062 {
1063 	idev->mc_gq_running = 0;
1064 	if (cancel_delayed_work(&idev->mc_gq_work))
1065 		__in6_dev_put(idev);
1066 }
1067 
1068 /* called with mc_lock */
mld_ifc_start_work(struct inet6_dev * idev,unsigned long delay)1069 static void mld_ifc_start_work(struct inet6_dev *idev, unsigned long delay)
1070 {
1071 	unsigned long tv = get_random_u32_below(delay);
1072 
1073 	if (!mod_delayed_work(mld_wq, &idev->mc_ifc_work, tv + 2))
1074 		in6_dev_hold(idev);
1075 }
1076 
1077 /* called with mc_lock */
mld_ifc_stop_work(struct inet6_dev * idev)1078 static void mld_ifc_stop_work(struct inet6_dev *idev)
1079 {
1080 	idev->mc_ifc_count = 0;
1081 	if (cancel_delayed_work(&idev->mc_ifc_work))
1082 		__in6_dev_put(idev);
1083 }
1084 
1085 /* called with mc_lock */
mld_dad_start_work(struct inet6_dev * idev,unsigned long delay)1086 static void mld_dad_start_work(struct inet6_dev *idev, unsigned long delay)
1087 {
1088 	unsigned long tv = get_random_u32_below(delay);
1089 
1090 	if (!mod_delayed_work(mld_wq, &idev->mc_dad_work, tv + 2))
1091 		in6_dev_hold(idev);
1092 }
1093 
mld_dad_stop_work(struct inet6_dev * idev)1094 static void mld_dad_stop_work(struct inet6_dev *idev)
1095 {
1096 	if (cancel_delayed_work(&idev->mc_dad_work))
1097 		__in6_dev_put(idev);
1098 }
1099 
mld_query_stop_work(struct inet6_dev * idev)1100 static void mld_query_stop_work(struct inet6_dev *idev)
1101 {
1102 	spin_lock_bh(&idev->mc_query_lock);
1103 	if (cancel_delayed_work(&idev->mc_query_work))
1104 		__in6_dev_put(idev);
1105 	spin_unlock_bh(&idev->mc_query_lock);
1106 }
1107 
mld_report_stop_work(struct inet6_dev * idev)1108 static void mld_report_stop_work(struct inet6_dev *idev)
1109 {
1110 	if (cancel_delayed_work_sync(&idev->mc_report_work))
1111 		__in6_dev_put(idev);
1112 }
1113 
1114 /*
1115  * IGMP handling (alias multicast ICMPv6 messages)
1116  * called with mc_lock
1117  */
igmp6_group_queried(struct ifmcaddr6 * ma,unsigned long resptime)1118 static void igmp6_group_queried(struct ifmcaddr6 *ma, unsigned long resptime)
1119 {
1120 	unsigned long delay = resptime;
1121 
1122 	/* Do not start work for these addresses */
1123 	if (ipv6_addr_is_ll_all_nodes(&ma->mca_addr) ||
1124 	    IPV6_ADDR_MC_SCOPE(&ma->mca_addr) < IPV6_ADDR_SCOPE_LINKLOCAL)
1125 		return;
1126 
1127 	if (cancel_delayed_work(&ma->mca_work)) {
1128 		refcount_dec(&ma->mca_refcnt);
1129 		delay = ma->mca_work.timer.expires - jiffies;
1130 	}
1131 
1132 	if (delay >= resptime)
1133 		delay = get_random_u32_below(resptime);
1134 
1135 	if (!mod_delayed_work(mld_wq, &ma->mca_work, delay))
1136 		refcount_inc(&ma->mca_refcnt);
1137 	ma->mca_flags |= MAF_TIMER_RUNNING;
1138 }
1139 
1140 /* mark EXCLUDE-mode sources
1141  * called with mc_lock
1142  */
mld_xmarksources(struct ifmcaddr6 * pmc,int nsrcs,const struct in6_addr * srcs)1143 static bool mld_xmarksources(struct ifmcaddr6 *pmc, int nsrcs,
1144 			     const struct in6_addr *srcs)
1145 {
1146 	struct ip6_sf_list *psf;
1147 	int i, scount;
1148 
1149 	scount = 0;
1150 	for_each_psf_mclock(pmc, psf) {
1151 		if (scount == nsrcs)
1152 			break;
1153 		for (i = 0; i < nsrcs; i++) {
1154 			/* skip inactive filters */
1155 			if (psf->sf_count[MCAST_INCLUDE] ||
1156 			    pmc->mca_sfcount[MCAST_EXCLUDE] !=
1157 			    psf->sf_count[MCAST_EXCLUDE])
1158 				break;
1159 			if (ipv6_addr_equal(&srcs[i], &psf->sf_addr)) {
1160 				scount++;
1161 				break;
1162 			}
1163 		}
1164 	}
1165 	pmc->mca_flags &= ~MAF_GSQUERY;
1166 	if (scount == nsrcs)	/* all sources excluded */
1167 		return false;
1168 	return true;
1169 }
1170 
1171 /* called with mc_lock */
mld_marksources(struct ifmcaddr6 * pmc,int nsrcs,const struct in6_addr * srcs)1172 static bool mld_marksources(struct ifmcaddr6 *pmc, int nsrcs,
1173 			    const struct in6_addr *srcs)
1174 {
1175 	struct ip6_sf_list *psf;
1176 	int i, scount;
1177 
1178 	if (pmc->mca_sfmode == MCAST_EXCLUDE)
1179 		return mld_xmarksources(pmc, nsrcs, srcs);
1180 
1181 	/* mark INCLUDE-mode sources */
1182 
1183 	scount = 0;
1184 	for_each_psf_mclock(pmc, psf) {
1185 		if (scount == nsrcs)
1186 			break;
1187 		for (i = 0; i < nsrcs; i++) {
1188 			if (ipv6_addr_equal(&srcs[i], &psf->sf_addr)) {
1189 				psf->sf_gsresp = 1;
1190 				scount++;
1191 				break;
1192 			}
1193 		}
1194 	}
1195 	if (!scount) {
1196 		pmc->mca_flags &= ~MAF_GSQUERY;
1197 		return false;
1198 	}
1199 	pmc->mca_flags |= MAF_GSQUERY;
1200 	return true;
1201 }
1202 
mld_force_mld_version(const struct inet6_dev * idev)1203 static int mld_force_mld_version(const struct inet6_dev *idev)
1204 {
1205 	/* Normally, both are 0 here. If enforcement to a particular is
1206 	 * being used, individual device enforcement will have a lower
1207 	 * precedence over 'all' device (.../conf/all/force_mld_version).
1208 	 */
1209 
1210 	if (dev_net(idev->dev)->ipv6.devconf_all->force_mld_version != 0)
1211 		return dev_net(idev->dev)->ipv6.devconf_all->force_mld_version;
1212 	else
1213 		return idev->cnf.force_mld_version;
1214 }
1215 
mld_in_v2_mode_only(const struct inet6_dev * idev)1216 static bool mld_in_v2_mode_only(const struct inet6_dev *idev)
1217 {
1218 	return mld_force_mld_version(idev) == 2;
1219 }
1220 
mld_in_v1_mode_only(const struct inet6_dev * idev)1221 static bool mld_in_v1_mode_only(const struct inet6_dev *idev)
1222 {
1223 	return mld_force_mld_version(idev) == 1;
1224 }
1225 
mld_in_v1_mode(const struct inet6_dev * idev)1226 static bool mld_in_v1_mode(const struct inet6_dev *idev)
1227 {
1228 	if (mld_in_v2_mode_only(idev))
1229 		return false;
1230 	if (mld_in_v1_mode_only(idev))
1231 		return true;
1232 	if (idev->mc_v1_seen && time_before(jiffies, idev->mc_v1_seen))
1233 		return true;
1234 
1235 	return false;
1236 }
1237 
mld_set_v1_mode(struct inet6_dev * idev)1238 static void mld_set_v1_mode(struct inet6_dev *idev)
1239 {
1240 	/* RFC3810, relevant sections:
1241 	 *  - 9.1. Robustness Variable
1242 	 *  - 9.2. Query Interval
1243 	 *  - 9.3. Query Response Interval
1244 	 *  - 9.12. Older Version Querier Present Timeout
1245 	 */
1246 	unsigned long switchback;
1247 
1248 	switchback = (idev->mc_qrv * idev->mc_qi) + idev->mc_qri;
1249 
1250 	idev->mc_v1_seen = jiffies + switchback;
1251 }
1252 
mld_update_qrv(struct inet6_dev * idev,const struct mld2_query * mlh2)1253 static void mld_update_qrv(struct inet6_dev *idev,
1254 			   const struct mld2_query *mlh2)
1255 {
1256 	/* RFC3810, relevant sections:
1257 	 *  - 5.1.8. QRV (Querier's Robustness Variable)
1258 	 *  - 9.1. Robustness Variable
1259 	 */
1260 
1261 	/* The value of the Robustness Variable MUST NOT be zero,
1262 	 * and SHOULD NOT be one. Catch this here if we ever run
1263 	 * into such a case in future.
1264 	 */
1265 	const int min_qrv = min(MLD_QRV_DEFAULT, sysctl_mld_qrv);
1266 	WARN_ON(idev->mc_qrv == 0);
1267 
1268 	if (mlh2->mld2q_qrv > 0)
1269 		idev->mc_qrv = mlh2->mld2q_qrv;
1270 
1271 	if (unlikely(idev->mc_qrv < min_qrv)) {
1272 		net_warn_ratelimited("IPv6: MLD: clamping QRV from %u to %u!\n",
1273 				     idev->mc_qrv, min_qrv);
1274 		idev->mc_qrv = min_qrv;
1275 	}
1276 }
1277 
mld_update_qi(struct inet6_dev * idev,const struct mld2_query * mlh2)1278 static void mld_update_qi(struct inet6_dev *idev,
1279 			  const struct mld2_query *mlh2)
1280 {
1281 	/* RFC3810, relevant sections:
1282 	 *  - 5.1.9. QQIC (Querier's Query Interval Code)
1283 	 *  - 9.2. Query Interval
1284 	 *  - 9.12. Older Version Querier Present Timeout
1285 	 *    (the [Query Interval] in the last Query received)
1286 	 */
1287 	unsigned long mc_qqi;
1288 
1289 	if (mlh2->mld2q_qqic < 128) {
1290 		mc_qqi = mlh2->mld2q_qqic;
1291 	} else {
1292 		unsigned long mc_man, mc_exp;
1293 
1294 		mc_exp = MLDV2_QQIC_EXP(mlh2->mld2q_qqic);
1295 		mc_man = MLDV2_QQIC_MAN(mlh2->mld2q_qqic);
1296 
1297 		mc_qqi = (mc_man | 0x10) << (mc_exp + 3);
1298 	}
1299 
1300 	idev->mc_qi = mc_qqi * HZ;
1301 }
1302 
mld_update_qri(struct inet6_dev * idev,const struct mld2_query * mlh2)1303 static void mld_update_qri(struct inet6_dev *idev,
1304 			   const struct mld2_query *mlh2)
1305 {
1306 	/* RFC3810, relevant sections:
1307 	 *  - 5.1.3. Maximum Response Code
1308 	 *  - 9.3. Query Response Interval
1309 	 */
1310 	idev->mc_qri = msecs_to_jiffies(mldv2_mrc(mlh2));
1311 }
1312 
mld_process_v1(struct inet6_dev * idev,struct mld_msg * mld,unsigned long * max_delay,bool v1_query)1313 static int mld_process_v1(struct inet6_dev *idev, struct mld_msg *mld,
1314 			  unsigned long *max_delay, bool v1_query)
1315 {
1316 	unsigned long mldv1_md;
1317 
1318 	/* Ignore v1 queries */
1319 	if (mld_in_v2_mode_only(idev))
1320 		return -EINVAL;
1321 
1322 	mldv1_md = ntohs(mld->mld_maxdelay);
1323 
1324 	/* When in MLDv1 fallback and a MLDv2 router start-up being
1325 	 * unaware of current MLDv1 operation, the MRC == MRD mapping
1326 	 * only works when the exponential algorithm is not being
1327 	 * used (as MLDv1 is unaware of such things).
1328 	 *
1329 	 * According to the RFC author, the MLDv2 implementations
1330 	 * he's aware of all use a MRC < 32768 on start up queries.
1331 	 *
1332 	 * Thus, should we *ever* encounter something else larger
1333 	 * than that, just assume the maximum possible within our
1334 	 * reach.
1335 	 */
1336 	if (!v1_query)
1337 		mldv1_md = min(mldv1_md, MLDV1_MRD_MAX_COMPAT);
1338 
1339 	*max_delay = max(msecs_to_jiffies(mldv1_md), 1UL);
1340 
1341 	/* MLDv1 router present: we need to go into v1 mode *only*
1342 	 * when an MLDv1 query is received as per section 9.12. of
1343 	 * RFC3810! And we know from RFC2710 section 3.7 that MLDv1
1344 	 * queries MUST be of exactly 24 octets.
1345 	 */
1346 	if (v1_query)
1347 		mld_set_v1_mode(idev);
1348 
1349 	/* cancel MLDv2 report work */
1350 	mld_gq_stop_work(idev);
1351 	/* cancel the interface change work */
1352 	mld_ifc_stop_work(idev);
1353 	/* clear deleted report items */
1354 	mld_clear_delrec(idev);
1355 
1356 	return 0;
1357 }
1358 
mld_process_v2(struct inet6_dev * idev,struct mld2_query * mld,unsigned long * max_delay)1359 static void mld_process_v2(struct inet6_dev *idev, struct mld2_query *mld,
1360 			   unsigned long *max_delay)
1361 {
1362 	*max_delay = max(msecs_to_jiffies(mldv2_mrc(mld)), 1UL);
1363 
1364 	mld_update_qrv(idev, mld);
1365 	mld_update_qi(idev, mld);
1366 	mld_update_qri(idev, mld);
1367 
1368 	idev->mc_maxdelay = *max_delay;
1369 
1370 	return;
1371 }
1372 
1373 /* called with rcu_read_lock() */
igmp6_event_query(struct sk_buff * skb)1374 void igmp6_event_query(struct sk_buff *skb)
1375 {
1376 	struct inet6_dev *idev = __in6_dev_get(skb->dev);
1377 
1378 	if (!idev || idev->dead)
1379 		goto out;
1380 
1381 	spin_lock_bh(&idev->mc_query_lock);
1382 	if (skb_queue_len(&idev->mc_query_queue) < MLD_MAX_SKBS) {
1383 		__skb_queue_tail(&idev->mc_query_queue, skb);
1384 		if (!mod_delayed_work(mld_wq, &idev->mc_query_work, 0))
1385 			in6_dev_hold(idev);
1386 		skb = NULL;
1387 	}
1388 	spin_unlock_bh(&idev->mc_query_lock);
1389 out:
1390 	kfree_skb(skb);
1391 }
1392 
__mld_query_work(struct sk_buff * skb)1393 static void __mld_query_work(struct sk_buff *skb)
1394 {
1395 	struct mld2_query *mlh2 = NULL;
1396 	const struct in6_addr *group;
1397 	unsigned long max_delay;
1398 	struct inet6_dev *idev;
1399 	struct ifmcaddr6 *ma;
1400 	struct mld_msg *mld;
1401 	int group_type;
1402 	int mark = 0;
1403 	int len, err;
1404 
1405 	if (!pskb_may_pull(skb, sizeof(struct in6_addr)))
1406 		goto kfree_skb;
1407 
1408 	/* compute payload length excluding extension headers */
1409 	len = ntohs(ipv6_hdr(skb)->payload_len) + sizeof(struct ipv6hdr);
1410 	len -= skb_network_header_len(skb);
1411 
1412 	/* RFC3810 6.2
1413 	 * Upon reception of an MLD message that contains a Query, the node
1414 	 * checks if the source address of the message is a valid link-local
1415 	 * address, if the Hop Limit is set to 1, and if the Router Alert
1416 	 * option is present in the Hop-By-Hop Options header of the IPv6
1417 	 * packet.  If any of these checks fails, the packet is dropped.
1418 	 */
1419 	if (!(ipv6_addr_type(&ipv6_hdr(skb)->saddr) & IPV6_ADDR_LINKLOCAL) ||
1420 	    ipv6_hdr(skb)->hop_limit != 1 ||
1421 	    !(IP6CB(skb)->flags & IP6SKB_ROUTERALERT) ||
1422 	    IP6CB(skb)->ra != htons(IPV6_OPT_ROUTERALERT_MLD))
1423 		goto kfree_skb;
1424 
1425 	idev = in6_dev_get(skb->dev);
1426 	if (!idev)
1427 		goto kfree_skb;
1428 
1429 	mld = (struct mld_msg *)icmp6_hdr(skb);
1430 	group = &mld->mld_mca;
1431 	group_type = ipv6_addr_type(group);
1432 
1433 	if (group_type != IPV6_ADDR_ANY &&
1434 	    !(group_type&IPV6_ADDR_MULTICAST))
1435 		goto out;
1436 
1437 	if (len < MLD_V1_QUERY_LEN) {
1438 		goto out;
1439 	} else if (len == MLD_V1_QUERY_LEN || mld_in_v1_mode(idev)) {
1440 		err = mld_process_v1(idev, mld, &max_delay,
1441 				     len == MLD_V1_QUERY_LEN);
1442 		if (err < 0)
1443 			goto out;
1444 	} else if (len >= MLD_V2_QUERY_LEN_MIN) {
1445 		int srcs_offset = sizeof(struct mld2_query) -
1446 				  sizeof(struct icmp6hdr);
1447 
1448 		if (!pskb_may_pull(skb, srcs_offset))
1449 			goto out;
1450 
1451 		mlh2 = (struct mld2_query *)skb_transport_header(skb);
1452 
1453 		mld_process_v2(idev, mlh2, &max_delay);
1454 
1455 		if (group_type == IPV6_ADDR_ANY) { /* general query */
1456 			if (mlh2->mld2q_nsrcs)
1457 				goto out; /* no sources allowed */
1458 
1459 			mld_gq_start_work(idev);
1460 			goto out;
1461 		}
1462 		/* mark sources to include, if group & source-specific */
1463 		if (mlh2->mld2q_nsrcs != 0) {
1464 			if (!pskb_may_pull(skb, srcs_offset +
1465 			    ntohs(mlh2->mld2q_nsrcs) * sizeof(struct in6_addr)))
1466 				goto out;
1467 
1468 			mlh2 = (struct mld2_query *)skb_transport_header(skb);
1469 			mark = 1;
1470 		}
1471 	} else {
1472 		goto out;
1473 	}
1474 
1475 	if (group_type == IPV6_ADDR_ANY) {
1476 		for_each_mc_mclock(idev, ma) {
1477 			igmp6_group_queried(ma, max_delay);
1478 		}
1479 	} else {
1480 		for_each_mc_mclock(idev, ma) {
1481 			if (!ipv6_addr_equal(group, &ma->mca_addr))
1482 				continue;
1483 			if (ma->mca_flags & MAF_TIMER_RUNNING) {
1484 				/* gsquery <- gsquery && mark */
1485 				if (!mark)
1486 					ma->mca_flags &= ~MAF_GSQUERY;
1487 			} else {
1488 				/* gsquery <- mark */
1489 				if (mark)
1490 					ma->mca_flags |= MAF_GSQUERY;
1491 				else
1492 					ma->mca_flags &= ~MAF_GSQUERY;
1493 			}
1494 			if (!(ma->mca_flags & MAF_GSQUERY) ||
1495 			    mld_marksources(ma, ntohs(mlh2->mld2q_nsrcs), mlh2->mld2q_srcs))
1496 				igmp6_group_queried(ma, max_delay);
1497 			break;
1498 		}
1499 	}
1500 
1501 out:
1502 	in6_dev_put(idev);
1503 kfree_skb:
1504 	consume_skb(skb);
1505 }
1506 
mld_query_work(struct work_struct * work)1507 static void mld_query_work(struct work_struct *work)
1508 {
1509 	struct inet6_dev *idev = container_of(to_delayed_work(work),
1510 					      struct inet6_dev,
1511 					      mc_query_work);
1512 	struct sk_buff_head q;
1513 	struct sk_buff *skb;
1514 	bool rework = false;
1515 	int cnt = 0;
1516 
1517 	skb_queue_head_init(&q);
1518 
1519 	spin_lock_bh(&idev->mc_query_lock);
1520 	while ((skb = __skb_dequeue(&idev->mc_query_queue))) {
1521 		__skb_queue_tail(&q, skb);
1522 
1523 		if (++cnt >= MLD_MAX_QUEUE) {
1524 			rework = true;
1525 			break;
1526 		}
1527 	}
1528 	spin_unlock_bh(&idev->mc_query_lock);
1529 
1530 	mutex_lock(&idev->mc_lock);
1531 	while ((skb = __skb_dequeue(&q)))
1532 		__mld_query_work(skb);
1533 	mutex_unlock(&idev->mc_lock);
1534 
1535 	if (rework && queue_delayed_work(mld_wq, &idev->mc_query_work, 0))
1536 		return;
1537 
1538 	in6_dev_put(idev);
1539 }
1540 
1541 /* called with rcu_read_lock() */
igmp6_event_report(struct sk_buff * skb)1542 void igmp6_event_report(struct sk_buff *skb)
1543 {
1544 	struct inet6_dev *idev = __in6_dev_get(skb->dev);
1545 
1546 	if (!idev || idev->dead)
1547 		goto out;
1548 
1549 	spin_lock_bh(&idev->mc_report_lock);
1550 	if (skb_queue_len(&idev->mc_report_queue) < MLD_MAX_SKBS) {
1551 		__skb_queue_tail(&idev->mc_report_queue, skb);
1552 		if (!mod_delayed_work(mld_wq, &idev->mc_report_work, 0))
1553 			in6_dev_hold(idev);
1554 		skb = NULL;
1555 	}
1556 	spin_unlock_bh(&idev->mc_report_lock);
1557 out:
1558 	kfree_skb(skb);
1559 }
1560 
__mld_report_work(struct sk_buff * skb)1561 static void __mld_report_work(struct sk_buff *skb)
1562 {
1563 	struct inet6_dev *idev;
1564 	struct ifmcaddr6 *ma;
1565 	struct mld_msg *mld;
1566 	int addr_type;
1567 
1568 	/* Our own report looped back. Ignore it. */
1569 	if (skb->pkt_type == PACKET_LOOPBACK)
1570 		goto kfree_skb;
1571 
1572 	/* send our report if the MC router may not have heard this report */
1573 	if (skb->pkt_type != PACKET_MULTICAST &&
1574 	    skb->pkt_type != PACKET_BROADCAST)
1575 		goto kfree_skb;
1576 
1577 	if (!pskb_may_pull(skb, sizeof(*mld) - sizeof(struct icmp6hdr)))
1578 		goto kfree_skb;
1579 
1580 	mld = (struct mld_msg *)icmp6_hdr(skb);
1581 
1582 	/* Drop reports with not link local source */
1583 	addr_type = ipv6_addr_type(&ipv6_hdr(skb)->saddr);
1584 	if (addr_type != IPV6_ADDR_ANY &&
1585 	    !(addr_type&IPV6_ADDR_LINKLOCAL))
1586 		goto kfree_skb;
1587 
1588 	idev = in6_dev_get(skb->dev);
1589 	if (!idev)
1590 		goto kfree_skb;
1591 
1592 	/*
1593 	 *	Cancel the work for this group
1594 	 */
1595 
1596 	for_each_mc_mclock(idev, ma) {
1597 		if (ipv6_addr_equal(&ma->mca_addr, &mld->mld_mca)) {
1598 			if (cancel_delayed_work(&ma->mca_work))
1599 				refcount_dec(&ma->mca_refcnt);
1600 			ma->mca_flags &= ~(MAF_LAST_REPORTER |
1601 					   MAF_TIMER_RUNNING);
1602 			break;
1603 		}
1604 	}
1605 
1606 	in6_dev_put(idev);
1607 kfree_skb:
1608 	consume_skb(skb);
1609 }
1610 
mld_report_work(struct work_struct * work)1611 static void mld_report_work(struct work_struct *work)
1612 {
1613 	struct inet6_dev *idev = container_of(to_delayed_work(work),
1614 					      struct inet6_dev,
1615 					      mc_report_work);
1616 	struct sk_buff_head q;
1617 	struct sk_buff *skb;
1618 	bool rework = false;
1619 	int cnt = 0;
1620 
1621 	skb_queue_head_init(&q);
1622 	spin_lock_bh(&idev->mc_report_lock);
1623 	while ((skb = __skb_dequeue(&idev->mc_report_queue))) {
1624 		__skb_queue_tail(&q, skb);
1625 
1626 		if (++cnt >= MLD_MAX_QUEUE) {
1627 			rework = true;
1628 			break;
1629 		}
1630 	}
1631 	spin_unlock_bh(&idev->mc_report_lock);
1632 
1633 	mutex_lock(&idev->mc_lock);
1634 	while ((skb = __skb_dequeue(&q)))
1635 		__mld_report_work(skb);
1636 	mutex_unlock(&idev->mc_lock);
1637 
1638 	if (rework && queue_delayed_work(mld_wq, &idev->mc_report_work, 0))
1639 		return;
1640 
1641 	in6_dev_put(idev);
1642 }
1643 
is_in(struct ifmcaddr6 * pmc,struct ip6_sf_list * psf,int type,int gdeleted,int sdeleted)1644 static bool is_in(struct ifmcaddr6 *pmc, struct ip6_sf_list *psf, int type,
1645 		  int gdeleted, int sdeleted)
1646 {
1647 	switch (type) {
1648 	case MLD2_MODE_IS_INCLUDE:
1649 	case MLD2_MODE_IS_EXCLUDE:
1650 		if (gdeleted || sdeleted)
1651 			return false;
1652 		if (!((pmc->mca_flags & MAF_GSQUERY) && !psf->sf_gsresp)) {
1653 			if (pmc->mca_sfmode == MCAST_INCLUDE)
1654 				return true;
1655 			/* don't include if this source is excluded
1656 			 * in all filters
1657 			 */
1658 			if (psf->sf_count[MCAST_INCLUDE])
1659 				return type == MLD2_MODE_IS_INCLUDE;
1660 			return pmc->mca_sfcount[MCAST_EXCLUDE] ==
1661 				psf->sf_count[MCAST_EXCLUDE];
1662 		}
1663 		return false;
1664 	case MLD2_CHANGE_TO_INCLUDE:
1665 		if (gdeleted || sdeleted)
1666 			return false;
1667 		return psf->sf_count[MCAST_INCLUDE] != 0;
1668 	case MLD2_CHANGE_TO_EXCLUDE:
1669 		if (gdeleted || sdeleted)
1670 			return false;
1671 		if (pmc->mca_sfcount[MCAST_EXCLUDE] == 0 ||
1672 		    psf->sf_count[MCAST_INCLUDE])
1673 			return false;
1674 		return pmc->mca_sfcount[MCAST_EXCLUDE] ==
1675 			psf->sf_count[MCAST_EXCLUDE];
1676 	case MLD2_ALLOW_NEW_SOURCES:
1677 		if (gdeleted || !psf->sf_crcount)
1678 			return false;
1679 		return (pmc->mca_sfmode == MCAST_INCLUDE) ^ sdeleted;
1680 	case MLD2_BLOCK_OLD_SOURCES:
1681 		if (pmc->mca_sfmode == MCAST_INCLUDE)
1682 			return gdeleted || (psf->sf_crcount && sdeleted);
1683 		return psf->sf_crcount && !gdeleted && !sdeleted;
1684 	}
1685 	return false;
1686 }
1687 
1688 static int
mld_scount(struct ifmcaddr6 * pmc,int type,int gdeleted,int sdeleted)1689 mld_scount(struct ifmcaddr6 *pmc, int type, int gdeleted, int sdeleted)
1690 {
1691 	struct ip6_sf_list *psf;
1692 	int scount = 0;
1693 
1694 	for_each_psf_mclock(pmc, psf) {
1695 		if (!is_in(pmc, psf, type, gdeleted, sdeleted))
1696 			continue;
1697 		scount++;
1698 	}
1699 	return scount;
1700 }
1701 
ip6_mc_hdr(const struct sock * sk,struct sk_buff * skb,struct net_device * dev,const struct in6_addr * saddr,const struct in6_addr * daddr,int proto,int len)1702 static void ip6_mc_hdr(const struct sock *sk, struct sk_buff *skb,
1703 		       struct net_device *dev, const struct in6_addr *saddr,
1704 		       const struct in6_addr *daddr, int proto, int len)
1705 {
1706 	struct ipv6hdr *hdr;
1707 
1708 	skb->protocol = htons(ETH_P_IPV6);
1709 	skb->dev = dev;
1710 
1711 	skb_reset_network_header(skb);
1712 	skb_put(skb, sizeof(struct ipv6hdr));
1713 	hdr = ipv6_hdr(skb);
1714 
1715 	ip6_flow_hdr(hdr, 0, 0);
1716 
1717 	hdr->payload_len = htons(len);
1718 	hdr->nexthdr = proto;
1719 	hdr->hop_limit = inet6_sk(sk)->hop_limit;
1720 
1721 	hdr->saddr = *saddr;
1722 	hdr->daddr = *daddr;
1723 }
1724 
mld_newpack(struct inet6_dev * idev,unsigned int mtu)1725 static struct sk_buff *mld_newpack(struct inet6_dev *idev, unsigned int mtu)
1726 {
1727 	u8 ra[8] = { IPPROTO_ICMPV6, 0, IPV6_TLV_ROUTERALERT,
1728 		     2, 0, 0, IPV6_TLV_PADN, 0 };
1729 	struct net_device *dev = idev->dev;
1730 	int hlen = LL_RESERVED_SPACE(dev);
1731 	int tlen = dev->needed_tailroom;
1732 	const struct in6_addr *saddr;
1733 	struct in6_addr addr_buf;
1734 	struct mld2_report *pmr;
1735 	struct sk_buff *skb;
1736 	unsigned int size;
1737 	struct sock *sk;
1738 	struct net *net;
1739 
1740 	/* we assume size > sizeof(ra) here
1741 	 * Also try to not allocate high-order pages for big MTU
1742 	 */
1743 	size = min_t(int, mtu, PAGE_SIZE / 2) + hlen + tlen;
1744 	skb = alloc_skb(size, GFP_KERNEL);
1745 	if (!skb)
1746 		return NULL;
1747 
1748 	skb->priority = TC_PRIO_CONTROL;
1749 	skb_reserve(skb, hlen);
1750 	skb_tailroom_reserve(skb, mtu, tlen);
1751 
1752 	rcu_read_lock();
1753 
1754 	net = dev_net_rcu(dev);
1755 	sk = net->ipv6.igmp_sk;
1756 	skb_set_owner_w(skb, sk);
1757 
1758 	if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
1759 		/* <draft-ietf-magma-mld-source-05.txt>:
1760 		 * use unspecified address as the source address
1761 		 * when a valid link-local address is not available.
1762 		 */
1763 		saddr = &in6addr_any;
1764 	} else
1765 		saddr = &addr_buf;
1766 
1767 	ip6_mc_hdr(sk, skb, dev, saddr, &mld2_all_mcr, NEXTHDR_HOP, 0);
1768 
1769 	rcu_read_unlock();
1770 
1771 	skb_put_data(skb, ra, sizeof(ra));
1772 
1773 	skb_set_transport_header(skb, skb_tail_pointer(skb) - skb->data);
1774 	skb_put(skb, sizeof(*pmr));
1775 	pmr = (struct mld2_report *)skb_transport_header(skb);
1776 	pmr->mld2r_type = ICMPV6_MLD2_REPORT;
1777 	pmr->mld2r_resv1 = 0;
1778 	pmr->mld2r_cksum = 0;
1779 	pmr->mld2r_resv2 = 0;
1780 	pmr->mld2r_ngrec = 0;
1781 	return skb;
1782 }
1783 
mld_sendpack(struct sk_buff * skb)1784 static void mld_sendpack(struct sk_buff *skb)
1785 {
1786 	struct ipv6hdr *pip6 = ipv6_hdr(skb);
1787 	struct mld2_report *pmr =
1788 			      (struct mld2_report *)skb_transport_header(skb);
1789 	int payload_len, mldlen;
1790 	struct inet6_dev *idev;
1791 	struct net *net = dev_net(skb->dev);
1792 	int err;
1793 	struct flowi6 fl6;
1794 	struct dst_entry *dst;
1795 
1796 	rcu_read_lock();
1797 	idev = __in6_dev_get(skb->dev);
1798 	IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTREQUESTS);
1799 
1800 	payload_len = (skb_tail_pointer(skb) - skb_network_header(skb)) -
1801 		sizeof(*pip6);
1802 	mldlen = skb_tail_pointer(skb) - skb_transport_header(skb);
1803 	pip6->payload_len = htons(payload_len);
1804 
1805 	pmr->mld2r_cksum = csum_ipv6_magic(&pip6->saddr, &pip6->daddr, mldlen,
1806 					   IPPROTO_ICMPV6,
1807 					   csum_partial(skb_transport_header(skb),
1808 							mldlen, 0));
1809 
1810 	icmpv6_flow_init(net->ipv6.igmp_sk, &fl6, ICMPV6_MLD2_REPORT,
1811 			 &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
1812 			 skb->dev->ifindex);
1813 	dst = icmp6_dst_alloc(skb->dev, &fl6);
1814 
1815 	err = 0;
1816 	if (IS_ERR(dst)) {
1817 		err = PTR_ERR(dst);
1818 		dst = NULL;
1819 	}
1820 	skb_dst_set(skb, dst);
1821 	if (err)
1822 		goto err_out;
1823 
1824 	err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
1825 		      net, net->ipv6.igmp_sk, skb, NULL, skb->dev,
1826 		      dst_output);
1827 out:
1828 	if (!err) {
1829 		ICMP6MSGOUT_INC_STATS(net, idev, ICMPV6_MLD2_REPORT);
1830 		ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
1831 	} else {
1832 		IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
1833 	}
1834 
1835 	rcu_read_unlock();
1836 	return;
1837 
1838 err_out:
1839 	kfree_skb(skb);
1840 	goto out;
1841 }
1842 
grec_size(struct ifmcaddr6 * pmc,int type,int gdel,int sdel)1843 static int grec_size(struct ifmcaddr6 *pmc, int type, int gdel, int sdel)
1844 {
1845 	return sizeof(struct mld2_grec) + 16 * mld_scount(pmc,type,gdel,sdel);
1846 }
1847 
add_grhead(struct sk_buff * skb,struct ifmcaddr6 * pmc,int type,struct mld2_grec ** ppgr,unsigned int mtu)1848 static struct sk_buff *add_grhead(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1849 	int type, struct mld2_grec **ppgr, unsigned int mtu)
1850 {
1851 	struct mld2_report *pmr;
1852 	struct mld2_grec *pgr;
1853 
1854 	if (!skb) {
1855 		skb = mld_newpack(pmc->idev, mtu);
1856 		if (!skb)
1857 			return NULL;
1858 	}
1859 	pgr = skb_put(skb, sizeof(struct mld2_grec));
1860 	pgr->grec_type = type;
1861 	pgr->grec_auxwords = 0;
1862 	pgr->grec_nsrcs = 0;
1863 	pgr->grec_mca = pmc->mca_addr;	/* structure copy */
1864 	pmr = (struct mld2_report *)skb_transport_header(skb);
1865 	pmr->mld2r_ngrec = htons(ntohs(pmr->mld2r_ngrec)+1);
1866 	*ppgr = pgr;
1867 	return skb;
1868 }
1869 
1870 #define AVAILABLE(skb)	((skb) ? skb_availroom(skb) : 0)
1871 
1872 /* called with mc_lock */
add_grec(struct sk_buff * skb,struct ifmcaddr6 * pmc,int type,int gdeleted,int sdeleted,int crsend)1873 static struct sk_buff *add_grec(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1874 				int type, int gdeleted, int sdeleted,
1875 				int crsend)
1876 {
1877 	struct ip6_sf_list *psf, *psf_prev, *psf_next;
1878 	int scount, stotal, first, isquery, truncate;
1879 	struct ip6_sf_list __rcu **psf_list;
1880 	struct inet6_dev *idev = pmc->idev;
1881 	struct net_device *dev = idev->dev;
1882 	struct mld2_grec *pgr = NULL;
1883 	struct mld2_report *pmr;
1884 	unsigned int mtu;
1885 
1886 	if (pmc->mca_flags & MAF_NOREPORT)
1887 		return skb;
1888 
1889 	mtu = READ_ONCE(dev->mtu);
1890 	if (mtu < IPV6_MIN_MTU)
1891 		return skb;
1892 
1893 	isquery = type == MLD2_MODE_IS_INCLUDE ||
1894 		  type == MLD2_MODE_IS_EXCLUDE;
1895 	truncate = type == MLD2_MODE_IS_EXCLUDE ||
1896 		    type == MLD2_CHANGE_TO_EXCLUDE;
1897 
1898 	stotal = scount = 0;
1899 
1900 	psf_list = sdeleted ? &pmc->mca_tomb : &pmc->mca_sources;
1901 
1902 	if (!rcu_access_pointer(*psf_list))
1903 		goto empty_source;
1904 
1905 	pmr = skb ? (struct mld2_report *)skb_transport_header(skb) : NULL;
1906 
1907 	/* EX and TO_EX get a fresh packet, if needed */
1908 	if (truncate) {
1909 		if (pmr && pmr->mld2r_ngrec &&
1910 		    AVAILABLE(skb) < grec_size(pmc, type, gdeleted, sdeleted)) {
1911 			if (skb)
1912 				mld_sendpack(skb);
1913 			skb = mld_newpack(idev, mtu);
1914 		}
1915 	}
1916 	first = 1;
1917 	psf_prev = NULL;
1918 	for (psf = mc_dereference(*psf_list, idev);
1919 	     psf;
1920 	     psf = psf_next) {
1921 		struct in6_addr *psrc;
1922 
1923 		psf_next = mc_dereference(psf->sf_next, idev);
1924 
1925 		if (!is_in(pmc, psf, type, gdeleted, sdeleted) && !crsend) {
1926 			psf_prev = psf;
1927 			continue;
1928 		}
1929 
1930 		/* Based on RFC3810 6.1. Should not send source-list change
1931 		 * records when there is a filter mode change.
1932 		 */
1933 		if (((gdeleted && pmc->mca_sfmode == MCAST_EXCLUDE) ||
1934 		     (!gdeleted && pmc->mca_crcount)) &&
1935 		    (type == MLD2_ALLOW_NEW_SOURCES ||
1936 		     type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount)
1937 			goto decrease_sf_crcount;
1938 
1939 		/* clear marks on query responses */
1940 		if (isquery)
1941 			psf->sf_gsresp = 0;
1942 
1943 		if (AVAILABLE(skb) < sizeof(*psrc) +
1944 		    first*sizeof(struct mld2_grec)) {
1945 			if (truncate && !first)
1946 				break;	 /* truncate these */
1947 			if (pgr)
1948 				pgr->grec_nsrcs = htons(scount);
1949 			if (skb)
1950 				mld_sendpack(skb);
1951 			skb = mld_newpack(idev, mtu);
1952 			first = 1;
1953 			scount = 0;
1954 		}
1955 		if (first) {
1956 			skb = add_grhead(skb, pmc, type, &pgr, mtu);
1957 			first = 0;
1958 		}
1959 		if (!skb)
1960 			return NULL;
1961 		psrc = skb_put(skb, sizeof(*psrc));
1962 		*psrc = psf->sf_addr;
1963 		scount++; stotal++;
1964 		if ((type == MLD2_ALLOW_NEW_SOURCES ||
1965 		     type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount) {
1966 decrease_sf_crcount:
1967 			psf->sf_crcount--;
1968 			if ((sdeleted || gdeleted) && psf->sf_crcount == 0) {
1969 				if (psf_prev)
1970 					rcu_assign_pointer(psf_prev->sf_next,
1971 							   mc_dereference(psf->sf_next, idev));
1972 				else
1973 					rcu_assign_pointer(*psf_list,
1974 							   mc_dereference(psf->sf_next, idev));
1975 				kfree_rcu(psf, rcu);
1976 				continue;
1977 			}
1978 		}
1979 		psf_prev = psf;
1980 	}
1981 
1982 empty_source:
1983 	if (!stotal) {
1984 		if (type == MLD2_ALLOW_NEW_SOURCES ||
1985 		    type == MLD2_BLOCK_OLD_SOURCES)
1986 			return skb;
1987 		if (pmc->mca_crcount || isquery || crsend) {
1988 			/* make sure we have room for group header */
1989 			if (skb && AVAILABLE(skb) < sizeof(struct mld2_grec)) {
1990 				mld_sendpack(skb);
1991 				skb = NULL; /* add_grhead will get a new one */
1992 			}
1993 			skb = add_grhead(skb, pmc, type, &pgr, mtu);
1994 		}
1995 	}
1996 	if (pgr)
1997 		pgr->grec_nsrcs = htons(scount);
1998 
1999 	if (isquery)
2000 		pmc->mca_flags &= ~MAF_GSQUERY;	/* clear query state */
2001 	return skb;
2002 }
2003 
2004 /* called with mc_lock */
mld_send_report(struct inet6_dev * idev,struct ifmcaddr6 * pmc)2005 static void mld_send_report(struct inet6_dev *idev, struct ifmcaddr6 *pmc)
2006 {
2007 	struct sk_buff *skb = NULL;
2008 	int type;
2009 
2010 	if (!pmc) {
2011 		for_each_mc_mclock(idev, pmc) {
2012 			if (pmc->mca_flags & MAF_NOREPORT)
2013 				continue;
2014 			if (pmc->mca_sfcount[MCAST_EXCLUDE])
2015 				type = MLD2_MODE_IS_EXCLUDE;
2016 			else
2017 				type = MLD2_MODE_IS_INCLUDE;
2018 			skb = add_grec(skb, pmc, type, 0, 0, 0);
2019 		}
2020 	} else {
2021 		if (pmc->mca_sfcount[MCAST_EXCLUDE])
2022 			type = MLD2_MODE_IS_EXCLUDE;
2023 		else
2024 			type = MLD2_MODE_IS_INCLUDE;
2025 		skb = add_grec(skb, pmc, type, 0, 0, 0);
2026 	}
2027 	if (skb)
2028 		mld_sendpack(skb);
2029 }
2030 
2031 /*
2032  * remove zero-count source records from a source filter list
2033  * called with mc_lock
2034  */
mld_clear_zeros(struct ip6_sf_list __rcu ** ppsf,struct inet6_dev * idev)2035 static void mld_clear_zeros(struct ip6_sf_list __rcu **ppsf, struct inet6_dev *idev)
2036 {
2037 	struct ip6_sf_list *psf_prev, *psf_next, *psf;
2038 
2039 	psf_prev = NULL;
2040 	for (psf = mc_dereference(*ppsf, idev);
2041 	     psf;
2042 	     psf = psf_next) {
2043 		psf_next = mc_dereference(psf->sf_next, idev);
2044 		if (psf->sf_crcount == 0) {
2045 			if (psf_prev)
2046 				rcu_assign_pointer(psf_prev->sf_next,
2047 						   mc_dereference(psf->sf_next, idev));
2048 			else
2049 				rcu_assign_pointer(*ppsf,
2050 						   mc_dereference(psf->sf_next, idev));
2051 			kfree_rcu(psf, rcu);
2052 		} else {
2053 			psf_prev = psf;
2054 		}
2055 	}
2056 }
2057 
2058 /* called with mc_lock */
mld_send_cr(struct inet6_dev * idev)2059 static void mld_send_cr(struct inet6_dev *idev)
2060 {
2061 	struct ifmcaddr6 *pmc, *pmc_prev, *pmc_next;
2062 	struct sk_buff *skb = NULL;
2063 	int type, dtype;
2064 
2065 	/* deleted MCA's */
2066 	pmc_prev = NULL;
2067 	for (pmc = mc_dereference(idev->mc_tomb, idev);
2068 	     pmc;
2069 	     pmc = pmc_next) {
2070 		pmc_next = mc_dereference(pmc->next, idev);
2071 		if (pmc->mca_sfmode == MCAST_INCLUDE) {
2072 			type = MLD2_BLOCK_OLD_SOURCES;
2073 			dtype = MLD2_BLOCK_OLD_SOURCES;
2074 			skb = add_grec(skb, pmc, type, 1, 0, 0);
2075 			skb = add_grec(skb, pmc, dtype, 1, 1, 0);
2076 		}
2077 		if (pmc->mca_crcount) {
2078 			if (pmc->mca_sfmode == MCAST_EXCLUDE) {
2079 				type = MLD2_CHANGE_TO_INCLUDE;
2080 				skb = add_grec(skb, pmc, type, 1, 0, 0);
2081 			}
2082 			pmc->mca_crcount--;
2083 			if (pmc->mca_crcount == 0) {
2084 				mld_clear_zeros(&pmc->mca_tomb, idev);
2085 				mld_clear_zeros(&pmc->mca_sources, idev);
2086 			}
2087 		}
2088 		if (pmc->mca_crcount == 0 &&
2089 		    !rcu_access_pointer(pmc->mca_tomb) &&
2090 		    !rcu_access_pointer(pmc->mca_sources)) {
2091 			if (pmc_prev)
2092 				rcu_assign_pointer(pmc_prev->next, pmc_next);
2093 			else
2094 				rcu_assign_pointer(idev->mc_tomb, pmc_next);
2095 			in6_dev_put(pmc->idev);
2096 			kfree_rcu(pmc, rcu);
2097 		} else
2098 			pmc_prev = pmc;
2099 	}
2100 
2101 	/* change recs */
2102 	for_each_mc_mclock(idev, pmc) {
2103 		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2104 			type = MLD2_BLOCK_OLD_SOURCES;
2105 			dtype = MLD2_ALLOW_NEW_SOURCES;
2106 		} else {
2107 			type = MLD2_ALLOW_NEW_SOURCES;
2108 			dtype = MLD2_BLOCK_OLD_SOURCES;
2109 		}
2110 		skb = add_grec(skb, pmc, type, 0, 0, 0);
2111 		skb = add_grec(skb, pmc, dtype, 0, 1, 0);	/* deleted sources */
2112 
2113 		/* filter mode changes */
2114 		if (pmc->mca_crcount) {
2115 			if (pmc->mca_sfmode == MCAST_EXCLUDE)
2116 				type = MLD2_CHANGE_TO_EXCLUDE;
2117 			else
2118 				type = MLD2_CHANGE_TO_INCLUDE;
2119 			skb = add_grec(skb, pmc, type, 0, 0, 0);
2120 			pmc->mca_crcount--;
2121 		}
2122 	}
2123 	if (!skb)
2124 		return;
2125 	(void) mld_sendpack(skb);
2126 }
2127 
igmp6_send(struct in6_addr * addr,struct net_device * dev,int type)2128 static void igmp6_send(struct in6_addr *addr, struct net_device *dev, int type)
2129 {
2130 	const struct in6_addr *snd_addr, *saddr;
2131 	int err, len, payload_len, full_len;
2132 	struct in6_addr addr_buf;
2133 	struct inet6_dev *idev;
2134 	struct sk_buff *skb;
2135 	struct mld_msg *hdr;
2136 	int hlen = LL_RESERVED_SPACE(dev);
2137 	int tlen = dev->needed_tailroom;
2138 	u8 ra[8] = { IPPROTO_ICMPV6, 0,
2139 		     IPV6_TLV_ROUTERALERT, 2, 0, 0,
2140 		     IPV6_TLV_PADN, 0 };
2141 	struct dst_entry *dst;
2142 	struct flowi6 fl6;
2143 	struct net *net;
2144 	struct sock *sk;
2145 
2146 	if (type == ICMPV6_MGM_REDUCTION)
2147 		snd_addr = &in6addr_linklocal_allrouters;
2148 	else
2149 		snd_addr = addr;
2150 
2151 	len = sizeof(struct icmp6hdr) + sizeof(struct in6_addr);
2152 	payload_len = len + sizeof(ra);
2153 	full_len = sizeof(struct ipv6hdr) + payload_len;
2154 
2155 	skb = alloc_skb(hlen + tlen + full_len, GFP_KERNEL);
2156 
2157 	rcu_read_lock();
2158 
2159 	net = dev_net_rcu(dev);
2160 	idev = __in6_dev_get(dev);
2161 	IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTREQUESTS);
2162 	if (!skb) {
2163 		IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
2164 		rcu_read_unlock();
2165 		return;
2166 	}
2167 	sk = net->ipv6.igmp_sk;
2168 	skb_set_owner_w(skb, sk);
2169 
2170 	skb->priority = TC_PRIO_CONTROL;
2171 	skb_reserve(skb, hlen);
2172 
2173 	if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
2174 		/* <draft-ietf-magma-mld-source-05.txt>:
2175 		 * use unspecified address as the source address
2176 		 * when a valid link-local address is not available.
2177 		 */
2178 		saddr = &in6addr_any;
2179 	} else
2180 		saddr = &addr_buf;
2181 
2182 	ip6_mc_hdr(sk, skb, dev, saddr, snd_addr, NEXTHDR_HOP, payload_len);
2183 
2184 	skb_put_data(skb, ra, sizeof(ra));
2185 
2186 	hdr = skb_put_zero(skb, sizeof(struct mld_msg));
2187 	hdr->mld_type = type;
2188 	hdr->mld_mca = *addr;
2189 
2190 	hdr->mld_cksum = csum_ipv6_magic(saddr, snd_addr, len,
2191 					 IPPROTO_ICMPV6,
2192 					 csum_partial(hdr, len, 0));
2193 
2194 	icmpv6_flow_init(sk, &fl6, type,
2195 			 &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
2196 			 skb->dev->ifindex);
2197 	dst = icmp6_dst_alloc(skb->dev, &fl6);
2198 	if (IS_ERR(dst)) {
2199 		err = PTR_ERR(dst);
2200 		goto err_out;
2201 	}
2202 
2203 	skb_dst_set(skb, dst);
2204 	err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
2205 		      net, sk, skb, NULL, skb->dev,
2206 		      dst_output);
2207 out:
2208 	if (!err) {
2209 		ICMP6MSGOUT_INC_STATS(net, idev, type);
2210 		ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
2211 	} else
2212 		IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
2213 
2214 	rcu_read_unlock();
2215 	return;
2216 
2217 err_out:
2218 	kfree_skb(skb);
2219 	goto out;
2220 }
2221 
2222 /* called with mc_lock */
mld_send_initial_cr(struct inet6_dev * idev)2223 static void mld_send_initial_cr(struct inet6_dev *idev)
2224 {
2225 	struct sk_buff *skb;
2226 	struct ifmcaddr6 *pmc;
2227 	int type;
2228 
2229 	if (mld_in_v1_mode(idev))
2230 		return;
2231 
2232 	skb = NULL;
2233 	for_each_mc_mclock(idev, pmc) {
2234 		if (pmc->mca_sfcount[MCAST_EXCLUDE])
2235 			type = MLD2_CHANGE_TO_EXCLUDE;
2236 		else
2237 			type = MLD2_ALLOW_NEW_SOURCES;
2238 		skb = add_grec(skb, pmc, type, 0, 0, 1);
2239 	}
2240 	if (skb)
2241 		mld_sendpack(skb);
2242 }
2243 
ipv6_mc_dad_complete(struct inet6_dev * idev)2244 void ipv6_mc_dad_complete(struct inet6_dev *idev)
2245 {
2246 	mutex_lock(&idev->mc_lock);
2247 	idev->mc_dad_count = idev->mc_qrv;
2248 	if (idev->mc_dad_count) {
2249 		mld_send_initial_cr(idev);
2250 		idev->mc_dad_count--;
2251 		if (idev->mc_dad_count)
2252 			mld_dad_start_work(idev,
2253 					   unsolicited_report_interval(idev));
2254 	}
2255 	mutex_unlock(&idev->mc_lock);
2256 }
2257 
mld_dad_work(struct work_struct * work)2258 static void mld_dad_work(struct work_struct *work)
2259 {
2260 	struct inet6_dev *idev = container_of(to_delayed_work(work),
2261 					      struct inet6_dev,
2262 					      mc_dad_work);
2263 	mutex_lock(&idev->mc_lock);
2264 	mld_send_initial_cr(idev);
2265 	if (idev->mc_dad_count) {
2266 		idev->mc_dad_count--;
2267 		if (idev->mc_dad_count)
2268 			mld_dad_start_work(idev,
2269 					   unsolicited_report_interval(idev));
2270 	}
2271 	mutex_unlock(&idev->mc_lock);
2272 	in6_dev_put(idev);
2273 }
2274 
2275 /* called with mc_lock */
ip6_mc_del1_src(struct ifmcaddr6 * pmc,int sfmode,const struct in6_addr * psfsrc)2276 static int ip6_mc_del1_src(struct ifmcaddr6 *pmc, int sfmode,
2277 	const struct in6_addr *psfsrc)
2278 {
2279 	struct ip6_sf_list *psf, *psf_prev;
2280 	int rv = 0;
2281 
2282 	psf_prev = NULL;
2283 	for_each_psf_mclock(pmc, psf) {
2284 		if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2285 			break;
2286 		psf_prev = psf;
2287 	}
2288 	if (!psf || psf->sf_count[sfmode] == 0) {
2289 		/* source filter not found, or count wrong =>  bug */
2290 		return -ESRCH;
2291 	}
2292 	psf->sf_count[sfmode]--;
2293 	if (!psf->sf_count[MCAST_INCLUDE] && !psf->sf_count[MCAST_EXCLUDE]) {
2294 		struct inet6_dev *idev = pmc->idev;
2295 
2296 		/* no more filters for this source */
2297 		if (psf_prev)
2298 			rcu_assign_pointer(psf_prev->sf_next,
2299 					   mc_dereference(psf->sf_next, idev));
2300 		else
2301 			rcu_assign_pointer(pmc->mca_sources,
2302 					   mc_dereference(psf->sf_next, idev));
2303 
2304 		if (psf->sf_oldin && !(pmc->mca_flags & MAF_NOREPORT) &&
2305 		    !mld_in_v1_mode(idev)) {
2306 			psf->sf_crcount = idev->mc_qrv;
2307 			rcu_assign_pointer(psf->sf_next,
2308 					   mc_dereference(pmc->mca_tomb, idev));
2309 			rcu_assign_pointer(pmc->mca_tomb, psf);
2310 			rv = 1;
2311 		} else {
2312 			kfree_rcu(psf, rcu);
2313 		}
2314 	}
2315 	return rv;
2316 }
2317 
2318 /* called with mc_lock */
ip6_mc_del_src(struct inet6_dev * idev,const struct in6_addr * pmca,int sfmode,int sfcount,const struct in6_addr * psfsrc,int delta)2319 static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2320 			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
2321 			  int delta)
2322 {
2323 	struct ifmcaddr6 *pmc;
2324 	int	changerec = 0;
2325 	int	i, err;
2326 
2327 	if (!idev)
2328 		return -ENODEV;
2329 
2330 	for_each_mc_mclock(idev, pmc) {
2331 		if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2332 			break;
2333 	}
2334 	if (!pmc)
2335 		return -ESRCH;
2336 
2337 	sf_markstate(pmc);
2338 	if (!delta) {
2339 		if (!pmc->mca_sfcount[sfmode])
2340 			return -EINVAL;
2341 
2342 		pmc->mca_sfcount[sfmode]--;
2343 	}
2344 	err = 0;
2345 	for (i = 0; i < sfcount; i++) {
2346 		int rv = ip6_mc_del1_src(pmc, sfmode, &psfsrc[i]);
2347 
2348 		changerec |= rv > 0;
2349 		if (!err && rv < 0)
2350 			err = rv;
2351 	}
2352 	if (pmc->mca_sfmode == MCAST_EXCLUDE &&
2353 	    pmc->mca_sfcount[MCAST_EXCLUDE] == 0 &&
2354 	    pmc->mca_sfcount[MCAST_INCLUDE]) {
2355 		struct ip6_sf_list *psf;
2356 
2357 		/* filter mode change */
2358 		pmc->mca_sfmode = MCAST_INCLUDE;
2359 		pmc->mca_crcount = idev->mc_qrv;
2360 		idev->mc_ifc_count = pmc->mca_crcount;
2361 		for_each_psf_mclock(pmc, psf)
2362 			psf->sf_crcount = 0;
2363 		mld_ifc_event(pmc->idev);
2364 	} else if (sf_setstate(pmc) || changerec) {
2365 		mld_ifc_event(pmc->idev);
2366 	}
2367 
2368 	return err;
2369 }
2370 
2371 /*
2372  * Add multicast single-source filter to the interface list
2373  * called with mc_lock
2374  */
ip6_mc_add1_src(struct ifmcaddr6 * pmc,int sfmode,const struct in6_addr * psfsrc)2375 static int ip6_mc_add1_src(struct ifmcaddr6 *pmc, int sfmode,
2376 	const struct in6_addr *psfsrc)
2377 {
2378 	struct ip6_sf_list *psf, *psf_prev;
2379 
2380 	psf_prev = NULL;
2381 	for_each_psf_mclock(pmc, psf) {
2382 		if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2383 			break;
2384 		psf_prev = psf;
2385 	}
2386 	if (!psf) {
2387 		psf = kzalloc(sizeof(*psf), GFP_KERNEL);
2388 		if (!psf)
2389 			return -ENOBUFS;
2390 
2391 		psf->sf_addr = *psfsrc;
2392 		if (psf_prev) {
2393 			rcu_assign_pointer(psf_prev->sf_next, psf);
2394 		} else {
2395 			rcu_assign_pointer(pmc->mca_sources, psf);
2396 		}
2397 	}
2398 	psf->sf_count[sfmode]++;
2399 	return 0;
2400 }
2401 
2402 /* called with mc_lock */
sf_markstate(struct ifmcaddr6 * pmc)2403 static void sf_markstate(struct ifmcaddr6 *pmc)
2404 {
2405 	struct ip6_sf_list *psf;
2406 	int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2407 
2408 	for_each_psf_mclock(pmc, psf) {
2409 		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2410 			psf->sf_oldin = mca_xcount ==
2411 				psf->sf_count[MCAST_EXCLUDE] &&
2412 				!psf->sf_count[MCAST_INCLUDE];
2413 		} else {
2414 			psf->sf_oldin = psf->sf_count[MCAST_INCLUDE] != 0;
2415 		}
2416 	}
2417 }
2418 
2419 /* called with mc_lock */
sf_setstate(struct ifmcaddr6 * pmc)2420 static int sf_setstate(struct ifmcaddr6 *pmc)
2421 {
2422 	struct ip6_sf_list *psf, *dpsf;
2423 	int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2424 	int qrv = pmc->idev->mc_qrv;
2425 	int new_in, rv;
2426 
2427 	rv = 0;
2428 	for_each_psf_mclock(pmc, psf) {
2429 		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2430 			new_in = mca_xcount == psf->sf_count[MCAST_EXCLUDE] &&
2431 				!psf->sf_count[MCAST_INCLUDE];
2432 		} else
2433 			new_in = psf->sf_count[MCAST_INCLUDE] != 0;
2434 		if (new_in) {
2435 			if (!psf->sf_oldin) {
2436 				struct ip6_sf_list *prev = NULL;
2437 
2438 				for_each_psf_tomb(pmc, dpsf) {
2439 					if (ipv6_addr_equal(&dpsf->sf_addr,
2440 					    &psf->sf_addr))
2441 						break;
2442 					prev = dpsf;
2443 				}
2444 				if (dpsf) {
2445 					if (prev)
2446 						rcu_assign_pointer(prev->sf_next,
2447 								   mc_dereference(dpsf->sf_next,
2448 										  pmc->idev));
2449 					else
2450 						rcu_assign_pointer(pmc->mca_tomb,
2451 								   mc_dereference(dpsf->sf_next,
2452 										  pmc->idev));
2453 					kfree_rcu(dpsf, rcu);
2454 				}
2455 				psf->sf_crcount = qrv;
2456 				rv++;
2457 			}
2458 		} else if (psf->sf_oldin) {
2459 			psf->sf_crcount = 0;
2460 			/*
2461 			 * add or update "delete" records if an active filter
2462 			 * is now inactive
2463 			 */
2464 
2465 			for_each_psf_tomb(pmc, dpsf)
2466 				if (ipv6_addr_equal(&dpsf->sf_addr,
2467 				    &psf->sf_addr))
2468 					break;
2469 			if (!dpsf) {
2470 				dpsf = kmalloc(sizeof(*dpsf), GFP_KERNEL);
2471 				if (!dpsf)
2472 					continue;
2473 				*dpsf = *psf;
2474 				rcu_assign_pointer(dpsf->sf_next,
2475 						   mc_dereference(pmc->mca_tomb, pmc->idev));
2476 				rcu_assign_pointer(pmc->mca_tomb, dpsf);
2477 			}
2478 			dpsf->sf_crcount = qrv;
2479 			rv++;
2480 		}
2481 	}
2482 	return rv;
2483 }
2484 
2485 /*
2486  * Add multicast source filter list to the interface list
2487  * called with mc_lock
2488  */
ip6_mc_add_src(struct inet6_dev * idev,const struct in6_addr * pmca,int sfmode,int sfcount,const struct in6_addr * psfsrc,int delta)2489 static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2490 			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
2491 			  int delta)
2492 {
2493 	struct ifmcaddr6 *pmc;
2494 	int	isexclude;
2495 	int	i, err;
2496 
2497 	if (!idev)
2498 		return -ENODEV;
2499 
2500 	for_each_mc_mclock(idev, pmc) {
2501 		if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2502 			break;
2503 	}
2504 	if (!pmc)
2505 		return -ESRCH;
2506 
2507 	sf_markstate(pmc);
2508 	isexclude = pmc->mca_sfmode == MCAST_EXCLUDE;
2509 	if (!delta)
2510 		pmc->mca_sfcount[sfmode]++;
2511 	err = 0;
2512 	for (i = 0; i < sfcount; i++) {
2513 		err = ip6_mc_add1_src(pmc, sfmode, &psfsrc[i]);
2514 		if (err)
2515 			break;
2516 	}
2517 	if (err) {
2518 		int j;
2519 
2520 		if (!delta)
2521 			pmc->mca_sfcount[sfmode]--;
2522 		for (j = 0; j < i; j++)
2523 			ip6_mc_del1_src(pmc, sfmode, &psfsrc[j]);
2524 	} else if (isexclude != (pmc->mca_sfcount[MCAST_EXCLUDE] != 0)) {
2525 		struct ip6_sf_list *psf;
2526 
2527 		/* filter mode change */
2528 		if (pmc->mca_sfcount[MCAST_EXCLUDE])
2529 			pmc->mca_sfmode = MCAST_EXCLUDE;
2530 		else if (pmc->mca_sfcount[MCAST_INCLUDE])
2531 			pmc->mca_sfmode = MCAST_INCLUDE;
2532 		/* else no filters; keep old mode for reports */
2533 
2534 		pmc->mca_crcount = idev->mc_qrv;
2535 		idev->mc_ifc_count = pmc->mca_crcount;
2536 		for_each_psf_mclock(pmc, psf)
2537 			psf->sf_crcount = 0;
2538 		mld_ifc_event(idev);
2539 	} else if (sf_setstate(pmc)) {
2540 		mld_ifc_event(idev);
2541 	}
2542 	return err;
2543 }
2544 
2545 /* called with mc_lock */
ip6_mc_clear_src(struct ifmcaddr6 * pmc)2546 static void ip6_mc_clear_src(struct ifmcaddr6 *pmc)
2547 {
2548 	struct ip6_sf_list *psf, *nextpsf;
2549 
2550 	for (psf = mc_dereference(pmc->mca_tomb, pmc->idev);
2551 	     psf;
2552 	     psf = nextpsf) {
2553 		nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2554 		kfree_rcu(psf, rcu);
2555 	}
2556 	RCU_INIT_POINTER(pmc->mca_tomb, NULL);
2557 	for (psf = mc_dereference(pmc->mca_sources, pmc->idev);
2558 	     psf;
2559 	     psf = nextpsf) {
2560 		nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2561 		kfree_rcu(psf, rcu);
2562 	}
2563 	RCU_INIT_POINTER(pmc->mca_sources, NULL);
2564 	pmc->mca_sfmode = MCAST_EXCLUDE;
2565 	pmc->mca_sfcount[MCAST_INCLUDE] = 0;
2566 	pmc->mca_sfcount[MCAST_EXCLUDE] = 1;
2567 }
2568 
2569 /* called with mc_lock */
igmp6_join_group(struct ifmcaddr6 * ma)2570 static void igmp6_join_group(struct ifmcaddr6 *ma)
2571 {
2572 	unsigned long delay;
2573 
2574 	if (ma->mca_flags & MAF_NOREPORT)
2575 		return;
2576 
2577 	igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2578 
2579 	delay = get_random_u32_below(unsolicited_report_interval(ma->idev));
2580 
2581 	if (cancel_delayed_work(&ma->mca_work)) {
2582 		refcount_dec(&ma->mca_refcnt);
2583 		delay = ma->mca_work.timer.expires - jiffies;
2584 	}
2585 
2586 	if (!mod_delayed_work(mld_wq, &ma->mca_work, delay))
2587 		refcount_inc(&ma->mca_refcnt);
2588 	ma->mca_flags |= MAF_TIMER_RUNNING | MAF_LAST_REPORTER;
2589 }
2590 
ip6_mc_leave_src(struct sock * sk,struct ipv6_mc_socklist * iml,struct inet6_dev * idev)2591 static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
2592 			    struct inet6_dev *idev)
2593 {
2594 	struct ip6_sf_socklist *psl;
2595 	int err;
2596 
2597 	psl = sock_dereference(iml->sflist, sk);
2598 
2599 	if (idev)
2600 		mutex_lock(&idev->mc_lock);
2601 
2602 	if (!psl) {
2603 		/* any-source empty exclude case */
2604 		err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode, 0, NULL, 0);
2605 	} else {
2606 		err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode,
2607 				     psl->sl_count, psl->sl_addr, 0);
2608 		RCU_INIT_POINTER(iml->sflist, NULL);
2609 		atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
2610 			   &sk->sk_omem_alloc);
2611 		kfree_rcu(psl, rcu);
2612 	}
2613 
2614 	if (idev)
2615 		mutex_unlock(&idev->mc_lock);
2616 
2617 	return err;
2618 }
2619 
2620 /* called with mc_lock */
igmp6_leave_group(struct ifmcaddr6 * ma)2621 static void igmp6_leave_group(struct ifmcaddr6 *ma)
2622 {
2623 	if (mld_in_v1_mode(ma->idev)) {
2624 		if (ma->mca_flags & MAF_LAST_REPORTER) {
2625 			igmp6_send(&ma->mca_addr, ma->idev->dev,
2626 				ICMPV6_MGM_REDUCTION);
2627 		}
2628 	} else {
2629 		mld_add_delrec(ma->idev, ma);
2630 		mld_ifc_event(ma->idev);
2631 	}
2632 }
2633 
mld_gq_work(struct work_struct * work)2634 static void mld_gq_work(struct work_struct *work)
2635 {
2636 	struct inet6_dev *idev = container_of(to_delayed_work(work),
2637 					      struct inet6_dev,
2638 					      mc_gq_work);
2639 
2640 	mutex_lock(&idev->mc_lock);
2641 	mld_send_report(idev, NULL);
2642 	idev->mc_gq_running = 0;
2643 	mutex_unlock(&idev->mc_lock);
2644 
2645 	in6_dev_put(idev);
2646 }
2647 
mld_ifc_work(struct work_struct * work)2648 static void mld_ifc_work(struct work_struct *work)
2649 {
2650 	struct inet6_dev *idev = container_of(to_delayed_work(work),
2651 					      struct inet6_dev,
2652 					      mc_ifc_work);
2653 
2654 	mutex_lock(&idev->mc_lock);
2655 	mld_send_cr(idev);
2656 
2657 	if (idev->mc_ifc_count) {
2658 		idev->mc_ifc_count--;
2659 		if (idev->mc_ifc_count)
2660 			mld_ifc_start_work(idev,
2661 					   unsolicited_report_interval(idev));
2662 	}
2663 	mutex_unlock(&idev->mc_lock);
2664 	in6_dev_put(idev);
2665 }
2666 
2667 /* called with mc_lock */
mld_ifc_event(struct inet6_dev * idev)2668 static void mld_ifc_event(struct inet6_dev *idev)
2669 {
2670 	if (mld_in_v1_mode(idev))
2671 		return;
2672 
2673 	idev->mc_ifc_count = idev->mc_qrv;
2674 	mld_ifc_start_work(idev, 1);
2675 }
2676 
mld_mca_work(struct work_struct * work)2677 static void mld_mca_work(struct work_struct *work)
2678 {
2679 	struct ifmcaddr6 *ma = container_of(to_delayed_work(work),
2680 					    struct ifmcaddr6, mca_work);
2681 
2682 	mutex_lock(&ma->idev->mc_lock);
2683 	if (mld_in_v1_mode(ma->idev))
2684 		igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2685 	else
2686 		mld_send_report(ma->idev, ma);
2687 	ma->mca_flags |=  MAF_LAST_REPORTER;
2688 	ma->mca_flags &= ~MAF_TIMER_RUNNING;
2689 	mutex_unlock(&ma->idev->mc_lock);
2690 
2691 	ma_put(ma);
2692 }
2693 
2694 /* Device changing type */
2695 
ipv6_mc_unmap(struct inet6_dev * idev)2696 void ipv6_mc_unmap(struct inet6_dev *idev)
2697 {
2698 	struct ifmcaddr6 *i;
2699 
2700 	/* Install multicast list, except for all-nodes (already installed) */
2701 
2702 	mutex_lock(&idev->mc_lock);
2703 	for_each_mc_mclock(idev, i)
2704 		igmp6_group_dropped(i);
2705 	mutex_unlock(&idev->mc_lock);
2706 }
2707 
ipv6_mc_remap(struct inet6_dev * idev)2708 void ipv6_mc_remap(struct inet6_dev *idev)
2709 {
2710 	ipv6_mc_up(idev);
2711 }
2712 
2713 /* Device going down */
ipv6_mc_down(struct inet6_dev * idev)2714 void ipv6_mc_down(struct inet6_dev *idev)
2715 {
2716 	struct ifmcaddr6 *i;
2717 
2718 	mutex_lock(&idev->mc_lock);
2719 	/* Withdraw multicast list */
2720 	for_each_mc_mclock(idev, i)
2721 		igmp6_group_dropped(i);
2722 	mutex_unlock(&idev->mc_lock);
2723 
2724 	/* Should stop work after group drop. or we will
2725 	 * start work again in mld_ifc_event()
2726 	 */
2727 	mld_query_stop_work(idev);
2728 	mld_report_stop_work(idev);
2729 
2730 	mutex_lock(&idev->mc_lock);
2731 	mld_ifc_stop_work(idev);
2732 	mld_gq_stop_work(idev);
2733 	mutex_unlock(&idev->mc_lock);
2734 
2735 	mld_dad_stop_work(idev);
2736 }
2737 
ipv6_mc_reset(struct inet6_dev * idev)2738 static void ipv6_mc_reset(struct inet6_dev *idev)
2739 {
2740 	idev->mc_qrv = sysctl_mld_qrv;
2741 	idev->mc_qi = MLD_QI_DEFAULT;
2742 	idev->mc_qri = MLD_QRI_DEFAULT;
2743 	idev->mc_v1_seen = 0;
2744 	idev->mc_maxdelay = unsolicited_report_interval(idev);
2745 }
2746 
2747 /* Device going up */
2748 
ipv6_mc_up(struct inet6_dev * idev)2749 void ipv6_mc_up(struct inet6_dev *idev)
2750 {
2751 	struct ifmcaddr6 *i;
2752 
2753 	/* Install multicast list, except for all-nodes (already installed) */
2754 
2755 	ipv6_mc_reset(idev);
2756 	mutex_lock(&idev->mc_lock);
2757 	for_each_mc_mclock(idev, i) {
2758 		mld_del_delrec(idev, i);
2759 		igmp6_group_added(i);
2760 	}
2761 	mutex_unlock(&idev->mc_lock);
2762 }
2763 
2764 /* IPv6 device initialization. */
2765 
ipv6_mc_init_dev(struct inet6_dev * idev)2766 void ipv6_mc_init_dev(struct inet6_dev *idev)
2767 {
2768 	idev->mc_gq_running = 0;
2769 	INIT_DELAYED_WORK(&idev->mc_gq_work, mld_gq_work);
2770 	RCU_INIT_POINTER(idev->mc_tomb, NULL);
2771 	idev->mc_ifc_count = 0;
2772 	INIT_DELAYED_WORK(&idev->mc_ifc_work, mld_ifc_work);
2773 	INIT_DELAYED_WORK(&idev->mc_dad_work, mld_dad_work);
2774 	INIT_DELAYED_WORK(&idev->mc_query_work, mld_query_work);
2775 	INIT_DELAYED_WORK(&idev->mc_report_work, mld_report_work);
2776 	skb_queue_head_init(&idev->mc_query_queue);
2777 	skb_queue_head_init(&idev->mc_report_queue);
2778 	spin_lock_init(&idev->mc_query_lock);
2779 	spin_lock_init(&idev->mc_report_lock);
2780 	mutex_init(&idev->mc_lock);
2781 	ipv6_mc_reset(idev);
2782 }
2783 
2784 /*
2785  *	Device is about to be destroyed: clean up.
2786  */
2787 
ipv6_mc_destroy_dev(struct inet6_dev * idev)2788 void ipv6_mc_destroy_dev(struct inet6_dev *idev)
2789 {
2790 	struct ifmcaddr6 *i;
2791 
2792 	/* Deactivate works */
2793 	ipv6_mc_down(idev);
2794 	mutex_lock(&idev->mc_lock);
2795 	mld_clear_delrec(idev);
2796 	mutex_unlock(&idev->mc_lock);
2797 	mld_clear_query(idev);
2798 	mld_clear_report(idev);
2799 
2800 	/* Delete all-nodes address. */
2801 	/* We cannot call ipv6_dev_mc_dec() directly, our caller in
2802 	 * addrconf.c has NULL'd out dev->ip6_ptr so in6_dev_get() will
2803 	 * fail.
2804 	 */
2805 	__ipv6_dev_mc_dec(idev, &in6addr_linklocal_allnodes);
2806 
2807 	if (idev->cnf.forwarding)
2808 		__ipv6_dev_mc_dec(idev, &in6addr_linklocal_allrouters);
2809 
2810 	mutex_lock(&idev->mc_lock);
2811 	while ((i = mc_dereference(idev->mc_list, idev))) {
2812 		rcu_assign_pointer(idev->mc_list, mc_dereference(i->next, idev));
2813 
2814 		ip6_mc_clear_src(i);
2815 		ma_put(i);
2816 	}
2817 	mutex_unlock(&idev->mc_lock);
2818 }
2819 
ipv6_mc_rejoin_groups(struct inet6_dev * idev)2820 static void ipv6_mc_rejoin_groups(struct inet6_dev *idev)
2821 {
2822 	struct ifmcaddr6 *pmc;
2823 
2824 	ASSERT_RTNL();
2825 
2826 	mutex_lock(&idev->mc_lock);
2827 	if (mld_in_v1_mode(idev)) {
2828 		for_each_mc_mclock(idev, pmc)
2829 			igmp6_join_group(pmc);
2830 	} else {
2831 		mld_send_report(idev, NULL);
2832 	}
2833 	mutex_unlock(&idev->mc_lock);
2834 }
2835 
ipv6_mc_netdev_event(struct notifier_block * this,unsigned long event,void * ptr)2836 static int ipv6_mc_netdev_event(struct notifier_block *this,
2837 				unsigned long event,
2838 				void *ptr)
2839 {
2840 	struct net_device *dev = netdev_notifier_info_to_dev(ptr);
2841 	struct inet6_dev *idev = __in6_dev_get(dev);
2842 
2843 	switch (event) {
2844 	case NETDEV_RESEND_IGMP:
2845 		if (idev)
2846 			ipv6_mc_rejoin_groups(idev);
2847 		break;
2848 	default:
2849 		break;
2850 	}
2851 
2852 	return NOTIFY_DONE;
2853 }
2854 
2855 static struct notifier_block igmp6_netdev_notifier = {
2856 	.notifier_call = ipv6_mc_netdev_event,
2857 };
2858 
2859 #ifdef CONFIG_PROC_FS
2860 struct igmp6_mc_iter_state {
2861 	struct seq_net_private p;
2862 	struct net_device *dev;
2863 	struct inet6_dev *idev;
2864 };
2865 
2866 #define igmp6_mc_seq_private(seq)	((struct igmp6_mc_iter_state *)(seq)->private)
2867 
igmp6_mc_get_first(struct seq_file * seq)2868 static inline struct ifmcaddr6 *igmp6_mc_get_first(struct seq_file *seq)
2869 {
2870 	struct ifmcaddr6 *im = NULL;
2871 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2872 	struct net *net = seq_file_net(seq);
2873 
2874 	state->idev = NULL;
2875 	for_each_netdev_rcu(net, state->dev) {
2876 		struct inet6_dev *idev;
2877 		idev = __in6_dev_get(state->dev);
2878 		if (!idev)
2879 			continue;
2880 
2881 		im = rcu_dereference(idev->mc_list);
2882 		if (im) {
2883 			state->idev = idev;
2884 			break;
2885 		}
2886 	}
2887 	return im;
2888 }
2889 
igmp6_mc_get_next(struct seq_file * seq,struct ifmcaddr6 * im)2890 static struct ifmcaddr6 *igmp6_mc_get_next(struct seq_file *seq, struct ifmcaddr6 *im)
2891 {
2892 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2893 
2894 	im = rcu_dereference(im->next);
2895 	while (!im) {
2896 		state->dev = next_net_device_rcu(state->dev);
2897 		if (!state->dev) {
2898 			state->idev = NULL;
2899 			break;
2900 		}
2901 		state->idev = __in6_dev_get(state->dev);
2902 		if (!state->idev)
2903 			continue;
2904 		im = rcu_dereference(state->idev->mc_list);
2905 	}
2906 	return im;
2907 }
2908 
igmp6_mc_get_idx(struct seq_file * seq,loff_t pos)2909 static struct ifmcaddr6 *igmp6_mc_get_idx(struct seq_file *seq, loff_t pos)
2910 {
2911 	struct ifmcaddr6 *im = igmp6_mc_get_first(seq);
2912 	if (im)
2913 		while (pos && (im = igmp6_mc_get_next(seq, im)) != NULL)
2914 			--pos;
2915 	return pos ? NULL : im;
2916 }
2917 
igmp6_mc_seq_start(struct seq_file * seq,loff_t * pos)2918 static void *igmp6_mc_seq_start(struct seq_file *seq, loff_t *pos)
2919 	__acquires(RCU)
2920 {
2921 	rcu_read_lock();
2922 	return igmp6_mc_get_idx(seq, *pos);
2923 }
2924 
igmp6_mc_seq_next(struct seq_file * seq,void * v,loff_t * pos)2925 static void *igmp6_mc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2926 {
2927 	struct ifmcaddr6 *im = igmp6_mc_get_next(seq, v);
2928 
2929 	++*pos;
2930 	return im;
2931 }
2932 
igmp6_mc_seq_stop(struct seq_file * seq,void * v)2933 static void igmp6_mc_seq_stop(struct seq_file *seq, void *v)
2934 	__releases(RCU)
2935 {
2936 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2937 
2938 	if (likely(state->idev))
2939 		state->idev = NULL;
2940 	state->dev = NULL;
2941 	rcu_read_unlock();
2942 }
2943 
igmp6_mc_seq_show(struct seq_file * seq,void * v)2944 static int igmp6_mc_seq_show(struct seq_file *seq, void *v)
2945 {
2946 	struct ifmcaddr6 *im = (struct ifmcaddr6 *)v;
2947 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2948 
2949 	seq_printf(seq,
2950 		   "%-4d %-15s %pi6 %5d %08X %ld\n",
2951 		   state->dev->ifindex, state->dev->name,
2952 		   &im->mca_addr,
2953 		   im->mca_users, im->mca_flags,
2954 		   (im->mca_flags & MAF_TIMER_RUNNING) ?
2955 		   jiffies_to_clock_t(im->mca_work.timer.expires - jiffies) : 0);
2956 	return 0;
2957 }
2958 
2959 static const struct seq_operations igmp6_mc_seq_ops = {
2960 	.start	=	igmp6_mc_seq_start,
2961 	.next	=	igmp6_mc_seq_next,
2962 	.stop	=	igmp6_mc_seq_stop,
2963 	.show	=	igmp6_mc_seq_show,
2964 };
2965 
2966 struct igmp6_mcf_iter_state {
2967 	struct seq_net_private p;
2968 	struct net_device *dev;
2969 	struct inet6_dev *idev;
2970 	struct ifmcaddr6 *im;
2971 };
2972 
2973 #define igmp6_mcf_seq_private(seq)	((struct igmp6_mcf_iter_state *)(seq)->private)
2974 
igmp6_mcf_get_first(struct seq_file * seq)2975 static inline struct ip6_sf_list *igmp6_mcf_get_first(struct seq_file *seq)
2976 {
2977 	struct ip6_sf_list *psf = NULL;
2978 	struct ifmcaddr6 *im = NULL;
2979 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
2980 	struct net *net = seq_file_net(seq);
2981 
2982 	state->idev = NULL;
2983 	state->im = NULL;
2984 	for_each_netdev_rcu(net, state->dev) {
2985 		struct inet6_dev *idev;
2986 		idev = __in6_dev_get(state->dev);
2987 		if (unlikely(idev == NULL))
2988 			continue;
2989 
2990 		im = rcu_dereference(idev->mc_list);
2991 		if (likely(im)) {
2992 			psf = rcu_dereference(im->mca_sources);
2993 			if (likely(psf)) {
2994 				state->im = im;
2995 				state->idev = idev;
2996 				break;
2997 			}
2998 		}
2999 	}
3000 	return psf;
3001 }
3002 
igmp6_mcf_get_next(struct seq_file * seq,struct ip6_sf_list * psf)3003 static struct ip6_sf_list *igmp6_mcf_get_next(struct seq_file *seq, struct ip6_sf_list *psf)
3004 {
3005 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3006 
3007 	psf = rcu_dereference(psf->sf_next);
3008 	while (!psf) {
3009 		state->im = rcu_dereference(state->im->next);
3010 		while (!state->im) {
3011 			state->dev = next_net_device_rcu(state->dev);
3012 			if (!state->dev) {
3013 				state->idev = NULL;
3014 				goto out;
3015 			}
3016 			state->idev = __in6_dev_get(state->dev);
3017 			if (!state->idev)
3018 				continue;
3019 			state->im = rcu_dereference(state->idev->mc_list);
3020 		}
3021 		if (!state->im)
3022 			break;
3023 		psf = rcu_dereference(state->im->mca_sources);
3024 	}
3025 out:
3026 	return psf;
3027 }
3028 
igmp6_mcf_get_idx(struct seq_file * seq,loff_t pos)3029 static struct ip6_sf_list *igmp6_mcf_get_idx(struct seq_file *seq, loff_t pos)
3030 {
3031 	struct ip6_sf_list *psf = igmp6_mcf_get_first(seq);
3032 	if (psf)
3033 		while (pos && (psf = igmp6_mcf_get_next(seq, psf)) != NULL)
3034 			--pos;
3035 	return pos ? NULL : psf;
3036 }
3037 
igmp6_mcf_seq_start(struct seq_file * seq,loff_t * pos)3038 static void *igmp6_mcf_seq_start(struct seq_file *seq, loff_t *pos)
3039 	__acquires(RCU)
3040 {
3041 	rcu_read_lock();
3042 	return *pos ? igmp6_mcf_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
3043 }
3044 
igmp6_mcf_seq_next(struct seq_file * seq,void * v,loff_t * pos)3045 static void *igmp6_mcf_seq_next(struct seq_file *seq, void *v, loff_t *pos)
3046 {
3047 	struct ip6_sf_list *psf;
3048 	if (v == SEQ_START_TOKEN)
3049 		psf = igmp6_mcf_get_first(seq);
3050 	else
3051 		psf = igmp6_mcf_get_next(seq, v);
3052 	++*pos;
3053 	return psf;
3054 }
3055 
igmp6_mcf_seq_stop(struct seq_file * seq,void * v)3056 static void igmp6_mcf_seq_stop(struct seq_file *seq, void *v)
3057 	__releases(RCU)
3058 {
3059 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3060 
3061 	if (likely(state->im))
3062 		state->im = NULL;
3063 	if (likely(state->idev))
3064 		state->idev = NULL;
3065 
3066 	state->dev = NULL;
3067 	rcu_read_unlock();
3068 }
3069 
igmp6_mcf_seq_show(struct seq_file * seq,void * v)3070 static int igmp6_mcf_seq_show(struct seq_file *seq, void *v)
3071 {
3072 	struct ip6_sf_list *psf = (struct ip6_sf_list *)v;
3073 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3074 
3075 	if (v == SEQ_START_TOKEN) {
3076 		seq_puts(seq, "Idx Device                Multicast Address                   Source Address    INC    EXC\n");
3077 	} else {
3078 		seq_printf(seq,
3079 			   "%3d %6.6s %pi6 %pi6 %6lu %6lu\n",
3080 			   state->dev->ifindex, state->dev->name,
3081 			   &state->im->mca_addr,
3082 			   &psf->sf_addr,
3083 			   psf->sf_count[MCAST_INCLUDE],
3084 			   psf->sf_count[MCAST_EXCLUDE]);
3085 	}
3086 	return 0;
3087 }
3088 
3089 static const struct seq_operations igmp6_mcf_seq_ops = {
3090 	.start	=	igmp6_mcf_seq_start,
3091 	.next	=	igmp6_mcf_seq_next,
3092 	.stop	=	igmp6_mcf_seq_stop,
3093 	.show	=	igmp6_mcf_seq_show,
3094 };
3095 
igmp6_proc_init(struct net * net)3096 static int __net_init igmp6_proc_init(struct net *net)
3097 {
3098 	int err;
3099 
3100 	err = -ENOMEM;
3101 	if (!proc_create_net("igmp6", 0444, net->proc_net, &igmp6_mc_seq_ops,
3102 			sizeof(struct igmp6_mc_iter_state)))
3103 		goto out;
3104 	if (!proc_create_net("mcfilter6", 0444, net->proc_net,
3105 			&igmp6_mcf_seq_ops,
3106 			sizeof(struct igmp6_mcf_iter_state)))
3107 		goto out_proc_net_igmp6;
3108 
3109 	err = 0;
3110 out:
3111 	return err;
3112 
3113 out_proc_net_igmp6:
3114 	remove_proc_entry("igmp6", net->proc_net);
3115 	goto out;
3116 }
3117 
igmp6_proc_exit(struct net * net)3118 static void __net_exit igmp6_proc_exit(struct net *net)
3119 {
3120 	remove_proc_entry("mcfilter6", net->proc_net);
3121 	remove_proc_entry("igmp6", net->proc_net);
3122 }
3123 #else
igmp6_proc_init(struct net * net)3124 static inline int igmp6_proc_init(struct net *net)
3125 {
3126 	return 0;
3127 }
igmp6_proc_exit(struct net * net)3128 static inline void igmp6_proc_exit(struct net *net)
3129 {
3130 }
3131 #endif
3132 
igmp6_net_init(struct net * net)3133 static int __net_init igmp6_net_init(struct net *net)
3134 {
3135 	int err;
3136 
3137 	err = inet_ctl_sock_create(&net->ipv6.igmp_sk, PF_INET6,
3138 				   SOCK_RAW, IPPROTO_ICMPV6, net);
3139 	if (err < 0) {
3140 		pr_err("Failed to initialize the IGMP6 control socket (err %d)\n",
3141 		       err);
3142 		goto out;
3143 	}
3144 
3145 	inet6_sk(net->ipv6.igmp_sk)->hop_limit = 1;
3146 	net->ipv6.igmp_sk->sk_allocation = GFP_KERNEL;
3147 
3148 	err = inet_ctl_sock_create(&net->ipv6.mc_autojoin_sk, PF_INET6,
3149 				   SOCK_RAW, IPPROTO_ICMPV6, net);
3150 	if (err < 0) {
3151 		pr_err("Failed to initialize the IGMP6 autojoin socket (err %d)\n",
3152 		       err);
3153 		goto out_sock_create;
3154 	}
3155 
3156 	err = igmp6_proc_init(net);
3157 	if (err)
3158 		goto out_sock_create_autojoin;
3159 
3160 	return 0;
3161 
3162 out_sock_create_autojoin:
3163 	inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3164 out_sock_create:
3165 	inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3166 out:
3167 	return err;
3168 }
3169 
igmp6_net_exit(struct net * net)3170 static void __net_exit igmp6_net_exit(struct net *net)
3171 {
3172 	inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3173 	inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3174 	igmp6_proc_exit(net);
3175 }
3176 
3177 static struct pernet_operations igmp6_net_ops = {
3178 	.init = igmp6_net_init,
3179 	.exit = igmp6_net_exit,
3180 };
3181 
igmp6_init(void)3182 int __init igmp6_init(void)
3183 {
3184 	int err;
3185 
3186 	err = register_pernet_subsys(&igmp6_net_ops);
3187 	if (err)
3188 		return err;
3189 
3190 	mld_wq = create_workqueue("mld");
3191 	if (!mld_wq) {
3192 		unregister_pernet_subsys(&igmp6_net_ops);
3193 		return -ENOMEM;
3194 	}
3195 
3196 	return err;
3197 }
3198 
igmp6_late_init(void)3199 int __init igmp6_late_init(void)
3200 {
3201 	return register_netdevice_notifier(&igmp6_netdev_notifier);
3202 }
3203 
igmp6_cleanup(void)3204 void igmp6_cleanup(void)
3205 {
3206 	unregister_pernet_subsys(&igmp6_net_ops);
3207 	destroy_workqueue(mld_wq);
3208 }
3209 
igmp6_late_cleanup(void)3210 void igmp6_late_cleanup(void)
3211 {
3212 	unregister_netdevice_notifier(&igmp6_netdev_notifier);
3213 }
3214