target/linux/generic/patches-4.3/645-bridge_multicast_to_unicast.patch

   1 From: Felix Fietkau <nbd@openwrt.org>
   2 Subject: [PATCH] bridge: multicast to unicast
   3
   4 Implement optinal multicast->unicast conversion for igmp snooping
   5 ---
   6 --- a/include/linux/if_bridge.h
   7 +++ b/include/linux/if_bridge.h
   8 @@ -46,6 +46,7 @@ struct br_ip_list {
   9  #define BR_LEARNING_SYNC       BIT(9)
  10  #define BR_PROXYARP_WIFI       BIT(10)
  11  #define BR_ISOLATE_MODE        BIT(11)
  12 +#define BR_MULTICAST_TO_UCAST  BIT(12)
  13
  14  extern void brioctl_set(int (*ioctl_hook)(struct net *, unsigned int, void __user *));
  15
  16 --- a/net/bridge/br_multicast.c
  17 +++ b/net/bridge/br_multicast.c
  18 @@ -42,12 +42,13 @@ static void br_multicast_add_router(stru
  19  static void br_ip4_multicast_leave_group(struct net_bridge *br,
  20                                          struct net_bridge_port *port,
  21                                          __be32 group,
  22 -                                        __u16 vid);
  23 +                                        __u16 vid,
  24 +                                        const unsigned char *src);
  25  #if IS_ENABLED(CONFIG_IPV6)
  26  static void br_ip6_multicast_leave_group(struct net_bridge *br,
  27                                          struct net_bridge_port *port,
  28                                          const struct in6_addr *group,
  29 -                                        __u16 vid);
  30 +                                        __u16 vid, const unsigned char *src);
  31  #endif
  32  unsigned int br_mdb_rehash_seq;
  33
  34 @@ -649,7 +650,8 @@ struct net_bridge_port_group *br_multica
  35                         struct net_bridge_port *port,
  36                         struct br_ip *group,
  37                         struct net_bridge_port_group __rcu *next,
  38 -                       unsigned char state)
  39 +                       unsigned char state,
  40 +                       const unsigned char *src)
  41  {
  42         struct net_bridge_port_group *p;
  43
  44 @@ -664,12 +666,33 @@ struct net_bridge_port_group *br_multica
  45         hlist_add_head(&p->mglist, &port->mglist);
  46         setup_timer(&p->timer, br_multicast_port_group_expired,
  47                     (unsigned long)p);
  48 +       if ((port->flags & BR_MULTICAST_TO_UCAST) && src) {
  49 +               memcpy(p->eth_addr, src, ETH_ALEN);
  50 +               p->unicast = true;
  51 +       }
  52         return p;
  53  }
  54
  55 +static bool br_port_group_equal(struct net_bridge_port_group *p,
  56 +                               struct net_bridge_port *port,
  57 +                               const unsigned char *src)
  58 +{
  59 +       if (p->port != port)
  60 +               return false;
  61 +
  62 +       if (!p->unicast)
  63 +               return true;
  64 +
  65 +       if (!src)
  66 +               return false;
  67 +
  68 +       return ether_addr_equal(src, p->eth_addr);
  69 +}
  70 +
  71  static int br_multicast_add_group(struct net_bridge *br,
  72                                   struct net_bridge_port *port,
  73 -                                 struct br_ip *group)
  74 +                                 struct br_ip *group,
  75 +                                 const unsigned char *src)
  76  {
  77         struct net_bridge_mdb_entry *mp;
  78         struct net_bridge_port_group *p;
  79 @@ -696,13 +719,13 @@ static int br_multicast_add_group(struct
  80         for (pp = &mp->ports;
  81              (p = mlock_dereference(*pp, br)) != NULL;
  82              pp = &p->next) {
  83 -               if (p->port == port)
  84 +               if (br_port_group_equal(p, port, src))
  85                         goto found;
  86                 if ((unsigned long)p->port < (unsigned long)port)
  87                         break;
  88         }
  89
  90 -       p = br_multicast_new_port_group(port, group, *pp, MDB_TEMPORARY);
  91 +       p = br_multicast_new_port_group(port, group, *pp, MDB_TEMPORARY, src);
  92         if (unlikely(!p))
  93                 goto err;
  94         rcu_assign_pointer(*pp, p);
  95 @@ -721,7 +744,7 @@ err:
  96  static int br_ip4_multicast_add_group(struct net_bridge *br,
  97                                       struct net_bridge_port *port,
  98                                       __be32 group,
  99 -                                     __u16 vid)
 100 +                                     __u16 vid, const unsigned char *src)
 101  {
 102         struct br_ip br_group;
 103
 104 @@ -732,14 +755,14 @@ static int br_ip4_multicast_add_group(st
 105         br_group.proto = htons(ETH_P_IP);
 106         br_group.vid = vid;
 107
 108 -       return br_multicast_add_group(br, port, &br_group);
 109 +       return br_multicast_add_group(br, port, &br_group, src);
 110  }
 111
 112  #if IS_ENABLED(CONFIG_IPV6)
 113  static int br_ip6_multicast_add_group(struct net_bridge *br,
 114                                       struct net_bridge_port *port,
 115                                       const struct in6_addr *group,
 116 -                                     __u16 vid)
 117 +                                     __u16 vid, const unsigned char *src)
 118  {
 119         struct br_ip br_group;
 120
 121 @@ -750,7 +773,7 @@ static int br_ip6_multicast_add_group(st
 122         br_group.proto = htons(ETH_P_IPV6);
 123         br_group.vid = vid;
 124
 125 -       return br_multicast_add_group(br, port, &br_group);
 126 +       return br_multicast_add_group(br, port, &br_group, src);
 127  }
 128  #endif
 129
 130 @@ -995,6 +1018,7 @@ static int br_ip4_multicast_igmp3_report
 131                                          struct sk_buff *skb,
 132                                          u16 vid)
 133  {
 134 +       const unsigned char *src;
 135         struct igmpv3_report *ih;
 136         struct igmpv3_grec *grec;
 137         int i;
 138 @@ -1038,9 +1062,10 @@ static int br_ip4_multicast_igmp3_report
 139                 if ((type == IGMPV3_CHANGE_TO_INCLUDE ||
 140                      type == IGMPV3_MODE_IS_INCLUDE) &&
 141                     ntohs(grec->grec_nsrcs) == 0) {
 142 -                       br_ip4_multicast_leave_group(br, port, group, vid);
 143 +                       br_ip4_multicast_leave_group(br, port, group, vid, src);
 144                 } else {
 145 -                       err = br_ip4_multicast_add_group(br, port, group, vid);
 146 +                       src = eth_hdr(skb)->h_source;
 147 +                       err = br_ip4_multicast_add_group(br, port, group, vid, src);
 148                         if (err)
 149                                 break;
 150                 }
 151 @@ -1055,6 +1080,7 @@ static int br_ip6_multicast_mld2_report(
 152                                         struct sk_buff *skb,
 153                                         u16 vid)
 154  {
 155 +       const unsigned char *src = eth_hdr(skb)->h_source;
 156         struct icmp6hdr *icmp6h;
 157         struct mld2_grec *grec;
 158         int i;
 159 @@ -1106,10 +1132,10 @@ static int br_ip6_multicast_mld2_report(
 160                      grec->grec_type == MLD2_MODE_IS_INCLUDE) &&
 161                     ntohs(*nsrcs) == 0) {
 162                         br_ip6_multicast_leave_group(br, port, &grec->grec_mca,
 163 -                                                    vid);
 164 +                                                    vid, src);
 165                 } else {
 166                         err = br_ip6_multicast_add_group(br, port,
 167 -                                                        &grec->grec_mca, vid);
 168 +                                                        &grec->grec_mca, vid, src);
 169                         if (!err)
 170                                 break;
 171                 }
 172 @@ -1422,7 +1448,8 @@ br_multicast_leave_group(struct net_brid
 173                          struct net_bridge_port *port,
 174                          struct br_ip *group,
 175                          struct bridge_mcast_other_query *other_query,
 176 -                        struct bridge_mcast_own_query *own_query)
 177 +                        struct bridge_mcast_own_query *own_query,
 178 +                        const unsigned char *src)
 179  {
 180         struct net_bridge_mdb_htable *mdb;
 181         struct net_bridge_mdb_entry *mp;
 182 @@ -1446,7 +1473,7 @@ br_multicast_leave_group(struct net_brid
 183                 for (pp = &mp->ports;
 184                      (p = mlock_dereference(*pp, br)) != NULL;
 185                      pp = &p->next) {
 186 -                       if (p->port != port)
 187 +                       if (!br_port_group_equal(p, port, src))
 188                                 continue;
 189
 190                         rcu_assign_pointer(*pp, p->next);
 191 @@ -1509,7 +1536,7 @@ br_multicast_leave_group(struct net_brid
 192         for (p = mlock_dereference(mp->ports, br);
 193              p != NULL;
 194              p = mlock_dereference(p->next, br)) {
 195 -               if (p->port != port)
 196 +               if (!br_port_group_equal(p, port, src))
 197                         continue;
 198
 199                 if (!hlist_unhashed(&p->mglist) &&
 200 @@ -1527,8 +1554,8 @@ out:
 201
 202  static void br_ip4_multicast_leave_group(struct net_bridge *br,
 203                                          struct net_bridge_port *port,
 204 -                                        __be32 group,
 205 -                                        __u16 vid)
 206 +                                        __be32 group, __u16 vid,
 207 +                                        const unsigned char *src)
 208  {
 209         struct br_ip br_group;
 210         struct bridge_mcast_own_query *own_query;
 211 @@ -1543,14 +1570,14 @@ static void br_ip4_multicast_leave_group
 212         br_group.vid = vid;
 213
 214         br_multicast_leave_group(br, port, &br_group, &br->ip4_other_query,
 215 -                                own_query);
 216 +                                own_query, src);
 217  }
 218
 219  #if IS_ENABLED(CONFIG_IPV6)
 220  static void br_ip6_multicast_leave_group(struct net_bridge *br,
 221                                          struct net_bridge_port *port,
 222                                          const struct in6_addr *group,
 223 -                                        __u16 vid)
 224 +                                        __u16 vid, const unsigned char *src)
 225  {
 226         struct br_ip br_group;
 227         struct bridge_mcast_own_query *own_query;
 228 @@ -1565,7 +1592,7 @@ static void br_ip6_multicast_leave_group
 229         br_group.vid = vid;
 230
 231         br_multicast_leave_group(br, port, &br_group, &br->ip6_other_query,
 232 -                                own_query);
 233 +                                own_query, src);
 234  }
 235  #endif
 236
 237 @@ -1574,6 +1601,7 @@ static int br_multicast_ipv4_rcv(struct
 238                                  struct sk_buff *skb,
 239                                  u16 vid)
 240  {
 241 +       const unsigned char *src;
 242         struct sk_buff *skb_trimmed = NULL;
 243         struct igmphdr *ih;
 244         int err;
 245 @@ -1590,12 +1618,13 @@ static int br_multicast_ipv4_rcv(struct
 246
 247         BR_INPUT_SKB_CB(skb)->igmp = 1;
 248         ih = igmp_hdr(skb);
 249 +       src = eth_hdr(skb)->h_source;
 250
 251         switch (ih->type) {
 252         case IGMP_HOST_MEMBERSHIP_REPORT:
 253         case IGMPV2_HOST_MEMBERSHIP_REPORT:
 254                 BR_INPUT_SKB_CB(skb)->mrouters_only = 1;
 255 -               err = br_ip4_multicast_add_group(br, port, ih->group, vid);
 256 +               err = br_ip4_multicast_add_group(br, port, ih->group, vid, src);
 257                 break;
 258         case IGMPV3_HOST_MEMBERSHIP_REPORT:
 259                 err = br_ip4_multicast_igmp3_report(br, port, skb_trimmed, vid);
 260 @@ -1604,7 +1633,7 @@ static int br_multicast_ipv4_rcv(struct
 261                 err = br_ip4_multicast_query(br, port, skb_trimmed, vid);
 262                 break;
 263         case IGMP_HOST_LEAVE_MESSAGE:
 264 -               br_ip4_multicast_leave_group(br, port, ih->group, vid);
 265 +               br_ip4_multicast_leave_group(br, port, ih->group, vid, src);
 266                 break;
 267         }
 268
 269 @@ -1620,6 +1649,7 @@ static int br_multicast_ipv6_rcv(struct
 270                                  struct sk_buff *skb,
 271                                  u16 vid)
 272  {
 273 +       const unsigned char *src;
 274         struct sk_buff *skb_trimmed = NULL;
 275         struct mld_msg *mld;
 276         int err;
 277 @@ -1639,8 +1669,9 @@ static int br_multicast_ipv6_rcv(struct
 278
 279         switch (mld->mld_type) {
 280         case ICMPV6_MGM_REPORT:
 281 +               src = eth_hdr(skb)->h_source;
 282                 BR_INPUT_SKB_CB(skb)->mrouters_only = 1;
 283 -               err = br_ip6_multicast_add_group(br, port, &mld->mld_mca, vid);
 284 +               err = br_ip6_multicast_add_group(br, port, &mld->mld_mca, vid, src);
 285                 break;
 286         case ICMPV6_MLD2_REPORT:
 287                 err = br_ip6_multicast_mld2_report(br, port, skb_trimmed, vid);
 288 @@ -1649,7 +1680,8 @@ static int br_multicast_ipv6_rcv(struct
 289                 err = br_ip6_multicast_query(br, port, skb_trimmed, vid);
 290                 break;
 291         case ICMPV6_MGM_REDUCTION:
 292 -               br_ip6_multicast_leave_group(br, port, &mld->mld_mca, vid);
 293 +               src = eth_hdr(skb)->h_source;
 294 +               br_ip6_multicast_leave_group(br, port, &mld->mld_mca, vid, src);
 295                 break;
 296         }
 297
 298 --- a/net/bridge/br_private.h
 299 +++ b/net/bridge/br_private.h
 300 @@ -114,6 +114,9 @@ struct net_bridge_port_group {
 301         struct timer_list               timer;
 302         struct br_ip                    addr;
 303         unsigned char                   state;
 304 +
 305 +       unsigned char                   eth_addr[ETH_ALEN];
 306 +       bool                            unicast;
 307  };
 308
 309  struct net_bridge_mdb_entry
 310 @@ -485,7 +488,8 @@ void br_multicast_free_pg(struct rcu_hea
 311  struct net_bridge_port_group *
 312  br_multicast_new_port_group(struct net_bridge_port *port, struct br_ip *group,
 313                             struct net_bridge_port_group __rcu *next,
 314 -                           unsigned char state);
 315 +                           unsigned char state,
 316 +                      const unsigned char *src);
 317  void br_mdb_init(void);
 318  void br_mdb_uninit(void);
 319  void br_mdb_notify(struct net_device *dev, struct net_bridge_port *port,
 320 --- a/net/bridge/br_mdb.c
 321 +++ b/net/bridge/br_mdb.c
 322 @@ -416,7 +416,7 @@ static int br_mdb_add_group(struct net_b
 323                         break;
 324         }
 325
 326 -       p = br_multicast_new_port_group(port, group, *pp, state);
 327 +       p = br_multicast_new_port_group(port, group, *pp, state, NULL);
 328         if (unlikely(!p))
 329                 return -ENOMEM;
 330         rcu_assign_pointer(*pp, p);
 331 --- a/net/bridge/br_forward.c
 332 +++ b/net/bridge/br_forward.c
 333 @@ -185,6 +185,34 @@ out:
 334         return p;
 335  }
 336
 337 +static struct net_bridge_port *maybe_deliver_addr(
 338 +       struct net_bridge_port *prev, struct net_bridge_port *p,
 339 +       struct sk_buff *skb, const unsigned char *addr,
 340 +       void (*__packet_hook)(const struct net_bridge_port *p,
 341 +                             struct sk_buff *skb))
 342 +{
 343 +       struct net_device *dev = BR_INPUT_SKB_CB(skb)->brdev;
 344 +       const unsigned char *src = eth_hdr(skb)->h_source;
 345 +
 346 +       if (!should_deliver(p, skb))
 347 +               return prev;
 348 +
 349 +       /* Even with hairpin, no soliloquies - prevent breaking IPv6 DAD */
 350 +       if (skb->dev == p->dev && ether_addr_equal(src, addr))
 351 +               return prev;
 352 +
 353 +       skb = skb_copy(skb, GFP_ATOMIC);
 354 +       if (!skb) {
 355 +               dev->stats.tx_dropped++;
 356 +               return prev;
 357 +       }
 358 +
 359 +       memcpy(eth_hdr(skb)->h_dest, addr, ETH_ALEN);
 360 +       __packet_hook(p, skb);
 361 +
 362 +       return prev;
 363 +}
 364 +
 365  /* called under bridge lock */
 366  static void br_flood(struct net_bridge *br, struct sk_buff *skb,
 367                      struct sk_buff *skb0,
 368 @@ -257,6 +285,7 @@ static void br_multicast_flood(struct ne
 369         struct net_bridge_port *prev = NULL;
 370         struct net_bridge_port_group *p;
 371         struct hlist_node *rp;
 372 +       const unsigned char *addr;
 373
 374         rp = rcu_dereference(hlist_first_rcu(&br->router_list));
 375         p = mdst ? rcu_dereference(mdst->ports) : NULL;
 376 @@ -267,10 +296,19 @@ static void br_multicast_flood(struct ne
 377                 rport = rp ? hlist_entry(rp, struct net_bridge_port, rlist) :
 378                              NULL;
 379
 380 -               port = (unsigned long)lport > (unsigned long)rport ?
 381 -                      lport : rport;
 382 -
 383 -               prev = maybe_deliver(prev, port, skb, __packet_hook);
 384 +               if ((unsigned long)lport > (unsigned long)rport) {
 385 +                       port = lport;
 386 +                       addr = p->unicast ? p->eth_addr : NULL;
 387 +               } else {
 388 +                       port = rport;
 389 +                       addr = NULL;
 390 +               }
 391 +
 392 +               if (addr)
 393 +                       prev = maybe_deliver_addr(prev, port, skb, addr,
 394 +                                                 __packet_hook);
 395 +               else
 396 +                       prev = maybe_deliver(prev, port, skb, __packet_hook);
 397                 if (IS_ERR(prev))
 398                         goto out;
 399
 400 --- a/net/bridge/br_sysfs_if.c
 401 +++ b/net/bridge/br_sysfs_if.c
 402 @@ -204,6 +204,7 @@ static BRPORT_ATTR(multicast_router, S_I
 403                    store_multicast_router);
 404
 405  BRPORT_ATTR_FLAG(multicast_fast_leave, BR_MULTICAST_FAST_LEAVE);
 406 +BRPORT_ATTR_FLAG(multicast_to_unicast, BR_MULTICAST_TO_UCAST);
 407  #endif
 408
 409  static const struct brport_attribute *brport_attrs[] = {
 410 @@ -230,6 +231,7 @@ static const struct brport_attribute *br
 411  #ifdef CONFIG_BRIDGE_IGMP_SNOOPING
 412         &brport_attr_multicast_router,
 413         &brport_attr_multicast_fast_leave,
 414 +       &brport_attr_multicast_to_unicast,
 415  #endif
 416         &brport_attr_proxyarp,
 417         &brport_attr_proxyarp_wifi,