X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_cloaking.cpp;fp=src%2Fmodules%2Fm_cloaking.cpp;h=3b00d3347aa5018f5edebbcdd5239d91c4f674a7;hb=3d0d64933da3d37866fadfa042e34a1125315db6;hp=756781ef052e6167dd5caf2dd70dc119b751d295;hpb=ee7db9bbfbb5f3bc7359d22e7021415aebddb9d2;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/m_cloaking.cpp b/src/modules/m_cloaking.cpp
index 756781ef0..3b00d3347 100644
--- a/src/modules/m_cloaking.cpp
+++ b/src/modules/m_cloaking.cpp
@@ -466,6 +466,24 @@ class ModuleCloaking : public Module
 		return chost;
 	}
 
+	void OnSetUserIP(LocalUser* user) CXX11_OVERRIDE
+	{
+		// Connecting users are handled in OnUserConnect not here.
+		if (user->registered != REG_ALL)
+			return;
+
+		// Remove the cloaks and generate new ones.
+		cu.ext.unset(user);
+		OnUserConnect(user);
+
+		// If a user is using a cloak then update it.
+		if (user->IsModeSet(cu))
+		{
+			CloakList* cloaklist = cu.ext.get(user);
+			user->ChangeDisplayedHost(cloaklist->front());
+		}
+	}
+
 	void OnUserConnect(LocalUser* dest) CXX11_OVERRIDE
 	{
 		if (cu.ext.get(dest))