在处理控制台的时候需要处理意外的退出事件以保存数据,对应的退出事件有CTRL_C_EVENT, CTRL_BREAK_EVENT, CTRL_CLOSE_EVENT, CTRL_LOGOFF_EVENT, CTRL_SHUTDOWN_EVENT。通过
BOOL WINAPI SetConsoleCtrlHandler(__in_opt PHANDLER_ROUTINE HandlerRoutine , __in BOOL Add );
可以设置一个回调函数,在控制台将要关闭时调用。在用户设置了ConsoleCtrlHandler时,WINDOWS会启动一个线程来执行这个函数。值得注意的是这个函数是由另外一个线程执行的。因此在意外关闭控制台前需要保存数据的话,需要注意线程的同步问题,可以这样处理:
//CtrlHandler回调函数 BOOL WINAPI ConsoleHandler(DWORD CEvent); HANDLE hQuitEvent; int _tmain(int argc, TCHAR* argv[], TCHAR* envp[]) { hQuitEvent = ::CreateEvent(NULL, TRUE, FALSE, NULL); //设置回调函数 if (SetConsoleCtrlHandler((PHANDLER_ROUTINE)ConsoleHandler, TRUE) == FALSE) { CloseHandle(hQuitEvent); return -1; }
//主程序代码,主程序代码部分需要处理用户退出的事件。
//清理工作
//通知回调线程结束
SetEvent(hQuitEvent); return 0; }
BOOL WINAPI ConsoleHandler(DWORD CEvent)
{
//处理对应的CTRL_事件,返回TRUE或者FALSE,在感兴趣的事件后面添加::WaitForSingleObject(hQuitEvent, INFINITE);等待主线程完成清理工作
switch (CEvent)
{
case CTRL_C_EVENT:
...
case CTRL_BREAK_EVENT:
...
case CTRL_CLOSE_EVENT:
...
case CTRL_LOGOFF_EVENT:
...
case CTRL_SHUTDOWN_EVENT:
...
default:
return FALSE;
}
return TRUE;
}