枚举系统中所有USB HUB设备

 

1. // HUB.cpp : 定义控制台应用程序的入口点。
2. //
3. 
   #include <stdio.h>
   #include <stdlib.h>
   #include <tchar.h>
   #include <windows.h>
4. #include <dbt.h>
   #include <initguid.h>
   #include <winioctl.h>
   #include <setupapi.h>
5. extern "C" {
   #include "cfgmgr32.h"
   #include "hidsdi.h"
   }
6. #pragma   comment(lib,"hid.lib")    
   #pragma   comment(lib,"cfgmgr32.lib")  
9. #define USB_DEVICE_DESCRIPTOR_TYPE                0x01
10. #define USB_CONFIGURATION_DESCRIPTOR_TYPE         0x02
11. #define USB_STRING_DESCRIPTOR_TYPE                0x03
12. #define USB_INTERFACE_DESCRIPTOR_TYPE             0x04
13. #define USB_ENDPOINT_DESCRIPTOR_TYPE              0x05
15. #define USB_GET_HUB_NAME            8
16. #define USB_GET_NODE_INFORMATION                    258
17. #define FILE_DEVICE_USB         FILE_DEVICE_UNKNOWN
19. #define FILE_DEVICE_USB_SCAN    0x8000
20. #define IOCTL_INDEX             0x0800
21. #define IOCTL_GET_USB_DESCRIPTOR        CTL_CODE(FILE_DEVICE_USB_SCAN,IOCTL_INDEX+8, METHOD_BUFFERED,FILE_ANY_ACCESS)
23. DEFINE_GUID(GUID_DEVINTERFACE_USB_HUB,    0xf18a0e88, 0xc30c, 0x11d0, 0x88, 0x15, 0x00, 0xa0, 0xc9, 0x06, 0xbe, 0xd8);
25. #define IOCTL_INTERNAL_USB_GET_HUB_NAME  CTL_CODE(FILE_DEVICE_USB,  /
26.                                                 USB_GET_HUB_NAME,  /
27.                                                 METHOD_BUFFERED,  /
28.                                                 FILE_ANY_ACCESS)
30. #define IOCTL_USB_GET_NODE_INFORMATION   CTL_CODE(FILE_DEVICE_USB,  /
31.                                                USB_GET_NODE_INFORMATION,  /
32.                                                METHOD_BUFFERED,  /
33.                                                FILE_ANY_ACCESS)
34. typedef struct _USBSCAN_GET_DESCRIPTOR {
35.   IN UCHAR  DescriptorType;
36.   IN UCHAR  Index;
37.   IN USHORT  LanguageId;
38. } USBSCAN_GET_DESCRIPTOR, *PUSBSCAN_GET_DESCRIPTOR;
40. typedef struct _USB_DEVICE_DESCRIPTOR { 
41.   UCHAR  bLength ;
42.   UCHAR  bDescriptorType ;
43.   USHORT  bcdUSB ;
44.   UCHAR  bDeviceClass ;
45.   UCHAR  bDeviceSubClass ;
46.   UCHAR  bDeviceProtocol ;
47.   UCHAR  bMaxPacketSize0 ;
48.   USHORT  idVendor ;
49.   USHORT  idProduct ;
50.   USHORT  bcdDevice ;
51.   UCHAR  iManufacturer ;
52.   UCHAR  iProduct ;
53.   UCHAR  iSerialNumber ;
54.   UCHAR  bNumConfigurations ;
55. } USB_DEVICE_DESCRIPTOR, *PUSB_DEVICE_DESCRIPTOR ;
57. typedef enum _USB_HUB_NODE {
58.     UsbHub,
59.     UsbMIParent
60. } USB_HUB_NODE;
62. typedef struct _USB_HUB_DESCRIPTOR {  UCHAR  bDescriptorLength;  UCHAR  bDescriptorType;  UCHAR  bNumberOfPorts;  USHORT  wHubCharacteristics;  UCHAR  bPowerOnToPowerGood;  UCHAR  bHubControlCurrent;  UCHAR  bRemoveAndPowerMask[64];       } USB_HUB_DESCRIPTOR, *PUSB_HUB_DESCRIPTOR;
64. typedef struct _USB_HUB_INFORMATION {
65.     /*
66.        copy of data from hub descriptor
67.     */
68.     USB_HUB_DESCRIPTOR HubDescriptor;
70.     BOOLEAN HubIsBusPowered;
72. } USB_HUB_INFORMATION, *PUSB_HUB_INFORMATION;
74. typedef struct _USB_MI_PARENT_INFORMATION {
75.     ULONG NumberOfInterfaces;
76. } USB_MI_PARENT_INFORMATION, *PUSB_MI_PARENT_INFORMATION;
78. typedef struct _USB_NODE_INFORMATION {
79.     USB_HUB_NODE NodeType;        /* hub, mi parent */
80.     union {
81.         USB_HUB_INFORMATION HubInformation;
82.         USB_MI_PARENT_INFORMATION MiParentInformation;
83.     } u;
84. } USB_NODE_INFORMATION, *PUSB_NODE_INFORMATION;
86. int _tmain(int argc, _TCHAR* argv[])
87. {
88.     HANDLE                              hHub;
89.     HDEVINFO                            hDeviceInfo;
90.     SP_DEVICE_INTERFACE_DATA            _sdid;
91.     PSP_DEVICE_INTERFACE_DETAIL_DATA    _psdidd;
93.     DWORD dwLength;
94.     DWORD dwBytesX;
96.     __try{
97.         hDeviceInfo = SetupDiGetClassDevs(
98.             &GUID_DEVINTERFACE_USB_HUB,
99.             NULL,
100.             NULL,
101.             DIGCF_PRESENT | DIGCF_DEVICEINTERFACE
102.         );
104.         if( hDeviceInfo == INVALID_HANDLE_VALUE ){
105.             printf("error:hDevInfo == INVALID_HANDLE_VALUE"); 
106.             return -1;
107.         }
109.         _sdid.cbSize    = sizeof(SP_DEVICE_INTERFACE_DATA);
111.         for( int i = 0; SetupDiEnumDeviceInterfaces( hDeviceInfo, NULL, &GUID_DEVINTERFACE_USB_HUB, i, &_sdid ); i++ ){
112.             SetupDiGetDeviceInterfaceDetail(
113.                 hDeviceInfo,
114.                 &_sdid,
115.                 NULL,
116.                 0,
117.                 &dwBytesX,
118.                 NULL
119.             );
121.             dwLength        = dwBytesX;
122.             _psdidd         = (PSP_DEVICE_INTERFACE_DETAIL_DATA)GlobalAlloc( GPTR, dwLength );
123.             if( _psdidd == NULL ){
124.                 continue;
125.             }
126.             
127.             _psdidd->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
129.             if( !SetupDiGetDeviceInterfaceDetail(
130.                 hDeviceInfo,
131.                 &_sdid,
132.                 _psdidd,
133.                 dwLength,
134.                 &dwBytesX,
135.                 NULL ) )
136.             {
137.                 GlobalFree( _psdidd );
138.                 continue;
139.             }
141.             hHub    = CreateFile(
142.                 _psdidd->DevicePath,
143.                 GENERIC_READ,
144.                 FILE_SHARE_READ | FILE_SHARE_WRITE,
145.                 NULL,
146.                 OPEN_EXISTING,
147.                 0,
148.                 NULL
149.             );
151.             GlobalFree( _psdidd );
153.             if( hDeviceInfo == INVALID_HANDLE_VALUE ){
154.                 continue;
155.             }
157.             USBSCAN_GET_DESCRIPTOR  _usb_nif;
158.             memset( &_usb_nif, 0, sizeof(USBSCAN_GET_DESCRIPTOR) );
159.             _usb_nif.DescriptorType = USB_DEVICE_DESCRIPTOR_TYPE;
160.             _usb_nif.Index          = 1;
161.             _usb_nif.LanguageId     = 0;
163.             USB_DEVICE_DESCRIPTOR   _usbdd;
164.             memset( &_usbdd, 0, sizeof(USB_DEVICE_DESCRIPTOR) );
165.             dwLength    = sizeof(USB_DEVICE_DESCRIPTOR);
167.             if( !DeviceIoControl(
168.                 hHub,
169.                 IOCTL_GET_USB_DESCRIPTOR,
170.                 &_usb_nif,
171.                 sizeof(USBSCAN_GET_DESCRIPTOR),
172.                 &_usbdd,
173.                 dwLength,
174.                 &dwBytesX,
175.                 NULL ) )
176.             {   
177.                 CloseHandle( hHub );
178.                 printf( "%d/n", GetLastError() );
179.                 continue;
180.             }
181.             
182.             wprintf( _T("%d,%d/n"), _usbdd.idVendor, _usbdd.iProduct );
184.             CloseHandle( hHub );
185.         }
187.     }
188.     __finally{
189.         if( hDeviceInfo != INVALID_HANDLE_VALUE ){
190.             SetupDiDestroyDeviceInfoList( hDeviceInfo );
191.         }
192.     }
194.     system( "pause" );
195.     return 0;
196. }