1 //
2 // Copyright (C) 2011 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16
17 #include "update_engine/chrome_browser_proxy_resolver.h"
18
19 #include <utility>
20
21 #include <base/bind.h>
22 #include <base/memory/ptr_util.h>
23 #include <base/strings/string_util.h>
24 #include <brillo/http/http_proxy.h>
25
26 #include "update_engine/dbus_connection.h"
27
28 namespace chromeos_update_engine {
29
ChromeBrowserProxyResolver()30 ChromeBrowserProxyResolver::ChromeBrowserProxyResolver()
31 : next_request_id_(kProxyRequestIdNull + 1), weak_ptr_factory_(this) {}
32
33 ChromeBrowserProxyResolver::~ChromeBrowserProxyResolver() = default;
34
GetProxiesForUrl(const std::string & url,const ProxiesResolvedFn & callback)35 ProxyRequestId ChromeBrowserProxyResolver::GetProxiesForUrl(
36 const std::string& url, const ProxiesResolvedFn& callback) {
37 const ProxyRequestId id = next_request_id_++;
38 brillo::http::GetChromeProxyServersAsync(
39 DBusConnection::Get()->GetDBus(),
40 url,
41 base::Bind(&ChromeBrowserProxyResolver::OnGetChromeProxyServers,
42 weak_ptr_factory_.GetWeakPtr(),
43 id));
44 pending_callbacks_[id] = callback;
45 return id;
46 }
47
CancelProxyRequest(ProxyRequestId request)48 bool ChromeBrowserProxyResolver::CancelProxyRequest(ProxyRequestId request) {
49 return pending_callbacks_.erase(request) != 0;
50 }
51
OnGetChromeProxyServers(ProxyRequestId request_id,bool success,const std::vector<std::string> & proxies)52 void ChromeBrowserProxyResolver::OnGetChromeProxyServers(
53 ProxyRequestId request_id,
54 bool success,
55 const std::vector<std::string>& proxies) {
56 // If |success| is false, |proxies| will still hold the direct proxy option
57 // which is what we do in our error case.
58 auto it = pending_callbacks_.find(request_id);
59 if (it == pending_callbacks_.end())
60 return;
61
62 ProxiesResolvedFn callback = it->second;
63 pending_callbacks_.erase(it);
64 callback.Run(std::deque<std::string>(proxies.begin(), proxies.end()));
65 }
66
67 } // namespace chromeos_update_engine
68