获取系统磁盘类型信息以及判断是否是U盘

使用WIN API函数实现

#include <iostream>
#include <vector>
#include <string>
#include <Windows.h>
#include <atlstr.h>
using namespace std;

#define MAX_DRIVER_STRING_LENGHT MAX_PATH

void getVidPid(wchar_t volNum)
{

wchar_t szDrvName[] = L"\\\\.\\A:";
szDrvName[4] = volNum;
HANDLE hDevice = NULL;

try
{
hDevice = CreateFile(szDrvName, GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, NULL, NULL);
}
catch (...)
{
//创建设备文件失败CloseHandle(hDevice);
return;
}

PSTORAGE_DEVICE_DESCRIPTOR pDevDesc;
STORAGE_PROPERTY_QUERY Query;
DWORD dwOutBytes = 0;
BOOL bResult;

pDevDesc = (PSTORAGE_DEVICE_DESCRIPTOR)new BYTE[sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1];
pDevDesc->Size = sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1;

char buffer [1000] = {0};
memset ((void *) & Query, 0, sizeof (Query));
memset (buffer, 0, sizeof (buffer));

Query.PropertyId = StorageDeviceProperty;
Query.QueryType = PropertyStandardQuery;

char* vid = NULL, *pid = NULL;
if ( DeviceIoControl (hDevice, IOCTL_STORAGE_QUERY_PROPERTY,
& Query,
sizeof (Query),
buffer,
sizeof (buffer),
& dwOutBytes, NULL) )
{
STORAGE_DEVICE_DESCRIPTOR * descrip = (STORAGE_DEVICE_DESCRIPTOR *) buffer;

if(descrip -> VendorIdOffset)
{
vid = &buffer [descrip -> VendorIdOffset];
}
if(buffer [descrip -> ProductIdOffset])
{
pid = &buffer [descrip -> ProductIdOffset];
}
}

try
{
bResult = ::DeviceIoControl(hDevice,
IOCTL_STORAGE_QUERY_PROPERTY,
&Query, sizeof(STORAGE_PROPERTY_QUERY),
pDevDesc, pDevDesc->Size,
&dwOutBytes,
(LPOVERLAPPED)NULL);
}
catch (...)
{
//MessageBox(NULL, "获取信息失败", "错误", MB_OK | MB_ICONERROR);
return ;
}

if(bResult)
{
unsigned char *p = NULL;
p = (unsigned char*)pDevDesc;

//BYTE *vID1 = (BYTE*)(pDevDesc+(*pDevDesc).VendorIdOffset);

unsigned char *vID = &p[pDevDesc->VendorIdOffset];
cout<<vID<<endl;
unsigned char *pID = &p[pDevDesc->ProductIdOffset];
cout<<pID<<endl;

//strcpy(GetUSBInfo->U_PROD,&p[pDevDesc->ProductIdOffset]);//此即为producter

//strcpy(GetUSBInfo->U_REV,&p[pDevDesc->ProductRevisionOffset]);//此即为version

//strcpy(GetUSBInfo->U_VEN ,&p[pDevDesc->VendorIdOffset]);//此即为vendor

}
else
{
//MessageBox(NULL, "获取信息失败", "错误", MB_OK | MB_ICONERROR);
return ;
}
}

int GetUsbDevice(vector<wstring>& strUsbDriverName)
{
wchar_t lpBuffer[MAX_DRIVER_STRING_LENGHT] = {0};
wchar_t* lpSingleDevice = NULL;

DWORD nBufferLength = MAX_DRIVER_STRING_LENGHT;
DWORD dwSignleDeviceBufferIndex =0, dwDeviceNum =0;

lpSingleDevice = (wchar_t*)malloc(sizeof(wchar_t)* MAX_DRIVER_STRING_LENGHT);
if(lpSingleDevice == NULL)
{
cout<<"系统分配内存错误"<<endl;
return 0;
}
memset(lpSingleDevice,0x0,MAX_DRIVER_STRING_LENGHT*2);
//memset(lpBuffer,0x0,MAX_DRIVER_STRING_LENGHT);

nBufferLength = GetLogicalDriveStrings(nBufferLength,lpBuffer); //列举盘符

for(DWORD i =0; i< nBufferLength; i++)
{
if(lpBuffer[i] != 0x00) //分离各个离盘符
{
lpSingleDevice[dwSignleDeviceBufferIndex] = lpBuffer[i];
dwSignleDeviceBufferIndex++;
continue;
}
lpSingleDevice[dwSignleDeviceBufferIndex] = 0x0;
dwSignleDeviceBufferIndex = 0;
if(*lpSingleDevice == L'A') //A盘 不需要
continue;
UINT flag = GetDriveType(lpSingleDevice);

if(flag == DRIVE_REMOVABLE) //获取移动磁盘
{
strUsbDriverName.push_back(lpSingleDevice);
//strUsbDriverName[dwDeviceNum] = lpSingleDevice ;
wcout<<L"DRIVE_REMOVABLE = "<<lpSingleDevice<<endl;
dwDeviceNum++;
getVidPid(*lpSingleDevice);
}
else if(flag == DRIVE_REMOTE)
{
strUsbDriverName.push_back(lpSingleDevice);
//strUsbDriverName[dwDeviceNum] = lpSingleDevice ;
wcout<<L"DRIVE_REMOTE = "<<lpSingleDevice<<endl;
dwDeviceNum++;
}
else if(flag == DRIVE_CDROM)
{
strUsbDriverName.push_back(lpSingleDevice);
//strUsbDriverName[dwDeviceNum] = lpSingleDevice ;
wcout<<"DRIVE_CDROM = "<<lpSingleDevice<<endl;
dwDeviceNum++;
}

}
free(lpSingleDevice);
lpSingleDevice = NULL;

return dwDeviceNum;
}

#include <boost/filesystem.hpp>

int main()
{
/*string from("F:/SVN提交模板.txt");
string to("\\\\192.190.20.200\\工具\\SVN提交模板.txt");
try
{
boost::filesystem::copy_file(from, to, boost::filesystem::copy_option::overwrite_if_exists);
}
catch (const std::exception &ex)
{
cout<<"error = "<<ex.what()<<endl;
}*/

vector<wstring> strUsbDriverName, strCdRomDriverName, strRemoteDriverName;
int deviceNum = 0;
deviceNum = GetUsbDevice(strUsbDriverName);
cout<<"deviceNum = "<<deviceNum<<endl;
return 0;
}