Improved strhashcomp with no allocations

[user/henk/code/inspircd.git] / src / modules / m_messageflood.cpp

diff --git a/src/modules/m_messageflood.cpp b/src/modules/m_messageflood.cpp
index 660c2128b895fd5b6496c7707fb7747d74edfcc4..282a42a3bb23447f503adeec4c70d02f77cf2309 100644 (file)
--- a/src/modules/m_messageflood.cpp
+++ b/src/modules/m_messageflood.cpp
@@ -172,47 +172,46 @@ class ModuleMsgFlood : public Module
                return 0;
        }
 
-       int ProcessMessages(userrec* user,chanrec* dest,std::string &text)
+       void ProcessMessages(userrec* user,chanrec* dest,std::string &text)
        {
-               floodsettings *f = (floodsettings*)dest->GetExt("flood");
-               if (f)
+               if (IS_LOCAL(user))
                {
-                       f->addmessage(user);
-                       if (f->shouldkick(user))
+                       floodsettings *f = (floodsettings*)dest->GetExt("flood");
+                       if (f)
                        {
-                               /* Youre outttta here! */
-                               f->clear(user);
-                               if (f->ban)
+                               f->addmessage(user);
+                               if (f->shouldkick(user))
                                {
-                                       char* parameters[3];
-                                       parameters[0] = dest->name;
-                                       parameters[1] = "+b";
-                                       parameters[2] = user->MakeWildHost();
-                                       Srv->SendMode(parameters,3,user);
+                                       /* Youre outttta here! */
+                                       f->clear(user);
+                                       if (f->ban)
+                                       {
+                                               char* parameters[3];
+                                               parameters[0] = dest->name;
+                                               parameters[1] = "+b";
+                                               parameters[2] = user->MakeWildHost();
+                                               Srv->SendMode(parameters,3,user);
+                                       }
+                                       Srv->KickUser(NULL, user, dest, "Channel flood triggered (mode +f)");
                                }
-                               Srv->KickUser(NULL, user, dest, "Channel flood triggered (mode +f)");
-                               return 1;
                        }
                }
-               return 0;
        }
 
-        virtual int OnUserPreMessage(userrec* user,void* dest,int target_type, std::string &text)
-        {
+        virtual void OnUserMessage(userrec* user, void* dest, int target_type, std::string text, char status)
+       {
                 if (target_type == TYPE_CHANNEL)
                 {
-                        return ProcessMessages(user,(chanrec*)dest,text);
+                        ProcessMessages(user,(chanrec*)dest,text);
                 }
-                else return 0;
        }
 
-       virtual int OnUserPreNotice(userrec* user,void* dest,int target_type, std::string &text)
+       virtual void OnUserNotice(userrec* user, void* dest, int target_type, std::string text, char status)
        {
                if (target_type == TYPE_CHANNEL)
                {
-                       return ProcessMessages(user,(chanrec*)dest,text);
+                       ProcessMessages(user,(chanrec*)dest,text);
                }
-               else return 0;
        }
 
        void OnChannelDelete(chanrec* chan)
@@ -227,28 +226,13 @@ class ModuleMsgFlood : public Module
 
        void Implements(char* List)
        {
-               List[I_On005Numeric] = List[I_OnExtendedMode] = List[I_OnChannelDelete] = List[I_OnUserPreNotice] = List[I_OnUserPreMessage] = 1;
+               List[I_On005Numeric] = List[I_OnExtendedMode] = List[I_OnChannelDelete] = List[I_OnUserNotice] = List[I_OnUserMessage] = 1;
        }
 
-        virtual void On005Numeric(std::string &output)
-        {
-                std::stringstream line(output);
-                std::string temp1, temp2;
-                while (!line.eof())
-                {
-                        line >> temp1;
-                        if (temp1.substr(0,10) == "CHANMODES=")
-                        {
-                               // By doing this we're *assuming* no other module has fucked up the CHANMODES=
-                               // section of the 005 numeric. If they have, we're going DOWN in a blaze of glory,
-                               // with a honking great EXCEPTION :)
-                               temp1.insert(temp1.find(",")+1,"f");
-                        }
-                        temp2 = temp2 + temp1 + " ";
-                }
-               if (temp2.length())
-                       output = temp2.substr(0,temp2.length()-1);
-        }
+       virtual void On005Numeric(std::string &output)
+       {
+               InsertMode(output, "f", 3);
+       }
 
        virtual ~ModuleMsgFlood()
        {