summaryrefslogtreecommitdiff
path: root/include/u_listmode.h
blob: a41f30e68e7356cfad423d77480e38a0077edf7f (plain)
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
#ifndef INSPIRCD_LISTMODE_PROVIDER
#define INSPIRCD_LISTMODE_PROVIDER

#include <stdio.h>
#include <string>
#include <sstream>
#include <vector>
#include "users.h"
#include "channels.h"
#include "modules.h"
#include "helperfuncs.h"

/* $ModDesc: Provides support for easily creating listmodes, stores the time set, the user, and a parameter. */

/* Updated to use the <banlist> config tag if it exists */
/* Written by Om <omster@gmail.com>, December 2005. */
/* Based on code previously written by Om - April 2005 */
/* Originally based on m_chanprotect and m_silence */

inline std::string stringtime()
{
	std::ostringstream TIME;
	TIME << time(NULL); 
	return TIME.str();
}

class ListItem
{
public:
	std::string nick;
	std::string mask;
	std::string time;
};

class ListLimit
{
public:
	std::string mask;
	unsigned int limit;
};

// Just defining the type we use for the exception list here...
typedef std::vector<ListItem> modelist;
typedef std::vector<ListLimit> limitlist;

class ListModeBaseModule : public Module
{
protected:
	char mode;
	std::string infokey;
	std::string listnumeric;
	std::string endoflistnumeric;
	std::string endofliststring;
	limitlist chanlimits;

	Server* Srv;
	ConfigReader* Conf;
public:
	ListModeBaseModule(Server* serv, char modechar, const std::string &eolstr, const std::string &lnum, const std::string &eolnum) : Module::Module(serv)
	{
		Srv = serv;
		Conf = new ConfigReader;
		mode = modechar;
		listnumeric = lnum;
		endoflistnumeric = eolnum;
		endofliststring = eolstr;
		
		OnRehash("");
		infokey = "exceptionbase_mode_" + std::string(1, mode) + "_list";
		Srv->AddExtendedListMode(modechar);		
	}
	
	virtual void OnRehash(const std::string &param)
	{
		delete Conf;
		Conf = new ConfigReader;
		
		chanlimits.clear();
		
		for(int i = 0; i < Conf->Enumerate("banlist"); i++)
		{
			// For each <banlist> tag
			ListLimit limit;
			limit.mask = Conf->ReadValue("banlist", "chan", i);
			limit.limit = Conf->ReadInteger("banlist", "limit", i, true);
			
			if(limit.mask.size() && limit.limit > 0)
			{
				chanlimits.push_back(limit);
				log(DEBUG, "m_exceptionbase.so: Read channel listmode limit of %u for mask '%s'", limit.limit, limit.mask.c_str());
			}
			else
			{
				log(DEBUG, "m_exceptionbase.so: Invalid tag");
			}
		}
		
		if(chanlimits.size() == 0)
		{
			ListLimit limit;
			limit.mask = "*";
			limit.limit = 64;
			chanlimits.push_back(limit);
		}
	}
	
	void DoImplements(char* List)
	{
		List[I_OnExtendedMode] = List[I_OnSendList] = List[I_OnChannelDelete] = List[I_OnSyncChannel] = List[I_OnCleanup] = List[I_OnRehash] = 1;
	}
	
	virtual int OnExtendedMode(userrec *user, void *target, char modechar, int type, bool mode_on, string_list &params)
	{
		// First, check it's our mode
		if ((modechar == mode) && (type == MT_CHANNEL))
		{
			Srv->Log(DEBUG, "m_exceptionbase.so: General listmode handler called, handling mode '" + std::string(1, mode) + "'");
			chanrec* chan = (chanrec*)target;

			// Try and grab the list
			modelist* el = (modelist*)chan->GetExt(infokey);

			if(mode_on)
			{
				// If there was no list
				if(!el)
				{
					// Make one
					Srv->Log(DEBUG, "m_exceptionbase.so: Creating new list");
					el = new modelist;
					chan->Extend(infokey, (char*)el);
				}
				
				if(!Srv->IsValidMask(params[0]))
				{	
					Srv->Log(DEBUG, "m_exceptionbase.so: Banmask was invalid, returning -1");
					return -1;
				}
					
				for (modelist::iterator it = el->begin(); it != el->end(); it++)
				{
					Srv->Log(DEBUG, "m_exceptionbase.so: Iterating over exception list, current mask: " + it->mask);
					if(params[0] == it->mask)
					{
						Srv->Log(DEBUG, "m_exceptionbase.so: Someone tried to set an exception which was already set, returning -1");
						return -1;
					}
				}
				
				unsigned int maxsize = 0;
				
				for(limitlist::iterator it = chanlimits.begin(); it != chanlimits.end(); it++)
				{
					if(Srv->MatchText(chan->name, it->mask))
					{
						// We have a pattern matching the channel...
						maxsize = el->size();
						if(maxsize < it->limit)
						{
							// And now add the mask onto the list...
							ListItem e;
							e.mask = params[0];
							e.nick = user->nick;
							e.time = stringtime();
				
							Srv->Log(DEBUG, "m_exceptionbase.so: All checks passed, adding exception mask to list and returning 1");
							el->push_back(e);
							return 1;
						}
					}
				}

				// List is full
				WriteServ(user->fd, "478 %s %s %s :Channel ban/ignore list is full", user->nick, chan->name, params[0].c_str());
				log(DEBUG, "m_exceptionbase.so: %s tried to set mask %s on %s but the list is full (max %d)", user->nick, params[0].c_str(), chan->name, maxsize);
				return -1;
			}
			else
 			{
				// We're taking the mode off
				if(el)
				{
					for (modelist::iterator it = el->begin(); it != el->end(); it++)
					{
						Srv->Log(DEBUG, "m_exceptionbase.so: Removing mode, iterating over exception list, current mask: " + it->mask);
						if(params[0] == it->mask)
						{
							Srv->Log(DEBUG, "m_exceptionbase.so: Found match for removal of exception, removing and returning 1");
							el->erase(it);
							if(el->size() == 0)
							{
								Srv->Log(DEBUG, "m_exceptionbase.so: Erased the last entry on the exception list, removing the list");
								chan->Shrink(infokey);
								delete el;
							}
							return 1;
						}
					}
					Srv->Log(DEBUG, "m_exceptionbase.so: No match found for attempted removing of exception, returning -1");
					return -1;
				}
				else
				{
					// Hmm, taking an exception off a non-existant list, DIE
					Srv->Log(DEBUG, "m_exceptionbase.so: Attempted removal of an exception, when there was no exception list created, returning -1");
					return -1;
				}
			}
		}
		
		return 0;
	}
	
	virtual void OnSendList(userrec* user, chanrec* chan, char modechar)
	{
		if(modechar == mode)
		{
			modelist* el = (modelist*)chan->GetExt(infokey);
			Srv->Log(DEBUG, "m_exceptionbase.so: " + std::string(user->nick)+" is listing listmodes on "+std::string(chan->name));
			if (el)
			{
				for(modelist::iterator it = el->begin(); it != el->end(); it++)
				{
					WriteServ(user->fd, "%s %s %s %s %s %s", listnumeric.c_str(), user->nick, chan->name, it->mask.c_str(), it->nick.c_str(), it->time.c_str());
				}
			}
			
			WriteServ(user->fd, "%s %s %s %s", endoflistnumeric.c_str(), user->nick, chan->name, endofliststring.c_str());
		}
	}

	virtual void OnChannelDelete(chanrec* chan)
	{
		modelist* list = (modelist*)chan->GetExt(infokey);
			
		if(list)
		{
			chan->Shrink(infokey);
			delete list;
		}
	}
	
	virtual void OnSyncChannel(chanrec* chan, Module* proto, void* opaque)
	{
		modelist* list = (modelist*)chan->GetExt(infokey);		
		if(list)
		{
			for(modelist::iterator it = list->begin(); it != list->end(); it++)
			{
				proto->ProtoSendMode(opaque, TYPE_CHANNEL, chan, "+" + std::string(1, mode) + " " + it->mask);
			}
		}
	}
	
	virtual void OnCleanup(int target_type, void* item)
	{
		if (target_type == TYPE_CHANNEL)
		{
			chanrec* chan = (chanrec*)item;
			
			modelist* list = (modelist*)chan->GetExt(infokey);
			
			if(list)
			{
				chan->Shrink(infokey);
				delete list;
			}
		}
	}
};

#endif