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 "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_configurator.h"
6
7 #include "base/prefs/pref_service.h"
8 #include "base/prefs/scoped_user_pref_update.h"
9 #include "base/sequenced_task_runner.h"
10 #include "base/strings/string_util.h"
11 #include "chrome/browser/prefs/proxy_prefs.h"
12 #include "chrome/common/pref_names.h"
13 #include "net/proxy/proxy_config.h"
14 #include "net/proxy/proxy_info.h"
15 #include "net/proxy/proxy_service.h"
16
DataReductionProxyChromeConfigurator(PrefService * prefs,scoped_refptr<base::SequencedTaskRunner> network_task_runner)17 DataReductionProxyChromeConfigurator::DataReductionProxyChromeConfigurator(
18 PrefService* prefs,
19 scoped_refptr<base::SequencedTaskRunner> network_task_runner)
20 : prefs_(prefs), network_task_runner_(network_task_runner) {
21 }
22
~DataReductionProxyChromeConfigurator()23 DataReductionProxyChromeConfigurator::~DataReductionProxyChromeConfigurator() {
24 }
25
Enable(bool primary_restricted,bool fallback_restricted,const std::string & primary_origin,const std::string & fallback_origin,const std::string & ssl_origin)26 void DataReductionProxyChromeConfigurator::Enable(
27 bool primary_restricted,
28 bool fallback_restricted,
29 const std::string& primary_origin,
30 const std::string& fallback_origin,
31 const std::string& ssl_origin) {
32 DCHECK(prefs_);
33 DictionaryPrefUpdate update(prefs_, prefs::kProxy);
34 // TODO(bengr): Consider relying on the proxy config for all data reduction
35 // proxy configuration.
36 base::DictionaryValue* dict = update.Get();
37
38 std::vector<std::string> proxies;
39 if (!primary_restricted) {
40 std::string trimmed_primary;
41 base::TrimString(primary_origin, "/", &trimmed_primary);
42 if (!trimmed_primary.empty())
43 proxies.push_back(trimmed_primary);
44 }
45 if (!fallback_restricted) {
46 std::string trimmed_fallback;
47 base::TrimString(fallback_origin, "/", &trimmed_fallback);
48 if (!trimmed_fallback.empty())
49 proxies.push_back(trimmed_fallback);
50 }
51 if (proxies.empty()) {
52 std::string mode;
53 // If already in a disabled mode, do nothing.
54 if (dict->GetString("mode", &mode))
55 if (ProxyModeToString(ProxyPrefs::MODE_SYSTEM) == mode)
56 return;
57 Disable();
58 return;
59 }
60
61 std::string trimmed_ssl;
62 base::TrimString(ssl_origin, "/", &trimmed_ssl);
63
64 std::string server = "http=" + JoinString(proxies, ",") + ",direct://;"
65 + (ssl_origin.empty() ? "" : ("https=" + trimmed_ssl + ",direct://;"));
66
67 dict->SetString("server", server);
68 dict->SetString("mode", ProxyModeToString(ProxyPrefs::MODE_FIXED_SERVERS));
69 dict->SetString("bypass_list", JoinString(bypass_rules_, ", "));
70
71 net::ProxyConfig config;
72 config.proxy_rules().ParseFromString(server);
73 config.proxy_rules().bypass_rules.ParseFromString(
74 JoinString(bypass_rules_, ", "));
75 // The ID is set to a bogus value. It cannot be left uninitialized, else the
76 // config will return invalid.
77 net::ProxyConfig::ID unused_id = 1;
78 config.set_id(unused_id);
79 network_task_runner_->PostTask(
80 FROM_HERE,
81 base::Bind(
82 &DataReductionProxyChromeConfigurator::UpdateProxyConfigOnIO,
83 base::Unretained(this),
84 config));
85 }
86
Disable()87 void DataReductionProxyChromeConfigurator::Disable() {
88 DCHECK(prefs_);
89 DictionaryPrefUpdate update(prefs_, prefs::kProxy);
90 base::DictionaryValue* dict = update.Get();
91 dict->SetString("mode", ProxyModeToString(ProxyPrefs::MODE_SYSTEM));
92 dict->SetString("server", "");
93 dict->SetString("bypass_list", "");
94 net::ProxyConfig config = net::ProxyConfig::CreateDirect();
95 network_task_runner_->PostTask(
96 FROM_HERE,
97 base::Bind(
98 &DataReductionProxyChromeConfigurator::UpdateProxyConfigOnIO,
99 base::Unretained(this),
100 config));
101 }
102
AddHostPatternToBypass(const std::string & pattern)103 void DataReductionProxyChromeConfigurator::AddHostPatternToBypass(
104 const std::string& pattern) {
105 bypass_rules_.push_back(pattern);
106 }
107
AddURLPatternToBypass(const std::string & pattern)108 void DataReductionProxyChromeConfigurator::AddURLPatternToBypass(
109 const std::string& pattern) {
110 size_t pos = pattern.find('/');
111 if (pattern.find('/', pos + 1) == pos + 1)
112 pos = pattern.find('/', pos + 2);
113
114 std::string host_pattern;
115 if (pos != std::string::npos)
116 host_pattern = pattern.substr(0, pos);
117 else
118 host_pattern = pattern;
119
120 AddHostPatternToBypass(host_pattern);
121 }
122
UpdateProxyConfigOnIO(const net::ProxyConfig & config)123 void DataReductionProxyChromeConfigurator::UpdateProxyConfigOnIO(
124 const net::ProxyConfig& config) {
125 config_ = config;
126 }
127
128 const net::ProxyConfig&
GetProxyConfigOnIO() const129 DataReductionProxyChromeConfigurator::GetProxyConfigOnIO() const {
130 return config_;
131 }
132