Added comment

[user/henk/code/inspircd.git] / src / dynamic.cpp

diff --git a/src/dynamic.cpp b/src/dynamic.cpp
index cc7aeb0b99e0ddb86a58f2572d2ee26e47ba14c4..8e63b7d03ee0729738a9a12cd677087cbad969be 100644 (file)
--- a/src/dynamic.cpp
+++ b/src/dynamic.cpp
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
+ *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
  *                       E-mail:
  *                <brain@chatspike.net>
  *               <Craig@chatspike.net>
@@ -16,67 +16,163 @@
 
 using namespace std;
 
+#include "inspircd_config.h"
+#include "inspircd_io.h"
 #include "globals.h"
-#include <dlfcn.h>
 #include "dynamic.h"
+
+#ifndef STATIC_LINK
+#include <dlfcn.h>
+#else
+#include "modlist.h"
+#endif
+
 #include "inspstring.h"
 #include "helperfuncs.h"
+#include <unistd.h>
+#include <sys/types.h>
+#include <stdio.h>
 
-DLLManager::DLLManager(const char *fname)
+extern ServerConfig* Config;
+
+DLLManager::DLLManager(char *fname)
 {
-    // Try to open the library now and get any error message.
+       if (!strstr(fname,".so"))
+       {
+               err = "This doesn't look like a module file to me...";
+               return;
+       }
+#ifdef STATIC_LINK
+       this->staticname[0] = '\0';
+       log(DEBUG,"Loading core-compiled module '%s'",fname);
+       for (int j = 0; modsyms[j].name; j++)
+       {
+               log(DEBUG,"Check %s",modsyms[j].name);
+               if (!strcmp(modsyms[j].name,fname))
+               {
+                       log(DEBUG,"Found %s",fname);
+                       strlcpy(this->staticname,fname,1020);
+                       err = 0;
+                       return;
+               }
+       }
+       err = "Module is not statically compiled into the ircd";
+#else
+#ifdef IS_CYGWIN
+       // Cygwin behaviour is handled slightly differently
+       // With the advent of dynamic modules. Because Windows
+       // wont let you overwrite a file which is currently in
+       // Use, we can safely attempt to load the module from its
+       // Current location :)
+
+       h = dlopen(fname, RTLD_NOW );
+       err = (char*)dlerror();
+
+#else
+       // Copy the library to a temp location, this makes recompiles
+       // a little safer if the ircd is running at the time as the
+       // shared libraries are mmap()ed and not doing this causes
+       // segfaults.
+       FILE* x = fopen(fname,"rb");
+       if (!x)
+       {
+               err = "Module file not found or cannot access, game over man!";
+               return;
+       }
+       char tmpfile_template[255];
+       char buffer[65536];
+       snprintf(tmpfile_template, 255, "%s/inspircd_file.so.%d.XXXXXXXXXX",Config->TempDir,getpid());
+       int fd = mkstemp(tmpfile_template);
+       while (!feof(x))
+       {
+               int n = fread(buffer, 1, 65535, x);
+               if (n)
+                       write(fd,buffer,n);
+       }
        
-       h=dlopen( fname, RTLD_NOW );
-       err=dlerror();
+       // Try to open the library now and get any error message.
+       
+       h = dlopen(tmpfile_template, RTLD_NOW );
+       err = (char*)dlerror();
+       close(fd);
+       // We can delete the tempfile once it's loaded, leaving just the inode.
+       if (!Config->debugging)
+               unlink(tmpfile_template);
+#endif
+#endif
 }
 
 DLLManager::~DLLManager()
 {
+#ifndef STATIC_LINK
        // close the library if it isn't null
-       if (h!=0)
-       dlclose(h);
+       if (h != 0)
+       dlclose(h);
+#endif
 }
 
 
-bool DLLManager::GetSymbol(void **v, const char *sym_name)
+
+#ifdef STATIC_LINK
+
+bool DLLManager::GetSymbol(initfunc* &v, char *sym_name)
+{
+       log(DEBUG,"Symbol search...");
+       for (int j = 0; modsyms[j].name; j++)
+       {
+               if (!strcmp(this->staticname,modsyms[j].name))
+               {
+                       log(DEBUG,"Loading symbol...");
+                       v = modsyms[j].value;
+                       err = 0;
+                       return true;
+               }
+       }
+       err = "Module symbol missing from the core";
+       return false;
+}
+
+#else
+
+bool DLLManager::GetSymbol(void **v, char *sym_name)
 {
        // try extract a symbol from the library
        // get any error message is there is any
        
-       if( h!=0 )
+       if(h != 0)
        {
                *v = dlsym( h, sym_name );
-       err=dlerror();
-           if( err==0 )
-                 return true;
-       else
-         return false;
+               err = (char*)dlerror();
+               if( err == 0 )
+                       return true;
+               else
+                       return false;
        }
        else
        {       
-        return false;
+               return false;
        }
-       
 }
 
+#endif
 
-DLLFactoryBase::DLLFactoryBase(const char *fname, const char *factory) : DLLManager(fname)
+DLLFactoryBase::DLLFactoryBase(char *fname, char *factory) : DLLManager(fname)
 {
        // try get the factory function if there is no error yet
        
-       factory_func=0;
+       factory_func = 0;
        
-       if( LastError()==0 )
-       {               
-       GetSymbol( (void **)&factory_func, factory ? factory : "init_module" );
+       if(LastError() == 0)
+       {
+#ifdef STATIC_LINK
+               GetSymbol( factory_func, factory ? factory : (char*)"init_module" );
+#else
+               GetSymbol( (void **)&factory_func, factory ? factory : (char*)"init_module" );
+#endif
        }
-       
 }
 
 
 DLLFactoryBase::~DLLFactoryBase()
 {
 }
-
-
-