fix some unitialised vectors and tidy up a bit.

[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 f6a9bbf3f190340f87a87f62585b95024cfff021..cdcbbde9e6b98a28a280aa92962b0a775f05e231 100644 (file)
--- a/src/modules/m_sethost.cpp
+++ b/src/modules/m_sethost.cpp
@@ -1,86 +1,104 @@
-/*
- *  SETHOST module for InspIRCD
- *  Author: brain
- *  Version: 1.0.0.0
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
  *
- *  Syntax: /SETHOST [new host]
- *  Changes the user's DHOST who issues the command
- *  (oper only)
- *  
+ *  InspIRCd: (C) 2002-2008 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.
+ *
+ * ---------------------------------------------------
  */
 
-#include <stdio.h>
-#include <string>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
+#include "inspircd.h"
 
 /* $ModDesc: Provides support for the SETHOST command */
 
-Server *Srv;
-        
-void handle_sethost(char **parameters, int pcnt, userrec *user)
+/** Handle /SETHOST
+ */
+class CommandSethost : public Command
 {
-       for (int x = 0; x < strlen(parameters[0]); x++)
+ private:
+       char* hostmap;
+ public:
+       CommandSethost (InspIRCd* Instance, char* hmap) : Command(Instance,"SETHOST","o",1), hostmap(hmap)
+       {
+               this->source = "m_sethost.so";
+               syntax = "<new-hostname>";
+               TRANSLATE2(TR_TEXT, TR_END);
+       }
+
+       CmdResult Handle (const std::vector<std::string>& parameters, User *user)
        {
-               if (((tolower(parameters[0][x]) < 'a') || (tolower(parameters[0][x]) > 'z')) && (parameters[0][x] != '.'))
+               size_t len = 0;
+               for (std::string::const_iterator x = parameters[0].begin(); x != parameters[0].end(); x++, len++)
                {
-                       if (((parameters[0][x] < '0') || (parameters[0][x]> '9')) && (parameters[0][x] != '-'))
+                       if (!hostmap[(const unsigned char)*x])
                        {
-                               Srv->SendTo(NULL,user,"NOTICE "+std::string(user->nick)+" :*** Invalid characters in hostname");
-                               return;
+                               user->WriteServ("NOTICE "+std::string(user->nick)+" :*** SETHOST: Invalid characters in hostname");
+                               return CMD_FAILURE;
                        }
                }
+               if (len == 0)
+               {
+                       user->WriteServ("NOTICE %s :*** SETHOST: Host must be specified", user->nick);
+                       return CMD_FAILURE;
+               }
+               if (len > 64)
+               {
+                       user->WriteServ("NOTICE %s :*** SETHOST: Host too long",user->nick);
+                       return CMD_FAILURE;
+               }
+               if (user->ChangeDisplayedHost(parameters[0].c_str()))
+               {
+                       ServerInstance->SNO->WriteToSnoMask('A', std::string(user->nick)+" used SETHOST to change their displayed host to "+user->dhost);
+                       return CMD_SUCCESS;
+               }
+
+               return CMD_FAILURE;
        }
-       Srv->ChangeHost(user,parameters[0]);
-       Srv->SendOpers(std::string(user->nick)+" used SETHOST to change their displayed host to "+std::string(parameters[0]));
-}
+};
 
 
 class ModuleSetHost : public Module
 {
+       CommandSethost* mycommand;
+       char hostmap[256];
  public:
-       ModuleSetHost()
-       {
-               Srv = new Server;
-               Srv->AddCommand("SETHOST",handle_sethost,'o',1);
+       ModuleSetHost(InspIRCd* Me)
+               : Module(Me)
+       {       
+               OnRehash(NULL,"");
+               mycommand = new CommandSethost(ServerInstance, hostmap);
+               ServerInstance->AddCommand(mycommand);
+               Implementation eventlist[] = { I_OnRehash };
+               ServerInstance->Modules->Attach(eventlist, this, 1);
        }
-       
-       virtual ~ModuleSetHost()
-       {
-               delete Srv;
-       }
-       
-       virtual Version GetVersion()
-       {
-               return Version(1,0,0,0);
-       }
-       
-};
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
 
-class ModuleSetHostFactory : public ModuleFactory
-{
- public:
-       ModuleSetHostFactory()
+       void OnRehash(User* 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;
        }
-       
-       ~ModuleSetHostFactory()
+
+       virtual ~ModuleSetHost()
        {
        }
        
-       virtual Module * CreateModule()
+       virtual Version GetVersion()
        {
-               return new ModuleSetHost;
+               return Version(1, 2, 0, 1, VF_COMMON | VF_VENDOR, API_VERSION);
        }
        
 };
 
-
-extern "C" void * init_module( void )
-{
-       return new ModuleSetHostFactory;
-}
-
+MODULE_INIT(ModuleSetHost)