VC++：结合设备安装类GUID和设备接口类GUID获取设备VIDPID

前言：

VID和PID常被用于厂家的软件加密，只有在系统中检测到某VID和PID的设备时，软件才能运行。因此获取某一类型设备或者全部设备的VID和PID集合至关重要。获取设备VID和PID的一般流程是通过设备接口类GUID创建设备信息集，然后从设备接口详细信息中获取设备路径，再调用HidD_GetAttributes从属性中读取VID和PID。该方法的缺点是需要事先知道设备接口类GUID，且每次只能获取一个设备接口类的VID和PID集合。本方法既可以指定设备安装类GUID和/或设备接口类GUID，也可以两者都不指定而直接创建设备信息集，然后通过设备信息数据获取设备实例ID，并直接从设备实例ID中提取出VID和PID，巧妙地避开了对设备IO的读写。

下载：

WDK_VidPidQuery.zip

源代码：

WDK_VidPidQuery.h

/* ----------------------------------------------------------
文件名称：WDK_VidPidQuery.h

作者：秦建辉

MSN：[email protected]
QQ：36748897

开发环境：
	Visual Studio V2010 
	WinDDK 7600.16385.1

版本历史：
	V1.0	2011年09月10日
			结合设备安装类GUID和设备接口类GUID获取设备VIDPID

接口函数：
	WDK_WhoAllVidPid
	WDK_isExistVidPid
------------------------------------------------------------ */
#pragma once

#include <windows.h>

#ifndef MACRO_HIDD_VIDPID
	#define MACRO_HIDD_VIDPID
	typedef struct _HIDD_VIDPID
	{
		USHORT	VendorID;
		USHORT	ProductID;
	} HIDD_VIDPID;
#endif

#ifdef __cplusplus
extern "C"
{
#endif

/*
功能：获取系统所有设备的VIDPID
入口参数：
	[out] pVidPid：存储返回的VIDPID，会自动过滤掉重复的VIDPID
	[in] iCapacity：存储单元的容量，建议为32
	[in] SetupClassGuid：设备安装类GUID，默认为NULL
	[in] InterfaceClassGuid：设备接口类GUID，默认为NULL
返回值：
	获取到的VID和PID数目
优点：
	直接通过设备实例ID提取VIDPID，从而无需获取设备路径来读写IO。
*/
INT WINAPI WDK_WhoAllVidPid( HIDD_VIDPID* pVidPid, INT iCapacity, const GUID* SetupClassGuid = NULL, const GUID* InterfaceClassGuid = NULL );

/*
功能：判断指定的VIDPID是否存在于列表中
入口参数：
	[in] VendorID：要查找的VID
	[in] ProductID：要查找的PID
	[in] pVidPid：用来检索的VIDPID库
	[in] iSize：VIDPID库大小
返回值：
	TRUE：存在
	FALSE：不存在
*/
BOOL WINAPI WDK_isExistVidPid( USHORT VendorID, USHORT ProductID, const HIDD_VIDPID* pVidPid, INT iSize );

#ifdef __cplusplus
}
#endif

WDK_VidPidQuery.cpp

#include "WDK_VidPidQuery.h"
#include <tchar.h>
#include <setupapi.h>

#pragma comment (lib, "Setupapi.lib")

#define DeviceInstanceIdSize 256	// 设备实例ID最大长度

// 获取系统的VID和PID集合
INT WINAPI WDK_WhoAllVidPid( HIDD_VIDPID* pVidPid, INT iCapacity, const GUID* SetupClassGuid, const GUID* InterfaceClassGuid )
{
	// 检测入口参数
	if (pVidPid == NULL || iCapacity <= 0) return 0;	

	// 根据设备安装类GUID创建空的设备信息集合
	HDEVINFO DeviceInfoSet = SetupDiCreateDeviceInfoList( SetupClassGuid, NULL );
	if (DeviceInfoSet == INVALID_HANDLE_VALUE) return -1;
	
	// 根据设备安装类GUID获取设备信息集合
	HDEVINFO hDevInfo;
	if(InterfaceClassGuid == NULL)
		hDevInfo = SetupDiGetClassDevsEx( NULL, NULL, NULL, DIGCF_ALLCLASSES | DIGCF_DEVICEINTERFACE | DIGCF_PRESENT, DeviceInfoSet, NULL, NULL );
	else
		hDevInfo = SetupDiGetClassDevsEx( InterfaceClassGuid, NULL, NULL, DIGCF_DEVICEINTERFACE | DIGCF_PRESENT, DeviceInfoSet, NULL, NULL );

	if (hDevInfo == INVALID_HANDLE_VALUE) return -1;

	// 存储得到的VID和PID数目
	INT iTotal = 0;

	// 存储设备实例ID	
	TCHAR DeviceInstanceId[DeviceInstanceIdSize];	

	// 存储设备信息数据
	SP_DEVINFO_DATA DeviceInfoData;					
	DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);

	// 获取设备信息数据
	DWORD DeviceIndex = 0;
	while (SetupDiEnumDeviceInfo( hDevInfo, DeviceIndex++, &DeviceInfoData))
	{
		// 获取设备实例ID
		if (SetupDiGetDeviceInstanceId(hDevInfo, &DeviceInfoData, DeviceInstanceId, DeviceInstanceIdSize, NULL))
		{
			// 从设备实例ID中提取VID和PID
			TCHAR* pVidIndex = _tcsstr(DeviceInstanceId, TEXT("VID_"));
			if (pVidIndex == NULL) continue;

			TCHAR* pPidIndex = _tcsstr(pVidIndex + 4, TEXT("PID_"));
			if (pPidIndex == NULL) continue;

			USHORT VendorID = _tcstoul(pVidIndex + 4, NULL, 16);
			USHORT ProductID = _tcstoul(pPidIndex + 4, NULL, 16);

			// 剔除重复的VID和PID
			if (!WDK_isExistVidPid( VendorID, ProductID, pVidPid, iTotal ))
			{
				pVidPid[iTotal].VendorID = VendorID;
				pVidPid[iTotal].ProductID = ProductID;
				if (++iTotal >= iCapacity) break;
			}
		}
	}

	return iTotal;
}

// 判断VID和PID是否已经存在
BOOL WINAPI WDK_isExistVidPid( USHORT VendorID, USHORT ProductID, const HIDD_VIDPID* pVidPid, INT iSize )
{
	if (pVidPid != NULL)
	{
		for (INT i = 0; i < iSize; i++)
		{
			if (pVidPid[i].VendorID == VendorID && pVidPid[i].ProductID == ProductID)
				return TRUE;
		}
	}

	return FALSE;
}