src/modules.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 /* $Core: libIRCDmodules */
  15
  16 #include "inspircd.h"
  17 #include "wildcard.h"
  18 #include "xline.h"
  19 #include "socket.h"
  20 #include "socketengine.h"
  21 #include "command_parse.h"
  22 #include "dns.h"
  23 #include "exitcodes.h"
  24
  25 #ifndef WIN32
  26         #include <dirent.h>
  27 #endif
  28
  29 // version is a simple class for holding a modules version number
  30 Version::Version(int major, int minor, int revision, int build, int flags, int api_ver)
  31 : Major(major), Minor(minor), Revision(revision), Build(build), Flags(flags), API(api_ver)
  32 {
  33 }
  34
  35 Request::Request(char* anydata, Module* src, Module* dst)
  36 : data(anydata), source(src), dest(dst)
  37 {
  38         /* Ensure that because this module doesnt support ID strings, it doesnt break modules that do
  39          * by passing them uninitialized pointers (could happen)
  40          */
  41         id = '\0';
  42 }
  43
  44 Request::Request(Module* src, Module* dst, const char* idstr)
  45 : id(idstr), source(src), dest(dst)
  46 {
  47 }
  48
  49 char* Request::GetData()
  50 {
  51         return this->data;
  52 }
  53
  54 const char* Request::GetId()
  55 {
  56         return this->id;
  57 }
  58
  59 Module* Request::GetSource()
  60 {
  61         return this->source;
  62 }
  63
  64 Module* Request::GetDest()
  65 {
  66         return this->dest;
  67 }
  68
  69 char* Request::Send()
  70 {
  71         if (this->dest)
  72         {
  73                 return dest->OnRequest(this);
  74         }
  75         else
  76         {
  77                 return NULL;
  78         }
  79 }
  80
  81 Event::Event(char* anydata, Module* src, const std::string &eventid) : data(anydata), source(src), id(eventid) { }
  82
  83 char* Event::GetData()
  84 {
  85         return (char*)this->data;
  86 }
  87
  88 Module* Event::GetSource()
  89 {
  90         return this->source;
  91 }
  92
  93 char* Event::Send(InspIRCd* ServerInstance)
  94 {
  95         FOREACH_MOD(I_OnEvent,OnEvent(this));
  96         return NULL;
  97 }
  98
  99 std::string Event::GetEventID()
 100 {
 101         return this->id;
 102 }
 103
 104
 105 // These declarations define the behavours of the base class Module (which does nothing at all)
 106
 107                 Module::Module(InspIRCd* Me) : ServerInstance(Me) { }
 108                 Module::~Module() { }
 109 void            Module::OnUserConnect(User*) { }
 110 void            Module::OnUserQuit(User*, const std::string&, const std::string&) { }
 111 void            Module::OnUserDisconnect(User*) { }
 112 void            Module::OnUserJoin(User*, Channel*, bool&) { }
 113 void            Module::OnPostJoin(User*, Channel*) { }
 114 void            Module::OnUserPart(User*, Channel*, const std::string&, bool&) { }
 115 void            Module::OnRehash(User*, const std::string&) { }
 116 void            Module::OnServerRaw(std::string&, bool, User*) { }
 117 int             Module::OnUserPreJoin(User*, Channel*, const char*, std::string&) { return 0; }
 118 void            Module::OnMode(User*, void*, int, const std::string&) { }
 119 Version         Module::GetVersion() { return Version(1,0,0,0,VF_VENDOR,-1); }
 120 void            Module::OnOper(User*, const std::string&) { }
 121 void            Module::OnPostOper(User*, const std::string&) { }
 122 void            Module::OnInfo(User*) { }
 123 void            Module::OnWhois(User*, User*) { }
 124 int             Module::OnUserPreInvite(User*, User*, Channel*) { return 0; }
 125 int             Module::OnUserPreMessage(User*, void*, int, std::string&, char, CUList&) { return 0; }
 126 int             Module::OnUserPreNotice(User*, void*, int, std::string&, char, CUList&) { return 0; }
 127 int             Module::OnUserPreNick(User*, const std::string&) { return 0; }
 128 void            Module::OnUserPostNick(User*, const std::string&) { }
 129 int             Module::OnAccessCheck(User*, User*, Channel*, int) { return ACR_DEFAULT; }
 130 void            Module::On005Numeric(std::string&) { }
 131 int             Module::OnKill(User*, User*, const std::string&) { return 0; }
 132 void            Module::OnLoadModule(Module*, const std::string&) { }
 133 void            Module::OnUnloadModule(Module*, const std::string&) { }
 134 void            Module::OnBackgroundTimer(time_t) { }
 135 int             Module::OnPreCommand(const std::string&, const char**, int, User *, bool, const std::string&) { return 0; }
 136 void            Module::OnPostCommand(const std::string&, const char**, int, User *, CmdResult, const std::string&) { }
 137 bool            Module::OnCheckReady(User*) { return true; }
 138 int             Module::OnUserRegister(User*) { return 0; }
 139 int             Module::OnUserPreKick(User*, User*, Channel*, const std::string&) { return 0; }
 140 void            Module::OnUserKick(User*, User*, Channel*, const std::string&, bool&) { }
 141 int             Module::OnCheckInvite(User*, Channel*) { return 0; }
 142 int             Module::OnCheckKey(User*, Channel*, const std::string&) { return 0; }
 143 int             Module::OnCheckLimit(User*, Channel*) { return 0; }
 144 int             Module::OnCheckBan(User*, Channel*) { return 0; }
 145 int             Module::OnStats(char, User*, string_list&) { return 0; }
 146 int             Module::OnChangeLocalUserHost(User*, const std::string&) { return 0; }
 147 int             Module::OnChangeLocalUserGECOS(User*, const std::string&) { return 0; }
 148 int             Module::OnLocalTopicChange(User*, Channel*, const std::string&) { return 0; }
 149 void            Module::OnEvent(Event*) { return; }
 150 char*           Module::OnRequest(Request*) { return NULL; }
 151 int             Module::OnOperCompare(const std::string&, const std::string&, int) { return 0; }
 152 void            Module::OnGlobalOper(User*) { }
 153 void            Module::OnPostConnect(User*) { }
 154 int             Module::OnAddBan(User*, Channel*, const std::string &) { return 0; }
 155 int             Module::OnDelBan(User*, Channel*, const std::string &) { return 0; }
 156 void            Module::OnRawSocketAccept(int, const std::string&, int) { }
 157 int             Module::OnRawSocketWrite(int, const char*, int) { return 0; }
 158 void            Module::OnRawSocketClose(int) { }
 159 void            Module::OnRawSocketConnect(int) { }
 160 int             Module::OnRawSocketRead(int, char*, unsigned int, int&) { return 0; }
 161 void            Module::OnUserMessage(User*, void*, int, const std::string&, char, const CUList&) { }
 162 void            Module::OnUserNotice(User*, void*, int, const std::string&, char, const CUList&) { }
 163 void            Module::OnRemoteKill(User*, User*, const std::string&, const std::string&) { }
 164 void            Module::OnUserInvite(User*, User*, Channel*) { }
 165 void            Module::OnPostLocalTopicChange(User*, Channel*, const std::string&) { }
 166 void            Module::OnGetServerDescription(const std::string&, std::string&) { }
 167 void            Module::OnSyncUser(User*, Module*, void*) { }
 168 void            Module::OnSyncChannel(Channel*, Module*, void*) { }
 169 void            Module::ProtoSendMode(void*, int, void*, const std::string&) { }
 170 void            Module::OnSyncChannelMetaData(Channel*, Module*, void*, const std::string&, bool) { }
 171 void            Module::OnSyncUserMetaData(User*, Module*, void*, const std::string&, bool) { }
 172 void            Module::OnSyncOtherMetaData(Module*, void*, bool) { }
 173 void            Module::OnDecodeMetaData(int, void*, const std::string&, const std::string&) { }
 174 void            Module::ProtoSendMetaData(void*, int, void*, const std::string&, const std::string&) { }
 175 void            Module::OnWallops(User*, const std::string&) { }
 176 void            Module::OnChangeHost(User*, const std::string&) { }
 177 void            Module::OnChangeName(User*, const std::string&) { }
 178 void            Module::OnAddLine(User*, XLine*) { }
 179 void            Module::OnDelLine(User*, XLine*) { }
 180 void            Module::OnCleanup(int, void*) { }
 181 void            Module::OnChannelDelete(Channel*) { }
 182 void            Module::OnSetAway(User*) { }
 183 void            Module::OnCancelAway(User*) { }
 184 int             Module::OnUserList(User*, Channel*, CUList*&) { return 0; }
 185 int             Module::OnWhoisLine(User*, User*, int&, std::string&) { return 0; }
 186 void            Module::OnBuildExemptList(MessageType, Channel*, User*, char, CUList&, const std::string&) { }
 187 void            Module::OnGarbageCollect() { }
 188 void            Module::OnBufferFlushed(User*) { }
 189 void            Module::OnText(User*, void*, int, const std::string&, char, CUList&) { }
 190
 191
 192 ModuleManager::ModuleManager(InspIRCd* Ins) : ModCount(0), Instance(Ins)
 193 {
 194         for (int n = I_BEGIN + 1; n != I_END; ++n)
 195                 EventHandlers.push_back(std::vector<Module*>());
 196 }
 197
 198 ModuleManager::~ModuleManager()
 199 {
 200 }
 201
 202 bool ModuleManager::Attach(Implementation i, Module* mod)
 203 {
 204         if (std::find(EventHandlers[i].begin(), EventHandlers[i].end(), mod) != EventHandlers[i].end())
 205                 return false;
 206
 207         EventHandlers[i].push_back(mod);
 208         return true;
 209 }
 210
 211 bool ModuleManager::Detach(Implementation i, Module* mod)
 212 {
 213         EventHandlerIter x = std::find(EventHandlers[i].begin(), EventHandlers[i].end(), mod);
 214
 215         if (x == EventHandlers[i].end())
 216                 return false;
 217
 218         EventHandlers[i].erase(x);
 219         return true;
 220 }
 221
 222 void ModuleManager::Attach(Implementation* i, Module* mod, size_t sz)
 223 {
 224         for (size_t n = 0; n < sz; ++n)
 225                 Attach(i[n], mod);
 226 }
 227
 228 void ModuleManager::DetachAll(Module* mod)
 229 {
 230         for (size_t n = I_BEGIN + 1; n != I_END; ++n)
 231                 Detach((Implementation)n, mod);
 232 }
 233
 234 bool ModuleManager::SetPriority(Module* mod, PriorityState s)
 235 {
 236         for (size_t n = I_BEGIN + 1; n != I_END; ++n)
 237                 SetPriority(mod, (Implementation)n, s);
 238
 239         return true;
 240 }
 241
 242 bool ModuleManager::SetPriority(Module* mod, Implementation i, PriorityState s, Module** modules, size_t sz)
 243 {
 244         if (GetModuleName(mod) != "m_spanningtree.so")
 245                 Instance->Log(DEBUG,"ModuleManager::SetPriority called by %s, priority state %s num_modules=%u", GetModuleName(mod).c_str(), s == PRIO_BEFORE ? "PRIO_BEFORE" :
 246                         s == PRIO_AFTER ? "PRIO_AFTER" :
 247                         s == PRIO_LAST ? "PRIO_LAST" :
 248                         s == PRIO_FIRST ? "PRIO_FIRST" : "<unknown!>",
 249                         sz);
 250
 251         size_t swap_pos;
 252         size_t source = 0;
 253         bool swap = true;
 254         bool found = false;
 255
 256         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
 257         {
 258                 if (EventHandlers[i][x] == mod)
 259                 {
 260                         source = x;
 261                         found = true;
 262                         break;
 263                 }
 264         }
 265
 266         if (!found)
 267                 return false;
 268
 269         Instance->Log(DEBUG,"ModuleManager::SetPriority: My position: %u", source);
 270
 271         if (modules)
 272         {
 273                 for (size_t n = 0; n < sz; ++n)
 274                 {
 275                         if (modules[n])
 276                                 Instance->Log(DEBUG,"    Listed Module: [%08x] %s", modules[n], GetModuleName(modules[n]).c_str());
 277                         else
 278                                 Instance->Log(DEBUG,"    [null module]");
 279                 }
 280         }
 281
 282         switch (s)
 283         {
 284                 case PRIO_DONTCARE:
 285                         swap = false;
 286                 break;
 287                 case PRIO_FIRST:
 288                         swap_pos = 0;
 289                 break;
 290                 case PRIO_LAST:
 291                         if (EventHandlers[i].empty())
 292                                 swap_pos = 0;
 293                         else
 294                                 swap_pos = EventHandlers[i].size() - 1;
 295                 break;
 296                 case PRIO_AFTER:
 297                 {
 298                         /* Find the latest possible position */
 299                         swap_pos = 0;
 300                         swap = false;
 301                         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
 302                         {
 303                                 for (size_t n = 0; n < sz; ++n)
 304                                 {
 305                                         if ((modules[n]) && (EventHandlers[i][x] == modules[n]) && (x >= swap_pos) && (source <= swap_pos))
 306                                         {
 307                                                 swap_pos = x;
 308                                                 swap = true;
 309                                         }
 310                                 }
 311                         }
 312                 }
 313                 break;
 314                 case PRIO_BEFORE:
 315                 {
 316                         swap_pos = EventHandlers[i].size() - 1;
 317                         swap = false;
 318                         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
 319                         {
 320                                 for (size_t n = 0; n < sz; ++n)
 321                                 {
 322                                         if ((modules[n]) && (EventHandlers[i][x] == modules[n]) && (x <= swap_pos) && (source >= swap_pos))
 323                                         {
 324                                                 swap = true;
 325                                                 swap_pos = x;
 326                                         }
 327                                 }
 328                         }
 329                 }
 330                 break;
 331         }
 332
 333         if (swap && (swap_pos != source))
 334         {
 335                 std::swap(EventHandlers[i][swap_pos], EventHandlers[i][source]);
 336                 Instance->Log(DEBUG,"Swap locations %u and %u", swap_pos, source);
 337         }
 338         else
 339                 Instance->Log(DEBUG,"No need to swap");
 340
 341         Instance->Log(DEBUG,"New ordering:");
 342         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
 343         {
 344                 Instance->Log(DEBUG,"  [%08x] %s", EventHandlers[i][x], GetModuleName(EventHandlers[i][x]).c_str());
 345         }
 346
 347         return true;
 348 }
 349
 350 const char* ModuleManager::LastError()
 351 {
 352         return MODERR;
 353 }
 354
 355 bool ModuleManager::Load(const char* filename)
 356 {
 357         /* Do we have a glob pattern in the filename?
 358          * The user wants to load multiple modules which
 359          * match the pattern.
 360          */
 361         if (strchr(filename,'*') || (strchr(filename,'?')))
 362         {
 363                 int n_match = 0;
 364                 DIR* library = opendir(Instance->Config->ModPath);
 365                 if (library)
 366                 {
 367                         /* Try and locate and load all modules matching the pattern */
 368                         dirent* entry = NULL;
 369                         while ((entry = readdir(library)))
 370                         {
 371                                 if (Instance->MatchText(entry->d_name, filename))
 372                                 {
 373                                         if (!this->Load(entry->d_name))
 374                                                 n_match++;
 375                                 }
 376                         }
 377                         closedir(library);
 378                 }
 379                 /* Loadmodule will now return false if any one of the modules failed
 380                  * to load (but wont abort when it encounters a bad one) and when 1 or
 381                  * more modules were actually loaded.
 382                  */
 383                 return (n_match > 0);
 384         }
 385
 386         char modfile[MAXBUF];
 387         snprintf(modfile,MAXBUF,"%s/%s",Instance->Config->ModPath,filename);
 388         std::string filename_str = filename;
 389
 390         if (!ServerConfig::DirValid(modfile))
 391         {
 392                 snprintf(MODERR, MAXBUF,"Module %s is not within the modules directory.", modfile);
 393                 Instance->Log(DEFAULT, MODERR);
 394                 return false;
 395         }
 396
 397         if (!ServerConfig::FileExists(modfile))
 398         {
 399                 snprintf(MODERR,MAXBUF,"Module file could not be found: %s", modfile);
 400                 Instance->Log(DEFAULT, MODERR);
 401                 return false;
 402         }
 403
 404         if(find(Instance->Config->module_names.begin(), Instance->Config->module_names.end(), filename_str) != Instance->Config->module_names.end())
 405         {
 406                 Instance->Log(DEFAULT,"Module %s is already loaded, cannot load a module twice!",modfile);
 407                 snprintf(MODERR, MAXBUF, "Module already loaded");
 408                 return false;
 409         }
 410
 411         Module* newmod;
 412         ircd_module* newhandle;
 413
 414         newmod = NULL;
 415         newhandle = NULL;
 416
 417         try
 418         {
 419                 /* This will throw a CoreException if there's a problem loading
 420                  * the module file or getting a pointer to the init_module symbol.
 421                  */
 422                 newhandle = new ircd_module(Instance, modfile, "init_module");
 423
 424                 handles[this->ModCount+1] = newhandle;
 425
 426                 newmod = handles[this->ModCount+1]->CallInit();
 427
 428                 if(newmod)
 429                 {
 430                         Version v = newmod->GetVersion();
 431
 432                         if (v.API != API_VERSION)
 433                         {
 434                                 delete newmod;
 435                                 Instance->Log(DEFAULT,"Unable to load %s: Incorrect module API version: %d (our version: %d)",modfile,v.API,API_VERSION);
 436                                 snprintf(MODERR,MAXBUF,"Loader/Linker error: Incorrect module API version: %d (our version: %d)",v.API,API_VERSION);
 437                                 return false;
 438                         }
 439                         else
 440                         {
 441                                 Instance->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]"));
 442                         }
 443
 444                         modules[this->ModCount+1] = newmod;
 445
 446                         /* save the module and the module's classfactory, if
 447                          * this isnt done, random crashes can occur :/ */
 448                         Instance->Config->module_names.push_back(filename);
 449                 }
 450                 else
 451                 {
 452                         Instance->Log(DEFAULT, "Unable to load %s",modfile);
 453                         snprintf(MODERR,MAXBUF, "Probably missing init_module() entrypoint, but dlsym() didn't notice a problem");
 454                         return false;
 455                 }
 456         }
 457         catch (LoadModuleException& modexcept)
 458         {
 459                 Instance->Log(DEFAULT,"Unable to load %s: %s", modfile, modexcept.GetReason());
 460                 snprintf(MODERR,MAXBUF,"Loader/Linker error: %s", modexcept.GetReason());
 461                 return false;
 462         }
 463         catch (FindSymbolException& modexcept)
 464         {
 465                 Instance->Log(DEFAULT,"Unable to load %s: %s", modfile, modexcept.GetReason());
 466                 snprintf(MODERR,MAXBUF,"Loader/Linker error: %s", modexcept.GetReason());
 467                 return false;
 468         }
 469         catch (CoreException& modexcept)
 470         {
 471                 Instance->Log(DEFAULT,"Unable to load %s: %s",modfile,modexcept.GetReason());
 472                 snprintf(MODERR,MAXBUF,"Factory function of %s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 473                 return false;
 474         }
 475
 476         this->ModCount++;
 477         FOREACH_MOD_I(Instance,I_OnLoadModule,OnLoadModule(modules[this->ModCount],filename_str));
 478
 479         for (int n = 0; n != this->ModCount+1; ++n)
 480                 modules[n]->Prioritize();
 481
 482         Instance->BuildISupport();
 483         return true;
 484 }
 485
 486 bool ModuleManager::EraseHandle(unsigned int j)
 487 {
 488         ModuleHandleList::iterator iter;
 489
 490         if (j >= handles.size())
 491         {
 492                 return false;
 493         }
 494
 495         iter = handles.begin() + j;
 496
 497         if(*iter)
 498         {
 499                 delete *iter;
 500                 handles.erase(iter);
 501                 handles.push_back(NULL);
 502         }
 503
 504         return true;
 505 }
 506
 507 bool ModuleManager::EraseModule(unsigned int j)
 508 {
 509         bool success = false;
 510
 511         ModuleList::iterator iter;
 512
 513         if (j >= modules.size())
 514         {
 515                 return false;
 516         }
 517
 518         iter = modules.begin() + j;
 519
 520         if (*iter)
 521         {
 522                 delete *iter;
 523                 modules.erase(iter);
 524                 modules.push_back(NULL);
 525                 success = true;
 526         }
 527
 528         std::vector<std::string>::iterator iter2;
 529
 530         if (j >= Instance->Config->module_names.size())
 531         {
 532                 return false;
 533         }
 534
 535         iter2 = Instance->Config->module_names.begin() + j;
 536
 537         Instance->Config->module_names.erase(iter2);
 538         success = true;
 539
 540         return success;
 541 }
 542
 543 bool ModuleManager::Unload(const char* filename)
 544 {
 545         std::string filename_str = filename;
 546         for (unsigned int j = 0; j != Instance->Config->module_names.size(); j++)
 547         {
 548                 if (Instance->Config->module_names[j] == filename_str)
 549                 {
 550                         if (modules[j]->GetVersion().Flags & VF_STATIC)
 551                         {
 552                                 Instance->Log(DEFAULT,"Failed to unload STATIC module %s",filename);
 553                                 snprintf(MODERR,MAXBUF,"Module not unloadable (marked static)");
 554                                 return false;
 555                         }
 556                         std::pair<int,std::string> intercount = GetInterfaceInstanceCount(modules[j]);
 557                         if (intercount.first > 0)
 558                         {
 559                                 Instance->Log(DEFAULT,"Failed to unload module %s, being used by %d other(s) via interface '%s'",filename, intercount.first, intercount.second.c_str());
 560                                 snprintf(MODERR,MAXBUF,"Module not unloadable (Still in use by %d other module%s which %s using its interface '%s') -- unload dependent modules first!",
 561                                                 intercount.first,
 562                                                 intercount.first > 1 ? "s" : "",
 563                                                 intercount.first > 1 ? "are" : "is",
 564                                                 intercount.second.c_str());
 565                                 return false;
 566                         }
 567                         /* Give the module a chance to tidy out all its metadata */
 568                         for (chan_hash::iterator c = Instance->chanlist->begin(); c != Instance->chanlist->end(); c++)
 569                         {
 570                                 modules[j]->OnCleanup(TYPE_CHANNEL,c->second);
 571                         }
 572                         for (user_hash::iterator u = Instance->clientlist->begin(); u != Instance->clientlist->end(); u++)
 573                         {
 574                                 modules[j]->OnCleanup(TYPE_USER,u->second);
 575                         }
 576
 577                         /* Tidy up any dangling resolvers */
 578                         Instance->Res->CleanResolvers(modules[j]);
 579
 580                         FOREACH_MOD_I(Instance,I_OnUnloadModule,OnUnloadModule(modules[j],Instance->Config->module_names[j]));
 581
 582                         this->DetachAll(modules[j]);
 583
 584                         // found the module
 585                         Instance->Parser->RemoveCommands(filename);
 586                         this->EraseModule(j);
 587                         this->EraseHandle(j);
 588                         Instance->Log(DEFAULT,"Module %s unloaded",filename);
 589                         this->ModCount--;
 590                         Instance->BuildISupport();
 591                         return true;
 592                 }
 593         }
 594         Instance->Log(DEFAULT,"Module %s is not loaded, cannot unload it!",filename);
 595         snprintf(MODERR,MAXBUF,"Module not loaded");
 596         return false;
 597 }
 598
 599 /* We must load the modules AFTER initializing the socket engine, now */
 600 void ModuleManager::LoadAll()
 601 {
 602         char configToken[MAXBUF];
 603         Instance->Config->module_names.clear();
 604         ModCount = -1;
 605
 606         for(int count = 0; count < Instance->Config->ConfValueEnum(Instance->Config->config_data, "module"); count++)
 607         {
 608                 Instance->Config->ConfValue(Instance->Config->config_data, "module", "name", count, configToken, MAXBUF);
 609                 printf_c("[\033[1;32m*\033[0m] Loading module:\t\033[1;32m%s\033[0m\n",configToken);
 610
 611                 if (!this->Load(configToken))
 612                 {
 613                         Instance->Log(DEFAULT,"There was an error loading the module '%s': %s", configToken, this->LastError());
 614                         printf_c("\n[\033[1;31m*\033[0m] There was an error loading the module '%s': %s\n\n", configToken, this->LastError());
 615                         Instance->Exit(EXIT_STATUS_MODULE);
 616                 }
 617         }
 618         printf_c("\nA total of \033[1;32m%d\033[0m module%s been loaded.\n", (this->GetCount()+1), (this->GetCount()+1) == 1 ? " has" : "s have");
 619         Instance->Log(DEFAULT,"Total loaded modules: %d", this->GetCount()+1);
 620 }
 621
 622 bool ModuleManager::PublishFeature(const std::string &FeatureName, Module* Mod)
 623 {
 624         if (Features.find(FeatureName) == Features.end())
 625         {
 626                 Features[FeatureName] = Mod;
 627                 return true;
 628         }
 629         return false;
 630 }
 631
 632 bool ModuleManager::UnpublishFeature(const std::string &FeatureName)
 633 {
 634         featurelist::iterator iter = Features.find(FeatureName);
 635
 636         if (iter == Features.end())
 637                 return false;
 638
 639         Features.erase(iter);
 640         return true;
 641 }
 642
 643 Module* ModuleManager::FindFeature(const std::string &FeatureName)
 644 {
 645         featurelist::iterator iter = Features.find(FeatureName);
 646
 647         if (iter == Features.end())
 648                 return NULL;
 649
 650         return iter->second;
 651 }
 652
 653 bool ModuleManager::PublishInterface(const std::string &InterfaceName, Module* Mod)
 654 {
 655         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 656
 657         if (iter == Interfaces.end())
 658         {
 659                 modulelist ml;
 660                 ml.push_back(Mod);
 661                 Interfaces[InterfaceName] = std::make_pair(0, ml);
 662                 return true;
 663         }
 664         else
 665         {
 666                 iter->second.second.push_back(Mod);
 667                 return true;
 668         }
 669         return false;
 670 }
 671
 672 bool ModuleManager::UnpublishInterface(const std::string &InterfaceName, Module* Mod)
 673 {
 674         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 675
 676         if (iter == Interfaces.end())
 677                 return false;
 678
 679         for (modulelist::iterator x = iter->second.second.begin(); x != iter->second.second.end(); x++)
 680         {
 681                 if (*x == Mod)
 682                 {
 683                         iter->second.second.erase(x);
 684                         if (iter->second.second.empty())
 685                                 Interfaces.erase(InterfaceName);
 686                         return true;
 687                 }
 688         }
 689         return false;
 690 }
 691
 692 modulelist* ModuleManager::FindInterface(const std::string &InterfaceName)
 693 {
 694         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 695         if (iter == Interfaces.end())
 696                 return NULL;
 697         else
 698                 return &(iter->second.second);
 699 }
 700
 701 void ModuleManager::UseInterface(const std::string &InterfaceName)
 702 {
 703         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 704         if (iter != Interfaces.end())
 705                 iter->second.first++;
 706
 707 }
 708
 709 void ModuleManager::DoneWithInterface(const std::string &InterfaceName)
 710 {
 711         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 712         if (iter != Interfaces.end())
 713                 iter->second.first--;
 714 }
 715
 716 std::pair<int,std::string> ModuleManager::GetInterfaceInstanceCount(Module* m)
 717 {
 718         for (interfacelist::iterator iter = Interfaces.begin(); iter != Interfaces.end(); iter++)
 719         {
 720                 for (modulelist::iterator x = iter->second.second.begin(); x != iter->second.second.end(); x++)
 721                 {
 722                         if (*x == m)
 723                         {
 724                                 return std::make_pair(iter->second.first, iter->first);
 725                         }
 726                 }
 727         }
 728         return std::make_pair(0, "");
 729 }
 730
 731 const std::string& ModuleManager::GetModuleName(Module* m)
 732 {
 733         static std::string nothing; /* Prevent compiler warning */
 734
 735         if (!this->GetCount())
 736                 return nothing;
 737
 738         for (int i = 0; i <= this->GetCount(); i++)
 739         {
 740                 if (this->modules[i] == m)
 741                 {
 742                         return Instance->Config->module_names[i];
 743                 }
 744         }
 745         return nothing; /* As above */
 746 }
 747
 748 /* This is ugly, yes, but hash_map's arent designed to be
 749  * addressed in this manner, and this is a bit of a kludge.
 750  * Luckily its a specialist function and rarely used by
 751  * many modules (in fact, it was specially created to make
 752  * m_safelist possible, initially).
 753  */
 754
 755 Channel* InspIRCd::GetChannelIndex(long index)
 756 {
 757         int target = 0;
 758         for (chan_hash::iterator n = this->chanlist->begin(); n != this->chanlist->end(); n++, target++)
 759         {
 760                 if (index == target)
 761                         return n->second;
 762         }
 763         return NULL;
 764 }
 765
 766 bool InspIRCd::MatchText(const std::string &sliteral, const std::string &spattern)
 767 {
 768         return match(sliteral.c_str(),spattern.c_str());
 769 }
 770
 771 CmdResult InspIRCd::CallCommandHandler(const std::string &commandname, const char** parameters, int pcnt, User* user)
 772 {
 773         return this->Parser->CallHandler(commandname,parameters,pcnt,user);
 774 }
 775
 776 bool InspIRCd::IsValidModuleCommand(const std::string &commandname, int pcnt, User* user)
 777 {
 778         return this->Parser->IsValidCommand(commandname, pcnt, user);
 779 }
 780
 781 void InspIRCd::AddCommand(Command *f)
 782 {
 783         if (!this->Parser->CreateCommand(f))
 784         {
 785                 ModuleException err("Command "+std::string(f->command)+" already exists.");
 786                 throw (err);
 787         }
 788 }
 789
 790 void InspIRCd::SendMode(const char** parameters, int pcnt, User *user)
 791 {
 792         this->Modes->Process(parameters,pcnt,user,true);
 793 }
 794
 795 void InspIRCd::DumpText(User* User, const std::string &LinePrefix, stringstream &TextStream)
 796 {
 797         std::string CompleteLine = LinePrefix;
 798         std::string Word;
 799         while (TextStream >> Word)
 800         {
 801                 if (CompleteLine.length() + Word.length() + 3 > 500)
 802                 {
 803                         User->WriteServ(CompleteLine);
 804                         CompleteLine = LinePrefix;
 805                 }
 806                 CompleteLine = CompleteLine + Word + " ";
 807         }
 808         User->WriteServ(CompleteLine);
 809 }
 810
 811 User* FindDescriptorHandler::Call(int socket)
 812 {
 813         return reinterpret_cast<User*>(Server->SE->GetRef(socket));
 814 }
 815
 816 bool InspIRCd::AddMode(ModeHandler* mh)
 817 {
 818         return this->Modes->AddMode(mh);
 819 }
 820
 821 bool InspIRCd::AddModeWatcher(ModeWatcher* mw)
 822 {
 823         return this->Modes->AddModeWatcher(mw);
 824 }
 825
 826 bool InspIRCd::DelModeWatcher(ModeWatcher* mw)
 827 {
 828         return this->Modes->DelModeWatcher(mw);
 829 }
 830
 831 bool InspIRCd::AddResolver(Resolver* r, bool cached)
 832 {
 833         if (!cached)
 834                 return this->Res->AddResolverClass(r);
 835         else
 836         {
 837                 r->TriggerCachedResult();
 838                 delete r;
 839                 return true;
 840         }
 841 }
 842
 843 Module* ModuleManager::Find(const std::string &name)
 844 {
 845         for (int i = 0; i <= this->GetCount(); i++)
 846         {
 847                 if (Instance->Config->module_names[i] == name)
 848                 {
 849                         return this->modules[i];
 850                 }
 851         }
 852         return NULL;
 853 }
 854
 855 ConfigReader::ConfigReader(InspIRCd* Instance) : ServerInstance(Instance)
 856 {
 857         /* Is there any reason to load the entire config file again here?
 858          * it's needed if they specify another config file, but using the
 859          * default one we can just use the global config data - pre-parsed!
 860          */
 861         this->errorlog = new std::ostringstream(std::stringstream::in | std::stringstream::out);
 862         this->error = CONF_NO_ERROR;
 863         this->data = &ServerInstance->Config->config_data;
 864         this->privatehash = false;
 865 }
 866
 867
 868 ConfigReader::~ConfigReader()
 869 {
 870         if (this->errorlog)
 871                 delete this->errorlog;
 872         if(this->privatehash)
 873                 delete this->data;
 874 }
 875
 876
 877 ConfigReader::ConfigReader(InspIRCd* Instance, const std::string &filename) : ServerInstance(Instance)
 878 {
 879         ServerInstance->Config->ClearStack();
 880
 881         this->error = CONF_NO_ERROR;
 882         this->data = new ConfigDataHash;
 883         this->privatehash = true;
 884         this->errorlog = new std::ostringstream(std::stringstream::in | std::stringstream::out);
 885         this->readerror = ServerInstance->Config->LoadConf(*this->data, filename, *this->errorlog);
 886         if (!this->readerror)
 887                 this->error = CONF_FILE_NOT_FOUND;
 888 }
 889
 890
 891 std::string ConfigReader::ReadValue(const std::string &tag, const std::string &name, const std::string &default_value, int index, bool allow_linefeeds)
 892 {
 893         /* Don't need to strlcpy() tag and name anymore, ReadConf() takes const char* */
 894         std::string result;
 895
 896         if (!ServerInstance->Config->ConfValue(*this->data, tag, name, default_value, index, result, allow_linefeeds))
 897         {
 898                 this->error = CONF_VALUE_NOT_FOUND;
 899         }
 900         return result;
 901 }
 902
 903 std::string ConfigReader::ReadValue(const std::string &tag, const std::string &name, int index, bool allow_linefeeds)
 904 {
 905         return ReadValue(tag, name, "", index, allow_linefeeds);
 906 }
 907
 908 bool ConfigReader::ReadFlag(const std::string &tag, const std::string &name, const std::string &default_value, int index)
 909 {
 910         return ServerInstance->Config->ConfValueBool(*this->data, tag, name, default_value, index);
 911 }
 912
 913 bool ConfigReader::ReadFlag(const std::string &tag, const std::string &name, int index)
 914 {
 915         return ReadFlag(tag, name, "", index);
 916 }
 917
 918
 919 int ConfigReader::ReadInteger(const std::string &tag, const std::string &name, const std::string &default_value, int index, bool need_positive)
 920 {
 921         int result;
 922
 923         if(!ServerInstance->Config->ConfValueInteger(*this->data, tag, name, default_value, index, result))
 924         {
 925                 this->error = CONF_VALUE_NOT_FOUND;
 926                 return 0;
 927         }
 928
 929         if ((need_positive) && (result < 0))
 930         {
 931                 this->error = CONF_INT_NEGATIVE;
 932                 return 0;
 933         }
 934
 935         return result;
 936 }
 937
 938 int ConfigReader::ReadInteger(const std::string &tag, const std::string &name, int index, bool need_positive)
 939 {
 940         return ReadInteger(tag, name, "", index, need_positive);
 941 }
 942
 943 long ConfigReader::GetError()
 944 {
 945         long olderr = this->error;
 946         this->error = 0;
 947         return olderr;
 948 }
 949
 950 void ConfigReader::DumpErrors(bool bail, User* user)
 951 {
 952         ServerInstance->Config->ReportConfigError(this->errorlog->str(), bail, user);
 953 }
 954
 955
 956 int ConfigReader::Enumerate(const std::string &tag)
 957 {
 958         return ServerInstance->Config->ConfValueEnum(*this->data, tag);
 959 }
 960
 961 int ConfigReader::EnumerateValues(const std::string &tag, int index)
 962 {
 963         return ServerInstance->Config->ConfVarEnum(*this->data, tag, index);
 964 }
 965
 966 bool ConfigReader::Verify()
 967 {
 968         return this->readerror;
 969 }
 970
 971
 972 FileReader::FileReader(InspIRCd* Instance, const std::string &filename) : ServerInstance(Instance)
 973 {
 974         LoadFile(filename);
 975 }
 976
 977 FileReader::FileReader(InspIRCd* Instance) : ServerInstance(Instance)
 978 {
 979 }
 980
 981 std::string FileReader::Contents()
 982 {
 983         std::string x;
 984         for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
 985         {
 986                 x.append(*a);
 987                 x.append("\r\n");
 988         }
 989         return x;
 990 }
 991
 992 unsigned long FileReader::ContentSize()
 993 {
 994         return this->contentsize;
 995 }
 996
 997 void FileReader::CalcSize()
 998 {
 999         unsigned long n = 0;
1000         for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
1001                 n += (a->length() + 2);
1002         this->contentsize = n;
1003 }
1004
1005 void FileReader::LoadFile(const std::string &filename)
1006 {
1007         file_cache c;
1008         c.clear();
1009         if (ServerInstance->Config->ReadFile(c,filename.c_str()))
1010         {
1011                 this->fc = c;
1012                 this->CalcSize();
1013         }
1014 }
1015
1016
1017 FileReader::~FileReader()
1018 {
1019 }
1020
1021 bool FileReader::Exists()
1022 {
1023         return (!(fc.size() == 0));
1024 }
1025
1026 std::string FileReader::GetLine(int x)
1027 {
1028         if ((x<0) || ((unsigned)x>fc.size()))
1029                 return "";
1030         return fc[x];
1031 }
1032
1033 int FileReader::FileSize()
1034 {
1035         return fc.size();
1036 }