Software `/` code `/` prosody

Annotate

plugins/mod_websocket.lua @ 10092:4b3c129e96f2 0.11

mod_websocket: Clone stanza before mutating (fixes #1398) Checking for `stanza.attr.xmlns == nil` to determine if the stanza object is an actual stanza (`<message>`, `<presence>` or `<iq>` in the `jabber:client` or `jabbber:server` namespace) or some other stream element. Since this mutation is not reverted, it may leak to other places and cause them to mistreat stanzas as stream elements. Especially in cases like MUC where a single stanza is broadcast to many recipients.

author	Kim Alvefur <zash@zash.se>
date	Sat, 20 Jul 2019 04:19:58 +0200
parent	9805:7bfc4269dc36
child	10097:1f45d316b222
child	10581:10d6d0d91f4e

rev	line source
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	1 -- Prosody IM
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	2 -- Copyright (C) 2012-2014 Florian Zeitz
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	3 --
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	4 -- This project is MIT/X11 licensed. Please see the
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	5 -- COPYING file in the source package for more information.
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	6 --
6894 f7203c7cb7ff mod_websocket: Silence luacheck warnings Kim Alvefur <zash@zash.se> parents: 6893 diff changeset	7 -- luacheck: ignore 431/log
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	8
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	9 module:set_global();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	10
6893 861790282dda mod_websocket: Import util.timer and session close timeout config option (thanks fairuz) Kim Alvefur <zash@zash.se> parents: 6793 diff changeset	11 local add_task = require "util.timer".add_task;
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	12 local add_filter = require "util.filters".add_filter;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	13 local sha1 = require "util.hashes".sha1;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	14 local base64 = require "util.encodings".base64.encode;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	15 local st = require "util.stanza";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	16 local parse_xml = require "util.xml".parse;
7761 e0e1f6d6fb4f mod_websocket: Use contains_token from util.http for checking if the requested WebSocket sub-protocols include XMPP Kim Alvefur <zash@zash.se> parents: 7760 diff changeset	17 local contains_token = require "util.http".contains_token;
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	18 local portmanager = require "core.portmanager";
6793 2bf1b7e2149a mod_websocket: Import sessionmanager (fixes traceback) Kim Alvefur <zash@zash.se> parents: 6397 diff changeset	19 local sm_destroy_session = require"core.sessionmanager".destroy_session;
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	20 local log = module._log;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	21
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	22 local websocket_frames = require"net.websocket.frames";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	23 local parse_frame = websocket_frames.parse;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	24 local build_frame = websocket_frames.build;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	25 local build_close = websocket_frames.build_close;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	26 local parse_close = websocket_frames.parse_close;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	27
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	28 local t_concat = table.concat;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	29
6893 861790282dda mod_websocket: Import util.timer and session close timeout config option (thanks fairuz) Kim Alvefur <zash@zash.se> parents: 6793 diff changeset	30 local stream_close_timeout = module:get_option_number("c2s_close_timeout", 5);
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	31 local consider_websocket_secure = module:get_option_boolean("consider_websocket_secure");
7762 2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	32 local cross_domain = module:get_option_set("cross_domain_websocket", {});
2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	33 if cross_domain:contains("*") or cross_domain:contains(true) then
2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	34 cross_domain = true;
2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	35 end
2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	36
2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	37 local function check_origin(origin)
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	38 if cross_domain == true then
7762 2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	39 return true;
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	40 end
7762 2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	41 return cross_domain:contains(origin);
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	42 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	43
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	44 local xmlns_framing = "urn:ietf:params:xml:ns:xmpp-framing";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	45 local xmlns_streams = "http://etherx.jabber.org/streams";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	46 local xmlns_client = "jabber:client";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	47 local stream_xmlns_attr = {xmlns='urn:ietf:params:xml:ns:xmpp-streams'};
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	48
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	49 module:depends("c2s")
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	50 local sessions = module:shared("c2s/sessions");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	51 local c2s_listener = portmanager.get_service("c2s").listener;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	52
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	53 --- Session methods
7938 3629f03817f8 mod_websocket: Make open_stream method behave like the one from util.xmppstream Kim Alvefur <zash@zash.se> parents: 7937 diff changeset	54 local function session_open_stream(session, from, to)
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	55 local attr = {
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	56 xmlns = xmlns_framing,
7937 5b03a8003659 mod_websocket: Include xml:lang attribute on stream <open> (fixes #840) Kim Alvefur <zash@zash.se> parents: 7914 diff changeset	57 ["xml:lang"] = "en",
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	58 version = "1.0",
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	59 id = session.streamid or "",
7938 3629f03817f8 mod_websocket: Make open_stream method behave like the one from util.xmppstream Kim Alvefur <zash@zash.se> parents: 7937 diff changeset	60 from = from or session.host, to = to,
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	61 };
7938 3629f03817f8 mod_websocket: Make open_stream method behave like the one from util.xmppstream Kim Alvefur <zash@zash.se> parents: 7937 diff changeset	62 if session.stream_attrs then
3629f03817f8 mod_websocket: Make open_stream method behave like the one from util.xmppstream Kim Alvefur <zash@zash.se> parents: 7937 diff changeset	63 session:stream_attrs(from, to, attr)
3629f03817f8 mod_websocket: Make open_stream method behave like the one from util.xmppstream Kim Alvefur <zash@zash.se> parents: 7937 diff changeset	64 end
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	65 session.send(st.stanza("open", attr));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	66 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	67
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	68 local function session_close(session, reason)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	69 local log = session.log or log;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	70 if session.conn then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	71 if session.notopen then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	72 session:open_stream();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	73 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	74 if reason then -- nil == no err, initiated by us, false == initiated by client
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	75 local stream_error = st.stanza("stream:error");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	76 if type(reason) == "string" then -- assume stream error
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	77 stream_error:tag(reason, {xmlns = 'urn:ietf:params:xml:ns:xmpp-streams' });
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	78 elseif type(reason) == "table" then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	79 if reason.condition then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	80 stream_error:tag(reason.condition, stream_xmlns_attr):up();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	81 if reason.text then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	82 stream_error:tag("text", stream_xmlns_attr):text(reason.text):up();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	83 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	84 if reason.extra then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	85 stream_error:add_child(reason.extra);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	86 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	87 elseif reason.name then -- a stanza
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	88 stream_error = reason;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	89 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	90 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	91 log("debug", "Disconnecting client, <stream:error> is: %s", tostring(stream_error));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	92 session.send(stream_error);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	93 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	94
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	95 session.send(st.stanza("close", { xmlns = xmlns_framing }));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	96 function session.send() return false; end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	97
9415 02155a10c5e9 mod_websocket: Silence the one warning instead of ignoring the entire file Kim Alvefur <zash@zash.se> parents: 9378 diff changeset	98 -- luacheck: ignore 422/reason
02155a10c5e9 mod_websocket: Silence the one warning instead of ignoring the entire file Kim Alvefur <zash@zash.se> parents: 9378 diff changeset	99 -- FIXME reason should be handled in common place
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	100 local reason = (reason and (reason.name or reason.text or reason.condition)) or reason;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	101 session.log("debug", "c2s stream for %s closed: %s", session.full_jid or ("<"..session.ip..">"), reason or "session closed");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	102
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	103 -- Authenticated incoming stream may still be sending us stanzas, so wait for </stream:stream> from remote
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	104 local conn = session.conn;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	105 if reason == nil and not session.notopen and session.type == "c2s" then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	106 -- Grace time to process data from authenticated cleanly-closed stream
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	107 add_task(stream_close_timeout, function ()
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	108 if not session.destroyed then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	109 session.log("warn", "Failed to receive a stream close response, closing connection anyway...");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	110 sm_destroy_session(session, reason);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	111 conn:write(build_close(1000, "Stream closed"));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	112 conn:close();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	113 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	114 end);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	115 else
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	116 sm_destroy_session(session, reason);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	117 conn:write(build_close(1000, "Stream closed"));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	118 conn:close();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	119 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	120 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	121 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	122
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	123
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	124 --- Filters
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	125 local function filter_open_close(data)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	126 if not data:find(xmlns_framing, 1, true) then return data; end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	127
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	128 local oc = parse_xml(data);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	129 if not oc then return data; end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	130 if oc.attr.xmlns ~= xmlns_framing then return data; end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	131 if oc.name == "close" then return "</stream:stream>"; end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	132 if oc.name == "open" then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	133 oc.name = "stream:stream";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	134 oc.attr.xmlns = nil;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	135 oc.attr["xmlns:stream"] = xmlns_streams;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	136 return oc:top_tag();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	137 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	138
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	139 return data;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	140 end
6894 f7203c7cb7ff mod_websocket: Silence luacheck warnings Kim Alvefur <zash@zash.se> parents: 6893 diff changeset	141 function handle_request(event)
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	142 local request, response = event.request, event.response;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	143 local conn = response.conn;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	144
7914 a6eb3b6bf903 mod_websocket: Set connections starttls method to false to prevent mod_tls from offering starttls (fixes #837) Kim Alvefur <zash@zash.se> parents: 7764 diff changeset	145 conn.starttls = false; -- Prevent mod_tls from believing starttls can be done
a6eb3b6bf903 mod_websocket: Set connections starttls method to false to prevent mod_tls from offering starttls (fixes #837) Kim Alvefur <zash@zash.se> parents: 7764 diff changeset	146
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	147 if not request.headers.sec_websocket_key then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	148 response.headers.content_type = "text/html";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	149 return [[<!DOCTYPE html><html><head><title>Websocket</title></head><body>
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	150 <p>It works! Now point your WebSocket client to this URL to connect to Prosody.</p>
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	151 </body></html>]];
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	152 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	153
7761 e0e1f6d6fb4f mod_websocket: Use contains_token from util.http for checking if the requested WebSocket sub-protocols include XMPP Kim Alvefur <zash@zash.se> parents: 7760 diff changeset	154 local wants_xmpp = contains_token(request.headers.sec_websocket_protocol or "", "xmpp");
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	155
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	156 if not wants_xmpp then
7760 801d4c8e0f58 mod_websocket: Add some debug messages Kim Alvefur <zash@zash.se> parents: 7716 diff changeset	157 module:log("debug", "Client didn't want to talk XMPP, list of protocols was %s", request.headers.sec_websocket_protocol or "(empty)");
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	158 return 501;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	159 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	160
7762 2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	161 if not check_origin(request.headers.origin or "") then
9800 5c5117d41133 mod_websocket: Include the value of cross_domain_websocket in debug message Kim Alvefur <zash@zash.se> parents: 9415 diff changeset	162 module:log("debug", "Origin %s is not allowed by 'cross_domain_websocket' [ %s ]", request.headers.origin or "(missing header)", cross_domain);
7762 2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	163 return 403;
2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	164 end
2208e6cd0d9f mod_websocket: Verify that the client-sent Origin header matches cross_domain_websocket (fixes #652) Kim Alvefur <zash@zash.se> parents: 7761 diff changeset	165
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	166 local function websocket_close(code, message)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	167 conn:write(build_close(code, message));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	168 conn:close();
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	169 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	170
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	171 local dataBuffer;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	172 local function handle_frame(frame)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	173 local opcode = frame.opcode;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	174 local length = frame.length;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	175 module:log("debug", "Websocket received frame: opcode=%0x, %i bytes", frame.opcode, #frame.data);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	176
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	177 -- Error cases
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	178 if frame.RSV1 or frame.RSV2 or frame.RSV3 then -- Reserved bits non zero
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	179 websocket_close(1002, "Reserved bits not zero");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	180 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	181 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	182
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	183 if opcode == 0x8 then -- close frame
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	184 if length == 1 then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	185 websocket_close(1002, "Close frame with payload, but too short for status code");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	186 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	187 elseif length >= 2 then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	188 local status_code = parse_close(frame.data)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	189 if status_code < 1000 then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	190 websocket_close(1002, "Closed with invalid status code");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	191 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	192 elseif ((status_code > 1003 and status_code < 1007) or status_code > 1011) and status_code < 3000 then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	193 websocket_close(1002, "Closed with reserved status code");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	194 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	195 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	196 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	197 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	198
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	199 if opcode >= 0x8 then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	200 if length > 125 then -- Control frame with too much payload
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	201 websocket_close(1002, "Payload too large");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	202 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	203 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	204
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	205 if not frame.FIN then -- Fragmented control frame
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	206 websocket_close(1002, "Fragmented control frame");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	207 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	208 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	209 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	210
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	211 if (opcode > 0x2 and opcode < 0x8) or (opcode > 0xA) then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	212 websocket_close(1002, "Reserved opcode");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	213 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	214 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	215
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	216 if opcode == 0x0 and not dataBuffer then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	217 websocket_close(1002, "Unexpected continuation frame");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	218 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	219 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	220
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	221 if (opcode == 0x1 or opcode == 0x2) and dataBuffer then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	222 websocket_close(1002, "Continuation frame expected");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	223 return false;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	224 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	225
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	226 -- Valid cases
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	227 if opcode == 0x0 then -- Continuation frame
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	228 dataBuffer[#dataBuffer+1] = frame.data;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	229 elseif opcode == 0x1 then -- Text frame
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	230 dataBuffer = {frame.data};
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	231 elseif opcode == 0x2 then -- Binary frame
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	232 websocket_close(1003, "Only text frames are supported");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	233 return;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	234 elseif opcode == 0x8 then -- Close request
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	235 websocket_close(1000, "Goodbye");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	236 return;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	237 elseif opcode == 0x9 then -- Ping frame
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	238 frame.opcode = 0xA;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	239 conn:write(build_frame(frame));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	240 return "";
7314 e327e5b592f5 mod_websocket: Remove warning about unsolicited pong frames "MAY be sent unsolicited" per RFC 6455 (thanks mt) Kim Alvefur <zash@zash.se> parents: 7294 diff changeset	241 elseif opcode == 0xA then -- Pong frame, MAY be sent unsolicited, eg as keepalive
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	242 return "";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	243 else
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	244 log("warn", "Received frame with unsupported opcode %i", opcode);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	245 return "";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	246 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	247
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	248 if frame.FIN then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	249 local data = t_concat(dataBuffer, "");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	250 dataBuffer = nil;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	251 return data;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	252 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	253 return "";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	254 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	255
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	256 conn:setlistener(c2s_listener);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	257 c2s_listener.onconnect(conn);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	258
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	259 local session = sessions[conn];
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	260
8595 d3bbff01df9d mod_websocket: Transfer IP address derived by mod_http Kim Alvefur <zash@zash.se> parents: 8145 diff changeset	261 -- Use upstream IP if a HTTP proxy was used
d3bbff01df9d mod_websocket: Transfer IP address derived by mod_http Kim Alvefur <zash@zash.se> parents: 8145 diff changeset	262 -- See mod_http and #540
d3bbff01df9d mod_websocket: Transfer IP address derived by mod_http Kim Alvefur <zash@zash.se> parents: 8145 diff changeset	263 session.ip = request.ip;
d3bbff01df9d mod_websocket: Transfer IP address derived by mod_http Kim Alvefur <zash@zash.se> parents: 8145 diff changeset	264
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	265 session.secure = consider_websocket_secure or session.secure;
8789 4ae8dd415e94 mod_websocket: Store the request object on the session for use by other modules Matthew Wild <mwild1@gmail.com> parents: 8145 diff changeset	266 session.websocket_request = request;
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	267
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	268 session.open_stream = session_open_stream;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	269 session.close = session_close;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	270
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	271 local frameBuffer = "";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	272 add_filter(session, "bytes/in", function(data)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	273 local cache = {};
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	274 frameBuffer = frameBuffer .. data;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	275 local frame, length = parse_frame(frameBuffer);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	276
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	277 while frame do
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	278 frameBuffer = frameBuffer:sub(length + 1);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	279 local result = handle_frame(frame);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	280 if not result then return; end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	281 cache[#cache+1] = filter_open_close(result);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	282 frame, length = parse_frame(frameBuffer);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	283 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	284 return t_concat(cache, "");
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	285 end);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	286
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	287 add_filter(session, "stanzas/out", function(stanza)
10092 4b3c129e96f2 mod_websocket: Clone stanza before mutating (fixes #1398) Kim Alvefur <zash@zash.se> parents: 9805 diff changeset	288 stanza = st.clone(stanza);
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	289 local attr = stanza.attr;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	290 attr.xmlns = attr.xmlns or xmlns_client;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	291 if stanza.name:find("^stream:") then
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	292 attr["xmlns:stream"] = attr["xmlns:stream"] or xmlns_streams;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	293 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	294 return stanza;
7294 5f4d0753c818 mod_websocket: Make sure stanza xmlns filter runs late in the chain Kim Alvefur <zash@zash.se> parents: 6894 diff changeset	295 end, -1000);
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	296
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	297 add_filter(session, "bytes/out", function(data)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	298 return build_frame({ FIN = true, opcode = 0x01, data = tostring(data)});
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	299 end);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	300
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	301 response.status_code = 101;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	302 response.headers.upgrade = "websocket";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	303 response.headers.connection = "Upgrade";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	304 response.headers.sec_webSocket_accept = base64(sha1(request.headers.sec_websocket_key .. "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"));
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	305 response.headers.sec_webSocket_protocol = "xmpp";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	306
7760 801d4c8e0f58 mod_websocket: Add some debug messages Kim Alvefur <zash@zash.se> parents: 7716 diff changeset	307 session.log("debug", "Sending WebSocket handshake");
801d4c8e0f58 mod_websocket: Add some debug messages Kim Alvefur <zash@zash.se> parents: 7716 diff changeset	308
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	309 return "";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	310 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	311
7315 4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	312 local function keepalive(event)
7340 7dea28dafc49 mod_websocket: Fix read timeout handler (thanks mt) Kim Alvefur <zash@zash.se> parents: 7315 diff changeset	313 local session = event.session;
7dea28dafc49 mod_websocket: Fix read timeout handler (thanks mt) Kim Alvefur <zash@zash.se> parents: 7315 diff changeset	314 if session.open_stream == session_open_stream then
7716 779a9ef6b4fd mod_websocket: Set FIN flag on ping frames (fixes #773) Kim Alvefur <zash@zash.se> parents: 7340 diff changeset	315 return session.conn:write(build_frame({ opcode = 0x9, FIN = true }));
7340 7dea28dafc49 mod_websocket: Fix read timeout handler (thanks mt) Kim Alvefur <zash@zash.se> parents: 7315 diff changeset	316 end
7315 4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	317 end
4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	318
4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	319 module:hook("c2s-read-timeout", keepalive, -0.9);
4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	320
9378 a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	321 module:depends("http");
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	322 module:provides("http", {
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	323 name = "websocket";
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	324 default_path = "xmpp-websocket";
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	325 route = {
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	326 ["GET"] = handle_request;
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	327 ["GET /"] = handle_request;
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	328 };
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	329 });
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	330
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	331 function module.add_host(module)
7315 4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	332 module:hook("c2s-read-timeout", keepalive, -0.9);
7763 c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	333
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	334 if cross_domain ~= true then
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	335 local url = require "socket.url";
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	336 local ws_url = module:http_url("websocket", "xmpp-websocket");
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	337 local url_components = url.parse(ws_url);
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	338 -- The 'Origin' consists of the base URL without path
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	339 url_components.path = nil;
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	340 local this_origin = url.build(url_components);
7764 d6b96e42e8e6 mod_websocket: Allow per-host cross_domain_websocket, defaulting to the base URL of the current host Kim Alvefur <zash@zash.se> parents: 7763 diff changeset	341 local local_cross_domain = module:get_option_set("cross_domain_websocket", { this_origin });
9805 7bfc4269dc36 mod_websocket: Log an error if cross_domain_websocket = true is set in a VirtualHost section Kim Alvefur <zash@zash.se> parents: 9800 diff changeset	342 if local_cross_domain:contains(true) then
7bfc4269dc36 mod_websocket: Log an error if cross_domain_websocket = true is set in a VirtualHost section Kim Alvefur <zash@zash.se> parents: 9800 diff changeset	343 module:log("error", "cross_domain_websocket = true only works in the global section");
7bfc4269dc36 mod_websocket: Log an error if cross_domain_websocket = true is set in a VirtualHost section Kim Alvefur <zash@zash.se> parents: 9800 diff changeset	344 return;
7bfc4269dc36 mod_websocket: Log an error if cross_domain_websocket = true is set in a VirtualHost section Kim Alvefur <zash@zash.se> parents: 9800 diff changeset	345 end
7bfc4269dc36 mod_websocket: Log an error if cross_domain_websocket = true is set in a VirtualHost section Kim Alvefur <zash@zash.se> parents: 9800 diff changeset	346
7764 d6b96e42e8e6 mod_websocket: Allow per-host cross_domain_websocket, defaulting to the base URL of the current host Kim Alvefur <zash@zash.se> parents: 7763 diff changeset	347 -- Don't add / remove something added by another host
d6b96e42e8e6 mod_websocket: Allow per-host cross_domain_websocket, defaulting to the base URL of the current host Kim Alvefur <zash@zash.se> parents: 7763 diff changeset	348 -- This might be weird with random load order
d6b96e42e8e6 mod_websocket: Allow per-host cross_domain_websocket, defaulting to the base URL of the current host Kim Alvefur <zash@zash.se> parents: 7763 diff changeset	349 local_cross_domain:exclude(cross_domain);
d6b96e42e8e6 mod_websocket: Allow per-host cross_domain_websocket, defaulting to the base URL of the current host Kim Alvefur <zash@zash.se> parents: 7763 diff changeset	350 cross_domain:include(local_cross_domain);
8145 4d0f5ea19851 mod_websocket: Convert set to string (syslog sink needs a better fix) Kim Alvefur <zash@zash.se> parents: 8052 diff changeset	351 module:log("debug", "cross_domain = %s", tostring(cross_domain));
7763 c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	352 function module.unload()
7764 d6b96e42e8e6 mod_websocket: Allow per-host cross_domain_websocket, defaulting to the base URL of the current host Kim Alvefur <zash@zash.se> parents: 7763 diff changeset	353 cross_domain:exclude(local_cross_domain);
7763 c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	354 end
c5ce14539fc4 mod_websocket: Add the base URL of each host module is enabled on to 'cross_domain_websocket' Kim Alvefur <zash@zash.se> parents: 7762 diff changeset	355 end
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	356 end

prosody

4b3c129e96f2

plugins/mod_websocket.lua

Software / code / prosody

Annotate

plugins/mod_websocket.lua @ 10092:4b3c129e96f2 0.11

Software `/` code `/` prosody