2

这个问题基于以下问题: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

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

假设处理函数如下所示:

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 函数终止进程。
  • 返回假。如果所有注册的处理程序函数都没有返回 TRUE,则默认处理程序终止进程。
  • 返回真。在这种情况下,不会调用其他处理函数并且系统终止

过程。

我错过了一些明显的东西吗?终止win32控制台进程执行其清理代码的正确方法是什么?

4

2 回答 2

3

这是一种方法,尽管我建议您使用事件 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 调试器运行它。如果您安装了处理程序,只需在通知您可以继续时选择“继续”,您可以这样做。

于 2013-09-13T06:53:39.273 回答
2

在 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 上一样。

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

于 2013-09-13T10:15:23.213 回答