Fix new millisec /map to compile on windows, by ifndef gettimeofday out reverting...

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

diff --git a/src/modules/m_banredirect.cpp b/src/modules/m_banredirect.cpp
index 88d4872cef4c2102d73a99b8da5539acbfa2fbaa..1764e6f56689b7f4dfb74af3804c6a2a9aa24aa1 100644 (file)
--- a/src/modules/m_banredirect.cpp
+++ b/src/modules/m_banredirect.cpp
@@ -11,11 +11,12 @@
  * ---------------------------------------------------
  */
 
+#include "inspircd.h"
 #include "mode.h"
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
-#include "inspircd.h"
+#include "u_listmode.h"
 
 /* $ModDesc: Allows an extended ban (+b) syntax redirecting banned users to another channel */
 
@@ -186,21 +187,25 @@ class BanRedirect : public ModeWatcher
 class ModuleBanRedirect : public Module
 {
        BanRedirect* re;
-       InspIRCd* Srv;
-       
+       bool nofollow;
+       Module* ExceptionModule;
+
  public:
        ModuleBanRedirect(InspIRCd* Me)
-       : Module::Module(Me), Srv(Me)
+       : Module(Me)
        {
                re = new BanRedirect(Me);
+               nofollow = false;
                
-               if(!Srv->AddModeWatcher(re))
+               if(!ServerInstance->AddModeWatcher(re))
                        throw ModuleException("Could not add mode watcher");
+
+               OnRehash(NULL, "");
        }
        
        void Implements(char* List)
        {
-               List[I_OnUserPreJoin] = List[I_OnChannelDelete] = List[I_OnCleanup] = 1;
+               List[I_OnRehash] = List[I_OnUserPreJoin] = List[I_OnChannelDelete] = List[I_OnCleanup] = 1;
        }
        
        virtual void OnChannelDelete(chanrec* chan)
@@ -219,8 +224,8 @@ class ModuleBanRedirect : public Module
                        {
                                irc::modestacker modestack(false);
                                StringDeque stackresult;
-                               const char* mode_junk[MAXMODES+1];
-                               userrec* myhorriblefakeuser = new userrec(Srv);
+                               const char* mode_junk[MAXMODES+2];
+                               userrec* myhorriblefakeuser = new userrec(ServerInstance);
                                myhorriblefakeuser->SetFd(FD_MAGIC_NUMBER);
                                
                                mode_junk[0] = chan->name;
@@ -243,7 +248,7 @@ class ModuleBanRedirect : public Module
                                                mode_junk[i+1] = stackresult[i].c_str();
                                        }
                                        
-                                       Srv->SendMode(mode_junk, stackresult.size() + 1, myhorriblefakeuser);
+                                       ServerInstance->SendMode(mode_junk, stackresult.size() + 1, myhorriblefakeuser);
                                }
                                
                                DELETE(myhorriblefakeuser);
@@ -253,8 +258,19 @@ class ModuleBanRedirect : public Module
                }
        }
 
+       virtual void OnRehash(userrec* user, const std::string &param)
+       {
+               ExceptionModule = ServerInstance->FindModule("m_banexception.so");
+       }
+
        virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname, std::string &privs)
        {
+               /* This prevents recursion when a user sets multiple ban redirects in a chain
+                * (thanks Potter)
+                */
+               if (nofollow)
+                       return 0;
+
                /* Return 1 to prevent the join, 0 to allow it */
                if (chan)
                {
@@ -267,17 +283,26 @@ class ModuleBanRedirect : public Module
                                /* This was replaced with user->MakeHostIP() when I had a snprintf(), but MakeHostIP() doesn't seem to add the nick.
                                 * Maybe we should have a GetFullIPHost() or something to match GetFullHost() and GetFullRealHost?
                                 */
+
+                               if (ExceptionModule)
+                               {
+                                       ListModeRequest n(this, ExceptionModule, user, chan);
+                                       /* Users with ban exceptions are allowed to join without being redirected */
+                                       if (n.Send())
+                                               return 0;
+                               }
+
                                std::string ipmask(user->nick);
                                ipmask.append(1, '!').append(user->MakeHostIP());
                                
                                for(BanRedirectList::iterator redir = redirects->begin(); redir != redirects->end(); redir++)
                                {
-                                       if(Srv->MatchText(user->GetFullRealHost(), redir->banmask) || Srv->MatchText(user->GetFullHost(), redir->banmask) || Srv->MatchText(ipmask, redir->banmask))
+                                       if(ServerInstance->MatchText(user->GetFullRealHost(), redir->banmask) || ServerInstance->MatchText(user->GetFullHost(), redir->banmask) || ServerInstance->MatchText(ipmask, redir->banmask))
                                        {
                                                /* tell them they're banned and are being transferred */
-                                               chanrec* destchan = Srv->FindChan(redir->targetchan);
+                                               chanrec* destchan = ServerInstance->FindChan(redir->targetchan);
                                                
-                                               if(destchan && Srv->FindModule("m_redirect.so") && destchan->IsModeSet('L') && destchan->limit && (destchan->GetUserCounter() >= destchan->limit))
+                                               if(destchan && ServerInstance->FindModule("m_redirect.so") && destchan->IsModeSet('L') && destchan->limit && (destchan->GetUserCounter() >= destchan->limit))
                                                {
                                                        user->WriteServ("474 %s %s :Cannot join channel (You are banned)", user->nick, chan->name);
                                                        return 1;
@@ -285,7 +310,9 @@ class ModuleBanRedirect : public Module
                                                else
                                                {
                                                        user->WriteServ("470 %s :You are banned from %s. You are being automatically redirected to %s", user->nick, chan->name, redir->targetchan.c_str());
-                                                       chanrec::JoinUser(Srv, user, redir->targetchan.c_str(), false);
+                                                       nofollow = true;
+                                                       chanrec::JoinUser(ServerInstance, user, redir->targetchan.c_str(), false, "", ServerInstance->Time(true));
+                                                       nofollow = false;
                                                        return 1;
                                                }
                                        }
@@ -297,7 +324,7 @@ class ModuleBanRedirect : public Module
 
        virtual ~ModuleBanRedirect()
        {
-               Srv->Modes->DelModeWatcher(re);
+               ServerInstance->Modes->DelModeWatcher(re);
                DELETE(re);
        }
        
@@ -305,29 +332,12 @@ class ModuleBanRedirect : public Module
        {
                return Version(1, 0, 0, 0, VF_COMMON | VF_VENDOR, API_VERSION);
        }
-};
-
-
-class ModuleBanRedirectFactory : public ModuleFactory
-{
- public:
-       ModuleBanRedirectFactory()
-       {
-       }
-       
-       ~ModuleBanRedirectFactory()
-       {
-       }
        
-       virtual Module * CreateModule(InspIRCd* Me)
+       Priority Prioritize()
        {
-               return new ModuleBanRedirect(Me);
+               return (Priority)ServerInstance->PriorityBefore("m_banexception.so");
        }
-       
 };
 
 
-extern "C" void * init_module( void )
-{
-       return new ModuleBanRedirectFactory;
-}
+MODULE_INIT(ModuleBanRedirect)