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