test/vpp_papi_provider.py

   1 import os
   2 import fnmatch
   3 import time
   4 from hook import Hook
   5 from collections import deque
   6
   7 # Sphinx creates auto-generated documentation by importing the python source
   8 # files and collecting the docstrings from them. The NO_VPP_PAPI flag allows
   9 # the vpp_papi_provider.py file to be importable without having to build
  10 # the whole vpp api if the user only wishes to generate the test documentation.
  11 do_import = True
  12 try:
  13     no_vpp_papi = os.getenv("NO_VPP_PAPI")
  14     if no_vpp_papi == "1":
  15         do_import = False
  16 except:
  17     pass
  18
  19 if do_import:
  20     from vpp_papi import VPP
  21
  22 # from vnet/vnet/mpls/mpls_types.h
  23 MPLS_IETF_MAX_LABEL = 0xfffff
  24 MPLS_LABEL_INVALID = MPLS_IETF_MAX_LABEL + 1
  25
  26
  27 class L2_VTR_OP:
  28     L2_POP_1 = 3
  29
  30
  31 class VppPapiProvider(object):
  32     """VPP-api provider using vpp-papi
  33
  34     @property hook: hook object providing before and after api/cli hooks
  35     """
  36
  37     _zero, _negative = range(2)
  38
  39     def __init__(self, name, shm_prefix, test_class):
  40         self.hook = Hook("vpp-papi-provider")
  41         self.name = name
  42         self.shm_prefix = shm_prefix
  43         self.test_class = test_class
  44         self._expect_api_retval = self._zero
  45         self._expect_stack = []
  46         jsonfiles = []
  47
  48         install_dir = os.getenv('VPP_TEST_INSTALL_PATH')
  49         for root, dirnames, filenames in os.walk(install_dir):
  50             for filename in fnmatch.filter(filenames, '*.api.json'):
  51                 jsonfiles.append(os.path.join(root, filename))
  52
  53         self.papi = VPP(jsonfiles)
  54         self._events = deque()
  55
  56     def __enter__(self):
  57         return self
  58
  59     def expect_negative_api_retval(self):
  60         """ Expect API failure """
  61         self._expect_stack.append(self._expect_api_retval)
  62         self._expect_api_retval = self._negative
  63         return self
  64
  65     def expect_zero_api_retval(self):
  66         """ Expect API success """
  67         self._expect_stack.append(self._expect_api_retval)
  68         self._expect_api_retval = self._zero
  69         return self
  70
  71     def __exit__(self, exc_type, exc_value, traceback):
  72         self._expect_api_retval = self._expect_stack.pop()
  73
  74     def register_hook(self, hook):
  75         """Replace hook registration with new hook
  76
  77         :param hook:
  78
  79         """
  80         self.hook = hook
  81
  82     def collect_events(self):
  83         """ Collect all events from the internal queue and clear the queue. """
  84         e = self._events
  85         self._events = deque()
  86         return e
  87
  88     def wait_for_event(self, timeout, name=None):
  89         """ Wait for and return next event. """
  90         if name:
  91             self.test_class.logger.debug("Expecting event within %ss",
  92                                          timeout)
  93         else:
  94             self.test_class.logger.debug("Expecting event '%s' within %ss",
  95                                          name, timeout)
  96         if self._events:
  97             self.test_class.logger.debug("Not waiting, event already queued")
  98         limit = time.time() + timeout
  99         while time.time() < limit:
 100             if self._events:
 101                 e = self._events.popleft()
 102                 if name and type(e).__name__ != name:
 103                     raise Exception(
 104                         "Unexpected event received: %s, expected: %s" %
 105                         (type(e).__name__, name))
 106                 self.test_class.logger.debug("Returning event %s:%s" %
 107                                              (name, e))
 108                 return e
 109             time.sleep(0)  # yield
 110         raise Exception("Event did not occur within timeout")
 111
 112     def __call__(self, name, event):
 113         """ Enqueue event in the internal event queue. """
 114         # FIXME use the name instead of relying on type(e).__name__ ?
 115         # FIXME #2 if this throws, it is eaten silently, Ole?
 116         self.test_class.logger.debug("New event: %s: %s" % (name, event))
 117         self._events.append(event)
 118
 119     def connect(self):
 120         """Connect the API to VPP"""
 121         self.papi.connect(self.name, self.shm_prefix)
 122         self.papi.register_event_callback(self)
 123
 124     def disconnect(self):
 125         """Disconnect the API from VPP"""
 126         self.papi.disconnect()
 127
 128     def api(self, api_fn, api_args, expected_retval=0):
 129         """ Call API function and check it's return value.
 130         Call the appropriate hooks before and after the API call
 131
 132         :param api_fn: API function to call
 133         :param api_args: tuple of API function arguments
 134         :param expected_retval: Expected return value (Default value = 0)
 135         :returns: reply from the API
 136
 137         """
 138         self.hook.before_api(api_fn.__name__, api_args)
 139         reply = api_fn(**api_args)
 140         if self._expect_api_retval == self._negative:
 141             if hasattr(reply, 'retval') and reply.retval >= 0:
 142                 msg = "API call passed unexpectedly: expected negative "\
 143                     "return value instead of %d in %s" % \
 144                     (reply.retval, repr(reply))
 145                 self.test_class.logger.info(msg)
 146                 raise Exception(msg)
 147         elif self._expect_api_retval == self._zero:
 148             if hasattr(reply, 'retval') and reply.retval != expected_retval:
 149                 msg = "API call failed, expected zero return value instead "\
 150                     "of %d in %s" % (expected_retval, repr(reply))
 151                 self.test_class.logger.info(msg)
 152                 raise Exception(msg)
 153         else:
 154             raise Exception("Internal error, unexpected value for "
 155                             "self._expect_api_retval %s" %
 156                             self._expect_api_retval)
 157         self.hook.after_api(api_fn.__name__, api_args)
 158         return reply
 159
 160     def cli(self, cli):
 161         """ Execute a CLI, calling the before/after hooks appropriately.
 162
 163         :param cli: CLI to execute
 164         :returns: CLI output
 165
 166         """
 167         self.hook.before_cli(cli)
 168         cli += '\n'
 169         r = self.papi.cli_inband(length=len(cli), cmd=cli)
 170         self.hook.after_cli(cli)
 171         if hasattr(r, 'reply'):
 172             return r.reply.decode().rstrip('\x00')
 173
 174     def ppcli(self, cli):
 175         """ Helper method to print CLI command in case of info logging level.
 176
 177         :param cli: CLI to execute
 178         :returns: CLI output
 179         """
 180         return cli + "\n" + str(self.cli(cli))
 181
 182     def _convert_mac(self, mac):
 183         return int(mac.replace(":", ""), 16) << 16
 184
 185     def show_version(self):
 186         """ """
 187         return self.papi.show_version()
 188
 189     def pg_create_interface(self, pg_index):
 190         """
 191
 192         :param pg_index:
 193
 194         """
 195         return self.api(self.papi.pg_create_interface,
 196                         {"interface_id": pg_index})
 197
 198     def sw_interface_dump(self, filter=None):
 199         """
 200
 201         :param filter:  (Default value = None)
 202
 203         """
 204         if filter is not None:
 205             args = {"name_filter_valid": 1, "name_filter": filter}
 206         else:
 207             args = {}
 208         return self.api(self.papi.sw_interface_dump, args)
 209
 210     def sw_interface_set_table(self, sw_if_index, is_ipv6, table_id):
 211         """ Set the IPvX Table-id for the Interface
 212
 213         :param sw_if_index:
 214         :param is_ipv6:
 215         :param table_id:
 216
 217         """
 218         return self.api(self.papi.sw_interface_set_table,
 219                         {'sw_if_index': sw_if_index, 'is_ipv6': is_ipv6,
 220                          'vrf_id': table_id})
 221
 222     def sw_interface_add_del_address(self, sw_if_index, addr, addr_len,
 223                                      is_ipv6=0, is_add=1, del_all=0):
 224         """
 225
 226         :param addr: param is_ipv6:  (Default value = 0)
 227         :param sw_if_index:
 228         :param addr_len:
 229         :param is_ipv6:  (Default value = 0)
 230         :param is_add:  (Default value = 1)
 231         :param del_all:  (Default value = 0)
 232
 233         """
 234         return self.api(self.papi.sw_interface_add_del_address,
 235                         {'sw_if_index': sw_if_index,
 236                          'is_add': is_add,
 237                          'is_ipv6': is_ipv6,
 238                          'del_all': del_all,
 239                          'address_length': addr_len,
 240                          'address': addr})
 241
 242     def sw_interface_enable_disable_mpls(self, sw_if_index,
 243                                          is_enable=1):
 244         """
 245         Enable/Disable MPLS on the interface
 246         :param sw_if_index:
 247         :param is_enable:  (Default value = 1)
 248
 249         """
 250         return self.api(self.papi.sw_interface_set_mpls_enable,
 251                         {'sw_if_index': sw_if_index,
 252                          'enable': is_enable})
 253
 254     def sw_interface_ra_suppress(self, sw_if_index):
 255         return self.api(self.papi.sw_interface_ip6nd_ra_config,
 256                         {'sw_if_index': sw_if_index})
 257
 258     def ip6_sw_interface_ra_config(self, sw_if_index,
 259                                    no,
 260                                    suppress,
 261                                    send_unicast):
 262         return self.api(self.papi.sw_interface_ip6nd_ra_config,
 263                         {'sw_if_index': sw_if_index,
 264                          'is_no': no,
 265                          'suppress': suppress,
 266                          'send_unicast': send_unicast})
 267
 268     def ip6_sw_interface_enable_disable(self, sw_if_index, enable):
 269         """
 270         Enable/Disable An interface for IPv6
 271         """
 272         return self.api(self.papi.sw_interface_ip6_enable_disable,
 273                         {'sw_if_index': sw_if_index,
 274                          'enable': enable})
 275
 276     def vxlan_add_del_tunnel(
 277             self,
 278             src_addr,
 279             dst_addr,
 280             mcast_sw_if_index=0xFFFFFFFF,
 281             is_add=1,
 282             is_ipv6=0,
 283             encap_vrf_id=0,
 284             decap_next_index=0xFFFFFFFF,
 285             vni=0):
 286         """
 287
 288         :param dst_addr:
 289         :param src_addr:
 290         :param is_add:  (Default value = 1)
 291         :param is_ipv6:  (Default value = 0)
 292         :param encap_vrf_id:  (Default value = 0)
 293         :param decap_next_index:  (Default value = 0xFFFFFFFF)
 294         :param mcast_sw_if_index:  (Default value = 0xFFFFFFFF)
 295         :param vni:  (Default value = 0)
 296
 297         """
 298         return self.api(self.papi.vxlan_add_del_tunnel,
 299                         {'is_add': is_add,
 300                          'is_ipv6': is_ipv6,
 301                          'src_address': src_addr,
 302                          'dst_address': dst_addr,
 303                          'mcast_sw_if_index': mcast_sw_if_index,
 304                          'encap_vrf_id': encap_vrf_id,
 305                          'decap_next_index': decap_next_index,
 306                          'vni': vni})
 307
 308     def bridge_domain_add_del(self, bd_id, flood=1, uu_flood=1, forward=1,
 309                               learn=1, arp_term=0, is_add=1):
 310         """Create/delete bridge domain.
 311
 312         :param int bd_id: Bridge domain index.
 313         :param int flood: Enable/disable bcast/mcast flooding in the BD.
 314             (Default value = 1)
 315         :param int uu_flood: Enable/disable unknown unicast flood in the BD.
 316             (Default value = 1)
 317         :param int forward: Enable/disable forwarding on all interfaces in
 318             the BD. (Default value = 1)
 319         :param int learn: Enable/disable learning on all interfaces in the BD.
 320             (Default value = 1)
 321         :param int arp_term: Enable/disable arp termination in the BD.
 322             (Default value = 1)
 323         :param int is_add: Add or delete flag. (Default value = 1)
 324         """
 325         return self.api(self.papi.bridge_domain_add_del,
 326                         {'bd_id': bd_id,
 327                          'flood': flood,
 328                          'uu_flood': uu_flood,
 329                          'forward': forward,
 330                          'learn': learn,
 331                          'arp_term': arp_term,
 332                          'is_add': is_add})
 333
 334     def l2fib_add_del(self, mac, bd_id, sw_if_index, is_add=1, static_mac=0,
 335                       filter_mac=0, bvi_mac=0):
 336         """Create/delete L2 FIB entry.
 337
 338         :param str mac: MAC address to create FIB entry for.
 339         :param int bd_id: Bridge domain index.
 340         :param int sw_if_index: Software interface index of the interface.
 341         :param int is_add: Add or delete flag. (Default value = 1)
 342         :param int static_mac: Set to 1 to create static MAC entry.
 343             (Default value = 0)
 344         :param int filter_mac: Set to 1 to drop packet that's source or
 345             destination MAC address contains defined MAC address.
 346             (Default value = 0)
 347         :param int bvi_mac: Set to 1 to create entry that points to BVI
 348             interface. (Default value = 0)
 349         """
 350         return self.api(self.papi.l2fib_add_del,
 351                         {'mac': self._convert_mac(mac),
 352                          'bd_id': bd_id,
 353                          'sw_if_index': sw_if_index,
 354                          'is_add': is_add,
 355                          'static_mac': static_mac,
 356                          'filter_mac': filter_mac,
 357                          'bvi_mac': bvi_mac})
 358
 359     def sw_interface_set_l2_bridge(self, sw_if_index, bd_id,
 360                                    shg=0, bvi=0, enable=1):
 361         """Add/remove interface to/from bridge domain.
 362
 363         :param int sw_if_index: Software interface index of the interface.
 364         :param int bd_id: Bridge domain index.
 365         :param int shg: Split-horizon group index. (Default value = 0)
 366         :param int bvi: Set interface as a bridge group virtual interface.
 367             (Default value = 0)
 368         :param int enable: Add or remove interface. (Default value = 1)
 369         """
 370         return self.api(self.papi.sw_interface_set_l2_bridge,
 371                         {'rx_sw_if_index': sw_if_index,
 372                          'bd_id': bd_id,
 373                          'shg': shg,
 374                          'bvi': bvi,
 375                          'enable': enable})
 376
 377     def bridge_flags(self, bd_id, is_set, feature_bitmap):
 378         """Enable/disable required feature of the bridge domain with defined ID.
 379
 380         :param int bd_id: Bridge domain ID.
 381         :param int is_set: Set to 1 to enable, set to 0 to disable the feature.
 382         :param int feature_bitmap: Bitmap value of the feature to be set:
 383             - learn (1 << 0),
 384             - forward (1 << 1),
 385             - flood (1 << 2),
 386             - uu-flood (1 << 3) or
 387             - arp-term (1 << 4).
 388         """
 389         return self.api(self.papi.bridge_flags,
 390                         {'bd_id': bd_id,
 391                          'is_set': is_set,
 392                          'feature_bitmap': feature_bitmap})
 393
 394     def bridge_domain_dump(self, bd_id=0):
 395         """
 396
 397         :param int bd_id: Bridge domain ID. (Default value = 0 => dump of all
 398             existing bridge domains returned)
 399         :return: Dictionary of bridge domain(s) data.
 400         """
 401         return self.api(self.papi.bridge_domain_dump,
 402                         {'bd_id': bd_id})
 403
 404     def sw_interface_set_l2_xconnect(self, rx_sw_if_index, tx_sw_if_index,
 405                                      enable):
 406         """Create or delete unidirectional cross-connect from Tx interface to
 407         Rx interface.
 408
 409         :param int rx_sw_if_index: Software interface index of Rx interface.
 410         :param int tx_sw_if_index: Software interface index of Tx interface.
 411         :param int enable: Create cross-connect if equal to 1, delete
 412             cross-connect if equal to 0.
 413
 414         """
 415         return self.api(self.papi.sw_interface_set_l2_xconnect,
 416                         {'rx_sw_if_index': rx_sw_if_index,
 417                          'tx_sw_if_index': tx_sw_if_index,
 418                          'enable': enable})
 419
 420     def sw_interface_set_l2_tag_rewrite(
 421             self,
 422             sw_if_index,
 423             vtr_oper,
 424             push=0,
 425             tag1=0,
 426             tag2=0):
 427         """L2 interface vlan tag rewrite configure request
 428         :param client_index - opaque cookie to identify the sender
 429         :param context - sender context, to match reply w/ request
 430         :param sw_if_index - interface the operation is applied to
 431         :param vtr_op - Choose from l2_vtr_op_t enum values
 432         :param push_dot1q - first pushed flag dot1q id set, else dot1ad
 433         :param tag1 - Needed for any push or translate vtr op
 434         :param tag2 - Needed for any push 2 or translate x-2 vtr ops
 435
 436         """
 437         return self.api(self.papi.l2_interface_vlan_tag_rewrite,
 438                         {'sw_if_index': sw_if_index,
 439                          'vtr_op': vtr_oper,
 440                          'push_dot1q': push,
 441                          'tag1': tag1,
 442                          'tag2': tag2})
 443
 444     def sw_interface_set_flags(self, sw_if_index, admin_up_down,
 445                                link_up_down=0, deleted=0):
 446         """
 447
 448         :param admin_up_down:
 449         :param sw_if_index:
 450         :param link_up_down:  (Default value = 0)
 451         :param deleted:  (Default value = 0)
 452
 453         """
 454         return self.api(self.papi.sw_interface_set_flags,
 455                         {'sw_if_index': sw_if_index,
 456                          'admin_up_down': admin_up_down,
 457                          'link_up_down': link_up_down,
 458                          'deleted': deleted})
 459
 460     def create_subif(self, sw_if_index, sub_id, outer_vlan, inner_vlan,
 461                      no_tags=0, one_tag=0, two_tags=0, dot1ad=0, exact_match=0,
 462                      default_sub=0, outer_vlan_id_any=0, inner_vlan_id_any=0):
 463         """Create subinterface
 464         from vpe.api: set dot1ad = 0 for dot1q, set dot1ad = 1 for dot1ad
 465
 466         :param sub_id: param inner_vlan:
 467         :param sw_if_index:
 468         :param outer_vlan:
 469         :param inner_vlan:
 470         :param no_tags:  (Default value = 0)
 471         :param one_tag:  (Default value = 0)
 472         :param two_tags:  (Default value = 0)
 473         :param dot1ad:  (Default value = 0)
 474         :param exact_match:  (Default value = 0)
 475         :param default_sub:  (Default value = 0)
 476         :param outer_vlan_id_any:  (Default value = 0)
 477         :param inner_vlan_id_any:  (Default value = 0)
 478
 479         """
 480         return self.api(
 481             self.papi.create_subif,
 482             {'sw_if_index': sw_if_index,
 483              'sub_id': sub_id,
 484              'no_tags': no_tags,
 485              'one_tag': one_tag,
 486              'two_tags': two_tags,
 487              'dot1ad': dot1ad,
 488              'exact_match': exact_match,
 489              'default_sub': default_sub,
 490              'outer_vlan_id_any': outer_vlan_id_any,
 491              'inner_vlan_id_any': inner_vlan_id_any,
 492              'outer_vlan_id': outer_vlan,
 493              'inner_vlan_id': inner_vlan})
 494
 495     def delete_subif(self, sw_if_index):
 496         """Delete subinterface
 497
 498         :param sw_if_index:
 499         """
 500         return self.api(self.papi.delete_subif,
 501                         {'sw_if_index': sw_if_index})
 502
 503     def create_vlan_subif(self, sw_if_index, vlan):
 504         """
 505
 506         :param vlan:
 507         :param sw_if_index:
 508
 509         """
 510         return self.api(self.papi.create_vlan_subif,
 511                         {'sw_if_index': sw_if_index,
 512                          'vlan_id': vlan})
 513
 514     def create_loopback(self, mac=''):
 515         """
 516
 517         :param mac: (Optional)
 518         """
 519         return self.api(self.papi.create_loopback,
 520                         {'mac_address': mac})
 521
 522     def delete_loopback(self, sw_if_index):
 523         return self.api(self.papi.delete_loopback,
 524                         {'sw_if_index': sw_if_index, })
 525
 526     def ip_add_del_route(
 527             self,
 528             dst_address,
 529             dst_address_length,
 530             next_hop_address,
 531             next_hop_sw_if_index=0xFFFFFFFF,
 532             table_id=0,
 533             next_hop_table_id=0,
 534             next_hop_weight=1,
 535             next_hop_n_out_labels=0,
 536             next_hop_out_label_stack=[],
 537             next_hop_via_label=MPLS_LABEL_INVALID,
 538             create_vrf_if_needed=0,
 539             is_resolve_host=0,
 540             is_resolve_attached=0,
 541             classify_table_index=0xFFFFFFFF,
 542             is_add=1,
 543             is_drop=0,
 544             is_unreach=0,
 545             is_prohibit=0,
 546             is_ipv6=0,
 547             is_local=0,
 548             is_classify=0,
 549             is_multipath=0,
 550             not_last=0):
 551         """
 552
 553         :param dst_address_length:
 554         :param next_hop_sw_if_index:  (Default value = 0xFFFFFFFF)
 555         :param dst_address:
 556         :param next_hop_address:
 557         :param next_hop_sw_if_index:  (Default value = 0xFFFFFFFF)
 558         :param vrf_id:  (Default value = 0)
 559         :param lookup_in_vrf:  (Default value = 0)
 560         :param classify_table_index:  (Default value = 0xFFFFFFFF)
 561         :param create_vrf_if_needed:  (Default value = 0)
 562         :param is_add:  (Default value = 1)
 563         :param is_drop:  (Default value = 0)
 564         :param is_ipv6:  (Default value = 0)
 565         :param is_local:  (Default value = 0)
 566         :param is_classify:  (Default value = 0)
 567         :param is_multipath:  (Default value = 0)
 568         :param is_resolve_host:  (Default value = 0)
 569         :param is_resolve_attached:  (Default value = 0)
 570         :param not_last:  (Default value = 0)
 571         :param next_hop_weight:  (Default value = 1)
 572
 573         """
 574
 575         return self.api(
 576             self.papi.ip_add_del_route,
 577             {'next_hop_sw_if_index': next_hop_sw_if_index,
 578              'table_id': table_id,
 579              'classify_table_index': classify_table_index,
 580              'next_hop_table_id': next_hop_table_id,
 581              'create_vrf_if_needed': create_vrf_if_needed,
 582              'is_add': is_add,
 583              'is_drop': is_drop,
 584              'is_unreach': is_unreach,
 585              'is_prohibit': is_prohibit,
 586              'is_ipv6': is_ipv6,
 587              'is_local': is_local,
 588              'is_classify': is_classify,
 589              'is_multipath': is_multipath,
 590              'is_resolve_host': is_resolve_host,
 591              'is_resolve_attached': is_resolve_attached,
 592              'not_last': not_last,
 593              'next_hop_weight': next_hop_weight,
 594              'dst_address_length': dst_address_length,
 595              'dst_address': dst_address,
 596              'next_hop_address': next_hop_address,
 597              'next_hop_n_out_labels': next_hop_n_out_labels,
 598              'next_hop_via_label': next_hop_via_label,
 599              'next_hop_out_label_stack': next_hop_out_label_stack})
 600
 601     def ip_fib_dump(self):
 602         return self.api(self.papi.ip_fib_dump, {})
 603
 604     def ip_neighbor_add_del(self,
 605                             sw_if_index,
 606                             mac_address,
 607                             dst_address,
 608                             vrf_id=0,
 609                             is_add=1,
 610                             is_ipv6=0,
 611                             is_static=0,
 612                             ):
 613         """ Add neighbor MAC to IPv4 or IPv6 address.
 614
 615         :param sw_if_index:
 616         :param mac_address:
 617         :param dst_address:
 618         :param vrf_id:  (Default value = 0)
 619         :param is_add:  (Default value = 1)
 620         :param is_ipv6:  (Default value = 0)
 621         :param is_static:  (Default value = 0)
 622         """
 623
 624         return self.api(
 625             self.papi.ip_neighbor_add_del,
 626             {'vrf_id': vrf_id,
 627              'sw_if_index': sw_if_index,
 628              'is_add': is_add,
 629              'is_ipv6': is_ipv6,
 630              'is_static': is_static,
 631              'mac_address': mac_address,
 632              'dst_address': dst_address
 633              }
 634         )
 635
 636     def reset_vrf(self,
 637                   vrf_id,
 638                   is_ipv6=0,
 639                   ):
 640         """ Reset VRF (remove all routes etc.) request.
 641
 642         :param int vrf_id: ID of the FIB table / VRF to reset.
 643         :param int is_ipv6: 1 for IPv6 neighbor, 0 for IPv4. (Default = 0)
 644         """
 645
 646         return self.api(
 647             self.papi.reset_vrf,
 648             {'vrf_id': vrf_id,
 649              'is_ipv6': is_ipv6,
 650              }
 651         )
 652
 653     def reset_fib(self,
 654                   vrf_id,
 655                   is_ipv6=0,
 656                   ):
 657         """ Reset VRF (remove all routes etc.) request.
 658
 659         :param int vrf_id: ID of the FIB table / VRF to reset.
 660         :param int is_ipv6: 1 for IPv6 neighbor, 0 for IPv4. (Default = 0)
 661         """
 662
 663         return self.api(
 664             self.papi.reset_fib,
 665             {'vrf_id': vrf_id,
 666              'is_ipv6': is_ipv6,
 667              }
 668         )
 669
 670     def ip_dump(self,
 671                 is_ipv6=0,
 672                 ):
 673         """ Return IP dump.
 674
 675         :param int is_ipv6: 1 for IPv6 neighbor, 0 for IPv4. (Default = 0)
 676         """
 677
 678         return self.api(
 679             self.papi.ip_dump,
 680             {'is_ipv6': is_ipv6,
 681              }
 682         )
 683
 684     def sw_interface_span_enable_disable(
 685             self, sw_if_index_from, sw_if_index_to, state=1):
 686         """
 687
 688         :param sw_if_index_from:
 689         :param sw_if_index_to:
 690         :param state:
 691         """
 692         return self.api(self.papi.sw_interface_span_enable_disable,
 693                         {'sw_if_index_from': sw_if_index_from,
 694                          'sw_if_index_to': sw_if_index_to,
 695                          'state': state})
 696
 697     def gre_tunnel_add_del(self,
 698                            src_address,
 699                            dst_address,
 700                            outer_fib_id=0,
 701                            is_teb=0,
 702                            is_add=1,
 703                            is_ip6=0):
 704         """ Add a GRE tunnel
 705
 706         :param src_address:
 707         :param dst_address:
 708         :param outer_fib_id:  (Default value = 0)
 709         :param is_add:  (Default value = 1)
 710         :param is_ipv6:  (Default value = 0)
 711         :param is_teb:  (Default value = 0)
 712         """
 713
 714         return self.api(
 715             self.papi.gre_add_del_tunnel,
 716             {'is_add': is_add,
 717              'is_ipv6': is_ip6,
 718              'teb': is_teb,
 719              'src_address': src_address,
 720              'dst_address': dst_address,
 721              'outer_fib_id': outer_fib_id}
 722         )
 723
 724     def mpls_fib_dump(self):
 725         return self.api(self.papi.mpls_fib_dump, {})
 726
 727     def mpls_route_add_del(
 728             self,
 729             label,
 730             eos,
 731             next_hop_proto_is_ip4,
 732             next_hop_address,
 733             next_hop_sw_if_index=0xFFFFFFFF,
 734             table_id=0,
 735             next_hop_table_id=0,
 736             next_hop_weight=1,
 737             next_hop_n_out_labels=0,
 738             next_hop_out_label_stack=[],
 739             next_hop_via_label=MPLS_LABEL_INVALID,
 740             create_vrf_if_needed=0,
 741             is_resolve_host=0,
 742             is_resolve_attached=0,
 743             is_add=1,
 744             is_drop=0,
 745             is_multipath=0,
 746             classify_table_index=0xFFFFFFFF,
 747             is_classify=0,
 748             not_last=0):
 749         """
 750
 751         :param dst_address_length:
 752         :param next_hop_sw_if_index:  (Default value = 0xFFFFFFFF)
 753         :param dst_address:
 754         :param next_hop_address:
 755         :param next_hop_sw_if_index:  (Default value = 0xFFFFFFFF)
 756         :param vrf_id:  (Default value = 0)
 757         :param lookup_in_vrf:  (Default value = 0)
 758         :param classify_table_index:  (Default value = 0xFFFFFFFF)
 759         :param create_vrf_if_needed:  (Default value = 0)
 760         :param is_add:  (Default value = 1)
 761         :param is_drop:  (Default value = 0)
 762         :param is_ipv6:  (Default value = 0)
 763         :param is_local:  (Default value = 0)
 764         :param is_classify:  (Default value = 0)
 765         :param is_multipath:  (Default value = 0)
 766         :param is_resolve_host:  (Default value = 0)
 767         :param is_resolve_attached:  (Default value = 0)
 768         :param not_last:  (Default value = 0)
 769         :param next_hop_weight:  (Default value = 1)
 770
 771         """
 772
 773         return self.api(
 774             self.papi.mpls_route_add_del,
 775             {'mr_label': label,
 776              'mr_eos': eos,
 777              'mr_table_id': table_id,
 778              'mr_classify_table_index': classify_table_index,
 779              'mr_create_table_if_needed': create_vrf_if_needed,
 780              'mr_is_add': is_add,
 781              'mr_is_classify': is_classify,
 782              'mr_is_multipath': is_multipath,
 783              'mr_is_resolve_host': is_resolve_host,
 784              'mr_is_resolve_attached': is_resolve_attached,
 785              'mr_next_hop_proto_is_ip4': next_hop_proto_is_ip4,
 786              'mr_next_hop_weight': next_hop_weight,
 787              'mr_next_hop': next_hop_address,
 788              'mr_next_hop_n_out_labels': next_hop_n_out_labels,
 789              'mr_next_hop_sw_if_index': next_hop_sw_if_index,
 790              'mr_next_hop_table_id': next_hop_table_id,
 791              'mr_next_hop_via_label': next_hop_via_label,
 792              'mr_next_hop_out_label_stack': next_hop_out_label_stack})
 793
 794     def mpls_ip_bind_unbind(
 795             self,
 796             label,
 797             dst_address,
 798             dst_address_length,
 799             table_id=0,
 800             ip_table_id=0,
 801             is_ip4=1,
 802             create_vrf_if_needed=0,
 803             is_bind=1):
 804         """
 805         """
 806         return self.api(
 807             self.papi.mpls_ip_bind_unbind,
 808             {'mb_mpls_table_id': table_id,
 809              'mb_label': label,
 810              'mb_ip_table_id': ip_table_id,
 811              'mb_create_table_if_needed': create_vrf_if_needed,
 812              'mb_is_bind': is_bind,
 813              'mb_is_ip4': is_ip4,
 814              'mb_address_length': dst_address_length,
 815              'mb_address': dst_address})
 816
 817     def mpls_tunnel_add_del(
 818             self,
 819             tun_sw_if_index,
 820             next_hop_proto_is_ip4,
 821             next_hop_address,
 822             next_hop_sw_if_index=0xFFFFFFFF,
 823             next_hop_table_id=0,
 824             next_hop_weight=1,
 825             next_hop_n_out_labels=0,
 826             next_hop_out_label_stack=[],
 827             next_hop_via_label=MPLS_LABEL_INVALID,
 828             create_vrf_if_needed=0,
 829             is_add=1,
 830             l2_only=0):
 831         """
 832
 833         :param dst_address_length:
 834         :param next_hop_sw_if_index:  (Default value = 0xFFFFFFFF)
 835         :param dst_address:
 836         :param next_hop_address:
 837         :param next_hop_sw_if_index:  (Default value = 0xFFFFFFFF)
 838         :param vrf_id:  (Default value = 0)
 839         :param lookup_in_vrf:  (Default value = 0)
 840         :param classify_table_index:  (Default value = 0xFFFFFFFF)
 841         :param create_vrf_if_needed:  (Default value = 0)
 842         :param is_add:  (Default value = 1)
 843         :param is_drop:  (Default value = 0)
 844         :param is_ipv6:  (Default value = 0)
 845         :param is_local:  (Default value = 0)
 846         :param is_classify:  (Default value = 0)
 847         :param is_multipath:  (Default value = 0)
 848         :param is_resolve_host:  (Default value = 0)
 849         :param is_resolve_attached:  (Default value = 0)
 850         :param not_last:  (Default value = 0)
 851         :param next_hop_weight:  (Default value = 1)
 852
 853         """
 854         return self.api(
 855             self.papi.mpls_tunnel_add_del,
 856             {'mt_sw_if_index': tun_sw_if_index,
 857              'mt_is_add': is_add,
 858              'mt_l2_only': l2_only,
 859              'mt_next_hop_proto_is_ip4': next_hop_proto_is_ip4,
 860              'mt_next_hop_weight': next_hop_weight,
 861              'mt_next_hop': next_hop_address,
 862              'mt_next_hop_n_out_labels': next_hop_n_out_labels,
 863              'mt_next_hop_sw_if_index': next_hop_sw_if_index,
 864              'mt_next_hop_table_id': next_hop_table_id,
 865              'mt_next_hop_out_label_stack': next_hop_out_label_stack})
 866
 867     def snat_interface_add_del_feature(
 868             self,
 869             sw_if_index,
 870             is_inside=1,
 871             is_add=1):
 872         """Enable/disable S-NAT feature on the interface
 873
 874         :param sw_if_index: Software index of the interface
 875         :param is_inside: 1 if inside, 0 if outside (Default value = 1)
 876         :param is_add: 1 if add, 0 if delete (Default value = 1)
 877         """
 878         return self.api(
 879             self.papi.snat_interface_add_del_feature,
 880             {'is_add': is_add,
 881              'is_inside': is_inside,
 882              'sw_if_index': sw_if_index})
 883
 884     def snat_add_static_mapping(
 885             self,
 886             local_ip,
 887             external_ip=0,
 888             external_sw_if_index=0xFFFFFFFF,
 889             local_port=0,
 890             external_port=0,
 891             addr_only=1,
 892             vrf_id=0,
 893             protocol=0,
 894             is_add=1,
 895             is_ip4=1):
 896         """Add/delete S-NAT static mapping
 897
 898         :param local_ip: Local IP address
 899         :param external_ip: External IP address
 900         :param external_sw_if_index: External interface instead of IP address
 901         :param local_port: Local port number (Default value = 0)
 902         :param external_port: External port number (Default value = 0)
 903         :param addr_only: 1 if address only mapping, 0 if address and port
 904         :param vrf_id: VRF ID
 905         :param protocol: IP protocol (Default value = 0)
 906         :param is_add: 1 if add, 0 if delete (Default value = 1)
 907         :param is_ip4: 1 if address type is IPv4 (Default value = 1)
 908         """
 909         return self.api(
 910             self.papi.snat_add_static_mapping,
 911             {'is_add': is_add,
 912              'is_ip4': is_ip4,
 913              'addr_only': addr_only,
 914              'local_ip_address': local_ip,
 915              'external_ip_address': external_ip,
 916              'local_port': local_port,
 917              'external_port': external_port,
 918              'external_sw_if_index': external_sw_if_index,
 919              'vrf_id': vrf_id,
 920              'protocol': protocol})
 921
 922     def snat_add_address_range(
 923             self,
 924             first_ip_address,
 925             last_ip_address,
 926             is_add=1,
 927             is_ip4=1):
 928         """Add/del S-NAT address range
 929
 930         :param first_ip_address: First IP address
 931         :param last_ip_address: Last IP address
 932         :param is_add: 1 if add, 0 if delete (Default value = 1)
 933         :param is_ip4: 1 if address type is IPv4 (Default value = 1)
 934         """
 935         return self.api(
 936             self.papi.snat_add_address_range,
 937             {'is_ip4': is_ip4,
 938              'first_ip_address': first_ip_address,
 939              'last_ip_address': last_ip_address,
 940              'is_add': is_add})
 941
 942     def snat_address_dump(self):
 943         """Dump S-NAT addresses
 944         :return: Dictionary of S-NAT addresses
 945         """
 946         return self.api(self.papi.snat_address_dump, {})
 947
 948     def snat_interface_dump(self):
 949         """Dump interfaces with S-NAT feature
 950         :return: Dictionary of interfaces with S-NAT feature
 951         """
 952         return self.api(self.papi.snat_interface_dump, {})
 953
 954     def snat_static_mapping_dump(self):
 955         """Dump S-NAT static mappings
 956         :return: Dictionary of S-NAT static mappings
 957         """
 958         return self.api(self.papi.snat_static_mapping_dump, {})
 959
 960     def snat_show_config(self):
 961         """Show S-NAT config
 962         :return: S-NAT config parameters
 963         """
 964         return self.api(self.papi.snat_show_config, {})
 965
 966     def snat_add_interface_addr(
 967             self,
 968             sw_if_index,
 969             is_add=1):
 970         """Add/del S-NAT address from interface
 971
 972         :param sw_if_index: Software index of the interface
 973         :param is_add: 1 if add, 0 if delete (Default value = 1)
 974         """
 975         return self.api(self.papi.snat_add_del_interface_addr,
 976                         {'is_add': is_add, 'sw_if_index': sw_if_index})
 977
 978     def snat_interface_addr_dump(self):
 979         """Dump S-NAT addresses interfaces
 980         :return: Dictionary of S-NAT addresses interfaces
 981         """
 982         return self.api(self.papi.snat_interface_addr_dump, {})
 983
 984     def snat_ipfix(
 985             self,
 986             domain_id=1,
 987             src_port=4739,
 988             enable=1):
 989         """Enable/disable S-NAT IPFIX logging
 990
 991         :param domain_id: Observation domain ID (Default value = 1)
 992         :param src_port: Source port number (Default value = 4739)
 993         :param enable: 1 if enable, 0 if disable (Default value = 1)
 994         """
 995         return self.api(
 996             self.papi.snat_ipfix_enable_disable,
 997             {'domain_id': domain_id,
 998              'src_port': src_port,
 999              'enable': enable})
1000
1001     def control_ping(self):
1002         self.api(self.papi.control_ping)
1003
1004     def bfd_udp_add(self, sw_if_index, desired_min_tx, required_min_rx,
1005                     detect_mult, local_addr, peer_addr, is_ipv6=0,
1006                     bfd_key_id=None, conf_key_id=None):
1007         if bfd_key_id is None:
1008             return self.api(self.papi.bfd_udp_add,
1009                             {
1010                                 'sw_if_index': sw_if_index,
1011                                 'desired_min_tx': desired_min_tx,
1012                                 'required_min_rx': required_min_rx,
1013                                 'local_addr': local_addr,
1014                                 'peer_addr': peer_addr,
1015                                 'is_ipv6': is_ipv6,
1016                                 'detect_mult': detect_mult,
1017                             })
1018         else:
1019             return self.api(self.papi.bfd_udp_add,
1020                             {
1021                                 'sw_if_index': sw_if_index,
1022                                 'desired_min_tx': desired_min_tx,
1023                                 'required_min_rx': required_min_rx,
1024                                 'local_addr': local_addr,
1025                                 'peer_addr': peer_addr,
1026                                 'is_ipv6': is_ipv6,
1027                                 'detect_mult': detect_mult,
1028                                 'is_authenticated': 1,
1029                                 'bfd_key_id': bfd_key_id,
1030                                 'conf_key_id': conf_key_id,
1031                             })
1032
1033     def bfd_udp_mod(self, sw_if_index, desired_min_tx, required_min_rx,
1034                     detect_mult, local_addr, peer_addr, is_ipv6=0):
1035         return self.api(self.papi.bfd_udp_mod,
1036                         {
1037                             'sw_if_index': sw_if_index,
1038                             'desired_min_tx': desired_min_tx,
1039                             'required_min_rx': required_min_rx,
1040                             'local_addr': local_addr,
1041                             'peer_addr': peer_addr,
1042                             'is_ipv6': is_ipv6,
1043                             'detect_mult': detect_mult,
1044                         })
1045
1046     def bfd_udp_auth_activate(self, sw_if_index, local_addr, peer_addr,
1047                               is_ipv6=0, bfd_key_id=None, conf_key_id=None,
1048                               is_delayed=False):
1049         return self.api(self.papi.bfd_udp_auth_activate,
1050                         {
1051                             'sw_if_index': sw_if_index,
1052                             'local_addr': local_addr,
1053                             'peer_addr': peer_addr,
1054                             'is_ipv6': is_ipv6,
1055                             'is_delayed': 1 if is_delayed else 0,
1056                             'bfd_key_id': bfd_key_id,
1057                             'conf_key_id': conf_key_id,
1058                         })
1059
1060     def bfd_udp_auth_deactivate(self, sw_if_index, local_addr, peer_addr,
1061                                 is_ipv6=0, is_delayed=False):
1062         return self.api(self.papi.bfd_udp_auth_deactivate,
1063                         {
1064                             'sw_if_index': sw_if_index,
1065                             'local_addr': local_addr,
1066                             'peer_addr': peer_addr,
1067                             'is_ipv6': is_ipv6,
1068                             'is_delayed': 1 if is_delayed else 0,
1069                         })
1070
1071     def bfd_udp_del(self, sw_if_index, local_addr, peer_addr, is_ipv6=0):
1072         return self.api(self.papi.bfd_udp_del,
1073                         {
1074                             'sw_if_index': sw_if_index,
1075                             'local_addr': local_addr,
1076                             'peer_addr': peer_addr,
1077                             'is_ipv6': is_ipv6,
1078                         })
1079
1080     def bfd_udp_session_dump(self):
1081         return self.api(self.papi.bfd_udp_session_dump, {})
1082
1083     def bfd_udp_session_set_flags(self, admin_up_down, sw_if_index, local_addr,
1084                                   peer_addr, is_ipv6=0):
1085         return self.api(self.papi.bfd_udp_session_set_flags, {
1086             'admin_up_down': admin_up_down,
1087             'sw_if_index': sw_if_index,
1088             'local_addr': local_addr,
1089             'peer_addr': peer_addr,
1090             'is_ipv6': is_ipv6,
1091         })
1092
1093     def want_bfd_events(self, enable_disable=1):
1094         return self.api(self.papi.want_bfd_events, {
1095             'enable_disable': enable_disable,
1096             'pid': os.getpid(),
1097         })
1098
1099     def bfd_auth_set_key(self, conf_key_id, auth_type, key):
1100         return self.api(self.papi.bfd_auth_set_key, {
1101             'conf_key_id': conf_key_id,
1102             'auth_type': auth_type,
1103             'key': key,
1104             'key_len': len(key),
1105         })
1106
1107     def bfd_auth_del_key(self, conf_key_id):
1108         return self.api(self.papi.bfd_auth_del_key, {
1109             'conf_key_id': conf_key_id,
1110         })
1111
1112     def bfd_auth_keys_dump(self):
1113         return self.api(self.papi.bfd_auth_keys_dump, {})
1114
1115     def classify_add_del_table(
1116             self,
1117             is_add,
1118             mask,
1119             match_n_vectors=1,
1120             table_index=0xFFFFFFFF,
1121             nbuckets=2,
1122             memory_size=2097152,
1123             skip_n_vectors=0,
1124             next_table_index=0xFFFFFFFF,
1125             miss_next_index=0xFFFFFFFF,
1126             current_data_flag=0,
1127             current_data_offset=0):
1128         """
1129         :param is_add:
1130         :param mask:
1131         :param match_n_vectors: (Default value = 1)
1132         :param table_index: (Default value = 0xFFFFFFFF)
1133         :param nbuckets:  (Default value = 2)
1134         :param memory_size:  (Default value = 2097152)
1135         :param skip_n_vectors:  (Default value = 0)
1136         :param next_table_index:  (Default value = 0xFFFFFFFF)
1137         :param miss_next_index:  (Default value = 0xFFFFFFFF)
1138         :param current_data_flag:  (Default value = 0)
1139         :param current_data_offset:  (Default value = 0)
1140         """
1141
1142         return self.api(
1143             self.papi.classify_add_del_table,
1144             {'is_add': is_add,
1145              'table_index': table_index,
1146              'nbuckets': nbuckets,
1147              'memory_size': memory_size,
1148              'skip_n_vectors': skip_n_vectors,
1149              'match_n_vectors': match_n_vectors,
1150              'next_table_index': next_table_index,
1151              'miss_next_index': miss_next_index,
1152              'current_data_flag': current_data_flag,
1153              'current_data_offset': current_data_offset,
1154              'mask': mask})
1155
1156     def classify_add_del_session(
1157             self,
1158             is_add,
1159             table_index,
1160             match,
1161             opaque_index=0xFFFFFFFF,
1162             hit_next_index=0xFFFFFFFF,
1163             advance=0,
1164             action=0,
1165             metadata=0):
1166         """
1167         :param is_add:
1168         :param table_index:
1169         :param match:
1170         :param opaque_index:  (Default value = 0xFFFFFFFF)
1171         :param hit_next_index:  (Default value = 0xFFFFFFFF)
1172         :param advance:  (Default value = 0)
1173         :param action:  (Default value = 0)
1174         :param metadata:  (Default value = 0)
1175         """
1176
1177         return self.api(
1178             self.papi.classify_add_del_session,
1179             {'is_add': is_add,
1180              'table_index': table_index,
1181              'hit_next_index': hit_next_index,
1182              'opaque_index': opaque_index,
1183              'advance': advance,
1184              'action': action,
1185              'metadata': metadata,
1186              'match': match})
1187
1188     def input_acl_set_interface(
1189             self,
1190             is_add,
1191             sw_if_index,
1192             ip4_table_index=0xFFFFFFFF,
1193             ip6_table_index=0xFFFFFFFF,
1194             l2_table_index=0xFFFFFFFF):
1195         """
1196         :param is_add:
1197         :param sw_if_index:
1198         :param ip4_table_index:  (Default value = 0xFFFFFFFF)
1199         :param ip6_table_index:  (Default value = 0xFFFFFFFF)
1200         :param l2_table_index:  (Default value = 0xFFFFFFFF)
1201         """
1202
1203         return self.api(
1204             self.papi.input_acl_set_interface,
1205             {'sw_if_index': sw_if_index,
1206              'ip4_table_index': ip4_table_index,
1207              'ip6_table_index': ip6_table_index,
1208              'l2_table_index': l2_table_index,
1209              'is_add': is_add})
1210
1211     def set_ipfix_exporter(
1212             self,
1213             collector_address,
1214             src_address,
1215             path_mtu,
1216             template_interval,
1217             vrf_id=0,
1218             collector_port=4739,
1219             udp_checksum=0):
1220         return self.api(
1221             self.papi.set_ipfix_exporter,
1222             {
1223                 'collector_address': collector_address,
1224                 'collector_port': collector_port,
1225                 'src_address': src_address,
1226                 'vrf_id': vrf_id,
1227                 'path_mtu': path_mtu,
1228                 'template_interval': template_interval,
1229                 'udp_checksum': udp_checksum,
1230             })
1231
1232     def dhcp_proxy_config(self,
1233                           dhcp_server,
1234                           dhcp_src_address,
1235                           rx_table_id=0,
1236                           server_table_id=0,
1237                           is_add=1,
1238                           is_ipv6=0,
1239                           insert_circuit_id=0):
1240         return self.api(
1241             self.papi.dhcp_proxy_config_2,
1242             {
1243                 'rx_vrf_id': rx_table_id,
1244                 'server_vrf_id': server_table_id,
1245                 'is_ipv6': is_ipv6,
1246                 'is_add': is_add,
1247                 'insert_circuit_id': insert_circuit_id,
1248                 'dhcp_server': dhcp_server,
1249                 'dhcp_src_address': dhcp_src_address,
1250             })
1251
1252     def dhcp_proxy_set_vss(self,
1253                            table_id,
1254                            fib_id,
1255                            oui,
1256                            is_add=1,
1257                            is_ip6=0):
1258         return self.api(
1259             self.papi.dhcp_proxy_set_vss,
1260             {
1261                 'tbl_id': table_id,
1262                 'fib_id': fib_id,
1263                 'is_ipv6': is_ip6,
1264                 'is_add': is_add,
1265                 'oui': oui,
1266             })
1267
1268     def ip_mroute_add_del(self,
1269                           src_address,
1270                           grp_address,
1271                           grp_address_length,
1272                           e_flags,
1273                           next_hop_sw_if_index,
1274                           i_flags,
1275                           table_id=0,
1276                           create_vrf_if_needed=0,
1277                           is_add=1,
1278                           is_ipv6=0,
1279                           is_local=0):
1280         """
1281         """
1282         return self.api(
1283             self.papi.ip_mroute_add_del,
1284             {'next_hop_sw_if_index': next_hop_sw_if_index,
1285              'entry_flags': e_flags,
1286              'itf_flags': i_flags,
1287              'create_vrf_if_needed': create_vrf_if_needed,
1288              'is_add': is_add,
1289              'is_ipv6': is_ipv6,
1290              'is_local': is_local,
1291              'grp_address_length': grp_address_length,
1292              'grp_address': grp_address,
1293              'src_address': src_address})
1294
1295     def mfib_signal_dump(self):
1296         return self.api(self.papi.mfib_signal_dump, {})
1297
1298     def ip_mfib_dump(self):
1299         return self.api(self.papi.ip_mfib_dump, {})