获取系统磁盘类型信息以及判断是否是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;
}