ipmr.c 44.4 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 <net/net_namespace.h>
L
Linus Torvalds 已提交
51 52 53
#include <net/ip.h>
#include <net/protocol.h>
#include <linux/skbuff.h>
54
#include <net/route.h>
L
Linus Torvalds 已提交
55 56 57 58 59 60 61 62 63
#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>
64
#include <net/netlink.h>
L
Linus Torvalds 已提交
65 66 67 68 69 70 71 72 73 74 75 76 77 78 79

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

/* 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
 */

80
#define VIF_EXISTS(_net, _idx) ((_net)->ipv4.vif_table[_idx].dev != NULL)
L
Linus Torvalds 已提交
81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100

static int mroute_do_assert;				/* Set in PIM assert	*/
static int mroute_do_pim;

static struct mfc_cache *mfc_cache_array[MFC_LINES];	/* Forwarding cache	*/

static struct mfc_cache *mfc_unres_queue;		/* Queue of unresolved entries */
static atomic_t cache_resolve_queue_len;		/* Size of unresolved	*/

/* 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.
 */

101
static struct kmem_cache *mrt_cachep __read_mostly;
L
Linus Torvalds 已提交
102 103 104 105 106 107 108 109 110 111 112 113 114

static int ip_mr_forward(struct sk_buff *skb, struct mfc_cache *cache, int local);
static int ipmr_cache_report(struct sk_buff *pkt, vifi_t vifi, int assert);
static int ipmr_fill_mroute(struct sk_buff *skb, struct mfc_cache *c, struct rtmsg *rtm);

#ifdef CONFIG_IP_PIMSM_V2
static struct net_protocol pim_protocol;
#endif

static struct timer_list ipmr_expire_timer;

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

115 116 117 118 119 120
static void ipmr_del_tunnel(struct net_device *dev, struct vifctl *v)
{
	dev_close(dev);

	dev = __dev_get_by_name(&init_net, "tunl0");
	if (dev) {
121
		const struct net_device_ops *ops = dev->netdev_ops;
122 123 124 125 126 127 128 129 130 131 132 133
		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;

134 135 136 137 138 139 140
		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);
		}
141 142 143
	}
}

L
Linus Torvalds 已提交
144 145 146 147 148
static
struct net_device *ipmr_new_tunnel(struct vifctl *v)
{
	struct net_device  *dev;

149
	dev = __dev_get_by_name(&init_net, "tunl0");
L
Linus Torvalds 已提交
150 151

	if (dev) {
152
		const struct net_device_ops *ops = dev->netdev_ops;
L
Linus Torvalds 已提交
153 154 155 156 157 158 159 160 161 162 163 164
		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 已提交
165
		ifr.ifr_ifru.ifru_data = (__force void __user *)&p;
L
Linus Torvalds 已提交
166

167 168 169 170 171 172 173 174
		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 已提交
175 176 177

		dev = NULL;

178
		if (err == 0 && (dev = __dev_get_by_name(&init_net, p.name)) != NULL) {
L
Linus Torvalds 已提交
179 180
			dev->flags |= IFF_MULTICAST;

181
			in_dev = __in_dev_get_rtnl(dev);
182
			if (in_dev == NULL)
L
Linus Torvalds 已提交
183
				goto failure;
184 185 186

			ipv4_devconf_setall(in_dev);
			IPV4_DEVCONF(in_dev->cnf, RP_FILTER) = 0;
L
Linus Torvalds 已提交
187 188 189

			if (dev_open(dev))
				goto failure;
190
			dev_hold(dev);
L
Linus Torvalds 已提交
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210
		}
	}
	return dev;

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

	unregister_netdevice(dev);
	return NULL;
}

#ifdef CONFIG_IP_PIMSM

static int reg_vif_num = -1;

static int reg_vif_xmit(struct sk_buff *skb, struct net_device *dev)
{
	read_lock(&mrt_lock);
211 212
	dev->stats.tx_bytes += skb->len;
	dev->stats.tx_packets++;
L
Linus Torvalds 已提交
213 214 215 216 217 218
	ipmr_cache_report(skb, reg_vif_num, IGMPMSG_WHOLEPKT);
	read_unlock(&mrt_lock);
	kfree_skb(skb);
	return 0;
}

219 220 221 222
static const struct net_device_ops reg_vif_netdev_ops = {
	.ndo_start_xmit	= reg_vif_xmit,
};

L
Linus Torvalds 已提交
223 224 225
static void reg_vif_setup(struct net_device *dev)
{
	dev->type		= ARPHRD_PIMREG;
226
	dev->mtu		= ETH_DATA_LEN - sizeof(struct iphdr) - 8;
L
Linus Torvalds 已提交
227
	dev->flags		= IFF_NOARP;
228
	dev->netdev_ops		= &reg_vif_netdev_ops,
L
Linus Torvalds 已提交
229 230 231 232 233 234 235 236
	dev->destructor		= free_netdev;
}

static struct net_device *ipmr_reg_vif(void)
{
	struct net_device *dev;
	struct in_device *in_dev;

237
	dev = alloc_netdev(0, "pimreg", reg_vif_setup);
L
Linus Torvalds 已提交
238 239 240 241 242 243 244 245 246 247

	if (dev == NULL)
		return NULL;

	if (register_netdevice(dev)) {
		free_netdev(dev);
		return NULL;
	}
	dev->iflink = 0;

248 249 250
	rcu_read_lock();
	if ((in_dev = __in_dev_get_rcu(dev)) == NULL) {
		rcu_read_unlock();
L
Linus Torvalds 已提交
251
		goto failure;
252
	}
L
Linus Torvalds 已提交
253

254 255 256
	ipv4_devconf_setall(in_dev);
	IPV4_DEVCONF(in_dev->cnf, RP_FILTER) = 0;
	rcu_read_unlock();
L
Linus Torvalds 已提交
257 258 259 260

	if (dev_open(dev))
		goto failure;

261 262
	dev_hold(dev);

L
Linus Torvalds 已提交
263 264 265 266 267 268 269 270 271 272 273 274 275 276
	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
277
 *	@notify: Set to 1, if the caller is a notifier_call
L
Linus Torvalds 已提交
278
 */
279

280
static int vif_delete(int vifi, int notify)
L
Linus Torvalds 已提交
281 282 283 284 285
{
	struct vif_device *v;
	struct net_device *dev;
	struct in_device *in_dev;

286
	if (vifi < 0 || vifi >= init_net.ipv4.maxvif)
L
Linus Torvalds 已提交
287 288
		return -EADDRNOTAVAIL;

289
	v = &init_net.ipv4.vif_table[vifi];
L
Linus Torvalds 已提交
290 291 292 293 294 295 296 297 298 299 300 301 302 303 304

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

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

#ifdef CONFIG_IP_PIMSM
	if (vifi == reg_vif_num)
		reg_vif_num = -1;
#endif

305
	if (vifi+1 == init_net.ipv4.maxvif) {
L
Linus Torvalds 已提交
306 307
		int tmp;
		for (tmp=vifi-1; tmp>=0; tmp--) {
308
			if (VIF_EXISTS(&init_net, tmp))
L
Linus Torvalds 已提交
309 310
				break;
		}
311
		init_net.ipv4.maxvif = tmp+1;
L
Linus Torvalds 已提交
312 313 314 315 316 317
	}

	write_unlock_bh(&mrt_lock);

	dev_set_allmulti(dev, -1);

318
	if ((in_dev = __in_dev_get_rtnl(dev)) != NULL) {
319
		IPV4_DEVCONF(in_dev->cnf, MC_FORWARDING)--;
L
Linus Torvalds 已提交
320 321 322
		ip_rt_multicast_event(in_dev);
	}

323
	if (v->flags&(VIFF_TUNNEL|VIFF_REGISTER) && !notify)
L
Linus Torvalds 已提交
324 325 326 327 328 329
		unregister_netdevice(dev);

	dev_put(dev);
	return 0;
}

330 331 332 333 334 335
static inline void ipmr_cache_free(struct mfc_cache *c)
{
	release_net(mfc_net(c));
	kmem_cache_free(mrt_cachep, c);
}

L
Linus Torvalds 已提交
336 337 338 339 340 341 342
/* Destroy an unresolved cache entry, killing queued skbs
   and reporting error to netlink readers.
 */

static void ipmr_destroy_unres(struct mfc_cache *c)
{
	struct sk_buff *skb;
343
	struct nlmsgerr *e;
L
Linus Torvalds 已提交
344 345 346

	atomic_dec(&cache_resolve_queue_len);

J
Jianjun Kong 已提交
347
	while ((skb = skb_dequeue(&c->mfc_un.unres.unresolved))) {
348
		if (ip_hdr(skb)->version == 0) {
L
Linus Torvalds 已提交
349 350 351 352
			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);
353 354 355
			e = NLMSG_DATA(nlh);
			e->error = -ETIMEDOUT;
			memset(&e->msg, 0, sizeof(e->msg));
356

357
			rtnl_unicast(skb, &init_net, NETLINK_CB(skb).pid);
L
Linus Torvalds 已提交
358 359 360 361
		} else
			kfree_skb(skb);
	}

362
	ipmr_cache_free(c);
L
Linus Torvalds 已提交
363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408
}


/* Single timer process for all the unresolved queue. */

static void ipmr_expire_process(unsigned long dummy)
{
	unsigned long now;
	unsigned long expires;
	struct mfc_cache *c, **cp;

	if (!spin_trylock(&mfc_unres_lock)) {
		mod_timer(&ipmr_expire_timer, jiffies+HZ/10);
		return;
	}

	if (atomic_read(&cache_resolve_queue_len) == 0)
		goto out;

	now = jiffies;
	expires = 10*HZ;
	cp = &mfc_unres_queue;

	while ((c=*cp) != NULL) {
		if (time_after(c->mfc_un.unres.expires, now)) {
			unsigned long interval = c->mfc_un.unres.expires - now;
			if (interval < expires)
				expires = interval;
			cp = &c->next;
			continue;
		}

		*cp = c->next;

		ipmr_destroy_unres(c);
	}

	if (atomic_read(&cache_resolve_queue_len))
		mod_timer(&ipmr_expire_timer, jiffies + expires);

out:
	spin_unlock(&mfc_unres_lock);
}

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

409
static void ipmr_update_thresholds(struct mfc_cache *cache, unsigned char *ttls)
L
Linus Torvalds 已提交
410 411 412 413 414 415 416
{
	int vifi;

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

417 418 419
	for (vifi = 0; vifi < init_net.ipv4.maxvif; vifi++) {
		if (VIF_EXISTS(&init_net, vifi) &&
		    ttls[vifi] && ttls[vifi] < 255) {
L
Linus Torvalds 已提交
420 421 422 423 424 425 426 427 428 429 430 431
			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;
		}
	}
}

static int vif_add(struct vifctl *vifc, int mrtsock)
{
	int vifi = vifc->vifc_vifi;
432
	struct vif_device *v = &init_net.ipv4.vif_table[vifi];
L
Linus Torvalds 已提交
433 434
	struct net_device *dev;
	struct in_device *in_dev;
435
	int err;
L
Linus Torvalds 已提交
436 437

	/* Is vif busy ? */
438
	if (VIF_EXISTS(&init_net, vifi))
L
Linus Torvalds 已提交
439 440 441 442 443 444 445 446 447 448 449 450 451 452
		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
		 */
		if (reg_vif_num >= 0)
			return -EADDRINUSE;
		dev = ipmr_reg_vif();
		if (!dev)
			return -ENOBUFS;
453 454 455
		err = dev_set_allmulti(dev, 1);
		if (err) {
			unregister_netdevice(dev);
456
			dev_put(dev);
457 458
			return err;
		}
L
Linus Torvalds 已提交
459 460
		break;
#endif
461
	case VIFF_TUNNEL:
L
Linus Torvalds 已提交
462 463 464
		dev = ipmr_new_tunnel(vifc);
		if (!dev)
			return -ENOBUFS;
465 466 467
		err = dev_set_allmulti(dev, 1);
		if (err) {
			ipmr_del_tunnel(dev, vifc);
468
			dev_put(dev);
469 470
			return err;
		}
L
Linus Torvalds 已提交
471 472
		break;
	case 0:
473
		dev = ip_dev_find(&init_net, vifc->vifc_lcl_addr.s_addr);
L
Linus Torvalds 已提交
474 475
		if (!dev)
			return -EADDRNOTAVAIL;
476
		err = dev_set_allmulti(dev, 1);
477 478
		if (err) {
			dev_put(dev);
479
			return err;
480
		}
L
Linus Torvalds 已提交
481 482 483 484 485
		break;
	default:
		return -EINVAL;
	}

486
	if ((in_dev = __in_dev_get_rtnl(dev)) == NULL)
L
Linus Torvalds 已提交
487
		return -EADDRNOTAVAIL;
488
	IPV4_DEVCONF(in_dev->cnf, MC_FORWARDING)++;
L
Linus Torvalds 已提交
489 490 491 492 493
	ip_rt_multicast_event(in_dev);

	/*
	 *	Fill in the VIF structures
	 */
J
Jianjun Kong 已提交
494 495 496 497
	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 已提交
498 499
	if (!mrtsock)
		v->flags |= VIFF_STATIC;
J
Jianjun Kong 已提交
500
	v->threshold = vifc->vifc_threshold;
L
Linus Torvalds 已提交
501 502 503 504 505 506 507 508 509 510
	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 已提交
511
	v->dev = dev;
L
Linus Torvalds 已提交
512 513 514 515
#ifdef CONFIG_IP_PIMSM
	if (v->flags&VIFF_REGISTER)
		reg_vif_num = vifi;
#endif
516 517
	if (vifi+1 > init_net.ipv4.maxvif)
		init_net.ipv4.maxvif = vifi+1;
L
Linus Torvalds 已提交
518 519 520 521
	write_unlock_bh(&mrt_lock);
	return 0;
}

A
Al Viro 已提交
522
static struct mfc_cache *ipmr_cache_find(__be32 origin, __be32 mcastgrp)
L
Linus Torvalds 已提交
523
{
J
Jianjun Kong 已提交
524
	int line = MFC_HASH(mcastgrp, origin);
L
Linus Torvalds 已提交
525 526 527 528 529 530 531 532 533 534 535 536
	struct mfc_cache *c;

	for (c=mfc_cache_array[line]; c; c = c->next) {
		if (c->mfc_origin==origin && c->mfc_mcastgrp==mcastgrp)
			break;
	}
	return c;
}

/*
 *	Allocate a multicast cache entry
 */
537
static struct mfc_cache *ipmr_cache_alloc(struct net *net)
L
Linus Torvalds 已提交
538
{
J
Jianjun Kong 已提交
539 540
	struct mfc_cache *c = kmem_cache_zalloc(mrt_cachep, GFP_KERNEL);
	if (c == NULL)
L
Linus Torvalds 已提交
541 542
		return NULL;
	c->mfc_un.res.minvif = MAXVIFS;
543
	mfc_net_set(c, net);
L
Linus Torvalds 已提交
544 545 546
	return c;
}

547
static struct mfc_cache *ipmr_cache_alloc_unres(struct net *net)
L
Linus Torvalds 已提交
548
{
J
Jianjun Kong 已提交
549 550
	struct mfc_cache *c = kmem_cache_zalloc(mrt_cachep, GFP_ATOMIC);
	if (c == NULL)
L
Linus Torvalds 已提交
551 552 553
		return NULL;
	skb_queue_head_init(&c->mfc_un.unres.unresolved);
	c->mfc_un.unres.expires = jiffies + 10*HZ;
554
	mfc_net_set(c, net);
L
Linus Torvalds 已提交
555 556 557 558 559 560
	return c;
}

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

L
Linus Torvalds 已提交
562 563 564
static void ipmr_cache_resolve(struct mfc_cache *uc, struct mfc_cache *c)
{
	struct sk_buff *skb;
565
	struct nlmsgerr *e;
L
Linus Torvalds 已提交
566 567 568 569 570

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

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

			if (ipmr_fill_mroute(skb, c, NLMSG_DATA(nlh)) > 0) {
576 577
				nlh->nlmsg_len = (skb_tail_pointer(skb) -
						  (u8 *)nlh);
L
Linus Torvalds 已提交
578 579 580 581
			} else {
				nlh->nlmsg_type = NLMSG_ERROR;
				nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct nlmsgerr));
				skb_trim(skb, nlh->nlmsg_len);
582 583 584
				e = NLMSG_DATA(nlh);
				e->error = -EMSGSIZE;
				memset(&e->msg, 0, sizeof(e->msg));
L
Linus Torvalds 已提交
585
			}
586

587
			rtnl_unicast(skb, &init_net, NETLINK_CB(skb).pid);
L
Linus Torvalds 已提交
588 589 590 591 592 593 594 595 596 597 598
		} else
			ip_mr_forward(skb, c, 0);
	}
}

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

L
Linus Torvalds 已提交
600 601 602
static int ipmr_cache_report(struct sk_buff *pkt, vifi_t vifi, int assert)
{
	struct sk_buff *skb;
603
	const int ihl = ip_hdrlen(pkt);
L
Linus Torvalds 已提交
604 605 606 607 608 609 610 611 612 613 614
	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 已提交
615
	if (!skb)
L
Linus Torvalds 已提交
616 617 618 619 620 621 622 623 624
		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" :-)
		 */
625 626
		skb_push(skb, sizeof(struct iphdr));
		skb_reset_network_header(skb);
627
		skb_reset_transport_header(skb);
628
		msg = (struct igmpmsg *)skb_network_header(skb);
629
		memcpy(msg, skb_network_header(pkt), sizeof(struct iphdr));
L
Linus Torvalds 已提交
630 631
		msg->im_msgtype = IGMPMSG_WHOLEPKT;
		msg->im_mbz = 0;
632
		msg->im_vif = reg_vif_num;
633 634 635
		ip_hdr(skb)->ihl = sizeof(struct iphdr) >> 2;
		ip_hdr(skb)->tot_len = htons(ntohs(ip_hdr(pkt)->tot_len) +
					     sizeof(struct iphdr));
636
	} else
L
Linus Torvalds 已提交
637
#endif
638 639
	{

L
Linus Torvalds 已提交
640 641 642 643
	/*
	 *	Copy the IP header
	 */

644
	skb->network_header = skb->tail;
645
	skb_put(skb, ihl);
646
	skb_copy_to_linear_data(skb, pkt->data, ihl);
647 648
	ip_hdr(skb)->protocol = 0;			/* Flag to the kernel this is a route add */
	msg = (struct igmpmsg *)skb_network_header(skb);
L
Linus Torvalds 已提交
649 650 651 652 653 654 655
	msg->im_vif = vifi;
	skb->dst = dst_clone(pkt->dst);

	/*
	 *	Add our header
	 */

J
Jianjun Kong 已提交
656
	igmp=(struct igmphdr *)skb_put(skb, sizeof(struct igmphdr));
L
Linus Torvalds 已提交
657 658 659
	igmp->type	=
	msg->im_msgtype = assert;
	igmp->code 	=	0;
660
	ip_hdr(skb)->tot_len = htons(skb->len);			/* Fix the length */
661
	skb->transport_header = skb->network_header;
662
	}
L
Linus Torvalds 已提交
663

664
	if (init_net.ipv4.mroute_sk == NULL) {
L
Linus Torvalds 已提交
665 666 667 668 669 670 671
		kfree_skb(skb);
		return -EINVAL;
	}

	/*
	 *	Deliver to mrouted
	 */
672 673
	ret = sock_queue_rcv_skb(init_net.ipv4.mroute_sk, skb);
	if (ret < 0) {
L
Linus Torvalds 已提交
674 675 676 677 678 679 680 681 682 683 684
		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!
 */
685

L
Linus Torvalds 已提交
686 687 688 689 690
static int
ipmr_cache_unresolved(vifi_t vifi, struct sk_buff *skb)
{
	int err;
	struct mfc_cache *c;
691
	const struct iphdr *iph = ip_hdr(skb);
L
Linus Torvalds 已提交
692 693 694

	spin_lock_bh(&mfc_unres_lock);
	for (c=mfc_unres_queue; c; c=c->next) {
695 696
		if (c->mfc_mcastgrp == iph->daddr &&
		    c->mfc_origin == iph->saddr)
L
Linus Torvalds 已提交
697 698 699 700 701 702 703 704
			break;
	}

	if (c == NULL) {
		/*
		 *	Create a new entry if allowable
		 */

J
Jianjun Kong 已提交
705
		if (atomic_read(&cache_resolve_queue_len) >= 10 ||
706
		    (c = ipmr_cache_alloc_unres(&init_net)) == NULL) {
L
Linus Torvalds 已提交
707 708 709 710 711 712 713 714 715
			spin_unlock_bh(&mfc_unres_lock);

			kfree_skb(skb);
			return -ENOBUFS;
		}

		/*
		 *	Fill in the new cache entry
		 */
716 717 718
		c->mfc_parent	= -1;
		c->mfc_origin	= iph->saddr;
		c->mfc_mcastgrp	= iph->daddr;
L
Linus Torvalds 已提交
719 720 721 722 723

		/*
		 *	Reflect first query at mrouted.
		 */
		if ((err = ipmr_cache_report(skb, vifi, IGMPMSG_NOCACHE))<0) {
724
			/* If the report failed throw the cache entry
L
Linus Torvalds 已提交
725 726 727 728
			   out - Brad Parker
			 */
			spin_unlock_bh(&mfc_unres_lock);

729
			ipmr_cache_free(c);
L
Linus Torvalds 已提交
730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747
			kfree_skb(skb);
			return err;
		}

		atomic_inc(&cache_resolve_queue_len);
		c->next = mfc_unres_queue;
		mfc_unres_queue = c;

		mod_timer(&ipmr_expire_timer, c->mfc_un.unres.expires);
	}

	/*
	 *	See if we can append the packet
	 */
	if (c->mfc_un.unres.unresolved.qlen>3) {
		kfree_skb(skb);
		err = -ENOBUFS;
	} else {
J
Jianjun Kong 已提交
748
		skb_queue_tail(&c->mfc_un.unres.unresolved, skb);
L
Linus Torvalds 已提交
749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764
		err = 0;
	}

	spin_unlock_bh(&mfc_unres_lock);
	return err;
}

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

static int ipmr_mfc_delete(struct mfcctl *mfc)
{
	int line;
	struct mfc_cache *c, **cp;

J
Jianjun Kong 已提交
765
	line = MFC_HASH(mfc->mfcc_mcastgrp.s_addr, mfc->mfcc_origin.s_addr);
L
Linus Torvalds 已提交
766 767 768 769 770 771 772 773

	for (cp=&mfc_cache_array[line]; (c=*cp) != NULL; cp = &c->next) {
		if (c->mfc_origin == mfc->mfcc_origin.s_addr &&
		    c->mfc_mcastgrp == mfc->mfcc_mcastgrp.s_addr) {
			write_lock_bh(&mrt_lock);
			*cp = c->next;
			write_unlock_bh(&mrt_lock);

774
			ipmr_cache_free(c);
L
Linus Torvalds 已提交
775 776 777 778 779 780 781 782 783 784 785
			return 0;
		}
	}
	return -ENOENT;
}

static int ipmr_mfc_add(struct mfcctl *mfc, int mrtsock)
{
	int line;
	struct mfc_cache *uc, *c, **cp;

J
Jianjun Kong 已提交
786
	line = MFC_HASH(mfc->mfcc_mcastgrp.s_addr, mfc->mfcc_origin.s_addr);
L
Linus Torvalds 已提交
787 788 789 790 791 792 793 794 795 796

	for (cp=&mfc_cache_array[line]; (c=*cp) != NULL; cp = &c->next) {
		if (c->mfc_origin == mfc->mfcc_origin.s_addr &&
		    c->mfc_mcastgrp == mfc->mfcc_mcastgrp.s_addr)
			break;
	}

	if (c != NULL) {
		write_lock_bh(&mrt_lock);
		c->mfc_parent = mfc->mfcc_parent;
797
		ipmr_update_thresholds(c, mfc->mfcc_ttls);
L
Linus Torvalds 已提交
798 799 800 801 802 803
		if (!mrtsock)
			c->mfc_flags |= MFC_STATIC;
		write_unlock_bh(&mrt_lock);
		return 0;
	}

804
	if (!ipv4_is_multicast(mfc->mfcc_mcastgrp.s_addr))
L
Linus Torvalds 已提交
805 806
		return -EINVAL;

807
	c = ipmr_cache_alloc(&init_net);
J
Jianjun Kong 已提交
808
	if (c == NULL)
L
Linus Torvalds 已提交
809 810
		return -ENOMEM;

J
Jianjun Kong 已提交
811 812 813
	c->mfc_origin = mfc->mfcc_origin.s_addr;
	c->mfc_mcastgrp = mfc->mfcc_mcastgrp.s_addr;
	c->mfc_parent = mfc->mfcc_parent;
814
	ipmr_update_thresholds(c, mfc->mfcc_ttls);
L
Linus Torvalds 已提交
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
	if (!mrtsock)
		c->mfc_flags |= MFC_STATIC;

	write_lock_bh(&mrt_lock);
	c->next = mfc_cache_array[line];
	mfc_cache_array[line] = c;
	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.
	 */
	spin_lock_bh(&mfc_unres_lock);
	for (cp = &mfc_unres_queue; (uc=*cp) != NULL;
	     cp = &uc->next) {
		if (uc->mfc_origin == c->mfc_origin &&
		    uc->mfc_mcastgrp == c->mfc_mcastgrp) {
			*cp = uc->next;
			if (atomic_dec_and_test(&cache_resolve_queue_len))
				del_timer(&ipmr_expire_timer);
			break;
		}
	}
	spin_unlock_bh(&mfc_unres_lock);

	if (uc) {
		ipmr_cache_resolve(uc, c);
842
		ipmr_cache_free(uc);
L
Linus Torvalds 已提交
843 844 845 846 847 848 849
	}
	return 0;
}

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

L
Linus Torvalds 已提交
851 852 853
static void mroute_clean_tables(struct sock *sk)
{
	int i;
854

L
Linus Torvalds 已提交
855 856 857
	/*
	 *	Shut down all active vif entries
	 */
858 859
	for (i = 0; i < init_net.ipv4.maxvif; i++) {
		if (!(init_net.ipv4.vif_table[i].flags&VIFF_STATIC))
860
			vif_delete(i, 0);
L
Linus Torvalds 已提交
861 862 863 864 865
	}

	/*
	 *	Wipe the cache
	 */
J
Jianjun Kong 已提交
866
	for (i=0; i<MFC_LINES; i++) {
L
Linus Torvalds 已提交
867 868 869 870 871 872 873 874 875 876 877 878
		struct mfc_cache *c, **cp;

		cp = &mfc_cache_array[i];
		while ((c = *cp) != NULL) {
			if (c->mfc_flags&MFC_STATIC) {
				cp = &c->next;
				continue;
			}
			write_lock_bh(&mrt_lock);
			*cp = c->next;
			write_unlock_bh(&mrt_lock);

879
			ipmr_cache_free(c);
L
Linus Torvalds 已提交
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902
		}
	}

	if (atomic_read(&cache_resolve_queue_len) != 0) {
		struct mfc_cache *c;

		spin_lock_bh(&mfc_unres_lock);
		while (mfc_unres_queue != NULL) {
			c = mfc_unres_queue;
			mfc_unres_queue = c->next;
			spin_unlock_bh(&mfc_unres_lock);

			ipmr_destroy_unres(c);

			spin_lock_bh(&mfc_unres_lock);
		}
		spin_unlock_bh(&mfc_unres_lock);
	}
}

static void mrtsock_destruct(struct sock *sk)
{
	rtnl_lock();
903
	if (sk == init_net.ipv4.mroute_sk) {
904
		IPV4_DEVCONF_ALL(sock_net(sk), MC_FORWARDING)--;
L
Linus Torvalds 已提交
905 906

		write_lock_bh(&mrt_lock);
907
		init_net.ipv4.mroute_sk = NULL;
L
Linus Torvalds 已提交
908 909 910 911 912 913 914 915 916 917 918 919 920
		write_unlock_bh(&mrt_lock);

		mroute_clean_tables(sk);
	}
	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.
 */
921

J
Jianjun Kong 已提交
922
int ip_mroute_setsockopt(struct sock *sk, int optname, char __user *optval, int optlen)
L
Linus Torvalds 已提交
923 924 925 926
{
	int ret;
	struct vifctl vif;
	struct mfcctl mfc;
927

S
Stephen Hemminger 已提交
928
	if (optname != MRT_INIT) {
929
		if (sk != init_net.ipv4.mroute_sk && !capable(CAP_NET_ADMIN))
L
Linus Torvalds 已提交
930 931 932
			return -EACCES;
	}

S
Stephen Hemminger 已提交
933 934 935 936 937
	switch (optname) {
	case MRT_INIT:
		if (sk->sk_type != SOCK_RAW ||
		    inet_sk(sk)->num != IPPROTO_IGMP)
			return -EOPNOTSUPP;
J
Jianjun Kong 已提交
938
		if (optlen != sizeof(int))
S
Stephen Hemminger 已提交
939
			return -ENOPROTOOPT;
L
Linus Torvalds 已提交
940

S
Stephen Hemminger 已提交
941
		rtnl_lock();
942
		if (init_net.ipv4.mroute_sk) {
L
Linus Torvalds 已提交
943
			rtnl_unlock();
S
Stephen Hemminger 已提交
944 945 946 947 948 949
			return -EADDRINUSE;
		}

		ret = ip_ra_control(sk, 1, mrtsock_destruct);
		if (ret == 0) {
			write_lock_bh(&mrt_lock);
950
			init_net.ipv4.mroute_sk = sk;
S
Stephen Hemminger 已提交
951 952
			write_unlock_bh(&mrt_lock);

953
			IPV4_DEVCONF_ALL(sock_net(sk), MC_FORWARDING)++;
S
Stephen Hemminger 已提交
954 955 956 957
		}
		rtnl_unlock();
		return ret;
	case MRT_DONE:
958
		if (sk != init_net.ipv4.mroute_sk)
S
Stephen Hemminger 已提交
959 960 961 962
			return -EACCES;
		return ip_ra_control(sk, 0, NULL);
	case MRT_ADD_VIF:
	case MRT_DEL_VIF:
J
Jianjun Kong 已提交
963
		if (optlen != sizeof(vif))
S
Stephen Hemminger 已提交
964
			return -EINVAL;
J
Jianjun Kong 已提交
965
		if (copy_from_user(&vif, optval, sizeof(vif)))
S
Stephen Hemminger 已提交
966 967 968 969
			return -EFAULT;
		if (vif.vifc_vifi >= MAXVIFS)
			return -ENFILE;
		rtnl_lock();
J
Jianjun Kong 已提交
970
		if (optname == MRT_ADD_VIF) {
971
			ret = vif_add(&vif, sk == init_net.ipv4.mroute_sk);
S
Stephen Hemminger 已提交
972
		} else {
973
			ret = vif_delete(vif.vifc_vifi, 0);
S
Stephen Hemminger 已提交
974 975 976
		}
		rtnl_unlock();
		return ret;
L
Linus Torvalds 已提交
977 978 979 980 981

		/*
		 *	Manipulate the forwarding caches. These live
		 *	in a sort of kernel/user symbiosis.
		 */
S
Stephen Hemminger 已提交
982 983
	case MRT_ADD_MFC:
	case MRT_DEL_MFC:
J
Jianjun Kong 已提交
984
		if (optlen != sizeof(mfc))
S
Stephen Hemminger 已提交
985
			return -EINVAL;
J
Jianjun Kong 已提交
986
		if (copy_from_user(&mfc, optval, sizeof(mfc)))
S
Stephen Hemminger 已提交
987 988
			return -EFAULT;
		rtnl_lock();
J
Jianjun Kong 已提交
989
		if (optname == MRT_DEL_MFC)
S
Stephen Hemminger 已提交
990 991
			ret = ipmr_mfc_delete(&mfc);
		else
992
			ret = ipmr_mfc_add(&mfc, sk == init_net.ipv4.mroute_sk);
S
Stephen Hemminger 已提交
993 994
		rtnl_unlock();
		return ret;
L
Linus Torvalds 已提交
995 996 997
		/*
		 *	Control PIM assert.
		 */
S
Stephen Hemminger 已提交
998 999 1000 1001 1002 1003 1004 1005
	case MRT_ASSERT:
	{
		int v;
		if (get_user(v,(int __user *)optval))
			return -EFAULT;
		mroute_do_assert=(v)?1:0;
		return 0;
	}
L
Linus Torvalds 已提交
1006
#ifdef CONFIG_IP_PIMSM
S
Stephen Hemminger 已提交
1007 1008
	case MRT_PIM:
	{
S
Stephen Hemminger 已提交
1009 1010
		int v;

S
Stephen Hemminger 已提交
1011 1012
		if (get_user(v,(int __user *)optval))
			return -EFAULT;
S
Stephen Hemminger 已提交
1013 1014
		v = (v) ? 1 : 0;

S
Stephen Hemminger 已提交
1015 1016 1017 1018 1019
		rtnl_lock();
		ret = 0;
		if (v != mroute_do_pim) {
			mroute_do_pim = v;
			mroute_do_assert = v;
L
Linus Torvalds 已提交
1020
#ifdef CONFIG_IP_PIMSM_V2
S
Stephen Hemminger 已提交
1021 1022 1023 1024 1025 1026 1027 1028
			if (mroute_do_pim)
				ret = inet_add_protocol(&pim_protocol,
							IPPROTO_PIM);
			else
				ret = inet_del_protocol(&pim_protocol,
							IPPROTO_PIM);
			if (ret < 0)
				ret = -EAGAIN;
L
Linus Torvalds 已提交
1029 1030
#endif
		}
S
Stephen Hemminger 已提交
1031 1032 1033
		rtnl_unlock();
		return ret;
	}
L
Linus Torvalds 已提交
1034
#endif
S
Stephen Hemminger 已提交
1035 1036 1037 1038 1039 1040
	/*
	 *	Spurious command, or MRT_VERSION which you cannot
	 *	set.
	 */
	default:
		return -ENOPROTOOPT;
L
Linus Torvalds 已提交
1041 1042 1043 1044 1045 1046
	}
}

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

J
Jianjun Kong 已提交
1048
int ip_mroute_getsockopt(struct sock *sk, int optname, char __user *optval, int __user *optlen)
L
Linus Torvalds 已提交
1049 1050 1051 1052
{
	int olr;
	int val;

J
Jianjun Kong 已提交
1053
	if (optname != MRT_VERSION &&
L
Linus Torvalds 已提交
1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065
#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;
1066

J
Jianjun Kong 已提交
1067
	if (put_user(olr, optlen))
L
Linus Torvalds 已提交
1068
		return -EFAULT;
J
Jianjun Kong 已提交
1069 1070
	if (optname == MRT_VERSION)
		val = 0x0305;
L
Linus Torvalds 已提交
1071
#ifdef CONFIG_IP_PIMSM
J
Jianjun Kong 已提交
1072 1073
	else if (optname == MRT_PIM)
		val = mroute_do_pim;
L
Linus Torvalds 已提交
1074 1075
#endif
	else
J
Jianjun Kong 已提交
1076 1077
		val = mroute_do_assert;
	if (copy_to_user(optval, &val, olr))
L
Linus Torvalds 已提交
1078 1079 1080 1081 1082 1083 1084
		return -EFAULT;
	return 0;
}

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

L
Linus Torvalds 已提交
1086 1087 1088 1089 1090 1091
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;
1092

S
Stephen Hemminger 已提交
1093 1094
	switch (cmd) {
	case SIOCGETVIFCNT:
J
Jianjun Kong 已提交
1095
		if (copy_from_user(&vr, arg, sizeof(vr)))
S
Stephen Hemminger 已提交
1096
			return -EFAULT;
1097
		if (vr.vifi >= init_net.ipv4.maxvif)
S
Stephen Hemminger 已提交
1098 1099
			return -EINVAL;
		read_lock(&mrt_lock);
1100 1101
		vif = &init_net.ipv4.vif_table[vr.vifi];
		if (VIF_EXISTS(&init_net, vr.vifi)) {
J
Jianjun Kong 已提交
1102 1103 1104 1105
			vr.icount = vif->pkt_in;
			vr.ocount = vif->pkt_out;
			vr.ibytes = vif->bytes_in;
			vr.obytes = vif->bytes_out;
L
Linus Torvalds 已提交
1106 1107
			read_unlock(&mrt_lock);

J
Jianjun Kong 已提交
1108
			if (copy_to_user(arg, &vr, sizeof(vr)))
S
Stephen Hemminger 已提交
1109 1110 1111 1112 1113 1114
				return -EFAULT;
			return 0;
		}
		read_unlock(&mrt_lock);
		return -EADDRNOTAVAIL;
	case SIOCGETSGCNT:
J
Jianjun Kong 已提交
1115
		if (copy_from_user(&sr, arg, sizeof(sr)))
S
Stephen Hemminger 已提交
1116 1117 1118 1119 1120 1121 1122 1123
			return -EFAULT;

		read_lock(&mrt_lock);
		c = ipmr_cache_find(sr.src.s_addr, sr.grp.s_addr);
		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 已提交
1124
			read_unlock(&mrt_lock);
S
Stephen Hemminger 已提交
1125

J
Jianjun Kong 已提交
1126
			if (copy_to_user(arg, &sr, sizeof(sr)))
S
Stephen Hemminger 已提交
1127 1128 1129 1130 1131 1132 1133
				return -EFAULT;
			return 0;
		}
		read_unlock(&mrt_lock);
		return -EADDRNOTAVAIL;
	default:
		return -ENOIOCTLCMD;
L
Linus Torvalds 已提交
1134 1135 1136 1137 1138 1139
	}
}


static int ipmr_device_event(struct notifier_block *this, unsigned long event, void *ptr)
{
1140
	struct net_device *dev = ptr;
L
Linus Torvalds 已提交
1141 1142
	struct vif_device *v;
	int ct;
1143

1144
	if (!net_eq(dev_net(dev), &init_net))
1145 1146
		return NOTIFY_DONE;

L
Linus Torvalds 已提交
1147 1148
	if (event != NETDEV_UNREGISTER)
		return NOTIFY_DONE;
1149 1150
	v = &init_net.ipv4.vif_table[0];
	for (ct = 0; ct < init_net.ipv4.maxvif; ct++, v++) {
J
Jianjun Kong 已提交
1151
		if (v->dev == dev)
1152
			vif_delete(ct, 1);
L
Linus Torvalds 已提交
1153 1154 1155 1156 1157
	}
	return NOTIFY_DONE;
}


J
Jianjun Kong 已提交
1158
static struct notifier_block ip_mr_notifier = {
L
Linus Torvalds 已提交
1159 1160 1161 1162 1163 1164 1165 1166
	.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.
 */
1167

A
Al Viro 已提交
1168
static void ip_encap(struct sk_buff *skb, __be32 saddr, __be32 daddr)
L
Linus Torvalds 已提交
1169
{
1170
	struct iphdr *iph;
1171
	struct iphdr *old_iph = ip_hdr(skb);
1172 1173

	skb_push(skb, sizeof(struct iphdr));
1174
	skb->transport_header = skb->network_header;
1175
	skb_reset_network_header(skb);
1176
	iph = ip_hdr(skb);
L
Linus Torvalds 已提交
1177 1178

	iph->version	= 	4;
1179 1180
	iph->tos	=	old_iph->tos;
	iph->ttl	=	old_iph->ttl;
L
Linus Torvalds 已提交
1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197
	iph->frag_off	=	0;
	iph->daddr	=	daddr;
	iph->saddr	=	saddr;
	iph->protocol	=	IPPROTO_IPIP;
	iph->ihl	=	5;
	iph->tot_len	=	htons(skb->len);
	ip_select_ident(iph, skb->dst, NULL);
	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);

1198
	IP_INC_STATS_BH(dev_net(skb->dst->dev), IPSTATS_MIB_OUTFORWDATAGRAMS);
L
Linus Torvalds 已提交
1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211

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

	return dst_output(skb);
}

/*
 *	Processing handlers for ipmr_forward
 */

static void ipmr_queue_xmit(struct sk_buff *skb, struct mfc_cache *c, int vifi)
{
1212
	const struct iphdr *iph = ip_hdr(skb);
1213
	struct vif_device *vif = &init_net.ipv4.vif_table[vifi];
L
Linus Torvalds 已提交
1214 1215 1216 1217 1218 1219 1220 1221 1222 1223
	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 已提交
1224
		vif->bytes_out += skb->len;
1225 1226
		vif->dev->stats.tx_bytes += skb->len;
		vif->dev->stats.tx_packets++;
L
Linus Torvalds 已提交
1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
		ipmr_cache_report(skb, vifi, IGMPMSG_WHOLEPKT);
		kfree_skb(skb);
		return;
	}
#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 };
1240
		if (ip_route_output_key(&init_net, &rt, &fl))
L
Linus Torvalds 已提交
1241 1242 1243 1244 1245 1246 1247 1248
			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 };
1249
		if (ip_route_output_key(&init_net, &rt, &fl))
L
Linus Torvalds 已提交
1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
			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.
		 */

1261
		IP_INC_STATS_BH(dev_net(dev), IPSTATS_MIB_FRAGFAILS);
L
Linus Torvalds 已提交
1262 1263 1264 1265 1266 1267 1268
		ip_rt_put(rt);
		goto out_free;
	}

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

	if (skb_cow(skb, encap)) {
1269
		ip_rt_put(rt);
L
Linus Torvalds 已提交
1270 1271 1272 1273
		goto out_free;
	}

	vif->pkt_out++;
J
Jianjun Kong 已提交
1274
	vif->bytes_out += skb->len;
L
Linus Torvalds 已提交
1275 1276 1277

	dst_release(skb->dst);
	skb->dst = &rt->u.dst;
1278
	ip_decrease_ttl(ip_hdr(skb));
L
Linus Torvalds 已提交
1279 1280 1281 1282 1283 1284

	/* 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 */
1285 1286
		vif->dev->stats.tx_packets++;
		vif->dev->stats.tx_bytes += skb->len;
L
Linus Torvalds 已提交
1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301
	}

	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.
	 */
1302
	NF_HOOK(PF_INET, NF_INET_FORWARD, skb, skb->dev, dev,
L
Linus Torvalds 已提交
1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313
		ipmr_forward_finish);
	return;

out_free:
	kfree_skb(skb);
	return;
}

static int ipmr_find_vif(struct net_device *dev)
{
	int ct;
1314 1315
	for (ct = init_net.ipv4.maxvif-1; ct >= 0; ct--) {
		if (init_net.ipv4.vif_table[ct].dev == dev)
L
Linus Torvalds 已提交
1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334
			break;
	}
	return ct;
}

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

static int ip_mr_forward(struct sk_buff *skb, struct mfc_cache *cache, int local)
{
	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.
	 */
1335
	if (init_net.ipv4.vif_table[vif].dev != skb->dev) {
L
Linus Torvalds 已提交
1336 1337
		int true_vifi;

1338
		if (skb->rtable->fl.iif == 0) {
L
Linus Torvalds 已提交
1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362
			/* 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++;
		true_vifi = ipmr_find_vif(skb->dev);

		if (true_vifi >= 0 && mroute_do_assert &&
		    /* 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
		     */
		    (mroute_do_pim || cache->mfc_un.res.ttls[true_vifi] < 255) &&
1363
		    time_after(jiffies,
L
Linus Torvalds 已提交
1364 1365 1366 1367 1368 1369 1370
			       cache->mfc_un.res.last_assert + MFC_ASSERT_THRESH)) {
			cache->mfc_un.res.last_assert = jiffies;
			ipmr_cache_report(skb, true_vifi, IGMPMSG_WRONGVIF);
		}
		goto dont_forward;
	}

1371 1372
	init_net.ipv4.vif_table[vif].pkt_in++;
	init_net.ipv4.vif_table[vif].bytes_in += skb->len;
L
Linus Torvalds 已提交
1373 1374 1375 1376 1377

	/*
	 *	Forward the frame
	 */
	for (ct = cache->mfc_un.res.maxvif-1; ct >= cache->mfc_un.res.minvif; ct--) {
1378
		if (ip_hdr(skb)->ttl > cache->mfc_un.res.ttls[ct]) {
L
Linus Torvalds 已提交
1379 1380 1381 1382 1383
			if (psend != -1) {
				struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
				if (skb2)
					ipmr_queue_xmit(skb2, cache, psend);
			}
J
Jianjun Kong 已提交
1384
			psend = ct;
L
Linus Torvalds 已提交
1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411
		}
	}
	if (psend != -1) {
		if (local) {
			struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
			if (skb2)
				ipmr_queue_xmit(skb2, cache, psend);
		} else {
			ipmr_queue_xmit(skb, cache, psend);
			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;
1412
	int local = skb->rtable->rt_flags&RTCF_LOCAL;
L
Linus Torvalds 已提交
1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423

	/* 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;

	if (!local) {
		    if (IPCB(skb)->opt.router_alert) {
			    if (ip_call_ra_chain(skb))
				    return 0;
1424
		    } else if (ip_hdr(skb)->protocol == IPPROTO_IGMP){
L
Linus Torvalds 已提交
1425 1426 1427 1428 1429 1430 1431
			    /* 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);
1432
			    if (init_net.ipv4.mroute_sk) {
1433
				    nf_reset(skb);
1434
				    raw_rcv(init_net.ipv4.mroute_sk, skb);
L
Linus Torvalds 已提交
1435 1436 1437 1438 1439 1440 1441 1442
				    read_unlock(&mrt_lock);
				    return 0;
			    }
			    read_unlock(&mrt_lock);
		    }
	}

	read_lock(&mrt_lock);
1443
	cache = ipmr_cache_find(ip_hdr(skb)->saddr, ip_hdr(skb)->daddr);
L
Linus Torvalds 已提交
1444 1445 1446 1447

	/*
	 *	No usable cache entry
	 */
J
Jianjun Kong 已提交
1448
	if (cache == NULL) {
L
Linus Torvalds 已提交
1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488
		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;
		}

		vif = ipmr_find_vif(skb->dev);
		if (vif >= 0) {
			int err = ipmr_cache_unresolved(vif, skb);
			read_unlock(&mrt_lock);

			return err;
		}
		read_unlock(&mrt_lock);
		kfree_skb(skb);
		return -ENODEV;
	}

	ip_mr_forward(skb, cache, local);

	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 已提交
1489 1490
#ifdef CONFIG_IP_PIMSM
static int __pim_rcv(struct sk_buff *skb, unsigned int pimlen)
L
Linus Torvalds 已提交
1491
{
I
Ilpo Järvinen 已提交
1492 1493
	struct net_device *reg_dev = NULL;
	struct iphdr *encap;
L
Linus Torvalds 已提交
1494

I
Ilpo Järvinen 已提交
1495
	encap = (struct iphdr *)(skb_transport_header(skb) + pimlen);
L
Linus Torvalds 已提交
1496 1497 1498 1499 1500 1501
	/*
	   Check that:
	   a. packet is really destinted to a multicast group
	   b. packet is not a NULL-REGISTER
	   c. packet is not truncated
	 */
1502
	if (!ipv4_is_multicast(encap->daddr) ||
L
Linus Torvalds 已提交
1503
	    encap->tot_len == 0 ||
I
Ilpo Järvinen 已提交
1504 1505
	    ntohs(encap->tot_len) + pimlen > skb->len)
		return 1;
L
Linus Torvalds 已提交
1506 1507 1508

	read_lock(&mrt_lock);
	if (reg_vif_num >= 0)
1509
		reg_dev = init_net.ipv4.vif_table[reg_vif_num].dev;
L
Linus Torvalds 已提交
1510 1511 1512 1513
	if (reg_dev)
		dev_hold(reg_dev);
	read_unlock(&mrt_lock);

1514
	if (reg_dev == NULL)
I
Ilpo Järvinen 已提交
1515
		return 1;
L
Linus Torvalds 已提交
1516

1517
	skb->mac_header = skb->network_header;
L
Linus Torvalds 已提交
1518
	skb_pull(skb, (u8*)encap - skb->data);
1519
	skb_reset_network_header(skb);
L
Linus Torvalds 已提交
1520 1521 1522 1523 1524 1525
	skb->dev = reg_dev;
	skb->protocol = htons(ETH_P_IP);
	skb->ip_summed = 0;
	skb->pkt_type = PACKET_HOST;
	dst_release(skb->dst);
	skb->dst = NULL;
1526 1527
	reg_dev->stats.rx_bytes += skb->len;
	reg_dev->stats.rx_packets++;
L
Linus Torvalds 已提交
1528 1529 1530
	nf_reset(skb);
	netif_rx(skb);
	dev_put(reg_dev);
I
Ilpo Järvinen 已提交
1531

L
Linus Torvalds 已提交
1532
	return 0;
I
Ilpo Järvinen 已提交
1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
}
#endif

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

int pim_rcv_v1(struct sk_buff * skb)
{
	struct igmphdr *pim;

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

	pim = igmp_hdr(skb);

	if (!mroute_do_pim ||
	    pim->group != PIM_V1_VERSION || pim->code != PIM_V1_REGISTER)
		goto drop;

	if (__pim_rcv(skb, sizeof(*pim))) {
drop:
		kfree_skb(skb);
	}
L
Linus Torvalds 已提交
1558 1559 1560 1561 1562 1563 1564 1565 1566
	return 0;
}
#endif

#ifdef CONFIG_IP_PIMSM_V2
static int pim_rcv(struct sk_buff * skb)
{
	struct pimreghdr *pim;

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

1570
	pim = (struct pimreghdr *)skb_transport_header(skb);
1571
	if (pim->type != ((PIM_VERSION<<4)|(PIM_REGISTER)) ||
L
Linus Torvalds 已提交
1572
	    (pim->flags&PIM_NULL_REGISTER) ||
1573
	    (ip_compute_csum((void *)pim, sizeof(*pim)) != 0 &&
1574
	     csum_fold(skb_checksum(skb, 0, skb->len, 0))))
L
Linus Torvalds 已提交
1575 1576
		goto drop;

I
Ilpo Järvinen 已提交
1577 1578 1579 1580
	if (__pim_rcv(skb, sizeof(*pim))) {
drop:
		kfree_skb(skb);
	}
L
Linus Torvalds 已提交
1581 1582 1583 1584 1585 1586 1587 1588 1589
	return 0;
}
#endif

static int
ipmr_fill_mroute(struct sk_buff *skb, struct mfc_cache *c, struct rtmsg *rtm)
{
	int ct;
	struct rtnexthop *nhp;
1590
	struct net_device *dev = init_net.ipv4.vif_table[c->mfc_parent].dev;
1591
	u8 *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
1592 1593 1594 1595 1596
	struct rtattr *mp_head;

	if (dev)
		RTA_PUT(skb, RTA_IIF, 4, &dev->ifindex);

J
Jianjun Kong 已提交
1597
	mp_head = (struct rtattr *)skb_put(skb, RTA_LENGTH(0));
L
Linus Torvalds 已提交
1598 1599 1600 1601 1602

	for (ct = c->mfc_un.res.minvif; ct < c->mfc_un.res.maxvif; ct++) {
		if (c->mfc_un.res.ttls[ct] < 255) {
			if (skb_tailroom(skb) < RTA_ALIGN(RTA_ALIGN(sizeof(*nhp)) + 4))
				goto rtattr_failure;
J
Jianjun Kong 已提交
1603
			nhp = (struct rtnexthop *)skb_put(skb, RTA_ALIGN(sizeof(*nhp)));
L
Linus Torvalds 已提交
1604 1605
			nhp->rtnh_flags = 0;
			nhp->rtnh_hops = c->mfc_un.res.ttls[ct];
1606
			nhp->rtnh_ifindex = init_net.ipv4.vif_table[ct].dev->ifindex;
L
Linus Torvalds 已提交
1607 1608 1609 1610
			nhp->rtnh_len = sizeof(*nhp);
		}
	}
	mp_head->rta_type = RTA_MULTIPATH;
1611
	mp_head->rta_len = skb_tail_pointer(skb) - (u8 *)mp_head;
L
Linus Torvalds 已提交
1612 1613 1614 1615
	rtm->rtm_type = RTN_MULTICAST;
	return 1;

rtattr_failure:
1616
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1617 1618 1619 1620 1621 1622 1623
	return -EMSGSIZE;
}

int ipmr_get_route(struct sk_buff *skb, struct rtmsg *rtm, int nowait)
{
	int err;
	struct mfc_cache *cache;
1624
	struct rtable *rt = skb->rtable;
L
Linus Torvalds 已提交
1625 1626 1627 1628

	read_lock(&mrt_lock);
	cache = ipmr_cache_find(rt->rt_src, rt->rt_dst);

J
Jianjun Kong 已提交
1629
	if (cache == NULL) {
1630
		struct sk_buff *skb2;
1631
		struct iphdr *iph;
L
Linus Torvalds 已提交
1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644
		struct net_device *dev;
		int vif;

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

		dev = skb->dev;
		if (dev == NULL || (vif = ipmr_find_vif(dev)) < 0) {
			read_unlock(&mrt_lock);
			return -ENODEV;
		}
1645 1646 1647 1648 1649 1650
		skb2 = skb_clone(skb, GFP_ATOMIC);
		if (!skb2) {
			read_unlock(&mrt_lock);
			return -ENOMEM;
		}

1651 1652
		skb_push(skb2, sizeof(struct iphdr));
		skb_reset_network_header(skb2);
1653 1654 1655 1656 1657
		iph = ip_hdr(skb2);
		iph->ihl = sizeof(struct iphdr) >> 2;
		iph->saddr = rt->rt_src;
		iph->daddr = rt->rt_dst;
		iph->version = 0;
1658
		err = ipmr_cache_unresolved(vif, skb2);
L
Linus Torvalds 已提交
1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669
		read_unlock(&mrt_lock);
		return err;
	}

	if (!nowait && (rtm->rtm_flags&RTM_F_NOTIFY))
		cache->mfc_flags |= MFC_NOTIFY;
	err = ipmr_fill_mroute(skb, cache, rtm);
	read_unlock(&mrt_lock);
	return err;
}

1670
#ifdef CONFIG_PROC_FS
L
Linus Torvalds 已提交
1671 1672 1673 1674 1675 1676 1677 1678 1679 1680
/*
 *	The /proc interfaces to multicast routing /proc/ip_mr_cache /proc/ip_mr_vif
 */
struct ipmr_vif_iter {
	int ct;
};

static struct vif_device *ipmr_vif_seq_idx(struct ipmr_vif_iter *iter,
					   loff_t pos)
{
1681 1682
	for (iter->ct = 0; iter->ct < init_net.ipv4.maxvif; ++iter->ct) {
		if (!VIF_EXISTS(&init_net, iter->ct))
L
Linus Torvalds 已提交
1683
			continue;
1684
		if (pos-- == 0)
1685
			return &init_net.ipv4.vif_table[iter->ct];
L
Linus Torvalds 已提交
1686 1687 1688 1689 1690
	}
	return NULL;
}

static void *ipmr_vif_seq_start(struct seq_file *seq, loff_t *pos)
S
Stephen Hemminger 已提交
1691
	__acquires(mrt_lock)
L
Linus Torvalds 已提交
1692 1693
{
	read_lock(&mrt_lock);
1694
	return *pos ? ipmr_vif_seq_idx(seq->private, *pos - 1)
L
Linus Torvalds 已提交
1695 1696 1697 1698 1699 1700 1701 1702 1703 1704
		: 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;

	++*pos;
	if (v == SEQ_START_TOKEN)
		return ipmr_vif_seq_idx(iter, 0);
1705

1706 1707
	while (++iter->ct < init_net.ipv4.maxvif) {
		if (!VIF_EXISTS(&init_net, iter->ct))
L
Linus Torvalds 已提交
1708
			continue;
1709
		return &init_net.ipv4.vif_table[iter->ct];
L
Linus Torvalds 已提交
1710 1711 1712 1713 1714
	}
	return NULL;
}

static void ipmr_vif_seq_stop(struct seq_file *seq, void *v)
S
Stephen Hemminger 已提交
1715
	__releases(mrt_lock)
L
Linus Torvalds 已提交
1716 1717 1718 1719 1720 1721 1722
{
	read_unlock(&mrt_lock);
}

static int ipmr_vif_seq_show(struct seq_file *seq, void *v)
{
	if (v == SEQ_START_TOKEN) {
1723
		seq_puts(seq,
L
Linus Torvalds 已提交
1724 1725 1726 1727 1728 1729 1730
			 "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",
1731
			   vif - init_net.ipv4.vif_table,
1732
			   name, vif->bytes_in, vif->pkt_in,
L
Linus Torvalds 已提交
1733 1734 1735 1736 1737 1738
			   vif->bytes_out, vif->pkt_out,
			   vif->flags, vif->local, vif->remote);
	}
	return 0;
}

1739
static const struct seq_operations ipmr_vif_seq_ops = {
L
Linus Torvalds 已提交
1740 1741 1742 1743 1744 1745 1746 1747
	.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)
{
1748 1749
	return seq_open_private(file, &ipmr_vif_seq_ops,
			sizeof(struct ipmr_vif_iter));
L
Linus Torvalds 已提交
1750 1751
}

1752
static const struct file_operations ipmr_vif_fops = {
L
Linus Torvalds 已提交
1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771
	.owner	 = THIS_MODULE,
	.open    = ipmr_vif_open,
	.read    = seq_read,
	.llseek  = seq_lseek,
	.release = seq_release_private,
};

struct ipmr_mfc_iter {
	struct mfc_cache **cache;
	int ct;
};


static struct mfc_cache *ipmr_mfc_seq_idx(struct ipmr_mfc_iter *it, loff_t pos)
{
	struct mfc_cache *mfc;

	it->cache = mfc_cache_array;
	read_lock(&mrt_lock);
1772
	for (it->ct = 0; it->ct < MFC_LINES; it->ct++)
S
Stephen Hemminger 已提交
1773
		for (mfc = mfc_cache_array[it->ct]; mfc; mfc = mfc->next)
1774
			if (pos-- == 0)
L
Linus Torvalds 已提交
1775 1776 1777 1778 1779
				return mfc;
	read_unlock(&mrt_lock);

	it->cache = &mfc_unres_queue;
	spin_lock_bh(&mfc_unres_lock);
S
Stephen Hemminger 已提交
1780
	for (mfc = mfc_unres_queue; mfc; mfc = mfc->next)
L
Linus Torvalds 已提交
1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794
		if (pos-- == 0)
			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;
	it->cache = NULL;
	it->ct = 0;
1795
	return *pos ? ipmr_mfc_seq_idx(seq->private, *pos - 1)
L
Linus Torvalds 已提交
1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810
		: 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;

	++*pos;

	if (v == SEQ_START_TOKEN)
		return ipmr_mfc_seq_idx(seq->private, 0);

	if (mfc->next)
		return mfc->next;
1811 1812

	if (it->cache == &mfc_unres_queue)
L
Linus Torvalds 已提交
1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826
		goto end_of_list;

	BUG_ON(it->cache != mfc_cache_array);

	while (++it->ct < MFC_LINES) {
		mfc = mfc_cache_array[it->ct];
		if (mfc)
			return mfc;
	}

	/* exhausted cache_array, show unresolved */
	read_unlock(&mrt_lock);
	it->cache = &mfc_unres_queue;
	it->ct = 0;
1827

L
Linus Torvalds 已提交
1828 1829
	spin_lock_bh(&mfc_unres_lock);
	mfc = mfc_unres_queue;
1830
	if (mfc)
L
Linus Torvalds 已提交
1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
		return mfc;

 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;

	if (it->cache == &mfc_unres_queue)
		spin_unlock_bh(&mfc_unres_lock);
	else if (it->cache == mfc_cache_array)
		read_unlock(&mrt_lock);
}

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

	if (v == SEQ_START_TOKEN) {
1855
		seq_puts(seq,
L
Linus Torvalds 已提交
1856 1857 1858 1859
		 "Group    Origin   Iif     Pkts    Bytes    Wrong Oifs\n");
	} else {
		const struct mfc_cache *mfc = v;
		const struct ipmr_mfc_iter *it = seq->private;
1860

1861
		seq_printf(seq, "%08lX %08lX %-3hd",
L
Linus Torvalds 已提交
1862 1863
			   (unsigned long) mfc->mfc_mcastgrp,
			   (unsigned long) mfc->mfc_origin,
1864
			   mfc->mfc_parent);
L
Linus Torvalds 已提交
1865 1866

		if (it->cache != &mfc_unres_queue) {
1867 1868 1869 1870
			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 已提交
1871 1872
			for (n = mfc->mfc_un.res.minvif;
			     n < mfc->mfc_un.res.maxvif; n++ ) {
1873 1874 1875
				if (VIF_EXISTS(&init_net, n) &&
				    mfc->mfc_un.res.ttls[n] < 255)
					seq_printf(seq,
1876
					   " %2d:%-3d",
L
Linus Torvalds 已提交
1877 1878
					   n, mfc->mfc_un.res.ttls[n]);
			}
1879 1880 1881 1882 1883
		} 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 已提交
1884 1885 1886 1887 1888 1889
		}
		seq_putc(seq, '\n');
	}
	return 0;
}

1890
static const struct seq_operations ipmr_mfc_seq_ops = {
L
Linus Torvalds 已提交
1891 1892 1893 1894 1895 1896 1897 1898
	.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)
{
1899 1900
	return seq_open_private(file, &ipmr_mfc_seq_ops,
			sizeof(struct ipmr_mfc_iter));
L
Linus Torvalds 已提交
1901 1902
}

1903
static const struct file_operations ipmr_mfc_fops = {
L
Linus Torvalds 已提交
1904 1905 1906 1907 1908 1909
	.owner	 = THIS_MODULE,
	.open    = ipmr_mfc_open,
	.read    = seq_read,
	.llseek  = seq_lseek,
	.release = seq_release_private,
};
1910
#endif
L
Linus Torvalds 已提交
1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921

#ifdef CONFIG_IP_PIMSM_V2
static struct net_protocol pim_protocol = {
	.handler	=	pim_rcv,
};
#endif


/*
 *	Setup for IP multicast routing
 */
1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944
static int __net_init ipmr_net_init(struct net *net)
{
	int err = 0;

	net->ipv4.vif_table = kcalloc(MAXVIFS, sizeof(struct vif_device),
				      GFP_KERNEL);
	if (!net->ipv4.vif_table) {
		err = -ENOMEM;
		goto fail;
	}
fail:
	return err;
}

static void __net_exit ipmr_net_exit(struct net *net)
{
	kfree(net->ipv4.vif_table);
}

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

W
Wang Chen 已提交
1946
int __init ip_mr_init(void)
L
Linus Torvalds 已提交
1947
{
W
Wang Chen 已提交
1948 1949
	int err;

L
Linus Torvalds 已提交
1950 1951
	mrt_cachep = kmem_cache_create("ip_mrt_cache",
				       sizeof(struct mfc_cache),
A
Alexey Dobriyan 已提交
1952
				       0, SLAB_HWCACHE_ALIGN|SLAB_PANIC,
1953
				       NULL);
W
Wang Chen 已提交
1954 1955 1956
	if (!mrt_cachep)
		return -ENOMEM;

1957 1958 1959 1960
	err = register_pernet_subsys(&ipmr_net_ops);
	if (err)
		goto reg_pernet_fail;

1961
	setup_timer(&ipmr_expire_timer, ipmr_expire_process, 0);
W
Wang Chen 已提交
1962 1963 1964
	err = register_netdevice_notifier(&ip_mr_notifier);
	if (err)
		goto reg_notif_fail;
1965
#ifdef CONFIG_PROC_FS
W
Wang Chen 已提交
1966 1967 1968 1969 1970
	err = -ENOMEM;
	if (!proc_net_fops_create(&init_net, "ip_mr_vif", 0, &ipmr_vif_fops))
		goto proc_vif_fail;
	if (!proc_net_fops_create(&init_net, "ip_mr_cache", 0, &ipmr_mfc_fops))
		goto proc_cache_fail;
1971
#endif
W
Wang Chen 已提交
1972 1973 1974 1975
	return 0;
#ifdef CONFIG_PROC_FS
proc_cache_fail:
	proc_net_remove(&init_net, "ip_mr_vif");
B
Benjamin Thery 已提交
1976 1977
proc_vif_fail:
	unregister_netdevice_notifier(&ip_mr_notifier);
W
Wang Chen 已提交
1978
#endif
B
Benjamin Thery 已提交
1979 1980
reg_notif_fail:
	del_timer(&ipmr_expire_timer);
1981 1982
	unregister_pernet_subsys(&ipmr_net_ops);
reg_pernet_fail:
B
Benjamin Thery 已提交
1983
	kmem_cache_destroy(mrt_cachep);
W
Wang Chen 已提交
1984
	return err;
L
Linus Torvalds 已提交
1985
}