src/inspircd.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
   6  * See: http://www.inspircd.org/wiki/index.php/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "configreader.h"
  16 #include <signal.h>
  17 #include <dirent.h>
  18 #include <exception>
  19 #include <fstream>
  20 #include <unistd.h>
  21 #include "modules.h"
  22 #include "mode.h"
  23 #include "xline.h"
  24 #include "socketengine.h"
  25 #include "inspircd_se_config.h"
  26 #include "socket.h"
  27 #include "typedefs.h"
  28 #include "command_parse.h"
  29 #include "exitcodes.h"
  30 #include <dlfcn.h>
  31 #include <getopt.h>
  32
  33
  34 using irc::sockets::NonBlocking;
  35 using irc::sockets::Blocking;
  36 using irc::sockets::insp_ntoa;
  37 using irc::sockets::insp_inaddr;
  38 using irc::sockets::insp_sockaddr;
  39
  40 InspIRCd* SI = NULL;
  41
  42 void InspIRCd::AddServerName(const std::string &servername)
  43 {
  44         if(find(servernames.begin(), servernames.end(), servername) == servernames.end())
  45                 servernames.push_back(servername); /* Wasn't already there. */
  46 }
  47
  48 const char* InspIRCd::FindServerNamePtr(const std::string &servername)
  49 {
  50         servernamelist::iterator iter = find(servernames.begin(), servernames.end(), servername);
  51
  52         if(iter == servernames.end())
  53         {
  54                 AddServerName(servername);
  55                 iter = --servernames.end();
  56         }
  57
  58         return iter->c_str();
  59 }
  60
  61 bool InspIRCd::FindServerName(const std::string &servername)
  62 {
  63         return (find(servernames.begin(), servernames.end(), servername) != servernames.end());
  64 }
  65
  66 void InspIRCd::Exit(int status)
  67 {
  68         if (SI)
  69         {
  70                 SI->SendError("Exiting with status " + ConvToStr(status) + " (" + std::string(ExitCodes[status]) + ")");
  71                 SI->Cleanup();
  72         }
  73         exit (status);
  74 }
  75
  76 void InspIRCd::Cleanup()
  77 {
  78         std::vector<std::string> mymodnames;
  79         int MyModCount = this->GetModuleCount();
  80
  81         for (unsigned int i = 0; i < stats->BoundPortCount; i++)
  82         {
  83                 /* This calls the constructor and closes the listening socket */
  84                 delete Config->openSockfd[i];
  85                 Config->openSockfd[i] = NULL;
  86         }
  87         stats->BoundPortCount = 0;
  88
  89         /* Close all client sockets, or the new process inherits them */
  90         for (std::vector<userrec*>::const_iterator i = this->local_users.begin(); i != this->local_users.end(); i++)
  91         {
  92                 (*i)->SetWriteError("Server shutdown");
  93                 (*i)->CloseSocket();
  94         }
  95
  96         /* We do this more than once, so that any service providers get a
  97          * chance to be unhooked by the modules using them, but then get
  98          * a chance to be removed themsleves.
  99          */
 100         for (int tries = 0; tries < 3; tries++)
 101         {
 102                 MyModCount = this->GetModuleCount();
 103                 mymodnames.clear();
 104
 105                 /* Unload all modules, so they get a chance to clean up their listeners */
 106                 for (int j = 0; j <= MyModCount; j++)
 107                         mymodnames.push_back(Config->module_names[j]);
 108
 109                 for (int k = 0; k <= MyModCount; k++)
 110                         this->UnloadModule(mymodnames[k].c_str());
 111         }
 112
 113         /* Close logging */
 114         this->Logger->Close();
 115 }
 116
 117 void InspIRCd::Restart(const std::string &reason)
 118 {
 119         /* SendError flushes each client's queue,
 120          * regardless of writeability state
 121          */
 122         this->SendError(reason);
 123
 124         this->Cleanup();
 125
 126         /* Figure out our filename (if theyve renamed it, we're boned) */
 127         std::string me = Config->MyDir + "/inspircd";
 128
 129         if (execv(me.c_str(), Config->argv) == -1)
 130         {
 131                 /* Will raise a SIGABRT if not trapped */
 132                 throw CoreException(std::string("Failed to execv()! error: ") + strerror(errno));
 133         }
 134 }
 135
 136 void InspIRCd::Start()
 137 {
 138         printf("\033[1;32mInspire Internet Relay Chat Server, compiled %s at %s\n",__DATE__,__TIME__);
 139         printf("(C) InspIRCd Development Team.\033[0m\n\n");
 140         printf("Developers:\t\t\033[1;32mBrain, FrostyCoolSlug, w00t, Om, Special, pippijn, peavey\033[0m\n");
 141         printf("Others:\t\t\t\033[1;32mSee /INFO Output\033[0m\n");
 142 }
 143
 144 void InspIRCd::Rehash(int status)
 145 {
 146         SI->WriteOpers("*** Rehashing config file %s due to SIGHUP",ServerConfig::CleanFilename(SI->ConfigFileName));
 147         SI->CloseLog();
 148         SI->OpenLog(SI->Config->argv, SI->Config->argc);
 149         SI->RehashUsersAndChans();
 150         FOREACH_MOD_I(SI, I_OnGarbageCollect, OnGarbageCollect());
 151         SI->Config->Read(false,NULL);
 152         SI->ResetMaxBans();
 153         SI->Res->Rehash();
 154         SI->BuildISupport();
 155         FOREACH_MOD_I(SI,I_OnRehash,OnRehash(NULL,""));
 156 }
 157
 158 void InspIRCd::ResetMaxBans()
 159 {
 160         for (chan_hash::const_iterator i = chanlist->begin(); i != chanlist->end(); i++)
 161                 i->second->ResetMaxBans();
 162 }
 163
 164
 165 /** Because hash_map doesnt free its buckets when we delete items (this is a 'feature')
 166  * we must occasionally rehash the hash (yes really).
 167  * We do this by copying the entries from the old hash to a new hash, causing all
 168  * empty buckets to be weeded out of the hash. We dont do this on a timer, as its
 169  * very expensive, so instead we do it when the user types /REHASH and expects a
 170  * short delay anyway.
 171  */
 172 void InspIRCd::RehashUsersAndChans()
 173 {
 174         user_hash* old_users = this->clientlist;
 175         chan_hash* old_chans = this->chanlist;
 176
 177         this->clientlist = new user_hash();
 178         this->chanlist = new chan_hash();
 179
 180         for (user_hash::const_iterator n = old_users->begin(); n != old_users->end(); n++)
 181                 this->clientlist->insert(*n);
 182
 183         delete old_users;
 184
 185         for (chan_hash::const_iterator n = old_chans->begin(); n != old_chans->end(); n++)
 186                 this->chanlist->insert(*n);
 187
 188         delete old_chans;
 189 }
 190
 191 void InspIRCd::CloseLog()
 192 {
 193         this->Logger->Close();
 194 }
 195
 196 void InspIRCd::SetSignals()
 197 {
 198         signal(SIGALRM, SIG_IGN);
 199         signal(SIGHUP, InspIRCd::Rehash);
 200         signal(SIGPIPE, SIG_IGN);
 201         signal(SIGTERM, InspIRCd::Exit);
 202         signal(SIGCHLD, SIG_IGN);
 203 }
 204
 205 void InspIRCd::QuickExit(int status)
 206 {
 207         exit(0);
 208 }
 209
 210 bool InspIRCd::DaemonSeed()
 211 {
 212         signal(SIGTERM, InspIRCd::QuickExit);
 213
 214         int childpid;
 215         if ((childpid = fork ()) < 0)
 216                 return false;
 217         else if (childpid > 0)
 218         {
 219                 /* We wait here for the child process to kill us,
 220                  * so that the shell prompt doesnt come back over
 221                  * the output.
 222                  * Sending a kill with a signal of 0 just checks
 223                  * if the child pid is still around. If theyre not,
 224                  * they threw an error and we should give up.
 225                  */
 226                 while (kill(childpid, 0) != -1)
 227                         sleep(1);
 228                 exit(0);
 229         }
 230         setsid ();
 231         umask (007);
 232         printf("InspIRCd Process ID: \033[1;32m%lu\033[0m\n",(unsigned long)getpid());
 233
 234         signal(SIGTERM, InspIRCd::Exit);
 235
 236         rlimit rl;
 237         if (getrlimit(RLIMIT_CORE, &rl) == -1)
 238         {
 239                 this->Log(DEFAULT,"Failed to getrlimit()!");
 240                 return false;
 241         }
 242         else
 243         {
 244                 rl.rlim_cur = rl.rlim_max;
 245                 if (setrlimit(RLIMIT_CORE, &rl) == -1)
 246                         this->Log(DEFAULT,"setrlimit() failed, cannot increase coredump size.");
 247         }
 248
 249         return true;
 250 }
 251
 252 void InspIRCd::WritePID(const std::string &filename)
 253 {
 254         std::string fname = (filename.empty() ? "inspircd.pid" : filename);
 255         if (*(fname.begin()) != '/')
 256         {
 257                 std::string::size_type pos;
 258                 std::string confpath = this->ConfigFileName;
 259                 if ((pos = confpath.rfind("/")) != std::string::npos)
 260                 {
 261                         /* Leaves us with just the path */
 262                         fname = confpath.substr(0, pos) + std::string("/") + fname;
 263                 }
 264         }
 265         std::ofstream outfile(fname.c_str());
 266         if (outfile.is_open())
 267         {
 268                 outfile << getpid();
 269                 outfile.close();
 270         }
 271         else
 272         {
 273                 printf("Failed to write PID-file '%s', exiting.\n",fname.c_str());
 274                 this->Log(DEFAULT,"Failed to write PID-file '%s', exiting.",fname.c_str());
 275                 Exit(EXIT_STATUS_PID);
 276         }
 277 }
 278
 279 std::string InspIRCd::GetRevision()
 280 {
 281         return REVISION;
 282 }
 283
 284 InspIRCd::InspIRCd(int argc, char** argv)
 285         : ModCount(-1), duration_m(60), duration_h(60*60), duration_d(60*60*24), duration_w(60*60*24*7), duration_y(60*60*24*365), GlobalCulls(this)
 286 {
 287         int found_ports = 0;
 288         FailedPortList pl;
 289         int do_version = 0, do_nofork = 0, do_debug = 0, do_nolog = 0, do_root = 0;    /* flag variables */
 290         char c = 0;
 291
 292         modules.resize(255);
 293         factory.resize(255);
 294
 295         this->unregistered_count = 0;
 296
 297         this->clientlist = new user_hash();
 298         this->chanlist = new chan_hash();
 299
 300         this->Config = new ServerConfig(this);
 301
 302         this->Config->argv = argv;
 303         this->Config->argc = argc;
 304
 305         this->Config->opertypes.clear();
 306         this->Config->operclass.clear();
 307         this->SNO = new SnomaskManager(this);
 308         this->TIME = this->OLDTIME = this->startup_time = time(NULL);
 309         this->time_delta = 0;
 310         this->next_call = this->TIME + 3;
 311         srand(this->TIME);
 312
 313         *this->LogFileName = 0;
 314         strlcpy(this->ConfigFileName, CONFIG_FILE, MAXBUF);
 315
 316         struct option longopts[] =
 317         {
 318                 { "nofork",     no_argument,            &do_nofork,     1       },
 319                 { "logfile",    required_argument,      NULL,           'f'     },
 320                 { "config",     required_argument,      NULL,           'c'     },
 321                 { "debug",      no_argument,            &do_debug,      1       },
 322                 { "nolog",      no_argument,            &do_nolog,      1       },
 323                 { "runasroot",  no_argument,            &do_root,       1       },
 324                 { "version",    no_argument,            &do_version,    1       },
 325                 { 0, 0, 0, 0 }
 326         };
 327
 328         while ((c = getopt_long_only(argc, argv, ":f:", longopts, NULL)) != -1)
 329         {
 330                 switch (c)
 331                 {
 332                         case 'f':
 333                                 /* Log filename was set */
 334                                 strlcpy(LogFileName, optarg, MAXBUF);
 335                                 printf("LOG: Setting logfile to %s\n", LogFileName);
 336                         break;
 337                         case 'c':
 338                                 /* Config filename was set */
 339                                 strlcpy(ConfigFileName, optarg, MAXBUF);
 340                                 printf("CONFIG: Setting config file to %s\n", ConfigFileName);
 341                         break;
 342                         case 0:
 343                                 /* getopt_long_only() set an int variable, just keep going */
 344                         break;
 345                         default:
 346                                 /* Unknown parameter! DANGER, INTRUDER.... err.... yeah. */
 347                                 printf("Usage: %s [--nofork] [--nolog] [--debug] [--logfile <filename>] [--runasroot] [--version] [--config <config>]\n", argv[0]);
 348                                 Exit(EXIT_STATUS_ARGV);
 349                         break;
 350                 }
 351         }
 352
 353         if (do_version)
 354         {
 355                 printf("\n%s r%s\n", VERSION, REVISION);
 356                 Exit(EXIT_STATUS_NOERROR);
 357         }
 358
 359         if (!ServerConfig::FileExists(this->ConfigFileName))
 360         {
 361                 printf("ERROR: Cannot open config file: %s\nExiting...\n", this->ConfigFileName);
 362                 this->Log(DEFAULT,"Unable to open config file %s", this->ConfigFileName);
 363                 Exit(EXIT_STATUS_CONFIG);
 364         }
 365
 366         this->Start();
 367
 368         /* Set the finished argument values */
 369         Config->nofork = do_nofork;
 370         Config->forcedebug = do_debug;
 371         Config->writelog = !do_nolog;
 372
 373         strlcpy(Config->MyExecutable,argv[0],MAXBUF);
 374
 375         this->OpenLog(argv, argc);
 376
 377         this->stats = new serverstats();
 378         this->Timers = new TimerManager(this);
 379         this->Parser = new CommandParser(this);
 380         this->XLines = new XLineManager(this);
 381         Config->ClearStack();
 382         Config->Read(true, NULL);
 383
 384         if (!do_root)
 385                 this->CheckRoot();
 386         else
 387         {
 388                 printf("* WARNING * WARNING * WARNING * WARNING * WARNING * \n\n");
 389                 printf("YOU ARE RUNNING INSPIRCD AS ROOT. THIS IS UNSUPPORTED\n");
 390                 printf("AND IF YOU ARE HACKED, CRACKED, SPINDLED OR MUTILATED\n");
 391                 printf("OR ANYTHING ELSE UNEXPECTED HAPPENS TO YOU OR YOUR\n");
 392                 printf("SERVER, THEN IT IS YOUR OWN FAULT. IF YOU DID NOT MEAN\n");
 393                 printf("TO START INSPIRCD AS ROOT, HIT CTRL+C NOW AND RESTART\n");
 394                 printf("THE PROGRAM AS A NORMAL USER. YOU HAVE BEEN WARNED!\n");
 395                 printf("\nInspIRCd starting in 20 seconds, ctrl+c to abort...\n");
 396                 sleep(20);
 397         }
 398
 399         this->Modes = new ModeParser(this);
 400         this->AddServerName(Config->ServerName);
 401         CheckDie();
 402         InitializeDisabledCommands(Config->DisabledCommands, this);
 403         stats->BoundPortCount = BindPorts(true, found_ports, pl);
 404
 405         for(int t = 0; t < 255; t++)
 406                 Config->global_implementation[t] = 0;
 407
 408         memset(&Config->implement_lists,0,sizeof(Config->implement_lists));
 409
 410         printf("\n");
 411         this->SetSignals();
 412
 413         if (!Config->nofork)
 414         {
 415                 if (!this->DaemonSeed())
 416                 {
 417                         printf("ERROR: could not go into daemon mode. Shutting down.\n");
 418                         Log(DEFAULT,"ERROR: could not go into daemon mode. Shutting down.");
 419                         Exit(EXIT_STATUS_FORK);
 420                 }
 421         }
 422
 423         /* Because of limitations in kqueue on freebsd, we must fork BEFORE we
 424          * initialize the socket engine.
 425          */
 426         SocketEngineFactory* SEF = new SocketEngineFactory();
 427         SE = SEF->Create(this);
 428         delete SEF;
 429
 430         this->Res = new DNS(this);
 431
 432         this->LoadAllModules();
 433         /* Just in case no modules were loaded - fix for bug #101 */
 434         this->BuildISupport();
 435
 436         if ((stats->BoundPortCount == 0) && (found_ports > 0))
 437         {
 438                 printf("\nERROR: I couldn't bind any ports! Are you sure you didn't start InspIRCd twice?\n");
 439                 Log(DEFAULT,"ERROR: I couldn't bind any ports! Are you sure you didn't start InspIRCd twice?");
 440                 Exit(EXIT_STATUS_BIND);
 441         }
 442
 443         if (stats->BoundPortCount != (unsigned int)found_ports)
 444         {
 445                 printf("\nWARNING: Not all your client ports could be bound --\nstarting anyway with %ld of %d client ports bound.\n\n", stats->BoundPortCount, found_ports);
 446                 printf("The following port%s failed to bind:\n", found_ports - stats->BoundPortCount != 1 ? "s" : "");
 447                 int j = 1;
 448                 for (FailedPortList::iterator i = pl.begin(); i != pl.end(); i++, j++)
 449                 {
 450                         printf("%d.\tIP: %s\tPort: %lu\n", j, i->first.empty() ? "<all>" : i->first.c_str(), (unsigned long)i->second);
 451                 }
 452         }
 453
 454         /* Add the listening sockets used for client inbound connections
 455          * to the socket engine
 456          */
 457         for (unsigned long count = 0; count < stats->BoundPortCount; count++)
 458         {
 459                 if (!SE->AddFd(Config->openSockfd[count]))
 460                 {
 461                         printf("\nEH? Could not add listener to socketengine. You screwed up, aborting.\n");
 462                         Log(DEFAULT,"EH? Could not add listener to socketengine. You screwed up, aborting.");
 463                         Exit(EXIT_STATUS_INTERNAL);
 464                 }
 465         }
 466
 467         if (!Config->nofork)
 468         {
 469                 if (kill(getppid(), SIGTERM) == -1)
 470                 {
 471                         printf("Error killing parent process: %s\n",strerror(errno));
 472                         Log(DEFAULT,"Error killing parent process: %s",strerror(errno));
 473                 }
 474         }
 475
 476         if (isatty(0) && isatty(1) && isatty(2))
 477         {
 478                 /* We didn't start from a TTY, we must have started from a background process -
 479                  * e.g. we are restarting, or being launched by cron. Dont kill parent, and dont
 480                  * close stdin/stdout
 481                  */
 482                 if (!do_nofork)
 483                 {
 484                         fclose(stdin);
 485                         fclose(stderr);
 486                         fclose(stdout);
 487                 }
 488                 else
 489                 {
 490                         Log(DEFAULT,"Keeping pseudo-tty open as we are running in the foreground.");
 491                 }
 492         }
 493
 494         printf("\nInspIRCd is now running!\n");
 495         Log(DEFAULT,"Startup complete.");
 496
 497         this->WritePID(Config->PID);
 498 }
 499
 500 std::string InspIRCd::GetVersionString()
 501 {
 502         char versiondata[MAXBUF];
 503         char dnsengine[] = "singlethread-object";
 504
 505         if (*Config->CustomVersion)
 506         {
 507                 snprintf(versiondata,MAXBUF,"%s %s :%s",VERSION,Config->ServerName,Config->CustomVersion);
 508         }
 509         else
 510         {
 511                 snprintf(versiondata,MAXBUF,"%s %s :%s [FLAGS=%s,%s,%s]",VERSION,Config->ServerName,SYSTEM,REVISION,SE->GetName().c_str(),dnsengine);
 512         }
 513         return versiondata;
 514 }
 515
 516 char* InspIRCd::ModuleError()
 517 {
 518         return MODERR;
 519 }
 520
 521 void InspIRCd::EraseFactory(int j)
 522 {
 523         int v = 0;
 524         for (std::vector<ircd_module*>::iterator t = factory.begin(); t != factory.end(); t++)
 525         {
 526                 if (v == j)
 527                 {
 528                         delete *t;
 529                         factory.erase(t);
 530                         factory.push_back(NULL);
 531                         return;
 532                 }
 533                 v++;
 534         }
 535 }
 536
 537 void InspIRCd::EraseModule(int j)
 538 {
 539         int v1 = 0;
 540         for (ModuleList::iterator m = modules.begin(); m!= modules.end(); m++)
 541         {
 542                 if (v1 == j)
 543                 {
 544                         DELETE(*m);
 545                         modules.erase(m);
 546                         modules.push_back(NULL);
 547                         break;
 548                 }
 549                 v1++;
 550         }
 551         int v2 = 0;
 552         for (std::vector<std::string>::iterator v = Config->module_names.begin(); v != Config->module_names.end(); v++)
 553         {
 554                 if (v2 == j)
 555                 {
 556                        Config->module_names.erase(v);
 557                        break;
 558                 }
 559                 v2++;
 560         }
 561
 562 }
 563
 564 void InspIRCd::MoveTo(std::string modulename,int slot)
 565 {
 566         unsigned int v2 = 256;
 567         for (unsigned int v = 0; v < Config->module_names.size(); v++)
 568         {
 569                 if (Config->module_names[v] == modulename)
 570                 {
 571                         // found an instance, swap it with the item at the end
 572                         v2 = v;
 573                         break;
 574                 }
 575         }
 576         if ((v2 != (unsigned int)slot) && (v2 < 256))
 577         {
 578                 // Swap the module names over
 579                 Config->module_names[v2] = Config->module_names[slot];
 580                 Config->module_names[slot] = modulename;
 581                 // now swap the module factories
 582                 ircd_module* temp = factory[v2];
 583                 factory[v2] = factory[slot];
 584                 factory[slot] = temp;
 585                 // now swap the module objects
 586                 Module* temp_module = modules[v2];
 587                 modules[v2] = modules[slot];
 588                 modules[slot] = temp_module;
 589                 // now swap the implement lists (we dont
 590                 // need to swap the global or recount it)
 591                 for (int n = 0; n < 255; n++)
 592                 {
 593                         char x = Config->implement_lists[v2][n];
 594                         Config->implement_lists[v2][n] = Config->implement_lists[slot][n];
 595                         Config->implement_lists[slot][n] = x;
 596                 }
 597         }
 598 }
 599
 600 void InspIRCd::MoveAfter(std::string modulename, std::string after)
 601 {
 602         for (unsigned int v = 0; v < Config->module_names.size(); v++)
 603         {
 604                 if (Config->module_names[v] == after)
 605                 {
 606                         MoveTo(modulename, v);
 607                         return;
 608                 }
 609         }
 610 }
 611
 612 void InspIRCd::MoveBefore(std::string modulename, std::string before)
 613 {
 614         for (unsigned int v = 0; v < Config->module_names.size(); v++)
 615         {
 616                 if (Config->module_names[v] == before)
 617                 {
 618                         if (v > 0)
 619                         {
 620                                 MoveTo(modulename, v-1);
 621                         }
 622                         else
 623                         {
 624                                 MoveTo(modulename, v);
 625                         }
 626                         return;
 627                 }
 628         }
 629 }
 630
 631 void InspIRCd::MoveToFirst(std::string modulename)
 632 {
 633         MoveTo(modulename,0);
 634 }
 635
 636 void InspIRCd::MoveToLast(std::string modulename)
 637 {
 638         MoveTo(modulename,this->GetModuleCount());
 639 }
 640
 641 void InspIRCd::BuildISupport()
 642 {
 643         // the neatest way to construct the initial 005 numeric, considering the number of configure constants to go in it...
 644         std::stringstream v;
 645         v << "WALLCHOPS WALLVOICES MODES=" << MAXMODES-1 << " CHANTYPES=# PREFIX=" << this->Modes->BuildPrefixes() << " MAP MAXCHANNELS=" << Config->MaxChans << " MAXBANS=60 VBANLIST NICKLEN=" << NICKMAX-1;
 646         v << " CASEMAPPING=rfc1459 STATUSMSG=@%+ CHARSET=ascii TOPICLEN=" << MAXTOPIC << " KICKLEN=" << MAXKICK << " MAXTARGETS=" << Config->MaxTargets << " AWAYLEN=";
 647         v << MAXAWAY << " CHANMODES=" << this->Modes->ChanModes() << " FNC NETWORK=" << Config->Network << " MAXPARA=32";
 648         Config->data005 = v.str();
 649         FOREACH_MOD_I(this,I_On005Numeric,On005Numeric(Config->data005));
 650         Config->Update005();
 651 }
 652
 653 bool InspIRCd::UnloadModule(const char* filename)
 654 {
 655         std::string filename_str = filename;
 656         for (unsigned int j = 0; j != Config->module_names.size(); j++)
 657         {
 658                 if (Config->module_names[j] == filename_str)
 659                 {
 660                         if (modules[j]->GetVersion().Flags & VF_STATIC)
 661                         {
 662                                 this->Log(DEFAULT,"Failed to unload STATIC module %s",filename);
 663                                 snprintf(MODERR,MAXBUF,"Module not unloadable (marked static)");
 664                                 return false;
 665                         }
 666                         std::pair<int,std::string> intercount = GetInterfaceInstanceCount(modules[j]);
 667                         if (intercount.first > 0)
 668                         {
 669                                 this->Log(DEFAULT,"Failed to unload module %s, being used by %d other(s) via interface '%s'",filename, intercount.first, intercount.second.c_str());
 670                                 snprintf(MODERR,MAXBUF,"Module not unloadable (Still in use by %d other module%s which %s using its interface '%s') -- unload dependent modules first!",
 671                                                 intercount.first,
 672                                                 intercount.first > 1 ? "s" : "",
 673                                                 intercount.first > 1 ? "are" : "is",
 674                                                 intercount.second.c_str());
 675                                 return false;
 676                         }
 677                         /* Give the module a chance to tidy out all its metadata */
 678                         for (chan_hash::iterator c = this->chanlist->begin(); c != this->chanlist->end(); c++)
 679                         {
 680                                 modules[j]->OnCleanup(TYPE_CHANNEL,c->second);
 681                         }
 682                         for (user_hash::iterator u = this->clientlist->begin(); u != this->clientlist->end(); u++)
 683                         {
 684                                 modules[j]->OnCleanup(TYPE_USER,u->second);
 685                         }
 686
 687                         /* Tidy up any dangling resolvers */
 688                         this->Res->CleanResolvers(modules[j]);
 689
 690                         FOREACH_MOD_I(this,I_OnUnloadModule,OnUnloadModule(modules[j],Config->module_names[j]));
 691
 692                         for(int t = 0; t < 255; t++)
 693                         {
 694                                 Config->global_implementation[t] -= Config->implement_lists[j][t];
 695                         }
 696
 697                         /* We have to renumber implement_lists after unload because the module numbers change!
 698                          */
 699                         for(int j2 = j; j2 < 254; j2++)
 700                         {
 701                                 for(int t = 0; t < 255; t++)
 702                                 {
 703                                         Config->implement_lists[j2][t] = Config->implement_lists[j2+1][t];
 704                                 }
 705                         }
 706
 707                         // found the module
 708                         Parser->RemoveCommands(filename);
 709                         this->EraseModule(j);
 710                         this->EraseFactory(j);
 711                         this->Log(DEFAULT,"Module %s unloaded",filename);
 712                         this->ModCount--;
 713                         BuildISupport();
 714                         return true;
 715                 }
 716         }
 717         this->Log(DEFAULT,"Module %s is not loaded, cannot unload it!",filename);
 718         snprintf(MODERR,MAXBUF,"Module not loaded");
 719         return false;
 720 }
 721
 722 bool InspIRCd::LoadModule(const char* filename)
 723 {
 724         /* Do we have a glob pattern in the filename?
 725          * The user wants to load multiple modules which
 726          * match the pattern.
 727          */
 728         if (strchr(filename,'*') || (strchr(filename,'?')))
 729         {
 730                 int n_match = 0;
 731                 DIR* library = opendir(Config->ModPath);
 732                 if (library)
 733                 {
 734                         /* Try and locate and load all modules matching the pattern */
 735                         dirent* entry = NULL;
 736                         while ((entry = readdir(library)))
 737                         {
 738                                 if (this->MatchText(entry->d_name, filename))
 739                                 {
 740                                         if (!this->LoadModule(entry->d_name))
 741                                                 n_match++;
 742                                 }
 743                         }
 744                         closedir(library);
 745                 }
 746                 /* Loadmodule will now return false if any one of the modules failed
 747                  * to load (but wont abort when it encounters a bad one) and when 1 or
 748                  * more modules were actually loaded.
 749                  */
 750                 return (n_match > 0);
 751         }
 752
 753         char modfile[MAXBUF];
 754         snprintf(modfile,MAXBUF,"%s/%s",Config->ModPath,filename);
 755         std::string filename_str = filename;
 756
 757         if (!ServerConfig::DirValid(modfile))
 758         {
 759                 this->Log(DEFAULT,"Module %s is not within the modules directory.",modfile);
 760                 snprintf(MODERR,MAXBUF,"Module %s is not within the modules directory.",modfile);
 761                 return false;
 762         }
 763         if (ServerConfig::FileExists(modfile))
 764         {
 765
 766                 for (unsigned int j = 0; j < Config->module_names.size(); j++)
 767                 {
 768                         if (Config->module_names[j] == filename_str)
 769                         {
 770                                 this->Log(DEFAULT,"Module %s is already loaded, cannot load a module twice!",modfile);
 771                                 snprintf(MODERR,MAXBUF,"Module already loaded");
 772                                 return false;
 773                         }
 774                 }
 775                 try
 776                 {
 777                         ircd_module* a = new ircd_module(this, modfile);
 778                         factory[this->ModCount+1] = a;
 779                         if (factory[this->ModCount+1]->LastError())
 780                         {
 781                                 this->Log(DEFAULT,"Unable to load %s: %s",modfile,factory[this->ModCount+1]->LastError());
 782                                 snprintf(MODERR,MAXBUF,"Loader/Linker error: %s",factory[this->ModCount+1]->LastError());
 783                                 return false;
 784                         }
 785                         if ((long)factory[this->ModCount+1]->factory != -1)
 786                         {
 787                                 Module* m = factory[this->ModCount+1]->factory->CreateModule(this);
 788
 789                                 Version v = m->GetVersion();
 790
 791                                 if (v.API != API_VERSION)
 792                                 {
 793                                         delete m;
 794                                         delete a;
 795                                         this->Log(DEFAULT,"Unable to load %s: Incorrect module API version: %d (our version: %d)",modfile,v.API,API_VERSION);
 796                                         snprintf(MODERR,MAXBUF,"Loader/Linker error: Incorrect module API version: %d (our version: %d)",v.API,API_VERSION);
 797                                         return false;
 798                                 }
 799                                 else
 800                                 {
 801                                         this->Log(DEFAULT,"New module introduced: %s (API version %d, Module version %d.%d.%d.%d)%s", filename, v.API, v.Major, v.Minor, v.Revision, v.Build, (!(v.Flags & VF_VENDOR) ? " [3rd Party]" : " [Vendor]"));
 802                                 }
 803
 804                                 modules[this->ModCount+1] = m;
 805                                 /* save the module and the module's classfactory, if
 806                                  * this isnt done, random crashes can occur :/ */
 807                                 Config->module_names.push_back(filename);
 808
 809                                 char* x = &Config->implement_lists[this->ModCount+1][0];
 810                                 for(int t = 0; t < 255; t++)
 811                                         x[t] = 0;
 812
 813                                 modules[this->ModCount+1]->Implements(x);
 814
 815                                 for(int t = 0; t < 255; t++)
 816                                         Config->global_implementation[t] += Config->implement_lists[this->ModCount+1][t];
 817                         }
 818                         else
 819                         {
 820                                 this->Log(DEFAULT,"Unable to load %s",modfile);
 821                                 snprintf(MODERR,MAXBUF,"Factory function failed: Probably missing init_module() entrypoint.");
 822                                 return false;
 823                         }
 824                 }
 825                 catch (CoreException& modexcept)
 826                 {
 827                         this->Log(DEFAULT,"Unable to load %s: %s",modfile,modexcept.GetReason());
 828                         snprintf(MODERR,MAXBUF,"Factory function of %s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 829                         return false;
 830                 }
 831         }
 832         else
 833         {
 834                 this->Log(DEFAULT,"InspIRCd: startup: Module Not Found %s",modfile);
 835                 snprintf(MODERR,MAXBUF,"Module file could not be found");
 836                 return false;
 837         }
 838         this->ModCount++;
 839         FOREACH_MOD_I(this,I_OnLoadModule,OnLoadModule(modules[this->ModCount],filename_str));
 840         // now work out which modules, if any, want to move to the back of the queue,
 841         // and if they do, move them there.
 842         std::vector<std::string> put_to_back;
 843         std::vector<std::string> put_to_front;
 844         std::map<std::string,std::string> put_before;
 845         std::map<std::string,std::string> put_after;
 846         for (unsigned int j = 0; j < Config->module_names.size(); j++)
 847         {
 848                 if (modules[j]->Prioritize() == PRIORITY_LAST)
 849                         put_to_back.push_back(Config->module_names[j]);
 850                 else if (modules[j]->Prioritize() == PRIORITY_FIRST)
 851                         put_to_front.push_back(Config->module_names[j]);
 852                 else if ((modules[j]->Prioritize() & 0xFF) == PRIORITY_BEFORE)
 853                         put_before[Config->module_names[j]] = Config->module_names[modules[j]->Prioritize() >> 8];
 854                 else if ((modules[j]->Prioritize() & 0xFF) == PRIORITY_AFTER)
 855                         put_after[Config->module_names[j]] = Config->module_names[modules[j]->Prioritize() >> 8];
 856         }
 857         for (unsigned int j = 0; j < put_to_back.size(); j++)
 858                 MoveToLast(put_to_back[j]);
 859         for (unsigned int j = 0; j < put_to_front.size(); j++)
 860                 MoveToFirst(put_to_front[j]);
 861         for (std::map<std::string,std::string>::iterator j = put_before.begin(); j != put_before.end(); j++)
 862                 MoveBefore(j->first,j->second);
 863         for (std::map<std::string,std::string>::iterator j = put_after.begin(); j != put_after.end(); j++)
 864                 MoveAfter(j->first,j->second);
 865         BuildISupport();
 866         return true;
 867 }
 868
 869 void InspIRCd::DoOneIteration(bool process_module_sockets)
 870 {
 871         static rusage ru;
 872
 873         /* time() seems to be a pretty expensive syscall, so avoid calling it too much.
 874          * Once per loop iteration is pleanty.
 875          */
 876         OLDTIME = TIME;
 877         TIME = time(NULL);
 878
 879         /* Run background module timers every few seconds
 880          * (the docs say modules shouldnt rely on accurate
 881          * timing using this event, so we dont have to
 882          * time this exactly).
 883          */
 884         if (TIME != OLDTIME)
 885         {
 886                 if (TIME < OLDTIME)
 887                         WriteOpers("*** \002EH?!\002 -- Time is flowing BACKWARDS in this dimension! Clock drifted backwards %d secs.",abs(OLDTIME-TIME));
 888                 if ((TIME % 3600) == 0)
 889                 {
 890                         this->RehashUsersAndChans();
 891                         FOREACH_MOD_I(this, I_OnGarbageCollect, OnGarbageCollect());
 892                 }
 893                 Timers->TickTimers(TIME);
 894                 this->DoBackgroundUserStuff(TIME);
 895
 896                 if ((TIME % 5) == 0)
 897                 {
 898                         XLines->expire_lines();
 899                         FOREACH_MOD_I(this,I_OnBackgroundTimer,OnBackgroundTimer(TIME));
 900                         Timers->TickMissedTimers(TIME);
 901                 }
 902
 903                 if (!getrusage(0, &ru))
 904                 {
 905                         gettimeofday(&this->stats->LastSampled, NULL);
 906                         this->stats->LastCPU = ru.ru_utime;
 907                 }
 908         }
 909
 910         /* Call the socket engine to wait on the active
 911          * file descriptors. The socket engine has everything's
 912          * descriptors in its list... dns, modules, users,
 913          * servers... so its nice and easy, just one call.
 914          * This will cause any read or write events to be
 915          * dispatched to their handlers.
 916          */
 917         SE->DispatchEvents();
 918
 919         /* if any users was quit, take them out */
 920         GlobalCulls.Apply();
 921
 922         /* If any inspsockets closed, remove them */
 923         for (std::map<InspSocket*,InspSocket*>::iterator x = SocketCull.begin(); x != SocketCull.end(); ++x)
 924         {
 925                 SE->DelFd(x->second);
 926                 x->second->Close();
 927                 delete x->second;
 928         }
 929         SocketCull.clear();
 930 }
 931
 932 bool InspIRCd::IsIdent(const char* n)
 933 {
 934         if (!n || !*n)
 935                 return false;
 936
 937         for (char* i = (char*)n; *i; i++)
 938         {
 939                 if ((*i >= 'A') && (*i <= '}'))
 940                 {
 941                         continue;
 942                 }
 943                 if (((*i >= '0') && (*i <= '9')) || (*i == '-') || (*i == '.'))
 944                 {
 945                         continue;
 946                 }
 947                 return false;
 948         }
 949         return true;
 950 }
 951
 952 int InspIRCd::Run()
 953 {
 954         while (true)
 955         {
 956                 DoOneIteration(true);
 957         }
 958         /* This is never reached -- we hope! */
 959         return 0;
 960 }
 961
 962 /**********************************************************************************/
 963
 964 /**
 965  * An ircd in four lines! bwahahaha. ahahahahaha. ahahah *cough*.
 966  */
 967
 968 int main(int argc, char** argv)
 969 {
 970         SI = new InspIRCd(argc, argv);
 971         SI->Run();
 972         delete SI;
 973         return 0;
 974 }
 975
 976 /* this returns true when all modules are satisfied that the user should be allowed onto the irc server
 977  * (until this returns true, a user will block in the waiting state, waiting to connect up to the
 978  * registration timeout maximum seconds)
 979  */
 980 bool InspIRCd::AllModulesReportReady(userrec* user)
 981 {
 982         if (!Config->global_implementation[I_OnCheckReady])
 983                 return true;
 984
 985         for (int i = 0; i <= this->GetModuleCount(); i++)
 986         {
 987                 if (Config->implement_lists[i][I_OnCheckReady])
 988                 {
 989                         int res = modules[i]->OnCheckReady(user);
 990                         if (!res)
 991                                 return false;
 992                 }
 993         }
 994         return true;
 995 }
 996
 997 int InspIRCd::GetModuleCount()
 998 {
 999         return this->ModCount;
1000 }
1001
1002 time_t InspIRCd::Time(bool delta)
1003 {
1004         if (delta)
1005                 return TIME + time_delta;
1006         return TIME;
1007 }
1008
1009 int InspIRCd::SetTimeDelta(int delta)
1010 {
1011         int old = time_delta;
1012         time_delta = delta;
1013         this->Log(DEBUG, "Time delta set to %d (was %d)", time_delta, old);
1014         return old;
1015 }
1016
1017 void InspIRCd::AddLocalClone(userrec* user)
1018 {
1019         clonemap::iterator x = local_clones.find(user->GetIPString());
1020         if (x != local_clones.end())
1021                 x->second++;
1022         else
1023                 local_clones[user->GetIPString()] = 1;
1024 }
1025
1026 void InspIRCd::AddGlobalClone(userrec* user)
1027 {
1028         clonemap::iterator y = global_clones.find(user->GetIPString());
1029         if (y != global_clones.end())
1030                 y->second++;
1031         else
1032                 global_clones[user->GetIPString()] = 1;
1033 }
1034
1035 int InspIRCd::GetTimeDelta()
1036 {
1037         return time_delta;
1038 }
1039
1040 bool FileLogger::Readable()
1041 {
1042         return false;
1043 }
1044
1045 void FileLogger::HandleEvent(EventType et, int errornum)
1046 {
1047         this->WriteLogLine("");
1048         if (log)
1049                 ServerInstance->SE->DelFd(this);
1050 }
1051
1052 void FileLogger::WriteLogLine(const std::string &line)
1053 {
1054         if (line.length())
1055                 buffer.append(line);
1056
1057         if (log)
1058         {
1059                 int written = fprintf(log,"%s",buffer.c_str());
1060                 if ((written >= 0) && (written < (int)buffer.length()))
1061                 {
1062                         buffer.erase(0, buffer.length());
1063                         ServerInstance->SE->AddFd(this);
1064                 }
1065                 else if (written == -1)
1066                 {
1067                         if (errno == EAGAIN)
1068                                 ServerInstance->SE->AddFd(this);
1069                 }
1070                 else
1071                 {
1072                         /* Wrote the whole buffer, and no need for write callback */
1073                         buffer = "";
1074                 }
1075
1076                 if (writeops++ % 20)
1077                 {
1078                         fflush(log);
1079                 }
1080         }
1081 }
1082
1083 void FileLogger::Close()
1084 {
1085         if (log)
1086         {
1087                 int flags = fcntl(fileno(log), F_GETFL, 0);
1088                 fcntl(fileno(log), F_SETFL, flags ^ O_NONBLOCK);
1089                 if (buffer.size())
1090                         fprintf(log,"%s",buffer.c_str());
1091
1092                 ServerInstance->SE->DelFd(this);
1093
1094                 fflush(log);
1095                 fclose(log);
1096         }
1097
1098         buffer = "";
1099 }
1100
1101 FileLogger::FileLogger(InspIRCd* Instance, FILE* logfile) : ServerInstance(Instance), log(logfile), writeops(0)
1102 {
1103         if (log)
1104         {
1105                 irc::sockets::NonBlocking(fileno(log));
1106                 this->SetFd(fileno(log));
1107                 buffer = "";
1108         }
1109 }
1110
1111 FileLogger::~FileLogger()
1112 {
1113         this->Close();
1114 }
1115