-/* +------------------------------------+\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
+/* +------------------------------------+
+ * | Inspire Internet Relay Chat Daemon |
+ * +------------------------------------+
+ *
+ * InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
+ *
+ * This program is free but copyrighted software; see
+ * the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
+
+#include "inspircd.h"
+
+/* $ModDesc: Provides channel mode +j (join flood protection) */
+
+/** Holds settings and state associated with channel mode +j
+ */
+class joinfloodsettings : public classbase
+{
+ private:
+ InspIRCd* ServerInstance;
+ public:
+ int secs;
+ int joins;
+ time_t reset;
+ time_t unlocktime;
+ int counter;
+ bool locked;
+
+ joinfloodsettings(InspIRCd *Instance, int b, int c) : ServerInstance(Instance), 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:
+ JoinFlood(InspIRCd* Instance) : ModeHandler(Instance, 'j', 1, 0, false, MODETYPE_CHANNEL, false) { }
+
+ ModePair ModeSet(User* source, User* dest, Channel* channel, const std::string ¶meter)
+ {
+ joinfloodsettings* x;
+ if (channel->GetExt("joinflood",x))
+ return std::make_pair(true, ConvToStr(x->joins)+":"+ConvToStr(x->secs));
+ else
+ return std::make_pair(false, parameter);
+ }
+
+ ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string ¶meter, bool adding, bool)
+ {
+ joinfloodsettings* dummy;
+
+ 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
+ {
+ if (!channel->GetExt("joinflood", dummy))
+ {
+ parameter = ConvToStr(njoins) + ":" +ConvToStr(nsecs);
+ joinfloodsettings *f = new joinfloodsettings(ServerInstance, nsecs, njoins);
+ channel->Extend("joinflood", 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
+ if ((nsecs > 0) && (njoins > 0))
+ {
+ joinfloodsettings* f;
+ channel->GetExt("joinflood", f);
+ delete f;
+ f = new joinfloodsettings(ServerInstance, nsecs, njoins);
+ channel->Shrink("joinflood");
+ channel->Extend("joinflood", f);
+ channel->SetModeParam('j', parameter);
+ return MODEACTION_ALLOW;
+ }
+ else
+ {
+ return MODEACTION_DENY;
+ }
+ }
+ }
+ }
+ }
+ else
+ {
+ source->WriteNumeric(608, "%s %s :Invalid flood parameter",source->nick.c_str(),channel->name.c_str());
+ return MODEACTION_DENY;
+ }
+ }
+ else
+ {
+ if (channel->GetExt("joinflood", dummy))
+ {
+ joinfloodsettings *f;
+ channel->GetExt("joinflood", f);
+ delete f;
+ channel->Shrink("joinflood");
+ channel->SetModeParam('j', "");
+ return MODEACTION_ALLOW;
+ }
+ }
+ return MODEACTION_DENY;
+ }
+};
+
+class ModuleJoinFlood : public Module
+{
+
+ JoinFlood jf;
+
+ public:
+
+ ModuleJoinFlood(InspIRCd* Me)
+ : Module(Me), jf(Me)
+ {
+
+ if (!ServerInstance->Modes->AddMode(&jf))
+ throw ModuleException("Could not add new modes!");
+ Implementation eventlist[] = { I_OnChannelDelete, I_OnUserPreJoin, I_OnUserJoin };
+ ServerInstance->Modules->Attach(eventlist, this, 3);
+ }
+
+ virtual int OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
+ {
+ if (chan)
+ {
+ joinfloodsettings *f;
+ if (chan->GetExt("joinflood", f))
+ {
+ if (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 1;
+ }
+ }
+ }
+ return 0;
+ }
+
+ virtual void OnUserJoin(User* user, Channel* channel, bool sync, bool &silent, bool created)
+ {
+ joinfloodsettings *f;
+
+ /* We arent interested in JOIN events caused by a network burst */
+ if (sync)
+ return;
+
+ /* But all others are OK */
+ if (channel->GetExt("joinflood",f))
+ {
+ f->addjoin();
+ if (f->shouldlock())
+ {
+ f->clear();
+ f->lock();
+ 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.c_str(), f->joins, f->secs);
+ }
+ }
+ }
+
+ void OnChannelDelete(Channel* chan)
+ {
+ joinfloodsettings *f;
+ if (chan->GetExt("joinflood",f))
+ {
+ delete f;
+ chan->Shrink("joinflood");
+ }
+ }
+
+
+ virtual ~ModuleJoinFlood()
+ {
+ ServerInstance->Modes->DelMode(&jf);
+ }
+
+ virtual Version GetVersion()
+ {
+ return Version("$Id$", VF_COMMON | VF_VENDOR, API_VERSION);
+ }
+};
+
+MODULE_INIT(ModuleJoinFlood)
projects / user / henk / code / inspircd.git / blobdiff