X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FIPv4Util.py;h=96572f5e1df40452ce7bec90066767e6578e4041;hp=31e6bf13f62d9aa30bfb791cf40ffad0f0419197;hb=a114591eac9f52502048db886da2fb228c62254d;hpb=8c12ff59f1a5e750151f5eb0e806dcc80e91c3c2

diff --git a/resources/libraries/python/IPv4Util.py b/resources/libraries/python/IPv4Util.py
index 31e6bf13f6..96572f5e1d 100644
--- a/resources/libraries/python/IPv4Util.py
+++ b/resources/libraries/python/IPv4Util.py
@@ -18,6 +18,7 @@ from robot.api.deco import keyword
 
 from resources.libraries.python.topology import Topology
 from resources.libraries.python.IPv4Setup import get_node
+from resources.libraries.python.ssh import exec_cmd
 
 
 class IPv4Util(object):
@@ -27,6 +28,15 @@ class IPv4Util(object):
     @keyword('From node "${node}" interface "${port}" ARP-ping '
              'IPv4 address "${ip_address}"')
     def arp_ping(node, interface, ip_address):
+        """Send an ARP ping from the specified node.
+
+        :param node: Node in topology.
+        :param ip_address: Destination IP address for the ARP packet.
+        :param interface: Name of an interface to send the ARP packet from.
+        :type node: dict
+        :type ip_address: str
+        :type interface: str
+        """
         log.debug('From node {} interface {} ARP-ping IPv4 address {}'.
                   format(Topology.get_node_hostname(node),
                          interface, ip_address))
@@ -51,9 +61,6 @@ class IPv4Util(object):
         get_node(node).set_ip(interface, address, int(prefix_length))
 
     @staticmethod
-    @keyword('Node "${node}" routes to IPv4 network "${network}" with prefix '
-             'length "${prefix_length}" using interface "${interface}" via '
-             '"${gateway}"')
     def set_route(node, network, prefix_length, interface, gateway):
         """See IPv4Node.set_route for more information.
 
@@ -91,8 +98,8 @@ class IPv4Util(object):
         :rtype: int
         """
         for net in nodes_addr.values():
-            for p in net['ports'].values():
-                if p['node'] == node['host'] and p['if'] == port:
+            for net_port in net['ports'].values():
+                if net_port['node'] == node['host'] and net_port['if'] == port:
                     return net['prefix']
 
         raise Exception('Subnet not found for node {n} port {p}'.
@@ -114,8 +121,8 @@ class IPv4Util(object):
         :rtype: str
         """
         for net in nodes_addr.values():
-            for p in net['ports'].values():
-                if p['node'] == node['host'] and p['if'] == port:
+            for net_port in net['ports'].values():
+                if net_port['node'] == node['host'] and net_port['if'] == port:
                     return net['net_addr']
 
         raise Exception('Subnet not found for node {n} port {p}'.
@@ -163,3 +170,60 @@ class IPv4Util(object):
         if net is None:
             raise ValueError('Link "{0}" not found'.format(link))
         return net.get('prefix')
+
+    @staticmethod
+    def send_ping_from_node_to_dst(node, destination, namespace=None,
+                                   ping_count=3, interface=None):
+        """Send a ping from node to destination. Optionally, you can define a
+        namespace and interface from where to send a ping.
+
+        :param node: Node to start ping on.
+        :param destination: IPv4 address where to send ping.
+        :param namespace: Namespace to send ping from. Optional
+        :param ping_count: Number of pings to send. Default 3
+        :param interface: Interface from where to send ping. Optional
+        :type node: dict
+        :type destination: str
+        :type namespace: str
+        :type ping_count: int
+        :type interface: str
+        :raises RuntimeError: If no response for ping, raise error
+        """
+        cmd = ''
+        if namespace is not None:
+            cmd = 'ip netns exec {0} ping -c{1} {2}'.format(
+                namespace, ping_count, destination)
+        elif interface is not None:
+            cmd = 'ping -I {0} -c{1} {2}'.format(
+                interface, ping_count, destination)
+        else:
+            cmd = 'ping -c{0} {1}'.format(ping_count, destination)
+        ret_code, _, _ = exec_cmd(node, cmd, sudo=True)
+        if ret_code != 0:
+            raise RuntimeError("Ping Not Successful")
+
+    @staticmethod
+    def set_linux_interface_arp(node, interface, ip_addr, mac, namespace=None):
+        """Set arp on interface in linux.
+
+        :param node: Node where to execute command.
+        :param interface: Interface in namespace.
+        :param ip_addr: IP address for ARP entry.
+        :param mac: MAC address.
+        :param namespace: Execute command in namespace. Optional
+        :type node: dict
+        :type interface: str
+        :type ip_addr: str
+        :type mac: str
+        :type namespace: str
+        :raises RuntimeError: Could not set ARP properly.
+        """
+        if namespace is not None:
+            cmd = 'ip netns exec {} arp -i {} -s {} {}'.format(
+                namespace, interface, ip_addr, mac)
+        else:
+            cmd = 'arp -i {} -s {} {}'.format(interface, ip_addr, mac)
+        ret_code, _, stderr = exec_cmd(node, cmd, sudo=True)
+        if ret_code != 0:
+            raise RuntimeError("Arp set not successful, reason:{}".
+                               format(stderr))