Main Page | Namespace List | Class Hierarchy | Alphabetical List | Class List | Directories | File List | Namespace Members | Class Members | File Members

home » brain » inspircd-cvs » inspircd » src » modules

m_hostchange.cpp

Go to the documentation of this file.

/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon |
 *       +------------------------------------+
 *
 *  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 "inspircd.h"

/* $ModDesc: Provides masking of user hostnames in a different way to m_cloaking */

class Host : public classbase
{
 public:
        std::string action;
        std::string newhost;
        std::string ports;
};

typedef std::map<std::string,Host*> hostchanges_t;

class ModuleHostChange : public Module
{
 private:
        hostchanges_t hostchanges;
        std::string MySuffix;
        std::string MyPrefix;
        std::string MySeparator;

 public:
        ModuleHostChange(InspIRCd* Me)
                : Module(Me)
        {
                OnRehash(NULL,"");
                Implementation eventlist[] = { I_OnRehash, I_OnUserConnect };
                ServerInstance->Modules->Attach(eventlist, this, 2);
        }

        virtual ~ModuleHostChange()
        {
                for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)
                {
                        delete i->second;
                }
                hostchanges.clear();
        }

        void Prioritize()
        {
                Module* cloak = ServerInstance->Modules->Find("m_cloaking.so");
                ServerInstance->Modules->SetPriority(this, I_OnUserConnect, PRIO_AFTER, &cloak);
        }


        virtual void OnRehash(User* user, const std::string &parameter)
        {
                ConfigReader Conf(ServerInstance);
                MySuffix = Conf.ReadValue("host","suffix",0);
                MyPrefix = Conf.ReadValue("host","prefix","",0);
                MySeparator = Conf.ReadValue("host","separator",".",0);
                for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)
                {
                        delete i->second;
                }
                hostchanges.clear();
                for (int index = 0; index < Conf.Enumerate("hostchange"); index++)
                {
                        std::string mask = Conf.ReadValue("hostchange", "mask", index);
                        std::string ports = Conf.ReadValue("hosthange", "ports", index);
                        std::string action = Conf.ReadValue("hostchange", "action", index);
                        std::string newhost = Conf.ReadValue("hostchange", "value", index);
                        Host* x = new Host;
                        x->action = action;
                        x->ports = ports;
                        x->newhost = newhost;
                        hostchanges[mask] = x;
                }
        }

        virtual Version GetVersion()
        {
                // returns the version number of the module to be
                // listed in /MODULES
                return Version("$Id: m_hostchange.cpp 10291 2008-08-25 20:35:51Z w00t $", VF_VENDOR, API_VERSION);
        }

        virtual void OnUserConnect(User* user)
        {
                for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)
                {
                        if (((InspIRCd::MatchCIDR(user->MakeHost(), i->first)) || (InspIRCd::MatchCIDR(user->MakeHostIP(), i->first))))
                        {
                                Host* h = i->second;

                                if (!h->ports.empty())
                                {
                                        irc::portparser portrange(h->ports, false);
                                        long portno = -1;
                                        bool foundany = false;

                                        while ((portno = portrange.GetToken()))
                                                if (portno == user->GetPort())
                                                        foundany = true;

                                        if (!foundany)
                                                continue;
                                }

                                // host of new user matches a hostchange tag's mask
                                std::string newhost;
                                if (h->action == "set")
                                {
                                        newhost = h->newhost;
                                }
                                else if (h->action == "suffix")
                                {
                                        newhost = MySuffix;
                                }
                                else if (h->action == "addnick")
                                {
                                        // first take their nick and strip out non-dns, leaving just [A-Z0-9\-]
                                        std::string complete;
                                        std::string old = user->nick;
                                        for (unsigned int j = 0; j < old.length(); j++)
                                        {
                                                if  (((old[j] >= 'A') && (old[j] <= 'Z')) ||
                                                    ((old[j] >= 'a') && (old[j] <= 'z')) ||
                                                    ((old[j] >= '0') && (old[j] <= '9')) ||
                                                    (old[j] == '-'))
                                                {
                                                        complete = complete + old[j];
                                                }
                                        }
                                        if (complete.empty())
                                                complete = "i-have-a-lame-nick";

                                        if (!MyPrefix.empty())
                                                newhost = MyPrefix + MySeparator + complete;
                                        else
                                                newhost = complete + MySeparator + MySuffix;
                                }
                                if (!newhost.empty())
                                {
                                        user->WriteServ("NOTICE "+std::string(user->nick)+" :Setting your virtual host: " + newhost);
                                        if (!user->ChangeDisplayedHost(newhost.c_str()))
                                                user->WriteServ("NOTICE "+std::string(user->nick)+" :Could not set your virtual host: " + newhost);
                                        return;
                                }
                        }
                }
        }
};

MODULE_INIT(ModuleHostChange)