[PATCH 6/8] KVM test: load/unload kvm module

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

From: Suqin Huang <shuang@redhat.com>

check whether kvm modules exist, unload them if exist, else load them.

Signed-off-by: Suqin Huang <shuang@redhat.com>
---
 client/tests/kvm/tests/module_probe.py |   53 ++++++++++++++++++++++++++++++++
 client/tests/kvm/tests_base.cfg.sample |    6 +++
 2 files changed, 59 insertions(+), 0 deletions(-)
 create mode 100644 client/tests/kvm/tests/module_probe.py

diff --git a/client/tests/kvm/tests/module_probe.py b/client/tests/kvm/tests/module_probe.py
new file mode 100644
index 0000000..519a916
--- /dev/null
+++ b/client/tests/kvm/tests/module_probe.py
@@ -0,0 +1,53 @@
+import re, commands, logging, os
+from autotest_lib.client.common_lib import error
+import kvm_subprocess, kvm_test_utils, kvm_utils
+
+def run_module_probe(test, params, env):
+    """
+    load/unload kvm modules several times.
+
+    Packet Loss Test:
+    1) check host cpu module
+    2) get module info
+    3) unload modules if they exist, else load them
+
+    @param test: Kvm test object
+    @param params: Dictionary with the test parameters
+    @param env: Dictionary with test environment.
+    """
+
+    def module_probe(name_list, arg=""):
+        for name in name_list:
+            cmd = "modprobe %s %s" % (arg, name)
+            logging.debug(cmd)
+            s, o = commands.getstatusoutput(cmd)
+            if s != 0:
+                logging.error("Failed to load/unload modules %s" % o)
+                return False
+        return True
+
+    #Check host cpu module
+    flags = file("/proc/cpuinfo").read()
+    arch_check = re.findall("%s\s" % "vmx", flags)
+    if arch_check:
+        arch = "kvm_intel"
+    else:
+        arch = "kvm_amd"
+
+    #Check whether ksm module exist
+    if os.path.exists("/sys/module/ksm"):
+        mod_list = ["ksm", arch, "kvm"]
+    else:
+        mod_list = [arch, "kvm"]
+
+    logging.debug(mod_list)
+    load_count = int(params.get("load_count", 100))
+
+    try:
+        for i in range(load_count):
+                if not module_probe(mod_list):
+                    raise error.TestFail("Failed to load module %s" % mod_list)
+                if not module_probe(mod_list, "-r"):
+                    raise error.TestFail("Failed to remove module %s" % mod_list)
+    finally:
+        module_probe(mod_list)
diff --git a/client/tests/kvm/tests_base.cfg.sample b/client/tests/kvm/tests_base.cfg.sample
index 1a260b4..8b2d752 100644
--- a/client/tests/kvm/tests_base.cfg.sample
+++ b/client/tests/kvm/tests_base.cfg.sample
@@ -748,6 +748,12 @@ variants:
             - vmexit:
                 case = vmexit
 
+    - module_probe:
+        type = module_probe
+        # You can specify your own module list, though it is not needed usually.
+        # mod_list = kvm
+        load_count = 100
+
     - ioquit:
         type = ioquit
         background_cmd = "for i in 1 2 3 4; do (nohup dd if=/dev/urandom of=/tmp/file bs=102400 count=10000000 &) done"
-- 
1.7.2.3