| OLD | NEW |
| (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 | |
| OLD | NEW |