4 object abstractions for representing IP routes in VPP
7 from vpp_object import *
8 from socket import inet_pton, inet_ntop, AF_INET, AF_INET6
11 # from vnet/vnet/mpls/mpls_types.h
12 MPLS_IETF_MAX_LABEL = 0xfffff
13 MPLS_LABEL_INVALID = MPLS_IETF_MAX_LABEL + 1
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
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
38 def ip_to_dpo_proto(addr):
40 return DpoProto.DPO_PROTO_IP6
42 return DpoProto.DPO_PROTO_IP4
45 def find_route(test, ip_addr, len, table_id=0, inet=AF_INET):
48 routes = test.vapi.ip_fib_dump()
51 routes = test.vapi.ip6_fib_dump()
53 route_addr = inet_pton(inet, ip_addr)
55 if route_addr == e.address[:s] \
56 and len == e.address_length \
57 and table_id == e.table_id:
62 def find_mroute(test, grp_addr, src_addr, grp_addr_len,
63 table_id=0, inet=AF_INET):
66 routes = test.vapi.ip_mfib_dump()
69 routes = test.vapi.ip6_mfib_dump()
70 gaddr = inet_pton(inet, grp_addr)
71 saddr = inet_pton(inet, src_addr)
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:
81 def find_mpls_route(test, table_id, label, eos_bit, paths=None):
82 dump = test.vapi.mpls_fib_dump()
85 and eos_bit == e.eos_bit \
86 and table_id == e.table_id:
90 if (len(paths) != len(e.path)):
92 for i in range(len(paths)):
93 if (paths[i] != e.path[i]):
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)
109 if a.prefix_length == length and \
110 a.sw_if_index == sw_if_index and \
111 a.ip[:n] == vp.bytes:
116 class VppIpTable(VppObject):
123 self.table_id = table_id
126 def add_vpp_config(self):
127 self._test.vapi.ip_table_add_del(
131 self._test.registry.register(self, self._test.logger)
133 def remove_vpp_config(self):
134 self._test.vapi.ip_table_add_del(
139 def query_vpp_config(self):
140 if self.table_id == 0:
141 # the default table always exists
143 # find the default route
144 return find_route(self._test,
145 "::" if self.is_ip6 else "0.0.0.0",
148 inet=AF_INET6 if self.is_ip6 == 1 else AF_INET)
151 return self.object_id()
154 return ("table-%s-%d" %
155 ("v6" if self.is_ip6 == 1 else "v4",
159 class VppIpInterfaceAddress(VppObject):
161 def __init__(self, test, intf, addr, len):
164 self.prefix = VppIpPrefix(addr, len)
166 def add_vpp_config(self):
167 self._test.vapi.sw_interface_add_del_address(
168 self.intf.sw_if_index,
172 is_ipv6=self.prefix.is_ip6)
173 self._test.registry.register(self, self._test.logger)
175 def remove_vpp_config(self):
176 self._test.vapi.sw_interface_add_del_address(
177 self.intf.sw_if_index,
181 is_ipv6=self.prefix.is_ip6)
183 def query_vpp_config(self):
184 return fib_interface_ip_prefix(self._test,
187 self.intf.sw_if_index)
190 return self.object_id()
193 return "interface-ip-%s-%s" % (self.intf, self.prefix)
196 class VppIpInterfaceBind(VppObject):
198 def __init__(self, test, intf, table):
203 def add_vpp_config(self):
204 if self.table.is_ip6:
205 self.intf.set_table_ip6(self.table.table_id)
207 self.intf.set_table_ip4(self.table.table_id)
208 self._test.registry.register(self, self._test.logger)
210 def remove_vpp_config(self):
211 if 0 == self.table.table_id:
213 if self.table.is_ip6:
214 self.intf.set_table_ip6(0)
216 self.intf.set_table_ip4(0)
218 def query_vpp_config(self):
219 if 0 == self.table.table_id:
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
226 return self.object_id()
229 return "interface-bind-%s-%s" % (self.intf, self.table)
232 class VppMplsLabel(object):
233 def __init__(self, value, mode=MplsLspMode.PIPE, ttl=64, exp=0):
240 is_uniform = 0 if self.mode is MplsLspMode.PIPE else 1
241 return {'label': self.value,
244 'is_uniform': is_uniform}
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)
260 def __ne__(self, other):
261 return not (self == other)
264 class VppRoutePath(object):
272 nh_via_label=MPLS_LABEL_INVALID,
276 is_resolve_attached=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
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)
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
303 self.is_udp_encap = is_udp_encap
304 self.next_hop_id = next_hop_id
307 def encode_labels(self):
309 for l in self.nh_labels:
310 if type(l) == VppMplsLabel:
311 lstack.append(l.encode())
313 lstack.append({'label': l,
318 return {'next_hop': self.nh_addr,
322 'table_id': self.nh_table_id,
323 'next_hop_id': self.next_hop_id,
324 'sw_if_index': self.nh_itf,
326 'is_udp_encap': self.is_udp_encap,
327 'n_labels': len(self.nh_labels),
328 'label_stack': self.encode_labels()}
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'):
335 if (len(self.nh_labels) != other.n_labels):
337 for i in range(len(self.nh_labels)):
338 if (self.nh_labels[i] != other.label_stack[i]):
340 return self.nh_itf == other.sw_if_index
344 def __ne__(self, other):
345 return not (self == other)
348 class VppMRoutePath(VppRoutePath):
350 def __init__(self, nh_sw_if_index, flags,
352 proto=DpoProto.DPO_PROTO_IP4,
355 nh = "::" if proto is DpoProto.DPO_PROTO_IP6 else "0.0.0.0"
356 super(VppMRoutePath, self).__init__(nh,
359 self.nh_i_flags = flags
360 self.bier_imp = bier_imp
363 class VppIpRoute(VppObject):
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):
373 self.dest_addr_len = dest_addr_len
374 self.table_id = table_id
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
382 self.dest_addr = inet_pton(AF_INET6, dest_addr)
384 self.dest_addr = inet_pton(AF_INET, dest_addr)
386 def modify(self, paths, is_local=0,
387 is_unreach=0, is_prohibit=0):
389 self.is_local = is_local
390 self.is_unreach = is_unreach
391 self.is_prohibit = is_prohibit
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(
399 inet_pton(AF_INET6, "::"),
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,
408 for path in self.paths:
409 lstack = path.encode_labels()
411 r = self._test.vapi.ip_add_del_route(
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,
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)
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(
438 inet_pton(AF_INET6, "::"),
440 is_local=self.is_local,
441 is_unreach=self.is_unreach,
442 is_prohibit=self.is_prohibit,
444 table_id=self.table_id,
447 for path in self.paths:
448 self._test.vapi.ip_add_del_route(
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,
458 is_udp_encap=path.is_udp_encap,
462 def query_vpp_config(self):
463 return find_route(self._test,
467 inet=AF_INET6 if self.is_ip6 == 1 else AF_INET)
470 return self.object_id()
478 def get_stats_to(self):
479 c = self._test.statistics.get_counter("/net/route/to")
480 return c[0][self.stats_index]
482 def get_stats_via(self):
483 c = self._test.statistics.get_counter("/net/route/via")
484 return c[0][self.stats_index]
487 class VppIpMRoute(VppObject):
492 def __init__(self, test, src_addr, grp_addr,
493 grp_addr_len, e_flags, paths, table_id=0,
497 self.grp_addr_len = grp_addr_len
498 self.table_id = table_id
499 self.e_flags = e_flags
503 self.grp_addr_p = grp_addr
504 self.src_addr_p = src_addr
506 self.grp_addr = inet_pton(AF_INET6, grp_addr)
507 self.src_addr = inet_pton(AF_INET6, src_addr)
509 self.grp_addr = inet_pton(AF_INET, grp_addr)
510 self.src_addr = inet_pton(AF_INET, src_addr)
512 def add_vpp_config(self):
513 for path in self.paths:
514 r = self._test.vapi.ip_mroute_add_del(self.src_addr,
522 bier_imp=path.bier_imp,
524 table_id=self.table_id,
526 self.stats_index = r.stats_index
527 self._test.registry.register(self, self._test.logger)
529 def remove_vpp_config(self):
530 for path in self.paths:
531 self._test.vapi.ip_mroute_add_del(self.src_addr,
539 table_id=self.table_id,
540 bier_imp=path.bier_imp,
544 def update_entry_flags(self, flags):
546 self._test.vapi.ip_mroute_add_del(self.src_addr,
554 table_id=self.table_id,
557 def update_rpf_id(self, rpf_id):
559 self._test.vapi.ip_mroute_add_del(self.src_addr,
568 table_id=self.table_id,
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
576 self._test.vapi.ip_mroute_add_del(self.src_addr,
584 table_id=self.table_id,
587 def query_vpp_config(self):
588 return find_mroute(self._test,
593 inet=AF_INET6 if self.is_ip6 == 1 else AF_INET)
596 return self.object_id()
600 return ("%d:(%s,%s/%d)"
602 inet_ntop(AF_INET6, self.src_addr),
603 inet_ntop(AF_INET6, self.grp_addr),
606 return ("%d:(%s,%s/%d)"
608 inet_ntop(AF_INET, self.src_addr),
609 inet_ntop(AF_INET, self.grp_addr),
613 c = self._test.statistics.get_counter("/net/mroute")
614 return c[0][self.stats_index]
617 class VppMFibSignal(object):
618 def __init__(self, test, route, interface, packet):
620 self.interface = interface
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):
634 self.test.assertEqual(self.route.src_addr[i],
635 signal.src_address[i])
638 class VppMplsIpBind(VppObject):
643 def __init__(self, test, local_label, dest_addr, dest_addr_len,
644 table_id=0, ip_table_id=0, is_ip6=0):
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
653 self.dest_addrn = inet_pton(AF_INET6, dest_addr)
655 self.dest_addrn = inet_pton(AF_INET, dest_addr)
657 def add_vpp_config(self):
658 self._test.vapi.mpls_ip_bind_unbind(self.local_label,
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)
666 def remove_vpp_config(self):
667 self._test.vapi.mpls_ip_bind_unbind(self.local_label,
670 table_id=self.table_id,
671 ip_table_id=self.ip_table_id,
673 is_ip4=(self.is_ip6 == 0))
675 def query_vpp_config(self):
676 dump = self._test.vapi.mpls_fib_dump()
678 if self.local_label == e.label \
679 and self.table_id == e.table_id:
684 return self.object_id()
687 return ("%d:%s binds %d:%s/%d"
695 class VppMplsTable(VppObject):
701 self.table_id = table_id
703 def add_vpp_config(self):
704 self._test.vapi.mpls_table_add_del(
707 self._test.registry.register(self, self._test.logger)
709 def remove_vpp_config(self):
710 self._test.vapi.mpls_table_add_del(
714 def query_vpp_config(self):
715 # find the default route
716 dump = self._test.vapi.mpls_fib_dump()
722 return self.object_id()
725 return ("table-mpls-%d" % (self.table_id))
728 class VppMplsRoute(VppObject):
733 def __init__(self, test, local_label, eos_bit, paths, table_id=0,
737 self.local_label = local_label
738 self.eos_bit = eos_bit
739 self.table_id = table_id
740 self.is_multicast = is_multicast
742 def add_vpp_config(self):
743 is_multipath = len(self.paths) > 1
744 for path in self.paths:
745 lstack = path.encode_labels()
747 r = self._test.vapi.mpls_route_add_del(
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)
765 def remove_vpp_config(self):
766 for path in self.paths:
767 self._test.vapi.mpls_route_add_del(self.local_label,
772 is_rpf_id=path.is_rpf_id,
773 table_id=self.table_id,
776 def query_vpp_config(self):
777 return find_mpls_route(self._test, self.table_id,
778 self.local_label, self.eos_bit)
781 return self.object_id()
789 def get_stats_to(self):
790 c = self._test.statistics.get_counter("/net/route/to")
791 return c[0][self.stats_index]
793 def get_stats_via(self):
794 c = self._test.statistics.get_counter("/net/route/via")
795 return c[0][self.stats_index]