Minor spelling errors in m_spanningtree.so

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

diff --git a/src/modules/m_mlock.cpp b/src/modules/m_mlock.cpp
index 7e891f1468c937883ecf3745cf846bbc86e37c20..d1df81354b65e7ce9c64b5b9cbbbe1b4026488e4 100644 (file)
--- a/src/modules/m_mlock.cpp
+++ b/src/modules/m_mlock.cpp
@@ -16,9 +16,13 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+
+/* $ModDesc: Implements the ability to have server-side MLOCK enforcement. */
+
 #include "inspircd.h"
 
-class ModuleMLock : public Module {
+class ModuleMLock : public Module
+{
 private:
        StringExtItem mlock;
 
@@ -27,8 +31,8 @@ public:
 
        void init()
        {
-               Implementation eventlist[] = { I_OnPreMode };
-               ServerInstance->Modules->Attach(eventlist, this, 1);
+               ServerInstance->Modules->Attach(I_OnRawMode, this);
+               ServerInstance->Modules->AddService(this->mlock);
        }
 
        Version GetVersion()
@@ -36,28 +40,24 @@ public:
                return Version("Implements the ability to have server-side MLOCK enforcement.", VF_VENDOR);
        }
 
-       void Prioritize()
-       {
-               ServerInstance->Modules->SetPriority(this, I_OnPreMode, PRIORITY_FIRST);
-       }
-
-       ModResult OnPreMode(User* source, User* dest, Channel* channel, const std::vector<std::string>& parameters)
+       ModResult OnRawMode(User* source, Channel* channel, const char mode, const std::string& parameter, bool adding, int pcnt)
        {
                if (!channel)
                        return MOD_RES_PASSTHRU;
 
+               if (!IS_LOCAL(source))
+                       return MOD_RES_PASSTHRU;
+
                std::string *mlock_str = mlock.get(channel);
-               if (mlock_str->empty())
+               if (!mlock_str)
                        return MOD_RES_PASSTHRU;
 
-               for (const char *modes = parameters[1].c_str(); *modes; modes++)
+               std::string::size_type p = mlock_str->find(mode);
+               if (p != std::string::npos)
                {
-                       if (mlock_str->find(*modes))
-                       {
-                               source->WriteNumeric(742, "%s %c %s :MODE cannot be set due to channel having an active MLOCK restriction policy",
-                                                    channel->name.c_str(), *modes, mlock_str->c_str());
-                               return MOD_RES_DENY;
-                       }
+                       source->WriteNumeric(742, "%s %c %s :MODE cannot be set due to channel having an active MLOCK restriction policy",
+                                            channel->name.c_str(), mode, mlock_str->c_str());
+                       return MOD_RES_DENY;
                }
 
                return MOD_RES_PASSTHRU;