src/modmanager_dynamic.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "xline.h"
  16 #include "socket.h"
  17 #include "socketengine.h"
  18 #include "command_parse.h"
  19 #include "dns.h"
  20 #include "exitcodes.h"
  21
  22 #ifndef WIN32
  23 #include <dirent.h>
  24 #endif
  25
  26 #ifndef PURE_STATIC
  27
  28 bool ModuleManager::Load(const char* filename)
  29 {
  30         /* Don't allow people to specify paths for modules, it doesn't work as expected */
  31         if (strchr(filename, '/'))
  32                 return false;
  33         /* Do we have a glob pattern in the filename?
  34          * The user wants to load multiple modules which
  35          * match the pattern.
  36          */
  37         if (strchr(filename,'*') || (strchr(filename,'?')))
  38         {
  39                 int n_match = 0;
  40                 DIR* library = opendir(ServerInstance->Config->ModPath.c_str());
  41                 if (library)
  42                 {
  43                         /* Try and locate and load all modules matching the pattern */
  44                         dirent* entry = NULL;
  45                         while (0 != (entry = readdir(library)))
  46                         {
  47                                 if (InspIRCd::Match(entry->d_name, filename, ascii_case_insensitive_map))
  48                                 {
  49                                         if (!this->Load(entry->d_name))
  50                                                 n_match++;
  51                                 }
  52                         }
  53                         closedir(library);
  54                 }
  55                 /* Loadmodule will now return false if any one of the modules failed
  56                  * to load (but wont abort when it encounters a bad one) and when 1 or
  57                  * more modules were actually loaded.
  58                  */
  59                 return (n_match > 0 ? false : true);
  60         }
  61
  62         char modfile[MAXBUF];
  63         snprintf(modfile,MAXBUF,"%s/%s",ServerInstance->Config->ModPath.c_str(),filename);
  64         std::string filename_str = filename;
  65
  66         if (!ServerConfig::FileExists(modfile))
  67         {
  68                 LastModuleError = "Module file could not be found: " + filename_str;
  69                 ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
  70                 return false;
  71         }
  72
  73         if (Modules.find(filename_str) != Modules.end())
  74         {
  75                 LastModuleError = "Module " + filename_str + " is already loaded, cannot load a module twice!";
  76                 ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
  77                 return false;
  78         }
  79
  80         Module* newmod = NULL;
  81         DLLManager* newhandle = new DLLManager(modfile);
  82
  83         try
  84         {
  85                 newmod = newhandle->callInit();
  86
  87                 if (newmod)
  88                 {
  89                         newmod->ModuleSourceFile = filename_str;
  90                         newmod->ModuleDLLManager = newhandle;
  91                         Version v = newmod->GetVersion();
  92
  93                         ServerInstance->Logs->Log("MODULE", DEFAULT,"New module introduced: %s (Module version %s)%s",
  94                                 filename, v.version.c_str(), (!(v.Flags & VF_VENDOR) ? " [3rd Party]" : " [Vendor]"));
  95
  96                         Modules[filename_str] = newmod;
  97                 }
  98                 else
  99                 {
 100                         LastModuleError = "Unable to load " + filename_str + ": " + newhandle->LastError();
 101                         ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
 102                         delete newhandle;
 103                         return false;
 104                 }
 105         }
 106         catch (CoreException& modexcept)
 107         {
 108                 // failure in module constructor
 109                 delete newmod;
 110                 delete newhandle;
 111                 LastModuleError = "Unable to load " + filename_str + ": " + modexcept.GetReason();
 112                 ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
 113                 return false;
 114         }
 115
 116         this->ModCount++;
 117         FOREACH_MOD(I_OnLoadModule,OnLoadModule(newmod));
 118
 119         /* We give every module a chance to re-prioritize when we introduce a new one,
 120          * not just the one thats loading, as the new module could affect the preference
 121          * of others
 122          */
 123         for(int tries = 0; tries < 20; tries++)
 124         {
 125                 prioritizationState = tries > 0 ? PRIO_STATE_LAST : PRIO_STATE_FIRST;
 126                 for (std::map<std::string, Module*>::iterator n = Modules.begin(); n != Modules.end(); ++n)
 127                         n->second->Prioritize();
 128
 129                 if (prioritizationState == PRIO_STATE_LAST)
 130                         break;
 131                 if (tries == 19)
 132                         ServerInstance->Logs->Log("MODULE", DEFAULT, "Hook priority dependency loop detected while loading " + filename_str);
 133         }
 134
 135         ServerInstance->BuildISupport();
 136         return true;
 137 }
 138
 139 bool ModuleManager::CanUnload(Module* mod)
 140 {
 141         std::map<std::string, Module*>::iterator modfind = Modules.find(mod->ModuleSourceFile);
 142
 143         if (modfind == Modules.end() || modfind->second != mod)
 144         {
 145                 LastModuleError = "Module " + mod->ModuleSourceFile + " is not loaded, cannot unload it!";
 146                 ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
 147                 return false;
 148         }
 149         if (mod->GetVersion().Flags & VF_STATIC)
 150         {
 151                 LastModuleError = "Module " + mod->ModuleSourceFile + " not unloadable (marked static)";
 152                 ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
 153                 return false;
 154         }
 155         return true;
 156 }
 157
 158 void ModuleManager::DoSafeUnload(Module* mod)
 159 {
 160         std::map<std::string, Module*>::iterator modfind = Modules.find(mod->ModuleSourceFile);
 161
 162         std::vector<reference<ExtensionItem> > items;
 163         ServerInstance->Extensions.BeginUnregister(modfind->second, items);
 164         /* Give the module a chance to tidy out all its metadata */
 165         for (chan_hash::iterator c = ServerInstance->chanlist->begin(); c != ServerInstance->chanlist->end(); c++)
 166         {
 167                 mod->OnCleanup(TYPE_CHANNEL,c->second);
 168                 c->second->doUnhookExtensions(items);
 169                 const UserMembList* users = c->second->GetUsers();
 170                 for(UserMembCIter mi = users->begin(); mi != users->end(); mi++)
 171                         mi->second->doUnhookExtensions(items);
 172         }
 173         for (user_hash::iterator u = ServerInstance->Users->clientlist->begin(); u != ServerInstance->Users->clientlist->end(); u++)
 174         {
 175                 mod->OnCleanup(TYPE_USER,u->second);
 176                 u->second->doUnhookExtensions(items);
 177         }
 178         for(char m='A'; m <= 'z'; m++)
 179         {
 180                 ModeHandler* mh;
 181                 mh = ServerInstance->Modes->FindMode(m, MODETYPE_USER);
 182                 if (mh && mh->creator == mod)
 183                         ServerInstance->Modes->DelMode(mh);
 184                 mh = ServerInstance->Modes->FindMode(m, MODETYPE_CHANNEL);
 185                 if (mh && mh->creator == mod)
 186                         ServerInstance->Modes->DelMode(mh);
 187         }
 188         for(std::multimap<std::string, ServiceProvider*>::iterator i = DataProviders.begin(); i != DataProviders.end(); )
 189         {
 190                 std::multimap<std::string, ServiceProvider*>::iterator curr = i++;
 191                 if (curr->second->creator == mod)
 192                         DataProviders.erase(curr);
 193         }
 194         for(unsigned int i = 0; i < ServerInstance->Modules->ActiveDynrefs.size(); i++)
 195                 ServerInstance->Modules->ActiveDynrefs[i]->ClearCache();
 196
 197         /* Tidy up any dangling resolvers */
 198         ServerInstance->Res->CleanResolvers(mod);
 199
 200         FOREACH_MOD(I_OnUnloadModule,OnUnloadModule(mod));
 201
 202         DetachAll(mod);
 203
 204         Modules.erase(modfind);
 205         ServerInstance->GlobalCulls.AddItem(mod);
 206
 207         ServerInstance->Logs->Log("MODULE", DEFAULT,"Module %s unloaded",mod->ModuleSourceFile.c_str());
 208         this->ModCount--;
 209         ServerInstance->BuildISupport();
 210 }
 211
 212 namespace {
 213         struct UnloadAction : public HandlerBase0<void>
 214         {
 215                 Module* const mod;
 216                 UnloadAction(Module* m) : mod(m) {}
 217                 void Call()
 218                 {
 219                         DLLManager* dll = mod->ModuleDLLManager;
 220                         ServerInstance->Modules->DoSafeUnload(mod);
 221                         ServerInstance->GlobalCulls.Apply();
 222                         delete dll;
 223                         ServerInstance->GlobalCulls.AddItem(this);
 224                 }
 225         };
 226
 227         struct ReloadAction : public HandlerBase0<void>
 228         {
 229                 Module* const mod;
 230                 HandlerBase1<void, bool>* const callback;
 231                 ReloadAction(Module* m, HandlerBase1<void, bool>* c)
 232                         : mod(m), callback(c) {}
 233                 void Call()
 234                 {
 235                         DLLManager* dll = mod->ModuleDLLManager;
 236                         std::string name = mod->ModuleSourceFile;
 237                         ServerInstance->Modules->DoSafeUnload(mod);
 238                         ServerInstance->GlobalCulls.Apply();
 239                         delete dll;
 240                         bool rv = ServerInstance->Modules->Load(name.c_str());
 241                         callback->Call(rv);
 242                         ServerInstance->GlobalCulls.AddItem(this);
 243                 }
 244         };
 245 }
 246
 247 bool ModuleManager::Unload(Module* mod)
 248 {
 249         if (!CanUnload(mod))
 250                 return false;
 251         ServerInstance->AtomicActions.AddAction(new UnloadAction(mod));
 252         return true;
 253 }
 254
 255 void ModuleManager::Reload(Module* mod, HandlerBase1<void, bool>* callback)
 256 {
 257         if (CanUnload(mod))
 258                 ServerInstance->AtomicActions.AddAction(new ReloadAction(mod, callback));
 259         else
 260                 callback->Call(false);
 261 }
 262
 263 /* We must load the modules AFTER initializing the socket engine, now */
 264 void ModuleManager::LoadAll()
 265 {
 266         ModCount = 0;
 267
 268         printf("\nLoading core commands");
 269         fflush(stdout);
 270
 271         DIR* library = opendir(ServerInstance->Config->ModPath.c_str());
 272         if (library)
 273         {
 274                 dirent* entry = NULL;
 275                 while (0 != (entry = readdir(library)))
 276                 {
 277                         if (InspIRCd::Match(entry->d_name, "cmd_*.so", ascii_case_insensitive_map))
 278                         {
 279                                 printf(".");
 280                                 fflush(stdout);
 281
 282                                 if (!Load(entry->d_name))
 283                                 {
 284                                         ServerInstance->Logs->Log("MODULE", DEFAULT, this->LastError());
 285                                         printf_c("\n[\033[1;31m*\033[0m] %s\n\n", this->LastError().c_str());
 286                                         ServerInstance->Exit(EXIT_STATUS_MODULE);
 287                                 }
 288                         }
 289                 }
 290                 closedir(library);
 291                 printf("\n");
 292         }
 293
 294         ConfigTagList tags = ServerInstance->Config->ConfTags("module");
 295         for(ConfigIter i = tags.first; i != tags.second; ++i)
 296         {
 297                 ConfigTag* tag = i->second;
 298                 std::string name = tag->getString("name");
 299                 printf_c("[\033[1;32m*\033[0m] Loading module:\t\033[1;32m%s\033[0m\n",name.c_str());
 300
 301                 if (!this->Load(name.c_str()))
 302                 {
 303                         ServerInstance->Logs->Log("MODULE", DEFAULT, this->LastError());
 304                         printf_c("\n[\033[1;31m*\033[0m] %s\n\n", this->LastError().c_str());
 305                         ServerInstance->Exit(EXIT_STATUS_MODULE);
 306                 }
 307         }
 308 }
 309
 310 void ModuleManager::UnloadAll()
 311 {
 312         /* We do this more than once, so that any service providers get a
 313          * chance to be unhooked by the modules using them, but then get
 314          * a chance to be removed themsleves.
 315          *
 316          * Note: this deliberately does NOT delete the DLLManager objects
 317          */
 318         for (int tries = 0; tries < 4; tries++)
 319         {
 320                 std::map<std::string, Module*>::iterator i = Modules.begin();
 321                 while (i != Modules.end())
 322                 {
 323                         std::map<std::string, Module*>::iterator me = i++;
 324                         if (CanUnload(me->second))
 325                         {
 326                                 DoSafeUnload(me->second);
 327                         }
 328                 }
 329                 ServerInstance->GlobalCulls.Apply();
 330         }
 331 }
 332
 333 #endif