Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(5)

Side by Side Diff: talk/media/base/turnutils.cc

Issue 1578323002: Add rtppacketuil.h/cc (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « talk/media/base/turnutils.h ('k') | talk/media/base/turnutils_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 /*
2 * libjingle
3 * Copyright 2016 Google Inc.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 *
8 * 1. Redistributions of source code must retain the above copyright notice,
9 * this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright notice,
11 * this list of conditions and the following disclaimer in the documentation
12 * and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28 #include "talk/media/base/turnutils.h"
29
30 #include "webrtc/base/byteorder.h"
31 #include "webrtc/base/checks.h"
32 #include "webrtc/p2p/base/stun.h"
33
34 namespace cricket {
35
36 namespace {
37
38 const size_t kTurnChannelHeaderLength = 4;
39
40 bool IsTurnChannelData(const uint8_t* data, size_t length) {
41 return length >= kTurnChannelHeaderLength && ((*data & 0xC0) == 0x40);
42 }
43
44 bool IsTurnSendIndicationPacket(const uint8_t* data, size_t length) {
45 if (length < kStunHeaderSize) {
46 return false;
47 }
48
49 uint16_t type = rtc::GetBE16(data);
50 return (type == TURN_SEND_INDICATION);
51 }
52
53 } // namespace
54
55 bool UnwrapTurnPacket(const uint8_t* packet,
56 size_t packet_size,
57 size_t* content_position,
58 size_t* content_size) {
59 if (IsTurnChannelData(packet, packet_size)) {
60 // Turn Channel Message header format.
61 // 0 1 2 3
62 // 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
63 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
64 // | Channel Number | Length |
65 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
66 // | |
67 // / Application Data /
68 // / /
69 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
70 size_t length = rtc::GetBE16(&packet[2]);
71 if (length + kTurnChannelHeaderLength > packet_size) {
72 return false;
73 }
74
75 *content_position = kTurnChannelHeaderLength;
76 *content_size = length;
77 return true;
78 }
79
80 if (IsTurnSendIndicationPacket(packet, packet_size)) {
81 // Validate STUN message length.
82 const size_t stun_message_length = rtc::GetBE16(&packet[2]);
83 if (stun_message_length + kStunHeaderSize != packet_size) {
84 return false;
85 }
86
87 // First skip mandatory stun header which is of 20 bytes.
88 size_t pos = kStunHeaderSize;
89 // Loop through STUN attributes until we find STUN DATA attribute.
90 while (pos < packet_size) {
91 // Keep reading STUN attributes until we hit DATA attribute.
92 // Attribute will be a TLV structure.
93 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
94 // | Type | Length |
95 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
96 // | Value (variable) ....
97 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
98 // The value in the length field MUST contain the length of the Value
99 // part of the attribute, prior to padding, measured in bytes. Since
100 // STUN aligns attributes on 32-bit boundaries, attributes whose content
101 // is not a multiple of 4 bytes are padded with 1, 2, or 3 bytes of
102 // padding so that its value contains a multiple of 4 bytes. The
103 // padding bits are ignored, and may be any value.
104 uint16_t attr_type, attr_length;
105 const int kAttrHeaderLength = sizeof(attr_type) + sizeof(attr_length);
106
107 if (packet_size < pos + kAttrHeaderLength) {
108 return false;
109 }
110
111 // Getting attribute type and length.
112 attr_type = rtc::GetBE16(&packet[pos]);
113 attr_length = rtc::GetBE16(&packet[pos + sizeof(attr_type)]);
114
115 pos += kAttrHeaderLength; // Skip STUN_DATA_ATTR header.
116
117 // Checking for bogus attribute length.
118 if (pos + attr_length > packet_size) {
119 return false;
120 }
121
122 if (attr_type == STUN_ATTR_DATA) {
123 *content_position = pos;
124 *content_size = attr_length;
125 return true;
126 }
127
128 pos += attr_length;
129 if ((attr_length % 4) != 0) {
130 pos += (4 - (attr_length % 4));
131 }
132 }
133
134 // There is no data attribute present in the message.
135 return false;
136 }
137
138 // This is not a TURN packet.
139 *content_position = 0;
140 *content_size = packet_size;
141 return true;
142 }
143
144 } // namespace cricket
OLDNEW
« no previous file with comments | « talk/media/base/turnutils.h ('k') | talk/media/base/turnutils_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698