| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2004 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/base/natsocketfactory.h" | 11 #include "webrtc/base/natsocketfactory.h" |
| 12 | 12 |
| 13 #include "webrtc/base/arraysize.h" | 13 #include "webrtc/base/arraysize.h" |
| 14 #include "webrtc/base/checks.h" |
| 14 #include "webrtc/base/logging.h" | 15 #include "webrtc/base/logging.h" |
| 15 #include "webrtc/base/natserver.h" | 16 #include "webrtc/base/natserver.h" |
| 16 #include "webrtc/base/virtualsocketserver.h" | 17 #include "webrtc/base/virtualsocketserver.h" |
| 17 | 18 |
| 18 namespace rtc { | 19 namespace rtc { |
| 19 | 20 |
| 20 // Packs the given socketaddress into the buffer in buf, in the quasi-STUN | 21 // Packs the given socketaddress into the buffer in buf, in the quasi-STUN |
| 21 // format that the natserver uses. | 22 // format that the natserver uses. |
| 22 // Returns 0 if an invalid address is passed. | 23 // Returns 0 if an invalid address is passed. |
| 23 size_t PackAddressForNAT(char* buf, size_t buf_size, | 24 size_t PackAddressForNAT(char* buf, size_t buf_size, |
| 24 const SocketAddress& remote_addr) { | 25 const SocketAddress& remote_addr) { |
| 25 const IPAddress& ip = remote_addr.ipaddr(); | 26 const IPAddress& ip = remote_addr.ipaddr(); |
| 26 int family = ip.family(); | 27 int family = ip.family(); |
| 27 buf[0] = 0; | 28 buf[0] = 0; |
| 28 buf[1] = family; | 29 buf[1] = family; |
| 29 // Writes the port. | 30 // Writes the port. |
| 30 *(reinterpret_cast<uint16_t*>(&buf[2])) = HostToNetwork16(remote_addr.port()); | 31 *(reinterpret_cast<uint16_t*>(&buf[2])) = HostToNetwork16(remote_addr.port()); |
| 31 if (family == AF_INET) { | 32 if (family == AF_INET) { |
| 32 ASSERT(buf_size >= kNATEncodedIPv4AddressSize); | 33 RTC_DCHECK(buf_size >= kNATEncodedIPv4AddressSize); |
| 33 in_addr v4addr = ip.ipv4_address(); | 34 in_addr v4addr = ip.ipv4_address(); |
| 34 memcpy(&buf[4], &v4addr, kNATEncodedIPv4AddressSize - 4); | 35 memcpy(&buf[4], &v4addr, kNATEncodedIPv4AddressSize - 4); |
| 35 return kNATEncodedIPv4AddressSize; | 36 return kNATEncodedIPv4AddressSize; |
| 36 } else if (family == AF_INET6) { | 37 } else if (family == AF_INET6) { |
| 37 ASSERT(buf_size >= kNATEncodedIPv6AddressSize); | 38 RTC_DCHECK(buf_size >= kNATEncodedIPv6AddressSize); |
| 38 in6_addr v6addr = ip.ipv6_address(); | 39 in6_addr v6addr = ip.ipv6_address(); |
| 39 memcpy(&buf[4], &v6addr, kNATEncodedIPv6AddressSize - 4); | 40 memcpy(&buf[4], &v6addr, kNATEncodedIPv6AddressSize - 4); |
| 40 return kNATEncodedIPv6AddressSize; | 41 return kNATEncodedIPv6AddressSize; |
| 41 } | 42 } |
| 42 return 0U; | 43 return 0U; |
| 43 } | 44 } |
| 44 | 45 |
| 45 // Decodes the remote address from a packet that has been encoded with the nat's | 46 // Decodes the remote address from a packet that has been encoded with the nat's |
| 46 // quasi-STUN format. Returns the length of the address (i.e., the offset into | 47 // quasi-STUN format. Returns the length of the address (i.e., the offset into |
| 47 // data where the original packet starts). | 48 // data where the original packet starts). |
| 48 size_t UnpackAddressFromNAT(const char* buf, size_t buf_size, | 49 size_t UnpackAddressFromNAT(const char* buf, size_t buf_size, |
| 49 SocketAddress* remote_addr) { | 50 SocketAddress* remote_addr) { |
| 50 ASSERT(buf_size >= 8); | 51 RTC_DCHECK(buf_size >= 8); |
| 51 ASSERT(buf[0] == 0); | 52 RTC_DCHECK(buf[0] == 0); |
| 52 int family = buf[1]; | 53 int family = buf[1]; |
| 53 uint16_t port = | 54 uint16_t port = |
| 54 NetworkToHost16(*(reinterpret_cast<const uint16_t*>(&buf[2]))); | 55 NetworkToHost16(*(reinterpret_cast<const uint16_t*>(&buf[2]))); |
| 55 if (family == AF_INET) { | 56 if (family == AF_INET) { |
| 56 const in_addr* v4addr = reinterpret_cast<const in_addr*>(&buf[4]); | 57 const in_addr* v4addr = reinterpret_cast<const in_addr*>(&buf[4]); |
| 57 *remote_addr = SocketAddress(IPAddress(*v4addr), port); | 58 *remote_addr = SocketAddress(IPAddress(*v4addr), port); |
| 58 return kNATEncodedIPv4AddressSize; | 59 return kNATEncodedIPv4AddressSize; |
| 59 } else if (family == AF_INET6) { | 60 } else if (family == AF_INET6) { |
| 60 ASSERT(buf_size >= 20); | 61 RTC_DCHECK(buf_size >= 20); |
| 61 const in6_addr* v6addr = reinterpret_cast<const in6_addr*>(&buf[4]); | 62 const in6_addr* v6addr = reinterpret_cast<const in6_addr*>(&buf[4]); |
| 62 *remote_addr = SocketAddress(IPAddress(*v6addr), port); | 63 *remote_addr = SocketAddress(IPAddress(*v6addr), port); |
| 63 return kNATEncodedIPv6AddressSize; | 64 return kNATEncodedIPv6AddressSize; |
| 64 } | 65 } |
| 65 return 0U; | 66 return 0U; |
| 66 } | 67 } |
| 67 | 68 |
| 68 | 69 |
| 69 // NATSocket | 70 // NATSocket |
| 70 class NATSocket : public AsyncSocket, public sigslot::has_slots<> { | 71 class NATSocket : public AsyncSocket, public sigslot::has_slots<> { |
| (...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 122 } | 123 } |
| 123 | 124 |
| 124 if (result >= 0) { | 125 if (result >= 0) { |
| 125 remote_addr_ = addr; | 126 remote_addr_ = addr; |
| 126 } | 127 } |
| 127 | 128 |
| 128 return result; | 129 return result; |
| 129 } | 130 } |
| 130 | 131 |
| 131 int Send(const void* data, size_t size) override { | 132 int Send(const void* data, size_t size) override { |
| 132 ASSERT(connected_); | 133 RTC_DCHECK(connected_); |
| 133 return SendTo(data, size, remote_addr_); | 134 return SendTo(data, size, remote_addr_); |
| 134 } | 135 } |
| 135 | 136 |
| 136 int SendTo(const void* data, | 137 int SendTo(const void* data, |
| 137 size_t size, | 138 size_t size, |
| 138 const SocketAddress& addr) override { | 139 const SocketAddress& addr) override { |
| 139 ASSERT(!connected_ || addr == remote_addr_); | 140 RTC_DCHECK(!connected_ || addr == remote_addr_); |
| 140 if (server_addr_.IsNil() || type_ == SOCK_STREAM) { | 141 if (server_addr_.IsNil() || type_ == SOCK_STREAM) { |
| 141 return socket_->SendTo(data, size, addr); | 142 return socket_->SendTo(data, size, addr); |
| 142 } | 143 } |
| 143 // This array will be too large for IPv4 packets, but only by 12 bytes. | 144 // This array will be too large for IPv4 packets, but only by 12 bytes. |
| 144 std::unique_ptr<char[]> buf(new char[size + kNATEncodedIPv6AddressSize]); | 145 std::unique_ptr<char[]> buf(new char[size + kNATEncodedIPv6AddressSize]); |
| 145 size_t addrlength = PackAddressForNAT(buf.get(), | 146 size_t addrlength = PackAddressForNAT(buf.get(), |
| 146 size + kNATEncodedIPv6AddressSize, | 147 size + kNATEncodedIPv6AddressSize, |
| 147 addr); | 148 addr); |
| 148 size_t encoded_size = size + addrlength; | 149 size_t encoded_size = size + addrlength; |
| 149 memcpy(buf.get() + addrlength, data, size); | 150 memcpy(buf.get() + addrlength, data, size); |
| 150 int result = socket_->SendTo(buf.get(), encoded_size, server_addr_); | 151 int result = socket_->SendTo(buf.get(), encoded_size, server_addr_); |
| 151 if (result >= 0) { | 152 if (result >= 0) { |
| 152 ASSERT(result == static_cast<int>(encoded_size)); | 153 RTC_DCHECK(result == static_cast<int>(encoded_size)); |
| 153 result = result - static_cast<int>(addrlength); | 154 result = result - static_cast<int>(addrlength); |
| 154 } | 155 } |
| 155 return result; | 156 return result; |
| 156 } | 157 } |
| 157 | 158 |
| 158 int Recv(void* data, size_t size, int64_t* timestamp) override { | 159 int Recv(void* data, size_t size, int64_t* timestamp) override { |
| 159 SocketAddress addr; | 160 SocketAddress addr; |
| 160 return RecvFrom(data, size, &addr, timestamp); | 161 return RecvFrom(data, size, &addr, timestamp); |
| 161 } | 162 } |
| 162 | 163 |
| 163 int RecvFrom(void* data, | 164 int RecvFrom(void* data, |
| 164 size_t size, | 165 size_t size, |
| 165 SocketAddress* out_addr, | 166 SocketAddress* out_addr, |
| 166 int64_t* timestamp) override { | 167 int64_t* timestamp) override { |
| 167 if (server_addr_.IsNil() || type_ == SOCK_STREAM) { | 168 if (server_addr_.IsNil() || type_ == SOCK_STREAM) { |
| 168 return socket_->RecvFrom(data, size, out_addr, timestamp); | 169 return socket_->RecvFrom(data, size, out_addr, timestamp); |
| 169 } | 170 } |
| 170 // Make sure we have enough room to read the requested amount plus the | 171 // Make sure we have enough room to read the requested amount plus the |
| 171 // largest possible header address. | 172 // largest possible header address. |
| 172 SocketAddress remote_addr; | 173 SocketAddress remote_addr; |
| 173 Grow(size + kNATEncodedIPv6AddressSize); | 174 Grow(size + kNATEncodedIPv6AddressSize); |
| 174 | 175 |
| 175 // Read the packet from the socket. | 176 // Read the packet from the socket. |
| 176 int result = socket_->RecvFrom(buf_, size_, &remote_addr, timestamp); | 177 int result = socket_->RecvFrom(buf_, size_, &remote_addr, timestamp); |
| 177 if (result >= 0) { | 178 if (result >= 0) { |
| 178 ASSERT(remote_addr == server_addr_); | 179 RTC_DCHECK(remote_addr == server_addr_); |
| 179 | 180 |
| 180 // TODO: we need better framing so we know how many bytes we can | 181 // TODO: we need better framing so we know how many bytes we can |
| 181 // return before we need to read the next address. For UDP, this will be | 182 // return before we need to read the next address. For UDP, this will be |
| 182 // fine as long as the reader always reads everything in the packet. | 183 // fine as long as the reader always reads everything in the packet. |
| 183 ASSERT((size_t)result < size_); | 184 RTC_DCHECK((size_t)result < size_); |
| 184 | 185 |
| 185 // Decode the wire packet into the actual results. | 186 // Decode the wire packet into the actual results. |
| 186 SocketAddress real_remote_addr; | 187 SocketAddress real_remote_addr; |
| 187 size_t addrlength = UnpackAddressFromNAT(buf_, result, &real_remote_addr); | 188 size_t addrlength = UnpackAddressFromNAT(buf_, result, &real_remote_addr); |
| 188 memcpy(data, buf_ + addrlength, result - addrlength); | 189 memcpy(data, buf_ + addrlength, result - addrlength); |
| 189 | 190 |
| 190 // Make sure this packet should be delivered before returning it. | 191 // Make sure this packet should be delivered before returning it. |
| 191 if (!connected_ || (real_remote_addr == remote_addr_)) { | 192 if (!connected_ || (real_remote_addr == remote_addr_)) { |
| 192 if (out_addr) | 193 if (out_addr) |
| 193 *out_addr = real_remote_addr; | 194 *out_addr = real_remote_addr; |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 228 int EstimateMTU(uint16_t* mtu) override { return socket_->EstimateMTU(mtu); } | 229 int EstimateMTU(uint16_t* mtu) override { return socket_->EstimateMTU(mtu); } |
| 229 int GetOption(Option opt, int* value) override { | 230 int GetOption(Option opt, int* value) override { |
| 230 return socket_->GetOption(opt, value); | 231 return socket_->GetOption(opt, value); |
| 231 } | 232 } |
| 232 int SetOption(Option opt, int value) override { | 233 int SetOption(Option opt, int value) override { |
| 233 return socket_->SetOption(opt, value); | 234 return socket_->SetOption(opt, value); |
| 234 } | 235 } |
| 235 | 236 |
| 236 void OnConnectEvent(AsyncSocket* socket) { | 237 void OnConnectEvent(AsyncSocket* socket) { |
| 237 // If we're NATed, we need to send a message with the real addr to use. | 238 // If we're NATed, we need to send a message with the real addr to use. |
| 238 ASSERT(socket == socket_); | 239 RTC_DCHECK(socket == socket_); |
| 239 if (server_addr_.IsNil()) { | 240 if (server_addr_.IsNil()) { |
| 240 connected_ = true; | 241 connected_ = true; |
| 241 SignalConnectEvent(this); | 242 SignalConnectEvent(this); |
| 242 } else { | 243 } else { |
| 243 SendConnectRequest(); | 244 SendConnectRequest(); |
| 244 } | 245 } |
| 245 } | 246 } |
| 246 void OnReadEvent(AsyncSocket* socket) { | 247 void OnReadEvent(AsyncSocket* socket) { |
| 247 // If we're NATed, we need to process the connect reply. | 248 // If we're NATed, we need to process the connect reply. |
| 248 ASSERT(socket == socket_); | 249 RTC_DCHECK(socket == socket_); |
| 249 if (type_ == SOCK_STREAM && !server_addr_.IsNil() && !connected_) { | 250 if (type_ == SOCK_STREAM && !server_addr_.IsNil() && !connected_) { |
| 250 HandleConnectReply(); | 251 HandleConnectReply(); |
| 251 } else { | 252 } else { |
| 252 SignalReadEvent(this); | 253 SignalReadEvent(this); |
| 253 } | 254 } |
| 254 } | 255 } |
| 255 void OnWriteEvent(AsyncSocket* socket) { | 256 void OnWriteEvent(AsyncSocket* socket) { |
| 256 ASSERT(socket == socket_); | 257 RTC_DCHECK(socket == socket_); |
| 257 SignalWriteEvent(this); | 258 SignalWriteEvent(this); |
| 258 } | 259 } |
| 259 void OnCloseEvent(AsyncSocket* socket, int error) { | 260 void OnCloseEvent(AsyncSocket* socket, int error) { |
| 260 ASSERT(socket == socket_); | 261 RTC_DCHECK(socket == socket_); |
| 261 SignalCloseEvent(this, error); | 262 SignalCloseEvent(this, error); |
| 262 } | 263 } |
| 263 | 264 |
| 264 private: | 265 private: |
| 265 // Makes sure the buffer is at least the given size. | 266 // Makes sure the buffer is at least the given size. |
| 266 void Grow(size_t new_size) { | 267 void Grow(size_t new_size) { |
| 267 if (size_ < new_size) { | 268 if (size_ < new_size) { |
| 268 delete[] buf_; | 269 delete[] buf_; |
| 269 size_ = new_size; | 270 size_ = new_size; |
| 270 buf_ = new char[size_]; | 271 buf_ = new char[size_]; |
| (...skipping 226 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 497 NATSocketServer::Translator* NATSocketServer::TranslatorMap::FindClient( | 498 NATSocketServer::Translator* NATSocketServer::TranslatorMap::FindClient( |
| 498 const SocketAddress& int_ip) { | 499 const SocketAddress& int_ip) { |
| 499 Translator* nat = NULL; | 500 Translator* nat = NULL; |
| 500 for (TranslatorMap::iterator it = begin(); it != end() && !nat; ++it) { | 501 for (TranslatorMap::iterator it = begin(); it != end() && !nat; ++it) { |
| 501 nat = it->second->FindClient(int_ip); | 502 nat = it->second->FindClient(int_ip); |
| 502 } | 503 } |
| 503 return nat; | 504 return nat; |
| 504 } | 505 } |
| 505 | 506 |
| 506 } // namespace rtc | 507 } // namespace rtc |
| OLD | NEW |