[PATCH 09/18] KVM test: Add a subtest of load/unload nic driver

[Lucas Meneghel Rodrigues <lmr@redhat.com>]

From: Amos Kong <akong@redhat.com>

Repeatedly load/unload nic driver, try to transfer file between guest and host
by threads at the same time, and check the md5sum.

Changes from v1:
- Use a new method to get nic driver name
- Use utils.hash_file() to get md5sum

Signed-off-by: Amos Kong <akong@redhat.com>
---
 client/tests/kvm/tests/file_transfer.py    |   19 +++--
 client/tests/kvm/tests/nicdriver_unload.py |  115 ++++++++++++++++++++++++++++
 client/tests/kvm/tests_base.cfg.sample     |   10 ++-
 3 files changed, 135 insertions(+), 9 deletions(-)
 create mode 100644 client/tests/kvm/tests/nicdriver_unload.py

diff --git a/client/tests/kvm/tests/file_transfer.py b/client/tests/kvm/tests/file_transfer.py
index c9a3476..a0c6cff 100644
--- a/client/tests/kvm/tests/file_transfer.py
+++ b/client/tests/kvm/tests/file_transfer.py
@@ -7,18 +7,19 @@ def run_file_transfer(test, params, env):
     """
     Test ethrnet device function by ethtool
 
-    1) Boot up a virtual machine
-    2) Create a large file by dd on host
-    3) Copy this file from host to guest
-    4) Copy this file from guest to host
-    5) Check if file transfers good
+    1) Boot up a VM.
+    2) Create a large file by dd on host.
+    3) Copy this file from host to guest.
+    4) Copy this file from guest to host.
+    5) Check if file transfers ended good.
 
-    @param test: Kvm test object
+    @param test: KVM test object.
     @param params: Dictionary with the test parameters.
     @param env: Dictionary with test environment.
     """
     vm = kvm_test_utils.get_living_vm(env, params.get("main_vm"))
     timeout=int(params.get("login_timeout", 360))
+
     logging.info("Trying to log into guest '%s' by serial", vm.name)
     session = kvm_utils.wait_for(lambda: vm.serial_login(),
                                  timeout, 0, step=2)
@@ -29,18 +30,19 @@ def run_file_transfer(test, params, env):
     scp_timeout = int(params.get("scp_timeout"))
     cmd = "dd if=/dev/urandom of=%s/a.out bs=1M count=%d" %  (dir, int(
                                          params.get("filesize", 4000)))
+
     try:
         logging.info("Create file by dd command on host, cmd: %s" % cmd)
         utils.run(cmd)
 
         logging.info("Transfer file from host to guest")
         if not vm.copy_files_to("%s/a.out" % dir, "/tmp/b.out",
-                                                        timeout=scp_timeout):
+                                timeout=scp_timeout):
             raise error.TestFail("Fail to transfer file from host to guest")
 
         logging.info("Transfer file from guest to host")
         if not vm.copy_files_from("/tmp/b.out", "%s/c.out" % dir,
-                                                        timeout=scp_timeout):
+                                  timeout=scp_timeout):
             raise error.TestFail("Fail to transfer file from guest to host")
 
         logging.debug(commands.getoutput("ls -l %s/[ac].out" % dir))
@@ -49,6 +51,7 @@ def run_file_transfer(test, params, env):
 
         if md5_orig != md5_new:
             raise error.TestFail("File changed after transfer")
+
     finally:
         session.get_command_status("rm -f /tmp/b.out")
         utils.run("rm -f %s/[ac].out" % dir)
diff --git a/client/tests/kvm/tests/nicdriver_unload.py b/client/tests/kvm/tests/nicdriver_unload.py
new file mode 100644
index 0000000..8630f88
--- /dev/null
+++ b/client/tests/kvm/tests/nicdriver_unload.py
@@ -0,0 +1,115 @@
+import logging, threading, os
+from autotest_lib.client.common_lib import error
+from autotest_lib.client.bin import utils
+import kvm_utils, kvm_test_utils
+
+def run_nicdriver_unload(test, params, env):
+    """
+    Test nic driver.
+
+    1) Boot a VM.
+    2) Get the NIC driver name.
+    3) Repeatedly unload/load NIC driver.
+    4) Multi-session TCP transfer on test interface.
+    5) Check whether the test interface should still work.
+
+    @param test: KVM test object.
+    @param params: Dictionary with the test parameters.
+    @param env: Dictionary with test environment.
+    """
+    timeout = int(params.get("login_timeout", 360))
+    vm = kvm_test_utils.get_living_vm(env, params.get("main_vm"))
+    session = kvm_test_utils.wait_for_login(vm, timeout=timeout)
+    logging.info("Trying to log into guest '%s' by serial", vm.name)
+    session2 = kvm_utils.wait_for(lambda: vm.serial_login(),
+                                  timeout, 0, step=2)
+    if not session2:
+        raise error.TestFail("Could not log into guest '%s'" % vm.name)
+
+    ethname = kvm_test_utils.get_linux_ifname(session, vm.get_macaddr(0))
+    temp_path = "/sys/class/net/%s/device/driver" % (ethname)
+    if os.path.islink(temp_path):
+        driver = os.path.split(os.path.realpath(temp_path))[-1]
+    else:
+        raise error.TestError("Could not find driver name")
+    logging.info("driver is %s", driver)
+
+    class ThreadScp(threading.Thread):
+        def run(self):
+            remote_file = '/tmp/' + self.getName()
+            file_list.append(remote_file)
+            ret = vm.copy_files_to(file_name, remote_file, timeout=scp_timeout)
+            if ret:
+                logging.debug("File %s was transfered successfuly", remote_file)
+            else:
+                logging.debug("Failed to transfer file %s", remote_file)
+
+    def compare(origin_file, receive_file):
+        cmd = "md5sum %s"
+        check_sum1 = utils.hash_file(origin_file, method="md5")
+        s, output2 = session.get_command_status_output(cmd % receive_file)
+        if s != 0:
+            logging.error("Could not get md5sum of receive_file")
+            return False
+        check_sum2 = output2.strip().split()[0]
+        logging.debug("original file md5: %s, received file md5: %s",
+                      check_sum1, check_sum2)
+        if check_sum1 != check_sum2:
+            logging.error("MD5 hash of origin and received files doesn't match")
+            return False
+        return True
+
+    #produce sized file in host
+    file_size = params.get("file_size")
+    file_name = "/tmp/nicdriver_unload_file"
+    cmd = "dd if=/dev/urandom of=%s bs=%sM count=1"
+    utils.system(cmd % (file_name, file_size))
+
+    file_list = []
+    connect_time = params.get("connect_time")
+    scp_timeout = int(params.get("scp_timeout"))
+    thread_num = int(params.get("thread_num"))
+    unload_load_cmd = ("sleep %s && ifconfig %s down && modprobe -r %s && "
+                       "sleep 1 && modprobe %s && ifconfig %s up" %
+                       (connect_time, ethname, driver, driver, ethname))
+    pid = os.fork()
+    if pid != 0:
+        logging.info("Unload/load NIC driver repeatedly in guest...")
+        while True:
+            logging.debug("Try to unload/load nic drive once")
+            if session2.get_command_status(unload_load_cmd, timeout=120) != 0:
+                session.get_command_output("rm -rf /tmp/Thread-*")
+                raise error.TestFail("Unload/load nic driver failed")
+            pid, s = os.waitpid(pid, os.WNOHANG)
+            status = os.WEXITSTATUS(s)
+            if (pid, status) != (0, 0):
+                logging.debug("Child process ending")
+                break
+    else:
+        logging.info("Multi-session TCP data transfer")
+        threads = []
+        for i in range(thread_num):
+            t = ThreadScp()
+            t.start()
+            threads.append(t)
+        for t in threads:
+            t.join(timeout = scp_timeout)
+        os._exit(0)
+
+    session2.close()
+
+    try:
+        logging.info("Check MD5 hash for received files in multi-session")
+        for f in file_list:
+            if not compare(file_name, f):
+                raise error.TestFail("Fail to compare (guest) file %s" % f)
+
+        logging.info("Test nic function after load/unload")
+        if not vm.copy_files_to(file_name, file_name):
+            raise error.TestFail("Fail to copy file from host to guest")
+        if not compare(file_name, file_name):
+            raise error.TestFail("Test nic function after load/unload fail")
+
+    finally:
+        session.get_command_output("rm -rf /tmp/Thread-*")
+        session.close()
diff --git a/client/tests/kvm/tests_base.cfg.sample b/client/tests/kvm/tests_base.cfg.sample
index 4b424da..8443fee 100644
--- a/client/tests/kvm/tests_base.cfg.sample
+++ b/client/tests/kvm/tests_base.cfg.sample
@@ -482,6 +482,14 @@ variants:
         filesize = 4000
         scp_timeout = 1000
 
+    - nicdriver_unload:  install setup unattended_install.cdrom
+        type = nicdriver_unload
+        nic_mode = tap
+        file_size = 100
+        connect_time = 4
+        scp_timeout = 300
+        thread_num = 10
+
     - physical_resources_check: install setup unattended_install.cdrom
         type = physical_resources_check
         catch_uuid_cmd = dmidecode | awk -F: '/UUID/ {print $2}'
@@ -1199,7 +1207,7 @@ variants:
 
     # Windows section
     - @Windows:
-        no autotest linux_s3 vlan_tag ioquit unattended_install.(url|nfs|remote_ks) jumbo file_transfer
+        no autotest linux_s3 vlan_tag ioquit unattended_install.(url|nfs|remote_ks) jumbo file_transfer nicdriver_unload
         shutdown_command = shutdown /s /f /t 0
         reboot_command = shutdown /r /f /t 0
         status_test_command = echo %errorlevel%
-- 
1.7.2.2