Software /
code /
prosody
Annotate
net/stun.lua @ 12367:7a2f036f73b3
net.stun: Hyphenate method names for consistency
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Sat, 05 Mar 2022 10:57:47 +0000 |
parent | 12360:0801db678f5e |
child | 12368:d82c8efc6dd5 |
rev | line source |
---|---|
12356
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"; |
12360
0801db678f5e
net.stun: Use util.bitcompat to deal with bit module variances across Lua versions
Kim Alvefur <zash@zash.se>
parents:
12359
diff
changeset
|
6 local bit32 = require"util.bitcompat"; |
12359 | 7 local sxor = require"util.strbitop".sxor; |
12356
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; |
12367
7a2f036f73b3
net.stun: Hyphenate method names for consistency
Matthew Wild <mwild1@gmail.com>
parents:
12360
diff
changeset
|
44 ["create-permission"] = 0x008; |
7a2f036f73b3
net.stun: Hyphenate method names for consistency
Matthew Wild <mwild1@gmail.com>
parents:
12360
diff
changeset
|
45 ["channel-bind"] = 0x009; |
12356
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 }; |