src/modules/m_sethost.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   5  *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
   6  *   Copyright (C) 2004-2006 Craig Edwards <craigedwards@brainbox.cc>
   7  *
   8  * This file is part of InspIRCd.  InspIRCd is free software: you can
   9  * redistribute it and/or modify it under the terms of the GNU General Public
  10  * License as published by the Free Software Foundation, version 2.
  11  *
  12  * This program is distributed in the hope that it will be useful, but WITHOUT
  13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  15  * details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21
  22 #include "inspircd.h"
  23
  24 /* $ModDesc: Provides support for the SETHOST command */
  25
  26 /** Handle /SETHOST
  27  */
  28 class CommandSethost : public Command
  29 {
  30  private:
  31         char* hostmap;
  32  public:
  33         CommandSethost(Module* Creator, char* hmap) : Command(Creator,"SETHOST", 1), hostmap(hmap)
  34         {
  35                 flags_needed = 'o'; syntax = "<new-hostname>";
  36                 TRANSLATE2(TR_TEXT, TR_END);
  37         }
  38
  39         CmdResult Handle (const std::vector<std::string>& parameters, User *user)
  40         {
  41                 size_t len = 0;
  42                 for (std::string::const_iterator x = parameters[0].begin(); x != parameters[0].end(); x++, len++)
  43                 {
  44                         if (!hostmap[(const unsigned char)*x])
  45                         {
  46                                 user->WriteServ("NOTICE "+user->nick+" :*** SETHOST: Invalid characters in hostname");
  47                                 return CMD_FAILURE;
  48                         }
  49                 }
  50                 if (len == 0)
  51                 {
  52                         user->WriteServ("NOTICE %s :*** SETHOST: Host must be specified", user->nick.c_str());
  53                         return CMD_FAILURE;
  54                 }
  55                 if (len > 64)
  56                 {
  57                         user->WriteServ("NOTICE %s :*** SETHOST: Host too long",user->nick.c_str());
  58                         return CMD_FAILURE;
  59                 }
  60
  61                 if (user->ChangeDisplayedHost(parameters[0].c_str()))
  62                 {
  63                         ServerInstance->SNO->WriteGlobalSno('a', user->nick+" used SETHOST to change their displayed host to "+user->dhost);
  64                         return CMD_SUCCESS;
  65                 }
  66
  67                 return CMD_FAILURE;
  68         }
  69 };
  70
  71
  72 class ModuleSetHost : public Module
  73 {
  74         CommandSethost cmd;
  75         char hostmap[256];
  76  public:
  77         ModuleSetHost()
  78                 : cmd(this, hostmap)
  79         {
  80         }
  81
  82         void init()
  83         {
  84                 OnRehash(NULL);
  85                 ServerInstance->AddCommand(&cmd);
  86                 Implementation eventlist[] = { I_OnRehash };
  87                 ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
  88         }
  89
  90         void OnRehash(User* user)
  91         {
  92                 std::string hmap = ServerInstance->Config->ConfValue("hostname")->getString("charmap", "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz.-_/0123456789");
  93
  94                 memset(hostmap, 0, sizeof(hostmap));
  95                 for (std::string::iterator n = hmap.begin(); n != hmap.end(); n++)
  96                         hostmap[(unsigned char)*n] = 1;
  97         }
  98
  99         virtual ~ModuleSetHost()
 100         {
 101         }
 102
 103         virtual Version GetVersion()
 104         {
 105                 return Version("Provides support for the SETHOST command", VF_VENDOR);
 106         }
 107
 108 };
 109
 110 MODULE_INIT(ModuleSetHost)