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 = "<password>";
  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         const std::string unregline = "ERROR :" + message;
  43         const UserManager::LocalList& list = ServerInstance->Users.GetLocalUsers();
  44         for (UserManager::LocalList::const_iterator i = list.begin(); i != list.end(); ++i)
  45         {
  46                 LocalUser* user = *i;
  47                 if (user->registered == REG_ALL)
  48                 {
  49                         user->WriteNotice(message);
  50                 }
  51                 else
  52                 {
  53                         // Unregistered connections receive ERROR, not a NOTICE
  54                         user->Write(unregline);
  55                 }
  56         }
  57 }
  58
  59 /** Handle /DIE
  60  */
  61 CmdResult CommandDie::Handle (const std::vector<std::string>& parameters, User *user)
  62 {
  63         if (DieRestart::CheckPass(user, parameters[0], "diepass"))
  64         {
  65                 {
  66                         std::string diebuf = "*** DIE command from " + user->GetFullHost() + ". Terminating.";
  67                         ServerInstance->Logs->Log("COMMAND", LOG_SPARSE, diebuf);
  68                         DieRestart::SendError(diebuf);
  69                 }
  70
  71                 QuitAll();
  72                 ServerInstance->Exit(EXIT_STATUS_DIE);
  73         }
  74         else
  75         {
  76                 ServerInstance->Logs->Log("COMMAND", LOG_SPARSE, "Failed /DIE command from %s", user->GetFullRealHost().c_str());
  77                 ServerInstance->SNO->WriteGlobalSno('a', "Failed DIE Command from %s.", user->GetFullRealHost().c_str());
  78                 return CMD_FAILURE;
  79         }
  80         return CMD_SUCCESS;
  81 }