TestDlg.h的代码 // TestUSBDlg.h : 头文件 // #pragma once // CTestUSBDlg 对话框 class CTestUSBDlg : public CDialog { // 构造 public: CTestUSBDlg(CWnd* pParent = NULL); // 标准构造函数 // 对话框数据 enum { IDD = IDD_TESTUSB_DIALOG }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: HICON m_hIcon; // 生成的消息映射函数 virtual BOOL OnInitDialog(); afx_msg void OnSysCommand(UINT nID, LPARAM lParam); afx_msg void OnPaint(); afx_msg HCURSOR OnQueryDragIcon(); DECLARE_MESSAGE_MAP() virtual LRESULT WindowProc(UINT message, WPARAM wParam, LPARAM lParam); char FirstDriveFromMask(DWORD uMask); }; TestDlg.cpp的代码 // TestUSBDlg.cpp : 实现文件 // #include "stdafx.h" #include "TestUSB.h" #include "TestUSBDlg.h" #include "dbt.h" //#pragma comment(lib, "user32.lib") #ifdef _DEBUG #define new DEBUG_NEW #endif // 用于应用程序“关于”菜单项的 CAboutDlg 对话框 class CAboutDlg : public CDialog { public: CAboutDlg(); // 对话框数据 enum { IDD = IDD_ABOUTBOX }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialog) END_MESSAGE_MAP() // CTestUSBDlg 对话框 CTestUSBDlg::CTestUSBDlg(CWnd* pParent /*=NULL*/) : CDialog(CTestUSBDlg::IDD, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CTestUSBDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CTestUSBDlg, CDialog) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() //}}AFX_MSG_MAP END_MESSAGE_MAP() // CTestUSBDlg 消息处理程序 BOOL CTestUSBDlg::OnInitDialog() { CDialog::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != NULL) { CString strAboutMenu; strAboutMenu.LoadString(IDS_ABOUTBOX); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。当应用程序主窗口不是对话框时,框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 // TODO: 在此添加额外的初始化代码 return TRUE; // 除非将焦点设置到控件,否则返回 TRUE } void CTestUSBDlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialog::OnSysCommand(nID, lParam); } } // 如果向对话框添加最小化按钮,则需要下面的代码 // 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序, // 这将由框架自动完成。 void CTestUSBDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使图标在工作矩形中居中 int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // 绘制图标 dc.DrawIcon(x, y, m_hIcon); } else { CDialog::OnPaint(); } } //当用户拖动最小化窗口时系统调用此函数取得光标显示。 // HCURSOR CTestUSBDlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } LRESULT CTestUSBDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) { // TODO: 在此添加专用代码和/或调用基类 switch (message) { case WM_DEVICECHANGE: { PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam; switch(wParam) { case WM_DEVICECHANGE: break; case DBT_DEVICEARRIVAL://设备检测结束 并且可以使用 { if(lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)//逻辑卷 { PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb; switch(lpdbv->dbcv_flags) { case 0: { CString decDriver; decDriver = FirstDriveFromMask(lpdbv->dbcv_unitmask); MessageBox(decDriver,L"u盘"); } break; case DBTF_MEDIA: MessageBox(CString(FirstDriveFromMask(lpdbv->dbcv_unitmask)),L"光盘插入"); 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); MessageBox(decDriver,L"检测到U盘:[%s]拔出!/n"); } break; case DBTF_MEDIA: //光盘 { MessageBox(CString(FirstDriveFromMask(lpdbv->dbcv_unitmask)),L"光盘弹出"); } break; } } } break; } } break; } return CDialog::WindowProc(message, wParam, lParam); } /*------------------------------------------------------------------ FirstDriveFromMask (uMask) Description Finds the first valid drive letter from a mask of drive letters. The mask must be in the format bit 0 = A, bit 1 = B, bit 2 = C, and so on. A valid drive letter is defined when the corresponding bit is set to 1. Returns the first drive letter that was found. --------------------------------------------------------------------*/ char CTestUSBDlg::FirstDriveFromMask( DWORD uMask ) { char i; for (i = 0; i < 26; ++i) { if (uMask & 0x1) break; uMask = uMask >> 1; } return (i + 'A'); }