控制台上的陷阱退出

Trap exit on console

我想在用户按下后调用清理函数 控制台 window.

右上角的小 "x"

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

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

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

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

windows,CTRL_CLOSE_EVENT事件被传递,你可以使用SetConsoleCtrlHandler winapi来处理这个。

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

程序注册了一个信号处理程序和一个atexit函数。然后它会休眠 10 秒。如果您在 10s 内没有关闭控制台,进程将正常终止并调用 atexit 处理程序。如果你在 10 秒前关闭 window,它会捕捉到信号。在 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);
}