c++ - 控制台上的陷阱退出

Question

我想在用户按下后调用清理函数控制台窗口右上角的小“x”。

我已经注册了一个 atexit 方法，但在这种情况下不会调用该方法。

解决方案需要在 Windows 和 Linux 上运行。

Answer 1

您不能使用atexit这里是因为，当进程正常终止时会调用它，而在您的情况下，进程是由信号终止的。

在 linux 上，当控制终端关闭时，SIGHUP(挂起信号)会发送到进程，您可以使用 POSIX 信号处理。

在windows上，会传递CTRL_CLOSE_EVENT事件，您可以使用SetConsoleCtrlHandler winapi 来处理这个问题。

因此，据我所知，c++ 中没有独立于平台的方法来处理此问题。您应该使用与平台相关的代码来执行此操作，如以下简单程序所示。我在 Windows 上用 VS 测试了它，在 ubuntu 上用 g++ 测试了它。请注意，为了简单起见，错误处理已被省略，并且 I/O 在信号处理程序中执行。

程序注册了一个信号处理程序和一个atexit函数。然后它会休眠 10 秒钟。如果您在 10 秒内没有关闭控制台，进程将正常终止，并调用 atexit 处理程序。如果您在 10 秒之前关闭窗口，它就会捕获信号。在 Linux 上，您不会看到信号处理程序被调用，但它确实被调用，您可以通过编写文件(或发出蜂鸣声？)来检查这一点，尽管我不推荐这样做。

#ifdef  WIN32
#include <windows.h> 
#else
#include <unistd.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#endif

#include <stdio.h>
#include <stdlib.h>


#ifdef  WIN32
BOOL sig_handler(DWORD signum) 
{ 
    switch( signum ) 
    { 
        case CTRL_CLOSE_EVENT: 
            printf( "Ctrl-Close event\n" );
            return( TRUE ); 

        default: 
            return FALSE; 
    } 
}
#else
void sig_handler(int signum)
{
    /* you won't see this printed, but it runs */
    printf("Received signal %d\n", signum);
}
#endif 


void exit_fn(void)
{
   printf("%s\n", __FUNCTION__);
}


void setup_signal_handler()
{
#ifdef  WIN32
    SetConsoleCtrlHandler((PHANDLER_ROUTINE)sig_handler, TRUE);
#else
    struct sigaction sa;
    sa.sa_handler = &sig_handler;
    sigfillset(&sa.sa_mask);
    sigaction(SIGHUP, &sa, NULL);
#endif  
}


int main(void)
{
    printf("%s\n", __FUNCTION__);
    /* setup signal handler */
    setup_signal_handler();
    /* setup function to be called at normal process termination */
    atexit(exit_fn);
    /* sleep for 10s */
#ifdef  WIN32
    Sleep(10000);
#else
    sleep(10);
#endif
    /* print message if process terminates normally */
    printf("Normal process termination\n");

    exit(EXIT_SUCCESS);
}