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