igmp.c
上传用户:jlfgdled
上传日期:2013-04-10
资源大小:33168k
文件大小:19k
源码类别:

Linux/Unix编程

开发平台:

Unix_Linux

  1. /*
  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.46 2001/07/27 09:27:29 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 initializedn"
  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.  */
  72. #include <linux/config.h>
  73. #include <asm/uaccess.h>
  74. #include <asm/system.h>
  75. #include <linux/types.h>
  76. #include <linux/kernel.h>
  77. #include <linux/sched.h>
  78. #include <linux/string.h>
  79. #include <linux/socket.h>
  80. #include <linux/sockios.h>
  81. #include <linux/in.h>
  82. #include <linux/inet.h>
  83. #include <linux/netdevice.h>
  84. #include <linux/skbuff.h>
  85. #include <linux/inetdevice.h>
  86. #include <linux/igmp.h>
  87. #include <linux/if_arp.h>
  88. #include <linux/rtnetlink.h>
  89. #include <net/ip.h>
  90. #include <net/protocol.h>
  91. #include <net/route.h>
  92. #include <net/sock.h>
  93. #include <net/checksum.h>
  94. #include <linux/netfilter_ipv4.h>
  95. #ifdef CONFIG_IP_MROUTE
  96. #include <linux/mroute.h>
  97. #endif
  98. #define IP_MAX_MEMBERSHIPS 20
  99. #ifdef CONFIG_IP_MULTICAST
  100. /* Parameter names and values are taken from igmp-v2-06 draft */
  101. #define IGMP_V1_Router_Present_Timeout (400*HZ)
  102. #define IGMP_Unsolicited_Report_Interval (10*HZ)
  103. #define IGMP_Query_Response_Interval (10*HZ)
  104. #define IGMP_Unsolicited_Report_Count 2
  105. #define IGMP_Initial_Report_Delay (1)
  106. /* IGMP_Initial_Report_Delay is not from IGMP specs!
  107.  * IGMP specs require to report membership immediately after
  108.  * joining a group, but we delay the first report by a
  109.  * small interval. It seems more natural and still does not
  110.  * contradict to specs provided this delay is small enough.
  111.  */
  112. #define IGMP_V1_SEEN(in_dev) ((in_dev)->mr_v1_seen && (long)(jiffies - (in_dev)->mr_v1_seen) < 0)
  113. #endif
  114. static void ip_ma_put(struct ip_mc_list *im)
  115. {
  116. if (atomic_dec_and_test(&im->refcnt)) {
  117. in_dev_put(im->interface);
  118. kfree(im);
  119. }
  120. }
  121. #ifdef CONFIG_IP_MULTICAST
  122. /*
  123.  * Timer management
  124.  */
  125. static __inline__ void igmp_stop_timer(struct ip_mc_list *im)
  126. {
  127. spin_lock_bh(&im->lock);
  128. if (del_timer(&im->timer))
  129. atomic_dec(&im->refcnt);
  130. im->tm_running=0;
  131. im->reporter = 0;
  132. im->unsolicit_count = 0;
  133. spin_unlock_bh(&im->lock);
  134. }
  135. /* It must be called with locked im->lock */
  136. static void igmp_start_timer(struct ip_mc_list *im, int max_delay)
  137. {
  138. int tv=net_random() % max_delay;
  139. im->tm_running=1;
  140. if (!mod_timer(&im->timer, jiffies+tv+2))
  141. atomic_inc(&im->refcnt);
  142. }
  143. static void igmp_mod_timer(struct ip_mc_list *im, int max_delay)
  144. {
  145. spin_lock_bh(&im->lock);
  146. im->unsolicit_count = 0;
  147. if (del_timer(&im->timer)) {
  148. if ((long)(im->timer.expires-jiffies) < max_delay) {
  149. add_timer(&im->timer);
  150. im->tm_running=1;
  151. spin_unlock_bh(&im->lock);
  152. return;
  153. }
  154. atomic_dec(&im->refcnt);
  155. }
  156. igmp_start_timer(im, max_delay);
  157. spin_unlock_bh(&im->lock);
  158. }
  159. /*
  160.  * Send an IGMP report.
  161.  */
  162. #define IGMP_SIZE (sizeof(struct igmphdr)+sizeof(struct iphdr)+4)
  163. /* Don't just hand NF_HOOK skb->dst->output, in case netfilter hook
  164.    changes route */
  165. static inline int
  166. output_maybe_reroute(struct sk_buff *skb)
  167. {
  168. return skb->dst->output(skb);
  169. }
  170. static int igmp_send_report(struct net_device *dev, u32 group, int type)
  171. {
  172. struct sk_buff *skb;
  173. struct iphdr *iph;
  174. struct igmphdr *ih;
  175. struct rtable *rt;
  176. u32 dst;
  177. /* According to IGMPv2 specs, LEAVE messages are
  178.  * sent to all-routers group.
  179.  */
  180. dst = group;
  181. if (type == IGMP_HOST_LEAVE_MESSAGE)
  182. dst = IGMP_ALL_ROUTER;
  183. if (ip_route_output(&rt, dst, 0, 0, dev->ifindex))
  184. return -1;
  185. if (rt->rt_src == 0) {
  186. ip_rt_put(rt);
  187. return -1;
  188. }
  189. skb=alloc_skb(IGMP_SIZE+dev->hard_header_len+15, GFP_ATOMIC);
  190. if (skb == NULL) {
  191. ip_rt_put(rt);
  192. return -1;
  193. }
  194. skb->dst = &rt->u.dst;
  195. skb_reserve(skb, (dev->hard_header_len+15)&~15);
  196. skb->nh.iph = iph = (struct iphdr *)skb_put(skb, sizeof(struct iphdr)+4);
  197. iph->version  = 4;
  198. iph->ihl      = (sizeof(struct iphdr)+4)>>2;
  199. iph->tos      = 0;
  200. iph->frag_off = htons(IP_DF);
  201. iph->ttl      = 1;
  202. iph->daddr    = dst;
  203. iph->saddr    = rt->rt_src;
  204. iph->protocol = IPPROTO_IGMP;
  205. iph->tot_len  = htons(IGMP_SIZE);
  206. ip_select_ident(iph, &rt->u.dst, NULL);
  207. ((u8*)&iph[1])[0] = IPOPT_RA;
  208. ((u8*)&iph[1])[1] = 4;
  209. ((u8*)&iph[1])[2] = 0;
  210. ((u8*)&iph[1])[3] = 0;
  211. ip_send_check(iph);
  212. ih = (struct igmphdr *)skb_put(skb, sizeof(struct igmphdr));
  213. ih->type=type;
  214. ih->code=0;
  215. ih->csum=0;
  216. ih->group=group;
  217. ih->csum=ip_compute_csum((void *)ih, sizeof(struct igmphdr));
  218. return NF_HOOK(PF_INET, NF_IP_LOCAL_OUT, skb, NULL, rt->u.dst.dev,
  219.        output_maybe_reroute);
  220. }
  221. static void igmp_timer_expire(unsigned long data)
  222. {
  223. struct ip_mc_list *im=(struct ip_mc_list *)data;
  224. struct in_device *in_dev = im->interface;
  225. int err;
  226. spin_lock(&im->lock);
  227. im->tm_running=0;
  228. if (IGMP_V1_SEEN(in_dev))
  229. err = igmp_send_report(in_dev->dev, im->multiaddr, IGMP_HOST_MEMBERSHIP_REPORT);
  230. else
  231. err = igmp_send_report(in_dev->dev, im->multiaddr, IGMP_HOST_NEW_MEMBERSHIP_REPORT);
  232. /* Failed. Retry later. */
  233. if (err) {
  234. if (!in_dev->dead)
  235. igmp_start_timer(im, IGMP_Unsolicited_Report_Interval);
  236. goto out;
  237. }
  238. if (im->unsolicit_count) {
  239. im->unsolicit_count--;
  240. igmp_start_timer(im, IGMP_Unsolicited_Report_Interval);
  241. }
  242. im->reporter = 1;
  243. out:
  244. spin_unlock(&im->lock);
  245. ip_ma_put(im);
  246. }
  247. static void igmp_heard_report(struct in_device *in_dev, u32 group)
  248. {
  249. struct ip_mc_list *im;
  250. /* Timers are only set for non-local groups */
  251. if (group == IGMP_ALL_HOSTS)
  252. return;
  253. read_lock(&in_dev->lock);
  254. for (im=in_dev->mc_list; im!=NULL; im=im->next) {
  255. if (im->multiaddr == group) {
  256. igmp_stop_timer(im);
  257. break;
  258. }
  259. }
  260. read_unlock(&in_dev->lock);
  261. }
  262. static void igmp_heard_query(struct in_device *in_dev, unsigned char max_resp_time,
  263.      u32 group)
  264. {
  265. struct ip_mc_list *im;
  266. int max_delay;
  267. max_delay = max_resp_time*(HZ/IGMP_TIMER_SCALE);
  268. if (max_resp_time == 0) {
  269. /* Alas, old v1 router presents here. */
  270. max_delay = IGMP_Query_Response_Interval;
  271. in_dev->mr_v1_seen = jiffies + IGMP_V1_Router_Present_Timeout;
  272. group = 0;
  273. }
  274. /*
  275.  * - Start the timers in all of our membership records
  276.  *   that the query applies to for the interface on
  277.  *   which the query arrived excl. those that belong
  278.  *   to a "local" group (224.0.0.X)
  279.  * - For timers already running check if they need to
  280.  *   be reset.
  281.  * - Use the igmp->igmp_code field as the maximum
  282.  *   delay possible
  283.  */
  284. read_lock(&in_dev->lock);
  285. for (im=in_dev->mc_list; im!=NULL; im=im->next) {
  286. if (group && group != im->multiaddr)
  287. continue;
  288. if (im->multiaddr == IGMP_ALL_HOSTS)
  289. continue;
  290. igmp_mod_timer(im, max_delay);
  291. }
  292. read_unlock(&in_dev->lock);
  293. }
  294. int igmp_rcv(struct sk_buff *skb)
  295. {
  296. /* This basically follows the spec line by line -- see RFC1112 */
  297. struct igmphdr *ih = skb->h.igmph;
  298. struct in_device *in_dev = in_dev_get(skb->dev);
  299. int len = skb->len;
  300. if (in_dev==NULL) {
  301. kfree_skb(skb);
  302. return 0;
  303. }
  304. if (skb_is_nonlinear(skb)) {
  305. if (skb_linearize(skb, GFP_ATOMIC) != 0) {
  306. kfree_skb(skb);
  307. return -ENOMEM;
  308. }
  309. ih = skb->h.igmph;
  310. }
  311. if (len < sizeof(struct igmphdr) || ip_compute_csum((void *)ih, len)) {
  312. in_dev_put(in_dev);
  313. kfree_skb(skb);
  314. return 0;
  315. }
  316. switch (ih->type) {
  317. case IGMP_HOST_MEMBERSHIP_QUERY:
  318. igmp_heard_query(in_dev, ih->code, ih->group);
  319. break;
  320. case IGMP_HOST_MEMBERSHIP_REPORT:
  321. case IGMP_HOST_NEW_MEMBERSHIP_REPORT:
  322. /* Is it our report looped back? */
  323. if (((struct rtable*)skb->dst)->key.iif == 0)
  324. break;
  325. igmp_heard_report(in_dev, ih->group);
  326. break;
  327. case IGMP_PIM:
  328. #ifdef CONFIG_IP_PIMSM_V1
  329. in_dev_put(in_dev);
  330. return pim_rcv_v1(skb);
  331. #endif
  332. case IGMP_DVMRP:
  333. case IGMP_TRACE:
  334. case IGMP_HOST_LEAVE_MESSAGE:
  335. case IGMP_MTRACE:
  336. case IGMP_MTRACE_RESP:
  337. break;
  338. default:
  339. NETDEBUG(printk(KERN_DEBUG "New IGMP type=%d, why we do not know about it?n", ih->type));
  340. }
  341. in_dev_put(in_dev);
  342. kfree_skb(skb);
  343. return 0;
  344. }
  345. #endif
  346. /*
  347.  * Add a filter to a device
  348.  */
  349. static void ip_mc_filter_add(struct in_device *in_dev, u32 addr)
  350. {
  351. char buf[MAX_ADDR_LEN];
  352. struct net_device *dev = in_dev->dev;
  353. /* Checking for IFF_MULTICAST here is WRONG-WRONG-WRONG.
  354.    We will get multicast token leakage, when IFF_MULTICAST
  355.    is changed. This check should be done in dev->set_multicast_list
  356.    routine. Something sort of:
  357.    if (dev->mc_list && dev->flags&IFF_MULTICAST) { do it; }
  358.    --ANK
  359.    */
  360. if (arp_mc_map(addr, buf, dev, 0) == 0)
  361. dev_mc_add(dev,buf,dev->addr_len,0);
  362. }
  363. /*
  364.  * Remove a filter from a device
  365.  */
  366. static void ip_mc_filter_del(struct in_device *in_dev, u32 addr)
  367. {
  368. char buf[MAX_ADDR_LEN];
  369. struct net_device *dev = in_dev->dev;
  370. if (arp_mc_map(addr, buf, dev, 0) == 0)
  371. dev_mc_delete(dev,buf,dev->addr_len,0);
  372. }
  373. static void igmp_group_dropped(struct ip_mc_list *im)
  374. {
  375. #ifdef CONFIG_IP_MULTICAST
  376. int reporter;
  377. #endif
  378. if (im->loaded) {
  379. im->loaded = 0;
  380. ip_mc_filter_del(im->interface, im->multiaddr);
  381. }
  382. #ifdef CONFIG_IP_MULTICAST
  383. if (im->multiaddr == IGMP_ALL_HOSTS)
  384. return;
  385. reporter = im->reporter;
  386. igmp_stop_timer(im);
  387. if (reporter && !IGMP_V1_SEEN(im->interface))
  388. igmp_send_report(im->interface->dev, im->multiaddr, IGMP_HOST_LEAVE_MESSAGE);
  389. #endif
  390. }
  391. static void igmp_group_added(struct ip_mc_list *im)
  392. {
  393. if (im->loaded == 0) {
  394. im->loaded = 1;
  395. ip_mc_filter_add(im->interface, im->multiaddr);
  396. }
  397. #ifdef CONFIG_IP_MULTICAST
  398. if (im->multiaddr == IGMP_ALL_HOSTS)
  399. return;
  400. spin_lock_bh(&im->lock);
  401. igmp_start_timer(im, IGMP_Initial_Report_Delay);
  402. spin_unlock_bh(&im->lock);
  403. #endif
  404. }
  405. /*
  406.  * Multicast list managers
  407.  */
  408. /*
  409.  * A socket has joined a multicast group on device dev.
  410.  */
  411. void ip_mc_inc_group(struct in_device *in_dev, u32 addr)
  412. {
  413. struct ip_mc_list *im;
  414. ASSERT_RTNL();
  415. for (im=in_dev->mc_list; im; im=im->next) {
  416. if (im->multiaddr == addr) {
  417. im->users++;
  418. goto out;
  419. }
  420. }
  421. im = (struct ip_mc_list *)kmalloc(sizeof(*im), GFP_KERNEL);
  422. if (!im)
  423. goto out;
  424. im->users=1;
  425. im->interface=in_dev;
  426. in_dev_hold(in_dev);
  427. im->multiaddr=addr;
  428. atomic_set(&im->refcnt, 1);
  429. spin_lock_init(&im->lock);
  430. #ifdef  CONFIG_IP_MULTICAST
  431. im->tm_running=0;
  432. init_timer(&im->timer);
  433. im->timer.data=(unsigned long)im;
  434. im->timer.function=&igmp_timer_expire;
  435. im->unsolicit_count = IGMP_Unsolicited_Report_Count;
  436. im->reporter = 0;
  437. #endif
  438. im->loaded = 0;
  439. write_lock_bh(&in_dev->lock);
  440. im->next=in_dev->mc_list;
  441. in_dev->mc_list=im;
  442. write_unlock_bh(&in_dev->lock);
  443. igmp_group_added(im);
  444. if (in_dev->dev->flags & IFF_UP)
  445. ip_rt_multicast_event(in_dev);
  446. out:
  447. return;
  448. }
  449. /*
  450.  * A socket has left a multicast group on device dev
  451.  */
  452. void ip_mc_dec_group(struct in_device *in_dev, u32 addr)
  453. {
  454. struct ip_mc_list *i, **ip;
  455. ASSERT_RTNL();
  456. for (ip=&in_dev->mc_list; (i=*ip)!=NULL; ip=&i->next) {
  457. if (i->multiaddr==addr) {
  458. if (--i->users == 0) {
  459. write_lock_bh(&in_dev->lock);
  460. *ip = i->next;
  461. write_unlock_bh(&in_dev->lock);
  462. igmp_group_dropped(i);
  463. if (in_dev->dev->flags & IFF_UP)
  464. ip_rt_multicast_event(in_dev);
  465. ip_ma_put(i);
  466. return;
  467. }
  468. break;
  469. }
  470. }
  471. }
  472. /* Device going down */
  473. void ip_mc_down(struct in_device *in_dev)
  474. {
  475. struct ip_mc_list *i;
  476. ASSERT_RTNL();
  477. for (i=in_dev->mc_list; i; i=i->next)
  478. igmp_group_dropped(i);
  479. ip_mc_dec_group(in_dev, IGMP_ALL_HOSTS);
  480. }
  481. /* Device going up */
  482. void ip_mc_up(struct in_device *in_dev)
  483. {
  484. struct ip_mc_list *i;
  485. ASSERT_RTNL();
  486. ip_mc_inc_group(in_dev, IGMP_ALL_HOSTS);
  487. for (i=in_dev->mc_list; i; i=i->next)
  488. igmp_group_added(i);
  489. }
  490. /*
  491.  * Device is about to be destroyed: clean up.
  492.  */
  493. void ip_mc_destroy_dev(struct in_device *in_dev)
  494. {
  495. struct ip_mc_list *i;
  496. ASSERT_RTNL();
  497. write_lock_bh(&in_dev->lock);
  498. while ((i = in_dev->mc_list) != NULL) {
  499. in_dev->mc_list = i->next;
  500. write_unlock_bh(&in_dev->lock);
  501. igmp_group_dropped(i);
  502. ip_ma_put(i);
  503. write_lock_bh(&in_dev->lock);
  504. }
  505. write_unlock_bh(&in_dev->lock);
  506. }
  507. static struct in_device * ip_mc_find_dev(struct ip_mreqn *imr)
  508. {
  509. struct rtable *rt;
  510. struct net_device *dev = NULL;
  511. struct in_device *idev = NULL;
  512. if (imr->imr_address.s_addr) {
  513. dev = ip_dev_find(imr->imr_address.s_addr);
  514. if (!dev)
  515. return NULL;
  516. __dev_put(dev);
  517. }
  518. if (!dev && !ip_route_output(&rt, imr->imr_multiaddr.s_addr, 0, 0, 0)) {
  519. dev = rt->u.dst.dev;
  520. ip_rt_put(rt);
  521. }
  522. if (dev) {
  523. imr->imr_ifindex = dev->ifindex;
  524. idev = __in_dev_get(dev);
  525. }
  526. return idev;
  527. }
  528. /*
  529.  * Join a socket to a group
  530.  */
  531. int sysctl_igmp_max_memberships = IP_MAX_MEMBERSHIPS;
  532. int ip_mc_join_group(struct sock *sk , struct ip_mreqn *imr)
  533. {
  534. int err;
  535. u32 addr = imr->imr_multiaddr.s_addr;
  536. struct ip_mc_socklist *iml, *i;
  537. struct in_device *in_dev;
  538. int count = 0;
  539. if (!MULTICAST(addr))
  540. return -EINVAL;
  541. rtnl_shlock();
  542. if (!imr->imr_ifindex)
  543. in_dev = ip_mc_find_dev(imr);
  544. else {
  545. in_dev = inetdev_by_index(imr->imr_ifindex);
  546. if (in_dev)
  547. __in_dev_put(in_dev);
  548. }
  549. if (!in_dev) {
  550. iml = NULL;
  551. err = -ENODEV;
  552. goto done;
  553. }
  554. iml = (struct ip_mc_socklist *)sock_kmalloc(sk, sizeof(*iml), GFP_KERNEL);
  555. err = -EADDRINUSE;
  556. for (i=sk->protinfo.af_inet.mc_list; i; i=i->next) {
  557. if (memcmp(&i->multi, imr, sizeof(*imr)) == 0) {
  558. /* New style additions are reference counted */
  559. if (imr->imr_address.s_addr == 0) {
  560. i->count++;
  561. err = 0;
  562. }
  563. goto done;
  564. }
  565. count++;
  566. }
  567. err = -ENOBUFS;
  568. if (iml == NULL || count >= sysctl_igmp_max_memberships)
  569. goto done;
  570. memcpy(&iml->multi, imr, sizeof(*imr));
  571. iml->next = sk->protinfo.af_inet.mc_list;
  572. iml->count = 1;
  573. sk->protinfo.af_inet.mc_list = iml;
  574. ip_mc_inc_group(in_dev, addr);
  575. iml = NULL;
  576. err = 0;
  577. done:
  578. rtnl_shunlock();
  579. if (iml)
  580. sock_kfree_s(sk, iml, sizeof(*iml));
  581. return err;
  582. }
  583. /*
  584.  * Ask a socket to leave a group.
  585.  */
  586. int ip_mc_leave_group(struct sock *sk, struct ip_mreqn *imr)
  587. {
  588. struct ip_mc_socklist *iml, **imlp;
  589. rtnl_lock();
  590. for (imlp=&sk->protinfo.af_inet.mc_list; (iml=*imlp)!=NULL; imlp=&iml->next) {
  591. if (iml->multi.imr_multiaddr.s_addr==imr->imr_multiaddr.s_addr &&
  592.     iml->multi.imr_address.s_addr==imr->imr_address.s_addr &&
  593.     (!imr->imr_ifindex || iml->multi.imr_ifindex==imr->imr_ifindex)) {
  594. struct in_device *in_dev;
  595. if (--iml->count) {
  596. rtnl_unlock();
  597. return 0;
  598. }
  599. *imlp = iml->next;
  600. in_dev = inetdev_by_index(iml->multi.imr_ifindex);
  601. if (in_dev) {
  602. ip_mc_dec_group(in_dev, imr->imr_multiaddr.s_addr);
  603. in_dev_put(in_dev);
  604. }
  605. rtnl_unlock();
  606. sock_kfree_s(sk, iml, sizeof(*iml));
  607. return 0;
  608. }
  609. }
  610. rtnl_unlock();
  611. return -EADDRNOTAVAIL;
  612. }
  613. /*
  614.  * A socket is closing.
  615.  */
  616. void ip_mc_drop_socket(struct sock *sk)
  617. {
  618. struct ip_mc_socklist *iml;
  619. if (sk->protinfo.af_inet.mc_list == NULL)
  620. return;
  621. rtnl_lock();
  622. while ((iml=sk->protinfo.af_inet.mc_list) != NULL) {
  623. struct in_device *in_dev;
  624. sk->protinfo.af_inet.mc_list = iml->next;
  625. if ((in_dev = inetdev_by_index(iml->multi.imr_ifindex)) != NULL) {
  626. ip_mc_dec_group(in_dev, iml->multi.imr_multiaddr.s_addr);
  627. in_dev_put(in_dev);
  628. }
  629. sock_kfree_s(sk, iml, sizeof(*iml));
  630. }
  631. rtnl_unlock();
  632. }
  633. int ip_check_mc(struct in_device *in_dev, u32 mc_addr)
  634. {
  635. struct ip_mc_list *im;
  636. read_lock(&in_dev->lock);
  637. for (im=in_dev->mc_list; im; im=im->next) {
  638. if (im->multiaddr == mc_addr) {
  639. read_unlock(&in_dev->lock);
  640. return 1;
  641. }
  642. }
  643. read_unlock(&in_dev->lock);
  644. return 0;
  645. }
  646. #ifdef CONFIG_IP_MULTICAST
  647.  
  648. int ip_mc_procinfo(char *buffer, char **start, off_t offset, int length)
  649. {
  650. off_t pos=0, begin=0;
  651. struct ip_mc_list *im;
  652. int len=0;
  653. struct net_device *dev;
  654. len=sprintf(buffer,"IdxtDevice    : Count QueriertGroup    Users TimertReportern");  
  655. read_lock(&dev_base_lock);
  656. for(dev = dev_base; dev; dev = dev->next) {
  657. struct in_device *in_dev = in_dev_get(dev);
  658. char   *querier = "NONE";
  659. if (in_dev == NULL)
  660. continue;
  661. querier = IGMP_V1_SEEN(in_dev) ? "V1" : "V2";
  662. len+=sprintf(buffer+len,"%dt%-10s: %5d %7sn",
  663.      dev->ifindex, dev->name, dev->mc_count, querier);
  664. read_lock(&in_dev->lock);
  665. for (im = in_dev->mc_list; im; im = im->next) {
  666. len+=sprintf(buffer+len,
  667.      "tttt%08lX %5d %d:%08lXtt%dn",
  668.      im->multiaddr, im->users,
  669.      im->tm_running, im->timer.expires-jiffies, im->reporter);
  670. pos=begin+len;
  671. if(pos<offset)
  672. {
  673. len=0;
  674. begin=pos;
  675. }
  676. if(pos>offset+length) {
  677. read_unlock(&in_dev->lock);
  678. in_dev_put(in_dev);
  679. goto done;
  680. }
  681. }
  682. read_unlock(&in_dev->lock);
  683. in_dev_put(in_dev);
  684. }
  685. done:
  686. read_unlock(&dev_base_lock);
  687. *start=buffer+(offset-begin);
  688. len-=(offset-begin);
  689. if(len>length)
  690. len=length;
  691. if(len<0)
  692. len=0;
  693. return len;
  694. }
  695. #endif