97077a3ab2
Update libyuv to r826. TEST=try bots R=niklas.enbom@webrtc.org Review URL: https://webrtc-codereview.appspot.com/2889004 git-svn-id: http://webrtc.googlecode.com/svn/trunk@5038 4adac7df-926f-26a2-2b94-8c16560cd09d
108 lines
3.3 KiB
C++
108 lines
3.3 KiB
C++
// Copyright 2010 Google Inc. All Rights Reserved,
|
|
//
|
|
// Author: Justin Uberti (juberti@google.com)
|
|
|
|
#ifndef TALK_P2P_CLIENT_FAKEPORTALLOCATOR_H_
|
|
#define TALK_P2P_CLIENT_FAKEPORTALLOCATOR_H_
|
|
|
|
#include <string>
|
|
#include "talk/base/scoped_ptr.h"
|
|
#include "talk/p2p/base/basicpacketsocketfactory.h"
|
|
#include "talk/p2p/base/portallocator.h"
|
|
#include "talk/p2p/base/udpport.h"
|
|
|
|
namespace talk_base {
|
|
class SocketFactory;
|
|
class Thread;
|
|
}
|
|
|
|
namespace cricket {
|
|
|
|
class FakePortAllocatorSession : public PortAllocatorSession {
|
|
public:
|
|
FakePortAllocatorSession(talk_base::Thread* worker_thread,
|
|
talk_base::PacketSocketFactory* factory,
|
|
const std::string& content_name,
|
|
int component,
|
|
const std::string& ice_ufrag,
|
|
const std::string& ice_pwd)
|
|
: PortAllocatorSession(content_name, component, ice_ufrag, ice_pwd,
|
|
cricket::PORTALLOCATOR_ENABLE_SHARED_UFRAG),
|
|
worker_thread_(worker_thread),
|
|
factory_(factory),
|
|
network_("network", "unittest",
|
|
talk_base::IPAddress(INADDR_LOOPBACK), 8),
|
|
port_(), running_(false),
|
|
port_config_count_(0) {
|
|
network_.AddIP(talk_base::IPAddress(INADDR_LOOPBACK));
|
|
}
|
|
|
|
virtual void StartGettingPorts() {
|
|
if (!port_) {
|
|
port_.reset(cricket::UDPPort::Create(worker_thread_, factory_,
|
|
&network_, network_.ip(), 0, 0,
|
|
username(),
|
|
password()));
|
|
AddPort(port_.get());
|
|
}
|
|
++port_config_count_;
|
|
running_ = true;
|
|
}
|
|
|
|
virtual void StopGettingPorts() { running_ = false; }
|
|
virtual bool IsGettingPorts() { return running_; }
|
|
int port_config_count() { return port_config_count_; }
|
|
|
|
void AddPort(cricket::Port* port) {
|
|
port->set_component(component_);
|
|
port->set_generation(0);
|
|
port->SignalPortComplete.connect(
|
|
this, &FakePortAllocatorSession::OnPortComplete);
|
|
port->PrepareAddress();
|
|
SignalPortReady(this, port);
|
|
}
|
|
void OnPortComplete(cricket::Port* port) {
|
|
SignalCandidatesReady(this, port->Candidates());
|
|
SignalCandidatesAllocationDone(this);
|
|
}
|
|
|
|
private:
|
|
talk_base::Thread* worker_thread_;
|
|
talk_base::PacketSocketFactory* factory_;
|
|
talk_base::Network network_;
|
|
talk_base::scoped_ptr<cricket::Port> port_;
|
|
bool running_;
|
|
int port_config_count_;
|
|
};
|
|
|
|
class FakePortAllocator : public cricket::PortAllocator {
|
|
public:
|
|
FakePortAllocator(talk_base::Thread* worker_thread,
|
|
talk_base::PacketSocketFactory* factory)
|
|
: worker_thread_(worker_thread), factory_(factory) {
|
|
if (factory_ == NULL) {
|
|
owned_factory_.reset(new talk_base::BasicPacketSocketFactory(
|
|
worker_thread_));
|
|
factory_ = owned_factory_.get();
|
|
}
|
|
}
|
|
|
|
virtual cricket::PortAllocatorSession* CreateSessionInternal(
|
|
const std::string& content_name,
|
|
int component,
|
|
const std::string& ice_ufrag,
|
|
const std::string& ice_pwd) {
|
|
return new FakePortAllocatorSession(
|
|
worker_thread_, factory_, content_name, component, ice_ufrag, ice_pwd);
|
|
}
|
|
|
|
private:
|
|
talk_base::Thread* worker_thread_;
|
|
talk_base::PacketSocketFactory* factory_;
|
|
talk_base::scoped_ptr<talk_base::BasicPacketSocketFactory> owned_factory_;
|
|
};
|
|
|
|
} // namespace cricket
|
|
|
|
#endif // TALK_P2P_CLIENT_FAKEPORTALLOCATOR_H_
|