VC检测USB接口中插拔外部设备

#include "StdAfx.h" #include "EnumDisks.h" using namespace Disk; using namespace Disk::Types; static RemovableDeviceType GetDeviceType( PSP_DEVINFO_DATA pDevInfoData ); static std::wstring GetDevicePath( PSP_DEVINFO_DATA pDevInfoData ); static std::wstring GetDeviceRegistryProperty( HDEVINFO hDevInfo, PSP_DEVINFO_DATA pDevInfoData, DWORD dwProperty ); static BOOL IsDeviceNumberEq(const std::wstring& sDevicePath, DWORD dwDeviceNumber); static BOOL GetDrivesDevInst(DWORD dwDeviceNumber, HDEVINFO hDevInfo, DWORD dwIndex, PDEVINST devInst); static DEVINST FindDrivesDevInstByPath(const std::wstring& wsPath); static DEVINST FindDrivesDevInstByDeviceNumber(DWORD dwDeviceNumber); static bool BuildPaths(const std::wstring& sPath, std::wstring& sDevice, std::wstring& sVolumeAccess); static DWORD GetDeviceNumber(const std::wstring & sVolumeAccess); static std::vector<unsigned int> GetHarddiskIndexes(const std::wstring& wsPath); static DiskGeometry GetDeviceGeometry(const std::wstring& wsDevInterfaceVolume); static std::wstring GetDevInterfaceVolume(PSP_DEVINFO_DATA pdevInfoData); std::wstring GetDevicePath(PSP_DEVINFO_DATA pDevInfoData) { if ( NULL == pDevInfoData ) { // Wrong PSP_DEVINFO_DATA parameter throw std::runtime_error("Wrong PSP_DEVINFO_DATA parameter"); } std::wstring wsPath(L"X:\\"); for (wchar_t wLeter = L'A'; wLeter <= L'Z'; ++wLeter) { wsPath[0] = wLeter; if ( pDevInfoData->DevInst == FindDrivesDevInstByPath(wsPath) ) { return wsPath; } } // Can't find device path throw std::runtime_error("Can't find device path"); } std::wstring GetDeviceRegistryProperty(HDEVINFO hDevInfo, PSP_DEVINFO_DATA pDevInfoData, DWORD dwProperty) { DWORD dwSize = 0; DWORD dwDataType = 0; if ( NULL == pDevInfoData ) { // Wrong PSP_DEVINFO_DATA parameter throw std::runtime_error("Wrong PSP_DEVINFO_DATA parameter"); } if ( FALSE == ::SetupDiGetDeviceRegistryPropertyW(hDevInfo, pDevInfoData, dwProperty, &dwDataType, NULL, 0, &dwSize) ) { if ( ERROR_INSUFFICIENT_BUFFER != ::GetLastError() ) { if ( SPDRP_FRIENDLYNAME == dwProperty ) { return L"Removable Disk"; } // Can't get Device Registry Property throw std::runtime_error("Can't get Device Registry Property"); } } std::vector<wchar_t> sResult(dwSize + sizeof(wchar_t), 0x00); if ( FALSE == ::SetupDiGetDeviceRegistryPropertyW(hDevInfo, pDevInfoData, dwProperty, &dwDataType, (PBYTE) &sResult.front(), (DWORD)sResult.size(), &dwSize) ) { // Can't get Device Registry Property throw std::runtime_error("Can't get Device Registry Property"); } return std::wstring(&sResult.at(0)); } BOOL IsDeviceNumberEq(const std::wstring& sDevicePath, DWORD dwDeviceNumber) { BOOL bResult = FALSE; HANDLE hDrive = ::CreateFileW(sDevicePath.c_str(), 0, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, NULL, NULL); if ( INVALID_HANDLE_VALUE == hDrive ) { return bResult; } DWORD dwRead = 0; STORAGE_DEVICE_NUMBER sdn; ZeroMemory(&sdn, sizeof(sdn)); if ( ::DeviceIoControl(hDrive, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL, 0, &sdn, sizeof(sdn), &dwRead, NULL) ) { bResult = ( dwDeviceNumber == sdn.DeviceNumber ); } ::CloseHandle(hDrive); return bResult; } BOOL GetDrivesDevInst(DWORD dwDeviceNumber, HDEVINFO hDevInfo, DWORD dwIndex, PDEVINST devInst) { if ( (INVALID_HANDLE_VALUE == hDevInfo) || (0 == devInst) ) { return FALSE; } *devInst = 0; DWORD dwSize = 0; BYTE bBuffer[0x1000] = {0}; SP_DEVICE_INTERFACE_DATA spDevInterfaceData; ZeroMemory(&spDevInterfaceData, sizeof(spDevInterfaceData)); spDevInterfaceData.cbSize = sizeof(spDevInterfaceData); if ( FALSE == ::SetupDiEnumDeviceInterfaces(hDevInfo, NULL, &GUID_DEVINTERFACE_DISK, dwIndex, &spDevInterfaceData) ) { return FALSE; } ::SetupDiGetDeviceInterfaceDetail(hDevInfo, &spDevInterfaceData, NULL, 0, &dwSize, NULL); if ( (0 == dwSize) || (sizeof(bBuffer) < dwSize) ) { return FALSE; } SP_DEVINFO_DATA spDevInfoData; ZeroMemory(&spDevInfoData, sizeof(spDevInfoData)); spDevInfoData.cbSize = sizeof(spDevInfoData); PSP_DEVICE_INTERFACE_DETAIL_DATA pspDevInterfaceDetailData = (PSP_DEVICE_INTERFACE_DETAIL_DATA) bBuffer; pspDevInterfaceDetailData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA); if ( FALSE == ::SetupDiGetDeviceInterfaceDetail( hDevInfo, &spDevInterfaceData, pspDevInterfaceDetailData, dwSize, &dwSize, &spDevInfoData) ) { return TRUE; } if ( IsDeviceNumberEq(pspDevInterfaceDetailData->DevicePath, dwDeviceNumber) ) { *devInst = spDevInfoData.DevInst; } return TRUE; } DEVINST FindDrivesDevInstByPath(const std::wstring& wsPath) { std::wstring sDevice; std::wstring sVolumeAccess; if ( !BuildPaths(wsPath, sDevice, sVolumeAccess) ) { return 0; } DWORD dwDeviceNumber = GetDeviceNumber(sVolumeAccess); if ( -1 == dwDeviceNumber ) { return 0; } wchar_t szDosDeviceName[MAX_PATH] = {0}; if ( FALSE == ::QueryDosDeviceW(sDevice.c_str(), szDosDeviceName, MAX_PATH) ) { return 0; } const std::wstring patern = L"Harddisk"; const std::wstring dosDeviceName = szDosDeviceName; if ( dosDeviceName.end() == std::search(dosDeviceName.begin(), dosDeviceName.end(), patern.begin(), patern.end()) ) { return 0; } return FindDrivesDevInstByDeviceNumber(dwDeviceNumber); } DEVINST FindDrivesDevInstByDeviceNumber(DWORD dwDeviceNumber) { CONST DEVINST devInstEmpty = 0; HDEVINFO hDevInfo = ::SetupDiGetClassDevs( &GUID_DEVINTERFACE_DISK, NULL, NULL, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE); DEVINST devInst = 0; BOOL bResult = TRUE; for ( DWORD dwIndex = 0; ((dwIndex < 0xFFFFFFFF) && (devInstEmpty == devInst) && bResult); ++dwIndex ) { bResult = GetDrivesDevInst(dwDeviceNumber, hDevInfo, dwIndex, &devInst); } ::SetupDiDestroyDeviceInfoList(hDevInfo); return devInst; } bool BuildPaths(const std::wstring& sPath, std::wstring& sDevice, std::wstring& sVolumeAccess) { if ( sPath.empty() ) { return false; } wchar_t wcDriveLetter = sPath[0]; wcDriveLetter &= ~0x20; if ( (wcDriveLetter < L'A') || (wcDriveLetter > L'Z') ) { return false; } sDevice.assign(L"X:"); sVolumeAccess.assign(L"\\\\.\\X:"); sDevice[0] = wcDriveLetter; sVolumeAccess[4] = wcDriveLetter; return true; } DWORD GetDeviceNumber(const std::wstring& sVolumeAccess) { DWORD dwDeviceNumber = (DWORD)(-1); HANDLE hVolume = ::CreateFileW(sVolumeAccess.c_str(), 0, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, NULL, NULL); if (hVolume == INVALID_HANDLE_VALUE) { return dwDeviceNumber; } DWORD dwBytesReturned = 0; STORAGE_DEVICE_NUMBER sdn; if ( ::DeviceIoControl(hVolume, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL, 0, &sdn, sizeof(sdn), &dwBytesReturned, NULL) ) { dwDeviceNumber = sdn.DeviceNumber; } ::CloseHandle(hVolume); return dwDeviceNumber; } std::vector<unsigned int> GetHarddiskIndexes(const std::wstring& wsPath) { AutoHandle handle( ::CreateFileW( wsPath.c_str(), GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_BACKUP_SEMANTICS, NULL) ); VOL