5a8123bda0
Add IP[46] MFIB dump. Change-Id: I4a2821f65e67a5416b291e4912c84f64989883b8 Signed-off-by: Neale Ranns <nranns@cisco.com>
347 lines
12 KiB
Python
347 lines
12 KiB
Python
"""
|
|
IP Routes
|
|
|
|
object abstractions for representing IP routes in VPP
|
|
"""
|
|
|
|
import socket
|
|
from vpp_object import *
|
|
|
|
# from vnet/vnet/mpls/mpls_types.h
|
|
MPLS_IETF_MAX_LABEL = 0xfffff
|
|
MPLS_LABEL_INVALID = MPLS_IETF_MAX_LABEL + 1
|
|
|
|
|
|
class VppRoutePath(object):
|
|
|
|
def __init__(
|
|
self,
|
|
nh_addr,
|
|
nh_sw_if_index,
|
|
nh_table_id=0,
|
|
labels=[],
|
|
nh_via_label=MPLS_LABEL_INVALID,
|
|
is_ip6=0):
|
|
self.nh_itf = nh_sw_if_index
|
|
self.nh_table_id = nh_table_id
|
|
self.nh_via_label = nh_via_label
|
|
self.nh_labels = labels
|
|
if is_ip6:
|
|
self.nh_addr = socket.inet_pton(socket.AF_INET6, nh_addr)
|
|
else:
|
|
self.nh_addr = socket.inet_pton(socket.AF_INET, nh_addr)
|
|
|
|
|
|
class VppMRoutePath(VppRoutePath):
|
|
|
|
def __init__(self, nh_sw_if_index, flags):
|
|
super(VppMRoutePath, self).__init__("0.0.0.0",
|
|
nh_sw_if_index)
|
|
self.nh_i_flags = flags
|
|
|
|
|
|
class VppIpRoute(VppObject):
|
|
"""
|
|
IP Route
|
|
"""
|
|
|
|
def __init__(self, test, dest_addr,
|
|
dest_addr_len, paths, table_id=0, is_ip6=0, is_local=0):
|
|
self._test = test
|
|
self.paths = paths
|
|
self.dest_addr_len = dest_addr_len
|
|
self.table_id = table_id
|
|
self.is_ip6 = is_ip6
|
|
self.is_local = is_local
|
|
if is_ip6:
|
|
self.dest_addr = socket.inet_pton(socket.AF_INET6, dest_addr)
|
|
else:
|
|
self.dest_addr = socket.inet_pton(socket.AF_INET, dest_addr)
|
|
|
|
def add_vpp_config(self):
|
|
if self.is_local:
|
|
self._test.vapi.ip_add_del_route(
|
|
self.dest_addr,
|
|
self.dest_addr_len,
|
|
socket.inet_pton(socket.AF_INET6, "::"),
|
|
0xffffffff,
|
|
is_local=1,
|
|
table_id=self.table_id,
|
|
is_ipv6=self.is_ip6)
|
|
else:
|
|
for path in self.paths:
|
|
self._test.vapi.ip_add_del_route(
|
|
self.dest_addr,
|
|
self.dest_addr_len,
|
|
path.nh_addr,
|
|
path.nh_itf,
|
|
table_id=self.table_id,
|
|
next_hop_out_label_stack=path.nh_labels,
|
|
next_hop_n_out_labels=len(
|
|
path.nh_labels),
|
|
next_hop_via_label=path.nh_via_label,
|
|
is_ipv6=self.is_ip6)
|
|
self._test.registry.register(self, self._test.logger)
|
|
|
|
def remove_vpp_config(self):
|
|
if self.is_local:
|
|
self._test.vapi.ip_add_del_route(
|
|
self.dest_addr,
|
|
self.dest_addr_len,
|
|
socket.inet_pton(socket.AF_INET6, "::"),
|
|
0xffffffff,
|
|
is_local=1,
|
|
is_add=0,
|
|
table_id=self.table_id,
|
|
is_ipv6=self.is_ip6)
|
|
else:
|
|
for path in self.paths:
|
|
self._test.vapi.ip_add_del_route(self.dest_addr,
|
|
self.dest_addr_len,
|
|
path.nh_addr,
|
|
path.nh_itf,
|
|
table_id=self.table_id,
|
|
is_add=0)
|
|
|
|
def query_vpp_config(self):
|
|
dump = self._test.vapi.ip_fib_dump()
|
|
for e in dump:
|
|
if self.dest_addr == e.address \
|
|
and self.dest_addr_len == e.address_length \
|
|
and self.table_id == e.table_id:
|
|
return True
|
|
return False
|
|
|
|
def __str__(self):
|
|
return self.object_id()
|
|
|
|
def object_id(self):
|
|
return ("%d:%s/%d"
|
|
% (self.table_id,
|
|
socket.inet_ntop(socket.AF_INET, self.dest_addr),
|
|
self.dest_addr_len))
|
|
|
|
|
|
class VppIpMRoute(VppObject):
|
|
"""
|
|
IP Multicast Route
|
|
"""
|
|
|
|
def __init__(self, test, src_addr, grp_addr,
|
|
grp_addr_len, e_flags, paths, table_id=0, is_ip6=0):
|
|
self._test = test
|
|
self.paths = paths
|
|
self.grp_addr_len = grp_addr_len
|
|
self.table_id = table_id
|
|
self.e_flags = e_flags
|
|
self.is_ip6 = is_ip6
|
|
|
|
if is_ip6:
|
|
self.grp_addr = socket.inet_pton(socket.AF_INET6, grp_addr)
|
|
self.src_addr = socket.inet_pton(socket.AF_INET6, src_addr)
|
|
else:
|
|
self.grp_addr = socket.inet_pton(socket.AF_INET, grp_addr)
|
|
self.src_addr = socket.inet_pton(socket.AF_INET, src_addr)
|
|
|
|
def add_vpp_config(self):
|
|
for path in self.paths:
|
|
self._test.vapi.ip_mroute_add_del(self.src_addr,
|
|
self.grp_addr,
|
|
self.grp_addr_len,
|
|
self.e_flags,
|
|
path.nh_itf,
|
|
path.nh_i_flags,
|
|
table_id=self.table_id,
|
|
is_ipv6=self.is_ip6)
|
|
self._test.registry.register(self, self._test.logger)
|
|
|
|
def remove_vpp_config(self):
|
|
for path in self.paths:
|
|
self._test.vapi.ip_mroute_add_del(self.src_addr,
|
|
self.grp_addr,
|
|
self.grp_addr_len,
|
|
self.e_flags,
|
|
path.nh_itf,
|
|
path.nh_i_flags,
|
|
table_id=self.table_id,
|
|
is_add=0,
|
|
is_ipv6=self.is_ip6)
|
|
|
|
def update_entry_flags(self, flags):
|
|
self.e_flags = flags
|
|
self._test.vapi.ip_mroute_add_del(self.src_addr,
|
|
self.grp_addr,
|
|
self.grp_addr_len,
|
|
self.e_flags,
|
|
0xffffffff,
|
|
0,
|
|
table_id=self.table_id,
|
|
is_ipv6=self.is_ip6)
|
|
|
|
def update_path_flags(self, itf, flags):
|
|
for path in self.paths:
|
|
if path.nh_itf == itf:
|
|
path.nh_i_flags = flags
|
|
break
|
|
self._test.vapi.ip_mroute_add_del(self.src_addr,
|
|
self.grp_addr,
|
|
self.grp_addr_len,
|
|
self.e_flags,
|
|
path.nh_itf,
|
|
path.nh_i_flags,
|
|
table_id=self.table_id,
|
|
is_ipv6=self.is_ip6)
|
|
|
|
def query_vpp_config(self):
|
|
dump = self._test.vapi.ip_fib_dump()
|
|
for e in dump:
|
|
if self.grp_addr == e.address \
|
|
and self.grp_addr_len == e.address_length \
|
|
and self.table_id == e.table_id:
|
|
return True
|
|
return False
|
|
|
|
def __str__(self):
|
|
return self.object_id()
|
|
|
|
def object_id(self):
|
|
if self.is_ip6:
|
|
return ("%d:(%s,%s/%d)"
|
|
% (self.table_id,
|
|
socket.inet_ntop(socket.AF_INET6, self.src_addr),
|
|
socket.inet_ntop(socket.AF_INET6, self.grp_addr),
|
|
self.grp_addr_len))
|
|
else:
|
|
return ("%d:(%s,%s/%d)"
|
|
% (self.table_id,
|
|
socket.inet_ntop(socket.AF_INET, self.src_addr),
|
|
socket.inet_ntop(socket.AF_INET, self.grp_addr),
|
|
self.grp_addr_len))
|
|
|
|
|
|
class VppMFibSignal(object):
|
|
def __init__(self, test, route, interface, packet):
|
|
self.route = route
|
|
self.interface = interface
|
|
self.packet = packet
|
|
self.test = test
|
|
|
|
def compare(self, signal):
|
|
self.test.assertEqual(self.interface, signal.sw_if_index)
|
|
self.test.assertEqual(self.route.table_id, signal.table_id)
|
|
self.test.assertEqual(self.route.grp_addr_len,
|
|
signal.grp_address_len)
|
|
for i in range(self.route.grp_addr_len / 8):
|
|
self.test.assertEqual(self.route.grp_addr[i],
|
|
signal.grp_address[i])
|
|
if (self.route.grp_addr_len > 32):
|
|
for i in range(4):
|
|
self.test.assertEqual(self.route.src_addr[i],
|
|
signal.src_address[i])
|
|
|
|
|
|
class VppMplsIpBind(VppObject):
|
|
"""
|
|
MPLS to IP Binding
|
|
"""
|
|
|
|
def __init__(self, test, local_label, dest_addr, dest_addr_len,
|
|
table_id=0, ip_table_id=0):
|
|
self._test = test
|
|
self.dest_addr = socket.inet_pton(socket.AF_INET, dest_addr)
|
|
self.dest_addr_len = dest_addr_len
|
|
self.local_label = local_label
|
|
self.table_id = table_id
|
|
self.ip_table_id = ip_table_id
|
|
|
|
def add_vpp_config(self):
|
|
self._test.vapi.mpls_ip_bind_unbind(self.local_label,
|
|
self.dest_addr,
|
|
self.dest_addr_len,
|
|
table_id=self.table_id,
|
|
ip_table_id=self.ip_table_id)
|
|
self._test.registry.register(self, self._test.logger)
|
|
|
|
def remove_vpp_config(self):
|
|
self._test.vapi.mpls_ip_bind_unbind(self.local_label,
|
|
self.dest_addr,
|
|
self.dest_addr_len,
|
|
is_bind=0)
|
|
|
|
def query_vpp_config(self):
|
|
dump = self._test.vapi.mpls_fib_dump()
|
|
for e in dump:
|
|
if self.local_label == e.label \
|
|
and self.eos_bit == e.eos_bit \
|
|
and self.table_id == e.table_id:
|
|
return True
|
|
return False
|
|
|
|
def __str__(self):
|
|
return self.object_id()
|
|
|
|
def object_id(self):
|
|
return ("%d:%s binds %d:%s/%d"
|
|
% (self.table_id,
|
|
self.local_label,
|
|
self.ip_table_id,
|
|
socket.inet_ntop(socket.AF_INET, self.dest_addr),
|
|
self.dest_addr_len))
|
|
|
|
|
|
class VppMplsRoute(VppObject):
|
|
"""
|
|
MPLS Route/LSP
|
|
"""
|
|
|
|
def __init__(self, test, local_label, eos_bit, paths, table_id=0):
|
|
self._test = test
|
|
self.paths = paths
|
|
self.local_label = local_label
|
|
self.eos_bit = eos_bit
|
|
self.table_id = table_id
|
|
|
|
def add_vpp_config(self):
|
|
for path in self.paths:
|
|
self._test.vapi.mpls_route_add_del(
|
|
self.local_label,
|
|
self.eos_bit,
|
|
1,
|
|
path.nh_addr,
|
|
path.nh_itf,
|
|
table_id=self.table_id,
|
|
next_hop_out_label_stack=path.nh_labels,
|
|
next_hop_n_out_labels=len(
|
|
path.nh_labels),
|
|
next_hop_via_label=path.nh_via_label,
|
|
next_hop_table_id=path.nh_table_id)
|
|
self._test.registry.register(self, self._test.logger)
|
|
|
|
def remove_vpp_config(self):
|
|
for path in self.paths:
|
|
self._test.vapi.mpls_route_add_del(self.local_label,
|
|
self.eos_bit,
|
|
1,
|
|
path.nh_addr,
|
|
path.nh_itf,
|
|
table_id=self.table_id,
|
|
is_add=0)
|
|
|
|
def query_vpp_config(self):
|
|
dump = self._test.vapi.mpls_fib_dump()
|
|
for e in dump:
|
|
if self.local_label == e.label \
|
|
and self.eos_bit == e.eos_bit \
|
|
and self.table_id == e.table_id:
|
|
return True
|
|
return False
|
|
|
|
def __str__(self):
|
|
return self.object_id()
|
|
|
|
def object_id(self):
|
|
return ("%d:%s/%d"
|
|
% (self.table_id,
|
|
self.local_label,
|
|
20+self.eos_bit))
|