From 6a53c10eab380ec1bd50a26b124c17b8eab4c448 Mon Sep 17 00:00:00 2001
From: Wei Mao <weimao1@gmail.com>
Date: Mon, 20 Mar 2023 16:43:40 -0700
Subject: [PATCH] Enumerate HostController with multiple DeviceInterfaces

---
 usb/usbview/enum.c | 119 ++++++++++++++++++++++-----------------------
 1 file changed, 58 insertions(+), 61 deletions(-)

diff --git a/usb/usbview/enum.c b/usb/usbview/enum.c
index fd427f688..297e0d320 100644
--- a/usb/usbview/enum.c
+++ b/usb/usbview/enum.c
@@ -296,77 +296,74 @@ EnumerateHostControllers (
     {
         deviceInterfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA);
 
-        success = SetupDiEnumDeviceInterfaces(deviceInfo,
-                                              0,
-                                              (LPGUID)&GUID_CLASS_USB_HOST_CONTROLLER,
-                                              index,
-                                              &deviceInterfaceData);
-
-        if (!success)
+        for (int devInterfaceIndex = 0;
+            SetupDiEnumDeviceInterfaces(deviceInfo,
+                                        &deviceInfoData,
+                                        (LPGUID)&GUID_CLASS_USB_HOST_CONTROLLER,
+                                        devInterfaceIndex,
+                                        &deviceInterfaceData);
+            devInterfaceIndex++)
         {
-            OOPS();
-            break;
-        }
-
-        success = SetupDiGetDeviceInterfaceDetail(deviceInfo,
-                                                  &deviceInterfaceData,
-                                                  NULL,
-                                                  0,
-                                                  &requiredLength,
-                                                  NULL);
+            success = SetupDiGetDeviceInterfaceDetail(deviceInfo,
+                                                      &deviceInterfaceData,
+                                                      NULL,
+                                                      0,
+                                                      &requiredLength,
+                                                      NULL);
 
-        if (!success && GetLastError() != ERROR_INSUFFICIENT_BUFFER)
-        {
-            OOPS();
-            break;
-        }
+            if (!success && GetLastError() != ERROR_INSUFFICIENT_BUFFER)
+            {
+                OOPS();
+                break;
+            }
 
-        deviceDetailData = ALLOC(requiredLength);
-        if (deviceDetailData == NULL)
-        {
-            OOPS();
-            break;
-        }
+            deviceDetailData = ALLOC(requiredLength);
+            if (deviceDetailData == NULL)
+            {
+                OOPS();
+                break;
+            }
 
-        deviceDetailData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
+            deviceDetailData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
 
-        success = SetupDiGetDeviceInterfaceDetail(deviceInfo,
-                                                  &deviceInterfaceData,
-                                                  deviceDetailData,
-                                                  requiredLength,
-                                                  &requiredLength,
-                                                  NULL);
+            success = SetupDiGetDeviceInterfaceDetail(deviceInfo,
+                                                      &deviceInterfaceData,
+                                                      deviceDetailData,
+                                                      requiredLength,
+                                                      &requiredLength,
+                                                      NULL);
 
-        if (!success)
-        {
-            OOPS();
-            break;
-        }
+            if (!success)
+            {
+                OOPS();
+                break;
+            }
 
-        hHCDev = CreateFile(deviceDetailData->DevicePath,
-                            GENERIC_WRITE,
-                            FILE_SHARE_WRITE,
-                            NULL,
-                            OPEN_EXISTING,
-                            0,
-                            NULL);
+            hHCDev = CreateFile(deviceDetailData->DevicePath,
+                                GENERIC_WRITE,
+                                FILE_SHARE_WRITE,
+                                NULL,
+                                OPEN_EXISTING,
+                                0,
+                                NULL);
+
+            // If the handle is valid, then we've successfully opened a Host
+            // Controller.  Display some info about the Host Controller itself,
+            // then enumerate the Root Hub attached to the Host Controller.
+            //
+            if (hHCDev != INVALID_HANDLE_VALUE)
+            {
+                EnumerateHostController(hTreeParent,
+                                        hHCDev,
+                                        deviceDetailData->DevicePath,
+                                        deviceInfo,
+                                        &deviceInfoData);
 
-        // If the handle is valid, then we've successfully opened a Host
-        // Controller.  Display some info about the Host Controller itself,
-        // then enumerate the Root Hub attached to the Host Controller.
-        //
-        if (hHCDev != INVALID_HANDLE_VALUE)
-        {
-            EnumerateHostController(hTreeParent,
-                                    hHCDev,
-                                    deviceDetailData->DevicePath,
-                                    deviceInfo,
-                                    &deviceInfoData);
+                CloseHandle(hHCDev);
+            }
 
-            CloseHandle(hHCDev);
+            FREE(deviceDetailData);
         }
-
-        FREE(deviceDetailData);
     }
 
     SetupDiDestroyDeviceInfoList(deviceInfo);