test/vpp_ip_route.py

   1 """
   2   IP Routes
   3
   4   object abstractions for representing IP routes in VPP
   5 """
   6
   7 from vpp_object import *
   8 from socket import inet_pton, inet_ntop, AF_INET, AF_INET6
   9 from vpp_ip import *
  10
  11 # from vnet/vnet/mpls/mpls_types.h
  12 MPLS_IETF_MAX_LABEL = 0xfffff
  13 MPLS_LABEL_INVALID = MPLS_IETF_MAX_LABEL + 1
  14
  15
  16 class MRouteItfFlags:
  17     MFIB_ITF_FLAG_NONE = 0
  18     MFIB_ITF_FLAG_NEGATE_SIGNAL = 1
  19     MFIB_ITF_FLAG_ACCEPT = 2
  20     MFIB_ITF_FLAG_FORWARD = 4
  21     MFIB_ITF_FLAG_SIGNAL_PRESENT = 8
  22     MFIB_ITF_FLAG_INTERNAL_COPY = 16
  23
  24
  25 class MRouteEntryFlags:
  26     MFIB_ENTRY_FLAG_NONE = 0
  27     MFIB_ENTRY_FLAG_SIGNAL = 1
  28     MFIB_ENTRY_FLAG_DROP = 2
  29     MFIB_ENTRY_FLAG_CONNECTED = 4
  30     MFIB_ENTRY_FLAG_INHERIT_ACCEPT = 8
  31
  32
  33 class MplsLspMode:
  34     PIPE = 0
  35     UNIFORM = 1
  36
  37
  38 def ip_to_dpo_proto(addr):
  39     if addr.version is 6:
  40         return DpoProto.DPO_PROTO_IP6
  41     else:
  42         return DpoProto.DPO_PROTO_IP4
  43
  44
  45 def find_route(test, ip_addr, len, table_id=0, inet=AF_INET):
  46     if inet == AF_INET:
  47         s = 4
  48         routes = test.vapi.ip_fib_dump()
  49     else:
  50         s = 16
  51         routes = test.vapi.ip6_fib_dump()
  52
  53     route_addr = inet_pton(inet, ip_addr)
  54     for e in routes:
  55         if route_addr == e.address[:s] \
  56            and len == e.address_length \
  57            and table_id == e.table_id:
  58             return True
  59     return False
  60
  61
  62 def find_mroute(test, grp_addr, src_addr, grp_addr_len,
  63                 table_id=0, inet=AF_INET):
  64     if inet == AF_INET:
  65         s = 4
  66         routes = test.vapi.ip_mfib_dump()
  67     else:
  68         s = 16
  69         routes = test.vapi.ip6_mfib_dump()
  70     gaddr = inet_pton(inet, grp_addr)
  71     saddr = inet_pton(inet, src_addr)
  72     for e in routes:
  73         if gaddr == e.grp_address[:s] \
  74            and grp_addr_len == e.address_length \
  75            and saddr == e.src_address[:s] \
  76            and table_id == e.table_id:
  77             return True
  78     return False
  79
  80
  81 def find_mpls_route(test, table_id, label, eos_bit, paths=None):
  82     dump = test.vapi.mpls_fib_dump()
  83     for e in dump:
  84         if label == e.label \
  85            and eos_bit == e.eos_bit \
  86            and table_id == e.table_id:
  87             if not paths:
  88                 return True
  89             else:
  90                 if (len(paths) != len(e.path)):
  91                     return False
  92                 for i in range(len(paths)):
  93                     if (paths[i] != e.path[i]):
  94                         return False
  95                 return True
  96     return False
  97
  98
  99 def fib_interface_ip_prefix(test, address, length, sw_if_index):
 100     vp = VppIpPrefix(address, length)
 101     addrs = test.vapi.ip_address_dump(sw_if_index, is_ipv6=vp.is_ip6)
 102
 103     if vp.is_ip6:
 104         n = 16
 105     else:
 106         n = 4
 107
 108     for a in addrs:
 109         if a.prefix_length == length and \
 110            a.sw_if_index == sw_if_index and \
 111            a.ip[:n] == vp.bytes:
 112             return True
 113     return False
 114
 115
 116 class VppIpTable(VppObject):
 117
 118     def __init__(self,
 119                  test,
 120                  table_id,
 121                  is_ip6=0):
 122         self._test = test
 123         self.table_id = table_id
 124         self.is_ip6 = is_ip6
 125
 126     def add_vpp_config(self):
 127         self._test.vapi.ip_table_add_del(
 128             self.table_id,
 129             is_ipv6=self.is_ip6,
 130             is_add=1)
 131         self._test.registry.register(self, self._test.logger)
 132
 133     def remove_vpp_config(self):
 134         self._test.vapi.ip_table_add_del(
 135             self.table_id,
 136             is_ipv6=self.is_ip6,
 137             is_add=0)
 138
 139     def query_vpp_config(self):
 140         if self.table_id == 0:
 141             # the default table always exists
 142             return False
 143         # find the default route
 144         return find_route(self._test,
 145                           "::" if self.is_ip6 else "0.0.0.0",
 146                           0,
 147                           self.table_id,
 148                           inet=AF_INET6 if self.is_ip6 == 1 else AF_INET)
 149
 150     def __str__(self):
 151         return self.object_id()
 152
 153     def object_id(self):
 154         return ("table-%s-%d" %
 155                 ("v6" if self.is_ip6 == 1 else "v4",
 156                  self.table_id))
 157
 158
 159 class VppIpInterfaceAddress(VppObject):
 160
 161     def __init__(self, test, intf, addr, len):
 162         self._test = test
 163         self.intf = intf
 164         self.prefix = VppIpPrefix(addr, len)
 165
 166     def add_vpp_config(self):
 167         self._test.vapi.sw_interface_add_del_address(
 168             self.intf.sw_if_index,
 169             self.prefix.bytes,
 170             self.prefix.length,
 171             is_add=1,
 172             is_ipv6=self.prefix.is_ip6)
 173         self._test.registry.register(self, self._test.logger)
 174
 175     def remove_vpp_config(self):
 176         self._test.vapi.sw_interface_add_del_address(
 177             self.intf.sw_if_index,
 178             self.prefix.bytes,
 179             self.prefix.length,
 180             is_add=0,
 181             is_ipv6=self.prefix.is_ip6)
 182
 183     def query_vpp_config(self):
 184         return fib_interface_ip_prefix(self._test,
 185                                        self.prefix.address,
 186                                        self.prefix.length,
 187                                        self.intf.sw_if_index)
 188
 189     def __str__(self):
 190         return self.object_id()
 191
 192     def object_id(self):
 193         return "interface-ip-%s-%s" % (self.intf, self.prefix)
 194
 195
 196 class VppIpInterfaceBind(VppObject):
 197
 198     def __init__(self, test, intf, table):
 199         self._test = test
 200         self.intf = intf
 201         self.table = table
 202
 203     def add_vpp_config(self):
 204         if self.table.is_ip6:
 205             self.intf.set_table_ip6(self.table.table_id)
 206         else:
 207             self.intf.set_table_ip4(self.table.table_id)
 208         self._test.registry.register(self, self._test.logger)
 209
 210     def remove_vpp_config(self):
 211         if 0 == self.table.table_id:
 212             return
 213         if self.table.is_ip6:
 214             self.intf.set_table_ip6(0)
 215         else:
 216             self.intf.set_table_ip4(0)
 217
 218     def query_vpp_config(self):
 219         if 0 == self.table.table_id:
 220             return False
 221         return self._test.vapi.sw_interface_get_table(
 222             self.intf.sw_if_index,
 223             self.table.is_ip6).vrf_id == self.table.table_id
 224
 225     def __str__(self):
 226         return self.object_id()
 227
 228     def object_id(self):
 229         return "interface-bind-%s-%s" % (self.intf, self.table)
 230
 231
 232 class VppMplsLabel(object):
 233     def __init__(self, value, mode=MplsLspMode.PIPE, ttl=64, exp=0):
 234         self.value = value
 235         self.mode = mode
 236         self.ttl = ttl
 237         self.exp = exp
 238
 239     def encode(self):
 240         is_uniform = 0 if self.mode is MplsLspMode.PIPE else 1
 241         return {'label': self.value,
 242                 'ttl': self.ttl,
 243                 'exp': self.exp,
 244                 'is_uniform': is_uniform}
 245
 246     def __eq__(self, other):
 247         if isinstance(other, self.__class__):
 248             return (self.value == other.value and
 249                     self.ttl == other.ttl and
 250                     self.exp == other.exp and
 251                     self.mode == other.mode)
 252         elif hasattr(other, 'label'):
 253             return (self.value == other.label and
 254                     self.ttl == other.ttl and
 255                     self.exp == other.exp and
 256                     (self.mode == MplsLspMode.UNIFORM) == other.is_uniform)
 257         else:
 258             return False
 259
 260     def __ne__(self, other):
 261         return not (self == other)
 262
 263
 264 class VppRoutePath(object):
 265
 266     def __init__(
 267             self,
 268             nh_addr,
 269             nh_sw_if_index,
 270             nh_table_id=0,
 271             labels=[],
 272             nh_via_label=MPLS_LABEL_INVALID,
 273             rpf_id=0,
 274             is_interface_rx=0,
 275             is_resolve_host=0,
 276             is_resolve_attached=0,
 277             is_source_lookup=0,
 278             is_udp_encap=0,
 279             is_dvr=0,
 280             next_hop_id=0xffffffff,
 281             proto=DpoProto.DPO_PROTO_IP4):
 282         self.nh_itf = nh_sw_if_index
 283         self.nh_table_id = nh_table_id
 284         self.nh_via_label = nh_via_label
 285         self.nh_labels = labels
 286         self.weight = 1
 287         self.rpf_id = rpf_id
 288         self.proto = proto
 289         if self.proto is DpoProto.DPO_PROTO_IP6:
 290             self.nh_addr = inet_pton(AF_INET6, nh_addr)
 291         elif self.proto is DpoProto.DPO_PROTO_IP4:
 292             self.nh_addr = inet_pton(AF_INET, nh_addr)
 293         else:
 294             self.nh_addr = inet_pton(AF_INET6, "::")
 295         self.is_resolve_host = is_resolve_host
 296         self.is_resolve_attached = is_resolve_attached
 297         self.is_interface_rx = is_interface_rx
 298         self.is_source_lookup = is_source_lookup
 299         self.is_rpf_id = 0
 300         if rpf_id != 0:
 301             self.is_rpf_id = 1
 302             self.nh_itf = rpf_id
 303         self.is_udp_encap = is_udp_encap
 304         self.next_hop_id = next_hop_id
 305         self.is_dvr = is_dvr
 306
 307     def encode_labels(self):
 308         lstack = []
 309         for l in self.nh_labels:
 310             if type(l) == VppMplsLabel:
 311                 lstack.append(l.encode())
 312             else:
 313                 lstack.append({'label': l,
 314                                'ttl': 255})
 315         return lstack
 316
 317     def encode(self):
 318         return {'next_hop': self.nh_addr,
 319                 'weight': 1,
 320                 'afi': 0,
 321                 'preference': 0,
 322                 'table_id': self.nh_table_id,
 323                 'next_hop_id': self.next_hop_id,
 324                 'sw_if_index': self.nh_itf,
 325                 'afi': self.proto,
 326                 'is_udp_encap': self.is_udp_encap,
 327                 'n_labels': len(self.nh_labels),
 328                 'label_stack': self.encode_labels()}
 329
 330     def __eq__(self, other):
 331         if isinstance(other, self.__class__):
 332             return self.nh_addr == other.nh_addr
 333         elif hasattr(other, 'sw_if_index'):
 334             # vl_api_fib_path_t
 335             if (len(self.nh_labels) != other.n_labels):
 336                 return False
 337             for i in range(len(self.nh_labels)):
 338                 if (self.nh_labels[i] != other.label_stack[i]):
 339                     return False
 340             return self.nh_itf == other.sw_if_index
 341         else:
 342             return False
 343
 344     def __ne__(self, other):
 345         return not (self == other)
 346
 347
 348 class VppMRoutePath(VppRoutePath):
 349
 350     def __init__(self, nh_sw_if_index, flags,
 351                  nh=None,
 352                  proto=DpoProto.DPO_PROTO_IP4,
 353                  bier_imp=0):
 354         if not nh:
 355             nh = "::" if proto is DpoProto.DPO_PROTO_IP6 else "0.0.0.0"
 356         super(VppMRoutePath, self).__init__(nh,
 357                                             nh_sw_if_index,
 358                                             proto=proto)
 359         self.nh_i_flags = flags
 360         self.bier_imp = bier_imp
 361
 362
 363 class VppIpRoute(VppObject):
 364     """
 365     IP Route
 366     """
 367
 368     def __init__(self, test, dest_addr,
 369                  dest_addr_len, paths, table_id=0, is_ip6=0, is_local=0,
 370                  is_unreach=0, is_prohibit=0, is_drop=0):
 371         self._test = test
 372         self.paths = paths
 373         self.dest_addr_len = dest_addr_len
 374         self.table_id = table_id
 375         self.is_ip6 = is_ip6
 376         self.is_local = is_local
 377         self.is_unreach = is_unreach
 378         self.is_prohibit = is_prohibit
 379         self.is_drop = is_drop
 380         self.dest_addr_p = dest_addr
 381         if is_ip6:
 382             self.dest_addr = inet_pton(AF_INET6, dest_addr)
 383         else:
 384             self.dest_addr = inet_pton(AF_INET, dest_addr)
 385
 386     def modify(self, paths, is_local=0,
 387                is_unreach=0, is_prohibit=0):
 388         self.paths = paths
 389         self.is_local = is_local
 390         self.is_unreach = is_unreach
 391         self.is_prohibit = is_prohibit
 392
 393     def add_vpp_config(self):
 394         if self.is_local or self.is_unreach or \
 395            self.is_prohibit or self.is_drop:
 396             r = self._test.vapi.ip_add_del_route(
 397                 self.dest_addr,
 398                 self.dest_addr_len,
 399                 inet_pton(AF_INET6, "::"),
 400                 0xffffffff,
 401                 is_local=self.is_local,
 402                 is_unreach=self.is_unreach,
 403                 is_prohibit=self.is_prohibit,
 404                 is_drop=self.is_drop,
 405                 table_id=self.table_id,
 406                 is_ipv6=self.is_ip6)
 407         else:
 408             for path in self.paths:
 409                 lstack = path.encode_labels()
 410
 411                 r = self._test.vapi.ip_add_del_route(
 412                     self.dest_addr,
 413                     self.dest_addr_len,
 414                     path.nh_addr,
 415                     path.nh_itf,
 416                     table_id=self.table_id,
 417                     next_hop_out_label_stack=lstack,
 418                     next_hop_n_out_labels=len(lstack),
 419                     next_hop_via_label=path.nh_via_label,
 420                     next_hop_table_id=path.nh_table_id,
 421                     next_hop_id=path.next_hop_id,
 422                     is_ipv6=self.is_ip6,
 423                     is_dvr=path.is_dvr,
 424                     is_resolve_host=path.is_resolve_host,
 425                     is_resolve_attached=path.is_resolve_attached,
 426                     is_source_lookup=path.is_source_lookup,
 427                     is_udp_encap=path.is_udp_encap,
 428                     is_multipath=1 if len(self.paths) > 1 else 0)
 429         self.stats_index = r.stats_index
 430         self._test.registry.register(self, self._test.logger)
 431
 432     def remove_vpp_config(self):
 433         if self.is_local or self.is_unreach or \
 434            self.is_prohibit or self.is_drop:
 435             self._test.vapi.ip_add_del_route(
 436                 self.dest_addr,
 437                 self.dest_addr_len,
 438                 inet_pton(AF_INET6, "::"),
 439                 0xffffffff,
 440                 is_local=self.is_local,
 441                 is_unreach=self.is_unreach,
 442                 is_prohibit=self.is_prohibit,
 443                 is_add=0,
 444                 table_id=self.table_id,
 445                 is_ipv6=self.is_ip6)
 446         else:
 447             for path in self.paths:
 448                 self._test.vapi.ip_add_del_route(
 449                     self.dest_addr,
 450                     self.dest_addr_len,
 451                     path.nh_addr,
 452                     path.nh_itf,
 453                     table_id=self.table_id,
 454                     next_hop_table_id=path.nh_table_id,
 455                     next_hop_via_label=path.nh_via_label,
 456                     next_hop_id=path.next_hop_id,
 457                     is_add=0,
 458                     is_udp_encap=path.is_udp_encap,
 459                     is_ipv6=self.is_ip6,
 460                     is_dvr=path.is_dvr)
 461
 462     def query_vpp_config(self):
 463         return find_route(self._test,
 464                           self.dest_addr_p,
 465                           self.dest_addr_len,
 466                           self.table_id,
 467                           inet=AF_INET6 if self.is_ip6 == 1 else AF_INET)
 468
 469     def __str__(self):
 470         return self.object_id()
 471
 472     def object_id(self):
 473         return ("%d:%s/%d"
 474                 % (self.table_id,
 475                    self.dest_addr_p,
 476                    self.dest_addr_len))
 477
 478     def get_stats_to(self):
 479         c = self._test.statistics.get_counter("/net/route/to")
 480         return c[0][self.stats_index]
 481
 482     def get_stats_via(self):
 483         c = self._test.statistics.get_counter("/net/route/via")
 484         return c[0][self.stats_index]
 485
 486
 487 class VppIpMRoute(VppObject):
 488     """
 489     IP Multicast Route
 490     """
 491
 492     def __init__(self, test, src_addr, grp_addr,
 493                  grp_addr_len, e_flags, paths, table_id=0,
 494                  rpf_id=0, is_ip6=0):
 495         self._test = test
 496         self.paths = paths
 497         self.grp_addr_len = grp_addr_len
 498         self.table_id = table_id
 499         self.e_flags = e_flags
 500         self.is_ip6 = is_ip6
 501         self.rpf_id = rpf_id
 502
 503         self.grp_addr_p = grp_addr
 504         self.src_addr_p = src_addr
 505         if is_ip6:
 506             self.grp_addr = inet_pton(AF_INET6, grp_addr)
 507             self.src_addr = inet_pton(AF_INET6, src_addr)
 508         else:
 509             self.grp_addr = inet_pton(AF_INET, grp_addr)
 510             self.src_addr = inet_pton(AF_INET, src_addr)
 511
 512     def add_vpp_config(self):
 513         for path in self.paths:
 514             r = self._test.vapi.ip_mroute_add_del(self.src_addr,
 515                                                   self.grp_addr,
 516                                                   self.grp_addr_len,
 517                                                   self.e_flags,
 518                                                   path.proto,
 519                                                   path.nh_itf,
 520                                                   path.nh_addr,
 521                                                   path.nh_i_flags,
 522                                                   bier_imp=path.bier_imp,
 523                                                   rpf_id=self.rpf_id,
 524                                                   table_id=self.table_id,
 525                                                   is_ipv6=self.is_ip6)
 526             self.stats_index = r.stats_index
 527         self._test.registry.register(self, self._test.logger)
 528
 529     def remove_vpp_config(self):
 530         for path in self.paths:
 531             self._test.vapi.ip_mroute_add_del(self.src_addr,
 532                                               self.grp_addr,
 533                                               self.grp_addr_len,
 534                                               self.e_flags,
 535                                               path.proto,
 536                                               path.nh_itf,
 537                                               path.nh_addr,
 538                                               path.nh_i_flags,
 539                                               table_id=self.table_id,
 540                                               bier_imp=path.bier_imp,
 541                                               is_add=0,
 542                                               is_ipv6=self.is_ip6)
 543
 544     def update_entry_flags(self, flags):
 545         self.e_flags = flags
 546         self._test.vapi.ip_mroute_add_del(self.src_addr,
 547                                           self.grp_addr,
 548                                           self.grp_addr_len,
 549                                           self.e_flags,
 550                                           0,
 551                                           0xffffffff,
 552                                           "",
 553                                           0,
 554                                           table_id=self.table_id,
 555                                           is_ipv6=self.is_ip6)
 556
 557     def update_rpf_id(self, rpf_id):
 558         self.rpf_id = rpf_id
 559         self._test.vapi.ip_mroute_add_del(self.src_addr,
 560                                           self.grp_addr,
 561                                           self.grp_addr_len,
 562                                           self.e_flags,
 563                                           0,
 564                                           0xffffffff,
 565                                           "",
 566                                           0,
 567                                           rpf_id=self.rpf_id,
 568                                           table_id=self.table_id,
 569                                           is_ipv6=self.is_ip6)
 570
 571     def update_path_flags(self, itf, flags):
 572         for path in self.paths:
 573             if path.nh_itf == itf:
 574                 path.nh_i_flags = flags
 575                 break
 576         self._test.vapi.ip_mroute_add_del(self.src_addr,
 577                                           self.grp_addr,
 578                                           self.grp_addr_len,
 579                                           self.e_flags,
 580                                           path.proto,
 581                                           path.nh_itf,
 582                                           path.nh_addr,
 583                                           path.nh_i_flags,
 584                                           table_id=self.table_id,
 585                                           is_ipv6=self.is_ip6)
 586
 587     def query_vpp_config(self):
 588         return find_mroute(self._test,
 589                            self.grp_addr_p,
 590                            self.src_addr_p,
 591                            self.grp_addr_len,
 592                            self.table_id,
 593                            inet=AF_INET6 if self.is_ip6 == 1 else AF_INET)
 594
 595     def __str__(self):
 596         return self.object_id()
 597
 598     def object_id(self):
 599         if self.is_ip6:
 600             return ("%d:(%s,%s/%d)"
 601                     % (self.table_id,
 602                        inet_ntop(AF_INET6, self.src_addr),
 603                        inet_ntop(AF_INET6, self.grp_addr),
 604                        self.grp_addr_len))
 605         else:
 606             return ("%d:(%s,%s/%d)"
 607                     % (self.table_id,
 608                        inet_ntop(AF_INET, self.src_addr),
 609                        inet_ntop(AF_INET, self.grp_addr),
 610                        self.grp_addr_len))
 611
 612     def get_stats(self):
 613         c = self._test.statistics.get_counter("/net/mroute")
 614         return c[0][self.stats_index]
 615
 616
 617 class VppMFibSignal(object):
 618     def __init__(self, test, route, interface, packet):
 619         self.route = route
 620         self.interface = interface
 621         self.packet = packet
 622         self.test = test
 623
 624     def compare(self, signal):
 625         self.test.assertEqual(self.interface, signal.sw_if_index)
 626         self.test.assertEqual(self.route.table_id, signal.table_id)
 627         self.test.assertEqual(self.route.grp_addr_len,
 628                               signal.grp_address_len)
 629         for i in range(self.route.grp_addr_len / 8):
 630             self.test.assertEqual(self.route.grp_addr[i],
 631                                   signal.grp_address[i])
 632         if (self.route.grp_addr_len > 32):
 633             for i in range(4):
 634                 self.test.assertEqual(self.route.src_addr[i],
 635                                       signal.src_address[i])
 636
 637
 638 class VppMplsIpBind(VppObject):
 639     """
 640     MPLS to IP Binding
 641     """
 642
 643     def __init__(self, test, local_label, dest_addr, dest_addr_len,
 644                  table_id=0, ip_table_id=0, is_ip6=0):
 645         self._test = test
 646         self.dest_addr_len = dest_addr_len
 647         self.dest_addr = dest_addr
 648         self.local_label = local_label
 649         self.table_id = table_id
 650         self.ip_table_id = ip_table_id
 651         self.is_ip6 = is_ip6
 652         if is_ip6:
 653             self.dest_addrn = inet_pton(AF_INET6, dest_addr)
 654         else:
 655             self.dest_addrn = inet_pton(AF_INET, dest_addr)
 656
 657     def add_vpp_config(self):
 658         self._test.vapi.mpls_ip_bind_unbind(self.local_label,
 659                                             self.dest_addrn,
 660                                             self.dest_addr_len,
 661                                             table_id=self.table_id,
 662                                             ip_table_id=self.ip_table_id,
 663                                             is_ip4=(self.is_ip6 == 0))
 664         self._test.registry.register(self, self._test.logger)
 665
 666     def remove_vpp_config(self):
 667         self._test.vapi.mpls_ip_bind_unbind(self.local_label,
 668                                             self.dest_addrn,
 669                                             self.dest_addr_len,
 670                                             table_id=self.table_id,
 671                                             ip_table_id=self.ip_table_id,
 672                                             is_bind=0,
 673                                             is_ip4=(self.is_ip6 == 0))
 674
 675     def query_vpp_config(self):
 676         dump = self._test.vapi.mpls_fib_dump()
 677         for e in dump:
 678             if self.local_label == e.label \
 679                and self.table_id == e.table_id:
 680                 return True
 681         return False
 682
 683     def __str__(self):
 684         return self.object_id()
 685
 686     def object_id(self):
 687         return ("%d:%s binds %d:%s/%d"
 688                 % (self.table_id,
 689                    self.local_label,
 690                    self.ip_table_id,
 691                    self.dest_addr,
 692                    self.dest_addr_len))
 693
 694
 695 class VppMplsTable(VppObject):
 696
 697     def __init__(self,
 698                  test,
 699                  table_id):
 700         self._test = test
 701         self.table_id = table_id
 702
 703     def add_vpp_config(self):
 704         self._test.vapi.mpls_table_add_del(
 705             self.table_id,
 706             is_add=1)
 707         self._test.registry.register(self, self._test.logger)
 708
 709     def remove_vpp_config(self):
 710         self._test.vapi.mpls_table_add_del(
 711             self.table_id,
 712             is_add=0)
 713
 714     def query_vpp_config(self):
 715         # find the default route
 716         dump = self._test.vapi.mpls_fib_dump()
 717         if len(dump):
 718             return True
 719         return False
 720
 721     def __str__(self):
 722         return self.object_id()
 723
 724     def object_id(self):
 725         return ("table-mpls-%d" % (self.table_id))
 726
 727
 728 class VppMplsRoute(VppObject):
 729     """
 730     MPLS Route/LSP
 731     """
 732
 733     def __init__(self, test, local_label, eos_bit, paths, table_id=0,
 734                  is_multicast=0):
 735         self._test = test
 736         self.paths = paths
 737         self.local_label = local_label
 738         self.eos_bit = eos_bit
 739         self.table_id = table_id
 740         self.is_multicast = is_multicast
 741
 742     def add_vpp_config(self):
 743         is_multipath = len(self.paths) > 1
 744         for path in self.paths:
 745             lstack = path.encode_labels()
 746
 747             r = self._test.vapi.mpls_route_add_del(
 748                 self.local_label,
 749                 self.eos_bit,
 750                 path.proto,
 751                 path.nh_addr,
 752                 path.nh_itf,
 753                 is_multicast=self.is_multicast,
 754                 is_multipath=is_multipath,
 755                 table_id=self.table_id,
 756                 is_interface_rx=path.is_interface_rx,
 757                 is_rpf_id=path.is_rpf_id,
 758                 next_hop_out_label_stack=lstack,
 759                 next_hop_n_out_labels=len(lstack),
 760                 next_hop_via_label=path.nh_via_label,
 761                 next_hop_table_id=path.nh_table_id)
 762         self.stats_index = r.stats_index
 763         self._test.registry.register(self, self._test.logger)
 764
 765     def remove_vpp_config(self):
 766         for path in self.paths:
 767             self._test.vapi.mpls_route_add_del(self.local_label,
 768                                                self.eos_bit,
 769                                                path.proto,
 770                                                path.nh_addr,
 771                                                path.nh_itf,
 772                                                is_rpf_id=path.is_rpf_id,
 773                                                table_id=self.table_id,
 774                                                is_add=0)
 775
 776     def query_vpp_config(self):
 777         return find_mpls_route(self._test, self.table_id,
 778                                self.local_label, self.eos_bit)
 779
 780     def __str__(self):
 781         return self.object_id()
 782
 783     def object_id(self):
 784         return ("%d:%s/%d"
 785                 % (self.table_id,
 786                    self.local_label,
 787                    20+self.eos_bit))
 788
 789     def get_stats_to(self):
 790         c = self._test.statistics.get_counter("/net/route/to")
 791         return c[0][self.stats_index]
 792
 793     def get_stats_via(self):
 794         c = self._test.statistics.get_counter("/net/route/via")
 795         return c[0][self.stats_index]