author | Kim Alvefur <zash@zash.se> |
Fri, 04 Mar 2022 16:58:28 +0100 | |
changeset 12364 | 0801db678f5e |
parent 12363 | f81488951f81 |
child 12371 | 7a2f036f73b3 |
permissions | -rw-r--r-- |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
1 |
local base64 = require "util.encodings".base64; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
2 |
local hashes = require "util.hashes"; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
3 |
local net = require "util.net"; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
4 |
local random = require "util.random"; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
5 |
local struct = require "util.struct"; |
12364
0801db678f5e
net.stun: Use util.bitcompat to deal with bit module variances across Lua versions
Kim Alvefur <zash@zash.se>
parents:
12363
diff
changeset
|
6 |
local bit32 = require"util.bitcompat"; |
12363 | 7 |
local sxor = require"util.strbitop".sxor; |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
8 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
9 |
--- Public helpers |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
10 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
11 |
-- Following draft-uberti-behave-turn-rest-00, convert a 'secret' string |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
12 |
-- into a username/password pair that can be used to auth to a TURN server |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
13 |
local function get_user_pass_from_secret(secret, ttl, opt_username) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
14 |
ttl = ttl or 86400; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
15 |
local username; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
16 |
if opt_username then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
17 |
username = ("%d:%s"):format(os.time() + ttl, opt_username); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
18 |
else |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
19 |
username = ("%d"):format(os.time() + ttl); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
20 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
21 |
local password = base64.encode(hashes.hmac_sha1(secret, username)); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
22 |
return username, password, ttl; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
23 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
24 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
25 |
-- Following RFC 8489 9.2, convert credentials to a HMAC key for signing |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
26 |
local function get_long_term_auth_key(realm, username, password) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
27 |
return hashes.md5(username..":"..realm..":"..password); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
28 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
29 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
30 |
--- Packet building/parsing |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
31 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
32 |
local packet_methods = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
33 |
local packet_mt = { __index = packet_methods }; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
34 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
35 |
local magic_cookie = string.char(0x21, 0x12, 0xA4, 0x42); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
36 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
37 |
local methods = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
38 |
binding = 0x001; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
39 |
-- TURN |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
40 |
allocate = 0x003; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
41 |
refresh = 0x004; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
42 |
send = 0x006; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
43 |
data = 0x007; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
44 |
create_permission = 0x008; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
45 |
channel_bind = 0x009; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
46 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
47 |
local method_lookup = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
48 |
for name, value in pairs(methods) do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
49 |
method_lookup[name] = value; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
50 |
method_lookup[value] = name; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
51 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
52 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
53 |
local classes = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
54 |
request = 0; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
55 |
indication = 1; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
56 |
success = 2; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
57 |
error = 3; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
58 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
59 |
local class_lookup = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
60 |
for name, value in pairs(classes) do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
61 |
class_lookup[name] = value; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
62 |
class_lookup[value] = name; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
63 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
64 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
65 |
local attributes = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
66 |
["mapped-address"] = 0x0001; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
67 |
["username"] = 0x0006; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
68 |
["message-integrity"] = 0x0008; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
69 |
["error-code"] = 0x0009; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
70 |
["unknown-attributes"] = 0x000A; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
71 |
["realm"] = 0x0014; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
72 |
["nonce"] = 0x0015; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
73 |
["xor-mapped-address"] = 0x0020; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
74 |
["software"] = 0x8022; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
75 |
["alternate-server"] = 0x8023; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
76 |
["fingerprint"] = 0x8028; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
77 |
["message-integrity-sha256"] = 0x001C; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
78 |
["password-algorithm"] = 0x001D; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
79 |
["userhash"] = 0x001E; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
80 |
["password-algorithms"] = 0x8002; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
81 |
["alternate-domains"] = 0x8003; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
82 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
83 |
-- TURN |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
84 |
["requested-transport"] = 0x0019; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
85 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
86 |
local attribute_lookup = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
87 |
for name, value in pairs(attributes) do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
88 |
attribute_lookup[name] = value; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
89 |
attribute_lookup[value] = name; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
90 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
91 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
92 |
function packet_methods:serialize_header(length) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
93 |
assert(#self.transaction_id == 12, "invalid transaction id length"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
94 |
local header = struct.pack(">I2I2", |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
95 |
self.type, |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
96 |
length |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
97 |
)..magic_cookie..self.transaction_id; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
98 |
return header; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
99 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
100 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
101 |
function packet_methods:serialize() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
102 |
local payload = table.concat(self.attributes); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
103 |
return self:serialize_header(#payload)..payload; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
104 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
105 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
106 |
function packet_methods:is_request() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
107 |
return bit32.band(self.type, 0x0110) == 0x0000; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
108 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
109 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
110 |
function packet_methods:is_indication() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
111 |
return bit32.band(self.type, 0x0110) == 0x0010; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
112 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
113 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
114 |
function packet_methods:is_success_resp() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
115 |
return bit32.band(self.type, 0x0110) == 0x0100; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
116 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
117 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
118 |
function packet_methods:is_err_resp() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
119 |
return bit32.band(self.type, 0x0110) == 0x0110; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
120 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
121 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
122 |
function packet_methods:get_method() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
123 |
local method = bit32.bor( |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
124 |
bit32.rshift(bit32.band(self.type, 0x3E00), 2), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
125 |
bit32.rshift(bit32.band(self.type, 0x00E0), 1), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
126 |
bit32.band(self.type, 0x000F) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
127 |
); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
128 |
return method, method_lookup[method]; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
129 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
130 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
131 |
function packet_methods:get_class() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
132 |
local class = bit32.bor( |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
133 |
bit32.rshift(bit32.band(self.type, 0x0100), 7), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
134 |
bit32.rshift(bit32.band(self.type, 0x0010), 4) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
135 |
); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
136 |
return class, class_lookup[class]; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
137 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
138 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
139 |
function packet_methods:set_type(method, class) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
140 |
if type(method) == "string" then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
141 |
method = assert(method_lookup[method:lower()], "unknown method: "..method); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
142 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
143 |
if type(class) == "string" then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
144 |
class = assert(classes[class], "unknown class: "..class); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
145 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
146 |
self.type = bit32.bor( |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
147 |
bit32.lshift(bit32.band(method, 0x1F80), 2), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
148 |
bit32.lshift(bit32.band(method, 0x0070), 1), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
149 |
bit32.band(method, 0x000F), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
150 |
bit32.lshift(bit32.band(class, 0x0002), 7), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
151 |
bit32.lshift(bit32.band(class, 0x0001), 4) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
152 |
); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
153 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
154 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
155 |
local function _serialize_attribute(attr_type, value) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
156 |
local len = #value; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
157 |
local padding = string.rep("\0", (4 - len)%4); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
158 |
return struct.pack(">I2I2", |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
159 |
attr_type, len |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
160 |
)..value..padding; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
161 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
162 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
163 |
function packet_methods:add_attribute(attr_type, value) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
164 |
if type(attr_type) == "string" then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
165 |
attr_type = assert(attributes[attr_type], "unknown attribute: "..attr_type); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
166 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
167 |
table.insert(self.attributes, _serialize_attribute(attr_type, value)); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
168 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
169 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
170 |
function packet_methods:deserialize(bytes) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
171 |
local type, len, cookie = struct.unpack(">I2I2I4", bytes); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
172 |
assert(#bytes == (len + 20), "incorrect packet length"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
173 |
assert(cookie == 0x2112A442, "invalid magic cookie"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
174 |
self.type = type; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
175 |
self.transaction_id = bytes:sub(9, 20); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
176 |
self.attributes = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
177 |
local pos = 21; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
178 |
while pos < #bytes do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
179 |
local attr_hdr = bytes:sub(pos, pos+3); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
180 |
assert(#attr_hdr == 4, "packet truncated in attribute header"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
181 |
local attr_type, attr_len = struct.unpack(">I2I2", attr_hdr); --luacheck: ignore 211/attr_type |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
182 |
if attr_len == 0 then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
183 |
table.insert(self.attributes, attr_hdr); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
184 |
pos = pos + 20; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
185 |
else |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
186 |
local data = bytes:sub(pos + 4, pos + 3 + attr_len); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
187 |
assert(#data == attr_len, "packet truncated in attribute value"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
188 |
table.insert(self.attributes, attr_hdr..data); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
189 |
local n_padding = (4 - attr_len)%4; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
190 |
pos = pos + 4 + attr_len + n_padding; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
191 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
192 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
193 |
return self; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
194 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
195 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
196 |
function packet_methods:get_attribute(attr_type) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
197 |
if type(attr_type) == "string" then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
198 |
attr_type = assert(attribute_lookup[attr_type:lower()], "unknown attribute: "..attr_type); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
199 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
200 |
for _, attribute in ipairs(self.attributes) do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
201 |
if struct.unpack(">I2", attribute) == attr_type then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
202 |
return attribute:sub(5); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
203 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
204 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
205 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
206 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
207 |
local addr_families = { "IPv4", "IPv6" }; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
208 |
function packet_methods:get_mapped_address() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
209 |
local data = self:get_attribute("mapped-address"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
210 |
if not data then return; end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
211 |
local family, port = struct.unpack("x>BI2", data); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
212 |
local addr = data:sub(5); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
213 |
return { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
214 |
family = addr_families[family] or "unknown"; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
215 |
port = port; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
216 |
address = net.ntop(addr); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
217 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
218 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
219 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
220 |
function packet_methods:get_xor_mapped_address() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
221 |
local data = self:get_attribute("xor-mapped-address"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
222 |
if not data then return; end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
223 |
local family, port = struct.unpack("x>BI2", data); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
224 |
local addr = sxor(data:sub(5), magic_cookie..self.transaction_id); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
225 |
return { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
226 |
family = addr_families[family] or "unknown"; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
227 |
port = bit32.bxor(port, 0x2112); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
228 |
address = net.ntop(addr); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
229 |
address_raw = data:sub(5); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
230 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
231 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
232 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
233 |
function packet_methods:add_message_integrity(key) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
234 |
-- Add attribute with a dummy value so we can artificially increase |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
235 |
-- the packet 'length' |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
236 |
self:add_attribute("message-integrity", string.rep("\0", 20)); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
237 |
-- Get the packet data, minus the message-integrity attribute itself |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
238 |
local pkt = self:serialize():sub(1, -25); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
239 |
local hash = hashes.hmac_sha1(key, pkt, false); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
240 |
self.attributes[#self.attributes] = nil; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
241 |
assert(#hash == 20, "invalid hash length"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
242 |
self:add_attribute("message-integrity", hash); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
243 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
244 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
245 |
do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
246 |
local transports = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
247 |
udp = 0x11; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
248 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
249 |
function packet_methods:add_requested_transport(transport) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
250 |
local transport_code = transports[transport]; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
251 |
assert(transport_code, "unsupported transport: "..tostring(transport)); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
252 |
self:add_attribute("requested-transport", string.char( |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
253 |
transport_code, 0x00, 0x00, 0x00 |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
254 |
)); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
255 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
256 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
257 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
258 |
function packet_methods:get_error() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
259 |
local err_attr = self:get_attribute("error-code"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
260 |
if not err_attr then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
261 |
return nil; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
262 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
263 |
local number = err_attr:byte(4); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
264 |
local class = bit32.band(0x07, err_attr:byte(3)); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
265 |
local msg = err_attr:sub(5); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
266 |
return (class*100)+number, msg; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
267 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
268 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
269 |
local function new_packet(method, class) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
270 |
local p = setmetatable({ |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
271 |
transaction_id = random.bytes(12); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
272 |
length = 0; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
273 |
attributes = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
274 |
}, packet_mt); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
275 |
p:set_type(method or "binding", class or "request"); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
276 |
return p; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
277 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
278 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
279 |
return { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
280 |
new_packet = new_packet; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
281 |
get_user_pass_from_secret = get_user_pass_from_secret; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
282 |
get_long_term_auth_key = get_long_term_auth_key; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
283 |
}; |