[Bridge] [PATCH net-next v2 09/16] net: bridge: mcast: when igmpv3/mldv2 are enabled lookup (S, G) first, then (*, G)

[Nikolay Aleksandrov]

From: Nikolay Aleksandrov <niko...@nvidia.com>

If (S,G) entries are enabled (igmpv3/mldv2) then look them up first. If
there isn't a present (S,G) entry then try to find (*,G).

Signed-off-by: Nikolay Aleksandrov <niko...@nvidia.com>
---
 net/bridge/br_multicast.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/net/bridge/br_multicast.c b/net/bridge/br_multicast.c
index e1fb822b9ddb..4fd690bc848f 100644
--- a/net/bridge/br_multicast.c
+++ b/net/bridge/br_multicast.c
@@ -127,10 +127,28 @@ struct net_bridge_mdb_entry *br_mdb_get(struct net_bridge 
*br,
        switch (skb->protocol) {
        case htons(ETH_P_IP):
                ip.dst.ip4 = ip_hdr(skb)->daddr;
+               if (br->multicast_igmp_version == 3) {
+                       struct net_bridge_mdb_entry *mdb;
+
+                       ip.src.ip4 = ip_hdr(skb)->saddr;
+                       mdb = br_mdb_ip_get_rcu(br, &ip);
+                       if (mdb)
+                               return mdb;
+                       ip.src.ip4 = 0;
+               }
                break;
 #if IS_ENABLED(CONFIG_IPV6)
        case htons(ETH_P_IPV6):
                ip.dst.ip6 = ipv6_hdr(skb)->daddr;
+               if (br->multicast_mld_version == 2) {
+                       struct net_bridge_mdb_entry *mdb;
+
+                       ip.src.ip6 = ipv6_hdr(skb)->saddr;
+                       mdb = br_mdb_ip_get_rcu(br, &ip);
+                       if (mdb)
+                               return mdb;
+                       memset(&ip.src.ip6, 0, sizeof(ip.src.ip6));
+               }
                break;
 #endif
        default:
-- 
2.25.4