Software `/` code `/` prosody

Annotate

plugins/mod_websocket.lua @ 9577:388b93b3eecf

server_select: Fix :lock method This always unlocks reading. I don't believe this is used anywhere. server_event does not implement this.

author	Kim Alvefur <zash@zash.se>
date	Sun, 28 Oct 2018 16:54:08 +0100
parent	9415:02155a10c5e9
child	9795:02735bc82126
child	9800:5c5117d41133

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
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	162 module:log("debug", "Origin %s is not allowed by 'cross_domain_websocket'", request.headers.origin or "(missing header)");
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)
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	288 local attr = stanza.attr;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	289 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	290 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	291 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	292 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	293 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	294 end, -1000);
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	295
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	296 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	297 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	298 end);
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	299
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	300 response.status_code = 101;
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	301 response.headers.upgrade = "websocket";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	302 response.headers.connection = "Upgrade";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	303 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	304 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	305
7760 801d4c8e0f58 mod_websocket: Add some debug messages Kim Alvefur <zash@zash.se> parents: 7716 diff changeset	306 session.log("debug", "Sending WebSocket handshake");
801d4c8e0f58 mod_websocket: Add some debug messages Kim Alvefur <zash@zash.se> parents: 7716 diff changeset	307
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	308 return "";
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	309 end
6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	310
7315 4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	311 local function keepalive(event)
7340 7dea28dafc49 mod_websocket: Fix read timeout handler (thanks mt) Kim Alvefur <zash@zash.se> parents: 7315 diff changeset	312 local session = event.session;
7dea28dafc49 mod_websocket: Fix read timeout handler (thanks mt) Kim Alvefur <zash@zash.se> parents: 7315 diff changeset	313 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	314 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	315 end
7315 4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	316 end
4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	317
4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	318 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	319
9378 a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	320 module:depends("http");
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	321 module:provides("http", {
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	322 name = "websocket";
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	323 default_path = "xmpp-websocket";
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	324 route = {
a6f54df39624 mod_websocket: Serve HTTP in global context Kim Alvefur <zash@zash.se> parents: 8794 diff changeset	325 ["GET"] = handle_request;
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 };
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
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	330 function module.add_host(module)
7315 4fd984d1e445 mod_websocket: Send a ping on read timeout Kim Alvefur <zash@zash.se> parents: 7314 diff changeset	331 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	332
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 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	334 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	335 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	336 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	337 -- 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	338 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	339 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	340 local local_cross_domain = module:get_option_set("cross_domain_websocket", { this_origin });
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 -- 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	342 -- 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	343 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	344 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	345 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	346 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	347 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	348 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	349 end
6397 6f75f8043936 mod_websocket: Initial commit (based on the prosody-modules version) Florian Zeitz <florob@babelmonkeys.de> parents: diff changeset	350 end

prosody

388b93b3eecf

plugins/mod_websocket.lua

Software / code / prosody

Annotate

plugins/mod_websocket.lua @ 9577:388b93b3eecf

Software `/` code `/` prosody