]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/inspircd.cpp
Merge pull request #545 from SaberUK/master+logging-cleanup
[user/henk/code/inspircd.git] / src / inspircd.cpp
1 /*
2  * InspIRCd -- Internet Relay Chat Daemon
3  *
4  *   Copyright (C) 2012 William Pitcock <nenolod@dereferenced.org>
5  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
6  *   Copyright (C) 2003-2008 Craig Edwards <craigedwards@brainbox.cc>
7  *   Copyright (C) 2008 Uli Schlachter <psychon@znc.in>
8  *   Copyright (C) 2006-2008 Robin Burchell <robin+git@viroteck.net>
9  *   Copyright (C) 2006-2007 Oliver Lupton <oliverlupton@gmail.com>
10  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
11  *   Copyright (C) 2007 Burlex <???@???>
12  *   Copyright (C) 2003 Craig McLure <craig@chatspike.net>
13  *   Copyright (C) 2003 randomdan <???@???>
14  *
15  * This file is part of InspIRCd.  InspIRCd is free software: you can
16  * redistribute it and/or modify it under the terms of the GNU General Public
17  * License as published by the Free Software Foundation, version 2.
18  *
19  * This program is distributed in the hope that it will be useful, but WITHOUT
20  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
21  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
22  * details.
23  *
24  * You should have received a copy of the GNU General Public License
25  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
26  */
27
28
29 /* $Core */
30 #include "inspircd.h"
31 #include <signal.h>
32
33 #ifndef _WIN32
34         #include <dirent.h>
35         #include <unistd.h>
36         #include <sys/resource.h>
37         #include <dlfcn.h>
38         #include <getopt.h>
39
40         /* Some systems don't define RUSAGE_SELF. This should fix them. */
41         #ifndef RUSAGE_SELF
42                 #define RUSAGE_SELF 0
43         #endif
44
45         #include <pwd.h> // setuid
46         #include <grp.h> // setgid
47 #else
48         WORD g_wOriginalColors;
49         WORD g_wBackgroundColor;
50         HANDLE g_hStdout;
51 #endif
52
53 #include <fstream>
54 #include <iostream>
55 #include "xline.h"
56 #include "bancache.h"
57 #include "socketengine.h"
58 #include "socket.h"
59 #include "command_parse.h"
60 #include "exitcodes.h"
61 #include "caller.h"
62 #include "testsuite.h"
63
64 InspIRCd* ServerInstance = NULL;
65
66 /** Seperate from the other casemap tables so that code *can* still exclusively rely on RFC casemapping
67  * if it must.
68  *
69  * This is provided as a pointer so that modules can change it to their custom mapping tables,
70  * e.g. for national character support.
71  */
72 unsigned const char *national_case_insensitive_map = rfc_case_insensitive_map;
73
74
75 /* Moved from exitcodes.h -- due to duplicate symbols -- Burlex
76  * XXX this is a bit ugly. -- w00t
77  */
78 const char* ExitCodes[] =
79 {
80                 "No error", /* 0 */
81                 "DIE command", /* 1 */
82                 "execv() failed", /* 2 */
83                 "Internal error", /* 3 */
84                 "Config file error", /* 4 */
85                 "Logfile error", /* 5 */
86                 "POSIX fork failed", /* 6 */
87                 "Bad commandline parameters", /* 7 */
88                 "No ports could be bound", /* 8 */
89                 "Can't write PID file", /* 9 */
90                 "SocketEngine could not initialize", /* 10 */
91                 "Refusing to start up as root", /* 11 */
92                 "Found a <die> tag!", /* 12 */
93                 "Couldn't load module on startup", /* 13 */
94                 "Could not create windows forked process", /* 14 */
95                 "Received SIGTERM", /* 15 */
96                 "Bad command handler loaded", /* 16 */
97                 "RegisterServiceCtrlHandler failed", /* 17 */
98                 "UpdateSCMStatus failed", /* 18 */
99                 "CreateEvent failed" /* 19 */
100 };
101
102 template<typename T> static void DeleteZero(T*&n)
103 {
104         T* t = n;
105         n = NULL;
106         delete t;
107 }
108
109 void InspIRCd::Cleanup()
110 {
111         for (unsigned int i = 0; i < ports.size(); i++)
112         {
113                 /* This calls the constructor and closes the listening socket */
114                 ports[i]->cull();
115                 delete ports[i];
116         }
117         ports.clear();
118
119         /* Close all client sockets, or the new process inherits them */
120         LocalUserList::reverse_iterator i = Users->local_users.rbegin();
121         while (i != this->Users->local_users.rend())
122         {
123                 User* u = *i++;
124                 Users->QuitUser(u, "Server shutdown");
125         }
126
127         GlobalCulls.Apply();
128         Modules->UnloadAll();
129
130         /* Delete objects dynamically allocated in constructor (destructor would be more appropriate, but we're likely exiting) */
131         /* Must be deleted before modes as it decrements modelines */
132         if (FakeClient)
133                 FakeClient->cull();
134         DeleteZero(this->FakeClient);
135         DeleteZero(this->Users);
136         DeleteZero(this->Modes);
137         DeleteZero(this->XLines);
138         DeleteZero(this->Parser);
139         DeleteZero(this->stats);
140         DeleteZero(this->Modules);
141         DeleteZero(this->BanCache);
142         DeleteZero(this->SNO);
143         DeleteZero(this->Config);
144         DeleteZero(this->chanlist);
145         DeleteZero(this->PI);
146         DeleteZero(this->Threads);
147         DeleteZero(this->Timers);
148         DeleteZero(this->SE);
149         /* Close logging */
150         this->Logs->CloseLogs();
151         DeleteZero(this->Logs);
152 }
153
154 void InspIRCd::Restart(const std::string &reason)
155 {
156         /* SendError flushes each client's queue,
157          * regardless of writeability state
158          */
159         this->SendError(reason);
160
161         /* Figure out our filename (if theyve renamed it, we're boned) */
162         std::string me;
163
164         char** argv = Config->cmdline.argv;
165
166 #ifdef _WIN32
167         char module[MAX_PATH];
168         if (GetModuleFileNameA(NULL, module, MAX_PATH))
169                 me = module;
170 #else
171         me = argv[0];
172 #endif
173
174         this->Cleanup();
175
176         if (execv(me.c_str(), argv) == -1)
177         {
178                 /* Will raise a SIGABRT if not trapped */
179                 throw CoreException(std::string("Failed to execv()! error: ") + strerror(errno));
180         }
181 }
182
183 void InspIRCd::SetSignals()
184 {
185 #ifndef _WIN32
186         signal(SIGALRM, SIG_IGN);
187         signal(SIGHUP, InspIRCd::SetSignal);
188         signal(SIGPIPE, SIG_IGN);
189         signal(SIGCHLD, SIG_IGN);
190         /* We want E2BIG not a signal! */
191         signal(SIGXFSZ, SIG_IGN);
192 #endif
193         signal(SIGTERM, InspIRCd::SetSignal);
194 }
195
196 void InspIRCd::QuickExit(int status)
197 {
198         exit(0);
199 }
200
201 bool InspIRCd::DaemonSeed()
202 {
203 #ifdef _WIN32
204         std::cout << "InspIRCd Process ID: " << con_green << GetCurrentProcessId() << con_reset << std::endl;
205         return true;
206 #else
207         signal(SIGTERM, InspIRCd::QuickExit);
208
209         int childpid;
210         if ((childpid = fork ()) < 0)
211                 return false;
212         else if (childpid > 0)
213         {
214                 /* We wait here for the child process to kill us,
215                  * so that the shell prompt doesnt come back over
216                  * the output.
217                  * Sending a kill with a signal of 0 just checks
218                  * if the child pid is still around. If theyre not,
219                  * they threw an error and we should give up.
220                  */
221                 while (kill(childpid, 0) != -1)
222                         sleep(1);
223                 exit(0);
224         }
225         setsid ();
226         std::cout << "InspIRCd Process ID: " << con_green << getpid() << con_reset << std::endl;
227
228         signal(SIGTERM, InspIRCd::SetSignal);
229
230         rlimit rl;
231         if (getrlimit(RLIMIT_CORE, &rl) == -1)
232         {
233                 this->Logs->Log("STARTUP", LOG_DEFAULT, "Failed to getrlimit()!");
234                 return false;
235         }
236         rl.rlim_cur = rl.rlim_max;
237
238         if (setrlimit(RLIMIT_CORE, &rl) == -1)
239                         this->Logs->Log("STARTUP", LOG_DEFAULT, "setrlimit() failed, cannot increase coredump size.");
240
241         return true;
242 #endif
243 }
244
245 void InspIRCd::WritePID(const std::string &filename)
246 {
247 #ifndef _WIN32
248         std::string fname(filename);
249         if (fname.empty())
250                 fname = DATA_PATH "/inspircd.pid";
251         std::ofstream outfile(fname.c_str());
252         if (outfile.is_open())
253         {
254                 outfile << getpid();
255                 outfile.close();
256         }
257         else
258         {
259                 std::cout << "Failed to write PID-file '" << fname << "', exiting." << std::endl;
260                 this->Logs->Log("STARTUP", LOG_DEFAULT, "Failed to write PID-file '%s', exiting.",fname.c_str());
261                 Exit(EXIT_STATUS_PID);
262         }
263 #endif
264 }
265
266 InspIRCd::InspIRCd(int argc, char** argv) :
267          ConfigFileName(CONFIG_PATH "/inspircd.conf"),
268
269          /* Functor pointer initialisation.
270           *
271           * THIS MUST MATCH THE ORDER OF DECLARATION OF THE FUNCTORS, e.g. the methods
272           * themselves within the class.
273           */
274          OperQuit("OperQuit", NULL),
275          GenRandom(&HandleGenRandom),
276          IsChannel(&HandleIsChannel),
277          Rehash(&HandleRehash),
278          IsNick(&HandleIsNick),
279          IsIdent(&HandleIsIdent),
280          OnCheckExemption(&HandleOnCheckExemption)
281 {
282         ServerInstance = this;
283
284         Extensions.Register(&OperQuit);
285
286         FailedPortList pl;
287         int do_version = 0, do_nofork = 0, do_debug = 0,
288             do_nolog = 0, do_root = 0, do_testsuite = 0;    /* flag variables */
289         int c = 0;
290
291         // Initialize so that if we exit before proper initialization they're not deleted
292         this->Logs = 0;
293         this->Threads = 0;
294         this->PI = 0;
295         this->Users = 0;
296         this->chanlist = 0;
297         this->Config = 0;
298         this->SNO = 0;
299         this->BanCache = 0;
300         this->Modules = 0;
301         this->stats = 0;
302         this->Timers = 0;
303         this->Parser = 0;
304         this->XLines = 0;
305         this->Modes = 0;
306         this->ConfigThread = NULL;
307         this->FakeClient = NULL;
308
309         UpdateTime();
310         this->startup_time = TIME.tv_sec;
311
312         // This must be created first, so other parts of Insp can use it while starting up
313         this->Logs = new LogManager;
314
315         SE = CreateSocketEngine();
316
317         this->Threads = new ThreadEngine;
318
319         /* Default implementation does nothing */
320         this->PI = new ProtocolInterface;
321
322         // Create base manager classes early, so nothing breaks
323         this->Users = new UserManager;
324
325         this->Users->unregistered_count = 0;
326
327         this->Users->clientlist = new user_hash();
328         this->Users->uuidlist = new user_hash();
329         this->chanlist = new chan_hash();
330
331         this->Config = new ServerConfig;
332         this->SNO = new SnomaskManager;
333         this->BanCache = new BanCacheManager;
334         this->Modules = new ModuleManager();
335         dynamic_reference_base::reset_all();
336         this->stats = new serverstats();
337         this->Timers = new TimerManager;
338         this->Parser = new CommandParser;
339         this->XLines = new XLineManager;
340
341         this->Config->cmdline.argv = argv;
342         this->Config->cmdline.argc = argc;
343
344 #ifdef _WIN32
345         srand(TIME.tv_nsec ^ TIME.tv_sec);
346
347         // Initialize the console values
348         g_hStdout = GetStdHandle(STD_OUTPUT_HANDLE);
349         CONSOLE_SCREEN_BUFFER_INFO bufinf;
350         if(GetConsoleScreenBufferInfo(g_hStdout, &bufinf))
351         {
352                 g_wOriginalColors = bufinf.wAttributes & 0x00FF;
353                 g_wBackgroundColor = bufinf.wAttributes & 0x00F0;
354         }
355         else
356         {
357                 g_wOriginalColors = FOREGROUND_RED|FOREGROUND_BLUE|FOREGROUND_GREEN;
358                 g_wBackgroundColor = 0;
359         }
360 #else
361         srandom(TIME.tv_nsec ^ TIME.tv_sec);
362 #endif
363
364         struct option longopts[] =
365         {
366                 { "nofork",     no_argument,            &do_nofork,     1       },
367                 { "config",     required_argument,      NULL,           'c'     },
368                 { "debug",      no_argument,            &do_debug,      1       },
369                 { "nolog",      no_argument,            &do_nolog,      1       },
370                 { "runasroot",  no_argument,            &do_root,       1       },
371                 { "version",    no_argument,            &do_version,    1       },
372                 { "testsuite",  no_argument,            &do_testsuite,  1       },
373                 { 0, 0, 0, 0 }
374         };
375
376         int index;
377         while ((c = getopt_long(argc, argv, ":c:", longopts, &index)) != -1)
378         {
379                 switch (c)
380                 {
381                         case 'c':
382                                 /* Config filename was set */
383                                 ConfigFileName = optarg;
384                         break;
385                         case 0:
386                                 /* getopt_long_only() set an int variable, just keep going */
387                         break;
388                         case '?':
389                                 /* Unknown parameter */
390                         default:
391                                 /* Fall through to handle other weird values too */
392                                 std::cout << "Unknown parameter '" << argv[optind-1] << "'" << std::endl;
393                                 std::cout << "Usage: " << argv[0] << " [--nofork] [--nolog] [--debug] [--config <config>]" << std::endl <<
394                                         std::string(static_cast<int>(8+strlen(argv[0])), ' ') << "[--runasroot] [--version] [--testsuite]" << std::endl;
395                                 Exit(EXIT_STATUS_ARGV);
396                         break;
397                 }
398         }
399
400         if (do_testsuite)
401                 do_nofork = do_debug = true;
402
403         if (do_version)
404         {
405                 std::cout << std::endl << VERSION << " r" << REVISION << std::endl;
406                 Exit(EXIT_STATUS_NOERROR);
407         }
408
409 #ifdef _WIN32
410         // Set up winsock
411         WSADATA wsadata;
412         WSAStartup(MAKEWORD(2,2), &wsadata);
413 #endif
414
415         /* Set the finished argument values */
416         Config->cmdline.nofork = (do_nofork != 0);
417         Config->cmdline.forcedebug = (do_debug != 0);
418         Config->cmdline.writelog = (!do_nolog != 0);
419         Config->cmdline.TestSuite = (do_testsuite != 0);
420
421         if (do_debug)
422         {
423                 FileWriter* fw = new FileWriter(stdout);
424                 FileLogStream* fls = new FileLogStream(LOG_RAWIO, fw);
425                 Logs->AddLogTypes("*", fls, true);
426         }
427
428         if (!ServerConfig::FileExists(ConfigFileName.c_str()))
429         {
430 #ifdef _WIN32
431                 /* Windows can (and defaults to) hide file extensions, so let's play a bit nice for windows users. */
432                 std::string txtconf = this->ConfigFileName;
433                 txtconf.append(".txt");
434
435                 if (ServerConfig::FileExists(txtconf.c_str()))
436                 {
437                         ConfigFileName = txtconf;
438                 }
439                 else
440 #endif
441                 {
442                         std::cout << "ERROR: Cannot open config file: " << ConfigFileName << std::endl << "Exiting..." << std::endl;
443                         this->Logs->Log("STARTUP", LOG_DEFAULT, "Unable to open config file %s", ConfigFileName.c_str());
444                         Exit(EXIT_STATUS_CONFIG);
445                 }
446         }
447
448         std::cout << con_green << "Inspire Internet Relay Chat Server" << con_reset << ", compiled on " __DATE__ " at " __TIME__ << std::endl;
449         std::cout << con_green << "(C) InspIRCd Development Team." << con_reset << std::endl << std::endl;
450         std::cout << "Developers:" << std::endl;
451         std::cout << con_green << "\tBrain, FrostyCoolSlug, w00t, Om, Special, peavey" << std::endl;
452         std::cout << "\taquanight, psychon, dz, danieldg, jackmcbarn" << con_reset << std::endl << std::endl;
453         std::cout << "Others:\t\t\t" << con_green << "See /INFO Output" << con_reset << std::endl;
454
455         this->Modes = new ModeParser;
456
457 #ifndef _WIN32
458         if (!do_root)
459                 this->CheckRoot();
460         else
461         {
462                 std::cout << "* WARNING * WARNING * WARNING * WARNING * WARNING *" << std::endl
463                 << "YOU ARE RUNNING INSPIRCD AS ROOT. THIS IS UNSUPPORTED" << std::endl
464                 << "AND IF YOU ARE HACKED, CRACKED, SPINDLED OR MUTILATED" << std::endl
465                 << "OR ANYTHING ELSE UNEXPECTED HAPPENS TO YOU OR YOUR" << std::endl
466                 << "SERVER, THEN IT IS YOUR OWN FAULT. IF YOU DID NOT MEAN" << std::endl
467                 << "TO START INSPIRCD AS ROOT, HIT CTRL+C NOW AND RESTART" << std::endl
468                 << "THE PROGRAM AS A NORMAL USER. YOU HAVE BEEN WARNED!" << std::endl << std::endl
469                 << "InspIRCd starting in 20 seconds, ctrl+c to abort..." << std::endl;
470                 sleep(20);
471         }
472 #endif
473
474         this->SetSignals();
475
476         if (!Config->cmdline.nofork)
477         {
478                 if (!this->DaemonSeed())
479                 {
480                         std::cout << "ERROR: could not go into daemon mode. Shutting down." << std::endl;
481                         Logs->Log("STARTUP", LOG_DEFAULT, "ERROR: could not go into daemon mode. Shutting down.");
482                         Exit(EXIT_STATUS_FORK);
483                 }
484         }
485
486         SE->RecoverFromFork();
487
488         /* During startup we don't actually initialize this
489          * in the thread engine.
490          */
491         this->Config->Read();
492         this->Config->Apply(NULL, "");
493         Logs->OpenFileLogs();
494         ModeParser::InitBuiltinModes();
495
496         // If we don't have a SID, generate one based on the server name and the server description
497         if (Config->sid.empty())
498                 Config->sid = UIDGenerator::GenerateSID(Config->ServerName, Config->ServerDesc);
499
500         // Initialize the UID generator with our sid
501         this->UIDGen.init(Config->sid);
502
503         /* set up fake client again this time with the correct uid */
504         this->FakeClient = new FakeUser(Config->sid, Config->ServerName);
505
506         // Get XLine to do it's thing.
507         this->XLines->CheckELines();
508         this->XLines->ApplyLines();
509
510         int bounditems = BindPorts(pl);
511
512         std::cout << std::endl;
513
514         this->Modules->LoadAll();
515
516         /* Just in case no modules were loaded - fix for bug #101 */
517         this->ISupport.Build();
518         Config->ApplyDisabledCommands(Config->DisabledCommands);
519
520         if (!pl.empty())
521         {
522                 std::cout << std::endl << "WARNING: Not all your client ports could be bound -- " << std::endl << "starting anyway with " << bounditems
523                         << " of " << bounditems + (int)pl.size() << " client ports bound." << std::endl << std::endl;
524                 std::cout << "The following port(s) failed to bind:" << std::endl << std::endl;
525                 int j = 1;
526                 for (FailedPortList::iterator i = pl.begin(); i != pl.end(); i++, j++)
527                 {
528                         std::cout << j << ".\tAddress: " << (i->first.empty() ? "<all>" : i->first) << " \tReason: " << i->second << std::endl;
529                 }
530
531                 std::cout << std::endl << "Hint: Try using a public IP instead of blank or *" << std::endl;
532         }
533
534         std::cout << "InspIRCd is now running as '" << Config->ServerName << "'[" << Config->GetSID() << "] with " << SE->GetMaxFds() << " max open sockets" << std::endl;
535
536 #ifndef _WIN32
537         if (!Config->cmdline.nofork)
538         {
539                 if (kill(getppid(), SIGTERM) == -1)
540                 {
541                         std::cout << "Error killing parent process: " << strerror(errno) << std::endl;
542                         Logs->Log("STARTUP", LOG_DEFAULT, "Error killing parent process: %s",strerror(errno));
543                 }
544         }
545
546         /* Explicitly shut down stdio's stdin/stdout/stderr.
547          *
548          * The previous logic here was to only do this if stdio was connected to a controlling
549          * terminal.  However, we must do this always to avoid information leaks and other
550          * problems related to stdio.
551          *
552          * The only exception is if we are in debug mode.
553          *
554          *    -- nenolod
555          */
556         if ((!do_nofork) && (!do_testsuite) && (!Config->cmdline.forcedebug))
557         {
558                 int fd = open("/dev/null", O_RDWR);
559
560                 fclose(stdin);
561                 fclose(stderr);
562                 fclose(stdout);
563
564                 if (dup2(fd, STDIN_FILENO) < 0)
565                         Logs->Log("STARTUP", LOG_DEFAULT, "Failed to dup /dev/null to stdin.");
566                 if (dup2(fd, STDOUT_FILENO) < 0)
567                         Logs->Log("STARTUP", LOG_DEFAULT, "Failed to dup /dev/null to stdout.");
568                 if (dup2(fd, STDERR_FILENO) < 0)
569                         Logs->Log("STARTUP", LOG_DEFAULT, "Failed to dup /dev/null to stderr.");
570                 close(fd);
571         }
572         else
573         {
574                 Logs->Log("STARTUP", LOG_DEFAULT, "Keeping pseudo-tty open as we are running in the foreground.");
575         }
576 #else
577         /* Set win32 service as running, if we are running as a service */
578         SetServiceRunning();
579
580         // Handle forking
581         if(!do_nofork)
582         {
583                 FreeConsole();
584         }
585
586         QueryPerformanceFrequency(&stats->QPFrequency);
587 #endif
588
589         Logs->Log("STARTUP", LOG_DEFAULT, "Startup complete as '%s'[%s], %d max open sockets", Config->ServerName.c_str(),Config->GetSID().c_str(), SE->GetMaxFds());
590
591 #ifndef _WIN32
592         std::string SetUser = Config->ConfValue("security")->getString("runasuser");
593         std::string SetGroup = Config->ConfValue("security")->getString("runasgroup");
594         if (!SetGroup.empty())
595         {
596                 int ret;
597
598                 // setgroups
599                 ret = setgroups(0, NULL);
600
601                 if (ret == -1)
602                 {
603                         this->Logs->Log("SETGROUPS", LOG_DEFAULT, "setgroups() failed (wtf?): %s", strerror(errno));
604                         this->QuickExit(0);
605                 }
606
607                 // setgid
608                 struct group *g;
609
610                 errno = 0;
611                 g = getgrnam(SetGroup.c_str());
612
613                 if (!g)
614                 {
615                         this->Logs->Log("SETGUID", LOG_DEFAULT, "getgrnam() failed (bad user?): %s", strerror(errno));
616                         this->QuickExit(0);
617                 }
618
619                 ret = setgid(g->gr_gid);
620
621                 if (ret == -1)
622                 {
623                         this->Logs->Log("SETGUID", LOG_DEFAULT, "setgid() failed (bad user?): %s", strerror(errno));
624                         this->QuickExit(0);
625                 }
626         }
627
628         if (!SetUser.empty())
629         {
630                 // setuid
631                 struct passwd *u;
632
633                 errno = 0;
634                 u = getpwnam(SetUser.c_str());
635
636                 if (!u)
637                 {
638                         this->Logs->Log("SETGUID", LOG_DEFAULT, "getpwnam() failed (bad user?): %s", strerror(errno));
639                         this->QuickExit(0);
640                 }
641
642                 int ret = setuid(u->pw_uid);
643
644                 if (ret == -1)
645                 {
646                         this->Logs->Log("SETGUID", LOG_DEFAULT, "setuid() failed (bad user?): %s", strerror(errno));
647                         this->QuickExit(0);
648                 }
649         }
650
651         this->WritePID(Config->PID);
652 #endif
653 }
654
655 void InspIRCd::UpdateTime()
656 {
657 #ifdef _WIN32
658         SYSTEMTIME st;
659         GetSystemTime(&st);
660
661         TIME.tv_sec = time(NULL);
662         TIME.tv_nsec = st.wMilliseconds;
663 #else
664         #ifdef HAS_CLOCK_GETTIME
665                 clock_gettime(CLOCK_REALTIME, &TIME);
666         #else
667                 struct timeval tv;
668                 gettimeofday(&tv, NULL);
669                 TIME.tv_sec = tv.tv_sec;
670                 TIME.tv_nsec = tv.tv_usec * 1000;
671         #endif
672 #endif
673 }
674
675 int InspIRCd::Run()
676 {
677         /* See if we're supposed to be running the test suite rather than entering the mainloop */
678         if (Config->cmdline.TestSuite)
679         {
680                 TestSuite* ts = new TestSuite;
681                 delete ts;
682                 Exit(0);
683         }
684
685         UpdateTime();
686         time_t OLDTIME = TIME.tv_sec;
687
688         while (true)
689         {
690 #ifndef _WIN32
691                 static rusage ru;
692 #endif
693
694                 /* Check if there is a config thread which has finished executing but has not yet been freed */
695                 if (this->ConfigThread && this->ConfigThread->IsDone())
696                 {
697                         /* Rehash has completed */
698                         this->Logs->Log("CONFIG", LOG_DEBUG, "Detected ConfigThread exiting, tidying up...");
699
700                         this->ConfigThread->Finish();
701
702                         ConfigThread->join();
703                         delete ConfigThread;
704                         ConfigThread = NULL;
705                 }
706
707                 UpdateTime();
708
709                 /* Run background module timers every few seconds
710                  * (the docs say modules shouldnt rely on accurate
711                  * timing using this event, so we dont have to
712                  * time this exactly).
713                  */
714                 if (TIME.tv_sec != OLDTIME)
715                 {
716 #ifndef _WIN32
717                         getrusage(RUSAGE_SELF, &ru);
718                         stats->LastSampled = TIME;
719                         stats->LastCPU = ru.ru_utime;
720 #else
721                         if(QueryPerformanceCounter(&stats->LastSampled))
722                         {
723                                 FILETIME CreationTime;
724                                 FILETIME ExitTime;
725                                 FILETIME KernelTime;
726                                 FILETIME UserTime;
727                                 GetProcessTimes(GetCurrentProcess(), &CreationTime, &ExitTime, &KernelTime, &UserTime);
728                                 stats->LastCPU.dwHighDateTime = KernelTime.dwHighDateTime + UserTime.dwHighDateTime;
729                                 stats->LastCPU.dwLowDateTime = KernelTime.dwLowDateTime + UserTime.dwLowDateTime;
730                         }
731 #endif
732
733                         /* Allow a buffer of two seconds drift on this so that ntpdate etc dont harass admins */
734                         if (TIME.tv_sec < OLDTIME - 2)
735                         {
736                                 SNO->WriteToSnoMask('d', "\002EH?!\002 -- Time is flowing BACKWARDS in this dimension! Clock drifted backwards %lu secs.", (unsigned long)OLDTIME-TIME.tv_sec);
737                         }
738                         else if (TIME.tv_sec > OLDTIME + 2)
739                         {
740                                 SNO->WriteToSnoMask('d', "\002EH?!\002 -- Time is jumping FORWARDS! Clock skipped %lu secs.", (unsigned long)TIME.tv_sec - OLDTIME);
741                         }
742
743                         OLDTIME = TIME.tv_sec;
744
745                         if ((TIME.tv_sec % 3600) == 0)
746                         {
747                                 Users->GarbageCollect();
748                                 FOREACH_MOD(I_OnGarbageCollect, OnGarbageCollect());
749                         }
750
751                         Timers->TickTimers(TIME.tv_sec);
752                         this->DoBackgroundUserStuff();
753
754                         if ((TIME.tv_sec % 5) == 0)
755                         {
756                                 FOREACH_MOD(I_OnBackgroundTimer,OnBackgroundTimer(TIME.tv_sec));
757                                 SNO->FlushSnotices();
758                         }
759                 }
760
761                 /* Call the socket engine to wait on the active
762                  * file descriptors. The socket engine has everything's
763                  * descriptors in its list... dns, modules, users,
764                  * servers... so its nice and easy, just one call.
765                  * This will cause any read or write events to be
766                  * dispatched to their handlers.
767                  */
768                 this->SE->DispatchTrialWrites();
769                 this->SE->DispatchEvents();
770
771                 /* if any users were quit, take them out */
772                 GlobalCulls.Apply();
773                 AtomicActions.Run();
774
775                 if (s_signal)
776                 {
777                         this->SignalHandler(s_signal);
778                         s_signal = 0;
779                 }
780         }
781
782         return 0;
783 }
784
785 /**********************************************************************************/
786
787 /* this returns true when all modules are satisfied that the user should be allowed onto the irc server
788  * (until this returns true, a user will block in the waiting state, waiting to connect up to the
789  * registration timeout maximum seconds)
790  */
791 bool InspIRCd::AllModulesReportReady(LocalUser* user)
792 {
793         ModResult res;
794         FIRST_MOD_RESULT(OnCheckReady, res, (user));
795         return (res == MOD_RES_PASSTHRU);
796 }
797
798 sig_atomic_t InspIRCd::s_signal = 0;
799
800 void InspIRCd::SetSignal(int signal)
801 {
802         s_signal = signal;
803 }
804
805 /* On posix systems, the flow of the program starts right here, with
806  * ENTRYPOINT being a #define that defines main(). On Windows, ENTRYPOINT
807  * defines smain() and the real main() is in the service code under
808  * win32service.cpp. This allows the service control manager to control
809  * the process where we are running as a windows service.
810  */
811 ENTRYPOINT
812 {
813         new InspIRCd(argc, argv);
814         ServerInstance->Run();
815         delete ServerInstance;
816         return 0;
817 }