src/modules/m_spanningtree/rsquit.cpp

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