nchebrov/azerothcore-wotlk
1
0
Fork 0
mirror of https://github.com/mod-playerbots/azerothcore-wotlk.git synced 2026-02-07 21:01:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(Core/SignalHandler): Replace ACE signal handling with std (#4877)

Browse Source
This commit is contained in:
Kargatum
2021-03-24 21:50:56 +07:00
committed by GitHub
parent e0d36be56e
commit c4c06a7734
3 changed files with 60 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/common/Utilities/SignalHandler.h
View File

@@ -4,26 +4,41 @@
* Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
*/
#ifndef __SIGNAL_HANDLER_H__
#define __SIGNAL_HANDLER_H__
#ifndef _SIGNAL_HANDLER_H_
#define _SIGNAL_HANDLER_H_
#include <ace/Event_Handler.h>
#include <csignal>
#include <unordered_set>
#include <mutex>
namespace acore
{
/// Handle termination signals
class SignalHandler : public ACE_Event_Handler
class SignalHandler
{
public:
int handle_signal(int SigNum, siginfo_t* = nullptr, ucontext_t* = nullptr) override
{
HandleSignal(SigNum);
return 0;
}
virtual void HandleSignal(int /*SigNum*/) { };
};
private:
std::unordered_set<int> _handled;
mutable std::mutex _mutex;
public:
bool handle_signal(int sig, void (*func)(int))
{
std::lock_guard lock(_mutex);
if (_handled.find(sig) != _handled.end())
return false;
_handled.insert(sig);
signal(sig, func);
return true;
}
~SignalHandler()
{
for (auto const& sig : _handled)
signal(sig, nullptr);
}
};
}
#endif /* __SIGNAL_HANDLER_H__ */
#endif // _SIGNAL_HANDLER_H_

28
src/server/authserver/Main.cpp
View File

@@ -15,7 +15,6 @@
#include <ace/Dev_Poll_Reactor.h>
#include <ace/TP_Reactor.h>
#include <ace/ACE.h>
#include <ace/Sig_Handler.h>
#include <openssl/opensslv.h>
#include <openssl/crypto.h>
@@ -46,22 +45,6 @@ bool stopEvent = false; // Setting it to tru
LoginDatabaseWorkerPool LoginDatabase; // Accessor to the authserver database
/// Handle authserver's termination signals
class AuthServerSignalHandler : public acore::SignalHandler
{
public:
void HandleSignal(int sigNum) override
{
switch (sigNum)
{
case SIGINT:
case SIGTERM:
stopEvent = true;
break;
}
}
};
/// Print out the usage string for this program on the console.
void usage(const char* prog)
{
@@ -180,12 +163,15 @@ extern int main(int argc, char** argv)
sLog->outString("Authserver listening to %s:%d", bind_ip.c_str(), rmport);
// Initialize the signal handlers
AuthServerSignalHandler SignalINT, SignalTERM;
acore::SignalHandler signalHandler;
auto const _handler = [](int) { stopEvent = true; };
// Register authservers's signal handlers
ACE_Sig_Handler Handler;
Handler.register_handler(SIGINT, &SignalINT);
Handler.register_handler(SIGTERM, &SignalTERM);
signalHandler.handle_signal(SIGINT, _handler);
signalHandler.handle_signal(SIGTERM, _handler);
#if AC_PLATFORM == AC_PLATFORM_WINDOWS
signalHandler.handle_signal(SIGBREAK, _handler);
#endif
#if defined(_WIN32) || defined(__linux__)

55
src/server/worldserver/Master.cpp
View File

@@ -43,30 +43,28 @@ extern int m_ServiceStatus;
#endif
/// Handle worldservers's termination signals
class WorldServerSignalHandler : public acore::SignalHandler
void HandleSignal(int sigNum)
{
public:
void HandleSignal(int sigNum) override
switch (sigNum)
{
switch (sigNum)
{
case SIGINT:
World::StopNow(RESTART_EXIT_CODE);
break;
case SIGTERM:
#ifdef _WIN32
case SIGBREAK:
if (m_ServiceStatus != 1)
case SIGINT:
World::StopNow(RESTART_EXIT_CODE);
break;
case SIGTERM:
#if AC_PLATFORM == AC_PLATFORM_WINDOWS
case SIGBREAK:
if (m_ServiceStatus != 1)
#endif
World::StopNow(SHUTDOWN_EXIT_CODE);
break;
/*case SIGSEGV:
sLog->outString("ZOMG! SIGSEGV handled!");
World::StopNow(SHUTDOWN_EXIT_CODE);
break;*/
}
World::StopNow(SHUTDOWN_EXIT_CODE);
break;
/*case SIGSEGV:
sLog->outString("ZOMG! SIGSEGV handled!");
World::StopNow(SHUTDOWN_EXIT_CODE);
break;*/
default:
break;
}
};
}
class FreezeDetectorRunnable : public acore::Runnable
{
@@ -164,19 +162,14 @@ int Master::Run()
sScriptMgr->OnStartup();
///- Initialize the signal handlers
WorldServerSignalHandler signalINT, signalTERM; //, signalSEGV
#ifdef _WIN32
WorldServerSignalHandler signalBREAK;
#endif /* _WIN32 */
acore::SignalHandler signalHandler;
///- Register worldserver's signal handlers
ACE_Sig_Handler handle;
handle.register_handler(SIGINT, &signalINT);
handle.register_handler(SIGTERM, &signalTERM);
#ifdef _WIN32
handle.register_handler(SIGBREAK, &signalBREAK);
signalHandler.handle_signal(SIGINT, &HandleSignal);
signalHandler.handle_signal(SIGTERM, &HandleSignal);
#if AC_PLATFORM == AC_PLATFORM_WINDOWS
signalHandler.handle_signal(SIGBREAK, &HandleSignal);
#endif
//handle.register_handler(SIGSEGV, &signalSEGV);
///- Launch WorldRunnable thread
acore::Thread worldThread(new WorldRunnable);