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),
32 weak_ptr_factory_(this) {}
33
34 ChromeBrowserProxyResolver::~ChromeBrowserProxyResolver() = default;
35
GetProxiesForUrl(const std::string & url,const ProxiesResolvedFn & callback)36 ProxyRequestId ChromeBrowserProxyResolver::GetProxiesForUrl(
37 const std::string& url, const ProxiesResolvedFn& callback) {
38 const ProxyRequestId id = next_request_id_++;
39 brillo::http::GetChromeProxyServersAsync(
40 DBusConnection::Get()->GetDBus(), url,
41 base::Bind(&ChromeBrowserProxyResolver::OnGetChromeProxyServers,
42 weak_ptr_factory_.GetWeakPtr(), id));
43 pending_callbacks_[id] = callback;
44 return id;
45 }
46
CancelProxyRequest(ProxyRequestId request)47 bool ChromeBrowserProxyResolver::CancelProxyRequest(ProxyRequestId request) {
48 return pending_callbacks_.erase(request) != 0;
49 }
50
OnGetChromeProxyServers(ProxyRequestId request_id,bool success,const std::vector<std::string> & proxies)51 void ChromeBrowserProxyResolver::OnGetChromeProxyServers(
52 ProxyRequestId request_id, bool success,
53 const std::vector<std::string>& proxies) {
54 // If |success| is false, |proxies| will still hold the direct proxy option
55 // which is what we do in our error case.
56 auto it = pending_callbacks_.find(request_id);
57 if (it == pending_callbacks_.end())
58 return;
59
60 ProxiesResolvedFn callback = it->second;
61 pending_callbacks_.erase(it);
62 callback.Run(std::deque<std::string>(proxies.begin(), proxies.end()));
63 }
64
65 } // namespace chromeos_update_engine
66