ipmr.c 53.1 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3
/*
 *	IP multicast routing support for mrouted 3.6/3.8
 *
4
 *		(c) 1995 Alan Cox, <alan@lxorguk.ukuu.org.uk>
L
Linus Torvalds 已提交
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
 *	  Linux Consultancy and Custom Driver Development
 *
 *	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.
 *
 *	Fixes:
 *	Michael Chastain	:	Incorrect size of copying.
 *	Alan Cox		:	Added the cache manager code
 *	Alan Cox		:	Fixed the clone/copy bug and device race.
 *	Mike McLagan		:	Routing by source
 *	Malcolm Beattie		:	Buffer handling fixes.
 *	Alexey Kuznetsov	:	Double buffer free and other fixes.
 *	SVR Anand		:	Fixed several multicast bugs and problems.
 *	Alexey Kuznetsov	:	Status, optimisations and more.
 *	Brad Parker		:	Better behaviour on mrouted upcall
 *					overflow.
 *      Carlos Picoto           :       PIMv1 Support
 *	Pavlin Ivanov Radoslavov:	PIMv2 Registers must checksum only PIM header
 *					Relax this requrement to work with older peers.
 *
 */

#include <asm/system.h>
#include <asm/uaccess.h>
#include <linux/types.h>
32
#include <linux/capability.h>
L
Linus Torvalds 已提交
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
#include <linux/errno.h>
#include <linux/timer.h>
#include <linux/mm.h>
#include <linux/kernel.h>
#include <linux/fcntl.h>
#include <linux/stat.h>
#include <linux/socket.h>
#include <linux/in.h>
#include <linux/inet.h>
#include <linux/netdevice.h>
#include <linux/inetdevice.h>
#include <linux/igmp.h>
#include <linux/proc_fs.h>
#include <linux/seq_file.h>
#include <linux/mroute.h>
#include <linux/init.h>
49
#include <linux/if_ether.h>
50
#include <linux/slab.h>
51
#include <net/net_namespace.h>
L
Linus Torvalds 已提交
52 53 54
#include <net/ip.h>
#include <net/protocol.h>
#include <linux/skbuff.h>
55
#include <net/route.h>
L
Linus Torvalds 已提交
56 57 58 59 60 61 62 63 64
#include <net/sock.h>
#include <net/icmp.h>
#include <net/udp.h>
#include <net/raw.h>
#include <linux/notifier.h>
#include <linux/if_arp.h>
#include <linux/netfilter_ipv4.h>
#include <net/ipip.h>
#include <net/checksum.h>
65
#include <net/netlink.h>
66
#include <net/fib_rules.h>
L
Linus Torvalds 已提交
67 68 69 70 71

#if defined(CONFIG_IP_PIMSM_V1) || defined(CONFIG_IP_PIMSM_V2)
#define CONFIG_IP_PIMSM	1
#endif

72
struct mr_table {
73
	struct list_head	list;
74 75 76
#ifdef CONFIG_NET_NS
	struct net		*net;
#endif
77
	u32			id;
78 79 80 81 82 83 84 85 86 87 88 89 90 91
	struct sock		*mroute_sk;
	struct timer_list	ipmr_expire_timer;
	struct list_head	mfc_unres_queue;
	struct list_head	mfc_cache_array[MFC_LINES];
	struct vif_device	vif_table[MAXVIFS];
	int			maxvif;
	atomic_t		cache_resolve_queue_len;
	int			mroute_do_assert;
	int			mroute_do_pim;
#if defined(CONFIG_IP_PIMSM_V1) || defined(CONFIG_IP_PIMSM_V2)
	int			mroute_reg_vif_num;
#endif
};

92 93 94 95 96 97 98 99
struct ipmr_rule {
	struct fib_rule		common;
};

struct ipmr_result {
	struct mr_table		*mrt;
};

L
Linus Torvalds 已提交
100 101 102 103 104 105 106 107 108 109
/* Big lock, protecting vif table, mrt cache and mroute socket state.
   Note that the changes are semaphored via rtnl_lock.
 */

static DEFINE_RWLOCK(mrt_lock);

/*
 *	Multicast router control variables
 */

110
#define VIF_EXISTS(_mrt, _idx) ((_mrt)->vif_table[_idx].dev != NULL)
L
Linus Torvalds 已提交
111 112 113 114 115 116 117 118 119 120 121 122

/* Special spinlock for queue of unresolved entries */
static DEFINE_SPINLOCK(mfc_unres_lock);

/* We return to original Alan's scheme. Hash table of resolved
   entries is changed only in process context and protected
   with weak lock mrt_lock. Queue of unresolved entries is protected
   with strong spinlock mfc_unres_lock.

   In this case data path is free of exclusive locks at all.
 */

123
static struct kmem_cache *mrt_cachep __read_mostly;
L
Linus Torvalds 已提交
124

125
static struct mr_table *ipmr_new_table(struct net *net, u32 id);
126 127 128 129
static int ip_mr_forward(struct net *net, struct mr_table *mrt,
			 struct sk_buff *skb, struct mfc_cache *cache,
			 int local);
static int ipmr_cache_report(struct mr_table *mrt,
130
			     struct sk_buff *pkt, vifi_t vifi, int assert);
131
static int ipmr_fill_mroute(struct mr_table *mrt, struct sk_buff *skb,
132
			    struct mfc_cache *c, struct rtmsg *rtm);
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313
static void ipmr_expire_process(unsigned long arg);

#ifdef CONFIG_IP_MROUTE_MULTIPLE_TABLES
#define ipmr_for_each_table(mrt, net) \
	list_for_each_entry_rcu(mrt, &net->ipv4.mr_tables, list)

static struct mr_table *ipmr_get_table(struct net *net, u32 id)
{
	struct mr_table *mrt;

	ipmr_for_each_table(mrt, net) {
		if (mrt->id == id)
			return mrt;
	}
	return NULL;
}

static int ipmr_fib_lookup(struct net *net, struct flowi *flp,
			   struct mr_table **mrt)
{
	struct ipmr_result res;
	struct fib_lookup_arg arg = { .result = &res, };
	int err;

	err = fib_rules_lookup(net->ipv4.mr_rules_ops, flp, 0, &arg);
	if (err < 0)
		return err;
	*mrt = res.mrt;
	return 0;
}

static int ipmr_rule_action(struct fib_rule *rule, struct flowi *flp,
			    int flags, struct fib_lookup_arg *arg)
{
	struct ipmr_result *res = arg->result;
	struct mr_table *mrt;

	switch (rule->action) {
	case FR_ACT_TO_TBL:
		break;
	case FR_ACT_UNREACHABLE:
		return -ENETUNREACH;
	case FR_ACT_PROHIBIT:
		return -EACCES;
	case FR_ACT_BLACKHOLE:
	default:
		return -EINVAL;
	}

	mrt = ipmr_get_table(rule->fr_net, rule->table);
	if (mrt == NULL)
		return -EAGAIN;
	res->mrt = mrt;
	return 0;
}

static int ipmr_rule_match(struct fib_rule *rule, struct flowi *fl, int flags)
{
	return 1;
}

static const struct nla_policy ipmr_rule_policy[FRA_MAX + 1] = {
	FRA_GENERIC_POLICY,
};

static int ipmr_rule_configure(struct fib_rule *rule, struct sk_buff *skb,
			       struct fib_rule_hdr *frh, struct nlattr **tb)
{
	return 0;
}

static int ipmr_rule_compare(struct fib_rule *rule, struct fib_rule_hdr *frh,
			     struct nlattr **tb)
{
	return 1;
}

static int ipmr_rule_fill(struct fib_rule *rule, struct sk_buff *skb,
			  struct fib_rule_hdr *frh)
{
	frh->dst_len = 0;
	frh->src_len = 0;
	frh->tos     = 0;
	return 0;
}

static struct fib_rules_ops ipmr_rules_ops_template = {
	.family		= FIB_RULES_IPMR,
	.rule_size	= sizeof(struct ipmr_rule),
	.addr_size	= sizeof(u32),
	.action		= ipmr_rule_action,
	.match		= ipmr_rule_match,
	.configure	= ipmr_rule_configure,
	.compare	= ipmr_rule_compare,
	.default_pref	= fib_default_rule_pref,
	.fill		= ipmr_rule_fill,
	.nlgroup	= RTNLGRP_IPV4_RULE,
	.policy		= ipmr_rule_policy,
	.owner		= THIS_MODULE,
};

static int __net_init ipmr_rules_init(struct net *net)
{
	struct fib_rules_ops *ops;
	struct mr_table *mrt;
	int err;

	ops = fib_rules_register(&ipmr_rules_ops_template, net);
	if (IS_ERR(ops))
		return PTR_ERR(ops);

	INIT_LIST_HEAD(&net->ipv4.mr_tables);

	mrt = ipmr_new_table(net, RT_TABLE_DEFAULT);
	if (mrt == NULL) {
		err = -ENOMEM;
		goto err1;
	}

	err = fib_default_rule_add(ops, 0x7fff, RT_TABLE_DEFAULT, 0);
	if (err < 0)
		goto err2;

	net->ipv4.mr_rules_ops = ops;
	return 0;

err2:
	kfree(mrt);
err1:
	fib_rules_unregister(ops);
	return err;
}

static void __net_exit ipmr_rules_exit(struct net *net)
{
	struct mr_table *mrt, *next;

	list_for_each_entry_safe(mrt, next, &net->ipv4.mr_tables, list)
		kfree(mrt);
	fib_rules_unregister(net->ipv4.mr_rules_ops);
}
#else
#define ipmr_for_each_table(mrt, net) \
	for (mrt = net->ipv4.mrt; mrt; mrt = NULL)

static struct mr_table *ipmr_get_table(struct net *net, u32 id)
{
	return net->ipv4.mrt;
}

static int ipmr_fib_lookup(struct net *net, struct flowi *flp,
			   struct mr_table **mrt)
{
	*mrt = net->ipv4.mrt;
	return 0;
}

static int __net_init ipmr_rules_init(struct net *net)
{
	net->ipv4.mrt = ipmr_new_table(net, RT_TABLE_DEFAULT);
	return net->ipv4.mrt ? 0 : -ENOMEM;
}

static void __net_exit ipmr_rules_exit(struct net *net)
{
	kfree(net->ipv4.mrt);
}
#endif

static struct mr_table *ipmr_new_table(struct net *net, u32 id)
{
	struct mr_table *mrt;
	unsigned int i;

	mrt = ipmr_get_table(net, id);
	if (mrt != NULL)
		return mrt;

	mrt = kzalloc(sizeof(*mrt), GFP_KERNEL);
	if (mrt == NULL)
		return NULL;
314
	write_pnet(&mrt->net, net);
315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333
	mrt->id = id;

	/* Forwarding cache */
	for (i = 0; i < MFC_LINES; i++)
		INIT_LIST_HEAD(&mrt->mfc_cache_array[i]);

	INIT_LIST_HEAD(&mrt->mfc_unres_queue);

	setup_timer(&mrt->ipmr_expire_timer, ipmr_expire_process,
		    (unsigned long)mrt);

#ifdef CONFIG_IP_PIMSM
	mrt->mroute_reg_vif_num = -1;
#endif
#ifdef CONFIG_IP_MROUTE_MULTIPLE_TABLES
	list_add_tail_rcu(&mrt->list, &net->ipv4.mr_tables);
#endif
	return mrt;
}
L
Linus Torvalds 已提交
334 335 336

/* Service routines creating virtual interfaces: DVMRP tunnels and PIMREG */

337 338
static void ipmr_del_tunnel(struct net_device *dev, struct vifctl *v)
{
339 340
	struct net *net = dev_net(dev);

341 342
	dev_close(dev);

343
	dev = __dev_get_by_name(net, "tunl0");
344
	if (dev) {
345
		const struct net_device_ops *ops = dev->netdev_ops;
346 347 348 349 350 351 352 353 354 355 356 357
		struct ifreq ifr;
		struct ip_tunnel_parm p;

		memset(&p, 0, sizeof(p));
		p.iph.daddr = v->vifc_rmt_addr.s_addr;
		p.iph.saddr = v->vifc_lcl_addr.s_addr;
		p.iph.version = 4;
		p.iph.ihl = 5;
		p.iph.protocol = IPPROTO_IPIP;
		sprintf(p.name, "dvmrp%d", v->vifc_vifi);
		ifr.ifr_ifru.ifru_data = (__force void __user *)&p;

358 359 360 361 362 363 364
		if (ops->ndo_do_ioctl) {
			mm_segment_t oldfs = get_fs();

			set_fs(KERNEL_DS);
			ops->ndo_do_ioctl(dev, &ifr, SIOCDELTUNNEL);
			set_fs(oldfs);
		}
365 366 367
	}
}

L
Linus Torvalds 已提交
368
static
369
struct net_device *ipmr_new_tunnel(struct net *net, struct vifctl *v)
L
Linus Torvalds 已提交
370 371 372
{
	struct net_device  *dev;

373
	dev = __dev_get_by_name(net, "tunl0");
L
Linus Torvalds 已提交
374 375

	if (dev) {
376
		const struct net_device_ops *ops = dev->netdev_ops;
L
Linus Torvalds 已提交
377 378 379 380 381 382 383 384 385 386 387 388
		int err;
		struct ifreq ifr;
		struct ip_tunnel_parm p;
		struct in_device  *in_dev;

		memset(&p, 0, sizeof(p));
		p.iph.daddr = v->vifc_rmt_addr.s_addr;
		p.iph.saddr = v->vifc_lcl_addr.s_addr;
		p.iph.version = 4;
		p.iph.ihl = 5;
		p.iph.protocol = IPPROTO_IPIP;
		sprintf(p.name, "dvmrp%d", v->vifc_vifi);
S
Stephen Hemminger 已提交
389
		ifr.ifr_ifru.ifru_data = (__force void __user *)&p;
L
Linus Torvalds 已提交
390

391 392 393 394 395 396 397 398
		if (ops->ndo_do_ioctl) {
			mm_segment_t oldfs = get_fs();

			set_fs(KERNEL_DS);
			err = ops->ndo_do_ioctl(dev, &ifr, SIOCADDTUNNEL);
			set_fs(oldfs);
		} else
			err = -EOPNOTSUPP;
L
Linus Torvalds 已提交
399 400 401

		dev = NULL;

402 403
		if (err == 0 &&
		    (dev = __dev_get_by_name(net, p.name)) != NULL) {
L
Linus Torvalds 已提交
404 405
			dev->flags |= IFF_MULTICAST;

406
			in_dev = __in_dev_get_rtnl(dev);
407
			if (in_dev == NULL)
L
Linus Torvalds 已提交
408
				goto failure;
409 410 411

			ipv4_devconf_setall(in_dev);
			IPV4_DEVCONF(in_dev->cnf, RP_FILTER) = 0;
L
Linus Torvalds 已提交
412 413 414

			if (dev_open(dev))
				goto failure;
415
			dev_hold(dev);
L
Linus Torvalds 已提交
416 417 418 419 420 421 422 423 424 425 426 427 428 429 430
		}
	}
	return dev;

failure:
	/* allow the register to be completed before unregistering. */
	rtnl_unlock();
	rtnl_lock();

	unregister_netdevice(dev);
	return NULL;
}

#ifdef CONFIG_IP_PIMSM

431
static netdev_tx_t reg_vif_xmit(struct sk_buff *skb, struct net_device *dev)
L
Linus Torvalds 已提交
432
{
433
	struct net *net = dev_net(dev);
434 435 436 437 438 439 440 441 442 443 444
	struct mr_table *mrt;
	struct flowi fl = {
		.oif		= dev->ifindex,
		.iif		= skb->skb_iif,
		.mark		= skb->mark,
	};
	int err;

	err = ipmr_fib_lookup(net, &fl, &mrt);
	if (err < 0)
		return err;
445

L
Linus Torvalds 已提交
446
	read_lock(&mrt_lock);
447 448
	dev->stats.tx_bytes += skb->len;
	dev->stats.tx_packets++;
449
	ipmr_cache_report(mrt, skb, mrt->mroute_reg_vif_num, IGMPMSG_WHOLEPKT);
L
Linus Torvalds 已提交
450 451
	read_unlock(&mrt_lock);
	kfree_skb(skb);
452
	return NETDEV_TX_OK;
L
Linus Torvalds 已提交
453 454
}

455 456 457 458
static const struct net_device_ops reg_vif_netdev_ops = {
	.ndo_start_xmit	= reg_vif_xmit,
};

L
Linus Torvalds 已提交
459 460 461
static void reg_vif_setup(struct net_device *dev)
{
	dev->type		= ARPHRD_PIMREG;
462
	dev->mtu		= ETH_DATA_LEN - sizeof(struct iphdr) - 8;
L
Linus Torvalds 已提交
463
	dev->flags		= IFF_NOARP;
464
	dev->netdev_ops		= &reg_vif_netdev_ops,
L
Linus Torvalds 已提交
465
	dev->destructor		= free_netdev;
T
Tom Goff 已提交
466
	dev->features		|= NETIF_F_NETNS_LOCAL;
L
Linus Torvalds 已提交
467 468
}

469
static struct net_device *ipmr_reg_vif(struct net *net, struct mr_table *mrt)
L
Linus Torvalds 已提交
470 471 472
{
	struct net_device *dev;
	struct in_device *in_dev;
473 474 475 476 477 478
	char name[IFNAMSIZ];

	if (mrt->id == RT_TABLE_DEFAULT)
		sprintf(name, "pimreg");
	else
		sprintf(name, "pimreg%u", mrt->id);
L
Linus Torvalds 已提交
479

480
	dev = alloc_netdev(0, name, reg_vif_setup);
L
Linus Torvalds 已提交
481 482 483 484

	if (dev == NULL)
		return NULL;

T
Tom Goff 已提交
485 486
	dev_net_set(dev, net);

L
Linus Torvalds 已提交
487 488 489 490 491 492
	if (register_netdevice(dev)) {
		free_netdev(dev);
		return NULL;
	}
	dev->iflink = 0;

493 494 495
	rcu_read_lock();
	if ((in_dev = __in_dev_get_rcu(dev)) == NULL) {
		rcu_read_unlock();
L
Linus Torvalds 已提交
496
		goto failure;
497
	}
L
Linus Torvalds 已提交
498

499 500 501
	ipv4_devconf_setall(in_dev);
	IPV4_DEVCONF(in_dev->cnf, RP_FILTER) = 0;
	rcu_read_unlock();
L
Linus Torvalds 已提交
502 503 504 505

	if (dev_open(dev))
		goto failure;

506 507
	dev_hold(dev);

L
Linus Torvalds 已提交
508 509 510 511 512 513 514 515 516 517 518 519 520 521
	return dev;

failure:
	/* allow the register to be completed before unregistering. */
	rtnl_unlock();
	rtnl_lock();

	unregister_netdevice(dev);
	return NULL;
}
#endif

/*
 *	Delete a VIF entry
522
 *	@notify: Set to 1, if the caller is a notifier_call
L
Linus Torvalds 已提交
523
 */
524

525
static int vif_delete(struct mr_table *mrt, int vifi, int notify,
526
		      struct list_head *head)
L
Linus Torvalds 已提交
527 528 529 530 531
{
	struct vif_device *v;
	struct net_device *dev;
	struct in_device *in_dev;

532
	if (vifi < 0 || vifi >= mrt->maxvif)
L
Linus Torvalds 已提交
533 534
		return -EADDRNOTAVAIL;

535
	v = &mrt->vif_table[vifi];
L
Linus Torvalds 已提交
536 537 538 539 540 541 542 543 544 545 546

	write_lock_bh(&mrt_lock);
	dev = v->dev;
	v->dev = NULL;

	if (!dev) {
		write_unlock_bh(&mrt_lock);
		return -EADDRNOTAVAIL;
	}

#ifdef CONFIG_IP_PIMSM
547 548
	if (vifi == mrt->mroute_reg_vif_num)
		mrt->mroute_reg_vif_num = -1;
L
Linus Torvalds 已提交
549 550
#endif

551
	if (vifi+1 == mrt->maxvif) {
L
Linus Torvalds 已提交
552 553
		int tmp;
		for (tmp=vifi-1; tmp>=0; tmp--) {
554
			if (VIF_EXISTS(mrt, tmp))
L
Linus Torvalds 已提交
555 556
				break;
		}
557
		mrt->maxvif = tmp+1;
L
Linus Torvalds 已提交
558 559 560 561 562 563
	}

	write_unlock_bh(&mrt_lock);

	dev_set_allmulti(dev, -1);

564
	if ((in_dev = __in_dev_get_rtnl(dev)) != NULL) {
565
		IPV4_DEVCONF(in_dev->cnf, MC_FORWARDING)--;
L
Linus Torvalds 已提交
566 567 568
		ip_rt_multicast_event(in_dev);
	}

569
	if (v->flags&(VIFF_TUNNEL|VIFF_REGISTER) && !notify)
570
		unregister_netdevice_queue(dev, head);
L
Linus Torvalds 已提交
571 572 573 574 575

	dev_put(dev);
	return 0;
}

576 577 578 579 580
static inline void ipmr_cache_free(struct mfc_cache *c)
{
	kmem_cache_free(mrt_cachep, c);
}

L
Linus Torvalds 已提交
581 582 583 584
/* Destroy an unresolved cache entry, killing queued skbs
   and reporting error to netlink readers.
 */

585
static void ipmr_destroy_unres(struct mr_table *mrt, struct mfc_cache *c)
L
Linus Torvalds 已提交
586
{
587
	struct net *net = read_pnet(&mrt->net);
L
Linus Torvalds 已提交
588
	struct sk_buff *skb;
589
	struct nlmsgerr *e;
L
Linus Torvalds 已提交
590

591
	atomic_dec(&mrt->cache_resolve_queue_len);
L
Linus Torvalds 已提交
592

J
Jianjun Kong 已提交
593
	while ((skb = skb_dequeue(&c->mfc_un.unres.unresolved))) {
594
		if (ip_hdr(skb)->version == 0) {
L
Linus Torvalds 已提交
595 596 597 598
			struct nlmsghdr *nlh = (struct nlmsghdr *)skb_pull(skb, sizeof(struct iphdr));
			nlh->nlmsg_type = NLMSG_ERROR;
			nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct nlmsgerr));
			skb_trim(skb, nlh->nlmsg_len);
599 600 601
			e = NLMSG_DATA(nlh);
			e->error = -ETIMEDOUT;
			memset(&e->msg, 0, sizeof(e->msg));
602

603
			rtnl_unicast(skb, net, NETLINK_CB(skb).pid);
L
Linus Torvalds 已提交
604 605 606 607
		} else
			kfree_skb(skb);
	}

608
	ipmr_cache_free(c);
L
Linus Torvalds 已提交
609 610 611
}


612
/* Timer process for the unresolved queue. */
L
Linus Torvalds 已提交
613

614
static void ipmr_expire_process(unsigned long arg)
L
Linus Torvalds 已提交
615
{
616
	struct mr_table *mrt = (struct mr_table *)arg;
L
Linus Torvalds 已提交
617 618
	unsigned long now;
	unsigned long expires;
619
	struct mfc_cache *c, *next;
L
Linus Torvalds 已提交
620 621

	if (!spin_trylock(&mfc_unres_lock)) {
622
		mod_timer(&mrt->ipmr_expire_timer, jiffies+HZ/10);
L
Linus Torvalds 已提交
623 624 625
		return;
	}

626
	if (list_empty(&mrt->mfc_unres_queue))
L
Linus Torvalds 已提交
627 628 629 630 631
		goto out;

	now = jiffies;
	expires = 10*HZ;

632
	list_for_each_entry_safe(c, next, &mrt->mfc_unres_queue, list) {
L
Linus Torvalds 已提交
633 634 635 636 637 638 639
		if (time_after(c->mfc_un.unres.expires, now)) {
			unsigned long interval = c->mfc_un.unres.expires - now;
			if (interval < expires)
				expires = interval;
			continue;
		}

640
		list_del(&c->list);
641
		ipmr_destroy_unres(mrt, c);
L
Linus Torvalds 已提交
642 643
	}

644 645
	if (!list_empty(&mrt->mfc_unres_queue))
		mod_timer(&mrt->ipmr_expire_timer, jiffies + expires);
L
Linus Torvalds 已提交
646 647 648 649 650 651 652

out:
	spin_unlock(&mfc_unres_lock);
}

/* Fill oifs list. It is called under write locked mrt_lock. */

653
static void ipmr_update_thresholds(struct mr_table *mrt, struct mfc_cache *cache,
654
				   unsigned char *ttls)
L
Linus Torvalds 已提交
655 656 657 658 659 660 661
{
	int vifi;

	cache->mfc_un.res.minvif = MAXVIFS;
	cache->mfc_un.res.maxvif = 0;
	memset(cache->mfc_un.res.ttls, 255, MAXVIFS);

662 663
	for (vifi = 0; vifi < mrt->maxvif; vifi++) {
		if (VIF_EXISTS(mrt, vifi) &&
664
		    ttls[vifi] && ttls[vifi] < 255) {
L
Linus Torvalds 已提交
665 666 667 668 669 670 671 672 673
			cache->mfc_un.res.ttls[vifi] = ttls[vifi];
			if (cache->mfc_un.res.minvif > vifi)
				cache->mfc_un.res.minvif = vifi;
			if (cache->mfc_un.res.maxvif <= vifi)
				cache->mfc_un.res.maxvif = vifi + 1;
		}
	}
}

674 675
static int vif_add(struct net *net, struct mr_table *mrt,
		   struct vifctl *vifc, int mrtsock)
L
Linus Torvalds 已提交
676 677
{
	int vifi = vifc->vifc_vifi;
678
	struct vif_device *v = &mrt->vif_table[vifi];
L
Linus Torvalds 已提交
679 680
	struct net_device *dev;
	struct in_device *in_dev;
681
	int err;
L
Linus Torvalds 已提交
682 683

	/* Is vif busy ? */
684
	if (VIF_EXISTS(mrt, vifi))
L
Linus Torvalds 已提交
685 686 687 688 689 690 691 692 693
		return -EADDRINUSE;

	switch (vifc->vifc_flags) {
#ifdef CONFIG_IP_PIMSM
	case VIFF_REGISTER:
		/*
		 * Special Purpose VIF in PIM
		 * All the packets will be sent to the daemon
		 */
694
		if (mrt->mroute_reg_vif_num >= 0)
L
Linus Torvalds 已提交
695
			return -EADDRINUSE;
696
		dev = ipmr_reg_vif(net, mrt);
L
Linus Torvalds 已提交
697 698
		if (!dev)
			return -ENOBUFS;
699 700 701
		err = dev_set_allmulti(dev, 1);
		if (err) {
			unregister_netdevice(dev);
702
			dev_put(dev);
703 704
			return err;
		}
L
Linus Torvalds 已提交
705 706
		break;
#endif
707
	case VIFF_TUNNEL:
708
		dev = ipmr_new_tunnel(net, vifc);
L
Linus Torvalds 已提交
709 710
		if (!dev)
			return -ENOBUFS;
711 712 713
		err = dev_set_allmulti(dev, 1);
		if (err) {
			ipmr_del_tunnel(dev, vifc);
714
			dev_put(dev);
715 716
			return err;
		}
L
Linus Torvalds 已提交
717
		break;
718 719

	case VIFF_USE_IFINDEX:
L
Linus Torvalds 已提交
720
	case 0:
721 722 723 724 725 726 727 728 729
		if (vifc->vifc_flags == VIFF_USE_IFINDEX) {
			dev = dev_get_by_index(net, vifc->vifc_lcl_ifindex);
			if (dev && dev->ip_ptr == NULL) {
				dev_put(dev);
				return -EADDRNOTAVAIL;
			}
		} else
			dev = ip_dev_find(net, vifc->vifc_lcl_addr.s_addr);

L
Linus Torvalds 已提交
730 731
		if (!dev)
			return -EADDRNOTAVAIL;
732
		err = dev_set_allmulti(dev, 1);
733 734
		if (err) {
			dev_put(dev);
735
			return err;
736
		}
L
Linus Torvalds 已提交
737 738 739 740 741
		break;
	default:
		return -EINVAL;
	}

742 743
	if ((in_dev = __in_dev_get_rtnl(dev)) == NULL) {
		dev_put(dev);
L
Linus Torvalds 已提交
744
		return -EADDRNOTAVAIL;
745
	}
746
	IPV4_DEVCONF(in_dev->cnf, MC_FORWARDING)++;
L
Linus Torvalds 已提交
747 748 749 750 751
	ip_rt_multicast_event(in_dev);

	/*
	 *	Fill in the VIF structures
	 */
J
Jianjun Kong 已提交
752 753 754 755
	v->rate_limit = vifc->vifc_rate_limit;
	v->local = vifc->vifc_lcl_addr.s_addr;
	v->remote = vifc->vifc_rmt_addr.s_addr;
	v->flags = vifc->vifc_flags;
L
Linus Torvalds 已提交
756 757
	if (!mrtsock)
		v->flags |= VIFF_STATIC;
J
Jianjun Kong 已提交
758
	v->threshold = vifc->vifc_threshold;
L
Linus Torvalds 已提交
759 760 761 762 763 764 765 766 767 768
	v->bytes_in = 0;
	v->bytes_out = 0;
	v->pkt_in = 0;
	v->pkt_out = 0;
	v->link = dev->ifindex;
	if (v->flags&(VIFF_TUNNEL|VIFF_REGISTER))
		v->link = dev->iflink;

	/* And finish update writing critical data */
	write_lock_bh(&mrt_lock);
J
Jianjun Kong 已提交
769
	v->dev = dev;
L
Linus Torvalds 已提交
770 771
#ifdef CONFIG_IP_PIMSM
	if (v->flags&VIFF_REGISTER)
772
		mrt->mroute_reg_vif_num = vifi;
L
Linus Torvalds 已提交
773
#endif
774 775
	if (vifi+1 > mrt->maxvif)
		mrt->maxvif = vifi+1;
L
Linus Torvalds 已提交
776 777 778 779
	write_unlock_bh(&mrt_lock);
	return 0;
}

780
static struct mfc_cache *ipmr_cache_find(struct mr_table *mrt,
781 782
					 __be32 origin,
					 __be32 mcastgrp)
L
Linus Torvalds 已提交
783
{
J
Jianjun Kong 已提交
784
	int line = MFC_HASH(mcastgrp, origin);
L
Linus Torvalds 已提交
785 786
	struct mfc_cache *c;

787
	list_for_each_entry(c, &mrt->mfc_cache_array[line], list) {
788 789
		if (c->mfc_origin == origin && c->mfc_mcastgrp == mcastgrp)
			return c;
L
Linus Torvalds 已提交
790
	}
791
	return NULL;
L
Linus Torvalds 已提交
792 793 794 795 796
}

/*
 *	Allocate a multicast cache entry
 */
797
static struct mfc_cache *ipmr_cache_alloc(void)
L
Linus Torvalds 已提交
798
{
J
Jianjun Kong 已提交
799 800
	struct mfc_cache *c = kmem_cache_zalloc(mrt_cachep, GFP_KERNEL);
	if (c == NULL)
L
Linus Torvalds 已提交
801 802 803 804 805
		return NULL;
	c->mfc_un.res.minvif = MAXVIFS;
	return c;
}

806
static struct mfc_cache *ipmr_cache_alloc_unres(void)
L
Linus Torvalds 已提交
807
{
J
Jianjun Kong 已提交
808 809
	struct mfc_cache *c = kmem_cache_zalloc(mrt_cachep, GFP_ATOMIC);
	if (c == NULL)
L
Linus Torvalds 已提交
810 811 812 813 814 815 816 817 818
		return NULL;
	skb_queue_head_init(&c->mfc_un.unres.unresolved);
	c->mfc_un.unres.expires = jiffies + 10*HZ;
	return c;
}

/*
 *	A cache entry has gone into a resolved state from queued
 */
819

820 821
static void ipmr_cache_resolve(struct net *net, struct mr_table *mrt,
			       struct mfc_cache *uc, struct mfc_cache *c)
L
Linus Torvalds 已提交
822 823
{
	struct sk_buff *skb;
824
	struct nlmsgerr *e;
L
Linus Torvalds 已提交
825 826 827 828 829

	/*
	 *	Play the pending entries through our router
	 */

J
Jianjun Kong 已提交
830
	while ((skb = __skb_dequeue(&uc->mfc_un.unres.unresolved))) {
831
		if (ip_hdr(skb)->version == 0) {
L
Linus Torvalds 已提交
832 833
			struct nlmsghdr *nlh = (struct nlmsghdr *)skb_pull(skb, sizeof(struct iphdr));

834
			if (ipmr_fill_mroute(mrt, skb, c, NLMSG_DATA(nlh)) > 0) {
835 836
				nlh->nlmsg_len = (skb_tail_pointer(skb) -
						  (u8 *)nlh);
L
Linus Torvalds 已提交
837 838 839 840
			} else {
				nlh->nlmsg_type = NLMSG_ERROR;
				nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct nlmsgerr));
				skb_trim(skb, nlh->nlmsg_len);
841 842 843
				e = NLMSG_DATA(nlh);
				e->error = -EMSGSIZE;
				memset(&e->msg, 0, sizeof(e->msg));
L
Linus Torvalds 已提交
844
			}
845

846
			rtnl_unicast(skb, net, NETLINK_CB(skb).pid);
L
Linus Torvalds 已提交
847
		} else
848
			ip_mr_forward(net, mrt, skb, c, 0);
L
Linus Torvalds 已提交
849 850 851 852 853 854 855 856 857
	}
}

/*
 *	Bounce a cache query up to mrouted. We could use netlink for this but mrouted
 *	expects the following bizarre scheme.
 *
 *	Called under mrt_lock.
 */
858

859
static int ipmr_cache_report(struct mr_table *mrt,
860
			     struct sk_buff *pkt, vifi_t vifi, int assert)
L
Linus Torvalds 已提交
861 862
{
	struct sk_buff *skb;
863
	const int ihl = ip_hdrlen(pkt);
L
Linus Torvalds 已提交
864 865 866 867 868 869 870 871 872 873 874
	struct igmphdr *igmp;
	struct igmpmsg *msg;
	int ret;

#ifdef CONFIG_IP_PIMSM
	if (assert == IGMPMSG_WHOLEPKT)
		skb = skb_realloc_headroom(pkt, sizeof(struct iphdr));
	else
#endif
		skb = alloc_skb(128, GFP_ATOMIC);

S
Stephen Hemminger 已提交
875
	if (!skb)
L
Linus Torvalds 已提交
876 877 878 879 880 881 882 883 884
		return -ENOBUFS;

#ifdef CONFIG_IP_PIMSM
	if (assert == IGMPMSG_WHOLEPKT) {
		/* Ugly, but we have no choice with this interface.
		   Duplicate old header, fix ihl, length etc.
		   And all this only to mangle msg->im_msgtype and
		   to set msg->im_mbz to "mbz" :-)
		 */
885 886
		skb_push(skb, sizeof(struct iphdr));
		skb_reset_network_header(skb);
887
		skb_reset_transport_header(skb);
888
		msg = (struct igmpmsg *)skb_network_header(skb);
889
		memcpy(msg, skb_network_header(pkt), sizeof(struct iphdr));
L
Linus Torvalds 已提交
890 891
		msg->im_msgtype = IGMPMSG_WHOLEPKT;
		msg->im_mbz = 0;
892
		msg->im_vif = mrt->mroute_reg_vif_num;
893 894 895
		ip_hdr(skb)->ihl = sizeof(struct iphdr) >> 2;
		ip_hdr(skb)->tot_len = htons(ntohs(ip_hdr(pkt)->tot_len) +
					     sizeof(struct iphdr));
896
	} else
L
Linus Torvalds 已提交
897
#endif
898 899
	{

L
Linus Torvalds 已提交
900 901 902 903
	/*
	 *	Copy the IP header
	 */

904
	skb->network_header = skb->tail;
905
	skb_put(skb, ihl);
906
	skb_copy_to_linear_data(skb, pkt->data, ihl);
907 908
	ip_hdr(skb)->protocol = 0;			/* Flag to the kernel this is a route add */
	msg = (struct igmpmsg *)skb_network_header(skb);
L
Linus Torvalds 已提交
909
	msg->im_vif = vifi;
E
Eric Dumazet 已提交
910
	skb_dst_set(skb, dst_clone(skb_dst(pkt)));
L
Linus Torvalds 已提交
911 912 913 914 915

	/*
	 *	Add our header
	 */

J
Jianjun Kong 已提交
916
	igmp=(struct igmphdr *)skb_put(skb, sizeof(struct igmphdr));
L
Linus Torvalds 已提交
917 918 919
	igmp->type	=
	msg->im_msgtype = assert;
	igmp->code 	=	0;
920
	ip_hdr(skb)->tot_len = htons(skb->len);			/* Fix the length */
921
	skb->transport_header = skb->network_header;
922
	}
L
Linus Torvalds 已提交
923

924
	if (mrt->mroute_sk == NULL) {
L
Linus Torvalds 已提交
925 926 927 928 929 930 931
		kfree_skb(skb);
		return -EINVAL;
	}

	/*
	 *	Deliver to mrouted
	 */
932
	ret = sock_queue_rcv_skb(mrt->mroute_sk, skb);
933
	if (ret < 0) {
L
Linus Torvalds 已提交
934 935 936 937 938 939 940 941 942 943 944
		if (net_ratelimit())
			printk(KERN_WARNING "mroute: pending queue full, dropping entries.\n");
		kfree_skb(skb);
	}

	return ret;
}

/*
 *	Queue a packet for resolution. It gets locked cache entry!
 */
945

L
Linus Torvalds 已提交
946
static int
947
ipmr_cache_unresolved(struct mr_table *mrt, vifi_t vifi, struct sk_buff *skb)
L
Linus Torvalds 已提交
948
{
949
	bool found = false;
L
Linus Torvalds 已提交
950 951
	int err;
	struct mfc_cache *c;
952
	const struct iphdr *iph = ip_hdr(skb);
L
Linus Torvalds 已提交
953 954

	spin_lock_bh(&mfc_unres_lock);
955
	list_for_each_entry(c, &mrt->mfc_unres_queue, list) {
956
		if (c->mfc_mcastgrp == iph->daddr &&
957 958
		    c->mfc_origin == iph->saddr) {
			found = true;
L
Linus Torvalds 已提交
959
			break;
960
		}
L
Linus Torvalds 已提交
961 962
	}

963
	if (!found) {
L
Linus Torvalds 已提交
964 965 966 967
		/*
		 *	Create a new entry if allowable
		 */

968
		if (atomic_read(&mrt->cache_resolve_queue_len) >= 10 ||
969
		    (c = ipmr_cache_alloc_unres()) == NULL) {
L
Linus Torvalds 已提交
970 971 972 973 974 975 976 977 978
			spin_unlock_bh(&mfc_unres_lock);

			kfree_skb(skb);
			return -ENOBUFS;
		}

		/*
		 *	Fill in the new cache entry
		 */
979 980 981
		c->mfc_parent	= -1;
		c->mfc_origin	= iph->saddr;
		c->mfc_mcastgrp	= iph->daddr;
L
Linus Torvalds 已提交
982 983 984 985

		/*
		 *	Reflect first query at mrouted.
		 */
986
		err = ipmr_cache_report(mrt, skb, vifi, IGMPMSG_NOCACHE);
987
		if (err < 0) {
988
			/* If the report failed throw the cache entry
L
Linus Torvalds 已提交
989 990 991 992
			   out - Brad Parker
			 */
			spin_unlock_bh(&mfc_unres_lock);

993
			ipmr_cache_free(c);
L
Linus Torvalds 已提交
994 995 996 997
			kfree_skb(skb);
			return err;
		}

998 999
		atomic_inc(&mrt->cache_resolve_queue_len);
		list_add(&c->list, &mrt->mfc_unres_queue);
L
Linus Torvalds 已提交
1000

1001
		mod_timer(&mrt->ipmr_expire_timer, c->mfc_un.unres.expires);
L
Linus Torvalds 已提交
1002 1003 1004 1005 1006 1007 1008 1009 1010
	}

	/*
	 *	See if we can append the packet
	 */
	if (c->mfc_un.unres.unresolved.qlen>3) {
		kfree_skb(skb);
		err = -ENOBUFS;
	} else {
J
Jianjun Kong 已提交
1011
		skb_queue_tail(&c->mfc_un.unres.unresolved, skb);
L
Linus Torvalds 已提交
1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022
		err = 0;
	}

	spin_unlock_bh(&mfc_unres_lock);
	return err;
}

/*
 *	MFC cache manipulation by user space mroute daemon
 */

1023
static int ipmr_mfc_delete(struct mr_table *mrt, struct mfcctl *mfc)
L
Linus Torvalds 已提交
1024 1025
{
	int line;
1026
	struct mfc_cache *c, *next;
L
Linus Torvalds 已提交
1027

J
Jianjun Kong 已提交
1028
	line = MFC_HASH(mfc->mfcc_mcastgrp.s_addr, mfc->mfcc_origin.s_addr);
L
Linus Torvalds 已提交
1029

1030
	list_for_each_entry_safe(c, next, &mrt->mfc_cache_array[line], list) {
L
Linus Torvalds 已提交
1031 1032 1033
		if (c->mfc_origin == mfc->mfcc_origin.s_addr &&
		    c->mfc_mcastgrp == mfc->mfcc_mcastgrp.s_addr) {
			write_lock_bh(&mrt_lock);
1034
			list_del(&c->list);
L
Linus Torvalds 已提交
1035 1036
			write_unlock_bh(&mrt_lock);

1037
			ipmr_cache_free(c);
L
Linus Torvalds 已提交
1038 1039 1040 1041 1042 1043
			return 0;
		}
	}
	return -ENOENT;
}

1044 1045
static int ipmr_mfc_add(struct net *net, struct mr_table *mrt,
			struct mfcctl *mfc, int mrtsock)
L
Linus Torvalds 已提交
1046
{
1047
	bool found = false;
L
Linus Torvalds 已提交
1048
	int line;
1049
	struct mfc_cache *uc, *c;
L
Linus Torvalds 已提交
1050

1051 1052 1053
	if (mfc->mfcc_parent >= MAXVIFS)
		return -ENFILE;

J
Jianjun Kong 已提交
1054
	line = MFC_HASH(mfc->mfcc_mcastgrp.s_addr, mfc->mfcc_origin.s_addr);
L
Linus Torvalds 已提交
1055

1056
	list_for_each_entry(c, &mrt->mfc_cache_array[line], list) {
L
Linus Torvalds 已提交
1057
		if (c->mfc_origin == mfc->mfcc_origin.s_addr &&
1058 1059
		    c->mfc_mcastgrp == mfc->mfcc_mcastgrp.s_addr) {
			found = true;
L
Linus Torvalds 已提交
1060
			break;
1061
		}
L
Linus Torvalds 已提交
1062 1063
	}

1064
	if (found) {
L
Linus Torvalds 已提交
1065 1066
		write_lock_bh(&mrt_lock);
		c->mfc_parent = mfc->mfcc_parent;
1067
		ipmr_update_thresholds(mrt, c, mfc->mfcc_ttls);
L
Linus Torvalds 已提交
1068 1069 1070 1071 1072 1073
		if (!mrtsock)
			c->mfc_flags |= MFC_STATIC;
		write_unlock_bh(&mrt_lock);
		return 0;
	}

1074
	if (!ipv4_is_multicast(mfc->mfcc_mcastgrp.s_addr))
L
Linus Torvalds 已提交
1075 1076
		return -EINVAL;

1077
	c = ipmr_cache_alloc();
J
Jianjun Kong 已提交
1078
	if (c == NULL)
L
Linus Torvalds 已提交
1079 1080
		return -ENOMEM;

J
Jianjun Kong 已提交
1081 1082 1083
	c->mfc_origin = mfc->mfcc_origin.s_addr;
	c->mfc_mcastgrp = mfc->mfcc_mcastgrp.s_addr;
	c->mfc_parent = mfc->mfcc_parent;
1084
	ipmr_update_thresholds(mrt, c, mfc->mfcc_ttls);
L
Linus Torvalds 已提交
1085 1086 1087 1088
	if (!mrtsock)
		c->mfc_flags |= MFC_STATIC;

	write_lock_bh(&mrt_lock);
1089
	list_add(&c->list, &mrt->mfc_cache_array[line]);
L
Linus Torvalds 已提交
1090 1091 1092 1093 1094 1095
	write_unlock_bh(&mrt_lock);

	/*
	 *	Check to see if we resolved a queued list. If so we
	 *	need to send on the frames and tidy up.
	 */
1096
	found = false;
L
Linus Torvalds 已提交
1097
	spin_lock_bh(&mfc_unres_lock);
1098
	list_for_each_entry(uc, &mrt->mfc_unres_queue, list) {
1099
		if (uc->mfc_origin == c->mfc_origin &&
L
Linus Torvalds 已提交
1100
		    uc->mfc_mcastgrp == c->mfc_mcastgrp) {
1101
			list_del(&uc->list);
1102
			atomic_dec(&mrt->cache_resolve_queue_len);
1103
			found = true;
L
Linus Torvalds 已提交
1104 1105 1106
			break;
		}
	}
1107 1108
	if (list_empty(&mrt->mfc_unres_queue))
		del_timer(&mrt->ipmr_expire_timer);
L
Linus Torvalds 已提交
1109 1110
	spin_unlock_bh(&mfc_unres_lock);

1111
	if (found) {
1112
		ipmr_cache_resolve(net, mrt, uc, c);
1113
		ipmr_cache_free(uc);
L
Linus Torvalds 已提交
1114 1115 1116 1117 1118 1119 1120
	}
	return 0;
}

/*
 *	Close the multicast socket, and clear the vif tables etc
 */
1121

1122
static void mroute_clean_tables(struct mr_table *mrt)
L
Linus Torvalds 已提交
1123 1124
{
	int i;
1125
	LIST_HEAD(list);
1126
	struct mfc_cache *c, *next;
1127

L
Linus Torvalds 已提交
1128 1129 1130
	/*
	 *	Shut down all active vif entries
	 */
1131 1132 1133
	for (i = 0; i < mrt->maxvif; i++) {
		if (!(mrt->vif_table[i].flags&VIFF_STATIC))
			vif_delete(mrt, i, 0, &list);
L
Linus Torvalds 已提交
1134
	}
1135
	unregister_netdevice_many(&list);
L
Linus Torvalds 已提交
1136 1137 1138 1139

	/*
	 *	Wipe the cache
	 */
1140
	for (i = 0; i < MFC_LINES; i++) {
1141
		list_for_each_entry_safe(c, next, &mrt->mfc_cache_array[i], list) {
1142
			if (c->mfc_flags&MFC_STATIC)
L
Linus Torvalds 已提交
1143 1144
				continue;
			write_lock_bh(&mrt_lock);
1145
			list_del(&c->list);
L
Linus Torvalds 已提交
1146 1147
			write_unlock_bh(&mrt_lock);

1148
			ipmr_cache_free(c);
L
Linus Torvalds 已提交
1149 1150 1151
		}
	}

1152
	if (atomic_read(&mrt->cache_resolve_queue_len) != 0) {
L
Linus Torvalds 已提交
1153
		spin_lock_bh(&mfc_unres_lock);
1154
		list_for_each_entry_safe(c, next, &mrt->mfc_unres_queue, list) {
1155
			list_del(&c->list);
1156
			ipmr_destroy_unres(mrt, c);
L
Linus Torvalds 已提交
1157 1158 1159 1160 1161 1162 1163
		}
		spin_unlock_bh(&mfc_unres_lock);
	}
}

static void mrtsock_destruct(struct sock *sk)
{
1164
	struct net *net = sock_net(sk);
1165
	struct mr_table *mrt;
1166

L
Linus Torvalds 已提交
1167
	rtnl_lock();
1168 1169 1170
	ipmr_for_each_table(mrt, net) {
		if (sk == mrt->mroute_sk) {
			IPV4_DEVCONF_ALL(net, MC_FORWARDING)--;
L
Linus Torvalds 已提交
1171

1172 1173 1174
			write_lock_bh(&mrt_lock);
			mrt->mroute_sk = NULL;
			write_unlock_bh(&mrt_lock);
L
Linus Torvalds 已提交
1175

1176 1177
			mroute_clean_tables(mrt);
		}
L
Linus Torvalds 已提交
1178 1179 1180 1181 1182 1183 1184 1185 1186 1187
	}
	rtnl_unlock();
}

/*
 *	Socket options and virtual interface manipulation. The whole
 *	virtual interface system is a complete heap, but unfortunately
 *	that's how BSD mrouted happens to think. Maybe one day with a proper
 *	MOSPF/PIM router set up we can clean this up.
 */
1188

1189
int ip_mroute_setsockopt(struct sock *sk, int optname, char __user *optval, unsigned int optlen)
L
Linus Torvalds 已提交
1190 1191 1192 1193
{
	int ret;
	struct vifctl vif;
	struct mfcctl mfc;
1194
	struct net *net = sock_net(sk);
1195 1196 1197 1198 1199
	struct mr_table *mrt;

	mrt = ipmr_get_table(net, raw_sk(sk)->ipmr_table ? : RT_TABLE_DEFAULT);
	if (mrt == NULL)
		return -ENOENT;
1200

S
Stephen Hemminger 已提交
1201
	if (optname != MRT_INIT) {
1202
		if (sk != mrt->mroute_sk && !capable(CAP_NET_ADMIN))
L
Linus Torvalds 已提交
1203 1204 1205
			return -EACCES;
	}

S
Stephen Hemminger 已提交
1206 1207 1208
	switch (optname) {
	case MRT_INIT:
		if (sk->sk_type != SOCK_RAW ||
E
Eric Dumazet 已提交
1209
		    inet_sk(sk)->inet_num != IPPROTO_IGMP)
S
Stephen Hemminger 已提交
1210
			return -EOPNOTSUPP;
J
Jianjun Kong 已提交
1211
		if (optlen != sizeof(int))
S
Stephen Hemminger 已提交
1212
			return -ENOPROTOOPT;
L
Linus Torvalds 已提交
1213

S
Stephen Hemminger 已提交
1214
		rtnl_lock();
1215
		if (mrt->mroute_sk) {
L
Linus Torvalds 已提交
1216
			rtnl_unlock();
S
Stephen Hemminger 已提交
1217 1218 1219 1220 1221 1222
			return -EADDRINUSE;
		}

		ret = ip_ra_control(sk, 1, mrtsock_destruct);
		if (ret == 0) {
			write_lock_bh(&mrt_lock);
1223
			mrt->mroute_sk = sk;
S
Stephen Hemminger 已提交
1224 1225
			write_unlock_bh(&mrt_lock);

1226
			IPV4_DEVCONF_ALL(net, MC_FORWARDING)++;
S
Stephen Hemminger 已提交
1227 1228 1229 1230
		}
		rtnl_unlock();
		return ret;
	case MRT_DONE:
1231
		if (sk != mrt->mroute_sk)
S
Stephen Hemminger 已提交
1232 1233 1234 1235
			return -EACCES;
		return ip_ra_control(sk, 0, NULL);
	case MRT_ADD_VIF:
	case MRT_DEL_VIF:
J
Jianjun Kong 已提交
1236
		if (optlen != sizeof(vif))
S
Stephen Hemminger 已提交
1237
			return -EINVAL;
J
Jianjun Kong 已提交
1238
		if (copy_from_user(&vif, optval, sizeof(vif)))
S
Stephen Hemminger 已提交
1239 1240 1241 1242
			return -EFAULT;
		if (vif.vifc_vifi >= MAXVIFS)
			return -ENFILE;
		rtnl_lock();
J
Jianjun Kong 已提交
1243
		if (optname == MRT_ADD_VIF) {
1244
			ret = vif_add(net, mrt, &vif, sk == mrt->mroute_sk);
S
Stephen Hemminger 已提交
1245
		} else {
1246
			ret = vif_delete(mrt, vif.vifc_vifi, 0, NULL);
S
Stephen Hemminger 已提交
1247 1248 1249
		}
		rtnl_unlock();
		return ret;
L
Linus Torvalds 已提交
1250 1251 1252 1253 1254

		/*
		 *	Manipulate the forwarding caches. These live
		 *	in a sort of kernel/user symbiosis.
		 */
S
Stephen Hemminger 已提交
1255 1256
	case MRT_ADD_MFC:
	case MRT_DEL_MFC:
J
Jianjun Kong 已提交
1257
		if (optlen != sizeof(mfc))
S
Stephen Hemminger 已提交
1258
			return -EINVAL;
J
Jianjun Kong 已提交
1259
		if (copy_from_user(&mfc, optval, sizeof(mfc)))
S
Stephen Hemminger 已提交
1260 1261
			return -EFAULT;
		rtnl_lock();
J
Jianjun Kong 已提交
1262
		if (optname == MRT_DEL_MFC)
1263
			ret = ipmr_mfc_delete(mrt, &mfc);
S
Stephen Hemminger 已提交
1264
		else
1265
			ret = ipmr_mfc_add(net, mrt, &mfc, sk == mrt->mroute_sk);
S
Stephen Hemminger 已提交
1266 1267
		rtnl_unlock();
		return ret;
L
Linus Torvalds 已提交
1268 1269 1270
		/*
		 *	Control PIM assert.
		 */
S
Stephen Hemminger 已提交
1271 1272 1273 1274 1275
	case MRT_ASSERT:
	{
		int v;
		if (get_user(v,(int __user *)optval))
			return -EFAULT;
1276
		mrt->mroute_do_assert = (v) ? 1 : 0;
S
Stephen Hemminger 已提交
1277 1278
		return 0;
	}
L
Linus Torvalds 已提交
1279
#ifdef CONFIG_IP_PIMSM
S
Stephen Hemminger 已提交
1280 1281
	case MRT_PIM:
	{
S
Stephen Hemminger 已提交
1282 1283
		int v;

S
Stephen Hemminger 已提交
1284 1285
		if (get_user(v,(int __user *)optval))
			return -EFAULT;
S
Stephen Hemminger 已提交
1286 1287
		v = (v) ? 1 : 0;

S
Stephen Hemminger 已提交
1288 1289
		rtnl_lock();
		ret = 0;
1290 1291 1292
		if (v != mrt->mroute_do_pim) {
			mrt->mroute_do_pim = v;
			mrt->mroute_do_assert = v;
L
Linus Torvalds 已提交
1293
		}
S
Stephen Hemminger 已提交
1294 1295 1296
		rtnl_unlock();
		return ret;
	}
1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
#endif
#ifdef CONFIG_IP_MROUTE_MULTIPLE_TABLES
	case MRT_TABLE:
	{
		u32 v;

		if (optlen != sizeof(u32))
			return -EINVAL;
		if (get_user(v, (u32 __user *)optval))
			return -EFAULT;
		if (sk == mrt->mroute_sk)
			return -EBUSY;

		rtnl_lock();
		ret = 0;
		if (!ipmr_new_table(net, v))
			ret = -ENOMEM;
		raw_sk(sk)->ipmr_table = v;
		rtnl_unlock();
		return ret;
	}
L
Linus Torvalds 已提交
1318
#endif
S
Stephen Hemminger 已提交
1319 1320 1321 1322 1323 1324
	/*
	 *	Spurious command, or MRT_VERSION which you cannot
	 *	set.
	 */
	default:
		return -ENOPROTOOPT;
L
Linus Torvalds 已提交
1325 1326 1327 1328 1329 1330
	}
}

/*
 *	Getsock opt support for the multicast routing system.
 */
1331

J
Jianjun Kong 已提交
1332
int ip_mroute_getsockopt(struct sock *sk, int optname, char __user *optval, int __user *optlen)
L
Linus Torvalds 已提交
1333 1334 1335
{
	int olr;
	int val;
1336
	struct net *net = sock_net(sk);
1337 1338 1339 1340 1341
	struct mr_table *mrt;

	mrt = ipmr_get_table(net, raw_sk(sk)->ipmr_table ? : RT_TABLE_DEFAULT);
	if (mrt == NULL)
		return -ENOENT;
L
Linus Torvalds 已提交
1342

J
Jianjun Kong 已提交
1343
	if (optname != MRT_VERSION &&
L
Linus Torvalds 已提交
1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355
#ifdef CONFIG_IP_PIMSM
	   optname!=MRT_PIM &&
#endif
	   optname!=MRT_ASSERT)
		return -ENOPROTOOPT;

	if (get_user(olr, optlen))
		return -EFAULT;

	olr = min_t(unsigned int, olr, sizeof(int));
	if (olr < 0)
		return -EINVAL;
1356

J
Jianjun Kong 已提交
1357
	if (put_user(olr, optlen))
L
Linus Torvalds 已提交
1358
		return -EFAULT;
J
Jianjun Kong 已提交
1359 1360
	if (optname == MRT_VERSION)
		val = 0x0305;
L
Linus Torvalds 已提交
1361
#ifdef CONFIG_IP_PIMSM
J
Jianjun Kong 已提交
1362
	else if (optname == MRT_PIM)
1363
		val = mrt->mroute_do_pim;
L
Linus Torvalds 已提交
1364 1365
#endif
	else
1366
		val = mrt->mroute_do_assert;
J
Jianjun Kong 已提交
1367
	if (copy_to_user(optval, &val, olr))
L
Linus Torvalds 已提交
1368 1369 1370 1371 1372 1373 1374
		return -EFAULT;
	return 0;
}

/*
 *	The IP multicast ioctl support routines.
 */
1375

L
Linus Torvalds 已提交
1376 1377 1378 1379 1380 1381
int ipmr_ioctl(struct sock *sk, int cmd, void __user *arg)
{
	struct sioc_sg_req sr;
	struct sioc_vif_req vr;
	struct vif_device *vif;
	struct mfc_cache *c;
1382
	struct net *net = sock_net(sk);
1383 1384 1385 1386 1387
	struct mr_table *mrt;

	mrt = ipmr_get_table(net, raw_sk(sk)->ipmr_table ? : RT_TABLE_DEFAULT);
	if (mrt == NULL)
		return -ENOENT;
1388

S
Stephen Hemminger 已提交
1389 1390
	switch (cmd) {
	case SIOCGETVIFCNT:
J
Jianjun Kong 已提交
1391
		if (copy_from_user(&vr, arg, sizeof(vr)))
S
Stephen Hemminger 已提交
1392
			return -EFAULT;
1393
		if (vr.vifi >= mrt->maxvif)
S
Stephen Hemminger 已提交
1394 1395
			return -EINVAL;
		read_lock(&mrt_lock);
1396 1397
		vif = &mrt->vif_table[vr.vifi];
		if (VIF_EXISTS(mrt, vr.vifi)) {
J
Jianjun Kong 已提交
1398 1399 1400 1401
			vr.icount = vif->pkt_in;
			vr.ocount = vif->pkt_out;
			vr.ibytes = vif->bytes_in;
			vr.obytes = vif->bytes_out;
L
Linus Torvalds 已提交
1402 1403
			read_unlock(&mrt_lock);

J
Jianjun Kong 已提交
1404
			if (copy_to_user(arg, &vr, sizeof(vr)))
S
Stephen Hemminger 已提交
1405 1406 1407 1408 1409 1410
				return -EFAULT;
			return 0;
		}
		read_unlock(&mrt_lock);
		return -EADDRNOTAVAIL;
	case SIOCGETSGCNT:
J
Jianjun Kong 已提交
1411
		if (copy_from_user(&sr, arg, sizeof(sr)))
S
Stephen Hemminger 已提交
1412 1413 1414
			return -EFAULT;

		read_lock(&mrt_lock);
1415
		c = ipmr_cache_find(mrt, sr.src.s_addr, sr.grp.s_addr);
S
Stephen Hemminger 已提交
1416 1417 1418 1419
		if (c) {
			sr.pktcnt = c->mfc_un.res.pkt;
			sr.bytecnt = c->mfc_un.res.bytes;
			sr.wrong_if = c->mfc_un.res.wrong_if;
L
Linus Torvalds 已提交
1420
			read_unlock(&mrt_lock);
S
Stephen Hemminger 已提交
1421

J
Jianjun Kong 已提交
1422
			if (copy_to_user(arg, &sr, sizeof(sr)))
S
Stephen Hemminger 已提交
1423 1424 1425 1426 1427 1428 1429
				return -EFAULT;
			return 0;
		}
		read_unlock(&mrt_lock);
		return -EADDRNOTAVAIL;
	default:
		return -ENOIOCTLCMD;
L
Linus Torvalds 已提交
1430 1431 1432 1433 1434 1435
	}
}


static int ipmr_device_event(struct notifier_block *this, unsigned long event, void *ptr)
{
1436
	struct net_device *dev = ptr;
1437
	struct net *net = dev_net(dev);
1438
	struct mr_table *mrt;
L
Linus Torvalds 已提交
1439 1440
	struct vif_device *v;
	int ct;
1441
	LIST_HEAD(list);
1442

L
Linus Torvalds 已提交
1443 1444
	if (event != NETDEV_UNREGISTER)
		return NOTIFY_DONE;
1445 1446 1447 1448 1449 1450 1451

	ipmr_for_each_table(mrt, net) {
		v = &mrt->vif_table[0];
		for (ct = 0; ct < mrt->maxvif; ct++, v++) {
			if (v->dev == dev)
				vif_delete(mrt, ct, 1, &list);
		}
L
Linus Torvalds 已提交
1452
	}
1453
	unregister_netdevice_many(&list);
L
Linus Torvalds 已提交
1454 1455 1456 1457
	return NOTIFY_DONE;
}


J
Jianjun Kong 已提交
1458
static struct notifier_block ip_mr_notifier = {
L
Linus Torvalds 已提交
1459 1460 1461 1462 1463 1464 1465 1466
	.notifier_call = ipmr_device_event,
};

/*
 * 	Encapsulate a packet by attaching a valid IPIP header to it.
 *	This avoids tunnel drivers and other mess and gives us the speed so
 *	important for multicast video.
 */
1467

A
Al Viro 已提交
1468
static void ip_encap(struct sk_buff *skb, __be32 saddr, __be32 daddr)
L
Linus Torvalds 已提交
1469
{
1470
	struct iphdr *iph;
1471
	struct iphdr *old_iph = ip_hdr(skb);
1472 1473

	skb_push(skb, sizeof(struct iphdr));
1474
	skb->transport_header = skb->network_header;
1475
	skb_reset_network_header(skb);
1476
	iph = ip_hdr(skb);
L
Linus Torvalds 已提交
1477 1478

	iph->version	= 	4;
1479 1480
	iph->tos	=	old_iph->tos;
	iph->ttl	=	old_iph->ttl;
L
Linus Torvalds 已提交
1481 1482 1483 1484 1485 1486
	iph->frag_off	=	0;
	iph->daddr	=	daddr;
	iph->saddr	=	saddr;
	iph->protocol	=	IPPROTO_IPIP;
	iph->ihl	=	5;
	iph->tot_len	=	htons(skb->len);
E
Eric Dumazet 已提交
1487
	ip_select_ident(iph, skb_dst(skb), NULL);
L
Linus Torvalds 已提交
1488 1489 1490 1491 1492 1493 1494 1495 1496 1497
	ip_send_check(iph);

	memset(&(IPCB(skb)->opt), 0, sizeof(IPCB(skb)->opt));
	nf_reset(skb);
}

static inline int ipmr_forward_finish(struct sk_buff *skb)
{
	struct ip_options * opt	= &(IPCB(skb)->opt);

E
Eric Dumazet 已提交
1498
	IP_INC_STATS_BH(dev_net(skb_dst(skb)->dev), IPSTATS_MIB_OUTFORWDATAGRAMS);
L
Linus Torvalds 已提交
1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509

	if (unlikely(opt->optlen))
		ip_forward_options(skb);

	return dst_output(skb);
}

/*
 *	Processing handlers for ipmr_forward
 */

1510 1511
static void ipmr_queue_xmit(struct net *net, struct mr_table *mrt,
			    struct sk_buff *skb, struct mfc_cache *c, int vifi)
L
Linus Torvalds 已提交
1512
{
1513
	const struct iphdr *iph = ip_hdr(skb);
1514
	struct vif_device *vif = &mrt->vif_table[vifi];
L
Linus Torvalds 已提交
1515 1516 1517 1518 1519 1520 1521 1522 1523 1524
	struct net_device *dev;
	struct rtable *rt;
	int    encap = 0;

	if (vif->dev == NULL)
		goto out_free;

#ifdef CONFIG_IP_PIMSM
	if (vif->flags & VIFF_REGISTER) {
		vif->pkt_out++;
J
Jianjun Kong 已提交
1525
		vif->bytes_out += skb->len;
1526 1527
		vif->dev->stats.tx_bytes += skb->len;
		vif->dev->stats.tx_packets++;
1528
		ipmr_cache_report(mrt, skb, vifi, IGMPMSG_WHOLEPKT);
1529
		goto out_free;
L
Linus Torvalds 已提交
1530 1531 1532 1533 1534 1535 1536 1537 1538 1539
	}
#endif

	if (vif->flags&VIFF_TUNNEL) {
		struct flowi fl = { .oif = vif->link,
				    .nl_u = { .ip4_u =
					      { .daddr = vif->remote,
						.saddr = vif->local,
						.tos = RT_TOS(iph->tos) } },
				    .proto = IPPROTO_IPIP };
1540
		if (ip_route_output_key(net, &rt, &fl))
L
Linus Torvalds 已提交
1541 1542 1543 1544 1545 1546 1547 1548
			goto out_free;
		encap = sizeof(struct iphdr);
	} else {
		struct flowi fl = { .oif = vif->link,
				    .nl_u = { .ip4_u =
					      { .daddr = iph->daddr,
						.tos = RT_TOS(iph->tos) } },
				    .proto = IPPROTO_IPIP };
1549
		if (ip_route_output_key(net, &rt, &fl))
L
Linus Torvalds 已提交
1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560
			goto out_free;
	}

	dev = rt->u.dst.dev;

	if (skb->len+encap > dst_mtu(&rt->u.dst) && (ntohs(iph->frag_off) & IP_DF)) {
		/* Do not fragment multicasts. Alas, IPv4 does not
		   allow to send ICMP, so that packets will disappear
		   to blackhole.
		 */

1561
		IP_INC_STATS_BH(dev_net(dev), IPSTATS_MIB_FRAGFAILS);
L
Linus Torvalds 已提交
1562 1563 1564 1565 1566 1567 1568
		ip_rt_put(rt);
		goto out_free;
	}

	encap += LL_RESERVED_SPACE(dev) + rt->u.dst.header_len;

	if (skb_cow(skb, encap)) {
1569
		ip_rt_put(rt);
L
Linus Torvalds 已提交
1570 1571 1572 1573
		goto out_free;
	}

	vif->pkt_out++;
J
Jianjun Kong 已提交
1574
	vif->bytes_out += skb->len;
L
Linus Torvalds 已提交
1575

E
Eric Dumazet 已提交
1576 1577
	skb_dst_drop(skb);
	skb_dst_set(skb, &rt->u.dst);
1578
	ip_decrease_ttl(ip_hdr(skb));
L
Linus Torvalds 已提交
1579 1580 1581 1582 1583 1584

	/* FIXME: forward and output firewalls used to be called here.
	 * What do we do with netfilter? -- RR */
	if (vif->flags & VIFF_TUNNEL) {
		ip_encap(skb, vif->local, vif->remote);
		/* FIXME: extra output firewall step used to be here. --RR */
1585 1586
		vif->dev->stats.tx_packets++;
		vif->dev->stats.tx_bytes += skb->len;
L
Linus Torvalds 已提交
1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601
	}

	IPCB(skb)->flags |= IPSKB_FORWARDED;

	/*
	 * RFC1584 teaches, that DVMRP/PIM router must deliver packets locally
	 * not only before forwarding, but after forwarding on all output
	 * interfaces. It is clear, if mrouter runs a multicasting
	 * program, it should receive packets not depending to what interface
	 * program is joined.
	 * If we will not make it, the program will have to join on all
	 * interfaces. On the other hand, multihoming host (or router, but
	 * not mrouter) cannot join to more than one interface - it will
	 * result in receiving multiple packets.
	 */
1602
	NF_HOOK(PF_INET, NF_INET_FORWARD, skb, skb->dev, dev,
L
Linus Torvalds 已提交
1603 1604 1605 1606 1607 1608 1609 1610
		ipmr_forward_finish);
	return;

out_free:
	kfree_skb(skb);
	return;
}

1611
static int ipmr_find_vif(struct mr_table *mrt, struct net_device *dev)
L
Linus Torvalds 已提交
1612 1613
{
	int ct;
1614 1615 1616

	for (ct = mrt->maxvif-1; ct >= 0; ct--) {
		if (mrt->vif_table[ct].dev == dev)
L
Linus Torvalds 已提交
1617 1618 1619 1620 1621 1622 1623
			break;
	}
	return ct;
}

/* "local" means that we should preserve one skb (for local delivery) */

1624 1625 1626
static int ip_mr_forward(struct net *net, struct mr_table *mrt,
			 struct sk_buff *skb, struct mfc_cache *cache,
			 int local)
L
Linus Torvalds 已提交
1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637
{
	int psend = -1;
	int vif, ct;

	vif = cache->mfc_parent;
	cache->mfc_un.res.pkt++;
	cache->mfc_un.res.bytes += skb->len;

	/*
	 * Wrong interface: drop packet and (maybe) send PIM assert.
	 */
1638
	if (mrt->vif_table[vif].dev != skb->dev) {
L
Linus Torvalds 已提交
1639 1640
		int true_vifi;

E
Eric Dumazet 已提交
1641
		if (skb_rtable(skb)->fl.iif == 0) {
L
Linus Torvalds 已提交
1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656
			/* It is our own packet, looped back.
			   Very complicated situation...

			   The best workaround until routing daemons will be
			   fixed is not to redistribute packet, if it was
			   send through wrong interface. It means, that
			   multicast applications WILL NOT work for
			   (S,G), which have default multicast route pointing
			   to wrong oif. In any case, it is not a good
			   idea to use multicasting applications on router.
			 */
			goto dont_forward;
		}

		cache->mfc_un.res.wrong_if++;
1657
		true_vifi = ipmr_find_vif(mrt, skb->dev);
L
Linus Torvalds 已提交
1658

1659
		if (true_vifi >= 0 && mrt->mroute_do_assert &&
L
Linus Torvalds 已提交
1660 1661 1662 1663 1664
		    /* pimsm uses asserts, when switching from RPT to SPT,
		       so that we cannot check that packet arrived on an oif.
		       It is bad, but otherwise we would need to move pretty
		       large chunk of pimd to kernel. Ough... --ANK
		     */
1665
		    (mrt->mroute_do_pim ||
1666
		     cache->mfc_un.res.ttls[true_vifi] < 255) &&
1667
		    time_after(jiffies,
L
Linus Torvalds 已提交
1668 1669
			       cache->mfc_un.res.last_assert + MFC_ASSERT_THRESH)) {
			cache->mfc_un.res.last_assert = jiffies;
1670
			ipmr_cache_report(mrt, skb, true_vifi, IGMPMSG_WRONGVIF);
L
Linus Torvalds 已提交
1671 1672 1673 1674
		}
		goto dont_forward;
	}

1675 1676
	mrt->vif_table[vif].pkt_in++;
	mrt->vif_table[vif].bytes_in += skb->len;
L
Linus Torvalds 已提交
1677 1678 1679 1680 1681

	/*
	 *	Forward the frame
	 */
	for (ct = cache->mfc_un.res.maxvif-1; ct >= cache->mfc_un.res.minvif; ct--) {
1682
		if (ip_hdr(skb)->ttl > cache->mfc_un.res.ttls[ct]) {
L
Linus Torvalds 已提交
1683 1684 1685
			if (psend != -1) {
				struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
				if (skb2)
1686 1687
					ipmr_queue_xmit(net, mrt, skb2, cache,
							psend);
L
Linus Torvalds 已提交
1688
			}
J
Jianjun Kong 已提交
1689
			psend = ct;
L
Linus Torvalds 已提交
1690 1691 1692 1693 1694 1695
		}
	}
	if (psend != -1) {
		if (local) {
			struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
			if (skb2)
1696
				ipmr_queue_xmit(net, mrt, skb2, cache, psend);
L
Linus Torvalds 已提交
1697
		} else {
1698
			ipmr_queue_xmit(net, mrt, skb, cache, psend);
L
Linus Torvalds 已提交
1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716
			return 0;
		}
	}

dont_forward:
	if (!local)
		kfree_skb(skb);
	return 0;
}


/*
 *	Multicast packets for forwarding arrive here
 */

int ip_mr_input(struct sk_buff *skb)
{
	struct mfc_cache *cache;
1717
	struct net *net = dev_net(skb->dev);
E
Eric Dumazet 已提交
1718
	int local = skb_rtable(skb)->rt_flags & RTCF_LOCAL;
1719 1720
	struct mr_table *mrt;
	int err;
L
Linus Torvalds 已提交
1721 1722 1723 1724 1725 1726 1727

	/* Packet is looped back after forward, it should not be
	   forwarded second time, but still can be delivered locally.
	 */
	if (IPCB(skb)->flags&IPSKB_FORWARDED)
		goto dont_forward;

1728 1729 1730 1731
	err = ipmr_fib_lookup(net, &skb_rtable(skb)->fl, &mrt);
	if (err < 0)
		return err;

L
Linus Torvalds 已提交
1732 1733 1734 1735
	if (!local) {
		    if (IPCB(skb)->opt.router_alert) {
			    if (ip_call_ra_chain(skb))
				    return 0;
1736
		    } else if (ip_hdr(skb)->protocol == IPPROTO_IGMP){
L
Linus Torvalds 已提交
1737 1738 1739 1740 1741 1742 1743
			    /* IGMPv1 (and broken IGMPv2 implementations sort of
			       Cisco IOS <= 11.2(8)) do not put router alert
			       option to IGMP packets destined to routable
			       groups. It is very bad, because it means
			       that we can forward NO IGMP messages.
			     */
			    read_lock(&mrt_lock);
1744
			    if (mrt->mroute_sk) {
1745
				    nf_reset(skb);
1746
				    raw_rcv(mrt->mroute_sk, skb);
L
Linus Torvalds 已提交
1747 1748 1749 1750 1751 1752 1753 1754
				    read_unlock(&mrt_lock);
				    return 0;
			    }
			    read_unlock(&mrt_lock);
		    }
	}

	read_lock(&mrt_lock);
1755
	cache = ipmr_cache_find(mrt, ip_hdr(skb)->saddr, ip_hdr(skb)->daddr);
L
Linus Torvalds 已提交
1756 1757 1758 1759

	/*
	 *	No usable cache entry
	 */
J
Jianjun Kong 已提交
1760
	if (cache == NULL) {
L
Linus Torvalds 已提交
1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772
		int vif;

		if (local) {
			struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
			ip_local_deliver(skb);
			if (skb2 == NULL) {
				read_unlock(&mrt_lock);
				return -ENOBUFS;
			}
			skb = skb2;
		}

1773
		vif = ipmr_find_vif(mrt, skb->dev);
L
Linus Torvalds 已提交
1774
		if (vif >= 0) {
1775
			int err2 = ipmr_cache_unresolved(mrt, vif, skb);
L
Linus Torvalds 已提交
1776 1777
			read_unlock(&mrt_lock);

1778
			return err2;
L
Linus Torvalds 已提交
1779 1780 1781 1782 1783 1784
		}
		read_unlock(&mrt_lock);
		kfree_skb(skb);
		return -ENODEV;
	}

1785
	ip_mr_forward(net, mrt, skb, cache, local);
L
Linus Torvalds 已提交
1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800

	read_unlock(&mrt_lock);

	if (local)
		return ip_local_deliver(skb);

	return 0;

dont_forward:
	if (local)
		return ip_local_deliver(skb);
	kfree_skb(skb);
	return 0;
}

I
Ilpo Järvinen 已提交
1801
#ifdef CONFIG_IP_PIMSM
1802 1803
static int __pim_rcv(struct mr_table *mrt, struct sk_buff *skb,
		     unsigned int pimlen)
L
Linus Torvalds 已提交
1804
{
I
Ilpo Järvinen 已提交
1805 1806
	struct net_device *reg_dev = NULL;
	struct iphdr *encap;
L
Linus Torvalds 已提交
1807

I
Ilpo Järvinen 已提交
1808
	encap = (struct iphdr *)(skb_transport_header(skb) + pimlen);
L
Linus Torvalds 已提交
1809 1810 1811 1812 1813 1814
	/*
	   Check that:
	   a. packet is really destinted to a multicast group
	   b. packet is not a NULL-REGISTER
	   c. packet is not truncated
	 */
1815
	if (!ipv4_is_multicast(encap->daddr) ||
L
Linus Torvalds 已提交
1816
	    encap->tot_len == 0 ||
I
Ilpo Järvinen 已提交
1817 1818
	    ntohs(encap->tot_len) + pimlen > skb->len)
		return 1;
L
Linus Torvalds 已提交
1819 1820

	read_lock(&mrt_lock);
1821 1822
	if (mrt->mroute_reg_vif_num >= 0)
		reg_dev = mrt->vif_table[mrt->mroute_reg_vif_num].dev;
L
Linus Torvalds 已提交
1823 1824 1825 1826
	if (reg_dev)
		dev_hold(reg_dev);
	read_unlock(&mrt_lock);

1827
	if (reg_dev == NULL)
I
Ilpo Järvinen 已提交
1828
		return 1;
L
Linus Torvalds 已提交
1829

1830
	skb->mac_header = skb->network_header;
L
Linus Torvalds 已提交
1831
	skb_pull(skb, (u8*)encap - skb->data);
1832
	skb_reset_network_header(skb);
L
Linus Torvalds 已提交
1833 1834 1835 1836
	skb->dev = reg_dev;
	skb->protocol = htons(ETH_P_IP);
	skb->ip_summed = 0;
	skb->pkt_type = PACKET_HOST;
E
Eric Dumazet 已提交
1837
	skb_dst_drop(skb);
1838 1839
	reg_dev->stats.rx_bytes += skb->len;
	reg_dev->stats.rx_packets++;
L
Linus Torvalds 已提交
1840 1841 1842
	nf_reset(skb);
	netif_rx(skb);
	dev_put(reg_dev);
I
Ilpo Järvinen 已提交
1843

L
Linus Torvalds 已提交
1844
	return 0;
I
Ilpo Järvinen 已提交
1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855
}
#endif

#ifdef CONFIG_IP_PIMSM_V1
/*
 * Handle IGMP messages of PIMv1
 */

int pim_rcv_v1(struct sk_buff * skb)
{
	struct igmphdr *pim;
1856
	struct net *net = dev_net(skb->dev);
1857
	struct mr_table *mrt;
I
Ilpo Järvinen 已提交
1858 1859 1860 1861 1862 1863

	if (!pskb_may_pull(skb, sizeof(*pim) + sizeof(struct iphdr)))
		goto drop;

	pim = igmp_hdr(skb);

1864 1865 1866
	if (ipmr_fib_lookup(net, &skb_rtable(skb)->fl, &mrt) < 0)
		goto drop;

1867
	if (!mrt->mroute_do_pim ||
I
Ilpo Järvinen 已提交
1868 1869 1870
	    pim->group != PIM_V1_VERSION || pim->code != PIM_V1_REGISTER)
		goto drop;

1871
	if (__pim_rcv(mrt, skb, sizeof(*pim))) {
I
Ilpo Järvinen 已提交
1872 1873 1874
drop:
		kfree_skb(skb);
	}
L
Linus Torvalds 已提交
1875 1876 1877 1878 1879 1880 1881 1882
	return 0;
}
#endif

#ifdef CONFIG_IP_PIMSM_V2
static int pim_rcv(struct sk_buff * skb)
{
	struct pimreghdr *pim;
1883 1884
	struct net *net = dev_net(skb->dev);
	struct mr_table *mrt;
L
Linus Torvalds 已提交
1885

I
Ilpo Järvinen 已提交
1886
	if (!pskb_may_pull(skb, sizeof(*pim) + sizeof(struct iphdr)))
L
Linus Torvalds 已提交
1887 1888
		goto drop;

1889
	pim = (struct pimreghdr *)skb_transport_header(skb);
1890
	if (pim->type != ((PIM_VERSION<<4)|(PIM_REGISTER)) ||
L
Linus Torvalds 已提交
1891
	    (pim->flags&PIM_NULL_REGISTER) ||
1892
	    (ip_compute_csum((void *)pim, sizeof(*pim)) != 0 &&
1893
	     csum_fold(skb_checksum(skb, 0, skb->len, 0))))
L
Linus Torvalds 已提交
1894 1895
		goto drop;

1896 1897 1898 1899
	if (ipmr_fib_lookup(net, &skb_rtable(skb)->fl, &mrt) < 0)
		goto drop;

	if (__pim_rcv(mrt, skb, sizeof(*pim))) {
I
Ilpo Järvinen 已提交
1900 1901 1902
drop:
		kfree_skb(skb);
	}
L
Linus Torvalds 已提交
1903 1904 1905 1906 1907
	return 0;
}
#endif

static int
1908
ipmr_fill_mroute(struct mr_table *mrt, struct sk_buff *skb, struct mfc_cache *c,
1909
		 struct rtmsg *rtm)
L
Linus Torvalds 已提交
1910 1911 1912
{
	int ct;
	struct rtnexthop *nhp;
1913
	u8 *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
1914 1915
	struct rtattr *mp_head;

1916 1917 1918 1919
	/* If cache is unresolved, don't try to parse IIF and OIF */
	if (c->mfc_parent > MAXVIFS)
		return -ENOENT;

1920 1921
	if (VIF_EXISTS(mrt, c->mfc_parent))
		RTA_PUT(skb, RTA_IIF, 4, &mrt->vif_table[c->mfc_parent].dev->ifindex);
L
Linus Torvalds 已提交
1922

J
Jianjun Kong 已提交
1923
	mp_head = (struct rtattr *)skb_put(skb, RTA_LENGTH(0));
L
Linus Torvalds 已提交
1924 1925

	for (ct = c->mfc_un.res.minvif; ct < c->mfc_un.res.maxvif; ct++) {
1926
		if (VIF_EXISTS(mrt, ct) && c->mfc_un.res.ttls[ct] < 255) {
L
Linus Torvalds 已提交
1927 1928
			if (skb_tailroom(skb) < RTA_ALIGN(RTA_ALIGN(sizeof(*nhp)) + 4))
				goto rtattr_failure;
J
Jianjun Kong 已提交
1929
			nhp = (struct rtnexthop *)skb_put(skb, RTA_ALIGN(sizeof(*nhp)));
L
Linus Torvalds 已提交
1930 1931
			nhp->rtnh_flags = 0;
			nhp->rtnh_hops = c->mfc_un.res.ttls[ct];
1932
			nhp->rtnh_ifindex = mrt->vif_table[ct].dev->ifindex;
L
Linus Torvalds 已提交
1933 1934 1935 1936
			nhp->rtnh_len = sizeof(*nhp);
		}
	}
	mp_head->rta_type = RTA_MULTIPATH;
1937
	mp_head->rta_len = skb_tail_pointer(skb) - (u8 *)mp_head;
L
Linus Torvalds 已提交
1938 1939 1940 1941
	rtm->rtm_type = RTN_MULTICAST;
	return 1;

rtattr_failure:
1942
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1943 1944 1945
	return -EMSGSIZE;
}

1946 1947
int ipmr_get_route(struct net *net,
		   struct sk_buff *skb, struct rtmsg *rtm, int nowait)
L
Linus Torvalds 已提交
1948 1949
{
	int err;
1950
	struct mr_table *mrt;
L
Linus Torvalds 已提交
1951
	struct mfc_cache *cache;
E
Eric Dumazet 已提交
1952
	struct rtable *rt = skb_rtable(skb);
L
Linus Torvalds 已提交
1953

1954 1955 1956 1957
	mrt = ipmr_get_table(net, RT_TABLE_DEFAULT);
	if (mrt == NULL)
		return -ENOENT;

L
Linus Torvalds 已提交
1958
	read_lock(&mrt_lock);
1959
	cache = ipmr_cache_find(mrt, rt->rt_src, rt->rt_dst);
L
Linus Torvalds 已提交
1960

J
Jianjun Kong 已提交
1961
	if (cache == NULL) {
1962
		struct sk_buff *skb2;
1963
		struct iphdr *iph;
L
Linus Torvalds 已提交
1964 1965 1966 1967 1968 1969 1970 1971 1972
		struct net_device *dev;
		int vif;

		if (nowait) {
			read_unlock(&mrt_lock);
			return -EAGAIN;
		}

		dev = skb->dev;
1973
		if (dev == NULL || (vif = ipmr_find_vif(mrt, dev)) < 0) {
L
Linus Torvalds 已提交
1974 1975 1976
			read_unlock(&mrt_lock);
			return -ENODEV;
		}
1977 1978 1979 1980 1981 1982
		skb2 = skb_clone(skb, GFP_ATOMIC);
		if (!skb2) {
			read_unlock(&mrt_lock);
			return -ENOMEM;
		}

1983 1984
		skb_push(skb2, sizeof(struct iphdr));
		skb_reset_network_header(skb2);
1985 1986 1987 1988 1989
		iph = ip_hdr(skb2);
		iph->ihl = sizeof(struct iphdr) >> 2;
		iph->saddr = rt->rt_src;
		iph->daddr = rt->rt_dst;
		iph->version = 0;
1990
		err = ipmr_cache_unresolved(mrt, vif, skb2);
L
Linus Torvalds 已提交
1991 1992 1993 1994 1995 1996
		read_unlock(&mrt_lock);
		return err;
	}

	if (!nowait && (rtm->rtm_flags&RTM_F_NOTIFY))
		cache->mfc_flags |= MFC_NOTIFY;
1997
	err = ipmr_fill_mroute(mrt, skb, cache, rtm);
L
Linus Torvalds 已提交
1998 1999 2000 2001
	read_unlock(&mrt_lock);
	return err;
}

2002
#ifdef CONFIG_PROC_FS
L
Linus Torvalds 已提交
2003 2004 2005 2006
/*
 *	The /proc interfaces to multicast routing /proc/ip_mr_cache /proc/ip_mr_vif
 */
struct ipmr_vif_iter {
2007
	struct seq_net_private p;
2008
	struct mr_table *mrt;
L
Linus Torvalds 已提交
2009 2010 2011
	int ct;
};

2012 2013
static struct vif_device *ipmr_vif_seq_idx(struct net *net,
					   struct ipmr_vif_iter *iter,
L
Linus Torvalds 已提交
2014 2015
					   loff_t pos)
{
2016
	struct mr_table *mrt = iter->mrt;
2017 2018 2019

	for (iter->ct = 0; iter->ct < mrt->maxvif; ++iter->ct) {
		if (!VIF_EXISTS(mrt, iter->ct))
L
Linus Torvalds 已提交
2020
			continue;
2021
		if (pos-- == 0)
2022
			return &mrt->vif_table[iter->ct];
L
Linus Torvalds 已提交
2023 2024 2025 2026 2027
	}
	return NULL;
}

static void *ipmr_vif_seq_start(struct seq_file *seq, loff_t *pos)
S
Stephen Hemminger 已提交
2028
	__acquires(mrt_lock)
L
Linus Torvalds 已提交
2029
{
2030
	struct ipmr_vif_iter *iter = seq->private;
2031
	struct net *net = seq_file_net(seq);
2032 2033 2034 2035 2036 2037 2038
	struct mr_table *mrt;

	mrt = ipmr_get_table(net, RT_TABLE_DEFAULT);
	if (mrt == NULL)
		return ERR_PTR(-ENOENT);

	iter->mrt = mrt;
2039

L
Linus Torvalds 已提交
2040
	read_lock(&mrt_lock);
2041
	return *pos ? ipmr_vif_seq_idx(net, seq->private, *pos - 1)
L
Linus Torvalds 已提交
2042 2043 2044 2045 2046 2047
		: SEQ_START_TOKEN;
}

static void *ipmr_vif_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
	struct ipmr_vif_iter *iter = seq->private;
2048
	struct net *net = seq_file_net(seq);
2049
	struct mr_table *mrt = iter->mrt;
L
Linus Torvalds 已提交
2050 2051 2052

	++*pos;
	if (v == SEQ_START_TOKEN)
2053
		return ipmr_vif_seq_idx(net, iter, 0);
2054

2055 2056
	while (++iter->ct < mrt->maxvif) {
		if (!VIF_EXISTS(mrt, iter->ct))
L
Linus Torvalds 已提交
2057
			continue;
2058
		return &mrt->vif_table[iter->ct];
L
Linus Torvalds 已提交
2059 2060 2061 2062 2063
	}
	return NULL;
}

static void ipmr_vif_seq_stop(struct seq_file *seq, void *v)
S
Stephen Hemminger 已提交
2064
	__releases(mrt_lock)
L
Linus Torvalds 已提交
2065 2066 2067 2068 2069 2070
{
	read_unlock(&mrt_lock);
}

static int ipmr_vif_seq_show(struct seq_file *seq, void *v)
{
2071 2072
	struct ipmr_vif_iter *iter = seq->private;
	struct mr_table *mrt = iter->mrt;
2073

L
Linus Torvalds 已提交
2074
	if (v == SEQ_START_TOKEN) {
2075
		seq_puts(seq,
L
Linus Torvalds 已提交
2076 2077 2078 2079 2080 2081 2082
			 "Interface      BytesIn  PktsIn  BytesOut PktsOut Flags Local    Remote\n");
	} else {
		const struct vif_device *vif = v;
		const char *name =  vif->dev ? vif->dev->name : "none";

		seq_printf(seq,
			   "%2Zd %-10s %8ld %7ld  %8ld %7ld %05X %08X %08X\n",
2083
			   vif - mrt->vif_table,
2084
			   name, vif->bytes_in, vif->pkt_in,
L
Linus Torvalds 已提交
2085 2086 2087 2088 2089 2090
			   vif->bytes_out, vif->pkt_out,
			   vif->flags, vif->local, vif->remote);
	}
	return 0;
}

2091
static const struct seq_operations ipmr_vif_seq_ops = {
L
Linus Torvalds 已提交
2092 2093 2094 2095 2096 2097 2098 2099
	.start = ipmr_vif_seq_start,
	.next  = ipmr_vif_seq_next,
	.stop  = ipmr_vif_seq_stop,
	.show  = ipmr_vif_seq_show,
};

static int ipmr_vif_open(struct inode *inode, struct file *file)
{
2100 2101
	return seq_open_net(inode, file, &ipmr_vif_seq_ops,
			    sizeof(struct ipmr_vif_iter));
L
Linus Torvalds 已提交
2102 2103
}

2104
static const struct file_operations ipmr_vif_fops = {
L
Linus Torvalds 已提交
2105 2106 2107 2108
	.owner	 = THIS_MODULE,
	.open    = ipmr_vif_open,
	.read    = seq_read,
	.llseek  = seq_lseek,
2109
	.release = seq_release_net,
L
Linus Torvalds 已提交
2110 2111 2112
};

struct ipmr_mfc_iter {
2113
	struct seq_net_private p;
2114
	struct mr_table *mrt;
2115
	struct list_head *cache;
L
Linus Torvalds 已提交
2116 2117 2118 2119
	int ct;
};


2120 2121
static struct mfc_cache *ipmr_mfc_seq_idx(struct net *net,
					  struct ipmr_mfc_iter *it, loff_t pos)
L
Linus Torvalds 已提交
2122
{
2123
	struct mr_table *mrt = it->mrt;
L
Linus Torvalds 已提交
2124 2125 2126
	struct mfc_cache *mfc;

	read_lock(&mrt_lock);
2127
	for (it->ct = 0; it->ct < MFC_LINES; it->ct++) {
2128
		it->cache = &mrt->mfc_cache_array[it->ct];
2129
		list_for_each_entry(mfc, it->cache, list)
2130
			if (pos-- == 0)
L
Linus Torvalds 已提交
2131
				return mfc;
2132
	}
L
Linus Torvalds 已提交
2133 2134 2135
	read_unlock(&mrt_lock);

	spin_lock_bh(&mfc_unres_lock);
2136
	it->cache = &mrt->mfc_unres_queue;
2137
	list_for_each_entry(mfc, it->cache, list)
2138
		if (pos-- == 0)
L
Linus Torvalds 已提交
2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149
			return mfc;
	spin_unlock_bh(&mfc_unres_lock);

	it->cache = NULL;
	return NULL;
}


static void *ipmr_mfc_seq_start(struct seq_file *seq, loff_t *pos)
{
	struct ipmr_mfc_iter *it = seq->private;
2150
	struct net *net = seq_file_net(seq);
2151 2152 2153 2154 2155
	struct mr_table *mrt;

	mrt = ipmr_get_table(net, RT_TABLE_DEFAULT);
	if (mrt == NULL)
		return ERR_PTR(-ENOENT);
2156

2157
	it->mrt = mrt;
L
Linus Torvalds 已提交
2158 2159
	it->cache = NULL;
	it->ct = 0;
2160
	return *pos ? ipmr_mfc_seq_idx(net, seq->private, *pos - 1)
L
Linus Torvalds 已提交
2161 2162 2163 2164 2165 2166 2167
		: SEQ_START_TOKEN;
}

static void *ipmr_mfc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
	struct mfc_cache *mfc = v;
	struct ipmr_mfc_iter *it = seq->private;
2168
	struct net *net = seq_file_net(seq);
2169
	struct mr_table *mrt = it->mrt;
L
Linus Torvalds 已提交
2170 2171 2172 2173

	++*pos;

	if (v == SEQ_START_TOKEN)
2174
		return ipmr_mfc_seq_idx(net, seq->private, 0);
L
Linus Torvalds 已提交
2175

2176 2177
	if (mfc->list.next != it->cache)
		return list_entry(mfc->list.next, struct mfc_cache, list);
2178

2179
	if (it->cache == &mrt->mfc_unres_queue)
L
Linus Torvalds 已提交
2180 2181
		goto end_of_list;

2182
	BUG_ON(it->cache != &mrt->mfc_cache_array[it->ct]);
L
Linus Torvalds 已提交
2183 2184

	while (++it->ct < MFC_LINES) {
2185
		it->cache = &mrt->mfc_cache_array[it->ct];
2186 2187 2188
		if (list_empty(it->cache))
			continue;
		return list_first_entry(it->cache, struct mfc_cache, list);
L
Linus Torvalds 已提交
2189 2190 2191 2192
	}

	/* exhausted cache_array, show unresolved */
	read_unlock(&mrt_lock);
2193
	it->cache = &mrt->mfc_unres_queue;
L
Linus Torvalds 已提交
2194
	it->ct = 0;
2195

L
Linus Torvalds 已提交
2196
	spin_lock_bh(&mfc_unres_lock);
2197 2198
	if (!list_empty(it->cache))
		return list_first_entry(it->cache, struct mfc_cache, list);
L
Linus Torvalds 已提交
2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209

 end_of_list:
	spin_unlock_bh(&mfc_unres_lock);
	it->cache = NULL;

	return NULL;
}

static void ipmr_mfc_seq_stop(struct seq_file *seq, void *v)
{
	struct ipmr_mfc_iter *it = seq->private;
2210
	struct mr_table *mrt = it->mrt;
L
Linus Torvalds 已提交
2211

2212
	if (it->cache == &mrt->mfc_unres_queue)
L
Linus Torvalds 已提交
2213
		spin_unlock_bh(&mfc_unres_lock);
2214
	else if (it->cache == &mrt->mfc_cache_array[it->ct])
L
Linus Torvalds 已提交
2215 2216 2217 2218 2219 2220 2221 2222
		read_unlock(&mrt_lock);
}

static int ipmr_mfc_seq_show(struct seq_file *seq, void *v)
{
	int n;

	if (v == SEQ_START_TOKEN) {
2223
		seq_puts(seq,
L
Linus Torvalds 已提交
2224 2225 2226 2227
		 "Group    Origin   Iif     Pkts    Bytes    Wrong Oifs\n");
	} else {
		const struct mfc_cache *mfc = v;
		const struct ipmr_mfc_iter *it = seq->private;
2228
		const struct mr_table *mrt = it->mrt;
2229

2230 2231 2232
		seq_printf(seq, "%08X %08X %-3hd",
			   (__force u32) mfc->mfc_mcastgrp,
			   (__force u32) mfc->mfc_origin,
2233
			   mfc->mfc_parent);
L
Linus Torvalds 已提交
2234

2235
		if (it->cache != &mrt->mfc_unres_queue) {
2236 2237 2238 2239
			seq_printf(seq, " %8lu %8lu %8lu",
				   mfc->mfc_un.res.pkt,
				   mfc->mfc_un.res.bytes,
				   mfc->mfc_un.res.wrong_if);
S
Stephen Hemminger 已提交
2240 2241
			for (n = mfc->mfc_un.res.minvif;
			     n < mfc->mfc_un.res.maxvif; n++ ) {
2242
				if (VIF_EXISTS(mrt, n) &&
2243 2244
				    mfc->mfc_un.res.ttls[n] < 255)
					seq_printf(seq,
2245
					   " %2d:%-3d",
L
Linus Torvalds 已提交
2246 2247
					   n, mfc->mfc_un.res.ttls[n]);
			}
2248 2249 2250 2251 2252
		} else {
			/* unresolved mfc_caches don't contain
			 * pkt, bytes and wrong_if values
			 */
			seq_printf(seq, " %8lu %8lu %8lu", 0ul, 0ul, 0ul);
L
Linus Torvalds 已提交
2253 2254 2255 2256 2257 2258
		}
		seq_putc(seq, '\n');
	}
	return 0;
}

2259
static const struct seq_operations ipmr_mfc_seq_ops = {
L
Linus Torvalds 已提交
2260 2261 2262 2263 2264 2265 2266 2267
	.start = ipmr_mfc_seq_start,
	.next  = ipmr_mfc_seq_next,
	.stop  = ipmr_mfc_seq_stop,
	.show  = ipmr_mfc_seq_show,
};

static int ipmr_mfc_open(struct inode *inode, struct file *file)
{
2268 2269
	return seq_open_net(inode, file, &ipmr_mfc_seq_ops,
			    sizeof(struct ipmr_mfc_iter));
L
Linus Torvalds 已提交
2270 2271
}

2272
static const struct file_operations ipmr_mfc_fops = {
L
Linus Torvalds 已提交
2273 2274 2275 2276
	.owner	 = THIS_MODULE,
	.open    = ipmr_mfc_open,
	.read    = seq_read,
	.llseek  = seq_lseek,
2277
	.release = seq_release_net,
L
Linus Torvalds 已提交
2278
};
2279
#endif
L
Linus Torvalds 已提交
2280 2281

#ifdef CONFIG_IP_PIMSM_V2
2282
static const struct net_protocol pim_protocol = {
L
Linus Torvalds 已提交
2283
	.handler	=	pim_rcv,
T
Tom Goff 已提交
2284
	.netns_ok	=	1,
L
Linus Torvalds 已提交
2285 2286 2287 2288 2289 2290 2291
};
#endif


/*
 *	Setup for IP multicast routing
 */
2292 2293
static int __net_init ipmr_net_init(struct net *net)
{
2294
	int err;
2295

2296 2297
	err = ipmr_rules_init(net);
	if (err < 0)
2298
		goto fail;
2299 2300 2301 2302 2303 2304 2305 2306

#ifdef CONFIG_PROC_FS
	err = -ENOMEM;
	if (!proc_net_fops_create(net, "ip_mr_vif", 0, &ipmr_vif_fops))
		goto proc_vif_fail;
	if (!proc_net_fops_create(net, "ip_mr_cache", 0, &ipmr_mfc_fops))
		goto proc_cache_fail;
#endif
2307 2308
	return 0;

2309 2310 2311 2312
#ifdef CONFIG_PROC_FS
proc_cache_fail:
	proc_net_remove(net, "ip_mr_vif");
proc_vif_fail:
2313
	ipmr_rules_exit(net);
2314
#endif
2315 2316 2317 2318 2319 2320
fail:
	return err;
}

static void __net_exit ipmr_net_exit(struct net *net)
{
2321 2322 2323 2324
#ifdef CONFIG_PROC_FS
	proc_net_remove(net, "ip_mr_cache");
	proc_net_remove(net, "ip_mr_vif");
#endif
2325
	ipmr_rules_exit(net);
2326 2327 2328 2329 2330 2331
}

static struct pernet_operations ipmr_net_ops = {
	.init = ipmr_net_init,
	.exit = ipmr_net_exit,
};
2332

W
Wang Chen 已提交
2333
int __init ip_mr_init(void)
L
Linus Torvalds 已提交
2334
{
W
Wang Chen 已提交
2335 2336
	int err;

L
Linus Torvalds 已提交
2337 2338
	mrt_cachep = kmem_cache_create("ip_mrt_cache",
				       sizeof(struct mfc_cache),
A
Alexey Dobriyan 已提交
2339
				       0, SLAB_HWCACHE_ALIGN|SLAB_PANIC,
2340
				       NULL);
W
Wang Chen 已提交
2341 2342 2343
	if (!mrt_cachep)
		return -ENOMEM;

2344 2345 2346 2347
	err = register_pernet_subsys(&ipmr_net_ops);
	if (err)
		goto reg_pernet_fail;

W
Wang Chen 已提交
2348 2349 2350
	err = register_netdevice_notifier(&ip_mr_notifier);
	if (err)
		goto reg_notif_fail;
T
Tom Goff 已提交
2351 2352 2353 2354 2355 2356 2357
#ifdef CONFIG_IP_PIMSM_V2
	if (inet_add_protocol(&pim_protocol, IPPROTO_PIM) < 0) {
		printk(KERN_ERR "ip_mr_init: can't add PIM protocol\n");
		err = -EAGAIN;
		goto add_proto_fail;
	}
#endif
W
Wang Chen 已提交
2358
	return 0;
2359

T
Tom Goff 已提交
2360 2361 2362 2363
#ifdef CONFIG_IP_PIMSM_V2
add_proto_fail:
	unregister_netdevice_notifier(&ip_mr_notifier);
#endif
B
Benjamin Thery 已提交
2364
reg_notif_fail:
2365 2366
	unregister_pernet_subsys(&ipmr_net_ops);
reg_pernet_fail:
B
Benjamin Thery 已提交
2367
	kmem_cache_destroy(mrt_cachep);
W
Wang Chen 已提交
2368
	return err;
L
Linus Torvalds 已提交
2369
}