X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_hideoper.cpp;h=f04d88809065669249f65532156e35ff873df4ca;hb=f35490427c17229e81b1ff266c7dcaf67d1a35b5;hp=2e0b388cf7eeec0630410a5f4fea687165b23882;hpb=565544fac966b14e046bb3042ab485f79bcf7c9e;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/m_hideoper.cpp b/src/modules/m_hideoper.cpp
index 2e0b388cf..f04d88809 100644
--- a/src/modules/m_hideoper.cpp
+++ b/src/modules/m_hideoper.cpp
@@ -20,6 +20,9 @@
 
 
 #include "inspircd.h"
+#include "modules/stats.h"
+#include "modules/who.h"
+#include "modules/whois.h"
 
 /** Handles user mode +H
  */
@@ -48,13 +51,21 @@ class HideOper : public SimpleUserModeHandler
 	}
 };
 
-class ModuleHideOper : public Module, public Whois::LineEventListener
+class ModuleHideOper
+	: public Module
+	, public Stats::EventListener
+	, public Who::EventListener
+	, public Whois::LineEventListener
 {
+ private:
 	HideOper hm;
 	bool active;
+
  public:
 	ModuleHideOper()
-		: Whois::LineEventListener(this)
+		: Stats::EventListener(this)
+		, Who::EventListener(this)
+		, Whois::LineEventListener(this)
 		, hm(this)
 		, active(false)
 	{
@@ -104,20 +115,37 @@ class ModuleHideOper : public Module, public Whois::LineEventListener
 		return MOD_RES_PASSTHRU;
 	}
 
-	ModResult OnSendWhoLine(User* source, const std::vector<std::string>& params, User* user, Membership* memb, Numeric::Numeric& numeric) CXX11_OVERRIDE
+	ModResult OnWhoLine(const Who::Request& request, LocalUser* source, User* user, Membership* memb, Numeric::Numeric& numeric) CXX11_OVERRIDE
 	{
 		if (user->IsModeSet(hm) && !source->HasPrivPermission("users/auspex"))
 		{
 			// Hide the line completely if doing a "/who * o" query
-			if ((params.size() > 1) && (params[1].find('o') != std::string::npos))
+			if (request.flags['o'])
 				return MOD_RES_DENY;
 
+			size_t flag_index = 5;
+			if (request.whox)
+			{
+				// We only need to fiddle with the flags if they are present.
+				if (!request.whox_fields['f'])
+					return MOD_RES_PASSTHRU;
+
+				// WHOX makes this a bit tricky as we need to work out the parameter which the flags are in.
+				flag_index = 0;
+				static const char* flags = "tcuihsn";
+				for (size_t i = 0; i < strlen(flags); ++i)
+				{
+					if (request.whox_fields[flags[i]])
+						flag_index += 1;
+				}
+			}
+
 			// hide the "*" that marks the user as an oper from the /WHO line
 			// #chan ident localhost insp22.test nick H@ :0 Attila
-			if (numeric.GetParams().size() < 6)
+			if (numeric.GetParams().size() <= flag_index)
 				return MOD_RES_PASSTHRU;
 
-			std::string& param = numeric.GetParams()[5];
+			std::string& param = numeric.GetParams()[flag_index];
 			const std::string::size_type pos = param.find('*');
 			if (pos != std::string::npos)
 				param.erase(pos, 1);