Software /
code /
prosody-modules
Annotate
mod_websocket/mod_websocket.lua @ 1354:93158d5758f3
mod_s2s_auth_dane: Skip TLSA lookups if SRV is insecure
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Tue, 18 Mar 2014 15:54:08 +0100 |
parent | 1340:3ffd64b4ab59 |
child | 1364:cc77341af5ee |
rev | line source |
---|---|
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
1 -- Prosody IM |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
2 -- Copyright (C) 2012 Florian Zeitz |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
3 -- |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
4 -- This project is MIT/X11 licensed. Please see the |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
5 -- COPYING file in the source package for more information. |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
6 -- |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
7 |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
8 module:set_global(); |
129 | 9 |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
10 local add_filter = require "util.filters".add_filter; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
11 local sha1 = require "util.hashes".sha1; |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
12 local base64 = require "util.encodings".base64.encode; |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
13 local softreq = require "util.dependencies".softreq; |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
14 local st = require "util.stanza"; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
15 local parse_xml = require "util.xml".parse; |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
16 local portmanager = require "core.portmanager"; |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
17 local sm_destroy_session = sessionmanager.destroy_session; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
18 local log = module._log; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
19 |
905
eae665bc2122
mod_websocket: Fix require to work with lua-bitop (Thanks István and Zash)
Florian Zeitz <florob@babelmonkeys.de>
parents:
895
diff
changeset
|
20 local bit; |
eae665bc2122
mod_websocket: Fix require to work with lua-bitop (Thanks István and Zash)
Florian Zeitz <florob@babelmonkeys.de>
parents:
895
diff
changeset
|
21 pcall(function() bit = require"bit"; end); |
eae665bc2122
mod_websocket: Fix require to work with lua-bitop (Thanks István and Zash)
Florian Zeitz <florob@babelmonkeys.de>
parents:
895
diff
changeset
|
22 bit = bit or softreq"bit32" |
eae665bc2122
mod_websocket: Fix require to work with lua-bitop (Thanks István and Zash)
Florian Zeitz <florob@babelmonkeys.de>
parents:
895
diff
changeset
|
23 if not bit then module:log("error", "No bit module found. Either LuaJIT 2, lua-bitop or Lua 5.2 is required"); end |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
24 local band = bit.band; |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
25 local bxor = bit.bxor; |
910
c469a2b2d77d
mod_websocket: Avoid floating point division
Florian Zeitz <florob@babelmonkeys.de>
parents:
909
diff
changeset
|
26 local rshift = bit.rshift; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
27 |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
28 local t_concat = table.concat; |
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
29 local s_byte = string.byte; |
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
30 local s_char= string.char; |
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
31 |
1193
bbe278a56b0a
mod_websocket: Add consider_websocket_secure option
Florian Zeitz <florob@babelmonkeys.de>
parents:
1029
diff
changeset
|
32 local consider_websocket_secure = module:get_option_boolean("consider_websocket_secure"); |
856
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
33 local cross_domain = module:get_option("cross_domain_websocket"); |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
34 if cross_domain then |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
35 if cross_domain == true then |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
36 cross_domain = "*"; |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
37 elseif type(cross_domain) == "table" then |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
38 cross_domain = t_concat(cross_domain, ", "); |
856
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
39 end |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
40 if type(cross_domain) ~= "string" then |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
41 cross_domain = nil; |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
42 end |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
43 end |
7eb54ed58856
mod_websocket: Add CORS support
Florian Zeitz <florob@babelmonkeys.de>
parents:
853
diff
changeset
|
44 |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
45 local xmlns_framing = "urn:ietf:params:xml:ns:xmpp-framing"; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
46 local xmlns_streams = "http://etherx.jabber.org/streams"; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
47 local xmlns_client = "jabber:client"; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
48 |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
49 module:depends("c2s") |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
50 local sessions = module:shared("c2s/sessions"); |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
51 local c2s_listener = portmanager.get_service("c2s").listener; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
52 |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
53 -- Websocket helpers |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
54 local function parse_frame(frame) |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
55 local result = {}; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
56 local pos = 1; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
57 local length_bytes = 0; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
58 local tmp_byte; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
59 |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
60 if #frame < 2 then return; end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
61 |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
62 tmp_byte = s_byte(frame, pos); |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
63 result.FIN = band(tmp_byte, 0x80) > 0; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
64 result.RSV1 = band(tmp_byte, 0x40) > 0; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
65 result.RSV2 = band(tmp_byte, 0x20) > 0; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
66 result.RSV3 = band(tmp_byte, 0x10) > 0; |
690
5acc203972f3
mod_websocket: Add fragmentation support
Florian Zeitz <florob@babelmonkeys.de>
parents:
689
diff
changeset
|
67 result.opcode = band(tmp_byte, 0x0F); |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
68 |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
69 pos = pos + 1; |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
70 tmp_byte = s_byte(frame, pos); |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
71 result.MASK = band(tmp_byte, 0x80) > 0; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
72 result.length = band(tmp_byte, 0x7F); |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
73 |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
74 if result.length == 126 then |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
75 length_bytes = 2; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
76 result.length = 0; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
77 elseif result.length == 127 then |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
78 length_bytes = 8; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
79 result.length = 0; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
80 end |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
81 |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
82 if #frame < (2 + length_bytes) then return; end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
83 |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
84 for i = 1, length_bytes do |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
85 pos = pos + 1; |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
86 result.length = result.length * 256 + s_byte(frame, pos); |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
87 end |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
88 |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
89 if #frame < (2 + length_bytes + (result.MASK and 4 or 0) + result.length) then return; end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
90 |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
91 if result.MASK then |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
92 local counter = 0; |
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
93 local data = {}; |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
94 local key = {s_byte(frame, pos+1), s_byte(frame, pos+2), |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
95 s_byte(frame, pos+3), s_byte(frame, pos+4)} |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
96 result.key = key; |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
97 |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
98 pos = pos + 5; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
99 for i = pos, pos + result.length - 1 do |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
100 data[#data+1] = s_char(bxor(key[counter+1], s_byte(frame, i))); |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
101 counter = (counter + 1) % 4; |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
102 end |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
103 result.data = t_concat(data, ""); |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
104 else |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
105 result.data = frame:sub(pos + 1, pos + result.length); |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
106 end |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
107 |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
108 return result, 2 + length_bytes + (result.MASK and 4 or 0) + result.length; |
677
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
109 end |
eeb41cd5e9f3
mod_websocket: Move frame handling into a separate function
Florian Zeitz <florob@babelmonkeys.de>
parents:
676
diff
changeset
|
110 |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
111 local function build_frame(desc) |
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
112 local length; |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
113 local result = {}; |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
114 local data = desc.data or ""; |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
115 |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
116 result[#result+1] = s_char(0x80 * (desc.FIN and 1 or 0) + desc.opcode); |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
117 |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
118 length = #data; |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
119 if length <= 125 then -- 7-bit length |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
120 result[#result+1] = s_char(length); |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
121 elseif length <= 0xFFFF then -- 2-byte length |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
122 result[#result+1] = s_char(126); |
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
123 result[#result+1] = s_char(rshift(length, 8)) .. s_char(length%0x100); |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
124 else -- 8-byte length |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
125 result[#result+1] = s_char(127); |
1025
b56a9aa171b3
mod_websocket: Fix length calculation
Florian Zeitz <florob@babelmonkeys.de>
parents:
969
diff
changeset
|
126 local length_bytes = {}; |
b56a9aa171b3
mod_websocket: Fix length calculation
Florian Zeitz <florob@babelmonkeys.de>
parents:
969
diff
changeset
|
127 for i = 8, 1, -1 do |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
128 length_bytes[i] = s_char(length % 0x100); |
1025
b56a9aa171b3
mod_websocket: Fix length calculation
Florian Zeitz <florob@babelmonkeys.de>
parents:
969
diff
changeset
|
129 length = rshift(length, 8); |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
130 end |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
131 result[#result+1] = t_concat(length_bytes, ""); |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
132 end |
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
133 |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
134 result[#result+1] = data; |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
135 |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
136 return t_concat(result, ""); |
687
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
137 end |
d141375ece4b
mod_websocket: Move frame building into a function
Florian Zeitz <florob@babelmonkeys.de>
parents:
677
diff
changeset
|
138 |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
139 --- Session methods |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
140 local function session_open_stream(session) |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
141 local attr = { |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
142 xmlns = xmlns_framing, |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
143 version = "1.0", |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
144 id = session.streamid or "", |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
145 from = session.host |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
146 }; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
147 session.send(st.stanza("open", attr)); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
148 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
149 |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
150 local function session_close(session, reason) |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
151 local log = session.log or log; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
152 if session.conn then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
153 if session.notopen then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
154 session:open_stream(); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
155 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
156 if reason then -- nil == no err, initiated by us, false == initiated by client |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
157 local stream_error = st.stanza("stream:error"); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
158 if type(reason) == "string" then -- assume stream error |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
159 stream_error:tag(reason, {xmlns = 'urn:ietf:params:xml:ns:xmpp-streams' }); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
160 elseif type(reason) == "table" then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
161 if reason.condition then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
162 stream_error:tag(reason.condition, stream_xmlns_attr):up(); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
163 if reason.text then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
164 stream_error:tag("text", stream_xmlns_attr):text(reason.text):up(); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
165 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
166 if reason.extra then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
167 stream_error:add_child(reason.extra); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
168 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
169 elseif reason.name then -- a stanza |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
170 stream_error = reason; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
171 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
172 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
173 stream_error = tostring(stream_error); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
174 log("debug", "Disconnecting client, <stream:error> is: %s", stream_error); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
175 session.send(stream_error); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
176 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
177 |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
178 session.send(st.stanza("close", { xmlns = xmlns_framing })); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
179 function session.send() return false; end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
180 |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
181 local reason = (reason and (reason.name or reason.text or reason.condition)) or reason; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
182 session.log("debug", "c2s stream for %s closed: %s", session.full_jid or ("<"..session.ip..">"), reason or "session closed"); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
183 |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
184 -- Authenticated incoming stream may still be sending us stanzas, so wait for </stream:stream> from remote |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
185 local conn = session.conn; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
186 if reason == nil and not session.notopen and session.type == "c2s" then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
187 -- Grace time to process data from authenticated cleanly-closed stream |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
188 add_task(stream_close_timeout, function () |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
189 if not session.destroyed then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
190 session.log("warn", "Failed to receive a stream close response, closing connection anyway..."); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
191 sm_destroy_session(session, reason); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
192 -- Sends close with code 1000 and message "Stream closed" |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
193 local data = s_char(0x03) .. s_char(0xe8) .. "Stream closed"; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
194 conn:write(build_frame({opcode = 0x8, FIN = true, data = data})); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
195 conn:close(); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
196 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
197 end); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
198 else |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
199 sm_destroy_session(session, reason); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
200 -- Sends close with code 1000 and message "Stream closed" |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
201 local data = s_char(0x03) .. s_char(0xe8) .. "Stream closed"; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
202 conn:write(build_frame({opcode = 0x8, FIN = true, data = data})); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
203 conn:close(); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
204 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
205 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
206 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
207 |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
208 |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
209 --- Filter stuff |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
210 local function filter_open_close(data) |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
211 if not data:find(xmlns_framing, 1, true) then return data; end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
212 |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
213 local oc = parse_xml(data); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
214 if not oc then return data; end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
215 if oc.attr.xmlns ~= xmlns_framing then return data; end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
216 if oc.name == "close" then return "</stream:stream>"; end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
217 if oc.name == "open" then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
218 oc.name = "stream:stream"; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
219 oc.attr.xmlns = nil; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
220 oc.attr["xmlns:stream"] = xmlns_streams; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
221 return oc:top_tag(); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
222 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
223 |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
224 return data; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
225 end |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
226 function handle_request(event, path) |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
227 local request, response = event.request, event.response; |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
228 local conn = response.conn; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
229 |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
230 if not request.headers.sec_websocket_key then |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
231 response.headers.content_type = "text/html"; |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
232 return [[<!DOCTYPE html><html><head><title>Websocket</title></head><body> |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
233 <p>It works! Now point your WebSocket client to this URL to connect to Prosody.</p> |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
234 </body></html>]]; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
235 end |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
236 |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
237 local wants_xmpp = false; |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
238 (request.headers.sec_websocket_protocol or ""):gsub("([^,]*),?", function (proto) |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
239 if proto == "xmpp" then wants_xmpp = true; end |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
240 end); |
129 | 241 |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
242 if not wants_xmpp then |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
243 return 501; |
129 | 244 end |
245 | |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
246 local function websocket_close(code, message) |
1026
e254cf49e14d
mod_websocket: Call string and table functions through locals
Florian Zeitz <florob@babelmonkeys.de>
parents:
1025
diff
changeset
|
247 local data = s_char(rshift(code, 8)) .. s_char(code%0x100) .. message; |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
248 conn:write(build_frame({opcode = 0x8, FIN = true, data = data})); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
249 conn:close(); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
250 end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
251 |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
252 local dataBuffer; |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
253 local function handle_frame(frame) |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
254 local opcode = frame.opcode; |
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
255 local length = frame.length; |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
256 module:log("debug", "Websocket received: %s (%i bytes)", frame.data, #frame.data); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
257 |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
258 -- Error cases |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
259 if frame.RSV1 or frame.RSV2 or frame.RSV3 then -- Reserved bits non zero |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
260 websocket_close(1002, "Reserved bits not zero"); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
261 return false; |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
262 end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
263 |
1029
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
264 if opcode == 0x8 then |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
265 if length == 1 then |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
266 websocket_close(1002, "Close frame with payload, but too short for status code"); |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
267 return false; |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
268 elseif length >= 2 then |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
269 local status_code = s_byte(frame.data, 1) * 256 + s_byte(frame.data, 2) |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
270 if status_code < 1000 then |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
271 websocket_close(1002, "Closed with invalid status code"); |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
272 return false; |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
273 elseif ((status_code > 1003 and status_code < 1007) or status_code > 1011) and status_code < 3000 then |
1272
717a08403b26
mod_websocket: Fix typo "Cosed" -> "Closed"
Florian Zeitz <florob@babelmonkeys.de>
parents:
1193
diff
changeset
|
274 websocket_close(1002, "Closed with reserved status code"); |
1029
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
275 return false; |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
276 end |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
277 end |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
278 end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
279 |
1029
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
280 if opcode >= 0x8 then |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
281 if length > 125 then -- Control frame with too much payload |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
282 websocket_close(1002, "Payload too large"); |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
283 return false; |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
284 end |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
285 |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
286 if not frame.FIN then -- Fragmented control frame |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
287 websocket_close(1002, "Fragmented control frame"); |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
288 return false; |
9d85aded2fb6
mod_websocket: Add some more error checks for close frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
1028
diff
changeset
|
289 end |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
290 end |
690
5acc203972f3
mod_websocket: Add fragmentation support
Florian Zeitz <florob@babelmonkeys.de>
parents:
689
diff
changeset
|
291 |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
292 if (opcode > 0x2 and opcode < 0x8) or (opcode > 0xA) then |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
293 websocket_close(1002, "Reserved opcode"); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
294 return false; |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
295 end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
296 |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
297 if opcode == 0x0 and not dataBuffer then |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
298 websocket_close(1002, "Unexpected continuation frame"); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
299 return false; |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
300 end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
301 |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
302 if (opcode == 0x1 or opcode == 0x2) and dataBuffer then |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
303 websocket_close(1002, "Continuation frame expected"); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
304 return false; |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
305 end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
306 |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
307 -- Valid cases |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
308 if opcode == 0x0 then -- Continuation frame |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
309 dataBuffer[#dataBuffer+1] = frame.data; |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
310 elseif opcode == 0x1 then -- Text frame |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
311 dataBuffer = {frame.data}; |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
312 elseif opcode == 0x2 then -- Binary frame |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
313 websocket_close(1003, "Only text frames are supported"); |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
314 return; |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
315 elseif opcode == 0x8 then -- Close request |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
316 websocket_close(1000, "Goodbye"); |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
317 return; |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
318 elseif opcode == 0x9 then -- Ping frame |
691
04662cc35280
mod_websocket: Answer ping frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
690
diff
changeset
|
319 frame.opcode = 0xA; |
04662cc35280
mod_websocket: Answer ping frames
Florian Zeitz <florob@babelmonkeys.de>
parents:
690
diff
changeset
|
320 conn:write(build_frame(frame)); |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
321 return ""; |
690
5acc203972f3
mod_websocket: Add fragmentation support
Florian Zeitz <florob@babelmonkeys.de>
parents:
689
diff
changeset
|
322 else |
1028
81065638299d
mod_websocket: Access some values via locales instead of through tables
Florian Zeitz <florob@babelmonkeys.de>
parents:
1027
diff
changeset
|
323 log("warn", "Received frame with unsupported opcode %i", opcode); |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
324 return ""; |
690
5acc203972f3
mod_websocket: Add fragmentation support
Florian Zeitz <florob@babelmonkeys.de>
parents:
689
diff
changeset
|
325 end |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
326 |
690
5acc203972f3
mod_websocket: Add fragmentation support
Florian Zeitz <florob@babelmonkeys.de>
parents:
689
diff
changeset
|
327 if frame.FIN then |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
328 local data = t_concat(dataBuffer, ""); |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
329 dataBuffer = nil; |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
330 return data; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
331 end |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
332 return ""; |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
333 end |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
334 |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
335 conn:setlistener(c2s_listener); |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
336 c2s_listener.onconnect(conn); |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
337 |
1193
bbe278a56b0a
mod_websocket: Add consider_websocket_secure option
Florian Zeitz <florob@babelmonkeys.de>
parents:
1029
diff
changeset
|
338 local session = sessions[conn]; |
bbe278a56b0a
mod_websocket: Add consider_websocket_secure option
Florian Zeitz <florob@babelmonkeys.de>
parents:
1029
diff
changeset
|
339 |
bbe278a56b0a
mod_websocket: Add consider_websocket_secure option
Florian Zeitz <florob@babelmonkeys.de>
parents:
1029
diff
changeset
|
340 session.secure = consider_websocket_secure or session.secure; |
bbe278a56b0a
mod_websocket: Add consider_websocket_secure option
Florian Zeitz <florob@babelmonkeys.de>
parents:
1029
diff
changeset
|
341 |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
342 session.open_stream = session_open_stream; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
343 session.close = session_close; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
344 |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
345 local frameBuffer = ""; |
1193
bbe278a56b0a
mod_websocket: Add consider_websocket_secure option
Florian Zeitz <florob@babelmonkeys.de>
parents:
1029
diff
changeset
|
346 add_filter(session, "bytes/in", function(data) |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
347 local cache = {}; |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
348 frameBuffer = frameBuffer .. data; |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
349 local frame, length = parse_frame(frameBuffer); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
350 |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
351 while frame do |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
352 frameBuffer = frameBuffer:sub(length + 1); |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
353 local result = handle_frame(frame); |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
354 if not result then return; end |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
355 cache[#cache+1] = filter_open_close(result); |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
356 frame, length = parse_frame(frameBuffer); |
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
357 end |
1027
6a2dfa8af421
mod_websocket: Optimize string concatenation
Florian Zeitz <florob@babelmonkeys.de>
parents:
1026
diff
changeset
|
358 return t_concat(cache, ""); |
844
a0987c0e4e1d
mod_websocket: Check whether the xmpp sub-protocol was requested
Florian Zeitz <florob@babelmonkeys.de>
parents:
842
diff
changeset
|
359 end); |
a0987c0e4e1d
mod_websocket: Check whether the xmpp sub-protocol was requested
Florian Zeitz <florob@babelmonkeys.de>
parents:
842
diff
changeset
|
360 |
1193
bbe278a56b0a
mod_websocket: Add consider_websocket_secure option
Florian Zeitz <florob@babelmonkeys.de>
parents:
1029
diff
changeset
|
361 add_filter(session, "bytes/out", function(data) |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
362 return build_frame({ FIN = true, opcode = 0x01, data = tostring(data)}); |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
363 end); |
842
32df4d13f178
mod_websocket: More robust frame handling
Florian Zeitz <florob@babelmonkeys.de>
parents:
840
diff
changeset
|
364 |
1340
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
365 add_filter(session, "stanzas/out", function(stanza) |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
366 local attr = stanza.attr; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
367 attr.xmlns = attr.xmlns or xmlns_client; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
368 if stanza.name:find("^stream:") then |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
369 attr["xmlns:stream"] = attr["xmlns:stream"] or xmlns_streams; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
370 end |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
371 return stanza; |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
372 end); |
3ffd64b4ab59
mod_websocket: Update to draft-ietf-xmpp-websocket-01
Florian Zeitz <florob@babelmonkeys.de>
parents:
1325
diff
changeset
|
373 |
969
8eba9d4809d2
mod_websocket: Set status_code instead of status
Florian Zeitz <florob@babelmonkeys.de>
parents:
910
diff
changeset
|
374 response.status_code = 101; |
895
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
375 response.headers.upgrade = "websocket"; |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
376 response.headers.connection = "Upgrade"; |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
377 response.headers.sec_webSocket_accept = base64(sha1(request.headers.sec_websocket_key .. "258EAFA5-E914-47DA-95CA-C5AB0DC85B11")); |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
378 response.headers.sec_webSocket_protocol = "xmpp"; |
1f4d77104da5
mod_websocket: Simplify by getting the c2s_listener from mod_c2s
Florian Zeitz <florob@babelmonkeys.de>
parents:
857
diff
changeset
|
379 response.headers.access_control_allow_origin = cross_domain; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
380 |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
381 return ""; |
129 | 382 end |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
383 |
688
e87678a52720
mod_websocket: Make this a shared module
Florian Zeitz <florob@babelmonkeys.de>
parents:
687
diff
changeset
|
384 function module.add_host(module) |
e87678a52720
mod_websocket: Make this a shared module
Florian Zeitz <florob@babelmonkeys.de>
parents:
687
diff
changeset
|
385 module:depends("http"); |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
386 module:provides("http", { |
909
ec4c6e8f277d
mod_websocket: Change the HTTP name to websocket, and override the default_path instead
Florian Zeitz <florob@babelmonkeys.de>
parents:
905
diff
changeset
|
387 name = "websocket"; |
ec4c6e8f277d
mod_websocket: Change the HTTP name to websocket, and override the default_path instead
Florian Zeitz <florob@babelmonkeys.de>
parents:
905
diff
changeset
|
388 default_path = "xmpp-websocket"; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
389 route = { |
849
6e2ec1825182
mod_websocket: Accept GET request without trailing /
Florian Zeitz <florob@babelmonkeys.de>
parents:
844
diff
changeset
|
390 ["GET"] = handle_request; |
6e2ec1825182
mod_websocket: Accept GET request without trailing /
Florian Zeitz <florob@babelmonkeys.de>
parents:
844
diff
changeset
|
391 ["GET /"] = handle_request; |
676
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
392 }; |
54fa9d6d7809
mod_websocket: New mod_c2s based version, still WIP
Florian Zeitz <florob@babelmonkeys.de>
parents:
129
diff
changeset
|
393 }); |
129 | 394 end |