src/modules/m_jumpserver.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2008 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
  16 /* $ModDesc: Provides support for the RPL_REDIR numeric */
  17
  18 /** Handle /JUMPSERVER
  19  */
  20 class CommandJumpserver : public Command
  21 {
  22  public:
  23         bool redirect_all_immediately;
  24         bool redirect_new_users;
  25         bool direction;
  26         std::string redirect_to;
  27         std::string reason;
  28         int port;
  29
  30         CommandJumpserver (InspIRCd* Instance) : Command(Instance, "JUMPSERVER", "o", 0)
  31         {
  32                 this->source = "m_jumpserver.so";
  33                 syntax = "[<server> <port> <+/-a> :<reason>]";
  34                 redirect_to.clear();
  35                 reason.clear();
  36                 port = 0;
  37                 redirect_all_immediately = redirect_new_users = false;
  38         }
  39
  40         CmdResult Handle (const std::vector<std::string> &parameters, User *user)
  41         {
  42                 int n_done = 0;
  43                 reason = (parameters.size() < 4) ? "Please use this server/port instead" : parameters[3];
  44                 redirect_all_immediately = false;
  45                 redirect_new_users = true;
  46                 direction = true;
  47                 std::string n_done_s;
  48
  49                 /* No parameters: jumpserver disabled */
  50                 if (!parameters.size())
  51                 {
  52                         if (port)
  53                                 user->WriteServ("NOTICE %s :*** Disabled jumpserver (previously set to '%s:%d')", user->nick.c_str(), redirect_to.c_str(), port);
  54                         else
  55                                 user->WriteServ("NOTICE %s :*** jumpserver was not enabled.", user->nick.c_str());
  56
  57                         port = 0;
  58                         redirect_to.clear();
  59                         return CMD_LOCALONLY;
  60                 }
  61
  62                 port = 0;
  63                 redirect_to.clear();
  64
  65                 if (parameters.size() >= 3)
  66                 {
  67                         for (const char* n = parameters[2].c_str(); *n; n++)
  68                         {
  69                                 switch (*n)
  70                                 {
  71                                         case '+':
  72                                                 direction = true;
  73                                         break;
  74                                         case '-':
  75                                                 direction = false;
  76                                         break;
  77                                         case 'a':
  78                                                 redirect_all_immediately = direction;
  79                                         break;
  80                                         case 'n':
  81                                                 redirect_new_users = direction;
  82                                         break;
  83                                 }
  84                         }
  85
  86                         if (redirect_all_immediately)
  87                         {
  88                                 /* Redirect everyone but the oper sending the command */
  89                                 for (std::vector<User*>::const_iterator i = ServerInstance->Users->local_users.begin(); i != ServerInstance->Users->local_users.end(); i++)
  90                                 {
  91                                         User* t = *i;
  92                                         if (!IS_OPER(t))
  93                                         {
  94                                                 t->WriteNumeric(10, "%s %s %s :Please use this Server/Port instead", user->nick.c_str(), parameters[0].c_str(), parameters[1].c_str());
  95                                                 ServerInstance->Users->QuitUser(t, reason);
  96                                                 n_done++;
  97                                         }
  98                                 }
  99                                 if (n_done)
 100                                 {
 101                                         n_done_s = ConvToStr(n_done);
 102                                 }
 103                         }
 104
 105                         if (redirect_new_users)
 106                         {
 107                                 redirect_to = parameters[0];
 108                                 port = atoi(parameters[1].c_str());
 109                         }
 110
 111                         user->WriteServ("NOTICE %s :*** Set jumpserver to server '%s' port '%s', flags '+%s%s'%s%s%s: %s", user->nick.c_str(), parameters[0].c_str(), parameters[1].c_str(),
 112                                         redirect_all_immediately ? "a" : "",
 113                                         redirect_new_users ? "n" : "",
 114                                         n_done ? " (" : "",
 115                                         n_done ? n_done_s.c_str() : "",
 116                                         n_done ? " user(s) redirected)" : "",
 117                                         reason.c_str());
 118                 }
 119
 120                 return CMD_LOCALONLY;
 121         }
 122 };
 123
 124
 125 class ModuleJumpServer : public Module
 126 {
 127         CommandJumpserver*      js;
 128  public:
 129         ModuleJumpServer(InspIRCd* Me)
 130                 : Module(Me)
 131         {
 132
 133                 js = new CommandJumpserver(ServerInstance);
 134                 ServerInstance->AddCommand(js);
 135                 Implementation eventlist[] = { I_OnUserRegister };
 136                 ServerInstance->Modules->Attach(eventlist, this, 1);
 137         }
 138
 139         virtual ~ModuleJumpServer()
 140         {
 141         }
 142
 143         virtual int OnUserRegister(User* user)
 144         {
 145                 if (js->port && js->redirect_new_users)
 146                 {
 147                         user->WriteNumeric(10, "%s %s %d :Please use this Server/Port instead", user->nick.c_str(), js->redirect_to.c_str(), js->port);
 148                         ServerInstance->Users->QuitUser(user, js->reason);
 149                         return 0;
 150                 }
 151                 return 0;
 152         }
 153
 154
 155         virtual Version GetVersion()
 156         {
 157                 return Version("$Id$", VF_VENDOR, API_VERSION);
 158         }
 159
 160 };
 161
 162 MODULE_INIT(ModuleJumpServer)