[vc++]枚举USB设备

#include 
#pragma comment(lib,"shlwapi.lib")
#include 
#pragma comment(lib,"setupapi.lib")
/*
* 函数名称: EnumUSBDev
* 函数功能: 枚举设备			
* 参    数: 
*			acDevPath:				存放设备路径
*			nDevNum:				设备数量
* 返 回 值: 成功返回TRUE,失败返回FALSE
*/
BOOL CEnumDevDlg::EnumUSBDev(OUT char acDevPath[MAX_PATH][MAX_PATH], OUT int &nDevNum)
{
	//入参检测
	nDevNum = 0;
	if (NULL == acDevPath)
	{
		return FALSE;
	}
	ZeroMemory(acDevPath, sizeof(acDevPath));
	BOOL								bRet = FALSE;
	SP_DEVICE_INTERFACE_DATA			DeviceInterfaceData;			// 保存设备接口信息
	DWORD								dwRequiredSize = 0;				// 得到详细信息时需要的内存大小
	PSP_INTERFACE_DEVICE_DETAIL_DATA	pDeviceDetailData = NULL;		// 详细信息
	HDEVINFO							hDeviceInfo;					// 设备信息集合的指针	
	GUID								InterfaceGUID = g_guidCdrom;
	DeviceInterfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA);
	
	//
	// 1.枚举设备
	//
	int i = 0;
	hDeviceInfo = SetupDiGetClassDevs(
		&InterfaceGUID, 
		NULL, 
		NULL, 
		DIGCF_DEVICEINTERFACE|DIGCF_PRESENT);
	if (hDeviceInfo == INVALID_HANDLE_VALUE)
	{
		return FALSE;
	}
	while (1)
	{
		bRet = SetupDiEnumDeviceInterfaces(hDeviceInfo, NULL, &InterfaceGUID, i++, &DeviceInterfaceData);
		if(!bRet) 
		{
			break;
		}
		// 下面需要得到Interface的详细信息
		// 先得到详细信息需要的内存大小
		bRet = SetupDiGetDeviceInterfaceDetail(hDeviceInfo, &DeviceInterfaceData, NULL, 0, &dwRequiredSize, NULL);
		if(bRet)  
		{
			break;
		}
		// 为详细信息分配内存
		if(pDeviceDetailData) 
		{
			free(pDeviceDetailData);
			pDeviceDetailData = NULL;
		}
		pDeviceDetailData = (PSP_INTERFACE_DEVICE_DETAIL_DATA)malloc(dwRequiredSize);
		if(pDeviceDetailData == NULL) 
		{
			break;
		}
		
		// 得到Interface的详细信息
		pDeviceDetailData->cbSize = sizeof(SP_INTERFACE_DEVICE_DETAIL_DATA);
		SP_DEVINFO_DATA DeviceInfoData;
		DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
		bRet = SetupDiGetDeviceInterfaceDetail(
			hDeviceInfo,
			&DeviceInterfaceData,
			pDeviceDetailData,
			dwRequiredSize,
			&dwRequiredSize,
			&DeviceInfoData);
		if ( !bRet ) 
		{
			break;
		}
		strcpy(acDevPath[nDevNum], pDeviceDetailData->DevicePath);
		nDevNum++;
	}
	if(pDeviceDetailData) 
	{
		free(pDeviceDetailData);
		pDeviceDetailData = NULL;
	}
	SetupDiDestroyDeviceInfoList(hDeviceInfo);
	return TRUE;
}

void CEnumDevDlg::OnButton1() 
{
	char acDevPath[MAX_PATH][MAX_PATH];
	int nDevNum = 0;
	EnumUSBDev(acDevPath, nDevNum);
	for (int i = 0; i < nDevNum; i++)
	{
		MessageBox(acDevPath[i], NULL, MB_OK);
	}
}