src/modules/m_lockserv.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 "users.h"
  16 #include "channels.h"
  17 #include "modules.h"
  18
  19 /* $ModDesc: Allows locking of the server to stop all incoming connections till unlocked again */
  20
  21 /** Adds numerics
  22  * 988 <nick> <servername> :Closed for new connections
  23  * 989 <nick> <servername> :Open for new connections
  24 */
  25
  26
  27 class cmd_lockserv : public command_t
  28 {
  29 private:
  30         bool& locked;
  31
  32 public:
  33         cmd_lockserv (InspIRCd* Instance, bool &lock)
  34         : command_t(Instance, "LOCKSERV", 'o', 0), locked(lock)
  35         {
  36                 this->source = "m_lockserv.so";
  37                 syntax.clear();
  38         }
  39
  40         CmdResult Handle (const char** parameters, int pcnt, userrec *user)
  41         {
  42                 locked = true;
  43                 user->WriteServ("988 %s %s :Closed for new connections", user->nick, user->server);
  44                 ServerInstance->WriteOpers("*** Oper %s used LOCKSERV to temporarily close for new connections", user->nick);
  45                 /* Dont send to the network */
  46                 return CMD_LOCALONLY;
  47         }
  48 };
  49
  50 class cmd_unlockserv : public command_t
  51 {
  52 private:
  53         bool& locked;
  54
  55 public:
  56         cmd_unlockserv (InspIRCd* Instance, bool &lock)
  57         : command_t(Instance, "UNLOCKSERV", 'o', 0), locked(lock)
  58         {
  59                 this->source = "m_lockserv.so";
  60                 syntax.clear();
  61         }
  62
  63         CmdResult Handle (const char** parameters, int pcnt, userrec *user)
  64         {
  65                 locked = false;
  66                 user->WriteServ("989 %s %s :Open for new connections", user->nick, user->server);
  67                 ServerInstance->WriteOpers("*** Oper %s used UNLOCKSERV to allow for new connections", user->nick);
  68                 /* Dont send to the network */
  69                 return CMD_LOCALONLY;
  70         }
  71 };
  72
  73 class ModuleLockserv : public Module
  74 {
  75 private:
  76         bool locked;
  77         cmd_lockserv* lockcommand;
  78         cmd_unlockserv* unlockcommand;
  79
  80         virtual void ResetLocked()
  81         {
  82                 locked = false;
  83         }
  84
  85 public:
  86         ModuleLockserv(InspIRCd* Me) : Module(Me)
  87         {
  88                 ResetLocked();
  89                 lockcommand = new cmd_lockserv(ServerInstance, locked);
  90                 ServerInstance->AddCommand(lockcommand);
  91
  92                 unlockcommand = new cmd_unlockserv(ServerInstance, locked);
  93                 ServerInstance->AddCommand(unlockcommand);
  94         }
  95
  96         virtual ~ModuleLockserv()
  97         {
  98         }
  99
 100         void Implements(char* List)
 101         {
 102                 List[I_OnUserRegister] = List[I_OnRehash] = List[I_OnCheckReady] = 1;
 103         }
 104
 105         virtual void OnRehash(userrec* user, const std::string &parameter)
 106         {
 107                 ResetLocked();
 108         }
 109
 110         virtual int OnUserRegister(userrec* user)
 111         {
 112                 if (locked)
 113                 {
 114                         userrec::QuitUser(ServerInstance, user, "Server is temporarily closed. Please try again later.");
 115                         return 1;
 116                 }
 117                 return 0;
 118         }
 119
 120         virtual bool OnCheckReady(userrec* user)
 121         {
 122                 return !locked;
 123         }
 124
 125         virtual Version GetVersion()
 126         {
 127                 return Version(1, 0, 0, 1, VF_VENDOR, API_VERSION);
 128         }
 129 };
 130
 131 MODULE_INIT(ModuleLockserv)