test/test_l2_flood.py

   1 #!/usr/bin/env python
   2
   3 import unittest
   4 import socket
   5
   6 from framework import VppTestCase, VppTestRunner
   7 from vpp_ip_route import VppIpRoute, VppRoutePath
   8 from vpp_papi_provider import L2_PORT_TYPE, BRIDGE_FLAGS
   9
  10 from scapy.packet import Raw
  11 from scapy.layers.l2 import Ether
  12 from scapy.layers.inet import IP, UDP
  13
  14
  15 class TestL2Flood(VppTestCase):
  16     """ L2-flood """
  17
  18     def setUp(self):
  19         super(TestL2Flood, self).setUp()
  20
  21         # 12 l2 interface and one l3
  22         self.create_pg_interfaces(range(13))
  23         self.create_loopback_interfaces(1)
  24
  25         for i in self.pg_interfaces:
  26             i.admin_up()
  27         for i in self.lo_interfaces:
  28             i.admin_up()
  29
  30         self.pg12.config_ip4()
  31         self.pg12.resolve_arp()
  32         self.loop0.config_ip4()
  33
  34     def tearDown(self):
  35         self.pg12.unconfig_ip4()
  36         self.loop0.unconfig_ip4()
  37
  38         for i in self.pg_interfaces:
  39             i.admin_down()
  40         for i in self.lo_interfaces:
  41             i.admin_down()
  42         super(TestL2Flood, self).tearDown()
  43
  44     def test_flood(self):
  45         """ L2 Flood Tests """
  46
  47         #
  48         # Create a single bridge Domain
  49         #
  50         self.vapi.bridge_domain_add_del(1)
  51
  52         #
  53         # add each interface to the BD. 3 interfaces per split horizon group
  54         #
  55         for i in self.pg_interfaces[0:4]:
  56             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, 0)
  57         for i in self.pg_interfaces[4:8]:
  58             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, 1)
  59         for i in self.pg_interfaces[8:12]:
  60             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, 2)
  61         for i in self.lo_interfaces:
  62             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, 2,
  63                                                  port_type=L2_PORT_TYPE.BVI)
  64
  65         p = (Ether(dst="ff:ff:ff:ff:ff:ff",
  66                    src="00:00:de:ad:be:ef") /
  67              IP(src="10.10.10.10", dst="1.1.1.1") /
  68              UDP(sport=1234, dport=1234) /
  69              Raw('\xa5' * 100))
  70
  71         #
  72         # input on pg0 expect copies on pg1->11
  73         # this is in SHG=0 so its flooded to all, expect the pg0 since that's
  74         # the ingress link
  75         #
  76         self.pg0.add_stream(p*65)
  77         self.pg_enable_capture(self.pg_interfaces)
  78         self.pg_start()
  79
  80         for i in self.pg_interfaces[1:12]:
  81             rx0 = i.get_capture(65, timeout=1)
  82
  83         #
  84         # input on pg4 (SHG=1) expect copies on pg0->3 (SHG=0)
  85         # and pg8->11 (SHG=2)
  86         #
  87         self.pg4.add_stream(p*65)
  88         self.pg_enable_capture(self.pg_interfaces)
  89         self.pg_start()
  90
  91         for i in self.pg_interfaces[:4]:
  92             rx0 = i.get_capture(65, timeout=1)
  93         for i in self.pg_interfaces[8:12]:
  94             rx0 = i.get_capture(65, timeout=1)
  95         for i in self.pg_interfaces[4:8]:
  96             i.assert_nothing_captured(remark="Different SH group")
  97
  98         #
  99         # An IP route so the packet that hits the BVI is sent out of pg12
 100         #
 101         ip_route = VppIpRoute(self, "1.1.1.1", 32,
 102                               [VppRoutePath(self.pg12.remote_ip4,
 103                                             self.pg12.sw_if_index)])
 104         ip_route.add_vpp_config()
 105
 106         self.logger.info(self.vapi.cli("sh bridge 1 detail"))
 107
 108         #
 109         # input on pg0 expect copies on pg1->12
 110         # this is in SHG=0 so its flooded to all, expect the pg0 since that's
 111         # the ingress link
 112         #
 113         self.pg0.add_stream(p*65)
 114         self.pg_enable_capture(self.pg_interfaces)
 115         self.pg_start()
 116
 117         for i in self.pg_interfaces[1:]:
 118             rx0 = i.get_capture(65, timeout=1)
 119
 120         #
 121         # input on pg4 (SHG=1) expect copies on pg0->3 (SHG=0)
 122         # and pg8->12 (SHG=2)
 123         #
 124         self.pg4.add_stream(p*65)
 125         self.pg_enable_capture(self.pg_interfaces)
 126         self.pg_start()
 127
 128         for i in self.pg_interfaces[:4]:
 129             rx0 = i.get_capture(65, timeout=1)
 130         for i in self.pg_interfaces[8:13]:
 131             rx0 = i.get_capture(65, timeout=1)
 132         for i in self.pg_interfaces[4:8]:
 133             i.assert_nothing_captured(remark="Different SH group")
 134
 135         #
 136         # cleanup
 137         #
 138         for i in self.pg_interfaces[:12]:
 139             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, enable=0)
 140         for i in self.lo_interfaces:
 141             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, 2,
 142                                                  port_type=L2_PORT_TYPE.BVI,
 143                                                  enable=0)
 144
 145         self.vapi.bridge_domain_add_del(1, is_add=0)
 146
 147     def test_uu_fwd(self):
 148         """ UU Flood """
 149
 150         #
 151         # Create a single bridge Domain
 152         #
 153         self.vapi.bridge_domain_add_del(1, uu_flood=1)
 154
 155         #
 156         # add each interface to the BD. 3 interfaces per split horizon group
 157         #
 158         for i in self.pg_interfaces[0:4]:
 159             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, 0)
 160
 161         #
 162         # an unknown unicast packet
 163         #
 164         p_uu = (Ether(dst="00:00:00:c1:5c:00",
 165                       src="00:00:de:ad:be:ef") /
 166                 IP(src="10.10.10.10", dst="1.1.1.1") /
 167                 UDP(sport=1234, dport=1234) /
 168                 Raw('\xa5' * 100))
 169
 170         #
 171         # input on pg0, expected copies on pg1->4
 172         #
 173         self.pg0.add_stream(p_uu*65)
 174         self.pg_enable_capture(self.pg_interfaces)
 175         self.pg_start()
 176
 177         for i in self.pg_interfaces[1:4]:
 178             rx0 = i.get_capture(65, timeout=1)
 179
 180         #
 181         # use pg8 as the uu-fwd interface
 182         #
 183         self.vapi.sw_interface_set_l2_bridge(self.pg8.sw_if_index, 1, 0,
 184                                              port_type=L2_PORT_TYPE.UU_FWD)
 185
 186         #
 187         # expect the UU packet on the uu-fwd interface and not be flooded
 188         #
 189         self.pg0.add_stream(p_uu*65)
 190         self.pg_enable_capture(self.pg_interfaces)
 191         self.pg_start()
 192
 193         rx0 = self.pg8.get_capture(65, timeout=1)
 194
 195         for i in self.pg_interfaces[0:4]:
 196             i.assert_nothing_captured(remark="UU not flooded")
 197
 198         #
 199         # remove the uu-fwd interface and expect UU to be flooded again
 200         #
 201         self.vapi.sw_interface_set_l2_bridge(self.pg8.sw_if_index, 1, 0,
 202                                              port_type=L2_PORT_TYPE.UU_FWD,
 203                                              enable=0)
 204
 205         self.pg0.add_stream(p_uu*65)
 206         self.pg_enable_capture(self.pg_interfaces)
 207         self.pg_start()
 208
 209         for i in self.pg_interfaces[1:4]:
 210             rx0 = i.get_capture(65, timeout=1)
 211
 212         #
 213         # change the BD config to not support UU-flood
 214         #
 215         self.vapi.bridge_flags(1, 0, BRIDGE_FLAGS.UU_FLOOD)
 216
 217         self.send_and_assert_no_replies(self.pg0, p_uu)
 218
 219         #
 220         # re-add the uu-fwd interface
 221         #
 222         self.vapi.sw_interface_set_l2_bridge(self.pg8.sw_if_index, 1, 0,
 223                                              port_type=L2_PORT_TYPE.UU_FWD)
 224         self.logger.info(self.vapi.cli("sh bridge 1 detail"))
 225
 226         self.pg0.add_stream(p_uu*65)
 227         self.pg_enable_capture(self.pg_interfaces)
 228         self.pg_start()
 229
 230         rx0 = self.pg8.get_capture(65, timeout=1)
 231
 232         for i in self.pg_interfaces[0:4]:
 233             i.assert_nothing_captured(remark="UU not flooded")
 234
 235         #
 236         # remove the uu-fwd interface
 237         #
 238         self.vapi.sw_interface_set_l2_bridge(self.pg8.sw_if_index, 1, 0,
 239                                              port_type=L2_PORT_TYPE.UU_FWD,
 240                                              enable=0)
 241         self.send_and_assert_no_replies(self.pg0, p_uu)
 242
 243         #
 244         # cleanup
 245         #
 246         for i in self.pg_interfaces[:4]:
 247             self.vapi.sw_interface_set_l2_bridge(i.sw_if_index, 1, enable=0)
 248
 249         self.vapi.bridge_domain_add_del(1, is_add=0)
 250
 251
 252 if __name__ == '__main__':
 253     unittest.main(testRunner=VppTestRunner)