mirror of
https://github.com/BeamMP/BeamMP-Server.git
synced 2025-07-01 15:26:59 +00:00
69 lines
1.8 KiB
C++
69 lines
1.8 KiB
C++
#pragma once
|
|
|
|
#include "Common.h"
|
|
|
|
#ifdef __unix
|
|
#include <csignal>
|
|
static void UnixSignalHandler(int sig) {
|
|
switch (sig) {
|
|
case SIGPIPE:
|
|
warn("ignoring SIGPIPE");
|
|
break;
|
|
case SIGTERM:
|
|
info("gracefully shutting down via SIGTERM");
|
|
Application::GracefullyShutdown();
|
|
break;
|
|
case SIGINT:
|
|
info("gracefully shutting down via SIGINT");
|
|
Application::GracefullyShutdown();
|
|
break;
|
|
default:
|
|
debug("unhandled signal: " + std::to_string(sig));
|
|
break;
|
|
}
|
|
}
|
|
#endif // __unix
|
|
|
|
#ifdef WIN32
|
|
#include <windows.h>
|
|
// return TRUE if handled, FALSE if not
|
|
BOOL WINAPI Win32CtrlC_Handler(DWORD CtrlType) {
|
|
switch (CtrlType) {
|
|
case CTRL_C_EVENT:
|
|
info("gracefully shutting down via CTRL+C");
|
|
Application::GracefullyShutdown();
|
|
return TRUE;
|
|
case CTRL_BREAK_EVENT:
|
|
info("gracefully shutting down via CTRL+BREAK");
|
|
Application::GracefullyShutdown();
|
|
return TRUE;
|
|
case CTRL_CLOSE_EVENT:
|
|
info("gracefully shutting down via close");
|
|
Application::GracefullyShutdown();
|
|
return TRUE;
|
|
}
|
|
// we dont care for any others like CTRL_LOGOFF_EVENT and CTRL_SHUTDOWN_EVENT
|
|
return FALSE;
|
|
}
|
|
#endif // WIN32
|
|
|
|
// clang-format off
|
|
static void SetupSignalHandlers() {
|
|
// signal handlers for unix
|
|
#ifdef __unix
|
|
trace("registering handlers for SIGINT, SIGTERM, SIGPIPE");
|
|
signal(SIGPIPE, UnixSignalHandler);
|
|
signal(SIGTERM, UnixSignalHandler);
|
|
#ifndef DEBUG
|
|
signal(SIGINT, UnixSignalHandler);
|
|
#endif // DEBUG
|
|
#endif // __unix
|
|
|
|
// signal handlers for win32
|
|
#ifdef WIN32
|
|
trace("registering handlers for CTRL_*_EVENTs");
|
|
SetConsoleCtrlHandler(Win32CtrlC_Handler, TRUE);
|
|
#endif // WIN32
|
|
}
|
|
// clang-format on
|