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