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 	struct net *net = dev_net(dev);
1733 	const struct in6_addr *saddr;
1734 	struct in6_addr addr_buf;
1735 	struct mld2_report *pmr;
1736 	struct sk_buff *skb;
1737 	unsigned int size;
1738 	struct sock *sk;
1739 	int err;
1740 
1741 	sk = net->ipv6.igmp_sk;
1742 	/* we assume size > sizeof(ra) here
1743 	 * Also try to not allocate high-order pages for big MTU
1744 	 */
1745 	size = min_t(int, mtu, PAGE_SIZE / 2) + hlen + tlen;
1746 	skb = sock_alloc_send_skb(sk, size, 1, &err);
1747 	if (!skb)
1748 		return NULL;
1749 
1750 	skb->priority = TC_PRIO_CONTROL;
1751 	skb_reserve(skb, hlen);
1752 	skb_tailroom_reserve(skb, mtu, tlen);
1753 
1754 	if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
1755 		/* <draft-ietf-magma-mld-source-05.txt>:
1756 		 * use unspecified address as the source address
1757 		 * when a valid link-local address is not available.
1758 		 */
1759 		saddr = &in6addr_any;
1760 	} else
1761 		saddr = &addr_buf;
1762 
1763 	ip6_mc_hdr(sk, skb, dev, saddr, &mld2_all_mcr, NEXTHDR_HOP, 0);
1764 
1765 	skb_put_data(skb, ra, sizeof(ra));
1766 
1767 	skb_set_transport_header(skb, skb_tail_pointer(skb) - skb->data);
1768 	skb_put(skb, sizeof(*pmr));
1769 	pmr = (struct mld2_report *)skb_transport_header(skb);
1770 	pmr->mld2r_type = ICMPV6_MLD2_REPORT;
1771 	pmr->mld2r_resv1 = 0;
1772 	pmr->mld2r_cksum = 0;
1773 	pmr->mld2r_resv2 = 0;
1774 	pmr->mld2r_ngrec = 0;
1775 	return skb;
1776 }
1777 
mld_sendpack(struct sk_buff * skb)1778 static void mld_sendpack(struct sk_buff *skb)
1779 {
1780 	struct ipv6hdr *pip6 = ipv6_hdr(skb);
1781 	struct mld2_report *pmr =
1782 			      (struct mld2_report *)skb_transport_header(skb);
1783 	int payload_len, mldlen;
1784 	struct inet6_dev *idev;
1785 	struct net *net = dev_net(skb->dev);
1786 	int err;
1787 	struct flowi6 fl6;
1788 	struct dst_entry *dst;
1789 
1790 	rcu_read_lock();
1791 	idev = __in6_dev_get(skb->dev);
1792 	IP6_UPD_PO_STATS(net, idev, IPSTATS_MIB_OUT, skb->len);
1793 
1794 	payload_len = (skb_tail_pointer(skb) - skb_network_header(skb)) -
1795 		sizeof(*pip6);
1796 	mldlen = skb_tail_pointer(skb) - skb_transport_header(skb);
1797 	pip6->payload_len = htons(payload_len);
1798 
1799 	pmr->mld2r_cksum = csum_ipv6_magic(&pip6->saddr, &pip6->daddr, mldlen,
1800 					   IPPROTO_ICMPV6,
1801 					   csum_partial(skb_transport_header(skb),
1802 							mldlen, 0));
1803 
1804 	icmpv6_flow_init(net->ipv6.igmp_sk, &fl6, ICMPV6_MLD2_REPORT,
1805 			 &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
1806 			 skb->dev->ifindex);
1807 	dst = icmp6_dst_alloc(skb->dev, &fl6);
1808 
1809 	err = 0;
1810 	if (IS_ERR(dst)) {
1811 		err = PTR_ERR(dst);
1812 		dst = NULL;
1813 	}
1814 	skb_dst_set(skb, dst);
1815 	if (err)
1816 		goto err_out;
1817 
1818 	err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
1819 		      net, net->ipv6.igmp_sk, skb, NULL, skb->dev,
1820 		      dst_output);
1821 out:
1822 	if (!err) {
1823 		ICMP6MSGOUT_INC_STATS(net, idev, ICMPV6_MLD2_REPORT);
1824 		ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
1825 	} else {
1826 		IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
1827 	}
1828 
1829 	rcu_read_unlock();
1830 	return;
1831 
1832 err_out:
1833 	kfree_skb(skb);
1834 	goto out;
1835 }
1836 
grec_size(struct ifmcaddr6 * pmc,int type,int gdel,int sdel)1837 static int grec_size(struct ifmcaddr6 *pmc, int type, int gdel, int sdel)
1838 {
1839 	return sizeof(struct mld2_grec) + 16 * mld_scount(pmc,type,gdel,sdel);
1840 }
1841 
add_grhead(struct sk_buff * skb,struct ifmcaddr6 * pmc,int type,struct mld2_grec ** ppgr,unsigned int mtu)1842 static struct sk_buff *add_grhead(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1843 	int type, struct mld2_grec **ppgr, unsigned int mtu)
1844 {
1845 	struct mld2_report *pmr;
1846 	struct mld2_grec *pgr;
1847 
1848 	if (!skb) {
1849 		skb = mld_newpack(pmc->idev, mtu);
1850 		if (!skb)
1851 			return NULL;
1852 	}
1853 	pgr = skb_put(skb, sizeof(struct mld2_grec));
1854 	pgr->grec_type = type;
1855 	pgr->grec_auxwords = 0;
1856 	pgr->grec_nsrcs = 0;
1857 	pgr->grec_mca = pmc->mca_addr;	/* structure copy */
1858 	pmr = (struct mld2_report *)skb_transport_header(skb);
1859 	pmr->mld2r_ngrec = htons(ntohs(pmr->mld2r_ngrec)+1);
1860 	*ppgr = pgr;
1861 	return skb;
1862 }
1863 
1864 #define AVAILABLE(skb)	((skb) ? skb_availroom(skb) : 0)
1865 
1866 /* called with mc_lock */
add_grec(struct sk_buff * skb,struct ifmcaddr6 * pmc,int type,int gdeleted,int sdeleted,int crsend)1867 static struct sk_buff *add_grec(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1868 				int type, int gdeleted, int sdeleted,
1869 				int crsend)
1870 {
1871 	struct ip6_sf_list *psf, *psf_prev, *psf_next;
1872 	int scount, stotal, first, isquery, truncate;
1873 	struct ip6_sf_list __rcu **psf_list;
1874 	struct inet6_dev *idev = pmc->idev;
1875 	struct net_device *dev = idev->dev;
1876 	struct mld2_grec *pgr = NULL;
1877 	struct mld2_report *pmr;
1878 	unsigned int mtu;
1879 
1880 	if (pmc->mca_flags & MAF_NOREPORT)
1881 		return skb;
1882 
1883 	mtu = READ_ONCE(dev->mtu);
1884 	if (mtu < IPV6_MIN_MTU)
1885 		return skb;
1886 
1887 	isquery = type == MLD2_MODE_IS_INCLUDE ||
1888 		  type == MLD2_MODE_IS_EXCLUDE;
1889 	truncate = type == MLD2_MODE_IS_EXCLUDE ||
1890 		    type == MLD2_CHANGE_TO_EXCLUDE;
1891 
1892 	stotal = scount = 0;
1893 
1894 	psf_list = sdeleted ? &pmc->mca_tomb : &pmc->mca_sources;
1895 
1896 	if (!rcu_access_pointer(*psf_list))
1897 		goto empty_source;
1898 
1899 	pmr = skb ? (struct mld2_report *)skb_transport_header(skb) : NULL;
1900 
1901 	/* EX and TO_EX get a fresh packet, if needed */
1902 	if (truncate) {
1903 		if (pmr && pmr->mld2r_ngrec &&
1904 		    AVAILABLE(skb) < grec_size(pmc, type, gdeleted, sdeleted)) {
1905 			if (skb)
1906 				mld_sendpack(skb);
1907 			skb = mld_newpack(idev, mtu);
1908 		}
1909 	}
1910 	first = 1;
1911 	psf_prev = NULL;
1912 	for (psf = mc_dereference(*psf_list, idev);
1913 	     psf;
1914 	     psf = psf_next) {
1915 		struct in6_addr *psrc;
1916 
1917 		psf_next = mc_dereference(psf->sf_next, idev);
1918 
1919 		if (!is_in(pmc, psf, type, gdeleted, sdeleted) && !crsend) {
1920 			psf_prev = psf;
1921 			continue;
1922 		}
1923 
1924 		/* Based on RFC3810 6.1. Should not send source-list change
1925 		 * records when there is a filter mode change.
1926 		 */
1927 		if (((gdeleted && pmc->mca_sfmode == MCAST_EXCLUDE) ||
1928 		     (!gdeleted && pmc->mca_crcount)) &&
1929 		    (type == MLD2_ALLOW_NEW_SOURCES ||
1930 		     type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount)
1931 			goto decrease_sf_crcount;
1932 
1933 		/* clear marks on query responses */
1934 		if (isquery)
1935 			psf->sf_gsresp = 0;
1936 
1937 		if (AVAILABLE(skb) < sizeof(*psrc) +
1938 		    first*sizeof(struct mld2_grec)) {
1939 			if (truncate && !first)
1940 				break;	 /* truncate these */
1941 			if (pgr)
1942 				pgr->grec_nsrcs = htons(scount);
1943 			if (skb)
1944 				mld_sendpack(skb);
1945 			skb = mld_newpack(idev, mtu);
1946 			first = 1;
1947 			scount = 0;
1948 		}
1949 		if (first) {
1950 			skb = add_grhead(skb, pmc, type, &pgr, mtu);
1951 			first = 0;
1952 		}
1953 		if (!skb)
1954 			return NULL;
1955 		psrc = skb_put(skb, sizeof(*psrc));
1956 		*psrc = psf->sf_addr;
1957 		scount++; stotal++;
1958 		if ((type == MLD2_ALLOW_NEW_SOURCES ||
1959 		     type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount) {
1960 decrease_sf_crcount:
1961 			psf->sf_crcount--;
1962 			if ((sdeleted || gdeleted) && psf->sf_crcount == 0) {
1963 				if (psf_prev)
1964 					rcu_assign_pointer(psf_prev->sf_next,
1965 							   mc_dereference(psf->sf_next, idev));
1966 				else
1967 					rcu_assign_pointer(*psf_list,
1968 							   mc_dereference(psf->sf_next, idev));
1969 				kfree_rcu(psf, rcu);
1970 				continue;
1971 			}
1972 		}
1973 		psf_prev = psf;
1974 	}
1975 
1976 empty_source:
1977 	if (!stotal) {
1978 		if (type == MLD2_ALLOW_NEW_SOURCES ||
1979 		    type == MLD2_BLOCK_OLD_SOURCES)
1980 			return skb;
1981 		if (pmc->mca_crcount || isquery || crsend) {
1982 			/* make sure we have room for group header */
1983 			if (skb && AVAILABLE(skb) < sizeof(struct mld2_grec)) {
1984 				mld_sendpack(skb);
1985 				skb = NULL; /* add_grhead will get a new one */
1986 			}
1987 			skb = add_grhead(skb, pmc, type, &pgr, mtu);
1988 		}
1989 	}
1990 	if (pgr)
1991 		pgr->grec_nsrcs = htons(scount);
1992 
1993 	if (isquery)
1994 		pmc->mca_flags &= ~MAF_GSQUERY;	/* clear query state */
1995 	return skb;
1996 }
1997 
1998 /* called with mc_lock */
mld_send_report(struct inet6_dev * idev,struct ifmcaddr6 * pmc)1999 static void mld_send_report(struct inet6_dev *idev, struct ifmcaddr6 *pmc)
2000 {
2001 	struct sk_buff *skb = NULL;
2002 	int type;
2003 
2004 	if (!pmc) {
2005 		for_each_mc_mclock(idev, pmc) {
2006 			if (pmc->mca_flags & MAF_NOREPORT)
2007 				continue;
2008 			if (pmc->mca_sfcount[MCAST_EXCLUDE])
2009 				type = MLD2_MODE_IS_EXCLUDE;
2010 			else
2011 				type = MLD2_MODE_IS_INCLUDE;
2012 			skb = add_grec(skb, pmc, type, 0, 0, 0);
2013 		}
2014 	} else {
2015 		if (pmc->mca_sfcount[MCAST_EXCLUDE])
2016 			type = MLD2_MODE_IS_EXCLUDE;
2017 		else
2018 			type = MLD2_MODE_IS_INCLUDE;
2019 		skb = add_grec(skb, pmc, type, 0, 0, 0);
2020 	}
2021 	if (skb)
2022 		mld_sendpack(skb);
2023 }
2024 
2025 /*
2026  * remove zero-count source records from a source filter list
2027  * called with mc_lock
2028  */
mld_clear_zeros(struct ip6_sf_list __rcu ** ppsf,struct inet6_dev * idev)2029 static void mld_clear_zeros(struct ip6_sf_list __rcu **ppsf, struct inet6_dev *idev)
2030 {
2031 	struct ip6_sf_list *psf_prev, *psf_next, *psf;
2032 
2033 	psf_prev = NULL;
2034 	for (psf = mc_dereference(*ppsf, idev);
2035 	     psf;
2036 	     psf = psf_next) {
2037 		psf_next = mc_dereference(psf->sf_next, idev);
2038 		if (psf->sf_crcount == 0) {
2039 			if (psf_prev)
2040 				rcu_assign_pointer(psf_prev->sf_next,
2041 						   mc_dereference(psf->sf_next, idev));
2042 			else
2043 				rcu_assign_pointer(*ppsf,
2044 						   mc_dereference(psf->sf_next, idev));
2045 			kfree_rcu(psf, rcu);
2046 		} else {
2047 			psf_prev = psf;
2048 		}
2049 	}
2050 }
2051 
2052 /* called with mc_lock */
mld_send_cr(struct inet6_dev * idev)2053 static void mld_send_cr(struct inet6_dev *idev)
2054 {
2055 	struct ifmcaddr6 *pmc, *pmc_prev, *pmc_next;
2056 	struct sk_buff *skb = NULL;
2057 	int type, dtype;
2058 
2059 	/* deleted MCA's */
2060 	pmc_prev = NULL;
2061 	for (pmc = mc_dereference(idev->mc_tomb, idev);
2062 	     pmc;
2063 	     pmc = pmc_next) {
2064 		pmc_next = mc_dereference(pmc->next, idev);
2065 		if (pmc->mca_sfmode == MCAST_INCLUDE) {
2066 			type = MLD2_BLOCK_OLD_SOURCES;
2067 			dtype = MLD2_BLOCK_OLD_SOURCES;
2068 			skb = add_grec(skb, pmc, type, 1, 0, 0);
2069 			skb = add_grec(skb, pmc, dtype, 1, 1, 0);
2070 		}
2071 		if (pmc->mca_crcount) {
2072 			if (pmc->mca_sfmode == MCAST_EXCLUDE) {
2073 				type = MLD2_CHANGE_TO_INCLUDE;
2074 				skb = add_grec(skb, pmc, type, 1, 0, 0);
2075 			}
2076 			pmc->mca_crcount--;
2077 			if (pmc->mca_crcount == 0) {
2078 				mld_clear_zeros(&pmc->mca_tomb, idev);
2079 				mld_clear_zeros(&pmc->mca_sources, idev);
2080 			}
2081 		}
2082 		if (pmc->mca_crcount == 0 &&
2083 		    !rcu_access_pointer(pmc->mca_tomb) &&
2084 		    !rcu_access_pointer(pmc->mca_sources)) {
2085 			if (pmc_prev)
2086 				rcu_assign_pointer(pmc_prev->next, pmc_next);
2087 			else
2088 				rcu_assign_pointer(idev->mc_tomb, pmc_next);
2089 			in6_dev_put(pmc->idev);
2090 			kfree_rcu(pmc, rcu);
2091 		} else
2092 			pmc_prev = pmc;
2093 	}
2094 
2095 	/* change recs */
2096 	for_each_mc_mclock(idev, pmc) {
2097 		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2098 			type = MLD2_BLOCK_OLD_SOURCES;
2099 			dtype = MLD2_ALLOW_NEW_SOURCES;
2100 		} else {
2101 			type = MLD2_ALLOW_NEW_SOURCES;
2102 			dtype = MLD2_BLOCK_OLD_SOURCES;
2103 		}
2104 		skb = add_grec(skb, pmc, type, 0, 0, 0);
2105 		skb = add_grec(skb, pmc, dtype, 0, 1, 0);	/* deleted sources */
2106 
2107 		/* filter mode changes */
2108 		if (pmc->mca_crcount) {
2109 			if (pmc->mca_sfmode == MCAST_EXCLUDE)
2110 				type = MLD2_CHANGE_TO_EXCLUDE;
2111 			else
2112 				type = MLD2_CHANGE_TO_INCLUDE;
2113 			skb = add_grec(skb, pmc, type, 0, 0, 0);
2114 			pmc->mca_crcount--;
2115 		}
2116 	}
2117 	if (!skb)
2118 		return;
2119 	(void) mld_sendpack(skb);
2120 }
2121 
igmp6_send(struct in6_addr * addr,struct net_device * dev,int type)2122 static void igmp6_send(struct in6_addr *addr, struct net_device *dev, int type)
2123 {
2124 	struct net *net = dev_net(dev);
2125 	struct sock *sk = net->ipv6.igmp_sk;
2126 	struct inet6_dev *idev;
2127 	struct sk_buff *skb;
2128 	struct mld_msg *hdr;
2129 	const struct in6_addr *snd_addr, *saddr;
2130 	struct in6_addr addr_buf;
2131 	int hlen = LL_RESERVED_SPACE(dev);
2132 	int tlen = dev->needed_tailroom;
2133 	int err, len, payload_len, full_len;
2134 	u8 ra[8] = { IPPROTO_ICMPV6, 0,
2135 		     IPV6_TLV_ROUTERALERT, 2, 0, 0,
2136 		     IPV6_TLV_PADN, 0 };
2137 	struct flowi6 fl6;
2138 	struct dst_entry *dst;
2139 
2140 	if (type == ICMPV6_MGM_REDUCTION)
2141 		snd_addr = &in6addr_linklocal_allrouters;
2142 	else
2143 		snd_addr = addr;
2144 
2145 	len = sizeof(struct icmp6hdr) + sizeof(struct in6_addr);
2146 	payload_len = len + sizeof(ra);
2147 	full_len = sizeof(struct ipv6hdr) + payload_len;
2148 
2149 	rcu_read_lock();
2150 	IP6_UPD_PO_STATS(net, __in6_dev_get(dev),
2151 		      IPSTATS_MIB_OUT, full_len);
2152 	rcu_read_unlock();
2153 
2154 	skb = sock_alloc_send_skb(sk, hlen + tlen + full_len, 1, &err);
2155 
2156 	if (!skb) {
2157 		rcu_read_lock();
2158 		IP6_INC_STATS(net, __in6_dev_get(dev),
2159 			      IPSTATS_MIB_OUTDISCARDS);
2160 		rcu_read_unlock();
2161 		return;
2162 	}
2163 	skb->priority = TC_PRIO_CONTROL;
2164 	skb_reserve(skb, hlen);
2165 
2166 	if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
2167 		/* <draft-ietf-magma-mld-source-05.txt>:
2168 		 * use unspecified address as the source address
2169 		 * when a valid link-local address is not available.
2170 		 */
2171 		saddr = &in6addr_any;
2172 	} else
2173 		saddr = &addr_buf;
2174 
2175 	ip6_mc_hdr(sk, skb, dev, saddr, snd_addr, NEXTHDR_HOP, payload_len);
2176 
2177 	skb_put_data(skb, ra, sizeof(ra));
2178 
2179 	hdr = skb_put_zero(skb, sizeof(struct mld_msg));
2180 	hdr->mld_type = type;
2181 	hdr->mld_mca = *addr;
2182 
2183 	hdr->mld_cksum = csum_ipv6_magic(saddr, snd_addr, len,
2184 					 IPPROTO_ICMPV6,
2185 					 csum_partial(hdr, len, 0));
2186 
2187 	rcu_read_lock();
2188 	idev = __in6_dev_get(skb->dev);
2189 
2190 	icmpv6_flow_init(sk, &fl6, type,
2191 			 &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
2192 			 skb->dev->ifindex);
2193 	dst = icmp6_dst_alloc(skb->dev, &fl6);
2194 	if (IS_ERR(dst)) {
2195 		err = PTR_ERR(dst);
2196 		goto err_out;
2197 	}
2198 
2199 	skb_dst_set(skb, dst);
2200 	err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
2201 		      net, sk, skb, NULL, skb->dev,
2202 		      dst_output);
2203 out:
2204 	if (!err) {
2205 		ICMP6MSGOUT_INC_STATS(net, idev, type);
2206 		ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
2207 	} else
2208 		IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
2209 
2210 	rcu_read_unlock();
2211 	return;
2212 
2213 err_out:
2214 	kfree_skb(skb);
2215 	goto out;
2216 }
2217 
2218 /* called with mc_lock */
mld_send_initial_cr(struct inet6_dev * idev)2219 static void mld_send_initial_cr(struct inet6_dev *idev)
2220 {
2221 	struct sk_buff *skb;
2222 	struct ifmcaddr6 *pmc;
2223 	int type;
2224 
2225 	if (mld_in_v1_mode(idev))
2226 		return;
2227 
2228 	skb = NULL;
2229 	for_each_mc_mclock(idev, pmc) {
2230 		if (pmc->mca_sfcount[MCAST_EXCLUDE])
2231 			type = MLD2_CHANGE_TO_EXCLUDE;
2232 		else
2233 			type = MLD2_ALLOW_NEW_SOURCES;
2234 		skb = add_grec(skb, pmc, type, 0, 0, 1);
2235 	}
2236 	if (skb)
2237 		mld_sendpack(skb);
2238 }
2239 
ipv6_mc_dad_complete(struct inet6_dev * idev)2240 void ipv6_mc_dad_complete(struct inet6_dev *idev)
2241 {
2242 	mutex_lock(&idev->mc_lock);
2243 	idev->mc_dad_count = idev->mc_qrv;
2244 	if (idev->mc_dad_count) {
2245 		mld_send_initial_cr(idev);
2246 		idev->mc_dad_count--;
2247 		if (idev->mc_dad_count)
2248 			mld_dad_start_work(idev,
2249 					   unsolicited_report_interval(idev));
2250 	}
2251 	mutex_unlock(&idev->mc_lock);
2252 }
2253 
mld_dad_work(struct work_struct * work)2254 static void mld_dad_work(struct work_struct *work)
2255 {
2256 	struct inet6_dev *idev = container_of(to_delayed_work(work),
2257 					      struct inet6_dev,
2258 					      mc_dad_work);
2259 	mutex_lock(&idev->mc_lock);
2260 	mld_send_initial_cr(idev);
2261 	if (idev->mc_dad_count) {
2262 		idev->mc_dad_count--;
2263 		if (idev->mc_dad_count)
2264 			mld_dad_start_work(idev,
2265 					   unsolicited_report_interval(idev));
2266 	}
2267 	mutex_unlock(&idev->mc_lock);
2268 	in6_dev_put(idev);
2269 }
2270 
2271 /* called with mc_lock */
ip6_mc_del1_src(struct ifmcaddr6 * pmc,int sfmode,const struct in6_addr * psfsrc)2272 static int ip6_mc_del1_src(struct ifmcaddr6 *pmc, int sfmode,
2273 	const struct in6_addr *psfsrc)
2274 {
2275 	struct ip6_sf_list *psf, *psf_prev;
2276 	int rv = 0;
2277 
2278 	psf_prev = NULL;
2279 	for_each_psf_mclock(pmc, psf) {
2280 		if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2281 			break;
2282 		psf_prev = psf;
2283 	}
2284 	if (!psf || psf->sf_count[sfmode] == 0) {
2285 		/* source filter not found, or count wrong =>  bug */
2286 		return -ESRCH;
2287 	}
2288 	psf->sf_count[sfmode]--;
2289 	if (!psf->sf_count[MCAST_INCLUDE] && !psf->sf_count[MCAST_EXCLUDE]) {
2290 		struct inet6_dev *idev = pmc->idev;
2291 
2292 		/* no more filters for this source */
2293 		if (psf_prev)
2294 			rcu_assign_pointer(psf_prev->sf_next,
2295 					   mc_dereference(psf->sf_next, idev));
2296 		else
2297 			rcu_assign_pointer(pmc->mca_sources,
2298 					   mc_dereference(psf->sf_next, idev));
2299 
2300 		if (psf->sf_oldin && !(pmc->mca_flags & MAF_NOREPORT) &&
2301 		    !mld_in_v1_mode(idev)) {
2302 			psf->sf_crcount = idev->mc_qrv;
2303 			rcu_assign_pointer(psf->sf_next,
2304 					   mc_dereference(pmc->mca_tomb, idev));
2305 			rcu_assign_pointer(pmc->mca_tomb, psf);
2306 			rv = 1;
2307 		} else {
2308 			kfree_rcu(psf, rcu);
2309 		}
2310 	}
2311 	return rv;
2312 }
2313 
2314 /* 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)2315 static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2316 			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
2317 			  int delta)
2318 {
2319 	struct ifmcaddr6 *pmc;
2320 	int	changerec = 0;
2321 	int	i, err;
2322 
2323 	if (!idev)
2324 		return -ENODEV;
2325 
2326 	for_each_mc_mclock(idev, pmc) {
2327 		if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2328 			break;
2329 	}
2330 	if (!pmc)
2331 		return -ESRCH;
2332 
2333 	sf_markstate(pmc);
2334 	if (!delta) {
2335 		if (!pmc->mca_sfcount[sfmode])
2336 			return -EINVAL;
2337 
2338 		pmc->mca_sfcount[sfmode]--;
2339 	}
2340 	err = 0;
2341 	for (i = 0; i < sfcount; i++) {
2342 		int rv = ip6_mc_del1_src(pmc, sfmode, &psfsrc[i]);
2343 
2344 		changerec |= rv > 0;
2345 		if (!err && rv < 0)
2346 			err = rv;
2347 	}
2348 	if (pmc->mca_sfmode == MCAST_EXCLUDE &&
2349 	    pmc->mca_sfcount[MCAST_EXCLUDE] == 0 &&
2350 	    pmc->mca_sfcount[MCAST_INCLUDE]) {
2351 		struct ip6_sf_list *psf;
2352 
2353 		/* filter mode change */
2354 		pmc->mca_sfmode = MCAST_INCLUDE;
2355 		pmc->mca_crcount = idev->mc_qrv;
2356 		idev->mc_ifc_count = pmc->mca_crcount;
2357 		for_each_psf_mclock(pmc, psf)
2358 			psf->sf_crcount = 0;
2359 		mld_ifc_event(pmc->idev);
2360 	} else if (sf_setstate(pmc) || changerec) {
2361 		mld_ifc_event(pmc->idev);
2362 	}
2363 
2364 	return err;
2365 }
2366 
2367 /*
2368  * Add multicast single-source filter to the interface list
2369  * called with mc_lock
2370  */
ip6_mc_add1_src(struct ifmcaddr6 * pmc,int sfmode,const struct in6_addr * psfsrc)2371 static int ip6_mc_add1_src(struct ifmcaddr6 *pmc, int sfmode,
2372 	const struct in6_addr *psfsrc)
2373 {
2374 	struct ip6_sf_list *psf, *psf_prev;
2375 
2376 	psf_prev = NULL;
2377 	for_each_psf_mclock(pmc, psf) {
2378 		if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2379 			break;
2380 		psf_prev = psf;
2381 	}
2382 	if (!psf) {
2383 		psf = kzalloc(sizeof(*psf), GFP_KERNEL);
2384 		if (!psf)
2385 			return -ENOBUFS;
2386 
2387 		psf->sf_addr = *psfsrc;
2388 		if (psf_prev) {
2389 			rcu_assign_pointer(psf_prev->sf_next, psf);
2390 		} else {
2391 			rcu_assign_pointer(pmc->mca_sources, psf);
2392 		}
2393 	}
2394 	psf->sf_count[sfmode]++;
2395 	return 0;
2396 }
2397 
2398 /* called with mc_lock */
sf_markstate(struct ifmcaddr6 * pmc)2399 static void sf_markstate(struct ifmcaddr6 *pmc)
2400 {
2401 	struct ip6_sf_list *psf;
2402 	int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2403 
2404 	for_each_psf_mclock(pmc, psf) {
2405 		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2406 			psf->sf_oldin = mca_xcount ==
2407 				psf->sf_count[MCAST_EXCLUDE] &&
2408 				!psf->sf_count[MCAST_INCLUDE];
2409 		} else {
2410 			psf->sf_oldin = psf->sf_count[MCAST_INCLUDE] != 0;
2411 		}
2412 	}
2413 }
2414 
2415 /* called with mc_lock */
sf_setstate(struct ifmcaddr6 * pmc)2416 static int sf_setstate(struct ifmcaddr6 *pmc)
2417 {
2418 	struct ip6_sf_list *psf, *dpsf;
2419 	int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2420 	int qrv = pmc->idev->mc_qrv;
2421 	int new_in, rv;
2422 
2423 	rv = 0;
2424 	for_each_psf_mclock(pmc, psf) {
2425 		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2426 			new_in = mca_xcount == psf->sf_count[MCAST_EXCLUDE] &&
2427 				!psf->sf_count[MCAST_INCLUDE];
2428 		} else
2429 			new_in = psf->sf_count[MCAST_INCLUDE] != 0;
2430 		if (new_in) {
2431 			if (!psf->sf_oldin) {
2432 				struct ip6_sf_list *prev = NULL;
2433 
2434 				for_each_psf_tomb(pmc, dpsf) {
2435 					if (ipv6_addr_equal(&dpsf->sf_addr,
2436 					    &psf->sf_addr))
2437 						break;
2438 					prev = dpsf;
2439 				}
2440 				if (dpsf) {
2441 					if (prev)
2442 						rcu_assign_pointer(prev->sf_next,
2443 								   mc_dereference(dpsf->sf_next,
2444 										  pmc->idev));
2445 					else
2446 						rcu_assign_pointer(pmc->mca_tomb,
2447 								   mc_dereference(dpsf->sf_next,
2448 										  pmc->idev));
2449 					kfree_rcu(dpsf, rcu);
2450 				}
2451 				psf->sf_crcount = qrv;
2452 				rv++;
2453 			}
2454 		} else if (psf->sf_oldin) {
2455 			psf->sf_crcount = 0;
2456 			/*
2457 			 * add or update "delete" records if an active filter
2458 			 * is now inactive
2459 			 */
2460 
2461 			for_each_psf_tomb(pmc, dpsf)
2462 				if (ipv6_addr_equal(&dpsf->sf_addr,
2463 				    &psf->sf_addr))
2464 					break;
2465 			if (!dpsf) {
2466 				dpsf = kmalloc(sizeof(*dpsf), GFP_KERNEL);
2467 				if (!dpsf)
2468 					continue;
2469 				*dpsf = *psf;
2470 				rcu_assign_pointer(dpsf->sf_next,
2471 						   mc_dereference(pmc->mca_tomb, pmc->idev));
2472 				rcu_assign_pointer(pmc->mca_tomb, dpsf);
2473 			}
2474 			dpsf->sf_crcount = qrv;
2475 			rv++;
2476 		}
2477 	}
2478 	return rv;
2479 }
2480 
2481 /*
2482  * Add multicast source filter list to the interface list
2483  * called with mc_lock
2484  */
ip6_mc_add_src(struct inet6_dev * idev,const struct in6_addr * pmca,int sfmode,int sfcount,const struct in6_addr * psfsrc,int delta)2485 static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2486 			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
2487 			  int delta)
2488 {
2489 	struct ifmcaddr6 *pmc;
2490 	int	isexclude;
2491 	int	i, err;
2492 
2493 	if (!idev)
2494 		return -ENODEV;
2495 
2496 	for_each_mc_mclock(idev, pmc) {
2497 		if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2498 			break;
2499 	}
2500 	if (!pmc)
2501 		return -ESRCH;
2502 
2503 	sf_markstate(pmc);
2504 	isexclude = pmc->mca_sfmode == MCAST_EXCLUDE;
2505 	if (!delta)
2506 		pmc->mca_sfcount[sfmode]++;
2507 	err = 0;
2508 	for (i = 0; i < sfcount; i++) {
2509 		err = ip6_mc_add1_src(pmc, sfmode, &psfsrc[i]);
2510 		if (err)
2511 			break;
2512 	}
2513 	if (err) {
2514 		int j;
2515 
2516 		if (!delta)
2517 			pmc->mca_sfcount[sfmode]--;
2518 		for (j = 0; j < i; j++)
2519 			ip6_mc_del1_src(pmc, sfmode, &psfsrc[j]);
2520 	} else if (isexclude != (pmc->mca_sfcount[MCAST_EXCLUDE] != 0)) {
2521 		struct ip6_sf_list *psf;
2522 
2523 		/* filter mode change */
2524 		if (pmc->mca_sfcount[MCAST_EXCLUDE])
2525 			pmc->mca_sfmode = MCAST_EXCLUDE;
2526 		else if (pmc->mca_sfcount[MCAST_INCLUDE])
2527 			pmc->mca_sfmode = MCAST_INCLUDE;
2528 		/* else no filters; keep old mode for reports */
2529 
2530 		pmc->mca_crcount = idev->mc_qrv;
2531 		idev->mc_ifc_count = pmc->mca_crcount;
2532 		for_each_psf_mclock(pmc, psf)
2533 			psf->sf_crcount = 0;
2534 		mld_ifc_event(idev);
2535 	} else if (sf_setstate(pmc)) {
2536 		mld_ifc_event(idev);
2537 	}
2538 	return err;
2539 }
2540 
2541 /* called with mc_lock */
ip6_mc_clear_src(struct ifmcaddr6 * pmc)2542 static void ip6_mc_clear_src(struct ifmcaddr6 *pmc)
2543 {
2544 	struct ip6_sf_list *psf, *nextpsf;
2545 
2546 	for (psf = mc_dereference(pmc->mca_tomb, pmc->idev);
2547 	     psf;
2548 	     psf = nextpsf) {
2549 		nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2550 		kfree_rcu(psf, rcu);
2551 	}
2552 	RCU_INIT_POINTER(pmc->mca_tomb, NULL);
2553 	for (psf = mc_dereference(pmc->mca_sources, pmc->idev);
2554 	     psf;
2555 	     psf = nextpsf) {
2556 		nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2557 		kfree_rcu(psf, rcu);
2558 	}
2559 	RCU_INIT_POINTER(pmc->mca_sources, NULL);
2560 	pmc->mca_sfmode = MCAST_EXCLUDE;
2561 	pmc->mca_sfcount[MCAST_INCLUDE] = 0;
2562 	pmc->mca_sfcount[MCAST_EXCLUDE] = 1;
2563 }
2564 
2565 /* called with mc_lock */
igmp6_join_group(struct ifmcaddr6 * ma)2566 static void igmp6_join_group(struct ifmcaddr6 *ma)
2567 {
2568 	unsigned long delay;
2569 
2570 	if (ma->mca_flags & MAF_NOREPORT)
2571 		return;
2572 
2573 	igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2574 
2575 	delay = get_random_u32_below(unsolicited_report_interval(ma->idev));
2576 
2577 	if (cancel_delayed_work(&ma->mca_work)) {
2578 		refcount_dec(&ma->mca_refcnt);
2579 		delay = ma->mca_work.timer.expires - jiffies;
2580 	}
2581 
2582 	if (!mod_delayed_work(mld_wq, &ma->mca_work, delay))
2583 		refcount_inc(&ma->mca_refcnt);
2584 	ma->mca_flags |= MAF_TIMER_RUNNING | MAF_LAST_REPORTER;
2585 }
2586 
ip6_mc_leave_src(struct sock * sk,struct ipv6_mc_socklist * iml,struct inet6_dev * idev)2587 static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
2588 			    struct inet6_dev *idev)
2589 {
2590 	struct ip6_sf_socklist *psl;
2591 	int err;
2592 
2593 	psl = sock_dereference(iml->sflist, sk);
2594 
2595 	if (idev)
2596 		mutex_lock(&idev->mc_lock);
2597 
2598 	if (!psl) {
2599 		/* any-source empty exclude case */
2600 		err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode, 0, NULL, 0);
2601 	} else {
2602 		err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode,
2603 				     psl->sl_count, psl->sl_addr, 0);
2604 		RCU_INIT_POINTER(iml->sflist, NULL);
2605 		atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
2606 			   &sk->sk_omem_alloc);
2607 		kfree_rcu(psl, rcu);
2608 	}
2609 
2610 	if (idev)
2611 		mutex_unlock(&idev->mc_lock);
2612 
2613 	return err;
2614 }
2615 
2616 /* called with mc_lock */
igmp6_leave_group(struct ifmcaddr6 * ma)2617 static void igmp6_leave_group(struct ifmcaddr6 *ma)
2618 {
2619 	if (mld_in_v1_mode(ma->idev)) {
2620 		if (ma->mca_flags & MAF_LAST_REPORTER) {
2621 			igmp6_send(&ma->mca_addr, ma->idev->dev,
2622 				ICMPV6_MGM_REDUCTION);
2623 		}
2624 	} else {
2625 		mld_add_delrec(ma->idev, ma);
2626 		mld_ifc_event(ma->idev);
2627 	}
2628 }
2629 
mld_gq_work(struct work_struct * work)2630 static void mld_gq_work(struct work_struct *work)
2631 {
2632 	struct inet6_dev *idev = container_of(to_delayed_work(work),
2633 					      struct inet6_dev,
2634 					      mc_gq_work);
2635 
2636 	mutex_lock(&idev->mc_lock);
2637 	mld_send_report(idev, NULL);
2638 	idev->mc_gq_running = 0;
2639 	mutex_unlock(&idev->mc_lock);
2640 
2641 	in6_dev_put(idev);
2642 }
2643 
mld_ifc_work(struct work_struct * work)2644 static void mld_ifc_work(struct work_struct *work)
2645 {
2646 	struct inet6_dev *idev = container_of(to_delayed_work(work),
2647 					      struct inet6_dev,
2648 					      mc_ifc_work);
2649 
2650 	mutex_lock(&idev->mc_lock);
2651 	mld_send_cr(idev);
2652 
2653 	if (idev->mc_ifc_count) {
2654 		idev->mc_ifc_count--;
2655 		if (idev->mc_ifc_count)
2656 			mld_ifc_start_work(idev,
2657 					   unsolicited_report_interval(idev));
2658 	}
2659 	mutex_unlock(&idev->mc_lock);
2660 	in6_dev_put(idev);
2661 }
2662 
2663 /* called with mc_lock */
mld_ifc_event(struct inet6_dev * idev)2664 static void mld_ifc_event(struct inet6_dev *idev)
2665 {
2666 	if (mld_in_v1_mode(idev))
2667 		return;
2668 
2669 	idev->mc_ifc_count = idev->mc_qrv;
2670 	mld_ifc_start_work(idev, 1);
2671 }
2672 
mld_mca_work(struct work_struct * work)2673 static void mld_mca_work(struct work_struct *work)
2674 {
2675 	struct ifmcaddr6 *ma = container_of(to_delayed_work(work),
2676 					    struct ifmcaddr6, mca_work);
2677 
2678 	mutex_lock(&ma->idev->mc_lock);
2679 	if (mld_in_v1_mode(ma->idev))
2680 		igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2681 	else
2682 		mld_send_report(ma->idev, ma);
2683 	ma->mca_flags |=  MAF_LAST_REPORTER;
2684 	ma->mca_flags &= ~MAF_TIMER_RUNNING;
2685 	mutex_unlock(&ma->idev->mc_lock);
2686 
2687 	ma_put(ma);
2688 }
2689 
2690 /* Device changing type */
2691 
ipv6_mc_unmap(struct inet6_dev * idev)2692 void ipv6_mc_unmap(struct inet6_dev *idev)
2693 {
2694 	struct ifmcaddr6 *i;
2695 
2696 	/* Install multicast list, except for all-nodes (already installed) */
2697 
2698 	mutex_lock(&idev->mc_lock);
2699 	for_each_mc_mclock(idev, i)
2700 		igmp6_group_dropped(i);
2701 	mutex_unlock(&idev->mc_lock);
2702 }
2703 
ipv6_mc_remap(struct inet6_dev * idev)2704 void ipv6_mc_remap(struct inet6_dev *idev)
2705 {
2706 	ipv6_mc_up(idev);
2707 }
2708 
2709 /* Device going down */
ipv6_mc_down(struct inet6_dev * idev)2710 void ipv6_mc_down(struct inet6_dev *idev)
2711 {
2712 	struct ifmcaddr6 *i;
2713 
2714 	mutex_lock(&idev->mc_lock);
2715 	/* Withdraw multicast list */
2716 	for_each_mc_mclock(idev, i)
2717 		igmp6_group_dropped(i);
2718 	mutex_unlock(&idev->mc_lock);
2719 
2720 	/* Should stop work after group drop. or we will
2721 	 * start work again in mld_ifc_event()
2722 	 */
2723 	synchronize_net();
2724 	mld_query_stop_work(idev);
2725 	mld_report_stop_work(idev);
2726 
2727 	mutex_lock(&idev->mc_lock);
2728 	mld_ifc_stop_work(idev);
2729 	mld_gq_stop_work(idev);
2730 	mutex_unlock(&idev->mc_lock);
2731 
2732 	mld_dad_stop_work(idev);
2733 }
2734 
ipv6_mc_reset(struct inet6_dev * idev)2735 static void ipv6_mc_reset(struct inet6_dev *idev)
2736 {
2737 	idev->mc_qrv = sysctl_mld_qrv;
2738 	idev->mc_qi = MLD_QI_DEFAULT;
2739 	idev->mc_qri = MLD_QRI_DEFAULT;
2740 	idev->mc_v1_seen = 0;
2741 	idev->mc_maxdelay = unsolicited_report_interval(idev);
2742 }
2743 
2744 /* Device going up */
2745 
ipv6_mc_up(struct inet6_dev * idev)2746 void ipv6_mc_up(struct inet6_dev *idev)
2747 {
2748 	struct ifmcaddr6 *i;
2749 
2750 	/* Install multicast list, except for all-nodes (already installed) */
2751 
2752 	ipv6_mc_reset(idev);
2753 	mutex_lock(&idev->mc_lock);
2754 	for_each_mc_mclock(idev, i) {
2755 		mld_del_delrec(idev, i);
2756 		igmp6_group_added(i);
2757 	}
2758 	mutex_unlock(&idev->mc_lock);
2759 }
2760 
2761 /* IPv6 device initialization. */
2762 
ipv6_mc_init_dev(struct inet6_dev * idev)2763 void ipv6_mc_init_dev(struct inet6_dev *idev)
2764 {
2765 	idev->mc_gq_running = 0;
2766 	INIT_DELAYED_WORK(&idev->mc_gq_work, mld_gq_work);
2767 	RCU_INIT_POINTER(idev->mc_tomb, NULL);
2768 	idev->mc_ifc_count = 0;
2769 	INIT_DELAYED_WORK(&idev->mc_ifc_work, mld_ifc_work);
2770 	INIT_DELAYED_WORK(&idev->mc_dad_work, mld_dad_work);
2771 	INIT_DELAYED_WORK(&idev->mc_query_work, mld_query_work);
2772 	INIT_DELAYED_WORK(&idev->mc_report_work, mld_report_work);
2773 	skb_queue_head_init(&idev->mc_query_queue);
2774 	skb_queue_head_init(&idev->mc_report_queue);
2775 	spin_lock_init(&idev->mc_query_lock);
2776 	spin_lock_init(&idev->mc_report_lock);
2777 	mutex_init(&idev->mc_lock);
2778 	ipv6_mc_reset(idev);
2779 }
2780 
2781 /*
2782  *	Device is about to be destroyed: clean up.
2783  */
2784 
ipv6_mc_destroy_dev(struct inet6_dev * idev)2785 void ipv6_mc_destroy_dev(struct inet6_dev *idev)
2786 {
2787 	struct ifmcaddr6 *i;
2788 
2789 	/* Deactivate works */
2790 	ipv6_mc_down(idev);
2791 	mutex_lock(&idev->mc_lock);
2792 	mld_clear_delrec(idev);
2793 	mutex_unlock(&idev->mc_lock);
2794 	mld_clear_query(idev);
2795 	mld_clear_report(idev);
2796 
2797 	/* Delete all-nodes address. */
2798 	/* We cannot call ipv6_dev_mc_dec() directly, our caller in
2799 	 * addrconf.c has NULL'd out dev->ip6_ptr so in6_dev_get() will
2800 	 * fail.
2801 	 */
2802 	__ipv6_dev_mc_dec(idev, &in6addr_linklocal_allnodes);
2803 
2804 	if (idev->cnf.forwarding)
2805 		__ipv6_dev_mc_dec(idev, &in6addr_linklocal_allrouters);
2806 
2807 	mutex_lock(&idev->mc_lock);
2808 	while ((i = mc_dereference(idev->mc_list, idev))) {
2809 		rcu_assign_pointer(idev->mc_list, mc_dereference(i->next, idev));
2810 
2811 		ip6_mc_clear_src(i);
2812 		ma_put(i);
2813 	}
2814 	mutex_unlock(&idev->mc_lock);
2815 }
2816 
ipv6_mc_rejoin_groups(struct inet6_dev * idev)2817 static void ipv6_mc_rejoin_groups(struct inet6_dev *idev)
2818 {
2819 	struct ifmcaddr6 *pmc;
2820 
2821 	ASSERT_RTNL();
2822 
2823 	mutex_lock(&idev->mc_lock);
2824 	if (mld_in_v1_mode(idev)) {
2825 		for_each_mc_mclock(idev, pmc)
2826 			igmp6_join_group(pmc);
2827 	} else {
2828 		mld_send_report(idev, NULL);
2829 	}
2830 	mutex_unlock(&idev->mc_lock);
2831 }
2832 
ipv6_mc_netdev_event(struct notifier_block * this,unsigned long event,void * ptr)2833 static int ipv6_mc_netdev_event(struct notifier_block *this,
2834 				unsigned long event,
2835 				void *ptr)
2836 {
2837 	struct net_device *dev = netdev_notifier_info_to_dev(ptr);
2838 	struct inet6_dev *idev = __in6_dev_get(dev);
2839 
2840 	switch (event) {
2841 	case NETDEV_RESEND_IGMP:
2842 		if (idev)
2843 			ipv6_mc_rejoin_groups(idev);
2844 		break;
2845 	default:
2846 		break;
2847 	}
2848 
2849 	return NOTIFY_DONE;
2850 }
2851 
2852 static struct notifier_block igmp6_netdev_notifier = {
2853 	.notifier_call = ipv6_mc_netdev_event,
2854 };
2855 
2856 #ifdef CONFIG_PROC_FS
2857 struct igmp6_mc_iter_state {
2858 	struct seq_net_private p;
2859 	struct net_device *dev;
2860 	struct inet6_dev *idev;
2861 };
2862 
2863 #define igmp6_mc_seq_private(seq)	((struct igmp6_mc_iter_state *)(seq)->private)
2864 
igmp6_mc_get_first(struct seq_file * seq)2865 static inline struct ifmcaddr6 *igmp6_mc_get_first(struct seq_file *seq)
2866 {
2867 	struct ifmcaddr6 *im = NULL;
2868 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2869 	struct net *net = seq_file_net(seq);
2870 
2871 	state->idev = NULL;
2872 	for_each_netdev_rcu(net, state->dev) {
2873 		struct inet6_dev *idev;
2874 		idev = __in6_dev_get(state->dev);
2875 		if (!idev)
2876 			continue;
2877 
2878 		im = rcu_dereference(idev->mc_list);
2879 		if (im) {
2880 			state->idev = idev;
2881 			break;
2882 		}
2883 	}
2884 	return im;
2885 }
2886 
igmp6_mc_get_next(struct seq_file * seq,struct ifmcaddr6 * im)2887 static struct ifmcaddr6 *igmp6_mc_get_next(struct seq_file *seq, struct ifmcaddr6 *im)
2888 {
2889 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2890 
2891 	im = rcu_dereference(im->next);
2892 	while (!im) {
2893 		state->dev = next_net_device_rcu(state->dev);
2894 		if (!state->dev) {
2895 			state->idev = NULL;
2896 			break;
2897 		}
2898 		state->idev = __in6_dev_get(state->dev);
2899 		if (!state->idev)
2900 			continue;
2901 		im = rcu_dereference(state->idev->mc_list);
2902 	}
2903 	return im;
2904 }
2905 
igmp6_mc_get_idx(struct seq_file * seq,loff_t pos)2906 static struct ifmcaddr6 *igmp6_mc_get_idx(struct seq_file *seq, loff_t pos)
2907 {
2908 	struct ifmcaddr6 *im = igmp6_mc_get_first(seq);
2909 	if (im)
2910 		while (pos && (im = igmp6_mc_get_next(seq, im)) != NULL)
2911 			--pos;
2912 	return pos ? NULL : im;
2913 }
2914 
igmp6_mc_seq_start(struct seq_file * seq,loff_t * pos)2915 static void *igmp6_mc_seq_start(struct seq_file *seq, loff_t *pos)
2916 	__acquires(RCU)
2917 {
2918 	rcu_read_lock();
2919 	return igmp6_mc_get_idx(seq, *pos);
2920 }
2921 
igmp6_mc_seq_next(struct seq_file * seq,void * v,loff_t * pos)2922 static void *igmp6_mc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2923 {
2924 	struct ifmcaddr6 *im = igmp6_mc_get_next(seq, v);
2925 
2926 	++*pos;
2927 	return im;
2928 }
2929 
igmp6_mc_seq_stop(struct seq_file * seq,void * v)2930 static void igmp6_mc_seq_stop(struct seq_file *seq, void *v)
2931 	__releases(RCU)
2932 {
2933 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2934 
2935 	if (likely(state->idev))
2936 		state->idev = NULL;
2937 	state->dev = NULL;
2938 	rcu_read_unlock();
2939 }
2940 
igmp6_mc_seq_show(struct seq_file * seq,void * v)2941 static int igmp6_mc_seq_show(struct seq_file *seq, void *v)
2942 {
2943 	struct ifmcaddr6 *im = (struct ifmcaddr6 *)v;
2944 	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2945 
2946 	seq_printf(seq,
2947 		   "%-4d %-15s %pi6 %5d %08X %ld\n",
2948 		   state->dev->ifindex, state->dev->name,
2949 		   &im->mca_addr,
2950 		   im->mca_users, im->mca_flags,
2951 		   (im->mca_flags & MAF_TIMER_RUNNING) ?
2952 		   jiffies_to_clock_t(im->mca_work.timer.expires - jiffies) : 0);
2953 	return 0;
2954 }
2955 
2956 static const struct seq_operations igmp6_mc_seq_ops = {
2957 	.start	=	igmp6_mc_seq_start,
2958 	.next	=	igmp6_mc_seq_next,
2959 	.stop	=	igmp6_mc_seq_stop,
2960 	.show	=	igmp6_mc_seq_show,
2961 };
2962 
2963 struct igmp6_mcf_iter_state {
2964 	struct seq_net_private p;
2965 	struct net_device *dev;
2966 	struct inet6_dev *idev;
2967 	struct ifmcaddr6 *im;
2968 };
2969 
2970 #define igmp6_mcf_seq_private(seq)	((struct igmp6_mcf_iter_state *)(seq)->private)
2971 
igmp6_mcf_get_first(struct seq_file * seq)2972 static inline struct ip6_sf_list *igmp6_mcf_get_first(struct seq_file *seq)
2973 {
2974 	struct ip6_sf_list *psf = NULL;
2975 	struct ifmcaddr6 *im = NULL;
2976 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
2977 	struct net *net = seq_file_net(seq);
2978 
2979 	state->idev = NULL;
2980 	state->im = NULL;
2981 	for_each_netdev_rcu(net, state->dev) {
2982 		struct inet6_dev *idev;
2983 		idev = __in6_dev_get(state->dev);
2984 		if (unlikely(idev == NULL))
2985 			continue;
2986 
2987 		im = rcu_dereference(idev->mc_list);
2988 		if (likely(im)) {
2989 			psf = rcu_dereference(im->mca_sources);
2990 			if (likely(psf)) {
2991 				state->im = im;
2992 				state->idev = idev;
2993 				break;
2994 			}
2995 		}
2996 	}
2997 	return psf;
2998 }
2999 
igmp6_mcf_get_next(struct seq_file * seq,struct ip6_sf_list * psf)3000 static struct ip6_sf_list *igmp6_mcf_get_next(struct seq_file *seq, struct ip6_sf_list *psf)
3001 {
3002 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3003 
3004 	psf = rcu_dereference(psf->sf_next);
3005 	while (!psf) {
3006 		state->im = rcu_dereference(state->im->next);
3007 		while (!state->im) {
3008 			state->dev = next_net_device_rcu(state->dev);
3009 			if (!state->dev) {
3010 				state->idev = NULL;
3011 				goto out;
3012 			}
3013 			state->idev = __in6_dev_get(state->dev);
3014 			if (!state->idev)
3015 				continue;
3016 			state->im = rcu_dereference(state->idev->mc_list);
3017 		}
3018 		if (!state->im)
3019 			break;
3020 		psf = rcu_dereference(state->im->mca_sources);
3021 	}
3022 out:
3023 	return psf;
3024 }
3025 
igmp6_mcf_get_idx(struct seq_file * seq,loff_t pos)3026 static struct ip6_sf_list *igmp6_mcf_get_idx(struct seq_file *seq, loff_t pos)
3027 {
3028 	struct ip6_sf_list *psf = igmp6_mcf_get_first(seq);
3029 	if (psf)
3030 		while (pos && (psf = igmp6_mcf_get_next(seq, psf)) != NULL)
3031 			--pos;
3032 	return pos ? NULL : psf;
3033 }
3034 
igmp6_mcf_seq_start(struct seq_file * seq,loff_t * pos)3035 static void *igmp6_mcf_seq_start(struct seq_file *seq, loff_t *pos)
3036 	__acquires(RCU)
3037 {
3038 	rcu_read_lock();
3039 	return *pos ? igmp6_mcf_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
3040 }
3041 
igmp6_mcf_seq_next(struct seq_file * seq,void * v,loff_t * pos)3042 static void *igmp6_mcf_seq_next(struct seq_file *seq, void *v, loff_t *pos)
3043 {
3044 	struct ip6_sf_list *psf;
3045 	if (v == SEQ_START_TOKEN)
3046 		psf = igmp6_mcf_get_first(seq);
3047 	else
3048 		psf = igmp6_mcf_get_next(seq, v);
3049 	++*pos;
3050 	return psf;
3051 }
3052 
igmp6_mcf_seq_stop(struct seq_file * seq,void * v)3053 static void igmp6_mcf_seq_stop(struct seq_file *seq, void *v)
3054 	__releases(RCU)
3055 {
3056 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3057 
3058 	if (likely(state->im))
3059 		state->im = NULL;
3060 	if (likely(state->idev))
3061 		state->idev = NULL;
3062 
3063 	state->dev = NULL;
3064 	rcu_read_unlock();
3065 }
3066 
igmp6_mcf_seq_show(struct seq_file * seq,void * v)3067 static int igmp6_mcf_seq_show(struct seq_file *seq, void *v)
3068 {
3069 	struct ip6_sf_list *psf = (struct ip6_sf_list *)v;
3070 	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3071 
3072 	if (v == SEQ_START_TOKEN) {
3073 		seq_puts(seq, "Idx Device                Multicast Address                   Source Address    INC    EXC\n");
3074 	} else {
3075 		seq_printf(seq,
3076 			   "%3d %6.6s %pi6 %pi6 %6lu %6lu\n",
3077 			   state->dev->ifindex, state->dev->name,
3078 			   &state->im->mca_addr,
3079 			   &psf->sf_addr,
3080 			   psf->sf_count[MCAST_INCLUDE],
3081 			   psf->sf_count[MCAST_EXCLUDE]);
3082 	}
3083 	return 0;
3084 }
3085 
3086 static const struct seq_operations igmp6_mcf_seq_ops = {
3087 	.start	=	igmp6_mcf_seq_start,
3088 	.next	=	igmp6_mcf_seq_next,
3089 	.stop	=	igmp6_mcf_seq_stop,
3090 	.show	=	igmp6_mcf_seq_show,
3091 };
3092 
igmp6_proc_init(struct net * net)3093 static int __net_init igmp6_proc_init(struct net *net)
3094 {
3095 	int err;
3096 
3097 	err = -ENOMEM;
3098 	if (!proc_create_net("igmp6", 0444, net->proc_net, &igmp6_mc_seq_ops,
3099 			sizeof(struct igmp6_mc_iter_state)))
3100 		goto out;
3101 	if (!proc_create_net("mcfilter6", 0444, net->proc_net,
3102 			&igmp6_mcf_seq_ops,
3103 			sizeof(struct igmp6_mcf_iter_state)))
3104 		goto out_proc_net_igmp6;
3105 
3106 	err = 0;
3107 out:
3108 	return err;
3109 
3110 out_proc_net_igmp6:
3111 	remove_proc_entry("igmp6", net->proc_net);
3112 	goto out;
3113 }
3114 
igmp6_proc_exit(struct net * net)3115 static void __net_exit igmp6_proc_exit(struct net *net)
3116 {
3117 	remove_proc_entry("mcfilter6", net->proc_net);
3118 	remove_proc_entry("igmp6", net->proc_net);
3119 }
3120 #else
igmp6_proc_init(struct net * net)3121 static inline int igmp6_proc_init(struct net *net)
3122 {
3123 	return 0;
3124 }
igmp6_proc_exit(struct net * net)3125 static inline void igmp6_proc_exit(struct net *net)
3126 {
3127 }
3128 #endif
3129 
igmp6_net_init(struct net * net)3130 static int __net_init igmp6_net_init(struct net *net)
3131 {
3132 	int err;
3133 
3134 	err = inet_ctl_sock_create(&net->ipv6.igmp_sk, PF_INET6,
3135 				   SOCK_RAW, IPPROTO_ICMPV6, net);
3136 	if (err < 0) {
3137 		pr_err("Failed to initialize the IGMP6 control socket (err %d)\n",
3138 		       err);
3139 		goto out;
3140 	}
3141 
3142 	inet6_sk(net->ipv6.igmp_sk)->hop_limit = 1;
3143 	net->ipv6.igmp_sk->sk_allocation = GFP_KERNEL;
3144 
3145 	err = inet_ctl_sock_create(&net->ipv6.mc_autojoin_sk, PF_INET6,
3146 				   SOCK_RAW, IPPROTO_ICMPV6, net);
3147 	if (err < 0) {
3148 		pr_err("Failed to initialize the IGMP6 autojoin socket (err %d)\n",
3149 		       err);
3150 		goto out_sock_create;
3151 	}
3152 
3153 	err = igmp6_proc_init(net);
3154 	if (err)
3155 		goto out_sock_create_autojoin;
3156 
3157 	return 0;
3158 
3159 out_sock_create_autojoin:
3160 	inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3161 out_sock_create:
3162 	inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3163 out:
3164 	return err;
3165 }
3166 
igmp6_net_exit(struct net * net)3167 static void __net_exit igmp6_net_exit(struct net *net)
3168 {
3169 	inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3170 	inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3171 	igmp6_proc_exit(net);
3172 }
3173 
3174 static struct pernet_operations igmp6_net_ops = {
3175 	.init = igmp6_net_init,
3176 	.exit = igmp6_net_exit,
3177 };
3178 
igmp6_init(void)3179 int __init igmp6_init(void)
3180 {
3181 	int err;
3182 
3183 	err = register_pernet_subsys(&igmp6_net_ops);
3184 	if (err)
3185 		return err;
3186 
3187 	mld_wq = create_workqueue("mld");
3188 	if (!mld_wq) {
3189 		unregister_pernet_subsys(&igmp6_net_ops);
3190 		return -ENOMEM;
3191 	}
3192 
3193 	return err;
3194 }
3195 
igmp6_late_init(void)3196 int __init igmp6_late_init(void)
3197 {
3198 	return register_netdevice_notifier(&igmp6_netdev_notifier);
3199 }
3200 
igmp6_cleanup(void)3201 void igmp6_cleanup(void)
3202 {
3203 	unregister_pernet_subsys(&igmp6_net_ops);
3204 	destroy_workqueue(mld_wq);
3205 }
3206 
igmp6_late_cleanup(void)3207 void igmp6_late_cleanup(void)
3208 {
3209 	unregister_netdevice_notifier(&igmp6_netdev_notifier);
3210 }
3211