[Bridge] [PATCH net-next mlxsw v2 14/16] selftests: forwarding: lib: Allow list of IPs for IGMPv3/MLDv2

[Petr Machata via Bridge]

The testsuite that checks for mcast_max_groups functionality will need
to generate IGMP and MLD packets with configurable number of (S,G)
addresses. To that end, further extend igmpv3_is_in_get() and
mldv2_is_in_get() to allow a list of IP addresses instead of one
address.

Signed-off-by: Petr Machata <pe...@nvidia.com>
Acked-by: Nikolay Aleksandrov <ra...@blackwall.org>
---
 tools/testing/selftests/net/forwarding/lib.sh | 22 +++++++++++++------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/tools/testing/selftests/net/forwarding/lib.sh 
b/tools/testing/selftests/net/forwarding/lib.sh
index 190e49e60508..12ef34ebcbbf 100755
--- a/tools/testing/selftests/net/forwarding/lib.sh
+++ b/tools/testing/selftests/net/forwarding/lib.sh
@@ -1789,11 +1789,12 @@ payload_template_nbytes()
 igmpv3_is_in_get()
 {
        local GRP=$1; shift
-       local IP=$1; shift
+       local sources=("$@")
 
        local igmpv3
+       local nsources=$(u16_to_bytes ${#sources[@]})
 
-       # IS_IN ( $IP )
+       # IS_IN ( $sources )
        igmpv3=$(:
                )"22:"$(                        : Type - Membership Report
                )"00:"$(                        : Reserved
@@ -1802,9 +1803,12 @@ igmpv3_is_in_get()
                )"00:01:"$(                     : Number of Group Records
                )"01:"$(                        : Record Type - IS_IN
                )"00:"$(                        : Aux Data Len
-               )"00:01:"$(                     : Number of Sources
+               )"${nsources}:"$(               : Number of Sources
                )"$(ipv4_to_bytes $GRP):"$(     : Multicast Address
-               )"$(ipv4_to_bytes $IP)"$(       : Source Address
+               )"$(for src in "${sources[@]}"; do
+                       ipv4_to_bytes $src
+                       echo -n :
+                   done)"$(                    : Source Addresses
                )
        local checksum=$(payload_template_calc_checksum "$igmpv3")
 
@@ -1815,10 +1819,11 @@ mldv2_is_in_get()
 {
        local SIP=$1; shift
        local GRP=$1; shift
-       local IP=$1; shift
+       local sources=("$@")
 
        local hbh
        local icmpv6
+       local nsources=$(u16_to_bytes ${#sources[@]})
 
        hbh=$(:
                )"3a:"$(                        : Next Header - ICMPv6
@@ -1834,9 +1839,12 @@ mldv2_is_in_get()
                )"00:01:"$(                     : Number of Group Records
                )"01:"$(                        : Record Type - IS_IN
                )"00:"$(                        : Aux Data Len
-               )"00:01:"$(                     : Number of Sources
+               )"${nsources}:"$(               : Number of Sources
                )"$(ipv6_to_bytes $GRP):"$(     : Multicast address
-               )"$(ipv6_to_bytes $IP):"$(      : Source Address
+               )"$(for src in "${sources[@]}"; do
+                       ipv6_to_bytes $src
+                       echo -n :
+                   done)"$(                    : Source Addresses
                )
 
        local len=$(u16_to_bytes $(payload_template_nbytes $icmpv6))
-- 
2.39.0