src/realmd/Main.cpp

   1 /*
   2  * Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 /// \addtogroup realmd Realm Daemon
  20 /// @{
  21 /// \file
  22
  23 #include "Common.h"
  24 #include "Database/DatabaseEnv.h"
  25 #include "RealmList.h"
  26
  27 #include "Config/ConfigEnv.h"
  28 #include "Log.h"
  29 #include "sockets/ListenSocket.h"
  30 #include "AuthSocket.h"
  31 #include "SystemConfig.h"
  32 #include "revision.h"
  33 #include "revision_nr.h"
  34 #include "Util.h"
  35
  36 #ifdef WIN32
  37 #include "ServiceWin32.h"
  38 char serviceName[] = "realmd";
  39 char serviceLongName[] = "MaNGOS realmd service";
  40 char serviceDescription[] = "Massive Network Game Object Server";
  41 /*
  42  * -1 - not in service mode
  43  *  0 - stopped
  44  *  1 - running
  45  *  2 - paused
  46  */
  47 int m_ServiceStatus = -1;
  48 #endif
  49
  50 bool StartDB(std::string &dbstring);
  51 void UnhookSignals();
  52 void HookSignals();
  53
  54 bool stopEvent = false;                                     ///< Setting it to true stops the server
  55 RealmList m_realmList;                                      ///< Holds the list of realms for this server
  56
  57 DatabaseType dbRealmServer;                                 ///< Accessor to the realm server database
  58
  59 /// Print out the usage string for this program on the console.
  60 void usage(const char *prog)
  61 {
  62     sLog.outString("Usage: \n %s [<options>]\n"
  63         "    --version                print version and exist\n\r"
  64         "    -c config_file           use config_file as configuration file\n\r"
  65         #ifdef WIN32
  66         "    Running as service functions:\n\r"
  67         "    --service                run as service\n\r"
  68         "    -s install               install service\n\r"
  69         "    -s uninstall             uninstall service\n\r"
  70         #endif
  71         ,prog);
  72 }
  73
  74 /// Launch the realm server
  75 extern int main(int argc, char **argv)
  76 {
  77     ///- Command line parsing to get the configuration file name
  78     char const* cfg_file = _REALMD_CONFIG;
  79     int c=1;
  80     while( c < argc )
  81     {
  82         if( strcmp(argv[c],"-c") == 0)
  83         {
  84             if( ++c >= argc )
  85             {
  86                 sLog.outError("Runtime-Error: -c option requires an input argument");
  87                 usage(argv[0]);
  88                 return 1;
  89             }
  90             else
  91                 cfg_file = argv[c];
  92         }
  93
  94         if( strcmp(argv[c],"--version") == 0)
  95         {
  96             printf("%s\n", _FULLVERSION(REVISION_DATE,REVISION_TIME,REVISION_NR,REVISION_ID));
  97             return 0;
  98         }
  99
 100         #ifdef WIN32
 101         ////////////
 102         //Services//
 103         ////////////
 104         if( strcmp(argv[c],"-s") == 0)
 105         {
 106             if( ++c >= argc )
 107             {
 108                 sLog.outError("Runtime-Error: -s option requires an input argument");
 109                 usage(argv[0]);
 110                 return 1;
 111             }
 112             if( strcmp(argv[c],"install") == 0)
 113             {
 114                 if (WinServiceInstall())
 115                     sLog.outString("Installing service");
 116                 return 1;
 117             }
 118             else if( strcmp(argv[c],"uninstall") == 0)
 119             {
 120                 if(WinServiceUninstall())
 121                     sLog.outString("Uninstalling service");
 122                 return 1;
 123             }
 124             else
 125             {
 126                 sLog.outError("Runtime-Error: unsupported option %s",argv[c]);
 127                 usage(argv[0]);
 128                 return 1;
 129             }
 130         }
 131         if( strcmp(argv[c],"--service") == 0)
 132         {
 133             WinServiceRun();
 134         }
 135         ////
 136         #endif
 137         ++c;
 138     }
 139
 140     if (!sConfig.SetSource(cfg_file))
 141     {
 142         sLog.outError("Could not find configuration file %s.", cfg_file);
 143         return 1;
 144     }
 145
 146     sLog.outString( "%s [realm-daemon]", _FULLVERSION(REVISION_DATE,REVISION_TIME,REVISION_NR,REVISION_ID) );
 147     sLog.outString( "<Ctrl-C> to stop.\n" );
 148     sLog.outString("Using configuration file %s.", cfg_file);
 149
 150     ///- Check the version of the configuration file
 151     uint32 confVersion = sConfig.GetIntDefault("ConfVersion", 0);
 152     if (confVersion < _REALMDCONFVERSION)
 153     {
 154         sLog.outError("*****************************************************************************");
 155         sLog.outError(" WARNING: Your realmd.conf version indicates your conf file is out of date!");
 156         sLog.outError("          Please check for updates, as your current default values may cause");
 157         sLog.outError("          strange behavior.");
 158         sLog.outError("*****************************************************************************");
 159         clock_t pause = 3000 + clock();
 160
 161         while (pause > clock()) {}
 162     }
 163
 164     /// realmd PID file creation
 165     std::string pidfile = sConfig.GetStringDefault("PidFile", "");
 166     if(!pidfile.empty())
 167     {
 168         uint32 pid = CreatePIDFile(pidfile);
 169         if( !pid )
 170         {
 171             sLog.outError( "Cannot create PID file %s.\n", pidfile.c_str() );
 172             return 1;
 173         }
 174
 175         sLog.outString( "Daemon PID: %u\n", pid );
 176     }
 177
 178     ///- Initialize the database connection
 179     std::string dbstring;
 180     if(!StartDB(dbstring))
 181         return 1;
 182
 183     ///- Get the list of realms for the server
 184     m_realmList.Initialize(sConfig.GetIntDefault("RealmsStateUpdateDelay", 20));
 185     if (m_realmList.size() == 0)
 186     {
 187         sLog.outError("No valid realms specified.");
 188         return 1;
 189     }
 190
 191     ///- Launch the listening network socket
 192     port_t rmport = sConfig.GetIntDefault( "RealmServerPort", DEFAULT_REALMSERVER_PORT );
 193     std::string bind_ip = sConfig.GetStringDefault("BindIP", "0.0.0.0");
 194
 195     SocketHandler h;
 196     ListenSocket<AuthSocket> authListenSocket(h);
 197     if ( authListenSocket.Bind(bind_ip.c_str(),rmport))
 198     {
 199         sLog.outError( "MaNGOS realmd can not bind to %s:%d",bind_ip.c_str(), rmport );
 200         return 1;
 201     }
 202
 203     h.Add(&authListenSocket);
 204
 205     ///- Catch termination signals
 206     HookSignals();
 207
 208     ///- Handle affinity for multiple processors and process priority on Windows
 209     #ifdef WIN32
 210     {
 211         HANDLE hProcess = GetCurrentProcess();
 212
 213         uint32 Aff = sConfig.GetIntDefault("UseProcessors", 0);
 214         if(Aff > 0)
 215         {
 216             ULONG_PTR appAff;
 217             ULONG_PTR sysAff;
 218
 219             if(GetProcessAffinityMask(hProcess,&appAff,&sysAff))
 220             {
 221                 ULONG_PTR curAff = Aff & appAff;            // remove non accessible processors
 222
 223                 if(!curAff )
 224                 {
 225                     sLog.outError("Processors marked in UseProcessors bitmask (hex) %x not accessible for realmd. Accessible processors bitmask (hex): %x",Aff,appAff);
 226                 }
 227                 else
 228                 {
 229                     if(SetProcessAffinityMask(hProcess,curAff))
 230                         sLog.outString("Using processors (bitmask, hex): %x", curAff);
 231                     else
 232                         sLog.outError("Can't set used processors (hex): %x", curAff);
 233                 }
 234             }
 235             sLog.outString();
 236         }
 237
 238         bool Prio = sConfig.GetBoolDefault("ProcessPriority", false);
 239
 240         if(Prio)
 241         {
 242             if(SetPriorityClass(hProcess,HIGH_PRIORITY_CLASS))
 243                 sLog.outString("realmd process priority class set to HIGH");
 244             else
 245                 sLog.outError("ERROR: Can't set realmd process priority class.");
 246             sLog.outString();
 247         }
 248     }
 249     #endif
 250
 251     // maximum counter for next ping
 252     uint32 numLoops = (sConfig.GetIntDefault( "MaxPingTime", 30 ) * (MINUTE * 1000000 / 100000));
 253     uint32 loopCounter = 0;
 254
 255     ///- Wait for termination signal
 256     while (!stopEvent)
 257     {
 258
 259         h.Select(0, 100000);
 260
 261         if( (++loopCounter) == numLoops )
 262         {
 263             loopCounter = 0;
 264             sLog.outDetail("Ping MySQL to keep connection alive");
 265             delete dbRealmServer.Query("SELECT 1 FROM realmlist LIMIT 1");
 266         }
 267 #ifdef WIN32
 268         if (m_ServiceStatus == 0) stopEvent = true;
 269         while (m_ServiceStatus == 2) Sleep(1000);
 270 #endif
 271     }
 272
 273     ///- Wait for the delay thread to exit
 274     dbRealmServer.HaltDelayThread();
 275
 276     ///- Remove signal handling before leaving
 277     UnhookSignals();
 278
 279     sLog.outString( "Halting process..." );
 280     return 0;
 281 }
 282
 283 /// Handle termination signals
 284 /** Put the global variable stopEvent to 'true' if a termination signal is caught **/
 285 void OnSignal(int s)
 286 {
 287     switch (s)
 288     {
 289         case SIGINT:
 290         case SIGTERM:
 291             stopEvent = true;
 292             break;
 293         #ifdef _WIN32
 294         case SIGBREAK:
 295             stopEvent = true;
 296             break;
 297         #endif
 298     }
 299
 300     signal(s, OnSignal);
 301 }
 302
 303 /// Initialize connection to the database
 304 bool StartDB(std::string &dbstring)
 305 {
 306     if(!sConfig.GetString("LoginDatabaseInfo", &dbstring))
 307     {
 308         sLog.outError("Database not specified");
 309         return false;
 310     }
 311
 312     sLog.outString("Database: %s", dbstring.c_str() );
 313     if(!dbRealmServer.Initialize(dbstring.c_str()))
 314     {
 315         sLog.outError("Cannot connect to database");
 316         return false;
 317     }
 318
 319     return true;
 320 }
 321
 322 /// Define hook 'OnSignal' for all termination signals
 323 void HookSignals()
 324 {
 325     signal(SIGINT, OnSignal);
 326     signal(SIGTERM, OnSignal);
 327     #ifdef _WIN32
 328     signal(SIGBREAK, OnSignal);
 329     #endif
 330 }
 331
 332 /// Unhook the signals before leaving
 333 void UnhookSignals()
 334 {
 335     signal(SIGINT, 0);
 336     signal(SIGTERM, 0);
 337     #ifdef _WIN32
 338     signal(SIGBREAK, 0);
 339     #endif
 340 }
 341
 342 /// @}