blob: f70ba622c856489b6382daf79ccf32a002ca8a22 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * Linux NET3: Internet Group Management Protocol [IGMP]
3 *
4 * This code implements the IGMP protocol as defined in RFC1112. There has
5 * been a further revision of this protocol since which is now supported.
6 *
7 * If you have trouble with this module be careful what gcc you have used,
8 * the older version didn't come out right using gcc 2.5.8, the newer one
9 * seems to fall out with gcc 2.6.2.
10 *
11 * Version: $Id: igmp.c,v 1.47 2002/02/01 22:01:03 davem Exp $
12 *
13 * Authors:
14 * Alan Cox <Alan.Cox@linux.org>
15 *
16 * This program is free software; you can redistribute it and/or
17 * modify it under the terms of the GNU General Public License
18 * as published by the Free Software Foundation; either version
19 * 2 of the License, or (at your option) any later version.
20 *
21 * Fixes:
22 *
23 * Alan Cox : Added lots of __inline__ to optimise
24 * the memory usage of all the tiny little
25 * functions.
26 * Alan Cox : Dumped the header building experiment.
27 * Alan Cox : Minor tweaks ready for multicast routing
28 * and extended IGMP protocol.
29 * Alan Cox : Removed a load of inline directives. Gcc 2.5.8
30 * writes utterly bogus code otherwise (sigh)
31 * fixed IGMP loopback to behave in the manner
32 * desired by mrouted, fixed the fact it has been
33 * broken since 1.3.6 and cleaned up a few minor
34 * points.
35 *
36 * Chih-Jen Chang : Tried to revise IGMP to Version 2
37 * Tsu-Sheng Tsao E-mail: chihjenc@scf.usc.edu and tsusheng@scf.usc.edu
38 * The enhancements are mainly based on Steve Deering's
39 * ipmulti-3.5 source code.
40 * Chih-Jen Chang : Added the igmp_get_mrouter_info and
41 * Tsu-Sheng Tsao igmp_set_mrouter_info to keep track of
42 * the mrouted version on that device.
43 * Chih-Jen Chang : Added the max_resp_time parameter to
44 * Tsu-Sheng Tsao igmp_heard_query(). Using this parameter
45 * to identify the multicast router version
46 * and do what the IGMP version 2 specified.
47 * Chih-Jen Chang : Added a timer to revert to IGMP V2 router
48 * Tsu-Sheng Tsao if the specified time expired.
49 * Alan Cox : Stop IGMP from 0.0.0.0 being accepted.
50 * Alan Cox : Use GFP_ATOMIC in the right places.
51 * Christian Daudt : igmp timer wasn't set for local group
52 * memberships but was being deleted,
53 * which caused a "del_timer() called
54 * from %p with timer not initialized\n"
55 * message (960131).
56 * Christian Daudt : removed del_timer from
57 * igmp_timer_expire function (960205).
58 * Christian Daudt : igmp_heard_report now only calls
59 * igmp_timer_expire if tm->running is
60 * true (960216).
61 * Malcolm Beattie : ttl comparison wrong in igmp_rcv made
62 * igmp_heard_query never trigger. Expiry
63 * miscalculation fixed in igmp_heard_query
64 * and random() made to return unsigned to
65 * prevent negative expiry times.
66 * Alexey Kuznetsov: Wrong group leaving behaviour, backport
67 * fix from pending 2.1.x patches.
68 * Alan Cox: Forget to enable FDDI support earlier.
69 * Alexey Kuznetsov: Fixed leaving groups on device down.
70 * Alexey Kuznetsov: Accordance to igmp-v2-06 draft.
71 * David L Stevens: IGMPv3 support, with help from
72 * Vinay Kulkarni
73 */
74
75#include <linux/config.h>
76#include <linux/module.h>
77#include <asm/uaccess.h>
78#include <asm/system.h>
79#include <linux/types.h>
80#include <linux/kernel.h>
81#include <linux/jiffies.h>
82#include <linux/string.h>
83#include <linux/socket.h>
84#include <linux/sockios.h>
85#include <linux/in.h>
86#include <linux/inet.h>
87#include <linux/netdevice.h>
88#include <linux/skbuff.h>
89#include <linux/inetdevice.h>
90#include <linux/igmp.h>
91#include <linux/if_arp.h>
92#include <linux/rtnetlink.h>
93#include <linux/times.h>
Arnaldo Carvalho de Melo14c85022005-12-27 02:43:12 -020094
95#include <net/arp.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070096#include <net/ip.h>
97#include <net/protocol.h>
98#include <net/route.h>
99#include <net/sock.h>
100#include <net/checksum.h>
101#include <linux/netfilter_ipv4.h>
102#ifdef CONFIG_IP_MROUTE
103#include <linux/mroute.h>
104#endif
105#ifdef CONFIG_PROC_FS
106#include <linux/proc_fs.h>
107#include <linux/seq_file.h>
108#endif
109
110#define IP_MAX_MEMBERSHIPS 20
111#define IP_MAX_MSF 10
112
113#ifdef CONFIG_IP_MULTICAST
114/* Parameter names and values are taken from igmp-v2-06 draft */
115
116#define IGMP_V1_Router_Present_Timeout (400*HZ)
117#define IGMP_V2_Router_Present_Timeout (400*HZ)
118#define IGMP_Unsolicited_Report_Interval (10*HZ)
119#define IGMP_Query_Response_Interval (10*HZ)
120#define IGMP_Unsolicited_Report_Count 2
121
122
123#define IGMP_Initial_Report_Delay (1)
124
125/* IGMP_Initial_Report_Delay is not from IGMP specs!
126 * IGMP specs require to report membership immediately after
127 * joining a group, but we delay the first report by a
128 * small interval. It seems more natural and still does not
129 * contradict to specs provided this delay is small enough.
130 */
131
132#define IGMP_V1_SEEN(in_dev) (ipv4_devconf.force_igmp_version == 1 || \
133 (in_dev)->cnf.force_igmp_version == 1 || \
134 ((in_dev)->mr_v1_seen && \
135 time_before(jiffies, (in_dev)->mr_v1_seen)))
136#define IGMP_V2_SEEN(in_dev) (ipv4_devconf.force_igmp_version == 2 || \
137 (in_dev)->cnf.force_igmp_version == 2 || \
138 ((in_dev)->mr_v2_seen && \
139 time_before(jiffies, (in_dev)->mr_v2_seen)))
140
141static void igmpv3_add_delrec(struct in_device *in_dev, struct ip_mc_list *im);
142static void igmpv3_del_delrec(struct in_device *in_dev, __u32 multiaddr);
143static void igmpv3_clear_delrec(struct in_device *in_dev);
144static int sf_setstate(struct ip_mc_list *pmc);
145static void sf_markstate(struct ip_mc_list *pmc);
146#endif
147static void ip_mc_clear_src(struct ip_mc_list *pmc);
148static int ip_mc_add_src(struct in_device *in_dev, __u32 *pmca, int sfmode,
149 int sfcount, __u32 *psfsrc, int delta);
150
151static void ip_ma_put(struct ip_mc_list *im)
152{
153 if (atomic_dec_and_test(&im->refcnt)) {
154 in_dev_put(im->interface);
155 kfree(im);
156 }
157}
158
159#ifdef CONFIG_IP_MULTICAST
160
161/*
162 * Timer management
163 */
164
165static __inline__ void igmp_stop_timer(struct ip_mc_list *im)
166{
167 spin_lock_bh(&im->lock);
168 if (del_timer(&im->timer))
169 atomic_dec(&im->refcnt);
170 im->tm_running=0;
171 im->reporter = 0;
172 im->unsolicit_count = 0;
173 spin_unlock_bh(&im->lock);
174}
175
176/* It must be called with locked im->lock */
177static void igmp_start_timer(struct ip_mc_list *im, int max_delay)
178{
179 int tv=net_random() % max_delay;
180
181 im->tm_running=1;
182 if (!mod_timer(&im->timer, jiffies+tv+2))
183 atomic_inc(&im->refcnt);
184}
185
186static void igmp_gq_start_timer(struct in_device *in_dev)
187{
188 int tv = net_random() % in_dev->mr_maxdelay;
189
190 in_dev->mr_gq_running = 1;
191 if (!mod_timer(&in_dev->mr_gq_timer, jiffies+tv+2))
192 in_dev_hold(in_dev);
193}
194
195static void igmp_ifc_start_timer(struct in_device *in_dev, int delay)
196{
197 int tv = net_random() % delay;
198
199 if (!mod_timer(&in_dev->mr_ifc_timer, jiffies+tv+2))
200 in_dev_hold(in_dev);
201}
202
203static void igmp_mod_timer(struct ip_mc_list *im, int max_delay)
204{
205 spin_lock_bh(&im->lock);
206 im->unsolicit_count = 0;
207 if (del_timer(&im->timer)) {
208 if ((long)(im->timer.expires-jiffies) < max_delay) {
209 add_timer(&im->timer);
210 im->tm_running=1;
211 spin_unlock_bh(&im->lock);
212 return;
213 }
214 atomic_dec(&im->refcnt);
215 }
216 igmp_start_timer(im, max_delay);
217 spin_unlock_bh(&im->lock);
218}
219
220
221/*
222 * Send an IGMP report.
223 */
224
225#define IGMP_SIZE (sizeof(struct igmphdr)+sizeof(struct iphdr)+4)
226
227
228static int is_in(struct ip_mc_list *pmc, struct ip_sf_list *psf, int type,
229 int gdeleted, int sdeleted)
230{
231 switch (type) {
232 case IGMPV3_MODE_IS_INCLUDE:
233 case IGMPV3_MODE_IS_EXCLUDE:
234 if (gdeleted || sdeleted)
235 return 0;
David L Stevensad125832006-01-18 14:20:56 -0800236 if (!(pmc->gsquery && !psf->sf_gsresp)) {
237 if (pmc->sfmode == MCAST_INCLUDE)
238 return 1;
239 /* don't include if this source is excluded
240 * in all filters
241 */
242 if (psf->sf_count[MCAST_INCLUDE])
243 return type == IGMPV3_MODE_IS_INCLUDE;
244 return pmc->sfcount[MCAST_EXCLUDE] ==
245 psf->sf_count[MCAST_EXCLUDE];
246 }
247 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700248 case IGMPV3_CHANGE_TO_INCLUDE:
249 if (gdeleted || sdeleted)
250 return 0;
251 return psf->sf_count[MCAST_INCLUDE] != 0;
252 case IGMPV3_CHANGE_TO_EXCLUDE:
253 if (gdeleted || sdeleted)
254 return 0;
255 if (pmc->sfcount[MCAST_EXCLUDE] == 0 ||
256 psf->sf_count[MCAST_INCLUDE])
257 return 0;
258 return pmc->sfcount[MCAST_EXCLUDE] ==
259 psf->sf_count[MCAST_EXCLUDE];
260 case IGMPV3_ALLOW_NEW_SOURCES:
261 if (gdeleted || !psf->sf_crcount)
262 return 0;
263 return (pmc->sfmode == MCAST_INCLUDE) ^ sdeleted;
264 case IGMPV3_BLOCK_OLD_SOURCES:
265 if (pmc->sfmode == MCAST_INCLUDE)
266 return gdeleted || (psf->sf_crcount && sdeleted);
267 return psf->sf_crcount && !gdeleted && !sdeleted;
268 }
269 return 0;
270}
271
272static int
273igmp_scount(struct ip_mc_list *pmc, int type, int gdeleted, int sdeleted)
274{
275 struct ip_sf_list *psf;
276 int scount = 0;
277
278 for (psf=pmc->sources; psf; psf=psf->sf_next) {
279 if (!is_in(pmc, psf, type, gdeleted, sdeleted))
280 continue;
281 scount++;
282 }
283 return scount;
284}
285
286static struct sk_buff *igmpv3_newpack(struct net_device *dev, int size)
287{
288 struct sk_buff *skb;
289 struct rtable *rt;
290 struct iphdr *pip;
291 struct igmpv3_report *pig;
292
293 skb = alloc_skb(size + LL_RESERVED_SPACE(dev), GFP_ATOMIC);
294 if (skb == NULL)
295 return NULL;
296
297 {
298 struct flowi fl = { .oif = dev->ifindex,
299 .nl_u = { .ip4_u = {
300 .daddr = IGMPV3_ALL_MCR } },
301 .proto = IPPROTO_IGMP };
302 if (ip_route_output_key(&rt, &fl)) {
303 kfree_skb(skb);
304 return NULL;
305 }
306 }
307 if (rt->rt_src == 0) {
308 kfree_skb(skb);
309 ip_rt_put(rt);
310 return NULL;
311 }
312
313 skb->dst = &rt->u.dst;
314 skb->dev = dev;
315
316 skb_reserve(skb, LL_RESERVED_SPACE(dev));
317
318 skb->nh.iph = pip =(struct iphdr *)skb_put(skb, sizeof(struct iphdr)+4);
319
320 pip->version = 4;
321 pip->ihl = (sizeof(struct iphdr)+4)>>2;
322 pip->tos = 0xc0;
323 pip->frag_off = htons(IP_DF);
324 pip->ttl = 1;
325 pip->daddr = rt->rt_dst;
326 pip->saddr = rt->rt_src;
327 pip->protocol = IPPROTO_IGMP;
328 pip->tot_len = 0; /* filled in later */
329 ip_select_ident(pip, &rt->u.dst, NULL);
330 ((u8*)&pip[1])[0] = IPOPT_RA;
331 ((u8*)&pip[1])[1] = 4;
332 ((u8*)&pip[1])[2] = 0;
333 ((u8*)&pip[1])[3] = 0;
334
335 pig =(struct igmpv3_report *)skb_put(skb, sizeof(*pig));
336 skb->h.igmph = (struct igmphdr *)pig;
337 pig->type = IGMPV3_HOST_MEMBERSHIP_REPORT;
338 pig->resv1 = 0;
339 pig->csum = 0;
340 pig->resv2 = 0;
341 pig->ngrec = 0;
342 return skb;
343}
344
345static int igmpv3_sendpack(struct sk_buff *skb)
346{
347 struct iphdr *pip = skb->nh.iph;
348 struct igmphdr *pig = skb->h.igmph;
349 int iplen, igmplen;
350
351 iplen = skb->tail - (unsigned char *)skb->nh.iph;
352 pip->tot_len = htons(iplen);
353 ip_send_check(pip);
354
355 igmplen = skb->tail - (unsigned char *)skb->h.igmph;
356 pig->csum = ip_compute_csum((void *)skb->h.igmph, igmplen);
357
358 return NF_HOOK(PF_INET, NF_IP_LOCAL_OUT, skb, NULL, skb->dev,
359 dst_output);
360}
361
362static int grec_size(struct ip_mc_list *pmc, int type, int gdel, int sdel)
363{
364 return sizeof(struct igmpv3_grec) + 4*igmp_scount(pmc,type,gdel,sdel);
365}
366
367static struct sk_buff *add_grhead(struct sk_buff *skb, struct ip_mc_list *pmc,
368 int type, struct igmpv3_grec **ppgr)
369{
370 struct net_device *dev = pmc->interface->dev;
371 struct igmpv3_report *pih;
372 struct igmpv3_grec *pgr;
373
374 if (!skb)
375 skb = igmpv3_newpack(dev, dev->mtu);
376 if (!skb)
377 return NULL;
378 pgr = (struct igmpv3_grec *)skb_put(skb, sizeof(struct igmpv3_grec));
379 pgr->grec_type = type;
380 pgr->grec_auxwords = 0;
381 pgr->grec_nsrcs = 0;
382 pgr->grec_mca = pmc->multiaddr;
383 pih = (struct igmpv3_report *)skb->h.igmph;
384 pih->ngrec = htons(ntohs(pih->ngrec)+1);
385 *ppgr = pgr;
386 return skb;
387}
388
389#define AVAILABLE(skb) ((skb) ? ((skb)->dev ? (skb)->dev->mtu - (skb)->len : \
390 skb_tailroom(skb)) : 0)
391
392static struct sk_buff *add_grec(struct sk_buff *skb, struct ip_mc_list *pmc,
393 int type, int gdeleted, int sdeleted)
394{
395 struct net_device *dev = pmc->interface->dev;
396 struct igmpv3_report *pih;
397 struct igmpv3_grec *pgr = NULL;
398 struct ip_sf_list *psf, *psf_next, *psf_prev, **psf_list;
David L Stevensad125832006-01-18 14:20:56 -0800399 int scount, stotal, first, isquery, truncate;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700400
401 if (pmc->multiaddr == IGMP_ALL_HOSTS)
402 return skb;
403
404 isquery = type == IGMPV3_MODE_IS_INCLUDE ||
405 type == IGMPV3_MODE_IS_EXCLUDE;
406 truncate = type == IGMPV3_MODE_IS_EXCLUDE ||
407 type == IGMPV3_CHANGE_TO_EXCLUDE;
408
David L Stevensad125832006-01-18 14:20:56 -0800409 stotal = scount = 0;
410
Linus Torvalds1da177e2005-04-16 15:20:36 -0700411 psf_list = sdeleted ? &pmc->tomb : &pmc->sources;
412
David L Stevensad125832006-01-18 14:20:56 -0800413 if (!*psf_list)
414 goto empty_source;
415
Linus Torvalds1da177e2005-04-16 15:20:36 -0700416 pih = skb ? (struct igmpv3_report *)skb->h.igmph : NULL;
417
418 /* EX and TO_EX get a fresh packet, if needed */
419 if (truncate) {
420 if (pih && pih->ngrec &&
421 AVAILABLE(skb) < grec_size(pmc, type, gdeleted, sdeleted)) {
422 if (skb)
423 igmpv3_sendpack(skb);
424 skb = igmpv3_newpack(dev, dev->mtu);
425 }
426 }
427 first = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700428 psf_prev = NULL;
429 for (psf=*psf_list; psf; psf=psf_next) {
430 u32 *psrc;
431
432 psf_next = psf->sf_next;
433
434 if (!is_in(pmc, psf, type, gdeleted, sdeleted)) {
435 psf_prev = psf;
436 continue;
437 }
438
439 /* clear marks on query responses */
440 if (isquery)
441 psf->sf_gsresp = 0;
442
443 if (AVAILABLE(skb) < sizeof(u32) +
444 first*sizeof(struct igmpv3_grec)) {
445 if (truncate && !first)
446 break; /* truncate these */
447 if (pgr)
448 pgr->grec_nsrcs = htons(scount);
449 if (skb)
450 igmpv3_sendpack(skb);
451 skb = igmpv3_newpack(dev, dev->mtu);
452 first = 1;
453 scount = 0;
454 }
455 if (first) {
456 skb = add_grhead(skb, pmc, type, &pgr);
457 first = 0;
458 }
459 psrc = (u32 *)skb_put(skb, sizeof(u32));
460 *psrc = psf->sf_inaddr;
David L Stevensad125832006-01-18 14:20:56 -0800461 scount++; stotal++;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700462 if ((type == IGMPV3_ALLOW_NEW_SOURCES ||
463 type == IGMPV3_BLOCK_OLD_SOURCES) && psf->sf_crcount) {
464 psf->sf_crcount--;
465 if ((sdeleted || gdeleted) && psf->sf_crcount == 0) {
466 if (psf_prev)
467 psf_prev->sf_next = psf->sf_next;
468 else
469 *psf_list = psf->sf_next;
470 kfree(psf);
471 continue;
472 }
473 }
474 psf_prev = psf;
475 }
David L Stevensad125832006-01-18 14:20:56 -0800476
477empty_source:
478 if (!stotal) {
479 if (type == IGMPV3_ALLOW_NEW_SOURCES ||
480 type == IGMPV3_BLOCK_OLD_SOURCES)
481 return skb;
482 if (pmc->crcount || isquery) {
483 /* make sure we have room for group header */
484 if (skb && AVAILABLE(skb)<sizeof(struct igmpv3_grec)) {
485 igmpv3_sendpack(skb);
486 skb = NULL; /* add_grhead will get a new one */
487 }
488 skb = add_grhead(skb, pmc, type, &pgr);
489 }
490 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700491 if (pgr)
492 pgr->grec_nsrcs = htons(scount);
493
494 if (isquery)
495 pmc->gsquery = 0; /* clear query state on report */
496 return skb;
497}
498
499static int igmpv3_send_report(struct in_device *in_dev, struct ip_mc_list *pmc)
500{
501 struct sk_buff *skb = NULL;
502 int type;
503
504 if (!pmc) {
505 read_lock(&in_dev->mc_list_lock);
506 for (pmc=in_dev->mc_list; pmc; pmc=pmc->next) {
507 if (pmc->multiaddr == IGMP_ALL_HOSTS)
508 continue;
509 spin_lock_bh(&pmc->lock);
510 if (pmc->sfcount[MCAST_EXCLUDE])
511 type = IGMPV3_MODE_IS_EXCLUDE;
512 else
513 type = IGMPV3_MODE_IS_INCLUDE;
514 skb = add_grec(skb, pmc, type, 0, 0);
515 spin_unlock_bh(&pmc->lock);
516 }
517 read_unlock(&in_dev->mc_list_lock);
518 } else {
519 spin_lock_bh(&pmc->lock);
520 if (pmc->sfcount[MCAST_EXCLUDE])
521 type = IGMPV3_MODE_IS_EXCLUDE;
522 else
523 type = IGMPV3_MODE_IS_INCLUDE;
524 skb = add_grec(skb, pmc, type, 0, 0);
525 spin_unlock_bh(&pmc->lock);
526 }
527 if (!skb)
528 return 0;
529 return igmpv3_sendpack(skb);
530}
531
532/*
533 * remove zero-count source records from a source filter list
534 */
535static void igmpv3_clear_zeros(struct ip_sf_list **ppsf)
536{
537 struct ip_sf_list *psf_prev, *psf_next, *psf;
538
539 psf_prev = NULL;
540 for (psf=*ppsf; psf; psf = psf_next) {
541 psf_next = psf->sf_next;
542 if (psf->sf_crcount == 0) {
543 if (psf_prev)
544 psf_prev->sf_next = psf->sf_next;
545 else
546 *ppsf = psf->sf_next;
547 kfree(psf);
548 } else
549 psf_prev = psf;
550 }
551}
552
553static void igmpv3_send_cr(struct in_device *in_dev)
554{
555 struct ip_mc_list *pmc, *pmc_prev, *pmc_next;
556 struct sk_buff *skb = NULL;
557 int type, dtype;
558
559 read_lock(&in_dev->mc_list_lock);
560 spin_lock_bh(&in_dev->mc_tomb_lock);
561
562 /* deleted MCA's */
563 pmc_prev = NULL;
564 for (pmc=in_dev->mc_tomb; pmc; pmc=pmc_next) {
565 pmc_next = pmc->next;
566 if (pmc->sfmode == MCAST_INCLUDE) {
567 type = IGMPV3_BLOCK_OLD_SOURCES;
568 dtype = IGMPV3_BLOCK_OLD_SOURCES;
569 skb = add_grec(skb, pmc, type, 1, 0);
570 skb = add_grec(skb, pmc, dtype, 1, 1);
571 }
572 if (pmc->crcount) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700573 if (pmc->sfmode == MCAST_EXCLUDE) {
574 type = IGMPV3_CHANGE_TO_INCLUDE;
575 skb = add_grec(skb, pmc, type, 1, 0);
576 }
David L Stevensad125832006-01-18 14:20:56 -0800577 pmc->crcount--;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700578 if (pmc->crcount == 0) {
579 igmpv3_clear_zeros(&pmc->tomb);
580 igmpv3_clear_zeros(&pmc->sources);
581 }
582 }
583 if (pmc->crcount == 0 && !pmc->tomb && !pmc->sources) {
584 if (pmc_prev)
585 pmc_prev->next = pmc_next;
586 else
587 in_dev->mc_tomb = pmc_next;
588 in_dev_put(pmc->interface);
589 kfree(pmc);
590 } else
591 pmc_prev = pmc;
592 }
593 spin_unlock_bh(&in_dev->mc_tomb_lock);
594
595 /* change recs */
596 for (pmc=in_dev->mc_list; pmc; pmc=pmc->next) {
597 spin_lock_bh(&pmc->lock);
598 if (pmc->sfcount[MCAST_EXCLUDE]) {
599 type = IGMPV3_BLOCK_OLD_SOURCES;
600 dtype = IGMPV3_ALLOW_NEW_SOURCES;
601 } else {
602 type = IGMPV3_ALLOW_NEW_SOURCES;
603 dtype = IGMPV3_BLOCK_OLD_SOURCES;
604 }
605 skb = add_grec(skb, pmc, type, 0, 0);
606 skb = add_grec(skb, pmc, dtype, 0, 1); /* deleted sources */
607
608 /* filter mode changes */
609 if (pmc->crcount) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700610 if (pmc->sfmode == MCAST_EXCLUDE)
611 type = IGMPV3_CHANGE_TO_EXCLUDE;
612 else
613 type = IGMPV3_CHANGE_TO_INCLUDE;
614 skb = add_grec(skb, pmc, type, 0, 0);
David L Stevensad125832006-01-18 14:20:56 -0800615 pmc->crcount--;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700616 }
617 spin_unlock_bh(&pmc->lock);
618 }
619 read_unlock(&in_dev->mc_list_lock);
620
621 if (!skb)
622 return;
623 (void) igmpv3_sendpack(skb);
624}
625
626static int igmp_send_report(struct in_device *in_dev, struct ip_mc_list *pmc,
627 int type)
628{
629 struct sk_buff *skb;
630 struct iphdr *iph;
631 struct igmphdr *ih;
632 struct rtable *rt;
633 struct net_device *dev = in_dev->dev;
634 u32 group = pmc ? pmc->multiaddr : 0;
635 u32 dst;
636
637 if (type == IGMPV3_HOST_MEMBERSHIP_REPORT)
638 return igmpv3_send_report(in_dev, pmc);
639 else if (type == IGMP_HOST_LEAVE_MESSAGE)
640 dst = IGMP_ALL_ROUTER;
641 else
642 dst = group;
643
644 {
645 struct flowi fl = { .oif = dev->ifindex,
646 .nl_u = { .ip4_u = { .daddr = dst } },
647 .proto = IPPROTO_IGMP };
648 if (ip_route_output_key(&rt, &fl))
649 return -1;
650 }
651 if (rt->rt_src == 0) {
652 ip_rt_put(rt);
653 return -1;
654 }
655
656 skb=alloc_skb(IGMP_SIZE+LL_RESERVED_SPACE(dev), GFP_ATOMIC);
657 if (skb == NULL) {
658 ip_rt_put(rt);
659 return -1;
660 }
661
662 skb->dst = &rt->u.dst;
663
664 skb_reserve(skb, LL_RESERVED_SPACE(dev));
665
666 skb->nh.iph = iph = (struct iphdr *)skb_put(skb, sizeof(struct iphdr)+4);
667
668 iph->version = 4;
669 iph->ihl = (sizeof(struct iphdr)+4)>>2;
670 iph->tos = 0xc0;
671 iph->frag_off = htons(IP_DF);
672 iph->ttl = 1;
673 iph->daddr = dst;
674 iph->saddr = rt->rt_src;
675 iph->protocol = IPPROTO_IGMP;
676 iph->tot_len = htons(IGMP_SIZE);
677 ip_select_ident(iph, &rt->u.dst, NULL);
678 ((u8*)&iph[1])[0] = IPOPT_RA;
679 ((u8*)&iph[1])[1] = 4;
680 ((u8*)&iph[1])[2] = 0;
681 ((u8*)&iph[1])[3] = 0;
682 ip_send_check(iph);
683
684 ih = (struct igmphdr *)skb_put(skb, sizeof(struct igmphdr));
685 ih->type=type;
686 ih->code=0;
687 ih->csum=0;
688 ih->group=group;
689 ih->csum=ip_compute_csum((void *)ih, sizeof(struct igmphdr));
690
691 return NF_HOOK(PF_INET, NF_IP_LOCAL_OUT, skb, NULL, rt->u.dst.dev,
692 dst_output);
693}
694
695static void igmp_gq_timer_expire(unsigned long data)
696{
697 struct in_device *in_dev = (struct in_device *)data;
698
699 in_dev->mr_gq_running = 0;
700 igmpv3_send_report(in_dev, NULL);
701 __in_dev_put(in_dev);
702}
703
704static void igmp_ifc_timer_expire(unsigned long data)
705{
706 struct in_device *in_dev = (struct in_device *)data;
707
708 igmpv3_send_cr(in_dev);
709 if (in_dev->mr_ifc_count) {
710 in_dev->mr_ifc_count--;
711 igmp_ifc_start_timer(in_dev, IGMP_Unsolicited_Report_Interval);
712 }
713 __in_dev_put(in_dev);
714}
715
716static void igmp_ifc_event(struct in_device *in_dev)
717{
718 if (IGMP_V1_SEEN(in_dev) || IGMP_V2_SEEN(in_dev))
719 return;
720 in_dev->mr_ifc_count = in_dev->mr_qrv ? in_dev->mr_qrv :
721 IGMP_Unsolicited_Report_Count;
722 igmp_ifc_start_timer(in_dev, 1);
723}
724
725
726static void igmp_timer_expire(unsigned long data)
727{
728 struct ip_mc_list *im=(struct ip_mc_list *)data;
729 struct in_device *in_dev = im->interface;
730
731 spin_lock(&im->lock);
732 im->tm_running=0;
733
734 if (im->unsolicit_count) {
735 im->unsolicit_count--;
736 igmp_start_timer(im, IGMP_Unsolicited_Report_Interval);
737 }
738 im->reporter = 1;
739 spin_unlock(&im->lock);
740
741 if (IGMP_V1_SEEN(in_dev))
742 igmp_send_report(in_dev, im, IGMP_HOST_MEMBERSHIP_REPORT);
743 else if (IGMP_V2_SEEN(in_dev))
744 igmp_send_report(in_dev, im, IGMPV2_HOST_MEMBERSHIP_REPORT);
745 else
746 igmp_send_report(in_dev, im, IGMPV3_HOST_MEMBERSHIP_REPORT);
747
748 ip_ma_put(im);
749}
750
David L Stevensad125832006-01-18 14:20:56 -0800751/* mark EXCLUDE-mode sources */
752static int igmp_xmarksources(struct ip_mc_list *pmc, int nsrcs, __u32 *srcs)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700753{
754 struct ip_sf_list *psf;
755 int i, scount;
756
757 scount = 0;
758 for (psf=pmc->sources; psf; psf=psf->sf_next) {
759 if (scount == nsrcs)
760 break;
David L Stevensad125832006-01-18 14:20:56 -0800761 for (i=0; i<nsrcs; i++) {
762 /* skip inactive filters */
763 if (pmc->sfcount[MCAST_INCLUDE] ||
764 pmc->sfcount[MCAST_EXCLUDE] !=
765 psf->sf_count[MCAST_EXCLUDE])
766 continue;
767 if (srcs[i] == psf->sf_inaddr) {
768 scount++;
769 break;
770 }
771 }
772 }
773 pmc->gsquery = 0;
774 if (scount == nsrcs) /* all sources excluded */
775 return 0;
776 return 1;
777}
778
779static int igmp_marksources(struct ip_mc_list *pmc, int nsrcs, __u32 *srcs)
780{
781 struct ip_sf_list *psf;
782 int i, scount;
783
784 if (pmc->sfmode == MCAST_EXCLUDE)
785 return igmp_xmarksources(pmc, nsrcs, srcs);
786
787 /* mark INCLUDE-mode sources */
788 scount = 0;
789 for (psf=pmc->sources; psf; psf=psf->sf_next) {
790 if (scount == nsrcs)
791 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700792 for (i=0; i<nsrcs; i++)
793 if (srcs[i] == psf->sf_inaddr) {
794 psf->sf_gsresp = 1;
795 scount++;
796 break;
797 }
798 }
David L Stevensad125832006-01-18 14:20:56 -0800799 if (!scount) {
800 pmc->gsquery = 0;
801 return 0;
802 }
803 pmc->gsquery = 1;
804 return 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700805}
806
807static void igmp_heard_report(struct in_device *in_dev, u32 group)
808{
809 struct ip_mc_list *im;
810
811 /* Timers are only set for non-local groups */
812
813 if (group == IGMP_ALL_HOSTS)
814 return;
815
816 read_lock(&in_dev->mc_list_lock);
817 for (im=in_dev->mc_list; im!=NULL; im=im->next) {
818 if (im->multiaddr == group) {
819 igmp_stop_timer(im);
820 break;
821 }
822 }
823 read_unlock(&in_dev->mc_list_lock);
824}
825
826static void igmp_heard_query(struct in_device *in_dev, struct sk_buff *skb,
827 int len)
828{
829 struct igmphdr *ih = skb->h.igmph;
830 struct igmpv3_query *ih3 = (struct igmpv3_query *)ih;
831 struct ip_mc_list *im;
832 u32 group = ih->group;
833 int max_delay;
834 int mark = 0;
835
836
837 if (len == 8) {
838 if (ih->code == 0) {
839 /* Alas, old v1 router presents here. */
840
841 max_delay = IGMP_Query_Response_Interval;
842 in_dev->mr_v1_seen = jiffies +
843 IGMP_V1_Router_Present_Timeout;
844 group = 0;
845 } else {
846 /* v2 router present */
847 max_delay = ih->code*(HZ/IGMP_TIMER_SCALE);
848 in_dev->mr_v2_seen = jiffies +
849 IGMP_V2_Router_Present_Timeout;
850 }
851 /* cancel the interface change timer */
852 in_dev->mr_ifc_count = 0;
853 if (del_timer(&in_dev->mr_ifc_timer))
854 __in_dev_put(in_dev);
855 /* clear deleted report items */
856 igmpv3_clear_delrec(in_dev);
857 } else if (len < 12) {
858 return; /* ignore bogus packet; freed by caller */
859 } else { /* v3 */
860 if (!pskb_may_pull(skb, sizeof(struct igmpv3_query)))
861 return;
862
863 ih3 = (struct igmpv3_query *) skb->h.raw;
864 if (ih3->nsrcs) {
865 if (!pskb_may_pull(skb, sizeof(struct igmpv3_query)
866 + ntohs(ih3->nsrcs)*sizeof(__u32)))
867 return;
868 ih3 = (struct igmpv3_query *) skb->h.raw;
869 }
870
871 max_delay = IGMPV3_MRC(ih3->code)*(HZ/IGMP_TIMER_SCALE);
872 if (!max_delay)
873 max_delay = 1; /* can't mod w/ 0 */
874 in_dev->mr_maxdelay = max_delay;
875 if (ih3->qrv)
876 in_dev->mr_qrv = ih3->qrv;
877 if (!group) { /* general query */
878 if (ih3->nsrcs)
879 return; /* no sources allowed */
880 igmp_gq_start_timer(in_dev);
881 return;
882 }
883 /* mark sources to include, if group & source-specific */
884 mark = ih3->nsrcs != 0;
885 }
886
887 /*
888 * - Start the timers in all of our membership records
889 * that the query applies to for the interface on
890 * which the query arrived excl. those that belong
891 * to a "local" group (224.0.0.X)
892 * - For timers already running check if they need to
893 * be reset.
894 * - Use the igmp->igmp_code field as the maximum
895 * delay possible
896 */
897 read_lock(&in_dev->mc_list_lock);
898 for (im=in_dev->mc_list; im!=NULL; im=im->next) {
David L Stevensad125832006-01-18 14:20:56 -0800899 int changed;
900
Linus Torvalds1da177e2005-04-16 15:20:36 -0700901 if (group && group != im->multiaddr)
902 continue;
903 if (im->multiaddr == IGMP_ALL_HOSTS)
904 continue;
905 spin_lock_bh(&im->lock);
906 if (im->tm_running)
907 im->gsquery = im->gsquery && mark;
908 else
909 im->gsquery = mark;
David L Stevensad125832006-01-18 14:20:56 -0800910 changed = !im->gsquery ||
911 igmp_marksources(im, ntohs(ih3->nsrcs), ih3->srcs);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700912 spin_unlock_bh(&im->lock);
David L Stevensad125832006-01-18 14:20:56 -0800913 if (changed)
914 igmp_mod_timer(im, max_delay);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700915 }
916 read_unlock(&in_dev->mc_list_lock);
917}
918
919int igmp_rcv(struct sk_buff *skb)
920{
921 /* This basically follows the spec line by line -- see RFC1112 */
922 struct igmphdr *ih;
923 struct in_device *in_dev = in_dev_get(skb->dev);
924 int len = skb->len;
925
926 if (in_dev==NULL) {
927 kfree_skb(skb);
928 return 0;
929 }
930
Herbert Xufb286bb2005-11-10 13:01:24 -0800931 if (!pskb_may_pull(skb, sizeof(struct igmphdr)))
932 goto drop;
933
934 switch (skb->ip_summed) {
935 case CHECKSUM_HW:
936 if (!(u16)csum_fold(skb->csum))
937 break;
938 /* fall through */
939 case CHECKSUM_NONE:
940 skb->csum = 0;
941 if (__skb_checksum_complete(skb))
942 goto drop;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700943 }
944
945 ih = skb->h.igmph;
946 switch (ih->type) {
947 case IGMP_HOST_MEMBERSHIP_QUERY:
948 igmp_heard_query(in_dev, skb, len);
949 break;
950 case IGMP_HOST_MEMBERSHIP_REPORT:
951 case IGMPV2_HOST_MEMBERSHIP_REPORT:
952 case IGMPV3_HOST_MEMBERSHIP_REPORT:
953 /* Is it our report looped back? */
954 if (((struct rtable*)skb->dst)->fl.iif == 0)
955 break;
David Stevens24c69272005-12-02 20:32:59 -0800956 /* don't rely on MC router hearing unicast reports */
957 if (skb->pkt_type == PACKET_MULTICAST ||
958 skb->pkt_type == PACKET_BROADCAST)
959 igmp_heard_report(in_dev, ih->group);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700960 break;
961 case IGMP_PIM:
962#ifdef CONFIG_IP_PIMSM_V1
963 in_dev_put(in_dev);
964 return pim_rcv_v1(skb);
965#endif
966 case IGMP_DVMRP:
967 case IGMP_TRACE:
968 case IGMP_HOST_LEAVE_MESSAGE:
969 case IGMP_MTRACE:
970 case IGMP_MTRACE_RESP:
971 break;
972 default:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700973 }
Herbert Xufb286bb2005-11-10 13:01:24 -0800974
975drop:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700976 in_dev_put(in_dev);
977 kfree_skb(skb);
978 return 0;
979}
980
981#endif
982
983
984/*
985 * Add a filter to a device
986 */
987
988static void ip_mc_filter_add(struct in_device *in_dev, u32 addr)
989{
990 char buf[MAX_ADDR_LEN];
991 struct net_device *dev = in_dev->dev;
992
993 /* Checking for IFF_MULTICAST here is WRONG-WRONG-WRONG.
994 We will get multicast token leakage, when IFF_MULTICAST
995 is changed. This check should be done in dev->set_multicast_list
996 routine. Something sort of:
997 if (dev->mc_list && dev->flags&IFF_MULTICAST) { do it; }
998 --ANK
999 */
1000 if (arp_mc_map(addr, buf, dev, 0) == 0)
1001 dev_mc_add(dev,buf,dev->addr_len,0);
1002}
1003
1004/*
1005 * Remove a filter from a device
1006 */
1007
1008static void ip_mc_filter_del(struct in_device *in_dev, u32 addr)
1009{
1010 char buf[MAX_ADDR_LEN];
1011 struct net_device *dev = in_dev->dev;
1012
1013 if (arp_mc_map(addr, buf, dev, 0) == 0)
1014 dev_mc_delete(dev,buf,dev->addr_len,0);
1015}
1016
1017#ifdef CONFIG_IP_MULTICAST
1018/*
1019 * deleted ip_mc_list manipulation
1020 */
1021static void igmpv3_add_delrec(struct in_device *in_dev, struct ip_mc_list *im)
1022{
1023 struct ip_mc_list *pmc;
1024
1025 /* this is an "ip_mc_list" for convenience; only the fields below
1026 * are actually used. In particular, the refcnt and users are not
1027 * used for management of the delete list. Using the same structure
1028 * for deleted items allows change reports to use common code with
1029 * non-deleted or query-response MCA's.
1030 */
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08001031 pmc = kmalloc(sizeof(*pmc), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001032 if (!pmc)
1033 return;
1034 memset(pmc, 0, sizeof(*pmc));
1035 spin_lock_bh(&im->lock);
1036 pmc->interface = im->interface;
1037 in_dev_hold(in_dev);
1038 pmc->multiaddr = im->multiaddr;
1039 pmc->crcount = in_dev->mr_qrv ? in_dev->mr_qrv :
1040 IGMP_Unsolicited_Report_Count;
1041 pmc->sfmode = im->sfmode;
1042 if (pmc->sfmode == MCAST_INCLUDE) {
1043 struct ip_sf_list *psf;
1044
1045 pmc->tomb = im->tomb;
1046 pmc->sources = im->sources;
1047 im->tomb = im->sources = NULL;
1048 for (psf=pmc->sources; psf; psf=psf->sf_next)
1049 psf->sf_crcount = pmc->crcount;
1050 }
1051 spin_unlock_bh(&im->lock);
1052
1053 spin_lock_bh(&in_dev->mc_tomb_lock);
1054 pmc->next = in_dev->mc_tomb;
1055 in_dev->mc_tomb = pmc;
1056 spin_unlock_bh(&in_dev->mc_tomb_lock);
1057}
1058
1059static void igmpv3_del_delrec(struct in_device *in_dev, __u32 multiaddr)
1060{
1061 struct ip_mc_list *pmc, *pmc_prev;
1062 struct ip_sf_list *psf, *psf_next;
1063
1064 spin_lock_bh(&in_dev->mc_tomb_lock);
1065 pmc_prev = NULL;
1066 for (pmc=in_dev->mc_tomb; pmc; pmc=pmc->next) {
1067 if (pmc->multiaddr == multiaddr)
1068 break;
1069 pmc_prev = pmc;
1070 }
1071 if (pmc) {
1072 if (pmc_prev)
1073 pmc_prev->next = pmc->next;
1074 else
1075 in_dev->mc_tomb = pmc->next;
1076 }
1077 spin_unlock_bh(&in_dev->mc_tomb_lock);
1078 if (pmc) {
1079 for (psf=pmc->tomb; psf; psf=psf_next) {
1080 psf_next = psf->sf_next;
1081 kfree(psf);
1082 }
1083 in_dev_put(pmc->interface);
1084 kfree(pmc);
1085 }
1086}
1087
1088static void igmpv3_clear_delrec(struct in_device *in_dev)
1089{
1090 struct ip_mc_list *pmc, *nextpmc;
1091
1092 spin_lock_bh(&in_dev->mc_tomb_lock);
1093 pmc = in_dev->mc_tomb;
1094 in_dev->mc_tomb = NULL;
1095 spin_unlock_bh(&in_dev->mc_tomb_lock);
1096
1097 for (; pmc; pmc = nextpmc) {
1098 nextpmc = pmc->next;
1099 ip_mc_clear_src(pmc);
1100 in_dev_put(pmc->interface);
1101 kfree(pmc);
1102 }
1103 /* clear dead sources, too */
1104 read_lock(&in_dev->mc_list_lock);
1105 for (pmc=in_dev->mc_list; pmc; pmc=pmc->next) {
1106 struct ip_sf_list *psf, *psf_next;
1107
1108 spin_lock_bh(&pmc->lock);
1109 psf = pmc->tomb;
1110 pmc->tomb = NULL;
1111 spin_unlock_bh(&pmc->lock);
1112 for (; psf; psf=psf_next) {
1113 psf_next = psf->sf_next;
1114 kfree(psf);
1115 }
1116 }
1117 read_unlock(&in_dev->mc_list_lock);
1118}
1119#endif
1120
1121static void igmp_group_dropped(struct ip_mc_list *im)
1122{
1123 struct in_device *in_dev = im->interface;
1124#ifdef CONFIG_IP_MULTICAST
1125 int reporter;
1126#endif
1127
1128 if (im->loaded) {
1129 im->loaded = 0;
1130 ip_mc_filter_del(in_dev, im->multiaddr);
1131 }
1132
1133#ifdef CONFIG_IP_MULTICAST
1134 if (im->multiaddr == IGMP_ALL_HOSTS)
1135 return;
1136
1137 reporter = im->reporter;
1138 igmp_stop_timer(im);
1139
1140 if (!in_dev->dead) {
1141 if (IGMP_V1_SEEN(in_dev))
1142 goto done;
1143 if (IGMP_V2_SEEN(in_dev)) {
1144 if (reporter)
1145 igmp_send_report(in_dev, im, IGMP_HOST_LEAVE_MESSAGE);
1146 goto done;
1147 }
1148 /* IGMPv3 */
1149 igmpv3_add_delrec(in_dev, im);
1150
1151 igmp_ifc_event(in_dev);
1152 }
1153done:
1154#endif
1155 ip_mc_clear_src(im);
1156}
1157
1158static void igmp_group_added(struct ip_mc_list *im)
1159{
1160 struct in_device *in_dev = im->interface;
1161
1162 if (im->loaded == 0) {
1163 im->loaded = 1;
1164 ip_mc_filter_add(in_dev, im->multiaddr);
1165 }
1166
1167#ifdef CONFIG_IP_MULTICAST
1168 if (im->multiaddr == IGMP_ALL_HOSTS)
1169 return;
1170
1171 if (in_dev->dead)
1172 return;
1173 if (IGMP_V1_SEEN(in_dev) || IGMP_V2_SEEN(in_dev)) {
1174 spin_lock_bh(&im->lock);
1175 igmp_start_timer(im, IGMP_Initial_Report_Delay);
1176 spin_unlock_bh(&im->lock);
1177 return;
1178 }
1179 /* else, v3 */
1180
1181 im->crcount = in_dev->mr_qrv ? in_dev->mr_qrv :
1182 IGMP_Unsolicited_Report_Count;
1183 igmp_ifc_event(in_dev);
1184#endif
1185}
1186
1187
1188/*
1189 * Multicast list managers
1190 */
1191
1192
1193/*
1194 * A socket has joined a multicast group on device dev.
1195 */
1196
1197void ip_mc_inc_group(struct in_device *in_dev, u32 addr)
1198{
1199 struct ip_mc_list *im;
1200
1201 ASSERT_RTNL();
1202
1203 for (im=in_dev->mc_list; im; im=im->next) {
1204 if (im->multiaddr == addr) {
1205 im->users++;
1206 ip_mc_add_src(in_dev, &addr, MCAST_EXCLUDE, 0, NULL, 0);
1207 goto out;
1208 }
1209 }
1210
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08001211 im = kmalloc(sizeof(*im), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001212 if (!im)
1213 goto out;
1214
1215 im->users=1;
1216 im->interface=in_dev;
1217 in_dev_hold(in_dev);
1218 im->multiaddr=addr;
1219 /* initial mode is (EX, empty) */
1220 im->sfmode = MCAST_EXCLUDE;
1221 im->sfcount[MCAST_INCLUDE] = 0;
1222 im->sfcount[MCAST_EXCLUDE] = 1;
1223 im->sources = NULL;
1224 im->tomb = NULL;
1225 im->crcount = 0;
1226 atomic_set(&im->refcnt, 1);
1227 spin_lock_init(&im->lock);
1228#ifdef CONFIG_IP_MULTICAST
1229 im->tm_running=0;
1230 init_timer(&im->timer);
1231 im->timer.data=(unsigned long)im;
1232 im->timer.function=&igmp_timer_expire;
1233 im->unsolicit_count = IGMP_Unsolicited_Report_Count;
1234 im->reporter = 0;
1235 im->gsquery = 0;
1236#endif
1237 im->loaded = 0;
1238 write_lock_bh(&in_dev->mc_list_lock);
1239 im->next=in_dev->mc_list;
1240 in_dev->mc_list=im;
1241 write_unlock_bh(&in_dev->mc_list_lock);
1242#ifdef CONFIG_IP_MULTICAST
1243 igmpv3_del_delrec(in_dev, im->multiaddr);
1244#endif
1245 igmp_group_added(im);
1246 if (!in_dev->dead)
1247 ip_rt_multicast_event(in_dev);
1248out:
1249 return;
1250}
1251
1252/*
1253 * A socket has left a multicast group on device dev
1254 */
1255
1256void ip_mc_dec_group(struct in_device *in_dev, u32 addr)
1257{
1258 struct ip_mc_list *i, **ip;
1259
1260 ASSERT_RTNL();
1261
1262 for (ip=&in_dev->mc_list; (i=*ip)!=NULL; ip=&i->next) {
1263 if (i->multiaddr==addr) {
1264 if (--i->users == 0) {
1265 write_lock_bh(&in_dev->mc_list_lock);
1266 *ip = i->next;
1267 write_unlock_bh(&in_dev->mc_list_lock);
1268 igmp_group_dropped(i);
1269
1270 if (!in_dev->dead)
1271 ip_rt_multicast_event(in_dev);
1272
1273 ip_ma_put(i);
1274 return;
1275 }
1276 break;
1277 }
1278 }
1279}
1280
1281/* Device going down */
1282
1283void ip_mc_down(struct in_device *in_dev)
1284{
1285 struct ip_mc_list *i;
1286
1287 ASSERT_RTNL();
1288
1289 for (i=in_dev->mc_list; i; i=i->next)
1290 igmp_group_dropped(i);
1291
1292#ifdef CONFIG_IP_MULTICAST
1293 in_dev->mr_ifc_count = 0;
1294 if (del_timer(&in_dev->mr_ifc_timer))
1295 __in_dev_put(in_dev);
1296 in_dev->mr_gq_running = 0;
1297 if (del_timer(&in_dev->mr_gq_timer))
1298 __in_dev_put(in_dev);
1299 igmpv3_clear_delrec(in_dev);
1300#endif
1301
1302 ip_mc_dec_group(in_dev, IGMP_ALL_HOSTS);
1303}
1304
1305void ip_mc_init_dev(struct in_device *in_dev)
1306{
1307 ASSERT_RTNL();
1308
1309 in_dev->mc_tomb = NULL;
1310#ifdef CONFIG_IP_MULTICAST
1311 in_dev->mr_gq_running = 0;
1312 init_timer(&in_dev->mr_gq_timer);
1313 in_dev->mr_gq_timer.data=(unsigned long) in_dev;
1314 in_dev->mr_gq_timer.function=&igmp_gq_timer_expire;
1315 in_dev->mr_ifc_count = 0;
1316 init_timer(&in_dev->mr_ifc_timer);
1317 in_dev->mr_ifc_timer.data=(unsigned long) in_dev;
1318 in_dev->mr_ifc_timer.function=&igmp_ifc_timer_expire;
1319 in_dev->mr_qrv = IGMP_Unsolicited_Report_Count;
1320#endif
1321
1322 rwlock_init(&in_dev->mc_list_lock);
1323 spin_lock_init(&in_dev->mc_tomb_lock);
1324}
1325
1326/* Device going up */
1327
1328void ip_mc_up(struct in_device *in_dev)
1329{
1330 struct ip_mc_list *i;
1331
1332 ASSERT_RTNL();
1333
1334 ip_mc_inc_group(in_dev, IGMP_ALL_HOSTS);
1335
1336 for (i=in_dev->mc_list; i; i=i->next)
1337 igmp_group_added(i);
1338}
1339
1340/*
1341 * Device is about to be destroyed: clean up.
1342 */
1343
1344void ip_mc_destroy_dev(struct in_device *in_dev)
1345{
1346 struct ip_mc_list *i;
1347
1348 ASSERT_RTNL();
1349
1350 /* Deactivate timers */
1351 ip_mc_down(in_dev);
1352
1353 write_lock_bh(&in_dev->mc_list_lock);
1354 while ((i = in_dev->mc_list) != NULL) {
1355 in_dev->mc_list = i->next;
1356 write_unlock_bh(&in_dev->mc_list_lock);
1357
1358 igmp_group_dropped(i);
1359 ip_ma_put(i);
1360
1361 write_lock_bh(&in_dev->mc_list_lock);
1362 }
1363 write_unlock_bh(&in_dev->mc_list_lock);
1364}
1365
1366static struct in_device * ip_mc_find_dev(struct ip_mreqn *imr)
1367{
1368 struct flowi fl = { .nl_u = { .ip4_u =
1369 { .daddr = imr->imr_multiaddr.s_addr } } };
1370 struct rtable *rt;
1371 struct net_device *dev = NULL;
1372 struct in_device *idev = NULL;
1373
1374 if (imr->imr_ifindex) {
1375 idev = inetdev_by_index(imr->imr_ifindex);
1376 if (idev)
1377 __in_dev_put(idev);
1378 return idev;
1379 }
1380 if (imr->imr_address.s_addr) {
1381 dev = ip_dev_find(imr->imr_address.s_addr);
1382 if (!dev)
1383 return NULL;
1384 __dev_put(dev);
1385 }
1386
1387 if (!dev && !ip_route_output_key(&rt, &fl)) {
1388 dev = rt->u.dst.dev;
1389 ip_rt_put(rt);
1390 }
1391 if (dev) {
1392 imr->imr_ifindex = dev->ifindex;
Herbert Xue5ed6392005-10-03 14:35:55 -07001393 idev = __in_dev_get_rtnl(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001394 }
1395 return idev;
1396}
1397
1398/*
1399 * Join a socket to a group
1400 */
1401int sysctl_igmp_max_memberships = IP_MAX_MEMBERSHIPS;
1402int sysctl_igmp_max_msf = IP_MAX_MSF;
1403
1404
1405static int ip_mc_del1_src(struct ip_mc_list *pmc, int sfmode,
1406 __u32 *psfsrc)
1407{
1408 struct ip_sf_list *psf, *psf_prev;
1409 int rv = 0;
1410
1411 psf_prev = NULL;
1412 for (psf=pmc->sources; psf; psf=psf->sf_next) {
1413 if (psf->sf_inaddr == *psfsrc)
1414 break;
1415 psf_prev = psf;
1416 }
1417 if (!psf || psf->sf_count[sfmode] == 0) {
1418 /* source filter not found, or count wrong => bug */
1419 return -ESRCH;
1420 }
1421 psf->sf_count[sfmode]--;
1422 if (psf->sf_count[sfmode] == 0) {
1423 ip_rt_multicast_event(pmc->interface);
1424 }
1425 if (!psf->sf_count[MCAST_INCLUDE] && !psf->sf_count[MCAST_EXCLUDE]) {
1426#ifdef CONFIG_IP_MULTICAST
1427 struct in_device *in_dev = pmc->interface;
1428#endif
1429
1430 /* no more filters for this source */
1431 if (psf_prev)
1432 psf_prev->sf_next = psf->sf_next;
1433 else
1434 pmc->sources = psf->sf_next;
1435#ifdef CONFIG_IP_MULTICAST
1436 if (psf->sf_oldin &&
1437 !IGMP_V1_SEEN(in_dev) && !IGMP_V2_SEEN(in_dev)) {
1438 psf->sf_crcount = in_dev->mr_qrv ? in_dev->mr_qrv :
1439 IGMP_Unsolicited_Report_Count;
1440 psf->sf_next = pmc->tomb;
1441 pmc->tomb = psf;
1442 rv = 1;
1443 } else
1444#endif
1445 kfree(psf);
1446 }
1447 return rv;
1448}
1449
1450#ifndef CONFIG_IP_MULTICAST
1451#define igmp_ifc_event(x) do { } while (0)
1452#endif
1453
1454static int ip_mc_del_src(struct in_device *in_dev, __u32 *pmca, int sfmode,
1455 int sfcount, __u32 *psfsrc, int delta)
1456{
1457 struct ip_mc_list *pmc;
1458 int changerec = 0;
1459 int i, err;
1460
1461 if (!in_dev)
1462 return -ENODEV;
1463 read_lock(&in_dev->mc_list_lock);
1464 for (pmc=in_dev->mc_list; pmc; pmc=pmc->next) {
1465 if (*pmca == pmc->multiaddr)
1466 break;
1467 }
1468 if (!pmc) {
1469 /* MCA not found?? bug */
1470 read_unlock(&in_dev->mc_list_lock);
1471 return -ESRCH;
1472 }
1473 spin_lock_bh(&pmc->lock);
1474 read_unlock(&in_dev->mc_list_lock);
1475#ifdef CONFIG_IP_MULTICAST
1476 sf_markstate(pmc);
1477#endif
1478 if (!delta) {
1479 err = -EINVAL;
1480 if (!pmc->sfcount[sfmode])
1481 goto out_unlock;
1482 pmc->sfcount[sfmode]--;
1483 }
1484 err = 0;
1485 for (i=0; i<sfcount; i++) {
1486 int rv = ip_mc_del1_src(pmc, sfmode, &psfsrc[i]);
1487
1488 changerec |= rv > 0;
1489 if (!err && rv < 0)
1490 err = rv;
1491 }
1492 if (pmc->sfmode == MCAST_EXCLUDE &&
1493 pmc->sfcount[MCAST_EXCLUDE] == 0 &&
1494 pmc->sfcount[MCAST_INCLUDE]) {
1495#ifdef CONFIG_IP_MULTICAST
1496 struct ip_sf_list *psf;
1497#endif
1498
1499 /* filter mode change */
1500 pmc->sfmode = MCAST_INCLUDE;
1501#ifdef CONFIG_IP_MULTICAST
1502 pmc->crcount = in_dev->mr_qrv ? in_dev->mr_qrv :
1503 IGMP_Unsolicited_Report_Count;
1504 in_dev->mr_ifc_count = pmc->crcount;
1505 for (psf=pmc->sources; psf; psf = psf->sf_next)
1506 psf->sf_crcount = 0;
1507 igmp_ifc_event(pmc->interface);
1508 } else if (sf_setstate(pmc) || changerec) {
1509 igmp_ifc_event(pmc->interface);
1510#endif
1511 }
1512out_unlock:
1513 spin_unlock_bh(&pmc->lock);
1514 return err;
1515}
1516
1517/*
1518 * Add multicast single-source filter to the interface list
1519 */
1520static int ip_mc_add1_src(struct ip_mc_list *pmc, int sfmode,
1521 __u32 *psfsrc, int delta)
1522{
1523 struct ip_sf_list *psf, *psf_prev;
1524
1525 psf_prev = NULL;
1526 for (psf=pmc->sources; psf; psf=psf->sf_next) {
1527 if (psf->sf_inaddr == *psfsrc)
1528 break;
1529 psf_prev = psf;
1530 }
1531 if (!psf) {
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08001532 psf = kmalloc(sizeof(*psf), GFP_ATOMIC);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001533 if (!psf)
1534 return -ENOBUFS;
1535 memset(psf, 0, sizeof(*psf));
1536 psf->sf_inaddr = *psfsrc;
1537 if (psf_prev) {
1538 psf_prev->sf_next = psf;
1539 } else
1540 pmc->sources = psf;
1541 }
1542 psf->sf_count[sfmode]++;
1543 if (psf->sf_count[sfmode] == 1) {
1544 ip_rt_multicast_event(pmc->interface);
1545 }
1546 return 0;
1547}
1548
1549#ifdef CONFIG_IP_MULTICAST
1550static void sf_markstate(struct ip_mc_list *pmc)
1551{
1552 struct ip_sf_list *psf;
1553 int mca_xcount = pmc->sfcount[MCAST_EXCLUDE];
1554
1555 for (psf=pmc->sources; psf; psf=psf->sf_next)
1556 if (pmc->sfcount[MCAST_EXCLUDE]) {
1557 psf->sf_oldin = mca_xcount ==
1558 psf->sf_count[MCAST_EXCLUDE] &&
1559 !psf->sf_count[MCAST_INCLUDE];
1560 } else
1561 psf->sf_oldin = psf->sf_count[MCAST_INCLUDE] != 0;
1562}
1563
1564static int sf_setstate(struct ip_mc_list *pmc)
1565{
David L Stevensad125832006-01-18 14:20:56 -08001566 struct ip_sf_list *psf, *dpsf;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001567 int mca_xcount = pmc->sfcount[MCAST_EXCLUDE];
1568 int qrv = pmc->interface->mr_qrv;
1569 int new_in, rv;
1570
1571 rv = 0;
1572 for (psf=pmc->sources; psf; psf=psf->sf_next) {
1573 if (pmc->sfcount[MCAST_EXCLUDE]) {
1574 new_in = mca_xcount == psf->sf_count[MCAST_EXCLUDE] &&
1575 !psf->sf_count[MCAST_INCLUDE];
1576 } else
1577 new_in = psf->sf_count[MCAST_INCLUDE] != 0;
David L Stevensad125832006-01-18 14:20:56 -08001578 if (new_in) {
1579 if (!psf->sf_oldin) {
1580 struct ip_sf_list *prev = 0;
1581
1582 for (dpsf=pmc->tomb; dpsf; dpsf=dpsf->sf_next) {
1583 if (dpsf->sf_inaddr == psf->sf_inaddr)
1584 break;
1585 prev = dpsf;
1586 }
1587 if (dpsf) {
1588 if (prev)
1589 prev->sf_next = dpsf->sf_next;
1590 else
1591 pmc->tomb = dpsf->sf_next;
1592 kfree(dpsf);
1593 }
1594 psf->sf_crcount = qrv;
1595 rv++;
1596 }
1597 } else if (psf->sf_oldin) {
1598
1599 psf->sf_crcount = 0;
1600 /*
1601 * add or update "delete" records if an active filter
1602 * is now inactive
1603 */
1604 for (dpsf=pmc->tomb; dpsf; dpsf=dpsf->sf_next)
1605 if (dpsf->sf_inaddr == psf->sf_inaddr)
1606 break;
1607 if (!dpsf) {
1608 dpsf = (struct ip_sf_list *)
1609 kmalloc(sizeof(*dpsf), GFP_ATOMIC);
1610 if (!dpsf)
1611 continue;
1612 *dpsf = *psf;
1613 /* pmc->lock held by callers */
1614 dpsf->sf_next = pmc->tomb;
1615 pmc->tomb = dpsf;
1616 }
1617 dpsf->sf_crcount = qrv;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001618 rv++;
1619 }
1620 }
1621 return rv;
1622}
1623#endif
1624
1625/*
1626 * Add multicast source filter list to the interface list
1627 */
1628static int ip_mc_add_src(struct in_device *in_dev, __u32 *pmca, int sfmode,
1629 int sfcount, __u32 *psfsrc, int delta)
1630{
1631 struct ip_mc_list *pmc;
1632 int isexclude;
1633 int i, err;
1634
1635 if (!in_dev)
1636 return -ENODEV;
1637 read_lock(&in_dev->mc_list_lock);
1638 for (pmc=in_dev->mc_list; pmc; pmc=pmc->next) {
1639 if (*pmca == pmc->multiaddr)
1640 break;
1641 }
1642 if (!pmc) {
1643 /* MCA not found?? bug */
1644 read_unlock(&in_dev->mc_list_lock);
1645 return -ESRCH;
1646 }
1647 spin_lock_bh(&pmc->lock);
1648 read_unlock(&in_dev->mc_list_lock);
1649
1650#ifdef CONFIG_IP_MULTICAST
1651 sf_markstate(pmc);
1652#endif
1653 isexclude = pmc->sfmode == MCAST_EXCLUDE;
1654 if (!delta)
1655 pmc->sfcount[sfmode]++;
1656 err = 0;
1657 for (i=0; i<sfcount; i++) {
1658 err = ip_mc_add1_src(pmc, sfmode, &psfsrc[i], delta);
1659 if (err)
1660 break;
1661 }
1662 if (err) {
1663 int j;
1664
1665 pmc->sfcount[sfmode]--;
1666 for (j=0; j<i; j++)
1667 (void) ip_mc_del1_src(pmc, sfmode, &psfsrc[i]);
1668 } else if (isexclude != (pmc->sfcount[MCAST_EXCLUDE] != 0)) {
1669#ifdef CONFIG_IP_MULTICAST
1670 struct in_device *in_dev = pmc->interface;
1671 struct ip_sf_list *psf;
1672#endif
1673
1674 /* filter mode change */
1675 if (pmc->sfcount[MCAST_EXCLUDE])
1676 pmc->sfmode = MCAST_EXCLUDE;
1677 else if (pmc->sfcount[MCAST_INCLUDE])
1678 pmc->sfmode = MCAST_INCLUDE;
1679#ifdef CONFIG_IP_MULTICAST
1680 /* else no filters; keep old mode for reports */
1681
1682 pmc->crcount = in_dev->mr_qrv ? in_dev->mr_qrv :
1683 IGMP_Unsolicited_Report_Count;
1684 in_dev->mr_ifc_count = pmc->crcount;
1685 for (psf=pmc->sources; psf; psf = psf->sf_next)
1686 psf->sf_crcount = 0;
1687 igmp_ifc_event(in_dev);
1688 } else if (sf_setstate(pmc)) {
1689 igmp_ifc_event(in_dev);
1690#endif
1691 }
1692 spin_unlock_bh(&pmc->lock);
1693 return err;
1694}
1695
1696static void ip_mc_clear_src(struct ip_mc_list *pmc)
1697{
1698 struct ip_sf_list *psf, *nextpsf;
1699
1700 for (psf=pmc->tomb; psf; psf=nextpsf) {
1701 nextpsf = psf->sf_next;
1702 kfree(psf);
1703 }
1704 pmc->tomb = NULL;
1705 for (psf=pmc->sources; psf; psf=nextpsf) {
1706 nextpsf = psf->sf_next;
1707 kfree(psf);
1708 }
1709 pmc->sources = NULL;
1710 pmc->sfmode = MCAST_EXCLUDE;
Denis Lukianovde9daad2005-09-14 20:53:42 -07001711 pmc->sfcount[MCAST_INCLUDE] = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001712 pmc->sfcount[MCAST_EXCLUDE] = 1;
1713}
1714
1715
1716/*
1717 * Join a multicast group
1718 */
1719int ip_mc_join_group(struct sock *sk , struct ip_mreqn *imr)
1720{
1721 int err;
1722 u32 addr = imr->imr_multiaddr.s_addr;
David L Stevensca9b9072005-07-08 17:38:07 -07001723 struct ip_mc_socklist *iml=NULL, *i;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001724 struct in_device *in_dev;
1725 struct inet_sock *inet = inet_sk(sk);
David L Stevensca9b9072005-07-08 17:38:07 -07001726 int ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001727 int count = 0;
1728
1729 if (!MULTICAST(addr))
1730 return -EINVAL;
1731
1732 rtnl_shlock();
1733
1734 in_dev = ip_mc_find_dev(imr);
1735
1736 if (!in_dev) {
1737 iml = NULL;
1738 err = -ENODEV;
1739 goto done;
1740 }
1741
Linus Torvalds1da177e2005-04-16 15:20:36 -07001742 err = -EADDRINUSE;
David L Stevensca9b9072005-07-08 17:38:07 -07001743 ifindex = imr->imr_ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001744 for (i = inet->mc_list; i; i = i->next) {
David L Stevensca9b9072005-07-08 17:38:07 -07001745 if (i->multi.imr_multiaddr.s_addr == addr &&
1746 i->multi.imr_ifindex == ifindex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001747 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001748 count++;
1749 }
1750 err = -ENOBUFS;
David L Stevensca9b9072005-07-08 17:38:07 -07001751 if (count >= sysctl_igmp_max_memberships)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001752 goto done;
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08001753 iml = sock_kmalloc(sk,sizeof(*iml),GFP_KERNEL);
David L Stevensca9b9072005-07-08 17:38:07 -07001754 if (iml == NULL)
1755 goto done;
1756
Linus Torvalds1da177e2005-04-16 15:20:36 -07001757 memcpy(&iml->multi, imr, sizeof(*imr));
1758 iml->next = inet->mc_list;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001759 iml->sflist = NULL;
1760 iml->sfmode = MCAST_EXCLUDE;
1761 inet->mc_list = iml;
1762 ip_mc_inc_group(in_dev, addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001763 err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001764done:
1765 rtnl_shunlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07001766 return err;
1767}
1768
1769static int ip_mc_leave_src(struct sock *sk, struct ip_mc_socklist *iml,
1770 struct in_device *in_dev)
1771{
1772 int err;
1773
1774 if (iml->sflist == 0) {
1775 /* any-source empty exclude case */
1776 return ip_mc_del_src(in_dev, &iml->multi.imr_multiaddr.s_addr,
1777 iml->sfmode, 0, NULL, 0);
1778 }
1779 err = ip_mc_del_src(in_dev, &iml->multi.imr_multiaddr.s_addr,
1780 iml->sfmode, iml->sflist->sl_count,
1781 iml->sflist->sl_addr, 0);
1782 sock_kfree_s(sk, iml->sflist, IP_SFLSIZE(iml->sflist->sl_max));
1783 iml->sflist = NULL;
1784 return err;
1785}
1786
1787/*
1788 * Ask a socket to leave a group.
1789 */
1790
1791int ip_mc_leave_group(struct sock *sk, struct ip_mreqn *imr)
1792{
1793 struct inet_sock *inet = inet_sk(sk);
1794 struct ip_mc_socklist *iml, **imlp;
David L Stevens84b42ba2005-07-08 17:48:38 -07001795 struct in_device *in_dev;
1796 u32 group = imr->imr_multiaddr.s_addr;
1797 u32 ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001798
1799 rtnl_lock();
David L Stevens84b42ba2005-07-08 17:48:38 -07001800 in_dev = ip_mc_find_dev(imr);
1801 if (!in_dev) {
1802 rtnl_unlock();
1803 return -ENODEV;
1804 }
1805 ifindex = imr->imr_ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001806 for (imlp = &inet->mc_list; (iml = *imlp) != NULL; imlp = &iml->next) {
David L Stevens84b42ba2005-07-08 17:48:38 -07001807 if (iml->multi.imr_multiaddr.s_addr == group &&
1808 iml->multi.imr_ifindex == ifindex) {
1809 (void) ip_mc_leave_src(sk, iml, in_dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001810
1811 *imlp = iml->next;
1812
David L Stevens84b42ba2005-07-08 17:48:38 -07001813 ip_mc_dec_group(in_dev, group);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001814 rtnl_unlock();
1815 sock_kfree_s(sk, iml, sizeof(*iml));
1816 return 0;
1817 }
1818 }
1819 rtnl_unlock();
1820 return -EADDRNOTAVAIL;
1821}
1822
1823int ip_mc_source(int add, int omode, struct sock *sk, struct
1824 ip_mreq_source *mreqs, int ifindex)
1825{
1826 int err;
1827 struct ip_mreqn imr;
1828 u32 addr = mreqs->imr_multiaddr;
1829 struct ip_mc_socklist *pmc;
1830 struct in_device *in_dev = NULL;
1831 struct inet_sock *inet = inet_sk(sk);
1832 struct ip_sf_socklist *psl;
David L Stevens8cdaaa12005-07-08 17:39:23 -07001833 int leavegroup = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001834 int i, j, rv;
1835
1836 if (!MULTICAST(addr))
1837 return -EINVAL;
1838
1839 rtnl_shlock();
1840
1841 imr.imr_multiaddr.s_addr = mreqs->imr_multiaddr;
1842 imr.imr_address.s_addr = mreqs->imr_interface;
1843 imr.imr_ifindex = ifindex;
1844 in_dev = ip_mc_find_dev(&imr);
1845
1846 if (!in_dev) {
1847 err = -ENODEV;
1848 goto done;
1849 }
1850 err = -EADDRNOTAVAIL;
1851
1852 for (pmc=inet->mc_list; pmc; pmc=pmc->next) {
David L Stevensca9b9072005-07-08 17:38:07 -07001853 if (pmc->multi.imr_multiaddr.s_addr == imr.imr_multiaddr.s_addr
1854 && pmc->multi.imr_ifindex == imr.imr_ifindex)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001855 break;
1856 }
David L Stevens917f2f12005-07-08 17:45:16 -07001857 if (!pmc) { /* must have a prior join */
1858 err = -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001859 goto done;
David L Stevens917f2f12005-07-08 17:45:16 -07001860 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001861 /* if a source filter was set, must be the same mode as before */
1862 if (pmc->sflist) {
David L Stevens917f2f12005-07-08 17:45:16 -07001863 if (pmc->sfmode != omode) {
1864 err = -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001865 goto done;
David L Stevens917f2f12005-07-08 17:45:16 -07001866 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001867 } else if (pmc->sfmode != omode) {
1868 /* allow mode switches for empty-set filters */
1869 ip_mc_add_src(in_dev, &mreqs->imr_multiaddr, omode, 0, NULL, 0);
1870 ip_mc_del_src(in_dev, &mreqs->imr_multiaddr, pmc->sfmode, 0,
1871 NULL, 0);
1872 pmc->sfmode = omode;
1873 }
1874
1875 psl = pmc->sflist;
1876 if (!add) {
1877 if (!psl)
David L Stevens917f2f12005-07-08 17:45:16 -07001878 goto done; /* err = -EADDRNOTAVAIL */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001879 rv = !0;
1880 for (i=0; i<psl->sl_count; i++) {
1881 rv = memcmp(&psl->sl_addr[i], &mreqs->imr_sourceaddr,
1882 sizeof(__u32));
1883 if (rv == 0)
1884 break;
1885 }
1886 if (rv) /* source not found */
David L Stevens917f2f12005-07-08 17:45:16 -07001887 goto done; /* err = -EADDRNOTAVAIL */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001888
David L Stevens8cdaaa12005-07-08 17:39:23 -07001889 /* special case - (INCLUDE, empty) == LEAVE_GROUP */
1890 if (psl->sl_count == 1 && omode == MCAST_INCLUDE) {
1891 leavegroup = 1;
1892 goto done;
1893 }
1894
Linus Torvalds1da177e2005-04-16 15:20:36 -07001895 /* update the interface filter */
1896 ip_mc_del_src(in_dev, &mreqs->imr_multiaddr, omode, 1,
1897 &mreqs->imr_sourceaddr, 1);
1898
1899 for (j=i+1; j<psl->sl_count; j++)
1900 psl->sl_addr[j-1] = psl->sl_addr[j];
1901 psl->sl_count--;
1902 err = 0;
1903 goto done;
1904 }
1905 /* else, add a new source to the filter */
1906
1907 if (psl && psl->sl_count >= sysctl_igmp_max_msf) {
1908 err = -ENOBUFS;
1909 goto done;
1910 }
1911 if (!psl || psl->sl_count == psl->sl_max) {
1912 struct ip_sf_socklist *newpsl;
1913 int count = IP_SFBLOCK;
1914
1915 if (psl)
1916 count += psl->sl_max;
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08001917 newpsl = sock_kmalloc(sk, IP_SFLSIZE(count), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001918 if (!newpsl) {
1919 err = -ENOBUFS;
1920 goto done;
1921 }
1922 newpsl->sl_max = count;
1923 newpsl->sl_count = count - IP_SFBLOCK;
1924 if (psl) {
1925 for (i=0; i<psl->sl_count; i++)
1926 newpsl->sl_addr[i] = psl->sl_addr[i];
1927 sock_kfree_s(sk, psl, IP_SFLSIZE(psl->sl_max));
1928 }
1929 pmc->sflist = psl = newpsl;
1930 }
1931 rv = 1; /* > 0 for insert logic below if sl_count is 0 */
1932 for (i=0; i<psl->sl_count; i++) {
1933 rv = memcmp(&psl->sl_addr[i], &mreqs->imr_sourceaddr,
1934 sizeof(__u32));
1935 if (rv == 0)
1936 break;
1937 }
1938 if (rv == 0) /* address already there is an error */
1939 goto done;
1940 for (j=psl->sl_count-1; j>=i; j--)
1941 psl->sl_addr[j+1] = psl->sl_addr[j];
1942 psl->sl_addr[i] = mreqs->imr_sourceaddr;
1943 psl->sl_count++;
1944 err = 0;
1945 /* update the interface list */
1946 ip_mc_add_src(in_dev, &mreqs->imr_multiaddr, omode, 1,
1947 &mreqs->imr_sourceaddr, 1);
1948done:
1949 rtnl_shunlock();
David L Stevens8cdaaa12005-07-08 17:39:23 -07001950 if (leavegroup)
1951 return ip_mc_leave_group(sk, &imr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001952 return err;
1953}
1954
1955int ip_mc_msfilter(struct sock *sk, struct ip_msfilter *msf, int ifindex)
1956{
David L Stevens9951f032005-07-08 17:47:28 -07001957 int err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001958 struct ip_mreqn imr;
1959 u32 addr = msf->imsf_multiaddr;
1960 struct ip_mc_socklist *pmc;
1961 struct in_device *in_dev;
1962 struct inet_sock *inet = inet_sk(sk);
1963 struct ip_sf_socklist *newpsl, *psl;
David L Stevens9951f032005-07-08 17:47:28 -07001964 int leavegroup = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001965
1966 if (!MULTICAST(addr))
1967 return -EINVAL;
1968 if (msf->imsf_fmode != MCAST_INCLUDE &&
1969 msf->imsf_fmode != MCAST_EXCLUDE)
1970 return -EINVAL;
1971
1972 rtnl_shlock();
1973
1974 imr.imr_multiaddr.s_addr = msf->imsf_multiaddr;
1975 imr.imr_address.s_addr = msf->imsf_interface;
1976 imr.imr_ifindex = ifindex;
1977 in_dev = ip_mc_find_dev(&imr);
1978
1979 if (!in_dev) {
1980 err = -ENODEV;
1981 goto done;
1982 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001983
David L Stevens9951f032005-07-08 17:47:28 -07001984 /* special case - (INCLUDE, empty) == LEAVE_GROUP */
1985 if (msf->imsf_fmode == MCAST_INCLUDE && msf->imsf_numsrc == 0) {
1986 leavegroup = 1;
1987 goto done;
1988 }
1989
Linus Torvalds1da177e2005-04-16 15:20:36 -07001990 for (pmc=inet->mc_list; pmc; pmc=pmc->next) {
1991 if (pmc->multi.imr_multiaddr.s_addr == msf->imsf_multiaddr &&
1992 pmc->multi.imr_ifindex == imr.imr_ifindex)
1993 break;
1994 }
David L Stevens917f2f12005-07-08 17:45:16 -07001995 if (!pmc) { /* must have a prior join */
1996 err = -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001997 goto done;
David L Stevens917f2f12005-07-08 17:45:16 -07001998 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001999 if (msf->imsf_numsrc) {
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08002000 newpsl = sock_kmalloc(sk, IP_SFLSIZE(msf->imsf_numsrc),
2001 GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002002 if (!newpsl) {
2003 err = -ENOBUFS;
2004 goto done;
2005 }
2006 newpsl->sl_max = newpsl->sl_count = msf->imsf_numsrc;
2007 memcpy(newpsl->sl_addr, msf->imsf_slist,
2008 msf->imsf_numsrc * sizeof(msf->imsf_slist[0]));
2009 err = ip_mc_add_src(in_dev, &msf->imsf_multiaddr,
2010 msf->imsf_fmode, newpsl->sl_count, newpsl->sl_addr, 0);
2011 if (err) {
2012 sock_kfree_s(sk, newpsl, IP_SFLSIZE(newpsl->sl_max));
2013 goto done;
2014 }
Yan Zheng8713dbf2005-10-28 08:02:08 +08002015 } else {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002016 newpsl = NULL;
Yan Zheng8713dbf2005-10-28 08:02:08 +08002017 (void) ip_mc_add_src(in_dev, &msf->imsf_multiaddr,
2018 msf->imsf_fmode, 0, NULL, 0);
2019 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002020 psl = pmc->sflist;
2021 if (psl) {
2022 (void) ip_mc_del_src(in_dev, &msf->imsf_multiaddr, pmc->sfmode,
2023 psl->sl_count, psl->sl_addr, 0);
2024 sock_kfree_s(sk, psl, IP_SFLSIZE(psl->sl_max));
2025 } else
2026 (void) ip_mc_del_src(in_dev, &msf->imsf_multiaddr, pmc->sfmode,
2027 0, NULL, 0);
2028 pmc->sflist = newpsl;
2029 pmc->sfmode = msf->imsf_fmode;
David L Stevens917f2f12005-07-08 17:45:16 -07002030 err = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002031done:
2032 rtnl_shunlock();
David L Stevens9951f032005-07-08 17:47:28 -07002033 if (leavegroup)
2034 err = ip_mc_leave_group(sk, &imr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002035 return err;
2036}
2037
2038int ip_mc_msfget(struct sock *sk, struct ip_msfilter *msf,
2039 struct ip_msfilter __user *optval, int __user *optlen)
2040{
2041 int err, len, count, copycount;
2042 struct ip_mreqn imr;
2043 u32 addr = msf->imsf_multiaddr;
2044 struct ip_mc_socklist *pmc;
2045 struct in_device *in_dev;
2046 struct inet_sock *inet = inet_sk(sk);
2047 struct ip_sf_socklist *psl;
2048
2049 if (!MULTICAST(addr))
2050 return -EINVAL;
2051
2052 rtnl_shlock();
2053
2054 imr.imr_multiaddr.s_addr = msf->imsf_multiaddr;
2055 imr.imr_address.s_addr = msf->imsf_interface;
2056 imr.imr_ifindex = 0;
2057 in_dev = ip_mc_find_dev(&imr);
2058
2059 if (!in_dev) {
2060 err = -ENODEV;
2061 goto done;
2062 }
2063 err = -EADDRNOTAVAIL;
2064
2065 for (pmc=inet->mc_list; pmc; pmc=pmc->next) {
2066 if (pmc->multi.imr_multiaddr.s_addr == msf->imsf_multiaddr &&
2067 pmc->multi.imr_ifindex == imr.imr_ifindex)
2068 break;
2069 }
2070 if (!pmc) /* must have a prior join */
2071 goto done;
2072 msf->imsf_fmode = pmc->sfmode;
2073 psl = pmc->sflist;
2074 rtnl_shunlock();
2075 if (!psl) {
2076 len = 0;
2077 count = 0;
2078 } else {
2079 count = psl->sl_count;
2080 }
2081 copycount = count < msf->imsf_numsrc ? count : msf->imsf_numsrc;
2082 len = copycount * sizeof(psl->sl_addr[0]);
2083 msf->imsf_numsrc = count;
2084 if (put_user(IP_MSFILTER_SIZE(copycount), optlen) ||
2085 copy_to_user(optval, msf, IP_MSFILTER_SIZE(0))) {
2086 return -EFAULT;
2087 }
2088 if (len &&
2089 copy_to_user(&optval->imsf_slist[0], psl->sl_addr, len))
2090 return -EFAULT;
2091 return 0;
2092done:
2093 rtnl_shunlock();
2094 return err;
2095}
2096
2097int ip_mc_gsfget(struct sock *sk, struct group_filter *gsf,
2098 struct group_filter __user *optval, int __user *optlen)
2099{
2100 int err, i, count, copycount;
2101 struct sockaddr_in *psin;
2102 u32 addr;
2103 struct ip_mc_socklist *pmc;
2104 struct inet_sock *inet = inet_sk(sk);
2105 struct ip_sf_socklist *psl;
2106
2107 psin = (struct sockaddr_in *)&gsf->gf_group;
2108 if (psin->sin_family != AF_INET)
2109 return -EINVAL;
2110 addr = psin->sin_addr.s_addr;
2111 if (!MULTICAST(addr))
2112 return -EINVAL;
2113
2114 rtnl_shlock();
2115
2116 err = -EADDRNOTAVAIL;
2117
2118 for (pmc=inet->mc_list; pmc; pmc=pmc->next) {
2119 if (pmc->multi.imr_multiaddr.s_addr == addr &&
2120 pmc->multi.imr_ifindex == gsf->gf_interface)
2121 break;
2122 }
2123 if (!pmc) /* must have a prior join */
2124 goto done;
2125 gsf->gf_fmode = pmc->sfmode;
2126 psl = pmc->sflist;
2127 rtnl_shunlock();
2128 count = psl ? psl->sl_count : 0;
2129 copycount = count < gsf->gf_numsrc ? count : gsf->gf_numsrc;
2130 gsf->gf_numsrc = count;
2131 if (put_user(GROUP_FILTER_SIZE(copycount), optlen) ||
2132 copy_to_user(optval, gsf, GROUP_FILTER_SIZE(0))) {
2133 return -EFAULT;
2134 }
2135 for (i=0; i<copycount; i++) {
2136 struct sockaddr_in *psin;
2137 struct sockaddr_storage ss;
2138
2139 psin = (struct sockaddr_in *)&ss;
2140 memset(&ss, 0, sizeof(ss));
2141 psin->sin_family = AF_INET;
2142 psin->sin_addr.s_addr = psl->sl_addr[i];
2143 if (copy_to_user(&optval->gf_slist[i], &ss, sizeof(ss)))
2144 return -EFAULT;
2145 }
2146 return 0;
2147done:
2148 rtnl_shunlock();
2149 return err;
2150}
2151
2152/*
2153 * check if a multicast source filter allows delivery for a given <src,dst,intf>
2154 */
2155int ip_mc_sf_allow(struct sock *sk, u32 loc_addr, u32 rmt_addr, int dif)
2156{
2157 struct inet_sock *inet = inet_sk(sk);
2158 struct ip_mc_socklist *pmc;
2159 struct ip_sf_socklist *psl;
2160 int i;
2161
2162 if (!MULTICAST(loc_addr))
2163 return 1;
2164
2165 for (pmc=inet->mc_list; pmc; pmc=pmc->next) {
2166 if (pmc->multi.imr_multiaddr.s_addr == loc_addr &&
2167 pmc->multi.imr_ifindex == dif)
2168 break;
2169 }
2170 if (!pmc)
2171 return 1;
2172 psl = pmc->sflist;
2173 if (!psl)
2174 return pmc->sfmode == MCAST_EXCLUDE;
2175
2176 for (i=0; i<psl->sl_count; i++) {
2177 if (psl->sl_addr[i] == rmt_addr)
2178 break;
2179 }
2180 if (pmc->sfmode == MCAST_INCLUDE && i >= psl->sl_count)
2181 return 0;
2182 if (pmc->sfmode == MCAST_EXCLUDE && i < psl->sl_count)
2183 return 0;
2184 return 1;
2185}
2186
2187/*
2188 * A socket is closing.
2189 */
2190
2191void ip_mc_drop_socket(struct sock *sk)
2192{
2193 struct inet_sock *inet = inet_sk(sk);
2194 struct ip_mc_socklist *iml;
2195
2196 if (inet->mc_list == NULL)
2197 return;
2198
2199 rtnl_lock();
2200 while ((iml = inet->mc_list) != NULL) {
2201 struct in_device *in_dev;
2202 inet->mc_list = iml->next;
2203
2204 if ((in_dev = inetdev_by_index(iml->multi.imr_ifindex)) != NULL) {
2205 (void) ip_mc_leave_src(sk, iml, in_dev);
2206 ip_mc_dec_group(in_dev, iml->multi.imr_multiaddr.s_addr);
2207 in_dev_put(in_dev);
2208 }
2209 sock_kfree_s(sk, iml, sizeof(*iml));
2210
2211 }
2212 rtnl_unlock();
2213}
2214
2215int ip_check_mc(struct in_device *in_dev, u32 mc_addr, u32 src_addr, u16 proto)
2216{
2217 struct ip_mc_list *im;
2218 struct ip_sf_list *psf;
2219 int rv = 0;
2220
2221 read_lock(&in_dev->mc_list_lock);
2222 for (im=in_dev->mc_list; im; im=im->next) {
2223 if (im->multiaddr == mc_addr)
2224 break;
2225 }
2226 if (im && proto == IPPROTO_IGMP) {
2227 rv = 1;
2228 } else if (im) {
2229 if (src_addr) {
2230 for (psf=im->sources; psf; psf=psf->sf_next) {
2231 if (psf->sf_inaddr == src_addr)
2232 break;
2233 }
2234 if (psf)
2235 rv = psf->sf_count[MCAST_INCLUDE] ||
2236 psf->sf_count[MCAST_EXCLUDE] !=
2237 im->sfcount[MCAST_EXCLUDE];
2238 else
2239 rv = im->sfcount[MCAST_EXCLUDE] != 0;
2240 } else
2241 rv = 1; /* unspecified source; tentatively allow */
2242 }
2243 read_unlock(&in_dev->mc_list_lock);
2244 return rv;
2245}
2246
2247#if defined(CONFIG_PROC_FS)
2248struct igmp_mc_iter_state {
2249 struct net_device *dev;
2250 struct in_device *in_dev;
2251};
2252
2253#define igmp_mc_seq_private(seq) ((struct igmp_mc_iter_state *)(seq)->private)
2254
2255static inline struct ip_mc_list *igmp_mc_get_first(struct seq_file *seq)
2256{
2257 struct ip_mc_list *im = NULL;
2258 struct igmp_mc_iter_state *state = igmp_mc_seq_private(seq);
2259
2260 for (state->dev = dev_base, state->in_dev = NULL;
2261 state->dev;
2262 state->dev = state->dev->next) {
2263 struct in_device *in_dev;
2264 in_dev = in_dev_get(state->dev);
2265 if (!in_dev)
2266 continue;
2267 read_lock(&in_dev->mc_list_lock);
2268 im = in_dev->mc_list;
2269 if (im) {
2270 state->in_dev = in_dev;
2271 break;
2272 }
2273 read_unlock(&in_dev->mc_list_lock);
2274 in_dev_put(in_dev);
2275 }
2276 return im;
2277}
2278
2279static struct ip_mc_list *igmp_mc_get_next(struct seq_file *seq, struct ip_mc_list *im)
2280{
2281 struct igmp_mc_iter_state *state = igmp_mc_seq_private(seq);
2282 im = im->next;
2283 while (!im) {
2284 if (likely(state->in_dev != NULL)) {
2285 read_unlock(&state->in_dev->mc_list_lock);
2286 in_dev_put(state->in_dev);
2287 }
2288 state->dev = state->dev->next;
2289 if (!state->dev) {
2290 state->in_dev = NULL;
2291 break;
2292 }
2293 state->in_dev = in_dev_get(state->dev);
2294 if (!state->in_dev)
2295 continue;
2296 read_lock(&state->in_dev->mc_list_lock);
2297 im = state->in_dev->mc_list;
2298 }
2299 return im;
2300}
2301
2302static struct ip_mc_list *igmp_mc_get_idx(struct seq_file *seq, loff_t pos)
2303{
2304 struct ip_mc_list *im = igmp_mc_get_first(seq);
2305 if (im)
2306 while (pos && (im = igmp_mc_get_next(seq, im)) != NULL)
2307 --pos;
2308 return pos ? NULL : im;
2309}
2310
2311static void *igmp_mc_seq_start(struct seq_file *seq, loff_t *pos)
2312{
2313 read_lock(&dev_base_lock);
2314 return *pos ? igmp_mc_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
2315}
2316
2317static void *igmp_mc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2318{
2319 struct ip_mc_list *im;
2320 if (v == SEQ_START_TOKEN)
2321 im = igmp_mc_get_first(seq);
2322 else
2323 im = igmp_mc_get_next(seq, v);
2324 ++*pos;
2325 return im;
2326}
2327
2328static void igmp_mc_seq_stop(struct seq_file *seq, void *v)
2329{
2330 struct igmp_mc_iter_state *state = igmp_mc_seq_private(seq);
2331 if (likely(state->in_dev != NULL)) {
2332 read_unlock(&state->in_dev->mc_list_lock);
2333 in_dev_put(state->in_dev);
2334 state->in_dev = NULL;
2335 }
2336 state->dev = NULL;
2337 read_unlock(&dev_base_lock);
2338}
2339
2340static int igmp_mc_seq_show(struct seq_file *seq, void *v)
2341{
2342 if (v == SEQ_START_TOKEN)
2343 seq_puts(seq,
2344 "Idx\tDevice : Count Querier\tGroup Users Timer\tReporter\n");
2345 else {
2346 struct ip_mc_list *im = (struct ip_mc_list *)v;
2347 struct igmp_mc_iter_state *state = igmp_mc_seq_private(seq);
2348 char *querier;
2349#ifdef CONFIG_IP_MULTICAST
2350 querier = IGMP_V1_SEEN(state->in_dev) ? "V1" :
2351 IGMP_V2_SEEN(state->in_dev) ? "V2" :
2352 "V3";
2353#else
2354 querier = "NONE";
2355#endif
2356
2357 if (state->in_dev->mc_list == im) {
2358 seq_printf(seq, "%d\t%-10s: %5d %7s\n",
2359 state->dev->ifindex, state->dev->name, state->dev->mc_count, querier);
2360 }
2361
2362 seq_printf(seq,
2363 "\t\t\t\t%08lX %5d %d:%08lX\t\t%d\n",
2364 im->multiaddr, im->users,
2365 im->tm_running, im->tm_running ?
2366 jiffies_to_clock_t(im->timer.expires-jiffies) : 0,
2367 im->reporter);
2368 }
2369 return 0;
2370}
2371
2372static struct seq_operations igmp_mc_seq_ops = {
2373 .start = igmp_mc_seq_start,
2374 .next = igmp_mc_seq_next,
2375 .stop = igmp_mc_seq_stop,
2376 .show = igmp_mc_seq_show,
2377};
2378
2379static int igmp_mc_seq_open(struct inode *inode, struct file *file)
2380{
2381 struct seq_file *seq;
2382 int rc = -ENOMEM;
2383 struct igmp_mc_iter_state *s = kmalloc(sizeof(*s), GFP_KERNEL);
2384
2385 if (!s)
2386 goto out;
2387 rc = seq_open(file, &igmp_mc_seq_ops);
2388 if (rc)
2389 goto out_kfree;
2390
2391 seq = file->private_data;
2392 seq->private = s;
2393 memset(s, 0, sizeof(*s));
2394out:
2395 return rc;
2396out_kfree:
2397 kfree(s);
2398 goto out;
2399}
2400
2401static struct file_operations igmp_mc_seq_fops = {
2402 .owner = THIS_MODULE,
2403 .open = igmp_mc_seq_open,
2404 .read = seq_read,
2405 .llseek = seq_lseek,
2406 .release = seq_release_private,
2407};
2408
2409struct igmp_mcf_iter_state {
2410 struct net_device *dev;
2411 struct in_device *idev;
2412 struct ip_mc_list *im;
2413};
2414
2415#define igmp_mcf_seq_private(seq) ((struct igmp_mcf_iter_state *)(seq)->private)
2416
2417static inline struct ip_sf_list *igmp_mcf_get_first(struct seq_file *seq)
2418{
2419 struct ip_sf_list *psf = NULL;
2420 struct ip_mc_list *im = NULL;
2421 struct igmp_mcf_iter_state *state = igmp_mcf_seq_private(seq);
2422
2423 for (state->dev = dev_base, state->idev = NULL, state->im = NULL;
2424 state->dev;
2425 state->dev = state->dev->next) {
2426 struct in_device *idev;
2427 idev = in_dev_get(state->dev);
2428 if (unlikely(idev == NULL))
2429 continue;
2430 read_lock(&idev->mc_list_lock);
2431 im = idev->mc_list;
2432 if (likely(im != NULL)) {
2433 spin_lock_bh(&im->lock);
2434 psf = im->sources;
2435 if (likely(psf != NULL)) {
2436 state->im = im;
2437 state->idev = idev;
2438 break;
2439 }
2440 spin_unlock_bh(&im->lock);
2441 }
2442 read_unlock(&idev->mc_list_lock);
2443 in_dev_put(idev);
2444 }
2445 return psf;
2446}
2447
2448static struct ip_sf_list *igmp_mcf_get_next(struct seq_file *seq, struct ip_sf_list *psf)
2449{
2450 struct igmp_mcf_iter_state *state = igmp_mcf_seq_private(seq);
2451
2452 psf = psf->sf_next;
2453 while (!psf) {
2454 spin_unlock_bh(&state->im->lock);
2455 state->im = state->im->next;
2456 while (!state->im) {
2457 if (likely(state->idev != NULL)) {
2458 read_unlock(&state->idev->mc_list_lock);
2459 in_dev_put(state->idev);
2460 }
2461 state->dev = state->dev->next;
2462 if (!state->dev) {
2463 state->idev = NULL;
2464 goto out;
2465 }
2466 state->idev = in_dev_get(state->dev);
2467 if (!state->idev)
2468 continue;
2469 read_lock(&state->idev->mc_list_lock);
2470 state->im = state->idev->mc_list;
2471 }
2472 if (!state->im)
2473 break;
2474 spin_lock_bh(&state->im->lock);
2475 psf = state->im->sources;
2476 }
2477out:
2478 return psf;
2479}
2480
2481static struct ip_sf_list *igmp_mcf_get_idx(struct seq_file *seq, loff_t pos)
2482{
2483 struct ip_sf_list *psf = igmp_mcf_get_first(seq);
2484 if (psf)
2485 while (pos && (psf = igmp_mcf_get_next(seq, psf)) != NULL)
2486 --pos;
2487 return pos ? NULL : psf;
2488}
2489
2490static void *igmp_mcf_seq_start(struct seq_file *seq, loff_t *pos)
2491{
2492 read_lock(&dev_base_lock);
2493 return *pos ? igmp_mcf_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
2494}
2495
2496static void *igmp_mcf_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2497{
2498 struct ip_sf_list *psf;
2499 if (v == SEQ_START_TOKEN)
2500 psf = igmp_mcf_get_first(seq);
2501 else
2502 psf = igmp_mcf_get_next(seq, v);
2503 ++*pos;
2504 return psf;
2505}
2506
2507static void igmp_mcf_seq_stop(struct seq_file *seq, void *v)
2508{
2509 struct igmp_mcf_iter_state *state = igmp_mcf_seq_private(seq);
2510 if (likely(state->im != NULL)) {
2511 spin_unlock_bh(&state->im->lock);
2512 state->im = NULL;
2513 }
2514 if (likely(state->idev != NULL)) {
2515 read_unlock(&state->idev->mc_list_lock);
2516 in_dev_put(state->idev);
2517 state->idev = NULL;
2518 }
2519 state->dev = NULL;
2520 read_unlock(&dev_base_lock);
2521}
2522
2523static int igmp_mcf_seq_show(struct seq_file *seq, void *v)
2524{
2525 struct ip_sf_list *psf = (struct ip_sf_list *)v;
2526 struct igmp_mcf_iter_state *state = igmp_mcf_seq_private(seq);
2527
2528 if (v == SEQ_START_TOKEN) {
2529 seq_printf(seq,
2530 "%3s %6s "
2531 "%10s %10s %6s %6s\n", "Idx",
2532 "Device", "MCA",
2533 "SRC", "INC", "EXC");
2534 } else {
2535 seq_printf(seq,
2536 "%3d %6.6s 0x%08x "
2537 "0x%08x %6lu %6lu\n",
2538 state->dev->ifindex, state->dev->name,
2539 ntohl(state->im->multiaddr),
2540 ntohl(psf->sf_inaddr),
2541 psf->sf_count[MCAST_INCLUDE],
2542 psf->sf_count[MCAST_EXCLUDE]);
2543 }
2544 return 0;
2545}
2546
2547static struct seq_operations igmp_mcf_seq_ops = {
2548 .start = igmp_mcf_seq_start,
2549 .next = igmp_mcf_seq_next,
2550 .stop = igmp_mcf_seq_stop,
2551 .show = igmp_mcf_seq_show,
2552};
2553
2554static int igmp_mcf_seq_open(struct inode *inode, struct file *file)
2555{
2556 struct seq_file *seq;
2557 int rc = -ENOMEM;
2558 struct igmp_mcf_iter_state *s = kmalloc(sizeof(*s), GFP_KERNEL);
2559
2560 if (!s)
2561 goto out;
2562 rc = seq_open(file, &igmp_mcf_seq_ops);
2563 if (rc)
2564 goto out_kfree;
2565
2566 seq = file->private_data;
2567 seq->private = s;
2568 memset(s, 0, sizeof(*s));
2569out:
2570 return rc;
2571out_kfree:
2572 kfree(s);
2573 goto out;
2574}
2575
2576static struct file_operations igmp_mcf_seq_fops = {
2577 .owner = THIS_MODULE,
2578 .open = igmp_mcf_seq_open,
2579 .read = seq_read,
2580 .llseek = seq_lseek,
2581 .release = seq_release_private,
2582};
2583
2584int __init igmp_mc_proc_init(void)
2585{
2586 proc_net_fops_create("igmp", S_IRUGO, &igmp_mc_seq_fops);
2587 proc_net_fops_create("mcfilter", S_IRUGO, &igmp_mcf_seq_fops);
2588 return 0;
2589}
2590#endif
2591
2592EXPORT_SYMBOL(ip_mc_dec_group);
2593EXPORT_SYMBOL(ip_mc_inc_group);
2594EXPORT_SYMBOL(ip_mc_join_group);