Adding the +e and +I modules m_banexception and m_inviteexception, also adding their shared utility header

git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@2884 e03df62e-2008-0410-955e-edbf42e46eb7

1 files changed, 269 insertions, 0 deletions

diff --git a/include/u_listmode.h b/include/u_listmode.h
new file mode 100644
index 000000000..b6cac9c50
--- /dev/null
+++ b/include/u_listmode.h
@@ -0,0 +1,269 @@
+#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. */
+
+/* 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 excpetion list here...
+typedef std::vector<ListItem> modelist;
+typedef std::vector<ListLimit> limitlist;
+
+class ListModeBaseModule : public Module
+{
+protected:
+	int maxlist;
+	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, std::string eolstr, std::string lnum, 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(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;
+					}
+				}
+				
+				for(limitlist::iterator it = chanlimits.begin(); it != chanlimits.end(); it++)
+				{
+					if(Srv->MatchText(chan->name, it->mask))
+					{
+						// We have a pattern matching the channel...
+						if(el->size() < 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, maxlist);
+				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