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