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

Unified Diff: webrtc/base/natsocketfactory.cc

Issue 2877023002: Move webrtc/{base => rtc_base} (Closed)
Patch Set: update presubmit.py and DEPS include rules Created 3 years, 6 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webrtc/base/natsocketfactory.h ('k') | webrtc/base/nattypes.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « webrtc/base/natsocketfactory.h ('k') | webrtc/base/nattypes.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698