[MFC]U盘检测

一、MFC中使用消息队列

　　　　1.1　　声明

virtual LRESULT WindowProc(UINT message, WPARAM wParam, LPARAM lParam);

　　　　1.2　　实现

LRESULT CTestDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{
     return  CDialog::WindowProc(message,wParam,lParam);
}

　　　　　　本示例是继承CDialog。

二、捕获U盘插入、卸载事件

LRESULT CTestDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) 
{
     switch  (message)
    {
         // WM_DEVICECHANGE，系统硬件改变发出的系统消息
         case  WM_DEVICECHANGE:
        {
            PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam;
             switch (wParam)
            {
                 case  WM_DEVICECHANGE:
                     break ;
                 case  DBT_DEVICEARRIVAL: // DBT_DEVICEARRIVAL，设备检测结束，并且可以使用
                {
                     if (lpdb -> dbch_devicetype  ==  DBT_DEVTYP_VOLUME) // 逻辑卷
                    {
                        PDEV_BROADCAST_VOLUME lpdbv  =   (PDEV_BROADCAST_VOLUME)lpdb;
                         switch (lpdbv -> dbcv_flags)
                        {
                             case   0 :                 // U盘
                            {
                                CString decDriver;
                                decDriver  =  FirstDriveFromMask(lpdbv  -> dbcv_unitmask);
                                TRACE1( " 检测到U盘:[%s]插入!\n " ,decDriver.GetBuffer( 0 ));
                            }
                                 break ;
                             case  DBTF_MEDIA:     // 光盘
                                TRACE1( " 检测到光盘:[%c]插入!\n " ,FirstDriveFromMask(lpdbv  -> dbcv_unitmask));
                                 break ;
                        }
                    }
                }
                     break ;
                 case  DBT_DEVICEREMOVECOMPLETE: // DBT_DEVICEREMOVECOMPLETE,设备卸载或者拔出
                {
                     if (lpdb -> dbch_devicetype  ==  DBT_DEVTYP_VOLUME) // 逻辑卷
                    {
                        PDEV_BROADCAST_VOLUME lpdbv  =   (PDEV_BROADCAST_VOLUME)lpdb;
                         switch (lpdbv -> dbcv_flags)
                        {
                             case   0 :                 // U盘
                            {
                                CString decDriver;
                                decDriver  =  FirstDriveFromMask(lpdbv  -> dbcv_unitmask);
                                TRACE1( " 检测到U盘:[%s]拔出!\n " ,decDriver.GetBuffer( 0 ));
                            }
                                 break ;
                             case  DBTF_MEDIA:     // 光盘
                                
                                 break ;
                        }
                    }
                }
                     break ;
            }
        }
         break ;
    }
     return  CDialog::WindowProc(message,wParam,lParam);  
}

三、效果图

　　 

四、文章修改

　　　　4.1　　2010-5-7　　MFC中使用不要使用SetWindowLong方式来获取消息队列，否则会出现无法关闭窗口的问题。

　　　　4.2　　2013-2-17　　感谢网友提醒，CString getBuffer后应调用ReleaseBuffer

本文转自博客园农民伯伯的博客，原文链接：[MFC]U盘检测，如需转载请自行联系原博主。