1116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch// Copyright 2014 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "remoting/protocol/chromium_port_allocator.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h" 95e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles)#include "base/strings/string_number_conversions.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_status_code.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context_getter.h" 14116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "remoting/protocol/chromium_socket_factory.h" 15116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "remoting/protocol/network_settings.h" 167dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "url/gurl.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 19116680a4aac90f2aa7413d9095a592090648e557Ben Murdochnamespace protocol { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 23eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdochclass ChromiumPortAllocatorSession 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : public cricket::HttpPortAllocatorSessionBase, 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public net::URLFetcherDelegate { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 27eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch ChromiumPortAllocatorSession( 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cricket::HttpPortAllocatorBase* allocator, 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& content_name, 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int component, 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& ice_username_fragment, 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& ice_password, 335f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) const std::vector<rtc::SocketAddress>& stun_hosts, 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<std::string>& relay_hosts, 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& relay, 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const scoped_refptr<net::URLRequestContextGetter>& url_context); 37eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual ~ChromiumPortAllocatorSession(); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // cricket::HttpPortAllocatorBase overrides. 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void ConfigReady(cricket::PortConfiguration* config) OVERRIDE; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SendSessionRequest(const std::string& host, int port) OVERRIDE; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // net::URLFetcherDelegate interface. 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnURLFetchComplete(const net::URLFetcher* url_fetcher) OVERRIDE; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<net::URLRequestContextGetter> url_context_; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::set<const net::URLFetcher*> url_fetchers_; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 50eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch DISALLOW_COPY_AND_ASSIGN(ChromiumPortAllocatorSession); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 53eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen MurdochChromiumPortAllocatorSession::ChromiumPortAllocatorSession( 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cricket::HttpPortAllocatorBase* allocator, 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& content_name, 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int component, 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& ice_username_fragment, 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& ice_password, 595f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) const std::vector<rtc::SocketAddress>& stun_hosts, 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<std::string>& relay_hosts, 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& relay, 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const scoped_refptr<net::URLRequestContextGetter>& url_context) 63c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : HttpPortAllocatorSessionBase(allocator, 64c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) content_name, 65c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) component, 66c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ice_username_fragment, 67c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ice_password, 68c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) stun_hosts, 69c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) relay_hosts, 70c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) relay, 71c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) std::string()), 72c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) url_context_(url_context) {} 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 74eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen MurdochChromiumPortAllocatorSession::~ChromiumPortAllocatorSession() { 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) STLDeleteElements(&url_fetchers_); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 78eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdochvoid ChromiumPortAllocatorSession::ConfigReady( 79eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch cricket::PortConfiguration* config) { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Filter out non-UDP relay ports, so that we don't try using TCP. 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (cricket::PortConfiguration::RelayList::iterator relay = 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config->relays.begin(); relay != config->relays.end(); ++relay) { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cricket::PortList filtered_ports; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (cricket::PortList::iterator port = 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relay->ports.begin(); port != relay->ports.end(); ++port) { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (port->proto == cricket::PROTO_UDP) { 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) filtered_ports.push_back(*port); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relay->ports = filtered_ports; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cricket::BasicPortAllocatorSession::ConfigReady(config); 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdochvoid ChromiumPortAllocatorSession::SendSessionRequest( 96eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch const std::string& host, 97eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch int port) { 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GURL url("https://" + host + ":" + base::IntToString(port) + 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetSessionRequestUrl() + "&sn=1"); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<net::URLFetcher> url_fetcher( 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::URLFetcher::Create(url, net::URLFetcher::GET, this)); 102868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) url_fetcher->SetRequestContext(url_context_.get()); 103868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) url_fetcher->AddExtraRequestHeader("X-Talk-Google-Relay-Auth: " + 104868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) relay_token()); 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) url_fetcher->AddExtraRequestHeader("X-Google-Relay-Auth: " + relay_token()); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) url_fetcher->AddExtraRequestHeader("X-Stream-Type: chromoting"); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) url_fetcher->Start(); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) url_fetchers_.insert(url_fetcher.release()); 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 111eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdochvoid ChromiumPortAllocatorSession::OnURLFetchComplete( 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const net::URLFetcher* source) { 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int response_code = source->GetResponseCode(); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string response; 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) source->GetResponseAsString(&response); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) url_fetchers_.erase(source); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete source; 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (response_code != net::HTTP_OK) { 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(WARNING) << "Received error when allocating relay session: " 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << response_code; 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TryCreateRelaySession(); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReceiveSessionResponse(response); 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 133eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdochscoped_ptr<ChromiumPortAllocator> ChromiumPortAllocator::Create( 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const scoped_refptr<net::URLRequestContextGetter>& url_context, 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NetworkSettings& network_settings) { 1365f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) scoped_ptr<rtc::NetworkManager> network_manager( 1375f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) new rtc::BasicNetworkManager()); 1385f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) scoped_ptr<rtc::PacketSocketFactory> socket_factory( 139116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch new ChromiumPacketSocketFactory()); 140eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch scoped_ptr<ChromiumPortAllocator> result( 141eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch new ChromiumPortAllocator(url_context, network_manager.Pass(), 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_factory.Pass())); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 14468043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // We always use PseudoTcp to provide a reliable channel. It provides poor 14568043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // performance when combined with TCP-based transport, so we have to disable 14668043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // TCP ports. ENABLE_SHARED_UFRAG flag is specified so that the same username 14768043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // fragment is shared between all candidates for this channel. 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int flags = cricket::PORTALLOCATOR_DISABLE_TCP | 14968043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) cricket::PORTALLOCATOR_ENABLE_SHARED_UFRAG | 15068043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) cricket::PORTALLOCATOR_ENABLE_IPV6; 1515c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1525c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu if (!(network_settings.flags & NetworkSettings::NAT_TRAVERSAL_STUN)) 1535c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu flags |= cricket::PORTALLOCATOR_DISABLE_STUN; 1545c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1555c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu if (!(network_settings.flags & NetworkSettings::NAT_TRAVERSAL_RELAY)) 1565c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu flags |= cricket::PORTALLOCATOR_DISABLE_RELAY; 1575c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result->set_flags(flags); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result->SetPortRange(network_settings.min_port, 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) network_settings.max_port); 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result.Pass(); 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen MurdochChromiumPortAllocator::ChromiumPortAllocator( 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const scoped_refptr<net::URLRequestContextGetter>& url_context, 1675f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) scoped_ptr<rtc::NetworkManager> network_manager, 1685f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) scoped_ptr<rtc::PacketSocketFactory> socket_factory) 169c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : HttpPortAllocatorBase(network_manager.get(), 170c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) socket_factory.get(), 171c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) std::string()), 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) url_context_(url_context), 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) network_manager_(network_manager.Pass()), 174c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) socket_factory_(socket_factory.Pass()) {} 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 176eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen MurdochChromiumPortAllocator::~ChromiumPortAllocator() { 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 179eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdochcricket::PortAllocatorSession* ChromiumPortAllocator::CreateSessionInternal( 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& content_name, 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int component, 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& ice_username_fragment, 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& ice_password) { 184eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch return new ChromiumPortAllocatorSession( 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, content_name, component, ice_username_fragment, ice_password, 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stun_hosts(), relay_hosts(), relay_token(), url_context_); 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 189116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} // namespace protocol 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 191