Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 1 | # NB NB NB NB NB NB NB NB NB NB NB |
| 2 | # |
| 3 | # NOTE: The API binary wrappers in this file are in the process of being |
| 4 | # deprecated. DO NOT ADD NEW WRAPPERS HERE. Call the functions using |
| 5 | # named arguments directly instead. |
| 6 | # |
| 7 | |
Paul Vinciguerra | 00671cf | 2018-11-25 12:47:04 -0800 | [diff] [blame] | 8 | import os |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 9 | import time |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 10 | from collections import deque |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 11 | import queue |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 12 | from six import moves, iteritems |
Paul Vinciguerra | 1954229 | 2019-03-17 17:34:46 -0700 | [diff] [blame] | 13 | from vpp_papi import VPPApiClient, mac_pton |
Paul Vinciguerra | 00671cf | 2018-11-25 12:47:04 -0800 | [diff] [blame] | 14 | from hook import Hook |
Paul Vinciguerra | b5f15bd | 2019-03-17 08:42:58 -0700 | [diff] [blame] | 15 | from vpp_ip_route import MPLS_IETF_MAX_LABEL, MPLS_LABEL_INVALID |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 16 | |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 17 | # |
| 18 | # Dictionary keyed on message name to override default values for |
| 19 | # named parameters |
| 20 | # |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 21 | defaultmapping = { |
Paul Vinciguerra | c0e9441 | 2020-04-28 01:12:04 -0400 | [diff] [blame] | 22 | |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 23 | 'acl_interface_add_del': {'is_add': 1, 'is_input': 1}, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 24 | 'bd_ip_mac_add_del': {'is_add': 1, }, |
Jakub Grajciar | 4682feb | 2019-09-02 13:28:52 +0200 | [diff] [blame] | 25 | 'bfd_udp_add': {'is_authenticated': False, 'bfd_key_id': None, |
| 26 | 'conf_key_id': None}, |
| 27 | 'bfd_udp_auth_activate': {'bfd_key_id': None, 'conf_key_id': None, |
| 28 | 'is_delayed': False}, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 29 | 'bier_disp_entry_add_del': {'next_hop_rpf_id': -1, 'next_hop_is_ip4': 1, |
| 30 | 'is_add': 1, }, |
| 31 | 'bier_disp_table_add_del': {'is_add': 1, }, |
| 32 | 'bier_imp_add': {'is_add': 1, }, |
| 33 | 'bier_route_add_del': {'is_add': 1, }, |
| 34 | 'bier_table_add_del': {'is_add': 1, }, |
| 35 | 'bridge_domain_add_del': {'flood': 1, 'uu_flood': 1, 'forward': 1, |
| 36 | 'learn': 1, 'is_add': 1, }, |
Paul Vinciguerra | 9a29f79 | 2019-04-30 20:44:25 -0700 | [diff] [blame] | 37 | 'bvi_delete': {}, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 38 | 'geneve_add_del_tunnel': {'mcast_sw_if_index': 4294967295, 'is_add': 1, |
| 39 | 'decap_next_index': 4294967295, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 40 | 'input_acl_set_interface': {'ip4_table_index': 4294967295, |
| 41 | 'ip6_table_index': 4294967295, |
| 42 | 'l2_table_index': 4294967295, }, |
| 43 | 'ip6_add_del_address_using_prefix': {'is_add': 1, }, |
| 44 | 'ip6nd_send_router_solicitation': {'irt': 1, 'mrt': 120, }, |
| 45 | 'ip_add_del_route': {'next_hop_sw_if_index': 4294967295, |
| 46 | 'next_hop_weight': 1, 'next_hop_via_label': 1048576, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 47 | 'classify_table_index': 4294967295, 'is_add': 1, }, |
| 48 | 'ip_mroute_add_del': {'is_add': 1, }, |
| 49 | 'ip_neighbor_add_del': {'is_add': 1, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 50 | 'ipsec_interface_add_del_spd': {'is_add': 1, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 51 | 'ipsec_spd_add_del': {'is_add': 1, }, |
| 52 | 'ipsec_spd_dump': {'sa_id': 4294967295, }, |
| 53 | 'ipsec_spd_entry_add_del': {'local_port_stop': 65535, |
| 54 | 'remote_port_stop': 65535, 'priority': 100, |
| 55 | 'is_outbound': 1, |
| 56 | 'is_add': 1, }, |
| 57 | 'ipsec_tunnel_if_add_del': {'is_add': 1, 'anti_replay': 1, }, |
| 58 | 'l2_emulation': {'enable': 1, }, |
| 59 | 'l2fib_add_del': {'is_add': 1, }, |
| 60 | 'lisp_add_del_adjacency': {'is_add': 1, }, |
| 61 | 'lisp_add_del_local_eid': {'is_add': 1, }, |
| 62 | 'lisp_add_del_locator': {'priority': 1, 'weight': 1, 'is_add': 1, }, |
| 63 | 'lisp_add_del_locator_set': {'is_add': 1, }, |
| 64 | 'lisp_add_del_remote_mapping': {'is_add': 1, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 65 | 'macip_acl_interface_add_del': {'is_add': 1, }, |
| 66 | 'mpls_ip_bind_unbind': {'is_ip4': 1, 'is_bind': 1, }, |
Ole Troan | a5b2eec | 2019-03-11 19:23:25 +0100 | [diff] [blame] | 67 | 'mpls_route_add_del': {'mr_next_hop_sw_if_index': 4294967295, |
| 68 | 'mr_next_hop_weight': 1, |
| 69 | 'mr_next_hop_via_label': 1048576, |
| 70 | 'mr_is_add': 1, |
| 71 | 'mr_classify_table_index': 4294967295, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 72 | 'mpls_table_add_del': {'is_add': 1, }, |
| 73 | 'mpls_tunnel_add_del': {'next_hop_sw_if_index': 4294967295, |
| 74 | 'next_hop_weight': 1, |
| 75 | 'next_hop_via_label': 1048576, |
| 76 | 'is_add': 1, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 77 | 'output_acl_set_interface': {'ip4_table_index': 4294967295, |
| 78 | 'ip6_table_index': 4294967295, |
| 79 | 'l2_table_index': 4294967295, }, |
| 80 | 'pppoe_add_del_session': {'is_add': 1, }, |
Jakub Grajciar | cd01fb4 | 2020-03-02 13:16:53 +0100 | [diff] [blame] | 81 | 'policer_add_del': {'is_add': 1, 'conform_action': {'type': 1}, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 82 | 'set_ipfix_exporter': {'collector_port': 4739, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 83 | 'sr_policy_add': {'weight': 1, 'is_encap': 1, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 84 | 'sw_interface_add_del_address': {'is_add': 1, }, |
| 85 | 'sw_interface_ip6nd_ra_prefix': {'val_lifetime': 4294967295, |
| 86 | 'pref_lifetime': 4294967295, }, |
| 87 | 'sw_interface_set_ip_directed_broadcast': {'enable': 1, }, |
| 88 | 'sw_interface_set_l2_bridge': {'enable': 1, }, |
| 89 | 'sw_interface_set_mpls_enable': {'enable': 1, }, |
| 90 | 'sw_interface_set_mtu': {'mtu': [0, 0, 0, 0], }, |
| 91 | 'sw_interface_set_unnumbered': {'is_add': 1, }, |
| 92 | 'sw_interface_span_enable_disable': {'state': 1, }, |
| 93 | 'vxlan_add_del_tunnel': {'mcast_sw_if_index': 4294967295, 'is_add': 1, |
| 94 | 'decap_next_index': 4294967295, |
| 95 | 'instance': 4294967295, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 96 | 'want_bfd_events': {'enable_disable': 1, }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 97 | 'want_igmp_events': {'enable': 1, }, |
| 98 | 'want_interface_events': {'enable_disable': 1, }, |
Jakub Grajciar | 145e330 | 2019-10-24 13:52:42 +0200 | [diff] [blame] | 99 | 'want_l2_macs_events': {'enable_disable': 1, 'pid': os.getpid(), }, |
Jerome Tollet | 0f8d100 | 2021-01-07 12:44:17 +0100 | [diff] [blame] | 100 | 'want_l2_macs_events2': {'enable_disable': 1, 'pid': os.getpid(), }, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 101 | } |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 102 | |
Matus Fabian | b4515b4 | 2018-11-19 04:25:32 -0800 | [diff] [blame] | 103 | |
Paul Vinciguerra | 0b0a84c | 2020-05-02 10:40:33 -0400 | [diff] [blame] | 104 | def as_fn_signature(d): |
| 105 | return ", ".join(f"{k}={v}" for k, v in d.items()) |
| 106 | |
| 107 | |
Paul Vinciguerra | 9673e3e | 2019-05-10 20:41:08 -0400 | [diff] [blame] | 108 | class CliFailedCommandError(Exception): |
| 109 | """ cli command failed.""" |
| 110 | |
| 111 | |
| 112 | class CliSyntaxError(Exception): |
| 113 | """ cli command had a syntax error.""" |
| 114 | |
| 115 | |
Klement Sekera | 7388448 | 2017-02-23 09:26:30 +0100 | [diff] [blame] | 116 | class UnexpectedApiReturnValueError(Exception): |
| 117 | """ exception raised when the API return value is unexpected """ |
| 118 | pass |
| 119 | |
| 120 | |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 121 | class VppPapiProvider(object): |
| 122 | """VPP-api provider using vpp-papi |
| 123 | |
| 124 | @property hook: hook object providing before and after api/cli hooks |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 125 | """ |
| 126 | |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 127 | _zero, _negative = range(2) |
| 128 | |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 129 | def __init__(self, name, test_class, read_timeout): |
Paul Vinciguerra | 895e2f8 | 2019-01-08 20:37:40 -0800 | [diff] [blame] | 130 | self.hook = Hook(test_class) |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 131 | self.name = name |
Klement Sekera | 7bb873a | 2016-11-18 07:38:42 +0100 | [diff] [blame] | 132 | self.test_class = test_class |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 133 | self._expect_api_retval = self._zero |
| 134 | self._expect_stack = [] |
Ole Troan | 7e3a875 | 2016-12-05 10:27:09 +0100 | [diff] [blame] | 135 | |
Paul Vinciguerra | 1954229 | 2019-03-17 17:34:46 -0700 | [diff] [blame] | 136 | # install_dir is a class attribute. We need to set it before |
| 137 | # calling the constructor. |
| 138 | VPPApiClient.apidir = os.getenv('VPP_INSTALL_PATH') |
Paul Vinciguerra | 04575c5 | 2019-01-13 10:26:05 -0800 | [diff] [blame] | 139 | |
Paul Vinciguerra | 1954229 | 2019-03-17 17:34:46 -0700 | [diff] [blame] | 140 | self.vpp = VPPApiClient(logger=test_class.logger, |
| 141 | read_timeout=read_timeout, |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 142 | use_socket=True, |
Klement Sekera | e263685 | 2021-03-16 12:52:12 +0100 | [diff] [blame] | 143 | server_address=test_class.get_api_sock_path()) |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 144 | self._events = queue.Queue() |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 145 | |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 146 | def __enter__(self): |
| 147 | return self |
| 148 | |
Klement Sekera | 7d6afb3 | 2018-11-08 11:52:04 +0100 | [diff] [blame] | 149 | def assert_negative_api_retval(self): |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 150 | """ Expect API failure - used with with, e.g.:: |
| 151 | |
| 152 | with self.vapi.assert_negative_api_retval(): |
| 153 | self.vapi.<api call expected to fail> |
| 154 | |
| 155 | .. |
Klement Sekera | 7d6afb3 | 2018-11-08 11:52:04 +0100 | [diff] [blame] | 156 | """ |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 157 | self._expect_stack.append(self._expect_api_retval) |
| 158 | self._expect_api_retval = self._negative |
| 159 | return self |
| 160 | |
Klement Sekera | 7d6afb3 | 2018-11-08 11:52:04 +0100 | [diff] [blame] | 161 | def assert_zero_api_retval(self): |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 162 | """ Expect API success - used with with, e.g.:: |
Klement Sekera | 7d6afb3 | 2018-11-08 11:52:04 +0100 | [diff] [blame] | 163 | |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 164 | with self.vapi.assert_negative_api_retval(): |
| 165 | self.vapi.<api call expected to succeed> |
| 166 | |
| 167 | :note: this is useful only inside another with block |
| 168 | as success is the default expected value |
Klement Sekera | 7d6afb3 | 2018-11-08 11:52:04 +0100 | [diff] [blame] | 169 | """ |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 170 | self._expect_stack.append(self._expect_api_retval) |
| 171 | self._expect_api_retval = self._zero |
| 172 | return self |
| 173 | |
| 174 | def __exit__(self, exc_type, exc_value, traceback): |
| 175 | self._expect_api_retval = self._expect_stack.pop() |
| 176 | |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 177 | def register_hook(self, hook): |
| 178 | """Replace hook registration with new hook |
| 179 | |
| 180 | :param hook: |
| 181 | |
| 182 | """ |
| 183 | self.hook = hook |
| 184 | |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 185 | def collect_events(self): |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 186 | """ Collect all events from the internal queue and clear the queue. """ |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 187 | result = [] |
| 188 | while True: |
| 189 | try: |
| 190 | e = self._events.get(block=False) |
| 191 | result.append(e) |
| 192 | except queue.Empty: |
| 193 | return result |
| 194 | return result |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 195 | |
| 196 | def wait_for_event(self, timeout, name=None): |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 197 | """ Wait for and return next event. """ |
Klement Sekera | acb9b8e | 2017-02-14 02:55:31 +0100 | [diff] [blame] | 198 | if name: |
Klement Sekera | acb9b8e | 2017-02-14 02:55:31 +0100 | [diff] [blame] | 199 | self.test_class.logger.debug("Expecting event '%s' within %ss", |
| 200 | name, timeout) |
Eyal Bari | 2019748 | 2017-09-13 12:29:08 +0300 | [diff] [blame] | 201 | else: |
| 202 | self.test_class.logger.debug("Expecting event within %ss", |
| 203 | timeout) |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 204 | try: |
| 205 | e = self._events.get(timeout=timeout) |
| 206 | except queue.Empty: |
| 207 | raise Exception("Event did not occur within timeout") |
| 208 | msgname = type(e).__name__ |
| 209 | if name and msgname != name: |
| 210 | raise Exception("Unexpected event received: %s, expected: %s" |
| 211 | % msgname) |
| 212 | self.test_class.logger.debug("Returning event %s:%s" % (name, e)) |
| 213 | return e |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 214 | |
| 215 | def __call__(self, name, event): |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 216 | """ Enqueue event in the internal event queue. """ |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 217 | self.test_class.logger.debug("New event: %s: %s" % (name, event)) |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 218 | self._events.put(event) |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 219 | |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 220 | def factory(self, name, apifn): |
| 221 | def f(*a, **ka): |
| 222 | fields = apifn._func.msg.fields |
| 223 | |
| 224 | # add positional and kw arguments |
| 225 | d = ka |
| 226 | for i, o in enumerate(fields[3:]): |
| 227 | try: |
| 228 | d[o] = a[i] |
Jakub Grajciar | 4682feb | 2019-09-02 13:28:52 +0200 | [diff] [blame] | 229 | except BaseException: |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 230 | break |
| 231 | |
| 232 | # Default override |
| 233 | if name in defaultmapping: |
| 234 | for k, v in iteritems(defaultmapping[name]): |
| 235 | if k in d: |
| 236 | continue |
| 237 | d[k] = v |
| 238 | return self.api(apifn, d) |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 239 | |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 240 | return f |
| 241 | |
Paul Vinciguerra | 7ab9970 | 2019-03-05 04:30:04 -0800 | [diff] [blame] | 242 | def __getattribute__(self, name): |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 243 | try: |
Paul Vinciguerra | 7ab9970 | 2019-03-05 04:30:04 -0800 | [diff] [blame] | 244 | method = super(VppPapiProvider, self).__getattribute__(name) |
| 245 | except AttributeError: |
| 246 | method = self.factory(name, getattr(self.papi, name)) |
| 247 | # lazily load the method so we don't need to call factory |
| 248 | # again for this name. |
| 249 | setattr(self, name, method) |
| 250 | return method |
Ole Troan | f159f58 | 2019-02-28 20:20:47 +0100 | [diff] [blame] | 251 | |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 252 | def connect(self): |
| 253 | """Connect the API to VPP""" |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 254 | # This might be called before VPP is prepared to listen to the socket |
| 255 | retries = 0 |
Klement Sekera | e263685 | 2021-03-16 12:52:12 +0100 | [diff] [blame] | 256 | while not os.path.exists(self.test_class.get_api_sock_path()): |
Ole Troan | 4376ab2 | 2021-03-03 10:40:05 +0100 | [diff] [blame] | 257 | time.sleep(0.5) |
| 258 | retries += 1 |
| 259 | if retries > 120: |
| 260 | break |
| 261 | self.vpp.connect(self.name[:63]) |
Klement Sekera | 7112c54 | 2017-03-01 09:53:19 +0100 | [diff] [blame] | 262 | self.papi = self.vpp.api |
| 263 | self.vpp.register_event_callback(self) |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 264 | |
| 265 | def disconnect(self): |
| 266 | """Disconnect the API from VPP""" |
Klement Sekera | 7112c54 | 2017-03-01 09:53:19 +0100 | [diff] [blame] | 267 | self.vpp.disconnect() |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 268 | |
| 269 | def api(self, api_fn, api_args, expected_retval=0): |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 270 | """ Call API function and check it's return value. |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 271 | Call the appropriate hooks before and after the API call |
| 272 | |
| 273 | :param api_fn: API function to call |
| 274 | :param api_args: tuple of API function arguments |
| 275 | :param expected_retval: Expected return value (Default value = 0) |
| 276 | :returns: reply from the API |
| 277 | |
| 278 | """ |
| 279 | self.hook.before_api(api_fn.__name__, api_args) |
Ole Troan | 7e3a875 | 2016-12-05 10:27:09 +0100 | [diff] [blame] | 280 | reply = api_fn(**api_args) |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 281 | if self._expect_api_retval == self._negative: |
| 282 | if hasattr(reply, 'retval') and reply.retval >= 0: |
Paul Vinciguerra | 0b0a84c | 2020-05-02 10:40:33 -0400 | [diff] [blame] | 283 | msg = "%s(%s) passed unexpectedly: expected negative " \ |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 284 | "return value instead of %d in %s" % \ |
Paul Vinciguerra | 0b0a84c | 2020-05-02 10:40:33 -0400 | [diff] [blame] | 285 | (api_fn.__name__, as_fn_signature(api_args), |
| 286 | reply.retval, |
| 287 | moves.reprlib.repr(reply)) |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 288 | self.test_class.logger.info(msg) |
Klement Sekera | 7388448 | 2017-02-23 09:26:30 +0100 | [diff] [blame] | 289 | raise UnexpectedApiReturnValueError(msg) |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 290 | elif self._expect_api_retval == self._zero: |
| 291 | if hasattr(reply, 'retval') and reply.retval != expected_retval: |
Paul Vinciguerra | 0b0a84c | 2020-05-02 10:40:33 -0400 | [diff] [blame] | 292 | msg = "%s(%s) failed, expected %d return value instead " \ |
| 293 | "of %d in %s" % (api_fn.__name__, |
| 294 | as_fn_signature(api_args), |
| 295 | expected_retval, reply.retval, |
Paul Vinciguerra | 1b534f5 | 2019-06-15 20:31:31 -0400 | [diff] [blame] | 296 | repr(reply)) |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 297 | self.test_class.logger.info(msg) |
Klement Sekera | 7388448 | 2017-02-23 09:26:30 +0100 | [diff] [blame] | 298 | raise UnexpectedApiReturnValueError(msg) |
Klement Sekera | e0545ef | 2017-01-25 08:00:40 +0100 | [diff] [blame] | 299 | else: |
| 300 | raise Exception("Internal error, unexpected value for " |
| 301 | "self._expect_api_retval %s" % |
| 302 | self._expect_api_retval) |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 303 | self.hook.after_api(api_fn.__name__, api_args) |
| 304 | return reply |
| 305 | |
Dave Barach | 5932ce1 | 2019-10-09 17:28:04 -0400 | [diff] [blame] | 306 | def cli_return_response(self, cli): |
| 307 | """ Execute a CLI, calling the before/after hooks appropriately. |
| 308 | Return the reply without examining it |
| 309 | |
| 310 | :param cli: CLI to execute |
| 311 | :returns: response object |
| 312 | |
| 313 | """ |
| 314 | self.hook.before_cli(cli) |
| 315 | cli += '\n' |
| 316 | r = self.papi.cli_inband(cmd=cli) |
| 317 | self.hook.after_cli(cli) |
| 318 | return r |
| 319 | |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 320 | def cli(self, cli): |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 321 | """ Execute a CLI, calling the before/after hooks appropriately. |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 322 | |
| 323 | :param cli: CLI to execute |
| 324 | :returns: CLI output |
| 325 | |
| 326 | """ |
Dave Barach | 5932ce1 | 2019-10-09 17:28:04 -0400 | [diff] [blame] | 327 | r = self.cli_return_response(cli) |
Paul Vinciguerra | 9673e3e | 2019-05-10 20:41:08 -0400 | [diff] [blame] | 328 | if r.retval == -156: |
| 329 | raise CliSyntaxError(r.reply) |
| 330 | if r.retval != 0: |
| 331 | raise CliFailedCommandError(r.reply) |
Jan | 49c0fca | 2016-10-26 15:44:27 +0200 | [diff] [blame] | 332 | if hasattr(r, 'reply'): |
Ole Troan | 413f4a5 | 2018-11-28 11:36:05 +0100 | [diff] [blame] | 333 | return r.reply |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 334 | |
Jan | 49c0fca | 2016-10-26 15:44:27 +0200 | [diff] [blame] | 335 | def ppcli(self, cli): |
Klement Sekera | e4504c6 | 2016-12-08 10:16:41 +0100 | [diff] [blame] | 336 | """ Helper method to print CLI command in case of info logging level. |
Jan | 49c0fca | 2016-10-26 15:44:27 +0200 | [diff] [blame] | 337 | |
| 338 | :param cli: CLI to execute |
| 339 | :returns: CLI output |
| 340 | """ |
Ole Troan | 6ed154f | 2019-10-15 19:31:55 +0200 | [diff] [blame] | 341 | return cli + "\n" + self.cli(cli) |
Jan | 49c0fca | 2016-10-26 15:44:27 +0200 | [diff] [blame] | 342 | |
Juraj Sloboda | 4b9669d | 2018-01-15 10:39:21 +0100 | [diff] [blame] | 343 | def ip6nd_send_router_solicitation(self, sw_if_index, irt=1, mrt=120, |
| 344 | mrc=0, mrd=0): |
| 345 | return self.api(self.papi.ip6nd_send_router_solicitation, |
| 346 | {'irt': irt, |
| 347 | 'mrt': mrt, |
| 348 | 'mrc': mrc, |
| 349 | 'mrd': mrd, |
| 350 | 'sw_if_index': sw_if_index}) |
| 351 | |
Juraj Sloboda | b3f9050 | 2018-10-04 15:15:16 +0200 | [diff] [blame] | 352 | def want_interface_events(self, enable_disable=1): |
| 353 | return self.api(self.papi.want_interface_events, |
| 354 | {'enable_disable': enable_disable, |
| 355 | 'pid': os.getpid(), }) |
| 356 | |
Neale Ranns | c819fc6 | 2018-02-16 02:44:05 -0800 | [diff] [blame] | 357 | def sw_interface_set_mac_address(self, sw_if_index, mac): |
| 358 | return self.api(self.papi.sw_interface_set_mac_address, |
| 359 | {'sw_if_index': sw_if_index, |
| 360 | 'mac_address': mac}) |
| 361 | |
Ole Troan | e1ade68 | 2019-03-04 23:55:43 +0100 | [diff] [blame] | 362 | def p2p_ethernet_add(self, sw_if_index, remote_mac, subif_id): |
Pavel Kotucek | 15ac81c | 2017-06-20 14:00:26 +0200 | [diff] [blame] | 363 | """Create p2p ethernet subinterface |
| 364 | |
| 365 | :param sw_if_index: main (parent) interface |
| 366 | :param remote_mac: client (remote) mac address |
| 367 | |
| 368 | """ |
| 369 | return self.api( |
| 370 | self.papi.p2p_ethernet_add, |
| 371 | {'parent_if_index': sw_if_index, |
| 372 | 'remote_mac': remote_mac, |
| 373 | 'subif_id': subif_id}) |
| 374 | |
Ole Troan | e1ade68 | 2019-03-04 23:55:43 +0100 | [diff] [blame] | 375 | def p2p_ethernet_del(self, sw_if_index, remote_mac): |
Pavel Kotucek | 15ac81c | 2017-06-20 14:00:26 +0200 | [diff] [blame] | 376 | """Delete p2p ethernet subinterface |
| 377 | |
| 378 | :param sw_if_index: main (parent) interface |
| 379 | :param remote_mac: client (remote) mac address |
| 380 | |
| 381 | """ |
| 382 | return self.api( |
| 383 | self.papi.p2p_ethernet_del, |
| 384 | {'parent_if_index': sw_if_index, |
| 385 | 'remote_mac': remote_mac}) |
| 386 | |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 387 | def create_vlan_subif(self, sw_if_index, vlan): |
| 388 | """ |
| 389 | |
| 390 | :param vlan: |
| 391 | :param sw_if_index: |
| 392 | |
| 393 | """ |
Ole Troan | 7e3a875 | 2016-12-05 10:27:09 +0100 | [diff] [blame] | 394 | return self.api(self.papi.create_vlan_subif, |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 395 | {'sw_if_index': sw_if_index, |
| 396 | 'vlan_id': vlan}) |
Klement Sekera | f62ae12 | 2016-10-11 11:47:09 +0200 | [diff] [blame] | 397 | |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 398 | def create_loopback(self, mac=''): |
| 399 | """ |
| 400 | |
| 401 | :param mac: (Optional) |
| 402 | """ |
Ole Troan | 7e3a875 | 2016-12-05 10:27:09 +0100 | [diff] [blame] | 403 | return self.api(self.papi.create_loopback, |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 404 | {'mac_address': mac}) |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 405 | |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 406 | def ip_route_dump(self, table_id, is_ip6=False): |
| 407 | return self.api(self.papi.ip_route_dump, |
| 408 | {'table': { |
| 409 | 'table_id': table_id, |
| 410 | 'is_ip6': is_ip6 |
| 411 | }}) |
| 412 | |
Neale Ranns | 976b259 | 2019-12-04 06:11:00 +0000 | [diff] [blame] | 413 | def ip_route_v2_dump(self, table_id, is_ip6=False, src=0): |
| 414 | return self.api(self.papi.ip_route_v2_dump, |
| 415 | { |
| 416 | 'src': src, |
| 417 | 'table': { |
| 418 | 'table_id': table_id, |
| 419 | 'is_ip6': is_ip6 |
| 420 | } |
| 421 | }) |
| 422 | |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 423 | def ip_neighbor_add_del(self, |
| 424 | sw_if_index, |
| 425 | mac_address, |
Neale Ranns | 3702930 | 2018-08-10 05:30:06 -0700 | [diff] [blame] | 426 | ip_address, |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 427 | is_add=1, |
Neale Ranns | 3702930 | 2018-08-10 05:30:06 -0700 | [diff] [blame] | 428 | flags=0): |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 429 | """ Add neighbor MAC to IPv4 or IPv6 address. |
| 430 | |
| 431 | :param sw_if_index: |
| 432 | :param mac_address: |
| 433 | :param dst_address: |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 434 | :param is_add: (Default value = 1) |
Neale Ranns | 3702930 | 2018-08-10 05:30:06 -0700 | [diff] [blame] | 435 | :param flags: (Default value = 0/NONE) |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 436 | """ |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 437 | return self.api( |
Ole Troan | 7e3a875 | 2016-12-05 10:27:09 +0100 | [diff] [blame] | 438 | self.papi.ip_neighbor_add_del, |
Neale Ranns | 3702930 | 2018-08-10 05:30:06 -0700 | [diff] [blame] | 439 | { |
| 440 | 'is_add': is_add, |
| 441 | 'neighbor': { |
| 442 | 'sw_if_index': sw_if_index, |
| 443 | 'flags': flags, |
| 444 | 'mac_address': mac_address, |
| 445 | 'ip_address': ip_address |
| 446 | } |
| 447 | } |
Matej Klotton | 0178d52 | 2016-11-04 11:11:44 +0100 | [diff] [blame] | 448 | ) |
Pavel Kotucek | f6e3dc4 | 2016-11-04 09:58:01 +0100 | [diff] [blame] | 449 | |
Neale Ranns | d0df49f | 2018-08-08 01:06:40 -0700 | [diff] [blame] | 450 | def udp_encap_add(self, |
Neale Ranns | d0df49f | 2018-08-08 01:06:40 -0700 | [diff] [blame] | 451 | src_ip, |
| 452 | dst_ip, |
| 453 | src_port, |
| 454 | dst_port, |
| 455 | table_id=0): |
Neale Ranns | 810086d | 2017-11-05 16:26:46 -0800 | [diff] [blame] | 456 | """ Add a GRE tunnel |
Neale Ranns | 810086d | 2017-11-05 16:26:46 -0800 | [diff] [blame] | 457 | :param src_ip: |
| 458 | :param dst_ip: |
| 459 | :param src_port: |
| 460 | :param dst_port: |
| 461 | :param outer_fib_id: (Default value = 0) |
Neale Ranns | 810086d | 2017-11-05 16:26:46 -0800 | [diff] [blame] | 462 | """ |
| 463 | |
| 464 | return self.api( |
Neale Ranns | d0df49f | 2018-08-08 01:06:40 -0700 | [diff] [blame] | 465 | self.papi.udp_encap_add, |
| 466 | { |
| 467 | 'udp_encap': { |
Neale Ranns | d0df49f | 2018-08-08 01:06:40 -0700 | [diff] [blame] | 468 | 'src_ip': src_ip, |
| 469 | 'dst_ip': dst_ip, |
| 470 | 'src_port': src_port, |
| 471 | 'dst_port': dst_port, |
| 472 | 'table_id': table_id |
| 473 | } |
| 474 | }) |
| 475 | |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 476 | def udp_encap_del(self, id): |
| 477 | return self.api(self.papi.udp_encap_del, {'id': id}) |
| 478 | |
| 479 | def udp_encap_dump(self): |
| 480 | return self.api(self.papi.udp_encap_dump, {}) |
| 481 | |
| 482 | def want_udp_encap_stats(self, enable=1): |
| 483 | return self.api(self.papi.want_udp_encap_stats, |
| 484 | {'enable': enable, |
| 485 | 'pid': os.getpid()}) |
| 486 | |
| 487 | def mpls_route_dump(self, table_id): |
| 488 | return self.api(self.papi.mpls_route_dump, |
| 489 | {'table': { |
| 490 | 'mt_table_id': table_id |
| 491 | }}) |
| 492 | |
| 493 | def mpls_table_dump(self): |
| 494 | return self.api(self.papi.mpls_table_dump, {}) |
| 495 | |
Neale Ranns | 1500254 | 2017-09-10 04:39:11 -0700 | [diff] [blame] | 496 | def mpls_table_add_del( |
| 497 | self, |
| 498 | table_id, |
| 499 | is_add=1): |
| 500 | """ |
| 501 | |
| 502 | :param table_id |
| 503 | :param is_add: (Default value = 1) |
| 504 | |
| 505 | """ |
| 506 | |
| 507 | return self.api( |
| 508 | self.papi.mpls_table_add_del, |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 509 | {'mt_table': |
| 510 | { |
| 511 | 'mt_table_id': table_id, |
| 512 | }, |
Neale Ranns | 1500254 | 2017-09-10 04:39:11 -0700 | [diff] [blame] | 513 | 'mt_is_add': is_add}) |
| 514 | |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 515 | def mpls_route_add_del(self, |
| 516 | table_id, |
| 517 | label, |
| 518 | eos, |
| 519 | eos_proto, |
| 520 | is_multicast, |
| 521 | paths, |
| 522 | is_add, |
| 523 | is_multipath): |
| 524 | """ MPLS Route add/del """ |
| 525 | return self.api( |
| 526 | self.papi.mpls_route_add_del, |
| 527 | {'mr_route': |
| 528 | { |
| 529 | 'mr_table_id': table_id, |
| 530 | 'mr_label': label, |
| 531 | 'mr_eos': eos, |
| 532 | 'mr_eos_proto': eos_proto, |
| 533 | 'mr_is_multicast': is_multicast, |
| 534 | 'mr_n_paths': len(paths), |
| 535 | 'mr_paths': paths, |
| 536 | }, |
| 537 | 'mr_is_add': is_add, |
| 538 | 'mr_is_multipath': is_multipath}) |
| 539 | |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 540 | def mpls_ip_bind_unbind( |
| 541 | self, |
| 542 | label, |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 543 | prefix, |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 544 | table_id=0, |
| 545 | ip_table_id=0, |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 546 | is_bind=1): |
| 547 | """ |
| 548 | """ |
| 549 | return self.api( |
Ole Troan | 7e3a875 | 2016-12-05 10:27:09 +0100 | [diff] [blame] | 550 | self.papi.mpls_ip_bind_unbind, |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 551 | {'mb_mpls_table_id': table_id, |
| 552 | 'mb_label': label, |
| 553 | 'mb_ip_table_id': ip_table_id, |
Klement Sekera | 0e3c0de | 2016-09-29 14:43:44 +0200 | [diff] [blame] | 554 | 'mb_is_bind': is_bind, |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 555 | 'mb_prefix': prefix}) |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 556 | |
| 557 | def mpls_tunnel_add_del( |
| 558 | self, |
| 559 | tun_sw_if_index, |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 560 | paths, |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 561 | is_add=1, |
Neale Ranns | 0f26c5a | 2017-03-01 15:12:11 -0800 | [diff] [blame] | 562 | l2_only=0, |
| 563 | is_multicast=0): |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 564 | """ |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 565 | """ |
Neale Ranns | ad422ed | 2016-11-02 14:20:04 +0000 | [diff] [blame] | 566 | return self.api( |
Ole Troan | 7e3a875 | 2016-12-05 10:27:09 +0100 | [diff] [blame] | 567 | self.papi.mpls_tunnel_add_del, |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 568 | {'mt_is_add': is_add, |
| 569 | 'mt_tunnel': |
| 570 | { |
| 571 | 'mt_sw_if_index': tun_sw_if_index, |
| 572 | 'mt_l2_only': l2_only, |
| 573 | 'mt_is_multicast': is_multicast, |
| 574 | 'mt_n_paths': len(paths), |
| 575 | 'mt_paths': paths, |
| 576 | }}) |
Matus Fabian | de88675 | 2016-12-07 03:38:19 -0800 | [diff] [blame] | 577 | |
Steve Shin | 7957d6e | 2016-12-19 09:24:50 -0800 | [diff] [blame] | 578 | def input_acl_set_interface( |
| 579 | self, |
| 580 | is_add, |
| 581 | sw_if_index, |
| 582 | ip4_table_index=0xFFFFFFFF, |
| 583 | ip6_table_index=0xFFFFFFFF, |
| 584 | l2_table_index=0xFFFFFFFF): |
| 585 | """ |
| 586 | :param is_add: |
| 587 | :param sw_if_index: |
| 588 | :param ip4_table_index: (Default value = 0xFFFFFFFF) |
| 589 | :param ip6_table_index: (Default value = 0xFFFFFFFF) |
| 590 | :param l2_table_index: (Default value = 0xFFFFFFFF) |
| 591 | """ |
| 592 | |
| 593 | return self.api( |
| 594 | self.papi.input_acl_set_interface, |
Klement Sekera | 77fabdb | 2017-01-02 07:46:14 +0100 | [diff] [blame] | 595 | {'sw_if_index': sw_if_index, |
| 596 | 'ip4_table_index': ip4_table_index, |
| 597 | 'ip6_table_index': ip6_table_index, |
| 598 | 'l2_table_index': l2_table_index, |
| 599 | 'is_add': is_add}) |
| 600 | |
Andrew Yourtchenko | 815d7d5 | 2018-02-07 11:37:02 +0100 | [diff] [blame] | 601 | def output_acl_set_interface( |
| 602 | self, |
| 603 | is_add, |
| 604 | sw_if_index, |
| 605 | ip4_table_index=0xFFFFFFFF, |
| 606 | ip6_table_index=0xFFFFFFFF, |
| 607 | l2_table_index=0xFFFFFFFF): |
| 608 | """ |
| 609 | :param is_add: |
| 610 | :param sw_if_index: |
| 611 | :param ip4_table_index: (Default value = 0xFFFFFFFF) |
| 612 | :param ip6_table_index: (Default value = 0xFFFFFFFF) |
| 613 | :param l2_table_index: (Default value = 0xFFFFFFFF) |
| 614 | """ |
| 615 | |
| 616 | return self.api( |
| 617 | self.papi.output_acl_set_interface, |
| 618 | {'sw_if_index': sw_if_index, |
| 619 | 'ip4_table_index': ip4_table_index, |
| 620 | 'ip6_table_index': ip6_table_index, |
| 621 | 'l2_table_index': l2_table_index, |
| 622 | 'is_add': is_add}) |
| 623 | |
Klement Sekera | 77fabdb | 2017-01-02 07:46:14 +0100 | [diff] [blame] | 624 | def set_ipfix_exporter( |
| 625 | self, |
| 626 | collector_address, |
| 627 | src_address, |
| 628 | path_mtu, |
| 629 | template_interval, |
| 630 | vrf_id=0, |
| 631 | collector_port=4739, |
| 632 | udp_checksum=0): |
| 633 | return self.api( |
| 634 | self.papi.set_ipfix_exporter, |
| 635 | { |
| 636 | 'collector_address': collector_address, |
| 637 | 'collector_port': collector_port, |
| 638 | 'src_address': src_address, |
| 639 | 'vrf_id': vrf_id, |
| 640 | 'path_mtu': path_mtu, |
| 641 | 'template_interval': template_interval, |
| 642 | 'udp_checksum': udp_checksum, |
| 643 | }) |
Neale Ranns | fca0c24 | 2017-01-13 07:57:46 -0800 | [diff] [blame] | 644 | |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 645 | def mfib_signal_dump(self): |
| 646 | return self.api(self.papi.mfib_signal_dump, {}) |
| 647 | |
| 648 | def ip_mroute_dump(self, table_id, is_ip6=False): |
| 649 | return self.api(self.papi.ip_mroute_dump, |
| 650 | {'table': { |
| 651 | 'table_id': table_id, |
| 652 | 'is_ip6': is_ip6 |
| 653 | }}) |
Neale Ranns | 32e1c01 | 2016-11-22 17:07:28 +0000 | [diff] [blame] | 654 | |
Neale Ranns | 79a05f5 | 2018-09-11 07:39:43 -0700 | [diff] [blame] | 655 | def vxlan_gbp_tunnel_dump(self, sw_if_index=0xffffffff): |
| 656 | return self.api(self.papi.vxlan_gbp_tunnel_dump, |
Neale Ranns | efd7bc2 | 2019-11-11 08:32:34 +0000 | [diff] [blame] | 657 | {'sw_if_index': sw_if_index}) |
Mohsin Kazmi | 61b94c6 | 2018-08-20 18:32:39 +0200 | [diff] [blame] | 658 | |
Hongjun Ni | 62f9cdd | 2017-07-04 20:11:57 +0800 | [diff] [blame] | 659 | def pppoe_add_del_session( |
| 660 | self, |
| 661 | client_ip, |
| 662 | client_mac, |
| 663 | session_id=0, |
| 664 | is_add=1, |
Hongjun Ni | 62f9cdd | 2017-07-04 20:11:57 +0800 | [diff] [blame] | 665 | decap_vrf_id=0): |
| 666 | """ |
| 667 | |
| 668 | :param is_add: (Default value = 1) |
| 669 | :param is_ipv6: (Default value = 0) |
| 670 | :param client_ip: |
| 671 | :param session_id: (Default value = 0) |
| 672 | :param client_mac: |
| 673 | :param decap_vrf_id: (Default value = 0) |
| 674 | |
| 675 | """ |
| 676 | return self.api(self.papi.pppoe_add_del_session, |
| 677 | {'is_add': is_add, |
Hongjun Ni | 62f9cdd | 2017-07-04 20:11:57 +0800 | [diff] [blame] | 678 | 'session_id': session_id, |
| 679 | 'client_ip': client_ip, |
| 680 | 'decap_vrf_id': decap_vrf_id, |
| 681 | 'client_mac': client_mac}) |
Kris Michielsen | 9107443 | 2017-06-22 13:00:20 +0200 | [diff] [blame] | 682 | |
Neale Ranns | 7c922dc | 2018-08-30 06:12:27 -0700 | [diff] [blame] | 683 | def sr_mpls_policy_add(self, bsid, weight, type, segments): |
| 684 | return self.api(self.papi.sr_mpls_policy_add, |
| 685 | {'bsid': bsid, |
| 686 | 'weight': weight, |
Jakub Grajciar | 00ec401 | 2020-01-31 10:17:29 +0100 | [diff] [blame] | 687 | 'is_spray': type, |
Neale Ranns | 7c922dc | 2018-08-30 06:12:27 -0700 | [diff] [blame] | 688 | 'n_segments': len(segments), |
| 689 | 'segments': segments}) |
| 690 | |
| 691 | def sr_mpls_policy_del(self, bsid): |
| 692 | return self.api(self.papi.sr_mpls_policy_del, |
| 693 | {'bsid': bsid}) |
| 694 | |
Neale Ranns | d792d9c | 2017-10-21 10:53:20 -0700 | [diff] [blame] | 695 | def bier_table_add_del(self, |
| 696 | bti, |
| 697 | mpls_label, |
| 698 | is_add=1): |
| 699 | """ BIER Table add/del """ |
| 700 | return self.api( |
| 701 | self.papi.bier_table_add_del, |
| 702 | {'bt_tbl_id': {"bt_set": bti.set_id, |
| 703 | "bt_sub_domain": bti.sub_domain_id, |
| 704 | "bt_hdr_len_id": bti.hdr_len_id}, |
| 705 | 'bt_label': mpls_label, |
| 706 | 'bt_is_add': is_add}) |
| 707 | |
| 708 | def bier_table_dump(self): |
| 709 | return self.api(self.papi.bier_table_dump, {}) |
| 710 | |
| 711 | def bier_route_add_del(self, |
| 712 | bti, |
| 713 | bp, |
Neale Ranns | 9128637 | 2017-12-05 13:24:04 -0800 | [diff] [blame] | 714 | paths, |
Neale Ranns | ef90ed0 | 2018-09-13 08:45:12 -0700 | [diff] [blame] | 715 | is_add=1, |
| 716 | is_replace=0): |
Neale Ranns | d792d9c | 2017-10-21 10:53:20 -0700 | [diff] [blame] | 717 | """ BIER Route add/del """ |
| 718 | return self.api( |
| 719 | self.papi.bier_route_add_del, |
Neale Ranns | 097fa66 | 2018-05-01 05:17:55 -0700 | [diff] [blame] | 720 | { |
| 721 | 'br_route': { |
| 722 | 'br_tbl_id': {"bt_set": bti.set_id, |
| 723 | "bt_sub_domain": bti.sub_domain_id, |
| 724 | "bt_hdr_len_id": bti.hdr_len_id}, |
| 725 | 'br_bp': bp, |
| 726 | 'br_n_paths': len(paths), |
| 727 | 'br_paths': paths, |
| 728 | }, |
| 729 | 'br_is_add': is_add, |
| 730 | 'br_is_replace': is_replace |
| 731 | }) |
Neale Ranns | d792d9c | 2017-10-21 10:53:20 -0700 | [diff] [blame] | 732 | |
| 733 | def bier_route_dump(self, bti): |
| 734 | return self.api( |
| 735 | self.papi.bier_route_dump, |
| 736 | {'br_tbl_id': {"bt_set": bti.set_id, |
| 737 | "bt_sub_domain": bti.sub_domain_id, |
| 738 | "bt_hdr_len_id": bti.hdr_len_id}}) |
| 739 | |
| 740 | def bier_imp_add(self, |
| 741 | bti, |
| 742 | src, |
| 743 | ibytes, |
| 744 | is_add=1): |
| 745 | """ BIER Imposition Add """ |
| 746 | return self.api( |
| 747 | self.papi.bier_imp_add, |
| 748 | {'bi_tbl_id': {"bt_set": bti.set_id, |
| 749 | "bt_sub_domain": bti.sub_domain_id, |
| 750 | "bt_hdr_len_id": bti.hdr_len_id}, |
| 751 | 'bi_src': src, |
| 752 | 'bi_n_bytes': len(ibytes), |
| 753 | 'bi_bytes': ibytes}) |
| 754 | |
| 755 | def bier_imp_del(self, bi_index): |
| 756 | """ BIER Imposition del """ |
| 757 | return self.api( |
| 758 | self.papi.bier_imp_del, |
| 759 | {'bi_index': bi_index}) |
| 760 | |
| 761 | def bier_imp_dump(self): |
| 762 | return self.api(self.papi.bier_imp_dump, {}) |
| 763 | |
| 764 | def bier_disp_table_add_del(self, |
| 765 | bdti, |
| 766 | is_add=1): |
| 767 | """ BIER Disposition Table add/del """ |
| 768 | return self.api( |
| 769 | self.papi.bier_disp_table_add_del, |
| 770 | {'bdt_tbl_id': bdti, |
| 771 | 'bdt_is_add': is_add}) |
| 772 | |
| 773 | def bier_disp_table_dump(self): |
| 774 | return self.api(self.papi.bier_disp_table_dump, {}) |
| 775 | |
| 776 | def bier_disp_entry_add_del(self, |
| 777 | bdti, |
| 778 | bp, |
| 779 | payload_proto, |
Neale Ranns | f051072 | 2018-01-31 11:35:41 -0800 | [diff] [blame] | 780 | next_hop_afi, |
Neale Ranns | d792d9c | 2017-10-21 10:53:20 -0700 | [diff] [blame] | 781 | next_hop, |
| 782 | next_hop_tbl_id=0, |
| 783 | next_hop_rpf_id=~0, |
| 784 | next_hop_is_ip4=1, |
| 785 | is_add=1): |
| 786 | """ BIER Route add/del """ |
Neale Ranns | 31ed744 | 2018-02-23 05:29:09 -0800 | [diff] [blame] | 787 | lstack = [] |
| 788 | while (len(lstack) < 16): |
| 789 | lstack.append({}) |
Neale Ranns | d792d9c | 2017-10-21 10:53:20 -0700 | [diff] [blame] | 790 | return self.api( |
| 791 | self.papi.bier_disp_entry_add_del, |
| 792 | {'bde_tbl_id': bdti, |
| 793 | 'bde_bp': bp, |
| 794 | 'bde_payload_proto': payload_proto, |
| 795 | 'bde_n_paths': 1, |
Paul Vinciguerra | 79a32a4 | 2019-07-11 19:09:30 -0400 | [diff] [blame] | 796 | 'bde_paths': [{'table_id': next_hop_tbl_id, |
Neale Ranns | d792d9c | 2017-10-21 10:53:20 -0700 | [diff] [blame] | 797 | 'rpf_id': next_hop_rpf_id, |
| 798 | 'n_labels': 0, |
Neale Ranns | 31ed744 | 2018-02-23 05:29:09 -0800 | [diff] [blame] | 799 | 'label_stack': lstack}], |
Neale Ranns | d792d9c | 2017-10-21 10:53:20 -0700 | [diff] [blame] | 800 | 'bde_is_add': is_add}) |
| 801 | |
| 802 | def bier_disp_entry_dump(self, bdti): |
| 803 | return self.api( |
| 804 | self.papi.bier_disp_entry_dump, |
| 805 | {'bde_tbl_id': bdti}) |
Gabriel Ganne | 8527f12 | 2017-10-02 11:41:24 +0200 | [diff] [blame] | 806 | |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 807 | def ipsec_spd_add_del(self, spd_id, is_add=1): |
| 808 | """ SPD add/del - Wrapper to add or del ipsec SPD |
| 809 | Sample CLI : 'ipsec spd add 1' |
| 810 | |
| 811 | :param spd_id - SPD ID to be created in the vpp . mandatory |
| 812 | :param is_add - create (1) or delete(0) SPD (Default 1 - add) . |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 813 | optional |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 814 | :returns: reply from the API |
| 815 | """ |
| 816 | return self.api( |
| 817 | self.papi.ipsec_spd_add_del, { |
| 818 | 'spd_id': spd_id, 'is_add': is_add}) |
| 819 | |
Neale Ranns | 311124e | 2019-01-24 04:52:25 -0800 | [diff] [blame] | 820 | def ipsec_spds_dump(self): |
| 821 | return self.api(self.papi.ipsec_spds_dump, {}) |
| 822 | |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 823 | def ipsec_interface_add_del_spd(self, spd_id, sw_if_index, is_add=1): |
| 824 | """ IPSEC interface SPD add/del - \ |
| 825 | Wrapper to associate/disassociate SPD to interface in VPP |
| 826 | Sample CLI : 'set interface ipsec spd GigabitEthernet0/6/0 1' |
| 827 | |
| 828 | :param spd_id - SPD ID to associate with the interface . mandatory |
| 829 | :param sw_if_index - Interface Index which needs to ipsec \ |
| 830 | association mandatory |
| 831 | :param is_add - add(1) or del(0) association with interface \ |
| 832 | (Default 1 - add) . optional |
| 833 | :returns: reply from the API |
| 834 | """ |
| 835 | return self.api( |
Klement Sekera | 4b089f2 | 2018-04-17 18:04:57 +0200 | [diff] [blame] | 836 | self.papi.ipsec_interface_add_del_spd, |
| 837 | {'spd_id': spd_id, 'sw_if_index': sw_if_index, 'is_add': is_add}) |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 838 | |
Neale Ranns | 311124e | 2019-01-24 04:52:25 -0800 | [diff] [blame] | 839 | def ipsec_spd_interface_dump(self, spd_index=None): |
| 840 | return self.api(self.papi.ipsec_spd_interface_dump, |
| 841 | {'spd_index': spd_index if spd_index else 0, |
| 842 | 'spd_index_valid': 1 if spd_index else 0}) |
| 843 | |
Neale Ranns | 17dcec0 | 2019-01-09 21:22:20 -0800 | [diff] [blame] | 844 | def ipsec_spd_entry_add_del(self, |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 845 | spd_id, |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 846 | sa_id, |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 847 | local_address_start, |
| 848 | local_address_stop, |
| 849 | remote_address_start, |
| 850 | remote_address_stop, |
| 851 | local_port_start=0, |
| 852 | local_port_stop=65535, |
| 853 | remote_port_start=0, |
| 854 | remote_port_stop=65535, |
| 855 | protocol=0, |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 856 | policy=0, |
| 857 | priority=100, |
| 858 | is_outbound=1, |
| 859 | is_add=1, |
Klement Sekera | 611864f | 2018-09-26 11:19:00 +0200 | [diff] [blame] | 860 | is_ipv6=0, |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 861 | is_ip_any=0): |
| 862 | """ IPSEC policy SPD add/del - |
| 863 | Wrapper to configure ipsec SPD policy entries in VPP |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 864 | |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 865 | :param spd_id: SPD ID for the policy |
| 866 | :param local_address_start: local-ip-range start address |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 867 | :param local_address_stop: local-ip-range stop address |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 868 | :param remote_address_start: remote-ip-range start address |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 869 | :param remote_address_stop: remote-ip-range stop address |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 870 | :param local_port_start: (Default value = 0) |
| 871 | :param local_port_stop: (Default value = 65535) |
| 872 | :param remote_port_start: (Default value = 0) |
| 873 | :param remote_port_stop: (Default value = 65535) |
| 874 | :param protocol: Any(0), AH(51) & ESP(50) protocol (Default value = 0) |
| 875 | :param sa_id: Security Association ID for mapping it to SPD |
| 876 | :param policy: bypass(0), discard(1), resolve(2) or protect(3) action |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 877 | (Default value = 0) |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 878 | :param priority: value for the spd action (Default value = 100) |
| 879 | :param is_outbound: flag for inbound(0) or outbound(1) |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 880 | (Default value = 1) |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 881 | :param is_add: (Default value = 1) |
“mystarrocks” | 23f0c45 | 2017-12-11 07:11:51 -0800 | [diff] [blame] | 882 | """ |
| 883 | return self.api( |
Neale Ranns | 17dcec0 | 2019-01-09 21:22:20 -0800 | [diff] [blame] | 884 | self.papi.ipsec_spd_entry_add_del, |
| 885 | { |
| 886 | 'is_add': is_add, |
| 887 | 'entry': |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 888 | { |
| 889 | 'spd_id': spd_id, |
| 890 | 'sa_id': sa_id, |
| 891 | 'local_address_start': local_address_start, |
| 892 | 'local_address_stop': local_address_stop, |
| 893 | 'remote_address_start': remote_address_start, |
| 894 | 'remote_address_stop': remote_address_stop, |
| 895 | 'local_port_start': local_port_start, |
| 896 | 'local_port_stop': local_port_stop, |
| 897 | 'remote_port_start': remote_port_start, |
| 898 | 'remote_port_stop': remote_port_stop, |
| 899 | 'protocol': protocol, |
| 900 | 'policy': policy, |
| 901 | 'priority': priority, |
| 902 | 'is_outbound': is_outbound, |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 903 | } |
Neale Ranns | 17dcec0 | 2019-01-09 21:22:20 -0800 | [diff] [blame] | 904 | }) |
Florin Coras | b795bd0 | 2017-12-14 11:30:48 -0800 | [diff] [blame] | 905 | |
Neale Ranns | 311124e | 2019-01-24 04:52:25 -0800 | [diff] [blame] | 906 | def ipsec_spd_dump(self, spd_id, sa_id=0xffffffff): |
| 907 | return self.api(self.papi.ipsec_spd_dump, |
| 908 | {'spd_id': spd_id, |
| 909 | 'sa_id': sa_id}) |
| 910 | |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 911 | def ipsec_tunnel_if_add_del(self, local_ip, remote_ip, local_spi, |
| 912 | remote_spi, crypto_alg, local_crypto_key, |
| 913 | remote_crypto_key, integ_alg, local_integ_key, |
Neale Ranns | 80f6fd5 | 2019-04-16 02:41:34 +0000 | [diff] [blame] | 914 | remote_integ_key, is_add=1, esn=0, salt=0, |
Neale Ranns | a6bee0a | 2019-06-14 01:13:25 -0700 | [diff] [blame] | 915 | anti_replay=1, renumber=0, |
Neale Ranns | 12989b5 | 2019-09-26 16:20:19 +0000 | [diff] [blame] | 916 | udp_encap=0, show_instance=0xffffffff): |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 917 | return self.api( |
| 918 | self.papi.ipsec_tunnel_if_add_del, |
Kingwel Xie | 1ba5bc8 | 2019-03-20 07:21:58 -0400 | [diff] [blame] | 919 | { |
| 920 | 'local_ip': local_ip, |
| 921 | 'remote_ip': remote_ip, |
| 922 | 'local_spi': local_spi, |
| 923 | 'remote_spi': remote_spi, |
| 924 | 'crypto_alg': crypto_alg, |
| 925 | 'local_crypto_key_len': len(local_crypto_key), |
| 926 | 'local_crypto_key': local_crypto_key, |
| 927 | 'remote_crypto_key_len': len(remote_crypto_key), |
| 928 | 'remote_crypto_key': remote_crypto_key, |
| 929 | 'integ_alg': integ_alg, |
| 930 | 'local_integ_key_len': len(local_integ_key), |
| 931 | 'local_integ_key': local_integ_key, |
| 932 | 'remote_integ_key_len': len(remote_integ_key), |
| 933 | 'remote_integ_key': remote_integ_key, |
| 934 | 'is_add': is_add, |
| 935 | 'esn': esn, |
| 936 | 'anti_replay': anti_replay, |
| 937 | 'renumber': renumber, |
Neale Ranns | 80f6fd5 | 2019-04-16 02:41:34 +0000 | [diff] [blame] | 938 | 'show_instance': show_instance, |
Neale Ranns | a6bee0a | 2019-06-14 01:13:25 -0700 | [diff] [blame] | 939 | 'udp_encap': udp_encap, |
Neale Ranns | 80f6fd5 | 2019-04-16 02:41:34 +0000 | [diff] [blame] | 940 | 'salt': salt |
Kingwel Xie | 1ba5bc8 | 2019-03-20 07:21:58 -0400 | [diff] [blame] | 941 | }) |
Klement Sekera | 31da2e3 | 2018-06-24 22:49:55 +0200 | [diff] [blame] | 942 | |
Klement Sekera | b4d3053 | 2018-11-08 13:00:02 +0100 | [diff] [blame] | 943 | def ipsec_select_backend(self, protocol, index): |
| 944 | return self.api(self.papi.ipsec_select_backend, |
| 945 | {'protocol': protocol, 'index': index}) |
| 946 | |
| 947 | def ipsec_backend_dump(self): |
| 948 | return self.api(self.papi.ipsec_backend_dump, {}) |
| 949 | |
Neale Ranns | 50f0ac0 | 2019-05-15 02:13:37 -0700 | [diff] [blame] | 950 | def punt_socket_register(self, reg, pathname, |
| 951 | header_version=1): |
Pavel Kotucek | e88865d | 2018-11-28 07:42:11 +0100 | [diff] [blame] | 952 | """ Register punt socket """ |
Klement Sekera | 75e7d13 | 2017-09-20 08:26:30 +0200 | [diff] [blame] | 953 | return self.api(self.papi.punt_socket_register, |
Pavel Kotucek | e88865d | 2018-11-28 07:42:11 +0100 | [diff] [blame] | 954 | {'header_version': header_version, |
Neale Ranns | 50f0ac0 | 2019-05-15 02:13:37 -0700 | [diff] [blame] | 955 | 'punt': reg, |
Pavel Kotucek | e88865d | 2018-11-28 07:42:11 +0100 | [diff] [blame] | 956 | 'pathname': pathname}) |
| 957 | |
Neale Ranns | 50f0ac0 | 2019-05-15 02:13:37 -0700 | [diff] [blame] | 958 | def punt_socket_deregister(self, reg): |
Pavel Kotucek | e88865d | 2018-11-28 07:42:11 +0100 | [diff] [blame] | 959 | """ Unregister punt socket """ |
| 960 | return self.api(self.papi.punt_socket_deregister, |
Neale Ranns | 50f0ac0 | 2019-05-15 02:13:37 -0700 | [diff] [blame] | 961 | {'punt': reg}) |
Klement Sekera | 75e7d13 | 2017-09-20 08:26:30 +0200 | [diff] [blame] | 962 | |
Neale Ranns | bc27d1b | 2018-02-05 01:13:38 -0800 | [diff] [blame] | 963 | def gbp_endpoint_dump(self): |
| 964 | """ GBP endpoint Dump """ |
Neale Ranns | efd7bc2 | 2019-11-11 08:32:34 +0000 | [diff] [blame] | 965 | return self.api(self.papi.gbp_endpoint_dump, {}) |
Neale Ranns | bc27d1b | 2018-02-05 01:13:38 -0800 | [diff] [blame] | 966 | |
Neale Ranns | 25b0494 | 2018-04-04 09:34:50 -0700 | [diff] [blame] | 967 | def gbp_recirc_dump(self): |
| 968 | """ GBP recirc Dump """ |
| 969 | return self.api(self.papi.gbp_recirc_dump, {}) |
| 970 | |
Neale Ranns | b6a4795 | 2018-11-21 05:44:35 -0800 | [diff] [blame] | 971 | def gbp_ext_itf_dump(self): |
| 972 | """ GBP recirc Dump """ |
| 973 | return self.api(self.papi.gbp_ext_itf_dump, {}) |
| 974 | |
Neale Ranns | 25b0494 | 2018-04-04 09:34:50 -0700 | [diff] [blame] | 975 | def gbp_subnet_dump(self): |
| 976 | """ GBP Subnet Dump """ |
Neale Ranns | efd7bc2 | 2019-11-11 08:32:34 +0000 | [diff] [blame] | 977 | return self.api(self.papi.gbp_subnet_dump, {}) |
Neale Ranns | 25b0494 | 2018-04-04 09:34:50 -0700 | [diff] [blame] | 978 | |
Neale Ranns | bc27d1b | 2018-02-05 01:13:38 -0800 | [diff] [blame] | 979 | def gbp_contract_dump(self): |
| 980 | """ GBP contract Dump """ |
| 981 | return self.api(self.papi.gbp_contract_dump, {}) |
Ole Troan | 6ee4051 | 2018-02-12 18:14:39 +0100 | [diff] [blame] | 982 | |
Neale Ranns | 93cc3ee | 2018-10-10 07:22:51 -0700 | [diff] [blame] | 983 | def gbp_vxlan_tunnel_dump(self): |
| 984 | """ GBP VXLAN tunnel add/del """ |
| 985 | return self.api(self.papi.gbp_vxlan_tunnel_dump, {}) |
| 986 | |
Neale Ranns | 947ea62 | 2018-06-07 23:48:20 -0700 | [diff] [blame] | 987 | def igmp_enable_disable(self, sw_if_index, enable, host): |
| 988 | """ Enable/disable IGMP on a given interface """ |
| 989 | return self.api(self.papi.igmp_enable_disable, |
| 990 | {'enable': enable, |
| 991 | 'mode': host, |
| 992 | 'sw_if_index': sw_if_index}) |
| 993 | |
Jakub Grajciar | 97748ca | 2018-10-04 11:05:35 +0200 | [diff] [blame] | 994 | def igmp_proxy_device_add_del(self, vrf_id, sw_if_index, add): |
| 995 | """ Add/del IGMP proxy device """ |
| 996 | return self.api(self.papi.igmp_proxy_device_add_del, |
| 997 | {'vrf_id': vrf_id, 'sw_if_index': sw_if_index, |
| 998 | 'add': add}) |
| 999 | |
| 1000 | def igmp_proxy_device_add_del_interface(self, vrf_id, sw_if_index, add): |
| 1001 | """ Add/del interface to/from IGMP proxy device """ |
| 1002 | return self.api(self.papi.igmp_proxy_device_add_del_interface, |
| 1003 | {'vrf_id': vrf_id, 'sw_if_index': sw_if_index, |
| 1004 | 'add': add}) |
| 1005 | |
Neale Ranns | 947ea62 | 2018-06-07 23:48:20 -0700 | [diff] [blame] | 1006 | def igmp_listen(self, filter, sw_if_index, saddrs, gaddr): |
Jakub Grajciar | 7b867a8 | 2017-12-08 16:28:42 +0100 | [diff] [blame] | 1007 | """ Listen for new (S,G) on specified interface |
| 1008 | |
Dave Wallace | d170681 | 2021-08-12 18:36:02 -0400 | [diff] [blame] | 1009 | :param enable: add/delas |
Jakub Grajciar | 7b867a8 | 2017-12-08 16:28:42 +0100 | [diff] [blame] | 1010 | :param sw_if_index: interface sw index |
| 1011 | :param saddr: source ip4 addr |
| 1012 | :param gaddr: group ip4 addr |
| 1013 | """ |
| 1014 | return self.api(self.papi.igmp_listen, |
Neale Ranns | 947ea62 | 2018-06-07 23:48:20 -0700 | [diff] [blame] | 1015 | { |
| 1016 | 'group': |
Ole Troan | 9a47537 | 2019-03-05 16:58:24 +0100 | [diff] [blame] | 1017 | { |
| 1018 | 'filter': filter, |
| 1019 | 'sw_if_index': sw_if_index, |
| 1020 | 'n_srcs': len(saddrs), |
| 1021 | 'saddrs': saddrs, |
| 1022 | 'gaddr': gaddr |
| 1023 | } |
Neale Ranns | 947ea62 | 2018-06-07 23:48:20 -0700 | [diff] [blame] | 1024 | }) |
Jakub Grajciar | 7b867a8 | 2017-12-08 16:28:42 +0100 | [diff] [blame] | 1025 | |
Jakub Grajciar | 7b867a8 | 2017-12-08 16:28:42 +0100 | [diff] [blame] | 1026 | def igmp_clear_interface(self, sw_if_index): |
| 1027 | """ Remove all (S,G)s from specified interface |
| 1028 | doesn't send IGMP report! |
| 1029 | """ |
| 1030 | return self.api( |
| 1031 | self.papi.igmp_clear_interface, { |
| 1032 | 'sw_if_index': sw_if_index}) |
| 1033 | |
| 1034 | def want_igmp_events(self, enable=1): |
| 1035 | return self.api(self.papi.want_igmp_events, {'enable': enable, |
| 1036 | 'pid': os.getpid()}) |