| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2013 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2013 The WebRTC Project Authors. All rights reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #include "webrtc/p2p/base/asyncstuntcpsocket.h" | 11 #include "webrtc/p2p/base/asyncstuntcpsocket.h" |
| 12 | 12 |
| 13 #include <string.h> | 13 #include <string.h> |
| 14 | 14 |
| 15 #include "webrtc/p2p/base/stun.h" | 15 #include "webrtc/p2p/base/stun.h" |
| 16 #include "webrtc/base/checks.h" |
| 16 #include "webrtc/base/common.h" | 17 #include "webrtc/base/common.h" |
| 17 #include "webrtc/base/logging.h" | 18 #include "webrtc/base/logging.h" |
| 18 | 19 |
| 19 namespace cricket { | 20 namespace cricket { |
| 20 | 21 |
| 21 static const size_t kMaxPacketSize = 64 * 1024; | 22 static const size_t kMaxPacketSize = 64 * 1024; |
| 22 | 23 |
| 23 typedef uint16_t PacketLength; | 24 typedef uint16_t PacketLength; |
| 24 static const size_t kPacketLenSize = sizeof(PacketLength); | 25 static const size_t kPacketLenSize = sizeof(PacketLength); |
| 25 static const size_t kPacketLenOffset = 2; | 26 static const size_t kPacketLenOffset = 2; |
| (...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 61 | 62 |
| 62 int pad_bytes; | 63 int pad_bytes; |
| 63 size_t expected_pkt_len = GetExpectedLength(pv, cb, &pad_bytes); | 64 size_t expected_pkt_len = GetExpectedLength(pv, cb, &pad_bytes); |
| 64 | 65 |
| 65 // Accepts only complete STUN/ChannelData packets. | 66 // Accepts only complete STUN/ChannelData packets. |
| 66 if (cb != expected_pkt_len) | 67 if (cb != expected_pkt_len) |
| 67 return -1; | 68 return -1; |
| 68 | 69 |
| 69 AppendToOutBuffer(pv, cb); | 70 AppendToOutBuffer(pv, cb); |
| 70 | 71 |
| 71 ASSERT(pad_bytes < 4); | 72 RTC_DCHECK(pad_bytes < 4); |
| 72 char padding[4] = {0}; | 73 char padding[4] = {0}; |
| 73 AppendToOutBuffer(padding, pad_bytes); | 74 AppendToOutBuffer(padding, pad_bytes); |
| 74 | 75 |
| 75 int res = FlushOutBuffer(); | 76 int res = FlushOutBuffer(); |
| 76 if (res <= 0) { | 77 if (res <= 0) { |
| 77 // drop packet if we made no progress | 78 // drop packet if we made no progress |
| 78 ClearOutBuffer(); | 79 ClearOutBuffer(); |
| 79 return res; | 80 return res; |
| 80 } | 81 } |
| 81 | 82 |
| (...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 144 // message (including padding) is (4 + Length) rounded up to the nearest | 145 // message (including padding) is (4 + Length) rounded up to the nearest |
| 145 // multiple of 4. Over UDP, the padding is not required but MAY be | 146 // multiple of 4. Over UDP, the padding is not required but MAY be |
| 146 // included. | 147 // included. |
| 147 if (expected_pkt_len % 4) | 148 if (expected_pkt_len % 4) |
| 148 *pad_bytes = 4 - (expected_pkt_len % 4); | 149 *pad_bytes = 4 - (expected_pkt_len % 4); |
| 149 } | 150 } |
| 150 return expected_pkt_len; | 151 return expected_pkt_len; |
| 151 } | 152 } |
| 152 | 153 |
| 153 } // namespace cricket | 154 } // namespace cricket |
| OLD | NEW |