Change SetClientIP to take a C++ string instead of a char array.

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

diff --git a/src/modules/m_hostcycle.cpp b/src/modules/m_hostcycle.cpp
index e8a0abbf15800d1f6871483a2ad80e55c07a28b4..621f06a279e53e715feaa36c58a053603fd0e93c 100644 (file)
--- a/src/modules/m_hostcycle.cpp
+++ b/src/modules/m_hostcycle.cpp
@@ -19,28 +19,33 @@
 
 
 #include "inspircd.h"
+#include "modules/cap.h"
 
 class ModuleHostCycle : public Module
 {
+       Cap::Reference chghostcap;
+
        /** Send fake quit/join/mode messages for host or ident cycle.
         */
-       static void DoHostCycle(User* user, const std::string& newident, const std::string& newhost, const char* quitmsg)
+       void DoHostCycle(User* user, const std::string& newident, const std::string& newhost, const char* quitmsg)
        {
                // GetFullHost() returns the original data at the time this function is called
                const std::string quitline = ":" + user->GetFullHost() + " QUIT :" + quitmsg;
 
-               already_sent_t silent_id = ++LocalUser::already_sent_id;
-               already_sent_t seen_id = ++LocalUser::already_sent_id;
+               already_sent_t silent_id = ServerInstance->Users.NextAlreadySentId();
+               already_sent_t seen_id = ServerInstance->Users.NextAlreadySentId();
 
                IncludeChanList include_chans(user->chans.begin(), user->chans.end());
                std::map<User*,bool> exceptions;
 
                FOREACH_MOD(OnBuildNeighborList, (user, include_chans, exceptions));
 
+               // Users shouldn't see themselves quitting when host cycling
+               exceptions.erase(user);
                for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
                {
                        LocalUser* u = IS_LOCAL(i->first);
-                       if (u && !u->quitting)
+                       if ((u) && (!u->quitting) && (!chghostcap.get(u)))
                        {
                                if (i->second)
                                {
@@ -80,6 +85,8 @@ class ModuleHostCycle : public Module
                                        continue;
                                if (u->already_sent == silent_id)
                                        continue;
+                               if (chghostcap.get(u))
+                                       continue;
 
                                if (u->already_sent != seen_id)
                                {
@@ -95,6 +102,11 @@ class ModuleHostCycle : public Module
        }
 
  public:
+       ModuleHostCycle()
+               : chghostcap(this, "chghost")
+       {
+       }
+
        void OnChangeIdent(User* user, const std::string& newident) CXX11_OVERRIDE
        {
                DoHostCycle(user, newident, user->dhost, "Changing ident");