net/ipv6/mcast.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  *      Multicast support for IPv6
   4  *      Linux INET6 implementation
   5  *
   6  *      Authors:
   7  *      Pedro Roque             <roque@di.fc.ul.pt>
   8  *
   9  *      Based on linux/ipv4/igmp.c and linux/ipv4/ip_sockglue.c
  10  */
  11
  12 /* Changes:
  13  *
  14  *      yoshfuji        : fix format of router-alert option
  15  *      YOSHIFUJI Hideaki @USAGI:
  16  *              Fixed source address for MLD message based on
  17  *              <draft-ietf-magma-mld-source-05.txt>.
  18  *      YOSHIFUJI Hideaki @USAGI:
  19  *              - Ignore Queries for invalid addresses.
  20  *              - MLD for link-local addresses.
  21  *      David L Stevens <dlstevens@us.ibm.com>:
  22  *              - MLDv2 support
  23  */
  24
  25 #include <linux/module.h>
  26 #include <linux/errno.h>
  27 #include <linux/types.h>
  28 #include <linux/string.h>
  29 #include <linux/socket.h>
  30 #include <linux/sockios.h>
  31 #include <linux/jiffies.h>
  32 #include <linux/net.h>
  33 #include <linux/in.h>
  34 #include <linux/in6.h>
  35 #include <linux/netdevice.h>
  36 #include <linux/if_arp.h>
  37 #include <linux/route.h>
  38 #include <linux/init.h>
  39 #include <linux/proc_fs.h>
  40 #include <linux/seq_file.h>
  41 #include <linux/slab.h>
  42 #include <linux/pkt_sched.h>
  43 #include <net/mld.h>
  44 #include <linux/workqueue.h>
  45
  46 #include <linux/netfilter.h>
  47 #include <linux/netfilter_ipv6.h>
  48
  49 #include <net/net_namespace.h>
  50 #include <net/sock.h>
  51 #include <net/snmp.h>
  52
  53 #include <net/ipv6.h>
  54 #include <net/protocol.h>
  55 #include <net/if_inet6.h>
  56 #include <net/ndisc.h>
  57 #include <net/addrconf.h>
  58 #include <net/ip6_route.h>
  59 #include <net/inet_common.h>
  60
  61 #include <net/ip6_checksum.h>
  62
  63 /* Ensure that we have struct in6_addr aligned on 32bit word. */
  64 static int __mld2_query_bugs[] __attribute__((__unused__)) = {
  65         BUILD_BUG_ON_ZERO(offsetof(struct mld2_query, mld2q_srcs) % 4),
  66         BUILD_BUG_ON_ZERO(offsetof(struct mld2_report, mld2r_grec) % 4),
  67         BUILD_BUG_ON_ZERO(offsetof(struct mld2_grec, grec_mca) % 4)
  68 };
  69
  70 static struct workqueue_struct *mld_wq;
  71 static struct in6_addr mld2_all_mcr = MLD2_ALL_MCR_INIT;
  72
  73 static void igmp6_join_group(struct ifmcaddr6 *ma);
  74 static void igmp6_leave_group(struct ifmcaddr6 *ma);
  75 static void mld_mca_work(struct work_struct *work);
  76
  77 static void mld_ifc_event(struct inet6_dev *idev);
  78 static bool mld_in_v1_mode(const struct inet6_dev *idev);
  79 static int sf_setstate(struct ifmcaddr6 *pmc);
  80 static void sf_markstate(struct ifmcaddr6 *pmc);
  81 static void ip6_mc_clear_src(struct ifmcaddr6 *pmc);
  82 static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
  83                           int sfmode, int sfcount, const struct in6_addr *psfsrc,
  84                           int delta);
  85 static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
  86                           int sfmode, int sfcount, const struct in6_addr *psfsrc,
  87                           int delta);
  88 static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
  89                             struct inet6_dev *idev);
  90 static int __ipv6_dev_mc_inc(struct net_device *dev,
  91                              const struct in6_addr *addr, unsigned int mode);
  92
  93 #define MLD_QRV_DEFAULT         2
  94 /* RFC3810, 9.2. Query Interval */
  95 #define MLD_QI_DEFAULT          (125 * HZ)
  96 /* RFC3810, 9.3. Query Response Interval */
  97 #define MLD_QRI_DEFAULT         (10 * HZ)
  98
  99 /* RFC3810, 8.1 Query Version Distinctions */
 100 #define MLD_V1_QUERY_LEN        24
 101 #define MLD_V2_QUERY_LEN_MIN    28
 102
 103 #define IPV6_MLD_MAX_MSF        64
 104
 105 int sysctl_mld_max_msf __read_mostly = IPV6_MLD_MAX_MSF;
 106 int sysctl_mld_qrv __read_mostly = MLD_QRV_DEFAULT;
 107
 108 /*
 109  *      socket join on multicast group
 110  */
 111 #define mc_dereference(e, idev) \
 112         rcu_dereference_protected(e, lockdep_is_held(&(idev)->mc_lock))
 113
 114 #define sock_dereference(e, sk) \
 115         rcu_dereference_protected(e, lockdep_sock_is_held(sk))
 116
 117 #define for_each_pmc_socklock(np, sk, pmc)                      \
 118         for (pmc = sock_dereference((np)->ipv6_mc_list, sk);    \
 119              pmc;                                               \
 120              pmc = sock_dereference(pmc->next, sk))
 121
 122 #define for_each_pmc_rcu(np, pmc)                               \
 123         for (pmc = rcu_dereference((np)->ipv6_mc_list);         \
 124              pmc;                                               \
 125              pmc = rcu_dereference(pmc->next))
 126
 127 #define for_each_psf_mclock(mc, psf)                            \
 128         for (psf = mc_dereference((mc)->mca_sources, mc->idev); \
 129              psf;                                               \
 130              psf = mc_dereference(psf->sf_next, mc->idev))
 131
 132 #define for_each_psf_rcu(mc, psf)                               \
 133         for (psf = rcu_dereference((mc)->mca_sources);          \
 134              psf;                                               \
 135              psf = rcu_dereference(psf->sf_next))
 136
 137 #define for_each_psf_tomb(mc, psf)                              \
 138         for (psf = mc_dereference((mc)->mca_tomb, mc->idev);    \
 139              psf;                                               \
 140              psf = mc_dereference(psf->sf_next, mc->idev))
 141
 142 #define for_each_mc_mclock(idev, mc)                            \
 143         for (mc = mc_dereference((idev)->mc_list, idev);        \
 144              mc;                                                \
 145              mc = mc_dereference(mc->next, idev))
 146
 147 #define for_each_mc_rcu(idev, mc)                               \
 148         for (mc = rcu_dereference((idev)->mc_list);             \
 149              mc;                                                \
 150              mc = rcu_dereference(mc->next))
 151
 152 #define for_each_mc_tomb(idev, mc)                              \
 153         for (mc = mc_dereference((idev)->mc_tomb, idev);        \
 154              mc;                                                \
 155              mc = mc_dereference(mc->next, idev))
 156
 157 static int unsolicited_report_interval(struct inet6_dev *idev)
 158 {
 159         int iv;
 160
 161         if (mld_in_v1_mode(idev))
 162                 iv = idev->cnf.mldv1_unsolicited_report_interval;
 163         else
 164                 iv = idev->cnf.mldv2_unsolicited_report_interval;
 165
 166         return iv > 0 ? iv : 1;
 167 }
 168
 169 static int __ipv6_sock_mc_join(struct sock *sk, int ifindex,
 170                                const struct in6_addr *addr, unsigned int mode)
 171 {
 172         struct net_device *dev = NULL;
 173         struct ipv6_mc_socklist *mc_lst;
 174         struct ipv6_pinfo *np = inet6_sk(sk);
 175         struct net *net = sock_net(sk);
 176         int err;
 177
 178         ASSERT_RTNL();
 179
 180         if (!ipv6_addr_is_multicast(addr))
 181                 return -EINVAL;
 182
 183         for_each_pmc_socklock(np, sk, mc_lst) {
 184                 if ((ifindex == 0 || mc_lst->ifindex == ifindex) &&
 185                     ipv6_addr_equal(&mc_lst->addr, addr))
 186                         return -EADDRINUSE;
 187         }
 188
 189         mc_lst = sock_kmalloc(sk, sizeof(struct ipv6_mc_socklist), GFP_KERNEL);
 190
 191         if (!mc_lst)
 192                 return -ENOMEM;
 193
 194         mc_lst->next = NULL;
 195         mc_lst->addr = *addr;
 196
 197         if (ifindex == 0) {
 198                 struct rt6_info *rt;
 199                 rt = rt6_lookup(net, addr, NULL, 0, NULL, 0);
 200                 if (rt) {
 201                         dev = rt->dst.dev;
 202                         ip6_rt_put(rt);
 203                 }
 204         } else
 205                 dev = __dev_get_by_index(net, ifindex);
 206
 207         if (!dev) {
 208                 sock_kfree_s(sk, mc_lst, sizeof(*mc_lst));
 209                 return -ENODEV;
 210         }
 211
 212         mc_lst->ifindex = dev->ifindex;
 213         mc_lst->sfmode = mode;
 214         RCU_INIT_POINTER(mc_lst->sflist, NULL);
 215
 216         /*
 217          *      now add/increase the group membership on the device
 218          */
 219
 220         err = __ipv6_dev_mc_inc(dev, addr, mode);
 221
 222         if (err) {
 223                 sock_kfree_s(sk, mc_lst, sizeof(*mc_lst));
 224                 return err;
 225         }
 226
 227         mc_lst->next = np->ipv6_mc_list;
 228         rcu_assign_pointer(np->ipv6_mc_list, mc_lst);
 229
 230         return 0;
 231 }
 232
 233 int ipv6_sock_mc_join(struct sock *sk, int ifindex, const struct in6_addr *addr)
 234 {
 235         return __ipv6_sock_mc_join(sk, ifindex, addr, MCAST_EXCLUDE);
 236 }
 237 EXPORT_SYMBOL(ipv6_sock_mc_join);
 238
 239 int ipv6_sock_mc_join_ssm(struct sock *sk, int ifindex,
 240                           const struct in6_addr *addr, unsigned int mode)
 241 {
 242         return __ipv6_sock_mc_join(sk, ifindex, addr, mode);
 243 }
 244
 245 /*
 246  *      socket leave on multicast group
 247  */
 248 int ipv6_sock_mc_drop(struct sock *sk, int ifindex, const struct in6_addr *addr)
 249 {
 250         struct ipv6_pinfo *np = inet6_sk(sk);
 251         struct ipv6_mc_socklist *mc_lst;
 252         struct ipv6_mc_socklist __rcu **lnk;
 253         struct net *net = sock_net(sk);
 254
 255         ASSERT_RTNL();
 256
 257         if (!ipv6_addr_is_multicast(addr))
 258                 return -EINVAL;
 259
 260         for (lnk = &np->ipv6_mc_list;
 261              (mc_lst = sock_dereference(*lnk, sk)) != NULL;
 262               lnk = &mc_lst->next) {
 263                 if ((ifindex == 0 || mc_lst->ifindex == ifindex) &&
 264                     ipv6_addr_equal(&mc_lst->addr, addr)) {
 265                         struct net_device *dev;
 266
 267                         *lnk = mc_lst->next;
 268
 269                         dev = __dev_get_by_index(net, mc_lst->ifindex);
 270                         if (dev) {
 271                                 struct inet6_dev *idev = __in6_dev_get(dev);
 272
 273                                 ip6_mc_leave_src(sk, mc_lst, idev);
 274                                 if (idev)
 275                                         __ipv6_dev_mc_dec(idev, &mc_lst->addr);
 276                         } else {
 277                                 ip6_mc_leave_src(sk, mc_lst, NULL);
 278                         }
 279
 280                         atomic_sub(sizeof(*mc_lst), &sk->sk_omem_alloc);
 281                         kfree_rcu(mc_lst, rcu);
 282                         return 0;
 283                 }
 284         }
 285
 286         return -EADDRNOTAVAIL;
 287 }
 288 EXPORT_SYMBOL(ipv6_sock_mc_drop);
 289
 290 static struct inet6_dev *ip6_mc_find_dev_rtnl(struct net *net,
 291                                               const struct in6_addr *group,
 292                                               int ifindex)
 293 {
 294         struct net_device *dev = NULL;
 295         struct inet6_dev *idev = NULL;
 296
 297         if (ifindex == 0) {
 298                 struct rt6_info *rt = rt6_lookup(net, group, NULL, 0, NULL, 0);
 299
 300                 if (rt) {
 301                         dev = rt->dst.dev;
 302                         ip6_rt_put(rt);
 303                 }
 304         } else {
 305                 dev = __dev_get_by_index(net, ifindex);
 306         }
 307
 308         if (!dev)
 309                 return NULL;
 310         idev = __in6_dev_get(dev);
 311         if (!idev)
 312                 return NULL;
 313         if (idev->dead)
 314                 return NULL;
 315         return idev;
 316 }
 317
 318 void __ipv6_sock_mc_close(struct sock *sk)
 319 {
 320         struct ipv6_pinfo *np = inet6_sk(sk);
 321         struct ipv6_mc_socklist *mc_lst;
 322         struct net *net = sock_net(sk);
 323
 324         ASSERT_RTNL();
 325
 326         while ((mc_lst = sock_dereference(np->ipv6_mc_list, sk)) != NULL) {
 327                 struct net_device *dev;
 328
 329                 np->ipv6_mc_list = mc_lst->next;
 330
 331                 dev = __dev_get_by_index(net, mc_lst->ifindex);
 332                 if (dev) {
 333                         struct inet6_dev *idev = __in6_dev_get(dev);
 334
 335                         ip6_mc_leave_src(sk, mc_lst, idev);
 336                         if (idev)
 337                                 __ipv6_dev_mc_dec(idev, &mc_lst->addr);
 338                 } else {
 339                         ip6_mc_leave_src(sk, mc_lst, NULL);
 340                 }
 341
 342                 atomic_sub(sizeof(*mc_lst), &sk->sk_omem_alloc);
 343                 kfree_rcu(mc_lst, rcu);
 344         }
 345 }
 346
 347 void ipv6_sock_mc_close(struct sock *sk)
 348 {
 349         struct ipv6_pinfo *np = inet6_sk(sk);
 350
 351         if (!rcu_access_pointer(np->ipv6_mc_list))
 352                 return;
 353
 354         rtnl_lock();
 355         lock_sock(sk);
 356         __ipv6_sock_mc_close(sk);
 357         release_sock(sk);
 358         rtnl_unlock();
 359 }
 360
 361 int ip6_mc_source(int add, int omode, struct sock *sk,
 362         struct group_source_req *pgsr)
 363 {
 364         struct in6_addr *source, *group;
 365         struct ipv6_mc_socklist *pmc;
 366         struct inet6_dev *idev;
 367         struct ipv6_pinfo *inet6 = inet6_sk(sk);
 368         struct ip6_sf_socklist *psl;
 369         struct net *net = sock_net(sk);
 370         int i, j, rv;
 371         int leavegroup = 0;
 372         int err;
 373
 374         source = &((struct sockaddr_in6 *)&pgsr->gsr_source)->sin6_addr;
 375         group = &((struct sockaddr_in6 *)&pgsr->gsr_group)->sin6_addr;
 376
 377         if (!ipv6_addr_is_multicast(group))
 378                 return -EINVAL;
 379
 380         idev = ip6_mc_find_dev_rtnl(net, group, pgsr->gsr_interface);
 381         if (!idev)
 382                 return -ENODEV;
 383
 384         err = -EADDRNOTAVAIL;
 385
 386         mutex_lock(&idev->mc_lock);
 387         for_each_pmc_socklock(inet6, sk, pmc) {
 388                 if (pgsr->gsr_interface && pmc->ifindex != pgsr->gsr_interface)
 389                         continue;
 390                 if (ipv6_addr_equal(&pmc->addr, group))
 391                         break;
 392         }
 393         if (!pmc) {             /* must have a prior join */
 394                 err = -EINVAL;
 395                 goto done;
 396         }
 397         /* if a source filter was set, must be the same mode as before */
 398         if (rcu_access_pointer(pmc->sflist)) {
 399                 if (pmc->sfmode != omode) {
 400                         err = -EINVAL;
 401                         goto done;
 402                 }
 403         } else if (pmc->sfmode != omode) {
 404                 /* allow mode switches for empty-set filters */
 405                 ip6_mc_add_src(idev, group, omode, 0, NULL, 0);
 406                 ip6_mc_del_src(idev, group, pmc->sfmode, 0, NULL, 0);
 407                 pmc->sfmode = omode;
 408         }
 409
 410         psl = sock_dereference(pmc->sflist, sk);
 411         if (!add) {
 412                 if (!psl)
 413                         goto done;      /* err = -EADDRNOTAVAIL */
 414                 rv = !0;
 415                 for (i = 0; i < psl->sl_count; i++) {
 416                         rv = !ipv6_addr_equal(&psl->sl_addr[i], source);
 417                         if (rv == 0)
 418                                 break;
 419                 }
 420                 if (rv)         /* source not found */
 421                         goto done;      /* err = -EADDRNOTAVAIL */
 422
 423                 /* special case - (INCLUDE, empty) == LEAVE_GROUP */
 424                 if (psl->sl_count == 1 && omode == MCAST_INCLUDE) {
 425                         leavegroup = 1;
 426                         goto done;
 427                 }
 428
 429                 /* update the interface filter */
 430                 ip6_mc_del_src(idev, group, omode, 1, source, 1);
 431
 432                 for (j = i+1; j < psl->sl_count; j++)
 433                         psl->sl_addr[j-1] = psl->sl_addr[j];
 434                 psl->sl_count--;
 435                 err = 0;
 436                 goto done;
 437         }
 438         /* else, add a new source to the filter */
 439
 440         if (psl && psl->sl_count >= sysctl_mld_max_msf) {
 441                 err = -ENOBUFS;
 442                 goto done;
 443         }
 444         if (!psl || psl->sl_count == psl->sl_max) {
 445                 struct ip6_sf_socklist *newpsl;
 446                 int count = IP6_SFBLOCK;
 447
 448                 if (psl)
 449                         count += psl->sl_max;
 450                 newpsl = sock_kmalloc(sk, struct_size(newpsl, sl_addr, count),
 451                                       GFP_KERNEL);
 452                 if (!newpsl) {
 453                         err = -ENOBUFS;
 454                         goto done;
 455                 }
 456                 newpsl->sl_max = count;
 457                 newpsl->sl_count = count - IP6_SFBLOCK;
 458                 if (psl) {
 459                         for (i = 0; i < psl->sl_count; i++)
 460                                 newpsl->sl_addr[i] = psl->sl_addr[i];
 461                         atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
 462                                    &sk->sk_omem_alloc);
 463                 }
 464                 rcu_assign_pointer(pmc->sflist, newpsl);
 465                 kfree_rcu(psl, rcu);
 466                 psl = newpsl;
 467         }
 468         rv = 1; /* > 0 for insert logic below if sl_count is 0 */
 469         for (i = 0; i < psl->sl_count; i++) {
 470                 rv = !ipv6_addr_equal(&psl->sl_addr[i], source);
 471                 if (rv == 0) /* There is an error in the address. */
 472                         goto done;
 473         }
 474         for (j = psl->sl_count-1; j >= i; j--)
 475                 psl->sl_addr[j+1] = psl->sl_addr[j];
 476         psl->sl_addr[i] = *source;
 477         psl->sl_count++;
 478         err = 0;
 479         /* update the interface list */
 480         ip6_mc_add_src(idev, group, omode, 1, source, 1);
 481 done:
 482         mutex_unlock(&idev->mc_lock);
 483         if (leavegroup)
 484                 err = ipv6_sock_mc_drop(sk, pgsr->gsr_interface, group);
 485         return err;
 486 }
 487
 488 int ip6_mc_msfilter(struct sock *sk, struct group_filter *gsf,
 489                     struct sockaddr_storage *list)
 490 {
 491         const struct in6_addr *group;
 492         struct ipv6_mc_socklist *pmc;
 493         struct inet6_dev *idev;
 494         struct ipv6_pinfo *inet6 = inet6_sk(sk);
 495         struct ip6_sf_socklist *newpsl, *psl;
 496         struct net *net = sock_net(sk);
 497         int leavegroup = 0;
 498         int i, err;
 499
 500         group = &((struct sockaddr_in6 *)&gsf->gf_group)->sin6_addr;
 501
 502         if (!ipv6_addr_is_multicast(group))
 503                 return -EINVAL;
 504         if (gsf->gf_fmode != MCAST_INCLUDE &&
 505             gsf->gf_fmode != MCAST_EXCLUDE)
 506                 return -EINVAL;
 507
 508         idev = ip6_mc_find_dev_rtnl(net, group, gsf->gf_interface);
 509         if (!idev)
 510                 return -ENODEV;
 511
 512         err = 0;
 513
 514         if (gsf->gf_fmode == MCAST_INCLUDE && gsf->gf_numsrc == 0) {
 515                 leavegroup = 1;
 516                 goto done;
 517         }
 518
 519         for_each_pmc_socklock(inet6, sk, pmc) {
 520                 if (pmc->ifindex != gsf->gf_interface)
 521                         continue;
 522                 if (ipv6_addr_equal(&pmc->addr, group))
 523                         break;
 524         }
 525         if (!pmc) {             /* must have a prior join */
 526                 err = -EINVAL;
 527                 goto done;
 528         }
 529         if (gsf->gf_numsrc) {
 530                 newpsl = sock_kmalloc(sk, struct_size(newpsl, sl_addr,
 531                                                       gsf->gf_numsrc),
 532                                       GFP_KERNEL);
 533                 if (!newpsl) {
 534                         err = -ENOBUFS;
 535                         goto done;
 536                 }
 537                 newpsl->sl_max = newpsl->sl_count = gsf->gf_numsrc;
 538                 for (i = 0; i < newpsl->sl_count; ++i, ++list) {
 539                         struct sockaddr_in6 *psin6;
 540
 541                         psin6 = (struct sockaddr_in6 *)list;
 542                         newpsl->sl_addr[i] = psin6->sin6_addr;
 543                 }
 544                 mutex_lock(&idev->mc_lock);
 545                 err = ip6_mc_add_src(idev, group, gsf->gf_fmode,
 546                                      newpsl->sl_count, newpsl->sl_addr, 0);
 547                 if (err) {
 548                         mutex_unlock(&idev->mc_lock);
 549                         sock_kfree_s(sk, newpsl, struct_size(newpsl, sl_addr,
 550                                                              newpsl->sl_max));
 551                         goto done;
 552                 }
 553                 mutex_unlock(&idev->mc_lock);
 554         } else {
 555                 newpsl = NULL;
 556                 mutex_lock(&idev->mc_lock);
 557                 ip6_mc_add_src(idev, group, gsf->gf_fmode, 0, NULL, 0);
 558                 mutex_unlock(&idev->mc_lock);
 559         }
 560
 561         mutex_lock(&idev->mc_lock);
 562         psl = sock_dereference(pmc->sflist, sk);
 563         if (psl) {
 564                 ip6_mc_del_src(idev, group, pmc->sfmode,
 565                                psl->sl_count, psl->sl_addr, 0);
 566                 atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
 567                            &sk->sk_omem_alloc);
 568         } else {
 569                 ip6_mc_del_src(idev, group, pmc->sfmode, 0, NULL, 0);
 570         }
 571         rcu_assign_pointer(pmc->sflist, newpsl);
 572         mutex_unlock(&idev->mc_lock);
 573         kfree_rcu(psl, rcu);
 574         pmc->sfmode = gsf->gf_fmode;
 575         err = 0;
 576 done:
 577         if (leavegroup)
 578                 err = ipv6_sock_mc_drop(sk, gsf->gf_interface, group);
 579         return err;
 580 }
 581
 582 int ip6_mc_msfget(struct sock *sk, struct group_filter *gsf,
 583                   sockptr_t optval, size_t ss_offset)
 584 {
 585         struct ipv6_pinfo *inet6 = inet6_sk(sk);
 586         const struct in6_addr *group;
 587         struct ipv6_mc_socklist *pmc;
 588         struct ip6_sf_socklist *psl;
 589         int i, count, copycount;
 590
 591         group = &((struct sockaddr_in6 *)&gsf->gf_group)->sin6_addr;
 592
 593         if (!ipv6_addr_is_multicast(group))
 594                 return -EINVAL;
 595
 596         /* changes to the ipv6_mc_list require the socket lock and
 597          * rtnl lock. We have the socket lock, so reading the list is safe.
 598          */
 599
 600         for_each_pmc_socklock(inet6, sk, pmc) {
 601                 if (pmc->ifindex != gsf->gf_interface)
 602                         continue;
 603                 if (ipv6_addr_equal(group, &pmc->addr))
 604                         break;
 605         }
 606         if (!pmc)               /* must have a prior join */
 607                 return -EADDRNOTAVAIL;
 608
 609         gsf->gf_fmode = pmc->sfmode;
 610         psl = sock_dereference(pmc->sflist, sk);
 611         count = psl ? psl->sl_count : 0;
 612
 613         copycount = count < gsf->gf_numsrc ? count : gsf->gf_numsrc;
 614         gsf->gf_numsrc = count;
 615         for (i = 0; i < copycount; i++) {
 616                 struct sockaddr_in6 *psin6;
 617                 struct sockaddr_storage ss;
 618
 619                 psin6 = (struct sockaddr_in6 *)&ss;
 620                 memset(&ss, 0, sizeof(ss));
 621                 psin6->sin6_family = AF_INET6;
 622                 psin6->sin6_addr = psl->sl_addr[i];
 623                 if (copy_to_sockptr_offset(optval, ss_offset, &ss, sizeof(ss)))
 624                         return -EFAULT;
 625                 ss_offset += sizeof(ss);
 626         }
 627         return 0;
 628 }
 629
 630 bool inet6_mc_check(const struct sock *sk, const struct in6_addr *mc_addr,
 631                     const struct in6_addr *src_addr)
 632 {
 633         const struct ipv6_pinfo *np = inet6_sk(sk);
 634         const struct ipv6_mc_socklist *mc;
 635         const struct ip6_sf_socklist *psl;
 636         bool rv = true;
 637
 638         rcu_read_lock();
 639         for_each_pmc_rcu(np, mc) {
 640                 if (ipv6_addr_equal(&mc->addr, mc_addr))
 641                         break;
 642         }
 643         if (!mc) {
 644                 rcu_read_unlock();
 645                 return np->mc_all;
 646         }
 647         psl = rcu_dereference(mc->sflist);
 648         if (!psl) {
 649                 rv = mc->sfmode == MCAST_EXCLUDE;
 650         } else {
 651                 int i;
 652
 653                 for (i = 0; i < psl->sl_count; i++) {
 654                         if (ipv6_addr_equal(&psl->sl_addr[i], src_addr))
 655                                 break;
 656                 }
 657                 if (mc->sfmode == MCAST_INCLUDE && i >= psl->sl_count)
 658                         rv = false;
 659                 if (mc->sfmode == MCAST_EXCLUDE && i < psl->sl_count)
 660                         rv = false;
 661         }
 662         rcu_read_unlock();
 663
 664         return rv;
 665 }
 666
 667 /* called with mc_lock */
 668 static void igmp6_group_added(struct ifmcaddr6 *mc)
 669 {
 670         struct net_device *dev = mc->idev->dev;
 671         char buf[MAX_ADDR_LEN];
 672
 673         if (IPV6_ADDR_MC_SCOPE(&mc->mca_addr) <
 674             IPV6_ADDR_SCOPE_LINKLOCAL)
 675                 return;
 676
 677         if (!(mc->mca_flags&MAF_LOADED)) {
 678                 mc->mca_flags |= MAF_LOADED;
 679                 if (ndisc_mc_map(&mc->mca_addr, buf, dev, 0) == 0)
 680                         dev_mc_add(dev, buf);
 681         }
 682
 683         if (!(dev->flags & IFF_UP) || (mc->mca_flags & MAF_NOREPORT))
 684                 return;
 685
 686         if (mld_in_v1_mode(mc->idev)) {
 687                 igmp6_join_group(mc);
 688                 return;
 689         }
 690         /* else v2 */
 691
 692         /* Based on RFC3810 6.1, for newly added INCLUDE SSM, we
 693          * should not send filter-mode change record as the mode
 694          * should be from IN() to IN(A).
 695          */
 696         if (mc->mca_sfmode == MCAST_EXCLUDE)
 697                 mc->mca_crcount = mc->idev->mc_qrv;
 698
 699         mld_ifc_event(mc->idev);
 700 }
 701
 702 /* called with mc_lock */
 703 static void igmp6_group_dropped(struct ifmcaddr6 *mc)
 704 {
 705         struct net_device *dev = mc->idev->dev;
 706         char buf[MAX_ADDR_LEN];
 707
 708         if (IPV6_ADDR_MC_SCOPE(&mc->mca_addr) <
 709             IPV6_ADDR_SCOPE_LINKLOCAL)
 710                 return;
 711
 712         if (mc->mca_flags&MAF_LOADED) {
 713                 mc->mca_flags &= ~MAF_LOADED;
 714                 if (ndisc_mc_map(&mc->mca_addr, buf, dev, 0) == 0)
 715                         dev_mc_del(dev, buf);
 716         }
 717
 718         if (mc->mca_flags & MAF_NOREPORT)
 719                 return;
 720
 721         if (!mc->idev->dead)
 722                 igmp6_leave_group(mc);
 723
 724         if (cancel_delayed_work(&mc->mca_work))
 725                 refcount_dec(&mc->mca_refcnt);
 726 }
 727
 728 /*
 729  * deleted ifmcaddr6 manipulation
 730  * called with mc_lock
 731  */
 732 static void mld_add_delrec(struct inet6_dev *idev, struct ifmcaddr6 *im)
 733 {
 734         struct ifmcaddr6 *pmc;
 735
 736         /* this is an "ifmcaddr6" for convenience; only the fields below
 737          * are actually used. In particular, the refcnt and users are not
 738          * used for management of the delete list. Using the same structure
 739          * for deleted items allows change reports to use common code with
 740          * non-deleted or query-response MCA's.
 741          */
 742         pmc = kzalloc(sizeof(*pmc), GFP_KERNEL);
 743         if (!pmc)
 744                 return;
 745
 746         pmc->idev = im->idev;
 747         in6_dev_hold(idev);
 748         pmc->mca_addr = im->mca_addr;
 749         pmc->mca_crcount = idev->mc_qrv;
 750         pmc->mca_sfmode = im->mca_sfmode;
 751         if (pmc->mca_sfmode == MCAST_INCLUDE) {
 752                 struct ip6_sf_list *psf;
 753
 754                 rcu_assign_pointer(pmc->mca_tomb,
 755                                    mc_dereference(im->mca_tomb, idev));
 756                 rcu_assign_pointer(pmc->mca_sources,
 757                                    mc_dereference(im->mca_sources, idev));
 758                 RCU_INIT_POINTER(im->mca_tomb, NULL);
 759                 RCU_INIT_POINTER(im->mca_sources, NULL);
 760
 761                 for_each_psf_mclock(pmc, psf)
 762                         psf->sf_crcount = pmc->mca_crcount;
 763         }
 764
 765         rcu_assign_pointer(pmc->next, idev->mc_tomb);
 766         rcu_assign_pointer(idev->mc_tomb, pmc);
 767 }
 768
 769 /* called with mc_lock */
 770 static void mld_del_delrec(struct inet6_dev *idev, struct ifmcaddr6 *im)
 771 {
 772         struct ip6_sf_list *psf, *sources, *tomb;
 773         struct in6_addr *pmca = &im->mca_addr;
 774         struct ifmcaddr6 *pmc, *pmc_prev;
 775
 776         pmc_prev = NULL;
 777         for_each_mc_tomb(idev, pmc) {
 778                 if (ipv6_addr_equal(&pmc->mca_addr, pmca))
 779                         break;
 780                 pmc_prev = pmc;
 781         }
 782         if (pmc) {
 783                 if (pmc_prev)
 784                         rcu_assign_pointer(pmc_prev->next, pmc->next);
 785                 else
 786                         rcu_assign_pointer(idev->mc_tomb, pmc->next);
 787         }
 788
 789         if (pmc) {
 790                 im->idev = pmc->idev;
 791                 if (im->mca_sfmode == MCAST_INCLUDE) {
 792                         tomb = rcu_replace_pointer(im->mca_tomb,
 793                                                    mc_dereference(pmc->mca_tomb, pmc->idev),
 794                                                    lockdep_is_held(&im->idev->mc_lock));
 795                         rcu_assign_pointer(pmc->mca_tomb, tomb);
 796
 797                         sources = rcu_replace_pointer(im->mca_sources,
 798                                                       mc_dereference(pmc->mca_sources, pmc->idev),
 799                                                       lockdep_is_held(&im->idev->mc_lock));
 800                         rcu_assign_pointer(pmc->mca_sources, sources);
 801                         for_each_psf_mclock(im, psf)
 802                                 psf->sf_crcount = idev->mc_qrv;
 803                 } else {
 804                         im->mca_crcount = idev->mc_qrv;
 805                 }
 806                 in6_dev_put(pmc->idev);
 807                 ip6_mc_clear_src(pmc);
 808                 kfree_rcu(pmc, rcu);
 809         }
 810 }
 811
 812 /* called with mc_lock */
 813 static void mld_clear_delrec(struct inet6_dev *idev)
 814 {
 815         struct ifmcaddr6 *pmc, *nextpmc;
 816
 817         pmc = mc_dereference(idev->mc_tomb, idev);
 818         RCU_INIT_POINTER(idev->mc_tomb, NULL);
 819
 820         for (; pmc; pmc = nextpmc) {
 821                 nextpmc = mc_dereference(pmc->next, idev);
 822                 ip6_mc_clear_src(pmc);
 823                 in6_dev_put(pmc->idev);
 824                 kfree_rcu(pmc, rcu);
 825         }
 826
 827         /* clear dead sources, too */
 828         for_each_mc_mclock(idev, pmc) {
 829                 struct ip6_sf_list *psf, *psf_next;
 830
 831                 psf = mc_dereference(pmc->mca_tomb, idev);
 832                 RCU_INIT_POINTER(pmc->mca_tomb, NULL);
 833                 for (; psf; psf = psf_next) {
 834                         psf_next = mc_dereference(psf->sf_next, idev);
 835                         kfree_rcu(psf, rcu);
 836                 }
 837         }
 838 }
 839
 840 static void mld_clear_query(struct inet6_dev *idev)
 841 {
 842         struct sk_buff *skb;
 843
 844         spin_lock_bh(&idev->mc_query_lock);
 845         while ((skb = __skb_dequeue(&idev->mc_query_queue)))
 846                 kfree_skb(skb);
 847         spin_unlock_bh(&idev->mc_query_lock);
 848 }
 849
 850 static void mld_clear_report(struct inet6_dev *idev)
 851 {
 852         struct sk_buff *skb;
 853
 854         spin_lock_bh(&idev->mc_report_lock);
 855         while ((skb = __skb_dequeue(&idev->mc_report_queue)))
 856                 kfree_skb(skb);
 857         spin_unlock_bh(&idev->mc_report_lock);
 858 }
 859
 860 static void mca_get(struct ifmcaddr6 *mc)
 861 {
 862         refcount_inc(&mc->mca_refcnt);
 863 }
 864
 865 static void ma_put(struct ifmcaddr6 *mc)
 866 {
 867         if (refcount_dec_and_test(&mc->mca_refcnt)) {
 868                 in6_dev_put(mc->idev);
 869                 kfree_rcu(mc, rcu);
 870         }
 871 }
 872
 873 /* called with mc_lock */
 874 static struct ifmcaddr6 *mca_alloc(struct inet6_dev *idev,
 875                                    const struct in6_addr *addr,
 876                                    unsigned int mode)
 877 {
 878         struct ifmcaddr6 *mc;
 879
 880         mc = kzalloc(sizeof(*mc), GFP_KERNEL);
 881         if (!mc)
 882                 return NULL;
 883
 884         INIT_DELAYED_WORK(&mc->mca_work, mld_mca_work);
 885
 886         mc->mca_addr = *addr;
 887         mc->idev = idev; /* reference taken by caller */
 888         mc->mca_users = 1;
 889         /* mca_stamp should be updated upon changes */
 890         mc->mca_cstamp = mc->mca_tstamp = jiffies;
 891         refcount_set(&mc->mca_refcnt, 1);
 892
 893         mc->mca_sfmode = mode;
 894         mc->mca_sfcount[mode] = 1;
 895
 896         if (ipv6_addr_is_ll_all_nodes(&mc->mca_addr) ||
 897             IPV6_ADDR_MC_SCOPE(&mc->mca_addr) < IPV6_ADDR_SCOPE_LINKLOCAL)
 898                 mc->mca_flags |= MAF_NOREPORT;
 899
 900         return mc;
 901 }
 902
 903 /*
 904  *      device multicast group inc (add if not found)
 905  */
 906 static int __ipv6_dev_mc_inc(struct net_device *dev,
 907                              const struct in6_addr *addr, unsigned int mode)
 908 {
 909         struct ifmcaddr6 *mc;
 910         struct inet6_dev *idev;
 911
 912         ASSERT_RTNL();
 913
 914         /* we need to take a reference on idev */
 915         idev = in6_dev_get(dev);
 916
 917         if (!idev)
 918                 return -EINVAL;
 919
 920         if (idev->dead) {
 921                 in6_dev_put(idev);
 922                 return -ENODEV;
 923         }
 924
 925         mutex_lock(&idev->mc_lock);
 926         for_each_mc_mclock(idev, mc) {
 927                 if (ipv6_addr_equal(&mc->mca_addr, addr)) {
 928                         mc->mca_users++;
 929                         ip6_mc_add_src(idev, &mc->mca_addr, mode, 0, NULL, 0);
 930                         mutex_unlock(&idev->mc_lock);
 931                         in6_dev_put(idev);
 932                         return 0;
 933                 }
 934         }
 935
 936         mc = mca_alloc(idev, addr, mode);
 937         if (!mc) {
 938                 mutex_unlock(&idev->mc_lock);
 939                 in6_dev_put(idev);
 940                 return -ENOMEM;
 941         }
 942
 943         rcu_assign_pointer(mc->next, idev->mc_list);
 944         rcu_assign_pointer(idev->mc_list, mc);
 945
 946         mca_get(mc);
 947
 948         mld_del_delrec(idev, mc);
 949         igmp6_group_added(mc);
 950         mutex_unlock(&idev->mc_lock);
 951         ma_put(mc);
 952         return 0;
 953 }
 954
 955 int ipv6_dev_mc_inc(struct net_device *dev, const struct in6_addr *addr)
 956 {
 957         return __ipv6_dev_mc_inc(dev, addr, MCAST_EXCLUDE);
 958 }
 959 EXPORT_SYMBOL(ipv6_dev_mc_inc);
 960
 961 /*
 962  * device multicast group del
 963  */
 964 int __ipv6_dev_mc_dec(struct inet6_dev *idev, const struct in6_addr *addr)
 965 {
 966         struct ifmcaddr6 *ma, __rcu **map;
 967
 968         ASSERT_RTNL();
 969
 970         mutex_lock(&idev->mc_lock);
 971         for (map = &idev->mc_list;
 972              (ma = mc_dereference(*map, idev));
 973              map = &ma->next) {
 974                 if (ipv6_addr_equal(&ma->mca_addr, addr)) {
 975                         if (--ma->mca_users == 0) {
 976                                 *map = ma->next;
 977
 978                                 igmp6_group_dropped(ma);
 979                                 ip6_mc_clear_src(ma);
 980                                 mutex_unlock(&idev->mc_lock);
 981
 982                                 ma_put(ma);
 983                                 return 0;
 984                         }
 985                         mutex_unlock(&idev->mc_lock);
 986                         return 0;
 987                 }
 988         }
 989
 990         mutex_unlock(&idev->mc_lock);
 991         return -ENOENT;
 992 }
 993
 994 int ipv6_dev_mc_dec(struct net_device *dev, const struct in6_addr *addr)
 995 {
 996         struct inet6_dev *idev;
 997         int err;
 998
 999         ASSERT_RTNL();
1000
1001         idev = __in6_dev_get(dev);
1002         if (!idev)
1003                 err = -ENODEV;
1004         else
1005                 err = __ipv6_dev_mc_dec(idev, addr);
1006
1007         return err;
1008 }
1009 EXPORT_SYMBOL(ipv6_dev_mc_dec);
1010
1011 /*
1012  *      check if the interface/address pair is valid
1013  */
1014 bool ipv6_chk_mcast_addr(struct net_device *dev, const struct in6_addr *group,
1015                          const struct in6_addr *src_addr)
1016 {
1017         struct inet6_dev *idev;
1018         struct ifmcaddr6 *mc;
1019         bool rv = false;
1020
1021         rcu_read_lock();
1022         idev = __in6_dev_get(dev);
1023         if (idev) {
1024                 for_each_mc_rcu(idev, mc) {
1025                         if (ipv6_addr_equal(&mc->mca_addr, group))
1026                                 break;
1027                 }
1028                 if (mc) {
1029                         if (src_addr && !ipv6_addr_any(src_addr)) {
1030                                 struct ip6_sf_list *psf;
1031
1032                                 for_each_psf_rcu(mc, psf) {
1033                                         if (ipv6_addr_equal(&psf->sf_addr, src_addr))
1034                                                 break;
1035                                 }
1036                                 if (psf)
1037                                         rv = psf->sf_count[MCAST_INCLUDE] ||
1038                                                 psf->sf_count[MCAST_EXCLUDE] !=
1039                                                 mc->mca_sfcount[MCAST_EXCLUDE];
1040                                 else
1041                                         rv = mc->mca_sfcount[MCAST_EXCLUDE] != 0;
1042                         } else
1043                                 rv = true; /* don't filter unspecified source */
1044                 }
1045         }
1046         rcu_read_unlock();
1047         return rv;
1048 }
1049
1050 /* called with mc_lock */
1051 static void mld_gq_start_work(struct inet6_dev *idev)
1052 {
1053         unsigned long tv = get_random_u32_below(idev->mc_maxdelay);
1054
1055         idev->mc_gq_running = 1;
1056         if (!mod_delayed_work(mld_wq, &idev->mc_gq_work, tv + 2))
1057                 in6_dev_hold(idev);
1058 }
1059
1060 /* called with mc_lock */
1061 static void mld_gq_stop_work(struct inet6_dev *idev)
1062 {
1063         idev->mc_gq_running = 0;
1064         if (cancel_delayed_work(&idev->mc_gq_work))
1065                 __in6_dev_put(idev);
1066 }
1067
1068 /* called with mc_lock */
1069 static void mld_ifc_start_work(struct inet6_dev *idev, unsigned long delay)
1070 {
1071         unsigned long tv = get_random_u32_below(delay);
1072
1073         if (!mod_delayed_work(mld_wq, &idev->mc_ifc_work, tv + 2))
1074                 in6_dev_hold(idev);
1075 }
1076
1077 /* called with mc_lock */
1078 static void mld_ifc_stop_work(struct inet6_dev *idev)
1079 {
1080         idev->mc_ifc_count = 0;
1081         if (cancel_delayed_work(&idev->mc_ifc_work))
1082                 __in6_dev_put(idev);
1083 }
1084
1085 /* called with mc_lock */
1086 static void mld_dad_start_work(struct inet6_dev *idev, unsigned long delay)
1087 {
1088         unsigned long tv = get_random_u32_below(delay);
1089
1090         if (!mod_delayed_work(mld_wq, &idev->mc_dad_work, tv + 2))
1091                 in6_dev_hold(idev);
1092 }
1093
1094 static void mld_dad_stop_work(struct inet6_dev *idev)
1095 {
1096         if (cancel_delayed_work(&idev->mc_dad_work))
1097                 __in6_dev_put(idev);
1098 }
1099
1100 static void mld_query_stop_work(struct inet6_dev *idev)
1101 {
1102         spin_lock_bh(&idev->mc_query_lock);
1103         if (cancel_delayed_work(&idev->mc_query_work))
1104                 __in6_dev_put(idev);
1105         spin_unlock_bh(&idev->mc_query_lock);
1106 }
1107
1108 static void mld_report_stop_work(struct inet6_dev *idev)
1109 {
1110         if (cancel_delayed_work_sync(&idev->mc_report_work))
1111                 __in6_dev_put(idev);
1112 }
1113
1114 /*
1115  * IGMP handling (alias multicast ICMPv6 messages)
1116  * called with mc_lock
1117  */
1118 static void igmp6_group_queried(struct ifmcaddr6 *ma, unsigned long resptime)
1119 {
1120         unsigned long delay = resptime;
1121
1122         /* Do not start work for these addresses */
1123         if (ipv6_addr_is_ll_all_nodes(&ma->mca_addr) ||
1124             IPV6_ADDR_MC_SCOPE(&ma->mca_addr) < IPV6_ADDR_SCOPE_LINKLOCAL)
1125                 return;
1126
1127         if (cancel_delayed_work(&ma->mca_work)) {
1128                 refcount_dec(&ma->mca_refcnt);
1129                 delay = ma->mca_work.timer.expires - jiffies;
1130         }
1131
1132         if (delay >= resptime)
1133                 delay = get_random_u32_below(resptime);
1134
1135         if (!mod_delayed_work(mld_wq, &ma->mca_work, delay))
1136                 refcount_inc(&ma->mca_refcnt);
1137         ma->mca_flags |= MAF_TIMER_RUNNING;
1138 }
1139
1140 /* mark EXCLUDE-mode sources
1141  * called with mc_lock
1142  */
1143 static bool mld_xmarksources(struct ifmcaddr6 *pmc, int nsrcs,
1144                              const struct in6_addr *srcs)
1145 {
1146         struct ip6_sf_list *psf;
1147         int i, scount;
1148
1149         scount = 0;
1150         for_each_psf_mclock(pmc, psf) {
1151                 if (scount == nsrcs)
1152                         break;
1153                 for (i = 0; i < nsrcs; i++) {
1154                         /* skip inactive filters */
1155                         if (psf->sf_count[MCAST_INCLUDE] ||
1156                             pmc->mca_sfcount[MCAST_EXCLUDE] !=
1157                             psf->sf_count[MCAST_EXCLUDE])
1158                                 break;
1159                         if (ipv6_addr_equal(&srcs[i], &psf->sf_addr)) {
1160                                 scount++;
1161                                 break;
1162                         }
1163                 }
1164         }
1165         pmc->mca_flags &= ~MAF_GSQUERY;
1166         if (scount == nsrcs)    /* all sources excluded */
1167                 return false;
1168         return true;
1169 }
1170
1171 /* called with mc_lock */
1172 static bool mld_marksources(struct ifmcaddr6 *pmc, int nsrcs,
1173                             const struct in6_addr *srcs)
1174 {
1175         struct ip6_sf_list *psf;
1176         int i, scount;
1177
1178         if (pmc->mca_sfmode == MCAST_EXCLUDE)
1179                 return mld_xmarksources(pmc, nsrcs, srcs);
1180
1181         /* mark INCLUDE-mode sources */
1182
1183         scount = 0;
1184         for_each_psf_mclock(pmc, psf) {
1185                 if (scount == nsrcs)
1186                         break;
1187                 for (i = 0; i < nsrcs; i++) {
1188                         if (ipv6_addr_equal(&srcs[i], &psf->sf_addr)) {
1189                                 psf->sf_gsresp = 1;
1190                                 scount++;
1191                                 break;
1192                         }
1193                 }
1194         }
1195         if (!scount) {
1196                 pmc->mca_flags &= ~MAF_GSQUERY;
1197                 return false;
1198         }
1199         pmc->mca_flags |= MAF_GSQUERY;
1200         return true;
1201 }
1202
1203 static int mld_force_mld_version(const struct inet6_dev *idev)
1204 {
1205         /* Normally, both are 0 here. If enforcement to a particular is
1206          * being used, individual device enforcement will have a lower
1207          * precedence over 'all' device (.../conf/all/force_mld_version).
1208          */
1209
1210         if (dev_net(idev->dev)->ipv6.devconf_all->force_mld_version != 0)
1211                 return dev_net(idev->dev)->ipv6.devconf_all->force_mld_version;
1212         else
1213                 return idev->cnf.force_mld_version;
1214 }
1215
1216 static bool mld_in_v2_mode_only(const struct inet6_dev *idev)
1217 {
1218         return mld_force_mld_version(idev) == 2;
1219 }
1220
1221 static bool mld_in_v1_mode_only(const struct inet6_dev *idev)
1222 {
1223         return mld_force_mld_version(idev) == 1;
1224 }
1225
1226 static bool mld_in_v1_mode(const struct inet6_dev *idev)
1227 {
1228         if (mld_in_v2_mode_only(idev))
1229                 return false;
1230         if (mld_in_v1_mode_only(idev))
1231                 return true;
1232         if (idev->mc_v1_seen && time_before(jiffies, idev->mc_v1_seen))
1233                 return true;
1234
1235         return false;
1236 }
1237
1238 static void mld_set_v1_mode(struct inet6_dev *idev)
1239 {
1240         /* RFC3810, relevant sections:
1241          *  - 9.1. Robustness Variable
1242          *  - 9.2. Query Interval
1243          *  - 9.3. Query Response Interval
1244          *  - 9.12. Older Version Querier Present Timeout
1245          */
1246         unsigned long switchback;
1247
1248         switchback = (idev->mc_qrv * idev->mc_qi) + idev->mc_qri;
1249
1250         idev->mc_v1_seen = jiffies + switchback;
1251 }
1252
1253 static void mld_update_qrv(struct inet6_dev *idev,
1254                            const struct mld2_query *mlh2)
1255 {
1256         /* RFC3810, relevant sections:
1257          *  - 5.1.8. QRV (Querier's Robustness Variable)
1258          *  - 9.1. Robustness Variable
1259          */
1260
1261         /* The value of the Robustness Variable MUST NOT be zero,
1262          * and SHOULD NOT be one. Catch this here if we ever run
1263          * into such a case in future.
1264          */
1265         const int min_qrv = min(MLD_QRV_DEFAULT, sysctl_mld_qrv);
1266         WARN_ON(idev->mc_qrv == 0);
1267
1268         if (mlh2->mld2q_qrv > 0)
1269                 idev->mc_qrv = mlh2->mld2q_qrv;
1270
1271         if (unlikely(idev->mc_qrv < min_qrv)) {
1272                 net_warn_ratelimited("IPv6: MLD: clamping QRV from %u to %u!\n",
1273                                      idev->mc_qrv, min_qrv);
1274                 idev->mc_qrv = min_qrv;
1275         }
1276 }
1277
1278 static void mld_update_qi(struct inet6_dev *idev,
1279                           const struct mld2_query *mlh2)
1280 {
1281         /* RFC3810, relevant sections:
1282          *  - 5.1.9. QQIC (Querier's Query Interval Code)
1283          *  - 9.2. Query Interval
1284          *  - 9.12. Older Version Querier Present Timeout
1285          *    (the [Query Interval] in the last Query received)
1286          */
1287         unsigned long mc_qqi;
1288
1289         if (mlh2->mld2q_qqic < 128) {
1290                 mc_qqi = mlh2->mld2q_qqic;
1291         } else {
1292                 unsigned long mc_man, mc_exp;
1293
1294                 mc_exp = MLDV2_QQIC_EXP(mlh2->mld2q_qqic);
1295                 mc_man = MLDV2_QQIC_MAN(mlh2->mld2q_qqic);
1296
1297                 mc_qqi = (mc_man | 0x10) << (mc_exp + 3);
1298         }
1299
1300         idev->mc_qi = mc_qqi * HZ;
1301 }
1302
1303 static void mld_update_qri(struct inet6_dev *idev,
1304                            const struct mld2_query *mlh2)
1305 {
1306         /* RFC3810, relevant sections:
1307          *  - 5.1.3. Maximum Response Code
1308          *  - 9.3. Query Response Interval
1309          */
1310         idev->mc_qri = msecs_to_jiffies(mldv2_mrc(mlh2));
1311 }
1312
1313 static int mld_process_v1(struct inet6_dev *idev, struct mld_msg *mld,
1314                           unsigned long *max_delay, bool v1_query)
1315 {
1316         unsigned long mldv1_md;
1317
1318         /* Ignore v1 queries */
1319         if (mld_in_v2_mode_only(idev))
1320                 return -EINVAL;
1321
1322         mldv1_md = ntohs(mld->mld_maxdelay);
1323
1324         /* When in MLDv1 fallback and a MLDv2 router start-up being
1325          * unaware of current MLDv1 operation, the MRC == MRD mapping
1326          * only works when the exponential algorithm is not being
1327          * used (as MLDv1 is unaware of such things).
1328          *
1329          * According to the RFC author, the MLDv2 implementations
1330          * he's aware of all use a MRC < 32768 on start up queries.
1331          *
1332          * Thus, should we *ever* encounter something else larger
1333          * than that, just assume the maximum possible within our
1334          * reach.
1335          */
1336         if (!v1_query)
1337                 mldv1_md = min(mldv1_md, MLDV1_MRD_MAX_COMPAT);
1338
1339         *max_delay = max(msecs_to_jiffies(mldv1_md), 1UL);
1340
1341         /* MLDv1 router present: we need to go into v1 mode *only*
1342          * when an MLDv1 query is received as per section 9.12. of
1343          * RFC3810! And we know from RFC2710 section 3.7 that MLDv1
1344          * queries MUST be of exactly 24 octets.
1345          */
1346         if (v1_query)
1347                 mld_set_v1_mode(idev);
1348
1349         /* cancel MLDv2 report work */
1350         mld_gq_stop_work(idev);
1351         /* cancel the interface change work */
1352         mld_ifc_stop_work(idev);
1353         /* clear deleted report items */
1354         mld_clear_delrec(idev);
1355
1356         return 0;
1357 }
1358
1359 static void mld_process_v2(struct inet6_dev *idev, struct mld2_query *mld,
1360                            unsigned long *max_delay)
1361 {
1362         *max_delay = max(msecs_to_jiffies(mldv2_mrc(mld)), 1UL);
1363
1364         mld_update_qrv(idev, mld);
1365         mld_update_qi(idev, mld);
1366         mld_update_qri(idev, mld);
1367
1368         idev->mc_maxdelay = *max_delay;
1369
1370         return;
1371 }
1372
1373 /* called with rcu_read_lock() */
1374 void igmp6_event_query(struct sk_buff *skb)
1375 {
1376         struct inet6_dev *idev = __in6_dev_get(skb->dev);
1377
1378         if (!idev || idev->dead)
1379                 goto out;
1380
1381         spin_lock_bh(&idev->mc_query_lock);
1382         if (skb_queue_len(&idev->mc_query_queue) < MLD_MAX_SKBS) {
1383                 __skb_queue_tail(&idev->mc_query_queue, skb);
1384                 if (!mod_delayed_work(mld_wq, &idev->mc_query_work, 0))
1385                         in6_dev_hold(idev);
1386                 skb = NULL;
1387         }
1388         spin_unlock_bh(&idev->mc_query_lock);
1389 out:
1390         kfree_skb(skb);
1391 }
1392
1393 static void __mld_query_work(struct sk_buff *skb)
1394 {
1395         struct mld2_query *mlh2 = NULL;
1396         const struct in6_addr *group;
1397         unsigned long max_delay;
1398         struct inet6_dev *idev;
1399         struct ifmcaddr6 *ma;
1400         struct mld_msg *mld;
1401         int group_type;
1402         int mark = 0;
1403         int len, err;
1404
1405         if (!pskb_may_pull(skb, sizeof(struct in6_addr)))
1406                 goto kfree_skb;
1407
1408         /* compute payload length excluding extension headers */
1409         len = ntohs(ipv6_hdr(skb)->payload_len) + sizeof(struct ipv6hdr);
1410         len -= skb_network_header_len(skb);
1411
1412         /* RFC3810 6.2
1413          * Upon reception of an MLD message that contains a Query, the node
1414          * checks if the source address of the message is a valid link-local
1415          * address, if the Hop Limit is set to 1, and if the Router Alert
1416          * option is present in the Hop-By-Hop Options header of the IPv6
1417          * packet.  If any of these checks fails, the packet is dropped.
1418          */
1419         if (!(ipv6_addr_type(&ipv6_hdr(skb)->saddr) & IPV6_ADDR_LINKLOCAL) ||
1420             ipv6_hdr(skb)->hop_limit != 1 ||
1421             !(IP6CB(skb)->flags & IP6SKB_ROUTERALERT) ||
1422             IP6CB(skb)->ra != htons(IPV6_OPT_ROUTERALERT_MLD))
1423                 goto kfree_skb;
1424
1425         idev = in6_dev_get(skb->dev);
1426         if (!idev)
1427                 goto kfree_skb;
1428
1429         mld = (struct mld_msg *)icmp6_hdr(skb);
1430         group = &mld->mld_mca;
1431         group_type = ipv6_addr_type(group);
1432
1433         if (group_type != IPV6_ADDR_ANY &&
1434             !(group_type&IPV6_ADDR_MULTICAST))
1435                 goto out;
1436
1437         if (len < MLD_V1_QUERY_LEN) {
1438                 goto out;
1439         } else if (len == MLD_V1_QUERY_LEN || mld_in_v1_mode(idev)) {
1440                 err = mld_process_v1(idev, mld, &max_delay,
1441                                      len == MLD_V1_QUERY_LEN);
1442                 if (err < 0)
1443                         goto out;
1444         } else if (len >= MLD_V2_QUERY_LEN_MIN) {
1445                 int srcs_offset = sizeof(struct mld2_query) -
1446                                   sizeof(struct icmp6hdr);
1447
1448                 if (!pskb_may_pull(skb, srcs_offset))
1449                         goto out;
1450
1451                 mlh2 = (struct mld2_query *)skb_transport_header(skb);
1452
1453                 mld_process_v2(idev, mlh2, &max_delay);
1454
1455                 if (group_type == IPV6_ADDR_ANY) { /* general query */
1456                         if (mlh2->mld2q_nsrcs)
1457                                 goto out; /* no sources allowed */
1458
1459                         mld_gq_start_work(idev);
1460                         goto out;
1461                 }
1462                 /* mark sources to include, if group & source-specific */
1463                 if (mlh2->mld2q_nsrcs != 0) {
1464                         if (!pskb_may_pull(skb, srcs_offset +
1465                             ntohs(mlh2->mld2q_nsrcs) * sizeof(struct in6_addr)))
1466                                 goto out;
1467
1468                         mlh2 = (struct mld2_query *)skb_transport_header(skb);
1469                         mark = 1;
1470                 }
1471         } else {
1472                 goto out;
1473         }
1474
1475         if (group_type == IPV6_ADDR_ANY) {
1476                 for_each_mc_mclock(idev, ma) {
1477                         igmp6_group_queried(ma, max_delay);
1478                 }
1479         } else {
1480                 for_each_mc_mclock(idev, ma) {
1481                         if (!ipv6_addr_equal(group, &ma->mca_addr))
1482                                 continue;
1483                         if (ma->mca_flags & MAF_TIMER_RUNNING) {
1484                                 /* gsquery <- gsquery && mark */
1485                                 if (!mark)
1486                                         ma->mca_flags &= ~MAF_GSQUERY;
1487                         } else {
1488                                 /* gsquery <- mark */
1489                                 if (mark)
1490                                         ma->mca_flags |= MAF_GSQUERY;
1491                                 else
1492                                         ma->mca_flags &= ~MAF_GSQUERY;
1493                         }
1494                         if (!(ma->mca_flags & MAF_GSQUERY) ||
1495                             mld_marksources(ma, ntohs(mlh2->mld2q_nsrcs), mlh2->mld2q_srcs))
1496                                 igmp6_group_queried(ma, max_delay);
1497                         break;
1498                 }
1499         }
1500
1501 out:
1502         in6_dev_put(idev);
1503 kfree_skb:
1504         consume_skb(skb);
1505 }
1506
1507 static void mld_query_work(struct work_struct *work)
1508 {
1509         struct inet6_dev *idev = container_of(to_delayed_work(work),
1510                                               struct inet6_dev,
1511                                               mc_query_work);
1512         struct sk_buff_head q;
1513         struct sk_buff *skb;
1514         bool rework = false;
1515         int cnt = 0;
1516
1517         skb_queue_head_init(&q);
1518
1519         spin_lock_bh(&idev->mc_query_lock);
1520         while ((skb = __skb_dequeue(&idev->mc_query_queue))) {
1521                 __skb_queue_tail(&q, skb);
1522
1523                 if (++cnt >= MLD_MAX_QUEUE) {
1524                         rework = true;
1525                         break;
1526                 }
1527         }
1528         spin_unlock_bh(&idev->mc_query_lock);
1529
1530         mutex_lock(&idev->mc_lock);
1531         while ((skb = __skb_dequeue(&q)))
1532                 __mld_query_work(skb);
1533         mutex_unlock(&idev->mc_lock);
1534
1535         if (rework && queue_delayed_work(mld_wq, &idev->mc_query_work, 0))
1536                 return;
1537
1538         in6_dev_put(idev);
1539 }
1540
1541 /* called with rcu_read_lock() */
1542 void igmp6_event_report(struct sk_buff *skb)
1543 {
1544         struct inet6_dev *idev = __in6_dev_get(skb->dev);
1545
1546         if (!idev || idev->dead)
1547                 goto out;
1548
1549         spin_lock_bh(&idev->mc_report_lock);
1550         if (skb_queue_len(&idev->mc_report_queue) < MLD_MAX_SKBS) {
1551                 __skb_queue_tail(&idev->mc_report_queue, skb);
1552                 if (!mod_delayed_work(mld_wq, &idev->mc_report_work, 0))
1553                         in6_dev_hold(idev);
1554                 skb = NULL;
1555         }
1556         spin_unlock_bh(&idev->mc_report_lock);
1557 out:
1558         kfree_skb(skb);
1559 }
1560
1561 static void __mld_report_work(struct sk_buff *skb)
1562 {
1563         struct inet6_dev *idev;
1564         struct ifmcaddr6 *ma;
1565         struct mld_msg *mld;
1566         int addr_type;
1567
1568         /* Our own report looped back. Ignore it. */
1569         if (skb->pkt_type == PACKET_LOOPBACK)
1570                 goto kfree_skb;
1571
1572         /* send our report if the MC router may not have heard this report */
1573         if (skb->pkt_type != PACKET_MULTICAST &&
1574             skb->pkt_type != PACKET_BROADCAST)
1575                 goto kfree_skb;
1576
1577         if (!pskb_may_pull(skb, sizeof(*mld) - sizeof(struct icmp6hdr)))
1578                 goto kfree_skb;
1579
1580         mld = (struct mld_msg *)icmp6_hdr(skb);
1581
1582         /* Drop reports with not link local source */
1583         addr_type = ipv6_addr_type(&ipv6_hdr(skb)->saddr);
1584         if (addr_type != IPV6_ADDR_ANY &&
1585             !(addr_type&IPV6_ADDR_LINKLOCAL))
1586                 goto kfree_skb;
1587
1588         idev = in6_dev_get(skb->dev);
1589         if (!idev)
1590                 goto kfree_skb;
1591
1592         /*
1593          *      Cancel the work for this group
1594          */
1595
1596         for_each_mc_mclock(idev, ma) {
1597                 if (ipv6_addr_equal(&ma->mca_addr, &mld->mld_mca)) {
1598                         if (cancel_delayed_work(&ma->mca_work))
1599                                 refcount_dec(&ma->mca_refcnt);
1600                         ma->mca_flags &= ~(MAF_LAST_REPORTER |
1601                                            MAF_TIMER_RUNNING);
1602                         break;
1603                 }
1604         }
1605
1606         in6_dev_put(idev);
1607 kfree_skb:
1608         consume_skb(skb);
1609 }
1610
1611 static void mld_report_work(struct work_struct *work)
1612 {
1613         struct inet6_dev *idev = container_of(to_delayed_work(work),
1614                                               struct inet6_dev,
1615                                               mc_report_work);
1616         struct sk_buff_head q;
1617         struct sk_buff *skb;
1618         bool rework = false;
1619         int cnt = 0;
1620
1621         skb_queue_head_init(&q);
1622         spin_lock_bh(&idev->mc_report_lock);
1623         while ((skb = __skb_dequeue(&idev->mc_report_queue))) {
1624                 __skb_queue_tail(&q, skb);
1625
1626                 if (++cnt >= MLD_MAX_QUEUE) {
1627                         rework = true;
1628                         break;
1629                 }
1630         }
1631         spin_unlock_bh(&idev->mc_report_lock);
1632
1633         mutex_lock(&idev->mc_lock);
1634         while ((skb = __skb_dequeue(&q)))
1635                 __mld_report_work(skb);
1636         mutex_unlock(&idev->mc_lock);
1637
1638         if (rework && queue_delayed_work(mld_wq, &idev->mc_report_work, 0))
1639                 return;
1640
1641         in6_dev_put(idev);
1642 }
1643
1644 static bool is_in(struct ifmcaddr6 *pmc, struct ip6_sf_list *psf, int type,
1645                   int gdeleted, int sdeleted)
1646 {
1647         switch (type) {
1648         case MLD2_MODE_IS_INCLUDE:
1649         case MLD2_MODE_IS_EXCLUDE:
1650                 if (gdeleted || sdeleted)
1651                         return false;
1652                 if (!((pmc->mca_flags & MAF_GSQUERY) && !psf->sf_gsresp)) {
1653                         if (pmc->mca_sfmode == MCAST_INCLUDE)
1654                                 return true;
1655                         /* don't include if this source is excluded
1656                          * in all filters
1657                          */
1658                         if (psf->sf_count[MCAST_INCLUDE])
1659                                 return type == MLD2_MODE_IS_INCLUDE;
1660                         return pmc->mca_sfcount[MCAST_EXCLUDE] ==
1661                                 psf->sf_count[MCAST_EXCLUDE];
1662                 }
1663                 return false;
1664         case MLD2_CHANGE_TO_INCLUDE:
1665                 if (gdeleted || sdeleted)
1666                         return false;
1667                 return psf->sf_count[MCAST_INCLUDE] != 0;
1668         case MLD2_CHANGE_TO_EXCLUDE:
1669                 if (gdeleted || sdeleted)
1670                         return false;
1671                 if (pmc->mca_sfcount[MCAST_EXCLUDE] == 0 ||
1672                     psf->sf_count[MCAST_INCLUDE])
1673                         return false;
1674                 return pmc->mca_sfcount[MCAST_EXCLUDE] ==
1675                         psf->sf_count[MCAST_EXCLUDE];
1676         case MLD2_ALLOW_NEW_SOURCES:
1677                 if (gdeleted || !psf->sf_crcount)
1678                         return false;
1679                 return (pmc->mca_sfmode == MCAST_INCLUDE) ^ sdeleted;
1680         case MLD2_BLOCK_OLD_SOURCES:
1681                 if (pmc->mca_sfmode == MCAST_INCLUDE)
1682                         return gdeleted || (psf->sf_crcount && sdeleted);
1683                 return psf->sf_crcount && !gdeleted && !sdeleted;
1684         }
1685         return false;
1686 }
1687
1688 static int
1689 mld_scount(struct ifmcaddr6 *pmc, int type, int gdeleted, int sdeleted)
1690 {
1691         struct ip6_sf_list *psf;
1692         int scount = 0;
1693
1694         for_each_psf_mclock(pmc, psf) {
1695                 if (!is_in(pmc, psf, type, gdeleted, sdeleted))
1696                         continue;
1697                 scount++;
1698         }
1699         return scount;
1700 }
1701
1702 static void ip6_mc_hdr(const struct sock *sk, struct sk_buff *skb,
1703                        struct net_device *dev, const struct in6_addr *saddr,
1704                        const struct in6_addr *daddr, int proto, int len)
1705 {
1706         struct ipv6hdr *hdr;
1707
1708         skb->protocol = htons(ETH_P_IPV6);
1709         skb->dev = dev;
1710
1711         skb_reset_network_header(skb);
1712         skb_put(skb, sizeof(struct ipv6hdr));
1713         hdr = ipv6_hdr(skb);
1714
1715         ip6_flow_hdr(hdr, 0, 0);
1716
1717         hdr->payload_len = htons(len);
1718         hdr->nexthdr = proto;
1719         hdr->hop_limit = inet6_sk(sk)->hop_limit;
1720
1721         hdr->saddr = *saddr;
1722         hdr->daddr = *daddr;
1723 }
1724
1725 static struct sk_buff *mld_newpack(struct inet6_dev *idev, unsigned int mtu)
1726 {
1727         u8 ra[8] = { IPPROTO_ICMPV6, 0, IPV6_TLV_ROUTERALERT,
1728                      2, 0, 0, IPV6_TLV_PADN, 0 };
1729         struct net_device *dev = idev->dev;
1730         int hlen = LL_RESERVED_SPACE(dev);
1731         int tlen = dev->needed_tailroom;
1732         struct net *net = dev_net(dev);
1733         const struct in6_addr *saddr;
1734         struct in6_addr addr_buf;
1735         struct mld2_report *pmr;
1736         struct sk_buff *skb;
1737         unsigned int size;
1738         struct sock *sk;
1739         int err;
1740
1741         sk = net->ipv6.igmp_sk;
1742         /* we assume size > sizeof(ra) here
1743          * Also try to not allocate high-order pages for big MTU
1744          */
1745         size = min_t(int, mtu, PAGE_SIZE / 2) + hlen + tlen;
1746         skb = sock_alloc_send_skb(sk, size, 1, &err);
1747         if (!skb)
1748                 return NULL;
1749
1750         skb->priority = TC_PRIO_CONTROL;
1751         skb_reserve(skb, hlen);
1752         skb_tailroom_reserve(skb, mtu, tlen);
1753
1754         if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
1755                 /* <draft-ietf-magma-mld-source-05.txt>:
1756                  * use unspecified address as the source address
1757                  * when a valid link-local address is not available.
1758                  */
1759                 saddr = &in6addr_any;
1760         } else
1761                 saddr = &addr_buf;
1762
1763         ip6_mc_hdr(sk, skb, dev, saddr, &mld2_all_mcr, NEXTHDR_HOP, 0);
1764
1765         skb_put_data(skb, ra, sizeof(ra));
1766
1767         skb_set_transport_header(skb, skb_tail_pointer(skb) - skb->data);
1768         skb_put(skb, sizeof(*pmr));
1769         pmr = (struct mld2_report *)skb_transport_header(skb);
1770         pmr->mld2r_type = ICMPV6_MLD2_REPORT;
1771         pmr->mld2r_resv1 = 0;
1772         pmr->mld2r_cksum = 0;
1773         pmr->mld2r_resv2 = 0;
1774         pmr->mld2r_ngrec = 0;
1775         return skb;
1776 }
1777
1778 static void mld_sendpack(struct sk_buff *skb)
1779 {
1780         struct ipv6hdr *pip6 = ipv6_hdr(skb);
1781         struct mld2_report *pmr =
1782                               (struct mld2_report *)skb_transport_header(skb);
1783         int payload_len, mldlen;
1784         struct inet6_dev *idev;
1785         struct net *net = dev_net(skb->dev);
1786         int err;
1787         struct flowi6 fl6;
1788         struct dst_entry *dst;
1789
1790         rcu_read_lock();
1791         idev = __in6_dev_get(skb->dev);
1792         IP6_UPD_PO_STATS(net, idev, IPSTATS_MIB_OUT, skb->len);
1793
1794         payload_len = (skb_tail_pointer(skb) - skb_network_header(skb)) -
1795                 sizeof(*pip6);
1796         mldlen = skb_tail_pointer(skb) - skb_transport_header(skb);
1797         pip6->payload_len = htons(payload_len);
1798
1799         pmr->mld2r_cksum = csum_ipv6_magic(&pip6->saddr, &pip6->daddr, mldlen,
1800                                            IPPROTO_ICMPV6,
1801                                            csum_partial(skb_transport_header(skb),
1802                                                         mldlen, 0));
1803
1804         icmpv6_flow_init(net->ipv6.igmp_sk, &fl6, ICMPV6_MLD2_REPORT,
1805                          &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
1806                          skb->dev->ifindex);
1807         dst = icmp6_dst_alloc(skb->dev, &fl6);
1808
1809         err = 0;
1810         if (IS_ERR(dst)) {
1811                 err = PTR_ERR(dst);
1812                 dst = NULL;
1813         }
1814         skb_dst_set(skb, dst);
1815         if (err)
1816                 goto err_out;
1817
1818         err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
1819                       net, net->ipv6.igmp_sk, skb, NULL, skb->dev,
1820                       dst_output);
1821 out:
1822         if (!err) {
1823                 ICMP6MSGOUT_INC_STATS(net, idev, ICMPV6_MLD2_REPORT);
1824                 ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
1825         } else {
1826                 IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
1827         }
1828
1829         rcu_read_unlock();
1830         return;
1831
1832 err_out:
1833         kfree_skb(skb);
1834         goto out;
1835 }
1836
1837 static int grec_size(struct ifmcaddr6 *pmc, int type, int gdel, int sdel)
1838 {
1839         return sizeof(struct mld2_grec) + 16 * mld_scount(pmc,type,gdel,sdel);
1840 }
1841
1842 static struct sk_buff *add_grhead(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1843         int type, struct mld2_grec **ppgr, unsigned int mtu)
1844 {
1845         struct mld2_report *pmr;
1846         struct mld2_grec *pgr;
1847
1848         if (!skb) {
1849                 skb = mld_newpack(pmc->idev, mtu);
1850                 if (!skb)
1851                         return NULL;
1852         }
1853         pgr = skb_put(skb, sizeof(struct mld2_grec));
1854         pgr->grec_type = type;
1855         pgr->grec_auxwords = 0;
1856         pgr->grec_nsrcs = 0;
1857         pgr->grec_mca = pmc->mca_addr;  /* structure copy */
1858         pmr = (struct mld2_report *)skb_transport_header(skb);
1859         pmr->mld2r_ngrec = htons(ntohs(pmr->mld2r_ngrec)+1);
1860         *ppgr = pgr;
1861         return skb;
1862 }
1863
1864 #define AVAILABLE(skb)  ((skb) ? skb_availroom(skb) : 0)
1865
1866 /* called with mc_lock */
1867 static struct sk_buff *add_grec(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1868                                 int type, int gdeleted, int sdeleted,
1869                                 int crsend)
1870 {
1871         struct ip6_sf_list *psf, *psf_prev, *psf_next;
1872         int scount, stotal, first, isquery, truncate;
1873         struct ip6_sf_list __rcu **psf_list;
1874         struct inet6_dev *idev = pmc->idev;
1875         struct net_device *dev = idev->dev;
1876         struct mld2_grec *pgr = NULL;
1877         struct mld2_report *pmr;
1878         unsigned int mtu;
1879
1880         if (pmc->mca_flags & MAF_NOREPORT)
1881                 return skb;
1882
1883         mtu = READ_ONCE(dev->mtu);
1884         if (mtu < IPV6_MIN_MTU)
1885                 return skb;
1886
1887         isquery = type == MLD2_MODE_IS_INCLUDE ||
1888                   type == MLD2_MODE_IS_EXCLUDE;
1889         truncate = type == MLD2_MODE_IS_EXCLUDE ||
1890                     type == MLD2_CHANGE_TO_EXCLUDE;
1891
1892         stotal = scount = 0;
1893
1894         psf_list = sdeleted ? &pmc->mca_tomb : &pmc->mca_sources;
1895
1896         if (!rcu_access_pointer(*psf_list))
1897                 goto empty_source;
1898
1899         pmr = skb ? (struct mld2_report *)skb_transport_header(skb) : NULL;
1900
1901         /* EX and TO_EX get a fresh packet, if needed */
1902         if (truncate) {
1903                 if (pmr && pmr->mld2r_ngrec &&
1904                     AVAILABLE(skb) < grec_size(pmc, type, gdeleted, sdeleted)) {
1905                         if (skb)
1906                                 mld_sendpack(skb);
1907                         skb = mld_newpack(idev, mtu);
1908                 }
1909         }
1910         first = 1;
1911         psf_prev = NULL;
1912         for (psf = mc_dereference(*psf_list, idev);
1913              psf;
1914              psf = psf_next) {
1915                 struct in6_addr *psrc;
1916
1917                 psf_next = mc_dereference(psf->sf_next, idev);
1918
1919                 if (!is_in(pmc, psf, type, gdeleted, sdeleted) && !crsend) {
1920                         psf_prev = psf;
1921                         continue;
1922                 }
1923
1924                 /* Based on RFC3810 6.1. Should not send source-list change
1925                  * records when there is a filter mode change.
1926                  */
1927                 if (((gdeleted && pmc->mca_sfmode == MCAST_EXCLUDE) ||
1928                      (!gdeleted && pmc->mca_crcount)) &&
1929                     (type == MLD2_ALLOW_NEW_SOURCES ||
1930                      type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount)
1931                         goto decrease_sf_crcount;
1932
1933                 /* clear marks on query responses */
1934                 if (isquery)
1935                         psf->sf_gsresp = 0;
1936
1937                 if (AVAILABLE(skb) < sizeof(*psrc) +
1938                     first*sizeof(struct mld2_grec)) {
1939                         if (truncate && !first)
1940                                 break;   /* truncate these */
1941                         if (pgr)
1942                                 pgr->grec_nsrcs = htons(scount);
1943                         if (skb)
1944                                 mld_sendpack(skb);
1945                         skb = mld_newpack(idev, mtu);
1946                         first = 1;
1947                         scount = 0;
1948                 }
1949                 if (first) {
1950                         skb = add_grhead(skb, pmc, type, &pgr, mtu);
1951                         first = 0;
1952                 }
1953                 if (!skb)
1954                         return NULL;
1955                 psrc = skb_put(skb, sizeof(*psrc));
1956                 *psrc = psf->sf_addr;
1957                 scount++; stotal++;
1958                 if ((type == MLD2_ALLOW_NEW_SOURCES ||
1959                      type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount) {
1960 decrease_sf_crcount:
1961                         psf->sf_crcount--;
1962                         if ((sdeleted || gdeleted) && psf->sf_crcount == 0) {
1963                                 if (psf_prev)
1964                                         rcu_assign_pointer(psf_prev->sf_next,
1965                                                            mc_dereference(psf->sf_next, idev));
1966                                 else
1967                                         rcu_assign_pointer(*psf_list,
1968                                                            mc_dereference(psf->sf_next, idev));
1969                                 kfree_rcu(psf, rcu);
1970                                 continue;
1971                         }
1972                 }
1973                 psf_prev = psf;
1974         }
1975
1976 empty_source:
1977         if (!stotal) {
1978                 if (type == MLD2_ALLOW_NEW_SOURCES ||
1979                     type == MLD2_BLOCK_OLD_SOURCES)
1980                         return skb;
1981                 if (pmc->mca_crcount || isquery || crsend) {
1982                         /* make sure we have room for group header */
1983                         if (skb && AVAILABLE(skb) < sizeof(struct mld2_grec)) {
1984                                 mld_sendpack(skb);
1985                                 skb = NULL; /* add_grhead will get a new one */
1986                         }
1987                         skb = add_grhead(skb, pmc, type, &pgr, mtu);
1988                 }
1989         }
1990         if (pgr)
1991                 pgr->grec_nsrcs = htons(scount);
1992
1993         if (isquery)
1994                 pmc->mca_flags &= ~MAF_GSQUERY; /* clear query state */
1995         return skb;
1996 }
1997
1998 /* called with mc_lock */
1999 static void mld_send_report(struct inet6_dev *idev, struct ifmcaddr6 *pmc)
2000 {
2001         struct sk_buff *skb = NULL;
2002         int type;
2003
2004         if (!pmc) {
2005                 for_each_mc_mclock(idev, pmc) {
2006                         if (pmc->mca_flags & MAF_NOREPORT)
2007                                 continue;
2008                         if (pmc->mca_sfcount[MCAST_EXCLUDE])
2009                                 type = MLD2_MODE_IS_EXCLUDE;
2010                         else
2011                                 type = MLD2_MODE_IS_INCLUDE;
2012                         skb = add_grec(skb, pmc, type, 0, 0, 0);
2013                 }
2014         } else {
2015                 if (pmc->mca_sfcount[MCAST_EXCLUDE])
2016                         type = MLD2_MODE_IS_EXCLUDE;
2017                 else
2018                         type = MLD2_MODE_IS_INCLUDE;
2019                 skb = add_grec(skb, pmc, type, 0, 0, 0);
2020         }
2021         if (skb)
2022                 mld_sendpack(skb);
2023 }
2024
2025 /*
2026  * remove zero-count source records from a source filter list
2027  * called with mc_lock
2028  */
2029 static void mld_clear_zeros(struct ip6_sf_list __rcu **ppsf, struct inet6_dev *idev)
2030 {
2031         struct ip6_sf_list *psf_prev, *psf_next, *psf;
2032
2033         psf_prev = NULL;
2034         for (psf = mc_dereference(*ppsf, idev);
2035              psf;
2036              psf = psf_next) {
2037                 psf_next = mc_dereference(psf->sf_next, idev);
2038                 if (psf->sf_crcount == 0) {
2039                         if (psf_prev)
2040                                 rcu_assign_pointer(psf_prev->sf_next,
2041                                                    mc_dereference(psf->sf_next, idev));
2042                         else
2043                                 rcu_assign_pointer(*ppsf,
2044                                                    mc_dereference(psf->sf_next, idev));
2045                         kfree_rcu(psf, rcu);
2046                 } else {
2047                         psf_prev = psf;
2048                 }
2049         }
2050 }
2051
2052 /* called with mc_lock */
2053 static void mld_send_cr(struct inet6_dev *idev)
2054 {
2055         struct ifmcaddr6 *pmc, *pmc_prev, *pmc_next;
2056         struct sk_buff *skb = NULL;
2057         int type, dtype;
2058
2059         /* deleted MCA's */
2060         pmc_prev = NULL;
2061         for (pmc = mc_dereference(idev->mc_tomb, idev);
2062              pmc;
2063              pmc = pmc_next) {
2064                 pmc_next = mc_dereference(pmc->next, idev);
2065                 if (pmc->mca_sfmode == MCAST_INCLUDE) {
2066                         type = MLD2_BLOCK_OLD_SOURCES;
2067                         dtype = MLD2_BLOCK_OLD_SOURCES;
2068                         skb = add_grec(skb, pmc, type, 1, 0, 0);
2069                         skb = add_grec(skb, pmc, dtype, 1, 1, 0);
2070                 }
2071                 if (pmc->mca_crcount) {
2072                         if (pmc->mca_sfmode == MCAST_EXCLUDE) {
2073                                 type = MLD2_CHANGE_TO_INCLUDE;
2074                                 skb = add_grec(skb, pmc, type, 1, 0, 0);
2075                         }
2076                         pmc->mca_crcount--;
2077                         if (pmc->mca_crcount == 0) {
2078                                 mld_clear_zeros(&pmc->mca_tomb, idev);
2079                                 mld_clear_zeros(&pmc->mca_sources, idev);
2080                         }
2081                 }
2082                 if (pmc->mca_crcount == 0 &&
2083                     !rcu_access_pointer(pmc->mca_tomb) &&
2084                     !rcu_access_pointer(pmc->mca_sources)) {
2085                         if (pmc_prev)
2086                                 rcu_assign_pointer(pmc_prev->next, pmc_next);
2087                         else
2088                                 rcu_assign_pointer(idev->mc_tomb, pmc_next);
2089                         in6_dev_put(pmc->idev);
2090                         kfree_rcu(pmc, rcu);
2091                 } else
2092                         pmc_prev = pmc;
2093         }
2094
2095         /* change recs */
2096         for_each_mc_mclock(idev, pmc) {
2097                 if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2098                         type = MLD2_BLOCK_OLD_SOURCES;
2099                         dtype = MLD2_ALLOW_NEW_SOURCES;
2100                 } else {
2101                         type = MLD2_ALLOW_NEW_SOURCES;
2102                         dtype = MLD2_BLOCK_OLD_SOURCES;
2103                 }
2104                 skb = add_grec(skb, pmc, type, 0, 0, 0);
2105                 skb = add_grec(skb, pmc, dtype, 0, 1, 0);       /* deleted sources */
2106
2107                 /* filter mode changes */
2108                 if (pmc->mca_crcount) {
2109                         if (pmc->mca_sfmode == MCAST_EXCLUDE)
2110                                 type = MLD2_CHANGE_TO_EXCLUDE;
2111                         else
2112                                 type = MLD2_CHANGE_TO_INCLUDE;
2113                         skb = add_grec(skb, pmc, type, 0, 0, 0);
2114                         pmc->mca_crcount--;
2115                 }
2116         }
2117         if (!skb)
2118                 return;
2119         (void) mld_sendpack(skb);
2120 }
2121
2122 static void igmp6_send(struct in6_addr *addr, struct net_device *dev, int type)
2123 {
2124         struct net *net = dev_net(dev);
2125         struct sock *sk = net->ipv6.igmp_sk;
2126         struct inet6_dev *idev;
2127         struct sk_buff *skb;
2128         struct mld_msg *hdr;
2129         const struct in6_addr *snd_addr, *saddr;
2130         struct in6_addr addr_buf;
2131         int hlen = LL_RESERVED_SPACE(dev);
2132         int tlen = dev->needed_tailroom;
2133         int err, len, payload_len, full_len;
2134         u8 ra[8] = { IPPROTO_ICMPV6, 0,
2135                      IPV6_TLV_ROUTERALERT, 2, 0, 0,
2136                      IPV6_TLV_PADN, 0 };
2137         struct flowi6 fl6;
2138         struct dst_entry *dst;
2139
2140         if (type == ICMPV6_MGM_REDUCTION)
2141                 snd_addr = &in6addr_linklocal_allrouters;
2142         else
2143                 snd_addr = addr;
2144
2145         len = sizeof(struct icmp6hdr) + sizeof(struct in6_addr);
2146         payload_len = len + sizeof(ra);
2147         full_len = sizeof(struct ipv6hdr) + payload_len;
2148
2149         rcu_read_lock();
2150         IP6_UPD_PO_STATS(net, __in6_dev_get(dev),
2151                       IPSTATS_MIB_OUT, full_len);
2152         rcu_read_unlock();
2153
2154         skb = sock_alloc_send_skb(sk, hlen + tlen + full_len, 1, &err);
2155
2156         if (!skb) {
2157                 rcu_read_lock();
2158                 IP6_INC_STATS(net, __in6_dev_get(dev),
2159                               IPSTATS_MIB_OUTDISCARDS);
2160                 rcu_read_unlock();
2161                 return;
2162         }
2163         skb->priority = TC_PRIO_CONTROL;
2164         skb_reserve(skb, hlen);
2165
2166         if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
2167                 /* <draft-ietf-magma-mld-source-05.txt>:
2168                  * use unspecified address as the source address
2169                  * when a valid link-local address is not available.
2170                  */
2171                 saddr = &in6addr_any;
2172         } else
2173                 saddr = &addr_buf;
2174
2175         ip6_mc_hdr(sk, skb, dev, saddr, snd_addr, NEXTHDR_HOP, payload_len);
2176
2177         skb_put_data(skb, ra, sizeof(ra));
2178
2179         hdr = skb_put_zero(skb, sizeof(struct mld_msg));
2180         hdr->mld_type = type;
2181         hdr->mld_mca = *addr;
2182
2183         hdr->mld_cksum = csum_ipv6_magic(saddr, snd_addr, len,
2184                                          IPPROTO_ICMPV6,
2185                                          csum_partial(hdr, len, 0));
2186
2187         rcu_read_lock();
2188         idev = __in6_dev_get(skb->dev);
2189
2190         icmpv6_flow_init(sk, &fl6, type,
2191                          &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
2192                          skb->dev->ifindex);
2193         dst = icmp6_dst_alloc(skb->dev, &fl6);
2194         if (IS_ERR(dst)) {
2195                 err = PTR_ERR(dst);
2196                 goto err_out;
2197         }
2198
2199         skb_dst_set(skb, dst);
2200         err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
2201                       net, sk, skb, NULL, skb->dev,
2202                       dst_output);
2203 out:
2204         if (!err) {
2205                 ICMP6MSGOUT_INC_STATS(net, idev, type);
2206                 ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
2207         } else
2208                 IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
2209
2210         rcu_read_unlock();
2211         return;
2212
2213 err_out:
2214         kfree_skb(skb);
2215         goto out;
2216 }
2217
2218 /* called with mc_lock */
2219 static void mld_send_initial_cr(struct inet6_dev *idev)
2220 {
2221         struct sk_buff *skb;
2222         struct ifmcaddr6 *pmc;
2223         int type;
2224
2225         if (mld_in_v1_mode(idev))
2226                 return;
2227
2228         skb = NULL;
2229         for_each_mc_mclock(idev, pmc) {
2230                 if (pmc->mca_sfcount[MCAST_EXCLUDE])
2231                         type = MLD2_CHANGE_TO_EXCLUDE;
2232                 else
2233                         type = MLD2_ALLOW_NEW_SOURCES;
2234                 skb = add_grec(skb, pmc, type, 0, 0, 1);
2235         }
2236         if (skb)
2237                 mld_sendpack(skb);
2238 }
2239
2240 void ipv6_mc_dad_complete(struct inet6_dev *idev)
2241 {
2242         mutex_lock(&idev->mc_lock);
2243         idev->mc_dad_count = idev->mc_qrv;
2244         if (idev->mc_dad_count) {
2245                 mld_send_initial_cr(idev);
2246                 idev->mc_dad_count--;
2247                 if (idev->mc_dad_count)
2248                         mld_dad_start_work(idev,
2249                                            unsolicited_report_interval(idev));
2250         }
2251         mutex_unlock(&idev->mc_lock);
2252 }
2253
2254 static void mld_dad_work(struct work_struct *work)
2255 {
2256         struct inet6_dev *idev = container_of(to_delayed_work(work),
2257                                               struct inet6_dev,
2258                                               mc_dad_work);
2259         mutex_lock(&idev->mc_lock);
2260         mld_send_initial_cr(idev);
2261         if (idev->mc_dad_count) {
2262                 idev->mc_dad_count--;
2263                 if (idev->mc_dad_count)
2264                         mld_dad_start_work(idev,
2265                                            unsolicited_report_interval(idev));
2266         }
2267         mutex_unlock(&idev->mc_lock);
2268         in6_dev_put(idev);
2269 }
2270
2271 /* called with mc_lock */
2272 static int ip6_mc_del1_src(struct ifmcaddr6 *pmc, int sfmode,
2273         const struct in6_addr *psfsrc)
2274 {
2275         struct ip6_sf_list *psf, *psf_prev;
2276         int rv = 0;
2277
2278         psf_prev = NULL;
2279         for_each_psf_mclock(pmc, psf) {
2280                 if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2281                         break;
2282                 psf_prev = psf;
2283         }
2284         if (!psf || psf->sf_count[sfmode] == 0) {
2285                 /* source filter not found, or count wrong =>  bug */
2286                 return -ESRCH;
2287         }
2288         psf->sf_count[sfmode]--;
2289         if (!psf->sf_count[MCAST_INCLUDE] && !psf->sf_count[MCAST_EXCLUDE]) {
2290                 struct inet6_dev *idev = pmc->idev;
2291
2292                 /* no more filters for this source */
2293                 if (psf_prev)
2294                         rcu_assign_pointer(psf_prev->sf_next,
2295                                            mc_dereference(psf->sf_next, idev));
2296                 else
2297                         rcu_assign_pointer(pmc->mca_sources,
2298                                            mc_dereference(psf->sf_next, idev));
2299
2300                 if (psf->sf_oldin && !(pmc->mca_flags & MAF_NOREPORT) &&
2301                     !mld_in_v1_mode(idev)) {
2302                         psf->sf_crcount = idev->mc_qrv;
2303                         rcu_assign_pointer(psf->sf_next,
2304                                            mc_dereference(pmc->mca_tomb, idev));
2305                         rcu_assign_pointer(pmc->mca_tomb, psf);
2306                         rv = 1;
2307                 } else {
2308                         kfree_rcu(psf, rcu);
2309                 }
2310         }
2311         return rv;
2312 }
2313
2314 /* called with mc_lock */
2315 static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2316                           int sfmode, int sfcount, const struct in6_addr *psfsrc,
2317                           int delta)
2318 {
2319         struct ifmcaddr6 *pmc;
2320         int     changerec = 0;
2321         int     i, err;
2322
2323         if (!idev)
2324                 return -ENODEV;
2325
2326         for_each_mc_mclock(idev, pmc) {
2327                 if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2328                         break;
2329         }
2330         if (!pmc)
2331                 return -ESRCH;
2332
2333         sf_markstate(pmc);
2334         if (!delta) {
2335                 if (!pmc->mca_sfcount[sfmode])
2336                         return -EINVAL;
2337
2338                 pmc->mca_sfcount[sfmode]--;
2339         }
2340         err = 0;
2341         for (i = 0; i < sfcount; i++) {
2342                 int rv = ip6_mc_del1_src(pmc, sfmode, &psfsrc[i]);
2343
2344                 changerec |= rv > 0;
2345                 if (!err && rv < 0)
2346                         err = rv;
2347         }
2348         if (pmc->mca_sfmode == MCAST_EXCLUDE &&
2349             pmc->mca_sfcount[MCAST_EXCLUDE] == 0 &&
2350             pmc->mca_sfcount[MCAST_INCLUDE]) {
2351                 struct ip6_sf_list *psf;
2352
2353                 /* filter mode change */
2354                 pmc->mca_sfmode = MCAST_INCLUDE;
2355                 pmc->mca_crcount = idev->mc_qrv;
2356                 idev->mc_ifc_count = pmc->mca_crcount;
2357                 for_each_psf_mclock(pmc, psf)
2358                         psf->sf_crcount = 0;
2359                 mld_ifc_event(pmc->idev);
2360         } else if (sf_setstate(pmc) || changerec) {
2361                 mld_ifc_event(pmc->idev);
2362         }
2363
2364         return err;
2365 }
2366
2367 /*
2368  * Add multicast single-source filter to the interface list
2369  * called with mc_lock
2370  */
2371 static int ip6_mc_add1_src(struct ifmcaddr6 *pmc, int sfmode,
2372         const struct in6_addr *psfsrc)
2373 {
2374         struct ip6_sf_list *psf, *psf_prev;
2375
2376         psf_prev = NULL;
2377         for_each_psf_mclock(pmc, psf) {
2378                 if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2379                         break;
2380                 psf_prev = psf;
2381         }
2382         if (!psf) {
2383                 psf = kzalloc(sizeof(*psf), GFP_KERNEL);
2384                 if (!psf)
2385                         return -ENOBUFS;
2386
2387                 psf->sf_addr = *psfsrc;
2388                 if (psf_prev) {
2389                         rcu_assign_pointer(psf_prev->sf_next, psf);
2390                 } else {
2391                         rcu_assign_pointer(pmc->mca_sources, psf);
2392                 }
2393         }
2394         psf->sf_count[sfmode]++;
2395         return 0;
2396 }
2397
2398 /* called with mc_lock */
2399 static void sf_markstate(struct ifmcaddr6 *pmc)
2400 {
2401         struct ip6_sf_list *psf;
2402         int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2403
2404         for_each_psf_mclock(pmc, psf) {
2405                 if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2406                         psf->sf_oldin = mca_xcount ==
2407                                 psf->sf_count[MCAST_EXCLUDE] &&
2408                                 !psf->sf_count[MCAST_INCLUDE];
2409                 } else {
2410                         psf->sf_oldin = psf->sf_count[MCAST_INCLUDE] != 0;
2411                 }
2412         }
2413 }
2414
2415 /* called with mc_lock */
2416 static int sf_setstate(struct ifmcaddr6 *pmc)
2417 {
2418         struct ip6_sf_list *psf, *dpsf;
2419         int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2420         int qrv = pmc->idev->mc_qrv;
2421         int new_in, rv;
2422
2423         rv = 0;
2424         for_each_psf_mclock(pmc, psf) {
2425                 if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2426                         new_in = mca_xcount == psf->sf_count[MCAST_EXCLUDE] &&
2427                                 !psf->sf_count[MCAST_INCLUDE];
2428                 } else
2429                         new_in = psf->sf_count[MCAST_INCLUDE] != 0;
2430                 if (new_in) {
2431                         if (!psf->sf_oldin) {
2432                                 struct ip6_sf_list *prev = NULL;
2433
2434                                 for_each_psf_tomb(pmc, dpsf) {
2435                                         if (ipv6_addr_equal(&dpsf->sf_addr,
2436                                             &psf->sf_addr))
2437                                                 break;
2438                                         prev = dpsf;
2439                                 }
2440                                 if (dpsf) {
2441                                         if (prev)
2442                                                 rcu_assign_pointer(prev->sf_next,
2443                                                                    mc_dereference(dpsf->sf_next,
2444                                                                                   pmc->idev));
2445                                         else
2446                                                 rcu_assign_pointer(pmc->mca_tomb,
2447                                                                    mc_dereference(dpsf->sf_next,
2448                                                                                   pmc->idev));
2449                                         kfree_rcu(dpsf, rcu);
2450                                 }
2451                                 psf->sf_crcount = qrv;
2452                                 rv++;
2453                         }
2454                 } else if (psf->sf_oldin) {
2455                         psf->sf_crcount = 0;
2456                         /*
2457                          * add or update "delete" records if an active filter
2458                          * is now inactive
2459                          */
2460
2461                         for_each_psf_tomb(pmc, dpsf)
2462                                 if (ipv6_addr_equal(&dpsf->sf_addr,
2463                                     &psf->sf_addr))
2464                                         break;
2465                         if (!dpsf) {
2466                                 dpsf = kmalloc(sizeof(*dpsf), GFP_KERNEL);
2467                                 if (!dpsf)
2468                                         continue;
2469                                 *dpsf = *psf;
2470                                 rcu_assign_pointer(dpsf->sf_next,
2471                                                    mc_dereference(pmc->mca_tomb, pmc->idev));
2472                                 rcu_assign_pointer(pmc->mca_tomb, dpsf);
2473                         }
2474                         dpsf->sf_crcount = qrv;
2475                         rv++;
2476                 }
2477         }
2478         return rv;
2479 }
2480
2481 /*
2482  * Add multicast source filter list to the interface list
2483  * called with mc_lock
2484  */
2485 static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2486                           int sfmode, int sfcount, const struct in6_addr *psfsrc,
2487                           int delta)
2488 {
2489         struct ifmcaddr6 *pmc;
2490         int     isexclude;
2491         int     i, err;
2492
2493         if (!idev)
2494                 return -ENODEV;
2495
2496         for_each_mc_mclock(idev, pmc) {
2497                 if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2498                         break;
2499         }
2500         if (!pmc)
2501                 return -ESRCH;
2502
2503         sf_markstate(pmc);
2504         isexclude = pmc->mca_sfmode == MCAST_EXCLUDE;
2505         if (!delta)
2506                 pmc->mca_sfcount[sfmode]++;
2507         err = 0;
2508         for (i = 0; i < sfcount; i++) {
2509                 err = ip6_mc_add1_src(pmc, sfmode, &psfsrc[i]);
2510                 if (err)
2511                         break;
2512         }
2513         if (err) {
2514                 int j;
2515
2516                 if (!delta)
2517                         pmc->mca_sfcount[sfmode]--;
2518                 for (j = 0; j < i; j++)
2519                         ip6_mc_del1_src(pmc, sfmode, &psfsrc[j]);
2520         } else if (isexclude != (pmc->mca_sfcount[MCAST_EXCLUDE] != 0)) {
2521                 struct ip6_sf_list *psf;
2522
2523                 /* filter mode change */
2524                 if (pmc->mca_sfcount[MCAST_EXCLUDE])
2525                         pmc->mca_sfmode = MCAST_EXCLUDE;
2526                 else if (pmc->mca_sfcount[MCAST_INCLUDE])
2527                         pmc->mca_sfmode = MCAST_INCLUDE;
2528                 /* else no filters; keep old mode for reports */
2529
2530                 pmc->mca_crcount = idev->mc_qrv;
2531                 idev->mc_ifc_count = pmc->mca_crcount;
2532                 for_each_psf_mclock(pmc, psf)
2533                         psf->sf_crcount = 0;
2534                 mld_ifc_event(idev);
2535         } else if (sf_setstate(pmc)) {
2536                 mld_ifc_event(idev);
2537         }
2538         return err;
2539 }
2540
2541 /* called with mc_lock */
2542 static void ip6_mc_clear_src(struct ifmcaddr6 *pmc)
2543 {
2544         struct ip6_sf_list *psf, *nextpsf;
2545
2546         for (psf = mc_dereference(pmc->mca_tomb, pmc->idev);
2547              psf;
2548              psf = nextpsf) {
2549                 nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2550                 kfree_rcu(psf, rcu);
2551         }
2552         RCU_INIT_POINTER(pmc->mca_tomb, NULL);
2553         for (psf = mc_dereference(pmc->mca_sources, pmc->idev);
2554              psf;
2555              psf = nextpsf) {
2556                 nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2557                 kfree_rcu(psf, rcu);
2558         }
2559         RCU_INIT_POINTER(pmc->mca_sources, NULL);
2560         pmc->mca_sfmode = MCAST_EXCLUDE;
2561         pmc->mca_sfcount[MCAST_INCLUDE] = 0;
2562         pmc->mca_sfcount[MCAST_EXCLUDE] = 1;
2563 }
2564
2565 /* called with mc_lock */
2566 static void igmp6_join_group(struct ifmcaddr6 *ma)
2567 {
2568         unsigned long delay;
2569
2570         if (ma->mca_flags & MAF_NOREPORT)
2571                 return;
2572
2573         igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2574
2575         delay = get_random_u32_below(unsolicited_report_interval(ma->idev));
2576
2577         if (cancel_delayed_work(&ma->mca_work)) {
2578                 refcount_dec(&ma->mca_refcnt);
2579                 delay = ma->mca_work.timer.expires - jiffies;
2580         }
2581
2582         if (!mod_delayed_work(mld_wq, &ma->mca_work, delay))
2583                 refcount_inc(&ma->mca_refcnt);
2584         ma->mca_flags |= MAF_TIMER_RUNNING | MAF_LAST_REPORTER;
2585 }
2586
2587 static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
2588                             struct inet6_dev *idev)
2589 {
2590         struct ip6_sf_socklist *psl;
2591         int err;
2592
2593         psl = sock_dereference(iml->sflist, sk);
2594
2595         if (idev)
2596                 mutex_lock(&idev->mc_lock);
2597
2598         if (!psl) {
2599                 /* any-source empty exclude case */
2600                 err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode, 0, NULL, 0);
2601         } else {
2602                 err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode,
2603                                      psl->sl_count, psl->sl_addr, 0);
2604                 RCU_INIT_POINTER(iml->sflist, NULL);
2605                 atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
2606                            &sk->sk_omem_alloc);
2607                 kfree_rcu(psl, rcu);
2608         }
2609
2610         if (idev)
2611                 mutex_unlock(&idev->mc_lock);
2612
2613         return err;
2614 }
2615
2616 /* called with mc_lock */
2617 static void igmp6_leave_group(struct ifmcaddr6 *ma)
2618 {
2619         if (mld_in_v1_mode(ma->idev)) {
2620                 if (ma->mca_flags & MAF_LAST_REPORTER) {
2621                         igmp6_send(&ma->mca_addr, ma->idev->dev,
2622                                 ICMPV6_MGM_REDUCTION);
2623                 }
2624         } else {
2625                 mld_add_delrec(ma->idev, ma);
2626                 mld_ifc_event(ma->idev);
2627         }
2628 }
2629
2630 static void mld_gq_work(struct work_struct *work)
2631 {
2632         struct inet6_dev *idev = container_of(to_delayed_work(work),
2633                                               struct inet6_dev,
2634                                               mc_gq_work);
2635
2636         mutex_lock(&idev->mc_lock);
2637         mld_send_report(idev, NULL);
2638         idev->mc_gq_running = 0;
2639         mutex_unlock(&idev->mc_lock);
2640
2641         in6_dev_put(idev);
2642 }
2643
2644 static void mld_ifc_work(struct work_struct *work)
2645 {
2646         struct inet6_dev *idev = container_of(to_delayed_work(work),
2647                                               struct inet6_dev,
2648                                               mc_ifc_work);
2649
2650         mutex_lock(&idev->mc_lock);
2651         mld_send_cr(idev);
2652
2653         if (idev->mc_ifc_count) {
2654                 idev->mc_ifc_count--;
2655                 if (idev->mc_ifc_count)
2656                         mld_ifc_start_work(idev,
2657                                            unsolicited_report_interval(idev));
2658         }
2659         mutex_unlock(&idev->mc_lock);
2660         in6_dev_put(idev);
2661 }
2662
2663 /* called with mc_lock */
2664 static void mld_ifc_event(struct inet6_dev *idev)
2665 {
2666         if (mld_in_v1_mode(idev))
2667                 return;
2668
2669         idev->mc_ifc_count = idev->mc_qrv;
2670         mld_ifc_start_work(idev, 1);
2671 }
2672
2673 static void mld_mca_work(struct work_struct *work)
2674 {
2675         struct ifmcaddr6 *ma = container_of(to_delayed_work(work),
2676                                             struct ifmcaddr6, mca_work);
2677
2678         mutex_lock(&ma->idev->mc_lock);
2679         if (mld_in_v1_mode(ma->idev))
2680                 igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2681         else
2682                 mld_send_report(ma->idev, ma);
2683         ma->mca_flags |=  MAF_LAST_REPORTER;
2684         ma->mca_flags &= ~MAF_TIMER_RUNNING;
2685         mutex_unlock(&ma->idev->mc_lock);
2686
2687         ma_put(ma);
2688 }
2689
2690 /* Device changing type */
2691
2692 void ipv6_mc_unmap(struct inet6_dev *idev)
2693 {
2694         struct ifmcaddr6 *i;
2695
2696         /* Install multicast list, except for all-nodes (already installed) */
2697
2698         mutex_lock(&idev->mc_lock);
2699         for_each_mc_mclock(idev, i)
2700                 igmp6_group_dropped(i);
2701         mutex_unlock(&idev->mc_lock);
2702 }
2703
2704 void ipv6_mc_remap(struct inet6_dev *idev)
2705 {
2706         ipv6_mc_up(idev);
2707 }
2708
2709 /* Device going down */
2710 void ipv6_mc_down(struct inet6_dev *idev)
2711 {
2712         struct ifmcaddr6 *i;
2713
2714         mutex_lock(&idev->mc_lock);
2715         /* Withdraw multicast list */
2716         for_each_mc_mclock(idev, i)
2717                 igmp6_group_dropped(i);
2718         mutex_unlock(&idev->mc_lock);
2719
2720         /* Should stop work after group drop. or we will
2721          * start work again in mld_ifc_event()
2722          */
2723         synchronize_net();
2724         mld_query_stop_work(idev);
2725         mld_report_stop_work(idev);
2726         mld_ifc_stop_work(idev);
2727         mld_gq_stop_work(idev);
2728         mld_dad_stop_work(idev);
2729 }
2730
2731 static void ipv6_mc_reset(struct inet6_dev *idev)
2732 {
2733         idev->mc_qrv = sysctl_mld_qrv;
2734         idev->mc_qi = MLD_QI_DEFAULT;
2735         idev->mc_qri = MLD_QRI_DEFAULT;
2736         idev->mc_v1_seen = 0;
2737         idev->mc_maxdelay = unsolicited_report_interval(idev);
2738 }
2739
2740 /* Device going up */
2741
2742 void ipv6_mc_up(struct inet6_dev *idev)
2743 {
2744         struct ifmcaddr6 *i;
2745
2746         /* Install multicast list, except for all-nodes (already installed) */
2747
2748         ipv6_mc_reset(idev);
2749         mutex_lock(&idev->mc_lock);
2750         for_each_mc_mclock(idev, i) {
2751                 mld_del_delrec(idev, i);
2752                 igmp6_group_added(i);
2753         }
2754         mutex_unlock(&idev->mc_lock);
2755 }
2756
2757 /* IPv6 device initialization. */
2758
2759 void ipv6_mc_init_dev(struct inet6_dev *idev)
2760 {
2761         idev->mc_gq_running = 0;
2762         INIT_DELAYED_WORK(&idev->mc_gq_work, mld_gq_work);
2763         RCU_INIT_POINTER(idev->mc_tomb, NULL);
2764         idev->mc_ifc_count = 0;
2765         INIT_DELAYED_WORK(&idev->mc_ifc_work, mld_ifc_work);
2766         INIT_DELAYED_WORK(&idev->mc_dad_work, mld_dad_work);
2767         INIT_DELAYED_WORK(&idev->mc_query_work, mld_query_work);
2768         INIT_DELAYED_WORK(&idev->mc_report_work, mld_report_work);
2769         skb_queue_head_init(&idev->mc_query_queue);
2770         skb_queue_head_init(&idev->mc_report_queue);
2771         spin_lock_init(&idev->mc_query_lock);
2772         spin_lock_init(&idev->mc_report_lock);
2773         mutex_init(&idev->mc_lock);
2774         ipv6_mc_reset(idev);
2775 }
2776
2777 /*
2778  *      Device is about to be destroyed: clean up.
2779  */
2780
2781 void ipv6_mc_destroy_dev(struct inet6_dev *idev)
2782 {
2783         struct ifmcaddr6 *i;
2784
2785         /* Deactivate works */
2786         ipv6_mc_down(idev);
2787         mutex_lock(&idev->mc_lock);
2788         mld_clear_delrec(idev);
2789         mutex_unlock(&idev->mc_lock);
2790         mld_clear_query(idev);
2791         mld_clear_report(idev);
2792
2793         /* Delete all-nodes address. */
2794         /* We cannot call ipv6_dev_mc_dec() directly, our caller in
2795          * addrconf.c has NULL'd out dev->ip6_ptr so in6_dev_get() will
2796          * fail.
2797          */
2798         __ipv6_dev_mc_dec(idev, &in6addr_linklocal_allnodes);
2799
2800         if (idev->cnf.forwarding)
2801                 __ipv6_dev_mc_dec(idev, &in6addr_linklocal_allrouters);
2802
2803         mutex_lock(&idev->mc_lock);
2804         while ((i = mc_dereference(idev->mc_list, idev))) {
2805                 rcu_assign_pointer(idev->mc_list, mc_dereference(i->next, idev));
2806
2807                 ip6_mc_clear_src(i);
2808                 ma_put(i);
2809         }
2810         mutex_unlock(&idev->mc_lock);
2811 }
2812
2813 static void ipv6_mc_rejoin_groups(struct inet6_dev *idev)
2814 {
2815         struct ifmcaddr6 *pmc;
2816
2817         ASSERT_RTNL();
2818
2819         mutex_lock(&idev->mc_lock);
2820         if (mld_in_v1_mode(idev)) {
2821                 for_each_mc_mclock(idev, pmc)
2822                         igmp6_join_group(pmc);
2823         } else {
2824                 mld_send_report(idev, NULL);
2825         }
2826         mutex_unlock(&idev->mc_lock);
2827 }
2828
2829 static int ipv6_mc_netdev_event(struct notifier_block *this,
2830                                 unsigned long event,
2831                                 void *ptr)
2832 {
2833         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
2834         struct inet6_dev *idev = __in6_dev_get(dev);
2835
2836         switch (event) {
2837         case NETDEV_RESEND_IGMP:
2838                 if (idev)
2839                         ipv6_mc_rejoin_groups(idev);
2840                 break;
2841         default:
2842                 break;
2843         }
2844
2845         return NOTIFY_DONE;
2846 }
2847
2848 static struct notifier_block igmp6_netdev_notifier = {
2849         .notifier_call = ipv6_mc_netdev_event,
2850 };
2851
2852 #ifdef CONFIG_PROC_FS
2853 struct igmp6_mc_iter_state {
2854         struct seq_net_private p;
2855         struct net_device *dev;
2856         struct inet6_dev *idev;
2857 };
2858
2859 #define igmp6_mc_seq_private(seq)       ((struct igmp6_mc_iter_state *)(seq)->private)
2860
2861 static inline struct ifmcaddr6 *igmp6_mc_get_first(struct seq_file *seq)
2862 {
2863         struct ifmcaddr6 *im = NULL;
2864         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2865         struct net *net = seq_file_net(seq);
2866
2867         state->idev = NULL;
2868         for_each_netdev_rcu(net, state->dev) {
2869                 struct inet6_dev *idev;
2870                 idev = __in6_dev_get(state->dev);
2871                 if (!idev)
2872                         continue;
2873
2874                 im = rcu_dereference(idev->mc_list);
2875                 if (im) {
2876                         state->idev = idev;
2877                         break;
2878                 }
2879         }
2880         return im;
2881 }
2882
2883 static struct ifmcaddr6 *igmp6_mc_get_next(struct seq_file *seq, struct ifmcaddr6 *im)
2884 {
2885         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2886
2887         im = rcu_dereference(im->next);
2888         while (!im) {
2889                 state->dev = next_net_device_rcu(state->dev);
2890                 if (!state->dev) {
2891                         state->idev = NULL;
2892                         break;
2893                 }
2894                 state->idev = __in6_dev_get(state->dev);
2895                 if (!state->idev)
2896                         continue;
2897                 im = rcu_dereference(state->idev->mc_list);
2898         }
2899         return im;
2900 }
2901
2902 static struct ifmcaddr6 *igmp6_mc_get_idx(struct seq_file *seq, loff_t pos)
2903 {
2904         struct ifmcaddr6 *im = igmp6_mc_get_first(seq);
2905         if (im)
2906                 while (pos && (im = igmp6_mc_get_next(seq, im)) != NULL)
2907                         --pos;
2908         return pos ? NULL : im;
2909 }
2910
2911 static void *igmp6_mc_seq_start(struct seq_file *seq, loff_t *pos)
2912         __acquires(RCU)
2913 {
2914         rcu_read_lock();
2915         return igmp6_mc_get_idx(seq, *pos);
2916 }
2917
2918 static void *igmp6_mc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2919 {
2920         struct ifmcaddr6 *im = igmp6_mc_get_next(seq, v);
2921
2922         ++*pos;
2923         return im;
2924 }
2925
2926 static void igmp6_mc_seq_stop(struct seq_file *seq, void *v)
2927         __releases(RCU)
2928 {
2929         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2930
2931         if (likely(state->idev))
2932                 state->idev = NULL;
2933         state->dev = NULL;
2934         rcu_read_unlock();
2935 }
2936
2937 static int igmp6_mc_seq_show(struct seq_file *seq, void *v)
2938 {
2939         struct ifmcaddr6 *im = (struct ifmcaddr6 *)v;
2940         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2941
2942         seq_printf(seq,
2943                    "%-4d %-15s %pi6 %5d %08X %ld\n",
2944                    state->dev->ifindex, state->dev->name,
2945                    &im->mca_addr,
2946                    im->mca_users, im->mca_flags,
2947                    (im->mca_flags & MAF_TIMER_RUNNING) ?
2948                    jiffies_to_clock_t(im->mca_work.timer.expires - jiffies) : 0);
2949         return 0;
2950 }
2951
2952 static const struct seq_operations igmp6_mc_seq_ops = {
2953         .start  =       igmp6_mc_seq_start,
2954         .next   =       igmp6_mc_seq_next,
2955         .stop   =       igmp6_mc_seq_stop,
2956         .show   =       igmp6_mc_seq_show,
2957 };
2958
2959 struct igmp6_mcf_iter_state {
2960         struct seq_net_private p;
2961         struct net_device *dev;
2962         struct inet6_dev *idev;
2963         struct ifmcaddr6 *im;
2964 };
2965
2966 #define igmp6_mcf_seq_private(seq)      ((struct igmp6_mcf_iter_state *)(seq)->private)
2967
2968 static inline struct ip6_sf_list *igmp6_mcf_get_first(struct seq_file *seq)
2969 {
2970         struct ip6_sf_list *psf = NULL;
2971         struct ifmcaddr6 *im = NULL;
2972         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
2973         struct net *net = seq_file_net(seq);
2974
2975         state->idev = NULL;
2976         state->im = NULL;
2977         for_each_netdev_rcu(net, state->dev) {
2978                 struct inet6_dev *idev;
2979                 idev = __in6_dev_get(state->dev);
2980                 if (unlikely(idev == NULL))
2981                         continue;
2982
2983                 im = rcu_dereference(idev->mc_list);
2984                 if (likely(im)) {
2985                         psf = rcu_dereference(im->mca_sources);
2986                         if (likely(psf)) {
2987                                 state->im = im;
2988                                 state->idev = idev;
2989                                 break;
2990                         }
2991                 }
2992         }
2993         return psf;
2994 }
2995
2996 static struct ip6_sf_list *igmp6_mcf_get_next(struct seq_file *seq, struct ip6_sf_list *psf)
2997 {
2998         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
2999
3000         psf = rcu_dereference(psf->sf_next);
3001         while (!psf) {
3002                 state->im = rcu_dereference(state->im->next);
3003                 while (!state->im) {
3004                         state->dev = next_net_device_rcu(state->dev);
3005                         if (!state->dev) {
3006                                 state->idev = NULL;
3007                                 goto out;
3008                         }
3009                         state->idev = __in6_dev_get(state->dev);
3010                         if (!state->idev)
3011                                 continue;
3012                         state->im = rcu_dereference(state->idev->mc_list);
3013                 }
3014                 if (!state->im)
3015                         break;
3016                 psf = rcu_dereference(state->im->mca_sources);
3017         }
3018 out:
3019         return psf;
3020 }
3021
3022 static struct ip6_sf_list *igmp6_mcf_get_idx(struct seq_file *seq, loff_t pos)
3023 {
3024         struct ip6_sf_list *psf = igmp6_mcf_get_first(seq);
3025         if (psf)
3026                 while (pos && (psf = igmp6_mcf_get_next(seq, psf)) != NULL)
3027                         --pos;
3028         return pos ? NULL : psf;
3029 }
3030
3031 static void *igmp6_mcf_seq_start(struct seq_file *seq, loff_t *pos)
3032         __acquires(RCU)
3033 {
3034         rcu_read_lock();
3035         return *pos ? igmp6_mcf_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
3036 }
3037
3038 static void *igmp6_mcf_seq_next(struct seq_file *seq, void *v, loff_t *pos)
3039 {
3040         struct ip6_sf_list *psf;
3041         if (v == SEQ_START_TOKEN)
3042                 psf = igmp6_mcf_get_first(seq);
3043         else
3044                 psf = igmp6_mcf_get_next(seq, v);
3045         ++*pos;
3046         return psf;
3047 }
3048
3049 static void igmp6_mcf_seq_stop(struct seq_file *seq, void *v)
3050         __releases(RCU)
3051 {
3052         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3053
3054         if (likely(state->im))
3055                 state->im = NULL;
3056         if (likely(state->idev))
3057                 state->idev = NULL;
3058
3059         state->dev = NULL;
3060         rcu_read_unlock();
3061 }
3062
3063 static int igmp6_mcf_seq_show(struct seq_file *seq, void *v)
3064 {
3065         struct ip6_sf_list *psf = (struct ip6_sf_list *)v;
3066         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3067
3068         if (v == SEQ_START_TOKEN) {
3069                 seq_puts(seq, "Idx Device                Multicast Address                   Source Address    INC    EXC\n");
3070         } else {
3071                 seq_printf(seq,
3072                            "%3d %6.6s %pi6 %pi6 %6lu %6lu\n",
3073                            state->dev->ifindex, state->dev->name,
3074                            &state->im->mca_addr,
3075                            &psf->sf_addr,
3076                            psf->sf_count[MCAST_INCLUDE],
3077                            psf->sf_count[MCAST_EXCLUDE]);
3078         }
3079         return 0;
3080 }
3081
3082 static const struct seq_operations igmp6_mcf_seq_ops = {
3083         .start  =       igmp6_mcf_seq_start,
3084         .next   =       igmp6_mcf_seq_next,
3085         .stop   =       igmp6_mcf_seq_stop,
3086         .show   =       igmp6_mcf_seq_show,
3087 };
3088
3089 static int __net_init igmp6_proc_init(struct net *net)
3090 {
3091         int err;
3092
3093         err = -ENOMEM;
3094         if (!proc_create_net("igmp6", 0444, net->proc_net, &igmp6_mc_seq_ops,
3095                         sizeof(struct igmp6_mc_iter_state)))
3096                 goto out;
3097         if (!proc_create_net("mcfilter6", 0444, net->proc_net,
3098                         &igmp6_mcf_seq_ops,
3099                         sizeof(struct igmp6_mcf_iter_state)))
3100                 goto out_proc_net_igmp6;
3101
3102         err = 0;
3103 out:
3104         return err;
3105
3106 out_proc_net_igmp6:
3107         remove_proc_entry("igmp6", net->proc_net);
3108         goto out;
3109 }
3110
3111 static void __net_exit igmp6_proc_exit(struct net *net)
3112 {
3113         remove_proc_entry("mcfilter6", net->proc_net);
3114         remove_proc_entry("igmp6", net->proc_net);
3115 }
3116 #else
3117 static inline int igmp6_proc_init(struct net *net)
3118 {
3119         return 0;
3120 }
3121 static inline void igmp6_proc_exit(struct net *net)
3122 {
3123 }
3124 #endif
3125
3126 static int __net_init igmp6_net_init(struct net *net)
3127 {
3128         int err;
3129
3130         err = inet_ctl_sock_create(&net->ipv6.igmp_sk, PF_INET6,
3131                                    SOCK_RAW, IPPROTO_ICMPV6, net);
3132         if (err < 0) {
3133                 pr_err("Failed to initialize the IGMP6 control socket (err %d)\n",
3134                        err);
3135                 goto out;
3136         }
3137
3138         inet6_sk(net->ipv6.igmp_sk)->hop_limit = 1;
3139         net->ipv6.igmp_sk->sk_allocation = GFP_KERNEL;
3140
3141         err = inet_ctl_sock_create(&net->ipv6.mc_autojoin_sk, PF_INET6,
3142                                    SOCK_RAW, IPPROTO_ICMPV6, net);
3143         if (err < 0) {
3144                 pr_err("Failed to initialize the IGMP6 autojoin socket (err %d)\n",
3145                        err);
3146                 goto out_sock_create;
3147         }
3148
3149         err = igmp6_proc_init(net);
3150         if (err)
3151                 goto out_sock_create_autojoin;
3152
3153         return 0;
3154
3155 out_sock_create_autojoin:
3156         inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3157 out_sock_create:
3158         inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3159 out:
3160         return err;
3161 }
3162
3163 static void __net_exit igmp6_net_exit(struct net *net)
3164 {
3165         inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3166         inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3167         igmp6_proc_exit(net);
3168 }
3169
3170 static struct pernet_operations igmp6_net_ops = {
3171         .init = igmp6_net_init,
3172         .exit = igmp6_net_exit,
3173 };
3174
3175 int __init igmp6_init(void)
3176 {
3177         int err;
3178
3179         err = register_pernet_subsys(&igmp6_net_ops);
3180         if (err)
3181                 return err;
3182
3183         mld_wq = create_workqueue("mld");
3184         if (!mld_wq) {
3185                 unregister_pernet_subsys(&igmp6_net_ops);
3186                 return -ENOMEM;
3187         }
3188
3189         return err;
3190 }
3191
3192 int __init igmp6_late_init(void)
3193 {
3194         return register_netdevice_notifier(&igmp6_netdev_notifier);
3195 }
3196
3197 void igmp6_cleanup(void)
3198 {
3199         unregister_pernet_subsys(&igmp6_net_ops);
3200         destroy_workqueue(mld_wq);
3201 }
3202
3203 void igmp6_late_cleanup(void)
3204 {
3205         unregister_netdevice_notifier(&igmp6_netdev_notifier);
3206 }