c++ - 为什么此代码在具有多字节字符集但不具有 unicode 字符集的 visual studio 中完美运行？

Question

当我在 g++ 上运行这段代码时，它运行流畅，但是当我在带有 unicode 字符集选项的 visual studio 上运行这段代码时，它不会打印产品 ID。你能解释一下如何解决这个问题以及为什么会这样吗？

#include <windows.h>
#include <stdio.h>
#include <iostream>
using namespace std;

wchar_t* GetRegistryKeyValue(const char* RegKey, const char* pPIDName)
{
    HKEY        Registry;
    long        ReturnStatus;
    DWORD       regType = 0;
    DWORD       regSize = 0;
    char*       pPID = 0;
    ReturnStatus = RegOpenKeyEx(HKEY_LOCAL_MACHINE, RegKey, 0, KEY_QUERY_VALUE | KEY_WOW64_64KEY, &Registry);
    if (ReturnStatus == ERROR_SUCCESS)
    {
        ReturnStatus = RegQueryValueEx(Registry, pPIDName, 0, &regType, 0, &regSize);
        pPID = new char[regSize];

        /* Get Value. */
        ReturnStatus = RegQueryValueEx(Registry, pPIDName, 0, &regType, (LPBYTE)pPID, &regSize);


        RegCloseKey(Registry);

        if (pPID[regSize] > 127 || pPID[regSize] < 32)
        {
            pPID[regSize] = '\0';
        }

        if (regSize > 1)
        {

            int s = 0;
            int i=0;
            while (pPID[i] != NULL)
            {
                s++;
                i++;
            }


            const size_t cSize = s ;
            wchar_t* wc = new wchar_t[cSize];
            mbstowcs(wc, pPID, cSize);  

            return wc;
        }
        else
        {
            printf("Size not > 1 (%d)\n", regSize);
            return NULL;
        }
    }
    else
    {
        RegCloseKey(Registry);
        return NULL;
    }
}
int main()
{
    wchar_t * resultData=NULL;  
    resultData = GetRegistryKeyValue("SOFTWARE\\MICROSOFT\\Windows NT\\CurrentVersion", "ProductId");
    wcout << resultData;
    cout << endl;
    delete resultData;
    system("PAUSE");
    return 0;
}

Answer 1

您正在使用基于 TCHAR 的 API，这些 API 在为 MBCS 编译时依赖于 char* 数据，在为 Unicode 编译时依赖于 wchar_t* 数据。您没有正确考虑这种差异。当设置为 Unicode 时，您的代码甚至不应该编译，因为您会将 ANSI 参数传递给 Unicode 函数。由于您想要返回 Unicode 字符串，因此您应该首先使用 Unicode API 函数。

您的代码中存在其他逻辑错误，例如内存泄漏、未为 wc 缓冲区分配正确的字节数、错误处理不充分等。

尝试更像这样的东西:

#include <windows.h>
#include <stdio.h>
#include <iostream>

using namespace std;

wchar_t* GetRegistryKeyValue(const wchar_t* RegKey, const wchar_t* pValueName)
{
    long        ReturnStatus;
    DWORD       regType = 0;
    DWORD       regSize = 0;
    DWORD dwFlags = RRF_RT_REG_SZ | RRF_RT_REG_MULTI_SZ | RRF_RT_REG_EXPAND_SZ | RRF_SUBKEY_WOW6464KEY;
    wchar_t* ws = 0;

    ReturnStatus = RegGetValueW(HKEY_LOCAL_MACHINE, RegKey, pValueName, dwFlags, &regType, 0, &regSize);
    if (ReturnStatus == ERROR_SUCCESS)
    {
        ws = new wchar_t[regSize / sizeof(WCHAR)];

        ReturnStatus = RegGetValueW(HKEY_LOCAL_MACHINE, RegKey, pValueName, dwFlags, &regType, ws, &regSize);
        if (ReturnStatus != ERROR_SUCCESS)
        {
            delete[] ws;
            ws = NULL;
        }
    }

    return ws;
}

int main()
{
    wchar_t* resultData = GetRegistryKeyValue(L"SOFTWARE\\MICROSOFT\\Windows NT\\CurrentVersion", L"ProductId");
    wcout << resultData << endl;
    delete[] resultData;
    system("PAUSE");
    return 0;
}

或者:

#include <windows.h>
#include <stdio.h>
#include <iostream>
#include <string>
#include <vector>

using namespace std;

wstring GetRegistryKeyValue(const wchar_t* RegKey, const wchar_t* pValueName)
{
    long        ReturnStatus;
    DWORD       regType = 0;
    DWORD       regSize = 0;
    DWORD dwFlags = RRF_RT_REG_SZ | RRF_RT_REG_MULTI_SZ | RRF_RT_REG_EXPAND_SZ | RRF_SUBKEY_WOW6464KEY;

    ReturnStatus = RegGetValueW(HKEY_LOCAL_MACHINE, RegKey, pValueName, dwFlags, &regType, 0, &regSize);
    if (ReturnStatus == ERROR_SUCCESS)
    {
        vector<BYTE> buf;
        buf.resize(regSize);

        ReturnStatus = RegGetValueW(HKEY_LOCAL_MACHINE, RegKey, pValueName, dwFlags, &regType, &buf[0], &regSize);
        if (ReturnStatus == ERROR_SUCCESS)
            return wstring((wchar_t*)&buf[0], (regSize / sizeof(WCHAR)) - 1);
    }

    return wstring();
}

int main()
{
    wcout << GetRegistryKeyValue(L"SOFTWARE\\MICROSOFT\\Windows NT\\CurrentVersion", L"ProductId") << endl;
    system("PAUSE");
    return 0;
}