mangos-mirror/server
0
0
Fork 0
mirror of https://github.com/mangosfour/server.git synced 2025-12-15 10:37:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[11500] Allow run mangos as daemon (linux/other posix)

Browse source 
Signed-off-by: VladimirMangos <vladimir@getmangos.com>
This commit is contained in:
finomen 2011-05-17 19:22:56 +04:00 • committed by VladimirMangos
parent b022e398b7
commit d00592f692
7 changed files with 238 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

41
src/mangosd/Main.cpp
View file

@ -46,6 +46,8 @@ char serviceDescription[] = "Massive Network Game Object Server";
* 2 - paused
*/
int m_ServiceStatus = -1;
#else
#include "PosixDaemon.h"
#endif
DatabaseType WorldDatabase; ///< Accessor to the world database
@ -65,6 +67,10 @@ void usage(const char *prog)
" -s run run as service\n\r"
" -s install install service\n\r"
" -s uninstall uninstall service\n\r"
#else
" Running as daemon functions:\n\r"
" -s run run as daemon\n\r"
" -s stop stop daemon\n\r"
#endif
,prog);
}
@ -75,15 +81,19 @@ extern int main(int argc, char **argv)
///- Command line parsing
char const* cfg_file = _MANGOSD_CONFIG;
#ifdef WIN32
char const *options = ":c:s:";
#else
char const *options = ":c:";
#endif
ACE_Get_Opt cmd_opts(argc, argv, options);
cmd_opts.long_option("version", 'v');
if (!sConfig.SetSource(cfg_file))
{
sLog.outError("Could not find configuration file %s.", cfg_file);
Log::WaitBeforeContinueIfNeed();
return 1;
}
int option;
while ((option = cmd_opts()) != EOF)
{
@ -95,9 +105,9 @@ extern int main(int argc, char **argv)
case 'v':
printf("%s\n", _FULLVERSION(REVISION_DATE,REVISION_TIME,REVISION_NR,REVISION_ID));
return 0;
#ifdef WIN32
case 's':
{
#ifdef WIN32
const char *mode = cmd_opts.opt_arg();
if (!strcmp(mode, "install"))
@ -122,8 +132,22 @@ extern int main(int argc, char **argv)
return 1;
}
break;
#else
const char *mode = cmd_opts.opt_arg();
if (!strcmp(mode, "run"))
startDaemon(120);
else if (!strcmp(mode, "stop"))
stopDaemon();
else
{
sLog.outError("Runtime-Error: -%c unsupported argument %s", cmd_opts.opt_opt(), mode);
usage(argv[0]);
Log::WaitBeforeContinueIfNeed();
return 1;
}
#endif
}
break;
case ':':
sLog.outError("Runtime-Error: -%c option requires an input argument", cmd_opts.opt_opt());
usage(argv[0]);
@ -137,13 +161,6 @@ extern int main(int argc, char **argv)
}
}
if (!sConfig.SetSource(cfg_file))
{
sLog.outError("Could not find configuration file %s.", cfg_file);
Log::WaitBeforeContinueIfNeed();
return 1;
}
sLog.outString( "%s [world-daemon]", _FULLVERSION(REVISION_DATE,REVISION_TIME,REVISION_NR,REVISION_ID) );
sLog.outString( "<Ctrl-C> to stop.\n\n" );

7
src/mangosd/Master.cpp
View file

@ -20,6 +20,10 @@
\ingroup mangosd
*/
#ifndef WIN32
#include "PosixDaemon.h"
#endif
#include "WorldSocketMgr.h"
#include "Common.h"
#include "Master.h"
@ -197,6 +201,9 @@ int Master::Run()
///- Initialize the World
sWorld.SetInitialWorldSettings();
#ifndef WIN32
detachDaemon();
#endif
//server loaded successfully => enable async DB requests
//this is done to forbid any async transactions during server startup!
CharacterDatabase.AllowAsyncTransactions();

45
src/realmd/Main.cpp
View file

@ -54,6 +54,8 @@ char serviceDescription[] = "Massive Network Game Object Server";
* 2 - paused
*/
int m_ServiceStatus = -1;
#else
#include "PosixDaemon.h"
#endif
bool StartDB();
@ -75,6 +77,10 @@ void usage(const char *prog)
" -s run run as service\n\r"
" -s install install service\n\r"
" -s uninstall uninstall service\n\r"
#else
" Running as daemon functions:\n\r"
" -s run run as daemon\n\r"
" -s stop stop daemon\n\r"
#endif
,prog);
}
@ -85,16 +91,20 @@ extern int main(int argc, char **argv)
///- Command line parsing
char const* cfg_file = _REALMD_CONFIG;
#ifdef WIN32
char const *options = ":c:s:";
#else
char const *options = ":c:";
#endif
ACE_Get_Opt cmd_opts(argc, argv, options);
cmd_opts.long_option("version", 'v');
int option;
if (!sConfig.SetSource(cfg_file))
{
sLog.outError("Could not find configuration file %s.", cfg_file);
Log::WaitBeforeContinueIfNeed();
return 1;
}
while ((option = cmd_opts()) != EOF)
{
switch (option)
@ -105,9 +115,10 @@ extern int main(int argc, char **argv)
case 'v':
printf("%s\n", _FULLVERSION(REVISION_DATE,REVISION_TIME,REVISION_NR,REVISION_ID));
return 0;
#ifdef WIN32
case 's':
{
#ifdef WIN32
const char *mode = cmd_opts.opt_arg();
if (!strcmp(mode, "install"))
@ -131,9 +142,22 @@ extern int main(int argc, char **argv)
Log::WaitBeforeContinueIfNeed();
return 1;
}
break;
#else
const char *mode = cmd_opts.opt_arg();
if (!strcmp(mode, "run"))
startDaemon();
else if (!strcmp(mode, "stop"))
stopDaemon();
else
{
sLog.outError("Runtime-Error: -%c unsupported argument %s", cmd_opts.opt_opt(), mode);
usage(argv[0]);
Log::WaitBeforeContinueIfNeed();
return 1;
}
#endif
break;
}
case ':':
sLog.outError("Runtime-Error: -%c option requires an input argument", cmd_opts.opt_opt());
usage(argv[0]);
@ -147,12 +171,6 @@ extern int main(int argc, char **argv)
}
}
if (!sConfig.SetSource(cfg_file))
{
sLog.outError("Could not find configuration file %s.", cfg_file);
Log::WaitBeforeContinueIfNeed();
return 1;
}
sLog.Initialize();
sLog.outString( "%s [realm-daemon]", _FULLVERSION(REVISION_DATE,REVISION_TIME,REVISION_NR,REVISION_ID) );
@ -294,6 +312,9 @@ extern int main(int argc, char **argv)
uint32 numLoops = (sConfig.GetIntDefault( "MaxPingTime", 30 ) * (MINUTE * 1000000 / 100000));
uint32 loopCounter = 0;
#ifndef WIN32
detachDaemon();
#endif
///- Wait for termination signal
while (!stopEvent)
{

7
src/shared/CMakeLists.txt
View file

@ -20,7 +20,12 @@
file(GLOB_RECURSE shared_SRCS RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} *.cpp *.h)
# Exclude Win32 files
if(NOT WIN32)
if(WIN32)
list(REMOVE_ITEM shared_SRCS
PosixDaemon.h
PosixDaemon.cpp
)
else()
list(REMOVE_ITEM shared_SRCS
WheatyExceptionReport.cpp
WheatyExceptionReport.h

136
src/shared/PosixDaemon.cpp Normal file
View file

@ -0,0 +1,136 @@
/*
* Copyright (C) 2005-2011 MaNGOS <http://getmangos.com/>
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#include "Config/Config.h"
#include "PosixDaemon.h"
#include <cstdio>
#include <iostream>
#include <fstream>
pid_t parent_pid = 0, sid = 0;
void daemonSignal(int s)
{
if (getpid() != parent_pid)
{
return;
}
if (s == SIGUSR1)
{
exit(EXIT_SUCCESS);
}
if (sid) {
kill(sid, s);
}
exit(EXIT_FAILURE);
}
void startDaemon(uint32_t timeout)
{
parent_pid = getpid();
pid_t pid;
signal(SIGUSR1, daemonSignal);
signal(SIGINT, daemonSignal);
signal(SIGTERM, daemonSignal);
signal(SIGALRM, daemonSignal);
sid = pid = fork();
if (pid < 0) {
exit(EXIT_FAILURE);
}
if (pid > 0) {
alarm(timeout);
pause();
exit(EXIT_FAILURE);
}
umask(0);
sid = setsid();
if (sid < 0) {
exit(EXIT_FAILURE);
}
if ((chdir("/")) < 0) {
exit(EXIT_FAILURE);
}
freopen("/dev/null", "rt", stdin);
freopen("/dev/null", "wt", stdout);
freopen("/dev/null", "wt", stderr);
}
void stopDaemon()
{
std::string pidfile = sConfig.GetStringDefault("PidFile", "");
if(!pidfile.empty())
{
std::fstream pf(pidfile.c_str(), std::ios::in);
uint32_t pid = 0;
pf >> pid;
if (kill(pid, SIGINT) < 0)
{
std::cerr << "Unable to stop daemon" << std::endl;
exit(EXIT_FAILURE);
}
pf.close();
}
else
{
std::cerr << "No pid file specified" << std::endl;
}
exit(EXIT_SUCCESS);
}
void detachDaemon()
{
if (parent_pid)
{
kill(parent_pid, SIGUSR1);
}
}
void exitDaemon()
{
if (parent_pid && parent_pid != getpid())
{
kill(parent_pid, SIGTERM);
}
}
struct WatchDog
{
~WatchDog()
{
exitDaemon();
}
};
WatchDog dog;

24
src/shared/PosixDaemon.h Normal file
View file

@ -0,0 +1,24 @@
/*
* Copyright (C) 2005-2011 MaNGOS <http://getmangos.com/>
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#include "Common.h"
#include "Log.h"
void startDaemon(uint32_t timeout = 10);
void stopDaemon();
void detachDaemon();

2
src/shared/revision_nr.h
View file

@ -1,4 +1,4 @@
#ifndef __REVISION_NR_H__
#define __REVISION_NR_H__
#define REVISION_NR "11499"
#define REVISION_NR "11500"
#endif // __REVISION_NR_H__