author | Kim Alvefur <zash@zash.se> |
Sun, 07 Apr 2024 18:36:00 +0200 | |
changeset 13480 | b9267ce1b183 |
parent 12978 | ba409c67353b |
permissions | -rw-r--r-- |
12978
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
1 |
local base64 = require "prosody.util.encodings".base64; |
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
2 |
local hashes = require "prosody.util.hashes"; |
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
3 |
local net = require "prosody.util.net"; |
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
4 |
local random = require "prosody.util.random"; |
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
5 |
local struct = require "prosody.util.struct"; |
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
6 |
local bit32 = require"prosody.util.bitcompat"; |
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
7 |
local sxor = require"prosody.util.strbitop".sxor; |
ba409c67353b
net: Prefix module imports with prosody namespace
Kim Alvefur <zash@zash.se>
parents:
12378
diff
changeset
|
8 |
local new_ip = require "prosody.util.ip".new_ip; |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
9 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
10 |
--- Public helpers |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
11 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
12 |
-- 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
|
13 |
-- 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
|
14 |
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
|
15 |
ttl = ttl or 86400; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
16 |
local username; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
17 |
if opt_username then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
18 |
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
|
19 |
else |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
20 |
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
|
21 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
22 |
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
|
23 |
return username, password, ttl; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
24 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
25 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
26 |
-- 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
|
27 |
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
|
28 |
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
|
29 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
30 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
31 |
--- Packet building/parsing |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
32 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
33 |
local packet_methods = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
34 |
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
|
35 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
36 |
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
|
37 |
|
12372
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
38 |
local function lookup_table(t) |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
39 |
local lookup = {}; |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
40 |
for k, v in pairs(t) do |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
41 |
lookup[k] = v; |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
42 |
lookup[v] = k; |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
43 |
end |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
44 |
return lookup; |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
45 |
end |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
46 |
|
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
47 |
local methods = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
48 |
binding = 0x001; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
49 |
-- TURN |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
50 |
allocate = 0x003; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
51 |
refresh = 0x004; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
52 |
send = 0x006; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
53 |
data = 0x007; |
12371
7a2f036f73b3
net.stun: Hyphenate method names for consistency
Matthew Wild <mwild1@gmail.com>
parents:
12364
diff
changeset
|
54 |
["create-permission"] = 0x008; |
7a2f036f73b3
net.stun: Hyphenate method names for consistency
Matthew Wild <mwild1@gmail.com>
parents:
12364
diff
changeset
|
55 |
["channel-bind"] = 0x009; |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
56 |
}; |
12372
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
57 |
local method_lookup = lookup_table(methods); |
12360
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 classes = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
60 |
request = 0; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
61 |
indication = 1; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
62 |
success = 2; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
63 |
error = 3; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
64 |
}; |
12372
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
65 |
local class_lookup = lookup_table(classes); |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
66 |
|
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
67 |
local addr_families = { "IPv4", "IPv6" }; |
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
68 |
local addr_family_lookup = lookup_table(addr_families); |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
69 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
70 |
local attributes = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
71 |
["mapped-address"] = 0x0001; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
72 |
["username"] = 0x0006; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
73 |
["message-integrity"] = 0x0008; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
74 |
["error-code"] = 0x0009; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
75 |
["unknown-attributes"] = 0x000A; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
76 |
["realm"] = 0x0014; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
77 |
["nonce"] = 0x0015; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
78 |
["xor-mapped-address"] = 0x0020; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
79 |
["software"] = 0x8022; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
80 |
["alternate-server"] = 0x8023; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
81 |
["fingerprint"] = 0x8028; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
82 |
["message-integrity-sha256"] = 0x001C; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
83 |
["password-algorithm"] = 0x001D; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
84 |
["userhash"] = 0x001E; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
85 |
["password-algorithms"] = 0x8002; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
86 |
["alternate-domains"] = 0x8003; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
87 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
88 |
-- TURN |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
89 |
["requested-transport"] = 0x0019; |
12373
f2ae9c6d1d9f
net.stun: Name some more attributes from TURN
Matthew Wild <mwild1@gmail.com>
parents:
12372
diff
changeset
|
90 |
["xor-peer-address"] = 0x0012; |
f2ae9c6d1d9f
net.stun: Name some more attributes from TURN
Matthew Wild <mwild1@gmail.com>
parents:
12372
diff
changeset
|
91 |
["data"] = 0x0013; |
f2ae9c6d1d9f
net.stun: Name some more attributes from TURN
Matthew Wild <mwild1@gmail.com>
parents:
12372
diff
changeset
|
92 |
["xor-relayed-address"] = 0x0016; |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
93 |
}; |
12372
d82c8efc6dd5
net.stun: Add lookup table generation helper, reduces code duplication
Matthew Wild <mwild1@gmail.com>
parents:
12371
diff
changeset
|
94 |
local attribute_lookup = lookup_table(attributes); |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
95 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
96 |
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
|
97 |
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
|
98 |
local header = struct.pack(">I2I2", |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
99 |
self.type, |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
100 |
length |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
101 |
)..magic_cookie..self.transaction_id; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
102 |
return header; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
103 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
104 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
105 |
function packet_methods:serialize() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
106 |
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
|
107 |
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
|
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_request() |
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) == 0x0000; |
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_indication() |
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) == 0x0010; |
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_success_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) == 0x0100; |
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:is_err_resp() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
123 |
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
|
124 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
125 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
126 |
function packet_methods:get_method() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
127 |
local method = bit32.bor( |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
128 |
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
|
129 |
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
|
130 |
bit32.band(self.type, 0x000F) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
131 |
); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
132 |
return method, method_lookup[method]; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
133 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
134 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
135 |
function packet_methods:get_class() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
136 |
local class = bit32.bor( |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
137 |
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
|
138 |
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
|
139 |
); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
140 |
return class, class_lookup[class]; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
141 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
142 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
143 |
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
|
144 |
if type(method) == "string" then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
145 |
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
|
146 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
147 |
if type(class) == "string" then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
148 |
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
|
149 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
150 |
self.type = bit32.bor( |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
151 |
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
|
152 |
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
|
153 |
bit32.band(method, 0x000F), |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
154 |
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
|
155 |
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
|
156 |
); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
157 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
158 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
159 |
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
|
160 |
local len = #value; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
161 |
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
|
162 |
return struct.pack(">I2I2", |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
163 |
attr_type, len |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
164 |
)..value..padding; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
165 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
166 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
167 |
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
|
168 |
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
|
169 |
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
|
170 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
171 |
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
|
172 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
173 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
174 |
function packet_methods:deserialize(bytes) |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
175 |
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
|
176 |
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
|
177 |
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
|
178 |
self.type = type; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
179 |
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
|
180 |
self.attributes = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
181 |
local pos = 21; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
182 |
while pos < #bytes do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
183 |
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
|
184 |
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
|
185 |
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
|
186 |
if attr_len == 0 then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
187 |
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
|
188 |
pos = pos + 20; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
189 |
else |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
190 |
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
|
191 |
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
|
192 |
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
|
193 |
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
|
194 |
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
|
195 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
196 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
197 |
return self; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
198 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
199 |
|
12378
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
200 |
function packet_methods:get_attribute(attr_type, idx) |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
201 |
idx = math.max(idx or 1, 1); |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
202 |
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
|
203 |
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
|
204 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
205 |
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
|
206 |
if struct.unpack(">I2", attribute) == attr_type then |
12378
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
207 |
if idx == 1 then |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
208 |
return attribute:sub(5); |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
209 |
else |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
210 |
idx = idx - 1; |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
211 |
end |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
212 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
213 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
214 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
215 |
|
12374
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
216 |
function packet_methods:_unpack_address(data, xor) |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
217 |
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
|
218 |
local addr = data:sub(5); |
12374
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
219 |
if xor then |
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
220 |
port = bit32.bxor(port, 0x2112); |
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
221 |
addr = sxor(addr, magic_cookie..self.transaction_id); |
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
222 |
end |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
223 |
return { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
224 |
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
|
225 |
port = port; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
226 |
address = net.ntop(addr); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
227 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
228 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
229 |
|
12375
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
230 |
function packet_methods:_pack_address(family, addr, port, xor) |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
231 |
if xor then |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
232 |
port = bit32.bxor(port, 0x2112); |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
233 |
addr = sxor(addr, magic_cookie..self.transaction_id); |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
234 |
end |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
235 |
local family_port = struct.pack("x>BI2", family, port); |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
236 |
return family_port..addr |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
237 |
end |
12374
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
238 |
|
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
239 |
function packet_methods:get_mapped_address() |
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
240 |
local data = self:get_attribute("mapped-address"); |
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
241 |
if not data then return; end |
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
242 |
return self:_unpack_address(data, false); |
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
243 |
end |
12375
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
244 |
|
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
245 |
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
|
246 |
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
|
247 |
if not data then return; end |
12374
9889b1815d31
net.stun: Factor out address unpack, an operation common to multiple attributes
Matthew Wild <mwild1@gmail.com>
parents:
12373
diff
changeset
|
248 |
return self:_unpack_address(data, true); |
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
249 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
250 |
|
12375
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
251 |
function packet_methods:add_xor_peer_address(address, port) |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
252 |
local parsed_ip = assert(new_ip(address)); |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
253 |
local family = assert(addr_family_lookup[parsed_ip.proto], "Unknown IP address family: "..parsed_ip.proto); |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
254 |
self:add_attribute("xor-peer-address", self:_pack_address(family, parsed_ip.packed, port or 0, true)); |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
255 |
end |
9a8b0c5b4b14
net.stun: Add xor-peer-address helper
Matthew Wild <mwild1@gmail.com>
parents:
12374
diff
changeset
|
256 |
|
12378
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
257 |
function packet_methods:get_xor_relayed_address(idx) |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
258 |
local data = self:get_attribute("xor-relayed-address", idx); |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
259 |
if not data then return; end |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
260 |
return self:_unpack_address(data, true); |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
261 |
end |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
262 |
|
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
263 |
function packet_methods:get_xor_relayed_addresses() |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
264 |
return { |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
265 |
self:get_xor_relayed_address(1); |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
266 |
self:get_xor_relayed_address(2); |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
267 |
}; |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
268 |
end |
0602245fc84e
net.stun: Support for xor-relayed-address attribute
Matthew Wild <mwild1@gmail.com>
parents:
12375
diff
changeset
|
269 |
|
12360
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
270 |
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
|
271 |
-- 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
|
272 |
-- the packet 'length' |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
273 |
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
|
274 |
-- 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
|
275 |
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
|
276 |
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
|
277 |
self.attributes[#self.attributes] = nil; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
278 |
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
|
279 |
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
|
280 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
281 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
282 |
do |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
283 |
local transports = { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
284 |
udp = 0x11; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
285 |
}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
286 |
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
|
287 |
local transport_code = transports[transport]; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
288 |
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
|
289 |
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
|
290 |
transport_code, 0x00, 0x00, 0x00 |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
291 |
)); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
292 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
293 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
294 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
295 |
function packet_methods:get_error() |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
296 |
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
|
297 |
if not err_attr then |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
298 |
return nil; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
299 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
300 |
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
|
301 |
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
|
302 |
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
|
303 |
return (class*100)+number, msg; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
304 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
305 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
306 |
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
|
307 |
local p = setmetatable({ |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
308 |
transaction_id = random.bytes(12); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
309 |
length = 0; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
310 |
attributes = {}; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
311 |
}, packet_mt); |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
312 |
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
|
313 |
return p; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
314 |
end |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
315 |
|
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
316 |
return { |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
317 |
new_packet = new_packet; |
0f77e28df5c8
net.stun: New library that implements STUN/TURN parsing/serialization
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
318 |
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
|
319 |
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
|
320 |
}; |