WINDOWS SDK 检测U盘插入

工作需要，需要随时检测U盘的插入或拔出状态，然后根据状态做出相关操作。在网上搜索到下列方法：

 

今天一位老兄要我帮忙做一个软件,需要把在U盘插入USB接口激活之后自动把U盘里的内容拷贝到硬上。拷贝文件不难，麻烦的就是不知道怎么检测U盘是否激活，激活后又是哪个盘符。
google了一下，在MSDN上看了几个例子，找到了检测U盘插入，并确定其盘符的方法。

 

下面给出窗口的回调函数：
LRESULT CALLBACK WndProc (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { switch (message) { case WM_CREATE:/*初始化*/ MessageBox(hWnd, "go!", "405 Studio", MB_OK); break; case WM_DEVICECHANGE: if(wParam == DBT_DEVICEARRIVAL) //设备激活 { PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam; PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb; char szMsg[80]; wsprintf (szMsg, "Drive %c: Media has arrived./n", FirstDriveFromMask(lpdbv ->dbcv_unitmask)); MessageBox (hWnd, szMsg, "WM_DEVICECHANGE", MB_OK); } break; default: return DefWindowProc (hWnd, message, wParam, lParam) ; } return 0; }
原理很简单检测U盘是否被插入是通过监听WM_DEVICECHANGE消息来实现的。18，19行是从WM_DEVICECHANGE的lParam里获取盘符。
FirstDriveFromMask是一个用返回盘符（A，B，C ....）的简单函数
char FirstDriveFromMask (ULONG unitmask) { char i; for (i = 0; i < 26; ++i) { if (unitmask & 0x1) break; unitmask = unitmask >> 1; } return (i + 'A'); } 

 

具体是哪个盘符的，我这边不太关系，主要是需要截获U盘插拔消息WM_DEVICECHANG，然后进行相关处理。最终在PreTranslateMessage(MSG* pMsg)里if (WM_DEVICECHANGE == pMsg->message)判断里来进行的处理。

 

至此，问题解决。

 

上述方式在win xp下运行是完全没问题的，可是在Win7或Vista下却根本没有WM_DEVICECHANG消息，在网络搜索后发现是由于win7的安全机制引起，于是采用下述方式来实现：

#include <windows.h> #include <ShlObj.h> #pragma comment(lib, "shell32.lib") #define WM_SHNOTIFY WM_USER + 1001 BEGIN_MESSAGE_MAP(CTvPlayerDlg, CDialog) ON_MESSAGE(WM_SHNOTIFY, &CTvPlayerDlg::OnSHWinProc) END_MESSAGE_MAP() //OnSHWinProc RESULT OnSHWinProc(WPARAM wParam, LPARAM lParam) { int res = CheckCard(); if(res != TRUE) { //退出程序 } return 0; } 

 

在Win7 64bit 下验证OK。