Merge pull request #1094 from SISheogorath/insp20+fixed-Override

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

diff --git a/src/dynamic.cpp b/src/dynamic.cpp
index 5d7759b4751761daefcff49ad083143db720867f..3a6a151cb5d047db124b4caca75d1a8063c0a1bf 100644 (file)
--- a/src/dynamic.cpp
+++ b/src/dynamic.cpp
@@ -1 +1,111 @@
-/*       +------------------------------------+\r *       | Inspire Internet Relay Chat Daemon |\r *       +------------------------------------+\r *\r *  InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r *            the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h"\r#include "configreader.h"\r#include "dynamic.h"\r#ifndef WIN32\r#include <dlfcn.h>\r#endif\r\rDLLManager::DLLManager(InspIRCd* ServerInstance, const char *fname)\r{\r       err = NULL;\r\r   if (!strstr(fname,".so"))\r      {\r              err = "This doesn't look like a module file to me...";\r         return;\r        }\r\r     h = dlopen(fname, RTLD_NOW|RTLD_LOCAL);\r        if (!h)\r        {\r              err = (char*)dlerror();\r                return;\r        }\r}\r\rDLLManager::~DLLManager()\r{\r       // close the library if it isn't null\r  if (h)\r         dlclose(h);\r}\r\r\r\rbool DLLManager::GetSymbol(void** v, const char* sym_name)\r{\r  // try extract a symbol from the library\r       // get any error message is there is any\r       \r       if (h)\r {\r              dlerror(); // clear value\r              *v = dlsym(h, sym_name);\r               err = (char*)dlerror();\r                if (!*v || err)\r                        return false;\r  }\r      \r       if (err)\r       {\r              return false;\r  }\r      else\r   {       \r               return true;\r   }\r}\r\rDLLFactoryBase::DLLFactoryBase(InspIRCd* Instance, const char* fname, const char* symbol) : DLLManager(Instance, fname)\r{\r // try get the factory function if there is no error yet\r       factory_func = 0;\r      \r       if (!LastError())\r      {\r              if (!GetSymbol( (void **)&factory_func, symbol ? symbol : "init_module"))\r              {\r                      throw ModuleException("Missing init_module() entrypoint!");\r            }\r      }\r}\r\rDLLFactoryBase::~DLLFactoryBase()\r{\r}\r\r
\ No newline at end of file
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ *   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/>.
+ */
+
+
+#include "inspircd.h"
+#include "dynamic.h"
+#ifndef _WIN32
+#include <dlfcn.h>
+#else
+#define dlopen(path, state) (void*)LoadLibraryA(path)
+#define dlsym(handle, export) (void*)GetProcAddress((HMODULE)handle, export)
+#define dlclose(handle) FreeLibrary((HMODULE)handle)
+#endif
+
+DLLManager::DLLManager(const char *fname)
+{
+       if (!strstr(fname,".so"))
+       {
+               err = "This doesn't look like a module file to me...";
+               h = NULL;
+               return;
+       }
+
+       h = dlopen(fname, RTLD_NOW|RTLD_LOCAL);
+       if (!h)
+       {
+#ifdef _WIN32
+               RetrieveLastError();
+#else
+               err = dlerror();
+#endif
+       }
+}
+
+DLLManager::~DLLManager()
+{
+       /* close the library */
+       if (h)
+               dlclose(h);
+}
+
+union init_t {
+       void* vptr;
+       Module* (*fptr)();
+};
+
+Module* DLLManager::CallInit()
+{
+       if (!h)
+               return NULL;
+
+       init_t initfn;
+       initfn.vptr = dlsym(h, MODULE_INIT_STR);
+       if (!initfn.vptr)
+       {
+#ifdef _WIN32
+               RetrieveLastError();
+#else
+               err = dlerror();
+#endif
+               return NULL;
+       }
+
+       return (*initfn.fptr)();
+}
+
+std::string DLLManager::GetVersion()
+{
+       if (!h)
+               return "";
+
+       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