src/mangosd/Master.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 /** \file
  20     \ingroup mangosd
  21 */
  22
  23 #include <ace/OS_NS_signal.h>
  24
  25 #include "WorldSocketMgr.h"
  26 #include "Common.h"
  27 #include "Master.h"
  28 #include "WorldSocket.h"
  29 #include "WorldRunnable.h"
  30 #include "World.h"
  31 #include "Log.h"
  32 #include "Timer.h"
  33 #include "Policies/SingletonImp.h"
  34 #include "SystemConfig.h"
  35 #include "Config/ConfigEnv.h"
  36 #include "Database/DatabaseEnv.h"
  37 #include "CliRunnable.h"
  38 #include "RASocket.h"
  39 #include "ScriptCalls.h"
  40 #include "Util.h"
  41
  42 #include "sockets/TcpSocket.h"
  43 #include "sockets/Utility.h"
  44 #include "sockets/Parse.h"
  45 #include "sockets/Socket.h"
  46 #include "sockets/SocketHandler.h"
  47 #include "sockets/ListenSocket.h"
  48
  49 #ifdef WIN32
  50 #include "ServiceWin32.h"
  51 extern int m_ServiceStatus;
  52 #endif
  53
  54 /// \todo Warning disabling not useful under VC++2005. Can somebody say on which compiler it is useful?
  55 #pragma warning(disable:4305)
  56
  57 INSTANTIATE_SINGLETON_1( Master );
  58
  59 volatile uint32 Master::m_masterLoopCounter = 0;
  60
  61 class FreezeDetectorRunnable : public ZThread::Runnable
  62 {
  63 public:
  64     FreezeDetectorRunnable() { _delaytime = 0; }
  65     uint32 m_loops, m_lastchange;
  66     uint32 w_loops, w_lastchange;
  67     uint32 _delaytime;
  68     void SetDelayTime(uint32 t) { _delaytime = t; }
  69     void run(void)
  70     {
  71         if(!_delaytime)
  72             return;
  73         sLog.outString("Starting up anti-freeze thread (%u seconds max stuck time)...",_delaytime/1000);
  74         m_loops = 0;
  75         w_loops = 0;
  76         m_lastchange = 0;
  77         w_lastchange = 0;
  78         while(!World::IsStopped())
  79         {
  80             ZThread::Thread::sleep(1000);
  81             uint32 curtime = getMSTime();
  82             //DEBUG_LOG("anti-freeze: time=%u, counters=[%u; %u]",curtime,Master::m_masterLoopCounter,World::m_worldLoopCounter);
  83
  84             // There is no Master anymore
  85             // TODO: clear the rest of the code
  86 //            // normal work
  87 //            if(m_loops != Master::m_masterLoopCounter)
  88 //            {
  89 //                m_lastchange = curtime;
  90 //                m_loops = Master::m_masterLoopCounter;
  91 //            }
  92 //            // possible freeze
  93 //            else if(getMSTimeDiff(m_lastchange,curtime) > _delaytime)
  94 //            {
  95 //                sLog.outError("Main/Sockets Thread hangs, kicking out server!");
  96 //                *((uint32 volatile*)NULL) = 0;                       // bang crash
  97 //            }
  98
  99             // normal work
 100             if(w_loops != World::m_worldLoopCounter)
 101             {
 102                 w_lastchange = curtime;
 103                 w_loops = World::m_worldLoopCounter;
 104             }
 105             // possible freeze
 106             else if(getMSTimeDiff(w_lastchange,curtime) > _delaytime)
 107             {
 108                 sLog.outError("World Thread hangs, kicking out server!");
 109                 *((uint32 volatile*)NULL) = 0;                       // bang crash
 110             }
 111         }
 112         sLog.outString("Anti-freeze thread exiting without problems.");
 113     }
 114 };
 115
 116 class RARunnable : public ZThread::Runnable
 117 {
 118 public:
 119   uint32 numLoops, loopCounter;
 120
 121   RARunnable ()
 122   {
 123     uint32 socketSelecttime = sWorld.getConfig (CONFIG_SOCKET_SELECTTIME);
 124     numLoops = (sConfig.GetIntDefault ("MaxPingTime", 30) * (MINUTE * 1000000 / socketSelecttime));
 125     loopCounter = 0;
 126   }
 127
 128   void
 129   checkping ()
 130   {
 131     // ping if need
 132     if ((++loopCounter) == numLoops)
 133       {
 134         loopCounter = 0;
 135         sLog.outDetail ("Ping MySQL to keep connection alive");
 136         delete WorldDatabase.Query ("SELECT 1 FROM command LIMIT 1");
 137         delete loginDatabase.Query ("SELECT 1 FROM realmlist LIMIT 1");
 138         delete CharacterDatabase.Query ("SELECT 1 FROM bugreport LIMIT 1");
 139       }
 140   }
 141
 142   void
 143   run (void)
 144   {
 145     SocketHandler h;
 146
 147     // Launch the RA listener socket
 148     ListenSocket<RASocket> RAListenSocket (h);
 149     bool usera = sConfig.GetBoolDefault ("Ra.Enable", false);
 150
 151     if (usera)
 152       {
 153         port_t raport = sConfig.GetIntDefault ("Ra.Port", 3443);
 154         std::string stringip = sConfig.GetStringDefault ("Ra.IP", "0.0.0.0");
 155         ipaddr_t raip;
 156         if (!Utility::u2ip (stringip, raip))
 157           sLog.outError ("MaNGOS RA can not bind to ip %s", stringip.c_str ());
 158         else if (RAListenSocket.Bind (raip, raport))
 159           sLog.outError ("MaNGOS RA can not bind to port %d on %s", raport, stringip.c_str ());
 160         else
 161           {
 162             h.Add (&RAListenSocket);
 163
 164             sLog.outString ("Starting Remote access listner on port %d on %s", raport, stringip.c_str ());
 165           }
 166       }
 167
 168     // Socket Selet time is in microseconds , not miliseconds!!
 169     uint32 socketSelecttime = sWorld.getConfig (CONFIG_SOCKET_SELECTTIME);
 170
 171     // if use ra spend time waiting for io, if not use ra ,just sleep
 172     if (usera)
 173       while (!World::IsStopped())
 174         {
 175           h.Select (0, socketSelecttime);
 176           checkping ();
 177         }
 178     else
 179       while (!World::IsStopped())
 180         {
 181           ZThread::Thread::sleep (static_cast<unsigned long> (socketSelecttime / 1000));
 182           checkping ();
 183         }
 184   }
 185 };
 186
 187 Master::Master()
 188 {
 189 }
 190
 191 Master::~Master()
 192 {
 193 }
 194
 195 /// Main function
 196 int Master::Run()
 197 {
 198     /// worldd PID file creation
 199     std::string pidfile = sConfig.GetStringDefault("PidFile", "");
 200     if(!pidfile.empty())
 201     {
 202         uint32 pid = CreatePIDFile(pidfile);
 203         if( !pid )
 204         {
 205             sLog.outError( "Cannot create PID file %s.\n", pidfile.c_str() );
 206             return 1;
 207         }
 208
 209         sLog.outString( "Daemon PID: %u\n", pid );
 210     }
 211
 212     ///- Start the databases
 213     if (!_StartDB())
 214         return 1;
 215
 216     ///- Initialize the World
 217     sWorld.SetInitialWorldSettings();
 218
 219     ///- Catch termination signals
 220     _HookSignals();
 221
 222     ///- Launch WorldRunnable thread
 223     ZThread::Thread t(new WorldRunnable);
 224     t.setPriority ((ZThread::Priority )2);
 225
 226     // set server online
 227     loginDatabase.PExecute("UPDATE realmlist SET color = 0, population = 0 WHERE id = '%d'",realmID);
 228
 229 #ifdef WIN32
 230     if (sConfig.GetBoolDefault("Console.Enable", true) && (m_ServiceStatus == -1)/* need disable console in service mode*/)
 231 #else
 232     if (sConfig.GetBoolDefault("Console.Enable", true))
 233 #endif
 234     {
 235         ///- Launch CliRunnable thread
 236         ZThread::Thread td1(new CliRunnable);
 237     }
 238
 239     ZThread::Thread td2(new RARunnable);
 240
 241     ///- Handle affinity for multiple processors and process priority on Windows
 242     #ifdef WIN32
 243     {
 244         HANDLE hProcess = GetCurrentProcess();
 245
 246         uint32 Aff = sConfig.GetIntDefault("UseProcessors", 0);
 247         if(Aff > 0)
 248         {
 249             ULONG_PTR appAff;
 250             ULONG_PTR sysAff;
 251
 252             if(GetProcessAffinityMask(hProcess,&appAff,&sysAff))
 253             {
 254                 ULONG_PTR curAff = Aff & appAff;            // remove non accessible processors
 255
 256                 if(!curAff )
 257                 {
 258                     sLog.outError("Processors marked in UseProcessors bitmask (hex) %x not accessible for mangosd. Accessible processors bitmask (hex): %x",Aff,appAff);
 259                 }
 260                 else
 261                 {
 262                     if(SetProcessAffinityMask(hProcess,curAff))
 263                         sLog.outString("Using processors (bitmask, hex): %x", curAff);
 264                     else
 265                         sLog.outError("Can't set used processors (hex): %x",curAff);
 266                 }
 267             }
 268             sLog.outString();
 269         }
 270
 271         bool Prio = sConfig.GetBoolDefault("ProcessPriority", false);
 272
 273 //        if(Prio && (m_ServiceStatus == -1)/* need set to default process priority class in service mode*/)
 274         if(Prio)
 275         {
 276             if(SetPriorityClass(hProcess,HIGH_PRIORITY_CLASS))
 277                 sLog.outString("mangosd process priority class set to HIGH");
 278             else
 279                 sLog.outError("ERROR: Can't set mangosd process priority class.");
 280             sLog.outString();
 281         }
 282     }
 283     #endif
 284
 285     uint32 realCurrTime, realPrevTime;
 286     realCurrTime = realPrevTime = getMSTime();
 287
 288     uint32 socketSelecttime = sWorld.getConfig(CONFIG_SOCKET_SELECTTIME);
 289
 290     // maximum counter for next ping
 291     uint32 numLoops = (sConfig.GetIntDefault( "MaxPingTime", 30 ) * (MINUTE * 1000000 / socketSelecttime));
 292     uint32 loopCounter = 0;
 293
 294     ///- Start up freeze catcher thread
 295     uint32 freeze_delay = sConfig.GetIntDefault("MaxCoreStuckTime", 0);
 296     if(freeze_delay)
 297     {
 298         FreezeDetectorRunnable *fdr = new FreezeDetectorRunnable();
 299         fdr->SetDelayTime(freeze_delay*1000);
 300         ZThread::Thread t(fdr);
 301         t.setPriority(ZThread::High);
 302     }
 303
 304     ///- Launch the world listener socket
 305   port_t wsport = sWorld.getConfig (CONFIG_PORT_WORLD);
 306   std::string bind_ip = sConfig.GetStringDefault ("BindIP", "0.0.0.0");
 307
 308   if (sWorldSocketMgr->StartNetwork (wsport, bind_ip.c_str ()) == -1)
 309     {
 310       sLog.outError ("Failed to start network");
 311       World::StopNow(ERROR_EXIT_CODE);
 312       // go down and shutdown the server
 313     }
 314
 315     sWorldSocketMgr->Wait ();
 316
 317     // set server offline
 318     loginDatabase.PExecute("UPDATE realmlist SET color = 2 WHERE id = '%d'",realmID);
 319
 320     ///- Remove signal handling before leaving
 321     _UnhookSignals();
 322
 323     // when the main thread closes the singletons get unloaded
 324     // since worldrunnable uses them, it will crash if unloaded after master
 325     t.wait();
 326     td2.wait ();
 327
 328     ///- Clean database before leaving
 329     clearOnlineAccounts();
 330
 331     ///- Wait for delay threads to end
 332     CharacterDatabase.HaltDelayThread();
 333     WorldDatabase.HaltDelayThread();
 334     loginDatabase.HaltDelayThread();
 335
 336     sLog.outString( "Halting process..." );
 337
 338     #ifdef WIN32
 339     if (sConfig.GetBoolDefault("Console.Enable", true))
 340     {
 341         // this only way to terminate CLI thread exist at Win32 (alt. way exist only in Windows Vista API)
 342         //_exit(1);
 343         // send keyboard input to safely unblock the CLI thread
 344         INPUT_RECORD b[5];
 345         HANDLE hStdIn = GetStdHandle(STD_INPUT_HANDLE);
 346         b[0].EventType = KEY_EVENT;
 347         b[0].Event.KeyEvent.bKeyDown = TRUE;
 348         b[0].Event.KeyEvent.uChar.AsciiChar = 'X';
 349         b[0].Event.KeyEvent.wVirtualKeyCode = 'X';
 350         b[0].Event.KeyEvent.wRepeatCount = 1;
 351
 352         b[1].EventType = KEY_EVENT;
 353         b[1].Event.KeyEvent.bKeyDown = FALSE;
 354         b[1].Event.KeyEvent.uChar.AsciiChar = 'X';
 355         b[1].Event.KeyEvent.wVirtualKeyCode = 'X';
 356         b[1].Event.KeyEvent.wRepeatCount = 1;
 357
 358         b[2].EventType = KEY_EVENT;
 359         b[2].Event.KeyEvent.bKeyDown = TRUE;
 360         b[2].Event.KeyEvent.dwControlKeyState = 0;
 361         b[2].Event.KeyEvent.uChar.AsciiChar = '\r';
 362         b[2].Event.KeyEvent.wVirtualKeyCode = VK_RETURN;
 363         b[2].Event.KeyEvent.wRepeatCount = 1;
 364         b[2].Event.KeyEvent.wVirtualScanCode = 0x1c;
 365
 366         b[3].EventType = KEY_EVENT;
 367         b[3].Event.KeyEvent.bKeyDown = FALSE;
 368         b[3].Event.KeyEvent.dwControlKeyState = 0;
 369         b[3].Event.KeyEvent.uChar.AsciiChar = '\r';
 370         b[3].Event.KeyEvent.wVirtualKeyCode = VK_RETURN;
 371         b[3].Event.KeyEvent.wVirtualScanCode = 0x1c;
 372         b[3].Event.KeyEvent.wRepeatCount = 1;
 373         DWORD numb;
 374         BOOL ret = WriteConsoleInput(hStdIn, b, 4, &numb);
 375     }
 376     #endif
 377
 378     // for some unknown reason, unloading scripts here and not in worldrunnable
 379     // fixes a memory leak related to detaching threads from the module
 380     UnloadScriptingModule();
 381
 382     // Exit the process with specified return value
 383     return World::GetExitCode();
 384 }
 385
 386 /// Initialize connection to the databases
 387 bool Master::_StartDB()
 388 {
 389     ///- Get world database info from configuration file
 390     std::string dbstring;
 391     if(!sConfig.GetString("WorldDatabaseInfo", &dbstring))
 392     {
 393         sLog.outError("Database not specified in configuration file");
 394         return false;
 395     }
 396     sLog.outString("World Database: %s", dbstring.c_str());
 397
 398     ///- Initialise the world database
 399     if(!WorldDatabase.Initialize(dbstring.c_str()))
 400     {
 401         sLog.outError("Cannot connect to world database %s",dbstring.c_str());
 402         return false;
 403     }
 404
 405     if(!sConfig.GetString("CharacterDatabaseInfo", &dbstring))
 406     {
 407         sLog.outError("Character Database not specified in configuration file");
 408         return false;
 409     }
 410     sLog.outString("Character Database: %s", dbstring.c_str());
 411
 412     ///- Initialise the Character database
 413     if(!CharacterDatabase.Initialize(dbstring.c_str()))
 414     {
 415         sLog.outError("Cannot connect to Character database %s",dbstring.c_str());
 416         return false;
 417     }
 418
 419     ///- Get login database info from configuration file
 420     if(!sConfig.GetString("LoginDatabaseInfo", &dbstring))
 421     {
 422         sLog.outError("Login database not specified in configuration file");
 423         return false;
 424     }
 425
 426     ///- Initialise the login database
 427     sLog.outString("Login Database: %s", dbstring.c_str() );
 428     if(!loginDatabase.Initialize(dbstring.c_str()))
 429     {
 430         sLog.outError("Cannot connect to login database %s",dbstring.c_str());
 431         return false;
 432     }
 433
 434     ///- Get the realm Id from the configuration file
 435     realmID = sConfig.GetIntDefault("RealmID", 0);
 436     if(!realmID)
 437     {
 438         sLog.outError("Realm ID not defined in configuration file");
 439         return false;
 440     }
 441     sLog.outString("Realm running as realm ID %d", realmID);
 442
 443     ///- Clean the database before starting
 444     clearOnlineAccounts();
 445
 446     sWorld.LoadDBVersion();
 447
 448     sLog.outString("Using %s", sWorld.GetDBVersion());
 449     return true;
 450 }
 451
 452 /// Clear 'online' status for all accounts with characters in this realm
 453 void Master::clearOnlineAccounts()
 454 {
 455     // Cleanup online status for characters hosted at current realm
 456     /// \todo Only accounts with characters logged on *this* realm should have online status reset. Move the online column from 'account' to 'realmcharacters'?
 457     loginDatabase.PExecute(
 458         "UPDATE account SET online = 0 WHERE online > 0 "
 459         "AND id IN (SELECT acctid FROM realmcharacters WHERE realmid = '%d')",realmID);
 460
 461
 462     CharacterDatabase.Execute("UPDATE characters SET online = 0 WHERE online<>0");
 463 }
 464
 465 /// Handle termination signals
 466 void Master::_OnSignal(int s)
 467 {
 468     switch (s)
 469     {
 470         case SIGINT:
 471             World::StopNow(RESTART_EXIT_CODE);
 472             break;
 473         case SIGTERM:
 474         #ifdef _WIN32
 475         case SIGBREAK:
 476         #endif
 477             World::StopNow(SHUTDOWN_EXIT_CODE);
 478             break;
 479     }
 480
 481     signal(s, _OnSignal);
 482 }
 483
 484 /// Define hook '_OnSignal' for all termination signals
 485 void Master::_HookSignals()
 486 {
 487     signal(SIGINT, _OnSignal);
 488     signal(SIGTERM, _OnSignal);
 489     #ifdef _WIN32
 490     signal(SIGBREAK, _OnSignal);
 491     #endif
 492 }
 493
 494 /// Unhook the signals before leaving
 495 void Master::_UnhookSignals()
 496 {
 497     signal(SIGINT, 0);
 498     signal(SIGTERM, 0);
 499     #ifdef _WIN32
 500     signal(SIGBREAK, 0);
 501     #endif
 502 }