Windows上USB设备检测-JoyCode 编程小战

枚举总线

主机控制器的驱动程序(HCD(Host Control Driver))。它位于USB主机控制器与USB系统软件之间。

可以用CreateFile打开名字为"\\\\.\\HCD1",\\\\.\\HCD2的文件来检测HCD总线。

打开句柄之后可以通过DeviceIoControl 传递IOCTL_GET_HCD_DRIVERKEY_NAME 参数来得到DriveName。

并可以用CM_系列函数遍历各节点来得到相应的驱动描述。

可以通过调用IOCTL_USB_GET_ROOT_HUB_NAME 为参数的DeviceIoControl 来得到此主机驱动器上的根HUB。

通过CreateFile打开名字为"\\\\.\\HUBNAME"形式的文件，来得到根HUB的句柄，然后通过将IOCTL_USB_GET_NODE_CONNECTION_INFORMATION 传递给函数DeviceIoControl来得到此HUB的连接信息，通过这个信息可以知道此HUB上有几个端口，以及每个端口的设备连接情况。还能得到连接端口设备的VID，PID，如果有的话。

如果HUB端口已经有设备连接，可以通过DeviceIoControl传递参数IOCTL_USB_GET_NODE_CONNECTION_DRIVERKEY_NAME 来得到连接设备的情况（传递参数为Hub句柄和当前的索引值，索引值要从1开始），如果当前端口连接的设备为另外一个HUB，那么可以通过像DevioceIoControl传递参数IOCTL_USB_GET_NODE_CONNECTION_NAME 来得到此HUB的名字，然后就可以枚举得到此子HUB的信息了。

查找设备

通过PID或者VID或者ClassGUID或者InterfaceGUID，可以得到符合条件的当前设备。

typedef struct _SSDevHandles

{

DWORD dwDevsCount;

HDEVINFO hDevInfoSet;

SP_DEVINFO_DATA *pDevDatas;

} SSDevHandles;

HANDLE STDCALL SSDevGetDevices(IN LPCTSTR lpVid /* = NULL */, IN LPCTSTR lpPID /* = NULL */,

IN const GUID* pSetupClassGuid /* = NULL */, IN const GUID* pInterfaceClassGuid /* = NULL */)

{

SSDevHandles *phDevs = NULL;

HDEVINFO hDevInfoSet = INVALID_HANDLE_VALUE;

check_int_begin

{

if(lpVid != NULL && _tcslen(lpVid) == 0)

lpVid = NULL;

if(lpPID != NULL && _tcslen(lpPID) == 0)

lpPID = NULL;

hDevInfoSet = SetupDiCreateDeviceInfoList(pSetupClassGuid, NULL);

check_int_bool(hDevInfoSet != INVALID_HANDLE_VALUE, SS_RC_NOT_FOUND);

if(pInterfaceClassGuid == NULL)

hDevInfoSet = SetupDiGetClassDevsEx(NULL, NULL, NULL,

DIGCF_ALLCLASSES|DIGCF_DEVICEINTERFACE|DIGCF_PRESENT,

hDevInfoSet, NULL, NULL);

else

hDevInfoSet = SetupDiGetClassDevsEx(pInterfaceClassGuid, NULL, NULL,

DIGCF_DEVICEINTERFACE|DIGCF_PRESENT,

hDevInfoSet, NULL, NULL);

check_int_bool(hDevInfoSet != INVALID_HANDLE_VALUE, SS_RC_NOT_FOUND);

SP_DEVINFO_DATA dtDevInfo = { sizeof(SP_DEVINFO_DATA) };

for(DWORD dwMemIdx = 0; ;dwMemIdx++)

{

if(!SetupDiEnumDeviceInfo(hDevInfoSet, dwMemIdx, &dtDevInfo))

{

if(::GetLastError() == ERROR_NO_MORE_ITEMS)

{

break;

}

continue;;

}

if(lpVid != NULL || lpPID != NULL)

{

TCHAR szInstanceId[MAX_INSTANCE_ID] = {0};

SetupDiGetDeviceInstanceId(hDevInfoSet, &dtDevInfo, szInstanceId, SS_DIMOF(szInstanceId), NULL);

if(_tcslen(szInstanceId) <= 0)

continue;

TCHAR szVID[5] = {0}, szPID[5] = {0};

if(!SSDevUtilGetVIDPID(szInstanceId, szVID, SS_DIMOF(szVID), szPID, SS_DIMOF(szPID)))

continue;

if(lpVid != NULL && _tcsicmp(lpVid, szVID) != 0)

continue;

if(lpPID != NULL && _tcsicmp(lpPID, szPID) != 0)

continue;

}

DWORD dwCapbilities = 0;

if(!SetupDiGetDeviceRegistryProperty(hDevInfoSet, &dtDevInfo, SPDRP_CAPABILITIES, NULL,\

(BYTE*)&dwCapbilities, sizeof(DWORD), NULL))

continue;

if(!SS_FLAG_ISSET(dwCapbilities, CM_DEVCAP_REMOVABLE))

continue;

if(phDevs == NULL)

{

phDevs = (SSDevHandles *)SS_MALLOC(sizeof(SSDevHandles));

memset(phDevs, 0, sizeof(SSDevHandles));

phDevs->hDevInfoSet = INVALID_HANDLE_VALUE;

}

phDevs->pDevDatas = (SP_DEVINFO_DATA*)SS_REALLOC(phDevs->pDevDatas, (phDevs->dwDevsCount + 1) *sizeof(SP_DEVINFO_DATA));

MP_ASSERT(phDevs->pDevDatas != NULL);

phDevs->pDevDatas[phDevs->dwDevsCount] = dtDevInfo;

phDevs->dwDevsCount++;

}

check_int_finally

{

if(phDevs != NULL && phDevs->dwDevsCount > 0)

{

phDevs->hDevInfoSet = hDevInfoSet;

hDevInfoSet = INVALID_HANDLE_VALUE;

}

if(hDevInfoSet != INVALID_HANDLE_VALUE)

{

SetupDiDestroyDeviceInfoList(hDevInfoSet);

hDevInfoSet = INVALID_HANDLE_VALUE;

}

return phDevs;

}

并通过setupDi函数SetupDiGetDeviceInstanceId ，SetupDiGetDeviceRegistryProperty 来得到设备信息。

监控USB设备插拔

注册设备通知事件

HANDLE STDCALL SSPNPRegisterDeviceNotifyToHwnd(IN HWND hWnd)

{

if(!::IsWindow(hWnd))

return NULL;

SSPNPDeviceNotifyHandle *pDeviceHandle = (SSPNPDeviceNotifyHandle *)SS_MALLOC(sizeof(SSPNPDeviceNotifyHandle));

memset(pDeviceHandle, 0, sizeof(SSPNPDeviceNotifyHandle));

DEV_BROADCAST_DEVICEINTERFACE NotificationFilter = {0};

NotificationFilter.dbcc_size = sizeof(NotificationFilter);

NotificationFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;

for(int nIdx=0; nIdx<SS_DIMOF(GUID_DEVINTERFACE_LIST); nIdx++)

{

NotificationFilter.dbcc_classguid = GUID_DEVINTERFACE_LIST[nIdx];

pDeviceHandle->hDevNotifies[pDeviceHandle->nCount++] = ::RegisterDeviceNotification(hWnd,

&NotificationFilter, DEVICE_NOTIFY_WINDOW_HANDLE);

}

if(pDeviceHandle->nCount <= 0)

{

SS_FREE(pDeviceHandle);

pDeviceHandle = NULL;

}

return (HANDLE)pDeviceHandle;

}

BOOL STDCALL SSPNPUnregisterDeviceNotify(IN HANDLE hDevNotify)

{

SSPNPDeviceNotifyHandle *pDeviceHandle = (SSPNPDeviceNotifyHandle *)hDevNotify;

if(pDeviceHandle == NULL)

return FALSE;

for(int nIdx=0; nIdx<SS_DIMOF(GUID_DEVINTERFACE_LIST); nIdx++)

{

if(pDeviceHandle->hDevNotifies[nIdx] != NULL)

UnregisterDeviceNotification(pDeviceHandle->hDevNotifies[nIdx]);

}

SS_FREE(pDeviceHandle);

pDeviceHandle = NULL;

return TRUE;

}

反注册设备通知事件

BOOL STDCALL SSPNPUnregisterDeviceNotify(IN HANDLE hDevNotify)

{

SSPNPDeviceNotifyHandle *pDeviceHandle = (SSPNPDeviceNotifyHandle *)hDevNotify;

if(pDeviceHandle == NULL)

return FALSE;

for(int nIdx=0; nIdx<SS_DIMOF(GUID_DEVINTERFACE_LIST); nIdx++)

{

if(pDeviceHandle->hDevNotifies[nIdx] != NULL)

UnregisterDeviceNotification(pDeviceHandle->hDevNotifies[nIdx]);

}

SS_FREE(pDeviceHandle);

pDeviceHandle = NULL;

return TRUE;

}

事件发生后会像窗口发送WM_DEVICECHANGE 消息，可以通过检测wParam是否为DBT_DEVICEARRIVAL 或者DBT_DEVICEREMOVECOMPLETE 来判断插拔。如果确实是插拔事件，且lParam不为空，则lParam指向一个结构DEV_BROADCAST_HDR ，通过判断它的域dbch_devicetype 来判断事件类型，例如

LPCTSTR STDCALL SSDevGetNameOnDeviceNotifyCB(WPARAM wParam, LPARAM lParam)

{

if(wParam != DBT_DEVICEARRIVAL && wParam != DBT_DEVICEREMOVECOMPLETE)

return NULL;

LPCTSTR lpReturnString = NULL;

DEV_BROADCAST_HDR *pDevBCHdr = (DEV_BROADCAST_HDR *)lParam;

MP_ASSERT(pDevBCHdr != NULL);

switch (pDevBCHdr->dbch_devicetype)

{

case DBT_DEVTYP_DEVICEINTERFACE:

{

DEV_BROADCAST_DEVICEINTERFACE *pDevBCInterface = (DEV_BROADCAST_DEVICEINTERFACE *)lParam;