Question

我正在尝试将我的软件添加到注册表中，我发现了一些我可以使用的代码但没有完整的代码C / C ++对我来说是新的，不能自己创建它。但这里有一个基本思路：检查是否设置了reg键，如果没有创建它。

我能够使用以下代码获取我的程序位置：

TCHAR szPath[MAX_PATH];
GetModuleFileName(NULL,szPath,MAX_PATH);

并且能够创建密钥:(不确定它是否正确）

HKEY newValue;
RegOpenKey(HKEY_CURRENT_USER,"Software\\Microsoft\\Windows\\CurrentVersion\\Run",&newValue);
RegSetValueEx(newValue,"myprogram",0,REG_SZ,(LPBYTE)szPath,sizeof(szPath));
RegCloseKey(newValue);
return 0;

缺少什么，小小的检查钥匙是否已经存在......

谢谢！

Answer 1

这里有一些代码可能会做你想要的。调用RegisterProgram让您的EXE自行注册，以便在用户登录时自动启动。此函数调用GetModuleFileName，然后调用另一个名为RegisterMyProgramForStartup的辅助函数来执行写入注册表中。

致电IsMyProgramRegisteredForStartup(L"My_Program")以检测注册是否确实存在且显示有效。

一个快速说明。在实际再次写出密钥之前检查密钥是否存在对性能的影响可以忽略不计。您可以盲目地调用RegisterProgram，如果密钥已经存在，它将覆盖密钥。检测是否存在注册对于初始化启用或禁用自动启动的UI复选框非常有用。（您正在为用户提供一个选择，对吧？因为我讨厌自动安装自己运行的应用程序而不给我一个选择。）

BOOL IsMyProgramRegisteredForStartup(PCWSTR pszAppName)
{
    HKEY hKey = NULL;
    LONG lResult = 0;
    BOOL fSuccess = TRUE;
    DWORD dwRegType = REG_SZ;
    wchar_t szPathToExe[MAX_PATH]  = {};
    DWORD dwSize = sizeof(szPathToExe);

    lResult = RegOpenKeyExW(HKEY_CURRENT_USER, L"Software\\Microsoft\\Windows\\CurrentVersion\\Run", 0, KEY_READ, &hKey);

    fSuccess = (lResult == 0);

    if (fSuccess)
    {
        lResult = RegGetValueW(hKey, NULL, pszAppName, RRF_RT_REG_SZ, &dwRegType, szPathToExe, &dwSize);
        fSuccess = (lResult == 0);
    }

    if (fSuccess)
    {
        fSuccess = (wcslen(szPathToExe) > 0) ? TRUE : FALSE;
    }

    if (hKey != NULL)
    {
        RegCloseKey(hKey);
        hKey = NULL;
    }

    return fSuccess;
}

BOOL RegisterMyProgramForStartup(PCWSTR pszAppName, PCWSTR pathToExe, PCWSTR args)
{
    HKEY hKey = NULL;
    LONG lResult = 0;
    BOOL fSuccess = TRUE;
    DWORD dwSize;

    const size_t count = MAX_PATH*2;
    wchar_t szValue[count] = {};


    wcscpy_s(szValue, count, L"\"");
    wcscat_s(szValue, count, pathToExe);
    wcscat_s(szValue, count, L"\" ");

    if (args != NULL)
    {
        // caller should make sure "args" is quoted if any single argument has a space
        // e.g. (L"-name \"Mark Voidale\"");
        wcscat_s(szValue, count, args);
    }

    lResult = RegCreateKeyExW(HKEY_CURRENT_USER, L"Software\\Microsoft\\Windows\\CurrentVersion\\Run", 0, NULL, 0, (KEY_WRITE | KEY_READ), NULL, &hKey, NULL);

    fSuccess = (lResult == 0);

    if (fSuccess)
    {
        dwSize = (wcslen(szValue)+1)*2;
        lResult = RegSetValueExW(hKey, pszAppName, 0, REG_SZ, (BYTE*)szValue, dwSize);
        fSuccess = (lResult == 0);
    }

    if (hKey != NULL)
    {
        RegCloseKey(hKey);
        hKey = NULL;
    }

    return fSuccess;
}

void RegisterProgram()
{
    wchar_t szPathToExe[MAX_PATH];

    GetModuleFileNameW(NULL, szPathToExe, MAX_PATH);
    RegisterMyProgramForStartup(L"My_Program", szPathToExe, L"-foobar");
}

int _tmain(int argc, _TCHAR* argv[])
{
    RegisterProgram();
    IsMyProgramRegisteredForStartup(L"My_Program");
    return 0;
}

Answer 2

要检查该值是否存在，请致电RegQueryValueEx。

LONG retval = RegQueryValueEx(hKey, "myprogram", NULL, NULL, NULL, NULL);

请注意，您所谓的newValue实际上是一个键而不是一个值。为了避免混淆，你应该这样命名。我使用了名称hKey。

然后，要检查该值是否存在，请按documentation中的说明将retval与ERROR_SUCCESS进行比较。

您的代码的另一个问题是绝对没有错误检查。我会留给你解决。

Answer 3

您忘了写有关安全性访问的论点

将应用程序添加到启动（注册表）

3 个答案: