USB移动存储设备的检测、写入及弹出

USB设备的基本操作涉及USB存储设备的插入检测及写入文件，同时可以在写入结束后自动弹出，下面介绍自己的实现方法。

（1）注册设备。

在检测之前需要对检测的设备进行注册。具体的代码如下：

//首先注册设备
void CUSBTestDlg::OnBtnRegister()
{
 // TODO: Add your control notification handler code here

 static const GUID GUID_DEVINTERFACE_LSIT[] =
 {
 //GUID_DEVINTERFACE_USB_DEVICE
 {0xA5DCBF10,0x6530,0x11D2,{0x90,0x1F,0x00,0xC0,0x4F,0xB9,0x51,0xED}},
 //GUID_DEVINTERFACE_DISK
 {0X53F56307,0XB6BF,0X11D0,{0X94,0XF2,0X00,0XA0,0XC9,0X1E,0XFB,0X8B}},
 //GUID_DEVINTERFACE_HID
 {0X4D1E55B2,0XF16F,0X11CF,{0X88,0XCB,0X00,0X11,0X11,0X00,0X00,0X30}},
 //GUID_NDIS_LAN_CLASS
 {0XAD498944,0X762F,0X11D0,{0X8D,0XCB,0X00,0X00,0X4F,0XC3,0X35,0X8C}}
 };
 //注册插拔事件
 HDEVNOTIFY hDevNotify;
 DEV_BROADCAST_DEVICEINTERFACE NotifacationFiler;
 ZeroMemory(&NotifacationFiler,sizeof(DEV_BROADCAST_DEVICEINTERFACE));
 NotifacationFiler.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
 NotifacationFiler.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
 
 for(int i=0;i<sizeof(GUID_DEVINTERFACE_LSIT)/sizeof(GUID);i++)
 {
  NotifacationFiler.dbcc_classguid = GUID_DEVINTERFACE_LSIT[i];//GetCurrentUSBGUID();//m_usb->GetDriverGUID();

  hDevNotify = RegisterDeviceNotification(m_hWnd,&NotifacationFiler,DEVICE_NOTIFY_WINDOW_HANDLE);
  if(!hDevNotify)
  {
   int Err = GetLastError();
   MessageBox("RegisterDeviceNotificationFailed");
  }
  else
  MessageBox("RegisterDeviceNotificationSuccess");
 } 
}

（2）利用ON_WM_DEVICECHANGE实现检测。

这里需要手动添加ON_WM_DEVICECHANGE消息。首先头文件中声明响应函数：

afx_msg BOOL OnDeviceChange(UINT nEventType, DWORD dwData);

定义响应函数：

BOOL CUSBTestDlg::OnDeviceChange(UINT nEventType, DWORD dwData)
{
 PDEV_BROADCAST_HDR pdb;
 PDEV_BROADCAST_VOLUME pdbv;
 CString strVolume="";

 pdb = (PDEV_BROADCAST_HDR)dwData;
 switch(nEventType)
 {
 case DBT_DEVICEARRIVAL:
  // Check whether a CD or DVD was inserted into a drive.
  if(pdb->dbch_devicetype==DBT_DEVTYP_VOLUME)
  {
   pdbv = (PDEV_BROADCAST_VOLUME)pdb;
   JudgeDeviceChangePath(pdbv->dbcv_unitmask,strVolume);//获取卷标
   g_strVolume = strVolume;
   IOFileToMobileDevice(strVolume);//写入数据
  }
  if(pdb->dbch_devicetype==DBT_DEVTYP_DEVICEINTERFACE)
  {

  }
  break;
 case DBT_DEVICEREMOVECOMPLETE:
  break;
 default:
  break;
 } 
 return TRUE;
}

（3）弹出USB设备。

BOOL CUSBTestDlg::UnInstallUsb(char szDriveName)
{
 DWORD accessMode = 0, shareMode = 0;
 HANDLE   hDevice;
 ULONG returned = 0,bResult = 0;
 DWORD   dwError; 
 shareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;  // default
 accessMode = GENERIC_WRITE | GENERIC_READ;       // default 
 if(szDriveName == 0) 
 { 
  return FALSE;
 }
 CString vol_string = "\\\\.\\A:";
 vol_string.SetAt(4, szDriveName);
 TRACE("%s",vol_string);
 hDevice = CreateFile(vol_string,accessMode,shareMode,NULL,OPEN_EXISTING,0,NULL);
    if(hDevice == INVALID_HANDLE_VALUE)
 {
  CString debugInfo;
  dwError   =   GetLastError();   
  debugInfo.Format("error = %d ", dwError); 
  TRACE("IOCTL_STORAGE_EJECT_MEDIA  errorcode = %d \n", dwError);
  return FALSE;
    }
 bResult  =  DeviceIoControl(hDevice,IOCTL_STORAGE_EJECT_MEDIA,NULL,0, NULL,0,&returned, NULL);
 if(!bResult)   //   IOCTL   failed     //经实验，有时虽然失败了，U盘还是能弹出 
 {  
  CString debugInfo;
  dwError   =   GetLastError();   
  debugInfo.Format("error = %d ", dwError); 
  TRACE("IOCTL_STORAGE_EJECT_MEDIA  errorcode = %d \n", dwError); 
 }
 bResult = CloseHandle(hDevice);
 if(!bResult)     
 {     
 CString debugInfo;
 dwError   =   GetLastError();   
 debugInfo.Format("error = %d ", dwError);
 TRACE("IOCTL_STORAGE_EJECT_MEDIA  errorcode = %d \n", dwError); 
 return FALSE;
 }   
 return TRUE;
}

这个函数在按钮的实现中，即把卷标传入即可：

void CUSBTestDlg::OnButton1()
{
 // TODO: Add your control notification handler code here
 UnInstallUsb((char)g_strVolume.GetAt(0));
}

（4）需要的读取卷标和写入的函数。

void CUSBTestDlg::JudgeDeviceChangePath(DWORD dwData, CString &strPath)
{
 for(int i=0;i<32;i++)
 {
  if(dwData&0x01)
  {
   //获取盘符
   CString str=char('A'+i);
   strPath+=str;
   break;
  }
  dwData = dwData>>1;
  if(dwData==0x0000)
   break;
 }
   
}

void CUSBTestDlg::IOFileToMobileDevice(CString str)
{
 FILE* stream = NULL;
 char paramfilepath[200];
 strcpy(paramfilepath,str);
 strcat(paramfilepath,":\\param.txt");
 if((stream = fopen(paramfilepath,"w"))!=NULL)
 {
  
  fprintf(stream,"%d\n",11);
 }

 CTime m_SysTime = CTime::GetCurrentTime();
 CString TimeStr;
 TimeStr = m_SysTime.Format("%Y%m%d%H%M%S");
 fprintf(stream,"%s\n",TimeStr);

 fclose(stream);
 stream = NULL;

 MessageBox("Write Success");
}

（5）注意事项。

头文件中需要包含：

#include <dbt.h>//
#include <winioctl.h>//IOCTL_STORAGE_EJECT_MEDIA未定义的问题

在StdAfx.h中定义如下：

#define WINVER 0X0500

这个宏定义的位置必须正确，在

#if !defined(AFX_STDAFX_H__30C2945E_D8FC_4DCD_B67A_099B447FE1AD__INCLUDED_)
#define AFX_STDAFX_H__30C2945E_D8FC_4DCD_B67A_099B447FE1AD__INCLUDED_

//solved 'DEV_BROADCAST_DEVICEINTERFACE' : undeclared identifier

#define WINVER 0X0500

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

 

 

下载地址：http://download.csdn.net/detail/zqckzqck/4980910