client/tests/kvm/tests/netperf.py

   1 import logging, os, signal
   2 from autotest_lib.client.common_lib import error
   3 from autotest_lib.client.bin import utils
   4 from autotest_lib.client.virt import aexpect, virt_utils
   5
   6 def run_netperf(test, params, env):
   7     """
   8     Network stress test with netperf.
   9
  10     1) Boot up a VM with multiple nics.
  11     2) Launch netserver on guest.
  12     3) Execute multiple netperf clients on host in parallel
  13        with different protocols.
  14     4) Output the test result.
  15
  16     @param test: KVM test object.
  17     @param params: Dictionary with the test parameters.
  18     @param env: Dictionary with test environment.
  19     """
  20     vm = env.get_vm(params["main_vm"])
  21     vm.verify_alive()
  22     login_timeout = int(params.get("login_timeout", 360))
  23     session = vm.wait_for_login(timeout=login_timeout)
  24     session.close()
  25     session_serial = vm.wait_for_serial_login(timeout=login_timeout)
  26
  27     netperf_dir = os.path.join(os.environ['AUTODIR'], "tests/netperf2")
  28     setup_cmd = params.get("setup_cmd")
  29
  30     firewall_flush = "iptables -F"
  31     session_serial.cmd_output(firewall_flush)
  32     try:
  33         utils.run("iptables -F")
  34     except:
  35         pass
  36
  37     for i in params.get("netperf_files").split():
  38         vm.copy_files_to(os.path.join(netperf_dir, i), "/tmp")
  39
  40     try:
  41         session_serial.cmd(firewall_flush)
  42     except aexpect.ShellError:
  43         logging.warning("Could not flush firewall rules on guest")
  44
  45     session_serial.cmd(setup_cmd % "/tmp", timeout=200)
  46     session_serial.cmd(params.get("netserver_cmd") % "/tmp")
  47
  48     tcpdump = env.get("tcpdump")
  49     pid = None
  50     if tcpdump:
  51         # Stop the background tcpdump process
  52         try:
  53             pid = int(utils.system_output("pidof tcpdump"))
  54             logging.debug("Stopping the background tcpdump")
  55             os.kill(pid, signal.SIGSTOP)
  56         except:
  57             pass
  58
  59     def netperf(i=0):
  60         guest_ip = vm.get_address(i)
  61         logging.info("Netperf_%s: netserver %s" % (i, guest_ip))
  62         result_file = os.path.join(test.resultsdir, "output_%s_%s"
  63                                    % (test.iteration, i ))
  64         list_fail = []
  65         result = open(result_file, "w")
  66         result.write("Netperf test results\n")
  67
  68         for p in params.get("protocols").split():
  69             packet_size = params.get("packet_size", "1500")
  70             for size in packet_size.split():
  71                 cmd = params.get("netperf_cmd") % (netperf_dir, p,
  72                                                    guest_ip, size)
  73                 logging.info("Netperf_%s: protocol %s" % (i, p))
  74                 try:
  75                     netperf_output = utils.system_output(cmd,
  76                                                          retain_output=True)
  77                     result.write("%s\n" % netperf_output)
  78                 except:
  79                     logging.error("Test of protocol %s failed", p)
  80                     list_fail.append(p)
  81
  82         result.close()
  83         if list_fail:
  84             raise error.TestFail("Some netperf tests failed: %s" %
  85                                  ", ".join(list_fail))
  86
  87     try:
  88         logging.info("Setup and run netperf clients on host")
  89         utils.run(setup_cmd % netperf_dir)
  90
  91         bg = []
  92         nic_num = len(params.get("nics").split())
  93         for i in range(nic_num):
  94             bg.append(virt_utils.Thread(netperf, (i,)))
  95             bg[i].start()
  96
  97         completed = False
  98         while not completed:
  99             completed = True
 100             for b in bg:
 101                 if b.is_alive():
 102                     completed = False
 103     finally:
 104         for b in bg:
 105             if b:
 106                 b.join()
 107         session_serial.cmd_output("killall netserver")
 108         if tcpdump and pid:
 109             logging.debug("Resuming the background tcpdump")
 110             logging.info("pid is %s" % pid)
 111             os.kill(pid, signal.SIGCONT)