Win32控制台程序中的清理代码

Question

本题基于以下问题：Handle CTRL+C on Win32

我正在开发一个多线程服务器，在 Linux 和 Windows 上运行。我不能使用 boost 或其他框架，只能使用 std c++。

我在 win32 端的清理代码有问题。 linux端工作正常：当我想关闭服务器时，我发送SIGINT（带有CTRL+C），信号处理程序设置一个全局变量，主pthread执行清理指令（加入其他pthread，释放堆内存等）。

在 Windows 上，获得相同的行为看起来并不那么简单。 我编写了一个简单的测试程序来了解信号处理程序在 windows 中的工作原理。

#include <iostream>
#include <windows.h>

bool running;

BOOL WINAPI consoleHandler(DWORD signal) {

    if (signal == CTRL_C_EVENT) {
        running = false;
        std::cout << "[CTRL+C]\n";
        return TRUE;
    }

    return FALSE;
} 

int main(int argc, char **argv) {

    running = true;

    if (!SetConsoleCtrlHandler(consoleHandler, TRUE)) {

            std::cerr << "Error: " << GetLastError() << '\n';
            return -1;
    }

    std::cout << "Main thread working hard...\n";
    while (running) { ; }

    for (int i = 0; i < 20; i++)
        std::cout << "This is the " << i << "th fake cleanup instruction\n";

    return 0;
}

输出如下：

$ test.exe
Main thread working hard...
[CTRL+C]
This is the 0th fake cleanup instruction
This is the 1th fake cleanup instruction

所以主线程很快就被杀死了，只在两条指令之后。在上一个问题中one of the suggestion 是在处理程序中移动清理代码，但并没有真正帮助：

假设处理函数如下所示：

BOOL WINAPI consoleHandler(DWORD signal) {

    if (signal == CTRL_C_EVENT) {
        running = false;
        std::cout << "[CTRL+C]\n";

        for (int i = 0; i < 20; i++)    
            std::cout << "This is the " << i << "th fake cleanup instruction\n";

        return TRUE;
    }

    return FALSE;
} 

现在行为更糟了！输出是：

$ test.exe
Main thread working hard...
[CTRL+C]
This is the

根据MSDN看来，进程总是被杀死：

一个 HandlerRoutine 可以执行任何必要的清理，然后取其中之一 以下操作：

• 调用ExitProcess 函数终止进程。
• 返回 FALSE。如果所有注册的处理程序函数都没有返回 TRUE，则默认处理程序终止进程。
• 返回真。在这种情况下，不会调用其他处理函数并且系统终止

过程。

我是否遗漏了一些明显的东西？ 终止 win32 控制台进程并执行其清理代码的正确方法是什么？

Answer 1

在 Windows 上，您也可以使用信号处理程序：

static void shutdown(int signum)
{
  printf("got signal #%d, terminating\n", signum);
  // cleanup
  _exit(1);
}

signal(SIGINT, shutdown);
signal(SIGTERM, shutdown);
signal(SIGSEGV, shutdown);

Ctrl-C 映射到SIGINT，就像在 Linux 上一样。

但是，这不会处理用户使用鼠标关闭控制台窗口。

Answer 2

这是一种方法，但我建议您使用事件 HANDLE 和 WaitForSingleObject，因为它往往会更加“屈服”。我把高速自旋循环留在这里只是为了让你在仍然看到处理程序被拦截的同时锁定你的一个核心。

我冒昧地修改了您的运行状态以分别进行原子评估和设置，因为我不希望优化器在主循环中抛出 eval。

#include <iostream>
#include <cstdlib>
#include <windows.h>

// using an event for monitoring
LONG running = 1;

BOOL WINAPI consoleHandler(DWORD signal)
{
    if (signal == CTRL_C_EVENT) 
    {
        std::out << "Received Ctrl-C; shutting down..." << std::endl;
        InterlockedExchange(&running, 0);
        return TRUE;
    }
    return FALSE;
} 

int main(int argc, char **argv) 
{
    if (!SetConsoleCtrlHandler(consoleHandler, TRUE)) 
    {
        std::cerr << "Error: " << GetLastError() << '\n';
        return EXIT_FAILURE;
    }

    std::cout << "Main thread working hard...\n";
    while (InterlockedCompareExchange(&running, 0, 0) == 1);

    std::cout << "Graceful shutdown received. Shutting down now." << std::endl;

    return 0;
}

输出（注意：我按了 ctrl-C，以防不明显）

Main thread working hard...
Received Ctrl-C; shutting down...
Graceful shutdown received. Shutting down now.

注意：我在 64 位和 32 位进程的调试 和 版本中对此进行了测试，没有问题。您可以从 VS 调试器运行它。如果您安装了处理程序，则只需在通知您可以继续时选择“继续”即可。