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

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

用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. void RegisterDeviceNotify()
4. {
5. HDEVNOTIFY hDevNotify;
6. DEV_BROADCAST_DEVICEINTERFACE NotificationFilter;
7. ZeroMemory( &NotificationFilter, sizeof(NotificationFilter) );
8. NotificationFilter.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
9. NotificationFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
10. NotificationFilter.dbcc_classguid = GUID_DEVINTERFACE_USB_DEVICE;
11. hDevNotify = RegisterDeviceNotification(hWnd, &NotificationFilter, DEVICE_NOTIFY_WINDOW_HANDLE);
12. }

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. return DefWindowProc(hWnd, message, wParam, lParam);
9. }

3.处理接收到的WM_DEVICECHANGE消息

[cpp] view plain copy

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

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