test/vpp_interface.py

   1 import binascii
   2 import socket
   3 import abc
   4
   5 import six
   6 from six import moves
   7
   8 from util import Host, mk_ll_addr
   9 from vpp_papi import mac_ntop, VppEnum
  10 from ipaddress import IPv4Network, IPv6Network
  11
  12 try:
  13     text_type = unicode
  14 except NameError:
  15     text_type = str
  16
  17
  18 @six.add_metaclass(abc.ABCMeta)
  19 class VppInterface(object):
  20     """Generic VPP interface."""
  21
  22     @property
  23     def sw_if_index(self):
  24         """Interface index assigned by VPP."""
  25         return self._sw_if_index
  26
  27     @property
  28     def remote_mac(self):
  29         """MAC-address of the remote interface "connected" to this interface"""
  30         return self._remote_hosts[0].mac
  31
  32     @property
  33     def local_mac(self):
  34         """MAC-address of the VPP interface."""
  35         return str(self._local_mac)
  36
  37     @property
  38     def local_addr(self):
  39         return self._local_addr
  40
  41     @property
  42     def remote_addr(self):
  43         return self._remote_addr
  44
  45     @property
  46     def local_ip4(self):
  47         """Local IPv4 address on VPP interface (string)."""
  48         return self._local_ip4
  49
  50     @local_ip4.setter
  51     def local_ip4(self, value):
  52         self._local_ip4 = value
  53
  54     @property
  55     def local_ip4_prefix_len(self):
  56         """Local IPv4 prefix length """
  57         return self._local_ip4_len
  58
  59     @local_ip4_prefix_len.setter
  60     def local_ip4_prefix_len(self, value):
  61         self._local_ip4_len = value
  62
  63     @property
  64     def local_ip4_prefix(self):
  65         """Local IPv4 prefix """
  66         return ("%s/%d" % (self._local_ip4, self._local_ip4_len))
  67
  68     @property
  69     def local_ip4n(self):
  70         """DEPRECATED """
  71         """Local IPv4 address - raw, suitable as API parameter."""
  72         return socket.inet_pton(socket.AF_INET, self._local_ip4)
  73
  74     @property
  75     def remote_ip4(self):
  76         """IPv4 address of remote peer "connected" to this interface."""
  77         return self._remote_hosts[0].ip4
  78
  79     @property
  80     def remote_ip4n(self):
  81         """DEPRECATED """
  82         """Local IPv6 address - raw, suitable as API parameter."""
  83         return socket.inet_pton(socket.AF_INET, self._remote_hosts[0].ip4)
  84
  85     @property
  86     def local_ip6(self):
  87         """Local IPv6 address on VPP interface (string)."""
  88         return self._local_ip6
  89
  90     @local_ip6.setter
  91     def local_ip6(self, value):
  92         self._local_ip6
  93
  94     @property
  95     def local_ip6_prefix_len(self):
  96         """Local IPv6 prefix length """
  97         return self._local_ip6_len
  98
  99     @local_ip6_prefix_len.setter
 100     def local_ip6_prefix_len(self, value):
 101         self._local_ip6_len = value
 102
 103     @property
 104     def local_ip6_prefix(self):
 105         """Local IPv4 prefix """
 106         return ("%s/%d" % (self._local_ip6, self._local_ip6_len))
 107
 108     @property
 109     def local_ip6n(self):
 110         """DEPRECATED """
 111         """Local IPv6 address - raw, suitable as API parameter."""
 112         return socket.inet_pton(socket.AF_INET6, self._local_ip6)
 113
 114     @property
 115     def remote_ip6(self):
 116         """IPv6 address of remote peer "connected" to this interface."""
 117         return self._remote_hosts[0].ip6
 118
 119     @property
 120     def remote_ip6n(self):
 121         """DEPRECATED """
 122         """Local IPv6 address - raw, suitable as API parameter."""
 123         return socket.inet_pton(socket.AF_INET6, self._remote_hosts[0].ip6)
 124
 125     @property
 126     def local_ip6_ll(self):
 127         """Local IPv6 link-local address on VPP interface (string)."""
 128         return self._local_ip6_ll
 129
 130     @property
 131     def local_ip6n_ll(self):
 132         """DEPRECATED """
 133         """Local IPv6 link-local address on VPP interface (string)."""
 134         return socket.inet_pton(socket.AF_INET6, self._local_ip6_ll.address)
 135
 136     @property
 137     def remote_ip6_ll(self):
 138         """Link-local IPv6 address of remote peer
 139         "connected" to this interface."""
 140         return self._remote_ip6_ll
 141
 142     @property
 143     def remote_ip6n_ll(self):
 144         """DEPRECATED """
 145         """Local IPv6 link-local address on VPP interface (string)."""
 146         return socket.inet_pton(socket.AF_INET6, self._remote_ip6_ll)
 147
 148     @property
 149     def name(self):
 150         """Name of the interface."""
 151         return self._name
 152
 153     @property
 154     def dump(self):
 155         """RAW result of sw_interface_dump for this interface."""
 156         return self._dump
 157
 158     @property
 159     def test(self):
 160         """Test case creating this interface."""
 161         return self._test
 162
 163     @property
 164     def remote_hosts(self):
 165         """Remote hosts list"""
 166         return self._remote_hosts
 167
 168     @remote_hosts.setter
 169     def remote_hosts(self, value):
 170         """
 171         :param list value: List of remote hosts.
 172         """
 173         self._remote_hosts = value
 174         self._hosts_by_mac = {}
 175         self._hosts_by_ip4 = {}
 176         self._hosts_by_ip6 = {}
 177         for host in self._remote_hosts:
 178             self._hosts_by_mac[host.mac] = host
 179             self._hosts_by_ip4[host.ip4] = host
 180             self._hosts_by_ip6[host.ip6] = host
 181
 182     def host_by_mac(self, mac):
 183         """
 184         :param mac: MAC address to find host by.
 185         :return: Host object assigned to interface.
 186         """
 187         return self._hosts_by_mac[mac]
 188
 189     def host_by_ip4(self, ip):
 190         """
 191         :param ip: IPv4 address to find host by.
 192         :return: Host object assigned to interface.
 193         """
 194         return self._hosts_by_ip4[ip]
 195
 196     def host_by_ip6(self, ip):
 197         """
 198         :param ip: IPv6 address to find host by.
 199         :return: Host object assigned to interface.
 200         """
 201         return self._hosts_by_ip6[ip]
 202
 203     def generate_remote_hosts(self, count=1):
 204         """Generate and add remote hosts for the interface.
 205
 206         :param int count: Number of generated remote hosts.
 207         """
 208         self._remote_hosts = []
 209         self._hosts_by_mac = {}
 210         self._hosts_by_ip4 = {}
 211         self._hosts_by_ip6 = {}
 212         for i in range(
 213                 2, count + 2):  # 0: network address, 1: local vpp address
 214             mac = "02:%02x:00:00:ff:%02x" % (self.sw_if_index, i)
 215             ip4 = "172.16.%u.%u" % (self.sw_if_index, i)
 216             ip6 = "fd01:%x::%x" % (self.sw_if_index, i)
 217             ip6_ll = mk_ll_addr(mac)
 218             host = Host(mac, ip4, ip6, ip6_ll)
 219             self._remote_hosts.append(host)
 220             self._hosts_by_mac[mac] = host
 221             self._hosts_by_ip4[ip4] = host
 222             self._hosts_by_ip6[ip6] = host
 223
 224     @abc.abstractmethod
 225     def __init__(self, test):
 226         self._test = test
 227
 228         self._remote_hosts = []
 229         self._hosts_by_mac = {}
 230         self._hosts_by_ip4 = {}
 231         self._hosts_by_ip6 = {}
 232
 233     def set_mac(self, mac):
 234         self._local_mac = str(mac)
 235         self._local_ip6_ll = mk_ll_addr(self._local_mac)
 236         self.test.vapi.sw_interface_set_mac_address(
 237             self.sw_if_index, mac.packed)
 238
 239     def set_sw_if_index(self, sw_if_index):
 240         if sw_if_index > 255:
 241             raise RuntimeError("Don't support sw_if_index values "
 242                                "greater than 255.")
 243         self._sw_if_index = sw_if_index
 244
 245         self.generate_remote_hosts()
 246
 247         self._local_ip4 = "172.16.%u.1" % self.sw_if_index
 248         self._local_ip4_len = 24
 249         self._local_ip4_subnet = "172.16.%u.0" % self.sw_if_index
 250         self._local_ip4_bcast = "172.16.%u.255" % self.sw_if_index
 251         self.has_ip4_config = False
 252         self.ip4_table_id = 0
 253
 254         self._local_ip6 = "fd01:%x::1" % self.sw_if_index
 255         self._local_ip6_len = 64
 256         self.has_ip6_config = False
 257         self.ip6_table_id = 0
 258
 259         self._local_addr = {socket.AF_INET: self.local_ip4,
 260                             socket.AF_INET6: self.local_ip6}
 261         self._remote_addr = {socket.AF_INET: self.remote_ip4,
 262                              socket.AF_INET6: self.remote_ip6}
 263
 264         r = self.test.vapi.sw_interface_dump(sw_if_index=self.sw_if_index)
 265         for intf in r:
 266             if intf.sw_if_index == self.sw_if_index:
 267                 self._name = intf.interface_name
 268                 self._local_mac = intf.l2_address
 269                 self._dump = intf
 270                 break
 271         else:
 272             raise Exception(
 273                 "Could not find interface with sw_if_index %d "
 274                 "in interface dump %s" %
 275                 (self.sw_if_index, moves.reprlib.repr(r)))
 276         self._local_ip6_ll = mk_ll_addr(self.local_mac)
 277         self._remote_ip6_ll = mk_ll_addr(self.remote_mac)
 278
 279     def config_ip4(self):
 280         """Configure IPv4 address on the VPP interface."""
 281         self.test.vapi.sw_interface_add_del_address(
 282             sw_if_index=self.sw_if_index, prefix=self.local_ip4_prefix)
 283         self.has_ip4_config = True
 284
 285     def unconfig_ip4(self):
 286         """Remove IPv4 address on the VPP interface."""
 287         try:
 288             if self.has_ip4_config:
 289                 self.test.vapi.sw_interface_add_del_address(
 290                     sw_if_index=self.sw_if_index,
 291                     prefix=self.local_ip4_prefix, is_add=0)
 292         except AttributeError:
 293             self.has_ip4_config = False
 294         self.has_ip4_config = False
 295
 296     def configure_ipv4_neighbors(self):
 297         """For every remote host assign neighbor's MAC to IPv4 addresses.
 298
 299         :param vrf_id: The FIB table / VRF ID. (Default value = 0)
 300         """
 301         for host in self._remote_hosts:
 302             self.test.vapi.ip_neighbor_add_del(self.sw_if_index,
 303                                                host.mac,
 304                                                host.ip4)
 305
 306     def config_ip6(self):
 307         """Configure IPv6 address on the VPP interface."""
 308         self.test.vapi.sw_interface_add_del_address(
 309             sw_if_index=self.sw_if_index, prefix=self.local_ip6_prefix)
 310         self.has_ip6_config = True
 311
 312     def unconfig_ip6(self):
 313         """Remove IPv6 address on the VPP interface."""
 314         try:
 315             if self.has_ip6_config:
 316                 self.test.vapi.sw_interface_add_del_address(
 317                     sw_if_index=self.sw_if_index,
 318                     prefix=self.local_ip6_prefix, is_add=0)
 319         except AttributeError:
 320             self.has_ip6_config = False
 321         self.has_ip6_config = False
 322
 323     def configure_ipv6_neighbors(self):
 324         """For every remote host assign neighbor's MAC to IPv6 addresses.
 325
 326         :param vrf_id: The FIB table / VRF ID. (Default value = 0)
 327         """
 328         for host in self._remote_hosts:
 329             self.test.vapi.ip_neighbor_add_del(self.sw_if_index,
 330                                                host.mac,
 331                                                host.ip6)
 332
 333     def unconfig(self):
 334         """Unconfigure IPv6 and IPv4 address on the VPP interface."""
 335         self.unconfig_ip4()
 336         self.unconfig_ip6()
 337
 338     def set_table_ip4(self, table_id):
 339         """Set the interface in a IPv4 Table.
 340
 341         .. note:: Must be called before configuring IP4 addresses.
 342         """
 343         self.ip4_table_id = table_id
 344         self.test.vapi.sw_interface_set_table(
 345             self.sw_if_index, 0, self.ip4_table_id)
 346
 347     def set_table_ip6(self, table_id):
 348         """Set the interface in a IPv6 Table.
 349
 350         .. note:: Must be called before configuring IP6 addresses.
 351         """
 352         self.ip6_table_id = table_id
 353         self.test.vapi.sw_interface_set_table(
 354             self.sw_if_index, 1, self.ip6_table_id)
 355
 356     def disable_ipv6_ra(self):
 357         """Configure IPv6 RA suppress on the VPP interface."""
 358         self.test.vapi.sw_interface_ip6nd_ra_config(
 359             sw_if_index=self.sw_if_index,
 360             suppress=1)
 361
 362     def ip6_ra_config(self, no=0, suppress=0, send_unicast=0):
 363         """Configure IPv6 RA suppress on the VPP interface."""
 364         self.test.vapi.sw_interface_ip6nd_ra_config(
 365             sw_if_index=self.sw_if_index,
 366             is_no=no,
 367             suppress=suppress,
 368             send_unicast=send_unicast)
 369
 370     def ip6_ra_prefix(self, prefix, is_no=0,
 371                       off_link=0, no_autoconfig=0, use_default=0):
 372         """Configure IPv6 RA suppress on the VPP interface.
 373
 374         prefix can be a string in the format of '<address>/<length_in_bits>'
 375         or ipaddress.ipnetwork object (if strict.)"""
 376
 377         self.test.vapi.sw_interface_ip6nd_ra_prefix(
 378             sw_if_index=self.sw_if_index,
 379             prefix=prefix,
 380             use_default=use_default,
 381             off_link=off_link, no_autoconfig=no_autoconfig,
 382             is_no=is_no)
 383
 384     def admin_up(self):
 385         """Put interface ADMIN-UP."""
 386         self.test.vapi.sw_interface_set_flags(
 387             self.sw_if_index,
 388             flags=VppEnum.vl_api_if_status_flags_t.IF_STATUS_API_FLAG_ADMIN_UP)
 389
 390     def admin_down(self):
 391         """Put interface ADMIN-down."""
 392         self.test.vapi.sw_interface_set_flags(self.sw_if_index,
 393                                               flags=0)
 394
 395     def link_up(self):
 396         """Put interface link-state-UP."""
 397         self.test.vapi.cli("test interface link-state %s up" % self.name)
 398
 399     def link_down(self):
 400         """Put interface link-state-down."""
 401         self.test.vapi.cli("test interface link-state %s down" % self.name)
 402
 403     def ip6_enable(self):
 404         """IPv6 Enable interface"""
 405         self.test.vapi.sw_interface_ip6_enable_disable(self.sw_if_index,
 406                                                        enable=1)
 407
 408     def ip6_disable(self):
 409         """Put interface ADMIN-DOWN."""
 410         self.test.vapi.sw_interface_ip6_enable_disable(self.sw_if_index,
 411                                                        enable=0)
 412
 413     def add_sub_if(self, sub_if):
 414         """Register a sub-interface with this interface.
 415
 416         :param sub_if: sub-interface
 417         """
 418         if not hasattr(self, 'sub_if'):
 419             self.sub_if = sub_if
 420         else:
 421             if isinstance(self.sub_if, list):
 422                 self.sub_if.append(sub_if)
 423             else:
 424                 self.sub_if = sub_if
 425
 426     def enable_mpls(self):
 427         """Enable MPLS on the VPP interface."""
 428         self.test.vapi.sw_interface_set_mpls_enable(self.sw_if_index)
 429
 430     def disable_mpls(self):
 431         """Enable MPLS on the VPP interface."""
 432         self.test.vapi.sw_interface_set_mpls_enable(self.sw_if_index, 0)
 433
 434     def is_ip4_entry_in_fib_dump(self, dump):
 435         for i in dump:
 436             n = IPv4Network(text_type("%s/%d" % (self.local_ip4,
 437                                                  self.local_ip4_prefix_len)))
 438             if i.route.prefix == n and \
 439                i.route.table_id == self.ip4_table_id:
 440                 return True
 441         return False
 442
 443     def set_unnumbered(self, ip_sw_if_index):
 444         """ Set the interface to unnumbered via ip_sw_if_index """
 445         self.test.vapi.sw_interface_set_unnumbered(ip_sw_if_index,
 446                                                    self.sw_if_index)
 447
 448     def unset_unnumbered(self, ip_sw_if_index):
 449         """ Unset the interface to unnumbered via ip_sw_if_index """
 450         self.test.vapi.sw_interface_set_unnumbered(ip_sw_if_index,
 451                                                    self.sw_if_index, is_add=0)
 452
 453     def set_proxy_arp(self, enable=1):
 454         """ Set the interface to enable/disable Proxy ARP """
 455         self.test.vapi.proxy_arp_intfc_enable_disable(
 456             self.sw_if_index,
 457             enable)
 458
 459     def query_vpp_config(self):
 460         dump = self.test.vapi.sw_interface_dump(sw_if_index=self.sw_if_index)
 461         return self.is_interface_config_in_dump(dump)
 462
 463     def get_interface_config_from_dump(self, dump):
 464         for i in dump:
 465             if i.interface_name.rstrip(' \t\r\n\0') == self.name and \
 466                     i.sw_if_index == self.sw_if_index:
 467                 return i
 468         else:
 469             return None
 470
 471     def is_interface_config_in_dump(self, dump):
 472         return self.get_interface_config_from_dump(dump) is not None
 473
 474     def assert_interface_state(self, admin_up_down, link_up_down,
 475                                expect_event=False):
 476         if expect_event:
 477             event = self.test.vapi.wait_for_event(timeout=1,
 478                                                   name='sw_interface_event')
 479             self.test.assert_equal(event.sw_if_index, self.sw_if_index,
 480                                    "sw_if_index")
 481             self.test.assert_equal((event.flags & 1), admin_up_down,
 482                                    "admin state")
 483             self.test.assert_equal((event.flags & 2), link_up_down,
 484                                    "link state")
 485         dump = self.test.vapi.sw_interface_dump()
 486         if_state = self.get_interface_config_from_dump(dump)
 487         self.test.assert_equal((if_state.flags & 1), admin_up_down,
 488                                "admin state")
 489         self.test.assert_equal((if_state.flags & 2), link_up_down,
 490                                "link state")
 491
 492     def __str__(self):
 493         return self.name
 494
 495     def get_rx_stats(self):
 496         c = self.test.statistics.get_counter("^/if/rx$")
 497         return c[0][self.sw_if_index]
 498
 499     def get_tx_stats(self):
 500         c = self.test.statistics.get_counter("^/if/tx$")
 501         return c[0][self.sw_if_index]
 502
 503     def set_l3_mtu(self, mtu):
 504         self.test.vapi.sw_interface_set_mtu(self.sw_if_index, [mtu, 0, 0, 0])
 505
 506     def set_ip4_mtu(self, mtu):
 507         self.test.vapi.sw_interface_set_mtu(self.sw_if_index, [0, mtu, 0, 0])
 508
 509     def set_ip6_mtu(self, mtu):
 510         self.test.vapi.sw_interface_set_mtu(self.sw_if_index, [0, 0, mtu, 0])
 511
 512     def set_mpls_mtu(self, mtu):
 513         self.test.vapi.sw_interface_set_mtu(self.sw_if_index, [0, 0, 0, mtu])