Windows下设置程序自动运行代码

#ifdef Q_WS_WIN
#include <windows.h>
#include <shlobj.h>
#endif

win32:LIBS += -lshell32 -lole32 -luser32

// 获取程序启动快捷方式
QString GetStartUpPath()//{{{
{
    QString strPath;
#ifdef Q_WS_WIN
    LPITEMIDLIST    pidl;
    LPMALLOC        pShellMalloc;
    wchar_t         szDir[1024];
    if (SUCCEEDED(SHGetMalloc(&pShellMalloc)))
    {
        if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_STARTUP, &pidl)))
        {
            if (SHGetPathFromIDList(pidl, szDir))
            {
                strPath = QString::fromWCharArray(szDir) + "\\MyApp.lnk";
            }
            pShellMalloc->Free(pidl);
        }
        pShellMalloc->Release();
    }
#endif
    return strPath;
}//}}}

// 设置是否自动运行
void setAutoRunStatus(bool bAutoRun)//{{{
{
    QString strExe = QApplication::applicationFilePath();
    QString strLink = GetStartUpPath();
    if (bAutoRun)
    {
#ifdef Q_WS_WIN
        HRESULT hr = CoInitialize(NULL);
        if (SUCCEEDED(hr))
        {
            IShellLink *pisl;
            hr = CoCreateInstance(CLSID_ShellLink, NULL,
                    CLSCTX_INPROC_SERVER, IID_IShellLink, (void**)&pisl);
            if (SUCCEEDED(hr))
            {
                IPersistFile* pIPF;

                //这里是我们要创建快捷方式的原始文件地址
                pisl->SetPath(strExe.toStdWString().c_str());
                hr = pisl->QueryInterface(IID_IPersistFile, (void**)&pIPF);
                if (SUCCEEDED(hr))
                {
                    //这里是我们要创建快捷方式的目标地址
                    pIPF->Save(strLink.toStdWString().c_str(), FALSE);
                    pIPF->Release();
                }
                pisl->Release();
            }
            CoUninitialize();
        }
#endif
    }
    else
    {
        QFile::remove(strLink);
    }
}//}}}

// 获取自动运行状态
bool getAutoRunStatus()//{{{
{
    QString strLink = GetStartUpPath();
    bool bAutoRun = QFile::exists(strLink);
    return bAutoRun;
}//}}}