U盘检测MFC

增加头文件 

#include <Dbt.h>

重写WindowProc

LRESULT CUDiskDetectionDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{
	// TODO: 在此添加专用代码和/或调用基类
	if(message == WM_DEVICECHANGE)
	{
		DealDeviceChange(wParam,lParam);
	}
	return CDialog::WindowProc(message, wParam, lParam);
}

增加

BOOL CUDiskDetectionDlg::DealDeviceChange(WPARAM wParam, LPARAM lParam)
{
	BOOL bRes = TRUE;

	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);
					}
					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);
					}
					break;
				case DBTF_MEDIA:    //光盘

					break;
				}
			}
		}
		break;
	}

	return bRes;
}

char CUDiskDetectionDlg::FirstDriveFromMask(ULONG unitmask)
{
	char i;

	for (i = 0; i < 26; ++i)
	{
		if (unitmask & 0x1)
			break;

		unitmask = unitmask >> 1;
	}

	return (i + 'A');
}