src/inspircd.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *          the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 /* $Core */
  15 #include "inspircd.h"
  16 #include "inspircd_version.h"
  17 #include <signal.h>
  18
  19 #ifndef WIN32
  20         #include <dirent.h>
  21         #include <unistd.h>
  22         #include <sys/resource.h>
  23         #include <dlfcn.h>
  24         #include <getopt.h>
  25
  26         /* Some systems don't define RUSAGE_SELF. This should fix them. */
  27         #ifndef RUSAGE_SELF
  28                 #define RUSAGE_SELF 0
  29         #endif
  30
  31         #include <pwd.h> // setuid
  32         #include <grp.h> // setgid
  33 #endif
  34
  35 #include <fstream>
  36 #include "xline.h"
  37 #include "bancache.h"
  38 #include "socketengine.h"
  39 #include "inspircd_se_config.h"
  40 #include "socket.h"
  41 #include "command_parse.h"
  42 #include "exitcodes.h"
  43 #include "caller.h"
  44 #include "testsuite.h"
  45
  46 InspIRCd* SI = NULL;
  47 int* mysig = NULL;
  48
  49 /** Seperate from the other casemap tables so that code *can* still exclusively rely on RFC casemapping
  50  * if it must.
  51  *
  52  * This is provided as a pointer so that modules can change it to their custom mapping tables,
  53  * e.g. for national character support.
  54  */
  55 unsigned const char *national_case_insensitive_map = rfc_case_insensitive_map;
  56
  57
  58 /* Moved from exitcodes.h -- due to duplicate symbols -- Burlex
  59  * XXX this is a bit ugly. -- w00t
  60  */
  61 const char* ExitCodes[] =
  62 {
  63                 "No error", /* 0 */
  64                 "DIE command", /* 1 */
  65                 "execv() failed", /* 2 */
  66                 "Internal error", /* 3 */
  67                 "Config file error", /* 4 */
  68                 "Logfile error", /* 5 */
  69                 "POSIX fork failed", /* 6 */
  70                 "Bad commandline parameters", /* 7 */
  71                 "No ports could be bound", /* 8 */
  72                 "Can't write PID file", /* 9 */
  73                 "SocketEngine could not initialize", /* 10 */
  74                 "Refusing to start up as root", /* 11 */
  75                 "Found a <die> tag!", /* 12 */
  76                 "Couldn't load module on startup", /* 13 */
  77                 "Could not create windows forked process", /* 14 */
  78                 "Received SIGTERM", /* 15 */
  79                 "Bad command handler loaded", /* 16 */
  80                 "RegisterServiceCtrlHandler failed", /* 17 */
  81                 "UpdateSCMStatus failed", /* 18 */
  82                 "CreateEvent failed" /* 19 */
  83 };
  84
  85 template<typename T> static void DeleteZero(T*&n)
  86 {
  87         if (n != NULL)
  88         {
  89                 delete n;
  90                 n = NULL;
  91         }
  92 }
  93
  94 void InspIRCd::Cleanup()
  95 {
  96         if (Config)
  97         {
  98                 for (unsigned int i = 0; i < ports.size(); i++)
  99                 {
 100                         /* This calls the constructor and closes the listening socket */
 101                         delete ports[i];
 102                 }
 103
 104                 ports.clear();
 105         }
 106
 107         /* Close all client sockets, or the new process inherits them */
 108         for (std::vector<User*>::const_iterator i = this->Users->local_users.begin(); i != this->Users->local_users.end(); i++)
 109         {
 110                 this->Users->QuitUser((*i), "Server shutdown");
 111                 (*i)->CloseSocket();
 112         }
 113
 114         /* We do this more than once, so that any service providers get a
 115          * chance to be unhooked by the modules using them, but then get
 116          * a chance to be removed themsleves.
 117          *
 118          * XXX there may be a better way to do this with 1.2
 119          */
 120         for (int tries = 0; tries < 3; tries++)
 121         {
 122                 std::vector<std::string> module_names = Modules->GetAllModuleNames(0);
 123                 for (std::vector<std::string>::iterator k = module_names.begin(); k != module_names.end(); ++k)
 124                 {
 125                         /* Unload all modules, so they get a chance to clean up their listeners */
 126                         this->Modules->Unload(k->c_str());
 127                 }
 128         }
 129         /* Remove core commands */
 130         Parser->RemoveRFCCommands();
 131
 132         /* Cleanup Server Names */
 133         for(servernamelist::iterator itr = servernames.begin(); itr != servernames.end(); ++itr)
 134                 delete (*itr);
 135
 136         /* Delete objects dynamically allocated in constructor (destructor would be more appropriate, but we're likely exiting) */
 137         /* Must be deleted before modes as it decrements modelines */
 138         DeleteZero(this->Users);
 139         DeleteZero(this->FakeClient);
 140         DeleteZero(this->Modes);
 141         DeleteZero(this->XLines);
 142         DeleteZero(this->Parser);
 143         DeleteZero(this->stats);
 144         DeleteZero(this->Modules);
 145         DeleteZero(this->BanCache);
 146         DeleteZero(this->SNO);
 147         DeleteZero(this->Config);
 148         DeleteZero(this->Res);
 149         DeleteZero(this->chanlist);
 150         DeleteZero(this->PI);
 151         DeleteZero(this->Threads);
 152         DeleteZero(this->Timers);
 153         DeleteZero(this->SE);
 154         /* Close logging */
 155         this->Logs->CloseLogs();
 156         DeleteZero(this->Logs);
 157 }
 158
 159 void InspIRCd::Restart(const std::string &reason)
 160 {
 161         /* SendError flushes each client's queue,
 162          * regardless of writeability state
 163          */
 164         this->SendError(reason);
 165
 166         /* Figure out our filename (if theyve renamed it, we're boned) */
 167         std::string me;
 168
 169 #ifdef WINDOWS
 170         char module[MAX_PATH];
 171         if (GetModuleFileName(NULL, module, MAX_PATH))
 172                 me = module;
 173 #else
 174         me = Config->MyDir + "/inspircd";
 175 #endif
 176
 177         char** argv = Config->argv;
 178
 179         this->Cleanup();
 180
 181         if (execv(me.c_str(), argv) == -1)
 182         {
 183                 /* Will raise a SIGABRT if not trapped */
 184                 throw CoreException(std::string("Failed to execv()! error: ") + strerror(errno));
 185         }
 186 }
 187
 188 void InspIRCd::ResetMaxBans()
 189 {
 190         for (chan_hash::const_iterator i = chanlist->begin(); i != chanlist->end(); i++)
 191                 i->second->ResetMaxBans();
 192 }
 193
 194 /** Because hash_map doesnt free its buckets when we delete items (this is a 'feature')
 195  * we must occasionally rehash the hash (yes really).
 196  * We do this by copying the entries from the old hash to a new hash, causing all
 197  * empty buckets to be weeded out of the hash. We dont do this on a timer, as its
 198  * very expensive, so instead we do it when the user types /REHASH and expects a
 199  * short delay anyway.
 200  */
 201 void InspIRCd::RehashUsersAndChans()
 202 {
 203         user_hash* old_users = this->Users->clientlist;
 204         user_hash* old_uuid  = this->Users->uuidlist;
 205         chan_hash* old_chans = this->chanlist;
 206
 207         this->Users->clientlist = new user_hash();
 208         this->Users->uuidlist = new user_hash();
 209         this->chanlist = new chan_hash();
 210
 211         for (user_hash::const_iterator n = old_users->begin(); n != old_users->end(); n++)
 212                 this->Users->clientlist->insert(*n);
 213
 214         delete old_users;
 215
 216         for (user_hash::const_iterator n = old_uuid->begin(); n != old_uuid->end(); n++)
 217                 this->Users->uuidlist->insert(*n);
 218
 219         delete old_uuid;
 220
 221         for (chan_hash::const_iterator n = old_chans->begin(); n != old_chans->end(); n++)
 222                 this->chanlist->insert(*n);
 223
 224         delete old_chans;
 225 }
 226
 227 void InspIRCd::SetSignals()
 228 {
 229 #ifndef WIN32
 230         signal(SIGALRM, SIG_IGN);
 231         signal(SIGHUP, InspIRCd::SetSignal);
 232         signal(SIGPIPE, SIG_IGN);
 233         signal(SIGCHLD, SIG_IGN);
 234         /* We want E2BIG not a signal! */
 235         signal(SIGXFSZ, SIG_IGN);
 236 #endif
 237         signal(SIGTERM, InspIRCd::SetSignal);
 238 }
 239
 240 void InspIRCd::QuickExit(int status)
 241 {
 242         exit(0);
 243 }
 244
 245 bool InspIRCd::DaemonSeed()
 246 {
 247 #ifdef WINDOWS
 248         printf_c("InspIRCd Process ID: \033[1;32m%lu\033[0m\n", GetCurrentProcessId());
 249         return true;
 250 #else
 251         signal(SIGTERM, InspIRCd::QuickExit);
 252
 253         int childpid;
 254         if ((childpid = fork ()) < 0)
 255                 return false;
 256         else if (childpid > 0)
 257         {
 258                 /* We wait here for the child process to kill us,
 259                  * so that the shell prompt doesnt come back over
 260                  * the output.
 261                  * Sending a kill with a signal of 0 just checks
 262                  * if the child pid is still around. If theyre not,
 263                  * they threw an error and we should give up.
 264                  */
 265                 while (kill(childpid, 0) != -1)
 266                         sleep(1);
 267                 exit(0);
 268         }
 269         setsid ();
 270         umask (007);
 271         printf("InspIRCd Process ID: \033[1;32m%lu\033[0m\n",(unsigned long)getpid());
 272
 273         signal(SIGTERM, InspIRCd::SetSignal);
 274
 275         rlimit rl;
 276         if (getrlimit(RLIMIT_CORE, &rl) == -1)
 277         {
 278                 this->Logs->Log("STARTUP",DEFAULT,"Failed to getrlimit()!");
 279                 return false;
 280         }
 281         else
 282         {
 283                 rl.rlim_cur = rl.rlim_max;
 284                 if (setrlimit(RLIMIT_CORE, &rl) == -1)
 285                         this->Logs->Log("STARTUP",DEFAULT,"setrlimit() failed, cannot increase coredump size.");
 286         }
 287
 288         return true;
 289 #endif
 290 }
 291
 292 void InspIRCd::WritePID(const std::string &filename)
 293 {
 294         std::string fname = (filename.empty() ? "inspircd.pid" : filename);
 295         std::replace(fname.begin(), fname.end(), '\\', '/');
 296         if ((fname[0] != '/') && (!Config->StartsWithWindowsDriveLetter(filename)))
 297         {
 298                 std::string::size_type pos;
 299                 std::string confpath = this->ConfigFileName;
 300                 if ((pos = confpath.rfind("/")) != std::string::npos)
 301                 {
 302                         /* Leaves us with just the path */
 303                         fname = confpath.substr(0, pos) + std::string("/") + fname;
 304                 }
 305         }
 306         std::ofstream outfile(fname.c_str());
 307         if (outfile.is_open())
 308         {
 309                 outfile << getpid();
 310                 outfile.close();
 311         }
 312         else
 313         {
 314                 printf("Failed to write PID-file '%s', exiting.\n",fname.c_str());
 315                 this->Logs->Log("STARTUP",DEFAULT,"Failed to write PID-file '%s', exiting.",fname.c_str());
 316                 Exit(EXIT_STATUS_PID);
 317         }
 318 }
 319
 320 InspIRCd::InspIRCd(int argc, char** argv)
 321         : GlobalCulls(this),
 322
 323          /* Functor initialisation. Note that the ordering here is very important.
 324           *
 325           * THIS MUST MATCH ORDER OF DECLARATION OF THE HandleWhateverFunc classes
 326           * within class InspIRCd.
 327           */
 328          HandleProcessUser(this),
 329          HandleIsNick(this),
 330          HandleIsIdent(this),
 331          HandleFloodQuitUser(this),
 332          HandleIsChannel(this),
 333          HandleIsSID(this),
 334          HandleRehash(this),
 335
 336          /* Functor pointer initialisation. Must match the order of the list above
 337           *
 338           * THIS MUST MATCH THE ORDER OF DECLARATION OF THE FUNCTORS, e.g. the methods
 339           * themselves within the class.
 340           */
 341          ProcessUser(&HandleProcessUser),
 342          IsChannel(&HandleIsChannel),
 343          IsSID(&HandleIsSID),
 344          Rehash(&HandleRehash),
 345          IsNick(&HandleIsNick),
 346          IsIdent(&HandleIsIdent),
 347          FloodQuitUser(&HandleFloodQuitUser)
 348
 349 {
 350 #ifdef WIN32
 351         // Strict, frequent checking of memory on debug builds
 352         _CrtSetDbgFlag ( _CRTDBG_CHECK_ALWAYS_DF | _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF );
 353
 354         // Avoid erroneous frees on early exit
 355         WindowsIPC = 0;
 356 #endif
 357         FailedPortList pl;
 358         int do_version = 0, do_nofork = 0, do_debug = 0,
 359             do_nolog = 0, do_root = 0, do_testsuite = 0;    /* flag variables */
 360         int c = 0;
 361
 362         // Initialize so that if we exit before proper initialization they're not deleted
 363         this->Logs = 0;
 364         this->Threads = 0;
 365         this->PI = 0;
 366         this->Users = 0;
 367         this->chanlist = 0;
 368         this->Config = 0;
 369         this->SNO = 0;
 370         this->BanCache = 0;
 371         this->Modules = 0;
 372         this->stats = 0;
 373         this->Timers = 0;
 374         this->Parser = 0;
 375         this->XLines = 0;
 376         this->Modes = 0;
 377         this->Res = 0;
 378         this->ConfigThread = NULL;
 379
 380         // Initialise TIME
 381         this->TIME = time(NULL);
 382
 383         // This must be created first, so other parts of Insp can use it while starting up
 384         this->Logs = new LogManager(this);
 385
 386         SocketEngineFactory* SEF = new SocketEngineFactory();
 387         SE = SEF->Create(this);
 388         delete SEF;
 389
 390         this->Threads = new ThreadEngine(this);
 391
 392         /* Default implementation does nothing */
 393         this->PI = new ProtocolInterface(this);
 394
 395         this->s_signal = 0;
 396
 397         // Create base manager classes early, so nothing breaks
 398         this->Users = new UserManager(this);
 399
 400         this->Users->unregistered_count = 0;
 401
 402         this->Users->clientlist = new user_hash();
 403         this->Users->uuidlist = new user_hash();
 404         this->chanlist = new chan_hash();
 405
 406         this->Config = new ServerConfig(this);
 407         this->SNO = new SnomaskManager(this);
 408         this->BanCache = new BanCacheManager(this);
 409         this->Modules = new ModuleManager(this);
 410         this->stats = new serverstats();
 411         this->Timers = new TimerManager(this);
 412         this->Parser = new CommandParser(this);
 413         this->XLines = new XLineManager(this);
 414
 415         this->Config->argv = argv;
 416         this->Config->argc = argc;
 417
 418         if (chdir(Config->GetFullProgDir().c_str()))
 419         {
 420                 printf("Unable to change to my directory: %s\nAborted.", strerror(errno));
 421                 exit(0);
 422         }
 423
 424         this->Config->opertypes.clear();
 425         this->Config->operclass.clear();
 426
 427         this->TIME = this->OLDTIME = this->startup_time = time(NULL);
 428         srand(this->TIME);
 429
 430         *this->LogFileName = 0;
 431         strlcpy(this->ConfigFileName, CONFIG_FILE, MAXBUF);
 432
 433         struct option longopts[] =
 434         {
 435                 { "nofork",     no_argument,            &do_nofork,     1       },
 436                 { "logfile",    required_argument,      NULL,           'f'     },
 437                 { "config",     required_argument,      NULL,           'c'     },
 438                 { "debug",      no_argument,            &do_debug,      1       },
 439                 { "nolog",      no_argument,            &do_nolog,      1       },
 440                 { "runasroot",  no_argument,            &do_root,       1       },
 441                 { "version",    no_argument,            &do_version,    1       },
 442                 { "testsuite",  no_argument,            &do_testsuite,  1       },
 443                 { 0, 0, 0, 0 }
 444         };
 445
 446         int index;
 447         while ((c = getopt_long_only(argc, argv, ":f:", longopts, &index)) != -1)
 448         {
 449                 switch (c)
 450                 {
 451                         case 'f':
 452                                 /* Log filename was set */
 453                                 strlcpy(LogFileName, optarg, MAXBUF);
 454                         break;
 455                         case 'c':
 456                                 /* Config filename was set */
 457                                 strlcpy(ConfigFileName, optarg, MAXBUF);
 458                         break;
 459                         case 0:
 460                                 /* getopt_long_only() set an int variable, just keep going */
 461                         break;
 462                         case '?':
 463                                 /* Unknown parameter */
 464                         default:
 465                                 /* Fall through to handle other weird values too */
 466                                 printf("Unknown parameter '%s'\n", argv[optind-1]);
 467                                 printf("Usage: %s [--nofork] [--nolog] [--debug] [--logfile <filename>]\n%*s[--runasroot] [--version] [--config <config>] [--testsuite]\n", argv[0], static_cast<int>(8+strlen(argv[0])), " ");
 468                                 Exit(EXIT_STATUS_ARGV);
 469                         break;
 470                 }
 471         }
 472
 473         if (do_testsuite)
 474                 do_nofork = do_debug = true;
 475
 476         if (do_version)
 477         {
 478                 printf("\n%s r%s\n", VERSION, REVISION);
 479                 Exit(EXIT_STATUS_NOERROR);
 480         }
 481
 482 #ifdef WIN32
 483
 484         // Handle forking
 485         if(!do_nofork)
 486         {
 487                 DWORD ExitCode = WindowsForkStart(this);
 488                 if(ExitCode)
 489                         exit(ExitCode);
 490         }
 491
 492         // Set up winsock
 493         WSADATA wsadata;
 494         WSAStartup(MAKEWORD(2,0), &wsadata);
 495         ChangeWindowsSpecificPointers(this);
 496 #endif
 497         Config->MyExecutable = argv[0];
 498
 499         /* Set the finished argument values */
 500         Config->nofork = do_nofork;
 501         Config->forcedebug = do_debug;
 502         Config->writelog = !do_nolog;
 503         Config->TestSuite = do_testsuite;
 504
 505         if (!this->OpenLog(argv, argc))
 506         {
 507                 printf("ERROR: Could not open logfile %s: %s\n\n", Config->logpath.c_str(), strerror(errno));
 508                 Exit(EXIT_STATUS_LOG);
 509         }
 510
 511         if (!ServerConfig::FileExists(this->ConfigFileName))
 512         {
 513 #ifdef WIN32
 514                 /* Windows can (and defaults to) hide file extensions, so let's play a bit nice for windows users. */
 515                 std::string txtconf = this->ConfigFileName;
 516                 txtconf.append(".txt");
 517
 518                 if (ServerConfig::FileExists(txtconf.c_str()))
 519                 {
 520                         strlcat(this->ConfigFileName, ".txt", MAXBUF);
 521                 }
 522                 else
 523 #endif
 524                 {
 525                         printf("ERROR: Cannot open config file: %s\nExiting...\n", this->ConfigFileName);
 526                         this->Logs->Log("STARTUP",DEFAULT,"Unable to open config file %s", this->ConfigFileName);
 527                         Exit(EXIT_STATUS_CONFIG);
 528                 }
 529         }
 530
 531         printf_c("\033[1;32mInspire Internet Relay Chat Server, compiled %s at %s\n",__DATE__,__TIME__);
 532         printf_c("(C) InspIRCd Development Team.\033[0m\n\n");
 533         printf_c("Developers:\n");
 534         printf_c("\t\033[1;32mBrain, FrostyCoolSlug, w00t, Om, Special\n");
 535         printf_c("\t\033[1;32mpeavey, aquanight, psychon, dz, danieldg\033[0m\n\n");
 536         printf_c("Others:\t\t\t\033[1;32mSee /INFO Output\033[0m\n");
 537
 538         this->Modes = new ModeParser(this);
 539
 540         if (!do_root)
 541                 this->CheckRoot();
 542         else
 543         {
 544                 printf("* WARNING * WARNING * WARNING * WARNING * WARNING * \n\n");
 545                 printf("YOU ARE RUNNING INSPIRCD AS ROOT. THIS IS UNSUPPORTED\n");
 546                 printf("AND IF YOU ARE HACKED, CRACKED, SPINDLED OR MUTILATED\n");
 547                 printf("OR ANYTHING ELSE UNEXPECTED HAPPENS TO YOU OR YOUR\n");
 548                 printf("SERVER, THEN IT IS YOUR OWN FAULT. IF YOU DID NOT MEAN\n");
 549                 printf("TO START INSPIRCD AS ROOT, HIT CTRL+C NOW AND RESTART\n");
 550                 printf("THE PROGRAM AS A NORMAL USER. YOU HAVE BEEN WARNED!\n");
 551                 printf("\nInspIRCd starting in 20 seconds, ctrl+c to abort...\n");
 552                 sleep(20);
 553         }
 554
 555         this->SetSignals();
 556
 557         if (!Config->nofork)
 558         {
 559                 if (!this->DaemonSeed())
 560                 {
 561                         printf("ERROR: could not go into daemon mode. Shutting down.\n");
 562                         Logs->Log("STARTUP", DEFAULT, "ERROR: could not go into daemon mode. Shutting down.");
 563                         Exit(EXIT_STATUS_FORK);
 564                 }
 565         }
 566
 567         SE->RecoverFromFork();
 568
 569         /* During startup we don't actually initialize this
 570          * in the thread engine.
 571          */
 572         this->Config->Read();
 573         this->Config->Apply(NULL, "");
 574         Logs->OpenFileLogs();
 575
 576         /** Note: This is safe, the method checks for user == NULL */
 577         this->Parser->SetupCommandTable();
 578
 579         this->Res = new DNS(this);
 580
 581         this->AddServerName(Config->ServerName);
 582
 583         /*
 584          * Initialise SID/UID.
 585          * For an explanation as to exactly how this works, and why it works this way, see GetUID().
 586          *   -- w00t
 587          */
 588         if (!*Config->sid)
 589         {
 590                 // Generate one
 591                 size_t sid = 0;
 592
 593                 for (const char* x = Config->ServerName; *x; ++x)
 594                         sid = 5 * sid + *x;
 595                 for (const char* y = Config->ServerDesc; *y; ++y)
 596                         sid = 5 * sid + *y;
 597                 sid = sid % 999;
 598
 599                 Config->sid[0] = (char)(sid / 100 + 48);
 600                 Config->sid[1] = (char)(((sid / 10) % 10) + 48);
 601                 Config->sid[2] = (char)(sid % 10 + 48);
 602                 Config->sid[3] = '\0';
 603         }
 604
 605         /* set up fake client again this time with the correct uid */
 606         this->FakeClient = new FakeUser(this, "!");
 607         this->FakeClient->SetFakeServer(Config->ServerName);
 608
 609         // Get XLine to do it's thing.
 610         this->XLines->CheckELines();
 611         this->XLines->ApplyLines();
 612
 613         int bounditems = BindPorts(pl);
 614
 615         printf("\n");
 616
 617         this->Modules->LoadAll();
 618
 619         /* Just in case no modules were loaded - fix for bug #101 */
 620         this->BuildISupport();
 621         Config->ApplyDisabledCommands(Config->DisabledCommands);
 622
 623         if (!pl.empty())
 624         {
 625                 printf("\nWARNING: Not all your client ports could be bound --\nstarting anyway with %d of %d client ports bound.\n\n",
 626                         bounditems, bounditems + (int)pl.size());
 627                 printf("The following port(s) failed to bind:\n");
 628                 printf("Hint: Try using a public IP instead of blank or *\n\n");
 629                 int j = 1;
 630                 for (FailedPortList::iterator i = pl.begin(); i != pl.end(); i++, j++)
 631                 {
 632                         printf("%d.\tAddress: %s\tReason: %s\n", j, i->first.empty() ? "<all>" : i->first.c_str(), i->second.c_str());
 633                 }
 634         }
 635
 636         printf("\nInspIRCd is now running as '%s'[%s] with %d max open sockets\n", Config->ServerName,Config->GetSID().c_str(), SE->GetMaxFds());
 637
 638 #ifndef WINDOWS
 639         if (!Config->nofork)
 640         {
 641                 if (kill(getppid(), SIGTERM) == -1)
 642                 {
 643                         printf("Error killing parent process: %s\n",strerror(errno));
 644                         Logs->Log("STARTUP", DEFAULT, "Error killing parent process: %s",strerror(errno));
 645                 }
 646         }
 647
 648         if (isatty(0) && isatty(1) && isatty(2))
 649         {
 650                 /* We didn't start from a TTY, we must have started from a background process -
 651                  * e.g. we are restarting, or being launched by cron. Dont kill parent, and dont
 652                  * close stdin/stdout
 653                  */
 654                 if ((!do_nofork) && (!do_testsuite))
 655                 {
 656                         fclose(stdin);
 657                         fclose(stderr);
 658                         fclose(stdout);
 659                 }
 660                 else
 661                 {
 662                         Logs->Log("STARTUP", DEFAULT,"Keeping pseudo-tty open as we are running in the foreground.");
 663                 }
 664         }
 665 #else
 666         WindowsIPC = new IPC(this);
 667         if(!Config->nofork)
 668         {
 669                 WindowsForkKillOwner(this);
 670                 FreeConsole();
 671         }
 672         /* Set win32 service as running, if we are running as a service */
 673         SetServiceRunning();
 674 #endif
 675
 676         Logs->Log("STARTUP", DEFAULT, "Startup complete as '%s'[%s], %d max open sockets", Config->ServerName,Config->GetSID().c_str(), SE->GetMaxFds());
 677
 678 #ifndef WIN32
 679         if (*(this->Config->SetGroup))
 680         {
 681                 int ret;
 682
 683                 // setgroups
 684                 ret = setgroups(0, NULL);
 685
 686                 if (ret == -1)
 687                 {
 688                         this->Logs->Log("SETGROUPS", DEFAULT, "setgroups() failed (wtf?): %s", strerror(errno));
 689                         this->QuickExit(0);
 690                 }
 691
 692                 // setgid
 693                 struct group *g;
 694
 695                 errno = 0;
 696                 g = getgrnam(this->Config->SetGroup);
 697
 698                 if (!g)
 699                 {
 700                         this->Logs->Log("SETGUID", DEFAULT, "getgrnam() failed (bad user?): %s", strerror(errno));
 701                         this->QuickExit(0);
 702                 }
 703
 704                 ret = setgid(g->gr_gid);
 705
 706                 if (ret == -1)
 707                 {
 708                         this->Logs->Log("SETGUID", DEFAULT, "setgid() failed (bad user?): %s", strerror(errno));
 709                         this->QuickExit(0);
 710                 }
 711         }
 712
 713         if (*(this->Config->SetUser))
 714         {
 715                 // setuid
 716                 struct passwd *u;
 717
 718                 errno = 0;
 719                 u = getpwnam(this->Config->SetUser);
 720
 721                 if (!u)
 722                 {
 723                         this->Logs->Log("SETGUID", DEFAULT, "getpwnam() failed (bad user?): %s", strerror(errno));
 724                         this->QuickExit(0);
 725                 }
 726
 727                 int ret = setuid(u->pw_uid);
 728
 729                 if (ret == -1)
 730                 {
 731                         this->Logs->Log("SETGUID", DEFAULT, "setuid() failed (bad user?): %s", strerror(errno));
 732                         this->QuickExit(0);
 733                 }
 734         }
 735 #endif
 736
 737         this->WritePID(Config->PID);
 738 }
 739
 740 int InspIRCd::Run()
 741 {
 742         /* See if we're supposed to be running the test suite rather than entering the mainloop */
 743         if (Config->TestSuite)
 744         {
 745                 TestSuite* ts = new TestSuite(this);
 746                 delete ts;
 747                 Exit(0);
 748         }
 749
 750         while (true)
 751         {
 752 #ifndef WIN32
 753                 static rusage ru;
 754 #else
 755                 static time_t uptime;
 756                 static struct tm * stime;
 757                 static char window_title[100];
 758 #endif
 759
 760                 /* Check if there is a config thread which has finished executing but has not yet been freed */
 761                 if (this->ConfigThread && this->ConfigThread->IsDone())
 762                 {
 763                         /* Rehash has completed */
 764                         this->Logs->Log("CONFIG",DEBUG,"Detected ConfigThread exiting, tidying up...");
 765
 766                         this->ConfigThread->Finish();
 767
 768                         ConfigThread->join();
 769                         delete ConfigThread;
 770                         ConfigThread = NULL;
 771                 }
 772
 773                 /* time() seems to be a pretty expensive syscall, so avoid calling it too much.
 774                  * Once per loop iteration is pleanty.
 775                  */
 776                 OLDTIME = TIME;
 777                 TIME = time(NULL);
 778
 779                 /* Run background module timers every few seconds
 780                  * (the docs say modules shouldnt rely on accurate
 781                  * timing using this event, so we dont have to
 782                  * time this exactly).
 783                  */
 784                 if (TIME != OLDTIME)
 785                 {
 786                         /* Allow a buffer of two seconds drift on this so that ntpdate etc dont harass admins */
 787                         if (TIME < OLDTIME - 2)
 788                         {
 789                                 SNO->WriteToSnoMask('d', "\002EH?!\002 -- Time is flowing BACKWARDS in this dimension! Clock drifted backwards %lu secs.", (unsigned long)OLDTIME-TIME);
 790                         }
 791                         else if (TIME > OLDTIME + 2)
 792                         {
 793                                 SNO->WriteToSnoMask('d', "\002EH?!\002 -- Time is jumping FORWARDS! Clock skipped %lu secs.", (unsigned long)TIME - OLDTIME);
 794                         }
 795
 796                         if ((TIME % 3600) == 0)
 797                         {
 798                                 this->RehashUsersAndChans();
 799                                 FOREACH_MOD_I(this, I_OnGarbageCollect, OnGarbageCollect());
 800                         }
 801
 802                         Timers->TickTimers(TIME);
 803                         this->DoBackgroundUserStuff();
 804
 805                         if ((TIME % 5) == 0)
 806                         {
 807                                 FOREACH_MOD_I(this,I_OnBackgroundTimer,OnBackgroundTimer(TIME));
 808                                 SNO->FlushSnotices();
 809                         }
 810 #ifndef WIN32
 811                         /* Same change as in cmd_stats.cpp, use RUSAGE_SELF rather than '0' -- Om */
 812                         if (!getrusage(RUSAGE_SELF, &ru))
 813                         {
 814                                 gettimeofday(&this->stats->LastSampled, NULL);
 815                                 this->stats->LastCPU = ru.ru_utime;
 816                         }
 817 #else
 818                         WindowsIPC->Check();
 819 #endif
 820                 }
 821
 822                 /* Call the socket engine to wait on the active
 823                  * file descriptors. The socket engine has everything's
 824                  * descriptors in its list... dns, modules, users,
 825                  * servers... so its nice and easy, just one call.
 826                  * This will cause any read or write events to be
 827                  * dispatched to their handlers.
 828                  */
 829                 this->SE->DispatchEvents();
 830
 831                 /* if any users were quit, take them out */
 832                 this->GlobalCulls.Apply();
 833
 834                 /* If any inspsockets closed, remove them */
 835                 this->BufferedSocketCull();
 836
 837                 if (this->s_signal)
 838                 {
 839                         this->SignalHandler(s_signal);
 840                         this->s_signal = 0;
 841                 }
 842         }
 843
 844         return 0;
 845 }
 846
 847 void InspIRCd::BufferedSocketCull()
 848 {
 849         for (std::map<BufferedSocket*,BufferedSocket*>::iterator x = SocketCull.begin(); x != SocketCull.end(); ++x)
 850         {
 851                 this->Logs->Log("MISC",DEBUG,"Cull socket");
 852                 SE->DelFd(x->second);
 853                 x->second->Close();
 854                 delete x->second;
 855         }
 856         SocketCull.clear();
 857 }
 858
 859 /**********************************************************************************/
 860
 861 /**
 862  * An ircd in five lines! bwahahaha. ahahahahaha. ahahah *cough*.
 863  */
 864
 865 /* this returns true when all modules are satisfied that the user should be allowed onto the irc server
 866  * (until this returns true, a user will block in the waiting state, waiting to connect up to the
 867  * registration timeout maximum seconds)
 868  */
 869 bool InspIRCd::AllModulesReportReady(User* user)
 870 {
 871         for (EventHandlerIter i = Modules->EventHandlers[I_OnCheckReady].begin(); i != Modules->EventHandlers[I_OnCheckReady].end(); ++i)
 872         {
 873                 if (!(*i)->OnCheckReady(user))
 874                         return false;
 875         }
 876         return true;
 877 }
 878
 879 time_t InspIRCd::Time()
 880 {
 881         return TIME;
 882 }
 883
 884 void InspIRCd::SetSignal(int signal)
 885 {
 886         *mysig = signal;
 887 }
 888
 889 /* On posix systems, the flow of the program starts right here, with
 890  * ENTRYPOINT being a #define that defines main(). On Windows, ENTRYPOINT
 891  * defines smain() and the real main() is in the service code under
 892  * win32service.cpp. This allows the service control manager to control
 893  * the process where we are running as a windows service.
 894  */
 895 ENTRYPOINT
 896 {
 897         SI = new InspIRCd(argc, argv);
 898         mysig = &SI->s_signal;
 899         SI->Run();
 900         delete SI;
 901         return 0;
 902 }