zoukankan      html  css  js  c++  java
  • 【转载】 用 Windows API “GetAdaptersInfo” 获取 MAC 时遇到的问题

      From:http://blog.csdn.net/weiyumingwww/article/details/17554461

     前段时间有个项目需要获取客户端的 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 位下均测试通过。

     
    如果问题解决起来不妥或者有更好的解决办法,麻烦请告知,帮助曾经和你一样的入门者,谢谢。
  • 相关阅读:
    Centos7上安装docker
    另类SQL拼接方法
    多平台Client TCP通讯组件
    Redis协议详解
    .net下简单快捷的数值高低位切换
    beetle 2.7海量消息广播测试
    FileSync文件同步更新工具
    简单实现TCP下的大文件高效传输
    感知机原理小结
    日志和告警数据挖掘经验谈
  • 原文地址:https://www.cnblogs.com/ourran/p/4968502.html
Copyright © 2011-2022 走看看