]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modmanager_dynamic.cpp
Don't try quite so hard to writev() the entire buffer in one go
[user/henk/code/inspircd.git] / src / modmanager_dynamic.cpp
index 0c63eefcd0cab6f918f8042387bcbf9b354ac86c..05c9740632731a62b998b9d3c7c49d3d36ea2cd2 100644 (file)
 
 #ifndef PURE_STATIC
 
-bool ModuleManager::Load(const char* filename)
+bool ModuleManager::Load(const std::string& filename, bool defer)
 {
        /* Don't allow people to specify paths for modules, it doesn't work as expected */
-       if (strchr(filename, '/'))
+       if (filename.find('/') != std::string::npos)
                return false;
        /* Do we have a glob pattern in the filename?
         * The user wants to load multiple modules which
         * match the pattern.
         */
-       if (strchr(filename,'*') || (strchr(filename,'?')))
+       if (strchr(filename.c_str(),'*') || (strchr(filename.c_str(),'?')))
        {
                int n_match = 0;
                DIR* library = opendir(ServerInstance->Config->ModPath.c_str());
@@ -44,9 +44,11 @@ bool ModuleManager::Load(const char* filename)
                        dirent* entry = NULL;
                        while (0 != (entry = readdir(library)))
                        {
-                               if (InspIRCd::Match(entry->d_name, filename, ascii_case_insensitive_map))
+                               if (entry->d_name[0] == '.')
+                                       continue;
+                               if (InspIRCd::Match(entry->d_name, filename.c_str(), ascii_case_insensitive_map))
                                {
-                                       if (!this->Load(entry->d_name))
+                                       if (!this->Load(entry->d_name, defer))
                                                n_match++;
                                }
                        }
@@ -60,19 +62,18 @@ bool ModuleManager::Load(const char* filename)
        }
 
        char modfile[MAXBUF];
-       snprintf(modfile,MAXBUF,"%s/%s",ServerInstance->Config->ModPath.c_str(),filename);
-       std::string filename_str = filename;
+       snprintf(modfile,MAXBUF,"%s/%s",ServerInstance->Config->ModPath.c_str(),filename.c_str());
 
        if (!ServerConfig::FileExists(modfile))
        {
-               LastModuleError = "Module file could not be found: " + filename_str;
+               LastModuleError = "Module file could not be found: " + filename;
                ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
                return false;
        }
 
-       if (Modules.find(filename_str) != Modules.end())
+       if (Modules.find(filename) != Modules.end())
        {
-               LastModuleError = "Module " + filename_str + " is already loaded, cannot load a module twice!";
+               LastModuleError = "Module " + filename + " is already loaded, cannot load a module twice!";
                ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
                return false;
        }
@@ -86,18 +87,25 @@ bool ModuleManager::Load(const char* filename)
 
                if (newmod)
                {
-                       newmod->ModuleSourceFile = filename_str;
+                       newmod->ModuleSourceFile = filename;
                        newmod->ModuleDLLManager = newhandle;
-                       Modules[filename_str] = newmod;
-                       newmod->init();
+                       Modules[filename] = newmod;
+                       if (defer)
+                       {
+                               ServerInstance->Logs->Log("MODULE", DEFAULT,"New module introduced: %s", filename.c_str());
+                       }
+                       else
+                       {
+                               newmod->init();
 
-                       Version v = newmod->GetVersion();
-                       ServerInstance->Logs->Log("MODULE", DEFAULT,"New module introduced: %s (Module version %s)%s",
-                               filename, newhandle->GetVersion().c_str(), (!(v.Flags & VF_VENDOR) ? " [3rd Party]" : " [Vendor]"));
+                               Version v = newmod->GetVersion();
+                               ServerInstance->Logs->Log("MODULE", DEFAULT,"New module introduced: %s (Module version %s)%s",
+                                       filename.c_str(), newhandle->GetVersion().c_str(), (!(v.Flags & VF_VENDOR) ? " [3rd Party]" : " [Vendor]"));
+                       }
                }
                else
                {
-                       LastModuleError = "Unable to load " + filename_str + ": " + newhandle->LastError();
+                       LastModuleError = "Unable to load " + filename + ": " + newhandle->LastError();
                        ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
                        delete newhandle;
                        return false;
@@ -108,16 +116,18 @@ bool ModuleManager::Load(const char* filename)
                // failure in module constructor
                if (newmod)
                        DoSafeUnload(newmod);
-               delete newmod;
-               delete newhandle;
-               LastModuleError = "Unable to load " + filename_str + ": " + modexcept.GetReason();
+               else
+                       delete newhandle;
+               LastModuleError = "Unable to load " + filename + ": " + modexcept.GetReason();
                ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
                return false;
        }
 
        this->ModCount++;
-       FOREACH_MOD(I_OnLoadModule,OnLoadModule(newmod));
+       if (defer)
+               return true;
 
+       FOREACH_MOD(I_OnLoadModule,OnLoadModule(newmod));
        /* We give every module a chance to re-prioritize when we introduce a new one,
         * not just the one thats loading, as the new module could affect the preference
         * of others
@@ -131,7 +141,7 @@ bool ModuleManager::Load(const char* filename)
                if (prioritizationState == PRIO_STATE_LAST)
                        break;
                if (tries == 19)
-                       ServerInstance->Logs->Log("MODULE", DEFAULT, "Hook priority dependency loop detected while loading " + filename_str);
+                       ServerInstance->Logs->Log("MODULE", DEFAULT, "Hook priority dependency loop detected while loading " + filename);
        }
 
        ServerInstance->BuildISupport();
@@ -167,7 +177,8 @@ namespace {
                        ServerInstance->GlobalCulls.Apply();
                        delete dll;
                        bool rv = ServerInstance->Modules->Load(name.c_str());
-                       callback->Call(rv);
+                       if (callback)
+                               callback->Call(rv);
                        ServerInstance->GlobalCulls.AddItem(this);
                }
        };
@@ -208,7 +219,7 @@ void ModuleManager::LoadAll()
                                printf(".");
                                fflush(stdout);
 
-                               if (!Load(entry->d_name))
+                               if (!Load(entry->d_name, true))
                                {
                                        ServerInstance->Logs->Log("MODULE", DEFAULT, this->LastError());
                                        printf_c("\n[\033[1;31m*\033[0m] %s\n\n", this->LastError().c_str());
@@ -227,13 +238,48 @@ void ModuleManager::LoadAll()
                std::string name = tag->getString("name");
                printf_c("[\033[1;32m*\033[0m] Loading module:\t\033[1;32m%s\033[0m\n",name.c_str());
 
-               if (!this->Load(name.c_str()))
+               if (!this->Load(name, true))
                {
                        ServerInstance->Logs->Log("MODULE", DEFAULT, this->LastError());
                        printf_c("\n[\033[1;31m*\033[0m] %s\n\n", this->LastError().c_str());
                        ServerInstance->Exit(EXIT_STATUS_MODULE);
                }
        }
+
+       for(std::map<std::string, Module*>::iterator i = Modules.begin(); i != Modules.end(); i++)
+       {
+               Module* mod = i->second;
+               try 
+               {
+                       mod->init();
+               }
+               catch (CoreException& modexcept)
+               {
+                       LastModuleError = "Unable to initialize " + mod->ModuleSourceFile + ": " + modexcept.GetReason();
+                       ServerInstance->Logs->Log("MODULE", DEFAULT, LastModuleError);
+                       printf_c("\n[\033[1;31m*\033[0m] %s\n\n", LastModuleError.c_str());
+                       ServerInstance->Exit(EXIT_STATUS_MODULE);
+               }
+       }
+
+       /* We give every module a chance to re-prioritize when we introduce a new one,
+        * not just the one thats loading, as the new module could affect the preference
+        * of others
+        */
+       for(int tries = 0; tries < 20; tries++)
+       {
+               prioritizationState = tries > 0 ? PRIO_STATE_LAST : PRIO_STATE_FIRST;
+               for (std::map<std::string, Module*>::iterator n = Modules.begin(); n != Modules.end(); ++n)
+                       n->second->Prioritize();
+
+               if (prioritizationState == PRIO_STATE_LAST)
+                       break;
+               if (tries == 19)
+               {
+                       ServerInstance->Logs->Log("MODULE", DEFAULT, "Hook priority dependency loop detected");
+                       ServerInstance->Exit(EXIT_STATUS_MODULE);
+               }
+       }
 }
 
 void ModuleManager::UnloadAll()