Convert more modules

[user/henk/code/inspircd.git] / src / modules / m_sethost.cpp

diff --git a/src/modules/m_sethost.cpp b/src/modules/m_sethost.cpp
index 6787867577d38ba997053e7d0f4d53bce81b480c..1cacef956edb8f785c2ed386c427dccffb43e83a 100644 (file)
--- a/src/modules/m_sethost.cpp
+++ b/src/modules/m_sethost.cpp
@@ -2,37 +2,30 @@
  *       | Inspire 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.
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include <stdio.h>
-#include <string>
+#include "inspircd.h"
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
 
-#include "inspircd.h"
-
 /* $ModDesc: Provides support for the SETHOST command */
 
-
-
-
+/** Handle /SETHOST
+ */
 class cmd_sethost : public command_t
 {
+ private:
+       char* hostmap;
  public:
-       cmd_sethost (InspIRCd* Instance) : command_t(Instance,"SETHOST",'o',1)
+       cmd_sethost (InspIRCd* Instance, char* hmap) : command_t(Instance,"SETHOST",'o',1), hostmap(hmap)
        {
                this->source = "m_sethost.so";
                syntax = "<new-hostname>";
@@ -43,13 +36,10 @@ class cmd_sethost : public command_t
                size_t len = 0;
                for (const char* x = parameters[0]; *x; x++, len++)
                {
-                       if (((tolower(*x) < 'a') || (tolower(*x) > 'z')) && (*x != '.'))
+                       if (!hostmap[(unsigned char)*x])
                        {
-                               if (((*x < '0') || (*x> '9')) && (*x != '-'))
-                               {
-                                       user->WriteServ("NOTICE "+std::string(user->nick)+" :*** Invalid characters in hostname");
-                                       return CMD_FAILURE;
-                               }
+                               user->WriteServ("NOTICE "+std::string(user->nick)+" :*** Invalid characters in hostname");
+                               return CMD_FAILURE;
                        }
                }
                if (len > 64)
@@ -70,29 +60,46 @@ class cmd_sethost : public command_t
 
 class ModuleSetHost : public Module
 {
-       cmd_sethost*    mycommand;
+       cmd_sethost* mycommand;
+       char hostmap[256];
  public:
        ModuleSetHost(InspIRCd* Me)
-               : Module::Module(Me)
-       {
-               
-               mycommand = new cmd_sethost(ServerInstance);
+               : Module(Me)
+       {       
+               OnRehash(NULL,"");
+               mycommand = new cmd_sethost(ServerInstance, hostmap);
                ServerInstance->AddCommand(mycommand);
        }
-       
+
+       void Implements(char* List)
+       {
+               List[I_OnRehash] = 1;
+       }
+
+       void OnRehash(userrec* user, const std::string &parameter)
+       {
+               ConfigReader Conf(ServerInstance);
+               std::string hmap = Conf.ReadValue("hostname", "charmap", 0);
+
+               if (hmap.empty())
+                       hmap = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz.-_/0123456789";
+
+               memset(&hostmap, 0, 255);
+               for (std::string::iterator n = hmap.begin(); n != hmap.end(); n++)
+                       hostmap[(unsigned char)*n] = 1;
+       }
+
        virtual ~ModuleSetHost()
        {
        }
        
        virtual Version GetVersion()
        {
-               return Version(1,0,0,1,VF_VENDOR,API_VERSION);
+               return Version(1,1,0,1,VF_VENDOR,API_VERSION);
        }
        
 };
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
-
 class ModuleSetHostFactory : public ModuleFactory
 {
  public:
@@ -112,7 +119,7 @@ class ModuleSetHostFactory : public ModuleFactory
 };
 
 
-extern "C" void * init_module( void )
+extern "C" DllExport void * init_module( void )
 {
        return new ModuleSetHostFactory;
 }