1 # Copyright (c) 2019 Cisco and/or its affiliates.
2 # Licensed under the Apache License, Version 2.0 (the "License");
3 # you may not use this file except in compliance with the License.
4 # You may obtain a copy of the License at:
6 # http://www.apache.org/licenses/LICENSE-2.0
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 # See the License for the specific language governing permissions and
12 # limitations under the License.
14 """Memif interface library."""
17 from enum import IntEnum
18 from robot.api import logger
20 from resources.libraries.python.topology import NodeType, Topology
21 from resources.libraries.python.PapiExecutor import PapiSocketExecutor
24 class MemifRole(IntEnum):
25 """Memif interface roles"""
31 """Memif interface class"""
37 def _memif_details(node):
38 """Get the memif dump details on the given node.
40 :param node: Given node to get Memif dump from.
42 :returns: List of memif interfaces extracted from Papi response.
46 with PapiSocketExecutor(node) as papi_exec:
47 details = papi_exec.add(cmd).get_details()
50 memif["hw_addr"] = str(memif["hw_addr"])
51 memif["role"] = memif["role"].value
52 memif["mode"] = memif["mode"].value
53 memif["flags"] = memif["flags"].value \
54 if hasattr(memif["flags"], 'value') else int(memif["flags"])
56 logger.debug("MEMIF details:\n{details}".format(details=details))
61 def _memif_socket_filename_add_del(node, is_add, filename, sid):
62 """Create Memif socket on the given node.
64 :param node: Given node to create Memif socket on.
65 :param is_add: If True, socket is added, otherwise deleted.
66 :param filename: Memif interface socket filename.
67 :param sid: Socket ID.
72 :returns: Verified data from PAPI response. In this case, the response
76 cmd = 'memif_socket_filename_add_del'
77 err_msg = 'Failed to create memif socket on host {host}'.format(
82 socket_filename=str('/tmp/' + filename)
84 with PapiSocketExecutor(node) as papi_exec:
85 return papi_exec.add(cmd, **args).get_reply(err_msg)
88 def _memif_create(node, mid, sid, rxq=1, txq=1, role=1):
89 """Create Memif interface on the given node, return its sw_if_index.
91 :param node: Given node to create Memif interface on.
92 :param mid: Memif interface ID.
93 :param sid: Socket ID.
94 :param rxq: Number of RX queues; 0 means do not set.
95 :param txq: Number of TX queues; 0 means do not set.
96 :param role: Memif interface role [master=0|slave=1]. Default is slave.
103 :returns: sw_if_index
107 err_msg = 'Failed to create memif interface on host {host}'.format(
117 with PapiSocketExecutor(node) as papi_exec:
118 return papi_exec.add(cmd, **args).get_sw_if_index(err_msg)
121 def create_memif_interface(node, filename, mid, sid, rxq=1, txq=1,
123 """Create Memif interface on the given node.
125 :param node: Given node to create Memif interface on.
126 :param filename: Memif interface socket filename.
127 :param mid: Memif interface ID.
128 :param sid: Socket ID.
129 :param rxq: Number of RX queues; 0 means do not set.
130 :param txq: Number of TX queues; 0 means do not set.
131 :param role: Memif interface role [master=0|slave=1]. Default is master.
139 :returns: SW interface index.
141 :raises ValueError: If command 'create memif' fails.
144 role = getattr(MemifRole, role.upper()).value
147 Memif._memif_socket_filename_add_del(node, True, filename, sid)
150 sw_if_index = Memif._memif_create(
151 node, mid, sid, rxq=rxq, txq=txq, role=role)
154 if_key = Topology.add_new_port(node, 'memif')
155 Topology.update_interface_sw_if_index(node, if_key, sw_if_index)
157 ifc_name = Memif.vpp_get_memif_interface_name(node, sw_if_index)
158 Topology.update_interface_name(node, if_key, ifc_name)
160 ifc_mac = Memif.vpp_get_memif_interface_mac(node, sw_if_index)
161 Topology.update_interface_mac_address(node, if_key, ifc_mac)
163 Topology.update_interface_memif_socket(node, if_key, '/tmp/' + filename)
164 Topology.update_interface_memif_id(node, if_key, mid)
165 Topology.update_interface_memif_role(node, if_key, str(role))
170 def show_memif(node):
171 """Show Memif data for the given node.
173 :param node: Given node to show Memif data on.
177 Memif._memif_details(node)
180 def show_memif_on_all_duts(nodes):
181 """Show Memif data on all DUTs.
183 :param nodes: Topology nodes.
186 for node in nodes.values():
187 if node['type'] == NodeType.DUT:
188 Memif.show_memif(node)
191 def vpp_get_memif_interface_name(node, sw_if_index):
192 """Get Memif interface name from Memif interfaces dump.
194 :param node: DUT node.
195 :param sw_if_index: DUT node.
197 :type sw_if_index: int
198 :returns: Memif interface name, or None if not found.
202 details = Memif._memif_details(node)
204 for memif in details:
205 if memif["sw_if_index"] == sw_if_index:
206 return memif["if_name"]
210 def vpp_get_memif_interface_mac(node, sw_if_index):
211 """Get Memif interface MAC address from Memif interfaces dump.
213 :param node: DUT node.
214 :param sw_if_index: DUT node.
216 :type sw_if_index: int
217 :returns: Memif interface MAC address, or None if not found.
221 details = Memif._memif_details(node)
223 for memif in details:
224 if memif["sw_if_index"] == sw_if_index:
225 return memif["hw_addr"]