usb_hotplug_unplug: add hotplug/hotunplug support for usb-storage

1.Add clean dmesg step before test in format_disk
2.Add clean partition step at the end of test in format_disk
3.Add hotplug/hotunplug support for usb-storage
Signed-off-by: Nybduan <yduan@redhat.com>

qemu/tests/cfg/usb.cfg

@@ -266,6 +266,28 @@
                 check_serial_option = no
                 check_removable_option = no
                 check_io_size_option = no
+                clean_partition_cmd = "echo select disk 1 > cmd &&"
+                clean_partition_cmd += " echo clean >> cmd &&"
+                clean_partition_cmd += " echo exit >> cmd &&"
+                clean_partition_cmd += " diskpart /s cmd"
+            variants:
+                - @default:
+                    with_hotplug_unplug = no
+                - hotplug_unplug_one_time:
+                    usb_repeat_times = 1
+                    with_hotplug_unplug = yes
+                    boot_drive_stg = no
+                    check_serial_option = no
+                    check_removable_option = no
+                    check_io_size_option = no
+                - hotplug_unplug_multi_times:
+                    no piix3-uhci, piix4-uhci, ich9-uhci
+                    usb_repeat_times = 300
+                    with_hotplug_unplug = yes
+                    boot_drive_stg = no
+                    check_serial_option = no
+                    check_removable_option = no
+                    check_io_size_option = no
         - usb_host:
             # must configure which device should be used
             #usb_host_device = "<vendorid>:<productid>"

qemu/tests/format_disk.py

@@ -33,6 +33,9 @@ def run(test, params, env):
     cmd_timeout = int(params.get("cmd_timeout", 360))
     os_type = params["os_type"]
 
+    if os_type == 'linux':
+        session.cmd("dmesg -C")
+
     drive_path = ""
     if os_type == 'linux':
         drive_name = params.objects("images")[-1]
@@ -151,6 +154,14 @@ def run(test, params, env):
             logging.debug("The mounted devices are: %s" % mount_list)
             test.fail("Failed to umount with error: %s" % output)
 
+    # Clean partition on disk
+    clean_partition_cmd = params.get("clean_partition_cmd")
+    if clean_partition_cmd:
+        status, output = session.cmd_status_output(clean_partition_cmd,
+                                                   timeout=cmd_timeout)
+        if status != 0:
+            test.fail("Failed to clean partition with error: %s" % output)
+
     output = ""
     try:
         output = session.cmd("dmesg -c")

qemu/tests/usb_storage.py

@@ -14,12 +14,14 @@ def run(test, params, env):
     Test usb storage devices in the guest.
 
     1) Create a image file by qemu-img
-    2) Boot up a guest add this image as a usb device
-    3) Check usb device information via monitor
-    4) Check usb information by executing guest command
-    5) Check usb serial option (optional)
-    6) Check usb removable option (optional)
-    7) Check usb min_io_size/opt_io_size option (optional)
+    2) Boot up a guest
+    3) Hotplug a usb storage (optional)
+    4) Check usb storage information via monitor
+    5) Check usb information by executing guest command
+    6) Check usb serial option (optional)
+    7) Check usb removable option (optional)
+    8) Check usb min_io_size/opt_io_size option (optional)
+    9) Hotunplug the usb storage (optional)
 
     :param test: QEMU test object
     :param params: Dictionary with the test parameters
@@ -193,6 +195,25 @@ def run(test, params, env):
     vm.verify_alive()
 
     login_timeout = int(params.get("login_timeout", 360))
+    hotplug_unplug = (params["with_hotplug_unplug"] == "yes")
+    repeat_times = int(params.get("usb_repeat_times", "1"))
+    for rt in range(1, repeat_times+1):
+        disk_hotplugged = []
+        if hotplug_unplug:
+            error_context.context("Hotplug the %s times." % rt, logging.info)
+            image_name = params.objects("images")[-1]
+            image_params = params.object_params(image_name)
+            devices = vm.devices.images_define_by_params(image_name,
+                                                         image_params,
+                                                         'disk', None,
+                                                         False, None)
+            for dev in devices:
+                ret = vm.devices.simple_hotplug(dev, vm.monitor)
+                if ret[1] is False:
+                    test.fail("Failed to hotplug device '%s'. Output:\n%s"
+                              % (dev, ret[0]))
+            disk_hotplugged.append(devices[-1])
+
         error_context.context("Check usb device information in monitor",
                               logging.info)
         output = str(vm.monitor.info("usb"))
@@ -213,7 +234,7 @@ def run(test, params, env):
         # this part is linux only
         if params.get("check_serial_option") == "yes":
             error_context.context("Check usb serial option", logging.info)
-        serial = str(uuid.uuid4())
+            serial = uuid.uuid4().hex
             regex_str = r'usb-storage.*?serial = "(.*?)"\s'
             _check_serial_option(serial, regex_str, serial)
 
@@ -226,7 +247,6 @@ def run(test, params, env):
 
             logging.info("Leave usb serial option blank")
             serial = "NO_EQUAL_STRING"
-        regex_str = r'usb-storage.*?serial = (.*?)\s'
             _check_serial_option(serial, regex_str, '"on"')
 
         if params.get("check_removable_option") == "yes":
@@ -242,7 +262,15 @@ def run(test, params, env):
         if params.get("check_io_size_option") == "yes":
             error_context.context("Check usb min/opt io_size option", logging.info)
             _check_io_size_option("0", "0")
-        # Guest can't recognize correct value which we set now,
+            # NOTE: Guest can't recognize correct value which we set now,
             # So comment these test temporary.
             # _check_io_size_option("1024", "1024")
             # _check_io_size_option("4096", "4096")
+
+        if hotplug_unplug:
+            error_context.context("Hotunplug the %s times." % rt, logging.info)
+            for dev in disk_hotplugged:
+                ret = vm.devices.simple_unplug(dev, vm.monitor)
+                if ret[1] is False:
+                    test.fail("Failed to unplug device '%s'. Output:\n%s"
+                              % (dev, ret[0]))