mcast.c 68.1 KB
Newer Older
L
Linus Torvalds 已提交
1 2
/*
 *	Multicast support for IPv6
3
 *	Linux INET6 implementation
L
Linus Torvalds 已提交
4 5
 *
 *	Authors:
6
 *	Pedro Roque		<roque@di.fc.ul.pt>
L
Linus Torvalds 已提交
7
 *
8
 *	Based on linux/ipv4/igmp.c and linux/ipv4/ip_sockglue.c
L
Linus Torvalds 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
 *
 *	This program is free software; you can redistribute it and/or
 *      modify it under the terms of the GNU General Public License
 *      as published by the Free Software Foundation; either version
 *      2 of the License, or (at your option) any later version.
 */

/* Changes:
 *
 *	yoshfuji	: fix format of router-alert option
 *	YOSHIFUJI Hideaki @USAGI:
 *		Fixed source address for MLD message based on
 *		<draft-ietf-magma-mld-source-05.txt>.
 *	YOSHIFUJI Hideaki @USAGI:
 *		- Ignore Queries for invalid addresses.
 *		- MLD for link-local addresses.
 *	David L Stevens <dlstevens@us.ibm.com>:
 *		- MLDv2 support
 */

#include <linux/module.h>
#include <linux/errno.h>
#include <linux/types.h>
#include <linux/string.h>
#include <linux/socket.h>
#include <linux/sockios.h>
#include <linux/jiffies.h>
#include <linux/times.h>
#include <linux/net.h>
#include <linux/in.h>
#include <linux/in6.h>
#include <linux/netdevice.h>
#include <linux/if_arp.h>
#include <linux/route.h>
#include <linux/init.h>
#include <linux/proc_fs.h>
#include <linux/seq_file.h>
46
#include <linux/slab.h>
47
#include <linux/pkt_sched.h>
48
#include <net/mld.h>
L
Linus Torvalds 已提交
49 50 51 52

#include <linux/netfilter.h>
#include <linux/netfilter_ipv6.h>

53
#include <net/net_namespace.h>
L
Linus Torvalds 已提交
54 55 56 57 58 59 60 61 62
#include <net/sock.h>
#include <net/snmp.h>

#include <net/ipv6.h>
#include <net/protocol.h>
#include <net/if_inet6.h>
#include <net/ndisc.h>
#include <net/addrconf.h>
#include <net/ip6_route.h>
63
#include <net/inet_common.h>
L
Linus Torvalds 已提交
64 65 66 67 68 69 70 71 72 73 74 75

#include <net/ip6_checksum.h>

/* Set to 3 to get tracing... */
#define MCAST_DEBUG 2

#if MCAST_DEBUG >= 3
#define MDBG(x) printk x
#else
#define MDBG(x)
#endif

76 77 78 79 80
/* Ensure that we have struct in6_addr aligned on 32bit word. */
static void *__mld2_query_bugs[] __attribute__((__unused__)) = {
	BUILD_BUG_ON_NULL(offsetof(struct mld2_query, mld2q_srcs) % 4),
	BUILD_BUG_ON_NULL(offsetof(struct mld2_report, mld2r_grec) % 4),
	BUILD_BUG_ON_NULL(offsetof(struct mld2_grec, grec_mca) % 4)
L
Linus Torvalds 已提交
81 82 83 84 85
};

static struct in6_addr mld2_all_mcr = MLD2_ALL_MCR_INIT;

/* Big mc list lock for all the sockets */
E
Eric Dumazet 已提交
86
static DEFINE_SPINLOCK(ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
87 88 89 90 91 92 93 94 95

static void igmp6_join_group(struct ifmcaddr6 *ma);
static void igmp6_leave_group(struct ifmcaddr6 *ma);
static void igmp6_timer_handler(unsigned long data);

static void mld_gq_timer_expire(unsigned long data);
static void mld_ifc_timer_expire(unsigned long data);
static void mld_ifc_event(struct inet6_dev *idev);
static void mld_add_delrec(struct inet6_dev *idev, struct ifmcaddr6 *pmc);
96
static void mld_del_delrec(struct inet6_dev *idev, const struct in6_addr *addr);
L
Linus Torvalds 已提交
97
static void mld_clear_delrec(struct inet6_dev *idev);
98
static bool mld_in_v1_mode(const struct inet6_dev *idev);
L
Linus Torvalds 已提交
99 100 101
static int sf_setstate(struct ifmcaddr6 *pmc);
static void sf_markstate(struct ifmcaddr6 *pmc);
static void ip6_mc_clear_src(struct ifmcaddr6 *pmc);
102 103
static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
L
Linus Torvalds 已提交
104
			  int delta);
105 106
static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
L
Linus Torvalds 已提交
107 108 109 110 111
			  int delta);
static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
			    struct inet6_dev *idev);

#define MLD_QRV_DEFAULT		2
112 113 114 115
/* RFC3810, 9.2. Query Interval */
#define MLD_QI_DEFAULT		(125 * HZ)
/* RFC3810, 9.3. Query Response Interval */
#define MLD_QRI_DEFAULT		(10 * HZ)
L
Linus Torvalds 已提交
116

117 118 119 120
/* RFC3810, 8.1 Query Version Distinctions */
#define MLD_V1_QUERY_LEN	24
#define MLD_V2_QUERY_LEN_MIN	28

121
#define IPV6_MLD_MAX_MSF	64
L
Linus Torvalds 已提交
122

123
int sysctl_mld_max_msf __read_mostly = IPV6_MLD_MAX_MSF;
L
Linus Torvalds 已提交
124 125 126 127 128

/*
 *	socket join on multicast group
 */

E
Eric Dumazet 已提交
129 130 131 132 133
#define for_each_pmc_rcu(np, pmc)				\
	for (pmc = rcu_dereference(np->ipv6_mc_list);		\
	     pmc != NULL;					\
	     pmc = rcu_dereference(pmc->next))

134 135 136 137
static int unsolicited_report_interval(struct inet6_dev *idev)
{
	int iv;

138
	if (mld_in_v1_mode(idev))
139 140 141 142 143 144 145
		iv = idev->cnf.mldv1_unsolicited_report_interval;
	else
		iv = idev->cnf.mldv2_unsolicited_report_interval;

	return iv > 0 ? iv : 1;
}

146
int ipv6_sock_mc_join(struct sock *sk, int ifindex, const struct in6_addr *addr)
L
Linus Torvalds 已提交
147 148 149 150
{
	struct net_device *dev = NULL;
	struct ipv6_mc_socklist *mc_lst;
	struct ipv6_pinfo *np = inet6_sk(sk);
151
	struct net *net = sock_net(sk);
L
Linus Torvalds 已提交
152 153 154 155 156
	int err;

	if (!ipv6_addr_is_multicast(addr))
		return -EINVAL;

E
Eric Dumazet 已提交
157 158
	rcu_read_lock();
	for_each_pmc_rcu(np, mc_lst) {
159 160
		if ((ifindex == 0 || mc_lst->ifindex == ifindex) &&
		    ipv6_addr_equal(&mc_lst->addr, addr)) {
E
Eric Dumazet 已提交
161
			rcu_read_unlock();
162 163 164
			return -EADDRINUSE;
		}
	}
E
Eric Dumazet 已提交
165
	rcu_read_unlock();
166

L
Linus Torvalds 已提交
167 168 169 170 171 172
	mc_lst = sock_kmalloc(sk, sizeof(struct ipv6_mc_socklist), GFP_KERNEL);

	if (mc_lst == NULL)
		return -ENOMEM;

	mc_lst->next = NULL;
A
Alexey Dobriyan 已提交
173
	mc_lst->addr = *addr;
L
Linus Torvalds 已提交
174

E
Eric Dumazet 已提交
175
	rcu_read_lock();
L
Linus Torvalds 已提交
176 177
	if (ifindex == 0) {
		struct rt6_info *rt;
178
		rt = rt6_lookup(net, addr, NULL, 0, 0);
L
Linus Torvalds 已提交
179
		if (rt) {
180
			dev = rt->dst.dev;
A
Amerigo Wang 已提交
181
			ip6_rt_put(rt);
L
Linus Torvalds 已提交
182 183
		}
	} else
E
Eric Dumazet 已提交
184
		dev = dev_get_by_index_rcu(net, ifindex);
L
Linus Torvalds 已提交
185 186

	if (dev == NULL) {
E
Eric Dumazet 已提交
187
		rcu_read_unlock();
L
Linus Torvalds 已提交
188 189 190 191 192 193
		sock_kfree_s(sk, mc_lst, sizeof(*mc_lst));
		return -ENODEV;
	}

	mc_lst->ifindex = dev->ifindex;
	mc_lst->sfmode = MCAST_EXCLUDE;
194
	rwlock_init(&mc_lst->sflock);
L
Linus Torvalds 已提交
195 196 197 198 199 200 201 202 203
	mc_lst->sflist = NULL;

	/*
	 *	now add/increase the group membership on the device
	 */

	err = ipv6_dev_mc_inc(dev, addr);

	if (err) {
E
Eric Dumazet 已提交
204
		rcu_read_unlock();
L
Linus Torvalds 已提交
205 206 207 208
		sock_kfree_s(sk, mc_lst, sizeof(*mc_lst));
		return err;
	}

E
Eric Dumazet 已提交
209
	spin_lock(&ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
210
	mc_lst->next = np->ipv6_mc_list;
E
Eric Dumazet 已提交
211 212
	rcu_assign_pointer(np->ipv6_mc_list, mc_lst);
	spin_unlock(&ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
213

E
Eric Dumazet 已提交
214
	rcu_read_unlock();
L
Linus Torvalds 已提交
215 216 217 218 219 220 221

	return 0;
}

/*
 *	socket leave on multicast group
 */
222
int ipv6_sock_mc_drop(struct sock *sk, int ifindex, const struct in6_addr *addr)
L
Linus Torvalds 已提交
223 224
{
	struct ipv6_pinfo *np = inet6_sk(sk);
E
Eric Dumazet 已提交
225 226
	struct ipv6_mc_socklist *mc_lst;
	struct ipv6_mc_socklist __rcu **lnk;
227
	struct net *net = sock_net(sk);
L
Linus Torvalds 已提交
228

229 230 231
	if (!ipv6_addr_is_multicast(addr))
		return -EINVAL;

E
Eric Dumazet 已提交
232 233 234 235 236
	spin_lock(&ipv6_sk_mc_lock);
	for (lnk = &np->ipv6_mc_list;
	     (mc_lst = rcu_dereference_protected(*lnk,
			lockdep_is_held(&ipv6_sk_mc_lock))) !=NULL ;
	      lnk = &mc_lst->next) {
L
Linus Torvalds 已提交
237 238 239 240 241
		if ((ifindex == 0 || mc_lst->ifindex == ifindex) &&
		    ipv6_addr_equal(&mc_lst->addr, addr)) {
			struct net_device *dev;

			*lnk = mc_lst->next;
E
Eric Dumazet 已提交
242
			spin_unlock(&ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
243

E
Eric Dumazet 已提交
244 245
			rcu_read_lock();
			dev = dev_get_by_index_rcu(net, mc_lst->ifindex);
246
			if (dev != NULL) {
E
Eric Dumazet 已提交
247
				struct inet6_dev *idev = __in6_dev_get(dev);
L
Linus Torvalds 已提交
248

249
				(void) ip6_mc_leave_src(sk, mc_lst, idev);
E
Eric Dumazet 已提交
250
				if (idev)
L
Linus Torvalds 已提交
251
					__ipv6_dev_mc_dec(idev, &mc_lst->addr);
252 253
			} else
				(void) ip6_mc_leave_src(sk, mc_lst, NULL);
E
Eric Dumazet 已提交
254
			rcu_read_unlock();
E
Eric Dumazet 已提交
255
			atomic_sub(sizeof(*mc_lst), &sk->sk_omem_alloc);
256
			kfree_rcu(mc_lst, rcu);
L
Linus Torvalds 已提交
257 258 259
			return 0;
		}
	}
E
Eric Dumazet 已提交
260
	spin_unlock(&ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
261

262
	return -EADDRNOTAVAIL;
L
Linus Torvalds 已提交
263 264
}

E
Eric Dumazet 已提交
265 266
/* called with rcu_read_lock() */
static struct inet6_dev *ip6_mc_find_dev_rcu(struct net *net,
267
					     const struct in6_addr *group,
E
Eric Dumazet 已提交
268
					     int ifindex)
L
Linus Torvalds 已提交
269 270 271 272 273
{
	struct net_device *dev = NULL;
	struct inet6_dev *idev = NULL;

	if (ifindex == 0) {
E
Eric Dumazet 已提交
274
		struct rt6_info *rt = rt6_lookup(net, group, NULL, 0, 0);
L
Linus Torvalds 已提交
275 276

		if (rt) {
277
			dev = rt->dst.dev;
A
Amerigo Wang 已提交
278
			ip6_rt_put(rt);
L
Linus Torvalds 已提交
279 280
		}
	} else
E
Eric Dumazet 已提交
281
		dev = dev_get_by_index_rcu(net, ifindex);
L
Linus Torvalds 已提交
282 283

	if (!dev)
E
Eric Dumazet 已提交
284 285
		return NULL;
	idev = __in6_dev_get(dev);
286
	if (!idev)
287
		return NULL;
L
Linus Torvalds 已提交
288
	read_lock_bh(&idev->lock);
E
Eric Dumazet 已提交
289 290 291 292
	if (idev->dead) {
		read_unlock_bh(&idev->lock);
		return NULL;
	}
L
Linus Torvalds 已提交
293 294 295 296 297 298 299
	return idev;
}

void ipv6_sock_mc_close(struct sock *sk)
{
	struct ipv6_pinfo *np = inet6_sk(sk);
	struct ipv6_mc_socklist *mc_lst;
300
	struct net *net = sock_net(sk);
L
Linus Torvalds 已提交
301

302 303 304
	if (!rcu_access_pointer(np->ipv6_mc_list))
		return;

E
Eric Dumazet 已提交
305 306 307
	spin_lock(&ipv6_sk_mc_lock);
	while ((mc_lst = rcu_dereference_protected(np->ipv6_mc_list,
				lockdep_is_held(&ipv6_sk_mc_lock))) != NULL) {
L
Linus Torvalds 已提交
308 309 310
		struct net_device *dev;

		np->ipv6_mc_list = mc_lst->next;
E
Eric Dumazet 已提交
311
		spin_unlock(&ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
312

E
Eric Dumazet 已提交
313 314
		rcu_read_lock();
		dev = dev_get_by_index_rcu(net, mc_lst->ifindex);
L
Linus Torvalds 已提交
315
		if (dev) {
E
Eric Dumazet 已提交
316
			struct inet6_dev *idev = __in6_dev_get(dev);
L
Linus Torvalds 已提交
317

318
			(void) ip6_mc_leave_src(sk, mc_lst, idev);
E
Eric Dumazet 已提交
319
			if (idev)
L
Linus Torvalds 已提交
320
				__ipv6_dev_mc_dec(idev, &mc_lst->addr);
321 322
		} else
			(void) ip6_mc_leave_src(sk, mc_lst, NULL);
E
Eric Dumazet 已提交
323
		rcu_read_unlock();
L
Linus Torvalds 已提交
324

E
Eric Dumazet 已提交
325
		atomic_sub(sizeof(*mc_lst), &sk->sk_omem_alloc);
326
		kfree_rcu(mc_lst, rcu);
E
Eric Dumazet 已提交
327 328

		spin_lock(&ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
329
	}
E
Eric Dumazet 已提交
330
	spin_unlock(&ipv6_sk_mc_lock);
L
Linus Torvalds 已提交
331 332 333 334 335 336 337 338 339 340
}

int ip6_mc_source(int add, int omode, struct sock *sk,
	struct group_source_req *pgsr)
{
	struct in6_addr *source, *group;
	struct ipv6_mc_socklist *pmc;
	struct inet6_dev *idev;
	struct ipv6_pinfo *inet6 = inet6_sk(sk);
	struct ip6_sf_socklist *psl;
341
	struct net *net = sock_net(sk);
L
Linus Torvalds 已提交
342
	int i, j, rv;
343
	int leavegroup = 0;
344
	int pmclocked = 0;
L
Linus Torvalds 已提交
345 346 347 348 349 350 351 352
	int err;

	source = &((struct sockaddr_in6 *)&pgsr->gsr_source)->sin6_addr;
	group = &((struct sockaddr_in6 *)&pgsr->gsr_group)->sin6_addr;

	if (!ipv6_addr_is_multicast(group))
		return -EINVAL;

E
Eric Dumazet 已提交
353 354 355 356
	rcu_read_lock();
	idev = ip6_mc_find_dev_rcu(net, group, pgsr->gsr_interface);
	if (!idev) {
		rcu_read_unlock();
L
Linus Torvalds 已提交
357
		return -ENODEV;
E
Eric Dumazet 已提交
358
	}
L
Linus Torvalds 已提交
359 360 361

	err = -EADDRNOTAVAIL;

E
Eric Dumazet 已提交
362
	for_each_pmc_rcu(inet6, pmc) {
L
Linus Torvalds 已提交
363 364 365 366 367
		if (pgsr->gsr_interface && pmc->ifindex != pgsr->gsr_interface)
			continue;
		if (ipv6_addr_equal(&pmc->addr, group))
			break;
	}
368 369
	if (!pmc) {		/* must have a prior join */
		err = -EINVAL;
L
Linus Torvalds 已提交
370
		goto done;
371
	}
L
Linus Torvalds 已提交
372 373
	/* if a source filter was set, must be the same mode as before */
	if (pmc->sflist) {
374 375
		if (pmc->sfmode != omode) {
			err = -EINVAL;
L
Linus Torvalds 已提交
376
			goto done;
377
		}
L
Linus Torvalds 已提交
378 379 380 381 382 383 384
	} else if (pmc->sfmode != omode) {
		/* allow mode switches for empty-set filters */
		ip6_mc_add_src(idev, group, omode, 0, NULL, 0);
		ip6_mc_del_src(idev, group, pmc->sfmode, 0, NULL, 0);
		pmc->sfmode = omode;
	}

E
Eric Dumazet 已提交
385
	write_lock(&pmc->sflock);
386 387
	pmclocked = 1;

L
Linus Torvalds 已提交
388 389 390
	psl = pmc->sflist;
	if (!add) {
		if (!psl)
391
			goto done;	/* err = -EADDRNOTAVAIL */
L
Linus Torvalds 已提交
392 393
		rv = !0;
		for (i=0; i<psl->sl_count; i++) {
394
			rv = !ipv6_addr_equal(&psl->sl_addr[i], source);
L
Linus Torvalds 已提交
395 396 397 398
			if (rv == 0)
				break;
		}
		if (rv)		/* source not found */
399
			goto done;	/* err = -EADDRNOTAVAIL */
L
Linus Torvalds 已提交
400

401 402 403 404 405 406
		/* special case - (INCLUDE, empty) == LEAVE_GROUP */
		if (psl->sl_count == 1 && omode == MCAST_INCLUDE) {
			leavegroup = 1;
			goto done;
		}

L
Linus Torvalds 已提交
407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427
		/* update the interface filter */
		ip6_mc_del_src(idev, group, omode, 1, source, 1);

		for (j=i+1; j<psl->sl_count; j++)
			psl->sl_addr[j-1] = psl->sl_addr[j];
		psl->sl_count--;
		err = 0;
		goto done;
	}
	/* else, add a new source to the filter */

	if (psl && psl->sl_count >= sysctl_mld_max_msf) {
		err = -ENOBUFS;
		goto done;
	}
	if (!psl || psl->sl_count == psl->sl_max) {
		struct ip6_sf_socklist *newpsl;
		int count = IP6_SFBLOCK;

		if (psl)
			count += psl->sl_max;
428
		newpsl = sock_kmalloc(sk, IP6_SFLSIZE(count), GFP_ATOMIC);
L
Linus Torvalds 已提交
429 430 431 432 433 434 435 436 437 438 439 440 441 442 443
		if (!newpsl) {
			err = -ENOBUFS;
			goto done;
		}
		newpsl->sl_max = count;
		newpsl->sl_count = count - IP6_SFBLOCK;
		if (psl) {
			for (i=0; i<psl->sl_count; i++)
				newpsl->sl_addr[i] = psl->sl_addr[i];
			sock_kfree_s(sk, psl, IP6_SFLSIZE(psl->sl_max));
		}
		pmc->sflist = psl = newpsl;
	}
	rv = 1;	/* > 0 for insert logic below if sl_count is 0 */
	for (i=0; i<psl->sl_count; i++) {
444
		rv = !ipv6_addr_equal(&psl->sl_addr[i], source);
445 446
		if (rv == 0) /* There is an error in the address. */
			goto done;
L
Linus Torvalds 已提交
447 448 449 450 451 452 453 454 455
	}
	for (j=psl->sl_count-1; j>=i; j--)
		psl->sl_addr[j+1] = psl->sl_addr[j];
	psl->sl_addr[i] = *source;
	psl->sl_count++;
	err = 0;
	/* update the interface list */
	ip6_mc_add_src(idev, group, omode, 1, source, 1);
done:
456
	if (pmclocked)
E
Eric Dumazet 已提交
457
		write_unlock(&pmc->sflock);
L
Linus Torvalds 已提交
458
	read_unlock_bh(&idev->lock);
E
Eric Dumazet 已提交
459
	rcu_read_unlock();
460 461
	if (leavegroup)
		return ipv6_sock_mc_drop(sk, pgsr->gsr_interface, group);
L
Linus Torvalds 已提交
462 463 464 465 466
	return err;
}

int ip6_mc_msfilter(struct sock *sk, struct group_filter *gsf)
{
467
	const struct in6_addr *group;
L
Linus Torvalds 已提交
468 469 470 471
	struct ipv6_mc_socklist *pmc;
	struct inet6_dev *idev;
	struct ipv6_pinfo *inet6 = inet6_sk(sk);
	struct ip6_sf_socklist *newpsl, *psl;
472
	struct net *net = sock_net(sk);
473
	int leavegroup = 0;
L
Linus Torvalds 已提交
474 475 476 477 478 479 480 481 482 483
	int i, err;

	group = &((struct sockaddr_in6 *)&gsf->gf_group)->sin6_addr;

	if (!ipv6_addr_is_multicast(group))
		return -EINVAL;
	if (gsf->gf_fmode != MCAST_INCLUDE &&
	    gsf->gf_fmode != MCAST_EXCLUDE)
		return -EINVAL;

E
Eric Dumazet 已提交
484 485
	rcu_read_lock();
	idev = ip6_mc_find_dev_rcu(net, group, gsf->gf_interface);
L
Linus Torvalds 已提交
486

E
Eric Dumazet 已提交
487 488
	if (!idev) {
		rcu_read_unlock();
L
Linus Torvalds 已提交
489
		return -ENODEV;
E
Eric Dumazet 已提交
490
	}
L
Linus Torvalds 已提交
491

492
	err = 0;
493

494 495 496 497 498
	if (gsf->gf_fmode == MCAST_INCLUDE && gsf->gf_numsrc == 0) {
		leavegroup = 1;
		goto done;
	}

E
Eric Dumazet 已提交
499
	for_each_pmc_rcu(inet6, pmc) {
L
Linus Torvalds 已提交
500 501 502 503 504
		if (pmc->ifindex != gsf->gf_interface)
			continue;
		if (ipv6_addr_equal(&pmc->addr, group))
			break;
	}
505 506
	if (!pmc) {		/* must have a prior join */
		err = -EINVAL;
L
Linus Torvalds 已提交
507
		goto done;
508
	}
L
Linus Torvalds 已提交
509
	if (gsf->gf_numsrc) {
510 511
		newpsl = sock_kmalloc(sk, IP6_SFLSIZE(gsf->gf_numsrc),
							  GFP_ATOMIC);
L
Linus Torvalds 已提交
512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528
		if (!newpsl) {
			err = -ENOBUFS;
			goto done;
		}
		newpsl->sl_max = newpsl->sl_count = gsf->gf_numsrc;
		for (i=0; i<newpsl->sl_count; ++i) {
			struct sockaddr_in6 *psin6;

			psin6 = (struct sockaddr_in6 *)&gsf->gf_slist[i];
			newpsl->sl_addr[i] = psin6->sin6_addr;
		}
		err = ip6_mc_add_src(idev, group, gsf->gf_fmode,
			newpsl->sl_count, newpsl->sl_addr, 0);
		if (err) {
			sock_kfree_s(sk, newpsl, IP6_SFLSIZE(newpsl->sl_max));
			goto done;
		}
529
	} else {
L
Linus Torvalds 已提交
530
		newpsl = NULL;
531 532
		(void) ip6_mc_add_src(idev, group, gsf->gf_fmode, 0, NULL, 0);
	}
533

E
Eric Dumazet 已提交
534
	write_lock(&pmc->sflock);
L
Linus Torvalds 已提交
535 536 537 538 539 540 541 542 543
	psl = pmc->sflist;
	if (psl) {
		(void) ip6_mc_del_src(idev, group, pmc->sfmode,
			psl->sl_count, psl->sl_addr, 0);
		sock_kfree_s(sk, psl, IP6_SFLSIZE(psl->sl_max));
	} else
		(void) ip6_mc_del_src(idev, group, pmc->sfmode, 0, NULL, 0);
	pmc->sflist = newpsl;
	pmc->sfmode = gsf->gf_fmode;
E
Eric Dumazet 已提交
544
	write_unlock(&pmc->sflock);
545
	err = 0;
L
Linus Torvalds 已提交
546 547
done:
	read_unlock_bh(&idev->lock);
E
Eric Dumazet 已提交
548
	rcu_read_unlock();
549 550
	if (leavegroup)
		err = ipv6_sock_mc_drop(sk, gsf->gf_interface, group);
L
Linus Torvalds 已提交
551 552 553 554 555 556 557
	return err;
}

int ip6_mc_msfget(struct sock *sk, struct group_filter *gsf,
	struct group_filter __user *optval, int __user *optlen)
{
	int err, i, count, copycount;
558
	const struct in6_addr *group;
L
Linus Torvalds 已提交
559 560 561 562
	struct ipv6_mc_socklist *pmc;
	struct inet6_dev *idev;
	struct ipv6_pinfo *inet6 = inet6_sk(sk);
	struct ip6_sf_socklist *psl;
563
	struct net *net = sock_net(sk);
L
Linus Torvalds 已提交
564 565 566 567 568 569

	group = &((struct sockaddr_in6 *)&gsf->gf_group)->sin6_addr;

	if (!ipv6_addr_is_multicast(group))
		return -EINVAL;

E
Eric Dumazet 已提交
570 571
	rcu_read_lock();
	idev = ip6_mc_find_dev_rcu(net, group, gsf->gf_interface);
L
Linus Torvalds 已提交
572

E
Eric Dumazet 已提交
573 574
	if (!idev) {
		rcu_read_unlock();
L
Linus Torvalds 已提交
575
		return -ENODEV;
E
Eric Dumazet 已提交
576
	}
L
Linus Torvalds 已提交
577 578

	err = -EADDRNOTAVAIL;
579 580 581 582 583
	/*
	 * changes to the ipv6_mc_list require the socket lock and
	 * a read lock on ip6_sk_mc_lock. We have the socket lock,
	 * so reading the list is safe.
	 */
L
Linus Torvalds 已提交
584

E
Eric Dumazet 已提交
585
	for_each_pmc_rcu(inet6, pmc) {
L
Linus Torvalds 已提交
586 587 588 589 590 591 592 593 594 595 596
		if (pmc->ifindex != gsf->gf_interface)
			continue;
		if (ipv6_addr_equal(group, &pmc->addr))
			break;
	}
	if (!pmc)		/* must have a prior join */
		goto done;
	gsf->gf_fmode = pmc->sfmode;
	psl = pmc->sflist;
	count = psl ? psl->sl_count : 0;
	read_unlock_bh(&idev->lock);
E
Eric Dumazet 已提交
597
	rcu_read_unlock();
L
Linus Torvalds 已提交
598 599 600 601 602 603 604

	copycount = count < gsf->gf_numsrc ? count : gsf->gf_numsrc;
	gsf->gf_numsrc = count;
	if (put_user(GROUP_FILTER_SIZE(copycount), optlen) ||
	    copy_to_user(optval, gsf, GROUP_FILTER_SIZE(0))) {
		return -EFAULT;
	}
605 606 607 608
	/* changes to psl require the socket lock, a read lock on
	 * on ipv6_sk_mc_lock and a write lock on pmc->sflock. We
	 * have the socket lock, so reading here is safe.
	 */
L
Linus Torvalds 已提交
609 610 611 612 613 614 615 616
	for (i=0; i<copycount; i++) {
		struct sockaddr_in6 *psin6;
		struct sockaddr_storage ss;

		psin6 = (struct sockaddr_in6 *)&ss;
		memset(&ss, 0, sizeof(ss));
		psin6->sin6_family = AF_INET6;
		psin6->sin6_addr = psl->sl_addr[i];
617
		if (copy_to_user(&optval->gf_slist[i], &ss, sizeof(ss)))
L
Linus Torvalds 已提交
618 619 620 621 622
			return -EFAULT;
	}
	return 0;
done:
	read_unlock_bh(&idev->lock);
E
Eric Dumazet 已提交
623
	rcu_read_unlock();
L
Linus Torvalds 已提交
624 625 626
	return err;
}

627 628
bool inet6_mc_check(struct sock *sk, const struct in6_addr *mc_addr,
		    const struct in6_addr *src_addr)
L
Linus Torvalds 已提交
629 630 631 632
{
	struct ipv6_pinfo *np = inet6_sk(sk);
	struct ipv6_mc_socklist *mc;
	struct ip6_sf_socklist *psl;
633
	bool rv = true;
L
Linus Torvalds 已提交
634

E
Eric Dumazet 已提交
635 636
	rcu_read_lock();
	for_each_pmc_rcu(np, mc) {
L
Linus Torvalds 已提交
637 638 639 640
		if (ipv6_addr_equal(&mc->addr, mc_addr))
			break;
	}
	if (!mc) {
E
Eric Dumazet 已提交
641
		rcu_read_unlock();
642
		return true;
L
Linus Torvalds 已提交
643
	}
644
	read_lock(&mc->sflock);
L
Linus Torvalds 已提交
645 646 647 648 649 650 651 652 653 654 655
	psl = mc->sflist;
	if (!psl) {
		rv = mc->sfmode == MCAST_EXCLUDE;
	} else {
		int i;

		for (i=0; i<psl->sl_count; i++) {
			if (ipv6_addr_equal(&psl->sl_addr[i], src_addr))
				break;
		}
		if (mc->sfmode == MCAST_INCLUDE && i >= psl->sl_count)
656
			rv = false;
L
Linus Torvalds 已提交
657
		if (mc->sfmode == MCAST_EXCLUDE && i < psl->sl_count)
658
			rv = false;
L
Linus Torvalds 已提交
659
	}
660
	read_unlock(&mc->sflock);
E
Eric Dumazet 已提交
661
	rcu_read_unlock();
L
Linus Torvalds 已提交
662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678

	return rv;
}

static void ma_put(struct ifmcaddr6 *mc)
{
	if (atomic_dec_and_test(&mc->mca_refcnt)) {
		in6_dev_put(mc->idev);
		kfree(mc);
	}
}

static void igmp6_group_added(struct ifmcaddr6 *mc)
{
	struct net_device *dev = mc->idev->dev;
	char buf[MAX_ADDR_LEN];

679 680 681 682
	if (IPV6_ADDR_MC_SCOPE(&mc->mca_addr) <
	    IPV6_ADDR_SCOPE_LINKLOCAL)
		return;

L
Linus Torvalds 已提交
683 684 685 686
	spin_lock_bh(&mc->mca_lock);
	if (!(mc->mca_flags&MAF_LOADED)) {
		mc->mca_flags |= MAF_LOADED;
		if (ndisc_mc_map(&mc->mca_addr, buf, dev, 0) == 0)
687
			dev_mc_add(dev, buf);
L
Linus Torvalds 已提交
688 689 690 691 692 693
	}
	spin_unlock_bh(&mc->mca_lock);

	if (!(dev->flags & IFF_UP) || (mc->mca_flags & MAF_NOREPORT))
		return;

694
	if (mld_in_v1_mode(mc->idev)) {
L
Linus Torvalds 已提交
695 696 697 698 699 700 701 702 703 704 705 706 707 708
		igmp6_join_group(mc);
		return;
	}
	/* else v2 */

	mc->mca_crcount = mc->idev->mc_qrv;
	mld_ifc_event(mc->idev);
}

static void igmp6_group_dropped(struct ifmcaddr6 *mc)
{
	struct net_device *dev = mc->idev->dev;
	char buf[MAX_ADDR_LEN];

709 710 711 712
	if (IPV6_ADDR_MC_SCOPE(&mc->mca_addr) <
	    IPV6_ADDR_SCOPE_LINKLOCAL)
		return;

L
Linus Torvalds 已提交
713 714 715 716
	spin_lock_bh(&mc->mca_lock);
	if (mc->mca_flags&MAF_LOADED) {
		mc->mca_flags &= ~MAF_LOADED;
		if (ndisc_mc_map(&mc->mca_addr, buf, dev, 0) == 0)
717
			dev_mc_del(dev, buf);
L
Linus Torvalds 已提交
718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747
	}

	if (mc->mca_flags & MAF_NOREPORT)
		goto done;
	spin_unlock_bh(&mc->mca_lock);

	if (!mc->idev->dead)
		igmp6_leave_group(mc);

	spin_lock_bh(&mc->mca_lock);
	if (del_timer(&mc->mca_timer))
		atomic_dec(&mc->mca_refcnt);
done:
	ip6_mc_clear_src(mc);
	spin_unlock_bh(&mc->mca_lock);
}

/*
 * deleted ifmcaddr6 manipulation
 */
static void mld_add_delrec(struct inet6_dev *idev, struct ifmcaddr6 *im)
{
	struct ifmcaddr6 *pmc;

	/* this is an "ifmcaddr6" for convenience; only the fields below
	 * are actually used. In particular, the refcnt and users are not
	 * used for management of the delete list. Using the same structure
	 * for deleted items allows change reports to use common code with
	 * non-deleted or query-response MCA's.
	 */
748
	pmc = kzalloc(sizeof(*pmc), GFP_ATOMIC);
L
Linus Torvalds 已提交
749 750
	if (!pmc)
		return;
751

L
Linus Torvalds 已提交
752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769
	spin_lock_bh(&im->mca_lock);
	spin_lock_init(&pmc->mca_lock);
	pmc->idev = im->idev;
	in6_dev_hold(idev);
	pmc->mca_addr = im->mca_addr;
	pmc->mca_crcount = idev->mc_qrv;
	pmc->mca_sfmode = im->mca_sfmode;
	if (pmc->mca_sfmode == MCAST_INCLUDE) {
		struct ip6_sf_list *psf;

		pmc->mca_tomb = im->mca_tomb;
		pmc->mca_sources = im->mca_sources;
		im->mca_tomb = im->mca_sources = NULL;
		for (psf=pmc->mca_sources; psf; psf=psf->sf_next)
			psf->sf_crcount = pmc->mca_crcount;
	}
	spin_unlock_bh(&im->mca_lock);

770
	spin_lock_bh(&idev->mc_lock);
L
Linus Torvalds 已提交
771 772
	pmc->next = idev->mc_tomb;
	idev->mc_tomb = pmc;
773
	spin_unlock_bh(&idev->mc_lock);
L
Linus Torvalds 已提交
774 775
}

776
static void mld_del_delrec(struct inet6_dev *idev, const struct in6_addr *pmca)
L
Linus Torvalds 已提交
777 778 779 780
{
	struct ifmcaddr6 *pmc, *pmc_prev;
	struct ip6_sf_list *psf, *psf_next;

781
	spin_lock_bh(&idev->mc_lock);
L
Linus Torvalds 已提交
782 783 784 785 786 787 788 789 790 791 792 793
	pmc_prev = NULL;
	for (pmc=idev->mc_tomb; pmc; pmc=pmc->next) {
		if (ipv6_addr_equal(&pmc->mca_addr, pmca))
			break;
		pmc_prev = pmc;
	}
	if (pmc) {
		if (pmc_prev)
			pmc_prev->next = pmc->next;
		else
			idev->mc_tomb = pmc->next;
	}
794 795
	spin_unlock_bh(&idev->mc_lock);

L
Linus Torvalds 已提交
796 797 798 799 800 801 802 803 804 805 806 807 808 809
	if (pmc) {
		for (psf=pmc->mca_tomb; psf; psf=psf_next) {
			psf_next = psf->sf_next;
			kfree(psf);
		}
		in6_dev_put(pmc->idev);
		kfree(pmc);
	}
}

static void mld_clear_delrec(struct inet6_dev *idev)
{
	struct ifmcaddr6 *pmc, *nextpmc;

810
	spin_lock_bh(&idev->mc_lock);
L
Linus Torvalds 已提交
811 812
	pmc = idev->mc_tomb;
	idev->mc_tomb = NULL;
813
	spin_unlock_bh(&idev->mc_lock);
L
Linus Torvalds 已提交
814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842

	for (; pmc; pmc = nextpmc) {
		nextpmc = pmc->next;
		ip6_mc_clear_src(pmc);
		in6_dev_put(pmc->idev);
		kfree(pmc);
	}

	/* clear dead sources, too */
	read_lock_bh(&idev->lock);
	for (pmc=idev->mc_list; pmc; pmc=pmc->next) {
		struct ip6_sf_list *psf, *psf_next;

		spin_lock_bh(&pmc->mca_lock);
		psf = pmc->mca_tomb;
		pmc->mca_tomb = NULL;
		spin_unlock_bh(&pmc->mca_lock);
		for (; psf; psf=psf_next) {
			psf_next = psf->sf_next;
			kfree(psf);
		}
	}
	read_unlock_bh(&idev->lock);
}


/*
 *	device multicast group inc (add if not found)
 */
843
int ipv6_dev_mc_inc(struct net_device *dev, const struct in6_addr *addr)
L
Linus Torvalds 已提交
844 845 846 847
{
	struct ifmcaddr6 *mc;
	struct inet6_dev *idev;

E
Eric Dumazet 已提交
848
	/* we need to take a reference on idev */
L
Linus Torvalds 已提交
849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875
	idev = in6_dev_get(dev);

	if (idev == NULL)
		return -EINVAL;

	write_lock_bh(&idev->lock);
	if (idev->dead) {
		write_unlock_bh(&idev->lock);
		in6_dev_put(idev);
		return -ENODEV;
	}

	for (mc = idev->mc_list; mc; mc = mc->next) {
		if (ipv6_addr_equal(&mc->mca_addr, addr)) {
			mc->mca_users++;
			write_unlock_bh(&idev->lock);
			ip6_mc_add_src(idev, &mc->mca_addr, MCAST_EXCLUDE, 0,
				NULL, 0);
			in6_dev_put(idev);
			return 0;
		}
	}

	/*
	 *	not found: create a new one.
	 */

876
	mc = kzalloc(sizeof(struct ifmcaddr6), GFP_ATOMIC);
L
Linus Torvalds 已提交
877 878 879 880 881 882 883

	if (mc == NULL) {
		write_unlock_bh(&idev->lock);
		in6_dev_put(idev);
		return -ENOMEM;
	}

884
	setup_timer(&mc->mca_timer, igmp6_timer_handler, (unsigned long)mc);
L
Linus Torvalds 已提交
885

A
Alexey Dobriyan 已提交
886
	mc->mca_addr = *addr;
E
Eric Dumazet 已提交
887
	mc->idev = idev; /* (reference taken) */
L
Linus Torvalds 已提交
888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914
	mc->mca_users = 1;
	/* mca_stamp should be updated upon changes */
	mc->mca_cstamp = mc->mca_tstamp = jiffies;
	atomic_set(&mc->mca_refcnt, 2);
	spin_lock_init(&mc->mca_lock);

	/* initial mode is (EX, empty) */
	mc->mca_sfmode = MCAST_EXCLUDE;
	mc->mca_sfcount[MCAST_EXCLUDE] = 1;

	if (ipv6_addr_is_ll_all_nodes(&mc->mca_addr) ||
	    IPV6_ADDR_MC_SCOPE(&mc->mca_addr) < IPV6_ADDR_SCOPE_LINKLOCAL)
		mc->mca_flags |= MAF_NOREPORT;

	mc->next = idev->mc_list;
	idev->mc_list = mc;
	write_unlock_bh(&idev->lock);

	mld_del_delrec(idev, &mc->mca_addr);
	igmp6_group_added(mc);
	ma_put(mc);
	return 0;
}

/*
 *	device multicast group del
 */
915
int __ipv6_dev_mc_dec(struct inet6_dev *idev, const struct in6_addr *addr)
L
Linus Torvalds 已提交
916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939
{
	struct ifmcaddr6 *ma, **map;

	write_lock_bh(&idev->lock);
	for (map = &idev->mc_list; (ma=*map) != NULL; map = &ma->next) {
		if (ipv6_addr_equal(&ma->mca_addr, addr)) {
			if (--ma->mca_users == 0) {
				*map = ma->next;
				write_unlock_bh(&idev->lock);

				igmp6_group_dropped(ma);

				ma_put(ma);
				return 0;
			}
			write_unlock_bh(&idev->lock);
			return 0;
		}
	}
	write_unlock_bh(&idev->lock);

	return -ENOENT;
}

940
int ipv6_dev_mc_dec(struct net_device *dev, const struct in6_addr *addr)
L
Linus Torvalds 已提交
941
{
E
Eric Dumazet 已提交
942
	struct inet6_dev *idev;
L
Linus Torvalds 已提交
943 944
	int err;

E
Eric Dumazet 已提交
945
	rcu_read_lock();
L
Linus Torvalds 已提交
946

E
Eric Dumazet 已提交
947 948 949 950 951
	idev = __in6_dev_get(dev);
	if (!idev)
		err = -ENODEV;
	else
		err = __ipv6_dev_mc_dec(idev, addr);
L
Linus Torvalds 已提交
952

E
Eric Dumazet 已提交
953
	rcu_read_unlock();
L
Linus Torvalds 已提交
954 955 956 957 958 959
	return err;
}

/*
 *	check if the interface/address pair is valid
 */
960 961
bool ipv6_chk_mcast_addr(struct net_device *dev, const struct in6_addr *group,
			 const struct in6_addr *src_addr)
L
Linus Torvalds 已提交
962 963 964
{
	struct inet6_dev *idev;
	struct ifmcaddr6 *mc;
965
	bool rv = false;
L
Linus Torvalds 已提交
966

E
Eric Dumazet 已提交
967 968
	rcu_read_lock();
	idev = __in6_dev_get(dev);
L
Linus Torvalds 已提交
969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991
	if (idev) {
		read_lock_bh(&idev->lock);
		for (mc = idev->mc_list; mc; mc=mc->next) {
			if (ipv6_addr_equal(&mc->mca_addr, group))
				break;
		}
		if (mc) {
			if (src_addr && !ipv6_addr_any(src_addr)) {
				struct ip6_sf_list *psf;

				spin_lock_bh(&mc->mca_lock);
				for (psf=mc->mca_sources;psf;psf=psf->sf_next) {
					if (ipv6_addr_equal(&psf->sf_addr, src_addr))
						break;
				}
				if (psf)
					rv = psf->sf_count[MCAST_INCLUDE] ||
						psf->sf_count[MCAST_EXCLUDE] !=
						mc->mca_sfcount[MCAST_EXCLUDE];
				else
					rv = mc->mca_sfcount[MCAST_EXCLUDE] !=0;
				spin_unlock_bh(&mc->mca_lock);
			} else
992
				rv = true; /* don't filter unspecified source */
L
Linus Torvalds 已提交
993 994 995
		}
		read_unlock_bh(&idev->lock);
	}
E
Eric Dumazet 已提交
996
	rcu_read_unlock();
L
Linus Torvalds 已提交
997 998 999 1000 1001
	return rv;
}

static void mld_gq_start_timer(struct inet6_dev *idev)
{
1002
	unsigned long tv = net_random() % idev->mc_maxdelay;
L
Linus Torvalds 已提交
1003 1004 1005 1006 1007 1008

	idev->mc_gq_running = 1;
	if (!mod_timer(&idev->mc_gq_timer, jiffies+tv+2))
		in6_dev_hold(idev);
}

1009 1010 1011 1012 1013 1014 1015
static void mld_gq_stop_timer(struct inet6_dev *idev)
{
	idev->mc_gq_running = 0;
	if (del_timer(&idev->mc_gq_timer))
		__in6_dev_put(idev);
}

1016
static void mld_ifc_start_timer(struct inet6_dev *idev, unsigned long delay)
L
Linus Torvalds 已提交
1017
{
1018
	unsigned long tv = net_random() % delay;
L
Linus Torvalds 已提交
1019 1020 1021 1022 1023

	if (!mod_timer(&idev->mc_ifc_timer, jiffies+tv+2))
		in6_dev_hold(idev);
}

1024 1025 1026 1027 1028 1029 1030
static void mld_ifc_stop_timer(struct inet6_dev *idev)
{
	idev->mc_ifc_count = 0;
	if (del_timer(&idev->mc_ifc_timer))
		__in6_dev_put(idev);
}

1031
static void mld_dad_start_timer(struct inet6_dev *idev, unsigned long delay)
1032
{
1033
	unsigned long tv = net_random() % delay;
1034 1035 1036 1037 1038

	if (!mod_timer(&idev->mc_dad_timer, jiffies+tv+2))
		in6_dev_hold(idev);
}

1039 1040 1041 1042 1043 1044
static void mld_dad_stop_timer(struct inet6_dev *idev)
{
	if (del_timer(&idev->mc_dad_timer))
		__in6_dev_put(idev);
}

L
Linus Torvalds 已提交
1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062
/*
 *	IGMP handling (alias multicast ICMPv6 messages)
 */

static void igmp6_group_queried(struct ifmcaddr6 *ma, unsigned long resptime)
{
	unsigned long delay = resptime;

	/* Do not start timer for these addresses */
	if (ipv6_addr_is_ll_all_nodes(&ma->mca_addr) ||
	    IPV6_ADDR_MC_SCOPE(&ma->mca_addr) < IPV6_ADDR_SCOPE_LINKLOCAL)
		return;

	if (del_timer(&ma->mca_timer)) {
		atomic_dec(&ma->mca_refcnt);
		delay = ma->mca_timer.expires - jiffies;
	}

1063 1064 1065
	if (delay >= resptime)
		delay = net_random() % resptime;

L
Linus Torvalds 已提交
1066 1067 1068 1069 1070 1071
	ma->mca_timer.expires = jiffies + delay;
	if (!mod_timer(&ma->mca_timer, jiffies + delay))
		atomic_inc(&ma->mca_refcnt);
	ma->mca_flags |= MAF_TIMER_RUNNING;
}

1072
/* mark EXCLUDE-mode sources */
1073 1074
static bool mld_xmarksources(struct ifmcaddr6 *pmc, int nsrcs,
			     const struct in6_addr *srcs)
L
Linus Torvalds 已提交
1075 1076 1077 1078 1079 1080 1081 1082
{
	struct ip6_sf_list *psf;
	int i, scount;

	scount = 0;
	for (psf=pmc->mca_sources; psf; psf=psf->sf_next) {
		if (scount == nsrcs)
			break;
1083 1084
		for (i=0; i<nsrcs; i++) {
			/* skip inactive filters */
1085
			if (psf->sf_count[MCAST_INCLUDE] ||
1086 1087
			    pmc->mca_sfcount[MCAST_EXCLUDE] !=
			    psf->sf_count[MCAST_EXCLUDE])
1088
				break;
1089 1090 1091 1092 1093 1094 1095 1096
			if (ipv6_addr_equal(&srcs[i], &psf->sf_addr)) {
				scount++;
				break;
			}
		}
	}
	pmc->mca_flags &= ~MAF_GSQUERY;
	if (scount == nsrcs)	/* all sources excluded */
1097 1098
		return false;
	return true;
1099 1100
}

1101 1102
static bool mld_marksources(struct ifmcaddr6 *pmc, int nsrcs,
			    const struct in6_addr *srcs)
1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116
{
	struct ip6_sf_list *psf;
	int i, scount;

	if (pmc->mca_sfmode == MCAST_EXCLUDE)
		return mld_xmarksources(pmc, nsrcs, srcs);

	/* mark INCLUDE-mode sources */

	scount = 0;
	for (psf=pmc->mca_sources; psf; psf=psf->sf_next) {
		if (scount == nsrcs)
			break;
		for (i=0; i<nsrcs; i++) {
L
Linus Torvalds 已提交
1117 1118 1119 1120 1121
			if (ipv6_addr_equal(&srcs[i], &psf->sf_addr)) {
				psf->sf_gsresp = 1;
				scount++;
				break;
			}
1122 1123 1124 1125
		}
	}
	if (!scount) {
		pmc->mca_flags &= ~MAF_GSQUERY;
1126
		return false;
L
Linus Torvalds 已提交
1127
	}
1128
	pmc->mca_flags |= MAF_GSQUERY;
1129
	return true;
L
Linus Torvalds 已提交
1130 1131
}

1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154
static int mld_force_mld_version(const struct inet6_dev *idev)
{
	/* Normally, both are 0 here. If enforcement to a particular is
	 * being used, individual device enforcement will have a lower
	 * precedence over 'all' device (.../conf/all/force_mld_version).
	 */

	if (dev_net(idev->dev)->ipv6.devconf_all->force_mld_version != 0)
		return dev_net(idev->dev)->ipv6.devconf_all->force_mld_version;
	else
		return idev->cnf.force_mld_version;
}

static bool mld_in_v2_mode_only(const struct inet6_dev *idev)
{
	return mld_force_mld_version(idev) == 2;
}

static bool mld_in_v1_mode_only(const struct inet6_dev *idev)
{
	return mld_force_mld_version(idev) == 1;
}

1155 1156
static bool mld_in_v1_mode(const struct inet6_dev *idev)
{
1157 1158 1159
	if (mld_in_v2_mode_only(idev))
		return false;
	if (mld_in_v1_mode_only(idev))
1160 1161 1162 1163 1164 1165 1166
		return true;
	if (idev->mc_v1_seen && time_before(jiffies, idev->mc_v1_seen))
		return true;

	return false;
}

1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237
static void mld_set_v1_mode(struct inet6_dev *idev)
{
	/* RFC3810, relevant sections:
	 *  - 9.1. Robustness Variable
	 *  - 9.2. Query Interval
	 *  - 9.3. Query Response Interval
	 *  - 9.12. Older Version Querier Present Timeout
	 */
	unsigned long switchback;

	switchback = (idev->mc_qrv * idev->mc_qi) + idev->mc_qri;

	idev->mc_v1_seen = jiffies + switchback;
}

static void mld_update_qrv(struct inet6_dev *idev,
			   const struct mld2_query *mlh2)
{
	/* RFC3810, relevant sections:
	 *  - 5.1.8. QRV (Querier's Robustness Variable)
	 *  - 9.1. Robustness Variable
	 */

	/* The value of the Robustness Variable MUST NOT be zero,
	 * and SHOULD NOT be one. Catch this here if we ever run
	 * into such a case in future.
	 */
	WARN_ON(idev->mc_qrv == 0);

	if (mlh2->mld2q_qrv > 0)
		idev->mc_qrv = mlh2->mld2q_qrv;

	if (unlikely(idev->mc_qrv < 2)) {
		net_warn_ratelimited("IPv6: MLD: clamping QRV from %u to %u!\n",
				     idev->mc_qrv, MLD_QRV_DEFAULT);
		idev->mc_qrv = MLD_QRV_DEFAULT;
	}
}

static void mld_update_qi(struct inet6_dev *idev,
			  const struct mld2_query *mlh2)
{
	/* RFC3810, relevant sections:
	 *  - 5.1.9. QQIC (Querier's Query Interval Code)
	 *  - 9.2. Query Interval
	 *  - 9.12. Older Version Querier Present Timeout
	 *    (the [Query Interval] in the last Query received)
	 */
	unsigned long mc_qqi;

	if (mlh2->mld2q_qqic < 128) {
		mc_qqi = mlh2->mld2q_qqic;
	} else {
		unsigned long mc_man, mc_exp;

		mc_exp = MLDV2_QQIC_EXP(mlh2->mld2q_qqic);
		mc_man = MLDV2_QQIC_MAN(mlh2->mld2q_qqic);

		mc_qqi = (mc_man | 0x10) << (mc_exp + 3);
	}

	idev->mc_qi = mc_qqi * HZ;
}

static void mld_update_qri(struct inet6_dev *idev,
			   const struct mld2_query *mlh2)
{
	/* RFC3810, relevant sections:
	 *  - 5.1.3. Maximum Response Code
	 *  - 9.3. Query Response Interval
	 */
1238
	idev->mc_qri = msecs_to_jiffies(mldv2_mrc(mlh2));
1239 1240
}

1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256
static int mld_process_v1(struct inet6_dev *idev, struct mld_msg *mld,
			  unsigned long *max_delay)
{
	unsigned long mldv1_md;

	/* Ignore v1 queries */
	if (mld_in_v2_mode_only(idev))
		return -EINVAL;

	/* MLDv1 router present */
	mldv1_md = ntohs(mld->mld_maxdelay);
	*max_delay = max(msecs_to_jiffies(mldv1_md), 1UL);

	mld_set_v1_mode(idev);

	/* cancel MLDv2 report timer */
1257
	mld_gq_stop_timer(idev);
1258
	/* cancel the interface change timer */
1259
	mld_ifc_stop_timer(idev);
1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283
	/* clear deleted report items */
	mld_clear_delrec(idev);

	return 0;
}

static int mld_process_v2(struct inet6_dev *idev, struct mld2_query *mld,
			  unsigned long *max_delay)
{
	/* hosts need to stay in MLDv1 mode, discard MLDv2 queries */
	if (mld_in_v1_mode(idev))
		return -EINVAL;

	*max_delay = max(msecs_to_jiffies(mldv2_mrc(mld)), 1UL);

	mld_update_qrv(idev, mld);
	mld_update_qi(idev, mld);
	mld_update_qri(idev, mld);

	idev->mc_maxdelay = *max_delay;

	return 0;
}

E
Eric Dumazet 已提交
1284
/* called with rcu_read_lock() */
L
Linus Torvalds 已提交
1285 1286
int igmp6_event_query(struct sk_buff *skb)
{
1287
	struct mld2_query *mlh2 = NULL;
L
Linus Torvalds 已提交
1288
	struct ifmcaddr6 *ma;
1289
	const struct in6_addr *group;
L
Linus Torvalds 已提交
1290 1291
	unsigned long max_delay;
	struct inet6_dev *idev;
1292
	struct mld_msg *mld;
L
Linus Torvalds 已提交
1293 1294
	int group_type;
	int mark = 0;
1295
	int len, err;
L
Linus Torvalds 已提交
1296 1297 1298 1299 1300

	if (!pskb_may_pull(skb, sizeof(struct in6_addr)))
		return -EINVAL;

	/* compute payload length excluding extension headers */
1301
	len = ntohs(ipv6_hdr(skb)->payload_len) + sizeof(struct ipv6hdr);
1302
	len -= skb_network_header_len(skb);
L
Linus Torvalds 已提交
1303 1304

	/* Drop queries with not link local source */
1305
	if (!(ipv6_addr_type(&ipv6_hdr(skb)->saddr) & IPV6_ADDR_LINKLOCAL))
L
Linus Torvalds 已提交
1306 1307
		return -EINVAL;

E
Eric Dumazet 已提交
1308
	idev = __in6_dev_get(skb->dev);
L
Linus Torvalds 已提交
1309 1310 1311
	if (idev == NULL)
		return 0;

1312 1313
	mld = (struct mld_msg *)icmp6_hdr(skb);
	group = &mld->mld_mca;
L
Linus Torvalds 已提交
1314 1315 1316
	group_type = ipv6_addr_type(group);

	if (group_type != IPV6_ADDR_ANY &&
E
Eric Dumazet 已提交
1317
	    !(group_type&IPV6_ADDR_MULTICAST))
L
Linus Torvalds 已提交
1318 1319
		return -EINVAL;

1320
	if (len == MLD_V1_QUERY_LEN) {
1321 1322 1323
		err = mld_process_v1(idev, mld, &max_delay);
		if (err < 0)
			return err;
1324
	} else if (len >= MLD_V2_QUERY_LEN_MIN) {
1325
		int srcs_offset = sizeof(struct mld2_query) -
1326
				  sizeof(struct icmp6hdr);
1327

E
Eric Dumazet 已提交
1328
		if (!pskb_may_pull(skb, srcs_offset))
1329
			return -EINVAL;
E
Eric Dumazet 已提交
1330

1331
		mlh2 = (struct mld2_query *)skb_transport_header(skb);
1332

1333 1334 1335
		err = mld_process_v2(idev, mlh2, &max_delay);
		if (err < 0)
			return err;
1336

L
Linus Torvalds 已提交
1337
		if (group_type == IPV6_ADDR_ANY) { /* general query */
E
Eric Dumazet 已提交
1338
			if (mlh2->mld2q_nsrcs)
L
Linus Torvalds 已提交
1339
				return -EINVAL; /* no sources allowed */
E
Eric Dumazet 已提交
1340

L
Linus Torvalds 已提交
1341 1342 1343 1344
			mld_gq_start_timer(idev);
			return 0;
		}
		/* mark sources to include, if group & source-specific */
1345
		if (mlh2->mld2q_nsrcs != 0) {
1346
			if (!pskb_may_pull(skb, srcs_offset +
E
Eric Dumazet 已提交
1347
			    ntohs(mlh2->mld2q_nsrcs) * sizeof(struct in6_addr)))
1348
				return -EINVAL;
E
Eric Dumazet 已提交
1349

1350
			mlh2 = (struct mld2_query *)skb_transport_header(skb);
1351 1352
			mark = 1;
		}
E
Eric Dumazet 已提交
1353
	} else
L
Linus Torvalds 已提交
1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364
		return -EINVAL;

	read_lock_bh(&idev->lock);
	if (group_type == IPV6_ADDR_ANY) {
		for (ma = idev->mc_list; ma; ma=ma->next) {
			spin_lock_bh(&ma->mca_lock);
			igmp6_group_queried(ma, max_delay);
			spin_unlock_bh(&ma->mca_lock);
		}
	} else {
		for (ma = idev->mc_list; ma; ma=ma->next) {
1365
			if (!ipv6_addr_equal(group, &ma->mca_addr))
L
Linus Torvalds 已提交
1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378
				continue;
			spin_lock_bh(&ma->mca_lock);
			if (ma->mca_flags & MAF_TIMER_RUNNING) {
				/* gsquery <- gsquery && mark */
				if (!mark)
					ma->mca_flags &= ~MAF_GSQUERY;
			} else {
				/* gsquery <- mark */
				if (mark)
					ma->mca_flags |= MAF_GSQUERY;
				else
					ma->mca_flags &= ~MAF_GSQUERY;
			}
1379
			if (!(ma->mca_flags & MAF_GSQUERY) ||
1380
			    mld_marksources(ma, ntohs(mlh2->mld2q_nsrcs), mlh2->mld2q_srcs))
1381
				igmp6_group_queried(ma, max_delay);
L
Linus Torvalds 已提交
1382
			spin_unlock_bh(&ma->mca_lock);
1383
			break;
L
Linus Torvalds 已提交
1384 1385 1386 1387 1388 1389 1390
		}
	}
	read_unlock_bh(&idev->lock);

	return 0;
}

E
Eric Dumazet 已提交
1391
/* called with rcu_read_lock() */
L
Linus Torvalds 已提交
1392 1393 1394 1395
int igmp6_event_report(struct sk_buff *skb)
{
	struct ifmcaddr6 *ma;
	struct inet6_dev *idev;
1396
	struct mld_msg *mld;
L
Linus Torvalds 已提交
1397 1398 1399 1400 1401 1402
	int addr_type;

	/* Our own report looped back. Ignore it. */
	if (skb->pkt_type == PACKET_LOOPBACK)
		return 0;

1403 1404 1405 1406 1407
	/* send our report if the MC router may not have heard this report */
	if (skb->pkt_type != PACKET_MULTICAST &&
	    skb->pkt_type != PACKET_BROADCAST)
		return 0;

1408
	if (!pskb_may_pull(skb, sizeof(*mld) - sizeof(struct icmp6hdr)))
L
Linus Torvalds 已提交
1409 1410
		return -EINVAL;

1411
	mld = (struct mld_msg *)icmp6_hdr(skb);
L
Linus Torvalds 已提交
1412 1413

	/* Drop reports with not link local source */
1414
	addr_type = ipv6_addr_type(&ipv6_hdr(skb)->saddr);
1415
	if (addr_type != IPV6_ADDR_ANY &&
L
Linus Torvalds 已提交
1416 1417 1418
	    !(addr_type&IPV6_ADDR_LINKLOCAL))
		return -EINVAL;

E
Eric Dumazet 已提交
1419
	idev = __in6_dev_get(skb->dev);
L
Linus Torvalds 已提交
1420 1421 1422 1423 1424 1425 1426 1427 1428
	if (idev == NULL)
		return -ENODEV;

	/*
	 *	Cancel the timer for this group
	 */

	read_lock_bh(&idev->lock);
	for (ma = idev->mc_list; ma; ma=ma->next) {
1429
		if (ipv6_addr_equal(&ma->mca_addr, &mld->mld_mca)) {
L
Linus Torvalds 已提交
1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441
			spin_lock(&ma->mca_lock);
			if (del_timer(&ma->mca_timer))
				atomic_dec(&ma->mca_refcnt);
			ma->mca_flags &= ~(MAF_LAST_REPORTER|MAF_TIMER_RUNNING);
			spin_unlock(&ma->mca_lock);
			break;
		}
	}
	read_unlock_bh(&idev->lock);
	return 0;
}

1442 1443
static bool is_in(struct ifmcaddr6 *pmc, struct ip6_sf_list *psf, int type,
		  int gdeleted, int sdeleted)
L
Linus Torvalds 已提交
1444 1445 1446 1447 1448
{
	switch (type) {
	case MLD2_MODE_IS_INCLUDE:
	case MLD2_MODE_IS_EXCLUDE:
		if (gdeleted || sdeleted)
1449
			return false;
1450 1451
		if (!((pmc->mca_flags & MAF_GSQUERY) && !psf->sf_gsresp)) {
			if (pmc->mca_sfmode == MCAST_INCLUDE)
1452
				return true;
1453 1454 1455 1456
			/* don't include if this source is excluded
			 * in all filters
			 */
			if (psf->sf_count[MCAST_INCLUDE])
1457
				return type == MLD2_MODE_IS_INCLUDE;
1458 1459 1460
			return pmc->mca_sfcount[MCAST_EXCLUDE] ==
				psf->sf_count[MCAST_EXCLUDE];
		}
1461
		return false;
L
Linus Torvalds 已提交
1462 1463
	case MLD2_CHANGE_TO_INCLUDE:
		if (gdeleted || sdeleted)
1464
			return false;
L
Linus Torvalds 已提交
1465 1466 1467
		return psf->sf_count[MCAST_INCLUDE] != 0;
	case MLD2_CHANGE_TO_EXCLUDE:
		if (gdeleted || sdeleted)
1468
			return false;
L
Linus Torvalds 已提交
1469 1470
		if (pmc->mca_sfcount[MCAST_EXCLUDE] == 0 ||
		    psf->sf_count[MCAST_INCLUDE])
1471
			return false;
L
Linus Torvalds 已提交
1472 1473 1474 1475
		return pmc->mca_sfcount[MCAST_EXCLUDE] ==
			psf->sf_count[MCAST_EXCLUDE];
	case MLD2_ALLOW_NEW_SOURCES:
		if (gdeleted || !psf->sf_crcount)
1476
			return false;
L
Linus Torvalds 已提交
1477 1478 1479 1480 1481 1482
		return (pmc->mca_sfmode == MCAST_INCLUDE) ^ sdeleted;
	case MLD2_BLOCK_OLD_SOURCES:
		if (pmc->mca_sfmode == MCAST_INCLUDE)
			return gdeleted || (psf->sf_crcount && sdeleted);
		return psf->sf_crcount && !gdeleted && !sdeleted;
	}
1483
	return false;
L
Linus Torvalds 已提交
1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499
}

static int
mld_scount(struct ifmcaddr6 *pmc, int type, int gdeleted, int sdeleted)
{
	struct ip6_sf_list *psf;
	int scount = 0;

	for (psf=pmc->mca_sources; psf; psf=psf->sf_next) {
		if (!is_in(pmc, psf, type, gdeleted, sdeleted))
			continue;
		scount++;
	}
	return scount;
}

1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524
static void ip6_mc_hdr(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)
{
	struct ipv6hdr *hdr;

	skb->protocol = htons(ETH_P_IPV6);
	skb->dev = dev;

	skb_reset_network_header(skb);
	skb_put(skb, sizeof(struct ipv6hdr));
	hdr = ipv6_hdr(skb);

	ip6_flow_hdr(hdr, 0, 0);

	hdr->payload_len = htons(len);
	hdr->nexthdr = proto;
	hdr->hop_limit = inet6_sk(sk)->hop_limit;

	hdr->saddr = *saddr;
	hdr->daddr = *daddr;
}

1525
static struct sk_buff *mld_newpack(struct inet6_dev *idev, int size)
L
Linus Torvalds 已提交
1526
{
1527
	struct net_device *dev = idev->dev;
1528
	struct net *net = dev_net(dev);
1529
	struct sock *sk = net->ipv6.igmp_sk;
L
Linus Torvalds 已提交
1530 1531 1532
	struct sk_buff *skb;
	struct mld2_report *pmr;
	struct in6_addr addr_buf;
1533
	const struct in6_addr *saddr;
1534 1535
	int hlen = LL_RESERVED_SPACE(dev);
	int tlen = dev->needed_tailroom;
L
Linus Torvalds 已提交
1536 1537 1538 1539 1540 1541
	int err;
	u8 ra[8] = { IPPROTO_ICMPV6, 0,
		     IPV6_TLV_ROUTERALERT, 2, 0, 0,
		     IPV6_TLV_PADN, 0 };

	/* we assume size > sizeof(ra) here */
1542
	size += hlen + tlen;
1543 1544 1545
	/* limit our allocations to order-0 page */
	size = min_t(int, size, SKB_MAX_ORDER(0, 0));
	skb = sock_alloc_send_skb(sk, size, 1, &err);
L
Linus Torvalds 已提交
1546

S
Stephen Hemminger 已提交
1547
	if (!skb)
L
Linus Torvalds 已提交
1548 1549
		return NULL;

1550
	skb->priority = TC_PRIO_CONTROL;
1551
	skb_reserve(skb, hlen);
L
Linus Torvalds 已提交
1552

1553
	if (__ipv6_get_lladdr(idev, &addr_buf, IFA_F_TENTATIVE)) {
L
Linus Torvalds 已提交
1554
		/* <draft-ietf-magma-mld-source-05.txt>:
1555
		 * use unspecified address as the source address
L
Linus Torvalds 已提交
1556 1557
		 * when a valid link-local address is not available.
		 */
1558 1559 1560
		saddr = &in6addr_any;
	} else
		saddr = &addr_buf;
L
Linus Torvalds 已提交
1561

1562
	ip6_mc_hdr(sk, skb, dev, saddr, &mld2_all_mcr, NEXTHDR_HOP, 0);
L
Linus Torvalds 已提交
1563 1564 1565

	memcpy(skb_put(skb, sizeof(ra)), ra, sizeof(ra));

1566
	skb_set_transport_header(skb, skb_tail_pointer(skb) - skb->data);
1567 1568
	skb_put(skb, sizeof(*pmr));
	pmr = (struct mld2_report *)skb_transport_header(skb);
1569 1570 1571 1572 1573
	pmr->mld2r_type = ICMPV6_MLD2_REPORT;
	pmr->mld2r_resv1 = 0;
	pmr->mld2r_cksum = 0;
	pmr->mld2r_resv2 = 0;
	pmr->mld2r_ngrec = 0;
L
Linus Torvalds 已提交
1574 1575 1576 1577 1578
	return skb;
}

static void mld_sendpack(struct sk_buff *skb)
{
1579
	struct ipv6hdr *pip6 = ipv6_hdr(skb);
1580 1581
	struct mld2_report *pmr =
			      (struct mld2_report *)skb_transport_header(skb);
L
Linus Torvalds 已提交
1582
	int payload_len, mldlen;
E
Eric Dumazet 已提交
1583
	struct inet6_dev *idev;
1584
	struct net *net = dev_net(skb->dev);
L
Linus Torvalds 已提交
1585
	int err;
1586
	struct flowi6 fl6;
E
Eric Dumazet 已提交
1587
	struct dst_entry *dst;
L
Linus Torvalds 已提交
1588

E
Eric Dumazet 已提交
1589 1590
	rcu_read_lock();
	idev = __in6_dev_get(skb->dev);
1591 1592
	IP6_UPD_PO_STATS(net, idev, IPSTATS_MIB_OUT, skb->len);

1593 1594 1595
	payload_len = (skb_tail_pointer(skb) - skb_network_header(skb)) -
		sizeof(*pip6);
	mldlen = skb_tail_pointer(skb) - skb_transport_header(skb);
L
Linus Torvalds 已提交
1596 1597
	pip6->payload_len = htons(payload_len);

1598 1599 1600 1601
	pmr->mld2r_cksum = csum_ipv6_magic(&pip6->saddr, &pip6->daddr, mldlen,
					   IPPROTO_ICMPV6,
					   csum_partial(skb_transport_header(skb),
							mldlen, 0));
1602

1603
	icmpv6_flow_init(net->ipv6.igmp_sk, &fl6, ICMPV6_MLD2_REPORT,
1604 1605
			 &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
			 skb->dev->ifindex);
1606
	dst = icmp6_dst_alloc(skb->dev, &fl6);
1607

1608 1609 1610 1611 1612
	err = 0;
	if (IS_ERR(dst)) {
		err = PTR_ERR(dst);
		dst = NULL;
	}
E
Eric Dumazet 已提交
1613
	skb_dst_set(skb, dst);
1614 1615 1616
	if (err)
		goto err_out;

1617 1618
	payload_len = skb->len;

1619
	err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT, skb, NULL, skb->dev,
1620 1621
		      dst_output);
out:
L
Linus Torvalds 已提交
1622
	if (!err) {
1623
		ICMP6MSGOUT_INC_STATS_BH(net, idev, ICMPV6_MLD2_REPORT);
1624
		ICMP6_INC_STATS_BH(net, idev, ICMP6_MIB_OUTMSGS);
1625
		IP6_UPD_PO_STATS_BH(net, idev, IPSTATS_MIB_OUTMCAST, payload_len);
L
Linus Torvalds 已提交
1626
	} else
1627
		IP6_INC_STATS_BH(net, idev, IPSTATS_MIB_OUTDISCARDS);
L
Linus Torvalds 已提交
1628

E
Eric Dumazet 已提交
1629
	rcu_read_unlock();
1630 1631 1632 1633 1634
	return;

err_out:
	kfree_skb(skb);
	goto out;
L
Linus Torvalds 已提交
1635 1636 1637 1638
}

static int grec_size(struct ifmcaddr6 *pmc, int type, int gdel, int sdel)
{
1639
	return sizeof(struct mld2_grec) + 16 * mld_scount(pmc,type,gdel,sdel);
L
Linus Torvalds 已提交
1640 1641 1642 1643 1644 1645 1646 1647 1648 1649
}

static struct sk_buff *add_grhead(struct sk_buff *skb, struct ifmcaddr6 *pmc,
	int type, struct mld2_grec **ppgr)
{
	struct net_device *dev = pmc->idev->dev;
	struct mld2_report *pmr;
	struct mld2_grec *pgr;

	if (!skb)
1650
		skb = mld_newpack(pmc->idev, dev->mtu);
L
Linus Torvalds 已提交
1651 1652 1653 1654 1655 1656 1657
	if (!skb)
		return NULL;
	pgr = (struct mld2_grec *)skb_put(skb, sizeof(struct mld2_grec));
	pgr->grec_type = type;
	pgr->grec_auxwords = 0;
	pgr->grec_nsrcs = 0;
	pgr->grec_mca = pmc->mca_addr;	/* structure copy */
1658
	pmr = (struct mld2_report *)skb_transport_header(skb);
1659
	pmr->mld2r_ngrec = htons(ntohs(pmr->mld2r_ngrec)+1);
L
Linus Torvalds 已提交
1660 1661 1662 1663 1664 1665 1666 1667 1668 1669
	*ppgr = pgr;
	return skb;
}

#define AVAILABLE(skb) ((skb) ? ((skb)->dev ? (skb)->dev->mtu - (skb)->len : \
	skb_tailroom(skb)) : 0)

static struct sk_buff *add_grec(struct sk_buff *skb, struct ifmcaddr6 *pmc,
	int type, int gdeleted, int sdeleted)
{
1670 1671
	struct inet6_dev *idev = pmc->idev;
	struct net_device *dev = idev->dev;
L
Linus Torvalds 已提交
1672 1673 1674
	struct mld2_report *pmr;
	struct mld2_grec *pgr = NULL;
	struct ip6_sf_list *psf, *psf_next, *psf_prev, **psf_list;
1675
	int scount, stotal, first, isquery, truncate;
L
Linus Torvalds 已提交
1676 1677 1678 1679 1680 1681 1682 1683 1684

	if (pmc->mca_flags & MAF_NOREPORT)
		return skb;

	isquery = type == MLD2_MODE_IS_INCLUDE ||
		  type == MLD2_MODE_IS_EXCLUDE;
	truncate = type == MLD2_MODE_IS_EXCLUDE ||
		    type == MLD2_CHANGE_TO_EXCLUDE;

1685 1686
	stotal = scount = 0;

L
Linus Torvalds 已提交
1687 1688
	psf_list = sdeleted ? &pmc->mca_tomb : &pmc->mca_sources;

1689 1690 1691
	if (!*psf_list)
		goto empty_source;

1692
	pmr = skb ? (struct mld2_report *)skb_transport_header(skb) : NULL;
L
Linus Torvalds 已提交
1693 1694 1695

	/* EX and TO_EX get a fresh packet, if needed */
	if (truncate) {
1696
		if (pmr && pmr->mld2r_ngrec &&
L
Linus Torvalds 已提交
1697 1698 1699
		    AVAILABLE(skb) < grec_size(pmc, type, gdeleted, sdeleted)) {
			if (skb)
				mld_sendpack(skb);
1700
			skb = mld_newpack(idev, dev->mtu);
L
Linus Torvalds 已提交
1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726
		}
	}
	first = 1;
	psf_prev = NULL;
	for (psf=*psf_list; psf; psf=psf_next) {
		struct in6_addr *psrc;

		psf_next = psf->sf_next;

		if (!is_in(pmc, psf, type, gdeleted, sdeleted)) {
			psf_prev = psf;
			continue;
		}

		/* clear marks on query responses */
		if (isquery)
			psf->sf_gsresp = 0;

		if (AVAILABLE(skb) < sizeof(*psrc) +
		    first*sizeof(struct mld2_grec)) {
			if (truncate && !first)
				break;	 /* truncate these */
			if (pgr)
				pgr->grec_nsrcs = htons(scount);
			if (skb)
				mld_sendpack(skb);
1727
			skb = mld_newpack(idev, dev->mtu);
L
Linus Torvalds 已提交
1728 1729 1730 1731 1732 1733 1734
			first = 1;
			scount = 0;
		}
		if (first) {
			skb = add_grhead(skb, pmc, type, &pgr);
			first = 0;
		}
1735 1736
		if (!skb)
			return NULL;
L
Linus Torvalds 已提交
1737 1738
		psrc = (struct in6_addr *)skb_put(skb, sizeof(*psrc));
		*psrc = psf->sf_addr;
1739
		scount++; stotal++;
L
Linus Torvalds 已提交
1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753
		if ((type == MLD2_ALLOW_NEW_SOURCES ||
		     type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount) {
			psf->sf_crcount--;
			if ((sdeleted || gdeleted) && psf->sf_crcount == 0) {
				if (psf_prev)
					psf_prev->sf_next = psf->sf_next;
				else
					*psf_list = psf->sf_next;
				kfree(psf);
				continue;
			}
		}
		psf_prev = psf;
	}
1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768

empty_source:
	if (!stotal) {
		if (type == MLD2_ALLOW_NEW_SOURCES ||
		    type == MLD2_BLOCK_OLD_SOURCES)
			return skb;
		if (pmc->mca_crcount || isquery) {
			/* make sure we have room for group header */
			if (skb && AVAILABLE(skb) < sizeof(struct mld2_grec)) {
				mld_sendpack(skb);
				skb = NULL; /* add_grhead will get a new one */
			}
			skb = add_grhead(skb, pmc, type, &pgr);
		}
	}
L
Linus Torvalds 已提交
1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781
	if (pgr)
		pgr->grec_nsrcs = htons(scount);

	if (isquery)
		pmc->mca_flags &= ~MAF_GSQUERY;	/* clear query state */
	return skb;
}

static void mld_send_report(struct inet6_dev *idev, struct ifmcaddr6 *pmc)
{
	struct sk_buff *skb = NULL;
	int type;

1782
	read_lock_bh(&idev->lock);
L
Linus Torvalds 已提交
1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803
	if (!pmc) {
		for (pmc=idev->mc_list; pmc; pmc=pmc->next) {
			if (pmc->mca_flags & MAF_NOREPORT)
				continue;
			spin_lock_bh(&pmc->mca_lock);
			if (pmc->mca_sfcount[MCAST_EXCLUDE])
				type = MLD2_MODE_IS_EXCLUDE;
			else
				type = MLD2_MODE_IS_INCLUDE;
			skb = add_grec(skb, pmc, type, 0, 0);
			spin_unlock_bh(&pmc->mca_lock);
		}
	} else {
		spin_lock_bh(&pmc->mca_lock);
		if (pmc->mca_sfcount[MCAST_EXCLUDE])
			type = MLD2_MODE_IS_EXCLUDE;
		else
			type = MLD2_MODE_IS_INCLUDE;
		skb = add_grec(skb, pmc, type, 0, 0);
		spin_unlock_bh(&pmc->mca_lock);
	}
1804
	read_unlock_bh(&idev->lock);
L
Linus Torvalds 已提交
1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836
	if (skb)
		mld_sendpack(skb);
}

/*
 * remove zero-count source records from a source filter list
 */
static void mld_clear_zeros(struct ip6_sf_list **ppsf)
{
	struct ip6_sf_list *psf_prev, *psf_next, *psf;

	psf_prev = NULL;
	for (psf=*ppsf; psf; psf = psf_next) {
		psf_next = psf->sf_next;
		if (psf->sf_crcount == 0) {
			if (psf_prev)
				psf_prev->sf_next = psf->sf_next;
			else
				*ppsf = psf->sf_next;
			kfree(psf);
		} else
			psf_prev = psf;
	}
}

static void mld_send_cr(struct inet6_dev *idev)
{
	struct ifmcaddr6 *pmc, *pmc_prev, *pmc_next;
	struct sk_buff *skb = NULL;
	int type, dtype;

	read_lock_bh(&idev->lock);
1837
	spin_lock(&idev->mc_lock);
L
Linus Torvalds 已提交
1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853

	/* deleted MCA's */
	pmc_prev = NULL;
	for (pmc=idev->mc_tomb; pmc; pmc=pmc_next) {
		pmc_next = pmc->next;
		if (pmc->mca_sfmode == MCAST_INCLUDE) {
			type = MLD2_BLOCK_OLD_SOURCES;
			dtype = MLD2_BLOCK_OLD_SOURCES;
			skb = add_grec(skb, pmc, type, 1, 0);
			skb = add_grec(skb, pmc, dtype, 1, 1);
		}
		if (pmc->mca_crcount) {
			if (pmc->mca_sfmode == MCAST_EXCLUDE) {
				type = MLD2_CHANGE_TO_INCLUDE;
				skb = add_grec(skb, pmc, type, 1, 0);
			}
1854
			pmc->mca_crcount--;
L
Linus Torvalds 已提交
1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870
			if (pmc->mca_crcount == 0) {
				mld_clear_zeros(&pmc->mca_tomb);
				mld_clear_zeros(&pmc->mca_sources);
			}
		}
		if (pmc->mca_crcount == 0 && !pmc->mca_tomb &&
		    !pmc->mca_sources) {
			if (pmc_prev)
				pmc_prev->next = pmc_next;
			else
				idev->mc_tomb = pmc_next;
			in6_dev_put(pmc->idev);
			kfree(pmc);
		} else
			pmc_prev = pmc;
	}
1871
	spin_unlock(&idev->mc_lock);
L
Linus Torvalds 已提交
1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892

	/* change recs */
	for (pmc=idev->mc_list; pmc; pmc=pmc->next) {
		spin_lock_bh(&pmc->mca_lock);
		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
			type = MLD2_BLOCK_OLD_SOURCES;
			dtype = MLD2_ALLOW_NEW_SOURCES;
		} else {
			type = MLD2_ALLOW_NEW_SOURCES;
			dtype = MLD2_BLOCK_OLD_SOURCES;
		}
		skb = add_grec(skb, pmc, type, 0, 0);
		skb = add_grec(skb, pmc, dtype, 0, 1);	/* deleted sources */

		/* filter mode changes */
		if (pmc->mca_crcount) {
			if (pmc->mca_sfmode == MCAST_EXCLUDE)
				type = MLD2_CHANGE_TO_EXCLUDE;
			else
				type = MLD2_CHANGE_TO_INCLUDE;
			skb = add_grec(skb, pmc, type, 0, 0);
1893
			pmc->mca_crcount--;
L
Linus Torvalds 已提交
1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904
		}
		spin_unlock_bh(&pmc->mca_lock);
	}
	read_unlock_bh(&idev->lock);
	if (!skb)
		return;
	(void) mld_sendpack(skb);
}

static void igmp6_send(struct in6_addr *addr, struct net_device *dev, int type)
{
1905
	struct net *net = dev_net(dev);
1906
	struct sock *sk = net->ipv6.igmp_sk;
L
Linus Torvalds 已提交
1907
	struct inet6_dev *idev;
1908
	struct sk_buff *skb;
1909
	struct mld_msg *hdr;
1910
	const struct in6_addr *snd_addr, *saddr;
L
Linus Torvalds 已提交
1911
	struct in6_addr addr_buf;
1912 1913
	int hlen = LL_RESERVED_SPACE(dev);
	int tlen = dev->needed_tailroom;
L
Linus Torvalds 已提交
1914 1915 1916 1917
	int err, len, payload_len, full_len;
	u8 ra[8] = { IPPROTO_ICMPV6, 0,
		     IPV6_TLV_ROUTERALERT, 2, 0, 0,
		     IPV6_TLV_PADN, 0 };
1918
	struct flowi6 fl6;
E
Eric Dumazet 已提交
1919
	struct dst_entry *dst;
L
Linus Torvalds 已提交
1920

1921 1922 1923 1924
	if (type == ICMPV6_MGM_REDUCTION)
		snd_addr = &in6addr_linklocal_allrouters;
	else
		snd_addr = addr;
L
Linus Torvalds 已提交
1925 1926 1927 1928 1929

	len = sizeof(struct icmp6hdr) + sizeof(struct in6_addr);
	payload_len = len + sizeof(ra);
	full_len = sizeof(struct ipv6hdr) + payload_len;

1930 1931 1932 1933 1934
	rcu_read_lock();
	IP6_UPD_PO_STATS(net, __in6_dev_get(dev),
		      IPSTATS_MIB_OUT, full_len);
	rcu_read_unlock();

1935
	skb = sock_alloc_send_skb(sk, hlen + tlen + full_len, 1, &err);
L
Linus Torvalds 已提交
1936 1937

	if (skb == NULL) {
1938
		rcu_read_lock();
1939
		IP6_INC_STATS(net, __in6_dev_get(dev),
1940 1941
			      IPSTATS_MIB_OUTDISCARDS);
		rcu_read_unlock();
L
Linus Torvalds 已提交
1942 1943
		return;
	}
1944
	skb->priority = TC_PRIO_CONTROL;
1945
	skb_reserve(skb, hlen);
L
Linus Torvalds 已提交
1946

1947
	if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
L
Linus Torvalds 已提交
1948
		/* <draft-ietf-magma-mld-source-05.txt>:
1949
		 * use unspecified address as the source address
L
Linus Torvalds 已提交
1950 1951
		 * when a valid link-local address is not available.
		 */
1952 1953 1954
		saddr = &in6addr_any;
	} else
		saddr = &addr_buf;
L
Linus Torvalds 已提交
1955

1956
	ip6_mc_hdr(sk, skb, dev, saddr, snd_addr, NEXTHDR_HOP, payload_len);
L
Linus Torvalds 已提交
1957 1958 1959

	memcpy(skb_put(skb, sizeof(ra)), ra, sizeof(ra));

1960 1961 1962
	hdr = (struct mld_msg *) skb_put(skb, sizeof(struct mld_msg));
	memset(hdr, 0, sizeof(struct mld_msg));
	hdr->mld_type = type;
A
Alexey Dobriyan 已提交
1963
	hdr->mld_mca = *addr;
L
Linus Torvalds 已提交
1964

1965 1966 1967
	hdr->mld_cksum = csum_ipv6_magic(saddr, snd_addr, len,
					 IPPROTO_ICMPV6,
					 csum_partial(hdr, len, 0));
L
Linus Torvalds 已提交
1968

E
Eric Dumazet 已提交
1969 1970
	rcu_read_lock();
	idev = __in6_dev_get(skb->dev);
L
Linus Torvalds 已提交
1971

1972
	icmpv6_flow_init(sk, &fl6, type,
1973 1974
			 &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
			 skb->dev->ifindex);
1975
	dst = icmp6_dst_alloc(skb->dev, &fl6);
1976 1977
	if (IS_ERR(dst)) {
		err = PTR_ERR(dst);
1978
		goto err_out;
1979
	}
1980

E
Eric Dumazet 已提交
1981
	skb_dst_set(skb, dst);
1982
	err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT, skb, NULL, skb->dev,
1983 1984
		      dst_output);
out:
L
Linus Torvalds 已提交
1985
	if (!err) {
1986
		ICMP6MSGOUT_INC_STATS(net, idev, type);
1987
		ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
1988
		IP6_UPD_PO_STATS(net, idev, IPSTATS_MIB_OUTMCAST, full_len);
L
Linus Torvalds 已提交
1989
	} else
1990
		IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
L
Linus Torvalds 已提交
1991

E
Eric Dumazet 已提交
1992
	rcu_read_unlock();
L
Linus Torvalds 已提交
1993
	return;
1994 1995 1996 1997

err_out:
	kfree_skb(skb);
	goto out;
L
Linus Torvalds 已提交
1998 1999
}

2000 2001
static void mld_resend_report(struct inet6_dev *idev)
{
2002
	if (mld_in_v1_mode(idev)) {
2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036
		struct ifmcaddr6 *mcaddr;
		read_lock_bh(&idev->lock);
		for (mcaddr = idev->mc_list; mcaddr; mcaddr = mcaddr->next) {
			if (!(mcaddr->mca_flags & MAF_NOREPORT))
				igmp6_send(&mcaddr->mca_addr, idev->dev,
					   ICMPV6_MGM_REPORT);
		}
		read_unlock_bh(&idev->lock);
	} else {
		mld_send_report(idev, NULL);
	}
}

void ipv6_mc_dad_complete(struct inet6_dev *idev)
{
	idev->mc_dad_count = idev->mc_qrv;
	if (idev->mc_dad_count) {
		mld_resend_report(idev);
		idev->mc_dad_count--;
		if (idev->mc_dad_count)
			mld_dad_start_timer(idev, idev->mc_maxdelay);
	}
}

static void mld_dad_timer_expire(unsigned long data)
{
	struct inet6_dev *idev = (struct inet6_dev *)data;

	mld_resend_report(idev);
	if (idev->mc_dad_count) {
		idev->mc_dad_count--;
		if (idev->mc_dad_count)
			mld_dad_start_timer(idev, idev->mc_maxdelay);
	}
2037
	in6_dev_put(idev);
2038 2039
}

L
Linus Torvalds 已提交
2040
static int ip6_mc_del1_src(struct ifmcaddr6 *pmc, int sfmode,
2041
	const struct in6_addr *psfsrc)
L
Linus Torvalds 已提交
2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065
{
	struct ip6_sf_list *psf, *psf_prev;
	int rv = 0;

	psf_prev = NULL;
	for (psf=pmc->mca_sources; psf; psf=psf->sf_next) {
		if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
			break;
		psf_prev = psf;
	}
	if (!psf || psf->sf_count[sfmode] == 0) {
		/* source filter not found, or count wrong =>  bug */
		return -ESRCH;
	}
	psf->sf_count[sfmode]--;
	if (!psf->sf_count[MCAST_INCLUDE] && !psf->sf_count[MCAST_EXCLUDE]) {
		struct inet6_dev *idev = pmc->idev;

		/* no more filters for this source */
		if (psf_prev)
			psf_prev->sf_next = psf->sf_next;
		else
			pmc->mca_sources = psf->sf_next;
		if (psf->sf_oldin && !(pmc->mca_flags & MAF_NOREPORT) &&
2066
		    !mld_in_v1_mode(idev)) {
L
Linus Torvalds 已提交
2067 2068 2069 2070 2071 2072 2073 2074 2075 2076
			psf->sf_crcount = idev->mc_qrv;
			psf->sf_next = pmc->mca_tomb;
			pmc->mca_tomb = psf;
			rv = 1;
		} else
			kfree(psf);
	}
	return rv;
}

2077 2078
static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
L
Linus Torvalds 已提交
2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137
			  int delta)
{
	struct ifmcaddr6 *pmc;
	int	changerec = 0;
	int	i, err;

	if (!idev)
		return -ENODEV;
	read_lock_bh(&idev->lock);
	for (pmc=idev->mc_list; pmc; pmc=pmc->next) {
		if (ipv6_addr_equal(pmca, &pmc->mca_addr))
			break;
	}
	if (!pmc) {
		/* MCA not found?? bug */
		read_unlock_bh(&idev->lock);
		return -ESRCH;
	}
	spin_lock_bh(&pmc->mca_lock);
	sf_markstate(pmc);
	if (!delta) {
		if (!pmc->mca_sfcount[sfmode]) {
			spin_unlock_bh(&pmc->mca_lock);
			read_unlock_bh(&idev->lock);
			return -EINVAL;
		}
		pmc->mca_sfcount[sfmode]--;
	}
	err = 0;
	for (i=0; i<sfcount; i++) {
		int rv = ip6_mc_del1_src(pmc, sfmode, &psfsrc[i]);

		changerec |= rv > 0;
		if (!err && rv < 0)
			err = rv;
	}
	if (pmc->mca_sfmode == MCAST_EXCLUDE &&
	    pmc->mca_sfcount[MCAST_EXCLUDE] == 0 &&
	    pmc->mca_sfcount[MCAST_INCLUDE]) {
		struct ip6_sf_list *psf;

		/* filter mode change */
		pmc->mca_sfmode = MCAST_INCLUDE;
		pmc->mca_crcount = idev->mc_qrv;
		idev->mc_ifc_count = pmc->mca_crcount;
		for (psf=pmc->mca_sources; psf; psf = psf->sf_next)
			psf->sf_crcount = 0;
		mld_ifc_event(pmc->idev);
	} else if (sf_setstate(pmc) || changerec)
		mld_ifc_event(pmc->idev);
	spin_unlock_bh(&pmc->mca_lock);
	read_unlock_bh(&idev->lock);
	return err;
}

/*
 * Add multicast single-source filter to the interface list
 */
static int ip6_mc_add1_src(struct ifmcaddr6 *pmc, int sfmode,
2138
	const struct in6_addr *psfsrc)
L
Linus Torvalds 已提交
2139 2140 2141 2142 2143 2144 2145 2146 2147 2148
{
	struct ip6_sf_list *psf, *psf_prev;

	psf_prev = NULL;
	for (psf=pmc->mca_sources; psf; psf=psf->sf_next) {
		if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
			break;
		psf_prev = psf;
	}
	if (!psf) {
2149
		psf = kzalloc(sizeof(*psf), GFP_ATOMIC);
L
Linus Torvalds 已提交
2150 2151
		if (!psf)
			return -ENOBUFS;
2152

L
Linus Torvalds 已提交
2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178
		psf->sf_addr = *psfsrc;
		if (psf_prev) {
			psf_prev->sf_next = psf;
		} else
			pmc->mca_sources = psf;
	}
	psf->sf_count[sfmode]++;
	return 0;
}

static void sf_markstate(struct ifmcaddr6 *pmc)
{
	struct ip6_sf_list *psf;
	int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];

	for (psf=pmc->mca_sources; psf; psf=psf->sf_next)
		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
			psf->sf_oldin = mca_xcount ==
				psf->sf_count[MCAST_EXCLUDE] &&
				!psf->sf_count[MCAST_INCLUDE];
		} else
			psf->sf_oldin = psf->sf_count[MCAST_INCLUDE] != 0;
}

static int sf_setstate(struct ifmcaddr6 *pmc)
{
2179
	struct ip6_sf_list *psf, *dpsf;
L
Linus Torvalds 已提交
2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190
	int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
	int qrv = pmc->idev->mc_qrv;
	int new_in, rv;

	rv = 0;
	for (psf=pmc->mca_sources; psf; psf=psf->sf_next) {
		if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
			new_in = mca_xcount == psf->sf_count[MCAST_EXCLUDE] &&
				!psf->sf_count[MCAST_INCLUDE];
		} else
			new_in = psf->sf_count[MCAST_INCLUDE] != 0;
2191 2192
		if (new_in) {
			if (!psf->sf_oldin) {
2193
				struct ip6_sf_list *prev = NULL;
2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222

				for (dpsf=pmc->mca_tomb; dpsf;
				     dpsf=dpsf->sf_next) {
					if (ipv6_addr_equal(&dpsf->sf_addr,
					    &psf->sf_addr))
						break;
					prev = dpsf;
				}
				if (dpsf) {
					if (prev)
						prev->sf_next = dpsf->sf_next;
					else
						pmc->mca_tomb = dpsf->sf_next;
					kfree(dpsf);
				}
				psf->sf_crcount = qrv;
				rv++;
			}
		} else if (psf->sf_oldin) {
			psf->sf_crcount = 0;
			/*
			 * add or update "delete" records if an active filter
			 * is now inactive
			 */
			for (dpsf=pmc->mca_tomb; dpsf; dpsf=dpsf->sf_next)
				if (ipv6_addr_equal(&dpsf->sf_addr,
				    &psf->sf_addr))
					break;
			if (!dpsf) {
2223
				dpsf = kmalloc(sizeof(*dpsf), GFP_ATOMIC);
2224 2225 2226 2227 2228 2229 2230 2231
				if (!dpsf)
					continue;
				*dpsf = *psf;
				/* pmc->mca_lock held by callers */
				dpsf->sf_next = pmc->mca_tomb;
				pmc->mca_tomb = dpsf;
			}
			dpsf->sf_crcount = qrv;
L
Linus Torvalds 已提交
2232 2233 2234 2235 2236 2237 2238 2239 2240
			rv++;
		}
	}
	return rv;
}

/*
 * Add multicast source filter list to the interface list
 */
2241 2242
static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
			  int sfmode, int sfcount, const struct in6_addr *psfsrc,
L
Linus Torvalds 已提交
2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268
			  int delta)
{
	struct ifmcaddr6 *pmc;
	int	isexclude;
	int	i, err;

	if (!idev)
		return -ENODEV;
	read_lock_bh(&idev->lock);
	for (pmc=idev->mc_list; pmc; pmc=pmc->next) {
		if (ipv6_addr_equal(pmca, &pmc->mca_addr))
			break;
	}
	if (!pmc) {
		/* MCA not found?? bug */
		read_unlock_bh(&idev->lock);
		return -ESRCH;
	}
	spin_lock_bh(&pmc->mca_lock);

	sf_markstate(pmc);
	isexclude = pmc->mca_sfmode == MCAST_EXCLUDE;
	if (!delta)
		pmc->mca_sfcount[sfmode]++;
	err = 0;
	for (i=0; i<sfcount; i++) {
2269
		err = ip6_mc_add1_src(pmc, sfmode, &psfsrc[i]);
L
Linus Torvalds 已提交
2270 2271 2272 2273 2274 2275 2276 2277 2278
		if (err)
			break;
	}
	if (err) {
		int j;

		if (!delta)
			pmc->mca_sfcount[sfmode]--;
		for (j=0; j<i; j++)
2279
			ip6_mc_del1_src(pmc, sfmode, &psfsrc[j]);
L
Linus Torvalds 已提交
2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316
	} else if (isexclude != (pmc->mca_sfcount[MCAST_EXCLUDE] != 0)) {
		struct ip6_sf_list *psf;

		/* filter mode change */
		if (pmc->mca_sfcount[MCAST_EXCLUDE])
			pmc->mca_sfmode = MCAST_EXCLUDE;
		else if (pmc->mca_sfcount[MCAST_INCLUDE])
			pmc->mca_sfmode = MCAST_INCLUDE;
		/* else no filters; keep old mode for reports */

		pmc->mca_crcount = idev->mc_qrv;
		idev->mc_ifc_count = pmc->mca_crcount;
		for (psf=pmc->mca_sources; psf; psf = psf->sf_next)
			psf->sf_crcount = 0;
		mld_ifc_event(idev);
	} else if (sf_setstate(pmc))
		mld_ifc_event(idev);
	spin_unlock_bh(&pmc->mca_lock);
	read_unlock_bh(&idev->lock);
	return err;
}

static void ip6_mc_clear_src(struct ifmcaddr6 *pmc)
{
	struct ip6_sf_list *psf, *nextpsf;

	for (psf=pmc->mca_tomb; psf; psf=nextpsf) {
		nextpsf = psf->sf_next;
		kfree(psf);
	}
	pmc->mca_tomb = NULL;
	for (psf=pmc->mca_sources; psf; psf=nextpsf) {
		nextpsf = psf->sf_next;
		kfree(psf);
	}
	pmc->mca_sources = NULL;
	pmc->mca_sfmode = MCAST_EXCLUDE;
2317
	pmc->mca_sfcount[MCAST_INCLUDE] = 0;
L
Linus Torvalds 已提交
2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330
	pmc->mca_sfcount[MCAST_EXCLUDE] = 1;
}


static void igmp6_join_group(struct ifmcaddr6 *ma)
{
	unsigned long delay;

	if (ma->mca_flags & MAF_NOREPORT)
		return;

	igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);

2331
	delay = net_random() % unsolicited_report_interval(ma->idev);
L
Linus Torvalds 已提交
2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349

	spin_lock_bh(&ma->mca_lock);
	if (del_timer(&ma->mca_timer)) {
		atomic_dec(&ma->mca_refcnt);
		delay = ma->mca_timer.expires - jiffies;
	}

	if (!mod_timer(&ma->mca_timer, jiffies + delay))
		atomic_inc(&ma->mca_refcnt);
	ma->mca_flags |= MAF_TIMER_RUNNING | MAF_LAST_REPORTER;
	spin_unlock_bh(&ma->mca_lock);
}

static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
			    struct inet6_dev *idev)
{
	int err;

2350 2351 2352
	/* callers have the socket lock and a write lock on ipv6_sk_mc_lock,
	 * so no other readers or writers of iml or its sflist
	 */
S
Stephen Hemminger 已提交
2353
	if (!iml->sflist) {
L
Linus Torvalds 已提交
2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365
		/* any-source empty exclude case */
		return ip6_mc_del_src(idev, &iml->addr, iml->sfmode, 0, NULL, 0);
	}
	err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode,
		iml->sflist->sl_count, iml->sflist->sl_addr, 0);
	sock_kfree_s(sk, iml->sflist, IP6_SFLSIZE(iml->sflist->sl_max));
	iml->sflist = NULL;
	return err;
}

static void igmp6_leave_group(struct ifmcaddr6 *ma)
{
2366
	if (mld_in_v1_mode(ma->idev)) {
L
Linus Torvalds 已提交
2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381
		if (ma->mca_flags & MAF_LAST_REPORTER)
			igmp6_send(&ma->mca_addr, ma->idev->dev,
				ICMPV6_MGM_REDUCTION);
	} else {
		mld_add_delrec(ma->idev, ma);
		mld_ifc_event(ma->idev);
	}
}

static void mld_gq_timer_expire(unsigned long data)
{
	struct inet6_dev *idev = (struct inet6_dev *)data;

	idev->mc_gq_running = 0;
	mld_send_report(idev, NULL);
2382
	in6_dev_put(idev);
L
Linus Torvalds 已提交
2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394
}

static void mld_ifc_timer_expire(unsigned long data)
{
	struct inet6_dev *idev = (struct inet6_dev *)data;

	mld_send_cr(idev);
	if (idev->mc_ifc_count) {
		idev->mc_ifc_count--;
		if (idev->mc_ifc_count)
			mld_ifc_start_timer(idev, idev->mc_maxdelay);
	}
2395
	in6_dev_put(idev);
L
Linus Torvalds 已提交
2396 2397 2398 2399
}

static void mld_ifc_event(struct inet6_dev *idev)
{
2400
	if (mld_in_v1_mode(idev))
L
Linus Torvalds 已提交
2401 2402 2403 2404 2405 2406 2407 2408 2409 2410
		return;
	idev->mc_ifc_count = idev->mc_qrv;
	mld_ifc_start_timer(idev, 1);
}


static void igmp6_timer_handler(unsigned long data)
{
	struct ifmcaddr6 *ma = (struct ifmcaddr6 *) data;

2411
	if (mld_in_v1_mode(ma->idev))
L
Linus Torvalds 已提交
2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422
		igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
	else
		mld_send_report(ma->idev, ma);

	spin_lock(&ma->mca_lock);
	ma->mca_flags |=  MAF_LAST_REPORTER;
	ma->mca_flags &= ~MAF_TIMER_RUNNING;
	spin_unlock(&ma->mca_lock);
	ma_put(ma);
}

2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441
/* Device changing type */

void ipv6_mc_unmap(struct inet6_dev *idev)
{
	struct ifmcaddr6 *i;

	/* Install multicast list, except for all-nodes (already installed) */

	read_lock_bh(&idev->lock);
	for (i = idev->mc_list; i; i = i->next)
		igmp6_group_dropped(i);
	read_unlock_bh(&idev->lock);
}

void ipv6_mc_remap(struct inet6_dev *idev)
{
	ipv6_mc_up(idev);
}

L
Linus Torvalds 已提交
2442 2443 2444 2445 2446 2447 2448 2449 2450
/* Device going down */

void ipv6_mc_down(struct inet6_dev *idev)
{
	struct ifmcaddr6 *i;

	/* Withdraw multicast list */

	read_lock_bh(&idev->lock);
2451 2452 2453
	mld_ifc_stop_timer(idev);
	mld_gq_stop_timer(idev);
	mld_dad_stop_timer(idev);
L
Linus Torvalds 已提交
2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481

	for (i = idev->mc_list; i; i=i->next)
		igmp6_group_dropped(i);
	read_unlock_bh(&idev->lock);

	mld_clear_delrec(idev);
}


/* Device going up */

void ipv6_mc_up(struct inet6_dev *idev)
{
	struct ifmcaddr6 *i;

	/* Install multicast list, except for all-nodes (already installed) */

	read_lock_bh(&idev->lock);
	for (i = idev->mc_list; i; i=i->next)
		igmp6_group_added(i);
	read_unlock_bh(&idev->lock);
}

/* IPv6 device initialization. */

void ipv6_mc_init_dev(struct inet6_dev *idev)
{
	write_lock_bh(&idev->lock);
2482
	spin_lock_init(&idev->mc_lock);
L
Linus Torvalds 已提交
2483
	idev->mc_gq_running = 0;
2484 2485
	setup_timer(&idev->mc_gq_timer, mld_gq_timer_expire,
			(unsigned long)idev);
L
Linus Torvalds 已提交
2486 2487
	idev->mc_tomb = NULL;
	idev->mc_ifc_count = 0;
2488 2489
	setup_timer(&idev->mc_ifc_timer, mld_ifc_timer_expire,
			(unsigned long)idev);
2490 2491
	setup_timer(&idev->mc_dad_timer, mld_dad_timer_expire,
		    (unsigned long)idev);
2492

L
Linus Torvalds 已提交
2493
	idev->mc_qrv = MLD_QRV_DEFAULT;
2494 2495 2496
	idev->mc_qi = MLD_QI_DEFAULT;
	idev->mc_qri = MLD_QRI_DEFAULT;

2497
	idev->mc_maxdelay = unsolicited_report_interval(idev);
L
Linus Torvalds 已提交
2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517
	idev->mc_v1_seen = 0;
	write_unlock_bh(&idev->lock);
}

/*
 *	Device is about to be destroyed: clean up.
 */

void ipv6_mc_destroy_dev(struct inet6_dev *idev)
{
	struct ifmcaddr6 *i;

	/* Deactivate timers */
	ipv6_mc_down(idev);

	/* Delete all-nodes address. */
	/* We cannot call ipv6_dev_mc_dec() directly, our caller in
	 * addrconf.c has NULL'd out dev->ip6_ptr so in6_dev_get() will
	 * fail.
	 */
2518
	__ipv6_dev_mc_dec(idev, &in6addr_linklocal_allnodes);
L
Linus Torvalds 已提交
2519

2520 2521
	if (idev->cnf.forwarding)
		__ipv6_dev_mc_dec(idev, &in6addr_linklocal_allrouters);
L
Linus Torvalds 已提交
2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537

	write_lock_bh(&idev->lock);
	while ((i = idev->mc_list) != NULL) {
		idev->mc_list = i->next;
		write_unlock_bh(&idev->lock);

		igmp6_group_dropped(i);
		ma_put(i);

		write_lock_bh(&idev->lock);
	}
	write_unlock_bh(&idev->lock);
}

#ifdef CONFIG_PROC_FS
struct igmp6_mc_iter_state {
2538
	struct seq_net_private p;
L
Linus Torvalds 已提交
2539 2540 2541 2542 2543 2544 2545 2546 2547 2548
	struct net_device *dev;
	struct inet6_dev *idev;
};

#define igmp6_mc_seq_private(seq)	((struct igmp6_mc_iter_state *)(seq)->private)

static inline struct ifmcaddr6 *igmp6_mc_get_first(struct seq_file *seq)
{
	struct ifmcaddr6 *im = NULL;
	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2549
	struct net *net = seq_file_net(seq);
L
Linus Torvalds 已提交
2550

2551
	state->idev = NULL;
2552
	for_each_netdev_rcu(net, state->dev) {
L
Linus Torvalds 已提交
2553
		struct inet6_dev *idev;
2554
		idev = __in6_dev_get(state->dev);
L
Linus Torvalds 已提交
2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573
		if (!idev)
			continue;
		read_lock_bh(&idev->lock);
		im = idev->mc_list;
		if (im) {
			state->idev = idev;
			break;
		}
		read_unlock_bh(&idev->lock);
	}
	return im;
}

static struct ifmcaddr6 *igmp6_mc_get_next(struct seq_file *seq, struct ifmcaddr6 *im)
{
	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);

	im = im->next;
	while (!im) {
2574
		if (likely(state->idev != NULL))
L
Linus Torvalds 已提交
2575
			read_unlock_bh(&state->idev->lock);
2576 2577

		state->dev = next_net_device_rcu(state->dev);
L
Linus Torvalds 已提交
2578 2579 2580 2581
		if (!state->dev) {
			state->idev = NULL;
			break;
		}
2582
		state->idev = __in6_dev_get(state->dev);
L
Linus Torvalds 已提交
2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600
		if (!state->idev)
			continue;
		read_lock_bh(&state->idev->lock);
		im = state->idev->mc_list;
	}
	return im;
}

static struct ifmcaddr6 *igmp6_mc_get_idx(struct seq_file *seq, loff_t pos)
{
	struct ifmcaddr6 *im = igmp6_mc_get_first(seq);
	if (im)
		while (pos && (im = igmp6_mc_get_next(seq, im)) != NULL)
			--pos;
	return pos ? NULL : im;
}

static void *igmp6_mc_seq_start(struct seq_file *seq, loff_t *pos)
2601
	__acquires(RCU)
L
Linus Torvalds 已提交
2602
{
2603
	rcu_read_lock();
L
Linus Torvalds 已提交
2604 2605 2606 2607 2608
	return igmp6_mc_get_idx(seq, *pos);
}

static void *igmp6_mc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
2609 2610
	struct ifmcaddr6 *im = igmp6_mc_get_next(seq, v);

L
Linus Torvalds 已提交
2611 2612 2613 2614 2615
	++*pos;
	return im;
}

static void igmp6_mc_seq_stop(struct seq_file *seq, void *v)
2616
	__releases(RCU)
L
Linus Torvalds 已提交
2617 2618
{
	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2619

L
Linus Torvalds 已提交
2620 2621 2622 2623 2624
	if (likely(state->idev != NULL)) {
		read_unlock_bh(&state->idev->lock);
		state->idev = NULL;
	}
	state->dev = NULL;
2625
	rcu_read_unlock();
L
Linus Torvalds 已提交
2626 2627 2628 2629 2630 2631 2632 2633
}

static int igmp6_mc_seq_show(struct seq_file *seq, void *v)
{
	struct ifmcaddr6 *im = (struct ifmcaddr6 *)v;
	struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);

	seq_printf(seq,
2634
		   "%-4d %-15s %pi6 %5d %08X %ld\n",
L
Linus Torvalds 已提交
2635
		   state->dev->ifindex, state->dev->name,
2636
		   &im->mca_addr,
L
Linus Torvalds 已提交
2637 2638 2639 2640 2641 2642
		   im->mca_users, im->mca_flags,
		   (im->mca_flags&MAF_TIMER_RUNNING) ?
		   jiffies_to_clock_t(im->mca_timer.expires-jiffies) : 0);
	return 0;
}

2643
static const struct seq_operations igmp6_mc_seq_ops = {
L
Linus Torvalds 已提交
2644 2645 2646 2647 2648 2649 2650 2651
	.start	=	igmp6_mc_seq_start,
	.next	=	igmp6_mc_seq_next,
	.stop	=	igmp6_mc_seq_stop,
	.show	=	igmp6_mc_seq_show,
};

static int igmp6_mc_seq_open(struct inode *inode, struct file *file)
{
2652 2653
	return seq_open_net(inode, file, &igmp6_mc_seq_ops,
			    sizeof(struct igmp6_mc_iter_state));
L
Linus Torvalds 已提交
2654 2655
}

2656
static const struct file_operations igmp6_mc_seq_fops = {
L
Linus Torvalds 已提交
2657 2658 2659 2660
	.owner		=	THIS_MODULE,
	.open		=	igmp6_mc_seq_open,
	.read		=	seq_read,
	.llseek		=	seq_lseek,
2661
	.release	=	seq_release_net,
L
Linus Torvalds 已提交
2662 2663 2664
};

struct igmp6_mcf_iter_state {
2665
	struct seq_net_private p;
L
Linus Torvalds 已提交
2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677
	struct net_device *dev;
	struct inet6_dev *idev;
	struct ifmcaddr6 *im;
};

#define igmp6_mcf_seq_private(seq)	((struct igmp6_mcf_iter_state *)(seq)->private)

static inline struct ip6_sf_list *igmp6_mcf_get_first(struct seq_file *seq)
{
	struct ip6_sf_list *psf = NULL;
	struct ifmcaddr6 *im = NULL;
	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
2678
	struct net *net = seq_file_net(seq);
L
Linus Torvalds 已提交
2679

2680 2681
	state->idev = NULL;
	state->im = NULL;
2682
	for_each_netdev_rcu(net, state->dev) {
L
Linus Torvalds 已提交
2683
		struct inet6_dev *idev;
2684
		idev = __in6_dev_get(state->dev);
L
Linus Torvalds 已提交
2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712
		if (unlikely(idev == NULL))
			continue;
		read_lock_bh(&idev->lock);
		im = idev->mc_list;
		if (likely(im != NULL)) {
			spin_lock_bh(&im->mca_lock);
			psf = im->mca_sources;
			if (likely(psf != NULL)) {
				state->im = im;
				state->idev = idev;
				break;
			}
			spin_unlock_bh(&im->mca_lock);
		}
		read_unlock_bh(&idev->lock);
	}
	return psf;
}

static struct ip6_sf_list *igmp6_mcf_get_next(struct seq_file *seq, struct ip6_sf_list *psf)
{
	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);

	psf = psf->sf_next;
	while (!psf) {
		spin_unlock_bh(&state->im->mca_lock);
		state->im = state->im->next;
		while (!state->im) {
2713
			if (likely(state->idev != NULL))
L
Linus Torvalds 已提交
2714
				read_unlock_bh(&state->idev->lock);
2715 2716

			state->dev = next_net_device_rcu(state->dev);
L
Linus Torvalds 已提交
2717 2718 2719 2720
			if (!state->dev) {
				state->idev = NULL;
				goto out;
			}
2721
			state->idev = __in6_dev_get(state->dev);
L
Linus Torvalds 已提交
2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745
			if (!state->idev)
				continue;
			read_lock_bh(&state->idev->lock);
			state->im = state->idev->mc_list;
		}
		if (!state->im)
			break;
		spin_lock_bh(&state->im->mca_lock);
		psf = state->im->mca_sources;
	}
out:
	return psf;
}

static struct ip6_sf_list *igmp6_mcf_get_idx(struct seq_file *seq, loff_t pos)
{
	struct ip6_sf_list *psf = igmp6_mcf_get_first(seq);
	if (psf)
		while (pos && (psf = igmp6_mcf_get_next(seq, psf)) != NULL)
			--pos;
	return pos ? NULL : psf;
}

static void *igmp6_mcf_seq_start(struct seq_file *seq, loff_t *pos)
2746
	__acquires(RCU)
L
Linus Torvalds 已提交
2747
{
2748
	rcu_read_lock();
L
Linus Torvalds 已提交
2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763
	return *pos ? igmp6_mcf_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
}

static void *igmp6_mcf_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
	struct ip6_sf_list *psf;
	if (v == SEQ_START_TOKEN)
		psf = igmp6_mcf_get_first(seq);
	else
		psf = igmp6_mcf_get_next(seq, v);
	++*pos;
	return psf;
}

static void igmp6_mcf_seq_stop(struct seq_file *seq, void *v)
2764
	__releases(RCU)
L
Linus Torvalds 已提交
2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775
{
	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
	if (likely(state->im != NULL)) {
		spin_unlock_bh(&state->im->mca_lock);
		state->im = NULL;
	}
	if (likely(state->idev != NULL)) {
		read_unlock_bh(&state->idev->lock);
		state->idev = NULL;
	}
	state->dev = NULL;
2776
	rcu_read_unlock();
L
Linus Torvalds 已提交
2777 2778 2779 2780 2781 2782 2783 2784
}

static int igmp6_mcf_seq_show(struct seq_file *seq, void *v)
{
	struct ip6_sf_list *psf = (struct ip6_sf_list *)v;
	struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);

	if (v == SEQ_START_TOKEN) {
2785
		seq_printf(seq,
L
Linus Torvalds 已提交
2786
			   "%3s %6s "
2787
			   "%32s %32s %6s %6s\n", "Idx",
L
Linus Torvalds 已提交
2788 2789 2790 2791
			   "Device", "Multicast Address",
			   "Source Address", "INC", "EXC");
	} else {
		seq_printf(seq,
2792
			   "%3d %6.6s %pi6 %pi6 %6lu %6lu\n",
L
Linus Torvalds 已提交
2793
			   state->dev->ifindex, state->dev->name,
2794 2795
			   &state->im->mca_addr,
			   &psf->sf_addr,
L
Linus Torvalds 已提交
2796 2797 2798 2799 2800 2801
			   psf->sf_count[MCAST_INCLUDE],
			   psf->sf_count[MCAST_EXCLUDE]);
	}
	return 0;
}

2802
static const struct seq_operations igmp6_mcf_seq_ops = {
L
Linus Torvalds 已提交
2803 2804 2805 2806 2807 2808 2809 2810
	.start	=	igmp6_mcf_seq_start,
	.next	=	igmp6_mcf_seq_next,
	.stop	=	igmp6_mcf_seq_stop,
	.show	=	igmp6_mcf_seq_show,
};

static int igmp6_mcf_seq_open(struct inode *inode, struct file *file)
{
2811 2812
	return seq_open_net(inode, file, &igmp6_mcf_seq_ops,
			    sizeof(struct igmp6_mcf_iter_state));
L
Linus Torvalds 已提交
2813 2814
}

2815
static const struct file_operations igmp6_mcf_seq_fops = {
L
Linus Torvalds 已提交
2816 2817 2818 2819
	.owner		=	THIS_MODULE,
	.open		=	igmp6_mcf_seq_open,
	.read		=	seq_read,
	.llseek		=	seq_lseek,
2820
	.release	=	seq_release_net,
L
Linus Torvalds 已提交
2821
};
2822

2823
static int __net_init igmp6_proc_init(struct net *net)
2824 2825 2826 2827
{
	int err;

	err = -ENOMEM;
2828
	if (!proc_create("igmp6", S_IRUGO, net->proc_net, &igmp6_mc_seq_fops))
2829
		goto out;
2830 2831
	if (!proc_create("mcfilter6", S_IRUGO, net->proc_net,
			 &igmp6_mcf_seq_fops))
2832 2833 2834 2835 2836 2837 2838
		goto out_proc_net_igmp6;

	err = 0;
out:
	return err;

out_proc_net_igmp6:
2839
	remove_proc_entry("igmp6", net->proc_net);
2840 2841 2842
	goto out;
}

2843
static void __net_exit igmp6_proc_exit(struct net *net)
2844
{
2845 2846
	remove_proc_entry("mcfilter6", net->proc_net);
	remove_proc_entry("igmp6", net->proc_net);
2847 2848
}
#else
2849
static inline int igmp6_proc_init(struct net *net)
2850 2851 2852
{
	return 0;
}
2853
static inline void igmp6_proc_exit(struct net *net)
2854 2855
{
}
L
Linus Torvalds 已提交
2856 2857
#endif

2858
static int __net_init igmp6_net_init(struct net *net)
L
Linus Torvalds 已提交
2859 2860 2861
{
	int err;

2862 2863
	err = inet_ctl_sock_create(&net->ipv6.igmp_sk, PF_INET6,
				   SOCK_RAW, IPPROTO_ICMPV6, net);
L
Linus Torvalds 已提交
2864
	if (err < 0) {
2865
		pr_err("Failed to initialize the IGMP6 control socket (err %d)\n",
L
Linus Torvalds 已提交
2866
		       err);
2867
		goto out;
L
Linus Torvalds 已提交
2868 2869
	}

2870
	inet6_sk(net->ipv6.igmp_sk)->hop_limit = 1;
L
Linus Torvalds 已提交
2871

2872 2873
	err = igmp6_proc_init(net);
	if (err)
2874 2875 2876 2877 2878
		goto out_sock_create;
out:
	return err;

out_sock_create:
2879
	inet_ctl_sock_destroy(net->ipv6.igmp_sk);
2880
	goto out;
L
Linus Torvalds 已提交
2881 2882
}

2883
static void __net_exit igmp6_net_exit(struct net *net)
L
Linus Torvalds 已提交
2884
{
2885
	inet_ctl_sock_destroy(net->ipv6.igmp_sk);
2886
	igmp6_proc_exit(net);
L
Linus Torvalds 已提交
2887
}
2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902

static struct pernet_operations igmp6_net_ops = {
	.init = igmp6_net_init,
	.exit = igmp6_net_exit,
};

int __init igmp6_init(void)
{
	return register_pernet_subsys(&igmp6_net_ops);
}

void igmp6_cleanup(void)
{
	unregister_pernet_subsys(&igmp6_net_ops);
}