Add support for blocking tag messages with the deaf mode.

[user/henk/code/inspircd.git] / src / modules / m_clones.cpp

diff --git a/src/modules/m_clones.cpp b/src/modules/m_clones.cpp
index b3e695bfde0c3c224a4d17ddb63cc825da6e127b..1210c7c126adddcf04922ace4ddf405a5399192e 100644 (file)
--- a/src/modules/m_clones.cpp
+++ b/src/modules/m_clones.cpp
@@ -1,9 +1,13 @@
 /*
  * InspIRCd -- Internet Relay Chat Daemon
  *
+ *   Copyright (C) 2013, 2018 Sadie Powell <sadie@witchery.services>
+ *   Copyright (C) 2012, 2019 Robby <robby@chatbelgie.be>
+ *   Copyright (C) 2012, 2014 Attila Molnar <attilamolnar@hush.com>
+ *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
+ *   Copyright (C) 2007, 2010 Craig Edwards <brain@inspircd.org>
  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
- *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
- *   Copyright (C) 2007 Craig Edwards <craigedwards@brainbox.cc>
  *
  * This file is part of InspIRCd.  InspIRCd is free software: you can
  * redistribute it and/or modify it under the terms of the GNU General Public
@@ -20,43 +24,64 @@
 
 
 #include "inspircd.h"
+#include "modules/ircv3_batch.h"
 
-/** Handle /CLONES
- */
-class CommandClones : public Command
+enum
+{
+       // InspIRCd-specific.
+       RPL_CLONES = 399
+};
+
+class CommandClones : public SplitCommand
 {
+ private:
+       IRCv3::Batch::API batchmanager;
+       IRCv3::Batch::Batch batch;
+
  public:
-       CommandClones(Module* Creator) : Command(Creator,"CLONES", 1)
+       CommandClones(Module* Creator)
+               : SplitCommand(Creator,"CLONES", 1)
+               , batchmanager(Creator)
+               , batch("inspircd.org/clones")
        {
-               flags_needed = 'o'; syntax = "<limit>";
+               flags_needed = 'o';
+               syntax = "<limit>";
        }
 
-       CmdResult Handle (const std::vector<std::string> &parameters, User *user)
+       CmdResult HandleLocal(LocalUser* user, const Params& parameters) CXX11_OVERRIDE
        {
+               unsigned int limit = ConvToNum<unsigned int>(parameters[0]);
 
-               std::string clonesstr = "CLONES ";
-
-               unsigned long limit = atoi(parameters[0].c_str());
-
-               /*
-                * Syntax of a /clones reply:
-                *  :server.name 304 target :CLONES START
-                *  :server.name 304 target :CLONES <count> <ip>
-                *  :server.name 304 target :CLONES END
-                */
+               // Syntax of a CLONES reply:
+               // :irc.example.com BATCH +<id> inspircd.org/clones :<min-count>
+               // @batch=<id> :irc.example.com 399 <client> <local-count> <remote-count> <cidr-mask>
+               /// :irc.example.com BATCH :-<id>
 
-               user->WriteNumeric(304, clonesstr + "START");
+               if (batchmanager)
+               {
+                       batchmanager->Start(batch);
+                       batch.GetBatchStartMessage().PushParam(ConvToStr(limit));
+               }
 
-               /* hostname or other */
                const UserManager::CloneMap& clonemap = ServerInstance->Users->GetCloneMap();
                for (UserManager::CloneMap::const_iterator i = clonemap.begin(); i != clonemap.end(); ++i)
                {
                        const UserManager::CloneCounts& counts = i->second;
-                       if (counts.global >= limit)
-                               user->WriteNumeric(304, clonesstr + ConvToStr(counts.global) + " " + i->first.str());
+                       if (counts.global < limit)
+                               continue;
+
+                       Numeric::Numeric numeric(RPL_CLONES);
+                       numeric.push(counts.local);
+                       numeric.push(counts.global);
+                       numeric.push(i->first.str());
+
+                       ClientProtocol::Messages::Numeric numericmsg(numeric, user);
+                       batch.AddToBatch(numericmsg);
+                       user->Send(ServerInstance->GetRFCEvents().numeric, numericmsg);
                }
 
-               user->WriteNumeric(304, clonesstr + "END");
+               if (batchmanager)
+                       batchmanager->End(batch);
 
                return CMD_SUCCESS;
        }
@@ -64,15 +89,18 @@ class CommandClones : public Command
 
 class ModuleClones : public Module
 {
+ public:
        CommandClones cmd;
+
  public:
-       ModuleClones() : cmd(this)
+       ModuleClones()
+               : cmd(this)
        {
        }
 
        Version GetVersion() CXX11_OVERRIDE
        {
-               return Version("Provides the /CLONES command to retrieve information on clones.", VF_VENDOR);
+               return Version("Adds the /CLONES command which allows server operators to view IP addresses from which there are more than a specified number of connections.", VF_VENDOR);
        }
 };