-/* +------------------------------------+\r * | Inspire Internet Relay Chat Daemon |\r * +------------------------------------+\r *\r * InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r * the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h"\r#include "users.h"\r#include "channels.h"\r#include "modules.h"\r\r/* $ModDesc: Provides channel mode +j (join flood protection) */\r\r/** Holds settings and state associated with channel mode +j\r */\rclass joinfloodsettings : public classbase\r{\r public:\r\r int secs;\r int joins;\r time_t reset;\r time_t unlocktime;\r int counter;\r bool locked;\r InspIRCd* ServerInstance;\r\r joinfloodsettings() : secs(0), joins(0) {};\r\r joinfloodsettings(int b, int c) : secs(b), joins(c)\r {\r reset = time(NULL) + secs;\r counter = 0;\r locked = false;\r };\r\r void addjoin()\r {\r counter++;\r if (time(NULL) > reset)\r {\r counter = 0;\r reset = time(NULL) + secs;\r }\r }\r\r bool shouldlock()\r {\r return (counter >= this->joins);\r }\r\r void clear()\r {\r counter = 0;\r }\r\r bool islocked()\r {\r if (locked)\r {\r if (time(NULL) > unlocktime)\r {\r locked = false;\r return false;\r }\r else\r {\r return true;\r }\r }\r return false;\r }\r\r void lock()\r {\r locked = true;\r unlocktime = time(NULL) + 60;\r }\r\r};\r\r/** Handles channel mode +j\r */\rclass JoinFlood : public ModeHandler\r{\r public:\r JoinFlood(InspIRCd* Instance) : ModeHandler(Instance, 'j', 1, 0, false, MODETYPE_CHANNEL, false) { }\r\r ModePair ModeSet(userrec* source, userrec* dest, chanrec* channel, const std::string ¶meter)\r {\r joinfloodsettings* x;\r if (channel->GetExt("joinflood",x))\r return std::make_pair(true, ConvToStr(x->joins)+":"+ConvToStr(x->secs));\r else\r return std::make_pair(false, parameter);\r } \r\r bool CheckTimeStamp(time_t theirs, time_t ours, const std::string &their_param, const std::string &our_param, chanrec* channel)\r {\r /* When TS is equal, the alphabetically later one wins */\r return (their_param < our_param);\r }\r\r ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string ¶meter, bool adding)\r {\r joinfloodsettings* dummy;\r\r if (adding)\r {\r char ndata[MAXBUF];\r char* data = ndata;\r strlcpy(ndata,parameter.c_str(),MAXBUF);\r char* joins = data;\r char* secs = NULL;\r while (*data)\r {\r if (*data == ':')\r {\r *data = 0;\r data++;\r secs = data;\r break;\r }\r else data++;\r }\r if (secs)\r\r {\r /* Set up the flood parameters for this channel */\r int njoins = atoi(joins);\r int nsecs = atoi(secs);\r if ((njoins<1) || (nsecs<1))\r {\r source->WriteServ("608 %s %s :Invalid flood parameter",source->nick,channel->name);\r parameter.clear();\r return MODEACTION_DENY;\r }\r else\r {\r if (!channel->GetExt("joinflood", dummy))\r {\r parameter = ConvToStr(njoins) + ":" +ConvToStr(nsecs);\r joinfloodsettings *f = new joinfloodsettings(nsecs,njoins);\r channel->Extend("joinflood", f);\r channel->SetMode('j', true);\r channel->SetModeParam('j', parameter.c_str(), true);\r return MODEACTION_ALLOW;\r }\r else\r {\r std::string cur_param = channel->GetModeParameter('j');\r parameter = ConvToStr(njoins) + ":" +ConvToStr(nsecs);\r if (cur_param == parameter)\r {\r // mode params match\r return MODEACTION_DENY;\r }\r else\r {\r // new mode param, replace old with new\r if ((nsecs > 0) && (njoins > 0))\r {\r joinfloodsettings* f;\r channel->GetExt("joinflood", f);\r delete f;\r f = new joinfloodsettings(nsecs,njoins);\r channel->Shrink("joinflood");\r channel->Extend("joinflood", f);\r channel->SetModeParam('j', cur_param.c_str(), false);\r channel->SetModeParam('j', parameter.c_str(), true);\r return MODEACTION_ALLOW;\r }\r else\r {\r return MODEACTION_DENY;\r }\r }\r }\r }\r }\r else\r {\r source->WriteServ("608 %s %s :Invalid flood parameter",source->nick,channel->name);\r return MODEACTION_DENY;\r }\r }\r else\r {\r if (channel->GetExt("joinflood", dummy))\r {\r joinfloodsettings *f;\r channel->GetExt("joinflood", f);\r DELETE(f);\r channel->Shrink("joinflood");\r channel->SetMode('j', false);\r return MODEACTION_ALLOW;\r }\r }\r return MODEACTION_DENY;\r }\r};\r\rclass ModuleJoinFlood : public Module\r{\r \r JoinFlood* jf;\r \r public:\r \r ModuleJoinFlood(InspIRCd* Me)\r : Module(Me)\r {\r \r jf = new JoinFlood(ServerInstance);\r if (!ServerInstance->AddMode(jf, 'j'))\r throw ModuleException("Could not add new modes!");\r }\r \r virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname, std::string &privs)\r {\r if (chan)\r {\r joinfloodsettings *f;\r if (chan->GetExt("joinflood", f))\r {\r if (f->islocked())\r {\r user->WriteServ("609 %s %s :This channel is temporarily unavailable (+j). Please try again later.",user->nick,chan->name);\r return 1;\r }\r }\r }\r return 0;\r }\r\r virtual void OnUserJoin(userrec* user, chanrec* channel, bool &silent)\r {\r joinfloodsettings *f;\r if (channel->GetExt("joinflood",f))\r {\r f->addjoin();\r if (f->shouldlock())\r {\r f->clear();\r f->lock();\r channel->WriteChannelWithServ((char*)ServerInstance->Config->ServerName, "NOTICE %s :This channel has been closed to new users for 60 seconds because there have been more than %d joins in %d seconds.", channel->name, f->joins, f->secs);\r }\r }\r }\r\r void OnChannelDelete(chanrec* chan)\r {\r joinfloodsettings *f;\r if (chan->GetExt("joinflood",f))\r {\r DELETE(f);\r chan->Shrink("joinflood");\r }\r }\r\r void Implements(char* List)\r {\r List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserJoin] = 1;\r }\r\r virtual ~ModuleJoinFlood()\r {\r ServerInstance->Modes->DelMode(jf);\r DELETE(jf);\r }\r \r virtual Version GetVersion()\r {\r return Version(1, 1, 0, 0, VF_COMMON | VF_VENDOR, API_VERSION);\r }\r};\r\rMODULE_INIT(ModuleJoinFlood)\r
\ No newline at end of file
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ * Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ * Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ * Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
+ * Copyright (C) 2006-2007 Craig Edwards <craigedwards@brainbox.cc>
+ * Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
+ *
+ * This file is part of InspIRCd. InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include "inspircd.h"
+
+/* $ModDesc: Provides channel mode +j (join flood protection) */
+
+/** Holds settings and state associated with channel mode +j
+ */
+class joinfloodsettings
+{
+ public:
+ int secs;
+ int joins;
+ time_t reset;
+ time_t unlocktime;
+ int counter;
+ bool locked;
+
+ joinfloodsettings(int b, int c) : secs(b), joins(c)
+ {
+ reset = ServerInstance->Time() + secs;
+ counter = 0;
+ locked = false;
+ };
+
+ void addjoin()
+ {
+ counter++;
+ if (ServerInstance->Time() > reset)
+ {
+ counter = 0;
+ reset = ServerInstance->Time() + secs;
+ }
+ }
+
+ bool shouldlock()
+ {
+ return (counter >= this->joins);
+ }
+
+ void clear()
+ {
+ counter = 0;
+ }
+
+ bool islocked()
+ {
+ if (locked)
+ {
+ if (ServerInstance->Time() > unlocktime)
+ {
+ locked = false;
+ return false;
+ }
+ else
+ {
+ return true;
+ }
+ }
+ return false;
+ }
+
+ void lock()
+ {
+ locked = true;
+ unlocktime = ServerInstance->Time() + 60;
+ }
+
+};
+
+/** Handles channel mode +j
+ */
+class JoinFlood : public ModeHandler
+{
+ public:
+ SimpleExtItem<joinfloodsettings> ext;
+ JoinFlood(Module* Creator) : ModeHandler(Creator, "joinflood", 'j', PARAM_SETONLY, MODETYPE_CHANNEL),
+ ext("joinflood", Creator) { }
+
+ ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string ¶meter, bool adding)
+ {
+ if (adding)
+ {
+ char ndata[MAXBUF];
+ char* data = ndata;
+ strlcpy(ndata,parameter.c_str(),MAXBUF);
+ char* joins = data;
+ char* secs = NULL;
+ while (*data)
+ {
+ if (*data == ':')
+ {
+ *data = 0;
+ data++;
+ secs = data;
+ break;
+ }
+ else data++;
+ }
+ if (secs)
+
+ {
+ /* Set up the flood parameters for this channel */
+ int njoins = atoi(joins);
+ int nsecs = atoi(secs);
+ if ((njoins<1) || (nsecs<1))
+ {
+ source->WriteNumeric(608, "%s %s :Invalid flood parameter",source->nick.c_str(),channel->name.c_str());
+ parameter.clear();
+ return MODEACTION_DENY;
+ }
+ else
+ {
+ joinfloodsettings* f = ext.get(channel);
+ if (!f)
+ {
+ parameter = ConvToStr(njoins) + ":" +ConvToStr(nsecs);
+ f = new joinfloodsettings(nsecs, njoins);
+ ext.set(channel, f);
+ channel->SetModeParam('j', parameter);
+ return MODEACTION_ALLOW;
+ }
+ else
+ {
+ std::string cur_param = channel->GetModeParameter('j');
+ parameter = ConvToStr(njoins) + ":" +ConvToStr(nsecs);
+ if (cur_param == parameter)
+ {
+ // mode params match
+ return MODEACTION_DENY;
+ }
+ else
+ {
+ // new mode param, replace old with new
+ f = new joinfloodsettings(nsecs, njoins);
+ ext.set(channel, f);
+ channel->SetModeParam('j', parameter);
+ return MODEACTION_ALLOW;
+ }
+ }
+ }
+ }
+ else
+ {
+ source->WriteNumeric(608, "%s %s :Invalid flood parameter",source->nick.c_str(),channel->name.c_str());
+ return MODEACTION_DENY;
+ }
+ }
+ else
+ {
+ if (channel->IsModeSet('j'))
+ {
+ ext.unset(channel);
+ channel->SetModeParam('j', "");
+ return MODEACTION_ALLOW;
+ }
+ }
+ return MODEACTION_DENY;
+ }
+};
+
+class ModuleJoinFlood : public Module
+{
+
+ JoinFlood jf;
+
+ public:
+
+ ModuleJoinFlood()
+ : jf(this)
+ {
+ }
+
+ void init()
+ {
+ ServerInstance->Modules->AddService(jf);
+ ServerInstance->Modules->AddService(jf.ext);
+ Implementation eventlist[] = { I_OnUserPreJoin, I_OnUserJoin };
+ ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
+ }
+
+ ModResult OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
+ {
+ if (chan)
+ {
+ joinfloodsettings *f = jf.ext.get(chan);
+ if (f && f->islocked())
+ {
+ user->WriteNumeric(609, "%s %s :This channel is temporarily unavailable (+j). Please try again later.",user->nick.c_str(),chan->name.c_str());
+ return MOD_RES_DENY;
+ }
+ }
+ return MOD_RES_PASSTHRU;
+ }
+
+ void OnUserJoin(Membership* memb, bool sync, bool created, CUList& excepts)
+ {
+ /* We arent interested in JOIN events caused by a network burst */
+ if (sync)
+ return;
+
+ joinfloodsettings *f = jf.ext.get(memb->chan);
+
+ /* But all others are OK */
+ if ((f) && (!f->islocked()))
+ {
+ f->addjoin();
+ if (f->shouldlock())
+ {
+ f->clear();
+ f->lock();
+ memb->chan->WriteChannelWithServ((char*)ServerInstance->Config->ServerName.c_str(), "NOTICE %s :This channel has been closed to new users for 60 seconds because there have been more than %d joins in %d seconds.", memb->chan->name.c_str(), f->joins, f->secs);
+ }
+ }
+ }
+
+ ~ModuleJoinFlood()
+ {
+ }
+
+ Version GetVersion()
+ {
+ return Version("Provides channel mode +j (join flood protection)", VF_VENDOR);
+ }
+};
+
+MODULE_INIT(ModuleJoinFlood)
projects / user / henk / code / inspircd.git / blobdiff