src/modules/m_conn_waitpong.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
   6  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   7  *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
   8  *   Copyright (C) 2006-2007 Oliver Lupton <oliverlupton@gmail.com>
   9  *   Copyright (C) 2006 Craig Edwards <craigedwards@brainbox.cc>
  10  *
  11  * This file is part of InspIRCd.  InspIRCd is free software: you can
  12  * redistribute it and/or modify it under the terms of the GNU General Public
  13  * License as published by the Free Software Foundation, version 2.
  14  *
  15  * This program is distributed in the hope that it will be useful, but WITHOUT
  16  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  17  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  18  * details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22  */
  23
  24
  25 #include "inspircd.h"
  26
  27 class ModuleWaitPong : public Module
  28 {
  29         bool sendsnotice;
  30         bool killonbadreply;
  31         LocalStringExt ext;
  32
  33  public:
  34         ModuleWaitPong()
  35          : ext("waitpong_pingstr", this)
  36         {
  37         }
  38
  39         void init() CXX11_OVERRIDE
  40         {
  41                 ServerInstance->Modules->AddService(ext);
  42         }
  43
  44         void ReadConfig(ConfigStatus& status) CXX11_OVERRIDE
  45         {
  46                 ConfigTag* tag = ServerInstance->Config->ConfValue("waitpong");
  47                 sendsnotice = tag->getBool("sendsnotice", true);
  48                 killonbadreply = tag->getBool("killonbadreply", true);
  49         }
  50
  51         ModResult OnUserRegister(LocalUser* user) CXX11_OVERRIDE
  52         {
  53                 std::string pingrpl = ServerInstance->GenRandomStr(10);
  54
  55                 user->Write("PING :%s", pingrpl.c_str());
  56
  57                 if(sendsnotice)
  58                         user->WriteNotice("*** If you are having problems connecting due to ping timeouts, please type /quote PONG " + pingrpl + " or /raw PONG " + pingrpl + " now.");
  59
  60                 ext.set(user, pingrpl);
  61                 return MOD_RES_PASSTHRU;
  62         }
  63
  64         ModResult OnPreCommand(std::string &command, std::vector<std::string> &parameters, LocalUser* user, bool validated, const std::string &original_line) CXX11_OVERRIDE
  65         {
  66                 if (command == "PONG")
  67                 {
  68                         std::string* pingrpl = ext.get(user);
  69
  70                         if (pingrpl)
  71                         {
  72                                 if (!parameters.empty() && *pingrpl == parameters[0])
  73                                 {
  74                                         ext.unset(user);
  75                                         return MOD_RES_DENY;
  76                                 }
  77                                 else
  78                                 {
  79                                         if(killonbadreply)
  80                                                 ServerInstance->Users->QuitUser(user, "Incorrect ping reply for registration");
  81                                         return MOD_RES_DENY;
  82                                 }
  83                         }
  84                 }
  85                 return MOD_RES_PASSTHRU;
  86         }
  87
  88         ModResult OnCheckReady(LocalUser* user) CXX11_OVERRIDE
  89         {
  90                 return ext.get(user) ? MOD_RES_DENY : MOD_RES_PASSTHRU;
  91         }
  92
  93         Version GetVersion() CXX11_OVERRIDE
  94         {
  95                 return Version("Require pong prior to registration", VF_VENDOR);
  96         }
  97 };
  98
  99 MODULE_INIT(ModuleWaitPong)