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::OnAddGLine(long, User*, const std::string&, const std::string&) { }
 179 void            Module::OnAddZLine(long, User*, const std::string&, const std::string&) { }
 180 void            Module::OnAddKLine(long, User*, const std::string&, const std::string&) { }
 181 void            Module::OnAddQLine(long, User*, const std::string&, const std::string&) { }
 182 void            Module::OnAddELine(long, User*, const std::string&, const std::string&) { }
 183 void            Module::OnDelGLine(User*, const std::string&) { }
 184 void            Module::OnDelZLine(User*, const std::string&) { }
 185 void            Module::OnDelKLine(User*, const std::string&) { }
 186 void            Module::OnDelQLine(User*, const std::string&) { }
 187 void            Module::OnDelELine(User*, const std::string&) { }
 188 void            Module::OnCleanup(int, void*) { }
 189 void            Module::Implements(char* Implements) { for (int j = 0; j < 255; j++) Implements[j] = 0; }
 190 void            Module::OnChannelDelete(Channel*) { }
 191 Priority        Module::Prioritize() { return PRIORITY_DONTCARE; }
 192 void            Module::OnSetAway(User*) { }
 193 void            Module::OnCancelAway(User*) { }
 194 int             Module::OnUserList(User*, Channel*, CUList*&) { return 0; }
 195 int             Module::OnWhoisLine(User*, User*, int&, std::string&) { return 0; }
 196 void            Module::OnBuildExemptList(MessageType, Channel*, User*, char, CUList&, const std::string&) { }
 197 void            Module::OnGarbageCollect() { }
 198 void            Module::OnBufferFlushed(User*) { }
 199
 200
 201 ModuleManager::ModuleManager(InspIRCd* Ins)
 202 : ModCount(0), Instance(Ins)
 203 {
 204 }
 205
 206 ModuleManager::~ModuleManager()
 207 {
 208 }
 209
 210 const char* ModuleManager::LastError()
 211 {
 212         return MODERR;
 213 }
 214
 215 bool ModuleManager::Load(const char* filename)
 216 {
 217         /* Do we have a glob pattern in the filename?
 218          * The user wants to load multiple modules which
 219          * match the pattern.
 220          */
 221         if (strchr(filename,'*') || (strchr(filename,'?')))
 222         {
 223                 int n_match = 0;
 224                 DIR* library = opendir(Instance->Config->ModPath);
 225                 if (library)
 226                 {
 227                         /* Try and locate and load all modules matching the pattern */
 228                         dirent* entry = NULL;
 229                         while ((entry = readdir(library)))
 230                         {
 231                                 if (Instance->MatchText(entry->d_name, filename))
 232                                 {
 233                                         if (!this->Load(entry->d_name))
 234                                                 n_match++;
 235                                 }
 236                         }
 237                         closedir(library);
 238                 }
 239                 /* Loadmodule will now return false if any one of the modules failed
 240                  * to load (but wont abort when it encounters a bad one) and when 1 or
 241                  * more modules were actually loaded.
 242                  */
 243                 return (n_match > 0);
 244         }
 245
 246         char modfile[MAXBUF];
 247         snprintf(modfile,MAXBUF,"%s/%s",Instance->Config->ModPath,filename);
 248         std::string filename_str = filename;
 249
 250         if (!ServerConfig::DirValid(modfile))
 251         {
 252                 snprintf(MODERR, MAXBUF,"Module %s is not within the modules directory.", modfile);
 253                 Instance->Log(DEFAULT, MODERR);
 254                 return false;
 255         }
 256
 257         if (!ServerConfig::FileExists(modfile))
 258         {
 259                 snprintf(MODERR,MAXBUF,"Module file could not be found: %s", modfile);
 260                 Instance->Log(DEFAULT, MODERR);
 261                 return false;
 262         }
 263
 264         if(find(Instance->Config->module_names.begin(), Instance->Config->module_names.end(), filename_str) != Instance->Config->module_names.end())
 265         {
 266                 Instance->Log(DEFAULT,"Module %s is already loaded, cannot load a module twice!",modfile);
 267                 snprintf(MODERR, MAXBUF, "Module already loaded");
 268                 return false;
 269         }
 270
 271         Module* newmod;
 272         ircd_module* newhandle;
 273
 274         newmod = NULL;
 275         newhandle = NULL;
 276
 277         try
 278         {
 279                 /* This will throw a CoreException if there's a problem loading
 280                  * the module file or getting a pointer to the init_module symbol.
 281                  */
 282                 newhandle = new ircd_module(Instance, modfile, "init_module");
 283
 284                 handles[this->ModCount+1] = newhandle;
 285
 286                 newmod = handles[this->ModCount+1]->CallInit();
 287
 288                 if(newmod)
 289                 {
 290                         Version v = newmod->GetVersion();
 291
 292                         if (v.API != API_VERSION)
 293                         {
 294                                 delete newmod;
 295                                 Instance->Log(DEFAULT,"Unable to load %s: Incorrect module API version: %d (our version: %d)",modfile,v.API,API_VERSION);
 296                                 snprintf(MODERR,MAXBUF,"Loader/Linker error: Incorrect module API version: %d (our version: %d)",v.API,API_VERSION);
 297                                 return false;
 298                         }
 299                         else
 300                         {
 301                                 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]"));
 302                         }
 303
 304                         modules[this->ModCount+1] = newmod;
 305
 306                         /* save the module and the module's classfactory, if
 307                          * this isnt done, random crashes can occur :/ */
 308                         Instance->Config->module_names.push_back(filename);
 309
 310                         char* x = &Instance->Config->implement_lists[this->ModCount+1][0];
 311                         for(int t = 0; t < 255; t++)
 312                                 x[t] = 0;
 313
 314                         modules[this->ModCount+1]->Implements(x);
 315
 316                         for(int t = 0; t < 255; t++)
 317                                 Instance->Config->global_implementation[t] += Instance->Config->implement_lists[this->ModCount+1][t];
 318                 }
 319                 else
 320                 {
 321                         Instance->Log(DEFAULT, "Unable to load %s",modfile);
 322                         snprintf(MODERR,MAXBUF, "Probably missing init_module() entrypoint, but dlsym() didn't notice a problem");
 323                         return false;
 324                 }
 325         }
 326         catch (LoadModuleException& modexcept)
 327         {
 328                 Instance->Log(DEFAULT,"Unable to load %s: %s", modfile, modexcept.GetReason());
 329                 snprintf(MODERR,MAXBUF,"Loader/Linker error: %s", modexcept.GetReason());
 330                 return false;
 331         }
 332         catch (FindSymbolException& modexcept)
 333         {
 334                 Instance->Log(DEFAULT,"Unable to load %s: %s", modfile, modexcept.GetReason());
 335                 snprintf(MODERR,MAXBUF,"Loader/Linker error: %s", modexcept.GetReason());
 336                 return false;
 337         }
 338         catch (CoreException& modexcept)
 339         {
 340                 Instance->Log(DEFAULT,"Unable to load %s: %s",modfile,modexcept.GetReason());
 341                 snprintf(MODERR,MAXBUF,"Factory function of %s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 342                 return false;
 343         }
 344
 345         this->ModCount++;
 346         FOREACH_MOD_I(Instance,I_OnLoadModule,OnLoadModule(modules[this->ModCount],filename_str));
 347         // now work out which modules, if any, want to move to the back of the queue,
 348         // and if they do, move them there.
 349         std::vector<std::string> put_to_back;
 350         std::vector<std::string> put_to_front;
 351         std::map<std::string,std::string> put_before;
 352         std::map<std::string,std::string> put_after;
 353         for (unsigned int j = 0; j < Instance->Config->module_names.size(); j++)
 354         {
 355                 if (modules[j]->Prioritize() == PRIORITY_LAST)
 356                         put_to_back.push_back(Instance->Config->module_names[j]);
 357                 else if (modules[j]->Prioritize() == PRIORITY_FIRST)
 358                         put_to_front.push_back(Instance->Config->module_names[j]);
 359                 else if ((modules[j]->Prioritize() & 0xFF) == PRIORITY_BEFORE)
 360                         put_before[Instance->Config->module_names[j]] = Instance->Config->module_names[modules[j]->Prioritize() >> 8];
 361                 else if ((modules[j]->Prioritize() & 0xFF) == PRIORITY_AFTER)
 362                         put_after[Instance->Config->module_names[j]] = Instance->Config->module_names[modules[j]->Prioritize() >> 8];
 363         }
 364         for (unsigned int j = 0; j < put_to_back.size(); j++)
 365                 MoveToLast(put_to_back[j]);
 366         for (unsigned int j = 0; j < put_to_front.size(); j++)
 367                 MoveToFirst(put_to_front[j]);
 368         for (std::map<std::string,std::string>::iterator j = put_before.begin(); j != put_before.end(); j++)
 369                 MoveBefore(j->first,j->second);
 370         for (std::map<std::string,std::string>::iterator j = put_after.begin(); j != put_after.end(); j++)
 371                 MoveAfter(j->first,j->second);
 372         Instance->BuildISupport();
 373         return true;
 374 }
 375
 376 bool ModuleManager::EraseHandle(unsigned int j)
 377 {
 378         ModuleHandleList::iterator iter;
 379
 380         if (j >= handles.size())
 381         {
 382                 return false;
 383         }
 384
 385         iter = handles.begin() + j;
 386
 387         if(*iter)
 388         {
 389                 delete *iter;
 390                 handles.erase(iter);
 391                 handles.push_back(NULL);
 392         }
 393
 394         return true;
 395 }
 396
 397 bool ModuleManager::EraseModule(unsigned int j)
 398 {
 399         bool success = false;
 400
 401         ModuleList::iterator iter;
 402
 403         if (j >= modules.size())
 404         {
 405                 return false;
 406         }
 407
 408         iter = modules.begin() + j;
 409
 410         if (*iter)
 411         {
 412                 delete *iter;
 413                 modules.erase(iter);
 414                 modules.push_back(NULL);
 415                 success = true;
 416         }
 417
 418         std::vector<std::string>::iterator iter2;
 419
 420         if (j >= Instance->Config->module_names.size())
 421         {
 422                 return false;
 423         }
 424
 425         iter2 = Instance->Config->module_names.begin() + j;
 426
 427         Instance->Config->module_names.erase(iter2);
 428         success = true;
 429
 430         return success;
 431 }
 432
 433 void ModuleManager::MoveTo(std::string modulename,int slot)
 434 {
 435         unsigned int v2 = 256;
 436
 437         for (unsigned int v = 0; v < Instance->Config->module_names.size(); v++)
 438         {
 439                 if (Instance->Config->module_names[v] == modulename)
 440                 {
 441                         // found an instance, swap it with the item at the end
 442                         v2 = v;
 443                         break;
 444                 }
 445         }
 446         if ((v2 != (unsigned int)slot) && (v2 < 256))
 447         {
 448                 // Swap the module names over
 449                 Instance->Config->module_names[v2] = Instance->Config->module_names[slot];
 450                 Instance->Config->module_names[slot] = modulename;
 451                 // now swap the module factories
 452                 ircd_module* temp = handles[v2];
 453                 handles[v2] = handles[slot];
 454                 handles[slot] = temp;
 455                 // now swap the module objects
 456                 Module* temp_module = modules[v2];
 457                 modules[v2] = modules[slot];
 458                 modules[slot] = temp_module;
 459                 // now swap the implement lists (we dont
 460                 // need to swap the global or recount it)
 461                 for (int n = 0; n < 255; n++)
 462                 {
 463                         char x = Instance->Config->implement_lists[v2][n];
 464                         Instance->Config->implement_lists[v2][n] = Instance->Config->implement_lists[slot][n];
 465                         Instance->Config->implement_lists[slot][n] = x;
 466                 }
 467         }
 468 }
 469
 470 void ModuleManager::MoveAfter(std::string modulename, std::string after)
 471 {
 472         for (unsigned int v = 0; v < Instance->Config->module_names.size(); v++)
 473         {
 474                 if (Instance->Config->module_names[v] == after)
 475                 {
 476                         MoveTo(modulename, v);
 477                         return;
 478                 }
 479         }
 480 }
 481
 482 void ModuleManager::MoveBefore(std::string modulename, std::string before)
 483 {
 484         for (unsigned int v = 0; v < Instance->Config->module_names.size(); v++)
 485         {
 486                 if (Instance->Config->module_names[v] == before)
 487                 {
 488                         if (v > 0)
 489                         {
 490                                 MoveTo(modulename, v-1);
 491                         }
 492                         else
 493                         {
 494                                 MoveTo(modulename, v);
 495                         }
 496                         return;
 497                 }
 498         }
 499 }
 500
 501 void ModuleManager::MoveToFirst(std::string modulename)
 502 {
 503         MoveTo(modulename,0);
 504 }
 505
 506 void ModuleManager::MoveToLast(std::string modulename)
 507 {
 508         MoveTo(modulename,this->GetCount());
 509 }
 510
 511 bool ModuleManager::Unload(const char* filename)
 512 {
 513         std::string filename_str = filename;
 514         for (unsigned int j = 0; j != Instance->Config->module_names.size(); j++)
 515         {
 516                 if (Instance->Config->module_names[j] == filename_str)
 517                 {
 518                         if (modules[j]->GetVersion().Flags & VF_STATIC)
 519                         {
 520                                 Instance->Log(DEFAULT,"Failed to unload STATIC module %s",filename);
 521                                 snprintf(MODERR,MAXBUF,"Module not unloadable (marked static)");
 522                                 return false;
 523                         }
 524                         std::pair<int,std::string> intercount = GetInterfaceInstanceCount(modules[j]);
 525                         if (intercount.first > 0)
 526                         {
 527                                 Instance->Log(DEFAULT,"Failed to unload module %s, being used by %d other(s) via interface '%s'",filename, intercount.first, intercount.second.c_str());
 528                                 snprintf(MODERR,MAXBUF,"Module not unloadable (Still in use by %d other module%s which %s using its interface '%s') -- unload dependent modules first!",
 529                                                 intercount.first,
 530                                                 intercount.first > 1 ? "s" : "",
 531                                                 intercount.first > 1 ? "are" : "is",
 532                                                 intercount.second.c_str());
 533                                 return false;
 534                         }
 535                         /* Give the module a chance to tidy out all its metadata */
 536                         for (chan_hash::iterator c = Instance->chanlist->begin(); c != Instance->chanlist->end(); c++)
 537                         {
 538                                 modules[j]->OnCleanup(TYPE_CHANNEL,c->second);
 539                         }
 540                         for (user_hash::iterator u = Instance->clientlist->begin(); u != Instance->clientlist->end(); u++)
 541                         {
 542                                 modules[j]->OnCleanup(TYPE_USER,u->second);
 543                         }
 544
 545                         /* Tidy up any dangling resolvers */
 546                         Instance->Res->CleanResolvers(modules[j]);
 547
 548                         FOREACH_MOD_I(Instance,I_OnUnloadModule,OnUnloadModule(modules[j],Instance->Config->module_names[j]));
 549
 550                         for(int t = 0; t < 255; t++)
 551                         {
 552                                 Instance->Config->global_implementation[t] -= Instance->Config->implement_lists[j][t];
 553                         }
 554
 555                         /* We have to renumber implement_lists after unload because the module numbers change!
 556                          */
 557                         for(int j2 = j; j2 < 254; j2++)
 558                         {
 559                                 for(int t = 0; t < 255; t++)
 560                                 {
 561                                         Instance->Config->implement_lists[j2][t] = Instance->Config->implement_lists[j2+1][t];
 562                                 }
 563                         }
 564
 565                         // found the module
 566                         Instance->Parser->RemoveCommands(filename);
 567                         this->EraseModule(j);
 568                         this->EraseHandle(j);
 569                         Instance->Log(DEFAULT,"Module %s unloaded",filename);
 570                         this->ModCount--;
 571                         Instance->BuildISupport();
 572                         return true;
 573                 }
 574         }
 575         Instance->Log(DEFAULT,"Module %s is not loaded, cannot unload it!",filename);
 576         snprintf(MODERR,MAXBUF,"Module not loaded");
 577         return false;
 578 }
 579
 580 /* We must load the modules AFTER initializing the socket engine, now */
 581 void ModuleManager::LoadAll()
 582 {
 583         char configToken[MAXBUF];
 584         Instance->Config->module_names.clear();
 585         ModCount = -1;
 586
 587         for(int count = 0; count < Instance->Config->ConfValueEnum(Instance->Config->config_data, "module"); count++)
 588         {
 589                 Instance->Config->ConfValue(Instance->Config->config_data, "module", "name", count, configToken, MAXBUF);
 590                 printf_c("[\033[1;32m*\033[0m] Loading module:\t\033[1;32m%s\033[0m\n",configToken);
 591
 592                 if (!this->Load(configToken))
 593                 {
 594                         Instance->Log(DEFAULT,"There was an error loading the module '%s': %s", configToken, this->LastError());
 595                         printf_c("\n[\033[1;31m*\033[0m] There was an error loading the module '%s': %s\n\n", configToken, this->LastError());
 596                         Instance->Exit(EXIT_STATUS_MODULE);
 597                 }
 598         }
 599         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");
 600         Instance->Log(DEFAULT,"Total loaded modules: %d", this->GetCount()+1);
 601 }
 602
 603 long ModuleManager::PriorityAfter(const std::string &modulename)
 604 {
 605         for (unsigned int j = 0; j < Instance->Config->module_names.size(); j++)
 606         {
 607                 if (Instance->Config->module_names[j] == modulename)
 608                 {
 609                         return ((j << 8) | PRIORITY_AFTER);
 610                 }
 611         }
 612         return PRIORITY_DONTCARE;
 613 }
 614
 615 long ModuleManager::PriorityBefore(const std::string &modulename)
 616 {
 617         for (unsigned int j = 0; j < Instance->Config->module_names.size(); j++)
 618         {
 619                 if (Instance->Config->module_names[j] == modulename)
 620                 {
 621                         return ((j << 8) | PRIORITY_BEFORE);
 622                 }
 623         }
 624         return PRIORITY_DONTCARE;
 625 }
 626
 627 bool ModuleManager::PublishFeature(const std::string &FeatureName, Module* Mod)
 628 {
 629         if (Features.find(FeatureName) == Features.end())
 630         {
 631                 Features[FeatureName] = Mod;
 632                 return true;
 633         }
 634         return false;
 635 }
 636
 637 bool ModuleManager::UnpublishFeature(const std::string &FeatureName)
 638 {
 639         featurelist::iterator iter = Features.find(FeatureName);
 640
 641         if (iter == Features.end())
 642                 return false;
 643
 644         Features.erase(iter);
 645         return true;
 646 }
 647
 648 Module* ModuleManager::FindFeature(const std::string &FeatureName)
 649 {
 650         featurelist::iterator iter = Features.find(FeatureName);
 651
 652         if (iter == Features.end())
 653                 return NULL;
 654
 655         return iter->second;
 656 }
 657
 658 bool ModuleManager::PublishInterface(const std::string &InterfaceName, Module* Mod)
 659 {
 660         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 661
 662         if (iter == Interfaces.end())
 663         {
 664                 modulelist ml;
 665                 ml.push_back(Mod);
 666                 Interfaces[InterfaceName] = std::make_pair(0, ml);
 667                 return true;
 668         }
 669         else
 670         {
 671                 iter->second.second.push_back(Mod);
 672                 return true;
 673         }
 674         return false;
 675 }
 676
 677 bool ModuleManager::UnpublishInterface(const std::string &InterfaceName, Module* Mod)
 678 {
 679         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 680
 681         if (iter == Interfaces.end())
 682                 return false;
 683
 684         for (modulelist::iterator x = iter->second.second.begin(); x != iter->second.second.end(); x++)
 685         {
 686                 if (*x == Mod)
 687                 {
 688                         iter->second.second.erase(x);
 689                         if (iter->second.second.empty())
 690                                 Interfaces.erase(InterfaceName);
 691                         return true;
 692                 }
 693         }
 694         return false;
 695 }
 696
 697 modulelist* ModuleManager::FindInterface(const std::string &InterfaceName)
 698 {
 699         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 700         if (iter == Interfaces.end())
 701                 return NULL;
 702         else
 703                 return &(iter->second.second);
 704 }
 705
 706 void ModuleManager::UseInterface(const std::string &InterfaceName)
 707 {
 708         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 709         if (iter != Interfaces.end())
 710                 iter->second.first++;
 711
 712 }
 713
 714 void ModuleManager::DoneWithInterface(const std::string &InterfaceName)
 715 {
 716         interfacelist::iterator iter = Interfaces.find(InterfaceName);
 717         if (iter != Interfaces.end())
 718                 iter->second.first--;
 719 }
 720
 721 std::pair<int,std::string> ModuleManager::GetInterfaceInstanceCount(Module* m)
 722 {
 723         for (interfacelist::iterator iter = Interfaces.begin(); iter != Interfaces.end(); iter++)
 724         {
 725                 for (modulelist::iterator x = iter->second.second.begin(); x != iter->second.second.end(); x++)
 726                 {
 727                         if (*x == m)
 728                         {
 729                                 return std::make_pair(iter->second.first, iter->first);
 730                         }
 731                 }
 732         }
 733         return std::make_pair(0, "");
 734 }
 735
 736 const std::string& ModuleManager::GetModuleName(Module* m)
 737 {
 738         static std::string nothing; /* Prevent compiler warning */
 739
 740         if (!this->GetCount())
 741                 return nothing;
 742
 743         for (int i = 0; i <= this->GetCount(); i++)
 744         {
 745                 if (this->modules[i] == m)
 746                 {
 747                         return Instance->Config->module_names[i];
 748                 }
 749         }
 750         return nothing; /* As above */
 751 }
 752
 753 /* This is ugly, yes, but hash_map's arent designed to be
 754  * addressed in this manner, and this is a bit of a kludge.
 755  * Luckily its a specialist function and rarely used by
 756  * many modules (in fact, it was specially created to make
 757  * m_safelist possible, initially).
 758  */
 759
 760 Channel* InspIRCd::GetChannelIndex(long index)
 761 {
 762         int target = 0;
 763         for (chan_hash::iterator n = this->chanlist->begin(); n != this->chanlist->end(); n++, target++)
 764         {
 765                 if (index == target)
 766                         return n->second;
 767         }
 768         return NULL;
 769 }
 770
 771 bool InspIRCd::MatchText(const std::string &sliteral, const std::string &spattern)
 772 {
 773         return match(sliteral.c_str(),spattern.c_str());
 774 }
 775
 776 CmdResult InspIRCd::CallCommandHandler(const std::string &commandname, const char** parameters, int pcnt, User* user)
 777 {
 778         return this->Parser->CallHandler(commandname,parameters,pcnt,user);
 779 }
 780
 781 bool InspIRCd::IsValidModuleCommand(const std::string &commandname, int pcnt, User* user)
 782 {
 783         return this->Parser->IsValidCommand(commandname, pcnt, user);
 784 }
 785
 786 void InspIRCd::AddCommand(Command *f)
 787 {
 788         if (!this->Parser->CreateCommand(f))
 789         {
 790                 ModuleException err("Command "+std::string(f->command)+" already exists.");
 791                 throw (err);
 792         }
 793 }
 794
 795 void InspIRCd::SendMode(const char** parameters, int pcnt, User *user)
 796 {
 797         this->Modes->Process(parameters,pcnt,user,true);
 798 }
 799
 800 void InspIRCd::DumpText(User* User, const std::string &LinePrefix, stringstream &TextStream)
 801 {
 802         std::string CompleteLine = LinePrefix;
 803         std::string Word;
 804         while (TextStream >> Word)
 805         {
 806                 if (CompleteLine.length() + Word.length() + 3 > 500)
 807                 {
 808                         User->WriteServ(CompleteLine);
 809                         CompleteLine = LinePrefix;
 810                 }
 811                 CompleteLine = CompleteLine + Word + " ";
 812         }
 813         User->WriteServ(CompleteLine);
 814 }
 815
 816 User* FindDescriptorHandler::Call(int socket)
 817 {
 818         return reinterpret_cast<User*>(Server->SE->GetRef(socket));
 819 }
 820
 821 bool InspIRCd::AddMode(ModeHandler* mh)
 822 {
 823         return this->Modes->AddMode(mh);
 824 }
 825
 826 bool InspIRCd::AddModeWatcher(ModeWatcher* mw)
 827 {
 828         return this->Modes->AddModeWatcher(mw);
 829 }
 830
 831 bool InspIRCd::DelModeWatcher(ModeWatcher* mw)
 832 {
 833         return this->Modes->DelModeWatcher(mw);
 834 }
 835
 836 bool InspIRCd::AddResolver(Resolver* r, bool cached)
 837 {
 838         if (!cached)
 839                 return this->Res->AddResolverClass(r);
 840         else
 841         {
 842                 r->TriggerCachedResult();
 843                 delete r;
 844                 return true;
 845         }
 846 }
 847
 848 void InspIRCd::AddGLine(long duration, const std::string &source, const std::string &reason, const std::string &hostmask)
 849 {
 850         XLines->add_gline(duration, source.c_str(), reason.c_str(), hostmask.c_str());
 851         XLines->apply_lines(APPLY_GLINES);
 852 }
 853
 854 void InspIRCd::AddQLine(long duration, const std::string &source, const std::string &reason, const std::string &nickname)
 855 {
 856         XLines->add_qline(duration, source.c_str(), reason.c_str(), nickname.c_str());
 857         XLines->apply_lines(APPLY_QLINES);
 858 }
 859
 860 void InspIRCd::AddZLine(long duration, const std::string &source, const std::string &reason, const std::string &ipaddr)
 861 {
 862         XLines->add_zline(duration, source.c_str(), reason.c_str(), ipaddr.c_str());
 863         XLines->apply_lines(APPLY_ZLINES);
 864 }
 865
 866 void InspIRCd::AddKLine(long duration, const std::string &source, const std::string &reason, const std::string &hostmask)
 867 {
 868         XLines->add_kline(duration, source.c_str(), reason.c_str(), hostmask.c_str());
 869         XLines->apply_lines(APPLY_KLINES);
 870 }
 871
 872 void InspIRCd::AddELine(long duration, const std::string &source, const std::string &reason, const std::string &hostmask)
 873 {
 874         XLines->add_eline(duration, source.c_str(), reason.c_str(), hostmask.c_str());
 875 }
 876
 877 bool InspIRCd::DelGLine(const std::string &hostmask)
 878 {
 879         return XLines->del_gline(hostmask.c_str());
 880 }
 881
 882 bool InspIRCd::DelQLine(const std::string &nickname)
 883 {
 884         return XLines->del_qline(nickname.c_str());
 885 }
 886
 887 bool InspIRCd::DelZLine(const std::string &ipaddr)
 888 {
 889         return XLines->del_zline(ipaddr.c_str());
 890 }
 891
 892 bool InspIRCd::DelKLine(const std::string &hostmask)
 893 {
 894         return XLines->del_kline(hostmask.c_str());
 895 }
 896
 897 bool InspIRCd::DelELine(const std::string &hostmask)
 898 {
 899         return XLines->del_eline(hostmask.c_str());
 900 }
 901
 902 Module* ModuleManager::Find(const std::string &name)
 903 {
 904         for (int i = 0; i <= this->GetCount(); i++)
 905         {
 906                 if (Instance->Config->module_names[i] == name)
 907                 {
 908                         return this->modules[i];
 909                 }
 910         }
 911         return NULL;
 912 }
 913
 914 ConfigReader::ConfigReader(InspIRCd* Instance) : ServerInstance(Instance)
 915 {
 916         /* Is there any reason to load the entire config file again here?
 917          * it's needed if they specify another config file, but using the
 918          * default one we can just use the global config data - pre-parsed!
 919          */
 920         this->errorlog = new std::ostringstream(std::stringstream::in | std::stringstream::out);
 921         this->error = CONF_NO_ERROR;
 922         this->data = &ServerInstance->Config->config_data;
 923         this->privatehash = false;
 924 }
 925
 926
 927 ConfigReader::~ConfigReader()
 928 {
 929         if (this->errorlog)
 930                 delete this->errorlog;
 931         if(this->privatehash)
 932                 delete this->data;
 933 }
 934
 935
 936 ConfigReader::ConfigReader(InspIRCd* Instance, const std::string &filename) : ServerInstance(Instance)
 937 {
 938         ServerInstance->Config->ClearStack();
 939
 940         this->error = CONF_NO_ERROR;
 941         this->data = new ConfigDataHash;
 942         this->privatehash = true;
 943         this->errorlog = new std::ostringstream(std::stringstream::in | std::stringstream::out);
 944         this->readerror = ServerInstance->Config->LoadConf(*this->data, filename, *this->errorlog);
 945         if (!this->readerror)
 946                 this->error = CONF_FILE_NOT_FOUND;
 947 }
 948
 949
 950 std::string ConfigReader::ReadValue(const std::string &tag, const std::string &name, const std::string &default_value, int index, bool allow_linefeeds)
 951 {
 952         /* Don't need to strlcpy() tag and name anymore, ReadConf() takes const char* */
 953         std::string result;
 954
 955         if (!ServerInstance->Config->ConfValue(*this->data, tag, name, default_value, index, result, allow_linefeeds))
 956         {
 957                 this->error = CONF_VALUE_NOT_FOUND;
 958         }
 959         return result;
 960 }
 961
 962 std::string ConfigReader::ReadValue(const std::string &tag, const std::string &name, int index, bool allow_linefeeds)
 963 {
 964         return ReadValue(tag, name, "", index, allow_linefeeds);
 965 }
 966
 967 bool ConfigReader::ReadFlag(const std::string &tag, const std::string &name, const std::string &default_value, int index)
 968 {
 969         return ServerInstance->Config->ConfValueBool(*this->data, tag, name, default_value, index);
 970 }
 971
 972 bool ConfigReader::ReadFlag(const std::string &tag, const std::string &name, int index)
 973 {
 974         return ReadFlag(tag, name, "", index);
 975 }
 976
 977
 978 int ConfigReader::ReadInteger(const std::string &tag, const std::string &name, const std::string &default_value, int index, bool need_positive)
 979 {
 980         int result;
 981
 982         if(!ServerInstance->Config->ConfValueInteger(*this->data, tag, name, default_value, index, result))
 983         {
 984                 this->error = CONF_VALUE_NOT_FOUND;
 985                 return 0;
 986         }
 987
 988         if ((need_positive) && (result < 0))
 989         {
 990                 this->error = CONF_INT_NEGATIVE;
 991                 return 0;
 992         }
 993
 994         return result;
 995 }
 996
 997 int ConfigReader::ReadInteger(const std::string &tag, const std::string &name, int index, bool need_positive)
 998 {
 999         return ReadInteger(tag, name, "", index, need_positive);
1000 }
1001
1002 long ConfigReader::GetError()
1003 {
1004         long olderr = this->error;
1005         this->error = 0;
1006         return olderr;
1007 }
1008
1009 void ConfigReader::DumpErrors(bool bail, User* user)
1010 {
1011         ServerInstance->Config->ReportConfigError(this->errorlog->str(), bail, user);
1012 }
1013
1014
1015 int ConfigReader::Enumerate(const std::string &tag)
1016 {
1017         return ServerInstance->Config->ConfValueEnum(*this->data, tag);
1018 }
1019
1020 int ConfigReader::EnumerateValues(const std::string &tag, int index)
1021 {
1022         return ServerInstance->Config->ConfVarEnum(*this->data, tag, index);
1023 }
1024
1025 bool ConfigReader::Verify()
1026 {
1027         return this->readerror;
1028 }
1029
1030
1031 FileReader::FileReader(InspIRCd* Instance, const std::string &filename) : ServerInstance(Instance)
1032 {
1033         LoadFile(filename);
1034 }
1035
1036 FileReader::FileReader(InspIRCd* Instance) : ServerInstance(Instance)
1037 {
1038 }
1039
1040 std::string FileReader::Contents()
1041 {
1042         std::string x;
1043         for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
1044         {
1045                 x.append(*a);
1046                 x.append("\r\n");
1047         }
1048         return x;
1049 }
1050
1051 unsigned long FileReader::ContentSize()
1052 {
1053         return this->contentsize;
1054 }
1055
1056 void FileReader::CalcSize()
1057 {
1058         unsigned long n = 0;
1059         for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
1060                 n += (a->length() + 2);
1061         this->contentsize = n;
1062 }
1063
1064 void FileReader::LoadFile(const std::string &filename)
1065 {
1066         file_cache c;
1067         c.clear();
1068         if (ServerInstance->Config->ReadFile(c,filename.c_str()))
1069         {
1070                 this->fc = c;
1071                 this->CalcSize();
1072         }
1073 }
1074
1075
1076 FileReader::~FileReader()
1077 {
1078 }
1079
1080 bool FileReader::Exists()
1081 {
1082         return (!(fc.size() == 0));
1083 }
1084
1085 std::string FileReader::GetLine(int x)
1086 {
1087         if ((x<0) || ((unsigned)x>fc.size()))
1088                 return "";
1089         return fc[x];
1090 }
1091
1092 int FileReader::FileSize()
1093 {
1094         return fc.size();
1095 }