Fix crash on propogation after routed squit has reached it's destination. Thx HiroP.

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

diff --git a/src/modules/m_noinvite.cpp b/src/modules/m_noinvite.cpp
index 9e4d37ff1b68e89f37d5d1d3f17757837e2c1c92..219c1c78fee0aeff85bb5f3a8e057c64b994151a 100644 (file)
--- a/src/modules/m_noinvite.cpp
+++ b/src/modules/m_noinvite.cpp
@@ -1,32 +1,23 @@
-/*   +------------------------------------+
- *   | Inspire Internet Relay Chat Daemon |
- *   +------------------------------------+
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                   E-mail:
- *            <brain@chatspike.net>
- *            <Craig@chatspike.net>
- * 
- * Written by Craig Edwards, Craig McLure, and others.
  * This program is free but copyrighted software; see
- * the file COPYING for details.
+ *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include <stdio.h>
+#include "inspircd.h"
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
 
-#include "inspircd.h"
-
 /* $ModDesc: Provides support for unreal-style channel mode +V */
 
-
-
 class NoInvite : public ModeHandler
 {
  public:
@@ -57,71 +48,63 @@ class NoInvite : public ModeHandler
 
 class ModuleNoInvite : public Module
 {
-       
        NoInvite *ni;
-       
-       public:
- 
-               ModuleNoInvite(InspIRCd* Me) : Module::Module(Me)
-               {
-                       
-                       ni = new NoInvite(ServerInstance);
-                       ServerInstance->AddMode(ni, 'V');
-               }
-
-               void Implements(char* List)
-               {
-                       List[I_On005Numeric] = List[I_OnUserPreInvite] = 1;
-               }
+ public:
 
-               virtual void On005Numeric(std::string &output)
-               {
-                       ServerInstance->Modes->InsertMode(output,"V",4);
-               }
+       ModuleNoInvite(InspIRCd* Me) : Module(Me)
+       {
+               ni = new NoInvite(ServerInstance);
+               if (!ServerInstance->AddMode(ni, 'V'))
+                       throw ModuleException("Could not add new modes!");
+       }
 
+       void Implements(char* List)
+       {
+               List[I_OnUserPreInvite] = 1;
+       }
 
-               virtual int OnUserPreInvite(userrec* user,userrec* dest,chanrec* channel)
+       virtual int OnUserPreInvite(userrec* user,userrec* dest,chanrec* channel)
+       {
+               if (channel->IsModeSet('V'))
                {
-                       if (channel->IsModeSet('V'))
-                       {
-                               user->WriteServ("492 %s %s :Can't invite %s to channel (+V set)",user->nick, channel->name, dest->nick);
-                               return 1;
-                       }
-                       return 0;
+                       user->WriteServ("492 %s %s :Can't invite %s to channel (+V set)",user->nick, channel->name, dest->nick);
+                       return 1;
                }
+               return 0;
+       }
 
-               virtual ~ModuleNoInvite()
-               {
-                       DELETE(ni);
-               }
+       virtual ~ModuleNoInvite()
+       {
+               ServerInstance->Modes->DelMode(ni);
+               DELETE(ni);
+       }
        
-               virtual Version GetVersion()
-               {
-                       return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
-               }
+       virtual Version GetVersion()
+       {
+               return Version(1,1,0,0,VF_COMMON|VF_VENDOR,API_VERSION);
+       }
 };
 
 
 class ModuleNoInviteFactory : public ModuleFactory
 {
-       public:
-               ModuleNoInviteFactory()
-               {
-               }
-       
-               ~ModuleNoInviteFactory()
-               {
-               }
-       
-               virtual Module * CreateModule(InspIRCd* Me)
-               {
-                       return new ModuleNoInvite(Me);
-               }
-       
+ public:
+       ModuleNoInviteFactory()
+       {
+       }
+
+       ~ModuleNoInviteFactory()
+       {
+       }
+
+       virtual Module * CreateModule(InspIRCd* Me)
+       {
+               return new ModuleNoInvite(Me);
+       }
 };
 
 
-extern "C" void * init_module( void )
+extern "C" DllExport void * init_module( void )
 {
        return new ModuleNoInviteFactory;
 }