fix some unitialised vectors and tidy up a bit.

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

diff --git a/src/modules/m_quitban.cpp b/src/modules/m_quitban.cpp
index 00a47beebfd8696526933518dae3e7c7266639a3..7cd5742d808cfee621d59502410cb5a76a430d2d 100644 (file)
--- a/src/modules/m_quitban.cpp
+++ b/src/modules/m_quitban.cpp
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
  * See: http://www.inspircd.org/wiki/index.php/Credits
  *
  * This program is free but copyrighted software; see
@@ -14,29 +14,29 @@
 #include "inspircd.h"
 #include "xline.h"
 
-/* $ModDesc: Creates a snomask with notices whenever a new channel is created */
+/* $ModDesc: Throttles the connections of any users who try quitflood the server */
 
-class ModuleChanCreate : public Module
+class ModuleQuitBan : public Module
 {
  private:
        clonemap quits;
        unsigned int threshold;
        unsigned int banduration;
  public:
-       ModuleChanCreate(InspIRCd* Me) : Module(Me)
+       ModuleQuitBan(InspIRCd* Me) : Module(Me)
        {
                Implementation eventlist[] = { I_OnUserDisconnect, I_OnGarbageCollect, I_OnRehash };
                ServerInstance->Modules->Attach(eventlist, this, 3);
                OnRehash(NULL, "");
        }
        
-       virtual ~ModuleChanCreate()
+       virtual ~ModuleQuitBan()
        {
        }
        
        virtual Version GetVersion()
        {
-               return Version(1,1,0,0,VF_VENDOR,API_VERSION);
+               return Version(1,2,0,0,VF_VENDOR,API_VERSION);
        }
 
        virtual void OnRehash(User* user, const std::string &parameter)
@@ -64,7 +64,7 @@ class ModuleChanCreate : public Module
                if (i != quits.end())
                {
                        i->second++;
-                       ServerInstance->Log(DEBUG, "quitban: Count for IP is now %d", i->second);
+                       ServerInstance->Logs->Log("m_quitban",DEBUG, "quitban: Count for IP is now %d", i->second);
 
                        if (i->second >= threshold)
                        {
@@ -76,20 +76,21 @@ class ModuleChanCreate : public Module
                                        delete zl;
 
                                ServerInstance->SNO->WriteToSnoMask('x', "Quit flooding from IP %s (%d)", u->GetIPString(), threshold);
+                               quits.erase(i);
                        }
                }
                else
                {
                        quits[u->GetIPString()] = 1;
-                       ServerInstance->Log(DEBUG, "quitban: Added new record");
+                       ServerInstance->Logs->Log("m_quitban",DEBUG, "quitban: Added new record");
                }
        }
 
        virtual void OnGarbageCollect()
        {
-               ServerInstance->Log(DEBUG, "quitban: Clearing map.");
+               ServerInstance->Logs->Log("m_quitban",DEBUG, "quitban: Clearing map.");
                quits.clear();
        }
 };
 
-MODULE_INIT(ModuleChanCreate)
+MODULE_INIT(ModuleQuitBan)