src/modules/m_spanningtree/rsquit.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2007 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 = "<remote-server-mask> [target-server-mask]";
  37 }
  38
  39 CmdResult cmd_rsquit::Handle (const char** parameters, int pcnt, User *user)
  40 {
  41         if (IS_LOCAL(user))
  42         {
  43                 if (!Utils->FindServerMask(parameters[0]))
  44                 {
  45                         user->WriteServ("NOTICE %s :*** RSQUIT: Server \002%s\002 isn't connected to the network!", user->nick, parameters[0]);
  46                         return CMD_FAILURE;
  47                 }
  48                 if (pcnt > 1)
  49                         user->WriteServ("NOTICE %s :*** RSQUIT: Sending remote squit to \002%s\002 to squit server \002%s\002.",user->nick,parameters[0],parameters[1]);
  50                 else
  51                         user->WriteServ("NOTICE %s :*** RSQUIT: Sending remote squit for server \002%s\002.",user->nick,parameters[0]);
  52         }
  53
  54         TreeServer* s = (pcnt > 1) ? Utils->FindServerMask(parameters[1]) : Utils->FindServerMask(parameters[0]);
  55
  56         if (pcnt > 1)
  57         {
  58                 if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameters[0]))
  59                 {
  60                         if (s)
  61                         {
  62                                 if (s == Utils->TreeRoot)
  63                                 {
  64                                         NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+ConvToStr(parameters[1])+" matches local server name)");
  65                                         return CMD_FAILURE;
  66                                 }
  67                                 TreeSocket* sock = s->GetSocket();
  68                                 if (!sock)
  69                                 {
  70                                         NoticeUser(user, "*** RSQUIT: Server \002"+ConvToStr(parameters[1])+"\002 isn't connected to \002"+ConvToStr(parameters[0])+"\002.");
  71                                         return CMD_FAILURE;
  72                                 }
  73                                 ServerInstance->SNO->WriteToSnoMask('l',"Remote SQUIT from %s matching \002%s\002, squitting server \002%s\002",user->nick,parameters[0],parameters[1]);
  74                                 const char* para[1];
  75                                 para[0] = parameters[1];
  76                                 std::string original_command = std::string("SQUIT ") + parameters[1];
  77                                 Creator->OnPreCommand("SQUIT", para, 1, user, true, original_command);
  78                                 return CMD_LOCALONLY;
  79                         }
  80                 }
  81         }
  82         else
  83         {
  84                 if (s)
  85                 {
  86                         if (s == Utils->TreeRoot)
  87                         {
  88                                 NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+ConvToStr(parameters[0])+" matches local server name)");
  89                                 return CMD_FAILURE;
  90                         }
  91                         TreeSocket* sock = s->GetSocket();
  92                         if (sock)
  93                         {
  94                                 ServerInstance->SNO->WriteToSnoMask('l',"RSQUIT: Server \002%s\002 removed from network by %s",parameters[0],user->nick);
  95                                 sock->Squit(s,std::string("Server quit by ") + user->GetFullRealHost());
  96                                 ServerInstance->SE->DelFd(sock);
  97                                 sock->Close();
  98                                 return CMD_LOCALONLY;
  99                         }
 100                 }
 101         }
 102
 103         return CMD_SUCCESS;
 104 }
 105
 106 void cmd_rsquit::NoticeUser(User* user, const std::string &msg)
 107 {
 108         if (IS_LOCAL(user))
 109         {
 110                 user->WriteServ("NOTICE %s :%s",user->nick,msg.c_str());
 111         }
 112         else
 113         {
 114                 std::deque<std::string> params;
 115                 params.push_back(user->nick);
 116                 params.push_back("NOTICE "+ConvToStr(user->nick)+" :"+msg);
 117                 Utils->DoOneToOne(ServerInstance->Config->GetSID(), "PUSH", params, user->server);
 118         }
 119 }
 120