关于HID设备与WM_DEVICECHANGE

LRESULT CHWDetectDlg::OnMyDeviceChange(WPARAM wParam, LPARAM lParam)

{

	if ( DBT_DEVICEARRIVAL == wParam || DBT_DEVICEREMOVECOMPLETE == wParam ) {

		PDEV_BROADCAST_HDR pHdr = (PDEV_BROADCAST_HDR)lParam;

		PDEV_BROADCAST_DEVICEINTERFACE pDevInf;

		PDEV_BROADCAST_HANDLE pDevHnd;

		PDEV_BROADCAST_OEM pDevOem;

		PDEV_BROADCAST_PORT pDevPort;

		PDEV_BROADCAST_VOLUME pDevVolume;

		switch( pHdr->dbch_devicetype ) {

			case DBT_DEVTYP_DEVICEINTERFACE:

				pDevInf = (PDEV_BROADCAST_DEVICEINTERFACE)pHdr;

				UpdateDevice(pDevInf, wParam);

				break;



			case DBT_DEVTYP_HANDLE:

				pDevHnd = (PDEV_BROADCAST_HANDLE)pHdr;

				break;



			case DBT_DEVTYP_OEM:

				pDevOem = (PDEV_BROADCAST_OEM)pHdr;

				break;



			case DBT_DEVTYP_PORT:

				pDevPort = (PDEV_BROADCAST_PORT)pHdr;

				break;



			case DBT_DEVTYP_VOLUME:

				pDevVolume = (PDEV_BROADCAST_VOLUME)pHdr;

				break;

		}

	}

	return 0;

}

void CHWDetectDlg::UpdateDevice(PDEV_BROADCAST_DEVICEINTERFACE pDevInf, WPARAM wParam)

{

	// pDevInf->dbcc_name: 

	// \\?\USB#Vid_04e8&Pid_503b#0002F9A9828E0F06#{a5dcbf10-6530-11d2-901f-00c04fb951ed}

	// szDevId: USB\Vid_04e8&Pid_503b\0002F9A9828E0F06

	// szClass: USB

	ASSERT(lstrlen(pDevInf->dbcc_name) > 4);

	CString szDevId = pDevInf->dbcc_name+4;

	int idx = szDevId.ReverseFind(_T('#'));

	ASSERT( -1 != idx );

	szDevId.Truncate(idx);

	szDevId.Replace(_T('#'), _T('\\'));

	szDevId.MakeUpper();



	CString szClass;

	idx = szDevId.Find(_T('\\'));

	ASSERT(-1 != idx );

	szClass = szDevId.Left(idx);

	CString szLog;

	m_ctrlEdit.GetWindowText(szLog);

	CString szTmp;

	if ( DBT_DEVICEARRIVAL == wParam ) {

		szTmp.Format(_T("Adding %s\r\n"), szDevId.GetBuffer());

		// TRACE("Adding %s\n", szDevId.GetBuffer());

	} else {

		szTmp.Format(_T("Removing %s\r\n"), szDevId.GetBuffer());

		// TRACE("Removing %s\n", szDevId.GetBuffer());

	}

	szLog.Append(szTmp);

	m_ctrlEdit.SetWindowText(szLog);



	// seems we should ignore "ROOT" type....

	if ( _T("ROOT") == szClass ) {

		return;

	}



	DWORD dwFlag = DBT_DEVICEARRIVAL != wParam ? 

		DIGCF_ALLCLASSES : (DIGCF_ALLCLASSES | DIGCF_PRESENT);

	HDEVINFO hDevInfo = SetupDiGetClassDevs(NULL,szClass,NULL,dwFlag);

	if( INVALID_HANDLE_VALUE == hDevInfo ) {

		AfxMessageBox(CString("SetupDiGetClassDevs(): ") 

			+ _com_error(GetLastError()).ErrorMessage(), MB_ICONEXCLAMATION);

		return;

	}



	SP_DEVINFO_DATA spDevInfoData;

	if ( FindDevice(hDevInfo, szDevId, spDevInfoData) ) {

		// OK, device found

		DWORD DataT ;

		TCHAR buf[MAX_PATH];

		DWORD nSize = 0;



		// get Friendly Name or Device Description

		if ( SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, 

			SPDRP_FRIENDLYNAME, &DataT, (PBYTE)buf, sizeof(buf), &nSize) ) {

		} else if ( SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, 

			SPDRP_DEVICEDESC, &DataT, (PBYTE)buf, sizeof(buf), &nSize) ) {

		} else {

			lstrcpy(buf, _T("Unknown"));

		}



		HTREEITEM hClass = GetClassItem(&(spDevInfoData.ClassGuid), wParam);

		// hClass can only be NULL for remove device and class node is not found

		if ( NULL == hClass ) return;

		// remove device

		HTREEITEM hChildItem = m_ctrlTree.GetChildItem(hClass);

		BOOL bFound = FALSE;

		while (hChildItem != NULL) {

			CString* pszData = (CString*)m_ctrlTree.GetItemData(hChildItem);

			if ( szDevId == *pszData  ) {

				bFound = TRUE;

				if ( DBT_DEVICEARRIVAL == wParam ) {

					// add device, but seems the device already exists

					break;

				} else {

					// remove device

					delete pszData;

					m_ctrlTree.DeleteItem(hChildItem);

					if ( !m_ctrlTree.ItemHasChildren(hClass) ) {

						HeapFree(GetProcessHeap(), 0, (LPVOID)m_ctrlTree.GetItemData(hClass));

						m_ctrlTree.DeleteItem(hClass);

					}

					break;

				}

			} else {

				hChildItem = m_ctrlTree.GetNextItem(hChildItem, TVGN_NEXT);

			}

		}

		if ( DBT_DEVICEARRIVAL == wParam && !bFound ) {

			// Add device

			int n = GetClassImgIndex(&(spDevInfoData.ClassGuid));

			HTREEITEM hItem = m_ctrlTree.InsertItem(buf, n, n, hClass);

			m_ctrlTree.SetItemData(hItem, (DWORD_PTR)(new CString(szDevId)));

			// set pspDevInfoData to NULL, otherwise it will be HeapFree

			m_ctrlTree.Expand(hClass, TVE_EXPAND);

		}

	}



	SetupDiDestroyDeviceInfoList(hDevInfo);

}