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