TransportController refactoring

webrtc/p2p/base/faketransportcontroller.h

 /*
  *  Copyright 2009 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.
  */
 
-#ifndef WEBRTC_P2P_BASE_FAKESESSION_H_
-#define WEBRTC_P2P_BASE_FAKESESSION_H_
+#ifndef WEBRTC_P2P_BASE_FAKETRANSPORTCONTROLLER_H_
+#define WEBRTC_P2P_BASE_FAKETRANSPORTCONTROLLER_H_
 
 #include <map>
 #include <string>
 #include <vector>
 
-#include "webrtc/p2p/base/session.h"
 #include "webrtc/p2p/base/transport.h"
 #include "webrtc/p2p/base/transportchannel.h"
+#include "webrtc/p2p/base/transportcontroller.h"
 #include "webrtc/p2p/base/transportchannelimpl.h"
+#include "webrtc/base/bind.h"
 #include "webrtc/base/buffer.h"
 #include "webrtc/base/fakesslidentity.h"
 #include "webrtc/base/messagequeue.h"
 #include "webrtc/base/sigslot.h"
 #include "webrtc/base/sslfingerprint.h"
+#include "webrtc/base/thread.h"
 
 namespace cricket {
 
 class FakeTransport;
 
 struct PacketMessageData : public rtc::MessageData {
   PacketMessageData(const char* data, size_t len) : packet(data, len) {
   }
   rtc::Buffer packet;
 };
 
 // Fake transport channel class, which can be passed to anything that needs a
 // transport channel. Can be informed of another FakeTransportChannel via
 // SetDestination.
 class FakeTransportChannel : public TransportChannelImpl,
                              public rtc::MessageHandler {
  public:
   explicit FakeTransportChannel(Transport* transport,
                                 const std::string& content_name,
                                 int component)
       : TransportChannelImpl(content_name, component),
         transport_(transport),
-        dest_(NULL),
+        dest_(nullptr),
         state_(STATE_INIT),
         async_(false),
-        identity_(NULL),
+        identity_(nullptr),
         do_dtls_(false),
         role_(ICEROLE_UNKNOWN),
         tiebreaker_(0),
         ice_proto_(ICEPROTO_HYBRID),
         remote_ice_mode_(ICEMODE_FULL),
-        dtls_fingerprint_("", NULL, 0),
+        dtls_fingerprint_("", nullptr, 0),
         ssl_role_(rtc::SSL_CLIENT),
-        connection_count_(0) {
+        connection_count_(0),
+        candidate_gathering_state_(Transport::kGatheringNew) {
   }
   ~FakeTransportChannel() {
     Reset();
   }
 
   uint64 IceTiebreaker() const { return tiebreaker_; }
   TransportProtocol protocol() const { return ice_proto_; }
   IceMode remote_ice_mode() const { return remote_ice_mode_; }
   const std::string& ice_ufrag() const { return ice_ufrag_; }
   const std::string& ice_pwd() const { return ice_pwd_; }
@@ skipping 55 matching lines @@
   virtual bool GetSslRole(rtc::SSLRole* role) const {
     *role = ssl_role_;
     return true;
   }
 
   virtual void Connect() {
     if (state_ == STATE_INIT) {
       state_ = STATE_CONNECTING;
     }
   }
+
+  virtual Transport::CandidateGatheringState candidate_gathering_state() const {
+    return candidate_gathering_state_;
+  }
+
   virtual void Reset() {
     if (state_ != STATE_INIT) {
       state_ = STATE_INIT;
       if (dest_) {
         dest_->state_ = STATE_INIT;
-        dest_->dest_ = NULL;
-        dest_ = NULL;
+        dest_->dest_ = nullptr;
+        dest_ = nullptr;
       }
     }
   }
 
   void SetWritable(bool writable) {
     set_writable(writable);
   }
 
   void SetDestination(FakeTransportChannel* dest) {
     if (state_ == STATE_CONNECTING && dest) {
       // This simulates the delivery of candidates.
       dest_ = dest;
       dest_->dest_ = this;
       if (identity_ && dest_->identity_) {
         do_dtls_ = true;
         dest_->do_dtls_ = true;
         NegotiateSrtpCiphers();
       }
       state_ = STATE_CONNECTED;
       dest_->state_ = STATE_CONNECTED;
       set_writable(true);
       dest_->set_writable(true);
     } else if (state_ == STATE_CONNECTED && !dest) {
       // Simulates loss of connectivity, by asymmetrically forgetting dest_.
-      dest_ = NULL;
+      dest_ = nullptr;
       state_ = STATE_CONNECTING;
       set_writable(false);
     }
   }
 
   void SetConnectionCount(size_t connection_count) {
     size_t old_connection_count = connection_count_;
     connection_count_ = connection_count;
     if (connection_count_ < old_connection_count)
       SignalConnectionRemoved(this);
   }
 
+  void SetCandidatesAllocationDone() {
+    if (candidate_gathering_state_ != Transport::kGatheringDone) {
+      candidate_gathering_state_ = Transport::kGatheringDone;
+      SignalCandidatesAllocationDone(this);
+    }
+  }
+
   void SetReceiving(bool receiving) {
     set_receiving(receiving);
   }
 
   void SetReceivingTimeout(int timeout) override {}
 
   virtual int SendPacket(const char* data, size_t len,
                          const rtc::PacketOptions& options, int flags) {
     if (state_ != STATE_CONNECTED) {
       return -1;
@@ skipping 14 matching lines @@
   virtual int SetOption(rtc::Socket::Option opt, int value) {
     return true;
   }
   virtual bool GetOption(rtc::Socket::Option opt, int* value) {
     return true;
   }
   virtual int GetError() {
     return 0;
   }
 
-  virtual void OnSignalingReady() {
-  }
   virtual void OnCandidate(const Candidate& candidate) {
   }
 
   virtual void OnMessage(rtc::Message* msg) {
     PacketMessageData* data = static_cast<PacketMessageData*>(
         msg->pdata);
     dest_->SignalReadPacket(dest_, data->packet.data<char>(),
                             data->packet.size(), rtc::CreatePacketTime(0), 0);
     delete data;
   }
@@ skipping 96 matching lines @@
   uint64 tiebreaker_;
   IceProtocolType ice_proto_;
   std::string ice_ufrag_;
   std::string ice_pwd_;
   std::string remote_ice_ufrag_;
   std::string remote_ice_pwd_;
   IceMode remote_ice_mode_;
   rtc::SSLFingerprint dtls_fingerprint_;
   rtc::SSLRole ssl_role_;
   size_t connection_count_;
+  Transport::CandidateGatheringState candidate_gathering_state_;
 };
 
 // Fake transport class, which can be passed to anything that needs a Transport.
 // Can be informed of another FakeTransport via SetDestination (low-tech way
 // of doing candidates)
 class FakeTransport : public Transport {
  public:
   typedef std::map<int, FakeTransportChannel*> ChannelMap;
-  FakeTransport(rtc::Thread* signaling_thread,
-                rtc::Thread* worker_thread,
-                const std::string& content_name,
-                PortAllocator* alllocator = NULL)
-      : Transport(signaling_thread, worker_thread,
-                  content_name, "test_type", NULL),
-      dest_(NULL),
+  FakeTransport(const std::string& content_name,
+                PortAllocator* alllocator = nullptr)
+      : Transport(content_name, "test_type", nullptr),
+      dest_(nullptr),
       async_(false),
-      identity_(NULL) {
+      identity_(nullptr) {
   }
+
   ~FakeTransport() {
     DestroyAllChannels();
   }
 
   const ChannelMap& channels() const { return channels_; }
 
   void SetAsync(bool async) { async_ = async; }
   void SetDestination(FakeTransport* dest) {
     dest_ = dest;
-    for (ChannelMap::iterator it = channels_.begin(); it != channels_.end();
-         ++it) {
-      it->second->SetLocalIdentity(identity_);
-      SetChannelDestination(it->first, it->second);
+    for (const auto& kv : channels_) {
+      kv.second->SetLocalIdentity(identity_);
+      SetChannelDestination(kv.first, kv.second);
     }
   }
 
   void SetWritable(bool writable) {
-    for (ChannelMap::iterator it = channels_.begin(); it != channels_.end();
-         ++it) {
-      it->second->SetWritable(writable);
+    for (const auto& kv : channels_) {
+      kv.second->SetWritable(writable);
     }
   }
 
-  void set_identity(rtc::SSLIdentity* identity) {
+  virtual void SetIdentity(rtc::SSLIdentity* identity) {
     identity_ = identity;
   }
+  virtual bool GetIdentity(rtc::SSLIdentity** identity) {
+    if (!identity_)
+      return false;
+
+    *identity = identity_->GetReference();
+    return true;
+  }
 
   using Transport::local_description;
   using Transport::remote_description;
 
  protected:
   virtual TransportChannelImpl* CreateTransportChannel(int component) {
     if (channels_.find(component) != channels_.end()) {
-      return NULL;
+      return nullptr;
     }
     FakeTransportChannel* channel =
         new FakeTransportChannel(this, content_name(), component);
     channel->SetAsync(async_);
     SetChannelDestination(component, channel);
     channels_[component] = channel;
     return channel;
   }
+
   virtual void DestroyTransportChannel(TransportChannelImpl* channel) {
     channels_.erase(channel->component());
     delete channel;
   }
-  virtual void SetIdentity_w(rtc::SSLIdentity* identity) {
-    identity_ = identity;
-  }
-  virtual bool GetIdentity_w(rtc::SSLIdentity** identity) {
-    if (!identity_)
-      return false;
-
-    *identity = identity_->GetReference();
-    return true;
-  }
 
  private:
   FakeTransportChannel* GetFakeChannel(int component) {
-    ChannelMap::iterator it = channels_.find(component);
-    return (it != channels_.end()) ? it->second : NULL;
+    auto it = channels_.find(component);
+    return (it != channels_.end()) ? it->second : nullptr;
   }
+
   void SetChannelDestination(int component,
                              FakeTransportChannel* channel) {
-    FakeTransportChannel* dest_channel = NULL;
+    FakeTransportChannel* dest_channel = nullptr;
     if (dest_) {
       dest_channel = dest_->GetFakeChannel(component);
       if (dest_channel) {
         dest_channel->SetLocalIdentity(dest_->identity_);
       }
     }
     channel->SetDestination(dest_channel);
   }
 
   // Note, this is distinct from the Channel map owned by Transport.
   // This map just tracks the FakeTransportChannels created by this class.
+  // It's mainly needed so that we can access a FakeTransportChannel directly,
+  // even if wrapped by a DtlsTransportChannelWrapper.
   ChannelMap channels_;
   FakeTransport* dest_;
   bool async_;
   rtc::SSLIdentity* identity_;
 };
 
-// Fake session class, which can be passed into a BaseChannel object for
-// test purposes. Can be connected to other FakeSessions via Connect().
-class FakeSession : public BaseSession {
+// Fake TransportController class, which can be passed into a BaseChannel object
+// for test purposes. Can be connected to other FakeTransportControllers via
+// Connect().
+class FakeTransportController : public TransportController {
  public:
-  explicit FakeSession()
-      : BaseSession(rtc::Thread::Current(),
-                    rtc::Thread::Current(),
-                    NULL, "", "", true),
-      fail_create_channel_(false) {
-  }
-  explicit FakeSession(bool initiator)
-      : BaseSession(rtc::Thread::Current(),
-                    rtc::Thread::Current(),
-                    NULL, "", "", initiator),
-      fail_create_channel_(false) {
-  }
-  FakeSession(rtc::Thread* worker_thread, bool initiator)
-      : BaseSession(rtc::Thread::Current(),
-                    worker_thread,
-                    NULL, "", "", initiator),
-      fail_create_channel_(false) {
+  FakeTransportController()
+      : TransportController(rtc::Thread::Current(),
+                            rtc::Thread::Current(),
+                            nullptr),
+        fail_create_channel_(false) {
   }
 
-  FakeTransport* GetTransport(const std::string& content_name) {
-    return static_cast<FakeTransport*>(
-        BaseSession::GetTransport(content_name));
+  explicit FakeTransportController(bool initiator)
+      : TransportController(rtc::Thread::Current(),
+                            rtc::Thread::Current(),
+                            nullptr),
+        fail_create_channel_(false) {
+    SetIceRole(initiator ? ICEROLE_CONTROLLING : ICEROLE_CONTROLLED);
   }
 
-  void Connect(FakeSession* dest) {
-    // Simulate the exchange of candidates.
-    CompleteNegotiation();
-    dest->CompleteNegotiation();
-    for (TransportMap::const_iterator it = transport_proxies().begin();
-        it != transport_proxies().end(); ++it) {
-      static_cast<FakeTransport*>(it->second->impl())->SetDestination(
-          dest->GetTransport(it->first));
-    }
+  FakeTransportController(rtc::Thread* worker_thread, bool initiator)
+      : TransportController(rtc::Thread::Current(),
+                            worker_thread,
+                            nullptr),
+        fail_create_channel_(false) {
+    SetIceRole(initiator ? ICEROLE_CONTROLLING : ICEROLE_CONTROLLED);
   }
 
-  virtual TransportChannel* CreateChannel(
-      const std::string& content_name,
+  FakeTransport* GetTransport_w(const std::string& transport_name) {
+    return static_cast<FakeTransport*>(
+        TransportController::GetTransport_w(transport_name));
+  }
+
+  void Connect(FakeTransportController* dest) {
+    worker_thread()->Invoke<void>(rtc::Bind(
+        &FakeTransportController::Connect_w, this, dest));
+  }
+
+  virtual TransportChannel* CreateTransportChannel_w(
+      const std::string& transport_name,
       int component) {
     if (fail_create_channel_) {
-      return NULL;
+      return nullptr;
     }
-    return BaseSession::CreateChannel(content_name, component);
+    return TransportController::CreateTransportChannel_w(transport_name,
+                                                         component);
   }
 
   void set_fail_channel_creation(bool fail_channel_creation) {
     fail_create_channel_ = fail_channel_creation;
   }
 
-  // TODO: Hoist this into Session when we re-work the Session code.
-  void set_ssl_identity(rtc::SSLIdentity* identity) {
-    for (TransportMap::const_iterator it = transport_proxies().begin();
-        it != transport_proxies().end(); ++it) {
-      // We know that we have a FakeTransport*
+ protected:
+  virtual Transport* CreateTransport_w(const std::string& transport_name) {
+    return new FakeTransport(transport_name);
+  }
 
-      static_cast<FakeTransport*>(it->second->impl())->set_identity
-          (identity);
+  void Connect_w(FakeTransportController* dest) {
+    // Simulate the exchange of candidates.
+    ConnectChannels_w();
+    dest->ConnectChannels_w();
+    for (auto& kv : transports()) {
+      FakeTransport* transport = static_cast<FakeTransport*>(kv.second);
+      transport->SetDestination(dest->GetTransport_w(kv.first));
     }
   }
 
- protected:
-  virtual Transport* CreateTransport(const std::string& content_name) {
-    return new FakeTransport(signaling_thread(), worker_thread(), content_name);
-  }
-
-  void CompleteNegotiation() {
-    for (TransportMap::const_iterator it = transport_proxies().begin();
-        it != transport_proxies().end(); ++it) {
-      it->second->CompleteNegotiation();
-      it->second->ConnectChannels();
+  void ConnectChannels_w() {
+    for (auto& kv : transports()) {
+      FakeTransport* transport = static_cast<FakeTransport*>(kv.second);
+      transport->ConnectChannels();
     }
   }
 
  private:
   bool fail_create_channel_;
 };
 
 }  // namespace cricket
 
-#endif  // WEBRTC_P2P_BASE_FAKESESSION_H_
+#endif  // WEBRTC_P2P_BASE_FAKETRANSPORTCONTROLLER_H_