学步园

开发语言：C/C++

支持平台：Windows

实现功能：

• 通过WMI获取网卡MAC地址、硬盘序列号、主板序列号、CPU ID、BIOS序列号

下载地址：

WMI_DeviceQuery.zip

版本历史：

V1.4 2010年05月17日  

• 修正了硬盘序列号处理中的错误。现在和EVEREST Ultimate Edition 5.5一致。 

V1.3 2010年05月11日  

• 增加了对网卡原生MAC地址的查询。

V1.2 2010年05月05日

• 增加对硬盘序列号的进一步处理。

V1.1 2010年04月30日

• 修正微软MSDN例子错误，并增加对虚拟网卡的判断。

V1.0 2010年04月27日

• 完成正式版本。

接口函数：

WMI_DeviceQuery 

接口文件：WMI_DeviceQuery.h

 /* ----------------------------------------------------------<br /> 文件名称：WMI_DeviceQuery.h</p> <p>作者：秦建辉</p> <p>MSN：splashcn@msn.com</p> <p>版本历史：<br /> V1.4 2010年05月17日<br /> 修正了硬盘序列号处理中的错误。现在和EVEREST Ultimate Edition 5.5一致。</p> <p> V1.3 2010年05月11日<br /> 增加了对网卡原生MAC地址的查询。</p> <p> V1.2 2010年05月05日<br /> 增加对硬盘序列号的进一步处理。</p> <p> V1.1 2010年04月30日<br /> 修正微软MSDN例子错误，并增加对虚拟机网卡的判断。</p> <p> V1.0 2010年04月27日<br /> 完成正式版本。</p> <p>功能描述：<br /> 基于WMI获取设备属性：<br /> 0：网卡原生MAC地址<br /> 1：硬盘序列号<br /> 2：主板序列号<br /> 3：CPU ID<br /> 4：BIOS序列号<br /> 5：主板型号<br /> 6：网卡当前MAC地址</p> <p>接口函数：<br /> WMI_DeviceQuery<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 WMI_QUERY_TYPENUM 7 // WMI查询支持的类型数</p> <p>#ifdef __cplusplus<br /> extern "C"<br /> {<br /> #endif</p> <p>/*<br /> 功能：通过WMI获取设备属性<br /> 参数说明：<br /> iQueryType：需要查询的设备属性<br /> 0：网卡原生MAC地址<br /> 1：硬盘序列号<br /> 2：主板序列号<br /> 3：CPU ID<br /> 4：BIOS序列号<br /> 5：主板型号<br /> 6：网卡当前MAC地址<br /> properties：存储设备属性值<br /> iSize：可存储的最大设备个数<br /> 返回值：<br /> -1：不支持的设备属性值<br /> -2：WMI连接失败<br /> -3：不正确的WQL查询语句<br /> >=0：获取的设备个数<br /> */<br /> INT WMI_DeviceQuery( INT iQueryType, T_DEVICE_PROPERTY *properties, INT iSize );</p> <p>#ifdef __cplusplus<br /> }<br /> #endif<br />   

实现文件：WMI_DeviceQuery.cpp

#include "WMI_DeviceQuery.h"<br /> #include <comutil.h><br /> #include <Wbemidl.h><br /> #include <tchar.h><br /> #include <strsafe.h><br /> #include <algorithm><br /> #include <atlconv.h><br /> #include <ntddndis.h></p> <p>#pragma comment (lib, "comsuppw.lib")<br /> #pragma comment (lib, "wbemuuid.lib")</p> <p>typedef struct _T_WQL_QUERY<br /> {<br /> CHAR* szSelect; // SELECT语句<br /> WCHAR* szProperty; // 属性字段<br /> } T_WQL_QUERY;</p> <p>// WQL查询语句<br /> const T_WQL_QUERY szWQLQuery[] = {<br /> // 网卡原生MAC地址<br /> "SELECT * FROM Win32_NetworkAdapter WHERE (MACAddress IS NOT NULL) AND (NOT (PNPDeviceID LIKE 'ROOT%'))",<br /> L"PNPDeviceID",</p> <p> // 硬盘序列号<br /> "SELECT * FROM Win32_DiskDrive WHERE (SerialNumber IS NOT NULL) AND (MediaType LIKE 'Fixed hard disk%')",<br /> L"SerialNumber",</p> <p> // 主板序列号<br /> "SELECT * FROM Win32_BaseBoard WHERE (SerialNumber IS NOT NULL)",<br /> L"SerialNumber", </p> <p> // 处理器ID<br /> "SELECT * FROM Win32_Processor WHERE (ProcessorId IS NOT NULL)",<br /> L"ProcessorId",</p> <p> // BIOS序列号<br /> "SELECT * FROM Win32_BIOS WHERE (SerialNumber IS NOT NULL)",<br /> L"SerialNumber",</p> <p> // 主板型号<br /> "SELECT * FROM Win32_BaseBoard WHERE (Product IS NOT NULL)",<br /> L"Product",</p> <p> // 网卡当前MAC地址<br /> "SELECT * FROM Win32_NetworkAdapter WHERE (MACAddress IS NOT NULL) AND (NOT (PNPDeviceID LIKE 'ROOT%'))",<br /> L"MACAddress",<br /> };</p> <p>// 通过“PNPDeviceID”获取网卡原生MAC地址<br /> static BOOL WMI_DoWithPNPDeviceID( const TCHAR *PNPDeviceID, TCHAR *MacAddress, UINT uSize )<br /> {<br /> TCHAR DevicePath[MAX_PATH];<br /> HANDLE hDeviceFile;<br /> BOOL isOK = FALSE;</p> <p> // 生成设备路径名<br /> StringCchCopy( DevicePath, MAX_PATH, TEXT("////.//") );<br /> StringCchCat( DevicePath, MAX_PATH, PNPDeviceID );<br /> StringCchCat( DevicePath, MAX_PATH, TEXT("#{ad498944-762f-11d0-8dcb-00c04fc3358c}") );</p> <p> // 将“PNPDeviceID”中的“/”替换成“#”，以获得真正的设备路径名<br /> std::replace( DevicePath + 4, DevicePath + 4 + _tcslen(PNPDeviceID), TEXT('//'), TEXT('#') ); </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 /> { // 将字节数组转换成16进制字符串<br /> for( DWORD i = 0; i < dwByteRet; i++ )<br /> {<br /> StringCchPrintf( MacAddress + (i << 1), uSize - (i << 1), TEXT("%02X"), ucData[i] );<br /> }</p> <p> MacAddress[dwByteRet << 1] = TEXT('/0'); // 写入字符串结束标记<br /> }</p> <p> CloseHandle( hDeviceFile );<br /> }</p> <p> return isOK;<br /> }</p> <p>static BOOL WMI_DoWithHarddiskSerialNumber( TCHAR *SerialNumber, UINT uSize )<br /> {<br /> UINT iLen;<br /> UINT i;</p> <p> iLen = _tcslen( SerialNumber );<br /> if( iLen == 40 ) // InterfaceType = "IDE"<br /> { // 需要将16进制编码串转换为字符串<br /> TCHAR ch, szBuf[32];<br /> BYTE b; </p> <p> for( i = 0; i < 20; i++ )<br /> { // 将16进制字符转换为高4位<br /> ch = SerialNumber[i * 2];<br /> if( (ch >= '0') && (ch <= '9') )<br /> {<br /> b = ch - '0';<br /> }<br /> else if( (ch >= 'A') && (ch <= 'F') )<br /> {<br /> b = ch - 'A' + 10;<br /> }<br /> else if( (ch >= 'a') && (ch <= 'f') )<br /> {<br /> b = ch - 'a' + 10;<br /> }<br /> else<br /> { // 非法字符<br /> break;<br /> }</p> <p> b <<= 4;</p> <p> // 将16进制字符转换为低4位<br /> ch = SerialNumber[i * 2 + 1];<br /> if( (ch >= '0') && (ch <= '9') )<br /> {<br /> b += ch - '0';<br /> }<br /> else if( (ch >= 'A') && (ch <= 'F') )<br /> {<br /> b += ch - 'A' + 10;<br /> }<br /> else if( (ch >= 'a') && (ch <= 'f') )<br /> {<br /> b += ch - 'a' + 10;<br /> }<br /> else<br /> { // 非法字符<br /> break;<br /> }</p> <p> szBuf[i] = b;<br /> }</p> <p> if( i == 20 )<br /> { // 转换成功<br /> szBuf[i] = L'/0';<br /> StringCchCopy( SerialNumber, uSize, szBuf );<br /> iLen = _tcslen( SerialNumber );<br /> }<br /> }</p> <p> // 每2个字符互换位置<br /> for( i = 0; i < iLen; i += 2 )<br /> {<br /> std::swap( SerialNumber[i], SerialNumber[i+1] );<br /> }</p> <p> // 去掉空格<br /> std::remove( SerialNumber, SerialNumber + _tcslen(SerialNumber) + 1, L' ' );</p> <p> return TRUE;<br /> }</p> <p>static BOOL WMI_DoWithProperty( INT iQueryType, TCHAR *szProperty, UINT uSize )<br /> {<br /> BOOL isOK = TRUE;</p> <p> switch( iQueryType )<br /> {<br /> case 0: // 网卡原生MAC地址<br /> isOK = WMI_DoWithPNPDeviceID( szProperty, szProperty, uSize );<br /> break;</p> <p> case 1: // 硬盘序列号<br /> isOK = WMI_DoWithHarddiskSerialNumber( szProperty, uSize );<br /> break;</p> <p> case 6: // 网卡当前MAC地址<br /> // 去掉冒号<br /> std::remove( szProperty, szProperty + _tcslen(szProperty) + 1, L':' );<br /> break;</p> <p> default:<br /> // 去掉空格<br /> std::remove( szProperty, szProperty + _tcslen(szProperty) + 1, L' ' );<br /> }</p> <p> return isOK;<br /> }</p> <p>// 基于Windows Management Instrumentation（Windows管理规范）<br /> INT WMI_DeviceQuery( INT iQueryType, T_DEVICE_PROPERTY *properties, INT iSize )<br /> {<br /> HRESULT hres;<br /> INT iTotal = 0;</p> <p> // 判断查询类型是否支持<br /> if( (iQueryType < 0) || (iQueryType >= sizeof(szWQLQuery)/sizeof(T_WQL_QUERY)) )<br /> {<br /> return -1; // 查询类型不支持<br /> }</p> <p> // 初始化COM<br /> hres = CoInitializeEx( NULL, COINIT_MULTITHREADED );<br /> if( FAILED(hres) )<br /> {<br /> return -2;<br /> }</p> <p> // 设置COM的安全认证级别<br /> hres = CoInitializeSecurity(<br /> NULL,<br /> -1,<br /> NULL,<br /> NULL,<br /> RPC_C_AUTHN_LEVEL_DEFAULT,<br /> RPC_C_IMP_LEVEL_IMPERSONATE,<br /> NULL,<br /> EOAC_NONE,<br /> NULL<br /> );<br /> if( FAILED(hres) )<br /> {<br /> CoUninitialize();<br /> return -2;<br /> }</p> <p> // 获得WMI连接COM接口<br /> IWbemLocator *pLoc = NULL;<br /> hres = CoCreateInstance(<br /> CLSID_WbemLocator,<br /> NULL,<br /> CLSCTX_INPROC_SERVER,<br /> IID_IWbemLocator,<br /> reinterpret_cast<LPVOID*>(&pLoc)<br /> );<br /> if( FAILED(hres) )<br /> {<br /> CoUninitialize();<br /> return -2;<br /> }</p> <p> // 通过连接接口连接WMI的内核对象名"ROOT//CIMV2"<br /> IWbemServices *pSvc = NULL;<br /> hres = pLoc->ConnectServer(<br /> _bstr_t( L"ROOT//CIMV2" ),<br /> NULL,<br /> NULL,<br /> NULL,<br /> 0,<br /> NULL,<br /> NULL,<br /> &pSvc<br /> );<br /> if( FAILED(hres) )<br /> {<br /> pLoc->Release();<br /> CoUninitialize();<br /> return -2;<br /> }</p> <p> // 设置请求代理的安全级别<br /> hres = CoSetProxyBlanket(<br /> pSvc,<br /> RPC_C_AUTHN_WINNT,<br /> RPC_C_AUTHZ_NONE,<br /> NULL,<br /> RPC_C_AUTHN_LEVEL_CALL,<br /> RPC_C_IMP_LEVEL_IMPERSONATE,<br /> NULL,<br /> EOAC_NONE<br /> );<br /> if( FAILED(hres) )<br /> {<br /> pSvc->Release();<br /> pLoc->Release();<br /> CoUninitialize();<br /> return -2;<br /> }</p> <p> // 通过请求代理来向WMI发送请求<br /> IEnumWbemClassObject *pEnumerator = NULL;<br /> hres = pSvc->ExecQuery(<br /> bstr_t("WQL"),<br /> bstr_t( szWQLQuery[iQueryType].szSelect ),<br /> WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,<br /> NULL,<br /> &pEnumerator<br /> );<br /> if( FAILED(hres) )<br /> {<br /> pSvc->Release();<br /> pLoc->Release();<br /> CoUninitialize();<br /> return -3;<br /> }</p> <p> // 循环枚举所有的结果对象<br /> while( pEnumerator )<br /> {<br /> IWbemClassObject *pclsObj = NULL;<br /> ULONG uReturn = 0;</p> <p> if( (properties != NULL) && (iTotal >= iSize) )<br /> {<br /> break;<br /> }</p> <p> pEnumerator->Next(<br /> WBEM_INFINITE,<br /> 1,<br /> &pclsObj,<br /> &uReturn<br /> );</p> <p> if( uReturn == 0 )<br /> {<br /> break;<br /> }</p> <p> if( properties != NULL )<br /> { // 获取属性值<br /> VARIANT vtProperty;</p> <p> VariantInit( &vtProperty );<br /> pclsObj->Get( szWQLQuery[iQueryType].szProperty, 0, &vtProperty, NULL, NULL );<br /> StringCchCopy( properties[iTotal].szProperty, PROPERTY_MAX_LEN, W2T(vtProperty.bstrVal) );<br /> VariantClear( &vtProperty );</p> <p> // 对属性值做进一步的处理<br /> if( WMI_DoWithProperty( iQueryType, properties[iTotal].szProperty, PROPERTY_MAX_LEN ) )<br /> {<br /> iTotal++;<br /> }<br /> }<br /> else<br /> {<br /> iTotal++;<br /> }</p> <p> pclsObj->Release();<br /> } // End While</p> <p> // 释放资源<br /> pEnumerator->Release();<br /> pSvc->Release();<br /> pLoc->Release();<br /> CoUninitialize();</p> <p> return iTotal;<br /> }<br />