Add module for anti-autorejoin channel mode +J

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

1 files changed, 177 insertions, 0 deletions

diff --git a/src/modules/m_kicknorejoin.cpp b/src/modules/m_kicknorejoin.cpp
new file mode 100644
index 000000000..bd8597a0f
--- /dev/null
+++ b/src/modules/m_kicknorejoin.cpp
@@ -0,0 +1,177 @@
+#include <time.h>
+#include <map>
+#include <vector>
+#include <sstream>
+#include "users.h"
+#include "channels.h"
+#include "modules.h"
+#include "helperfuncs.h"
+
+/* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
+
+inline int strtoint(const std::string &str)
+{
+	std::istringstream ss(str);
+	int result;
+	ss >> result;
+	return result;
+}
+
+typedef std::map<userrec*, time_t> delaylist;
+
+class ModuleKickNoRejoin : public Module
+{
+	Server *Srv;
+	
+public:
+ 
+	ModuleKickNoRejoin(Server* Me)
+		: Module::Module(Me)
+	{
+		Srv = Me;
+		
+		Srv->AddExtendedMode('J', MT_CHANNEL, false, 1, 0);
+	}
+	
+	virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list &params)
+	{
+		if ((modechar == 'J') && (type == MT_CHANNEL))
+		{
+			if(!mode_on)
+			{
+				// Taking the mode off, we need to clean up.
+				chanrec* c = (chanrec*)target;
+				
+				delaylist* dl = (delaylist*)c->GetExt("norejoinusers");
+				
+				if(dl)
+				{
+					delete dl;
+					c->Shrink("norejoinusers");
+				}
+			}
+			return 1;
+		}
+		return 0;
+	}
+
+	virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname)
+	{
+		if(chan)
+		{
+			delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
+			log(DEBUG, "m_kicknorejoin.so: tried to grab delay list");
+			
+			if(dl)
+			{
+				log(DEBUG, "m_kicknorejoin.so: got delay list, iterating over it");
+				std::vector<userrec*> itemstoremove;
+			
+				for(delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
+				{
+					log(DEBUG, "m_kicknorejoin.so:\t[%s] => %d", iter->first->nick, iter->second);
+					if(iter->second > time(NULL))
+					{
+						log(DEBUG, "m_kicknorejoin.so: still inside time slot");
+						if(iter->first == user)					
+						{
+							log(DEBUG, "m_kicknorejoin.so: and we have the right user");
+							WriteServ(user->fd, "495 %s %s :You cannot rejoin this channel yet after being kicked (+J)", user->nick, chan->name);
+							return 1;
+						}
+					}
+					else
+					{
+						// Expired record, remove.
+						log(DEBUG, "m_kicknorejoin.so: record expired");
+						itemstoremove.push_back(iter->first);
+					}
+				}
+				
+				for(unsigned int i = 0; i < itemstoremove.size(); i++)
+					dl->erase(itemstoremove[i]);
+																	
+				if(!dl->size())
+				{
+					// Now it's empty..
+					delete dl;
+					chan->Shrink("norejoinusers");
+				}
+			}
+		}
+		return 0;
+	}
+	
+	virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, std::string reason)
+	{
+		if(chan->IsCustomModeSet('J'))
+		{
+			delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
+			
+			if(!dl)
+			{
+				dl = new delaylist;
+				chan->Extend("norejoinusers", (char*)dl);
+			}
+			
+			log(DEBUG, "m_kicknorejoin.so: setting record for %s, %d second delay", user->nick, strtoint(chan->GetModeParameter('J')));
+			(*dl)[user] = time(NULL) + strtoint(chan->GetModeParameter('J'));
+		}
+	}
+	
+	virtual void OnChannelDelete(chanrec* chan)
+	{
+		delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
+		
+		if(dl)
+		{
+			delete dl;
+			chan->Shrink("norejoinusers");
+		}
+	}
+
+	virtual void Implements(char* List)
+	{
+		List[I_On005Numeric] = List[I_OnExtendedMode] = List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserKick] = 1;
+	}
+
+	virtual void On005Numeric(std::string &output)
+	{
+		InsertMode(output, "J", 3);
+	}
+
+	virtual ~ModuleKickNoRejoin()
+	{
+	}
+	
+	virtual Version GetVersion()
+	{
+		return Version(1, 0, 0, 0, VF_STATIC | VF_VENDOR);
+	}
+};
+
+
+class ModuleKickNoRejoinFactory : public ModuleFactory
+{
+ public:
+	ModuleKickNoRejoinFactory()
+	{
+	}
+	
+	~ModuleKickNoRejoinFactory()
+	{
+	}
+	
+	virtual Module * CreateModule(Server* Me)
+	{
+		return new ModuleKickNoRejoin(Me);
+	}
+	
+};
+
+
+extern "C" void * init_module( void )
+{
+	return new ModuleKickNoRejoinFactory;
+}
+