【Ｃ＋＋】Windows作業系統MAC位址擷取

// ConsoleApplication1.cpp: 定義控制台應用程式的入口點。
//

#include "stdafx.h"
#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 頭檔案
#include <iostream>
#define MAX_SIZE 50
#define BUF_SIZE 50

//
// 功能：擷取擴充卡特性
// 參數： 
//   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]);
}

int main() {
    char str[50];
    GetMacAddress(str);
    std::cout << str << std::endl;
    std::cin.get();
}
           

注：本人編譯環境為：window10，visual studio2017