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
|
/* +------------------------------------+
* | 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"
#include "xline.h"
/* $ModDesc: Throttles the connections of any users who try connect flood */
class ModuleQuitBan : public Module
{
private:
clonemap connects;
unsigned int threshold;
unsigned int banduration;
public:
ModuleQuitBan(InspIRCd* Me) : Module(Me)
{
Implementation eventlist[] = { I_OnUserConnect, I_OnGarbageCollect, I_OnRehash };
ServerInstance->Modules->Attach(eventlist, this, 3);
OnRehash(NULL, "");
}
virtual ~ModuleQuitBan()
{
}
virtual Version GetVersion()
{
return Version(1,2,0,0,VF_VENDOR,API_VERSION);
}
virtual void OnRehash(User* user, const std::string ¶meter)
{
ConfigReader Conf(ServerInstance);
std::string duration;
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(User *u)
{
clonemap::iterator i = connects.find(u->GetIPString());
if (i != connects.end())
{
i->second++;
ServerInstance->Logs->Log("m_connectban",DEBUG, "Count for IP is now %d", i->second);
if (i->second >= threshold)
{
// Create zline for set duration.
ZLine* zl = new ZLine(ServerInstance, ServerInstance->Time(), banduration, ServerInstance->Config->ServerName, "Connect flooding", u->GetIPString());
if (ServerInstance->XLines->AddLine(zl,NULL))
ServerInstance->XLines->ApplyLines();
else
delete zl;
ServerInstance->SNO->WriteToSnoMask('x', "Connect flooding from IP %s (%d)", u->GetIPString(), threshold);
connects.erase(i);
}
}
else
{
connects[u->GetIPString()] = 1;
ServerInstance->Logs->Log("m_quitban",DEBUG, "Added new record");
}
}
virtual void OnGarbageCollect()
{
ServerInstance->Logs->Log("m_quitban",DEBUG, "Clearing map.");
connects.clear();
}
};
MODULE_INIT(ModuleQuitBan)
|
