]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/dynamic.cpp
Merge pull request #1157 from SaberUK/insp20+fix-cron-restart
[user/henk/code/inspircd.git] / src / dynamic.cpp
index a4542760b024a52f346be6b3fcdc2d36996cebb1..3a6a151cb5d047db124b4caca75d1a8063c0a1bf 100644 (file)
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
- * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2007 Oliver Lupton <oliverlupton@gmail.com>
+ *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
+ *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *   Copyright (C) 2003, 2006 Craig Edwards <craigedwards@brainbox.cc>
  *
- * ---------------------------------------------------
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-using namespace std;
 
-#include "inspircd_config.h"
-#include "globals.h"
+#include "inspircd.h"
 #include "dynamic.h"
-
-#ifndef STATIC_LINK
+#ifndef _WIN32
 #include <dlfcn.h>
 #else
-#include "modlist.h"
+#define dlopen(path, state) (void*)LoadLibraryA(path)
+#define dlsym(handle, export) (void*)GetProcAddress((HMODULE)handle, export)
+#define dlclose(handle) FreeLibrary((HMODULE)handle)
 #endif
 
-#include "inspstring.h"
-#include "helperfuncs.h"
-#include <unistd.h>
-#include <sys/types.h>
-#include <stdio.h>
-
-DLLManager::DLLManager(char *fname)
+DLLManager::DLLManager(const char *fname)
 {
-#ifdef STATIC_LINK
-       this->staticname[0] = '\0';
-       log(DEBUG,"Loading core-compiled module '%s'",fname);
-       for (int j = 0; modsyms[j].name; j++)
+       if (!strstr(fname,".so"))
        {
-               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 = "This doesn't look like a module file to me...";
+               h = NULL;
+               return;
        }
-       err = "Module is not statically compiled into the ircd";
-#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");
-       char tmpfile_template[255];
-       char buffer[65536];
-       snprintf(tmpfile_template, 255, "/tmp/inspircd_file.so.%d.XXXXXXXXXX",getpid());
-       int fd = mkstemp(tmpfile_template);
-       while (!feof(x))
+
+       h = dlopen(fname, RTLD_NOW|RTLD_LOCAL);
+       if (!h)
        {
-               int n = fread(buffer, 1, 65535, x);
-               if (n)
-                       write(fd,buffer,n);
-       }
-       
-       // Try to open the library now and get any error message.
-       
-       h = dlopen(tmpfile_template, RTLD_NOW );
-       err = (char*)dlerror();
-       close(fd);
+#ifdef _WIN32
+               RetrieveLastError();
+#else
+               err = dlerror();
 #endif
+       }
 }
 
 DLLManager::~DLLManager()
 {
-#ifndef STATIC_LINK
-       // close the library if it isn't null
-       if (h != 0)
-       dlclose(h);
-#endif
+       /* close the library */
+       if (h)
+               dlclose(h);
 }
 
+union init_t {
+       void* vptr;
+       Module* (*fptr)();
+};
 
-
-#ifdef STATIC_LINK
-
-bool DLLManager::GetSymbol(initfunc* &v, char *sym_name)
+Module* DLLManager::CallInit()
 {
-       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;
-}
+       if (!h)
+               return NULL;
 
-#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)
+       init_t initfn;
+       initfn.vptr = dlsym(h, MODULE_INIT_STR);
+       if (!initfn.vptr)
        {
-               *v = dlsym( h, sym_name );
-               err = (char*)dlerror();
-               if( err == 0 )
-                       return true;
-               else
-                       return false;
-       }
-       else
-       {       
-               return false;
-       }
-}
-
-#endif
-
-DLLFactoryBase::DLLFactoryBase(char *fname, char *factory) : DLLManager(fname)
-{
-       // try get the factory function if there is no error yet
-       
-       factory_func = 0;
-       
-       if(LastError() == 0)
-       {
-#ifdef STATIC_LINK
-               GetSymbol( factory_func, factory ? factory : (char*)"init_module" );
+#ifdef _WIN32
+               RetrieveLastError();
 #else
-               GetSymbol( (void **)&factory_func, factory ? factory : (char*)"init_module" );
+               err = dlerror();
 #endif
+               return NULL;
        }
+
+       return (*initfn.fptr)();
 }
 
+std::string DLLManager::GetVersion()
+{
+       if (!h)
+               return "";
 
-DLLFactoryBase::~DLLFactoryBase()
+       const char* srcver = (char*)dlsym(h, "inspircd_src_version");
+       if (srcver)
+               return srcver;
+       return "Unversioned module";
+}
+
+#ifdef _WIN32
+void DLLManager::RetrieveLastError()
 {
+       char errmsg[500];
+       DWORD dwErrorCode = GetLastError();
+       if (FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, dwErrorCode, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPSTR)errmsg, _countof(errmsg), NULL) == 0)
+               sprintf_s(errmsg, _countof(errmsg), "Error code: %u", dwErrorCode);
+       SetLastError(ERROR_SUCCESS);
+       err = errmsg;
+
+       std::string::size_type p;
+       while ((p = err.find_last_of("\r\n")) != std::string::npos)
+               err.erase(p, 1);
 }
+#endif