mirror of
https://github.com/polserver/polserver
synced 2026-08-13 08:23:08 -04:00
0e675c516a
* use new formatter for ERROR_PRINT, added ERROR_PRINTLN * remove old log define * fix comments
81 lines
1.6 KiB
C++
81 lines
1.6 KiB
C++
/** @file
|
|
*
|
|
* @par History
|
|
*/
|
|
|
|
|
|
#include "esignal.h"
|
|
#ifdef WINDOWS
|
|
#include "Header_Windows.h"
|
|
#include "logfacility.h"
|
|
#else
|
|
#include <signal.h>
|
|
#endif
|
|
namespace Pol
|
|
{
|
|
namespace Clib
|
|
{
|
|
std::atomic<bool> exit_signalled( false );
|
|
std::atomic<int> exit_code( 0 );
|
|
|
|
void signal_exit( int code )
|
|
{
|
|
if ( exit_code == 0 ) // do not allow to "fix" the error code
|
|
exit_code = code;
|
|
exit_signalled = true;
|
|
}
|
|
|
|
#ifdef WINDOWS
|
|
BOOL WINAPI control_handler( DWORD dwCtrlType )
|
|
{
|
|
switch ( dwCtrlType )
|
|
{
|
|
case CTRL_BREAK_EVENT: // use Ctrl+C or Ctrl+Break to simulate
|
|
case CTRL_C_EVENT: // SERVICE_CONTROL_STOP in debug mode
|
|
ERROR_PRINTLN( "Ctrl-C detected. Stopping..." );
|
|
signal_exit();
|
|
return TRUE;
|
|
|
|
// Consider these three new signals?
|
|
#if 0
|
|
case CTRL_CLOSE_EVENT: // console window closing
|
|
ERROR_PRINTLN("Console window closing. Stopping...");
|
|
signal_exit();
|
|
return TRUE;
|
|
|
|
case CTRL_LOGOFF_EVENT: // User logging off system
|
|
ERROR_PRINTLN("User logging off system. Stopping...");
|
|
signal_exit();
|
|
return TRUE;
|
|
|
|
case CTRL_SHUTDOWN_EVENT: // User shutting down system
|
|
ERROR_PRINTLN("System shutting down. Stopping...");
|
|
signal_exit();
|
|
return TRUE;
|
|
#endif
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
void enable_exit_signaller()
|
|
{
|
|
SetConsoleCtrlHandler( control_handler, TRUE );
|
|
}
|
|
|
|
#else
|
|
|
|
|
|
void ctrlc_handler( int x )
|
|
{
|
|
(void)x;
|
|
signal_exit();
|
|
}
|
|
void enable_exit_signaller()
|
|
{
|
|
signal( SIGINT, ctrlc_handler );
|
|
signal( SIGQUIT, ctrlc_handler );
|
|
signal( SIGTERM, ctrlc_handler );
|
|
}
|
|
#endif
|
|
} // namespace Clib
|
|
} // namespace Pol
|