test/test_bfd.py

   1 #!/usr/bin/env python
   2
   3 import unittest
   4 import time
   5 from random import randint
   6 from bfd import *
   7 from framework import *
   8 from util import ppp
   9
  10
  11 class BFDCLITestCase(VppTestCase):
  12     """Bidirectional Forwarding Detection (BFD) - CLI"""
  13
  14     @classmethod
  15     def setUpClass(cls):
  16         super(BFDCLITestCase, cls).setUpClass()
  17
  18         try:
  19             cls.create_pg_interfaces([0])
  20             cls.pg0.config_ip4()
  21             cls.pg0.resolve_arp()
  22
  23         except Exception:
  24             super(BFDCLITestCase, cls).tearDownClass()
  25             raise
  26
  27     def test_add_bfd(self):
  28         """ create a BFD session """
  29         session = VppBFDUDPSession(self, self.pg0, self.pg0.remote_ip4)
  30         session.add_vpp_config()
  31         self.logger.debug("Session state is %s" % str(session.state))
  32         session.remove_vpp_config()
  33         session = VppBFDUDPSession(self, self.pg0, self.pg0.remote_ip4)
  34         session.add_vpp_config()
  35         self.logger.debug("Session state is %s" % str(session.state))
  36         session.remove_vpp_config()
  37
  38     def test_double_add(self):
  39         """ create the same BFD session twice (negative case) """
  40         session = VppBFDUDPSession(self, self.pg0, self.pg0.remote_ip4)
  41         session.add_vpp_config()
  42         try:
  43             session.add_vpp_config()
  44         except:
  45             session.remove_vpp_config()
  46             return
  47         session.remove_vpp_config()
  48         raise Exception("Expected failure while adding duplicate "
  49                         "configuration")
  50
  51
  52 def create_packet(interface, ttl=255, src_port=50000, **kwargs):
  53     p = (Ether(src=interface.remote_mac, dst=interface.local_mac) /
  54          IP(src=interface.remote_ip4, dst=interface.local_ip4, ttl=ttl) /
  55          UDP(sport=src_port, dport=BFD.udp_dport) /
  56          BFD(*kwargs))
  57     return p
  58
  59
  60 def verify_ip(test, packet, local_ip, remote_ip):
  61     """ Verify correctness of IP layer. """
  62     ip = packet[IP]
  63     test.assert_equal(ip.src, local_ip, "IP source address")
  64     test.assert_equal(ip.dst, remote_ip, "IP destination address")
  65     test.assert_equal(ip.ttl, 255, "IP TTL")
  66
  67
  68 def verify_udp(test, packet):
  69     """ Verify correctness of UDP layer. """
  70     udp = packet[UDP]
  71     test.assert_equal(udp.dport, BFD.udp_dport, "UDP destination port")
  72     test.assert_in_range(udp.sport, BFD.udp_sport_min, BFD.udp_sport_max,
  73                          "UDP source port")
  74
  75
  76 class BFDTestSession(object):
  77
  78     def __init__(self, test, interface, detect_mult=3):
  79         self.test = test
  80         self.interface = interface
  81         self.bfd_values = {
  82             'my_discriminator': 0,
  83             'desired_min_tx_interval': 500000,
  84             'detect_mult': detect_mult,
  85             'diag': BFDDiagCode.no_diagnostic,
  86         }
  87
  88     def update(self, **kwargs):
  89         self.bfd_values.update(kwargs)
  90
  91     def create_packet(self):
  92         packet = create_packet(self.interface)
  93         for name, value in self.bfd_values.iteritems():
  94             packet[BFD].setfieldval(name, value)
  95         return packet
  96
  97     def send_packet(self):
  98         p = self.create_packet()
  99         self.test.logger.debug(ppp("Sending packet:", p))
 100         self.test.pg0.add_stream([p])
 101         self.test.pg_start()
 102
 103     def verify_packet(self, packet):
 104         """ Verify correctness of BFD layer. """
 105         bfd = packet[BFD]
 106         self.test.assert_equal(bfd.version, 1, "BFD version")
 107         self.test.assert_equal(bfd.your_discriminator,
 108                                self.bfd_values['my_discriminator'],
 109                                "BFD - your discriminator")
 110
 111
 112 class BFDTestCase(VppTestCase):
 113     """Bidirectional Forwarding Detection (BFD)"""
 114
 115     @classmethod
 116     def setUpClass(cls):
 117         super(BFDTestCase, cls).setUpClass()
 118         try:
 119             cls.create_pg_interfaces([0, 1])
 120             cls.pg0.config_ip4()
 121             cls.pg0.generate_remote_hosts()
 122             cls.pg0.configure_ipv4_neighbors()
 123             cls.pg0.admin_up()
 124             cls.pg0.resolve_arp()
 125
 126         except Exception:
 127             super(BFDTestCase, cls).tearDownClass()
 128             raise
 129
 130     def setUp(self):
 131         self.vapi.want_bfd_events()
 132         self.vpp_session = VppBFDUDPSession(self, self.pg0, self.pg0.remote_ip4)
 133         self.vpp_session.add_vpp_config()
 134         self.vpp_session.admin_up()
 135         self.test_session = BFDTestSession(self, self.pg0)
 136
 137     def tearDown(self):
 138         self.vapi.want_bfd_events(enable_disable=0)
 139         if not self.vpp_dead:
 140             self.vpp_session.remove_vpp_config()
 141         super(BFDTestCase, self).tearDown()
 142
 143     def verify_event(self, event, expected_state):
 144         """ Verify correctness of event values. """
 145         e = event
 146         self.logger.debug("Event: %s" % repr(e))
 147         self.assert_equal(e.bs_index, self.vpp_session.bs_index,
 148                           "BFD session index")
 149         self.assert_equal(e.sw_if_index, self.vpp_session.interface.sw_if_index,
 150                           "BFD interface index")
 151         is_ipv6 = 0
 152         if self.vpp_session.af == AF_INET6:
 153             is_ipv6 = 1
 154         self.assert_equal(e.is_ipv6, is_ipv6, "is_ipv6")
 155         if self.vpp_session.af == AF_INET:
 156             self.assert_equal(e.local_addr[:4], self.vpp_session.local_addr_n,
 157                               "Local IPv4 address")
 158             self.assert_equal(e.peer_addr[:4], self.vpp_session.peer_addr_n,
 159                               "Peer IPv4 address")
 160         else:
 161             self.assert_equal(e.local_addr, self.vpp_session.local_addr_n,
 162                               "Local IPv6 address")
 163             self.assert_equal(e.peer_addr, self.vpp_session.peer_addr_n,
 164                               "Peer IPv6 address")
 165         self.assert_equal(e.state, expected_state, BFDState)
 166
 167     def wait_for_bfd_packet(self, timeout=1):
 168         p = self.pg0.wait_for_packet(timeout=timeout)
 169         bfd = p[BFD]
 170         if bfd is None:
 171             raise Exception(ppp("Unexpected or invalid BFD packet:", p))
 172         if bfd.payload:
 173             raise Exception(ppp("Unexpected payload in BFD packet:", bfd))
 174         verify_ip(self, p, self.pg0.local_ip4, self.pg0.remote_ip4)
 175         verify_udp(self, p)
 176         self.test_session.verify_packet(p)
 177         return p
 178
 179     def test_slow_timer(self):
 180         """ Slow timer """
 181
 182         self.pg_enable_capture([self.pg0])
 183         expected_packets = 10
 184         self.logger.info("Waiting for %d BFD packets" % expected_packets)
 185         self.wait_for_bfd_packet()
 186         for i in range(expected_packets):
 187             before = time.time()
 188             self.wait_for_bfd_packet()
 189             after = time.time()
 190             self.assert_in_range(
 191                 after - before, 0.75, 1, "time between slow packets")
 192             before = after
 193
 194     def test_zero_remote_min_rx(self):
 195         """ Zero RemoteMinRxInterval """
 196         self.pg_enable_capture([self.pg0])
 197         p = self.wait_for_bfd_packet()
 198         self.test_session.update(my_discriminator=randint(0, 40000000),
 199                                  your_discriminator=p[BFD].my_discriminator,
 200                                  state=BFDState.init,
 201                                  required_min_rx_interval=0)
 202         self.test_session.send_packet()
 203         e = self.vapi.wait_for_event(1, "bfd_udp_session_details")
 204         self.verify_event(e, expected_state=BFDState.up)
 205
 206         try:
 207             p = self.pg0.wait_for_packet(timeout=1)
 208         except:
 209             return
 210         raise Exception(ppp("Received unexpected BFD packet:", p))
 211
 212     def bfd_conn_up(self):
 213         self.pg_enable_capture([self.pg0])
 214         self.logger.info("Waiting for slow hello")
 215         p = self.wait_for_bfd_packet()
 216         self.logger.info("Sending Init")
 217         self.test_session.update(my_discriminator=randint(0, 40000000),
 218                                  your_discriminator=p[BFD].my_discriminator,
 219                                  state=BFDState.init,
 220                                  required_min_rx_interval=500000)
 221         self.test_session.send_packet()
 222         self.logger.info("Waiting for event")
 223         e = self.vapi.wait_for_event(1, "bfd_udp_session_details")
 224         self.verify_event(e, expected_state=BFDState.up)
 225         self.logger.info("Session is Up")
 226         self.test_session.update(state=BFDState.up)
 227
 228     def test_conn_up(self):
 229         """ Basic connection up """
 230         self.bfd_conn_up()
 231
 232     def test_hold_up(self):
 233         """ Hold BFD up """
 234         self.bfd_conn_up()
 235         for i in range(5):
 236             self.wait_for_bfd_packet()
 237             self.test_session.send_packet()
 238
 239     def test_conn_down(self):
 240         """ Session down after inactivity """
 241         self.bfd_conn_up()
 242         self.wait_for_bfd_packet()
 243         self.assert_equal(
 244             0, len(self.vapi.collect_events()),
 245             "number of bfd events")
 246         self.wait_for_bfd_packet()
 247         self.assert_equal(
 248             0, len(self.vapi.collect_events()),
 249             "number of bfd events")
 250         e = self.vapi.wait_for_event(1, "bfd_udp_session_details")
 251         self.verify_event(e, expected_state=BFDState.down)
 252
 253     @unittest.skip("this test is not working yet")
 254     def test_large_required_min_rx(self):
 255         self.bfd_conn_up()
 256         interval = 5000000
 257         self.test_session.update(required_min_rx_interval=interval)
 258         self.test_session.send_packet()
 259         now = time.time()
 260         count = 1
 261         while time.time() < now + interval / 1000000:
 262             try:
 263                 p = self.wait_for_bfd_packet()
 264                 if count > 1:
 265                     self.logger.error(ppp("Received unexpected packet:", p))
 266                 count += 1
 267             except:
 268                 pass
 269         self.assert_equal(count, 1, "number of packets received")
 270
 271
 272 if __name__ == '__main__':
 273     unittest.main(testRunner=VppTestRunner)