用 Windows API “GetAdaptersInfo” 获取 MAC 时遇到的问题

        前段时间有个项目需要获取客户端的 MAC 地址，用作统计去重的参考数据。从网上查到的获取 MAC 地址的代码，大多是用同一段代码修改的。于是我也用了那段代码。代码如下：

void GetMAC(BYTE mac[BUF_SIZE])
{
	ULONG size_pointer;
	PIP_ADAPTER_INFO pip_adapter_info = NULL;

	if(ERROR_BUFFER_OVERFLOW != GetAdaptersInfo(NULL, &size_pointer))
	{
		wsprintfA((LPSTR)mac, "GetMAC Failed! ErrorCode: %d", GetLastError());
		return;
	}

	pip_adapter_info = (PIP_ADAPTER_INFO)calloc(size_pointer, sizeof(BYTE));
	if(NULL == pip_adapter_info)
	{
		lstrcpyA((LPSTR)mac, "GetMAC Failed! Because calloc failed!");
		return;
	}

	if(ERROR_SUCCESS != GetAdaptersInfo(pip_adapter_info, &size_pointer))
	{
		wsprintfA((LPSTR)mac, "GetMAC Failed! ErrorCode: %d", GetLastError());
		free(pip_adapter_info);
		return;
	}

	for(int i = 0; i < 6; ++i)
	{
		wsprintfA((LPSTR)mac, "%02X", pip_adapter_info->Address[i]);
		mac += 2;
	}
}

        在自己的电脑上、虚拟机上测试了一番，没有发现问题，觉得一切 OK 之后将产品发布出去，结果发现许多机器返回的 MAC 为 NULL！！！

        问题严重，又重新研究了一番，发现之所以 MAC 返回 NULL 是因为这段代码并不完整：用 GetAdaptersInfo 获取到的不一定是网卡信息，有可能是别的适配器信息，不是网卡就没有 MAC，结果当然为 NULL！花了点时间把代码完善了一下，加了个检测适配器特性的函数，完成了该功能，最终的代码如下：

#include <windows.h>
#include <iphlpapi.h>       // API GetAdaptersInfo 头文件
#include <shlwapi.h>        // API StrCmpIA 头文件
#pragma comment(lib, "iphlpapi.lib")
#pragma comment(lib, "shlwapi.lib")
#include <Strsafe.h>        // API StringCbPrintfA 头文件
#include <shellapi.h>       // API lstrcpyA 头文件

//
// 功能：获取适配器特性
// 参数： 
//   adapter_name 适配器 ID
// 返回值：成功则返回由参数指定的适配器的特性标志，是一个 DWORD 值，失败返回 0
//
UINT GetAdapterCharacteristics(char* adapter_name)
{
    if(adapter_name == NULL || adapter_name[0] == 0)
        return 0;

    HKEY root = NULL;
    // 打开存储适配器信息的注册表根键
    if(ERROR_SUCCESS != RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Control\\Class\\{4D36E972-E325-11CE-BFC1-08002BE10318}", 0, KEY_READ, &root))
        return 0;

    DWORD subkeys = 0;
    // 获取该键下的子键数
    if(ERROR_SUCCESS != RegQueryInfoKeyA(root, NULL, NULL, NULL, &subkeys, NULL, NULL, NULL, NULL, NULL, NULL, NULL))
        subkeys = 100;

    DWORD ret_value = 0;
    for(DWORD i = 0; i < subkeys; i++)
    {
        // 每个适配器用一个子键存储，子键名为从 0 开始的 4 位数
        char subkey[MAX_SIZE];
        memset(subkey, 0, MAX_SIZE);
        StringCbPrintfA(subkey, MAX_SIZE, "%04u", i);

        // 打开该子键
        HKEY hKey = NULL;
        if(ERROR_SUCCESS != RegOpenKeyExA(root, subkey, 0, KEY_READ, &hKey))
            continue;

        // 获取该子键对应的适配器 ID，存于 name 中
        char name[MAX_PATH];
        DWORD type = 0;
        DWORD size = MAX_PATH;
        if(ERROR_SUCCESS != RegQueryValueExA(hKey, "NetCfgInstanceId", NULL, &type, (LPBYTE)name, &size))
        {
            RegCloseKey(hKey);
            continue;
        }

        // 对比该适配器 ID 是不是要获取特性的适配器 ID
        if(StrCmpIA(name, adapter_name) != 0)
        {
            RegCloseKey(hKey);
            continue;
        }

        // 读取该适配器的特性标志，该标志存储于值 Characteristics 中
        DWORD val = 0;
        size = 4;
        LSTATUS ls = RegQueryValueExA(hKey, "Characteristics", NULL, &type, (LPBYTE)&val, &size);
        RegCloseKey(hKey);

        if(ERROR_SUCCESS == ls)
        {
            ret_value = val;
            break;
        }
    }

    RegCloseKey(root);
    return ret_value;
}

//
// 功能：获取 Mac 地址的二进制数据
// 参数：
//   mac 用于输出 Mac 地址的二进制数据的缓冲区指针
// 返回值：成功返回 mac 地址的长度，失败返回 0，失败时 mac 中保存一些简单的错误信息，可适当修改，用于调试
//
int GetMAC(BYTE mac[BUF_SIZE])
{
#define NCF_PHYSICAL 0x4
    DWORD AdapterInfoSize = 0;
    if(ERROR_BUFFER_OVERFLOW != GetAdaptersInfo(NULL, &AdapterInfoSize))
    {
        StringCbPrintfA((LPSTR)mac, BUF_SIZE, "GetMAC Failed! ErrorCode: %d", GetLastError());
        return 0;
    }

    void* buffer = malloc(AdapterInfoSize);
    if(buffer == NULL)
    {
        lstrcpyA((LPSTR)mac, "GetMAC Failed! Because malloc failed!");
        return 0;
    }

    PIP_ADAPTER_INFO pAdapt = (PIP_ADAPTER_INFO)buffer;
    if(ERROR_SUCCESS != GetAdaptersInfo(pAdapt, &AdapterInfoSize))
    {
        StringCbPrintfA((LPSTR)mac, BUF_SIZE, "GetMAC Failed! ErrorCode: %d", GetLastError());
        free(buffer);
        return 0;
    }

    int mac_length = 0;
    while(pAdapt)
    {
        if(pAdapt->AddressLength >= 6 && pAdapt->AddressLength <= 8)
        {
            memcpy(mac, pAdapt->Address, pAdapt->AddressLength);
            mac_length = pAdapt->AddressLength;

            UINT flag = GetAdapterCharacteristics(pAdapt->AdapterName);
            bool is_physical = ((flag & NCF_PHYSICAL) == NCF_PHYSICAL);
            if(is_physical)
                break;
        }
        pAdapt = pAdapt->Next;
    }
    free(buffer);
    return mac_length;
}

//
// 功能：获取 Mac 地址，使用时直接调用此函数即可
// 参数：
//   mac 用于存储 Mac 地址的缓冲区指针
// 返回值：无返回值，函数执行完后会把 Mac 地址以16进制的形式存于参数指定的缓冲区中，若有错误，缓冲区中保存的是错误信息
//
void GetMacAddress( char* mac )
{
    BYTE buf[BUF_SIZE];
    memset(buf, 0, BUF_SIZE);

    int len = GetMAC(buf);
    if(len <= 0)
    {
        lstrcpyA(mac, (LPCSTR)buf);
        return;
    }

    if(len == 6)
        StringCbPrintfA(mac, BUF_SIZE, "%02X-%02X-%02X-%02X-%02X-%02X", buf[0], buf[1], buf[2], buf[3], buf[4], buf[5]);
    else
        StringCbPrintfA(mac, BUF_SIZE, "%02X-%02X-%02X-%02X-%02X-%02X-%02X-%02X", buf[0], buf[1], buf[2], buf[3], buf[4], buf[5], buf[6], buf[7]);
}

        编译环境： VS2008 + Windows SDK 7.1

        函数功能在 Windows 2000、Windows XP、Windows 2003、Vista、 Win7 32位和 Win7 64 位下均测试通过。