]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree/rsquit.cpp
Replace std::deque with std::vector in spanningtree and related modules
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / rsquit.cpp
index cfe2a8c90e11995ae7a3bc0eadfda407a58a175b..a5dd317528993251ea6b7d3bec83cc9f93c12523 100644 (file)
@@ -2,8 +2,8 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  */
 
 #include "inspircd.h"
-#include "configreader.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
 #include "commands/cmd_whois.h"
 #include "commands/cmd_stats.h"
 #include "socket.h"
-#include "wildcard.h"
 #include "xline.h"
-#include "transport.h"
+#include "../transport.h"
 
-#include "m_spanningtree/timesynctimer.h"
-#include "m_spanningtree/resolvers.h"
-#include "m_spanningtree/main.h"
-#include "m_spanningtree/utils.h"
-#include "m_spanningtree/treeserver.h"
-#include "m_spanningtree/link.h"
-#include "m_spanningtree/treesocket.h"
-#include "m_spanningtree/rsquit.h"
+#include "main.h"
+#include "utils.h"
+#include "treeserver.h"
+#include "treesocket.h"
+#include "rsquit.h"
 
-/* $ModDep: m_spanningtree/timesynctimer.h m_spanningtree/resolvers.h m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/link.h m_spanningtree/treesocket.h m_spanningtree/rsquit.h */
+/* $ModDep: m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/treesocket.h m_spanningtree/rsquit.h */
 
-cmd_rsquit::cmd_rsquit (InspIRCd* Instance, Module* Callback, SpanningTreeUtilities* Util) : command_t(Instance, "RSQUIT", 'o', 1), Creator(Callback), Utils(Util)
+CommandRSQuit::CommandRSQuit (InspIRCd* Instance, Module* Callback, SpanningTreeUtilities* Util) : Command(Instance, "RSQUIT", "o", 1), Creator(Callback), Utils(Util)
 {
        this->source = "m_spanningtree.so";
-       syntax = "<remote-server-mask> [target-server-mask]";
+       syntax = "<target-server-mask> [reason]";
 }
 
-CmdResult cmd_rsquit::Handle (const char** parameters, int pcnt, userrec *user)
+CmdResult CommandRSQuit::Handle (const std::vector<std::string>& parameters, User *user)
 {
-       if (IS_LOCAL(user))
+       TreeServer *server_target; // Server to squit
+       TreeServer *server_linked; // Server target is linked to
+
+       server_target = Utils->FindServerMask(parameters[0]);
+       if (!server_target)
        {
-               if (!Utils->FindServerMask(parameters[0]))
-               {
-                       user->WriteServ("NOTICE %s :*** RSQUIT: Server \002%s\002 isn't connected to the network!", user->nick, parameters[0]);
-                       return CMD_FAILURE;
-               }
-               if (pcnt > 1)
-                       user->WriteServ("NOTICE %s :*** RSQUIT: Sending remote squit to \002%s\002 to squit server \002%s\002.",user->nick,parameters[0],parameters[1]);
-               else
-                       user->WriteServ("NOTICE %s :*** RSQUIT: Sending remote squit for server \002%s\002.",user->nick,parameters[0]);
+               user->WriteServ("NOTICE %s :*** RSQUIT: Server \002%s\002 isn't connected to the network!", user->nick.c_str(), parameters[0].c_str());
+               return CMD_FAILURE;
        }
 
-       TreeServer* s = (pcnt > 1) ? Utils->FindServerMask(parameters[1]) : Utils->FindServerMask(parameters[0]);
-
-       if (pcnt > 1)
+       if (server_target == Utils->TreeRoot)
        {
-               if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameters[0]))
-               {
-                       if (s == Utils->TreeRoot)
-                       {
-                               NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+ConvToStr(parameters[1])+" matches local server name)");
-                               return CMD_FAILURE;
-                       }
-                       TreeSocket* sock = s->GetSocket();
-                       if (!sock)
-                       {
-                               NoticeUser(user, "*** RSQUIT: Server \002"+ConvToStr(parameters[1])+"\002 isn't connected to \002"+ConvToStr(parameters[0])+"\002.");
-                               return CMD_FAILURE;
-                       }
-                       ServerInstance->SNO->WriteToSnoMask('l',"Remote SQUIT from %s matching \002%s\002, squitting server \002%s\002",user->nick,parameters[0],parameters[1]);
-                       const char* para[1];
-                       para[0] = parameters[1];
-                       std::string original_command = std::string("SQUIT ") + parameters[1];
-                       Creator->OnPreCommand("SQUIT", para, 1, user, true, original_command);
-               }
+               NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+parameters[0]+" matches local server name)");
+               return CMD_FAILURE;
        }
-       else
+
+       server_linked = server_target->GetParent();
+
+       if (server_linked == Utils->TreeRoot)
        {
-               if (s == Utils->TreeRoot)
-               {
-                       NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+ConvToStr(parameters[0])+" matches local server name)");
-                       return CMD_FAILURE;
-               }
-               TreeSocket* sock = s->GetSocket();
+               // We have been asked to remove server_target.
+               TreeSocket* sock = server_target->GetSocket();
                if (sock)
                {
-                       ServerInstance->SNO->WriteToSnoMask('l',"RSQUIT: Server \002%s\002 removed from network by %s",parameters[0],user->nick);
-                       sock->Squit(s,std::string("Server quit by ") + user->GetFullRealHost());
+                       const char *reason = parameters.size() == 2 ? parameters[1].c_str() : "No reason";
+                       ServerInstance->SNO->WriteToSnoMask('l',"RSQUIT: Server \002%s\002 removed from network by %s (%s)", parameters[0].c_str(), user->nick.c_str(), reason);
+                       sock->Squit(server_target, std::string("Server quit by ") + user->GetFullRealHost() + " (" + reason + ")");
                        ServerInstance->SE->DelFd(sock);
                        sock->Close();
-                       delete sock;
+                       return CMD_LOCALONLY;
                }
        }
 
        return CMD_SUCCESS;
 }
 
-void cmd_rsquit::NoticeUser(userrec* user, const std::string &msg)
+// XXX use protocol interface instead of rolling our own :)
+void CommandRSQuit::NoticeUser(User* user, const std::string &msg)
 {
        if (IS_LOCAL(user))
        {
-               user->WriteServ("NOTICE %s :%s",user->nick,msg.c_str());
+               user->WriteServ("NOTICE %s :%s",user->nick.c_str(),msg.c_str());
        }
        else
        {
-               std::deque<std::string> params;
+               parameterlist params;
                params.push_back(user->nick);
                params.push_back("NOTICE "+ConvToStr(user->nick)+" :"+msg);
-               Utils->DoOneToOne(ServerInstance->Config->ServerName, "PUSH", params, user->server);
+               Utils->DoOneToOne(ServerInstance->Config->GetSID(), "PUSH", params, user->server);
        }
 }
+