Fix these too!

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

diff --git a/src/modules/m_redirect.cpp b/src/modules/m_redirect.cpp
index 0cdde51c8e6adbc932e40887018f7b362e9ef2f3..a3a03b4ceb8bd851471c285df1c8baf5851c69c7 100644 (file)
--- a/src/modules/m_redirect.cpp
+++ b/src/modules/m_redirect.cpp
@@ -1,73 +1,154 @@
-#include <stdio.h>
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  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.
+ *
+ * ---------------------------------------------------
+ */
+
+using namespace std;
 
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
+#include "inspircd.h"
 
 /* $ModDesc: Provides channel mode +L (limit redirection) */
 
-
-class ModuleRedirect : public Module
+/** Handle channel mode +L
+ */
+class Redirect : public ModeHandler
 {
-       Server *Srv;
-       
  public:
- 
-       ModuleRedirect()
+       Redirect(InspIRCd* Instance) : ModeHandler(Instance, 'L', 1, 0, false, MODETYPE_CHANNEL, false) { }
+
+        ModePair ModeSet(userrec* source, userrec* dest, chanrec* channel, const std::string &parameter)
+        {
+                if (channel->IsModeSet('L'))
+                        return std::make_pair(true, channel->GetModeParameter('L'));
+                else
+                        return std::make_pair(false, parameter);
+        }
+
+        bool CheckTimeStamp(time_t theirs, time_t ours, const std::string &their_param, const std::string &our_param, chanrec* channel)
        {
-               Srv = new Server;
-               Srv->AddExtendedMode('L',MT_CHANNEL,false,1,0);
-       }
+               /* When TS is equal, the alphabetically later one wins */
+               return (their_param < our_param);
+        }
        
-       virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list &params)
+       ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
        {
-               
-               if ((modechar == 'L') && (type == MT_CHANNEL))
+               if (adding)
                {
-                       chanrec* c = Srv->Findchannel(params[0]);
+                       chanrec* c = NULL;
+
+                       if (!ServerInstance->IsChannel(parameter.c_str()))
+                       {
+                               source->WriteServ("403 %s %s :Invalid channel name",source->nick, parameter.c_str());
+                               parameter = "";
+                               return MODEACTION_DENY;
+                       }
+
+                       c = ServerInstance->FindChan(parameter);
                        if (c)
                        {
-                               if (c->IsCustomModeSet('L'))
+                               /* Fix by brain: Dont let a channel be linked to *itself* either */
+                               if (IS_LOCAL(source))
                                {
-                                       WriteServ(user->fd,"690 %s :Circular redirection, mode +L to %s not allowed.",user->nick,params[0].c_str());
-                                       return 0;
+                                       if ((c == channel) || (c->IsModeSet('L')))
+                                       {
+                                               source->WriteServ("690 %s :Circular or chained +L to %s not allowed (Channel already has +L). Pack of wild dogs has been unleashed.",source->nick,parameter.c_str());
+                                               parameter = "";
+                                               return MODEACTION_DENY;
+                                       }
+                                       else
+                                       {
+                                               for (chan_hash::const_iterator i = ServerInstance->chanlist.begin(); i != ServerInstance->chanlist.end(); i++)
+                                               {
+                                                       if ((i->second != channel) && (i->second->IsModeSet('L')) && (irc::string(i->second->GetModeParameter('L').c_str()) == irc::string(channel->name)))
+                                                       {
+                                                               source->WriteServ("690 %s :Circular or chained +L to %s not allowed (Already forwarded here from %s). Angry monkeys dispatched.",source->nick,parameter.c_str(),i->second->name);
+                                                               return MODEACTION_DENY;
+                                                       }
+                                               }
+                                       }
                                }
                        }
-                       return 1;
+
+                       channel->SetMode('L', true);
+                       channel->SetModeParam('L', parameter.c_str(), true);
+                       return MODEACTION_ALLOW;
                }
+               else
+               {
+                       if (channel->IsModeSet('L'))
+                       {
+                               channel->SetMode('L', false);
+                               return MODEACTION_ALLOW;
+                       }
+               }
+
+               return MODEACTION_DENY;
+               
        }
+};
+
+class ModuleRedirect : public Module
+{
        
-       virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname)
+       Redirect* re;
+       
+ public:
+ 
+       ModuleRedirect(InspIRCd* Me)
+               : Module::Module(Me)
+       {
+               
+               re = new Redirect(ServerInstance);
+               ServerInstance->AddMode(re, 'L');
+       }
+       
+       void Implements(char* List)
+       {
+               List[I_OnUserPreJoin] = 1;
+       }
+
+       virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname, std::string &privs)
        {
                if (chan)
                {
-                       if (chan->IsCustomModeSet('L'))
+                       if (chan->IsModeSet('L') && chan->limit)
                        {
-                               if (chan->limit >= Srv->CountUsers(chan))
+                               if (chan->GetUserCounter() >= chan->limit)
                                {
-                                       char* channel = chan->GetModeParameter('L');
-                                       WriteServ(user->fd,"470 %s :%s has become full, so you are automatically being transferred to the linked channel %s",user->nick,cname,channel);
-                                       Srv->JoinUserToChannel(user,channel,"");
+                                       std::string channel = chan->GetModeParameter('L');
+                                       user->WriteServ("470 %s :%s has become full, so you are automatically being transferred to the linked channel %s",user->nick,cname,channel.c_str());
+                                       chanrec::JoinUser(ServerInstance, user, channel.c_str(), false);
                                        return 1;
                                }
                        }
                }
+               return 0;
        }
 
        virtual ~ModuleRedirect()
        {
-               delete Srv;
+               ServerInstance->Modes->DelMode(re);
+               DELETE(re);
        }
        
        virtual Version GetVersion()
        {
-               return Version(1,0,0,0);
+               return Version(1, 1, 0, 0, VF_COMMON | VF_VENDOR, API_VERSION);
        }
-       
-       virtual void OnUserConnect(userrec* user)
-       {
-       }
-
 };
 
 
@@ -82,9 +163,9 @@ class ModuleRedirectFactory : public ModuleFactory
        {
        }
        
-       virtual Module * CreateModule()
+       virtual Module * CreateModule(InspIRCd* Me)
        {
-               return new ModuleRedirect;
+               return new ModuleRedirect(Me);
        }
        
 };