用win32 API监听U盘插拔并取得其盘符/取得当前插入U盘的盘符

Posted 2020-09-29

版权声明：本文为博主原创文章，未经博主允许不得转载。

用win32 API监听U盘插拔并取得其盘符

1.使用RegisterDeviceNotification()函数注册

 

[cpp] view plain copy

 

1. static const GUID GUID_DEVINTERFACE_USB_DEVICE =   
2.     {0xA5DCBF10, 0x6530, 0x11D2, {0x90, 0x1F, 0x00, 0xC0, 0x4F, 0xB9, 0x51, 0xED}};  
3.   
4. void RegisterDeviceNotify()  
5. {  
6.     HDEVNOTIFY hDevNotify;  
7.     DEV_BROADCAST_DEVICEINTERFACE NotificationFilter;  
8.     ZeroMemory( &NotificationFilter, sizeof(NotificationFilter) );  
9.     NotificationFilter.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);  
10.     NotificationFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;  
11.     NotificationFilter.dbcc_classguid = GUID_DEVINTERFACE_USB_DEVICE;  
12.     hDevNotify = RegisterDeviceNotification(hWnd, &NotificationFilter, DEVICE_NOTIFY_WINDOW_HANDLE);  
13. }  

2.在WndProc()函数中接收WM_DEVICECHANGE消息

 

 

[cpp] view plain copy

 

1. LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)  
2. {  
3.     switch(message)  
4.     {  
5.     case WM_DEVICECHANGE:  
6.         return DeviceChange(message, wParam, lParam);  
7.     }  
8.   
9.     return DefWindowProc(hWnd, message, wParam, lParam);  
10. }  

3.处理接收到的WM_DEVICECHANGE消息

 

 

[cpp] view plain copy

 

1. char FirstDriveFromMask(ULONG unitmask)  
2. {  
3.    char i;  
4.   
5.    for (i = 0; i < 26; ++i)  
6.    {  
7.       if (unitmask & 0x1)  
8.          break;  
9.       unitmask >>= 1;  
10.    }  
11.   
12.    return (i + ‘A‘);  
13. }  
14.   
15. LRESULT DeviceChange(UINT message, WPARAM wParam, LPARAM lParam)  
16. {  
17.     if ( DBT_DEVICEARRIVAL == wParam || DBT_DEVICEREMOVECOMPLETE == wParam )   
18.     {  
19.         PDEV_BROADCAST_HDR pHdr = (PDEV_BROADCAST_HDR)lParam;  
20.         if (pHdr->dbch_devicetype == DBT_DEVTYP_VOLUME)   
21.         {  
22.             PDEV_BROADCAST_VOLUME pDevVolume = (PDEV_BROADCAST_VOLUME)lParam;  
23.             char driverLabel = FirstDriveFromMask(pDevVolume->dbcv_unitmask);  
24.             if (wParam == DBT_DEVICEARRIVAL) {  
25.                 printf("add %c\r\n", driverLabel);  
26.             } else {  
27.                 printf("remove %c\r\n", driverLabel);  
28.             }  
29.         }  
30.     }  
31.     return 0;  
32. }  

用win32 API取得当前插入U盘的盘符

1.使用GetLogicalDrives()取得代表各分区的掩码

 

[cpp] view plain copy

 

1. DWORD mask = GetLogicalDrives();  

2.遍历掩码的每一位，判断对应的分区是否是U盘

 

 

[cpp] view plain copy

 

1. bool IsUsbDevice(wchar_t letter)  
2. {  
3.     wchar_t volumeAccessPath[] = L"\\\\.\\X:";  
4.     volumeAccessPath[4] = letter;  
5.   
6.     HANDLE deviceHandle = CreateFile(  
7.         volumeAccessPath,  
8.         0,                // no access to the drive  
9.         FILE_SHARE_READ | // share mode  
10.         FILE_SHARE_WRITE,   
11.         NULL,             // default security attributes  
12.         OPEN_EXISTING,    // disposition  
13.         0,                // file attributes  
14.         NULL);            // do not copy file attributes  
15.   
16.     // setup query  
17.     STORAGE_PROPERTY_QUERY query;  
18.     memset(&query, 0, sizeof(query));  
19.     query.PropertyId = StorageDeviceProperty;  
20.     query.QueryType = PropertyStandardQuery;  
21.   
22.     // issue query  
23.     DWORD bytes;  
24.     STORAGE_DEVICE_DESCRIPTOR devd;  
25.     STORAGE_BUS_TYPE busType = BusTypeUnknown;  
26.   
27.     if (DeviceIoControl(deviceHandle,  
28.         IOCTL_STORAGE_QUERY_PROPERTY,  
29.         &query, sizeof(query),  
30.         &devd, sizeof(devd),  
31.         &bytes, NULL))  
32.     {  
33.     busType = devd.BusType;  
34.     }  
35.   
36.     CloseHandle(deviceHandle);  
37.   
38.     return BusTypeUsb == busType;  
39. }  
40.   
41. // 查找U盘  
42. // 参数:  _letter 存储U盘盘符  
43. // 返回值:true 当前有U盘  
44. //       false 当前无U盘  
45. bool findUSBStorage(char* _letter)  
46. {  
47.     DWORD mask = GetLogicalDrives();  
48.     int count = 0;  
49.     while (mask != 0)  
50.     {  
51.         if ((mask & 0x01) == 1)  
52.         {  
53.             wchar_t letter = L‘A‘ + count;  
54.   
55.                 // 判断取得的盘符是否是U盘  
56.             if (IsUsbDevice(letter))  
57.             {  
58.                     wcstombs(_letter, &letter, 1);  
59.                     return true;  
60.             }  
61.         }  
62.         count++;  
63.         mask = mask >> 1;  
64.     }  
65.     return false;  
66. }