diff mod_manifesto/mod_manifesto.lua @ 1283:3e96889c0c36

mod_manifesto: Check state of incoming connections
author Florian Zeitz <florob@babelmonkeys.de>
date Thu, 23 Jan 2014 19:15:47 +0100
parents 17cb51496469
children e36f82d7baae
line wrap: on
line diff
--- a/mod_manifesto/mod_manifesto.lua	Thu Jan 23 17:23:04 2014 +0100
+++ b/mod_manifesto/mod_manifesto.lua	Thu Jan 23 19:15:47 2014 +0100
@@ -9,6 +9,7 @@
 local hosts = prosody.hosts;
 local host = module.host;
 local host_session = hosts[host];
+local incoming_s2s = prosody.incoming_s2s;
 
 local default_tpl = [[
 Hello there.
@@ -53,24 +54,34 @@
 		return
 	end
 
-	notified[session.username] = now;
 	timer.add_task(15, function ()
 		local bad_contacts, bad_hosts = {}, {};
 		for contact_jid, item in pairs(session.roster) do
-			local _, host = jid_split(contact_jid);
-			local remote_host_session = host_session.s2sout[host];
+			local _, contact_host = jid_split(contact_jid);
+			local bad = false;
+			local remote_host_session = host_session.s2sout[contact_host];
 			if remote_host_session and remote_host_session.type == "s2sout" then -- Only check remote hosts we have completed s2s connections to
 				if not remote_host_session.secure then
-					local contact_name = item.name;
-					if contact_name then
-						table.insert(bad_contacts, contact_name.." <"..contact_jid..">");
-					else
-						table.insert(bad_contacts, contact_jid);
+					bad = true;
+				end
+			end
+			for session in pairs(incoming_s2s) do
+				if session.to_host == host and session.from_host == contact_host then
+					if not session.secure then
+						bad = true;
 					end
-					if not bad_hosts[host] then
-						bad_hosts[host] = true;
-						table.insert(bad_hosts, host);
-					end
+				end
+			end
+			if bad then
+				local contact_name = item.name;
+				if contact_name then
+					table.insert(bad_contacts, contact_name.." <"..contact_jid..">");
+				else
+					table.insert(bad_contacts, contact_jid);
+				end
+				if not bad_hosts[contact_host] then
+					bad_hosts[contact_host] = true;
+					table.insert(bad_hosts, contact_host);
 				end
 			end
 		end
@@ -83,6 +94,7 @@
 			};
 			session.send(st.message({ type = "headline", from = host }):tag("body"):text(message:gsub("$(%w+)", vars)));
 		end
+		notified[session.username] = now;
 	end);
 end);