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                         kfree_rcu(psl, rcu);
 464                 }
 465                 psl = newpsl;
 466                 rcu_assign_pointer(pmc->sflist, psl);
 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                 kfree_rcu(psl, rcu);
 569         } else {
 570                 ip6_mc_del_src(idev, group, pmc->sfmode, 0, NULL, 0);
 571         }
 572         mutex_unlock(&idev->mc_lock);
 573         rcu_assign_pointer(pmc->sflist, newpsl);
 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                   struct sockaddr_storage __user *p)
 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
 616         for (i = 0; i < copycount; i++, p++) {
 617                 struct sockaddr_in6 *psin6;
 618                 struct sockaddr_storage ss;
 619
 620                 psin6 = (struct sockaddr_in6 *)&ss;
 621                 memset(&ss, 0, sizeof(ss));
 622                 psin6->sin6_family = AF_INET6;
 623                 psin6->sin6_addr = psl->sl_addr[i];
 624                 if (copy_to_user(p, &ss, sizeof(ss)))
 625                         return -EFAULT;
 626         }
 627         return 0;
 628 }
 629
 630 bool inet6_mc_check(struct sock *sk, const struct in6_addr *mc_addr,
 631                     const struct in6_addr *src_addr)
 632 {
 633         struct ipv6_pinfo *np = inet6_sk(sk);
 634         struct ipv6_mc_socklist *mc;
 635         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 = prandom_u32() % 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 = prandom_u32() % 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 = prandom_u32() % 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 = prandom_u32() % 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 int 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 0;
1371 }
1372
1373 /* called with rcu_read_lock() */
1374 int igmp6_event_query(struct sk_buff *skb)
1375 {
1376         struct inet6_dev *idev = __in6_dev_get(skb->dev);
1377
1378         if (!idev)
1379                 return -EINVAL;
1380
1381         if (idev->dead) {
1382                 kfree_skb(skb);
1383                 return -ENODEV;
1384         }
1385
1386         spin_lock_bh(&idev->mc_query_lock);
1387         if (skb_queue_len(&idev->mc_query_queue) < MLD_MAX_SKBS) {
1388                 __skb_queue_tail(&idev->mc_query_queue, skb);
1389                 if (!mod_delayed_work(mld_wq, &idev->mc_query_work, 0))
1390                         in6_dev_hold(idev);
1391         }
1392         spin_unlock_bh(&idev->mc_query_lock);
1393
1394         return 0;
1395 }
1396
1397 static void __mld_query_work(struct sk_buff *skb)
1398 {
1399         struct mld2_query *mlh2 = NULL;
1400         const struct in6_addr *group;
1401         unsigned long max_delay;
1402         struct inet6_dev *idev;
1403         struct ifmcaddr6 *ma;
1404         struct mld_msg *mld;
1405         int group_type;
1406         int mark = 0;
1407         int len, err;
1408
1409         if (!pskb_may_pull(skb, sizeof(struct in6_addr)))
1410                 goto kfree_skb;
1411
1412         /* compute payload length excluding extension headers */
1413         len = ntohs(ipv6_hdr(skb)->payload_len) + sizeof(struct ipv6hdr);
1414         len -= skb_network_header_len(skb);
1415
1416         /* RFC3810 6.2
1417          * Upon reception of an MLD message that contains a Query, the node
1418          * checks if the source address of the message is a valid link-local
1419          * address, if the Hop Limit is set to 1, and if the Router Alert
1420          * option is present in the Hop-By-Hop Options header of the IPv6
1421          * packet.  If any of these checks fails, the packet is dropped.
1422          */
1423         if (!(ipv6_addr_type(&ipv6_hdr(skb)->saddr) & IPV6_ADDR_LINKLOCAL) ||
1424             ipv6_hdr(skb)->hop_limit != 1 ||
1425             !(IP6CB(skb)->flags & IP6SKB_ROUTERALERT) ||
1426             IP6CB(skb)->ra != htons(IPV6_OPT_ROUTERALERT_MLD))
1427                 goto kfree_skb;
1428
1429         idev = in6_dev_get(skb->dev);
1430         if (!idev)
1431                 goto kfree_skb;
1432
1433         mld = (struct mld_msg *)icmp6_hdr(skb);
1434         group = &mld->mld_mca;
1435         group_type = ipv6_addr_type(group);
1436
1437         if (group_type != IPV6_ADDR_ANY &&
1438             !(group_type&IPV6_ADDR_MULTICAST))
1439                 goto out;
1440
1441         if (len < MLD_V1_QUERY_LEN) {
1442                 goto out;
1443         } else if (len == MLD_V1_QUERY_LEN || mld_in_v1_mode(idev)) {
1444                 err = mld_process_v1(idev, mld, &max_delay,
1445                                      len == MLD_V1_QUERY_LEN);
1446                 if (err < 0)
1447                         goto out;
1448         } else if (len >= MLD_V2_QUERY_LEN_MIN) {
1449                 int srcs_offset = sizeof(struct mld2_query) -
1450                                   sizeof(struct icmp6hdr);
1451
1452                 if (!pskb_may_pull(skb, srcs_offset))
1453                         goto out;
1454
1455                 mlh2 = (struct mld2_query *)skb_transport_header(skb);
1456
1457                 err = mld_process_v2(idev, mlh2, &max_delay);
1458                 if (err < 0)
1459                         goto out;
1460
1461                 if (group_type == IPV6_ADDR_ANY) { /* general query */
1462                         if (mlh2->mld2q_nsrcs)
1463                                 goto out; /* no sources allowed */
1464
1465                         mld_gq_start_work(idev);
1466                         goto out;
1467                 }
1468                 /* mark sources to include, if group & source-specific */
1469                 if (mlh2->mld2q_nsrcs != 0) {
1470                         if (!pskb_may_pull(skb, srcs_offset +
1471                             ntohs(mlh2->mld2q_nsrcs) * sizeof(struct in6_addr)))
1472                                 goto out;
1473
1474                         mlh2 = (struct mld2_query *)skb_transport_header(skb);
1475                         mark = 1;
1476                 }
1477         } else {
1478                 goto out;
1479         }
1480
1481         if (group_type == IPV6_ADDR_ANY) {
1482                 for_each_mc_mclock(idev, ma) {
1483                         igmp6_group_queried(ma, max_delay);
1484                 }
1485         } else {
1486                 for_each_mc_mclock(idev, ma) {
1487                         if (!ipv6_addr_equal(group, &ma->mca_addr))
1488                                 continue;
1489                         if (ma->mca_flags & MAF_TIMER_RUNNING) {
1490                                 /* gsquery <- gsquery && mark */
1491                                 if (!mark)
1492                                         ma->mca_flags &= ~MAF_GSQUERY;
1493                         } else {
1494                                 /* gsquery <- mark */
1495                                 if (mark)
1496                                         ma->mca_flags |= MAF_GSQUERY;
1497                                 else
1498                                         ma->mca_flags &= ~MAF_GSQUERY;
1499                         }
1500                         if (!(ma->mca_flags & MAF_GSQUERY) ||
1501                             mld_marksources(ma, ntohs(mlh2->mld2q_nsrcs), mlh2->mld2q_srcs))
1502                                 igmp6_group_queried(ma, max_delay);
1503                         break;
1504                 }
1505         }
1506
1507 out:
1508         in6_dev_put(idev);
1509 kfree_skb:
1510         consume_skb(skb);
1511 }
1512
1513 static void mld_query_work(struct work_struct *work)
1514 {
1515         struct inet6_dev *idev = container_of(to_delayed_work(work),
1516                                               struct inet6_dev,
1517                                               mc_query_work);
1518         struct sk_buff_head q;
1519         struct sk_buff *skb;
1520         bool rework = false;
1521         int cnt = 0;
1522
1523         skb_queue_head_init(&q);
1524
1525         spin_lock_bh(&idev->mc_query_lock);
1526         while ((skb = __skb_dequeue(&idev->mc_query_queue))) {
1527                 __skb_queue_tail(&q, skb);
1528
1529                 if (++cnt >= MLD_MAX_QUEUE) {
1530                         rework = true;
1531                         schedule_delayed_work(&idev->mc_query_work, 0);
1532                         break;
1533                 }
1534         }
1535         spin_unlock_bh(&idev->mc_query_lock);
1536
1537         mutex_lock(&idev->mc_lock);
1538         while ((skb = __skb_dequeue(&q)))
1539                 __mld_query_work(skb);
1540         mutex_unlock(&idev->mc_lock);
1541
1542         if (!rework)
1543                 in6_dev_put(idev);
1544 }
1545
1546 /* called with rcu_read_lock() */
1547 int igmp6_event_report(struct sk_buff *skb)
1548 {
1549         struct inet6_dev *idev = __in6_dev_get(skb->dev);
1550
1551         if (!idev)
1552                 return -EINVAL;
1553
1554         if (idev->dead) {
1555                 kfree_skb(skb);
1556                 return -ENODEV;
1557         }
1558
1559         spin_lock_bh(&idev->mc_report_lock);
1560         if (skb_queue_len(&idev->mc_report_queue) < MLD_MAX_SKBS) {
1561                 __skb_queue_tail(&idev->mc_report_queue, skb);
1562                 if (!mod_delayed_work(mld_wq, &idev->mc_report_work, 0))
1563                         in6_dev_hold(idev);
1564         }
1565         spin_unlock_bh(&idev->mc_report_lock);
1566
1567         return 0;
1568 }
1569
1570 static void __mld_report_work(struct sk_buff *skb)
1571 {
1572         struct inet6_dev *idev;
1573         struct ifmcaddr6 *ma;
1574         struct mld_msg *mld;
1575         int addr_type;
1576
1577         /* Our own report looped back. Ignore it. */
1578         if (skb->pkt_type == PACKET_LOOPBACK)
1579                 goto kfree_skb;
1580
1581         /* send our report if the MC router may not have heard this report */
1582         if (skb->pkt_type != PACKET_MULTICAST &&
1583             skb->pkt_type != PACKET_BROADCAST)
1584                 goto kfree_skb;
1585
1586         if (!pskb_may_pull(skb, sizeof(*mld) - sizeof(struct icmp6hdr)))
1587                 goto kfree_skb;
1588
1589         mld = (struct mld_msg *)icmp6_hdr(skb);
1590
1591         /* Drop reports with not link local source */
1592         addr_type = ipv6_addr_type(&ipv6_hdr(skb)->saddr);
1593         if (addr_type != IPV6_ADDR_ANY &&
1594             !(addr_type&IPV6_ADDR_LINKLOCAL))
1595                 goto kfree_skb;
1596
1597         idev = in6_dev_get(skb->dev);
1598         if (!idev)
1599                 goto kfree_skb;
1600
1601         /*
1602          *      Cancel the work for this group
1603          */
1604
1605         for_each_mc_mclock(idev, ma) {
1606                 if (ipv6_addr_equal(&ma->mca_addr, &mld->mld_mca)) {
1607                         if (cancel_delayed_work(&ma->mca_work))
1608                                 refcount_dec(&ma->mca_refcnt);
1609                         ma->mca_flags &= ~(MAF_LAST_REPORTER |
1610                                            MAF_TIMER_RUNNING);
1611                         break;
1612                 }
1613         }
1614
1615         in6_dev_put(idev);
1616 kfree_skb:
1617         consume_skb(skb);
1618 }
1619
1620 static void mld_report_work(struct work_struct *work)
1621 {
1622         struct inet6_dev *idev = container_of(to_delayed_work(work),
1623                                               struct inet6_dev,
1624                                               mc_report_work);
1625         struct sk_buff_head q;
1626         struct sk_buff *skb;
1627         bool rework = false;
1628         int cnt = 0;
1629
1630         skb_queue_head_init(&q);
1631         spin_lock_bh(&idev->mc_report_lock);
1632         while ((skb = __skb_dequeue(&idev->mc_report_queue))) {
1633                 __skb_queue_tail(&q, skb);
1634
1635                 if (++cnt >= MLD_MAX_QUEUE) {
1636                         rework = true;
1637                         schedule_delayed_work(&idev->mc_report_work, 0);
1638                         break;
1639                 }
1640         }
1641         spin_unlock_bh(&idev->mc_report_lock);
1642
1643         mutex_lock(&idev->mc_lock);
1644         while ((skb = __skb_dequeue(&q)))
1645                 __mld_report_work(skb);
1646         mutex_unlock(&idev->mc_lock);
1647
1648         if (!rework)
1649                 in6_dev_put(idev);
1650 }
1651
1652 static bool is_in(struct ifmcaddr6 *pmc, struct ip6_sf_list *psf, int type,
1653                   int gdeleted, int sdeleted)
1654 {
1655         switch (type) {
1656         case MLD2_MODE_IS_INCLUDE:
1657         case MLD2_MODE_IS_EXCLUDE:
1658                 if (gdeleted || sdeleted)
1659                         return false;
1660                 if (!((pmc->mca_flags & MAF_GSQUERY) && !psf->sf_gsresp)) {
1661                         if (pmc->mca_sfmode == MCAST_INCLUDE)
1662                                 return true;
1663                         /* don't include if this source is excluded
1664                          * in all filters
1665                          */
1666                         if (psf->sf_count[MCAST_INCLUDE])
1667                                 return type == MLD2_MODE_IS_INCLUDE;
1668                         return pmc->mca_sfcount[MCAST_EXCLUDE] ==
1669                                 psf->sf_count[MCAST_EXCLUDE];
1670                 }
1671                 return false;
1672         case MLD2_CHANGE_TO_INCLUDE:
1673                 if (gdeleted || sdeleted)
1674                         return false;
1675                 return psf->sf_count[MCAST_INCLUDE] != 0;
1676         case MLD2_CHANGE_TO_EXCLUDE:
1677                 if (gdeleted || sdeleted)
1678                         return false;
1679                 if (pmc->mca_sfcount[MCAST_EXCLUDE] == 0 ||
1680                     psf->sf_count[MCAST_INCLUDE])
1681                         return false;
1682                 return pmc->mca_sfcount[MCAST_EXCLUDE] ==
1683                         psf->sf_count[MCAST_EXCLUDE];
1684         case MLD2_ALLOW_NEW_SOURCES:
1685                 if (gdeleted || !psf->sf_crcount)
1686                         return false;
1687                 return (pmc->mca_sfmode == MCAST_INCLUDE) ^ sdeleted;
1688         case MLD2_BLOCK_OLD_SOURCES:
1689                 if (pmc->mca_sfmode == MCAST_INCLUDE)
1690                         return gdeleted || (psf->sf_crcount && sdeleted);
1691                 return psf->sf_crcount && !gdeleted && !sdeleted;
1692         }
1693         return false;
1694 }
1695
1696 static int
1697 mld_scount(struct ifmcaddr6 *pmc, int type, int gdeleted, int sdeleted)
1698 {
1699         struct ip6_sf_list *psf;
1700         int scount = 0;
1701
1702         for_each_psf_mclock(pmc, psf) {
1703                 if (!is_in(pmc, psf, type, gdeleted, sdeleted))
1704                         continue;
1705                 scount++;
1706         }
1707         return scount;
1708 }
1709
1710 static void ip6_mc_hdr(struct sock *sk, struct sk_buff *skb,
1711                        struct net_device *dev,
1712                        const struct in6_addr *saddr,
1713                        const struct in6_addr *daddr,
1714                        int proto, int len)
1715 {
1716         struct ipv6hdr *hdr;
1717
1718         skb->protocol = htons(ETH_P_IPV6);
1719         skb->dev = dev;
1720
1721         skb_reset_network_header(skb);
1722         skb_put(skb, sizeof(struct ipv6hdr));
1723         hdr = ipv6_hdr(skb);
1724
1725         ip6_flow_hdr(hdr, 0, 0);
1726
1727         hdr->payload_len = htons(len);
1728         hdr->nexthdr = proto;
1729         hdr->hop_limit = inet6_sk(sk)->hop_limit;
1730
1731         hdr->saddr = *saddr;
1732         hdr->daddr = *daddr;
1733 }
1734
1735 static struct sk_buff *mld_newpack(struct inet6_dev *idev, unsigned int mtu)
1736 {
1737         u8 ra[8] = { IPPROTO_ICMPV6, 0, IPV6_TLV_ROUTERALERT,
1738                      2, 0, 0, IPV6_TLV_PADN, 0 };
1739         struct net_device *dev = idev->dev;
1740         int hlen = LL_RESERVED_SPACE(dev);
1741         int tlen = dev->needed_tailroom;
1742         struct net *net = dev_net(dev);
1743         const struct in6_addr *saddr;
1744         struct in6_addr addr_buf;
1745         struct mld2_report *pmr;
1746         struct sk_buff *skb;
1747         unsigned int size;
1748         struct sock *sk;
1749         int err;
1750
1751         sk = net->ipv6.igmp_sk;
1752         /* we assume size > sizeof(ra) here
1753          * Also try to not allocate high-order pages for big MTU
1754          */
1755         size = min_t(int, mtu, PAGE_SIZE / 2) + hlen + tlen;
1756         skb = sock_alloc_send_skb(sk, size, 1, &err);
1757         if (!skb)
1758                 return NULL;
1759
1760         skb->priority = TC_PRIO_CONTROL;
1761         skb_reserve(skb, hlen);
1762         skb_tailroom_reserve(skb, mtu, tlen);
1763
1764         if (__ipv6_get_lladdr(idev, &addr_buf, IFA_F_TENTATIVE)) {
1765                 /* <draft-ietf-magma-mld-source-05.txt>:
1766                  * use unspecified address as the source address
1767                  * when a valid link-local address is not available.
1768                  */
1769                 saddr = &in6addr_any;
1770         } else
1771                 saddr = &addr_buf;
1772
1773         ip6_mc_hdr(sk, skb, dev, saddr, &mld2_all_mcr, NEXTHDR_HOP, 0);
1774
1775         skb_put_data(skb, ra, sizeof(ra));
1776
1777         skb_set_transport_header(skb, skb_tail_pointer(skb) - skb->data);
1778         skb_put(skb, sizeof(*pmr));
1779         pmr = (struct mld2_report *)skb_transport_header(skb);
1780         pmr->mld2r_type = ICMPV6_MLD2_REPORT;
1781         pmr->mld2r_resv1 = 0;
1782         pmr->mld2r_cksum = 0;
1783         pmr->mld2r_resv2 = 0;
1784         pmr->mld2r_ngrec = 0;
1785         return skb;
1786 }
1787
1788 static void mld_sendpack(struct sk_buff *skb)
1789 {
1790         struct ipv6hdr *pip6 = ipv6_hdr(skb);
1791         struct mld2_report *pmr =
1792                               (struct mld2_report *)skb_transport_header(skb);
1793         int payload_len, mldlen;
1794         struct inet6_dev *idev;
1795         struct net *net = dev_net(skb->dev);
1796         int err;
1797         struct flowi6 fl6;
1798         struct dst_entry *dst;
1799
1800         rcu_read_lock();
1801         idev = __in6_dev_get(skb->dev);
1802         IP6_UPD_PO_STATS(net, idev, IPSTATS_MIB_OUT, skb->len);
1803
1804         payload_len = (skb_tail_pointer(skb) - skb_network_header(skb)) -
1805                 sizeof(*pip6);
1806         mldlen = skb_tail_pointer(skb) - skb_transport_header(skb);
1807         pip6->payload_len = htons(payload_len);
1808
1809         pmr->mld2r_cksum = csum_ipv6_magic(&pip6->saddr, &pip6->daddr, mldlen,
1810                                            IPPROTO_ICMPV6,
1811                                            csum_partial(skb_transport_header(skb),
1812                                                         mldlen, 0));
1813
1814         icmpv6_flow_init(net->ipv6.igmp_sk, &fl6, ICMPV6_MLD2_REPORT,
1815                          &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
1816                          skb->dev->ifindex);
1817         dst = icmp6_dst_alloc(skb->dev, &fl6);
1818
1819         err = 0;
1820         if (IS_ERR(dst)) {
1821                 err = PTR_ERR(dst);
1822                 dst = NULL;
1823         }
1824         skb_dst_set(skb, dst);
1825         if (err)
1826                 goto err_out;
1827
1828         err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
1829                       net, net->ipv6.igmp_sk, skb, NULL, skb->dev,
1830                       dst_output);
1831 out:
1832         if (!err) {
1833                 ICMP6MSGOUT_INC_STATS(net, idev, ICMPV6_MLD2_REPORT);
1834                 ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
1835         } else {
1836                 IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
1837         }
1838
1839         rcu_read_unlock();
1840         return;
1841
1842 err_out:
1843         kfree_skb(skb);
1844         goto out;
1845 }
1846
1847 static int grec_size(struct ifmcaddr6 *pmc, int type, int gdel, int sdel)
1848 {
1849         return sizeof(struct mld2_grec) + 16 * mld_scount(pmc,type,gdel,sdel);
1850 }
1851
1852 static struct sk_buff *add_grhead(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1853         int type, struct mld2_grec **ppgr, unsigned int mtu)
1854 {
1855         struct mld2_report *pmr;
1856         struct mld2_grec *pgr;
1857
1858         if (!skb) {
1859                 skb = mld_newpack(pmc->idev, mtu);
1860                 if (!skb)
1861                         return NULL;
1862         }
1863         pgr = skb_put(skb, sizeof(struct mld2_grec));
1864         pgr->grec_type = type;
1865         pgr->grec_auxwords = 0;
1866         pgr->grec_nsrcs = 0;
1867         pgr->grec_mca = pmc->mca_addr;  /* structure copy */
1868         pmr = (struct mld2_report *)skb_transport_header(skb);
1869         pmr->mld2r_ngrec = htons(ntohs(pmr->mld2r_ngrec)+1);
1870         *ppgr = pgr;
1871         return skb;
1872 }
1873
1874 #define AVAILABLE(skb)  ((skb) ? skb_availroom(skb) : 0)
1875
1876 /* called with mc_lock */
1877 static struct sk_buff *add_grec(struct sk_buff *skb, struct ifmcaddr6 *pmc,
1878                                 int type, int gdeleted, int sdeleted,
1879                                 int crsend)
1880 {
1881         struct ip6_sf_list *psf, *psf_prev, *psf_next;
1882         int scount, stotal, first, isquery, truncate;
1883         struct ip6_sf_list __rcu **psf_list;
1884         struct inet6_dev *idev = pmc->idev;
1885         struct net_device *dev = idev->dev;
1886         struct mld2_grec *pgr = NULL;
1887         struct mld2_report *pmr;
1888         unsigned int mtu;
1889
1890         if (pmc->mca_flags & MAF_NOREPORT)
1891                 return skb;
1892
1893         mtu = READ_ONCE(dev->mtu);
1894         if (mtu < IPV6_MIN_MTU)
1895                 return skb;
1896
1897         isquery = type == MLD2_MODE_IS_INCLUDE ||
1898                   type == MLD2_MODE_IS_EXCLUDE;
1899         truncate = type == MLD2_MODE_IS_EXCLUDE ||
1900                     type == MLD2_CHANGE_TO_EXCLUDE;
1901
1902         stotal = scount = 0;
1903
1904         psf_list = sdeleted ? &pmc->mca_tomb : &pmc->mca_sources;
1905
1906         if (!rcu_access_pointer(*psf_list))
1907                 goto empty_source;
1908
1909         pmr = skb ? (struct mld2_report *)skb_transport_header(skb) : NULL;
1910
1911         /* EX and TO_EX get a fresh packet, if needed */
1912         if (truncate) {
1913                 if (pmr && pmr->mld2r_ngrec &&
1914                     AVAILABLE(skb) < grec_size(pmc, type, gdeleted, sdeleted)) {
1915                         if (skb)
1916                                 mld_sendpack(skb);
1917                         skb = mld_newpack(idev, mtu);
1918                 }
1919         }
1920         first = 1;
1921         psf_prev = NULL;
1922         for (psf = mc_dereference(*psf_list, idev);
1923              psf;
1924              psf = psf_next) {
1925                 struct in6_addr *psrc;
1926
1927                 psf_next = mc_dereference(psf->sf_next, idev);
1928
1929                 if (!is_in(pmc, psf, type, gdeleted, sdeleted) && !crsend) {
1930                         psf_prev = psf;
1931                         continue;
1932                 }
1933
1934                 /* Based on RFC3810 6.1. Should not send source-list change
1935                  * records when there is a filter mode change.
1936                  */
1937                 if (((gdeleted && pmc->mca_sfmode == MCAST_EXCLUDE) ||
1938                      (!gdeleted && pmc->mca_crcount)) &&
1939                     (type == MLD2_ALLOW_NEW_SOURCES ||
1940                      type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount)
1941                         goto decrease_sf_crcount;
1942
1943                 /* clear marks on query responses */
1944                 if (isquery)
1945                         psf->sf_gsresp = 0;
1946
1947                 if (AVAILABLE(skb) < sizeof(*psrc) +
1948                     first*sizeof(struct mld2_grec)) {
1949                         if (truncate && !first)
1950                                 break;   /* truncate these */
1951                         if (pgr)
1952                                 pgr->grec_nsrcs = htons(scount);
1953                         if (skb)
1954                                 mld_sendpack(skb);
1955                         skb = mld_newpack(idev, mtu);
1956                         first = 1;
1957                         scount = 0;
1958                 }
1959                 if (first) {
1960                         skb = add_grhead(skb, pmc, type, &pgr, mtu);
1961                         first = 0;
1962                 }
1963                 if (!skb)
1964                         return NULL;
1965                 psrc = skb_put(skb, sizeof(*psrc));
1966                 *psrc = psf->sf_addr;
1967                 scount++; stotal++;
1968                 if ((type == MLD2_ALLOW_NEW_SOURCES ||
1969                      type == MLD2_BLOCK_OLD_SOURCES) && psf->sf_crcount) {
1970 decrease_sf_crcount:
1971                         psf->sf_crcount--;
1972                         if ((sdeleted || gdeleted) && psf->sf_crcount == 0) {
1973                                 if (psf_prev)
1974                                         rcu_assign_pointer(psf_prev->sf_next,
1975                                                            mc_dereference(psf->sf_next, idev));
1976                                 else
1977                                         rcu_assign_pointer(*psf_list,
1978                                                            mc_dereference(psf->sf_next, idev));
1979                                 kfree_rcu(psf, rcu);
1980                                 continue;
1981                         }
1982                 }
1983                 psf_prev = psf;
1984         }
1985
1986 empty_source:
1987         if (!stotal) {
1988                 if (type == MLD2_ALLOW_NEW_SOURCES ||
1989                     type == MLD2_BLOCK_OLD_SOURCES)
1990                         return skb;
1991                 if (pmc->mca_crcount || isquery || crsend) {
1992                         /* make sure we have room for group header */
1993                         if (skb && AVAILABLE(skb) < sizeof(struct mld2_grec)) {
1994                                 mld_sendpack(skb);
1995                                 skb = NULL; /* add_grhead will get a new one */
1996                         }
1997                         skb = add_grhead(skb, pmc, type, &pgr, mtu);
1998                 }
1999         }
2000         if (pgr)
2001                 pgr->grec_nsrcs = htons(scount);
2002
2003         if (isquery)
2004                 pmc->mca_flags &= ~MAF_GSQUERY; /* clear query state */
2005         return skb;
2006 }
2007
2008 /* called with mc_lock */
2009 static void mld_send_report(struct inet6_dev *idev, struct ifmcaddr6 *pmc)
2010 {
2011         struct sk_buff *skb = NULL;
2012         int type;
2013
2014         if (!pmc) {
2015                 for_each_mc_mclock(idev, pmc) {
2016                         if (pmc->mca_flags & MAF_NOREPORT)
2017                                 continue;
2018                         if (pmc->mca_sfcount[MCAST_EXCLUDE])
2019                                 type = MLD2_MODE_IS_EXCLUDE;
2020                         else
2021                                 type = MLD2_MODE_IS_INCLUDE;
2022                         skb = add_grec(skb, pmc, type, 0, 0, 0);
2023                 }
2024         } else {
2025                 if (pmc->mca_sfcount[MCAST_EXCLUDE])
2026                         type = MLD2_MODE_IS_EXCLUDE;
2027                 else
2028                         type = MLD2_MODE_IS_INCLUDE;
2029                 skb = add_grec(skb, pmc, type, 0, 0, 0);
2030         }
2031         if (skb)
2032                 mld_sendpack(skb);
2033 }
2034
2035 /*
2036  * remove zero-count source records from a source filter list
2037  * called with mc_lock
2038  */
2039 static void mld_clear_zeros(struct ip6_sf_list __rcu **ppsf, struct inet6_dev *idev)
2040 {
2041         struct ip6_sf_list *psf_prev, *psf_next, *psf;
2042
2043         psf_prev = NULL;
2044         for (psf = mc_dereference(*ppsf, idev);
2045              psf;
2046              psf = psf_next) {
2047                 psf_next = mc_dereference(psf->sf_next, idev);
2048                 if (psf->sf_crcount == 0) {
2049                         if (psf_prev)
2050                                 rcu_assign_pointer(psf_prev->sf_next,
2051                                                    mc_dereference(psf->sf_next, idev));
2052                         else
2053                                 rcu_assign_pointer(*ppsf,
2054                                                    mc_dereference(psf->sf_next, idev));
2055                         kfree_rcu(psf, rcu);
2056                 } else {
2057                         psf_prev = psf;
2058                 }
2059         }
2060 }
2061
2062 /* called with mc_lock */
2063 static void mld_send_cr(struct inet6_dev *idev)
2064 {
2065         struct ifmcaddr6 *pmc, *pmc_prev, *pmc_next;
2066         struct sk_buff *skb = NULL;
2067         int type, dtype;
2068
2069         /* deleted MCA's */
2070         pmc_prev = NULL;
2071         for (pmc = mc_dereference(idev->mc_tomb, idev);
2072              pmc;
2073              pmc = pmc_next) {
2074                 pmc_next = mc_dereference(pmc->next, idev);
2075                 if (pmc->mca_sfmode == MCAST_INCLUDE) {
2076                         type = MLD2_BLOCK_OLD_SOURCES;
2077                         dtype = MLD2_BLOCK_OLD_SOURCES;
2078                         skb = add_grec(skb, pmc, type, 1, 0, 0);
2079                         skb = add_grec(skb, pmc, dtype, 1, 1, 0);
2080                 }
2081                 if (pmc->mca_crcount) {
2082                         if (pmc->mca_sfmode == MCAST_EXCLUDE) {
2083                                 type = MLD2_CHANGE_TO_INCLUDE;
2084                                 skb = add_grec(skb, pmc, type, 1, 0, 0);
2085                         }
2086                         pmc->mca_crcount--;
2087                         if (pmc->mca_crcount == 0) {
2088                                 mld_clear_zeros(&pmc->mca_tomb, idev);
2089                                 mld_clear_zeros(&pmc->mca_sources, idev);
2090                         }
2091                 }
2092                 if (pmc->mca_crcount == 0 &&
2093                     !rcu_access_pointer(pmc->mca_tomb) &&
2094                     !rcu_access_pointer(pmc->mca_sources)) {
2095                         if (pmc_prev)
2096                                 rcu_assign_pointer(pmc_prev->next, pmc_next);
2097                         else
2098                                 rcu_assign_pointer(idev->mc_tomb, pmc_next);
2099                         in6_dev_put(pmc->idev);
2100                         kfree_rcu(pmc, rcu);
2101                 } else
2102                         pmc_prev = pmc;
2103         }
2104
2105         /* change recs */
2106         for_each_mc_mclock(idev, pmc) {
2107                 if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2108                         type = MLD2_BLOCK_OLD_SOURCES;
2109                         dtype = MLD2_ALLOW_NEW_SOURCES;
2110                 } else {
2111                         type = MLD2_ALLOW_NEW_SOURCES;
2112                         dtype = MLD2_BLOCK_OLD_SOURCES;
2113                 }
2114                 skb = add_grec(skb, pmc, type, 0, 0, 0);
2115                 skb = add_grec(skb, pmc, dtype, 0, 1, 0);       /* deleted sources */
2116
2117                 /* filter mode changes */
2118                 if (pmc->mca_crcount) {
2119                         if (pmc->mca_sfmode == MCAST_EXCLUDE)
2120                                 type = MLD2_CHANGE_TO_EXCLUDE;
2121                         else
2122                                 type = MLD2_CHANGE_TO_INCLUDE;
2123                         skb = add_grec(skb, pmc, type, 0, 0, 0);
2124                         pmc->mca_crcount--;
2125                 }
2126         }
2127         if (!skb)
2128                 return;
2129         (void) mld_sendpack(skb);
2130 }
2131
2132 static void igmp6_send(struct in6_addr *addr, struct net_device *dev, int type)
2133 {
2134         struct net *net = dev_net(dev);
2135         struct sock *sk = net->ipv6.igmp_sk;
2136         struct inet6_dev *idev;
2137         struct sk_buff *skb;
2138         struct mld_msg *hdr;
2139         const struct in6_addr *snd_addr, *saddr;
2140         struct in6_addr addr_buf;
2141         int hlen = LL_RESERVED_SPACE(dev);
2142         int tlen = dev->needed_tailroom;
2143         int err, len, payload_len, full_len;
2144         u8 ra[8] = { IPPROTO_ICMPV6, 0,
2145                      IPV6_TLV_ROUTERALERT, 2, 0, 0,
2146                      IPV6_TLV_PADN, 0 };
2147         struct flowi6 fl6;
2148         struct dst_entry *dst;
2149
2150         if (type == ICMPV6_MGM_REDUCTION)
2151                 snd_addr = &in6addr_linklocal_allrouters;
2152         else
2153                 snd_addr = addr;
2154
2155         len = sizeof(struct icmp6hdr) + sizeof(struct in6_addr);
2156         payload_len = len + sizeof(ra);
2157         full_len = sizeof(struct ipv6hdr) + payload_len;
2158
2159         rcu_read_lock();
2160         IP6_UPD_PO_STATS(net, __in6_dev_get(dev),
2161                       IPSTATS_MIB_OUT, full_len);
2162         rcu_read_unlock();
2163
2164         skb = sock_alloc_send_skb(sk, hlen + tlen + full_len, 1, &err);
2165
2166         if (!skb) {
2167                 rcu_read_lock();
2168                 IP6_INC_STATS(net, __in6_dev_get(dev),
2169                               IPSTATS_MIB_OUTDISCARDS);
2170                 rcu_read_unlock();
2171                 return;
2172         }
2173         skb->priority = TC_PRIO_CONTROL;
2174         skb_reserve(skb, hlen);
2175
2176         if (ipv6_get_lladdr(dev, &addr_buf, IFA_F_TENTATIVE)) {
2177                 /* <draft-ietf-magma-mld-source-05.txt>:
2178                  * use unspecified address as the source address
2179                  * when a valid link-local address is not available.
2180                  */
2181                 saddr = &in6addr_any;
2182         } else
2183                 saddr = &addr_buf;
2184
2185         ip6_mc_hdr(sk, skb, dev, saddr, snd_addr, NEXTHDR_HOP, payload_len);
2186
2187         skb_put_data(skb, ra, sizeof(ra));
2188
2189         hdr = skb_put_zero(skb, sizeof(struct mld_msg));
2190         hdr->mld_type = type;
2191         hdr->mld_mca = *addr;
2192
2193         hdr->mld_cksum = csum_ipv6_magic(saddr, snd_addr, len,
2194                                          IPPROTO_ICMPV6,
2195                                          csum_partial(hdr, len, 0));
2196
2197         rcu_read_lock();
2198         idev = __in6_dev_get(skb->dev);
2199
2200         icmpv6_flow_init(sk, &fl6, type,
2201                          &ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
2202                          skb->dev->ifindex);
2203         dst = icmp6_dst_alloc(skb->dev, &fl6);
2204         if (IS_ERR(dst)) {
2205                 err = PTR_ERR(dst);
2206                 goto err_out;
2207         }
2208
2209         skb_dst_set(skb, dst);
2210         err = NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT,
2211                       net, sk, skb, NULL, skb->dev,
2212                       dst_output);
2213 out:
2214         if (!err) {
2215                 ICMP6MSGOUT_INC_STATS(net, idev, type);
2216                 ICMP6_INC_STATS(net, idev, ICMP6_MIB_OUTMSGS);
2217         } else
2218                 IP6_INC_STATS(net, idev, IPSTATS_MIB_OUTDISCARDS);
2219
2220         rcu_read_unlock();
2221         return;
2222
2223 err_out:
2224         kfree_skb(skb);
2225         goto out;
2226 }
2227
2228 /* called with mc_lock */
2229 static void mld_send_initial_cr(struct inet6_dev *idev)
2230 {
2231         struct sk_buff *skb;
2232         struct ifmcaddr6 *pmc;
2233         int type;
2234
2235         if (mld_in_v1_mode(idev))
2236                 return;
2237
2238         skb = NULL;
2239         for_each_mc_mclock(idev, pmc) {
2240                 if (pmc->mca_sfcount[MCAST_EXCLUDE])
2241                         type = MLD2_CHANGE_TO_EXCLUDE;
2242                 else
2243                         type = MLD2_ALLOW_NEW_SOURCES;
2244                 skb = add_grec(skb, pmc, type, 0, 0, 1);
2245         }
2246         if (skb)
2247                 mld_sendpack(skb);
2248 }
2249
2250 void ipv6_mc_dad_complete(struct inet6_dev *idev)
2251 {
2252         mutex_lock(&idev->mc_lock);
2253         idev->mc_dad_count = idev->mc_qrv;
2254         if (idev->mc_dad_count) {
2255                 mld_send_initial_cr(idev);
2256                 idev->mc_dad_count--;
2257                 if (idev->mc_dad_count)
2258                         mld_dad_start_work(idev,
2259                                            unsolicited_report_interval(idev));
2260         }
2261         mutex_unlock(&idev->mc_lock);
2262 }
2263
2264 static void mld_dad_work(struct work_struct *work)
2265 {
2266         struct inet6_dev *idev = container_of(to_delayed_work(work),
2267                                               struct inet6_dev,
2268                                               mc_dad_work);
2269         mutex_lock(&idev->mc_lock);
2270         mld_send_initial_cr(idev);
2271         if (idev->mc_dad_count) {
2272                 idev->mc_dad_count--;
2273                 if (idev->mc_dad_count)
2274                         mld_dad_start_work(idev,
2275                                            unsolicited_report_interval(idev));
2276         }
2277         mutex_unlock(&idev->mc_lock);
2278         in6_dev_put(idev);
2279 }
2280
2281 /* called with mc_lock */
2282 static int ip6_mc_del1_src(struct ifmcaddr6 *pmc, int sfmode,
2283         const struct in6_addr *psfsrc)
2284 {
2285         struct ip6_sf_list *psf, *psf_prev;
2286         int rv = 0;
2287
2288         psf_prev = NULL;
2289         for_each_psf_mclock(pmc, psf) {
2290                 if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2291                         break;
2292                 psf_prev = psf;
2293         }
2294         if (!psf || psf->sf_count[sfmode] == 0) {
2295                 /* source filter not found, or count wrong =>  bug */
2296                 return -ESRCH;
2297         }
2298         psf->sf_count[sfmode]--;
2299         if (!psf->sf_count[MCAST_INCLUDE] && !psf->sf_count[MCAST_EXCLUDE]) {
2300                 struct inet6_dev *idev = pmc->idev;
2301
2302                 /* no more filters for this source */
2303                 if (psf_prev)
2304                         rcu_assign_pointer(psf_prev->sf_next,
2305                                            mc_dereference(psf->sf_next, idev));
2306                 else
2307                         rcu_assign_pointer(pmc->mca_sources,
2308                                            mc_dereference(psf->sf_next, idev));
2309
2310                 if (psf->sf_oldin && !(pmc->mca_flags & MAF_NOREPORT) &&
2311                     !mld_in_v1_mode(idev)) {
2312                         psf->sf_crcount = idev->mc_qrv;
2313                         rcu_assign_pointer(psf->sf_next,
2314                                            mc_dereference(pmc->mca_tomb, idev));
2315                         rcu_assign_pointer(pmc->mca_tomb, psf);
2316                         rv = 1;
2317                 } else {
2318                         kfree_rcu(psf, rcu);
2319                 }
2320         }
2321         return rv;
2322 }
2323
2324 /* called with mc_lock */
2325 static int ip6_mc_del_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2326                           int sfmode, int sfcount, const struct in6_addr *psfsrc,
2327                           int delta)
2328 {
2329         struct ifmcaddr6 *pmc;
2330         int     changerec = 0;
2331         int     i, err;
2332
2333         if (!idev)
2334                 return -ENODEV;
2335
2336         for_each_mc_mclock(idev, pmc) {
2337                 if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2338                         break;
2339         }
2340         if (!pmc)
2341                 return -ESRCH;
2342
2343         sf_markstate(pmc);
2344         if (!delta) {
2345                 if (!pmc->mca_sfcount[sfmode])
2346                         return -EINVAL;
2347
2348                 pmc->mca_sfcount[sfmode]--;
2349         }
2350         err = 0;
2351         for (i = 0; i < sfcount; i++) {
2352                 int rv = ip6_mc_del1_src(pmc, sfmode, &psfsrc[i]);
2353
2354                 changerec |= rv > 0;
2355                 if (!err && rv < 0)
2356                         err = rv;
2357         }
2358         if (pmc->mca_sfmode == MCAST_EXCLUDE &&
2359             pmc->mca_sfcount[MCAST_EXCLUDE] == 0 &&
2360             pmc->mca_sfcount[MCAST_INCLUDE]) {
2361                 struct ip6_sf_list *psf;
2362
2363                 /* filter mode change */
2364                 pmc->mca_sfmode = MCAST_INCLUDE;
2365                 pmc->mca_crcount = idev->mc_qrv;
2366                 idev->mc_ifc_count = pmc->mca_crcount;
2367                 for_each_psf_mclock(pmc, psf)
2368                         psf->sf_crcount = 0;
2369                 mld_ifc_event(pmc->idev);
2370         } else if (sf_setstate(pmc) || changerec) {
2371                 mld_ifc_event(pmc->idev);
2372         }
2373
2374         return err;
2375 }
2376
2377 /*
2378  * Add multicast single-source filter to the interface list
2379  * called with mc_lock
2380  */
2381 static int ip6_mc_add1_src(struct ifmcaddr6 *pmc, int sfmode,
2382         const struct in6_addr *psfsrc)
2383 {
2384         struct ip6_sf_list *psf, *psf_prev;
2385
2386         psf_prev = NULL;
2387         for_each_psf_mclock(pmc, psf) {
2388                 if (ipv6_addr_equal(&psf->sf_addr, psfsrc))
2389                         break;
2390                 psf_prev = psf;
2391         }
2392         if (!psf) {
2393                 psf = kzalloc(sizeof(*psf), GFP_KERNEL);
2394                 if (!psf)
2395                         return -ENOBUFS;
2396
2397                 psf->sf_addr = *psfsrc;
2398                 if (psf_prev) {
2399                         rcu_assign_pointer(psf_prev->sf_next, psf);
2400                 } else {
2401                         rcu_assign_pointer(pmc->mca_sources, psf);
2402                 }
2403         }
2404         psf->sf_count[sfmode]++;
2405         return 0;
2406 }
2407
2408 /* called with mc_lock */
2409 static void sf_markstate(struct ifmcaddr6 *pmc)
2410 {
2411         struct ip6_sf_list *psf;
2412         int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2413
2414         for_each_psf_mclock(pmc, psf) {
2415                 if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2416                         psf->sf_oldin = mca_xcount ==
2417                                 psf->sf_count[MCAST_EXCLUDE] &&
2418                                 !psf->sf_count[MCAST_INCLUDE];
2419                 } else {
2420                         psf->sf_oldin = psf->sf_count[MCAST_INCLUDE] != 0;
2421                 }
2422         }
2423 }
2424
2425 /* called with mc_lock */
2426 static int sf_setstate(struct ifmcaddr6 *pmc)
2427 {
2428         struct ip6_sf_list *psf, *dpsf;
2429         int mca_xcount = pmc->mca_sfcount[MCAST_EXCLUDE];
2430         int qrv = pmc->idev->mc_qrv;
2431         int new_in, rv;
2432
2433         rv = 0;
2434         for_each_psf_mclock(pmc, psf) {
2435                 if (pmc->mca_sfcount[MCAST_EXCLUDE]) {
2436                         new_in = mca_xcount == psf->sf_count[MCAST_EXCLUDE] &&
2437                                 !psf->sf_count[MCAST_INCLUDE];
2438                 } else
2439                         new_in = psf->sf_count[MCAST_INCLUDE] != 0;
2440                 if (new_in) {
2441                         if (!psf->sf_oldin) {
2442                                 struct ip6_sf_list *prev = NULL;
2443
2444                                 for_each_psf_tomb(pmc, dpsf) {
2445                                         if (ipv6_addr_equal(&dpsf->sf_addr,
2446                                             &psf->sf_addr))
2447                                                 break;
2448                                         prev = dpsf;
2449                                 }
2450                                 if (dpsf) {
2451                                         if (prev)
2452                                                 rcu_assign_pointer(prev->sf_next,
2453                                                                    mc_dereference(dpsf->sf_next,
2454                                                                                   pmc->idev));
2455                                         else
2456                                                 rcu_assign_pointer(pmc->mca_tomb,
2457                                                                    mc_dereference(dpsf->sf_next,
2458                                                                                   pmc->idev));
2459                                         kfree_rcu(dpsf, rcu);
2460                                 }
2461                                 psf->sf_crcount = qrv;
2462                                 rv++;
2463                         }
2464                 } else if (psf->sf_oldin) {
2465                         psf->sf_crcount = 0;
2466                         /*
2467                          * add or update "delete" records if an active filter
2468                          * is now inactive
2469                          */
2470
2471                         for_each_psf_tomb(pmc, dpsf)
2472                                 if (ipv6_addr_equal(&dpsf->sf_addr,
2473                                     &psf->sf_addr))
2474                                         break;
2475                         if (!dpsf) {
2476                                 dpsf = kmalloc(sizeof(*dpsf), GFP_KERNEL);
2477                                 if (!dpsf)
2478                                         continue;
2479                                 *dpsf = *psf;
2480                                 rcu_assign_pointer(dpsf->sf_next,
2481                                                    mc_dereference(pmc->mca_tomb, pmc->idev));
2482                                 rcu_assign_pointer(pmc->mca_tomb, dpsf);
2483                         }
2484                         dpsf->sf_crcount = qrv;
2485                         rv++;
2486                 }
2487         }
2488         return rv;
2489 }
2490
2491 /*
2492  * Add multicast source filter list to the interface list
2493  * called with mc_lock
2494  */
2495 static int ip6_mc_add_src(struct inet6_dev *idev, const struct in6_addr *pmca,
2496                           int sfmode, int sfcount, const struct in6_addr *psfsrc,
2497                           int delta)
2498 {
2499         struct ifmcaddr6 *pmc;
2500         int     isexclude;
2501         int     i, err;
2502
2503         if (!idev)
2504                 return -ENODEV;
2505
2506         for_each_mc_mclock(idev, pmc) {
2507                 if (ipv6_addr_equal(pmca, &pmc->mca_addr))
2508                         break;
2509         }
2510         if (!pmc)
2511                 return -ESRCH;
2512
2513         sf_markstate(pmc);
2514         isexclude = pmc->mca_sfmode == MCAST_EXCLUDE;
2515         if (!delta)
2516                 pmc->mca_sfcount[sfmode]++;
2517         err = 0;
2518         for (i = 0; i < sfcount; i++) {
2519                 err = ip6_mc_add1_src(pmc, sfmode, &psfsrc[i]);
2520                 if (err)
2521                         break;
2522         }
2523         if (err) {
2524                 int j;
2525
2526                 if (!delta)
2527                         pmc->mca_sfcount[sfmode]--;
2528                 for (j = 0; j < i; j++)
2529                         ip6_mc_del1_src(pmc, sfmode, &psfsrc[j]);
2530         } else if (isexclude != (pmc->mca_sfcount[MCAST_EXCLUDE] != 0)) {
2531                 struct ip6_sf_list *psf;
2532
2533                 /* filter mode change */
2534                 if (pmc->mca_sfcount[MCAST_EXCLUDE])
2535                         pmc->mca_sfmode = MCAST_EXCLUDE;
2536                 else if (pmc->mca_sfcount[MCAST_INCLUDE])
2537                         pmc->mca_sfmode = MCAST_INCLUDE;
2538                 /* else no filters; keep old mode for reports */
2539
2540                 pmc->mca_crcount = idev->mc_qrv;
2541                 idev->mc_ifc_count = pmc->mca_crcount;
2542                 for_each_psf_mclock(pmc, psf)
2543                         psf->sf_crcount = 0;
2544                 mld_ifc_event(idev);
2545         } else if (sf_setstate(pmc)) {
2546                 mld_ifc_event(idev);
2547         }
2548         return err;
2549 }
2550
2551 /* called with mc_lock */
2552 static void ip6_mc_clear_src(struct ifmcaddr6 *pmc)
2553 {
2554         struct ip6_sf_list *psf, *nextpsf;
2555
2556         for (psf = mc_dereference(pmc->mca_tomb, pmc->idev);
2557              psf;
2558              psf = nextpsf) {
2559                 nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2560                 kfree_rcu(psf, rcu);
2561         }
2562         RCU_INIT_POINTER(pmc->mca_tomb, NULL);
2563         for (psf = mc_dereference(pmc->mca_sources, pmc->idev);
2564              psf;
2565              psf = nextpsf) {
2566                 nextpsf = mc_dereference(psf->sf_next, pmc->idev);
2567                 kfree_rcu(psf, rcu);
2568         }
2569         RCU_INIT_POINTER(pmc->mca_sources, NULL);
2570         pmc->mca_sfmode = MCAST_EXCLUDE;
2571         pmc->mca_sfcount[MCAST_INCLUDE] = 0;
2572         pmc->mca_sfcount[MCAST_EXCLUDE] = 1;
2573 }
2574
2575 /* called with mc_lock */
2576 static void igmp6_join_group(struct ifmcaddr6 *ma)
2577 {
2578         unsigned long delay;
2579
2580         if (ma->mca_flags & MAF_NOREPORT)
2581                 return;
2582
2583         igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2584
2585         delay = prandom_u32() % unsolicited_report_interval(ma->idev);
2586
2587         if (cancel_delayed_work(&ma->mca_work)) {
2588                 refcount_dec(&ma->mca_refcnt);
2589                 delay = ma->mca_work.timer.expires - jiffies;
2590         }
2591
2592         if (!mod_delayed_work(mld_wq, &ma->mca_work, delay))
2593                 refcount_inc(&ma->mca_refcnt);
2594         ma->mca_flags |= MAF_TIMER_RUNNING | MAF_LAST_REPORTER;
2595 }
2596
2597 static int ip6_mc_leave_src(struct sock *sk, struct ipv6_mc_socklist *iml,
2598                             struct inet6_dev *idev)
2599 {
2600         struct ip6_sf_socklist *psl;
2601         int err;
2602
2603         psl = sock_dereference(iml->sflist, sk);
2604
2605         if (idev)
2606                 mutex_lock(&idev->mc_lock);
2607
2608         if (!psl) {
2609                 /* any-source empty exclude case */
2610                 err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode, 0, NULL, 0);
2611         } else {
2612                 err = ip6_mc_del_src(idev, &iml->addr, iml->sfmode,
2613                                      psl->sl_count, psl->sl_addr, 0);
2614                 RCU_INIT_POINTER(iml->sflist, NULL);
2615                 atomic_sub(struct_size(psl, sl_addr, psl->sl_max),
2616                            &sk->sk_omem_alloc);
2617                 kfree_rcu(psl, rcu);
2618         }
2619
2620         if (idev)
2621                 mutex_unlock(&idev->mc_lock);
2622
2623         return err;
2624 }
2625
2626 /* called with mc_lock */
2627 static void igmp6_leave_group(struct ifmcaddr6 *ma)
2628 {
2629         if (mld_in_v1_mode(ma->idev)) {
2630                 if (ma->mca_flags & MAF_LAST_REPORTER) {
2631                         igmp6_send(&ma->mca_addr, ma->idev->dev,
2632                                 ICMPV6_MGM_REDUCTION);
2633                 }
2634         } else {
2635                 mld_add_delrec(ma->idev, ma);
2636                 mld_ifc_event(ma->idev);
2637         }
2638 }
2639
2640 static void mld_gq_work(struct work_struct *work)
2641 {
2642         struct inet6_dev *idev = container_of(to_delayed_work(work),
2643                                               struct inet6_dev,
2644                                               mc_gq_work);
2645
2646         mutex_lock(&idev->mc_lock);
2647         mld_send_report(idev, NULL);
2648         idev->mc_gq_running = 0;
2649         mutex_unlock(&idev->mc_lock);
2650
2651         in6_dev_put(idev);
2652 }
2653
2654 static void mld_ifc_work(struct work_struct *work)
2655 {
2656         struct inet6_dev *idev = container_of(to_delayed_work(work),
2657                                               struct inet6_dev,
2658                                               mc_ifc_work);
2659
2660         mutex_lock(&idev->mc_lock);
2661         mld_send_cr(idev);
2662
2663         if (idev->mc_ifc_count) {
2664                 idev->mc_ifc_count--;
2665                 if (idev->mc_ifc_count)
2666                         mld_ifc_start_work(idev,
2667                                            unsolicited_report_interval(idev));
2668         }
2669         mutex_unlock(&idev->mc_lock);
2670         in6_dev_put(idev);
2671 }
2672
2673 /* called with mc_lock */
2674 static void mld_ifc_event(struct inet6_dev *idev)
2675 {
2676         if (mld_in_v1_mode(idev))
2677                 return;
2678
2679         idev->mc_ifc_count = idev->mc_qrv;
2680         mld_ifc_start_work(idev, 1);
2681 }
2682
2683 static void mld_mca_work(struct work_struct *work)
2684 {
2685         struct ifmcaddr6 *ma = container_of(to_delayed_work(work),
2686                                             struct ifmcaddr6, mca_work);
2687
2688         mutex_lock(&ma->idev->mc_lock);
2689         if (mld_in_v1_mode(ma->idev))
2690                 igmp6_send(&ma->mca_addr, ma->idev->dev, ICMPV6_MGM_REPORT);
2691         else
2692                 mld_send_report(ma->idev, ma);
2693         ma->mca_flags |=  MAF_LAST_REPORTER;
2694         ma->mca_flags &= ~MAF_TIMER_RUNNING;
2695         mutex_unlock(&ma->idev->mc_lock);
2696
2697         ma_put(ma);
2698 }
2699
2700 /* Device changing type */
2701
2702 void ipv6_mc_unmap(struct inet6_dev *idev)
2703 {
2704         struct ifmcaddr6 *i;
2705
2706         /* Install multicast list, except for all-nodes (already installed) */
2707
2708         mutex_lock(&idev->mc_lock);
2709         for_each_mc_mclock(idev, i)
2710                 igmp6_group_dropped(i);
2711         mutex_unlock(&idev->mc_lock);
2712 }
2713
2714 void ipv6_mc_remap(struct inet6_dev *idev)
2715 {
2716         ipv6_mc_up(idev);
2717 }
2718
2719 /* Device going down */
2720 void ipv6_mc_down(struct inet6_dev *idev)
2721 {
2722         struct ifmcaddr6 *i;
2723
2724         mutex_lock(&idev->mc_lock);
2725         /* Withdraw multicast list */
2726         for_each_mc_mclock(idev, i)
2727                 igmp6_group_dropped(i);
2728         mutex_unlock(&idev->mc_lock);
2729
2730         /* Should stop work after group drop. or we will
2731          * start work again in mld_ifc_event()
2732          */
2733         synchronize_net();
2734         mld_query_stop_work(idev);
2735         mld_report_stop_work(idev);
2736         mld_ifc_stop_work(idev);
2737         mld_gq_stop_work(idev);
2738         mld_dad_stop_work(idev);
2739 }
2740
2741 static void ipv6_mc_reset(struct inet6_dev *idev)
2742 {
2743         idev->mc_qrv = sysctl_mld_qrv;
2744         idev->mc_qi = MLD_QI_DEFAULT;
2745         idev->mc_qri = MLD_QRI_DEFAULT;
2746         idev->mc_v1_seen = 0;
2747         idev->mc_maxdelay = unsolicited_report_interval(idev);
2748 }
2749
2750 /* Device going up */
2751
2752 void ipv6_mc_up(struct inet6_dev *idev)
2753 {
2754         struct ifmcaddr6 *i;
2755
2756         /* Install multicast list, except for all-nodes (already installed) */
2757
2758         ipv6_mc_reset(idev);
2759         mutex_lock(&idev->mc_lock);
2760         for_each_mc_mclock(idev, i) {
2761                 mld_del_delrec(idev, i);
2762                 igmp6_group_added(i);
2763         }
2764         mutex_unlock(&idev->mc_lock);
2765 }
2766
2767 /* IPv6 device initialization. */
2768
2769 void ipv6_mc_init_dev(struct inet6_dev *idev)
2770 {
2771         idev->mc_gq_running = 0;
2772         INIT_DELAYED_WORK(&idev->mc_gq_work, mld_gq_work);
2773         RCU_INIT_POINTER(idev->mc_tomb, NULL);
2774         idev->mc_ifc_count = 0;
2775         INIT_DELAYED_WORK(&idev->mc_ifc_work, mld_ifc_work);
2776         INIT_DELAYED_WORK(&idev->mc_dad_work, mld_dad_work);
2777         INIT_DELAYED_WORK(&idev->mc_query_work, mld_query_work);
2778         INIT_DELAYED_WORK(&idev->mc_report_work, mld_report_work);
2779         skb_queue_head_init(&idev->mc_query_queue);
2780         skb_queue_head_init(&idev->mc_report_queue);
2781         spin_lock_init(&idev->mc_query_lock);
2782         spin_lock_init(&idev->mc_report_lock);
2783         mutex_init(&idev->mc_lock);
2784         ipv6_mc_reset(idev);
2785 }
2786
2787 /*
2788  *      Device is about to be destroyed: clean up.
2789  */
2790
2791 void ipv6_mc_destroy_dev(struct inet6_dev *idev)
2792 {
2793         struct ifmcaddr6 *i;
2794
2795         /* Deactivate works */
2796         ipv6_mc_down(idev);
2797         mutex_lock(&idev->mc_lock);
2798         mld_clear_delrec(idev);
2799         mutex_unlock(&idev->mc_lock);
2800         mld_clear_query(idev);
2801         mld_clear_report(idev);
2802
2803         /* Delete all-nodes address. */
2804         /* We cannot call ipv6_dev_mc_dec() directly, our caller in
2805          * addrconf.c has NULL'd out dev->ip6_ptr so in6_dev_get() will
2806          * fail.
2807          */
2808         __ipv6_dev_mc_dec(idev, &in6addr_linklocal_allnodes);
2809
2810         if (idev->cnf.forwarding)
2811                 __ipv6_dev_mc_dec(idev, &in6addr_linklocal_allrouters);
2812
2813         mutex_lock(&idev->mc_lock);
2814         while ((i = mc_dereference(idev->mc_list, idev))) {
2815                 rcu_assign_pointer(idev->mc_list, mc_dereference(i->next, idev));
2816
2817                 ip6_mc_clear_src(i);
2818                 ma_put(i);
2819         }
2820         mutex_unlock(&idev->mc_lock);
2821 }
2822
2823 static void ipv6_mc_rejoin_groups(struct inet6_dev *idev)
2824 {
2825         struct ifmcaddr6 *pmc;
2826
2827         ASSERT_RTNL();
2828
2829         mutex_lock(&idev->mc_lock);
2830         if (mld_in_v1_mode(idev)) {
2831                 for_each_mc_mclock(idev, pmc)
2832                         igmp6_join_group(pmc);
2833         } else {
2834                 mld_send_report(idev, NULL);
2835         }
2836         mutex_unlock(&idev->mc_lock);
2837 }
2838
2839 static int ipv6_mc_netdev_event(struct notifier_block *this,
2840                                 unsigned long event,
2841                                 void *ptr)
2842 {
2843         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
2844         struct inet6_dev *idev = __in6_dev_get(dev);
2845
2846         switch (event) {
2847         case NETDEV_RESEND_IGMP:
2848                 if (idev)
2849                         ipv6_mc_rejoin_groups(idev);
2850                 break;
2851         default:
2852                 break;
2853         }
2854
2855         return NOTIFY_DONE;
2856 }
2857
2858 static struct notifier_block igmp6_netdev_notifier = {
2859         .notifier_call = ipv6_mc_netdev_event,
2860 };
2861
2862 #ifdef CONFIG_PROC_FS
2863 struct igmp6_mc_iter_state {
2864         struct seq_net_private p;
2865         struct net_device *dev;
2866         struct inet6_dev *idev;
2867 };
2868
2869 #define igmp6_mc_seq_private(seq)       ((struct igmp6_mc_iter_state *)(seq)->private)
2870
2871 static inline struct ifmcaddr6 *igmp6_mc_get_first(struct seq_file *seq)
2872 {
2873         struct ifmcaddr6 *im = NULL;
2874         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2875         struct net *net = seq_file_net(seq);
2876
2877         state->idev = NULL;
2878         for_each_netdev_rcu(net, state->dev) {
2879                 struct inet6_dev *idev;
2880                 idev = __in6_dev_get(state->dev);
2881                 if (!idev)
2882                         continue;
2883
2884                 im = rcu_dereference(idev->mc_list);
2885                 if (im) {
2886                         state->idev = idev;
2887                         break;
2888                 }
2889         }
2890         return im;
2891 }
2892
2893 static struct ifmcaddr6 *igmp6_mc_get_next(struct seq_file *seq, struct ifmcaddr6 *im)
2894 {
2895         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2896
2897         im = rcu_dereference(im->next);
2898         while (!im) {
2899                 state->dev = next_net_device_rcu(state->dev);
2900                 if (!state->dev) {
2901                         state->idev = NULL;
2902                         break;
2903                 }
2904                 state->idev = __in6_dev_get(state->dev);
2905                 if (!state->idev)
2906                         continue;
2907                 im = rcu_dereference(state->idev->mc_list);
2908         }
2909         return im;
2910 }
2911
2912 static struct ifmcaddr6 *igmp6_mc_get_idx(struct seq_file *seq, loff_t pos)
2913 {
2914         struct ifmcaddr6 *im = igmp6_mc_get_first(seq);
2915         if (im)
2916                 while (pos && (im = igmp6_mc_get_next(seq, im)) != NULL)
2917                         --pos;
2918         return pos ? NULL : im;
2919 }
2920
2921 static void *igmp6_mc_seq_start(struct seq_file *seq, loff_t *pos)
2922         __acquires(RCU)
2923 {
2924         rcu_read_lock();
2925         return igmp6_mc_get_idx(seq, *pos);
2926 }
2927
2928 static void *igmp6_mc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2929 {
2930         struct ifmcaddr6 *im = igmp6_mc_get_next(seq, v);
2931
2932         ++*pos;
2933         return im;
2934 }
2935
2936 static void igmp6_mc_seq_stop(struct seq_file *seq, void *v)
2937         __releases(RCU)
2938 {
2939         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2940
2941         if (likely(state->idev))
2942                 state->idev = NULL;
2943         state->dev = NULL;
2944         rcu_read_unlock();
2945 }
2946
2947 static int igmp6_mc_seq_show(struct seq_file *seq, void *v)
2948 {
2949         struct ifmcaddr6 *im = (struct ifmcaddr6 *)v;
2950         struct igmp6_mc_iter_state *state = igmp6_mc_seq_private(seq);
2951
2952         seq_printf(seq,
2953                    "%-4d %-15s %pi6 %5d %08X %ld\n",
2954                    state->dev->ifindex, state->dev->name,
2955                    &im->mca_addr,
2956                    im->mca_users, im->mca_flags,
2957                    (im->mca_flags & MAF_TIMER_RUNNING) ?
2958                    jiffies_to_clock_t(im->mca_work.timer.expires - jiffies) : 0);
2959         return 0;
2960 }
2961
2962 static const struct seq_operations igmp6_mc_seq_ops = {
2963         .start  =       igmp6_mc_seq_start,
2964         .next   =       igmp6_mc_seq_next,
2965         .stop   =       igmp6_mc_seq_stop,
2966         .show   =       igmp6_mc_seq_show,
2967 };
2968
2969 struct igmp6_mcf_iter_state {
2970         struct seq_net_private p;
2971         struct net_device *dev;
2972         struct inet6_dev *idev;
2973         struct ifmcaddr6 *im;
2974 };
2975
2976 #define igmp6_mcf_seq_private(seq)      ((struct igmp6_mcf_iter_state *)(seq)->private)
2977
2978 static inline struct ip6_sf_list *igmp6_mcf_get_first(struct seq_file *seq)
2979 {
2980         struct ip6_sf_list *psf = NULL;
2981         struct ifmcaddr6 *im = NULL;
2982         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
2983         struct net *net = seq_file_net(seq);
2984
2985         state->idev = NULL;
2986         state->im = NULL;
2987         for_each_netdev_rcu(net, state->dev) {
2988                 struct inet6_dev *idev;
2989                 idev = __in6_dev_get(state->dev);
2990                 if (unlikely(idev == NULL))
2991                         continue;
2992
2993                 im = rcu_dereference(idev->mc_list);
2994                 if (likely(im)) {
2995                         psf = rcu_dereference(im->mca_sources);
2996                         if (likely(psf)) {
2997                                 state->im = im;
2998                                 state->idev = idev;
2999                                 break;
3000                         }
3001                 }
3002         }
3003         return psf;
3004 }
3005
3006 static struct ip6_sf_list *igmp6_mcf_get_next(struct seq_file *seq, struct ip6_sf_list *psf)
3007 {
3008         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3009
3010         psf = rcu_dereference(psf->sf_next);
3011         while (!psf) {
3012                 state->im = rcu_dereference(state->im->next);
3013                 while (!state->im) {
3014                         state->dev = next_net_device_rcu(state->dev);
3015                         if (!state->dev) {
3016                                 state->idev = NULL;
3017                                 goto out;
3018                         }
3019                         state->idev = __in6_dev_get(state->dev);
3020                         if (!state->idev)
3021                                 continue;
3022                         state->im = rcu_dereference(state->idev->mc_list);
3023                 }
3024                 if (!state->im)
3025                         break;
3026                 psf = rcu_dereference(state->im->mca_sources);
3027         }
3028 out:
3029         return psf;
3030 }
3031
3032 static struct ip6_sf_list *igmp6_mcf_get_idx(struct seq_file *seq, loff_t pos)
3033 {
3034         struct ip6_sf_list *psf = igmp6_mcf_get_first(seq);
3035         if (psf)
3036                 while (pos && (psf = igmp6_mcf_get_next(seq, psf)) != NULL)
3037                         --pos;
3038         return pos ? NULL : psf;
3039 }
3040
3041 static void *igmp6_mcf_seq_start(struct seq_file *seq, loff_t *pos)
3042         __acquires(RCU)
3043 {
3044         rcu_read_lock();
3045         return *pos ? igmp6_mcf_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
3046 }
3047
3048 static void *igmp6_mcf_seq_next(struct seq_file *seq, void *v, loff_t *pos)
3049 {
3050         struct ip6_sf_list *psf;
3051         if (v == SEQ_START_TOKEN)
3052                 psf = igmp6_mcf_get_first(seq);
3053         else
3054                 psf = igmp6_mcf_get_next(seq, v);
3055         ++*pos;
3056         return psf;
3057 }
3058
3059 static void igmp6_mcf_seq_stop(struct seq_file *seq, void *v)
3060         __releases(RCU)
3061 {
3062         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3063
3064         if (likely(state->im))
3065                 state->im = NULL;
3066         if (likely(state->idev))
3067                 state->idev = NULL;
3068
3069         state->dev = NULL;
3070         rcu_read_unlock();
3071 }
3072
3073 static int igmp6_mcf_seq_show(struct seq_file *seq, void *v)
3074 {
3075         struct ip6_sf_list *psf = (struct ip6_sf_list *)v;
3076         struct igmp6_mcf_iter_state *state = igmp6_mcf_seq_private(seq);
3077
3078         if (v == SEQ_START_TOKEN) {
3079                 seq_puts(seq, "Idx Device                Multicast Address                   Source Address    INC    EXC\n");
3080         } else {
3081                 seq_printf(seq,
3082                            "%3d %6.6s %pi6 %pi6 %6lu %6lu\n",
3083                            state->dev->ifindex, state->dev->name,
3084                            &state->im->mca_addr,
3085                            &psf->sf_addr,
3086                            psf->sf_count[MCAST_INCLUDE],
3087                            psf->sf_count[MCAST_EXCLUDE]);
3088         }
3089         return 0;
3090 }
3091
3092 static const struct seq_operations igmp6_mcf_seq_ops = {
3093         .start  =       igmp6_mcf_seq_start,
3094         .next   =       igmp6_mcf_seq_next,
3095         .stop   =       igmp6_mcf_seq_stop,
3096         .show   =       igmp6_mcf_seq_show,
3097 };
3098
3099 static int __net_init igmp6_proc_init(struct net *net)
3100 {
3101         int err;
3102
3103         err = -ENOMEM;
3104         if (!proc_create_net("igmp6", 0444, net->proc_net, &igmp6_mc_seq_ops,
3105                         sizeof(struct igmp6_mc_iter_state)))
3106                 goto out;
3107         if (!proc_create_net("mcfilter6", 0444, net->proc_net,
3108                         &igmp6_mcf_seq_ops,
3109                         sizeof(struct igmp6_mcf_iter_state)))
3110                 goto out_proc_net_igmp6;
3111
3112         err = 0;
3113 out:
3114         return err;
3115
3116 out_proc_net_igmp6:
3117         remove_proc_entry("igmp6", net->proc_net);
3118         goto out;
3119 }
3120
3121 static void __net_exit igmp6_proc_exit(struct net *net)
3122 {
3123         remove_proc_entry("mcfilter6", net->proc_net);
3124         remove_proc_entry("igmp6", net->proc_net);
3125 }
3126 #else
3127 static inline int igmp6_proc_init(struct net *net)
3128 {
3129         return 0;
3130 }
3131 static inline void igmp6_proc_exit(struct net *net)
3132 {
3133 }
3134 #endif
3135
3136 static int __net_init igmp6_net_init(struct net *net)
3137 {
3138         int err;
3139
3140         err = inet_ctl_sock_create(&net->ipv6.igmp_sk, PF_INET6,
3141                                    SOCK_RAW, IPPROTO_ICMPV6, net);
3142         if (err < 0) {
3143                 pr_err("Failed to initialize the IGMP6 control socket (err %d)\n",
3144                        err);
3145                 goto out;
3146         }
3147
3148         inet6_sk(net->ipv6.igmp_sk)->hop_limit = 1;
3149         net->ipv6.igmp_sk->sk_allocation = GFP_KERNEL;
3150
3151         err = inet_ctl_sock_create(&net->ipv6.mc_autojoin_sk, PF_INET6,
3152                                    SOCK_RAW, IPPROTO_ICMPV6, net);
3153         if (err < 0) {
3154                 pr_err("Failed to initialize the IGMP6 autojoin socket (err %d)\n",
3155                        err);
3156                 goto out_sock_create;
3157         }
3158
3159         err = igmp6_proc_init(net);
3160         if (err)
3161                 goto out_sock_create_autojoin;
3162
3163         return 0;
3164
3165 out_sock_create_autojoin:
3166         inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3167 out_sock_create:
3168         inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3169 out:
3170         return err;
3171 }
3172
3173 static void __net_exit igmp6_net_exit(struct net *net)
3174 {
3175         inet_ctl_sock_destroy(net->ipv6.igmp_sk);
3176         inet_ctl_sock_destroy(net->ipv6.mc_autojoin_sk);
3177         igmp6_proc_exit(net);
3178 }
3179
3180 static struct pernet_operations igmp6_net_ops = {
3181         .init = igmp6_net_init,
3182         .exit = igmp6_net_exit,
3183 };
3184
3185 int __init igmp6_init(void)
3186 {
3187         int err;
3188
3189         err = register_pernet_subsys(&igmp6_net_ops);
3190         if (err)
3191                 return err;
3192
3193         mld_wq = create_workqueue("mld");
3194         if (!mld_wq) {
3195                 unregister_pernet_subsys(&igmp6_net_ops);
3196                 return -ENOMEM;
3197         }
3198
3199         return err;
3200 }
3201
3202 int __init igmp6_late_init(void)
3203 {
3204         return register_netdevice_notifier(&igmp6_netdev_notifier);
3205 }
3206
3207 void igmp6_cleanup(void)
3208 {
3209         unregister_pernet_subsys(&igmp6_net_ops);
3210         destroy_workqueue(mld_wq);
3211 }
3212
3213 void igmp6_late_cleanup(void)
3214 {
3215         unregister_netdevice_notifier(&igmp6_netdev_notifier);
3216 }