Delete modewatchers when unloading modules that use them to keep the server from...

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

diff --git a/src/modules/m_connectban.cpp b/src/modules/m_connectban.cpp
index b676fa681df8bd8506a878688a46fd857965f851..276962b2dc1b4c6be889cb8d226cf6b0e6d3779f 100644 (file)
--- a/src/modules/m_connectban.cpp
+++ b/src/modules/m_connectban.cpp
@@ -37,7 +37,7 @@ class ModuleConnectBan : public Module
 
        virtual Version GetVersion()
        {
-               return Version("Throttles the connections of any users who try connect flood", VF_VENDOR,API_VERSION);
+               return Version("Throttles the connections of any users who try connect flood", VF_VENDOR);
        }
 
        virtual void OnRehash(User* user)
@@ -66,7 +66,7 @@ class ModuleConnectBan : public Module
                banduration = ServerInstance->Duration(duration);
        }
 
-       virtual void OnUserConnect(User *u)
+       virtual void OnUserConnect(LocalUser *u)
        {
                int range = 32;
                clonemap::iterator i;
@@ -81,7 +81,8 @@ class ModuleConnectBan : public Module
                        break;
                }
 
-               i = connects.find(u->GetCIDRMask(range));
+               irc::sockets::cidr_mask mask(u->client_sa, range);
+               i = connects.find(mask);
 
                if (i != connects.end())
                {
@@ -90,21 +91,21 @@ class ModuleConnectBan : public Module
                        if (i->second >= threshold)
                        {
                                // Create zline for set duration.
-                               ZLine* zl = new ZLine(ServerInstance->Time(), banduration, ServerInstance->Config->ServerName.c_str(), "Connect flooding", u->GetCIDRMask(range));
+                               ZLine* zl = new ZLine(ServerInstance->Time(), banduration, ServerInstance->Config->ServerName.c_str(), "Connect flooding", mask.str().c_str());
                                if (ServerInstance->XLines->AddLine(zl,NULL))
                                        ServerInstance->XLines->ApplyLines();
                                else
                                        delete zl;
 
                                ServerInstance->SNO->WriteGlobalSno('x',"Module m_connectban added Z:line on *@%s to expire on %s: Connect flooding", 
-                                       u->GetCIDRMask(range), ServerInstance->TimeString(zl->expiry).c_str());
-                               ServerInstance->SNO->WriteGlobalSno('a', "Connect flooding from IP range %s (%d)", u->GetCIDRMask(range), threshold);
+                                       mask.str().c_str(), ServerInstance->TimeString(zl->expiry).c_str());
+                               ServerInstance->SNO->WriteGlobalSno('a', "Connect flooding from IP range %s (%d)", mask.str().c_str(), threshold);
                                connects.erase(i);
                        }
                }
                else
                {
-                       connects[u->GetCIDRMask(range)] = 1;
+                       connects[mask] = 1;
                }
        }