| Index: webrtc/base/natsocketfactory.cc
|
| diff --git a/webrtc/base/natsocketfactory.cc b/webrtc/base/natsocketfactory.cc
|
| deleted file mode 100644
|
| index 95e64872199db3bf059614fcb8f8d7733e571e31..0000000000000000000000000000000000000000
|
| --- a/webrtc/base/natsocketfactory.cc
|
| +++ /dev/null
|
| @@ -1,530 +0,0 @@
|
| -/*
|
| - * Copyright 2004 The WebRTC Project Authors. All rights reserved.
|
| - *
|
| - * Use of this source code is governed by a BSD-style license
|
| - * that can be found in the LICENSE file in the root of the source
|
| - * tree. An additional intellectual property rights grant can be found
|
| - * in the file PATENTS. All contributing project authors may
|
| - * be found in the AUTHORS file in the root of the source tree.
|
| - */
|
| -
|
| -#include "webrtc/base/natsocketfactory.h"
|
| -
|
| -#include "webrtc/base/arraysize.h"
|
| -#include "webrtc/base/checks.h"
|
| -#include "webrtc/base/logging.h"
|
| -#include "webrtc/base/natserver.h"
|
| -#include "webrtc/base/virtualsocketserver.h"
|
| -
|
| -namespace rtc {
|
| -
|
| -// Packs the given socketaddress into the buffer in buf, in the quasi-STUN
|
| -// format that the natserver uses.
|
| -// Returns 0 if an invalid address is passed.
|
| -size_t PackAddressForNAT(char* buf, size_t buf_size,
|
| - const SocketAddress& remote_addr) {
|
| - const IPAddress& ip = remote_addr.ipaddr();
|
| - int family = ip.family();
|
| - buf[0] = 0;
|
| - buf[1] = family;
|
| - // Writes the port.
|
| - *(reinterpret_cast<uint16_t*>(&buf[2])) = HostToNetwork16(remote_addr.port());
|
| - if (family == AF_INET) {
|
| - RTC_DCHECK(buf_size >= kNATEncodedIPv4AddressSize);
|
| - in_addr v4addr = ip.ipv4_address();
|
| - memcpy(&buf[4], &v4addr, kNATEncodedIPv4AddressSize - 4);
|
| - return kNATEncodedIPv4AddressSize;
|
| - } else if (family == AF_INET6) {
|
| - RTC_DCHECK(buf_size >= kNATEncodedIPv6AddressSize);
|
| - in6_addr v6addr = ip.ipv6_address();
|
| - memcpy(&buf[4], &v6addr, kNATEncodedIPv6AddressSize - 4);
|
| - return kNATEncodedIPv6AddressSize;
|
| - }
|
| - return 0U;
|
| -}
|
| -
|
| -// Decodes the remote address from a packet that has been encoded with the nat's
|
| -// quasi-STUN format. Returns the length of the address (i.e., the offset into
|
| -// data where the original packet starts).
|
| -size_t UnpackAddressFromNAT(const char* buf, size_t buf_size,
|
| - SocketAddress* remote_addr) {
|
| - RTC_DCHECK(buf_size >= 8);
|
| - RTC_DCHECK(buf[0] == 0);
|
| - int family = buf[1];
|
| - uint16_t port =
|
| - NetworkToHost16(*(reinterpret_cast<const uint16_t*>(&buf[2])));
|
| - if (family == AF_INET) {
|
| - const in_addr* v4addr = reinterpret_cast<const in_addr*>(&buf[4]);
|
| - *remote_addr = SocketAddress(IPAddress(*v4addr), port);
|
| - return kNATEncodedIPv4AddressSize;
|
| - } else if (family == AF_INET6) {
|
| - RTC_DCHECK(buf_size >= 20);
|
| - const in6_addr* v6addr = reinterpret_cast<const in6_addr*>(&buf[4]);
|
| - *remote_addr = SocketAddress(IPAddress(*v6addr), port);
|
| - return kNATEncodedIPv6AddressSize;
|
| - }
|
| - return 0U;
|
| -}
|
| -
|
| -
|
| -// NATSocket
|
| -class NATSocket : public AsyncSocket, public sigslot::has_slots<> {
|
| - public:
|
| - explicit NATSocket(NATInternalSocketFactory* sf, int family, int type)
|
| - : sf_(sf),
|
| - family_(family),
|
| - type_(type),
|
| - connected_(false),
|
| - socket_(nullptr),
|
| - buf_(nullptr),
|
| - size_(0) {}
|
| -
|
| - ~NATSocket() override {
|
| - delete socket_;
|
| - delete[] buf_;
|
| - }
|
| -
|
| - SocketAddress GetLocalAddress() const override {
|
| - return (socket_) ? socket_->GetLocalAddress() : SocketAddress();
|
| - }
|
| -
|
| - SocketAddress GetRemoteAddress() const override {
|
| - return remote_addr_; // will be NIL if not connected
|
| - }
|
| -
|
| - int Bind(const SocketAddress& addr) override {
|
| - if (socket_) { // already bound, bubble up error
|
| - return -1;
|
| - }
|
| -
|
| - return BindInternal(addr);
|
| - }
|
| -
|
| - int Connect(const SocketAddress& addr) override {
|
| - int result = 0;
|
| - // If we're not already bound (meaning |socket_| is null), bind to ANY
|
| - // address.
|
| - if (!socket_) {
|
| - result = BindInternal(SocketAddress(GetAnyIP(family_), 0));
|
| - if (result < 0) {
|
| - return result;
|
| - }
|
| - }
|
| -
|
| - if (type_ == SOCK_STREAM) {
|
| - result = socket_->Connect(server_addr_.IsNil() ? addr : server_addr_);
|
| - } else {
|
| - connected_ = true;
|
| - }
|
| -
|
| - if (result >= 0) {
|
| - remote_addr_ = addr;
|
| - }
|
| -
|
| - return result;
|
| - }
|
| -
|
| - int Send(const void* data, size_t size) override {
|
| - RTC_DCHECK(connected_);
|
| - return SendTo(data, size, remote_addr_);
|
| - }
|
| -
|
| - int SendTo(const void* data,
|
| - size_t size,
|
| - const SocketAddress& addr) override {
|
| - RTC_DCHECK(!connected_ || addr == remote_addr_);
|
| - if (server_addr_.IsNil() || type_ == SOCK_STREAM) {
|
| - return socket_->SendTo(data, size, addr);
|
| - }
|
| - // This array will be too large for IPv4 packets, but only by 12 bytes.
|
| - std::unique_ptr<char[]> buf(new char[size + kNATEncodedIPv6AddressSize]);
|
| - size_t addrlength = PackAddressForNAT(buf.get(),
|
| - size + kNATEncodedIPv6AddressSize,
|
| - addr);
|
| - size_t encoded_size = size + addrlength;
|
| - memcpy(buf.get() + addrlength, data, size);
|
| - int result = socket_->SendTo(buf.get(), encoded_size, server_addr_);
|
| - if (result >= 0) {
|
| - RTC_DCHECK(result == static_cast<int>(encoded_size));
|
| - result = result - static_cast<int>(addrlength);
|
| - }
|
| - return result;
|
| - }
|
| -
|
| - int Recv(void* data, size_t size, int64_t* timestamp) override {
|
| - SocketAddress addr;
|
| - return RecvFrom(data, size, &addr, timestamp);
|
| - }
|
| -
|
| - int RecvFrom(void* data,
|
| - size_t size,
|
| - SocketAddress* out_addr,
|
| - int64_t* timestamp) override {
|
| - if (server_addr_.IsNil() || type_ == SOCK_STREAM) {
|
| - return socket_->RecvFrom(data, size, out_addr, timestamp);
|
| - }
|
| - // Make sure we have enough room to read the requested amount plus the
|
| - // largest possible header address.
|
| - SocketAddress remote_addr;
|
| - Grow(size + kNATEncodedIPv6AddressSize);
|
| -
|
| - // Read the packet from the socket.
|
| - int result = socket_->RecvFrom(buf_, size_, &remote_addr, timestamp);
|
| - if (result >= 0) {
|
| - RTC_DCHECK(remote_addr == server_addr_);
|
| -
|
| - // TODO: we need better framing so we know how many bytes we can
|
| - // return before we need to read the next address. For UDP, this will be
|
| - // fine as long as the reader always reads everything in the packet.
|
| - RTC_DCHECK((size_t)result < size_);
|
| -
|
| - // Decode the wire packet into the actual results.
|
| - SocketAddress real_remote_addr;
|
| - size_t addrlength = UnpackAddressFromNAT(buf_, result, &real_remote_addr);
|
| - memcpy(data, buf_ + addrlength, result - addrlength);
|
| -
|
| - // Make sure this packet should be delivered before returning it.
|
| - if (!connected_ || (real_remote_addr == remote_addr_)) {
|
| - if (out_addr)
|
| - *out_addr = real_remote_addr;
|
| - result = result - static_cast<int>(addrlength);
|
| - } else {
|
| - LOG(LS_ERROR) << "Dropping packet from unknown remote address: "
|
| - << real_remote_addr.ToString();
|
| - result = 0; // Tell the caller we didn't read anything
|
| - }
|
| - }
|
| -
|
| - return result;
|
| - }
|
| -
|
| - int Close() override {
|
| - int result = 0;
|
| - if (socket_) {
|
| - result = socket_->Close();
|
| - if (result >= 0) {
|
| - connected_ = false;
|
| - remote_addr_ = SocketAddress();
|
| - delete socket_;
|
| - socket_ = nullptr;
|
| - }
|
| - }
|
| - return result;
|
| - }
|
| -
|
| - int Listen(int backlog) override { return socket_->Listen(backlog); }
|
| - AsyncSocket* Accept(SocketAddress* paddr) override {
|
| - return socket_->Accept(paddr);
|
| - }
|
| - int GetError() const override {
|
| - return socket_ ? socket_->GetError() : error_;
|
| - }
|
| - void SetError(int error) override {
|
| - if (socket_) {
|
| - socket_->SetError(error);
|
| - } else {
|
| - error_ = error;
|
| - }
|
| - }
|
| - ConnState GetState() const override {
|
| - return connected_ ? CS_CONNECTED : CS_CLOSED;
|
| - }
|
| - int GetOption(Option opt, int* value) override {
|
| - return socket_->GetOption(opt, value);
|
| - }
|
| - int SetOption(Option opt, int value) override {
|
| - return socket_->SetOption(opt, value);
|
| - }
|
| -
|
| - void OnConnectEvent(AsyncSocket* socket) {
|
| - // If we're NATed, we need to send a message with the real addr to use.
|
| - RTC_DCHECK(socket == socket_);
|
| - if (server_addr_.IsNil()) {
|
| - connected_ = true;
|
| - SignalConnectEvent(this);
|
| - } else {
|
| - SendConnectRequest();
|
| - }
|
| - }
|
| - void OnReadEvent(AsyncSocket* socket) {
|
| - // If we're NATed, we need to process the connect reply.
|
| - RTC_DCHECK(socket == socket_);
|
| - if (type_ == SOCK_STREAM && !server_addr_.IsNil() && !connected_) {
|
| - HandleConnectReply();
|
| - } else {
|
| - SignalReadEvent(this);
|
| - }
|
| - }
|
| - void OnWriteEvent(AsyncSocket* socket) {
|
| - RTC_DCHECK(socket == socket_);
|
| - SignalWriteEvent(this);
|
| - }
|
| - void OnCloseEvent(AsyncSocket* socket, int error) {
|
| - RTC_DCHECK(socket == socket_);
|
| - SignalCloseEvent(this, error);
|
| - }
|
| -
|
| - private:
|
| - int BindInternal(const SocketAddress& addr) {
|
| - RTC_DCHECK(!socket_);
|
| -
|
| - int result;
|
| - socket_ = sf_->CreateInternalSocket(family_, type_, addr, &server_addr_);
|
| - result = (socket_) ? socket_->Bind(addr) : -1;
|
| - if (result >= 0) {
|
| - socket_->SignalConnectEvent.connect(this, &NATSocket::OnConnectEvent);
|
| - socket_->SignalReadEvent.connect(this, &NATSocket::OnReadEvent);
|
| - socket_->SignalWriteEvent.connect(this, &NATSocket::OnWriteEvent);
|
| - socket_->SignalCloseEvent.connect(this, &NATSocket::OnCloseEvent);
|
| - } else {
|
| - server_addr_.Clear();
|
| - delete socket_;
|
| - socket_ = nullptr;
|
| - }
|
| -
|
| - return result;
|
| - }
|
| -
|
| - // Makes sure the buffer is at least the given size.
|
| - void Grow(size_t new_size) {
|
| - if (size_ < new_size) {
|
| - delete[] buf_;
|
| - size_ = new_size;
|
| - buf_ = new char[size_];
|
| - }
|
| - }
|
| -
|
| - // Sends the destination address to the server to tell it to connect.
|
| - void SendConnectRequest() {
|
| - char buf[kNATEncodedIPv6AddressSize];
|
| - size_t length = PackAddressForNAT(buf, arraysize(buf), remote_addr_);
|
| - socket_->Send(buf, length);
|
| - }
|
| -
|
| - // Handles the byte sent back from the server and fires the appropriate event.
|
| - void HandleConnectReply() {
|
| - char code;
|
| - socket_->Recv(&code, sizeof(code), nullptr);
|
| - if (code == 0) {
|
| - connected_ = true;
|
| - SignalConnectEvent(this);
|
| - } else {
|
| - Close();
|
| - SignalCloseEvent(this, code);
|
| - }
|
| - }
|
| -
|
| - NATInternalSocketFactory* sf_;
|
| - int family_;
|
| - int type_;
|
| - bool connected_;
|
| - SocketAddress remote_addr_;
|
| - SocketAddress server_addr_; // address of the NAT server
|
| - AsyncSocket* socket_;
|
| - // Need to hold error in case it occurs before the socket is created.
|
| - int error_ = 0;
|
| - char* buf_;
|
| - size_t size_;
|
| -};
|
| -
|
| -// NATSocketFactory
|
| -NATSocketFactory::NATSocketFactory(SocketFactory* factory,
|
| - const SocketAddress& nat_udp_addr,
|
| - const SocketAddress& nat_tcp_addr)
|
| - : factory_(factory), nat_udp_addr_(nat_udp_addr),
|
| - nat_tcp_addr_(nat_tcp_addr) {
|
| -}
|
| -
|
| -Socket* NATSocketFactory::CreateSocket(int type) {
|
| - return CreateSocket(AF_INET, type);
|
| -}
|
| -
|
| -Socket* NATSocketFactory::CreateSocket(int family, int type) {
|
| - return new NATSocket(this, family, type);
|
| -}
|
| -
|
| -AsyncSocket* NATSocketFactory::CreateAsyncSocket(int type) {
|
| - return CreateAsyncSocket(AF_INET, type);
|
| -}
|
| -
|
| -AsyncSocket* NATSocketFactory::CreateAsyncSocket(int family, int type) {
|
| - return new NATSocket(this, family, type);
|
| -}
|
| -
|
| -AsyncSocket* NATSocketFactory::CreateInternalSocket(int family, int type,
|
| - const SocketAddress& local_addr, SocketAddress* nat_addr) {
|
| - if (type == SOCK_STREAM) {
|
| - *nat_addr = nat_tcp_addr_;
|
| - } else {
|
| - *nat_addr = nat_udp_addr_;
|
| - }
|
| - return factory_->CreateAsyncSocket(family, type);
|
| -}
|
| -
|
| -// NATSocketServer
|
| -NATSocketServer::NATSocketServer(SocketServer* server)
|
| - : server_(server), msg_queue_(nullptr) {}
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::GetTranslator(
|
| - const SocketAddress& ext_ip) {
|
| - return nats_.Get(ext_ip);
|
| -}
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::AddTranslator(
|
| - const SocketAddress& ext_ip, const SocketAddress& int_ip, NATType type) {
|
| - // Fail if a translator already exists with this extternal address.
|
| - if (nats_.Get(ext_ip))
|
| - return nullptr;
|
| -
|
| - return nats_.Add(ext_ip, new Translator(this, type, int_ip, server_, ext_ip));
|
| -}
|
| -
|
| -void NATSocketServer::RemoveTranslator(
|
| - const SocketAddress& ext_ip) {
|
| - nats_.Remove(ext_ip);
|
| -}
|
| -
|
| -Socket* NATSocketServer::CreateSocket(int type) {
|
| - return CreateSocket(AF_INET, type);
|
| -}
|
| -
|
| -Socket* NATSocketServer::CreateSocket(int family, int type) {
|
| - return new NATSocket(this, family, type);
|
| -}
|
| -
|
| -AsyncSocket* NATSocketServer::CreateAsyncSocket(int type) {
|
| - return CreateAsyncSocket(AF_INET, type);
|
| -}
|
| -
|
| -AsyncSocket* NATSocketServer::CreateAsyncSocket(int family, int type) {
|
| - return new NATSocket(this, family, type);
|
| -}
|
| -
|
| -void NATSocketServer::SetMessageQueue(MessageQueue* queue) {
|
| - msg_queue_ = queue;
|
| - server_->SetMessageQueue(queue);
|
| -}
|
| -
|
| -bool NATSocketServer::Wait(int cms, bool process_io) {
|
| - return server_->Wait(cms, process_io);
|
| -}
|
| -
|
| -void NATSocketServer::WakeUp() {
|
| - server_->WakeUp();
|
| -}
|
| -
|
| -AsyncSocket* NATSocketServer::CreateInternalSocket(int family, int type,
|
| - const SocketAddress& local_addr, SocketAddress* nat_addr) {
|
| - AsyncSocket* socket = nullptr;
|
| - Translator* nat = nats_.FindClient(local_addr);
|
| - if (nat) {
|
| - socket = nat->internal_factory()->CreateAsyncSocket(family, type);
|
| - *nat_addr = (type == SOCK_STREAM) ?
|
| - nat->internal_tcp_address() : nat->internal_udp_address();
|
| - } else {
|
| - socket = server_->CreateAsyncSocket(family, type);
|
| - }
|
| - return socket;
|
| -}
|
| -
|
| -// NATSocketServer::Translator
|
| -NATSocketServer::Translator::Translator(
|
| - NATSocketServer* server, NATType type, const SocketAddress& int_ip,
|
| - SocketFactory* ext_factory, const SocketAddress& ext_ip)
|
| - : server_(server) {
|
| - // Create a new private network, and a NATServer running on the private
|
| - // network that bridges to the external network. Also tell the private
|
| - // network to use the same message queue as us.
|
| - VirtualSocketServer* internal_server = new VirtualSocketServer();
|
| - internal_server->SetMessageQueue(server_->queue());
|
| - internal_factory_.reset(internal_server);
|
| - nat_server_.reset(new NATServer(type, internal_server, int_ip, int_ip,
|
| - ext_factory, ext_ip));
|
| -}
|
| -
|
| -NATSocketServer::Translator::~Translator() = default;
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::Translator::GetTranslator(
|
| - const SocketAddress& ext_ip) {
|
| - return nats_.Get(ext_ip);
|
| -}
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::Translator::AddTranslator(
|
| - const SocketAddress& ext_ip, const SocketAddress& int_ip, NATType type) {
|
| - // Fail if a translator already exists with this extternal address.
|
| - if (nats_.Get(ext_ip))
|
| - return nullptr;
|
| -
|
| - AddClient(ext_ip);
|
| - return nats_.Add(ext_ip,
|
| - new Translator(server_, type, int_ip, server_, ext_ip));
|
| -}
|
| -void NATSocketServer::Translator::RemoveTranslator(
|
| - const SocketAddress& ext_ip) {
|
| - nats_.Remove(ext_ip);
|
| - RemoveClient(ext_ip);
|
| -}
|
| -
|
| -bool NATSocketServer::Translator::AddClient(
|
| - const SocketAddress& int_ip) {
|
| - // Fail if a client already exists with this internal address.
|
| - if (clients_.find(int_ip) != clients_.end())
|
| - return false;
|
| -
|
| - clients_.insert(int_ip);
|
| - return true;
|
| -}
|
| -
|
| -void NATSocketServer::Translator::RemoveClient(
|
| - const SocketAddress& int_ip) {
|
| - std::set<SocketAddress>::iterator it = clients_.find(int_ip);
|
| - if (it != clients_.end()) {
|
| - clients_.erase(it);
|
| - }
|
| -}
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::Translator::FindClient(
|
| - const SocketAddress& int_ip) {
|
| - // See if we have the requested IP, or any of our children do.
|
| - return (clients_.find(int_ip) != clients_.end()) ?
|
| - this : nats_.FindClient(int_ip);
|
| -}
|
| -
|
| -// NATSocketServer::TranslatorMap
|
| -NATSocketServer::TranslatorMap::~TranslatorMap() {
|
| - for (TranslatorMap::iterator it = begin(); it != end(); ++it) {
|
| - delete it->second;
|
| - }
|
| -}
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::TranslatorMap::Get(
|
| - const SocketAddress& ext_ip) {
|
| - TranslatorMap::iterator it = find(ext_ip);
|
| - return (it != end()) ? it->second : nullptr;
|
| -}
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::TranslatorMap::Add(
|
| - const SocketAddress& ext_ip, Translator* nat) {
|
| - (*this)[ext_ip] = nat;
|
| - return nat;
|
| -}
|
| -
|
| -void NATSocketServer::TranslatorMap::Remove(
|
| - const SocketAddress& ext_ip) {
|
| - TranslatorMap::iterator it = find(ext_ip);
|
| - if (it != end()) {
|
| - delete it->second;
|
| - erase(it);
|
| - }
|
| -}
|
| -
|
| -NATSocketServer::Translator* NATSocketServer::TranslatorMap::FindClient(
|
| - const SocketAddress& int_ip) {
|
| - Translator* nat = nullptr;
|
| - for (TranslatorMap::iterator it = begin(); it != end() && !nat; ++it) {
|
| - nat = it->second->FindClient(int_ip);
|
| - }
|
| - return nat;
|
| -}
|
| -
|
| -} // namespace rtc
|
|
|