utils_net/utils_misc: add PowerShell support for windows nic attribute

Signed-off-by: Wenli Quan <wquan@redhat.com>

Author: quanwenli

virttest/utils_misc.py

@@ -2336,6 +2336,19 @@ def __get_unit_index(M):
         return ("%.20f" % data_size).rstrip("0")
 
 
+def is_wmic_available(session):
+    """
+    Check if 'wmic' is available on the Windows guest.
+    :param session: Windows guest session
+    :return: True if wmic can be executed, False otherwise
+    """
+    try:
+        status, _ = session.cmd_status_output("wmic /?", timeout=10)
+        return status == 0
+    except Exception:
+        return False
+
+
 def get_free_disk(session, mount):
     """
     Get FreeSpace for given mount point.
@@ -2474,7 +2487,14 @@ def get_win_disk_vol(session, condition="VolumeName='WIN_UTILS'"):
 
     :return: volume ID.
     """
-    cmd = "wmic logicaldisk where (%s) get DeviceID" % condition
+    if is_wmic_available(session):
+        cmd = "wmic logicaldisk where (%s) get DeviceID" % condition
+    else:
+        cmd = (
+            "PowerShell -NoProfile -Command "
+            '"(Get-CimInstance Win32_LogicalDisk | Where-Object { $_.%s }).DeviceID"'
+            % condition.replace("=", "-eq")
+        )
     output = session.cmd(cmd, timeout=120)
     device = re.search(r"(\w):", output, re.M)
     if not device:

virttest/utils_net.py

@@ -3952,7 +3952,14 @@ def get_windows_nic_attribute(
     :param target: which nic attribute you want to get.
 
     """
-    cmd = 'wmic %s where %s="%s" get %s' % (global_switch, key, value, target)
+    if utils_misc.is_wmic_available(session):
+        cmd = 'wmic %s where %s="%s" get %s' % (global_switch, key, value, target)
+    else:
+        value_ps = value.replace(":", "-").upper()
+        cmd = (
+            "PowerShell -NoProfile -Command "
+            "\"(Get-NetAdapter | Where-Object {{$_.{key} -eq '{value}'}}).Name\""
+        ).format(key=key, value=value_ps)
     status, out = session.cmd_status_output(cmd, timeout=timeout)
     if status != 0:
         err_msg = "Execute guest shell command('%s') " "failed with error: '%s'" % (
@@ -3961,8 +3968,9 @@ def get_windows_nic_attribute(
         )
         raise exceptions.TestError(err_msg)
     lines = [l.strip() for l in out.splitlines() if l.strip()]
-    # First line is header, return second line
-    return lines[1]
+    # WMIC output has a header (value is in lines[1])
+    # PowerShell output has no header, so return out.strip()
+    return lines[1] if len(lines) > 1 else out.strip()
 
 
 def set_win_guest_nic_status(session, connection_id, status, timeout=240):