#include "inspircd.h"
#include "xline.h"
-/* $ModDesc: Throttles the connections of any users who try connect flood */
+/* $ModDesc: Throttles the connections of IP ranges who try to connect flood. */
class ModuleConnectBan : public Module
{
- private:
clonemap connects;
unsigned int threshold;
unsigned int banduration;
unsigned int ipv4_cidr;
unsigned int ipv6_cidr;
- public:
- ModuleConnectBan() {
- Implementation eventlist[] = { I_OnUserConnect, I_OnGarbageCollect, I_OnRehash };
- ServerInstance->Modules->Attach(eventlist, this, 3);
- OnRehash(NULL);
- }
- virtual ~ModuleConnectBan()
+ public:
+ void init() CXX11_OVERRIDE
{
+ Implementation eventlist[] = { I_OnSetUserIP, I_OnGarbageCollect, I_OnRehash };
+ ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
+ OnRehash(NULL);
}
- virtual Version GetVersion()
+ Version GetVersion() CXX11_OVERRIDE
{
- return Version("Throttles the connections of any users who try connect flood", VF_VENDOR);
+ return Version("Throttles the connections of IP ranges who try to connect flood.", VF_VENDOR);
}
- virtual void OnRehash(User* user)
+ void OnRehash(User* user) CXX11_OVERRIDE
{
- ConfigReader Conf;
- std::string duration;
+ ConfigTag* tag = ServerInstance->Config->ConfValue("connectban");
- ipv4_cidr = Conf.ReadInteger("connectban", "ipv4cidr", 0, true);
+ ipv4_cidr = tag->getInt("ipv4cidr", 32);
if (ipv4_cidr == 0)
ipv4_cidr = 32;
- ipv6_cidr = Conf.ReadInteger("connectban", "ipv6cidr", 0, true);
+ ipv6_cidr = tag->getInt("ipv6cidr", 128);
if (ipv6_cidr == 0)
ipv6_cidr = 128;
- threshold = Conf.ReadInteger("connectban", "threshold", 0, true);
-
+ threshold = tag->getInt("threshold", 10);
if (threshold == 0)
threshold = 10;
- duration = Conf.ReadValue("connectban", "duration", 0, true);
-
- if (duration.empty())
- duration = "10m";
-
- banduration = ServerInstance->Duration(duration);
+ banduration = InspIRCd::Duration(tag->getString("duration", "10m"));
+ if (banduration == 0)
+ banduration = 10*60;
}
- virtual void OnUserConnect(LocalUser *u)
+ void OnSetUserIP(LocalUser* u) CXX11_OVERRIDE
{
+ if (u->exempt)
+ return;
+
int range = 32;
clonemap::iterator i;
else
delete zl;
- ServerInstance->SNO->WriteGlobalSno('x',"Module m_connectban added Z:line on *@%s to expire on %s: Connect flooding",
- mask.str().c_str(), ServerInstance->TimeString(zl->expiry).c_str());
- ServerInstance->SNO->WriteGlobalSno('a', "Connect flooding from IP range %s (%d)", mask.str().c_str(), threshold);
+ std::string maskstr = mask.str();
+ std::string timestr = ServerInstance->TimeString(zl->expiry);
+ ServerInstance->SNO->WriteGlobalSno('x',"Module m_connectban added Z:line on *@%s to expire on %s: Connect flooding",
+ maskstr.c_str(), timestr.c_str());
+ ServerInstance->SNO->WriteGlobalSno('a', "Connect flooding from IP range %s (%d)", maskstr.c_str(), threshold);
connects.erase(i);
}
}
}
}
- virtual void OnGarbageCollect()
+ void OnGarbageCollect()
{
- ServerInstance->Logs->Log("m_connectban",DEBUG, "Clearing map.");
+ ServerInstance->Logs->Log("m_connectban", LOG_DEBUG, "Clearing map.");
connects.clear();
}
};