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