core_hostname_lookup: find answer record of the correct type instead of assuming...

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

diff --git a/src/modules/m_vhost.cpp b/src/modules/m_vhost.cpp
index 238de02e0ea6679119bef1851e58b52b6e096aca..53910fdbe5c6f65e14f82fd76feb7ecda9fe7b5f 100644 (file)
--- a/src/modules/m_vhost.cpp
+++ b/src/modules/m_vhost.cpp
@@ -1,124 +1,77 @@
-/*       +------------------------------------+
- *       | 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 Dennis Friis <peavey@inspircd.org>
+ *   Copyright (C) 2006-2007 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2006 Robin Burchell <robin+git@viroteck.net>
  *
- * ---------------------------------------------------
+ * 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 <stdio.h>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-
-/* $ModDesc: Provides masking of user hostnames via traditional /VHOST command */
-
-ConfigReader *Conf;
-Server* Srv;
-
-class cmd_vhost : public command_t
-{
- public:
-        cmd_vhost() : command_t("VHOST", 0, 2)
-        {
-                this->source = "m_vhost.so";
-        }
 
-        void Handle (char **parameters, int pcnt, userrec *user)
-        {
-                for (int index = 0; index < Conf->Enumerate("vhost"); index++)
-                {
-                        std::string mask = Conf->ReadValue("vhost","host",index);
-                       std::string user = Conf->ReadValue("vhost","user",index);
-                       std::string pass = Conf->ReadValue("vhost","pass",index);
-                        if ((!strcmp(parameters[0],user.c_str())) && (!strcmp(parameters[1],pass.c_str())))
-                        {
-                                if (mask != "")
-                                {
-                                        Srv->SendServ(user->fd,"NOTICE "+std::string(user->nick)+" :Setting your VHost: " + mask);
-                                        Srv->ChangeHost(user,mask);
-                                        return;
-                                }
-                        }
-                }
-               Srv->SendServ(user->fd,"NOTICE "+std::string(user->nick)+" :Invalid username or password.");
-        }
-};
+#include "inspircd.h"
 
-class ModuleVHost : public Module
+/** Handle /VHOST
+ */
+class CommandVhost : public Command
 {
- private:
-
-       cmd_vhost* mycommand;
-        
  public:
-       ModuleVHost(Server* Me)
-               : Module::Module(Me)
-       {
-               Srv = Me;
-                Conf = new ConfigReader;
-                mycommand = new cmd_vhost();
-                Srv->AddCommand(mycommand);
-       }
-       
-       virtual ~ModuleVHost()
+       CommandVhost(Module* Creator) : Command(Creator,"VHOST", 2)
        {
-               delete Conf;
+               syntax = "<username> <password>";
        }
 
-       void Implements(char* List)
+       CmdResult Handle (const std::vector<std::string> &parameters, User *user)
        {
-               List[I_OnRehash] = 1;
-       }
+               ConfigTagList tags = ServerInstance->Config->ConfTags("vhost");
+               for(ConfigIter i = tags.first; i != tags.second; ++i)
+               {
+                       ConfigTag* tag = i->second;
+                       std::string mask = tag->getString("host");
+                       std::string username = tag->getString("user");
+                       std::string pass = tag->getString("pass");
+                       std::string hash = tag->getString("hash");
 
-       virtual void OnRehash(std::string parameter)
-       {
-               delete Conf;
-               Conf = new ConfigReader;
-       }
-       
-       virtual Version GetVersion()
-       {
-               // returns the version number of the module to be
-               // listed in /MODULES
-               return Version(1,0,0,1,VF_VENDOR);
+                       if (parameters[0] == username && ServerInstance->PassCompare(user, pass, parameters[1], hash))
+                       {
+                               if (!mask.empty())
+                               {
+                                       user->WriteNotice("Setting your VHost: " + mask);
+                                       user->ChangeDisplayedHost(mask);
+                                       return CMD_SUCCESS;
+                               }
+                       }
+               }
+
+               user->WriteNotice("Invalid username or password.");
+               return CMD_FAILURE;
        }
-       
 };
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
-
-class ModuleVHostFactory : public ModuleFactory
+class ModuleVHost : public Module
 {
+       CommandVhost cmd;
+
  public:
-       ModuleVHostFactory()
-       {
-       }
-       
-       ~ModuleVHostFactory()
+       ModuleVHost() : cmd(this)
        {
        }
-       
-       virtual Module * CreateModule(Server* Me)
+
+       Version GetVersion() CXX11_OVERRIDE
        {
-               return new ModuleVHost(Me);
+               return Version("Provides masking of user hostnames via traditional /VHOST command",VF_VENDOR);
        }
-       
 };
 
-
-extern "C" void * init_module( void )
-{
-       return new ModuleVHostFactory;
-}
-
+MODULE_INIT(ModuleVHost)