From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from [140.186.70.92] (port=51381 helo=eggs.gnu.org) by lists.gnu.org with esmtp (Exim 4.43) id 1Ob1jt-000158-IS for qemu-devel@nongnu.org; Mon, 19 Jul 2010 21:35:22 -0400 Received: from Debian-exim by eggs.gnu.org with spam-scanned (Exim 4.69) (envelope-from ) id 1Ob1jr-0003Ou-RV for qemu-devel@nongnu.org; Mon, 19 Jul 2010 21:35:21 -0400 Received: from mx1.redhat.com ([209.132.183.28]:45591) by eggs.gnu.org with esmtp (Exim 4.69) (envelope-from ) id 1Ob1jr-0003Og-KO for qemu-devel@nongnu.org; Mon, 19 Jul 2010 21:35:19 -0400 From: Amos Kong Date: Tue, 20 Jul 2010 09:35:15 +0800 Message-ID: <20100720013515.2212.66387.stgit@z> In-Reply-To: <20100720013414.2212.13476.stgit@z> References: <20100720013414.2212.13476.stgit@z> MIME-Version: 1.0 Content-Type: text/plain; charset="utf-8" Content-Transfer-Encoding: 7bit Subject: [Qemu-devel] [RFC PATCH 03/14] KVM Test: Add a common ping module for network related tests List-Id: qemu-devel.nongnu.org List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , To: lmr@redhat.com Cc: autotest@test.kernel.org, qemu-devel@nongnu.org, kvm@vger.kernel.org The kvm_net_utils.py is a just a place that wraps common network related commands which is used to do the network-related tests. Use -1 as the packet ratio for loss analysis. Use quiet mode when doing the flood ping. Signed-off-by: Jason Wang Signed-off-by: Amos Kong --- 0 files changed, 0 insertions(+), 0 deletions(-) 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): """