OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "remoting/protocol/chromium_port_allocator.h" | |
6 | |
7 #include <utility> | |
8 | |
9 #include "base/bind.h" | |
10 #include "base/macros.h" | |
11 #include "base/stl_util.h" | |
12 #include "base/strings/string_number_conversions.h" | |
13 #include "net/http/http_status_code.h" | |
14 #include "net/url_request/url_fetcher.h" | |
15 #include "net/url_request/url_fetcher_delegate.h" | |
16 #include "net/url_request/url_request_context_getter.h" | |
17 #include "remoting/protocol/chromium_socket_factory.h" | |
18 #include "remoting/protocol/transport_context.h" | |
19 #include "url/gurl.h" | |
20 | |
21 namespace remoting { | |
22 namespace protocol { | |
23 | |
24 namespace { | |
25 | |
26 class ChromiumPortAllocatorSession : public PortAllocatorSessionBase, | |
27 public net::URLFetcherDelegate { | |
28 public: | |
29 ChromiumPortAllocatorSession(ChromiumPortAllocator* allocator, | |
30 const std::string& content_name, | |
31 int component, | |
32 const std::string& ice_username_fragment, | |
33 const std::string& ice_password); | |
34 ~ChromiumPortAllocatorSession() override; | |
35 | |
36 // PortAllocatorBase overrides. | |
37 void SendSessionRequest(const std::string& host) override; | |
38 | |
39 // net::URLFetcherDelegate interface. | |
40 void OnURLFetchComplete(const net::URLFetcher* url_fetcher) override; | |
41 | |
42 private: | |
43 scoped_refptr<net::URLRequestContextGetter> url_context_; | |
44 std::set<const net::URLFetcher*> url_fetchers_; | |
45 | |
46 DISALLOW_COPY_AND_ASSIGN(ChromiumPortAllocatorSession); | |
47 }; | |
48 | |
49 ChromiumPortAllocatorSession::ChromiumPortAllocatorSession( | |
50 ChromiumPortAllocator* allocator, | |
51 const std::string& content_name, | |
52 int component, | |
53 const std::string& ice_username_fragment, | |
54 const std::string& ice_password) | |
55 : PortAllocatorSessionBase(allocator, | |
56 content_name, | |
57 component, | |
58 ice_username_fragment, | |
59 ice_password), | |
60 url_context_(allocator->url_context()) {} | |
61 | |
62 ChromiumPortAllocatorSession::~ChromiumPortAllocatorSession() { | |
63 STLDeleteElements(&url_fetchers_); | |
64 } | |
65 | |
66 void ChromiumPortAllocatorSession::SendSessionRequest(const std::string& host) { | |
67 GURL url("https://" + host + GetSessionRequestUrl() + "&sn=1"); | |
68 scoped_ptr<net::URLFetcher> url_fetcher = | |
69 net::URLFetcher::Create(url, net::URLFetcher::GET, this); | |
70 url_fetcher->SetRequestContext(url_context_.get()); | |
71 url_fetcher->AddExtraRequestHeader("X-Talk-Google-Relay-Auth: " + | |
72 relay_token()); | |
73 url_fetcher->AddExtraRequestHeader("X-Google-Relay-Auth: " + relay_token()); | |
74 url_fetcher->AddExtraRequestHeader("X-Stream-Type: chromoting"); | |
75 url_fetcher->Start(); | |
76 url_fetchers_.insert(url_fetcher.release()); | |
77 } | |
78 | |
79 void ChromiumPortAllocatorSession::OnURLFetchComplete( | |
80 const net::URLFetcher* source) { | |
81 int response_code = source->GetResponseCode(); | |
82 std::string response; | |
83 source->GetResponseAsString(&response); | |
84 | |
85 url_fetchers_.erase(source); | |
86 delete source; | |
87 | |
88 if (response_code != net::HTTP_OK) { | |
89 LOG(WARNING) << "Received error when allocating relay session: " | |
90 << response_code; | |
91 TryCreateRelaySession(); | |
92 return; | |
93 } | |
94 | |
95 ReceiveSessionResponse(response); | |
96 } | |
97 | |
98 } // namespace | |
99 | |
100 ChromiumPortAllocator::ChromiumPortAllocator( | |
101 scoped_ptr<rtc::NetworkManager> network_manager, | |
102 scoped_ptr<rtc::PacketSocketFactory> socket_factory, | |
103 scoped_refptr<TransportContext> transport_context, | |
104 scoped_refptr<net::URLRequestContextGetter> url_context) | |
105 : PortAllocatorBase(std::move(network_manager), | |
106 std::move(socket_factory), | |
107 transport_context), | |
108 url_context_(url_context) {} | |
109 | |
110 ChromiumPortAllocator::~ChromiumPortAllocator() {} | |
111 | |
112 cricket::PortAllocatorSession* ChromiumPortAllocator::CreateSessionInternal( | |
113 const std::string& content_name, | |
114 int component, | |
115 const std::string& ice_username_fragment, | |
116 const std::string& ice_password) { | |
117 return new ChromiumPortAllocatorSession(this, content_name, component, | |
118 ice_username_fragment, ice_password); | |
119 } | |
120 | |
121 ChromiumPortAllocatorFactory::ChromiumPortAllocatorFactory( | |
122 scoped_refptr<net::URLRequestContextGetter> url_request_context_getter) | |
123 : url_request_context_getter_(url_request_context_getter) {} | |
124 | |
125 ChromiumPortAllocatorFactory::~ChromiumPortAllocatorFactory() {} | |
126 | |
127 scoped_ptr<cricket::PortAllocator> | |
128 ChromiumPortAllocatorFactory::CreatePortAllocator( | |
129 scoped_refptr<TransportContext> transport_context) { | |
130 scoped_ptr<rtc::NetworkManager> network_manager( | |
131 new rtc::BasicNetworkManager()); | |
132 scoped_ptr<rtc::PacketSocketFactory> socket_factory( | |
133 new ChromiumPacketSocketFactory()); | |
134 return make_scoped_ptr(new ChromiumPortAllocator( | |
135 std::move(network_manager), std::move(socket_factory), transport_context, | |
136 url_request_context_getter_)); | |
137 } | |
138 | |
139 } // namespace protocol | |
140 } // namespace remoting | |
OLD | NEW |