0ba1533fdb
For WebRTC there are instances where it may be desirable to provide information to the STUN/TURN server about the website that initiated a peer connection. This modification allows an origin string to be included in the MediaConstraints object provided by the browser, which is then passed as a STUN header in communications with the server. A separate change will be submitted to the Chromium project that uses and is dependent on this change, implementing IETF draft http://tools.ietf.org/html/draft-johnston-tram-stun-origin-02 Originally a patch from skobalt@gmail.com. (https://webrtc-codereview.appspot.com/12839005/edit) R=juberti@webrtc.org Review URL: https://webrtc-codereview.appspot.com/41409004 git-svn-id: http://webrtc.googlecode.com/svn/trunk@8035 4adac7df-926f-26a2-2b94-8c16560cd09d
123 lines
4.0 KiB
C++
123 lines
4.0 KiB
C++
/*
|
|
* Copyright 2010 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_CLIENT_FAKEPORTALLOCATOR_H_
|
|
#define WEBRTC_P2P_CLIENT_FAKEPORTALLOCATOR_H_
|
|
|
|
#include <string>
|
|
#include "webrtc/p2p/base/basicpacketsocketfactory.h"
|
|
#include "webrtc/p2p/base/portallocator.h"
|
|
#include "webrtc/p2p/base/udpport.h"
|
|
#include "webrtc/base/scoped_ptr.h"
|
|
|
|
namespace rtc {
|
|
class SocketFactory;
|
|
class Thread;
|
|
}
|
|
|
|
namespace cricket {
|
|
|
|
class FakePortAllocatorSession : public PortAllocatorSession {
|
|
public:
|
|
FakePortAllocatorSession(rtc::Thread* worker_thread,
|
|
rtc::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",
|
|
rtc::IPAddress(INADDR_LOOPBACK), 8),
|
|
port_(), running_(false),
|
|
port_config_count_(0) {
|
|
network_.AddIP(rtc::IPAddress(INADDR_LOOPBACK));
|
|
}
|
|
|
|
virtual void StartGettingPorts() {
|
|
if (!port_) {
|
|
port_.reset(cricket::UDPPort::Create(worker_thread_,
|
|
factory_,
|
|
&network_,
|
|
#ifdef USE_WEBRTC_DEV_BRANCH
|
|
network_.GetBestIP(),
|
|
#else // USE_WEBRTC_DEV_BRANCH
|
|
network_.ip(),
|
|
#endif // USE_WEBRTC_DEV_BRANCH
|
|
0,
|
|
0,
|
|
username(),
|
|
password(),
|
|
std::string()));
|
|
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:
|
|
rtc::Thread* worker_thread_;
|
|
rtc::PacketSocketFactory* factory_;
|
|
rtc::Network network_;
|
|
rtc::scoped_ptr<cricket::Port> port_;
|
|
bool running_;
|
|
int port_config_count_;
|
|
};
|
|
|
|
class FakePortAllocator : public cricket::PortAllocator {
|
|
public:
|
|
FakePortAllocator(rtc::Thread* worker_thread,
|
|
rtc::PacketSocketFactory* factory)
|
|
: worker_thread_(worker_thread), factory_(factory) {
|
|
if (factory_ == NULL) {
|
|
owned_factory_.reset(new rtc::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:
|
|
rtc::Thread* worker_thread_;
|
|
rtc::PacketSocketFactory* factory_;
|
|
rtc::scoped_ptr<rtc::BasicPacketSocketFactory> owned_factory_;
|
|
};
|
|
|
|
} // namespace cricket
|
|
|
|
#endif // WEBRTC_P2P_CLIENT_FAKEPORTALLOCATOR_H_
|