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                                 //ServerInstance->SNO->WriteToSnoMask('l',"RSQUIT: %s told me to SQUIT myself! (%s matches local server name)",user->nick,parameters[1]);
  67                                 NoticeUser(user, "*** RSQUIT: Foolish mortal, you cannot make a server SQUIT itself! ("+ConvToStr(parameters[1])+" matches local server name)");
  68                                 return CMD_FAILURE;
  69                         }
  70                         TreeSocket* sock = s->GetSocket();
  71                         if (!sock)
  72                         {
  73                                 //ServerInstance->SNO->WriteToSnoMask('l',"RSQUIT: %s told me to SQUIT \002%s\002 but the server isn't linked here.",user->nick,parameters[1]);
  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         else
  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                         delete sock;
  99                 }
 100         }
 101
 102         return CMD_SUCCESS;
 103 }
 104
 105 void cmd_rsquit::NoticeUser(userrec* user, const std::string &msg)
 106 {
 107         if (IS_LOCAL(user))
 108         {
 109                 user->WriteServ("NOTICE %s :%s",user->nick,msg.c_str());
 110         }
 111         else
 112         {
 113                 //bool DoOneToOne(const std::string &prefix, const std::string &command, std::deque<std::string> &params, std::string target);
 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 }