Register a few extensions that weren't registered

[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 e0b30163c28c5af598e895c69973167ad4ff8e08..814eee5ef7e8757c08aff551fbb11586486e7fca 100644 (file)
--- a/src/modules/m_messageflood.cpp
+++ b/src/modules/m_messageflood.cpp
@@ -75,8 +75,6 @@ class MsgFlood : public ModeHandler
 
        ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
        {
-               floodsettings *f = ext.get(channel);
-
                if (adding)
                {
                        std::string::size_type colon = parameter.find(':');
@@ -97,6 +95,7 @@ class MsgFlood : public ModeHandler
                                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;
@@ -108,15 +107,13 @@ class MsgFlood : public ModeHandler
                }
                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;
+               }
        }
 };
 
@@ -129,11 +126,14 @@ class ModuleMsgFlood : public Module
        ModuleMsgFlood()
                : mf(this)
        {
-               if (!ServerInstance->Modes->AddMode(&mf))
-                       throw ModuleException("Could not add new modes!");
-               ServerInstance->Extensions.Register(&mf.ext);
+       }
+
+       void init()
+       {
+               ServerInstance->Modules->AddService(mf);
+               ServerInstance->Modules->AddService(mf.ext);
                Implementation eventlist[] = { I_OnUserPreNotice, I_OnUserPreMessage };
-               ServerInstance->Modules->Attach(eventlist, this, 2);
+               ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
        }
 
        ModResult ProcessMessages(User* user,Channel* dest, const std::string &text)