First step in providing a UdpTransportChannel.

webrtc/p2p/base/udptransportchannel.cc

+/*
+ *  Copyright 2016 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/p2p/base/udptransportchannel.h"
+
+#include <string>
+
+#include "webrtc/base/asyncudpsocket.h"
+#include "webrtc/base/asyncpacketsocket.h"
+#include "webrtc/base/logging.h"
+#include "webrtc/base/physicalsocketserver.h"
+#include "webrtc/base/socketaddress.h"
+#include "webrtc/base/thread.h"
+#include "webrtc/base/thread_checker.h"
+#include "webrtc/p2p/base/basicpacketsocketfactory.h"
+#include "webrtc/p2p/base/packettransportinterface.h"
+
+namespace cricket {
+
+UdpTransportChannel::UdpTransportChannel(const std::string& transport_name,
+                                         int component)
+    : UdpTransportChannel(transport_name,
+                          component,
+                          rtc::Thread::Current()->socketserver()) {}
+
+UdpTransportChannel::UdpTransportChannel(const std::string& transport_name,
+                                         int component,
+                                         rtc::SocketServer* socket_server)
+    : transport_name_(transport_name),
+      component_(component),
+      socket_server_(socket_server) {
+  UpdateDebugName();
+}
+
+UdpTransportChannel::~UdpTransportChannel() {
+  RTC_DCHECK_RUN_ON(&network_thread_checker_);
+}
+
+void UdpTransportChannel::OnSocketReadPacket(
+    rtc::AsyncPacketSocket* socket,
+    const char* data,
+    size_t len,
+    const rtc::SocketAddress& remote_addr,
+    const rtc::PacketTime& packet_time) {
+  // No thread_checker in high frequency network function.
+  SignalReadPacket(this, data, len, packet_time, 0);
+}
+
+void UdpTransportChannel::OnSocketSentPacket(rtc::AsyncPacketSocket* socket,
+                                             const rtc::SentPacket& packet) {
+  RTC_DCHECK_EQ(socket_.get(), socket);
+  SignalSentPacket(this, packet);
+}
+
+void UdpTransportChannel::UpdateDebugName() {
+  debug_name_ = transport_name_ + " " + std::to_string(component_);
+}
+
+bool UdpTransportChannel::writable() const {
+  return state_ == UDPTRANSPORT_STATE_COMPLETED;
+}
+
+int UdpTransportChannel::SendPacket(const char* data,
+                                    size_t len,
+                                    const rtc::PacketOptions& options,
+                                    int flags) {
+  // No thread_checker in high frequency network function.
+  if (!remote_parameters_) {
+    LOG(LS_WARNING) << "Remote parameters not set.";
+    send_error_ = ENOTCONN;
+    return -1;
+  }
+  const rtc::SocketAddress& remote_addr_ = *remote_parameters_;
+  int result = socket_->SendTo((const void*)data, len, remote_addr_, options);
+  if (result <= 0) {
+    LOG(LS_VERBOSE) << "SendPacket() " << result;
+  }
+  return result;
+}
+
+void UdpTransportChannel::CreateSocket() {
+  RTC_DCHECK_RUN_ON(&network_thread_checker_);
+  if ((state_ == UDPTRANSPORT_STATE_CONNECTING) ||
+      (state_ == UDPTRANSPORT_STATE_COMPLETED)) {
+    LOG(LS_WARNING) << "Local socket already allocated.";
+    return;
+  }
+  static constexpr uint16_t kMaxTries = 100;
+  static constexpr uint16_t kMinPortNumber = 2000;
+  // TODO(johan) provide configuration option for kMinPortNumber.
+  rtc::SocketAddress socket_addr("0.0.0.0", 0);
+  // TODO(johan): Replace BasicPacketSocketFactory by something that honors RFC
+  // 3550 Section 11 port number requirements like
+  // {port_{RTP} is even, port_{RTCP} := port{RTP} + 1}.
+  // This could be a SetSocket(...) method for sockets allocated at a higher
+  // control level.
+  rtc::BasicPacketSocketFactory socket_factory(socket_server_);
+  socket_.reset(socket_factory.CreateUdpSocket(socket_addr, kMinPortNumber,
+                                               kMinPortNumber + kMaxTries));
+  if (socket_) {
+    uint16_t port = socket_->GetLocalAddress().port();
+    LOG(INFO) << "Created UDP socket with port " << port << ".";
+    socket_->SignalReadPacket.connect(this,
+                                      &UdpTransportChannel::OnSocketReadPacket);
+    socket_->SignalSentPacket.connect(this,
+                                      &UdpTransportChannel::OnSocketSentPacket);
+    socket_addr.SetPort(port);
+    local_parameters_ = rtc::Optional<rtc::SocketAddress>(socket_addr);
+  } else {
+    LOG(INFO) << "Local socket allocation failure";
+  }
+  UpdateState();
+  return;
+}
+
+void UdpTransportChannel::UpdateState() {
+  RTC_DCHECK_RUN_ON(&network_thread_checker_);
+  UdpTransportState state;
+  if (local_parameters_ && !socket_) {
+    LOG(INFO) << "local address is set, but socket is null";
+    state = UDPTRANSPORT_STATE_FAILED;
+  } else if (!local_parameters_ && !remote_parameters_) {
+    state = UDPTRANSPORT_STATE_INIT;
+  } else if (local_parameters_ && !remote_parameters_) {
+    state = UDPTRANSPORT_STATE_CONNECTING;
+  } else if (local_parameters_ && remote_parameters_) {
+    state = UDPTRANSPORT_STATE_COMPLETED;
+  } else {
+    state = UDPTRANSPORT_STATE_FAILED;
+  }
+  SetTransportChannelState(state);
+}
+
+void UdpTransportChannel::SetRemoteParameters(const rtc::SocketAddress& addr) {
+  RTC_DCHECK_RUN_ON(&network_thread_checker_);
+  if (!addr.IsComplete()) {
+    LOG(INFO) << "remote address not complete";
+    return;
+  }
+  // TODO(johan) check for ipv4, other settings.
+  remote_parameters_ = rtc::Optional<rtc::SocketAddress>(addr);
+  UpdateState();
+}
+
+const rtc::SocketAddress& UdpTransportChannel::local_parameters() {
+  static const rtc::SocketAddress default_addr;
+  return local_parameters_.value_or(default_addr);
+}
+
+void UdpTransportChannel::SetTransportChannelState(UdpTransportState state) {
+  RTC_DCHECK_RUN_ON(&network_thread_checker_);
+  if (state_ == state) {
+    return;
+  }
+  state_ = state;
+  if (state == UDPTRANSPORT_STATE_COMPLETED) {
+    SignalWritableState(this);
+    SignalReadyToSend(this);
+  }
+}
+}  // namespace cricket