]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_messageflood.cpp
Introduce ModeProcessFlags, can be passed to ModeParser::Process() to indicate local...
[user/henk/code/inspircd.git] / src / modules / m_messageflood.cpp
index db17ee09dc10b76887ed4564c8f1941a4c43fea6..86296094b3151af2c81828a5b32c4bbaa6018747 100644 (file)
@@ -1,69 +1,62 @@
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
- * See: http://wiki.inspircd.org/Credits
+ *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
+ *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
+ *   Copyright (C) 2007 John Brooks <john.brooks@dereferenced.net>
+ *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *   Copyright (C) 2006 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
  *
- * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ * 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 +f (message flood protection) */
 
 /** Holds flood settings and state for mode +f
  */
-class floodsettings : public classbase
+class floodsettings
 {
- private:
-       InspIRCd *ServerInstance;
  public:
        bool ban;
-       int secs;
-       int lines;
+       unsigned int secs;
+       unsigned int lines;
        time_t reset;
-       std::map<User*,int> counters;
+       std::map<User*, unsigned int> counters;
 
-       floodsettings(InspIRCd *Instance, bool a, int b, int c) : ServerInstance(Instance), ban(a), secs(b), lines(c)
+       floodsettings(bool a, int b, int c) : ban(a), secs(b), lines(c)
        {
                reset = ServerInstance->Time() + secs;
-       };
+       }
 
-       void addmessage(User* who)
+       bool addmessage(User* who)
        {
-               std::map<User*,int>::iterator iter = counters.find(who);
-               if (iter != counters.end())
-               {
-                       iter->second++;
-               }
-               else
-               {
-                       counters[who] = 1;
-               }
                if (ServerInstance->Time() > reset)
                {
                        counters.clear();
                        reset = ServerInstance->Time() + secs;
                }
-       }
 
-       bool shouldkick(User* who)
-       {
-               std::map<User*,int>::iterator iter = counters.find(who);
-               if (iter != counters.end())
-               {
-                       return (iter->second >= this->lines);
-               }
-               else return false;
+               return (++counters[who] >= this->lines);
        }
 
        void clear(User* who)
        {
-               std::map<User*,int>::iterator iter = counters.find(who);
+               std::map<User*, unsigned int>::iterator iter = counters.find(who);
                if (iter != counters.end())
                {
                        counters.erase(iter);
@@ -77,115 +70,50 @@ class MsgFlood : public ModeHandler
 {
  public:
        SimpleExtItem<floodsettings> ext;
-       MsgFlood(InspIRCd* Instance, Module* Creator) : ModeHandler(Instance, Creator, 'f', 1, 0, false, MODETYPE_CHANNEL, false),
+       MsgFlood(Module* Creator) : ModeHandler(Creator, "flood", 'f', PARAM_SETONLY, MODETYPE_CHANNEL),
                ext("messageflood", Creator) { }
 
-       ModePair ModeSet(User* source, User* dest, Channel* channel, const std::string &parameter)
-       {
-               floodsettings* x = ext.get(channel);
-               if (x)
-                       return std::make_pair(true, (x->ban ? "*" : "")+ConvToStr(x->lines)+":"+ConvToStr(x->secs));
-               else
-                       return std::make_pair(false, parameter);
-       }
-
        ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
        {
-               floodsettings *f = ext.get(channel);
-
                if (adding)
                {
-                       char ndata[MAXBUF];
-                       char* data = ndata;
-                       strlcpy(ndata,parameter.c_str(),MAXBUF);
-                       char* lines = data;
-                       char* secs = NULL;
-                       bool ban = false;
-                       if (*data == '*')
-                       {
-                               ban = true;
-                               lines++;
-                       }
-                       else
-                       {
-                               ban = false;
-                       }
-                       while (*data)
+                       std::string::size_type colon = parameter.find(':');
+                       if ((colon == std::string::npos) || (parameter.find('-') != std::string::npos))
                        {
-                               if (*data == ':')
-                               {
-                                       *data = 0;
-                                       data++;
-                                       secs = data;
-                                       break;
-                               }
-                               else data++;
-                       }
-                       if (secs)
-                       {
-                               /* Set up the flood parameters for this channel */
-                               int nlines = atoi(lines);
-                               int nsecs = atoi(secs);
-                               if ((nlines<2) || (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 (!f)
-                                       {
-                                               parameter = std::string(ban ? "*" : "") + ConvToStr(nlines) + ":" +ConvToStr(nsecs);
-                                               f = new floodsettings(ServerInstance,ban,nsecs,nlines);
-                                               ext.set(channel, f);
-                                               channel->SetModeParam('f', parameter);
-                                               return MODEACTION_ALLOW;
-                                       }
-                                       else
-                                       {
-                                               std::string cur_param = channel->GetModeParameter('f');
-                                               parameter = std::string(ban ? "*" : "") + ConvToStr(nlines) + ":" +ConvToStr(nsecs);
-                                               if (cur_param == parameter)
-                                               {
-                                                       // mode params match
-                                                       return MODEACTION_DENY;
-                                               }
-                                               else
-                                               {
-                                                       if ((((nlines != f->lines) || (nsecs != f->secs) || (ban != f->ban))) && (((nsecs > 0) && (nlines > 0))))
-                                                       {
-                                                               floodsettings *fs = new floodsettings(ServerInstance,ban,nsecs,nlines);
-                                                               ext.set(channel, fs);
-                                                               channel->SetModeParam('f', parameter);
-                                                               return MODEACTION_ALLOW;
-                                                       }
-                                                       else
-                                                       {
-                                                               return MODEACTION_DENY;
-                                                       }
-                                               }
-                                       }
-                               }
+                               source->WriteNumeric(608, "%s %s :Invalid flood parameter",source->nick.c_str(),channel->name.c_str());
+                               return MODEACTION_DENY;
                        }
-                       else
+
+                       /* Set up the flood parameters for this channel */
+                       bool ban = (parameter[0] == '*');
+                       unsigned int nlines = ConvToInt(parameter.substr(ban ? 1 : 0, ban ? colon-1 : colon));
+                       unsigned int nsecs = ConvToInt(parameter.substr(colon+1));
+
+                       if ((nlines<2) || (nsecs<1))
                        {
                                source->WriteNumeric(608, "%s %s :Invalid flood parameter",source->nick.c_str(),channel->name.c_str());
-                               parameter.clear();
                                return MODEACTION_DENY;
                        }
+
+                       floodsettings* f = ext.get(channel);
+                       if ((f) && (nlines == f->lines) && (nsecs == f->secs) && (ban == f->ban))
+                               // mode params match
+                               return MODEACTION_DENY;
+
+                       ext.set(channel, new floodsettings(ban, nsecs, nlines));
+                       parameter = std::string(ban ? "*" : "") + ConvToStr(nlines) + ":" + ConvToStr(nsecs);
+                       channel->SetModeParam('f', parameter);
+                       return MODEACTION_ALLOW;
                }
                else
                {
-                       if (f)
-                       {
-                               ext.unset(channel);
-                               channel->SetModeParam('f', "");
-                               return MODEACTION_ALLOW;
-                       }
-               }
+                       if (!channel->IsModeSet('f'))
+                               return MODEACTION_DENY;
 
-               return MODEACTION_DENY;
+                       ext.unset(channel);
+                       channel->SetModeParam('f', "");
+                       return MODEACTION_ALLOW;
+               }
        }
 };
 
@@ -195,28 +123,30 @@ class ModuleMsgFlood : public Module
 
  public:
 
-       ModuleMsgFlood(InspIRCd* Me)
-               : Module(Me), mf(Me, this)
+       ModuleMsgFlood()
+               : mf(this)
+       {
+       }
+
+       void init() CXX11_OVERRIDE
        {
-               if (!ServerInstance->Modes->AddMode(&mf))
-                       throw ModuleException("Could not add new modes!");
-               Extensible::Register(&mf.ext);
-               Implementation eventlist[] = { I_OnUserPreNotice, I_OnUserPreMessage };
-               ServerInstance->Modules->Attach(eventlist, this, 2);
+               ServerInstance->Modules->AddService(mf);
+               ServerInstance->Modules->AddService(mf.ext);
+               ServerInstance->Modules->Attach(I_OnUserPreMessage, this);
        }
 
        ModResult ProcessMessages(User* user,Channel* dest, const std::string &text)
        {
-               if (!IS_LOCAL(user) || (CHANOPS_EXEMPT(ServerInstance, 'f') && dest->GetPrefixValue(user) == OP_VALUE))
-               {
+               if ((!IS_LOCAL(user)) || !dest->IsModeSet('f'))
+                       return MOD_RES_PASSTHRU;
+
+               if (ServerInstance->OnCheckExemption(user,dest,"flood") == MOD_RES_ALLOW)
                        return MOD_RES_PASSTHRU;
-               }
 
                floodsettings *f = mf.ext.get(dest);
                if (f)
                {
-                       f->addmessage(user);
-                       if (f->shouldkick(user))
+                       if (f->addmessage(user))
                        {
                                /* Youre outttta here! */
                                f->clear(user);
@@ -225,19 +155,14 @@ class ModuleMsgFlood : public Module
                                        std::vector<std::string> parameters;
                                        parameters.push_back(dest->name);
                                        parameters.push_back("+b");
-                                       parameters.push_back(user->MakeWildHost());
-                                       ServerInstance->SendMode(parameters, ServerInstance->FakeClient);
-
-                                       ServerInstance->PI->SendModeStr(dest->name, std::string("+b ") + user->MakeWildHost());
+                                       parameters.push_back("*!*@" + user->dhost);
+                                       ServerInstance->Modes->Process(parameters, ServerInstance->FakeClient);
                                }
 
-                               char kickmessage[MAXBUF];
-                               snprintf(kickmessage, MAXBUF, "Channel flood triggered (limit is %d lines in %d secs)", f->lines, f->secs);
+                               const std::string kickMessage = "Channel flood triggered (limit is " + ConvToStr(f->lines) +
+                                       " in " + ConvToStr(f->secs) + " secs)";
 
-                               if (!dest->ServerKickUser(user, kickmessage))
-                               {
-                                       delete dest;
-                               }
+                               dest->KickUser(ServerInstance->FakeClient, user, kickMessage);
 
                                return MOD_RES_DENY;
                        }
@@ -246,15 +171,7 @@ class ModuleMsgFlood : public Module
                return MOD_RES_PASSTHRU;
        }
 
-       ModResult OnUserPreMessage(User *user, void *dest, int target_type, std::string &text, char status, CUList &exempt_list)
-       {
-               if (target_type == TYPE_CHANNEL)
-                       return ProcessMessages(user,(Channel*)dest,text);
-
-               return MOD_RES_PASSTHRU;
-       }
-
-       ModResult OnUserPreNotice(User *user, void *dest, int target_type, std::string &text, char status, CUList &exempt_list)
+       ModResult OnUserPreMessage(User* user, void* dest, int target_type, std::string& text, char status, CUList& exempt_list, MessageType msgtype) CXX11_OVERRIDE
        {
                if (target_type == TYPE_CHANNEL)
                        return ProcessMessages(user,(Channel*)dest,text);
@@ -262,14 +179,15 @@ class ModuleMsgFlood : public Module
                return MOD_RES_PASSTHRU;
        }
 
-       ~ModuleMsgFlood()
+       void Prioritize()
        {
-               ServerInstance->Modes->DelMode(&mf);
+               // we want to be after all modules that might deny the message (e.g. m_muteban, m_noctcp, m_blockcolor, etc.)
+               ServerInstance->Modules->SetPriority(this, I_OnUserPreMessage, PRIORITY_LAST);
        }
 
-       Version GetVersion()
+       Version GetVersion() CXX11_OVERRIDE
        {
-               return Version("$Id$", VF_COMMON | VF_VENDOR, API_VERSION);
+               return Version("Provides channel mode +f (message flood protection)", VF_VENDOR);
        }
 };