控制台上的陷阱退出

Question

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

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

解决方案需要在 windows 和 linux 上运行。

Answer 1

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

在linux，SIGHUP（信号挂断）在其控制终端关闭时发送给进程，您可以使用POSIX信号处理。

在windows，传递CTRL_CLOSE_EVENT事件，你可以使用SetConsoleCtrlHandlerwinapi来处理。

因此，据我所知，c++ 中没有独立于平台的方法来处理此问题。您应该使用平台相关代码来执行此操作，如以下简单程序所示。我在带有 VS 的 windows 和带有 g++ 的 ubuntu 上对其进行了测试。请注意，为简单起见，错误处理已被省略，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);
}