mirror of
https://github.com/mangosfour/server.git
synced 2025-12-25 22:37:04 +00:00
147 lines
2.9 KiB
C++
147 lines
2.9 KiB
C++
/**
|
|
* MaNGOS is a full featured server for World of Warcraft, supporting
|
|
* the following clients: 1.12.x, 2.4.3, 3.3.5a, 4.3.4a and 5.4.8
|
|
*
|
|
* Copyright (C) 2005-2020 MaNGOS <https://getmangos.eu>
|
|
*
|
|
* 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
|
|
*
|
|
* World of Warcraft, and all World of Warcraft or Warcraft art, images,
|
|
* and lore are copyrighted by Blizzard Entertainment, Inc.
|
|
*/
|
|
|
|
#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);
|
|
}
|
|
|
|
if (!freopen("/dev/null", "rt", stdin))
|
|
exit(EXIT_FAILURE);
|
|
if (!freopen("/dev/null", "wt", stdout))
|
|
exit(EXIT_FAILURE);
|
|
if (!freopen("/dev/null", "wt", stderr))
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
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;
|