| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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/search_engines/search_provider_install_data.h" | |
| 6 | |
| 7 #include <algorithm> | |
| 8 #include <functional> | |
| 9 #include <utility> | |
| 10 #include <vector> | |
| 11 | |
| 12 #include "base/bind.h" | |
| 13 #include "base/bind_helpers.h" | |
| 14 #include "base/logging.h" | |
| 15 #include "base/macros.h" | |
| 16 #include "base/memory/ref_counted.h" | |
| 17 #include "base/sequenced_task_runner_helpers.h" | |
| 18 #include "components/google/core/browser/google_url_tracker.h" | |
| 19 #include "components/search_engines/search_host_to_urls_map.h" | |
| 20 #include "components/search_engines/search_terms_data.h" | |
| 21 #include "components/search_engines/template_url.h" | |
| 22 #include "components/search_engines/template_url_service.h" | |
| 23 #include "components/search_engines/util.h" | |
| 24 #include "content/public/browser/browser_thread.h" | |
| 25 #include "content/public/browser/render_process_host.h" | |
| 26 #include "content/public/browser/render_process_host_observer.h" | |
| 27 | |
| 28 using content::BrowserThread; | |
| 29 | |
| 30 typedef SearchHostToURLsMap::TemplateURLSet TemplateURLSet; | |
| 31 | |
| 32 namespace { | |
| 33 | |
| 34 void LoadDataOnUIThread(TemplateURLService* template_url_service, | |
| 35 const base::Callback<void(ScopedVector<TemplateURL>, | |
| 36 TemplateURL*)>& callback) { | |
| 37 ScopedVector<TemplateURL> template_url_copies; | |
| 38 TemplateURL* default_provider_copy = NULL; | |
| 39 TemplateURLService::TemplateURLVector original_template_urls = | |
| 40 template_url_service->GetTemplateURLs(); | |
| 41 TemplateURL* original_default_provider = | |
| 42 template_url_service->GetDefaultSearchProvider(); | |
| 43 for (TemplateURLService::TemplateURLVector::const_iterator it = | |
| 44 original_template_urls.begin(); | |
| 45 it != original_template_urls.end(); | |
| 46 ++it) { | |
| 47 template_url_copies.push_back(new TemplateURL((*it)->data())); | |
| 48 if (*it == original_default_provider) | |
| 49 default_provider_copy = template_url_copies.back(); | |
| 50 } | |
| 51 BrowserThread::PostTask( | |
| 52 BrowserThread::IO, FROM_HERE, | |
| 53 base::Bind(callback, base::Passed(std::move(template_url_copies)), | |
| 54 base::Unretained(default_provider_copy))); | |
| 55 } | |
| 56 | |
| 57 // Implementation of SearchTermsData that may be used on the I/O thread. | |
| 58 class IOThreadSearchTermsData : public SearchTermsData { | |
| 59 public: | |
| 60 explicit IOThreadSearchTermsData(const std::string& google_base_url); | |
| 61 | |
| 62 // Implementation of SearchTermsData. | |
| 63 std::string GoogleBaseURLValue() const override; | |
| 64 | |
| 65 private: | |
| 66 std::string google_base_url_; | |
| 67 | |
| 68 DISALLOW_COPY_AND_ASSIGN(IOThreadSearchTermsData); | |
| 69 }; | |
| 70 | |
| 71 IOThreadSearchTermsData::IOThreadSearchTermsData( | |
| 72 const std::string& google_base_url) : google_base_url_(google_base_url) { | |
| 73 } | |
| 74 | |
| 75 std::string IOThreadSearchTermsData::GoogleBaseURLValue() const { | |
| 76 return google_base_url_; | |
| 77 } | |
| 78 | |
| 79 // Handles telling SearchProviderInstallData about changes to the google base | |
| 80 // url. (Ensure that this is deleted on the I/O thread so that the WeakPtr is | |
| 81 // deleted on the correct thread.) | |
| 82 class GoogleURLChangeNotifier | |
| 83 : public base::RefCountedThreadSafe<GoogleURLChangeNotifier, | |
| 84 BrowserThread::DeleteOnIOThread> { | |
| 85 public: | |
| 86 explicit GoogleURLChangeNotifier( | |
| 87 const base::WeakPtr<SearchProviderInstallData>& install_data); | |
| 88 | |
| 89 // Called on the I/O thread with the Google base URL whenever the value | |
| 90 // changes. | |
| 91 void OnChange(const std::string& google_base_url); | |
| 92 | |
| 93 private: | |
| 94 friend struct BrowserThread::DeleteOnThread<BrowserThread::IO>; | |
| 95 friend class base::DeleteHelper<GoogleURLChangeNotifier>; | |
| 96 | |
| 97 ~GoogleURLChangeNotifier() {} | |
| 98 | |
| 99 base::WeakPtr<SearchProviderInstallData> install_data_; | |
| 100 | |
| 101 DISALLOW_COPY_AND_ASSIGN(GoogleURLChangeNotifier); | |
| 102 }; | |
| 103 | |
| 104 GoogleURLChangeNotifier::GoogleURLChangeNotifier( | |
| 105 const base::WeakPtr<SearchProviderInstallData>& install_data) | |
| 106 : install_data_(install_data) { | |
| 107 } | |
| 108 | |
| 109 void GoogleURLChangeNotifier::OnChange(const std::string& google_base_url) { | |
| 110 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 111 if (install_data_.get()) | |
| 112 install_data_->OnGoogleURLChange(google_base_url); | |
| 113 } | |
| 114 | |
| 115 // Notices changes in the Google base URL and sends them along | |
| 116 // to the SearchProviderInstallData on the I/O thread. | |
| 117 class GoogleURLObserver : public content::RenderProcessHostObserver { | |
| 118 public: | |
| 119 GoogleURLObserver(GoogleURLTracker* google_url_tracker, | |
| 120 GoogleURLChangeNotifier* change_notifier, | |
| 121 content::RenderProcessHost* host); | |
| 122 | |
| 123 // Implementation of content::RenderProcessHostObserver. | |
| 124 void RenderProcessHostDestroyed(content::RenderProcessHost* host) override; | |
| 125 | |
| 126 private: | |
| 127 ~GoogleURLObserver() override {} | |
| 128 | |
| 129 // Callback that is called when the Google URL is updated. | |
| 130 void OnGoogleURLUpdated(); | |
| 131 | |
| 132 GoogleURLTracker* google_url_tracker_; | |
| 133 scoped_refptr<GoogleURLChangeNotifier> change_notifier_; | |
| 134 | |
| 135 std::unique_ptr<GoogleURLTracker::Subscription> | |
| 136 google_url_updated_subscription_; | |
| 137 | |
| 138 DISALLOW_COPY_AND_ASSIGN(GoogleURLObserver); | |
| 139 }; | |
| 140 | |
| 141 GoogleURLObserver::GoogleURLObserver( | |
| 142 GoogleURLTracker* google_url_tracker, | |
| 143 GoogleURLChangeNotifier* change_notifier, | |
| 144 content::RenderProcessHost* host) | |
| 145 : google_url_tracker_(google_url_tracker), | |
| 146 change_notifier_(change_notifier) { | |
| 147 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
| 148 google_url_updated_subscription_ = | |
| 149 google_url_tracker_->RegisterCallback(base::Bind( | |
| 150 &GoogleURLObserver::OnGoogleURLUpdated, base::Unretained(this))); | |
| 151 host->AddObserver(this); | |
| 152 } | |
| 153 | |
| 154 void GoogleURLObserver::OnGoogleURLUpdated() { | |
| 155 BrowserThread::PostTask(BrowserThread::IO, | |
| 156 FROM_HERE, | |
| 157 base::Bind(&GoogleURLChangeNotifier::OnChange, | |
| 158 change_notifier_.get(), | |
| 159 google_url_tracker_->google_url().spec())); | |
| 160 } | |
| 161 | |
| 162 void GoogleURLObserver::RenderProcessHostDestroyed( | |
| 163 content::RenderProcessHost* host) { | |
| 164 delete this; | |
| 165 } | |
| 166 | |
| 167 // Indicates if the two inputs have the same security origin. | |
| 168 // |requested_origin| should only be a security origin (no path, etc.). | |
| 169 // It is ok if |template_url| is NULL. | |
| 170 static bool IsSameOrigin(const GURL& requested_origin, | |
| 171 TemplateURL* template_url, | |
| 172 const SearchTermsData& search_terms_data) { | |
| 173 DCHECK(requested_origin == requested_origin.GetOrigin()); | |
| 174 DCHECK(template_url->GetType() != TemplateURL::OMNIBOX_API_EXTENSION); | |
| 175 return requested_origin == | |
| 176 template_url->GenerateSearchURL(search_terms_data).GetOrigin(); | |
| 177 } | |
| 178 | |
| 179 } // namespace | |
| 180 | |
| 181 SearchProviderInstallData::SearchProviderInstallData( | |
| 182 TemplateURLService* template_url_service, | |
| 183 const std::string& google_base_url, | |
| 184 GoogleURLTracker* google_url_tracker, | |
| 185 content::RenderProcessHost* host) | |
| 186 : template_url_service_(template_url_service), | |
| 187 google_base_url_(google_base_url), | |
| 188 weak_factory_(this) { | |
| 189 // GoogleURLTracker is not created in tests. | |
| 190 if (google_url_tracker) { | |
| 191 // GoogleURLObserver is responsible for killing itself when | |
| 192 // the given notification occurs. | |
| 193 new GoogleURLObserver( | |
| 194 google_url_tracker, | |
| 195 new GoogleURLChangeNotifier(weak_factory_.GetWeakPtr()), | |
| 196 host); | |
| 197 } | |
| 198 } | |
| 199 | |
| 200 SearchProviderInstallData::~SearchProviderInstallData() { | |
| 201 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 202 } | |
| 203 | |
| 204 void SearchProviderInstallData::CallWhenLoaded(const base::Closure& closure) { | |
| 205 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 206 | |
| 207 if (provider_map_.get()) { | |
| 208 closure.Run(); | |
| 209 return; | |
| 210 } | |
| 211 | |
| 212 bool do_load = closure_queue_.empty(); | |
| 213 closure_queue_.push_back(closure); | |
| 214 | |
| 215 // If the queue wasn't empty, there was already a load in progress. | |
| 216 if (!do_load) | |
| 217 return; | |
| 218 | |
| 219 if (template_url_service_) { | |
| 220 BrowserThread::PostTask( | |
| 221 BrowserThread::UI, | |
| 222 FROM_HERE, | |
| 223 base::Bind(&LoadDataOnUIThread, | |
| 224 template_url_service_, | |
| 225 base::Bind(&SearchProviderInstallData::OnTemplateURLsLoaded, | |
| 226 weak_factory_.GetWeakPtr()))); | |
| 227 } else { | |
| 228 OnLoadFailed(); | |
| 229 } | |
| 230 } | |
| 231 | |
| 232 SearchProviderInstallData::State SearchProviderInstallData::GetInstallState( | |
| 233 const GURL& requested_origin) { | |
| 234 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 235 DCHECK(provider_map_.get()); | |
| 236 | |
| 237 // First check to see if the origin is the default search provider. | |
| 238 if (requested_origin.spec() == default_search_origin_) | |
| 239 return INSTALLED_AS_DEFAULT; | |
| 240 | |
| 241 // Is the url any search provider? | |
| 242 const TemplateURLSet* urls = provider_map_->GetURLsForHost( | |
| 243 requested_origin.host()); | |
| 244 if (!urls) | |
| 245 return NOT_INSTALLED; | |
| 246 | |
| 247 IOThreadSearchTermsData search_terms_data(google_base_url_); | |
| 248 for (TemplateURLSet::const_iterator i = urls->begin(); | |
| 249 i != urls->end(); ++i) { | |
| 250 if (IsSameOrigin(requested_origin, *i, search_terms_data)) | |
| 251 return INSTALLED_BUT_NOT_DEFAULT; | |
| 252 } | |
| 253 return NOT_INSTALLED; | |
| 254 } | |
| 255 | |
| 256 void SearchProviderInstallData::OnGoogleURLChange( | |
| 257 const std::string& google_base_url) { | |
| 258 google_base_url_ = google_base_url; | |
| 259 } | |
| 260 | |
| 261 void SearchProviderInstallData::OnTemplateURLsLoaded( | |
| 262 ScopedVector<TemplateURL> template_urls, | |
| 263 TemplateURL* default_provider) { | |
| 264 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 265 | |
| 266 template_urls_ = std::move(template_urls); | |
| 267 | |
| 268 IOThreadSearchTermsData search_terms_data(google_base_url_); | |
| 269 provider_map_.reset(new SearchHostToURLsMap()); | |
| 270 provider_map_->Init(template_urls_.get(), search_terms_data); | |
| 271 SetDefault(default_provider); | |
| 272 NotifyLoaded(); | |
| 273 } | |
| 274 | |
| 275 void SearchProviderInstallData::SetDefault(const TemplateURL* template_url) { | |
| 276 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 277 | |
| 278 if (!template_url) { | |
| 279 default_search_origin_.clear(); | |
| 280 return; | |
| 281 } | |
| 282 | |
| 283 DCHECK(template_url->GetType() != TemplateURL::OMNIBOX_API_EXTENSION); | |
| 284 | |
| 285 IOThreadSearchTermsData search_terms_data(google_base_url_); | |
| 286 const GURL url(template_url->GenerateSearchURL(search_terms_data)); | |
| 287 if (!url.is_valid() || !url.has_host()) { | |
| 288 default_search_origin_.clear(); | |
| 289 return; | |
| 290 } | |
| 291 default_search_origin_ = url.GetOrigin().spec(); | |
| 292 } | |
| 293 | |
| 294 void SearchProviderInstallData::OnLoadFailed() { | |
| 295 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 296 | |
| 297 provider_map_.reset(new SearchHostToURLsMap()); | |
| 298 IOThreadSearchTermsData search_terms_data(google_base_url_); | |
| 299 provider_map_->Init(template_urls_.get(), search_terms_data); | |
| 300 SetDefault(NULL); | |
| 301 NotifyLoaded(); | |
| 302 } | |
| 303 | |
| 304 void SearchProviderInstallData::NotifyLoaded() { | |
| 305 DCHECK_CURRENTLY_ON(BrowserThread::IO); | |
| 306 | |
| 307 std::vector<base::Closure> closure_queue; | |
| 308 closure_queue.swap(closure_queue_); | |
| 309 | |
| 310 for (auto& closure : closure_queue) | |
| 311 closure.Run(); | |
| 312 | |
| 313 // Since we expect this request to be rare, clear out the information. This | |
| 314 // also keeps the responses current as the search providers change. | |
| 315 provider_map_.reset(); | |
| 316 SetDefault(NULL); | |
| 317 } | |
| OLD | NEW |