CSIT-158: Tap interface tests
[csit.git] / resources / libraries / python / InterfaceUtil.py
1 # Copyright (c) 2016 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:
5 #
6 #     http://www.apache.org/licenses/LICENSE-2.0
7 #
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.
13
14 """Interface util library"""
15
16 from time import time, sleep
17
18 from robot.api import logger
19
20 from resources.libraries.python.ssh import SSH
21 from resources.libraries.python.IPUtil import convert_ipv4_netmask_prefix
22 from resources.libraries.python.ssh import exec_cmd_no_error
23 from resources.libraries.python.topology import NodeType, Topology
24 from resources.libraries.python.VatExecutor import VatExecutor, VatTerminal
25 from resources.libraries.python.VatJsonUtil import VatJsonUtil
26 from resources.libraries.python.parsers.JsonParser import JsonParser
27
28
29 class InterfaceUtil(object):
30     """General utilities for managing interfaces"""
31
32     __UDEV_IF_RULES_FILE = '/etc/udev/rules.d/10-network.rules'
33
34     @staticmethod
35     def set_interface_state(node, interface, state, if_type="key"):
36         """Set interface state on a node.
37
38         Function can be used for DUTs as well as for TGs.
39
40         :param node: Node where the interface is.
41         :param interface: Interface key or sw_if_index or name.
42         :param state: One of 'up' or 'down'.
43         :param if_type: Interface type
44         :type node: dict
45         :type interface: str or int
46         :type state: str
47         :type if_type: str
48         :return: nothing
49         """
50
51         if if_type == "key":
52             if isinstance(interface, basestring):
53                 sw_if_index = Topology.get_interface_sw_index(node, interface)
54                 iface_name = Topology.get_interface_name(node, interface)
55             else:
56                 sw_if_index = interface
57         elif if_type == "name":
58             iface_key = Topology.get_interface_by_name(node, interface)
59             if iface_key is not None:
60                 sw_if_index = Topology.get_interface_sw_index(node, iface_key)
61             iface_name = interface
62         else:
63             raise ValueError("if_type unknown: {}".format(if_type))
64
65         if node['type'] == NodeType.DUT:
66             if state == 'up':
67                 state = 'admin-up'
68             elif state == 'down':
69                 state = 'admin-down'
70             else:
71                 raise ValueError('Unexpected interface state: {}'.format(state))
72             VatExecutor.cmd_from_template(node, 'set_if_state.vat',
73                                           sw_if_index=sw_if_index, state=state)
74         elif node['type'] == NodeType.TG or node['type'] == NodeType.VM:
75             cmd = 'ip link set {} {}'.format(iface_name, state)
76             exec_cmd_no_error(node, cmd, sudo=True)
77         else:
78             raise Exception('Node {} has unknown NodeType: "{}"'.
79                             format(node['host'], node['type']))
80
81     @staticmethod
82     def set_interface_ethernet_mtu(node, iface_key, mtu):
83         """Set Ethernet MTU for specified interface.
84
85         Function can be used only for TGs.
86
87         :param node: Node where the interface is.
88         :param interface: Interface key from topology file.
89         :param mtu: MTU to set.
90         :type node: dict
91         :type iface_key: str
92         :type mtu: int
93         :return: nothing
94         """
95         if node['type'] == NodeType.DUT:
96             ValueError('Node {}: Setting Ethernet MTU for interface '
97                        'on DUT nodes not supported', node['host'])
98         elif node['type'] == NodeType.TG:
99             iface_name = Topology.get_interface_name(node, iface_key)
100             cmd = 'ip link set {} mtu {}'.format(iface_name, mtu)
101             exec_cmd_no_error(node, cmd, sudo=True)
102         else:
103             raise ValueError('Node {} has unknown NodeType: "{}"'.
104                              format(node['host'], node['type']))
105
106     @staticmethod
107     def set_default_ethernet_mtu_on_all_interfaces_on_node(node):
108         """Set default Ethernet MTU on all interfaces on node.
109
110         Function can be used only for TGs.
111
112         :param node: Node where to set default MTU.
113         :type node: dict
114         :return: nothing
115         """
116         for ifc in node['interfaces']:
117             InterfaceUtil.set_interface_ethernet_mtu(node, ifc, 1500)
118
119     @staticmethod
120     def vpp_node_interfaces_ready_wait(node, timeout=10):
121         """Wait until all interfaces with admin-up are in link-up state.
122
123         :param node: Node to wait on.
124         :param timeout: Waiting timeout in seconds (optional, default 10s).
125         :type node: dict
126         :type timeout: int
127         :raises: RuntimeError if the timeout period value has elapsed.
128         """
129         if_ready = False
130         not_ready = []
131         start = time()
132         while not if_ready:
133             out = InterfaceUtil.vpp_get_interface_data(node)
134             if time() - start > timeout:
135                 for interface in out:
136                     if interface.get('admin_up_down') == 1:
137                         if interface.get('link_up_down') != 1:
138                             logger.debug('{0} link-down'.format(
139                                 interface.get('interface_name')))
140                 raise RuntimeError('timeout, not up {0}'.format(not_ready))
141             not_ready = []
142             for interface in out:
143                 if interface.get('admin_up_down') == 1:
144                     if interface.get('link_up_down') != 1:
145                         not_ready.append(interface.get('interface_name'))
146             if not not_ready:
147                 if_ready = True
148             else:
149                 logger.debug('Interfaces still in link-down state: {0}, '
150                              'waiting...'.format(not_ready))
151                 sleep(1)
152
153     @staticmethod
154     def vpp_nodes_interfaces_ready_wait(nodes, timeout=10):
155         """Wait until all interfaces with admin-up are in link-up state for
156         listed nodes.
157
158         :param nodes: List of nodes to wait on.
159         :param timeout: Seconds to wait per node for all interfaces to come up.
160         :type nodes: list
161         :type timeout: int
162         :raises: RuntimeError if the timeout period value has elapsed.
163         """
164         for node in nodes:
165             InterfaceUtil.vpp_node_interfaces_ready_wait(node, timeout)
166
167     @staticmethod
168     def all_vpp_interfaces_ready_wait(nodes, timeout=10):
169         """Wait until all interfaces with admin-up are in link-up state for all
170         nodes in the topology.
171
172         :param nodes: Nodes in the topology.
173         :param timeout: Seconds to wait per node for all interfaces to come up.
174         :type nodes: dict
175         :type timeout: int
176         :raises: RuntimeError if the timeout period value has elapsed.
177         """
178         for node in nodes.values():
179             if node['type'] == NodeType.DUT:
180                 InterfaceUtil.vpp_node_interfaces_ready_wait(node, timeout)
181
182     @staticmethod
183     def vpp_get_interface_data(node, interface=None):
184         """Get all interface data from a VPP node. If a name or
185         sw_interface_index is provided, return only data for the matching
186         interface.
187
188         :param node: VPP node to get interface data from.
189         :param interface: Numeric index or name string of a specific interface.
190         :type node: dict
191         :type interface: int or str
192         :return: List of dictionaries containing data for each interface, or a
193         single dictionary for the specified interface.
194         :rtype: list or dict
195         """
196         with VatTerminal(node) as vat:
197             response = vat.vat_terminal_exec_cmd_from_template(
198                 "interface_dump.vat")
199
200         data = response[0]
201
202         if interface is not None:
203             if isinstance(interface, basestring):
204                 param = "interface_name"
205             elif isinstance(interface, int):
206                 param = "sw_if_index"
207             else:
208                 raise TypeError
209             for data_if in data:
210                 if data_if[param] == interface:
211                     return data_if
212             return dict()
213         return data
214
215     @staticmethod
216     def vpp_get_interface_ip_addresses(node, interface, ip_version):
217         """Get list of IP addresses from an interface on a VPP node.
218
219          :param node: VPP node to get data from.
220          :param interface: Name of an interface on the VPP node.
221          :param ip_version: IP protocol version (ipv4 or ipv6).
222          :type node: dict
223          :type interface: str
224          :type ip_version: str
225          :return: List of dictionaries, each containing IP address, subnet
226          prefix length and also the subnet mask for ipv4 addresses.
227          Note: A single interface may have multiple IP addresses assigned.
228          :rtype: list
229         """
230         sw_if_index = InterfaceUtil.get_sw_if_index(node, interface)
231
232         with VatTerminal(node) as vat:
233             response = vat.vat_terminal_exec_cmd_from_template(
234                 "ip_address_dump.vat", ip_version=ip_version,
235                 sw_if_index=sw_if_index)
236
237         data = response[0]
238
239         if ip_version == "ipv4":
240             for item in data:
241                 item["netmask"] = convert_ipv4_netmask_prefix(
242                     item["prefix_length"])
243         return data
244
245     @staticmethod
246     def tg_set_interface_driver(node, pci_addr, driver):
247         """Set interface driver on the TG node.
248
249         :param node: Node to set interface driver on (must be TG node).
250         :param pci_addr: PCI address of the interface.
251         :param driver: Driver name.
252         :type node: dict
253         :type pci_addr: str
254         :type driver: str
255         """
256         old_driver = InterfaceUtil.tg_get_interface_driver(node, pci_addr)
257         if old_driver == driver:
258             return
259
260         ssh = SSH()
261         ssh.connect(node)
262
263         # Unbind from current driver
264         if old_driver is not None:
265             cmd = 'sh -c "echo {0} > /sys/bus/pci/drivers/{1}/unbind"'.format(
266                 pci_addr, old_driver)
267             (ret_code, _, _) = ssh.exec_command_sudo(cmd)
268             if int(ret_code) != 0:
269                 raise Exception("'{0}' failed on '{1}'".format(cmd,
270                                                                node['host']))
271
272         # Bind to the new driver
273         cmd = 'sh -c "echo {0} > /sys/bus/pci/drivers/{1}/bind"'.format(
274             pci_addr, driver)
275         (ret_code, _, _) = ssh.exec_command_sudo(cmd)
276         if int(ret_code) != 0:
277             raise Exception("'{0}' failed on '{1}'".format(cmd, node['host']))
278
279     @staticmethod
280     def tg_get_interface_driver(node, pci_addr):
281         """Get interface driver from the TG node.
282
283         :param node: Node to get interface driver on (must be TG node).
284         :param pci_addr: PCI address of the interface.
285         :type node: dict
286         :type pci_addr: str
287         :return: Interface driver or None if not found.
288         :rtype: str
289
290         .. note::
291             # lspci -vmmks 0000:00:05.0
292             Slot:   00:05.0
293             Class:  Ethernet controller
294             Vendor: Red Hat, Inc
295             Device: Virtio network device
296             SVendor:        Red Hat, Inc
297             SDevice:        Device 0001
298             PhySlot:        5
299             Driver: virtio-pci
300         """
301         ssh = SSH()
302         ssh.connect(node)
303
304         cmd = 'lspci -vmmks {0}'.format(pci_addr)
305
306         (ret_code, stdout, _) = ssh.exec_command(cmd)
307         if int(ret_code) != 0:
308             raise Exception("'{0}' failed on '{1}'".format(cmd, node['host']))
309
310         for line in stdout.splitlines():
311             if len(line) == 0:
312                 continue
313             (name, value) = line.split("\t", 1)
314             if name == 'Driver:':
315                 return value
316
317         return None
318
319     @staticmethod
320     def tg_set_interfaces_udev_rules(node):
321         """Set udev rules for interfaces.
322
323         Create udev rules file in /etc/udev/rules.d where are rules for each
324         interface used by TG node, based on MAC interface has specific name.
325         So after unbind and bind again to kernel driver interface has same
326         name as before. This must be called after TG has set name for each
327         port in topology dictionary.
328         udev rule example
329         SUBSYSTEM=="net", ACTION=="add", ATTR{address}=="52:54:00:e1:8a:0f",
330         NAME="eth1"
331
332         :param node: Node to set udev rules on (must be TG node).
333         :type node: dict
334         """
335         ssh = SSH()
336         ssh.connect(node)
337
338         cmd = 'rm -f {0}'.format(InterfaceUtil.__UDEV_IF_RULES_FILE)
339         (ret_code, _, _) = ssh.exec_command_sudo(cmd)
340         if int(ret_code) != 0:
341             raise Exception("'{0}' failed on '{1}'".format(cmd, node['host']))
342
343         for interface in node['interfaces'].values():
344             rule = 'SUBSYSTEM==\\"net\\", ACTION==\\"add\\", ATTR{address}' + \
345                    '==\\"' + interface['mac_address'] + '\\", NAME=\\"' + \
346                    interface['name'] + '\\"'
347             cmd = 'sh -c "echo \'{0}\' >> {1}"'.format(
348                 rule, InterfaceUtil.__UDEV_IF_RULES_FILE)
349             (ret_code, _, _) = ssh.exec_command_sudo(cmd)
350             if int(ret_code) != 0:
351                 raise Exception("'{0}' failed on '{1}'".format(cmd,
352                                                                node['host']))
353
354         cmd = '/etc/init.d/udev restart'
355         ssh.exec_command_sudo(cmd)
356
357     @staticmethod
358     def tg_set_interfaces_default_driver(node):
359         """Set interfaces default driver specified in topology yaml file.
360
361         :param node: Node to setup interfaces driver on (must be TG node).
362         :type node: dict
363         """
364         for interface in node['interfaces'].values():
365             InterfaceUtil.tg_set_interface_driver(node,
366                                                   interface['pci_address'],
367                                                   interface['driver'])
368
369     @staticmethod
370     def update_vpp_interface_data_on_node(node):
371         """Update vpp generated interface data for a given node in DICT__nodes.
372
373         Updates interface names, software if index numbers and any other details
374         generated specifically by vpp that are unknown before testcase run.
375         It does this by dumping interface list to JSON output from all
376         devices using vpp_api_test, and pairing known information from topology
377         (mac address/pci address of interface) to state from VPP.
378
379         :param node: Node selected from DICT__nodes.
380         :type node: dict
381         """
382         vat_executor = VatExecutor()
383         vat_executor.execute_script_json_out("dump_interfaces.vat", node)
384         interface_dump_json = vat_executor.get_script_stdout()
385         VatJsonUtil.update_vpp_interface_data_from_json(node,
386                                                         interface_dump_json)
387
388     @staticmethod
389     def update_tg_interface_data_on_node(node):
390         """Update interface name for TG/linux node in DICT__nodes.
391
392         :param node: Node selected from DICT__nodes.
393         :type node: dict
394
395         .. note::
396             # for dev in `ls /sys/class/net/`;
397             > do echo "\"`cat /sys/class/net/$dev/address`\": \"$dev\""; done
398             "52:54:00:9f:82:63": "eth0"
399             "52:54:00:77:ae:a9": "eth1"
400             "52:54:00:e1:8a:0f": "eth2"
401             "00:00:00:00:00:00": "lo"
402
403         .. todo:: parse lshw -json instead
404         """
405         # First setup interface driver specified in yaml file
406         InterfaceUtil.tg_set_interfaces_default_driver(node)
407
408         # Get interface names
409         ssh = SSH()
410         ssh.connect(node)
411
412         cmd = ('for dev in `ls /sys/class/net/`; do echo "\\"`cat '
413                '/sys/class/net/$dev/address`\\": \\"$dev\\""; done;')
414
415         (ret_code, stdout, _) = ssh.exec_command(cmd)
416         if int(ret_code) != 0:
417             raise Exception('Get interface name and MAC failed')
418         tmp = "{" + stdout.rstrip().replace('\n', ',') + "}"
419         interfaces = JsonParser().parse_data(tmp)
420         for interface in node['interfaces'].values():
421             name = interfaces.get(interface['mac_address'])
422             if name is None:
423                 continue
424             interface['name'] = name
425
426         # Set udev rules for interfaces
427         InterfaceUtil.tg_set_interfaces_udev_rules(node)
428
429     @staticmethod
430     def update_all_interface_data_on_all_nodes(nodes):
431         """Update interface names on all nodes in DICT__nodes.
432
433         This method updates the topology dictionary by querying interface lists
434         of all nodes mentioned in the topology dictionary.
435
436         :param nodes: Nodes in the topology.
437         :type nodes: dict
438         """
439         for node_data in nodes.values():
440             if node_data['type'] == NodeType.DUT:
441                 InterfaceUtil.update_vpp_interface_data_on_node(node_data)
442             elif node_data['type'] == NodeType.TG:
443                 InterfaceUtil.update_tg_interface_data_on_node(node_data)
444
445     @staticmethod
446     def create_vlan_subinterface(node, interface, vlan):
447         """Create VLAN subinterface on node.
448
449         :param node: Node to add VLAN subinterface on.
450         :param interface: Interface name on which create VLAN subinterface.
451         :param vlan: VLAN ID of the subinterface to be created.
452         :type node: dict
453         :type interface: str
454         :type vlan: int
455         :return: Name and index of created subinterface.
456         :rtype: tuple
457         """
458         iface_key = Topology.get_interface_by_name(node, interface)
459         sw_if_index = Topology.get_interface_sw_index(node, iface_key)
460
461         output = VatExecutor.cmd_from_template(node, "create_vlan_subif.vat",
462                                                sw_if_index=sw_if_index,
463                                                vlan=vlan)
464         if output[0]["retval"] == 0:
465             sw_subif_index = output[0]["sw_if_index"]
466             logger.trace('VLAN subinterface with sw_if_index {} and VLAN ID {} '
467                          'created on node {}'.format(sw_subif_index,
468                                                      vlan, node['host']))
469         else:
470             raise RuntimeError('Unable to create VLAN subinterface on node {}'
471                                .format(node['host']))
472
473         with VatTerminal(node, False) as vat:
474             vat.vat_terminal_exec_cmd('exec show interfaces')
475
476         return '{}.{}'.format(interface, vlan), sw_subif_index
477
478     @staticmethod
479     def create_vxlan_interface(node, vni, source_ip, destination_ip):
480         """Create VXLAN interface and return sw if index of created interface.
481
482         Executes "vxlan_add_del_tunnel src {src} dst {dst} vni {vni}" VAT
483         command on the node.
484
485         :param node: Node where to create VXLAN interface.
486         :param vni: VXLAN Network Identifier.
487         :param source_ip: Source IP of a VXLAN Tunnel End Point.
488         :param destination_ip: Destination IP of a VXLAN Tunnel End Point.
489         :type node: dict
490         :type vni: int
491         :type source_ip: str
492         :type destination_ip: str
493         :return: SW IF INDEX of created interface.
494         :rtype: int
495         """
496         output = VatExecutor.cmd_from_template(node, "vxlan_create.vat",
497                                                src=source_ip,
498                                                dst=destination_ip,
499                                                vni=vni)
500         output = output[0]
501
502         if output["retval"] == 0:
503             return output["sw_if_index"]
504         else:
505             raise RuntimeError('Unable to create VXLAN interface on node {0}'
506                                .format(node))
507
508     @staticmethod
509     def vxlan_dump(node, interface=None):
510         """Get VxLAN data for the given interface.
511
512         :param node: VPP node to get interface data from.
513         :param interface: Numeric index or name string of a specific interface.
514         If None, information about all VxLAN interfaces is returned.
515         :type node: dict
516         :type interface: int or str
517         :return: Dictionary containing data for the given VxLAN interface or if
518         interface=None, the list of dictionaries with all VxLAN interfaces.
519         :rtype: dict or list
520         """
521         param = "sw_if_index"
522         if interface is None:
523             param = ''
524             sw_if_index = ''
525         elif isinstance(interface, basestring):
526             sw_if_index = Topology.get_interface_sw_index(node, interface)
527         elif isinstance(interface, int):
528             sw_if_index = interface
529         else:
530             raise Exception("Wrong interface format {0}".format(interface))
531
532         with VatTerminal(node) as vat:
533             response = vat.vat_terminal_exec_cmd_from_template(
534                 "vxlan_dump.vat", param=param, sw_if_index=sw_if_index)
535
536         if sw_if_index:
537             for vxlan in response[0]:
538                 if vxlan["sw_if_index"] == sw_if_index:
539                     return vxlan
540             return {}
541         return response[0]
542
543     @staticmethod
544     def vhost_user_dump(node):
545         """Get vhost-user data for the given node.
546
547         :param node: VPP node to get interface data from.
548         :type node: dict
549         :return: List of dictionaries with all vhost-user interfaces.
550         :rtype: list
551         """
552         with VatTerminal(node) as vat:
553             response = vat.vat_terminal_exec_cmd_from_template(
554                 "vhost_user_dump.vat")
555
556         return response[0]
557
558     @staticmethod
559     def tap_dump(node, name=None):
560         """Get all TAP interface data from the given node, or data about
561         a specific TAP interface.
562
563         :param node: VPP node to get data from.
564         :param name: Optional name of a specific TAP interface.
565         :type node: dict
566         :type name: str
567         :return: Dictionary of information about a specific TAP interface, or
568         a List of dictionaries containing all TAP data for the given node.
569         :rtype: dict or list
570         """
571         with VatTerminal(node) as vat:
572             response = vat.vat_terminal_exec_cmd_from_template(
573                 "tap_dump.vat")
574         if name is None:
575             return response[0]
576         else:
577             for item in response[0]:
578                 if name == item['dev_name']:
579                     return item
580             return {}
581
582     @staticmethod
583     def create_subinterface(node, interface, sub_id, outer_vlan_id=None,
584                             inner_vlan_id=None, type_subif=None):
585         """Create sub-interface on node. It is possible to set required
586         sub-interface type and VLAN tag(s).
587
588         :param node: Node to add sub-interface.
589         :param interface: Interface name on which create sub-interface.
590         :param sub_id: ID of the sub-interface to be created.
591         :param outer_vlan_id: Optional outer VLAN ID.
592         :param inner_vlan_id: Optional inner VLAN ID.
593         :param type_subif: Optional type of sub-interface. Values supported by
594         VPP: [no_tags] [one_tag] [two_tags] [dot1ad] [exact_match] [default_sub]
595         :type node: dict
596         :type interface: str or int
597         :type sub_id: int
598         :type outer_vlan_id: int
599         :type inner_vlan_id: int
600         :type type_subif: str
601         :return: Name and index of created sub-interface.
602         :rtype: tuple
603         :raises RuntimeError: If it is not possible to create sub-interface.
604         """
605
606         outer_vlan_id = 'outer_vlan_id {0}'.format(outer_vlan_id)\
607             if outer_vlan_id else ''
608
609         inner_vlan_id = 'inner_vlan_id {0}'.format(inner_vlan_id)\
610             if inner_vlan_id else ''
611
612         if type_subif is None:
613             type_subif = ''
614
615         if isinstance(interface, basestring):
616             iface_key = Topology.get_interface_by_name(node, interface)
617             sw_if_index = Topology.get_interface_sw_index(node, iface_key)
618         else:
619             sw_if_index = interface
620
621         output = VatExecutor.cmd_from_template(node, "create_sub_interface.vat",
622                                                sw_if_index=sw_if_index,
623                                                sub_id=sub_id,
624                                                outer_vlan_id=outer_vlan_id,
625                                                inner_vlan_id=inner_vlan_id,
626                                                type_subif=type_subif)
627
628         if output[0]["retval"] == 0:
629             sw_subif_index = output[0]["sw_if_index"]
630             logger.trace('Created subinterface with index {}'
631                          .format(sw_subif_index))
632         else:
633             raise RuntimeError('Unable to create sub-interface on node {}'
634                                .format(node['host']))
635
636         with VatTerminal(node, json_param=False) as vat:
637             vat.vat_terminal_exec_cmd('exec show interfaces')
638
639         name = '{}.{}'.format(interface, sub_id)
640         return name, sw_subif_index
641
642     @staticmethod
643     def create_gre_tunnel_interface(node, source_ip, destination_ip):
644         """Create GRE tunnel interface on node.
645
646         :param node: VPP node to add tunnel interface.
647         :param source_ip: Source of the GRE tunnel.
648         :param destination_ip: Destination of the GRE tunnel.
649         :type node: dict
650         :type source_ip: str
651         :type destination_ip: str
652         :return: Name and index of created GRE tunnel interface.
653         :rtype: tuple
654         :raises RuntimeError: If unable to create GRE tunnel interface.
655         """
656         output = VatExecutor.cmd_from_template(node, "create_gre.vat",
657                                                src=source_ip,
658                                                dst=destination_ip)
659         output = output[0]
660
661         if output["retval"] == 0:
662             sw_if_index = output["sw_if_index"]
663
664             vat_executor = VatExecutor()
665             vat_executor.execute_script_json_out("dump_interfaces.vat", node)
666             interface_dump_json = vat_executor.get_script_stdout()
667             name = VatJsonUtil.get_interface_name_from_json(
668                 interface_dump_json, sw_if_index)
669             return name, sw_if_index
670         else:
671             raise RuntimeError('Unable to create GRE tunnel on node {}.'
672                                .format(node))
673
674     @staticmethod
675     def vpp_create_loopback(node):
676         """Create loopback interface on VPP node.
677
678         :param node: Node to create loopback interface on.
679         :type node: dict
680         :return: SW interface index.
681         :rtype: int
682         """
683         out = VatExecutor.cmd_from_template(node, "create_loopback.vat")
684         if out[0].get('retval') == 0:
685             return out[0].get('sw_if_index')
686         else:
687             raise RuntimeError('Create loopback failed on node "{}"'
688                                .format(node['host']))
689
690     @staticmethod
691     def vpp_enable_input_acl_interface(node, interface, ip_version,
692                                        table_index):
693         """Enable input acl on interface.
694
695         :param node: VPP node to setup interface for input acl.
696         :param interface: Interface to setup input acl.
697         :param ip_version: Version of IP protocol.
698         :param table_index: Classify table index.
699         :type node: dict
700         :type interface: str or int
701         :type ip_version: str
702         :type table_index: int
703         """
704         if isinstance(interface, basestring):
705             sw_if_index = Topology.get_interface_sw_index(node, interface)
706         else:
707             sw_if_index = interface
708
709         with VatTerminal(node) as vat:
710             vat.vat_terminal_exec_cmd_from_template("input_acl_int.vat",
711                                                     sw_if_index=sw_if_index,
712                                                     ip_version=ip_version,
713                                                     table_index=table_index)
714
715     @staticmethod
716     def get_interface_classify_table(node, interface):
717         """Get name of classify table for the given interface.
718
719         :param node: VPP node to get data from.
720         :param interface: Name or sw_if_index of a specific interface.
721         :type node: dict
722         :type interface: str or int
723         :return: Classify table name.
724         :rtype: str
725         """
726         if isinstance(interface, basestring):
727             sw_if_index = InterfaceUtil.get_sw_if_index(node, interface)
728         else:
729             sw_if_index = interface
730
731         with VatTerminal(node) as vat:
732             data = vat.vat_terminal_exec_cmd_from_template(
733                 "classify_interface_table.vat",
734                 sw_if_index=sw_if_index
735             )
736         return data[0]
737
738     @staticmethod
739     def get_sw_if_index(node, interface_name):
740         """Get sw_if_index for the given interface from actual interface dump.
741
742         :param node: VPP node to get interface data from.
743         :param interface_name: Name of the specific interface.
744         :type node: dict
745         :type interface_name: str
746         :return: sw_if_index of the given interface.
747         :rtype: str
748         """
749
750         with VatTerminal(node) as vat:
751             if_data = vat.vat_terminal_exec_cmd_from_template(
752                 "interface_dump.vat")
753         for interface in if_data[0]:
754             if interface["interface_name"] == interface_name:
755                 return interface["sw_if_index"]
756
757         return None
758
759     @staticmethod
760     def vxlan_gpe_dump(node, interface_name=None):
761         """Get VxLAN GPE data for the given interface.
762
763         :param node: VPP node to get interface data from.
764         :param interface_name: Name of the specific interface. If None,
765         information about all VxLAN GPE interfaces is returned.
766         :type node: dict
767         :type interface_name: str
768         :return: Dictionary containing data for the given VxLAN GPE interface or
769         if interface=None, the list of dictionaries with all VxLAN GPE
770         interfaces.
771         :rtype: dict or list
772         """
773
774         with VatTerminal(node) as vat:
775             vxlan_gpe_data = vat.vat_terminal_exec_cmd_from_template(
776                 "vxlan_gpe_dump.vat")
777
778         if interface_name:
779             sw_if_index = InterfaceUtil.get_sw_if_index(node, interface_name)
780             if sw_if_index:
781                 for vxlan_gpe in vxlan_gpe_data[0]:
782                     if vxlan_gpe["sw_if_index"] == sw_if_index:
783                         return vxlan_gpe
784             return {}
785
786         return vxlan_gpe_data[0]
787
788     @staticmethod
789     def vpp_proxy_arp_interface_enable(node, interface):
790         """Enable proxy ARP on interface.
791
792         :param node: VPP node to enable proxy ARP on interface.
793         :param interface: Interface to enable proxy ARP.
794         :type node: dict
795         :type interface: str or int
796         """
797         if isinstance(interface, basestring):
798             sw_if_index = InterfaceUtil.get_sw_if_index(node, interface)
799         else:
800             sw_if_index = interface
801
802         with VatTerminal(node) as vat:
803             vat.vat_terminal_exec_cmd_from_template(
804                 "proxy_arp_intfc_enable.vat",
805                 sw_if_index=sw_if_index)
806
807     @staticmethod
808     def vpp_ip_source_check_setup(node, interface):
809         """Setup Reverse Path Forwarding source check on interface.
810
811         :param node: Node to setup RPF source check.
812         :param interface: Interface name to setup RPF source check.
813         :type node: dict
814         :type interface: str
815         """
816         with VatTerminal(node) as vat:
817             vat.vat_terminal_exec_cmd_from_template("ip_source_check.vat",
818                                                     interface_name=interface)
819
820     @staticmethod
821     def assign_interface_to_fib_table(node, interface, table_id):
822         """Assign VPP interface to specific VRF/FIB table.
823
824         :param node: VPP node where the FIB and interface are located.
825         :param interface: Interface to be assigned to FIB.
826         :param table_id: VRF table ID.
827         :type node: dict
828         :type interface: str or int
829         :type table_id: int
830         """
831         if isinstance(interface, basestring):
832             sw_if_index = Topology.get_interface_sw_index(node, interface)
833         else:
834             sw_if_index = interface
835
836         with VatTerminal(node) as vat:
837             vat.vat_terminal_exec_cmd_from_template("set_fib_to_interface.vat",
838                                                     sw_index=sw_if_index,
839                                                     vrf=table_id)
840
841     @staticmethod
842     def set_linux_interface_mac(node, interface, mac, namespace=None):
843         """Set MAC address for interface in linux.
844
845         :param node: Node where to execute command.
846         :param interface: Interface in namespace.
847         :param mac: MAC to be assigned to interface.
848         :param namespace: Execute command in namespace. Optional
849         :type node: dict
850         :type interface: str
851         :type mac: str
852         :type namespace: str
853         """
854         if namespace is not None:
855             cmd = 'ip netns exec {} ip link set {} address {}'.format(
856                 namespace, interface, mac)
857         else:
858             cmd = 'ip link set {} address {}'.format(interface, mac)
859         exec_cmd_no_error(node, cmd, sudo=True)