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