WinAPI技术问题：如何在强制退出时执行清理代码

任务管理器的「结束进程」操作是通过TerminateProcess API强制终止进程，这种场景下进程没有任何机会执行用户态的清理代码（包括全局对象的析构函数）。但对于其他终止场景（如用户关闭窗口、系统关机/注销、任务管理器「结束任务」（先发送关闭信号）），我们可以通过针对性处理确保清理逻辑执行。

1. 处理窗口关闭与系统关机消息 #

对于GUI程序，系统在关闭窗口、关机或注销时会发送WM_QUERYENDSESSION和WM_ENDSESSION消息，我们需要在窗口过程中捕获这些消息并执行清理。首先需要为程序添加窗口（原代码无窗口，无法接收系统消息）：

#include <windows.h>

// 统一的清理函数，将所有需要执行的清理逻辑集中在这里
void Cleanup() {
    // 替换为实际清理操作：释放资源、写入日志等
    // 注意：终止场景下尽量避免弹窗（如MessageBox），可能无法正常显示
    MessageBoxA(NULL, "Cleanup executed", "", MB_OK);
}

class Conf {
public:
    Conf(int i) : m_i{ i } {
        MessageBoxA(NULL, "Conf", "", MB_OK);
    }

    ~Conf() {
        Cleanup();
    }

private:
    int m_i;
};

Conf conf(26);

// 窗口过程函数，处理系统消息
LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    switch (uMsg) {
        case WM_QUERYENDSESSION:
            // 系统询问是否允许终止进程，提前执行清理
            Cleanup();
            return TRUE; // 返回TRUE表示允许终止
        case WM_ENDSESSION:
            // 系统即将终止进程，再次执行清理（双重保险）
            if (wParam) Cleanup();
            break;
        case WM_CLOSE:
            // 用户主动关闭窗口时执行清理
            Cleanup();
            DestroyWindow(hwnd);
            break;
        case WM_DESTROY:
            PostQuitMessage(0);
            break;
        default:
            return DefWindowProc(hwnd, uMsg, wParam, lParam);
    }
    return 0;
}

int WinMain(
    HINSTANCE hInstance,
    HINSTANCE hPrevInstance,
    PSTR pCmdLine,
    int nCmdShow) {

    // 注册窗口类
    const wchar_t CLASS_NAME[] = L"Cleanup Demo Class";
    WNDCLASS wc = {0};
    wc.lpfnWndProc = WindowProc;
    wc.hInstance = hInstance;
    wc.lpszClassName = CLASS_NAME;
    wc.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
    RegisterClass(&wc);

    // 创建窗口
    HWND hwnd = CreateWindowEx(
        0,
        CLASS_NAME,
        L"Cleanup Demo",
        WS_OVERLAPPEDWINDOW,
        CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
        NULL, NULL, hInstance, NULL
    );

    if (!hwnd) return 0;
    ShowWindow(hwnd, nCmdShow);

    MSG msg;
    while (GetMessage(&msg, NULL, NULL, NULL)) {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }

    return 13;
}

2. 注册控制台控制处理函数（处理控制台终止信号） #

如果程序可能以控制台模式运行，或需要处理Ctrl+C、Ctrl+Break等信号，可以注册SetConsoleCtrlHandler捕获这些事件：

// 控制台信号处理函数
BOOL WINAPI ConsoleCtrlHandler(DWORD dwCtrlType) {
    switch (dwCtrlType) {
        case CTRL_C_EVENT:
        case CTRL_BREAK_EVENT:
        case CTRL_CLOSE_EVENT:
        case CTRL_LOGOFF_EVENT:
        case CTRL_SHUTDOWN_EVENT:
            Cleanup();
            return TRUE; // 表示已处理该信号
        default:
            return FALSE;
    }
}

// 在WinMain开头注册处理函数
int WinMain(...) {
    SetConsoleCtrlHandler(ConsoleCtrlHandler, TRUE);
    // ... 其余窗口创建、消息循环代码
}

3. 关于「结束进程」的说明 #

任务管理器的「结束进程」是硬终止操作，系统会直接调用TerminateProcess强制杀死进程，这种场景下无法执行任何用户态清理代码——这是Windows的设计，目的是确保能强制结束无响应进程，没有技术手段可以在该场景下执行自定义逻辑。

• 清理代码要简洁、无外部依赖，避免调用可能失败的API（如弹窗），建议通过写入日志文件记录清理操作。
• 全局对象的析构仅在进程正常退出时执行，因此必须将清理逻辑抽离为独立函数，在多个终止场景中主动调用，确保覆盖尽可能多的情况。

内容的提问来源于stack exchange，提问作者coderodde