学步园

开发语言：C/C++

开发工具：

• Visual Studio 2010
• Windows Driver Kit Version 7600.16385.1

支持平台：Windows

实现功能：

• 通过WDK获取设备的VID-PID、网卡原生MAC地址、硬盘序列号

下载地址：

WDK_DeviceQuery.zip

版本历史：

V1.3 2010年05月22日

• 增加获取硬盘序列号（需要系统管理员权限）。

V1.2 2010年05月20日

• 更新接口函数。
• 增加获取网卡原生MAC地址。

V1.1 2010年04月27日

• 优化代码。

V1.0 2010年04月15日

• 完成正式版本。获取设备的VID-PID。

接口函数：

WDK_DeviceQuery_Property

WDK_DeviceQuery_VIDPID

接口文件：WDK_DeviceQuery.h

 /* ----------------------------------------------------------<br /> 文件名称：WDK_DeviceQuery.h</p> <p>作者：秦建辉</p> <p>MSN：splashcn@msn.com</p> <p>版本历史：<br /> V1.3 2010年05月22日<br /> 增加获取硬盘序列号（需要系统管理员权限）。</p> <p> V1.2 2010年05月20日<br /> 更新接口函数。<br /> 增加获取网卡原生MAC地址。</p> <p> V1.1 2010年04月27日<br /> 优化代码。</p> <p> V1.0 2010年04月15日<br /> 完成正式版本。获取设备的VID-PID。</p> <p>功能描述：<br /> 基于WDK获取设备属性值。</p> <p>接口函数：<br /> WDK_DeviceQuery_Property<br /> WDK_DeviceQuery_VIDPID<br /> ------------------------------------------------------------ */<br /> #pragma once</p> <p>#include <windows.h></p> <p>#ifndef MACRO_T_DEVICE_PROPERTY<br /> #define MACRO_T_DEVICE_PROPERTY</p> <p> #define PROPERTY_MAX_LEN 128 // 属性字段最大长度<br /> typedef struct _T_DEVICE_PROPERTY<br /> {<br /> TCHAR szProperty[PROPERTY_MAX_LEN];<br /> } T_DEVICE_PROPERTY;<br /> #endif</p> <p>#define WDK_QUERY_TYPENUM 4 // WDK查询支持的类型数</p> <p>#ifndef MACRO_HIDD_VIDPID<br /> #define MACRO_HIDD_VIDPID</p> <p> typedef struct _HIDD_VIDPID<br /> {<br /> USHORT VendorID;<br /> USHORT ProductID;<br /> } HIDD_VIDPID;<br /> #endif</p> <p>#ifdef __cplusplus<br /> extern "C"<br /> {<br /> #endif</p> <p>/*<br /> 功能：基于WDK获取设备属性值<br /> 参数说明：<br /> iQueryType：需要查询的设备属性<br /> 0：人机交互设备的VID-PID<br /> 1：网卡原生MAC地址（包含USB网卡）<br /> 2：网卡原生MAC地址（剔除USB网卡）<br /> 3：硬盘序列号（需要系统管理员权限）<br /> properties：存储设备属性值<br /> iSize：可存储的最大设备数<br /> 返回值：<br /> -1：获取设备信息表失败<br /> -2：不支持的查询类型<br /> >=0：实际获取的设备数<br /> */<br /> INT WDK_DeviceQuery_Property( INT iQueryType, T_DEVICE_PROPERTY *properties, INT iSize );</p> <p>/*<br /> 功能：获取指定设备接口类相关设备的VID和PID<br /> 参数说明：<br /> pGuid：指向设备接口类GUID<br /> pVIDPID：存储设备的VID和PID<br /> iSize：可存储的最大设备数<br /> 返回值：<br /> -1：获取设备信息表失败<br /> >=0：实际获取的设备数<br /> */<br /> INT WDK_DeviceQuery_VIDPID( const GUID* pGuid, HIDD_VIDPID* pVIDPID, INT iSize );</p> <p>#ifdef __cplusplus<br /> }<br /> #endif<br />

实现代码：WDK_DeviceQuery.cpp

#include "WDK_DeviceQuery.h"<br /> #include <tchar.h><br /> #include <strsafe.h><br /> #include <setupapi.h><br /> #include <ntddndis.h><br /> #include <ntdddisk.h><br /> #include <algorithm></p> <p>#ifdef __cplusplus<br /> extern "C"<br /> {<br /> #endif</p> <p>#include <hidsdi.h></p> <p>#ifdef __cplusplus<br /> }<br /> #endif</p> <p>#pragma comment (lib, "Setupapi.lib")<br /> #pragma comment (lib, "hid.lib")</p> <p>const GUID GUID_QUERYSET[] = {<br /> // 人机交互设备VID-PID<br /> {0x4D1E55B2, 0xF16F, 0x11CF, 0x88, 0xCB, 0x00, 0x11, 0x11, 0x00, 0x00, 0x30},</p> <p> // 网卡原生MAC地址（包含USB网卡）<br /> {0xAD498944, 0x762F, 0x11D0, 0x8D, 0xCB, 0x00, 0xC0, 0x4F, 0xC3, 0x35, 0x8C},</p> <p> // 网卡原生MAC地址（剔除USB网卡）<br /> {0xAD498944, 0x762F, 0x11D0, 0x8D, 0xCB, 0x00, 0xC0, 0x4F, 0xC3, 0x35, 0x8C},</p> <p> // 硬盘序列号<br /> {0x53F56307, 0xB6BF, 0x11D0, 0x94, 0xF2, 0x00, 0xA0, 0xC9, 0x1E, 0xFB, 0x8B},<br /> };</p> <p>// 获取设备的VID-PID<br /> static BOOL WDK_GetVIDPID( TCHAR* DevicePath, TCHAR* szProperty, UINT uSize )<br /> {<br /> HANDLE hDeviceFile;<br /> BOOL isOK = FALSE;</p> <p> // 获取设备句柄<br /> hDeviceFile = CreateFile( DevicePath,<br /> 0,<br /> FILE_SHARE_READ | FILE_SHARE_WRITE,<br /> NULL,<br /> OPEN_EXISTING,<br /> 0,<br /> NULL);</p> <p> if( hDeviceFile != INVALID_HANDLE_VALUE )<br /> { // 获取设备属性<br /> HIDD_ATTRIBUTES Attributes;</p> <p> Attributes.Size = sizeof(HIDD_ATTRIBUTES);<br /> isOK = HidD_GetAttributes( hDeviceFile, &Attributes );<br /> if( isOK )<br /> {<br /> if( szProperty != NULL )<br /> {<br /> StringCchPrintf( szProperty, uSize, TEXT("%04X-%04X"), Attributes.VendorID, Attributes.ProductID );<br /> }<br /> }</p> <p> CloseHandle( hDeviceFile );<br /> }</p> <p> return isOK;<br /> }</p> <p>// 获取网卡原生MAC地址<br /> static BOOL WDK_GetMacAddress( TCHAR* DevicePath, TCHAR* szProperty, UINT uSize, BOOL isIncludeUSB )<br /> {<br /> HANDLE hDeviceFile;<br /> BOOL isOK = FALSE;</p> <p> // 剔除虚拟网卡<br /> if( _tcsnicmp( DevicePath + 4, TEXT("root"), 4 ) == 0 )<br /> {<br /> return FALSE;<br /> }</p> <p> if( !isIncludeUSB )<br /> { // 剔除USB网卡<br /> if( _tcsnicmp( DevicePath + 4, TEXT("usb"), 4 ) == 0 )<br /> {<br /> return FALSE;<br /> }<br /> }</p> <p> // 获取设备句柄<br /> hDeviceFile = CreateFile( DevicePath,<br /> 0,<br /> FILE_SHARE_READ | FILE_SHARE_WRITE,<br /> NULL,<br /> OPEN_EXISTING,<br /> 0,<br /> NULL);</p> <p> if( hDeviceFile != INVALID_HANDLE_VALUE )<br /> {<br /> ULONG dwID;<br /> BYTE ucData[8];<br /> DWORD dwByteRet; </p> <p> // 获取原生MAC地址<br /> dwID = OID_802_3_PERMANENT_ADDRESS;<br /> isOK = DeviceIoControl( hDeviceFile, IOCTL_NDIS_QUERY_GLOBAL_STATS, &dwID, sizeof(dwID), ucData, sizeof(ucData), &dwByteRet, NULL );<br /> if( isOK )<br /> {<br /> if( szProperty != NULL )<br /> { // 将字节数组转换成16进制字符串<br /> for( DWORD i = 0; i < dwByteRet; i++ )<br /> {<br /> StringCchPrintf( szProperty + (i << 1), uSize - (i << 1), TEXT("%02X"), ucData[i] );<br /> }<br /> }<br /> }</p> <p> CloseHandle( hDeviceFile );<br /> }</p> <p> return isOK;<br /> }</p> <p>// 获取硬盘序列号（需要系统管理员权限）<br /> static BOOL WDK_GetHarddiskSerialNumberWithAdminRights( TCHAR* DevicePath, TCHAR* szProperty, UINT uSize )<br /> {<br /> HANDLE hDeviceFile;<br /> BOOL isOK = FALSE;</p> <p> // 获取设备句柄<br /> hDeviceFile = CreateFile( DevicePath,<br /> GENERIC_READ | GENERIC_WRITE, // 注意：需要系统管理员权限才能执行读写操作<br /> FILE_SHARE_READ | FILE_SHARE_WRITE,<br /> NULL,<br /> OPEN_EXISTING,<br /> 0,<br /> NULL);</p> <p> if( hDeviceFile != INVALID_HANDLE_VALUE )<br /> {<br /> GETVERSIONINPARAMS VersionInParams;<br /> DWORD dwByteRet;</p> <p> // 获取硬盘属性<br /> isOK = DeviceIoControl( hDeviceFile, SMART_GET_VERSION, NULL, 0, &VersionInParams, sizeof(GETVERSIONINPARAMS), &dwByteRet, NULL );<br /> if( isOK )<br /> {<br /> isOK = FALSE;<br /> if( VersionInParams.fCapabilities & CAP_ATA_ID_CMD )<br /> {<br /> BYTE abSCIP[sizeof(SENDCMDINPARAMS) - 1] = {0};<br /> BYTE abSCOP[sizeof(SENDCMDOUTPARAMS) - 1 + 512] = {0};<br /> SENDCMDINPARAMS *scip = reinterpret_cast<SENDCMDINPARAMS*>(abSCIP);<br /> SENDCMDOUTPARAMS *scop = reinterpret_cast<SENDCMDOUTPARAMS*>(abSCOP);</p> <p> // 设置输入参数<br /> scip->irDriveRegs.bCommandReg = ID_CMD;</p> <p> // 设置输出参数<br /> scop->cBufferSize = 512;</p> <p> // 获取硬盘详细信息<br /> isOK = DeviceIoControl( hDeviceFile, SMART_RCV_DRIVE_DATA, scip, sizeof(abSCIP), scop, sizeof(abSCOP), &dwByteRet, NULL );<br /> if( isOK )<br /> {<br /> if( szProperty != NULL )<br /> { // 提取硬盘序列号<br /> for( UINT i = 0; i < 20; i += 2 )<br /> { // 颠倒高低位字节<br /> szProperty[i] = scop->bBuffer[i+21];<br /> szProperty[i + 1] = scop->bBuffer[i+20];<br /> }</p> <p> szProperty[20] = TEXT('/0');</p> <p> // 去掉空格<br /> std::remove( szProperty, szProperty + _tcslen(szProperty) + 1, L' ' );<br /> }<br /> }<br /> }<br /> }</p> <p> CloseHandle( hDeviceFile );<br /> }</p> <p> return isOK;<br /> }</p> <p>static BOOL WDK_GetProperty( TCHAR* DevicePath, INT iQueryType, TCHAR* szProperty, UINT uSize )<br /> {<br /> BOOL isOK = FALSE;</p> <p> switch( iQueryType )<br /> {<br /> case 0:<br /> // 获取设备的VID-PID<br /> isOK = WDK_GetVIDPID( DevicePath, szProperty, uSize );<br /> break;</p> <p> case 1: // 网卡原生MAC地址（包含USB网卡）<br /> isOK = WDK_GetMacAddress( DevicePath, szProperty, uSize, TRUE );<br /> break;</p> <p> case 2: // 网卡原生MAC地址（剔除USB网卡）<br /> isOK = WDK_GetMacAddress( DevicePath, szProperty, uSize, FALSE );<br /> break;</p> <p> case 3: // 硬盘序列号（需要系统管理员权限）<br /> isOK = WDK_GetHarddiskSerialNumberWithAdminRights( DevicePath, szProperty, uSize );<br /> break;</p> <p> default:<br /> break;<br /> }</p> <p> return isOK;<br /> }</p> <p>INT WDK_DeviceQuery_Property( INT iQueryType, T_DEVICE_PROPERTY *properties, INT iSize )<br /> {<br /> HDEVINFO hDevInfo;<br /> DWORD MemberIndex, RequiredSize;<br /> SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;<br /> PSP_DEVICE_INTERFACE_DETAIL_DATA DeviceInterfaceDetailData;<br /> INT iTotal = 0;</p> <p> // 判断查询类型是否支持<br /> if( (iQueryType < 0) || (iQueryType >= sizeof(GUID_QUERYSET)/sizeof(GUID)) )<br /> {<br /> return -2; // 查询类型不支持<br /> }</p> <p> // 获取设备信息集<br /> hDevInfo = SetupDiGetClassDevs( GUID_QUERYSET + iQueryType, NULL, NULL, DIGCF_PRESENT | DIGCF_INTERFACEDEVICE );<br /> if( hDevInfo == INVALID_HANDLE_VALUE )<br /> {<br /> return -1;<br /> }</p> <p> // 枚举设备信息集中所有设备<br /> DeviceInterfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA);<br /> for( MemberIndex = 0; ((properties == NULL) || (iTotal < iSize)); MemberIndex++ )<br /> { // 获取设备接口<br /> if( !SetupDiEnumDeviceInterfaces( hDevInfo, NULL, GUID_QUERYSET + iQueryType, MemberIndex, &DeviceInterfaceData ) )<br /> { // 设备枚举完毕<br /> break;<br /> }</p> <p> // 获取接收缓冲区大小，函数返回值为FALSE，GetLastError()=ERROR_INSUFFICIENT_BUFFER<br /> SetupDiGetDeviceInterfaceDetail( hDevInfo, &DeviceInterfaceData, NULL, 0, &RequiredSize, NULL );</p> <p> // 申请接收缓冲区<br /> DeviceInterfaceDetailData = (PSP_DEVICE_INTERFACE_DETAIL_DATA)malloc( RequiredSize );<br /> DeviceInterfaceDetailData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);</p> <p> // 获取设备细节信息<br /> if( SetupDiGetDeviceInterfaceDetail( hDevInfo, &DeviceInterfaceData, DeviceInterfaceDetailData, RequiredSize, NULL, NULL ) )<br /> {<br /> if( properties != NULL )<br /> {<br /> if( WDK_GetProperty( DeviceInterfaceDetailData->DevicePath, iQueryType, properties[iTotal].szProperty, PROPERTY_MAX_LEN ) )<br /> {<br /> if( iTotal > 0 )<br /> { // 剔除重复设备<br /> if( _tcscmp( properties[iTotal].szProperty, properties[iTotal - 1].szProperty ) != 0 )<br /> {<br /> iTotal++;<br /> }<br /> }<br /> else<br /> {<br /> iTotal++;<br /> }<br /> }<br /> }<br /> else<br /> {<br /> iTotal++;<br /> }<br /> }</p> <p> free( DeviceInterfaceDetailData );<br /> }</p> <p> SetupDiDestroyDeviceInfoList( hDevInfo );</p> <p> return iTotal;<br /> }</p> <p>INT WDK_DeviceQuery_VIDPID( const GUID* pGuid, HIDD_VIDPID* pVIDPID, INT iSize )<br /> {<br /> HDEVINFO hDevInfo;<br /> DWORD MemberIndex, RequiredSize;<br /> SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;<br /> PSP_DEVICE_INTERFACE_DETAIL_DATA DeviceInterfaceDetailData;<br /> INT iTotal = 0;</p> <p> // 获取设备信息集<br /> hDevInfo = SetupDiGetClassDevs( pGuid, NULL, NULL, DIGCF_PRESENT | DIGCF_INTERFACEDEVICE );<br /> if( hDevInfo == INVALID_HANDLE_VALUE )<br /> {<br /> return -1;<br /> }</p> <p> // 枚举设备信息集中所有设备<br /> DeviceInterfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA);<br /> for( MemberIndex = 0; ((pVIDPID == NULL) || (iTotal < iSize)); MemberIndex++ )<br /> { // 获取设备接口<br /> if( !SetupDiEnumDeviceInterfaces( hDevInfo, NULL, pGuid, MemberIndex, &DeviceInterfaceData ) )<br /> { // 设备枚举完毕<br /> break;<br /> }</p> <p> // 获取接收缓冲区大小，函数返回值为FALSE，GetLastError()=ERROR_INSUFFICIENT_BUFFER<br /> SetupDiGetDeviceInterfaceDetail( hDevInfo, &DeviceInterfaceData, NULL, 0, &RequiredSize, NULL );</p> <p> // 申请接收缓冲区<br /> DeviceInterfaceDetailData = (PSP_DEVICE_INTERFACE_DETAIL_DATA)malloc( RequiredSize );<br /> DeviceInterfaceDetailData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);</p> <p> // 获取设备细节信息<br /> if( SetupDiGetDeviceInterfaceDetail( hDevInfo, &DeviceInterfaceData, DeviceInterfaceDetailData, RequiredSize, NULL, NULL ) )<br /> { // 获取设备句柄<br /> HANDLE hDeviceFile; </p> <p> hDeviceFile = CreateFile( DeviceInterfaceDetailData->DevicePath,<br /> 0,<br /> FILE_SHARE_READ | FILE_SHARE_WRITE,<br /> NULL,<br /> OPEN_EXISTING,<br /> 0,<br /> NULL);</p> <p> if( hDeviceFile != INVALID_HANDLE_VALUE )<br /> { // 获取设备属性<br /> HIDD_ATTRIBUTES Attributes;</p> <p> Attributes.Size = sizeof(HIDD_ATTRIBUTES);<br /> if( HidD_GetAttributes( hDeviceFile, &Attributes ) )<br /> {<br /> if( pVIDPID != NULL )<br /> {<br /> pVIDPID[iTotal].VendorID = Attributes.VendorID;<br /> pVIDPID[iTotal].ProductID = Attributes.ProductID;<br /> if( iTotal > 0 )<br /> { // 剔除重复设备<br /> if( (pVIDPID[iTotal].VendorID != pVIDPID[iTotal - 1].VendorID) || (pVIDPID[iTotal].ProductID != pVIDPID[iTotal - 1].ProductID) )<br /> {<br /> iTotal++;<br /> }<br /> }<br /> else<br /> {<br /> iTotal++;<br /> }<br /> }<br /> else<br /> {<br /> iTotal++;<br /> }<br /> } </p> <p> CloseHandle( hDeviceFile );<br /> }<br /> }</p> <p> free( DeviceInterfaceDetailData );<br /> }</p> <p> SetupDiDestroyDeviceInfoList( hDevInfo );</p> <p> return iTotal;<br /> }<br />