1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/* +------------------------------------+
* | Inspire Internet Relay Chat Daemon |
* +------------------------------------+
*
* InspIRCd: (C) 2002-2010 InspIRCd Development Team
* See: http://wiki.inspircd.org/Credits
*
* This program is free but copyrighted software; see
* the file COPYING for details.
*
* ---------------------------------------------------
*/
#include "inspircd.h"
#include "xline.h"
/* $ModDesc: Throttles the connections of any users who try 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()
{
}
virtual Version GetVersion()
{
return Version("Throttles the connections of any users who try connect flood", VF_VENDOR);
}
virtual void OnRehash(User* user)
{
ConfigReader Conf;
std::string duration;
ipv4_cidr = Conf.ReadInteger("connectban", "ipv4cidr", 0, true);
if (ipv4_cidr == 0)
ipv4_cidr = 32;
ipv6_cidr = Conf.ReadInteger("connectban", "ipv6cidr", 0, true);
if (ipv6_cidr == 0)
ipv6_cidr = 128;
threshold = Conf.ReadInteger("connectban", "threshold", 0, true);
if (threshold == 0)
threshold = 10;
duration = Conf.ReadValue("connectban", "duration", 0, true);
if (duration.empty())
duration = "10m";
banduration = ServerInstance->Duration(duration);
}
virtual void OnUserConnect(LocalUser *u)
{
int range = 32;
clonemap::iterator i;
switch (u->client_sa.sa.sa_family)
{
case AF_INET6:
range = ipv6_cidr;
break;
case AF_INET:
range = ipv4_cidr;
break;
}
irc::sockets::cidr_mask mask(u->client_sa, range);
i = connects.find(mask);
if (i != connects.end())
{
i->second++;
if (i->second >= threshold)
{
// Create zline for set duration.
ZLine* zl = new ZLine(ServerInstance->Time(), banduration, ServerInstance->Config->ServerName.c_str(), "Connect flooding", mask.str().c_str());
if (ServerInstance->XLines->AddLine(zl,NULL))
ServerInstance->XLines->ApplyLines();
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);
connects.erase(i);
}
}
else
{
connects[mask] = 1;
}
}
virtual void OnGarbageCollect()
{
ServerInstance->Logs->Log("m_connectban",DEBUG, "Clearing map.");
connects.clear();
}
};
MODULE_INIT(ModuleConnectBan)
|
