diff mbox

[RFC,03/14] KVM Test: Add a common ping module for network related tests

Message ID 20100720013515.2212.66387.stgit@z (mailing list archive)
State New, archived
Headers show

Commit Message

Amos Kong July 20, 2010, 1:35 a.m. UTC
None
diff mbox

Patch

diff --git a/client/tests/kvm/kvm_net_utils.py b/client/tests/kvm/kvm_net_utils.py
index ede4965..8a71858 100644
--- a/client/tests/kvm/kvm_net_utils.py
+++ b/client/tests/kvm/kvm_net_utils.py
@@ -1,4 +1,114 @@ 
-import re
+import logging, re, signal
+from autotest_lib.client.common_lib import error
+import kvm_subprocess, kvm_utils
+
+def get_loss_ratio(output):
+    """
+    Get the packet loss ratio from the output of ping
+
+    @param output
+    """
+    try:
+        return int(re.findall('(\d+)% packet loss', output)[0])
+    except IndexError:
+        logging.debug(output)
+        return -1
+
+def raw_ping(command, timeout, session, output_func):
+    """
+    Low-level ping command execution.
+
+    @param command: ping command
+    @param timeout: timeout of the ping command
+    @param session: local executon hint or session to execute the ping command
+    """
+    if session == "localhost":
+        process = kvm_subprocess.run_bg(command, output_func=output_func,
+                                        timeout=timeout)
+
+        # Send SIGINT singal to notify the timeout of running ping process,
+        # Because ping have the ability to catch the SIGINT signal so we can
+        # always get the packet loss ratio even if timeout.
+        if process.is_alive():
+            kvm_utils.kill_process_tree(process.get_pid(), signal.SIGINT)
+
+        status = process.get_status()
+        output = process.get_output()
+
+        process.close()
+        return status, output
+    else:
+        session.sendline(command)
+        status, output = session.read_up_to_prompt(timeout=timeout,
+                                                   print_func=output_func)
+        if status is False:
+            # Send ctrl+c (SIGINT) through ssh session
+            session.sendline("\003")
+            status, output2 = session.read_up_to_prompt(print_func=output_func)
+            output += output2
+            if status is False:
+                # We also need to use this session to query the return value
+                session.sendline("\003")
+
+        session.sendline(session.status_test_command)
+        s2, o2 = session.read_up_to_prompt()
+        if s2 is False:
+            status = -1
+        else:
+            try:
+                status = int(re.findall("\d+", o2)[0])
+            except:
+                status = -1
+
+        return status, output
+
+def ping(dest = "localhost", count = None, interval = None, interface = None,
+         packetsize = None, ttl = None, hint = None, adaptive = False,
+         broadcast = False, flood = False, timeout = 0,
+         output_func = logging.debug, session = "localhost"):
+    """
+    Wrapper of ping.
+
+    @param dest: destination address
+    @param count: count of icmp packet
+    @param interval: interval of two icmp echo request
+    @param interface: specified interface of the source address
+    @param packetsize: packet size of icmp
+    @param ttl: ip time to live
+    @param hint: path mtu discovery hint
+    @param adaptive: adaptive ping flag
+    @param broadcast: broadcast ping flag
+    @param flood: flood ping flag
+    @param timeout: timeout for the ping command
+    @param output_func: function used to log the result of ping
+    @param session: local executon hint or session to execute the ping command
+    """
+
+    command = "ping %s " % dest
+
+    if count is not None:
+        command += " -c %s" % count
+    if interval is not None:
+        command += " -i %s" % interval
+    if interface is not None:
+        command += " -I %s" % interface
+    if packetsize is not None:
+        command += " -s %s" % packetsize
+    if ttl is not None:
+        command += " -t %s" % ttl
+    if hint is not None:
+        command += " -M %s" % hint
+    if adaptive is True:
+        command += " -A"
+    if broadcast is True:
+        command += " -b"
+    if flood is True:
+        # temporary workaround as the kvm_subprocess may not properly handle
+        # the timeout for the output of flood ping
+        command += " -f -q"
+        output_func = None
+
+    return raw_ping(command, timeout, session, output_func)
 
 def get_linux_ifname(session, mac_address):
     """