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::OnReadConfig(ServerConfig*, ConfigReader*) { }
 110 int             Module::OnDownloadFile(const std::string&, std::stringstream&) { return 0; }
 111 void            Module::OnUserConnect(User*) { }
 112 void            Module::OnUserQuit(User*, const std::string&, const std::string&) { }
 113 void            Module::OnUserDisconnect(User*) { }
 114 void            Module::OnUserJoin(User*, Channel*, bool, bool&) { }
 115 void            Module::OnPostJoin(User*, Channel*) { }
 116 void            Module::OnUserPart(User*, Channel*, const std::string&, bool&) { }
 117 void            Module::OnRehash(User*, const std::string&) { }
 118 void            Module::OnServerRaw(std::string&, bool, User*) { }
 119 int             Module::OnUserPreJoin(User*, Channel*, const char*, std::string&) { return 0; }
 120 void            Module::OnMode(User*, void*, int, const std::string&) { }
 121 Version         Module::GetVersion() { return Version(1,0,0,0,VF_VENDOR,-1); }
 122 void            Module::OnOper(User*, const std::string&) { }
 123 void            Module::OnPostOper(User*, const std::string&) { }
 124 void            Module::OnInfo(User*) { }
 125 void            Module::OnWhois(User*, User*) { }
 126 int             Module::OnUserPreInvite(User*, User*, Channel*) { return 0; }
 127 int             Module::OnUserPreMessage(User*, void*, int, std::string&, char, CUList&) { return 0; }
 128 int             Module::OnUserPreNotice(User*, void*, int, std::string&, char, CUList&) { return 0; }
 129 int             Module::OnUserPreNick(User*, const std::string&) { return 0; }
 130 void            Module::OnUserPostNick(User*, const std::string&) { }
 131 int             Module::OnAccessCheck(User*, User*, Channel*, int) { return ACR_DEFAULT; }
 132 void            Module::On005Numeric(std::string&) { }
 133 int             Module::OnKill(User*, User*, const std::string&) { return 0; }
 134 void            Module::OnLoadModule(Module*, const std::string&) { }
 135 void            Module::OnUnloadModule(Module*, const std::string&) { }
 136 void            Module::OnBackgroundTimer(time_t) { }
 137 int             Module::OnPreCommand(const std::string&, const char**, int, User *, bool, const std::string&) { return 0; }
 138 void            Module::OnPostCommand(const std::string&, const char**, int, User *, CmdResult, const std::string&) { }
 139 bool            Module::OnCheckReady(User*) { return true; }
 140 int             Module::OnUserRegister(User*) { return 0; }
 141 int             Module::OnUserPreKick(User*, User*, Channel*, const std::string&) { return 0; }
 142 void            Module::OnUserKick(User*, User*, Channel*, const std::string&, bool&) { }
 143 int             Module::OnCheckInvite(User*, Channel*) { return 0; }
 144 int             Module::OnCheckKey(User*, Channel*, const std::string&) { return 0; }
 145 int             Module::OnCheckLimit(User*, Channel*) { return 0; }
 146 int             Module::OnCheckBan(User*, Channel*) { return 0; }
 147 int             Module::OnStats(char, User*, string_list&) { return 0; }
 148 int             Module::OnChangeLocalUserHost(User*, const std::string&) { return 0; }
 149 int             Module::OnChangeLocalUserGECOS(User*, const std::string&) { return 0; }
 150 int             Module::OnLocalTopicChange(User*, Channel*, const std::string&) { return 0; }
 151 void            Module::OnEvent(Event*) { return; }
 152 char*           Module::OnRequest(Request*) { return NULL; }
 153 int             Module::OnOperCompare(const std::string&, const std::string&, int) { return 0; }
 154 void            Module::OnGlobalOper(User*) { }
 155 void            Module::OnPostConnect(User*) { }
 156 int             Module::OnAddBan(User*, Channel*, const std::string &) { return 0; }
 157 int             Module::OnDelBan(User*, Channel*, const std::string &) { return 0; }
 158 void            Module::OnRawSocketAccept(int, const std::string&, int) { }
 159 int             Module::OnRawSocketWrite(int, const char*, int) { return 0; }
 160 void            Module::OnRawSocketClose(int) { }
 161 void            Module::OnRawSocketConnect(int) { }
 162 int             Module::OnRawSocketRead(int, char*, unsigned int, int&) { return 0; }
 163 void            Module::OnUserMessage(User*, void*, int, const std::string&, char, const CUList&) { }
 164 void            Module::OnUserNotice(User*, void*, int, const std::string&, char, const CUList&) { }
 165 void            Module::OnRemoteKill(User*, User*, const std::string&, const std::string&) { }
 166 void            Module::OnUserInvite(User*, User*, Channel*) { }
 167 void            Module::OnPostLocalTopicChange(User*, Channel*, const std::string&) { }
 168 void            Module::OnGetServerDescription(const std::string&, std::string&) { }
 169 void            Module::OnSyncUser(User*, Module*, void*) { }
 170 void            Module::OnSyncChannel(Channel*, Module*, void*) { }
 171 void            Module::ProtoSendMode(void*, int, void*, const std::string&) { }
 172 void            Module::OnSyncChannelMetaData(Channel*, Module*, void*, const std::string&, bool) { }
 173 void            Module::OnSyncUserMetaData(User*, Module*, void*, const std::string&, bool) { }
 174 void            Module::OnSyncOtherMetaData(Module*, void*, bool) { }
 175 void            Module::OnDecodeMetaData(int, void*, const std::string&, const std::string&) { }
 176 void            Module::ProtoSendMetaData(void*, int, void*, const std::string&, const std::string&) { }
 177 void            Module::OnWallops(User*, const std::string&) { }
 178 void            Module::OnChangeHost(User*, const std::string&) { }
 179 void            Module::OnChangeName(User*, const std::string&) { }
 180 void            Module::OnAddLine(User*, XLine*) { }
 181 void            Module::OnDelLine(User*, XLine*) { }
 182 void            Module::OnCleanup(int, void*) { }
 183 void            Module::OnChannelDelete(Channel*) { }
 184 void            Module::OnSetAway(User*) { }
 185 void            Module::OnCancelAway(User*) { }
 186 int             Module::OnUserList(User*, Channel*, CUList*&) { return 0; }
 187 int             Module::OnWhoisLine(User*, User*, int&, std::string&) { return 0; }
 188 void            Module::OnBuildExemptList(MessageType, Channel*, User*, char, CUList&, const std::string&) { }
 189 void            Module::OnGarbageCollect() { }
 190 void            Module::OnBufferFlushed(User*) { }
 191 void            Module::OnText(User*, void*, int, const std::string&, char, CUList&) { }
 192
 193
 194 ModuleManager::ModuleManager(InspIRCd* Ins) : ModCount(0), Instance(Ins)
 195 {
 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         /** To change the priority of a module, we first find its position in the vector,
 245          * then we find the position of the other modules in the vector that this module
 246          * wants to be before/after. We pick off either the first or last of these depending
 247          * on which they want, and we make sure our module is *at least* before or after
 248          * the first or last of this subset, depending again on the type of priority.
 249          */
 250         size_t swap_pos;
 251         size_t source = 0;
 252         bool swap = true;
 253         bool found = false;
 254
 255         /* Locate our module. This is O(n) but it only occurs on module load so we're
 256          * not too bothered about it
 257          */
 258         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
 259         {
 260                 if (EventHandlers[i][x] == mod)
 261                 {
 262                         source = x;
 263                         found = true;
 264                         break;
 265                 }
 266         }
 267
 268         /* Eh? this module doesnt exist, probably trying to set priority on an event
 269          * theyre not attached to.
 270          */
 271         if (!found)
 272                 return false;
 273
 274         switch (s)
 275         {
 276                 /* Dummy value */
 277                 case PRIO_DONTCARE:
 278                         swap = false;
 279                 break;
 280                 /* Module wants to be first, sod everything else */
 281                 case PRIO_FIRST:
 282                         swap_pos = 0;
 283                 break;
 284                 /* Module is submissive and wants to be last... awww. */
 285                 case PRIO_LAST:
 286                         if (EventHandlers[i].empty())
 287                                 swap_pos = 0;
 288                         else
 289                                 swap_pos = EventHandlers[i].size() - 1;
 290                 break;
 291                 /* Place this module after a set of other modules */
 292                 case PRIO_AFTER:
 293                 {
 294                         /* Find the latest possible position */
 295                         swap_pos = 0;
 296                         swap = false;
 297                         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
 298                         {
 299                                 for (size_t n = 0; n < sz; ++n)
 300                                 {
 301                                         if ((modules[n]) && (EventHandlers[i][x] == modules[n]) && (x >= swap_pos) && (source <= swap_pos))
 302                                         {
 303                                                 swap_pos = x;
 304                                                 swap = true;
 305                                         }
 306                                 }
 307                         }
 308                 }
 309                 break;
 310                 /* Place this module before a set of other modules */
 311                 case PRIO_BEFORE:
 312                 {
 313                         swap_pos = EventHandlers[i].size() - 1;
 314                         swap = false;
 315                         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
 316                         {
 317                                 for (size_t n = 0; n < sz; ++n)
 318                                 {
 319                                         if ((modules[n]) && (EventHandlers[i][x] == modules[n]) && (x <= swap_pos) && (source >= swap_pos))
 320                                         {
 321                                                 swap = true;
 322                                                 swap_pos = x;
 323                                         }
 324                                 }
 325                         }
 326                 }
 327                 break;
 328         }
 329
 330         /* Do we need to swap? */
 331         if (swap && (swap_pos != source))
 332                 std::swap(EventHandlers[i][swap_pos], EventHandlers[i][source]);
 333
 334         return true;
 335 }
 336
 337 std::string& ModuleManager::LastError()
 338 {
 339         return LastModuleError;
 340 }
 341
 342 bool ModuleManager::Load(const char* filename)
 343 {
 344         /* Do we have a glob pattern in the filename?
 345          * The user wants to load multiple modules which
 346          * match the pattern.
 347          */
 348         if (strchr(filename,'*') || (strchr(filename,'?')))
 349         {
 350                 int n_match = 0;
 351                 DIR* library = opendir(Instance->Config->ModPath);
 352                 if (library)
 353                 {
 354                         /* Try and locate and load all modules matching the pattern */
 355                         dirent* entry = NULL;
 356                         while ((entry = readdir(library)))
 357                         {
 358                                 if (Instance->MatchText(entry->d_name, filename))
 359                                 {
 360                                         if (!this->Load(entry->d_name))
 361                                                 n_match++;
 362                                 }
 363                         }
 364                         closedir(library);
 365                 }
 366                 /* Loadmodule will now return false if any one of the modules failed
 367                  * to load (but wont abort when it encounters a bad one) and when 1 or
 368                  * more modules were actually loaded.
 369                  */
 370                 return (n_match > 0);
 371         }
 372
 373         char modfile[MAXBUF];
 374         snprintf(modfile,MAXBUF,"%s/%s",Instance->Config->ModPath,filename);
 375         std::string filename_str = filename;
 376
 377         if (!ServerConfig::DirValid(modfile))
 378         {
 379                 LastModuleError = "Module " + filename_str + " is not within the modules directory.";
 380                 Instance->Log(DEFAULT, LastModuleError);
 381                 return false;
 382         }
 383
 384         if (!ServerConfig::FileExists(modfile))
 385         {
 386                 LastModuleError = "Module file could not be found: " + filename_str;
 387                 Instance->Log(DEFAULT, LastModuleError);
 388                 return false;
 389         }
 390
 391         if (Modules.find(filename_str) != Modules.end())
 392         {
 393                 LastModuleError = "Module " + filename_str + " is already loaded, cannot load a module twice!";
 394                 Instance->Log(DEFAULT, LastModuleError);
 395                 return false;
 396         }
 397
 398         Module* newmod = NULL;
 399         ircd_module* newhandle = NULL;
 400
 401         try
 402         {
 403                 /* This will throw a CoreException if there's a problem loading
 404                  * the module file or getting a pointer to the init_module symbol.
 405                  */
 406                 newhandle = new ircd_module(Instance, modfile, "init_module");
 407                 newmod = newhandle->CallInit();
 408
 409                 if (newmod)
 410                 {
 411                         Version v = newmod->GetVersion();
 412
 413                         if (v.API != API_VERSION)
 414                         {
 415                                 DetachAll(newmod);
 416                                 delete newmod;
 417                                 delete newhandle;
 418                                 LastModuleError = "Unable to load " + filename_str + ": Incorrect module API version: " + ConvToStr(v.API) + " (our version: " + ConvToStr(API_VERSION) + ")";
 419                                 Instance->Log(DEFAULT, LastModuleError);
 420                                 return false;
 421                         }
 422                         else
 423                         {
 424                                 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]"));
 425                         }
 426
 427                         Modules[filename_str] = std::make_pair(newhandle, newmod);
 428                 }
 429                 else
 430                 {
 431                         delete newhandle;
 432                         LastModuleError = "Unable to load " + filename_str + ": Probably missing init_module() entrypoint, but dlsym() didn't notice a problem";
 433                         Instance->Log(DEFAULT, LastModuleError);
 434                         return false;
 435                 }
 436         }
 437         /** XXX: Is there anything we can do about this mess? -- Brain */
 438         catch (LoadModuleException& modexcept)
 439         {
 440                 DetachAll(newmod);
 441                 if (newmod)
 442                         delete newmod;
 443                 if (newhandle)
 444                         delete newhandle;
 445                 LastModuleError = "Unable to load " + filename_str + ": Error when loading: " + modexcept.GetReason();
 446                 Instance->Log(DEFAULT, LastModuleError);
 447                 return false;
 448         }
 449         catch (FindSymbolException& modexcept)
 450         {
 451                 DetachAll(newmod);
 452                 if (newmod)
 453                         delete newmod;
 454                 if (newhandle)
 455                         delete newhandle;
 456                 LastModuleError = "Unable to load " + filename_str + ": Error finding symbol: " + modexcept.GetReason();
 457                 Instance->Log(DEFAULT, LastModuleError);
 458                 return false;
 459         }
 460         catch (CoreException& modexcept)
 461         {
 462                 DetachAll(newmod);
 463                 if (newmod)
 464                         delete newmod;
 465                 if (newhandle)
 466                         delete newhandle;
 467                 LastModuleError = "Unable to load " + filename_str + ": " + modexcept.GetReason();
 468                 Instance->Log(DEFAULT, LastModuleError);
 469                 return false;
 470         }
 471
 472         this->ModCount++;
 473         FOREACH_MOD_I(Instance,I_OnLoadModule,OnLoadModule(newmod, filename_str));
 474
 475         /* We give every module a chance to re-prioritize when we introduce a new one,
 476          * not just the one thats loading, as the new module could affect the preference
 477          * of others
 478          */
 479         for (std::map<std::string, std::pair<ircd_module*, Module*> >::iterator n = Modules.begin(); n != Modules.end(); ++n)
 480                 n->second.second->Prioritize();
 481
 482         Instance->BuildISupport();
 483         return true;
 484 }
 485
 486 bool ModuleManager::Unload(const char* filename)
 487 {
 488         std::string filename_str(filename);
 489         std::map<std::string, std::pair<ircd_module*, Module*> >::iterator modfind = Modules.find(filename);
 490
 491         if (modfind != Modules.end())
 492         {
 493                 if (modfind->second.second->GetVersion().Flags & VF_STATIC)
 494                 {
 495                         LastModuleError = "Module " + filename_str + " not unloadable (marked static)";
 496                         Instance->Log(DEFAULT, LastModuleError);
 497                         return false;
 498                 }
 499                 std::pair<int,std::string> intercount = GetInterfaceInstanceCount(modfind->second.second);
 500                 if (intercount.first > 0)
 501                 {
 502                         LastModuleError = "Failed to unload module " + filename_str + ", being used by " + ConvToStr(intercount.first) + " other(s) via interface '" + intercount.second + "'";
 503                         Instance->Log(DEFAULT, LastModuleError);
 504                         return false;
 505                 }
 506
 507                 /* Give the module a chance to tidy out all its metadata */
 508                 for (chan_hash::iterator c = Instance->chanlist->begin(); c != Instance->chanlist->end(); c++)
 509                 {
 510                         modfind->second.second->OnCleanup(TYPE_CHANNEL,c->second);
 511                 }
 512                 for (user_hash::iterator u = Instance->clientlist->begin(); u != Instance->clientlist->end(); u++)
 513                 {
 514                         modfind->second.second->OnCleanup(TYPE_USER,u->second);
 515                 }
 516
 517                 /* Tidy up any dangling resolvers */
 518                 Instance->Res->CleanResolvers(modfind->second.second);
 519
 520
 521                 FOREACH_MOD_I(Instance,I_OnUnloadModule,OnUnloadModule(modfind->second.second, modfind->first));
 522
 523                 this->DetachAll(modfind->second.second);
 524
 525                 Instance->Parser->RemoveCommands(filename);
 526
 527                 delete modfind->second.second;
 528                 delete modfind->second.first;
 529                 Modules.erase(modfind);
 530
 531                 Instance->Log(DEFAULT,"Module %s unloaded",filename);
 532                 this->ModCount--;
 533                 Instance->BuildISupport();
 534                 return true;
 535         }
 536
 537         LastModuleError = "Module " + filename_str + " is not loaded, cannot unload it!";
 538         Instance->Log(DEFAULT, LastModuleError);
 539         return false;
 540 }
 541
 542 /* We must load the modules AFTER initializing the socket engine, now */
 543 void ModuleManager::LoadAll()
 544 {
 545         char configToken[MAXBUF];
 546         ModCount = -1;
 547
 548         for(int count = 0; count < Instance->Config->ConfValueEnum(Instance->Config->config_data, "module"); count++)
 549         {
 550                 Instance->Config->ConfValue(Instance->Config->config_data, "module", "name", count, configToken, MAXBUF);
 551                 printf_c("[\033[1;32m*\033[0m] Loading module:\t\033[1;32m%s\033[0m\n",configToken);
 552
 553                 if (!this->Load(configToken))
 554                 {
 555                         Instance->Log(DEFAULT, this->LastError());
 556                         printf_c("\n[\033[1;31m*\033[0m] %s\n\n", this->LastError().c_str());
 557                         Instance->Exit(EXIT_STATUS_MODULE);
 558                 }
 559         }
 560         printf_c("\nA total of \033[1;32m%d\033[0m module%s been loaded.\n", (this->GetCount()), (this->GetCount()) == 1 ? " has" : "s have");
 561         Instance->Log(DEFAULT,"Total loaded modules: %d", this->GetCount());
 562 }
 563
 564 bool ModuleManager::PublishFeature(const std::string &FeatureName, Module* Mod)
 565 {
 566         if (Features.find(FeatureName) == Features.end())
 567         {
 568                 Features[FeatureName] = Mod;
 569                 return true;
 570         }
 571         return false;
 572 }
 573
 574 bool ModuleManager::UnpublishFeature(const std::string &FeatureName)
 575 {
 576         featurelist::iterator iter = Features.find(FeatureName);
 577
 578         if (iter == Features.end())
 579                 return false;
 580
 581         Features.erase(iter);
 582         return true;
 583 }
 584
 585 Module* ModuleManager::FindFeature(const std::string &FeatureName)
 586 {
 587         featurelist::iterator iter = Features.find(FeatureName);
 588
 589         if (iter == Features.end())
 590                 return NULL;
 591
 592         return iter->second;
 593 }
 594
 595 bool ModuleManager::PublishInterface(const std::string &InterfaceName, Module* Mod)
 596 {
 597         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 598
 599         if (iter == Interfaces.end())
 600         {
 601                 modulelist ml;
 602                 ml.push_back(Mod);
 603                 Interfaces[InterfaceName] = std::make_pair(0, ml);
 604                 return true;
 605         }
 606         else
 607         {
 608                 iter->second.second.push_back(Mod);
 609                 return true;
 610         }
 611         return false;
 612 }
 613
 614 bool ModuleManager::UnpublishInterface(const std::string &InterfaceName, Module* Mod)
 615 {
 616         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 617
 618         if (iter == Interfaces.end())
 619                 return false;
 620
 621         for (modulelist::iterator x = iter->second.second.begin(); x != iter->second.second.end(); x++)
 622         {
 623                 if (*x == Mod)
 624                 {
 625                         iter->second.second.erase(x);
 626                         if (iter->second.second.empty())
 627                                 Interfaces.erase(InterfaceName);
 628                         return true;
 629                 }
 630         }
 631         return false;
 632 }
 633
 634 modulelist* ModuleManager::FindInterface(const std::string &InterfaceName)
 635 {
 636         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 637         if (iter == Interfaces.end())
 638                 return NULL;
 639         else
 640                 return &(iter->second.second);
 641 }
 642
 643 void ModuleManager::UseInterface(const std::string &InterfaceName)
 644 {
 645         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 646         if (iter != Interfaces.end())
 647                 iter->second.first++;
 648
 649 }
 650
 651 void ModuleManager::DoneWithInterface(const std::string &InterfaceName)
 652 {
 653         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 654         if (iter != Interfaces.end())
 655                 iter->second.first--;
 656 }
 657
 658 std::pair<int,std::string> ModuleManager::GetInterfaceInstanceCount(Module* m)
 659 {
 660         for (interfacelist::iterator iter = Interfaces.begin(); iter != Interfaces.end(); iter++)
 661         {
 662                 for (modulelist::iterator x = iter->second.second.begin(); x != iter->second.second.end(); x++)
 663                 {
 664                         if (*x == m)
 665                         {
 666                                 return std::make_pair(iter->second.first, iter->first);
 667                         }
 668                 }
 669         }
 670         return std::make_pair(0, "");
 671 }
 672
 673 const std::string& ModuleManager::GetModuleName(Module* m)
 674 {
 675         static std::string nothing;
 676
 677         for (std::map<std::string, std::pair<ircd_module*, Module*> >::iterator n = Modules.begin(); n != Modules.end(); ++n)
 678         {
 679                 if (n->second.second == m)
 680                         return n->first;
 681         }
 682
 683         return nothing;
 684 }
 685
 686 /* This is ugly, yes, but hash_map's arent designed to be
 687  * addressed in this manner, and this is a bit of a kludge.
 688  * Luckily its a specialist function and rarely used by
 689  * many modules (in fact, it was specially created to make
 690  * m_safelist possible, initially).
 691  */
 692
 693 Channel* InspIRCd::GetChannelIndex(long index)
 694 {
 695         int target = 0;
 696         for (chan_hash::iterator n = this->chanlist->begin(); n != this->chanlist->end(); n++, target++)
 697         {
 698                 if (index == target)
 699                         return n->second;
 700         }
 701         return NULL;
 702 }
 703
 704 bool InspIRCd::MatchText(const std::string &sliteral, const std::string &spattern)
 705 {
 706         return match(sliteral.c_str(),spattern.c_str());
 707 }
 708
 709 CmdResult InspIRCd::CallCommandHandler(const std::string &commandname, const char** parameters, int pcnt, User* user)
 710 {
 711         return this->Parser->CallHandler(commandname,parameters,pcnt,user);
 712 }
 713
 714 bool InspIRCd::IsValidModuleCommand(const std::string &commandname, int pcnt, User* user)
 715 {
 716         return this->Parser->IsValidCommand(commandname, pcnt, user);
 717 }
 718
 719 void InspIRCd::AddCommand(Command *f)
 720 {
 721         if (!this->Parser->CreateCommand(f))
 722         {
 723                 ModuleException err("Command "+std::string(f->command)+" already exists.");
 724                 throw (err);
 725         }
 726 }
 727
 728 void InspIRCd::SendMode(const char** parameters, int pcnt, User *user)
 729 {
 730         this->Modes->Process(parameters,pcnt,user,true);
 731 }
 732
 733 void InspIRCd::DumpText(User* User, const std::string &LinePrefix, stringstream &TextStream)
 734 {
 735         std::string CompleteLine = LinePrefix;
 736         std::string Word;
 737         while (TextStream >> Word)
 738         {
 739                 if (CompleteLine.length() + Word.length() + 3 > 500)
 740                 {
 741                         User->WriteServ(CompleteLine);
 742                         CompleteLine = LinePrefix;
 743                 }
 744                 CompleteLine = CompleteLine + Word + " ";
 745         }
 746         User->WriteServ(CompleteLine);
 747 }
 748
 749 User* FindDescriptorHandler::Call(int socket)
 750 {
 751         return reinterpret_cast<User*>(Server->SE->GetRef(socket));
 752 }
 753
 754 bool InspIRCd::AddMode(ModeHandler* mh)
 755 {
 756         return this->Modes->AddMode(mh);
 757 }
 758
 759 bool InspIRCd::AddModeWatcher(ModeWatcher* mw)
 760 {
 761         return this->Modes->AddModeWatcher(mw);
 762 }
 763
 764 bool InspIRCd::DelModeWatcher(ModeWatcher* mw)
 765 {
 766         return this->Modes->DelModeWatcher(mw);
 767 }
 768
 769 bool InspIRCd::AddResolver(Resolver* r, bool cached)
 770 {
 771         if (!cached)
 772                 return this->Res->AddResolverClass(r);
 773         else
 774         {
 775                 r->TriggerCachedResult();
 776                 delete r;
 777                 return true;
 778         }
 779 }
 780
 781 Module* ModuleManager::Find(const std::string &name)
 782 {
 783         std::map<std::string, std::pair<ircd_module*, Module*> >::iterator modfind = Modules.find(name);
 784
 785         if (modfind == Modules.end())
 786                 return NULL;
 787         else
 788                 return modfind->second.second;
 789 }
 790
 791 const std::vector<std::string> ModuleManager::GetAllModuleNames(int filter)
 792 {
 793         std::vector<std::string> retval;
 794         for (std::map<std::string, std::pair<ircd_module*, Module*> >::iterator x = Modules.begin(); x != Modules.end(); ++x)
 795                 if (!filter || (x->second.second->GetVersion().Flags & filter))
 796                         retval.push_back(x->first);
 797         return retval;
 798 }
 799
 800 ConfigReader::ConfigReader(InspIRCd* Instance) : ServerInstance(Instance)
 801 {
 802         /* Is there any reason to load the entire config file again here?
 803          * it's needed if they specify another config file, but using the
 804          * default one we can just use the global config data - pre-parsed!
 805          */
 806         this->errorlog = new std::ostringstream(std::stringstream::in | std::stringstream::out);
 807         this->error = CONF_NO_ERROR;
 808         this->data = &ServerInstance->Config->config_data;
 809         this->privatehash = false;
 810 }
 811
 812
 813 ConfigReader::~ConfigReader()
 814 {
 815         if (this->errorlog)
 816                 delete this->errorlog;
 817         if(this->privatehash)
 818                 delete this->data;
 819 }
 820
 821
 822 ConfigReader::ConfigReader(InspIRCd* Instance, const std::string &filename) : ServerInstance(Instance)
 823 {
 824         ServerInstance->Config->ClearStack();
 825
 826         this->error = CONF_NO_ERROR;
 827         this->data = new ConfigDataHash;
 828         this->privatehash = true;
 829         this->errorlog = new std::ostringstream(std::stringstream::in | std::stringstream::out);
 830         for (int pass = 0; pass < 2; pass++)
 831         {
 832                 /*** XXX: Can return a 'not ready yet!' code! */
 833                 this->readerror = ServerInstance->Config->LoadConf(*this->data, filename, *this->errorlog, pass);
 834         }
 835         if (!this->readerror)
 836                 this->error = CONF_FILE_NOT_FOUND;
 837 }
 838
 839
 840 std::string ConfigReader::ReadValue(const std::string &tag, const std::string &name, const std::string &default_value, int index, bool allow_linefeeds)
 841 {
 842         /* Don't need to strlcpy() tag and name anymore, ReadConf() takes const char* */
 843         std::string result;
 844
 845         if (!ServerInstance->Config->ConfValue(*this->data, tag, name, default_value, index, result, allow_linefeeds))
 846         {
 847                 this->error = CONF_VALUE_NOT_FOUND;
 848         }
 849         return result;
 850 }
 851
 852 std::string ConfigReader::ReadValue(const std::string &tag, const std::string &name, int index, bool allow_linefeeds)
 853 {
 854         return ReadValue(tag, name, "", index, allow_linefeeds);
 855 }
 856
 857 bool ConfigReader::ReadFlag(const std::string &tag, const std::string &name, const std::string &default_value, int index)
 858 {
 859         return ServerInstance->Config->ConfValueBool(*this->data, tag, name, default_value, index);
 860 }
 861
 862 bool ConfigReader::ReadFlag(const std::string &tag, const std::string &name, int index)
 863 {
 864         return ReadFlag(tag, name, "", index);
 865 }
 866
 867
 868 int ConfigReader::ReadInteger(const std::string &tag, const std::string &name, const std::string &default_value, int index, bool need_positive)
 869 {
 870         int result;
 871
 872         if(!ServerInstance->Config->ConfValueInteger(*this->data, tag, name, default_value, index, result))
 873         {
 874                 this->error = CONF_VALUE_NOT_FOUND;
 875                 return 0;
 876         }
 877
 878         if ((need_positive) && (result < 0))
 879         {
 880                 this->error = CONF_INT_NEGATIVE;
 881                 return 0;
 882         }
 883
 884         return result;
 885 }
 886
 887 int ConfigReader::ReadInteger(const std::string &tag, const std::string &name, int index, bool need_positive)
 888 {
 889         return ReadInteger(tag, name, "", index, need_positive);
 890 }
 891
 892 long ConfigReader::GetError()
 893 {
 894         long olderr = this->error;
 895         this->error = 0;
 896         return olderr;
 897 }
 898
 899 void ConfigReader::DumpErrors(bool bail, User* user)
 900 {
 901         ServerInstance->Config->ReportConfigError(this->errorlog->str(), bail, user);
 902 }
 903
 904
 905 int ConfigReader::Enumerate(const std::string &tag)
 906 {
 907         return ServerInstance->Config->ConfValueEnum(*this->data, tag);
 908 }
 909
 910 int ConfigReader::EnumerateValues(const std::string &tag, int index)
 911 {
 912         return ServerInstance->Config->ConfVarEnum(*this->data, tag, index);
 913 }
 914
 915 bool ConfigReader::Verify()
 916 {
 917         return this->readerror;
 918 }
 919
 920
 921 FileReader::FileReader(InspIRCd* Instance, const std::string &filename) : ServerInstance(Instance)
 922 {
 923         LoadFile(filename);
 924 }
 925
 926 FileReader::FileReader(InspIRCd* Instance) : ServerInstance(Instance)
 927 {
 928 }
 929
 930 std::string FileReader::Contents()
 931 {
 932         std::string x;
 933         for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
 934         {
 935                 x.append(*a);
 936                 x.append("\r\n");
 937         }
 938         return x;
 939 }
 940
 941 unsigned long FileReader::ContentSize()
 942 {
 943         return this->contentsize;
 944 }
 945
 946 void FileReader::CalcSize()
 947 {
 948         unsigned long n = 0;
 949         for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
 950                 n += (a->length() + 2);
 951         this->contentsize = n;
 952 }
 953
 954 void FileReader::LoadFile(const std::string &filename)
 955 {
 956         file_cache c;
 957         c.clear();
 958         if (ServerInstance->Config->ReadFile(c,filename.c_str()))
 959         {
 960                 this->fc = c;
 961                 this->CalcSize();
 962         }
 963 }
 964
 965
 966 FileReader::~FileReader()
 967 {
 968 }
 969
 970 bool FileReader::Exists()
 971 {
 972         return (!(fc.size() == 0));
 973 }
 974
 975 std::string FileReader::GetLine(int x)
 976 {
 977         if ((x<0) || ((unsigned)x>fc.size()))
 978                 return "";
 979         return fc[x];
 980 }
 981
 982 int FileReader::FileSize()
 983 {
 984         return fc.size();
 985 }