C++ 程序自我重啟技術實現：基於 CreateProcess 的優雅重啟方案

前言

在 C++ 應用開發中，程序自我重啟是常見需求（如配置生效、插件更新、錯誤恢復、權限提升）。本文介紹一種基於 CreateProcess 在主入口點實現的優雅重啟方案，通過全局標誌位協調舊進程退出與新進程啟動，確保無縫切換。

一、核心原理

1. 用戶觸發重啟請求（點擊「重啟」按鈕）
2. 設置全局標誌 g_bIsRunAgain = true
3. 正常關閉主窗口，進入 WinMain 尾部
4. 檢測標誌位，調用 CreateProcess 啟動自身新實例
5. 舊進程退出，新進程接管

二、完整實現代碼

2.1 項目入口文件

// Project1.cpp
#include "Unit1.h"

bool g_bIsRunAgain = false; // 全局重啟標誌

WINAPI WinMain(HINSTANCE, HINSTANCE, LPSTR, int) {
    try {
        Application->Initialize();
        Application->CreateForm(__classid(TfrmMain), &frmMain);
        Application->Run();
    }
    catch (Exception &exception) {
        Application->ShowException(&exception);
    }
    catch (...) {
        try { throw Exception(""); }
        catch (Exception &exception) {
            Application->ShowException(&exception);
        }
    }

    // 關鍵：重啟邏輯
    if (g_bIsRunAgain) {
        AnsiString strPath = Application->ExeName; // 獲取自身完整路徑
        STARTUPINFO si = { sizeof(STARTUPINFO) };
        PROCESS_INFORMATION pi;
        
        // 創建新進程，優先級正常
        if (CreateProcess(
            (LPCTSTR)strPath.c_str(), // 應用程序名
            NULL,                     // 命令行參數
            NULL, NULL,               // 進程/線程安全屬性
            FALSE,                    // 不繼承句柄
            NORMAL_PRIORITY_CLASS,    // 優先級
            NULL, NULL,               // 環境塊/當前目錄
            &si, &pi)) {               // 啟動信息/進程信息
            
            CloseHandle(pi.hThread);
            CloseHandle(pi.hProcess);
        }
    }
    return 0;
}

2.2 主窗口頭文件

// Unit1.h
extern bool g_bIsRunAgain; // 聲明全局標誌

class TfrmMain : public TForm {
    // ...
    void __fastcall btnCloseClick(TObject *Sender);   // 關閉按鈕
    void __fastcall btnReExcuteClick(TObject *Sender); // 重啟按鈕
};

2.3 主窗口實現文件

// Unit1.cpp
#include "Unit1.h"

bool g_bIsRunAgain = false; // 定義全局標誌

void __fastcall TfrmMain::btnCloseClick(TObject *Sender) {
    Close(); // 正常關閉窗口
}

void __fastcall TfrmMain::btnReExcuteClick(TObject *Sender) {
    g_bIsRunAgain = true; // 設置重啟標誌
    Close();              // 觸發關閉流程
}

三、關鍵技術細節解析

四、進階優化方案

4.1 傳遞命令行參數保持狀態

if (g_bIsRunAgain) {
    AnsiString strPath = Application->ExeName;
    AnsiString strParams = ""; // 收集需傳遞的參數
    // 例如：strParams = "--restarted --config=new.cfg";
    
    STARTUPINFO si = { sizeof(STARTUPINFO) };
    PROCESS_INFORMATION pi;
    
    // 使用命令行參數版本
    AnsiString cmdLine = strPath + " " + strParams;
    if (CreateProcess(NULL, (LPSTR)cmdLine.c_str(), NULL, NULL, FALSE,
        NORMAL_PRIORITY_CLASS, NULL, NULL, &si, &pi)) {
        CloseHandle(pi.hThread);
        CloseHandle(pi.hProcess);
    }
}

4.2 互斥體防止多實例衝突

// WinMain 開頭
HANDLE hMutex = CreateMutex(NULL, TRUE, "Global\\MyApp_Restart_Mutex");
if (GetLastError() == ERROR_ALREADY_EXISTS) {
    // 已有實例運行，可選：激活舊窗口、傳遞參數給舊實例、或直接退出
    CloseHandle(hMutex);
    return 0;
}
// ... 正常運行 ...
// 退出前
ReleaseMutex(hMutex);
CloseHandle(hMutex);

4.3 管理員權限重啟（UAC 提升）

if (g_bIsRunAgain && NeedAdminRights()) {
    SHELLEXECUTEINFO sei = { sizeof(sei) };
    sei.lpVerb = "runas"; // 關鍵：請求管理員權限
    sei.lpFile = Application->ExeName.c_str();
    sei.lpParameters = "--restarted";
    sei.nShow = SW_SHOWNORMAL;
    ShellExecuteEx(&sei);
    return 0; // 當前進程退出
}

4.4 無縫重啟：保持窗口位置/狀態

// 重啟前保存狀態
void __fastcall TfrmMain::btnReExcuteClick(TObject *Sender) {
    // 保存窗口位置、大小、用戶數據等
    TIniFile* ini = new TIniFile(ChangeFileExt(Application->ExeName, ".ini"));
    ini->WriteInteger("Window", "Left", Left);
    ini->WriteInteger("Window", "Top", Top);
    ini->WriteInteger("Window", "Width", Width);
    ini->WriteInteger("Window", "Height", Height);
    // 保存其他業務狀態...
    delete ini;
    
    g_bIsRunAgain = true;
    Close();
}

// 新進程啟動時恢復
// 在 FormCreate 或 FormShow 中讀取 ini 恢復狀態

五、常見問題與解決方案

六、總結

基於 WinMain 尾部檢測全局標誌調用 CreateProcess 的自我重啟方案，具有實現簡單、無需額外啟動器、參數傳遞靈活、易於擴展（UAC提升、狀態保存、互斥體保護）等優點。關鍵在於：標誌位設置時機、STARTUPINFO 正確初始化、進程句柄及時關閉、工作目錄與參數正確傳遞。結合具體業務場景選擇基礎版或進階版，即可實現穩定可靠的程序自我重啟功能。