]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/mode.cpp
Switch <stdint.h> test to use a test file too.
[user/henk/code/inspircd.git] / src / mode.cpp
index 5efc98250d7e1851b93d88cce54d6a900e1e6145..89ff37fa1e7fd0b709bd411c220abdf15cf6b1f4 100644 (file)
@@ -1,6 +1,7 @@
 /*
  * InspIRCd -- Internet Relay Chat Daemon
  *
+ *   Copyright (C) 2012 Shawn Smith <shawn@inspircd.org>
  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
  *   Copyright (C) 2007, 2009 Dennis Friis <peavey@inspircd.org>
  *   Copyright (C) 2006-2008 Robin Burchell <robin+git@viroteck.net>
 
 
 #include "inspircd.h"
-#include "inspstring.h"
 
 /* +s (secret) */
-#include "modes/cmode_s.h"
 /* +p (private) */
-#include "modes/cmode_p.h"
-/* +b (bans) */
-#include "modes/cmode_b.h"
 /* +m (moderated) */
-#include "modes/cmode_m.h"
 /* +t (only (half) ops can change topic) */
-#include "modes/cmode_t.h"
 /* +n (no external messages) */
-#include "modes/cmode_n.h"
 /* +i (invite only) */
-#include "modes/cmode_i.h"
+/* +w (see wallops) */
+/* +i (invisible) */
+#include "modes/simplemodes.h"
+/* +b (bans) */
+#include "modes/cmode_b.h"
 /* +k (keyed channel) */
 #include "modes/cmode_k.h"
 /* +l (channel user limit) */
 #include "modes/cmode_o.h"
 /* +v (channel voice) */
 #include "modes/cmode_v.h"
-/* +w (see wallops) */
-#include "modes/umode_w.h"
-/* +i (invisible) */
-#include "modes/umode_i.h"
 /* +o (operator) */
 #include "modes/umode_o.h"
 /* +s (server notice masks) */
@@ -132,47 +125,39 @@ bool ModeHandler::ResolveModeConflict(std::string& theirs, const std::string& ou
 
 ModeAction SimpleUserModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
 {
-       if (adding)
-       {
-               if (!dest->IsModeSet(this->GetModeChar()))
-               {
-                       dest->SetMode(this->GetModeChar(),true);
-                       return MODEACTION_ALLOW;
-               }
-       }
-       else
-       {
-               if (dest->IsModeSet(this->GetModeChar()))
-               {
-                       dest->SetMode(this->GetModeChar(),false);
-                       return MODEACTION_ALLOW;
-               }
-       }
+       /* We're either trying to add a mode we already have or
+               remove a mode we don't have, deny. */
+       if (dest->IsModeSet(this->GetModeChar()) == adding)
+               return MODEACTION_DENY;
 
-       return MODEACTION_DENY;
+       /* adding will be either true or false, depending on if we
+               are adding or removing the mode, since we already checked
+               to make sure we aren't adding a mode we have or that we
+               aren't removing a mode we don't have, we don't have to do any
+               other checks here to see if it's true or false, just add or
+               remove the mode */
+       dest->SetMode(this->GetModeChar(), adding);
+
+       return MODEACTION_ALLOW;
 }
 
 
 ModeAction SimpleChannelModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
 {
-       if (adding)
-       {
-               if (!channel->IsModeSet(this->GetModeChar()))
-               {
-                       channel->SetMode(this->GetModeChar(),true);
-                       return MODEACTION_ALLOW;
-               }
-       }
-       else
-       {
-               if (channel->IsModeSet(this->GetModeChar()))
-               {
-                       channel->SetMode(this->GetModeChar(),false);
-                       return MODEACTION_ALLOW;
-               }
-       }
+       /* We're either trying to add a mode we already have or
+               remove a mode we don't have, deny. */
+       if (channel->IsModeSet(this->GetModeChar()) == adding)
+               return MODEACTION_DENY;
 
-       return MODEACTION_DENY;
+       /* adding will be either true or false, depending on if we
+               are adding or removing the mode, since we already checked
+               to make sure we aren't adding a mode we have or that we
+               aren't removing a mode we don't have, we don't have to do any
+               other checks here to see if it's true or false, just add or
+               remove the mode */
+       channel->SetMode(this->GetModeChar(), adding);
+
+       return MODEACTION_ALLOW;
 }
 
 ModeAction ParamChannelModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
@@ -361,10 +346,19 @@ ModeAction ModeParser::TryMode(User* user, User* targetuser, Channel* chan, bool
                return MODEACTION_DENY;
        }
 
-       if (mh->GetTranslateType() == TR_NICK && !ServerInstance->FindNick(parameter))
+       if (mh->GetTranslateType() == TR_NICK)
        {
-               user->WriteNumeric(ERR_NOSUCHNICK, "%s %s :No such nick/channel", user->nick.c_str(), parameter.c_str());
-               return MODEACTION_DENY;
+               User* prefixtarget;
+               if (IS_LOCAL(user))
+                       prefixtarget = ServerInstance->FindNickOnly(parameter);
+               else
+                       prefixtarget = ServerInstance->FindNick(parameter);
+
+               if (!prefixtarget)
+               {
+                       user->WriteNumeric(ERR_NOSUCHNICK, "%s %s :No such nick/channel", user->nick.c_str(), parameter.c_str());
+                       return MODEACTION_DENY;
+               }
        }
 
        if (mh->GetPrefixRank() && chan)
@@ -393,16 +387,23 @@ ModeAction ModeParser::TryMode(User* user, User* targetuser, Channel* chan, bool
 
 void ModeParser::Process(const std::vector<std::string>& parameters, User *user, bool merge)
 {
-       std::string target = parameters[0];
+       const std::string& target = parameters[0];
        Channel* targetchannel = ServerInstance->FindChan(target);
-       User* targetuser  = ServerInstance->FindNick(target);
+       User* targetuser = NULL;
+       if (!targetchannel)
+       {
+               if (IS_LOCAL(user))
+                       targetuser = ServerInstance->FindNickOnly(target);
+               else
+                       targetuser = ServerInstance->FindNick(target);
+       }
        ModeType type = targetchannel ? MODETYPE_CHANNEL : MODETYPE_USER;
 
        LastParse.clear();
        LastParseParams.clear();
        LastParseTranslate.clear();
 
-       if (!targetchannel && !targetuser)
+       if ((!targetchannel) && ((!targetuser) || (IS_SERVER(targetuser))))
        {
                user->WriteNumeric(ERR_NOSUCHNICK, "%s %s :No such nick/channel",user->nick.c_str(),target.c_str());
                return;
@@ -457,7 +458,7 @@ void ModeParser::Process(const std::vector<std::string>& parameters, User *user,
                        continue;
                }
 
-               std::string parameter = "";
+               std::string parameter;
                int pcnt = mh->GetNumParams(adding);
                if (pcnt && param_at == parameters.size())
                {
@@ -469,7 +470,7 @@ void ModeParser::Process(const std::vector<std::string>& parameters, User *user,
                {
                        parameter = parameters[param_at++];
                        /* Make sure the user isn't trying to slip in an invalid parameter */
-                       if ((parameter.find(':') == 0) || (parameter.rfind(' ') != std::string::npos))
+                       if ((parameter.empty()) || (parameter.find(':') == 0) || (parameter.rfind(' ') != std::string::npos))
                                continue;
                        if (merge && targetchannel && targetchannel->IsModeSet(modechar) && !mh->IsListMode())
                        {
@@ -690,15 +691,20 @@ bool ModeParser::DelMode(ModeHandler* mh)
        switch (mh->GetModeType())
        {
                case MODETYPE_USER:
-                       for (user_hash::iterator i = ServerInstance->Users->clientlist->begin(); i != ServerInstance->Users->clientlist->end(); i++)
+                       for (user_hash::iterator i = ServerInstance->Users->clientlist->begin(); i != ServerInstance->Users->clientlist->end(); )
                        {
-                               mh->RemoveMode(i->second);
+                               User* user = i->second;
+                               ++i;
+                               mh->RemoveMode(user);
                        }
                break;
                case MODETYPE_CHANNEL:
-                       for (chan_hash::iterator i = ServerInstance->chanlist->begin(); i != ServerInstance->chanlist->end(); i++)
+                       for (chan_hash::iterator i = ServerInstance->chanlist->begin(); i != ServerInstance->chanlist->end(); )
                        {
-                               mh->RemoveMode(i->second);
+                               // The channel may not be in the hash after RemoveMode(), see m_permchannels
+                               Channel* chan = i->second;
+                               ++i;
+                               mh->RemoveMode(chan);
                        }
                break;
        }
@@ -890,7 +896,7 @@ bool ModeParser::DelModeWatcher(ModeWatcher* mw)
        mw->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
        pos = (mw->GetModeChar()-65) | mask;
 
-       ModeWatchIter a = find(modewatchers[pos].begin(),modewatchers[pos].end(),mw);
+       ModeWatchIter a = std::find(modewatchers[pos].begin(),modewatchers[pos].end(),mw);
 
        if (a == modewatchers[pos].end())
        {
@@ -906,9 +912,6 @@ bool ModeParser::DelModeWatcher(ModeWatcher* mw)
  */
 void ModeHandler::RemoveMode(User* user, irc::modestacker* stack)
 {
-       char moderemove[MAXBUF];
-       std::vector<std::string> parameters;
-
        if (user->IsModeSet(this->GetModeChar()))
        {
                if (stack)
@@ -917,9 +920,10 @@ void ModeHandler::RemoveMode(User* user, irc::modestacker* stack)
                }
                else
                {
-                       sprintf(moderemove,"-%c",this->GetModeChar());
+                       std::vector<std::string> parameters;
                        parameters.push_back(user->nick);
-                       parameters.push_back(moderemove);
+                       parameters.push_back("-");
+                       parameters[1].push_back(this->GetModeChar());
                        ServerInstance->Modes->Process(parameters, ServerInstance->FakeClient);
                }
        }
@@ -930,9 +934,6 @@ void ModeHandler::RemoveMode(User* user, irc::modestacker* stack)
  */
 void ModeHandler::RemoveMode(Channel* channel, irc::modestacker* stack)
 {
-       char moderemove[MAXBUF];
-       std::vector<std::string> parameters;
-
        if (channel->IsModeSet(this->GetModeChar()))
        {
                if (stack)
@@ -941,9 +942,10 @@ void ModeHandler::RemoveMode(Channel* channel, irc::modestacker* stack)
                }
                else
                {
-                       sprintf(moderemove,"-%c",this->GetModeChar());
+                       std::vector<std::string> parameters;
                        parameters.push_back(channel->name);
-                       parameters.push_back(moderemove);
+                       parameters.push_back("-");
+                       parameters[1].push_back(this->GetModeChar());
                        ServerInstance->SendMode(parameters, ServerInstance->FakeClient);
                }
        }