C++ – Windows下实现程序开机自启动功能或者关闭开机自启动

1 Windows下程序实现开机自启动的原理

在Windows系统下要实现程序开机自启，主要是在系统注册表路径HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Run写入需要开机自启动的程序路径。

Windows在系统启动时会去系统注册表下的HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Run读取所有的键值key-value，启动value中保存路径的程序。

2 C++实现程序开机自启动或者关闭程序开机自启动

第1节中说明了Windows下实现程序开机自启动的原理，使用代码实现实际上就是一个在注册表中添加键值对以及删除键值对的过程。具体的实现代码如下

/// <summary>
/// 设置当前程序开机自启或者关闭程序开机自启
/// </summary>
/// <param name="b_auto_run">true为设置程序开机自启，false则取消程序开机自启</param>
/// <param name="key">注册表键值</param>
/// <returns></returns>
bool SetBootStartup(bool b_auto_run,std::string key)
{
    HKEY hKey;
    if (RegOpenKeyEx(HKEY_CURRENT_USER, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run", 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS) ///打开启动项  
    {
        // 获取当前程序的路径
        char pFileName[MAX_PATH] = { 0 };
        DWORD dwRet = GetModuleFileName(NULL, (LPSTR)pFileName, MAX_PATH);

        TCHAR strDir[MAX_PATH] = {};
        DWORD max_length = MAX_PATH;

        // 判断当前的键是否存在
        long result = RegGetValue(hKey, nullptr, key.c_str(), RRF_RT_REG_SZ, 0, strDir, &max_length);

        if (b_auto_run)
        {
            // 如果键值不存在
            if (result != ERROR_SUCCESS)
            {
                RegSetValueEx(
                    hKey,
                    key.c_str(),
                    0,
                    REG_SZ,
                    (const BYTE*)(LPCSTR)pFileName,
                    MAX_PATH
                );
            }

        }
        else 
        {
            // 如果键值存在
            if (result == ERROR_SUCCESS)
            {
                RegDeleteValueA(hKey, key.c_str());
            }
        }

        RegCloseKey(hKey);

        return true;
    }
    else 
    {
        return false;
    }

    return true;
}