src/coremods/core_oper/cmd_die.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
   6  *
   7  * This file is part of InspIRCd.  InspIRCd is free software: you can
   8  * redistribute it and/or modify it under the terms of the GNU General Public
   9  * License as published by the Free Software Foundation, version 2.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  14  * details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include "inspircd.h"
  22 #include "exitcodes.h"
  23 #include "core_oper.h"
  24
  25 CommandDie::CommandDie(Module* parent)
  26         : Command(parent, "DIE", 1)
  27 {
  28         flags_needed = 'o';
  29         syntax = "<server>";
  30 }
  31
  32 static void QuitAll()
  33 {
  34         const std::string quitmsg = "Server shutdown";
  35         const UserManager::LocalList& list = ServerInstance->Users.GetLocalUsers();
  36         while (!list.empty())
  37                 ServerInstance->Users.QuitUser(list.front(), quitmsg);
  38 }
  39
  40 void DieRestart::SendError(const std::string& message)
  41 {
  42         ClientProtocol::Messages::Error errormsg(message);
  43         ClientProtocol::Event errorevent(ServerInstance->GetRFCEvents().error, errormsg);
  44         const UserManager::LocalList& list = ServerInstance->Users.GetLocalUsers();
  45         for (UserManager::LocalList::const_iterator i = list.begin(); i != list.end(); ++i)
  46         {
  47                 LocalUser* user = *i;
  48                 if (user->registered == REG_ALL)
  49                 {
  50                         user->WriteNotice(message);
  51                 }
  52                 else
  53                 {
  54                         // Unregistered connections receive ERROR, not a NOTICE
  55                         user->Send(errorevent);
  56                 }
  57         }
  58 }
  59
  60 /** Handle /DIE
  61  */
  62 CmdResult CommandDie::Handle(User* user, const Params& parameters)
  63 {
  64         if (DieRestart::CheckPass(user, parameters[0], "diepass"))
  65         {
  66                 {
  67                         std::string diebuf = "*** DIE command from " + user->GetFullHost() + ". Terminating.";
  68                         ServerInstance->Logs->Log(MODNAME, LOG_SPARSE, diebuf);
  69                         DieRestart::SendError(diebuf);
  70                 }
  71
  72                 QuitAll();
  73                 ServerInstance->Exit(EXIT_STATUS_DIE);
  74         }
  75         else
  76         {
  77                 ServerInstance->Logs->Log(MODNAME, LOG_SPARSE, "Failed /DIE command from %s", user->GetFullRealHost().c_str());
  78                 ServerInstance->SNO->WriteGlobalSno('a', "Failed DIE Command from %s.", user->GetFullRealHost().c_str());
  79                 return CMD_FAILURE;
  80         }
  81         return CMD_SUCCESS;
  82 }