曲径通幽论坛

标题: SetConsoleCtrlHandler() -- 设置控制台信号处理函数 [打印本页]

作者: beyes    时间: 2014-1-20 13:40
标题: SetConsoleCtrlHandler() -- 设置控制台信号处理函数
当你在命令行里工作时,经常会输入 Ctrl-C 这个组合键以停止程序的运行。SetConsoleCtrlHandler() 函数可以让一个或多个特定函数在接收到 Ctrl-C, Ctrl-break 这些与控制台有关的信号时得到执行。
  1. BOOL WINAPI SetConsoleCtrlHandler(
  2.   _In_opt_  PHANDLER_ROUTINE HandlerRoutine,
  3.   _In_      BOOL Add
  4. );
复制代码
函数有两个参数,第 1 个参数是要安装的信号处理函数,当第 2 个参数 Add 设置为 TRUE 时,那么就安装第 1 个参数所表示的处理函数;否则,就从控制台控制例程列表中将之删除。需要注意的是,这里没有指定信号,测试到底是什么信号,是处理程序所负责的事情。

函数的返回值是个布尔值,当函数执行成功时,返回非 0 值;反之,返回 0 值,具体的出错原因,可以用 GetLastError() 函数来获得。

如果第 1 个参数设置为 NULL,且 Add 为 TRUE 时,那么 Ctrl-C 信号会被忽略。

如果已经使用 SetConsoleMode() 设置了 ENABLE_PROCESSED_INPUT 标志,那么系统会将 Ctrl-C 作为键盘输入处理,而不是作为信号处理。

处理程序例程实际上是以进程内的独立线程执行的,正常的程序将继续操作。

处理程序:
  1. BOOL WINAPI HandlerRoutine(
  2.   _In_  DWORD dwCtrlType
  3. );
复制代码
带有一个 DWORD 类型值 dwCtrlType ,它用来标识信号(或事件),可以是以下 5 个值之一:

1) CTRL_C_EVENT 表示键盘输入了 Ctrl-C 。
2) CTRL_CLOSE_EVENT 表示控制台窗口正在被关闭。
3) CTRL_BREAK_EVENT 表示 Ctrl_break 信号。
4) CTRL_LOGOFF_EVENT 表示用户正在注销。
5) CTRL_SHUTDOWN_EVENT 表示 Windows 正在关闭。

信号处理程序可以像异常或终止程序那样执行清理操作。当信号处理程序返回 TRUE 时,表示函数已经处理了信号;若是返回 FALSE ,那么就去执行列表中下一个处理程序函数。信号处理程序与设置顺序相反的顺序执行,也就是说,最后一个设置程序将被首先执行,而系统的处理程序在最后执行。

示例程序:
[C] 纯文本查看 复制代码
static BOOL WINAPI Handler(DWORD cntrlEvent);
static BOOL exitFlag = FALSE;

int _tmain(int argc, LPTSTR argv[])

/* Beep periodically until signaled to stop. */
{
                /* 添加信号或事件处理函数 */
        if (!SetConsoleCtrlHandler(Handler, TRUE))
                ReportError(_T("Error setting event handler"), 1, TRUE);
       
        while (!exitFlag) { /* This flag is detected right after a beep, before a handler exits */
                Sleep(4750);
                Beep(1000);
        }
        _tprintf(_T("Stopping the main program as requested.\n"));
        return 0;
}       

BOOL WINAPI Handler(DWORD cntrlEvent)
{
        switch (cntrlEvent) {
                /* The signal timing will determine if you see the second handler message */
                case CTRL_C_EVENT:
                        _tprintf(_T("Ctrl-C received by handler. Leaving in 5 seconds or less.\n"));
                        exitFlag = TRUE;
                        Sleep(4000); /* Decrease this time to get a different effect */
                        _tprintf(_T("Leaving handler in 1 second or less.\n"));
                        return TRUE; /* TRUE indicates that the signal was handled. */
                case CTRL_CLOSE_EVENT:
                        _tprintf(_T("Close event received by handler. Leaving the handler in 5 seconds or less.\n"));
                        exitFlag = TRUE;
                        Sleep(4000); /* Decrease this time to get a different effect */
                        _tprintf(_T("Leaving handler in 1 second or less.\n"));
                        return TRUE; /* Try returning FALSE. Any difference? */
                default:
                        _tprintf(_T("Event: %d received by handler. Leaving in 5 seconds or less.\n"), cntrlEvent);
                        exitFlag = TRUE;
                        Sleep(4000); /* Decrease this time to get a different effect */
                        _tprintf(_T("Leaving handler in 1 seconds or less.\n"));
                        return TRUE; /* TRUE indicates that the signal was handled. */
        }
}





欢迎光临 曲径通幽论坛 (http://www.groad.net/bbs/) Powered by Discuz! X3.2