开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 嵌入式/单片机编程 > 嵌入式Linux > 查看源码
igmp.c
资源名称：SBC-2410X_kernel.tar.gz　[点击查看]
上传用户：lgb322
上传日期：2013-02-24
资源大小：30529k
文件大小：19k
源码类别：

嵌入式Linux

开发平台：

Unix_Linux

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