aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python
diff options
context:
space:
mode:
Diffstat (limited to 'resources/libraries/python')
-rw-r--r--resources/libraries/python/L2Util.py424
-rw-r--r--resources/libraries/python/PapiExecutor.py6
2 files changed, 257 insertions, 173 deletions
diff --git a/resources/libraries/python/L2Util.py b/resources/libraries/python/L2Util.py
index 52d138552b..03ba5640c9 100644
--- a/resources/libraries/python/L2Util.py
+++ b/resources/libraries/python/L2Util.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2018 Cisco and/or its affiliates.
+# Copyright (c) 2019 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
@@ -13,15 +13,29 @@
"""L2 Utilities Library."""
+import binascii
from textwrap import wrap
-from robot.api.deco import keyword
+from enum import IntEnum
+from resources.libraries.python.PapiExecutor import PapiExecutor
from resources.libraries.python.topology import Topology
-from resources.libraries.python.VatExecutor import VatExecutor, VatTerminal
from resources.libraries.python.ssh import exec_cmd_no_error
+class L2VtrOp(IntEnum):
+ """VLAN tag rewrite operation."""
+ L2_VTR_DISABLED = 0
+ L2_VTR_PUSH_1 = 1
+ L2_VTR_PUSH_2 = 2
+ L2_VTR_POP_1 = 3
+ L2_VTR_POP_2 = 4
+ L2_VTR_TRANSLATE_1_1 = 5
+ L2_VTR_TRANSLATE_1_2 = 6
+ L2_VTR_TRANSLATE_2_1 = 7
+ L2_VTR_TRANSLATE_2_2 = 8
+
+
class L2Util(object):
"""Utilities for l2 configuration."""
@@ -50,117 +64,146 @@ class L2Util(object):
return ':'.join(wrap("{:012x}".format(mac_int), width=2))
@staticmethod
- def vpp_add_l2fib_entry(node, mac, interface, bd_id):
- """ Create a static L2FIB entry on a vpp node.
+ def mac_to_bin(mac_str):
+ """Convert MAC address from string format (e.g. 01:02:03:04:05:06) to
+ binary representation (\x01\x02\x03\x04\x05\x06).
+
+ :param mac_str: MAC address in string representation.
+ :type mac_str: str
+ :returns: Binary representation of MAC address.
+ :rtype: binary
+ """
+ return binascii.unhexlify(mac_str.replace(':', ''))
+
+ @staticmethod
+ def bin_to_mac(mac_bin):
+ """Convert MAC address from binary representation
+ (\x01\x02\x03\x04\x05\x06) to string format (e.g. 01:02:03:04:05:06).
+
+ :param mac_bin: MAC address in binary representation.
+ :type mac_bin: binary
+ :returns: String representation of MAC address.
+ :rtype: str
+ """
+ x = ':'.join(binascii.hexlify(mac_bin)[i:i + 2]
+ for i in range(0, 12, 2))
+ return str(x.decode('ascii'))
+
+ @staticmethod
+ def vpp_add_l2fib_entry(node, mac, interface, bd_id, static_mac=1,
+ filter_mac=0, bvi_mac=0):
+ """ Create a static L2FIB entry on a VPP node.
:param node: Node to add L2FIB entry on.
- :param mac: Destination mac address.
+ :param mac: Destination mac address in string format 01:02:03:04:05:06.
:param interface: Interface name or sw_if_index.
- :param bd_id: Bridge domain id.
+ :param bd_id: Bridge domain index.
+ :param static_mac: Set to 1 to create static MAC entry.
+ (Default value = 1)
+ :param filter_mac: Set to 1 to drop packet that's source or destination
+ MAC address contains defined MAC address. (Default value = 0)
+ :param bvi_mac: Set to 1 to create entry that points to BVI interface.
+ (Default value = 0)
:type node: dict
:type mac: str
:type interface: str or int
- :type bd_id: int
+ :type bd_id: int or str
+ :type static_mac: int or str
+ :type filter_mac: int or str
+ :type bvi_mac: int or str
"""
+
if isinstance(interface, basestring):
sw_if_index = Topology.get_interface_sw_index(node, interface)
else:
sw_if_index = interface
- VatExecutor.cmd_from_template(node, "add_l2_fib_entry.vat",
- mac=mac, bd=bd_id,
- interface=sw_if_index)
+
+ cmd = 'l2fib_add_del'
+ err_msg = 'Failed to add L2FIB entry on host {host}'.format(
+ host=node['host'])
+ args = dict(mac=L2Util.mac_to_bin(mac),
+ bd_id=int(bd_id),
+ sw_if_index=sw_if_index,
+ is_add=1,
+ static_mac=int(static_mac),
+ filter_mac=int(filter_mac),
+ bvi_mac=int(bvi_mac))
+ with PapiExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_replies(err_msg).\
+ verify_reply(err_msg=err_msg)
@staticmethod
def create_l2_bd(node, bd_id, flood=1, uu_flood=1, forward=1, learn=1,
arp_term=0):
- """Create a l2 bridge domain on the chosen VPP node
-
- Execute "bridge_domain_add_del bd_id {bd_id} flood {flood} uu-flood 1
- forward {forward} learn {learn} arp-term {arp_term}" VAT command on
- the node.
-
- :param node: Node where we wish to crate the l2 bridge domain.
- :param bd_id: Bridge domain index number.
- :param flood: Enable flooding.
- :param uu_flood: Enable uu_flood.
- :param forward: Enable forwarding.
- :param learn: Enable mac address learning to fib.
- :param arp_term: Enable arp_termination.
+ """Create an L2 bridge domain on a VPP node.
+
+ :param node: Node where we wish to crate the L2 bridge domain.
+ :param bd_id: Bridge domain index.
+ :param flood: Enable/disable bcast/mcast flooding in the BD.
+ (Default value = 1)
+ :param uu_flood: Enable/disable unknown unicast flood in the BD.
+ (Default value = 1)
+ :param forward: Enable/disable forwarding on all interfaces in
+ the BD. (Default value = 1)
+ :param learn: Enable/disable MAC learning on all interfaces in the BD.
+ (Default value = 1)
+ :param arp_term: Enable/disable arp termination in the BD.
+ (Default value = 1)
:type node: dict
- :type bd_id: int
- :type flood: bool
- :type uu_flood: bool
- :type forward: bool
- :type learn: bool
- :type arp_term: bool
+ :type bd_id: int or str
+ :type flood: int or str
+ :type uu_flood: int or str
+ :type forward: int or str
+ :type learn: int or str
+ :type arp_term: int or str
"""
- VatExecutor.cmd_from_template(node, "l2_bd_create.vat",
- bd_id=bd_id, flood=flood,
- uu_flood=uu_flood, forward=forward,
- learn=learn, arp_term=arp_term)
+
+ cmd = 'bridge_domain_add_del'
+ err_msg = 'Failed to create L2 bridge domain on host {host}'.format(
+ host=node['host'])
+ args = dict(bd_id=int(bd_id),
+ flood=int(flood),
+ uu_flood=int(uu_flood),
+ forward=int(forward),
+ learn=int(learn),
+ arp_term=int(arp_term),
+ is_add=1)
+ with PapiExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_replies(err_msg).\
+ verify_reply(err_msg=err_msg)
@staticmethod
- def add_interface_to_l2_bd(node, interface, bd_id, shg=0):
- """Add a interface to the l2 bridge domain.
+ def add_interface_to_l2_bd(node, interface, bd_id, shg=0, port_type=0):
+ """Add an interface to the L2 bridge domain.
Get SW IF ID and add it to the bridge domain.
:param node: Node where we want to execute the command that does this.
:param interface: Interface name.
- :param bd_id: Bridge domain index number to add Interface name to.
- :param shg: Split horizon group.
+ :param bd_id: Bridge domain index.
+ :param shg: Split-horizon group index. (Default value = 0)
+ :param port_type: Port mode: 0 - normal, 1 - BVI, 2 - UU_FWD.
+ (Default value = 0)
:type node: dict
:type interface: str
- :type bd_id: int
- :type shg: int
+ :type bd_id: int or str
+ :type shg: int or str
+ :type port_type: int or str
"""
- sw_if_index = Topology.get_interface_sw_index(node, interface)
- L2Util.add_sw_if_index_to_l2_bd(node, sw_if_index, bd_id, shg)
-
- @staticmethod
- def add_sw_if_index_to_l2_bd(node, sw_if_index, bd_id, shg=0):
- """Add interface with sw_if_index to l2 bridge domain.
-
- Execute the "sw_interface_set_l2_bridge sw_if_index {sw_if_index}
- bd_id {bd_id} shg {shg} enable" VAT command on the given node.
- :param node: Node where we want to execute the command that does this.
- :param sw_if_index: Interface index.
- :param bd_id: Bridge domain index number to add SW IF ID to.
- :param shg: Split horizon group.
- :type node: dict
- :type sw_if_index: int
- :type bd_id: int
- :type shg: int
- """
- VatExecutor.cmd_from_template(node, "l2_bd_add_sw_if_index.vat",
- bd_id=bd_id, sw_if_index=sw_if_index,
- shg=shg)
+ sw_if_index = Topology.get_interface_sw_index(node, interface)
- @staticmethod
- @keyword('Create dict used in bridge domain template file for node '
- '"${node}" with links "${link_names}" and bd_id "${bd_id}"')
- def create_bridge_domain_vat_dict(node, link_names, bd_id):
- """Create dictionary that can be used in l2 bridge domain template.
-
- The resulting dictionary looks like this:
- 'interface1': interface name of first interface
- 'interface2': interface name of second interface
- 'bd_id': bridge domain index
-
- :param node: Node data dictionary.
- :param link_names: List of names of links the bridge domain should be
- connecting.
- :param bd_id: Bridge domain index number.
- :type node: dict
- :type link_names: list
- :returns: Dictionary used to generate l2 bridge domain VAT configuration
- from template file.
- :rtype: dict
- """
- bd_dict = Topology().get_interfaces_by_link_names(node, link_names)
- bd_dict['bd_id'] = bd_id
- return bd_dict
+ cmd = 'sw_interface_set_l2_bridge'
+ err_msg = 'Failed to add interface {ifc} to L2 bridge domain on host ' \
+ '{host}'.format(ifc=interface, host=node['host'])
+ args = dict(rx_sw_if_index=sw_if_index,
+ bd_id=int(bd_id),
+ shg=int(shg),
+ port_type=int(port_type),
+ enable=1)
+ with PapiExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_replies(err_msg).\
+ verify_reply(err_msg=err_msg)
@staticmethod
def vpp_add_l2_bridge_domain(node, bd_id, port_1, port_2, learn=True):
@@ -177,14 +220,39 @@ class L2Util(object):
:type port_2: str
:type learn: bool
"""
+
sw_if_index1 = Topology.get_interface_sw_index(node, port_1)
sw_if_index2 = Topology.get_interface_sw_index(node, port_2)
- VatExecutor.cmd_from_template(node,
- 'l2_bridge_domain.vat',
- sw_if_id1=sw_if_index1,
- sw_if_id2=sw_if_index2,
- bd_id=bd_id,
- learn=int(learn))
+ learn_int = 1 if learn else 0
+
+ cmd1 = 'bridge_domain_add_del'
+ args1 = dict(bd_id=int(bd_id),
+ flood=1,
+ uu_flood=1,
+ forward=1,
+ learn=learn_int,
+ arp_term=0,
+ is_add=1)
+
+ cmd2 = 'sw_interface_set_l2_bridge'
+ args2 = dict(rx_sw_if_index=sw_if_index1,
+ bd_id=int(bd_id),
+ shg=0,
+ port_type=0,
+ enable=1)
+
+ args3 = dict(rx_sw_if_index=sw_if_index2,
+ bd_id=int(bd_id),
+ shg=0,
+ port_type=0,
+ enable=1)
+
+ err_msg = 'Failed to add L2 bridge domain with 2 interfaces on host' \
+ ' {host}'.format(host=node['host'])
+
+ with PapiExecutor(node) as papi_exec:
+ papi_exec.add(cmd1, **args1).add(cmd2, **args2).add(cmd2, **args3).\
+ get_replies(err_msg).verify_replies(err_msg=err_msg)
@staticmethod
def vpp_setup_bidirectional_cross_connect(node, interface1, interface2):
@@ -208,13 +276,20 @@ class L2Util(object):
else:
sw_iface2 = interface2
- with VatTerminal(node) as vat:
- vat.vat_terminal_exec_cmd_from_template('l2_xconnect.vat',
- interface1=sw_iface1,
- interface2=sw_iface2)
- vat.vat_terminal_exec_cmd_from_template('l2_xconnect.vat',
- interface1=sw_iface2,
- interface2=sw_iface1)
+ cmd = 'sw_interface_set_l2_xconnect'
+ args1 = dict(rx_sw_if_index=sw_iface1,
+ tx_sw_if_index=sw_iface2,
+ enable=1)
+ args2 = dict(rx_sw_if_index=sw_iface2,
+ tx_sw_if_index=sw_iface1,
+ enable=1)
+
+ err_msg = 'Failed to add L2 cross-connect between two interfaces on' \
+ ' host {host}'.format(host=node['host'])
+
+ with PapiExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args1).add(cmd, **args2).get_replies(err_msg).\
+ verify_replies(err_msg=err_msg)
@staticmethod
def vpp_setup_bidirectional_l2_patch(node, interface1, interface2):
@@ -238,13 +313,20 @@ class L2Util(object):
else:
sw_iface2 = interface2
- with VatTerminal(node) as vat:
- vat.vat_terminal_exec_cmd_from_template('l2_patch.vat',
- interface1=sw_iface1,
- interface2=sw_iface2)
- vat.vat_terminal_exec_cmd_from_template('l2_patch.vat',
- interface1=sw_iface2,
- interface2=sw_iface1)
+ cmd = 'l2_patch_add_del'
+ args1 = dict(rx_sw_if_index=sw_iface1,
+ tx_sw_if_index=sw_iface2,
+ is_add=1)
+ args2 = dict(rx_sw_if_index=sw_iface2,
+ tx_sw_if_index=sw_iface1,
+ is_add=1)
+
+ err_msg = 'Failed to add L2 patch between two interfaces on' \
+ ' host {host}'.format(host=node['host'])
+
+ with PapiExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args1).add(cmd, **args2).get_replies(err_msg).\
+ verify_replies(err_msg=err_msg)
@staticmethod
def linux_add_bridge(node, br_name, if_1, if_2, set_up=True):
@@ -262,6 +344,7 @@ class L2Util(object):
:type if_2: str
:type set_up: bool
"""
+
cmd = 'brctl addbr {0}'.format(br_name)
exec_cmd_no_error(node, cmd, sudo=True)
cmd = 'brctl addif {0} {1}'.format(br_name, if_1)
@@ -287,6 +370,7 @@ class L2Util(object):
:type br_name: str
:type set_down: bool
"""
+
if set_down:
cmd = 'ip link set dev {0} down'.format(br_name)
exec_cmd_no_error(node, cmd, sudo=True)
@@ -294,7 +378,7 @@ class L2Util(object):
exec_cmd_no_error(node, cmd, sudo=True)
@staticmethod
- def vpp_get_bridge_domain_data(node, bd_id=None):
+ def vpp_get_bridge_domain_data(node, bd_id=0xffffffff):
"""Get all bridge domain data from a VPP node. If a domain ID number is
provided, return only data for the matching bridge domain.
@@ -306,18 +390,30 @@ class L2Util(object):
or a single dictionary for the specified bridge domain.
:rtype: list or dict
"""
- with VatTerminal(node) as vat:
- response = vat.vat_terminal_exec_cmd_from_template("l2_bd_dump.vat")
- data = response[0]
+ # TODO: set following variable per whole suite when planned FIB API
+ # changes are merged in VPP
+ bitwise_non_zero = 0xffffffff # equals to ~0 used in vpp code
+
+ cmd = 'bridge_domain_dump'
+ cmd_reply = 'bridge_domain_details'
+ args = dict(bd_id=bd_id if isinstance(bd_id, int) else int(bd_id))
+ err_msg = 'Failed to get L2FIB dump on host {host}'.format(
+ host=node['host'])
+ with PapiExecutor(node) as papi_exec:
+ papi_resp = papi_exec.add(cmd, **args).get_dump(err_msg)
- if bd_id is not None:
- for bridge_domain in data:
- if bridge_domain["bd_id"] == bd_id:
+ data = papi_resp.reply[0]['api_reply']
- return bridge_domain
+ bd_data = list() if bd_id == bitwise_non_zero else dict()
+ for bridge_domain in data:
+ if bd_id == bitwise_non_zero:
+ bd_data.append(bridge_domain[cmd_reply])
+ else:
+ if bridge_domain[cmd_reply]['bd_id'] == bd_id:
+ return bridge_domain[cmd_reply]
- return data
+ return bd_data
@staticmethod
def l2_vlan_tag_rewrite(node, interface, tag_rewrite_method,
@@ -338,10 +434,12 @@ class L2Util(object):
:type tag1_id: int
:type tag2_id: int
"""
- push_dot1q = 'push_dot1q 0' if not push_dot1q else ''
- tag1_id = 'tag1 {0}'.format(tag1_id) if tag1_id else ''
- tag2_id = 'tag2 {0}'.format(tag2_id) if tag2_id else ''
+ tag1_id = int(tag1_id) if tag1_id else 0
+ tag2_id = int(tag2_id) if tag2_id else 0
+
+ vtr_oper = getattr(L2VtrOp, 'L2_VTR_{}'.format(
+ tag_rewrite_method.replace('-', '_').upper()))
if isinstance(interface, basestring):
iface_key = Topology.get_interface_by_name(node, interface)
@@ -349,73 +447,51 @@ class L2Util(object):
else:
sw_if_index = interface
- with VatTerminal(node) as vat:
- vat.vat_terminal_exec_cmd_from_template("l2_vlan_tag_rewrite.vat",
- sw_if_index=sw_if_index,
- tag_rewrite_method=
- tag_rewrite_method,
- push_dot1q=push_dot1q,
- tag1_optional=tag1_id,
- tag2_optional=tag2_id)
-
- @staticmethod
- def delete_bridge_domain_vat(node, bd_id):
- """Delete the specified bridge domain from the node.
-
- :param node: VPP node to delete a bridge domain from.
- :param bd_id: Bridge domain ID.
- :type node: dict
- :type bd_id: int
- """
-
- with VatTerminal(node) as vat:
- vat.vat_terminal_exec_cmd_from_template(
- "l2_bridge_domain_delete.vat", bd_id=bd_id)
+ cmd = 'l2_interface_vlan_tag_rewrite'
+ args = dict(sw_if_index=sw_if_index,
+ vtr_op=int(vtr_oper),
+ push_dot1q=int(push_dot1q),
+ tag1=tag1_id,
+ tag2=tag2_id)
+ err_msg = 'Failed to set VLAN TAG rewrite on host {host}'.format(
+ host=node['host'])
+ with PapiExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_replies(err_msg).\
+ verify_reply(err_msg=err_msg)
@staticmethod
- def delete_l2_fib_entry(node, bd_id, mac):
- """Delete the specified L2 FIB entry.
+ def get_l2_fib_table(node, bd_id):
+ """Retrieves the L2 FIB table.
:param node: VPP node.
- :param bd_id: Bridge domain ID.
- :param mac: MAC address used as the key in L2 FIB entry.
+ :param bd_id: Index of the bridge domain.
:type node: dict
:type bd_id: int
- :type mac: str
- """
-
- with VatTerminal(node) as vat:
- vat.vat_terminal_exec_cmd_from_template("l2_fib_entry_delete.vat",
- mac=mac,
- bd_id=bd_id)
-
- @staticmethod
- def get_l2_fib_table_vat(node, bd_index):
- """Retrieves the L2 FIB table using VAT.
-
- :param node: VPP node.
- :param bd_index: Index of the bridge domain.
- :type node: dict
- :type bd_index: int
:returns: L2 FIB table.
:rtype: list
"""
- bd_data = L2Util.vpp_get_bridge_domain_data(node)
- bd_id = bd_data[bd_index-1]["bd_id"]
+ cmd = 'l2_fib_table_dump'
+ cmd_reply = 'l2_fib_table_details'
+ args = dict(bd_id=int(bd_id))
+ err_msg = 'Failed to get L2FIB dump on host {host}'.format(
+ host=node['host'])
+ with PapiExecutor(node) as papi_exec:
+ papi_resp = papi_exec.add(cmd, **args).get_dump(err_msg)
- try:
- with VatTerminal(node) as vat:
- table = vat.vat_terminal_exec_cmd_from_template(
- "l2_fib_table_dump.vat", bd_id=bd_id)
+ data = papi_resp.reply[0]['api_reply']
- return table[0]
- except ValueError:
- return []
+ fib_data = list()
+ for fib in data:
+ fib_item = fib[cmd_reply]
+ fib_item['mac'] = L2Util.bin_to_mac(fib_item['mac'])
+ fib_data.append(fib_item)
+
+ return fib_data
@staticmethod
- def get_l2_fib_entry_vat(node, bd_index, mac):
- """Retrieves the L2 FIB entry specified by MAC address using VAT.
+ def get_l2_fib_entry_by_mac(node, bd_index, mac):
+ """Retrieves the L2 FIB entry specified by MAC address using PAPI.
:param node: VPP node.
:param bd_index: Index of the bridge domain.
@@ -427,8 +503,12 @@ class L2Util(object):
:rtype: dict
"""
- table = L2Util.get_l2_fib_table_vat(node, bd_index)
+ bd_data = L2Util.vpp_get_bridge_domain_data(node)
+ bd_id = bd_data[bd_index-1]['bd_id']
+
+ table = L2Util.get_l2_fib_table(node, bd_id)
+
for entry in table:
- if entry["mac"] == mac:
+ if entry['mac'] == mac:
return entry
return {}
diff --git a/resources/libraries/python/PapiExecutor.py b/resources/libraries/python/PapiExecutor.py
index f6df2004f4..a0261992e9 100644
--- a/resources/libraries/python/PapiExecutor.py
+++ b/resources/libraries/python/PapiExecutor.py
@@ -457,7 +457,8 @@ class PapiExecutor(object):
reply_value = dict()
for reply_key, reply_v in api_r.iteritems():
for a_k, a_v in reply_v.iteritems():
- reply_value[a_k] = a_v
+ reply_value[a_k] = binascii.unhexlify(a_v) \
+ if isinstance(a_v, unicode) else a_v
reply_dict[reply_key] = reply_value
return reply_dict
@@ -581,6 +582,9 @@ class PapiExecutor(object):
raise
papi_reply.append(api_reply_processed)
+ # Log processed papi reply to be able to check API replies changes
+ logger.debug("Processed PAPI reply: {reply}".format(reply=papi_reply))
+
return PapiResponse(papi_reply=papi_reply,
stdout=stdout,
stderr=stderr,