#!/usr/bin/env python3
import time
import unittest
from scapy.contrib.lacp import LACP, SlowProtocol, MarkerProtocol
from scapy.layers.l2 import Ether
from framework import VppTestCase, VppTestRunner
from vpp_memif import remove_all_memif_vpp_config, VppSocketFilename, VppMemif
from vpp_bond_interface import VppBondInterface
from vpp_papi import VppEnum, MACAddress
bond_mac = "02:02:02:02:02:02"
lacp_dst_mac = '01:80:c2:00:00:02'
LACP_COLLECTION_AND_DISTRIBUTION_STATE = 63
class TestMarker(VppTestCase):
"""LACP Marker Protocol Test Case
"""
@classmethod
def setUpClass(cls):
super().setUpClass()
# Test variables
cls.pkts_per_burst = 257 # Number of packets per burst
# create 3 pg interfaces
cls.create_pg_interfaces(range(1))
# packet sizes
cls.pg_if_packet_sizes = [64, 512, 1518] # , 9018]
# setup all interfaces
for i in cls.pg_interfaces:
i.admin_up()
@classmethod
def tearDownClass(cls):
super().tearDownClass()
def setUp(self):
super().setUp()
def tearDown(self):
super().tearDown()
def show_commands_at_teardown(self):
self.logger.info(self.vapi.ppcli("show interface"))
def test_marker_request(self):
""" Marker Request test """
# topology
#
# +-+ +-+
# memif1 -----|B| |B|---- memif11
# |o| |o|
# |n|------|n|
# |d| |d|
# pg0 -----|0| |1|
# +-+ +-+
socket1 = VppSocketFilename(
self,
socket_id=1,
socket_filename="%s/memif.sock1" % self.tempdir)
socket1.add_vpp_config()
socket11 = VppSocketFilename(
self,
socket_id=2,
socket_filename="%s/memif.sock1" % self.tempdir)
socket11.add_vpp_config()
memif1 = VppMemif(
self,
role=VppEnum.vl_api_memif_role_t.MEMIF_ROLE_API_MASTER,
mode=VppEnum.vl_api_memif_mode_t.MEMIF_MODE_API_ETHERNET,
socket_id=1)
memif1.add_vpp_config()
memif1.admin_up()
memif11 = VppMemif(
self,
role=VppEnum.vl_api_memif_role_t.MEMIF_ROLE_API_SLAVE,
mode=VppEnum.vl_api_memif_mode_t.MEMIF_MODE_API_ETHERNET,
socket_id=2)
memif11.add_vpp_config()
memif11.admin_up()
bond0 = VppBondInterface(
self,
mode=VppEnum.vl_api_bond_mode_t.BOND_API_MODE_LACP,
use_custom_mac=1,
mac_address=bond_mac)
bond0.add_vpp_config()
bond0.admin_up()
bond1 = VppBondInterface(
self,
mode=VppEnum.vl_api_bond_mode_t.BOND_API_MODE_LACP)
bond1.add_vpp_config()
bond1.admin_up()
bond0.add_member_vpp_bond_interface(sw_if_index=memif1.sw_if_index)
bond1.add_member_vpp_bond_interface(sw_if_index=memif11.sw_if_index)
# wait for memif protocol exchange and hardware carrier to come up
self.assertEqual(memif1.wait_for_link_up(10), True)
self.assertEqual(memif11.wait_for_link_up(10), True)
# verify memif1 in bond0
intfs = self.vapi.sw_member_interface_dump(
sw_if_index=bond0.sw_if_index)
for intf in intfs:
self.assertEqual(intf.sw_if_index, memif1.sw_if_index)
# verify memif11 in bond1
intfs = self.vapi.sw_member_interface_dump(
sw_if_index=bond1.sw_if_index)
for intf in intfs:
self.assertEqual(intf.sw_if_index, memif11.sw_if_index)
self.vapi.ppcli("trace add memif-input 100")
# create marker request
marker = (Ether(src=bond_mac, dst=lacp_dst_mac) /
SlowProtocol() /
MarkerProtocol(marker_type=1,
requester_port=1,
requester_system=bond_mac,
requester_transaction_id=1))
bond1.add_member_vpp_bond_interface(sw_if_index=self.pg0.sw_if_index)
self.pg0.add_stream(marker)
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
show_trace = self.vapi.ppcli("show trace max 100")
self.assertIn("Marker Information TLV:", show_trace)
bond0.remove_vpp_config()
bond1.remove_vpp_config()
class TestLACP(VppTestCase):
"""LACP Test Case
"""
@classmethod
def setUpClass(cls):
super().setUpClass()
@classmethod
def tearDownClass(cls):
super().tearDownClass()
def setUp(self):
super().setUp()
def tearDown(self):
super().tearDown()
def show_commands_at_teardown(self):
self.logger.info(self.vapi.ppcli("show interface"))
def wait_for_lacp_connect(self, timeout, step=1):
while 1:
intfs = self.vapi.sw_interface_lacp_dump()
all_good = 1
for intf in intfs:
if ((intf.actor_state !=
LACP_COLLECTION_AND_DISTRIBUTION_STATE) or
(intf.partner_state !=
LACP_COLLECTION_AND_DISTRIBUTION_STATE)):
all_good = 0
if (all_good == 1):
return 1
self.sleep(step)
timeout -= step
if timeout <= 0:
return 0
def wait_for_member_detach(self, bond, timeout, count, step=1):
while 1:
intfs = self.vapi.sw_bond_interface_dump(
sw_if_index=bond.sw_if_index)
for intf in intfs:
if ((intf.members == count) and
(intf.active_members == count)):
return 1
else:
self.sleep(1)
timeout -= step
if (timeouut <= 0):
return 0
def test_lacp_connect(self):
""" LACP protocol connect test """
# topology
#
# +-+ +-+
# memif1 -----|B| |B|---- memif11
# |o| |o|
# |n|------|n|
# |d| |d|
# memif2 -----|0| |1|---- memif12
# +-+ +-+
socket1 = VppSocketFilename(
self,
socket_id=1,
socket_filename="%s/memif.sock1" % self.tempdir)
socket1.add_vpp_config()
socket11 = VppSocketFilename(
self,
socket_id=2,
socket_filename="%s/memif.sock1" % self.tempdir)
socket11.add_vpp_config()
socket2 = VppSocketFilename(
self,
socket_id=3,
socket_filename="%s/memif.sock2" % self.tempdir)
socket2.add_vpp_config()
socket22 = VppSocketFilename(
self,
socket_id=4,
socket_filename="%s/memif.sock2" % self.tempdir)
socket22.add_vpp_config()
memif1 = VppMemif(
self,
role=VppEnum.vl_api_memif_role_t.MEMIF_ROLE_API_MASTER,
mode=VppEnum.vl_api_memif_mode_t.MEMIF_MODE_API_ETHERNET,
socket_id=1)
memif1.add_vpp_config()
memif1.admin_up()
memif11 = VppMemif(
self,
role=VppEnum.vl_api_memif_role_t.MEMIF_ROLE_API_SLAVE,
mode=VppEnum.vl_api_memif_mode_t.MEMIF_MODE_API_ETHERNET,
socket_id=2)
memif11.add_vpp_config()
memif11.admin_up()
memif2 = VppMemif(
self,
role=VppEnum.vl_api_memif_role_t.MEMIF_ROLE_API_MASTER,
mode=VppEnum.vl_api_memif_mode_t.MEMIF_MODE_API_ETHERNET,
socket_id=3)
memif2.add_vpp_config()
memif2.admin_up()
memif12 = VppMemif(
self,
role=VppEnum.vl_api_memif_role_t.MEMIF_ROLE_API_SLAVE,
mode=VppEnum.vl_api_memif_mode_t.MEMIF_MODE_API_ETHERNET,
socket_id=4)
memif12.add_vpp_config()
memif12.admin_up()
self.logger.info(self.vapi.ppcli("debug lacp on"))
bond0 = VppBondInterface(
self,
mode=VppEnum.vl_api_bond_mode_t.BOND_API_MODE_LACP,
use_custom_mac=1,
mac_address=bond_mac)
bond0.add_vpp_config()
bond0.admin_up()
bond1 = VppBondInterface(
self,
mode=VppEnum.vl_api_bond_mode_t.BOND_API_MODE_LACP)
bond1.add_vpp_config()
bond1.admin_up()
# add member memif1 and memif2 to bond0
bond0.add_member_vpp_bond_interface(sw_if_index=memif1.sw_if_index)
bond0.add_member_vpp_bond_interface(sw_if_index=memif2.sw_if_index)
# add member memif11 and memif12 to bond1
bond1.add_member_vpp_bond_interface(sw_if_index=memif11.sw_if_index)
bond1.add_member_vpp_bond_interface(sw_if_index=memif12.sw_if_index)
# wait for memif protocol exchange and hardware carrier to come up
self.assertEqual(memif1.wait_for_link_up(10), True)
self.assertEqual(memif2.wait_for_link_up(10), True)
self.assertEqual(memif11.wait_for_link_up(10), True)
self.assertEqual(memif12.wait_for_link_up(10), True)
# verify memif1 and memif2 in bond0
intfs = self.vapi.sw_member_interface_dump(
sw_if_index=bond0.sw_if_index)
for intf in intfs:
self.assertIn(
intf.sw_if_index, (memif1.sw_if_index, memif2.sw_if_index))
# verify memif11 and memif12 in bond1
intfs = self.vapi.sw_member_interface_dump(
sw_if_index=bond1.sw_if_index)
for intf in intfs:
self.assertIn(
intf.sw_if_index, (memif11.sw_if_index, memif12.sw_if_index))
self.assertEqual(intf.is_long_timeout, 0)
self.assertEqual(intf.is_passive, 0)
# Let LACP create the bundle
self.wait_for_lacp_connect(30)
intfs = self.vapi.sw_interface_lacp_dump()
for intf in intfs:
self.assertEqual(
intf.actor_state, LACP_COLLECTION_AND_DISTRIBUTION_STATE)
self.assertEqual(
intf.partner_state, LACP_COLLECTION_AND_DISTRIBUTION_STATE)
intfs = self.vapi.sw_bond_interface_dump(sw_if_index=0xFFFFFFFF)
for intf in intfs:
self.assertEqual(intf.members, 2)
self.assertEqual(intf.active_members, 2)
self.assertEqual(
intf.mode, VppEnum.vl_api_bond_mode_t.BOND_API_MODE_LACP)
self.logger.info(self.vapi.ppcli("show lacp"))
self.logger.info(self.vapi.ppcli("show lacp details"))
# detach member memif1
bond0.detach_vpp_bond_interface(sw_if_index=memif1.sw_if_index)
self.wait_for_member_detach(bond0, timeout=10, count=1)
intfs = self.vapi.sw_bond_interface_dump(
sw_if_index=bond0.sw_if_index)
for intf in intfs:
self.assertEqual(intf.members, 1)
self.assertEqual(intf.active_members, 1)
self.assertEqual(
intf.mode, VppEnum.vl_api_bond_mode_t.BOND_API_MODE_LACP)
# detach member memif2
bond0.detach_vpp_bond_interface(sw_if_index=memif2.sw_if_index)
self.wait_for_member_detach(bond0, timeout=10, count=0)
intfs = self.vapi.sw_bond_interface_dump(
sw_if_index=bond0.sw_if_index)
for intf in intfs:
self.assertEqual(intf.members, 0)
self.assertEqual(intf.active_members, 0)
bond0.remove_vpp_config()
bond1.remove_vpp_config()
if __name__ == '__main__':
unittest.main(testRunner=VppTestRunner)
