-/* +------------------------------------+
- * | Inspire Internet Relay Chat Daemon |
- * +------------------------------------+
- *
- * InspIRCd: (C) 2002-2007 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
- *
- * This program is free but copyrighted software; see
- * the file COPYING for details.
- *
- * ---------------------------------------------------
- */
-
-#include "inspircd.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-
-/* $ModDesc: Provides channel mode +j (join flood protection) */
-
-/** Holds settings and state associated with channel mode +j
- */
-class joinfloodsettings : public classbase
-{
- public:
-
- int secs;
- int joins;
- time_t reset;
- time_t unlocktime;
- int counter;
- bool locked;
- InspIRCd* ServerInstance;
-
- joinfloodsettings() : secs(0), joins(0) {};
-
- joinfloodsettings(int b, int c) : secs(b), joins(c)
- {
- reset = time(NULL) + secs;
- counter = 0;
- locked = false;
- };
-
- void addjoin()
- {
- counter++;
- if (time(NULL) > reset)
- {
- counter = 0;
- reset = time(NULL) + secs;
- }
- }
-
- bool shouldlock()
- {
- return (counter >= this->joins);
- }
-
- void clear()
- {
- counter = 0;
- }
-
- bool islocked()
- {
- if (locked)
- {
- if (time(NULL) > unlocktime)
- {
- locked = false;
- return false;
- }
- else
- {
- return true;
- }
- }
- return false;
- }
-
- void lock()
- {
- locked = true;
- unlocktime = time(NULL) + 60;
- }
-
-};
-
-/** Handles channel mode +j
- */
-class JoinFlood : public ModeHandler
-{
- public:
- JoinFlood(InspIRCd* Instance) : ModeHandler(Instance, 'j', 1, 0, false, MODETYPE_CHANNEL, false) { }
-
- ModePair ModeSet(userrec* source, userrec* dest, chanrec* 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);
- }
-
- bool CheckTimeStamp(time_t theirs, time_t ours, const std::string &their_param, const std::string &our_param, chanrec* channel)
- {
- /* When TS is equal, the alphabetically later one wins */
- return (their_param < our_param);
- }
-
- ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string ¶meter, bool adding)
- {
- 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->WriteServ("608 %s %s :Invalid flood parameter",source->nick,channel->name);
- parameter.clear();
- return MODEACTION_DENY;
- }
- else
- {
- if (!channel->GetExt("joinflood", dummy))
- {
- parameter = ConvToStr(njoins) + ":" +ConvToStr(nsecs);
- joinfloodsettings *f = new joinfloodsettings(nsecs,njoins);
- channel->Extend("joinflood", f);
- channel->SetMode('j', true);
- channel->SetModeParam('j', parameter.c_str(), true);
- 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(nsecs,njoins);
- channel->Shrink("joinflood");
- channel->Extend("joinflood", f);
- channel->SetModeParam('j', cur_param.c_str(), false);
- channel->SetModeParam('j', parameter.c_str(), true);
- return MODEACTION_ALLOW;
- }
- else
- {
- return MODEACTION_DENY;
- }
- }
- }
- }
- }
- else
- {
- source->WriteServ("608 %s %s :Invalid flood parameter",source->nick,channel->name);
- return MODEACTION_DENY;
- }
- }
- else
- {
- if (channel->GetExt("joinflood", dummy))
- {
- joinfloodsettings *f;
- channel->GetExt("joinflood", f);
- DELETE(f);
- channel->Shrink("joinflood");
- channel->SetMode('j', false);
- return MODEACTION_ALLOW;
- }
- }
- return MODEACTION_DENY;
- }
-};
-
-class ModuleJoinFlood : public Module
-{
-
- JoinFlood* jf;
-
- public:
-
- ModuleJoinFlood(InspIRCd* Me)
- : Module(Me)
- {
-
- jf = new JoinFlood(ServerInstance);
- if (!ServerInstance->AddMode(jf, 'j'))
- throw ModuleException("Could not add new modes!");
- }
-
- virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname, std::string &privs)
- {
- if (chan)
- {
- joinfloodsettings *f;
- if (chan->GetExt("joinflood", f))
- {
- if (f->islocked())
- {
- user->WriteServ("609 %s %s :This channel is temporarily unavailable (+j). Please try again later.",user->nick,chan->name);
- return 1;
- }
- }
- }
- return 0;
- }
-
- virtual void OnUserJoin(userrec* user, chanrec* channel, bool &silent)
- {
- joinfloodsettings *f;
- 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, f->joins, f->secs);
- }
- }
- }
-
- void OnChannelDelete(chanrec* chan)
- {
- joinfloodsettings *f;
- if (chan->GetExt("joinflood",f))
- {
- DELETE(f);
- chan->Shrink("joinflood");
- }
- }
-
- void Implements(char* List)
- {
- List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserJoin] = 1;
- }
-
- virtual ~ModuleJoinFlood()
- {
- ServerInstance->Modes->DelMode(jf);
- DELETE(jf);
- }
-
- virtual Version GetVersion()
- {
- return Version(1, 1, 0, 0, VF_COMMON | VF_VENDOR, API_VERSION);
- }
-};
-
-MODULE_INIT(ModuleJoinFlood)
+/* +------------------------------------+\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