Simone
d074cc7c07
Plugin: ns7zip v2.0.0 Architectures: x86-ansi, x86-unicode, amd64-unicode License: LGPL-2.1-or-later
99 lines
1.8 KiB
C++
99 lines
1.8 KiB
C++
// ConsoleClose.cpp
|
|
|
|
#include "StdAfx.h"
|
|
|
|
#include "ConsoleClose.h"
|
|
|
|
#ifndef UNDER_CE
|
|
|
|
#ifdef _WIN32
|
|
#include "../../../Common/MyWindows.h"
|
|
#else
|
|
#include <stdlib.h>
|
|
#include <signal.h>
|
|
#endif
|
|
|
|
namespace NConsoleClose {
|
|
|
|
unsigned g_BreakCounter = 0;
|
|
static const unsigned kBreakAbortThreshold = 3;
|
|
|
|
#ifdef _WIN32
|
|
|
|
static BOOL WINAPI HandlerRoutine(DWORD ctrlType)
|
|
{
|
|
if (ctrlType == CTRL_LOGOFF_EVENT)
|
|
{
|
|
// printf("\nCTRL_LOGOFF_EVENT\n");
|
|
return TRUE;
|
|
}
|
|
|
|
if (++g_BreakCounter < kBreakAbortThreshold)
|
|
return TRUE;
|
|
return FALSE;
|
|
/*
|
|
switch (ctrlType)
|
|
{
|
|
case CTRL_C_EVENT:
|
|
case CTRL_BREAK_EVENT:
|
|
if (g_BreakCounter < kBreakAbortThreshold)
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
*/
|
|
}
|
|
|
|
CCtrlHandlerSetter::CCtrlHandlerSetter()
|
|
{
|
|
if (!SetConsoleCtrlHandler(HandlerRoutine, TRUE))
|
|
throw 1019; // "SetConsoleCtrlHandler fails";
|
|
}
|
|
|
|
CCtrlHandlerSetter::~CCtrlHandlerSetter()
|
|
{
|
|
if (!SetConsoleCtrlHandler(HandlerRoutine, FALSE))
|
|
{
|
|
// warning for throw in destructor.
|
|
// throw "SetConsoleCtrlHandler fails";
|
|
}
|
|
}
|
|
|
|
#else // _WIN32
|
|
|
|
static void HandlerRoutine(int)
|
|
{
|
|
if (++g_BreakCounter < kBreakAbortThreshold)
|
|
return;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
CCtrlHandlerSetter::CCtrlHandlerSetter()
|
|
{
|
|
memo_sig_int = signal(SIGINT, HandlerRoutine); // CTRL-C
|
|
if (memo_sig_int == SIG_ERR)
|
|
throw "SetConsoleCtrlHandler fails (SIGINT)";
|
|
memo_sig_term = signal(SIGTERM, HandlerRoutine); // for kill -15 (before "kill -9")
|
|
if (memo_sig_term == SIG_ERR)
|
|
throw "SetConsoleCtrlHandler fails (SIGTERM)";
|
|
}
|
|
|
|
CCtrlHandlerSetter::~CCtrlHandlerSetter()
|
|
{
|
|
signal(SIGINT, memo_sig_int); // CTRL-C
|
|
signal(SIGTERM, memo_sig_term); // kill {pid}
|
|
}
|
|
|
|
#endif // _WIN32
|
|
|
|
/*
|
|
void CheckCtrlBreak()
|
|
{
|
|
if (TestBreakSignal())
|
|
throw CCtrlBreakException();
|
|
}
|
|
*/
|
|
|
|
}
|
|
|
|
#endif
|