USB View工具源码

/*++ Copyright (c) 1997-1998 Microsoft Corporation Module Name: ENUM.C Abstract: This source file contains the routines which enumerate the USB bus and populate the TreeView control. The enumeration process goes like this: (1) Enumerate Host Controllers and Root Hubs Host controllers currently have symbolic link names of the form HCDx, where x starts at 0. Use CreateFile() to open each host controller symbolic link. Create a node in the TreeView to represent each host controller. After a host controller has been opened, send the host controller an IOCTL_USB_GET_ROOT_HUB_NAME request to get the symbolic link name of the root hub that is part of the host controller. (2) Enumerate Hubs (Root Hubs and External Hubs) Given the name of a hub, use CreateFile() to hub the hub. Send the hub an IOCTL_USB_GET_NODE_INFORMATION request to get info about the hub, such as the number of downstream ports. Create a node in the TreeView to represent each hub. (3) Enumerate Downstream Ports Given an handle to an open hub and the number of downstream ports on the hub, send the hub an IOCTL_USB_GET_NODE_CONNECTION_INFORMATION request for each downstream port of the hub to get info about the device (if any) attached to each port. If there is a device attached to a port, send the hub an IOCTL_USB_GET_NODE_CONNECTION_NAME request to get the symbolic link name of the hub attached to the downstream port. If there is a hub attached to the downstream port, recurse to step (2). Create a node in the TreeView to represent each hub port and attached device. Environment: user mode Revision History: 04-25-97 : created --*/ //***************************************************************************** // I N C L U D E S //***************************************************************************** #include <windows.h> #include <basetyps.h> #include <winioctl.h> #include <string.h> #include "usbview.h" //***************************************************************************** // D E F I N E S //***************************************************************************** #define NUM_HCS_TO_CHECK 10 //***************************************************************************** // L O C A L F U N C T I O N P R O T O T Y P E S //***************************************************************************** VOID EnumerateHub ( HTREEITEM hTreeParent, PCHAR HubName, PUSB_NODE_CONNECTION_INFORMATION ConnectionInfo, PUSB_DESCRIPTOR_REQUEST ConfigDesc, PSTRING_DESCRIPTOR_NODE StringDescs, PCHAR DeviceDesc ); VOID EnumerateHubPorts ( HTREEITEM hTreeParent, HANDLE hHubDevice, ULONG NumPorts ); PCHAR GetRootHubName ( HANDLE HostController ); PCHAR GetExternalHubName ( HANDLE Hub, ULONG ConnectionIndex ); PCHAR GetHCDDriverKeyName ( HANDLE HCD ); PCHAR GetDriverKeyName ( HANDLE Hub, ULONG ConnectionIndex ); PUSB_DESCRIPTOR_REQUEST GetConfigDescriptor ( HANDLE hHubDevice, ULONG ConnectionIndex, UCHAR DescriptorIndex ); BOOL AreThereStringDescriptors ( PUSB_DEVICE_DESCRIPTOR DeviceDesc, PUSB_CONFIGURATION_DESCRIPTOR ConfigDesc ); PSTRING_DESCRIPTOR_NODE GetAllStringDescriptors ( HANDLE hHubDevice, ULONG ConnectionIndex, PUSB_DEVICE_DESCRIPTOR DeviceDesc, PUSB_CONFIGURATION_DESCRIPTOR ConfigDesc ); PSTRING_DESCRIPTOR_NODE GetStringDescriptor ( HANDLE hHubDevice, ULONG ConnectionIndex, UCHAR DescriptorIndex, USHORT LanguageID ); PSTRING_DESCRIPTOR_NODE GetStringDescriptors ( HANDLE hHubDevice, ULONG ConnectionIndex, UCHAR DescriptorIndex, ULONG NumLanguageIDs, USHORT *LanguageIDs, PSTRING_DESCRIPTOR_NODE StringDescNodeTail ); //***************************************************************************** // G L O B A L S P R I V A T E T O T H I S F I L E //***************************************************************************** PCHAR ConnectionStatuses[] = { "NoDeviceConnected", "DeviceConnected", "DeviceFailedEnumeration", "DeviceGeneralFailure", "DeviceCausedOvercurrent", "DeviceNotEnoughPower" }; ULONG TotalDevicesConnected; //***************************************************************************** // // EnumerateHostControllers() // // hTreeParent - Handle of the TreeView item under which host controllers // should be added. // //***************************************************************************** VOID EnumerateHostControllers ( HTREEITEM hTreeParent, ULONG *DevicesConnected ) { char HCName[16]; int HCNum; HANDLE hHCDev; HTREEITEM hHCItem; PCHAR rootHubName; PCHAR leafName; TotalDevicesConnected = 0; TotalHubs = 0; // Iterate over some Host Controller names and try to open them. // for (HCNum = 0; HCNum < NUM_HCS_TO_CHECK; HCNum++) { wsprintf(HCName, "\\\\.\\HCD%d", HCNum); hHCDev = CreateFile(HCName, 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) { PCHAR driverKeyName, deviceDesc; driverKeyName = GetHCDDriverKeyName(hHCDev); leafName = HCName + sizeof("\\\\.\\") - sizeof(""); if (driverKeyName) { deviceDesc = DriverNameToDeviceDesc(driverKeyName); if (deviceDesc) { leafName = deviceDesc; } FREE(driverKeyName); } hHCItem = AddLeaf(hTreeParent, 0, leafName ); if (hHCItem) { rootHubName = GetRootHubName(hHCDev); if (rootHubName != NULL) { EnumerateHub(hHCItem, rootHubName, NULL, // ConnectionInfo NULL, // ConfigDesc NULL, // StringDescs "RootHub" // DeviceDesc ); } } CloseHandle(hHCDev); } } *DevicesConnected = TotalDevicesConnected; } //***************************************************************************** // // EnumerateHub() // // hTreeParent - Handle of the TreeView item under which this hub should be // added. // // HubName - Name of this hub. This pointer is kept so the caller can neither // free nor reuse this memory. // // ConnectionInfo - NULL if this is a root hub, else this is the connection // info for an external hub. This pointer is kept so the caller can neither // free nor reuse this memory. // // ConfigDesc - NULL if this is a root hub, else this is the Configuration // Descriptor for an external hub. This pointer is kept so th