src/modules/m_spanningtree/rsquit.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
   6  * See: http://www.inspircd.org/wiki/index.php/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "commands/cmd_whois.h"
  16 #include "commands/cmd_stats.h"
  17 #include "socket.h"
  18 #include "wildcard.h"
  19 #include "xline.h"
  20 #include "transport.h"
  21
  22 #include "m_spanningtree/timesynctimer.h"
  23 #include "m_spanningtree/resolvers.h"
  24 #include "m_spanningtree/main.h"
  25 #include "m_spanningtree/utils.h"
  26 #include "m_spanningtree/treeserver.h"
  27 #include "m_spanningtree/link.h"
  28 #include "m_spanningtree/treesocket.h"
  29 #include "m_spanningtree/rsquit.h"
  30
  31 /* $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 */
  32
  33 cmd_rsquit::cmd_rsquit (InspIRCd* Instance, Module* Callback, SpanningTreeUtilities* Util) : Command(Instance, "RSQUIT", "o", 1), Creator(Callback), Utils(Util)
  34 {
  35         this->source = "m_spanningtree.so";
  36         syntax = "<target-server-mask> [reason]";
  37 }
  38
  39 CmdResult cmd_rsquit::Handle (const std::vector<std::string>& parameters, User *user)
  40 {
  41         TreeServer *server_target; // Server to squit
  42         TreeServer *server_linked; // Server target is linked to
  43
  44         server_target = Utils->FindServerMask(parameters[0]);
  45         if (!server_target)
  46         {
  47                 user->WriteServ("NOTICE %s :*** RSQUIT: Server \002%s\002 isn't connected to the network!", user->nick, parameters[0].c_str());
  48                 return CMD_FAILURE;
  49         }
  50
  51         if (server_target == Utils->TreeRoot)
  52         {
  53                 NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+parameters[0]+" matches local server name)");
  54                 return CMD_FAILURE;
  55         }
  56
  57         server_linked = server_target->GetParent();
  58
  59         if (server_linked == Utils->TreeRoot)
  60         {
  61                 // We have been asked to remove server_target.
  62                 TreeSocket* sock = server_target->GetSocket();
  63                 if (sock)
  64                 {
  65                         const char *reason = parameters.size() == 2 ? parameters[1].c_str() : "No reason";
  66                         ServerInstance->SNO->WriteToSnoMask('l',"RSQUIT: Server \002%s\002 removed from network by %s (%s)", parameters[0].c_str(), user->nick, reason);
  67                         sock->Squit(server_target, std::string("Server quit by ") + user->GetFullRealHost() + " (" + reason + ")");
  68                         ServerInstance->SE->DelFd(sock);
  69                         sock->Close();
  70                         return CMD_LOCALONLY;
  71                 }
  72         }
  73
  74         return CMD_SUCCESS;
  75 }
  76
  77 // XXX use protocol interface instead of rolling our own :)
  78 void cmd_rsquit::NoticeUser(User* user, const std::string &msg)
  79 {
  80         if (IS_LOCAL(user))
  81         {
  82                 user->WriteServ("NOTICE %s :%s",user->nick,msg.c_str());
  83         }
  84         else
  85         {
  86                 std::deque<std::string> params;
  87                 params.push_back(user->nick);
  88                 params.push_back("NOTICE "+ConvToStr(user->nick)+" :"+msg);
  89                 Utils->DoOneToOne(ServerInstance->Config->GetSID(), "PUSH", params, user->server);
  90         }
  91 }
  92