34fa0ce8f7
Check and skip VPP_EXCLUDED_PLUGINS tests for most of plugins. Type: improvement Signed-off-by: Dmitry Valter <d-valter@yandex-team.com> Change-Id: I23fd3666729251c639aa8da72a676058e3f5bb4e
925 lines
32 KiB
Python
925 lines
32 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import unittest
|
|
|
|
from framework import VppTestCase
|
|
from asfframework import VppTestRunner
|
|
from vpp_ip_route import VppIpRoute, VppRoutePath
|
|
|
|
import scapy.compat
|
|
from scapy.packet import Raw
|
|
from scapy.layers.l2 import Ether, Dot1Q
|
|
from scapy.layers.inet6 import IPv6, UDP, IPv6ExtHdrSegmentRouting
|
|
from scapy.layers.inet import IP, UDP
|
|
from config import config
|
|
|
|
from util import ppp
|
|
|
|
|
|
@unittest.skipIf("srv6-as" in config.excluded_plugins, "Exclude srv6-as plugin tests")
|
|
class TestSRv6As(VppTestCase):
|
|
"""SRv6 Static Proxy plugin Test Case"""
|
|
|
|
@classmethod
|
|
def setUpClass(self):
|
|
super(TestSRv6As, self).setUpClass()
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
super(TestSRv6As, cls).tearDownClass()
|
|
|
|
def setUp(self):
|
|
"""Perform test setup before each test case."""
|
|
super(TestSRv6As, self).setUp()
|
|
|
|
# packet sizes, inclusive L2 overhead
|
|
self.pg_packet_sizes = [64, 512, 1518, 9018]
|
|
|
|
# reset packet_infos
|
|
self.reset_packet_infos()
|
|
|
|
def tearDown(self):
|
|
"""Clean up test setup after each test case."""
|
|
self.teardown_interfaces()
|
|
|
|
super(TestSRv6As, self).tearDown()
|
|
|
|
def configure_interface(
|
|
self, interface, ipv6=False, ipv4=False, ipv6_table_id=0, ipv4_table_id=0
|
|
):
|
|
"""Configure interface.
|
|
:param ipv6: configure IPv6 on interface
|
|
:param ipv4: configure IPv4 on interface
|
|
:param ipv6_table_id: FIB table_id for IPv6
|
|
:param ipv4_table_id: FIB table_id for IPv4
|
|
"""
|
|
self.logger.debug("Configuring interface %s" % (interface.name))
|
|
if ipv6:
|
|
self.logger.debug("Configuring IPv6")
|
|
interface.set_table_ip6(ipv6_table_id)
|
|
interface.config_ip6()
|
|
interface.resolve_ndp(timeout=5)
|
|
if ipv4:
|
|
self.logger.debug("Configuring IPv4")
|
|
interface.set_table_ip4(ipv4_table_id)
|
|
interface.config_ip4()
|
|
interface.resolve_arp()
|
|
interface.admin_up()
|
|
|
|
def setup_interfaces(self, ipv6=[], ipv4=[], ipv6_table_id=[], ipv4_table_id=[]):
|
|
"""Create and configure interfaces.
|
|
|
|
:param ipv6: list of interface IPv6 capabilities
|
|
:param ipv4: list of interface IPv4 capabilities
|
|
:param ipv6_table_id: list of intf IPv6 FIB table_ids
|
|
:param ipv4_table_id: list of intf IPv4 FIB table_ids
|
|
:returns: List of created interfaces.
|
|
"""
|
|
# how many interfaces?
|
|
if len(ipv6):
|
|
count = len(ipv6)
|
|
else:
|
|
count = len(ipv4)
|
|
self.logger.debug("Creating and configuring %d interfaces" % (count))
|
|
|
|
# fill up ipv6 and ipv4 lists if needed
|
|
# not enabled (False) is the default
|
|
if len(ipv6) < count:
|
|
ipv6 += (count - len(ipv6)) * [False]
|
|
if len(ipv4) < count:
|
|
ipv4 += (count - len(ipv4)) * [False]
|
|
|
|
# fill up table_id lists if needed
|
|
# table_id 0 (global) is the default
|
|
if len(ipv6_table_id) < count:
|
|
ipv6_table_id += (count - len(ipv6_table_id)) * [0]
|
|
if len(ipv4_table_id) < count:
|
|
ipv4_table_id += (count - len(ipv4_table_id)) * [0]
|
|
|
|
# create 'count' pg interfaces
|
|
self.create_pg_interfaces(range(count))
|
|
|
|
# setup all interfaces
|
|
for i in range(count):
|
|
intf = self.pg_interfaces[i]
|
|
self.configure_interface(
|
|
intf, ipv6[i], ipv4[i], ipv6_table_id[i], ipv4_table_id[i]
|
|
)
|
|
|
|
if any(ipv6):
|
|
self.logger.debug(self.vapi.cli("show ip6 neighbors"))
|
|
if any(ipv4):
|
|
self.logger.debug(self.vapi.cli("show ip4 neighbors"))
|
|
self.logger.debug(self.vapi.cli("show interface"))
|
|
self.logger.debug(self.vapi.cli("show hardware"))
|
|
|
|
return self.pg_interfaces
|
|
|
|
def teardown_interfaces(self):
|
|
"""Unconfigure and bring down interface."""
|
|
self.logger.debug("Tearing down interfaces")
|
|
# tear down all interfaces
|
|
# AFAIK they cannot be deleted
|
|
for i in self.pg_interfaces:
|
|
self.logger.debug("Tear down interface %s" % (i.name))
|
|
i.admin_down()
|
|
i.unconfig()
|
|
i.set_table_ip4(0)
|
|
i.set_table_ip6(0)
|
|
|
|
def test_SRv6_End_AS_IPv6_noSRH(self):
|
|
"""Test SRv6 End.AS behavior with IPv6 traffic and no SRH rewrite."""
|
|
self.run_SRv6_End_AS_IPv6(
|
|
sid_list=["a1::", "a2::a6", "a3::"],
|
|
test_sid_index=1,
|
|
rewrite_src_addr="a2::",
|
|
)
|
|
|
|
def test_SRv6_End_AS_IPv6_SRH(self):
|
|
"""Test SRv6 End.AS behavior with IPv6 traffic and SRH rewrite."""
|
|
self.run_SRv6_End_AS_IPv6(
|
|
sid_list=["a1::a6", "a2::", "a3::"],
|
|
test_sid_index=0,
|
|
rewrite_src_addr="a1::",
|
|
)
|
|
|
|
def test_SRv6_End_AS_IPv4_noSRH(self):
|
|
"""Test SRv6 End.AS behavior with IPv4 traffic and no SRH rewrite."""
|
|
self.run_SRv6_End_AS_IPv4(
|
|
sid_list=["a1::", "a2::a6", "a3::"],
|
|
test_sid_index=1,
|
|
rewrite_src_addr="a2::",
|
|
)
|
|
|
|
def test_SRv6_End_AS_IPv4_SRH(self):
|
|
"""Test SRv6 End.AS behavior with IPv4 traffic and SRH rewrite."""
|
|
self.run_SRv6_End_AS_IPv4(
|
|
sid_list=["a1::a6", "a2::", "a3::"],
|
|
test_sid_index=0,
|
|
rewrite_src_addr="a1::",
|
|
)
|
|
|
|
def test_SRv6_End_AS_L2_noSRH(self):
|
|
"""Test SRv6 End.AS behavior with L2 traffic and no SRH rewrite."""
|
|
self.run_SRv6_End_AS_L2(
|
|
sid_list=["a1::", "a2::a6", "a3::"],
|
|
test_sid_index=1,
|
|
rewrite_src_addr="a2::",
|
|
)
|
|
|
|
def test_SRv6_End_AS_L2_SRH(self):
|
|
"""Test SRv6 End.AS behavior with L2 traffic and SRH rewrite."""
|
|
self.run_SRv6_End_AS_L2(
|
|
sid_list=["a1::a6", "a2::", "a3::"],
|
|
test_sid_index=0,
|
|
rewrite_src_addr="a1::",
|
|
)
|
|
|
|
def run_SRv6_End_AS_L2(self, sid_list, test_sid_index, rewrite_src_addr):
|
|
"""Run SRv6 End.AS test with L2 traffic."""
|
|
self.rewrite_src_addr = rewrite_src_addr
|
|
self.rewrite_sid_list = sid_list[test_sid_index + 1 : :]
|
|
|
|
# send traffic to one destination interface
|
|
# source and destination interfaces are IPv6 only
|
|
self.setup_interfaces(ipv6=[True, False])
|
|
|
|
# configure route to next segment
|
|
route = VppIpRoute(
|
|
self,
|
|
sid_list[test_sid_index + 1],
|
|
128,
|
|
[VppRoutePath(self.pg0.remote_ip6, self.pg0.sw_if_index)],
|
|
)
|
|
route.add_vpp_config()
|
|
|
|
# configure SRv6 localSID behavior
|
|
cli_str = (
|
|
"sr localsid address "
|
|
+ sid_list[test_sid_index]
|
|
+ " behavior end.as"
|
|
+ " oif "
|
|
+ self.pg1.name
|
|
+ " iif "
|
|
+ self.pg1.name
|
|
+ " src "
|
|
+ self.rewrite_src_addr
|
|
)
|
|
for s in self.rewrite_sid_list:
|
|
cli_str += " next " + s
|
|
self.vapi.cli(cli_str)
|
|
|
|
# log the localsids
|
|
self.logger.debug(self.vapi.cli("show sr localsid"))
|
|
|
|
# send one packet per packet size
|
|
count = len(self.pg_packet_sizes)
|
|
|
|
# prepare L2 in SRv6 headers
|
|
packet_header1 = self.create_packet_header_IPv6_SRH_L2(
|
|
sidlist=sid_list[::-1], segleft=len(sid_list) - test_sid_index - 1, vlan=0
|
|
)
|
|
|
|
# generate packets (pg0->pg1)
|
|
pkts1 = self.create_stream(
|
|
self.pg0, self.pg1, packet_header1, self.pg_packet_sizes, count
|
|
)
|
|
|
|
# send packets and verify received packets
|
|
self.send_and_verify_pkts(
|
|
self.pg0, pkts1, self.pg1, self.compare_rx_tx_packet_End_AS_L2_out
|
|
)
|
|
|
|
# log the localsid counters
|
|
self.logger.info(self.vapi.cli("show sr localsid"))
|
|
|
|
# prepare L2 header for returning packets
|
|
packet_header2 = self.create_packet_header_L2()
|
|
|
|
# generate returning packets (pg1->pg0)
|
|
pkts2 = self.create_stream(
|
|
self.pg1, self.pg0, packet_header2, self.pg_packet_sizes, count
|
|
)
|
|
|
|
# send packets and verify received packets
|
|
self.send_and_verify_pkts(
|
|
self.pg1, pkts2, self.pg0, self.compare_rx_tx_packet_End_AS_L2_in
|
|
)
|
|
|
|
# log the localsid counters
|
|
self.logger.info(self.vapi.cli("show sr localsid"))
|
|
|
|
# remove SRv6 localSIDs
|
|
self.vapi.cli("sr localsid del address " + sid_list[test_sid_index])
|
|
|
|
# cleanup interfaces
|
|
self.teardown_interfaces()
|
|
|
|
def run_SRv6_End_AS_IPv6(self, sid_list, test_sid_index, rewrite_src_addr):
|
|
"""Run SRv6 End.AS test with IPv6 traffic."""
|
|
self.rewrite_src_addr = rewrite_src_addr
|
|
self.rewrite_sid_list = sid_list[test_sid_index + 1 : :]
|
|
|
|
# send traffic to one destination interface
|
|
# source and destination interfaces are IPv6 only
|
|
self.setup_interfaces(ipv6=[True, True])
|
|
|
|
# configure route to next segment
|
|
route = VppIpRoute(
|
|
self,
|
|
sid_list[test_sid_index + 1],
|
|
128,
|
|
[VppRoutePath(self.pg0.remote_ip6, self.pg0.sw_if_index)],
|
|
)
|
|
route.add_vpp_config()
|
|
|
|
# configure SRv6 localSID behavior
|
|
cli_str = (
|
|
"sr localsid address "
|
|
+ sid_list[test_sid_index]
|
|
+ " behavior end.as"
|
|
+ " nh "
|
|
+ self.pg1.remote_ip6
|
|
+ " oif "
|
|
+ self.pg1.name
|
|
+ " iif "
|
|
+ self.pg1.name
|
|
+ " src "
|
|
+ self.rewrite_src_addr
|
|
)
|
|
for s in self.rewrite_sid_list:
|
|
cli_str += " next " + s
|
|
self.vapi.cli(cli_str)
|
|
|
|
# log the localsids
|
|
self.logger.debug(self.vapi.cli("show sr localsid"))
|
|
|
|
# send one packet per packet size
|
|
count = len(self.pg_packet_sizes)
|
|
|
|
# prepare IPv6 in SRv6 headers
|
|
packet_header1 = self.create_packet_header_IPv6_SRH_IPv6(
|
|
sidlist=sid_list[::-1], segleft=len(sid_list) - test_sid_index - 1
|
|
)
|
|
|
|
# generate packets (pg0->pg1)
|
|
pkts1 = self.create_stream(
|
|
self.pg0, self.pg1, packet_header1, self.pg_packet_sizes, count
|
|
)
|
|
|
|
# send packets and verify received packets
|
|
self.send_and_verify_pkts(
|
|
self.pg0, pkts1, self.pg1, self.compare_rx_tx_packet_End_AS_IPv6_out
|
|
)
|
|
|
|
# log the localsid counters
|
|
self.logger.info(self.vapi.cli("show sr localsid"))
|
|
|
|
# prepare IPv6 header for returning packets
|
|
packet_header2 = self.create_packet_header_IPv6()
|
|
|
|
# generate returning packets (pg1->pg0)
|
|
pkts2 = self.create_stream(
|
|
self.pg1, self.pg0, packet_header2, self.pg_packet_sizes, count
|
|
)
|
|
|
|
# send packets and verify received packets
|
|
self.send_and_verify_pkts(
|
|
self.pg1, pkts2, self.pg0, self.compare_rx_tx_packet_End_AS_IPv6_in
|
|
)
|
|
|
|
# log the localsid counters
|
|
self.logger.info(self.vapi.cli("show sr localsid"))
|
|
|
|
# remove SRv6 localSIDs
|
|
self.vapi.cli("sr localsid del address " + sid_list[test_sid_index])
|
|
|
|
# cleanup interfaces
|
|
self.teardown_interfaces()
|
|
|
|
def run_SRv6_End_AS_IPv4(self, sid_list, test_sid_index, rewrite_src_addr):
|
|
"""Run SRv6 End.AS test with IPv4 traffic."""
|
|
self.rewrite_src_addr = rewrite_src_addr
|
|
self.rewrite_sid_list = sid_list[test_sid_index + 1 : :]
|
|
|
|
# send traffic to one destination interface
|
|
# source and destination interfaces are IPv6 only
|
|
self.setup_interfaces(ipv6=[True, False], ipv4=[True, True])
|
|
|
|
# configure route to next segment
|
|
route = VppIpRoute(
|
|
self,
|
|
sid_list[test_sid_index + 1],
|
|
128,
|
|
[VppRoutePath(self.pg0.remote_ip6, self.pg0.sw_if_index)],
|
|
)
|
|
route.add_vpp_config()
|
|
|
|
# configure SRv6 localSID behavior
|
|
cli_str = (
|
|
"sr localsid address "
|
|
+ sid_list[test_sid_index]
|
|
+ " behavior end.as"
|
|
+ " nh "
|
|
+ self.pg1.remote_ip4
|
|
+ " oif "
|
|
+ self.pg1.name
|
|
+ " iif "
|
|
+ self.pg1.name
|
|
+ " src "
|
|
+ self.rewrite_src_addr
|
|
)
|
|
for s in self.rewrite_sid_list:
|
|
cli_str += " next " + s
|
|
self.vapi.cli(cli_str)
|
|
|
|
# log the localsids
|
|
self.logger.debug(self.vapi.cli("show sr localsid"))
|
|
|
|
# send one packet per packet size
|
|
count = len(self.pg_packet_sizes)
|
|
|
|
# prepare IPv4 in SRv6 headers
|
|
packet_header1 = self.create_packet_header_IPv6_SRH_IPv4(
|
|
sidlist=sid_list[::-1], segleft=len(sid_list) - test_sid_index - 1
|
|
)
|
|
|
|
# generate packets (pg0->pg1)
|
|
pkts1 = self.create_stream(
|
|
self.pg0, self.pg1, packet_header1, self.pg_packet_sizes, count
|
|
)
|
|
|
|
# send packets and verify received packets
|
|
self.send_and_verify_pkts(
|
|
self.pg0, pkts1, self.pg1, self.compare_rx_tx_packet_End_AS_IPv4_out
|
|
)
|
|
|
|
# log the localsid counters
|
|
self.logger.info(self.vapi.cli("show sr localsid"))
|
|
|
|
# prepare IPv6 header for returning packets
|
|
packet_header2 = self.create_packet_header_IPv4()
|
|
|
|
# generate returning packets (pg1->pg0)
|
|
pkts2 = self.create_stream(
|
|
self.pg1, self.pg0, packet_header2, self.pg_packet_sizes, count
|
|
)
|
|
|
|
# send packets and verify received packets
|
|
self.send_and_verify_pkts(
|
|
self.pg1, pkts2, self.pg0, self.compare_rx_tx_packet_End_AS_IPv4_in
|
|
)
|
|
|
|
# log the localsid counters
|
|
self.logger.info(self.vapi.cli("show sr localsid"))
|
|
|
|
# remove SRv6 localSIDs
|
|
self.vapi.cli("sr localsid del address " + sid_list[test_sid_index])
|
|
|
|
# cleanup interfaces
|
|
self.teardown_interfaces()
|
|
|
|
def compare_rx_tx_packet_End_AS_IPv6_in(self, tx_pkt, rx_pkt):
|
|
"""Compare input and output packet after passing End.AS
|
|
|
|
:param tx_pkt: transmitted packet
|
|
:param rx_pkt: received packet
|
|
"""
|
|
|
|
# get first (outer) IPv6 header of rx'ed packet
|
|
rx_ip = rx_pkt.getlayer(IPv6)
|
|
rx_srh = None
|
|
|
|
tx_ip = tx_pkt.getlayer(IPv6)
|
|
|
|
# expected segment-list (SRH order)
|
|
tx_seglist = self.rewrite_sid_list[::-1]
|
|
|
|
# received ip.src should be equal to SR Policy source
|
|
self.assertEqual(rx_ip.src, self.rewrite_src_addr)
|
|
# received ip.dst should be equal to expected sidlist[lastentry]
|
|
self.assertEqual(rx_ip.dst, tx_seglist[-1])
|
|
|
|
if len(tx_seglist) > 1:
|
|
# rx'ed packet should have SRH
|
|
self.assertTrue(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
# get SRH
|
|
rx_srh = rx_pkt.getlayer(IPv6ExtHdrSegmentRouting)
|
|
# rx'ed seglist should be equal to expected seglist
|
|
self.assertEqual(rx_srh.addresses, tx_seglist)
|
|
# segleft should be equal to size expected seglist-1
|
|
self.assertEqual(rx_srh.segleft, len(tx_seglist) - 1)
|
|
# segleft should be equal to lastentry
|
|
self.assertEqual(rx_srh.segleft, rx_srh.lastentry)
|
|
# get payload
|
|
payload = rx_srh.payload
|
|
else:
|
|
# rx'ed packet should NOT have SRH
|
|
self.assertFalse(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
# get payload
|
|
payload = rx_ip.payload
|
|
|
|
# the whole rx'ed pkt beyond SRH should be equal to tx'ed pkt
|
|
# except for the hop-limit field
|
|
# -> update tx'ed hlim to the expected hlim
|
|
tx_ip.hlim = tx_ip.hlim - 1
|
|
|
|
self.assertEqual(payload, tx_ip)
|
|
|
|
self.logger.debug("packet verification: SUCCESS")
|
|
|
|
def compare_rx_tx_packet_End_AS_IPv4_in(self, tx_pkt, rx_pkt):
|
|
"""Compare input and output packet after passing End.AS
|
|
|
|
:param tx_pkt: transmitted packet
|
|
:param rx_pkt: received packet
|
|
"""
|
|
|
|
# get first (outer) IPv6 header of rx'ed packet
|
|
rx_ip = rx_pkt.getlayer(IPv6)
|
|
rx_srh = None
|
|
|
|
tx_ip = tx_pkt.getlayer(IP)
|
|
|
|
# expected segment-list (SRH order)
|
|
tx_seglist = self.rewrite_sid_list[::-1]
|
|
|
|
# received ip.src should be equal to SR Policy source
|
|
self.assertEqual(rx_ip.src, self.rewrite_src_addr)
|
|
# received ip.dst should be equal to expected sidlist[lastentry]
|
|
self.assertEqual(rx_ip.dst, tx_seglist[-1])
|
|
|
|
if len(tx_seglist) > 1:
|
|
# rx'ed packet should have SRH and IPv4 header
|
|
self.assertTrue(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
self.assertTrue(rx_ip.payload.haslayer(IP))
|
|
# get SRH
|
|
rx_srh = rx_pkt.getlayer(IPv6ExtHdrSegmentRouting)
|
|
# rx'ed seglist should be equal to seglist
|
|
self.assertEqual(rx_srh.addresses, tx_seglist)
|
|
# segleft should be equal to size seglist-1
|
|
self.assertEqual(rx_srh.segleft, len(tx_seglist) - 1)
|
|
# segleft should be equal to lastentry
|
|
self.assertEqual(rx_srh.segleft, rx_srh.lastentry)
|
|
payload = rx_srh.payload
|
|
else:
|
|
# rx'ed packet should NOT have SRH
|
|
self.assertFalse(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
# get payload
|
|
payload = rx_ip.payload
|
|
|
|
# the whole rx'ed pkt beyond SRH should be equal to tx'ed pkt
|
|
# except for the ttl field and ip checksum
|
|
# -> adjust tx'ed ttl to expected ttl
|
|
tx_ip.ttl = tx_ip.ttl - 1
|
|
# -> set tx'ed ip checksum to None and let scapy recompute
|
|
tx_ip.chksum = None
|
|
# read back the pkt (with str()) to force computing these fields
|
|
# probably other ways to accomplish this are possible
|
|
tx_ip = IP(scapy.compat.raw(tx_ip))
|
|
|
|
self.assertEqual(payload, tx_ip)
|
|
|
|
self.logger.debug("packet verification: SUCCESS")
|
|
|
|
def compare_rx_tx_packet_End_AS_L2_in(self, tx_pkt, rx_pkt):
|
|
"""Compare input and output packet after passing End.AS
|
|
|
|
:param tx_pkt: transmitted packet
|
|
:param rx_pkt: received packet
|
|
"""
|
|
|
|
# get first (outer) IPv6 header of rx'ed packet
|
|
rx_ip = rx_pkt.getlayer(IPv6)
|
|
rx_srh = None
|
|
|
|
tx_ether = tx_pkt.getlayer(Ether)
|
|
|
|
# expected segment-list (SRH order)
|
|
tx_seglist = self.rewrite_sid_list[::-1]
|
|
|
|
# received ip.src should be equal to SR Policy source
|
|
self.assertEqual(rx_ip.src, self.rewrite_src_addr)
|
|
# received ip.dst should be equal to expected sidlist[lastentry]
|
|
self.assertEqual(rx_ip.dst, tx_seglist[-1])
|
|
|
|
if len(tx_seglist) > 1:
|
|
# rx'ed packet should have SRH
|
|
self.assertTrue(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
# get SRH
|
|
rx_srh = rx_pkt.getlayer(IPv6ExtHdrSegmentRouting)
|
|
# rx'ed seglist should be equal to seglist
|
|
self.assertEqual(rx_srh.addresses, tx_seglist)
|
|
# segleft should be equal to size seglist-1
|
|
self.assertEqual(rx_srh.segleft, len(tx_seglist) - 1)
|
|
# segleft should be equal to lastentry
|
|
self.assertEqual(rx_srh.segleft, rx_srh.lastentry)
|
|
# nh should be "No Next Header" (143)
|
|
self.assertEqual(rx_srh.nh, 143)
|
|
# get payload
|
|
payload = rx_srh.payload
|
|
else:
|
|
# rx'ed packet should NOT have SRH
|
|
self.assertFalse(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
# get payload
|
|
payload = rx_ip.payload
|
|
|
|
# the whole rx'ed pkt beyond SRH should be equal to tx'ed pkt
|
|
self.assertEqual(Ether(scapy.compat.raw(payload)), tx_ether)
|
|
|
|
self.logger.debug("packet verification: SUCCESS")
|
|
|
|
def compare_rx_tx_packet_End_AS_IPv6_out(self, tx_pkt, rx_pkt):
|
|
"""Compare input and output packet after passing End.AS with IPv6
|
|
|
|
:param tx_pkt: transmitted packet
|
|
:param rx_pkt: received packet
|
|
"""
|
|
|
|
# get first (outer) IPv6 header of rx'ed packet
|
|
rx_ip = rx_pkt.getlayer(IPv6)
|
|
|
|
tx_ip = tx_pkt.getlayer(IPv6)
|
|
tx_ip2 = tx_pkt.getlayer(IPv6, 2)
|
|
|
|
# verify if rx'ed packet has no SRH
|
|
self.assertFalse(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
|
|
# the whole rx_ip pkt should be equal to tx_ip2
|
|
# except for the hlim field
|
|
# -> adjust tx'ed hlim to expected hlim
|
|
tx_ip2.hlim = tx_ip2.hlim - 1
|
|
|
|
self.assertEqual(rx_ip, tx_ip2)
|
|
|
|
self.logger.debug("packet verification: SUCCESS")
|
|
|
|
def compare_rx_tx_packet_End_AS_IPv4_out(self, tx_pkt, rx_pkt):
|
|
"""Compare input and output packet after passing End.AS with IPv4
|
|
|
|
:param tx_pkt: transmitted packet
|
|
:param rx_pkt: received packet
|
|
"""
|
|
|
|
# get IPv4 header of rx'ed packet
|
|
rx_ip = rx_pkt.getlayer(IP)
|
|
|
|
tx_ip = tx_pkt.getlayer(IPv6)
|
|
tx_ip2 = tx_pkt.getlayer(IP)
|
|
|
|
# verify if rx'ed packet has no SRH
|
|
self.assertFalse(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
|
|
# the whole rx_ip pkt should be equal to tx_ip2
|
|
# except for the ttl field and ip checksum
|
|
# -> adjust tx'ed ttl to expected ttl
|
|
tx_ip2.ttl = tx_ip2.ttl - 1
|
|
# -> set tx'ed ip checksum to None and let scapy recompute
|
|
tx_ip2.chksum = None
|
|
# read back the pkt (with str()) to force computing these fields
|
|
# probably other ways to accomplish this are possible
|
|
tx_ip2 = IP(scapy.compat.raw(tx_ip2))
|
|
|
|
self.assertEqual(rx_ip, tx_ip2)
|
|
|
|
self.logger.debug("packet verification: SUCCESS")
|
|
|
|
def compare_rx_tx_packet_End_AS_L2_out(self, tx_pkt, rx_pkt):
|
|
"""Compare input and output packet after passing End.AS with L2
|
|
|
|
:param tx_pkt: transmitted packet
|
|
:param rx_pkt: received packet
|
|
"""
|
|
|
|
# get IPv4 header of rx'ed packet
|
|
rx_eth = rx_pkt.getlayer(Ether)
|
|
|
|
tx_ip = tx_pkt.getlayer(IPv6)
|
|
# we can't just get the 2nd Ether layer
|
|
# get the Raw content and dissect it as Ether
|
|
tx_eth1 = Ether(scapy.compat.raw(tx_pkt[Raw]))
|
|
|
|
# verify if rx'ed packet has no SRH
|
|
self.assertFalse(rx_pkt.haslayer(IPv6ExtHdrSegmentRouting))
|
|
|
|
# the whole rx_eth pkt should be equal to tx_eth1
|
|
self.assertEqual(rx_eth, tx_eth1)
|
|
|
|
self.logger.debug("packet verification: SUCCESS")
|
|
|
|
def create_stream(self, src_if, dst_if, packet_header, packet_sizes, count):
|
|
"""Create SRv6 input packet stream for defined interface.
|
|
|
|
:param VppInterface src_if: Interface to create packet stream for
|
|
:param VppInterface dst_if: destination interface of packet stream
|
|
:param packet_header: Layer3 scapy packet headers,
|
|
L2 is added when not provided,
|
|
Raw(payload) with packet_info is added
|
|
:param list packet_sizes: packet stream pckt sizes,sequentially applied
|
|
to packets in stream have
|
|
:param int count: number of packets in packet stream
|
|
:return: list of packets
|
|
"""
|
|
self.logger.info("Creating packets")
|
|
pkts = []
|
|
for i in range(0, count - 1):
|
|
payload_info = self.create_packet_info(src_if, dst_if)
|
|
self.logger.debug("Creating packet with index %d" % (payload_info.index))
|
|
payload = self.info_to_payload(payload_info)
|
|
# add L2 header if not yet provided in packet_header
|
|
if packet_header.getlayer(0).name == "Ethernet":
|
|
p = packet_header / Raw(payload)
|
|
else:
|
|
p = (
|
|
Ether(dst=src_if.local_mac, src=src_if.remote_mac)
|
|
/ packet_header
|
|
/ Raw(payload)
|
|
)
|
|
size = packet_sizes[i % len(packet_sizes)]
|
|
self.logger.debug("Packet size %d" % (size))
|
|
self.extend_packet(p, size)
|
|
# we need to store the packet with the automatic fields computed
|
|
# read back the dumped packet (with str())
|
|
# to force computing these fields
|
|
# probably other ways are possible
|
|
p = Ether(scapy.compat.raw(p))
|
|
payload_info.data = p.copy()
|
|
self.logger.debug(ppp("Created packet:", p))
|
|
pkts.append(p)
|
|
self.logger.info("Done creating packets")
|
|
return pkts
|
|
|
|
def send_and_verify_pkts(self, input, pkts, output, compare_func):
|
|
"""Send packets and verify received packets using compare_func
|
|
|
|
:param input: ingress interface of DUT
|
|
:param pkts: list of packets to transmit
|
|
:param output: egress interface of DUT
|
|
:param compare_func: function to compare in and out packets
|
|
"""
|
|
# add traffic stream to input interface
|
|
input.add_stream(pkts)
|
|
|
|
# enable capture on all interfaces
|
|
self.pg_enable_capture(self.pg_interfaces)
|
|
|
|
# start traffic
|
|
self.logger.info("Starting traffic")
|
|
self.pg_start()
|
|
|
|
# get output capture
|
|
self.logger.info("Getting packet capture")
|
|
capture = output.get_capture()
|
|
|
|
# assert nothing was captured on input interface
|
|
# input.assert_nothing_captured()
|
|
|
|
# verify captured packets
|
|
self.verify_captured_pkts(output, capture, compare_func)
|
|
|
|
def create_packet_header_IPv6(self):
|
|
"""Create packet header: IPv6 header, UDP header
|
|
|
|
:param dst: IPv6 destination address
|
|
|
|
IPv6 source address is 1234::1
|
|
IPv6 destination address is 4321::1
|
|
UDP source port and destination port are 1234
|
|
"""
|
|
|
|
p = IPv6(src="1234::1", dst="4321::1") / UDP(sport=1234, dport=1234)
|
|
return p
|
|
|
|
def create_packet_header_IPv6_SRH_IPv6(self, sidlist, segleft):
|
|
"""Create packet header: IPv6 encapsulated in SRv6:
|
|
IPv6 header with SRH, IPv6 header, UDP header
|
|
|
|
:param list sidlist: segment list of outer IPv6 SRH
|
|
:param int segleft: segments-left field of outer IPv6 SRH
|
|
|
|
Outer IPv6 source address is set to 5678::1
|
|
Outer IPv6 destination address is set to sidlist[segleft]
|
|
IPv6 source addresses is 1234::1
|
|
IPv6 destination address is 4321::1
|
|
UDP source port and destination port are 1234
|
|
"""
|
|
|
|
p = (
|
|
IPv6(src="5678::1", dst=sidlist[segleft])
|
|
/ IPv6ExtHdrSegmentRouting(addresses=sidlist, segleft=segleft, nh=41)
|
|
/ IPv6(src="1234::1", dst="4321::1")
|
|
/ UDP(sport=1234, dport=1234)
|
|
)
|
|
return p
|
|
|
|
def create_packet_header_IPv4(self):
|
|
"""Create packet header: IPv4 header, UDP header
|
|
|
|
:param dst: IPv4 destination address
|
|
|
|
IPv4 source address is 123.1.1.1
|
|
IPv4 destination address is 124.1.1.1
|
|
UDP source port and destination port are 1234
|
|
"""
|
|
|
|
p = IP(src="123.1.1.1", dst="124.1.1.1") / UDP(sport=1234, dport=1234)
|
|
return p
|
|
|
|
def create_packet_header_IPv6_SRH_IPv4(self, sidlist, segleft):
|
|
"""Create packet header: IPv4 encapsulated in SRv6:
|
|
IPv6 header with SRH, IPv4 header, UDP header
|
|
|
|
:param ipv4address dst: inner IPv4 destination address
|
|
:param list sidlist: segment list of outer IPv6 SRH
|
|
:param int segleft: segments-left field of outer IPv6 SRH
|
|
|
|
Outer IPv6 destination address is set to sidlist[segleft]
|
|
IPv6 source address is 1234::1
|
|
IPv4 source address is 123.1.1.1
|
|
IPv4 destination address is 124.1.1.1
|
|
UDP source port and destination port are 1234
|
|
"""
|
|
|
|
p = (
|
|
IPv6(src="1234::1", dst=sidlist[segleft])
|
|
/ IPv6ExtHdrSegmentRouting(addresses=sidlist, segleft=segleft, nh=4)
|
|
/ IP(src="123.1.1.1", dst="124.1.1.1")
|
|
/ UDP(sport=1234, dport=1234)
|
|
)
|
|
return p
|
|
|
|
def create_packet_header_L2(self, vlan=0):
|
|
"""Create packet header: L2 header
|
|
|
|
:param vlan: if vlan!=0 then add 802.1q header
|
|
"""
|
|
# Note: the dst addr ('00:55:44:33:22:11') is used in
|
|
# the compare function compare_rx_tx_packet_T_Encaps_L2
|
|
# to detect presence of L2 in SRH payload
|
|
p = Ether(src="00:11:22:33:44:55", dst="00:55:44:33:22:11")
|
|
etype = 0x8137 # IPX
|
|
if vlan:
|
|
# add 802.1q layer
|
|
p /= Dot1Q(vlan=vlan, type=etype)
|
|
else:
|
|
p.type = etype
|
|
return p
|
|
|
|
def create_packet_header_IPv6_SRH_L2(self, sidlist, segleft, vlan=0):
|
|
"""Create packet header: L2 encapsulated in SRv6:
|
|
IPv6 header with SRH, L2
|
|
|
|
:param list sidlist: segment list of outer IPv6 SRH
|
|
:param int segleft: segments-left field of outer IPv6 SRH
|
|
:param vlan: L2 vlan; if vlan!=0 then add 802.1q header
|
|
|
|
Outer IPv6 destination address is set to sidlist[segleft]
|
|
IPv6 source address is 1234::1
|
|
"""
|
|
eth = Ether(src="00:11:22:33:44:55", dst="00:55:44:33:22:11")
|
|
etype = 0x8137 # IPX
|
|
if vlan:
|
|
# add 802.1q layer
|
|
eth /= Dot1Q(vlan=vlan, type=etype)
|
|
else:
|
|
eth.type = etype
|
|
|
|
p = (
|
|
IPv6(src="1234::1", dst=sidlist[segleft])
|
|
/ IPv6ExtHdrSegmentRouting(addresses=sidlist, segleft=segleft, nh=143)
|
|
/ eth
|
|
)
|
|
return p
|
|
|
|
def get_payload_info(self, packet):
|
|
"""Extract the payload_info from the packet"""
|
|
# in most cases, payload_info is in packet[Raw]
|
|
# but packet[Raw] gives the complete payload
|
|
# (incl L2 header) for the T.Encaps L2 case
|
|
try:
|
|
payload_info = self.payload_to_info(packet[Raw])
|
|
|
|
except:
|
|
# remote L2 header from packet[Raw]:
|
|
# take packet[Raw], convert it to an Ether layer
|
|
# and then extract Raw from it
|
|
payload_info = self.payload_to_info(
|
|
Ether(scapy.compat.raw(packet[Raw]))[Raw]
|
|
)
|
|
|
|
return payload_info
|
|
|
|
def verify_captured_pkts(self, dst_if, capture, compare_func):
|
|
"""
|
|
Verify captured packet stream for specified interface.
|
|
Compare ingress with egress packets using the specified compare fn
|
|
|
|
:param dst_if: egress interface of DUT
|
|
:param capture: captured packets
|
|
:param compare_func: function to compare in and out packet
|
|
"""
|
|
self.logger.info(
|
|
"Verifying capture on interface %s using function %s"
|
|
% (dst_if.name, compare_func.__name__)
|
|
)
|
|
|
|
last_info = dict()
|
|
for i in self.pg_interfaces:
|
|
last_info[i.sw_if_index] = None
|
|
dst_sw_if_index = dst_if.sw_if_index
|
|
|
|
for packet in capture:
|
|
try:
|
|
# extract payload_info from packet's payload
|
|
payload_info = self.get_payload_info(packet)
|
|
packet_index = payload_info.index
|
|
|
|
self.logger.debug("Verifying packet with index %d" % (packet_index))
|
|
# packet should have arrived on the expected interface
|
|
self.assertEqual(payload_info.dst, dst_sw_if_index)
|
|
self.logger.debug(
|
|
"Got packet on interface %s: src=%u (idx=%u)"
|
|
% (dst_if.name, payload_info.src, packet_index)
|
|
)
|
|
|
|
# search for payload_info with same src and dst if_index
|
|
# this will give us the transmitted packet
|
|
next_info = self.get_next_packet_info_for_interface2(
|
|
payload_info.src, dst_sw_if_index, last_info[payload_info.src]
|
|
)
|
|
last_info[payload_info.src] = next_info
|
|
# next_info should not be None
|
|
self.assertTrue(next_info is not None)
|
|
# index of tx and rx packets should be equal
|
|
self.assertEqual(packet_index, next_info.index)
|
|
# data field of next_info contains the tx packet
|
|
txed_packet = next_info.data
|
|
|
|
self.logger.debug(
|
|
ppp("Transmitted packet:", txed_packet)
|
|
) # ppp=Pretty Print Packet
|
|
|
|
self.logger.debug(ppp("Received packet:", packet))
|
|
|
|
# compare rcvd packet with expected packet using compare_func
|
|
compare_func(txed_packet, packet)
|
|
|
|
except:
|
|
self.logger.error(ppp("Unexpected or invalid packet:", packet))
|
|
raise
|
|
|
|
# have all expected packets arrived?
|
|
for i in self.pg_interfaces:
|
|
remaining_packet = self.get_next_packet_info_for_interface2(
|
|
i.sw_if_index, dst_sw_if_index, last_info[i.sw_if_index]
|
|
)
|
|
self.assertTrue(
|
|
remaining_packet is None,
|
|
"Interface %s: Packet expected from interface %s "
|
|
"didn't arrive" % (dst_if.name, i.name),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(testRunner=VppTestRunner)
|