plugins/mod_compression.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2009 Tobias Markmann
   3 --
   4 -- This project is MIT/X11 licensed. Please see the
   5 -- COPYING file in the source package for more information.
   6 --
   7
   8 local st = require "util.stanza";
   9 local print = print
  10
  11 local xmlns_compression_feature = "http://jabber.org/features/compress"
  12 local xmlns_compression_protocol = "http://jabber.org/protocol/compress"
  13 local compression_stream_feature = st.stanza("compression", {xmlns=xmlns_compression_feature}):tag("method"):text("zlib"):up();
  14
  15
  16 module:add_event_hook("stream-features",
  17                 function (session, features)
  18                         features:add_child(compression_stream_feature);
  19                 end
  20 );
  21
  22 module:add_handler("c2s_unauthed", "compress", xmlns_compression_protocol,
  23                 function(session, stanza)
  24                         -- checking if the compression method is supported
  25                         local method = stanza:child_with_name("method")[1];
  26                         if method == "zlib" then
  27                                 session.log("info", method.." compression selected.");
  28                                 session.send(st.stanza("compressed", {xmlns=xmlns_compression_protocol}));
  29                         else
  30                                 session.log("info", method.." compression selected. But we don't support it.");
  31                                 local error_st = st.stanza("failure", {xmlns=xmlns_compression_protocol}):tag("unsupported-method");
  32                                 session.send(error_st);
  33                         end
  34                 end
  35 );