c++ - RegSetValueEx - 错误路径，64 位，以管理员身份运行

Question

我有一些关于注册表相关功能的问题。我使用 RegSetValueEx 函数来创建注册表数据，但不幸的是我肯定做错了，但我不确定错误在哪里。我想在 HKEY_LOCAL_MACHINE - SOFTWARE\Microsoft\Windows\CurrentVersion\Run 中创建注册表，但它仅在 HKEY_LOCAL_MACHINE 键中创建数据，仅此而已。

TCHAR name[UNLEN + 1];
DWORD size = UNLEN + 1;
GetUserName((TCHAR*)name, &size);
string namep = name;
string path = "C:\\Users\\" + namep + "\\AppData\\Roaming\\MyProgram\\MyProgram.exe";
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run", 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS)
{
    cout << "OPENED!";
    if (RegQueryValueEx(HKEY_LOCAL_MACHINE, "MyRegistry", 0, NULL, NULL, NULL) == ERROR_FILE_NOT_FOUND)
    {
        cout << "NOT FOUND!"; // If registry not found, then create it?
        if (RegSetValueEx(HKEY_LOCAL_MACHINE, "MyRegistry", 0, REG_SZ, (const BYTE*)path.c_str(), path.size()) == ERROR_SUCCESS)
            cout << "REGISTRY WRITTEN";
        RegCloseKey(hKey);
    }
}

好的，这就是代码，有什么问题吗？另外我想问一下怎么做，所以我不必以管理员身份运行它来创建注册表，因为如果我不以管理员身份运行它，它就不会创建注册表。64 位版本的操作系统如何？我在 32 位上编译，所以我必须做些什么才能在 64 位机器上运行吗？

谢谢。 =)

Answer 1

您的代码有几个问题。

首先，不要硬编码文件路径，尤其是包含系统文件夹的路径。在这种情况下，您应该使用 SHGetFolderPath(CSIDL_APPDATA) 或 SHGetKnownFolderPath(FOLDERID_RoamingAppData) 来发现当前用户的 AppData\Roaming 文件夹的位置。或者，如果 MyProgram.exe 是当前正在运行的应用程序，您可以改用 GetModuleFileName(0)。

其次，非管理员用户没有对 HKEY_LOCAL_MACHINE 的写入权限，只有对 HKEY_CURRENT_USER 的写入权限。如果您需要写入 HKEY_LOCAL_MACHINE，您的应用需要以提升的管理员权限运行。

第三，您没有正确使用 RegQueryValueEx() 和 RegSetValueEx()。具体来说:

1. 您在第一个参数中传递了错误的 HKEY。您需要传递 RegOpenKeyEx() 返回给您的 HKEY。

2. 对于 RegSetValueEx()，REG_SZ 值必须包含空终止符，但是 path.size() 不算在内空终止符，因此您需要改用 size()+1。

第四，您正在泄露 RegOpenKeyEx() 返回给您的 HKEY，因为您调用 RegCloseKey() 仅当 RegQueryValueEx() 成功。

综上所述，尝试更像这样的事情:

if (RegOpenKeyEx(HKEY_CURRENT_USER, TEXT("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run"), 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS)
{
    cout << "OPENED! ";

    LONG lRet = RegQueryValueEx(hKey, TEXT("MyRegistry"), 0, NULL, NULL, NULL);
    if (lRet == ERROR_FILE_NOT_FOUND)
    {
        cout << "NOT FOUND! ";

        // assuming you want to store the calling process's
        // filename, otherwise adjust this as needed ...
        TCHAR filename[MAX_PATH + 1];
        DWORD len = GetModuleFileName(NULL, filename, MAX_PATH);
        filename[len] = 0;

        if (RegSetValueEx(hKey, TEXT("MyRegistry"), 0, REG_SZ, (const BYTE*)filename, (len+1) * sizeof(TCHAR)) == ERROR_SUCCESS)
            cout << "WRITTEN!";
        else
            cout << "NOT WRITTEN!";
    }
    else if (lRet == 0)
    {
        cout << "FOUND!";
    }
    else
        cout << "NOT QUERIED!";

    RegCloseKey(hKey);
}
else
    cout << "NOT OPENED!";

如果您只想确保注册表包含最新的文件名，您可以省略 RegQueryValueEx() 并让 RegSetValueEx() 覆盖现有值(如果存在):

if (RegOpenKeyEx(HKEY_CURRENT_USER, TEXT("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run"), 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS)
{
    cout << "OPENED! ";

    // assuming you want to store the calling process's
    // filename, otherwise adjust this as needed ...
    TCHAR filename[MAX_PATH + 1];
    DWORD len = GetModuleFileName(NULL, filename, MAX_PATH);
    filename[len] = 0;

    if (RegSetValueEx(hKey, TEXT("MyRegistry"), 0, REG_SZ, (const BYTE*)filename, (len+1) * sizeof(TCHAR)) == ERROR_SUCCESS)
        cout << "WRITTEN!";
    else
        cout << "NOT WRITTEN!";

    RegCloseKey(hKey);
}
else
    cout << "NOT OPENED!";