mod_tlsfail: Add c2s support
authorKim Alvefur <zash@zash.se>
Sun, 03 Oct 2021 17:04:05 +0200
changeset 4706 7009e16192fa
parent 4705 efdc3e4dc5df
child 4707 94df7986a874
mod_tlsfail: Add c2s support
mod_tlsfail/mod_tlsfail.lua
--- a/mod_tlsfail/mod_tlsfail.lua	Fri Oct 01 16:58:37 2021 +0200
+++ b/mod_tlsfail/mod_tlsfail.lua	Sun Oct 03 17:04:05 2021 +0200
@@ -5,6 +5,11 @@
 local s2s_feature = st.stanza("starttls", starttls_attr);
 local starttls_failure = st.stanza("failure", starttls_attr);
 
+module:hook("stream-features", function(event)
+	local features = event.features;
+	features:add_child(s2s_feature);
+end);
+
 module:hook("s2s-stream-features", function(event)
 	local features = event.features;
 	features:add_child(s2s_feature);
@@ -13,7 +18,7 @@
 -- Hook <starttls/>
 module:hook("stanza/urn:ietf:params:xml:ns:xmpp-tls:starttls", function(event)
 	local origin = event.origin;
-	origin.sends2s(starttls_failure);
+	(origin.sends2s or origin.send)(starttls_failure);
 	origin:close();
 	return true;
 end);