WINDOWS SDK 检测U盘插入

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

 

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

 

下面给出窗口的回调函数:

  1. LRESULT CALLBACK WndProc (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)  
  2.  {  
  3.        
  4.      switch (message)  
  5.      {  
  6.            
  7.      case WM_CREATE:/*初始化*/  
  8.            
  9.          MessageBox(hWnd, "go!""405 Studio", MB_OK);  
  10.            
  11.          break;  
  12.            
  13.      case WM_DEVICECHANGE:  
  14.            
  15.          if(wParam == DBT_DEVICEARRIVAL) //设备激活   
  16.          {  
  17.              PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam;  
  18.               
  19.              PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;  
  20.               
  21.              char szMsg[80];  
  22.              wsprintf (szMsg, "Drive %c: Media has arrived./n",  
  23.                  FirstDriveFromMask(lpdbv ->dbcv_unitmask));  
  24.               
  25.              MessageBox (hWnd, szMsg, "WM_DEVICECHANGE", MB_OK);  
  26.               
  27.          }  
  28.            
  29.          break;  
  30.            
  31.      default:  
  32.          return DefWindowProc (hWnd, message, wParam, lParam) ;  
  33.      }  
  34.        
  35.      return 0;  
  36.       
  37.  }  

原理很简单检测U盘是否被插入是通过监听WM_DEVICECHANGE消息来实现的。18,19行是从WM_DEVICECHANGE的lParam里获取盘符。
FirstDriveFromMask是一个用返回盘符(A,B,C ....)的简单函数
  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 = unitmask >> 1;  
  10.     }  
  11.     return (i + 'A');  
  12. }  
 

 

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

 

至此,问题解决。

 

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

[cpp:nogutter] view plaincopyprint?
  1. #include <windows.h>   
  2. #include <ShlObj.h>   
  3. #pragma  comment(lib, "shell32.lib")   
  4. #define WM_SHNOTIFY WM_USER + 1001   
  5. BEGIN_MESSAGE_MAP(CTvPlayerDlg, CDialog)  
  6.     ON_MESSAGE(WM_SHNOTIFY, &CTvPlayerDlg::OnSHWinProc)  
  7. END_MESSAGE_MAP()  
  8. //OnSHWinProc   
  9. RESULT OnSHWinProc(WPARAM wParam, LPARAM lParam)  
  10. {  
  11.     int res = CheckCard();  
  12.     if(res != TRUE)  
  13.     {  
  14. //退出程序   
  15. }  
  16. return 0;  
  17. }  
 

 

在Win7 64bit 下验证OK

发布了0 篇原创文章 · 获赞 10 · 访问量 17万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览