C++ 实现Windows系统设备枚举

1.枚举系统设备

HDEVINFO hDevInfo;

SP_DEVINFO_DATA DeviceInfoData;
DWORD dwDevID;

// 得到所有设备 HDEVINFO 
hDevInfo = SetupDiGetClassDevs(NULL, 0, 0, DIGCF_PRESENT | DIGCF_ALLCLASSES); //根据此处参数不同可分类枚举设备，这里枚举了所有设备
if (hDevInfo == INVALID_HANDLE_VALUE)
{
return FALSE;
}
DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
for (dwDevID = 0; SetupDiEnumDeviceInfo(hDevInfo, dwDevID, &DeviceInfoData); dwDevID++)
{
DWORD DataT;
LPTSTR buffer = NULL;
DWORD buffersize = 0;
while (!SetupDiGetDeviceRegistryProperty(
hDevInfo,
&DeviceInfoData,
SPDRP_DEVICEDESC,
&DataT,
(PBYTE)buffer,//设备名
buffersize,
&buffersize))
{
if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
{
// 内存不足 
if (buffer)
HeapFree(GetProcessHeap(), 0, buffer);
buffer = (LPTSTR)HeapAlloc(GetProcessHeap(), 0, buffersize);
}
else
break;
}
TCHAR  strguid[512] = _T("");//设备GUID
_stprintf_s(strguid, _T("%x-%x-%x-%x%x%x%x%x%x%x%x "),
DeviceInfoData.ClassGuid.Data1,
DeviceInfoData.ClassGuid.Data2,
DeviceInfoData.ClassGuid.Data3,
DeviceInfoData.ClassGuid.Data4[0],
DeviceInfoData.ClassGuid.Data4[1],
DeviceInfoData.ClassGuid.Data4[2],
DeviceInfoData.ClassGuid.Data4[3],
DeviceInfoData.ClassGuid.Data4[4],
DeviceInfoData.ClassGuid.Data4[5],
DeviceInfoData.ClassGuid.Data4[6],
DeviceInfoData.ClassGuid.Data4[7]);
deviceinfo.DevID = dwDevID;//设备ID
deviceinfo.bDisabled = IsDisabled(dwDevID, hDevInfo);//是否已禁用
deviceinfo.bDisableable = IsDisableable(dwDevID, hDevInfo);//是否可禁用
wcscpy_s(deviceinfo.DeviceName, buffer);,//设备名
if (buffer)
HeapFree(GetProcessHeap(), 0, buffer);
}
if (GetLastError() != NO_ERROR &&
GetLastError() != ERROR_NO_MORE_ITEMS)
{
return FALSE;
}
SetupDiDestroyDeviceInfoList(hDevInfo);

return TRUE;

//根据设备ID判断指定设备是否可禁用

BOOL IsDisableable(DWORD dwDevID, HDEVINFO hDevInfo)
{
SP_DEVINFO_DATA DevInfoData = { sizeof(SP_DEVINFO_DATA) };
DWORD    dwDevStatus, dwProblem;
if (!SetupDiEnumDeviceInfo(hDevInfo, dwDevID, &DevInfoData))
{
OutputDebugString(_T("SetupDiEnumDeviceInfo FAILED"));
return FALSE;
}
//查询设备状态
if (CM_Get_DevNode_Status(&dwDevStatus, &dwProblem, DevInfoData.DevInst, 0) != CR_SUCCESS)
{
OutputDebugString(_T("CM_GET_DevNode_Status FAILED"));
return FALSE;
}
return ((dwDevStatus & DN_DISABLEABLE) && (dwProblem != CM_PROB_HARDWARE_DISABLED));
}

//根据设备ID判断指定设备是否已禁用
BOOL IsDisabled(DWORD dwDevID, HDEVINFO hDevInfo)
{
SP_DEVINFO_DATA DevInfoData = { sizeof(SP_DEVINFO_DATA) };
DWORD    dwDevStatus, dwProblem;
if (!SetupDiEnumDeviceInfo(hDevInfo, dwDevID, &DevInfoData))
{
OutputDebugString(_T("SetupDiEnumDeviceInfo FAILED"));
return FALSE;
}


//查询设备状态
if (CM_Get_DevNode_Status(&dwDevStatus, &dwProblem, DevInfoData.DevInst, 0) != CR_SUCCESS)
{
OutputDebugString(_T("CM_GET_DevNode_Status FAILED"));
return FALSE;
}
return ((dwDevStatus & DN_HAS_PROBLEM) && (dwProblem == CM_PROB_DISABLED));
}

//根据设备ID以及dwNewState：DICS_ENABLE，DICS_DISABLE两个参数来启用，禁用设备
BOOL StateChange(DWORD dwNewState, DWORD dwDevID, HDEVINFO hDevInfo)
{
SP_PROPCHANGE_PARAMS PropChangeParams;
SP_DEVINFO_DATA        DevInfoData = { sizeof(SP_DEVINFO_DATA) };
SP_DEVINSTALL_PARAMS devParams;


//查询设备信息
if (!SetupDiEnumDeviceInfo(hDevInfo, dwDevID, &DevInfoData))
{
OutputDebugString(_T("SetupDiEnumDeviceInfo FAILED"));
return FALSE;
}


//设置设备属性变化参数
PropChangeParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
PropChangeParams.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
PropChangeParams.Scope = DICS_FLAG_GLOBAL; //使修改的属性保存在所有的硬件属性文件
PropChangeParams.StateChange = dwNewState;
PropChangeParams.HwProfile = 0;


//改变设备属性
if (!SetupDiSetClassInstallParams(hDevInfo,
&DevInfoData,
(SP_CLASSINSTALL_HEADER *)&PropChangeParams,
sizeof(PropChangeParams)))
{
OutputDebugString(_T("SetupDiSetClassInstallParams FAILED"));
return FALSE;
}


PropChangeParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
PropChangeParams.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
PropChangeParams.Scope = DICS_FLAG_CONFIGSPECIFIC;//使修改的属性保存在指定的属性文件
PropChangeParams.StateChange = dwNewState;
PropChangeParams.HwProfile = 0;


//改变设备属性并调用安装服务
if (!SetupDiSetClassInstallParams(hDevInfo,
&DevInfoData,
(SP_CLASSINSTALL_HEADER *)&PropChangeParams,
sizeof(PropChangeParams)) ||
!SetupDiCallClassInstaller(DIF_PROPERTYCHANGE, hDevInfo, &DevInfoData))
{
OutputDebugString(_T("SetupDiSetClassInstallParams or SetupDiCallClassInstaller FAILED"));
return TRUE;
}
else
{
//判断是否需要重新启动
devParams.cbSize = sizeof(devParams);
if (!SetupDiGetDeviceInstallParams(hDevInfo, &DevInfoData, &devParams))
{
OutputDebugString(_T("SetupDiGetDeviceInstallParams FAILED"));
return FALSE;
}


if (devParams.Flags & (DI_NEEDRESTART | DI_NEEDREBOOT))
{
OutputDebugString(_T("Need Restart Computer"));
return TRUE;
}
return TRUE;
}




}