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 "configreader.h"
  16 #include "users.h"
  17 #include "channels.h"
  18 #include "modules.h"
  19 #include "commands/cmd_whois.h"
  20 #include "commands/cmd_stats.h"
  21 #include "socket.h"
  22 #include "wildcard.h"
  23 #include "xline.h"
  24 #include "transport.h"
  25
  26 #include "m_spanningtree/timesynctimer.h"
  27 #include "m_spanningtree/resolvers.h"
  28 #include "m_spanningtree/main.h"
  29 #include "m_spanningtree/utils.h"
  30 #include "m_spanningtree/treeserver.h"
  31 #include "m_spanningtree/link.h"
  32 #include "m_spanningtree/treesocket.h"
  33 #include "m_spanningtree/rsquit.h"
  34
  35 /* $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 */
  36
  37 cmd_rsquit::cmd_rsquit (InspIRCd* Instance, Module* Callback, SpanningTreeUtilities* Util) : command_t(Instance, "RSQUIT", 'o', 1), Creator(Callback), Utils(Util)
  38 {
  39         this->source = "m_spanningtree.so";
  40         syntax = "<remote-server-mask> [target-server-mask]";
  41 }
  42
  43 CmdResult cmd_rsquit::Handle (const char** parameters, int pcnt, userrec *user)
  44 {
  45         if (IS_LOCAL(user))
  46         {
  47                 if (!Utils->FindServerMask(parameters[0]))
  48                 {
  49                         user->WriteServ("NOTICE %s :*** RSQUIT: Server \002%s\002 isn't connected to the network!", user->nick, parameters[0]);
  50                         return CMD_FAILURE;
  51                 }
  52                 if (pcnt > 1)
  53                         user->WriteServ("NOTICE %s :*** RSQUIT: Sending remote squit to \002%s\002 to squit server \002%s\002.",user->nick,parameters[0],parameters[1]);
  54                 else
  55                         user->WriteServ("NOTICE %s :*** RSQUIT: Sending remote squit for server \002%s\002.",user->nick,parameters[0]);
  56         }
  57
  58         TreeServer* s = (pcnt > 1) ? Utils->FindServerMask(parameters[1]) : Utils->FindServerMask(parameters[0]);
  59
  60         if (pcnt > 1)
  61         {
  62                 if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameters[0]))
  63                 {
  64                         if (s == Utils->TreeRoot)
  65                         {
  66                                 NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+ConvToStr(parameters[1])+" matches local server name)");
  67                                 return CMD_FAILURE;
  68                         }
  69                         if (s)
  70                         {
  71                                 TreeSocket* sock = s->GetSocket();
  72                                 if (!sock)
  73                                 {
  74                                         NoticeUser(user, "*** RSQUIT: Server \002"+ConvToStr(parameters[1])+"\002 isn't connected to \002"+ConvToStr(parameters[0])+"\002.");
  75                                         return CMD_FAILURE;
  76                                 }
  77                                 ServerInstance->SNO->WriteToSnoMask('l',"Remote SQUIT from %s matching \002%s\002, squitting server \002%s\002",user->nick,parameters[0],parameters[1]);
  78                                 const char* para[1];
  79                                 para[0] = parameters[1];
  80                                 std::string original_command = std::string("SQUIT ") + parameters[1];
  81                                 Creator->OnPreCommand("SQUIT", para, 1, user, true, original_command);
  82                         }
  83                 }
  84         }
  85         else
  86         {
  87                 if (s == Utils->TreeRoot)
  88                 {
  89                         NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+ConvToStr(parameters[0])+" matches local server name)");
  90                         return CMD_FAILURE;
  91                 }
  92                 TreeSocket* sock = s->GetSocket();
  93                 if (sock)
  94                 {
  95                         ServerInstance->SNO->WriteToSnoMask('l',"RSQUIT: Server \002%s\002 removed from network by %s",parameters[0],user->nick);
  96                         sock->Squit(s,std::string("Server quit by ") + user->GetFullRealHost());
  97                         ServerInstance->SE->DelFd(sock);
  98                         sock->Close();
  99                         delete sock;
 100                 }
 101         }
 102
 103         return CMD_SUCCESS;
 104 }
 105
 106 void cmd_rsquit::NoticeUser(userrec* 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->ServerName, "PUSH", params, user->server);
 118         }
 119 }