Changeset

3269:b0628bc93acf

mod_ipcheck: Simplify iq handling by hooking on iq-get/ instead of iq/.
author Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date Fri, 24 Aug 2018 21:15:38 +0200
parents 3268:4cdd1ddae72c
children 3270:7776c9dc5f37
files mod_ipcheck/mod_ipcheck.lua
diffstat 1 files changed, 24 insertions(+), 28 deletions(-) [+]
line wrap: on
line diff
--- a/mod_ipcheck/mod_ipcheck.lua	Fri Aug 24 20:55:10 2018 +0200
+++ b/mod_ipcheck/mod_ipcheck.lua	Fri Aug 24 21:15:38 2018 +0200
@@ -6,41 +6,37 @@
 
 module:add_feature("urn:xmpp:sic:0");
 
-module:hook("iq/bare/urn:xmpp:sic:0:ip", function(event)
+module:hook("iq-get/bare/urn:xmpp:sic:0:ip", function(event)
 	local origin, stanza = event.origin, event.stanza;
-	if stanza.attr.type == "get" then
-		if stanza.attr.to then
-			origin.send(st.error_reply(stanza, "auth", "forbidden", "You can only ask about your own IP address"));
-		elseif origin.ip then
-			origin.send(st.reply(stanza):tag("ip", {xmlns='urn:xmpp:sic:0'}):text(origin.ip));
-		else
-			-- IP addresses should normally be available, but in case they are not
-			origin.send(st.error_reply(stanza, "cancel", "service-unavailable", "IP address for this session is not available"));
-		end
-		return true;
+	if stanza.attr.to then
+		origin.send(st.error_reply(stanza, "auth", "forbidden", "You can only ask about your own IP address"));
+	elseif origin.ip then
+		origin.send(st.reply(stanza):tag("ip", {xmlns='urn:xmpp:sic:0'}):text(origin.ip));
+	else
+		-- IP addresses should normally be available, but in case they are not
+		origin.send(st.error_reply(stanza, "cancel", "service-unavailable", "IP address for this session is not available"));
 	end
+	return true;
 end);
 
 module:add_feature("urn:xmpp:sic:1");
 
-module:hook("iq/bare/urn:xmpp:sic:1:address", function(event)
+module:hook("iq-get/bare/urn:xmpp:sic:1:address", function(event)
 	local origin, stanza = event.origin, event.stanza;
-	if stanza.attr.type == "get" then
-		if stanza.attr.to then
-			origin.send(st.error_reply(stanza, "auth", "forbidden", "You can only ask about your own IP address"));
-		elseif origin.ip then
-			local reply = st.reply(stanza):tag("address", {xmlns='urn:xmpp:sic:1'})
-				:tag("ip"):text(origin.ip):up()
-			if origin.conn and origin.conn.port then -- server_event
-				reply:tag("port"):text(tostring(origin.conn:port()))
-			elseif origin.conn and origin.conn.clientport then -- server_select
-				reply:tag("port"):text(tostring(origin.conn:clientport()))
-			end
-			origin.send(reply);
-		else
-			-- IP addresses should normally be available, but in case they are not
-			origin.send(st.error_reply(stanza, "cancel", "service-unavailable", "IP address for this session is not available"));
+	if stanza.attr.to then
+		origin.send(st.error_reply(stanza, "auth", "forbidden", "You can only ask about your own IP address"));
+	elseif origin.ip then
+		local reply = st.reply(stanza):tag("address", {xmlns='urn:xmpp:sic:1'})
+			:tag("ip"):text(origin.ip):up()
+		if origin.conn and origin.conn.port then -- server_event
+			reply:tag("port"):text(tostring(origin.conn:port()))
+		elseif origin.conn and origin.conn.clientport then -- server_select
+			reply:tag("port"):text(tostring(origin.conn:clientport()))
 		end
-		return true;
+		origin.send(reply);
+	else
+		-- IP addresses should normally be available, but in case they are not
+		origin.send(st.error_reply(stanza, "cancel", "service-unavailable", "IP address for this session is not available"));
 	end
+	return true;
 end);