Software `/` code `/` prosody

Annotate

plugins/mod_compression.lua @ 2285:3dd7fdee9035

mod_compression: Some further refactoring.

author	Tobias Markmann <tm@ayena.de>
date	Sun, 29 Nov 2009 23:43:12 +0100
parent	2284:491b89d6816b
child	2286:e0b2d934f316

rev	line source
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	1 -- Prosody IM
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	2 -- Copyright (C) 2009 Tobias Markmann
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	3 --
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	4 -- This project is MIT/X11 licensed. Please see the
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	5 -- COPYING file in the source package for more information.
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	6 --
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	7
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	8 local st = require "util.stanza";
1671 d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	9 local zlib = require "zlib";
1678 79eb903d0e67 Using pcall to make save the rest from zlib fails. Emit errors on those fails. Tobias Markmann <tm@ayena.de> parents: 1677 diff changeset	10 local pcall = pcall;
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	11 local xmlns_compression_feature = "http://jabber.org/features/compress"
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	12 local xmlns_compression_protocol = "http://jabber.org/protocol/compress"
2280 0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	13 local xmlns_stream = "http://etherx.jabber.org/streams";
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	14 local compression_stream_feature = st.stanza("compression", {xmlns=xmlns_compression_feature}):tag("method"):text("zlib"):up();
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	15
1676 719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	16 local compression_level = module:get_option("compression_level");
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	17 -- if not defined assume admin wants best compression
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	18 if compression_level == nil then compression_level = 9 end;
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	19
2279 49bc4c7bdef8 Fixing some typos. Tobias Markmann <tm@ayena.de> parents: 1728 diff changeset	20
1676 719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	21 compression_level = tonumber(compression_level);
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	22 if not compression_level or compression_level < 1 or compression_level > 9 then
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	23 module:log("warn", "Invalid compression level in config: %s", tostring(compression_level));
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	24 module:log("warn", "Module loading aborted. Compression won't be available.");
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	25 return;
719350956714 Add config option handling. Tobias Markmann <tm@ayena.de> parents: 1674 diff changeset	26 end
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	27
1670 23bb280c5eac Remove unwanted spaces. Tobias Markmann <tm@ayena.de> parents: 1669 diff changeset	28 module:add_event_hook("stream-features",
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	29 function (session, features)
1673 5f81cd6d4a92 Remove space at the end of a line. Tobias Markmann <tm@ayena.de> parents: 1672 diff changeset	30 if not session.compressed then
1674 250bddde4162 Add a TODO for s2s compression support. Tobias Markmann <tm@ayena.de> parents: 1673 diff changeset	31 -- FIXME only advertise compression support when TLS layer has no compression enabled
1672 614623f393c6 Add FIXME to remember TLS compression detection. Tobias Markmann <tm@ayena.de> parents: 1671 diff changeset	32 features:add_child(compression_stream_feature);
614623f393c6 Add FIXME to remember TLS compression detection. Tobias Markmann <tm@ayena.de> parents: 1671 diff changeset	33 end
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	34 end
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	35 );
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	36
2280 0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	37 module:hook("s2s-stream-features",
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	38 function (data)
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	39 local session, features = data.session, data.features;
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	40 -- FIXME only advertise compression support when TLS layer has no compression enabled
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	41 features:add_child(compression_stream_feature);
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	42 end
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	43 );
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	44
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	45 -- S2Sout handling aka the client perspective in the S2S connection
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	46 module:hook_stanza(xmlns_stream, "features",
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	47 function (session, stanza)
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	48 if not session.compressed then
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	49 module:log("debug", "FEATURES: "..stanza:pretty_print())
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	50 -- does remote server support compression?
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	51 local comp_st = stanza:child_with_name("compression");
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	52 if comp_st then
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	53 -- do we support the mechanism
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	54 for a in comp_st:children() do
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	55 local algorithm = a[1]
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	56 if algorithm == "zlib" then
2282 6f54dac3ec2d mod_compression: Prepare activating of compression on s2s. Tobias Markmann <tm@ayena.de> parents: 2280 diff changeset	57 session.sends2s(st.stanza("compress", {xmlns=xmlns_compression_protocol}):tag("method"):text("zlib"))
2280 0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	58 session.log("info", "Enabled compression using zlib.")
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	59 return true;
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	60 end
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	61 end
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	62 session.log("debug", "Remote server supports no compression algorithm we support.")
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	63 end
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	64 end
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	65 end
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	66 , 250);
0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	67
2285 3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	68
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	69 -- returns either nil or a fully functional ready to use inflate stream
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	70 local function get_deflate_stream(session)
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	71 local status, deflate_stream = pcall(zlib.deflate, compression_level);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	72 if status == false then
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	73 local error_st = st.stanza("failure", {xmlns=xmlns_compression_protocol}):tag("setup-failed");
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	74 (session.sends2s or session.send)(error_st);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	75 session.log("error", "Failed to create zlib.deflate filter.");
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	76 module:log("error", deflate_stream);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	77 return
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	78 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	79 return deflate_stream
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	80 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	81
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	82 -- returns either nil or a fully functional ready to use inflate stream
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	83 local function get_inflate_stream(session)
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	84 local status, inflate_stream = pcall(zlib.inflate);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	85 if status == false then
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	86 local error_st = st.stanza("failure", {xmlns=xmlns_compression_protocol}):tag("setup-failed");
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	87 (session.sends2s or session.send)(error_st);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	88 session.log("error", "Failed to create zlib.deflate filter.");
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	89 module:log("error", inflate_stream);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	90 return
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	91 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	92 return inflate_stream
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	93 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	94
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	95 -- setup compression for a stream
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	96 local function setup_compression(session, deflate_stream)
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	97 local old_send = (session.sends2s or session.send);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	98
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	99 local new_send = function(t)
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	100 local status, compressed, eof = pcall(deflate_stream, tostring(t), 'sync');
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	101 if status == false then
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	102 session:close({
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	103 condition = "undefined-condition";
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	104 text = compressed;
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	105 extra = st.stanza("failure", {xmlns="http://jabber.org/protocol/compress"}):tag("processing-failed");
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	106 });
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	107 module:log("warn", compressed);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	108 return;
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	109 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	110 old_send(compressed);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	111 end;
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	112
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	113 if session.sends2s then session.sends2s = new_send
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	114 elseif session.send then session.send = new_send end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	115 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	116
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	117 -- setup decompression for a stream
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	118 local function setup_decompression(session, inflate_stream)
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	119 local old_data = session.data
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	120 session.data = function(conn, data)
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	121 local status, decompressed, eof = pcall(inflate_stream, data);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	122 if status == false then
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	123 session:close({
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	124 condition = "undefined-condition";
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	125 text = decompressed;
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	126 extra = st.stanza("failure", {xmlns="http://jabber.org/protocol/compress"}):tag("processing-failed");
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	127 });
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	128 module:log("warn", decompressed);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	129 return;
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	130 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	131 old_data(conn, decompressed);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	132 end;
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	133 end
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	134
1674 250bddde4162 Add a TODO for s2s compression support. Tobias Markmann <tm@ayena.de> parents: 1673 diff changeset	135 -- TODO Support compression on S2S level too.
2282 6f54dac3ec2d mod_compression: Prepare activating of compression on s2s. Tobias Markmann <tm@ayena.de> parents: 2280 diff changeset	136 module:add_handler({"s2sout_unauthed", "s2sout"}, "compressed", xmlns_compression_protocol,
6f54dac3ec2d mod_compression: Prepare activating of compression on s2s. Tobias Markmann <tm@ayena.de> parents: 2280 diff changeset	137 function(session ,stanza)
6f54dac3ec2d mod_compression: Prepare activating of compression on s2s. Tobias Markmann <tm@ayena.de> parents: 2280 diff changeset	138 session.log("debug", "Activating compression...")
6f54dac3ec2d mod_compression: Prepare activating of compression on s2s. Tobias Markmann <tm@ayena.de> parents: 2280 diff changeset	139 end
6f54dac3ec2d mod_compression: Prepare activating of compression on s2s. Tobias Markmann <tm@ayena.de> parents: 2280 diff changeset	140 );
6f54dac3ec2d mod_compression: Prepare activating of compression on s2s. Tobias Markmann <tm@ayena.de> parents: 2280 diff changeset	141
2280 0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	142 module:add_handler({"c2s_unauthed", "c2s", "s2sin_unauthed", "s2sin"}, "compress", xmlns_compression_protocol,
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	143 function(session, stanza)
1719 cf103398e643 Don't allow double compression. Tobias Markmann <tm@ayena.de> parents: 1718 diff changeset	144 -- fail if we are already compressed
cf103398e643 Don't allow double compression. Tobias Markmann <tm@ayena.de> parents: 1718 diff changeset	145 if session.compressed then
cf103398e643 Don't allow double compression. Tobias Markmann <tm@ayena.de> parents: 1718 diff changeset	146 local error_st = st.stanza("failure", {xmlns=xmlns_compression_protocol}):tag("unsupported-method");
cf103398e643 Don't allow double compression. Tobias Markmann <tm@ayena.de> parents: 1718 diff changeset	147 session.send(error_st);
2279 49bc4c7bdef8 Fixing some typos. Tobias Markmann <tm@ayena.de> parents: 1728 diff changeset	148 session.log("warn", "Tried to establish another compression layer.");
1719 cf103398e643 Don't allow double compression. Tobias Markmann <tm@ayena.de> parents: 1718 diff changeset	149 end
cf103398e643 Don't allow double compression. Tobias Markmann <tm@ayena.de> parents: 1718 diff changeset	150
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	151 -- checking if the compression method is supported
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	152 local method = stanza:child_with_name("method")[1];
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	153 if method == "zlib" then
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	154 session.log("info", method.." compression selected.");
2280 0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	155 (session.sends2s or session.send)(st.stanza("compressed", {xmlns=xmlns_compression_protocol}));
1671 d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	156 session:reset_stream();
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	157
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	158 -- create deflate and inflate streams
2285 3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	159 deflate_stream = get_deflate_stream(session);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	160 if not deflate_stream then return end
1678 79eb903d0e67 Using pcall to make save the rest from zlib fails. Emit errors on those fails. Tobias Markmann <tm@ayena.de> parents: 1677 diff changeset	161
2285 3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	162 inflate_stream = get_inflate_stream(session);
3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	163 if not inflate_stream then return end
1671 d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	164
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	165 -- setup compression for session.w
2285 3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	166 setup_compression(session, deflate_stream);
1671 d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	167
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	168 -- setup decompression for session.data
2285 3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	169 setup_decompression(session, inflate_stream);
1671 d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	170
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	171 local session_reset_stream = session.reset_stream;
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	172 session.reset_stream = function(session)
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	173 session_reset_stream(session);
2285 3dd7fdee9035 mod_compression: Some further refactoring. Tobias Markmann <tm@ayena.de> parents: 2284 diff changeset	174 setup_decompression(session, inflate_stream);
1671 d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	175 return true;
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	176 end;
d196ac213104 Actually inject de- and compression into the reading/writing functions. Tobias Markmann <tm@ayena.de> parents: 1670 diff changeset	177 session.compressed = true;
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	178 else
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	179 session.log("info", method.." compression selected. But we don't support it.");
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	180 local error_st = st.stanza("failure", {xmlns=xmlns_compression_protocol}):tag("unsupported-method");
2280 0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	181 (session.sends2s or session.send)(error_st);
1669 b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	182 end
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	183 end
b8eec163a823 Commit initial version of mod_compression. Tobias Markmann <tm@ayena.de> parents: diff changeset	184 );
2280 0b0fe49e5251 Enable one way stream compression on s2s links. Tobias Markmann <tm@ayena.de> parents: 2279 diff changeset	185

prosody

3dd7fdee9035

plugins/mod_compression.lua

Software / code / prosody

Annotate

plugins/mod_compression.lua @ 2285:3dd7fdee9035

Software `/` code `/` prosody