src/modules/m_jumpserver.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/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, 4)
  31         {
  32                 this->source = "m_jumpserver.so";
  33                 syntax = "[<server> <port> <+/-an> <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                                         default:
  84                                                 user->WriteServ("NOTICE %s :*** Invalid JUMPSERVER flag: %c", user->nick.c_str(), *n);
  85                                                 return CMD_FAILURE;
  86                                         break;
  87                                 }
  88                         }
  89
  90                         if (!atoi(parameters[1].c_str()))
  91                         {
  92                                 user->WriteServ("NOTICE %s :*** Invalid port number", user->nick.c_str());
  93                                 return CMD_FAILURE;
  94                         }
  95
  96                         if (redirect_all_immediately)
  97                         {
  98                                 /* Redirect everyone but the oper sending the command */
  99                                 for (std::vector<User*>::const_iterator i = ServerInstance->Users->local_users.begin(); i != ServerInstance->Users->local_users.end(); i++)
 100                                 {
 101                                         User* t = *i;
 102                                         if (!IS_OPER(t))
 103                                         {
 104                                                 t->WriteNumeric(10, "%s %s %s :Please use this Server/Port instead", user->nick.c_str(), parameters[0].c_str(), parameters[1].c_str());
 105                                                 ServerInstance->Users->QuitUser(t, reason);
 106                                                 n_done++;
 107                                         }
 108                                 }
 109                                 if (n_done)
 110                                 {
 111                                         n_done_s = ConvToStr(n_done);
 112                                 }
 113                         }
 114
 115                         if (redirect_new_users)
 116                         {
 117                                 redirect_to = parameters[0];
 118                                 port = atoi(parameters[1].c_str());
 119                         }
 120
 121                         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(),
 122                                         redirect_all_immediately ? "a" : "",
 123                                         redirect_new_users ? "n" : "",
 124                                         n_done ? " (" : "",
 125                                         n_done ? n_done_s.c_str() : "",
 126                                         n_done ? " user(s) redirected)" : "",
 127                                         reason.c_str());
 128                 }
 129
 130                 return CMD_LOCALONLY;
 131         }
 132 };
 133
 134
 135 class ModuleJumpServer : public Module
 136 {
 137         CommandJumpserver js;
 138  public:
 139         ModuleJumpServer(InspIRCd* Me)
 140                 : Module(Me), js(Me)
 141         {
 142                 ServerInstance->AddCommand(&js);
 143                 Implementation eventlist[] = { I_OnUserRegister };
 144                 ServerInstance->Modules->Attach(eventlist, this, 1);
 145         }
 146
 147         virtual ~ModuleJumpServer()
 148         {
 149         }
 150
 151         virtual int OnUserRegister(User* user)
 152         {
 153                 if (js.port && js.redirect_new_users)
 154                 {
 155                         user->WriteNumeric(10, "%s %s %d :Please use this Server/Port instead",
 156                                 user->nick.c_str(), js.redirect_to.c_str(), js.port);
 157                         ServerInstance->Users->QuitUser(user, js.reason);
 158                         return 0;
 159                 }
 160                 return 0;
 161         }
 162
 163
 164         virtual Version GetVersion()
 165         {
 166                 return Version("$Id$", VF_VENDOR, API_VERSION);
 167         }
 168
 169 };
 170
 171 MODULE_INIT(ModuleJumpServer)