| Index: webrtc/p2p/base/faketransportcontroller.h
|
| diff --git a/webrtc/p2p/base/faketransportcontroller.h b/webrtc/p2p/base/faketransportcontroller.h
|
| deleted file mode 100644
|
| index e356e814b754a1b2b6f88f987b7bd33cb46e943a..0000000000000000000000000000000000000000
|
| --- a/webrtc/p2p/base/faketransportcontroller.h
|
| +++ /dev/null
|
| @@ -1,537 +0,0 @@
|
| -/*
|
| - * 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_FAKETRANSPORTCONTROLLER_H_
|
| -#define WEBRTC_P2P_BASE_FAKETRANSPORTCONTROLLER_H_
|
| -
|
| -#include <map>
|
| -#include <string>
|
| -#include <vector>
|
| -
|
| -#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.
|
| -// TODO(hbos): Move implementation to .cc file, this and other classes in file.
|
| -class FakeTransportChannel : public TransportChannelImpl,
|
| - public rtc::MessageHandler {
|
| - public:
|
| - explicit FakeTransportChannel(Transport* transport,
|
| - const std::string& name,
|
| - int component)
|
| - : TransportChannelImpl(name, component),
|
| - transport_(transport),
|
| - dtls_fingerprint_("", nullptr, 0) {}
|
| - ~FakeTransportChannel() { Reset(); }
|
| -
|
| - uint64 IceTiebreaker() const { return tiebreaker_; }
|
| - 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_; }
|
| - const std::string& remote_ice_ufrag() const { return remote_ice_ufrag_; }
|
| - const std::string& remote_ice_pwd() const { return remote_ice_pwd_; }
|
| - const rtc::SSLFingerprint& dtls_fingerprint() const {
|
| - return dtls_fingerprint_;
|
| - }
|
| -
|
| - // If async, will send packets by "Post"-ing to message queue instead of
|
| - // synchronously "Send"-ing.
|
| - void SetAsync(bool async) { async_ = async; }
|
| -
|
| - Transport* GetTransport() override { return transport_; }
|
| -
|
| - TransportChannelState GetState() const override {
|
| - if (connection_count_ == 0) {
|
| - return had_connection_ ? TransportChannelState::STATE_FAILED
|
| - : TransportChannelState::STATE_INIT;
|
| - }
|
| -
|
| - if (connection_count_ == 1) {
|
| - return TransportChannelState::STATE_COMPLETED;
|
| - }
|
| -
|
| - return TransportChannelState::STATE_CONNECTING;
|
| - }
|
| -
|
| - void SetIceRole(IceRole role) override { role_ = role; }
|
| - IceRole GetIceRole() const override { return role_; }
|
| - void SetIceTiebreaker(uint64 tiebreaker) override {
|
| - tiebreaker_ = tiebreaker;
|
| - }
|
| - void SetIceCredentials(const std::string& ice_ufrag,
|
| - const std::string& ice_pwd) override {
|
| - ice_ufrag_ = ice_ufrag;
|
| - ice_pwd_ = ice_pwd;
|
| - }
|
| - void SetRemoteIceCredentials(const std::string& ice_ufrag,
|
| - const std::string& ice_pwd) override {
|
| - remote_ice_ufrag_ = ice_ufrag;
|
| - remote_ice_pwd_ = ice_pwd;
|
| - }
|
| -
|
| - void SetRemoteIceMode(IceMode mode) override { remote_ice_mode_ = mode; }
|
| - bool SetRemoteFingerprint(const std::string& alg,
|
| - const uint8* digest,
|
| - size_t digest_len) override {
|
| - dtls_fingerprint_ = rtc::SSLFingerprint(alg, digest, digest_len);
|
| - return true;
|
| - }
|
| - bool SetSslRole(rtc::SSLRole role) override {
|
| - ssl_role_ = role;
|
| - return true;
|
| - }
|
| - bool GetSslRole(rtc::SSLRole* role) const override {
|
| - *role = ssl_role_;
|
| - return true;
|
| - }
|
| -
|
| - void Connect() override {
|
| - if (state_ == STATE_INIT) {
|
| - state_ = STATE_CONNECTING;
|
| - }
|
| - }
|
| -
|
| - void MaybeStartGathering() override {
|
| - if (gathering_state_ == kIceGatheringNew) {
|
| - gathering_state_ = kIceGatheringGathering;
|
| - SignalGatheringState(this);
|
| - }
|
| - }
|
| -
|
| - IceGatheringState gathering_state() const override {
|
| - return gathering_state_;
|
| - }
|
| -
|
| - void Reset() {
|
| - if (state_ != STATE_INIT) {
|
| - state_ = STATE_INIT;
|
| - if (dest_) {
|
| - dest_->state_ = STATE_INIT;
|
| - 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 (local_cert_ && dest_->local_cert_) {
|
| - 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_ = 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)
|
| - had_connection_ = true;
|
| - if (connection_count_ < old_connection_count)
|
| - SignalConnectionRemoved(this);
|
| - }
|
| -
|
| - void SetCandidatesGatheringComplete() {
|
| - if (gathering_state_ != kIceGatheringComplete) {
|
| - gathering_state_ = kIceGatheringComplete;
|
| - SignalGatheringState(this);
|
| - }
|
| - }
|
| -
|
| - void SetReceiving(bool receiving) { set_receiving(receiving); }
|
| -
|
| - void SetReceivingTimeout(int timeout) override {
|
| - receiving_timeout_ = timeout;
|
| - }
|
| -
|
| - int receiving_timeout() const { return receiving_timeout_; }
|
| -
|
| - int SendPacket(const char* data,
|
| - size_t len,
|
| - const rtc::PacketOptions& options,
|
| - int flags) override {
|
| - if (state_ != STATE_CONNECTED) {
|
| - return -1;
|
| - }
|
| -
|
| - if (flags != PF_SRTP_BYPASS && flags != 0) {
|
| - return -1;
|
| - }
|
| -
|
| - PacketMessageData* packet = new PacketMessageData(data, len);
|
| - if (async_) {
|
| - rtc::Thread::Current()->Post(this, 0, packet);
|
| - } else {
|
| - rtc::Thread::Current()->Send(this, 0, packet);
|
| - }
|
| - return static_cast<int>(len);
|
| - }
|
| - int SetOption(rtc::Socket::Option opt, int value) override { return true; }
|
| - bool GetOption(rtc::Socket::Option opt, int* value) override { return true; }
|
| - int GetError() override { return 0; }
|
| -
|
| - void AddRemoteCandidate(const Candidate& candidate) override {
|
| - remote_candidates_.push_back(candidate);
|
| - }
|
| - const Candidates& remote_candidates() const { return remote_candidates_; }
|
| -
|
| - void OnMessage(rtc::Message* msg) override {
|
| - PacketMessageData* data = static_cast<PacketMessageData*>(msg->pdata);
|
| - dest_->SignalReadPacket(dest_, data->packet.data<char>(),
|
| - data->packet.size(), rtc::CreatePacketTime(0), 0);
|
| - delete data;
|
| - }
|
| -
|
| - bool SetLocalCertificate(
|
| - const rtc::scoped_refptr<rtc::RTCCertificate>& certificate) {
|
| - local_cert_ = certificate;
|
| - return true;
|
| - }
|
| -
|
| - void SetRemoteSSLCertificate(rtc::FakeSSLCertificate* cert) {
|
| - remote_cert_ = cert;
|
| - }
|
| -
|
| - bool IsDtlsActive() const override { return do_dtls_; }
|
| -
|
| - bool SetSrtpCiphers(const std::vector<std::string>& ciphers) override {
|
| - srtp_ciphers_ = ciphers;
|
| - return true;
|
| - }
|
| -
|
| - bool GetSrtpCipher(std::string* cipher) override {
|
| - if (!chosen_srtp_cipher_.empty()) {
|
| - *cipher = chosen_srtp_cipher_;
|
| - return true;
|
| - }
|
| - return false;
|
| - }
|
| -
|
| - bool GetSslCipher(std::string* cipher) override { return false; }
|
| -
|
| - rtc::scoped_refptr<rtc::RTCCertificate> GetLocalCertificate() const {
|
| - return local_cert_;
|
| - }
|
| -
|
| - bool GetRemoteSSLCertificate(rtc::SSLCertificate** cert) const override {
|
| - if (!remote_cert_)
|
| - return false;
|
| -
|
| - *cert = remote_cert_->GetReference();
|
| - return true;
|
| - }
|
| -
|
| - bool ExportKeyingMaterial(const std::string& label,
|
| - const uint8* context,
|
| - size_t context_len,
|
| - bool use_context,
|
| - uint8* result,
|
| - size_t result_len) override {
|
| - if (!chosen_srtp_cipher_.empty()) {
|
| - memset(result, 0xff, result_len);
|
| - return true;
|
| - }
|
| -
|
| - return false;
|
| - }
|
| -
|
| - void NegotiateSrtpCiphers() {
|
| - for (std::vector<std::string>::const_iterator it1 = srtp_ciphers_.begin();
|
| - it1 != srtp_ciphers_.end(); ++it1) {
|
| - for (std::vector<std::string>::const_iterator it2 =
|
| - dest_->srtp_ciphers_.begin();
|
| - it2 != dest_->srtp_ciphers_.end(); ++it2) {
|
| - if (*it1 == *it2) {
|
| - chosen_srtp_cipher_ = *it1;
|
| - dest_->chosen_srtp_cipher_ = *it2;
|
| - return;
|
| - }
|
| - }
|
| - }
|
| - }
|
| -
|
| - bool GetStats(ConnectionInfos* infos) override {
|
| - ConnectionInfo info;
|
| - infos->clear();
|
| - infos->push_back(info);
|
| - return true;
|
| - }
|
| -
|
| - void set_ssl_max_protocol_version(rtc::SSLProtocolVersion version) {
|
| - ssl_max_version_ = version;
|
| - }
|
| - rtc::SSLProtocolVersion ssl_max_protocol_version() const {
|
| - return ssl_max_version_;
|
| - }
|
| -
|
| - private:
|
| - enum State { STATE_INIT, STATE_CONNECTING, STATE_CONNECTED };
|
| - Transport* transport_;
|
| - FakeTransportChannel* dest_ = nullptr;
|
| - State state_ = STATE_INIT;
|
| - bool async_ = false;
|
| - Candidates remote_candidates_;
|
| - rtc::scoped_refptr<rtc::RTCCertificate> local_cert_;
|
| - rtc::FakeSSLCertificate* remote_cert_ = nullptr;
|
| - bool do_dtls_ = false;
|
| - std::vector<std::string> srtp_ciphers_;
|
| - std::string chosen_srtp_cipher_;
|
| - int receiving_timeout_ = -1;
|
| - IceRole role_ = ICEROLE_UNKNOWN;
|
| - uint64 tiebreaker_ = 0;
|
| - std::string ice_ufrag_;
|
| - std::string ice_pwd_;
|
| - std::string remote_ice_ufrag_;
|
| - std::string remote_ice_pwd_;
|
| - IceMode remote_ice_mode_ = ICEMODE_FULL;
|
| - rtc::SSLProtocolVersion ssl_max_version_ = rtc::SSL_PROTOCOL_DTLS_10;
|
| - rtc::SSLFingerprint dtls_fingerprint_;
|
| - rtc::SSLRole ssl_role_ = rtc::SSL_CLIENT;
|
| - size_t connection_count_ = 0;
|
| - IceGatheringState gathering_state_ = kIceGatheringNew;
|
| - bool had_connection_ = false;
|
| -};
|
| -
|
| -// 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;
|
| -
|
| - explicit FakeTransport(const std::string& name) : Transport(name, nullptr) {}
|
| -
|
| - // Note that we only have a constructor with the allocator parameter so it can
|
| - // be wrapped by a DtlsTransport.
|
| - FakeTransport(const std::string& name, PortAllocator* allocator)
|
| - : Transport(name, nullptr) {}
|
| -
|
| - ~FakeTransport() { DestroyAllChannels(); }
|
| -
|
| - const ChannelMap& channels() const { return channels_; }
|
| -
|
| - // If async, will send packets by "Post"-ing to message queue instead of
|
| - // synchronously "Send"-ing.
|
| - void SetAsync(bool async) { async_ = async; }
|
| - void SetDestination(FakeTransport* dest) {
|
| - dest_ = dest;
|
| - for (const auto& kv : channels_) {
|
| - kv.second->SetLocalCertificate(certificate_);
|
| - SetChannelDestination(kv.first, kv.second);
|
| - }
|
| - }
|
| -
|
| - void SetWritable(bool writable) {
|
| - for (const auto& kv : channels_) {
|
| - kv.second->SetWritable(writable);
|
| - }
|
| - }
|
| -
|
| - void SetLocalCertificate(
|
| - const rtc::scoped_refptr<rtc::RTCCertificate>& certificate) override {
|
| - certificate_ = certificate;
|
| - }
|
| - bool GetLocalCertificate(
|
| - rtc::scoped_refptr<rtc::RTCCertificate>* certificate) override {
|
| - if (!certificate_)
|
| - return false;
|
| -
|
| - *certificate = certificate_;
|
| - return true;
|
| - }
|
| -
|
| - bool GetSslRole(rtc::SSLRole* role) const override {
|
| - if (channels_.empty()) {
|
| - return false;
|
| - }
|
| - return channels_.begin()->second->GetSslRole(role);
|
| - }
|
| -
|
| - bool SetSslMaxProtocolVersion(rtc::SSLProtocolVersion version) override {
|
| - ssl_max_version_ = version;
|
| - for (const auto& kv : channels_) {
|
| - kv.second->set_ssl_max_protocol_version(ssl_max_version_);
|
| - }
|
| - return true;
|
| - }
|
| - rtc::SSLProtocolVersion ssl_max_protocol_version() const {
|
| - return ssl_max_version_;
|
| - }
|
| -
|
| - using Transport::local_description;
|
| - using Transport::remote_description;
|
| -
|
| - protected:
|
| - TransportChannelImpl* CreateTransportChannel(int component) override {
|
| - if (channels_.find(component) != channels_.end()) {
|
| - return nullptr;
|
| - }
|
| - FakeTransportChannel* channel =
|
| - new FakeTransportChannel(this, name(), component);
|
| - channel->set_ssl_max_protocol_version(ssl_max_version_);
|
| - channel->SetAsync(async_);
|
| - SetChannelDestination(component, channel);
|
| - channels_[component] = channel;
|
| - return channel;
|
| - }
|
| -
|
| - void DestroyTransportChannel(TransportChannelImpl* channel) override {
|
| - channels_.erase(channel->component());
|
| - delete channel;
|
| - }
|
| -
|
| - private:
|
| - FakeTransportChannel* GetFakeChannel(int component) {
|
| - auto it = channels_.find(component);
|
| - return (it != channels_.end()) ? it->second : nullptr;
|
| - }
|
| -
|
| - void SetChannelDestination(int component, FakeTransportChannel* channel) {
|
| - FakeTransportChannel* dest_channel = nullptr;
|
| - if (dest_) {
|
| - dest_channel = dest_->GetFakeChannel(component);
|
| - if (dest_channel) {
|
| - dest_channel->SetLocalCertificate(dest_->certificate_);
|
| - }
|
| - }
|
| - 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_ = nullptr;
|
| - bool async_ = false;
|
| - rtc::scoped_refptr<rtc::RTCCertificate> certificate_;
|
| - rtc::SSLProtocolVersion ssl_max_version_ = rtc::SSL_PROTOCOL_DTLS_10;
|
| -};
|
| -
|
| -// Fake TransportController class, which can be passed into a BaseChannel object
|
| -// for test purposes. Can be connected to other FakeTransportControllers via
|
| -// Connect().
|
| -//
|
| -// This fake is unusual in that for the most part, it's implemented with the
|
| -// real TransportController code, but with fake TransportChannels underneath.
|
| -class FakeTransportController : public TransportController {
|
| - public:
|
| - FakeTransportController()
|
| - : TransportController(rtc::Thread::Current(),
|
| - rtc::Thread::Current(),
|
| - nullptr),
|
| - fail_create_channel_(false) {}
|
| -
|
| - explicit FakeTransportController(IceRole role)
|
| - : TransportController(rtc::Thread::Current(),
|
| - rtc::Thread::Current(),
|
| - nullptr),
|
| - fail_create_channel_(false) {
|
| - SetIceRole(role);
|
| - }
|
| -
|
| - explicit FakeTransportController(rtc::Thread* worker_thread)
|
| - : TransportController(rtc::Thread::Current(), worker_thread, nullptr),
|
| - fail_create_channel_(false) {}
|
| -
|
| - FakeTransportController(rtc::Thread* worker_thread, IceRole role)
|
| - : TransportController(rtc::Thread::Current(), worker_thread, nullptr),
|
| - fail_create_channel_(false) {
|
| - SetIceRole(role);
|
| - }
|
| -
|
| - 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));
|
| - }
|
| -
|
| - TransportChannel* CreateTransportChannel_w(const std::string& transport_name,
|
| - int component) override {
|
| - if (fail_create_channel_) {
|
| - return nullptr;
|
| - }
|
| - return TransportController::CreateTransportChannel_w(transport_name,
|
| - component);
|
| - }
|
| -
|
| - void set_fail_channel_creation(bool fail_channel_creation) {
|
| - fail_create_channel_ = fail_channel_creation;
|
| - }
|
| -
|
| - protected:
|
| - Transport* CreateTransport_w(const std::string& transport_name) override {
|
| - return new FakeTransport(transport_name);
|
| - }
|
| -
|
| - 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));
|
| - }
|
| - }
|
| -
|
| - void ConnectChannels_w() {
|
| - for (auto& kv : transports()) {
|
| - FakeTransport* transport = static_cast<FakeTransport*>(kv.second);
|
| - transport->ConnectChannels();
|
| - transport->MaybeStartGathering();
|
| - }
|
| - }
|
| -
|
| - private:
|
| - bool fail_create_channel_;
|
| -};
|
| -
|
| -} // namespace cricket
|
| -
|
| -#endif // WEBRTC_P2P_BASE_FAKETRANSPORTCONTROLLER_H_
|
|
|