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_connflood.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: Connection throttle */

int conns = 0, throttled = 0;

class ModuleConnFlood : public Module
{
private:
        int seconds, maxconns, timeout, boot_wait;
        time_t first;
        std::string quitmsg;

        ConfigReader* conf;


public:
        ModuleConnFlood(InspIRCd* Me) : Module(Me)
        {

                InitConf();
                Implementation eventlist[] = { I_OnRehash, I_OnUserRegister };
                ServerInstance->Modules->Attach(eventlist, this, 2);
        }

        virtual ~ModuleConnFlood()
        {
        }

        virtual Version GetVersion()
        {
                return Version("$Id: m_connflood.cpp 10291 2008-08-25 20:35:51Z w00t $", VF_VENDOR,API_VERSION);
        }

        void InitConf()
        {
                /* read configuration variables */
                conf = new ConfigReader(ServerInstance);
                /* throttle configuration */
                seconds = conf->ReadInteger("connflood", "seconds", 0, true);
                maxconns = conf->ReadInteger("connflood", "maxconns", 0, true);
                timeout = conf->ReadInteger("connflood", "timeout", 0, true);
                quitmsg = conf->ReadValue("connflood", "quitmsg", 0);

                /* seconds to wait when the server just booted */
                boot_wait = conf->ReadInteger("connflood", "bootwait", 0, true);

                first = ServerInstance->Time();
        }

        virtual int OnUserRegister(User* user)
        {
                time_t next = ServerInstance->Time();

                if ((ServerInstance->startup_time + boot_wait) > next)
                        return 0;

                /* time difference between first and latest connection */
                time_t tdiff = next - first;

                /* increase connection count */
                conns++;

                if (throttled == 1)
                {
                        if (tdiff > seconds + timeout)
                        {
                                /* expire throttle */
                                throttled = 0;
                                ServerInstance->SNO->WriteToSnoMask('A', "Connection throttle deactivated");
                                return 0;
                        }

                        ServerInstance->Users->QuitUser(user, quitmsg);
                        return 1;
                }

                if (tdiff <= seconds)
                {
                        if (conns >= maxconns)
                        {
                                throttled = 1;
                                ServerInstance->SNO->WriteToSnoMask('A', "Connection throttle activated");
                                ServerInstance->Users->QuitUser(user, quitmsg);
                                return 1;
                        }
                }
                else
                {
                        conns = 1;
                        first = next;
                }
                return 0;
        }

        virtual void OnRehash(User* user, const std::string &parameter)
        {
                InitConf();
        }

};

MODULE_INIT(ModuleConnFlood)