Main Page | Namespace List | Class Hierarchy | Alphabetical List | Class List | Directories | File List | Namespace Members | Class Members | File Members

home » brain » inspircd-cvs » inspircd » src » modules

m_lockserv.cpp

Go to the documentation of this file.

/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon |
 *       +------------------------------------+
 *
 *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
 * See: http://www.inspircd.org/wiki/index.php/Credits
 *
 * This program is free but copyrighted software; see
 *            the file COPYING for details.
 *
 * ---------------------------------------------------
 */

#include "inspircd.h"

/* $ModDesc: Allows locking of the server to stop all incoming connections till unlocked again */

class CommandLockserv : public Command
{
private:
        bool& locked;

public:
        CommandLockserv (InspIRCd* Instance, bool &lock)
        : Command(Instance, "LOCKSERV", "o", 0), locked(lock)
        {
                this->source = "m_lockserv.so";
                syntax.clear();
        }

        CmdResult Handle (const std::vector<std::string> &parameters, User *user)
        {
                locked = true;
                user->WriteNumeric(988, "%s %s :Closed for new connections", user->nick.c_str(), user->server);
                ServerInstance->SNO->WriteToSnoMask('A', "Oper %s used LOCKSERV to temporarily close for new connections", user->nick.c_str());
                /* Dont send to the network */
                return CMD_LOCALONLY;
        }
};

class CommandUnlockserv : public Command
{
private:
        bool& locked;

public:
        CommandUnlockserv (InspIRCd* Instance, bool &lock)
        : Command(Instance, "UNLOCKSERV", "o", 0), locked(lock)
        {
                this->source = "m_lockserv.so";
                syntax.clear();
        }

        CmdResult Handle (const std::vector<std::string> &parameters, User *user)
        {
                locked = false;
                user->WriteNumeric(989, "%s %s :Open for new connections", user->nick.c_str(), user->server);
                ServerInstance->SNO->WriteToSnoMask('A', "Oper %s used UNLOCKSERV to allow for new connections", user->nick.c_str());
                /* Dont send to the network */
                return CMD_LOCALONLY;
        }
};

class ModuleLockserv : public Module
{
private:
        bool locked;
        CommandLockserv* lockcommand;
        CommandUnlockserv* unlockcommand;

        virtual void ResetLocked()
        {
                locked = false;
        }

public:
        ModuleLockserv(InspIRCd* Me) : Module(Me)
        {
                ResetLocked();
                lockcommand = new CommandLockserv(ServerInstance, locked);
                ServerInstance->AddCommand(lockcommand);

                unlockcommand = new CommandUnlockserv(ServerInstance, locked);
                ServerInstance->AddCommand(unlockcommand);
                Implementation eventlist[] = { I_OnUserRegister, I_OnRehash, I_OnCheckReady };
                ServerInstance->Modules->Attach(eventlist, this, 3);
        }

        virtual ~ModuleLockserv()
        {
        }


        virtual void OnRehash(User* user, const std::string &parameter)
        {
                ResetLocked();
        }

        virtual int OnUserRegister(User* user)
        {
                if (locked)
                {
                        ServerInstance->Users->QuitUser(user, "Server is temporarily closed. Please try again later.");
                        return 1;
                }
                return 0;
        }

        virtual bool OnCheckReady(User* user)
        {
                return !locked;
        }

        virtual Version GetVersion()
        {
                return Version("$Id: m_lockserv.cpp 10291 2008-08-25 20:35:51Z w00t $", VF_VENDOR, API_VERSION);
        }
};

MODULE_INIT(ModuleLockserv)