• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright 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/autocomplete/search_provider.h"
6 
7 #include <algorithm>
8 #include <cmath>
9 
10 #include "base/base64.h"
11 #include "base/callback.h"
12 #include "base/command_line.h"
13 #include "base/i18n/break_iterator.h"
14 #include "base/i18n/case_conversion.h"
15 #include "base/json/json_string_value_serializer.h"
16 #include "base/message_loop/message_loop.h"
17 #include "base/metrics/histogram.h"
18 #include "base/prefs/pref_service.h"
19 #include "base/rand_util.h"
20 #include "base/strings/string_util.h"
21 #include "base/strings/utf_string_conversions.h"
22 #include "chrome/browser/autocomplete/autocomplete_classifier.h"
23 #include "chrome/browser/autocomplete/autocomplete_classifier_factory.h"
24 #include "chrome/browser/autocomplete/autocomplete_provider_listener.h"
25 #include "chrome/browser/autocomplete/autocomplete_result.h"
26 #include "chrome/browser/autocomplete/keyword_provider.h"
27 #include "chrome/browser/history/history_service.h"
28 #include "chrome/browser/history/history_service_factory.h"
29 #include "chrome/browser/history/in_memory_database.h"
30 #include "chrome/browser/metrics/variations/variations_http_header_provider.h"
31 #include "chrome/browser/omnibox/omnibox_field_trial.h"
32 #include "chrome/browser/profiles/profile.h"
33 #include "chrome/browser/search/search.h"
34 #include "chrome/browser/search_engines/template_url_prepopulate_data.h"
35 #include "chrome/browser/search_engines/template_url_service.h"
36 #include "chrome/browser/search_engines/template_url_service_factory.h"
37 #include "chrome/browser/ui/search/instant_controller.h"
38 #include "chrome/common/chrome_switches.h"
39 #include "chrome/common/pref_names.h"
40 #include "components/autocomplete/url_prefix.h"
41 #include "components/google/core/browser/google_util.h"
42 #include "components/metrics/proto/omnibox_input_type.pb.h"
43 #include "content/public/browser/user_metrics.h"
44 #include "grit/generated_resources.h"
45 #include "net/base/escape.h"
46 #include "net/base/load_flags.h"
47 #include "net/base/net_util.h"
48 #include "net/http/http_request_headers.h"
49 #include "net/url_request/url_fetcher.h"
50 #include "net/url_request/url_request_status.h"
51 #include "ui/base/l10n/l10n_util.h"
52 #include "url/url_constants.h"
53 #include "url/url_util.h"
54 
55 // Helpers --------------------------------------------------------------------
56 
57 namespace {
58 
59 // We keep track in a histogram how many suggest requests we send, how
60 // many suggest requests we invalidate (e.g., due to a user typing
61 // another character), and how many replies we receive.
62 // *** ADD NEW ENUMS AFTER ALL PREVIOUSLY DEFINED ONES! ***
63 //     (excluding the end-of-list enum value)
64 // We do not want values of existing enums to change or else it screws
65 // up the statistics.
66 enum SuggestRequestsHistogramValue {
67   REQUEST_SENT = 1,
68   REQUEST_INVALIDATED,
69   REPLY_RECEIVED,
70   MAX_SUGGEST_REQUEST_HISTOGRAM_VALUE
71 };
72 
73 // The verbatim score for an input which is not an URL.
74 const int kNonURLVerbatimRelevance = 1300;
75 
76 // Increments the appropriate value in the histogram by one.
LogOmniboxSuggestRequest(SuggestRequestsHistogramValue request_value)77 void LogOmniboxSuggestRequest(
78     SuggestRequestsHistogramValue request_value) {
79   UMA_HISTOGRAM_ENUMERATION("Omnibox.SuggestRequests", request_value,
80                             MAX_SUGGEST_REQUEST_HISTOGRAM_VALUE);
81 }
82 
HasMultipleWords(const base::string16 & text)83 bool HasMultipleWords(const base::string16& text) {
84   base::i18n::BreakIterator i(text, base::i18n::BreakIterator::BREAK_WORD);
85   bool found_word = false;
86   if (i.Init()) {
87     while (i.Advance()) {
88       if (i.IsWord()) {
89         if (found_word)
90           return true;
91         found_word = true;
92       }
93     }
94   }
95   return false;
96 }
97 
98 }  // namespace
99 
100 // SearchProvider::Providers --------------------------------------------------
101 
Providers(TemplateURLService * template_url_service)102 SearchProvider::Providers::Providers(TemplateURLService* template_url_service)
103     : template_url_service_(template_url_service) {}
104 
GetDefaultProviderURL() const105 const TemplateURL* SearchProvider::Providers::GetDefaultProviderURL() const {
106   return default_provider_.empty() ? NULL :
107       template_url_service_->GetTemplateURLForKeyword(default_provider_);
108 }
109 
GetKeywordProviderURL() const110 const TemplateURL* SearchProvider::Providers::GetKeywordProviderURL() const {
111   return keyword_provider_.empty() ? NULL :
112       template_url_service_->GetTemplateURLForKeyword(keyword_provider_);
113 }
114 
115 
116 // SearchProvider::CompareScoredResults ---------------------------------------
117 
118 class SearchProvider::CompareScoredResults {
119  public:
operator ()(const Result & a,const Result & b)120   bool operator()(const Result& a, const Result& b) {
121     // Sort in descending relevance order.
122     return a.relevance() > b.relevance();
123   }
124 };
125 
126 
127 // SearchProvider -------------------------------------------------------------
128 
129 // static
130 int SearchProvider::kMinimumTimeBetweenSuggestQueriesMs = 100;
131 
SearchProvider(AutocompleteProviderListener * listener,Profile * profile)132 SearchProvider::SearchProvider(AutocompleteProviderListener* listener,
133                                Profile* profile)
134     : BaseSearchProvider(listener, profile, AutocompleteProvider::TYPE_SEARCH),
135       providers_(TemplateURLServiceFactory::GetForProfile(profile)) {
136 }
137 
138 // static
GetSuggestMetadata(const AutocompleteMatch & match)139 std::string SearchProvider::GetSuggestMetadata(const AutocompleteMatch& match) {
140   return match.GetAdditionalInfo(kSuggestMetadataKey);
141 }
142 
ResetSession()143 void SearchProvider::ResetSession() {
144   field_trial_triggered_in_session_ = false;
145 }
146 
~SearchProvider()147 SearchProvider::~SearchProvider() {
148 }
149 
UpdateMatchContentsClass(const base::string16 & input_text,Results * results)150 void SearchProvider::UpdateMatchContentsClass(const base::string16& input_text,
151                                               Results* results) {
152   for (SuggestResults::iterator sug_it = results->suggest_results.begin();
153        sug_it != results->suggest_results.end(); ++sug_it) {
154     sug_it->ClassifyMatchContents(false, input_text);
155   }
156   const std::string languages(
157       profile_->GetPrefs()->GetString(prefs::kAcceptLanguages));
158   for (NavigationResults::iterator nav_it = results->navigation_results.begin();
159        nav_it != results->navigation_results.end(); ++nav_it) {
160     nav_it->CalculateAndClassifyMatchContents(false, input_text, languages);
161   }
162 }
163 
164 // static
CalculateRelevanceForKeywordVerbatim(metrics::OmniboxInputType::Type type,bool prefer_keyword)165 int SearchProvider::CalculateRelevanceForKeywordVerbatim(
166     metrics::OmniboxInputType::Type type,
167     bool prefer_keyword) {
168   // This function is responsible for scoring verbatim query matches
169   // for non-extension keywords.  KeywordProvider::CalculateRelevance()
170   // scores verbatim query matches for extension keywords, as well as
171   // for keyword matches (i.e., suggestions of a keyword itself, not a
172   // suggestion of a query on a keyword search engine).  These two
173   // functions are currently in sync, but there's no reason we
174   // couldn't decide in the future to score verbatim matches
175   // differently for extension and non-extension keywords.  If you
176   // make such a change, however, you should update this comment to
177   // describe it, so it's clear why the functions diverge.
178   if (prefer_keyword)
179     return 1500;
180   return (type == metrics::OmniboxInputType::QUERY) ? 1450 : 1100;
181 }
182 
Start(const AutocompleteInput & input,bool minimal_changes)183 void SearchProvider::Start(const AutocompleteInput& input,
184                            bool minimal_changes) {
185   // Do our best to load the model as early as possible.  This will reduce
186   // odds of having the model not ready when really needed (a non-empty input).
187   TemplateURLService* model = providers_.template_url_service();
188   DCHECK(model);
189   model->Load();
190 
191   matches_.clear();
192   field_trial_triggered_ = false;
193 
194   // Can't return search/suggest results for bogus input or without a profile.
195   if (!profile_ || (input.type() == metrics::OmniboxInputType::INVALID)) {
196     Stop(true);
197     return;
198   }
199 
200   keyword_input_ = input;
201   const TemplateURL* keyword_provider =
202       KeywordProvider::GetSubstitutingTemplateURLForInput(model,
203                                                           &keyword_input_);
204   if (keyword_provider == NULL)
205     keyword_input_.Clear();
206   else if (keyword_input_.text().empty())
207     keyword_provider = NULL;
208 
209   const TemplateURL* default_provider = model->GetDefaultSearchProvider();
210   if (default_provider &&
211       !default_provider->SupportsReplacement(model->search_terms_data()))
212     default_provider = NULL;
213 
214   if (keyword_provider == default_provider)
215     default_provider = NULL;  // No use in querying the same provider twice.
216 
217   if (!default_provider && !keyword_provider) {
218     // No valid providers.
219     Stop(true);
220     return;
221   }
222 
223   // If we're still running an old query but have since changed the query text
224   // or the providers, abort the query.
225   base::string16 default_provider_keyword(default_provider ?
226       default_provider->keyword() : base::string16());
227   base::string16 keyword_provider_keyword(keyword_provider ?
228       keyword_provider->keyword() : base::string16());
229   if (!minimal_changes ||
230       !providers_.equal(default_provider_keyword, keyword_provider_keyword)) {
231     // Cancel any in-flight suggest requests.
232     if (!done_)
233       Stop(false);
234   }
235 
236   providers_.set(default_provider_keyword, keyword_provider_keyword);
237 
238   if (input.text().empty()) {
239     // User typed "?" alone.  Give them a placeholder result indicating what
240     // this syntax does.
241     if (default_provider) {
242       AutocompleteMatch match;
243       match.provider = this;
244       match.contents.assign(l10n_util::GetStringUTF16(IDS_EMPTY_KEYWORD_VALUE));
245       match.contents_class.push_back(
246           ACMatchClassification(0, ACMatchClassification::NONE));
247       match.keyword = providers_.default_provider();
248       match.allowed_to_be_default_match = true;
249       matches_.push_back(match);
250     }
251     Stop(true);
252     return;
253   }
254 
255   input_ = input;
256 
257   DoHistoryQuery(minimal_changes);
258   StartOrStopSuggestQuery(minimal_changes);
259   UpdateMatches();
260 }
261 
SortResults(bool is_keyword,const base::ListValue * relevances,Results * results)262 void SearchProvider::SortResults(bool is_keyword,
263                                  const base::ListValue* relevances,
264                                  Results* results) {
265   // Ignore suggested scores for non-keyword matches in keyword mode; if the
266   // server is allowed to score these, it could interfere with the user's
267   // ability to get good keyword results.
268   const bool abandon_suggested_scores =
269       !is_keyword && !providers_.keyword_provider().empty();
270   // Apply calculated relevance scores to suggestions if a valid list was
271   // not provided or we're abandoning suggested scores entirely.
272   if ((relevances == NULL) || abandon_suggested_scores) {
273     ApplyCalculatedSuggestRelevance(&results->suggest_results);
274     ApplyCalculatedNavigationRelevance(&results->navigation_results);
275     // If abandoning scores entirely, also abandon the verbatim score.
276     if (abandon_suggested_scores)
277       results->verbatim_relevance = -1;
278   }
279 
280   // Keep the result lists sorted.
281   const CompareScoredResults comparator = CompareScoredResults();
282   std::stable_sort(results->suggest_results.begin(),
283                    results->suggest_results.end(),
284                    comparator);
285   std::stable_sort(results->navigation_results.begin(),
286                    results->navigation_results.end(),
287                    comparator);
288 }
289 
GetTemplateURL(bool is_keyword) const290 const TemplateURL* SearchProvider::GetTemplateURL(bool is_keyword) const {
291   return is_keyword ? providers_.GetKeywordProviderURL()
292                     : providers_.GetDefaultProviderURL();
293 }
294 
GetInput(bool is_keyword) const295 const AutocompleteInput SearchProvider::GetInput(bool is_keyword) const {
296   return is_keyword ? keyword_input_ : input_;
297 }
298 
GetResultsToFill(bool is_keyword)299 BaseSearchProvider::Results* SearchProvider::GetResultsToFill(bool is_keyword) {
300   return is_keyword ? &keyword_results_ : &default_results_;
301 }
302 
ShouldAppendExtraParams(const SuggestResult & result) const303 bool SearchProvider::ShouldAppendExtraParams(
304     const SuggestResult& result) const {
305   return !result.from_keyword_provider() ||
306       providers_.default_provider().empty();
307 }
308 
StopSuggest()309 void SearchProvider::StopSuggest() {
310   // Increment the appropriate field in the histogram by the number of
311   // pending requests that were invalidated.
312   for (int i = 0; i < suggest_results_pending_; ++i)
313     LogOmniboxSuggestRequest(REQUEST_INVALIDATED);
314   suggest_results_pending_ = 0;
315   timer_.Stop();
316   // Stop any in-progress URL fetches.
317   keyword_fetcher_.reset();
318   default_fetcher_.reset();
319 }
320 
ClearAllResults()321 void SearchProvider::ClearAllResults() {
322   keyword_results_.Clear();
323   default_results_.Clear();
324 }
325 
GetDefaultResultRelevance() const326 int SearchProvider::GetDefaultResultRelevance() const {
327   return -1;
328 }
329 
RecordDeletionResult(bool success)330 void SearchProvider::RecordDeletionResult(bool success) {
331   if (success) {
332     content::RecordAction(
333         base::UserMetricsAction("Omnibox.ServerSuggestDelete.Success"));
334   } else {
335     content::RecordAction(
336         base::UserMetricsAction("Omnibox.ServerSuggestDelete.Failure"));
337   }
338 }
339 
LogFetchComplete(bool success,bool is_keyword)340 void SearchProvider::LogFetchComplete(bool success, bool is_keyword) {
341   LogOmniboxSuggestRequest(REPLY_RECEIVED);
342   // Record response time for suggest requests sent to Google.  We care
343   // only about the common case: the Google default provider used in
344   // non-keyword mode.
345   const TemplateURL* default_url = providers_.GetDefaultProviderURL();
346   if (!is_keyword && default_url &&
347       (TemplateURLPrepopulateData::GetEngineType(
348           *default_url,
349           providers_.template_url_service()->search_terms_data()) ==
350        SEARCH_ENGINE_GOOGLE)) {
351     const base::TimeDelta elapsed_time =
352         base::TimeTicks::Now() - time_suggest_request_sent_;
353     if (success) {
354       UMA_HISTOGRAM_TIMES("Omnibox.SuggestRequest.Success.GoogleResponseTime",
355                           elapsed_time);
356     } else {
357       UMA_HISTOGRAM_TIMES("Omnibox.SuggestRequest.Failure.GoogleResponseTime",
358                           elapsed_time);
359     }
360   }
361 }
362 
IsKeywordFetcher(const net::URLFetcher * fetcher) const363 bool SearchProvider::IsKeywordFetcher(const net::URLFetcher* fetcher) const {
364   return fetcher == keyword_fetcher_.get();
365 }
366 
UpdateMatches()367 void SearchProvider::UpdateMatches() {
368   ConvertResultsToAutocompleteMatches();
369 
370   // Check constraints that may be violated by suggested relevances.
371   if (!matches_.empty() &&
372       (default_results_.HasServerProvidedScores() ||
373        keyword_results_.HasServerProvidedScores())) {
374     // These blocks attempt to repair undesirable behavior by suggested
375     // relevances with minimal impact, preserving other suggested relevances.
376 
377     if (!HasKeywordDefaultMatchInKeywordMode()) {
378       // In keyword mode, disregard the keyword verbatim suggested relevance
379       // if necessary so there at least one keyword match that's allowed to
380       // be the default match.
381       keyword_results_.verbatim_relevance = -1;
382       ConvertResultsToAutocompleteMatches();
383     }
384     if (IsTopMatchSearchWithURLInput()) {
385       // Disregard the suggested search and verbatim relevances if the input
386       // type is URL and the top match is a highly-ranked search suggestion.
387       // For example, prevent a search for "foo.com" from outranking another
388       // provider's navigation for "foo.com" or "foo.com/url_from_history".
389       ApplyCalculatedSuggestRelevance(&keyword_results_.suggest_results);
390       ApplyCalculatedSuggestRelevance(&default_results_.suggest_results);
391       default_results_.verbatim_relevance = -1;
392       keyword_results_.verbatim_relevance = -1;
393       ConvertResultsToAutocompleteMatches();
394     }
395     if (FindTopMatch() == matches_.end()) {
396       // Guarantee that SearchProvider returns a legal default match.  (The
397       // omnibox always needs at least one legal default match, and it relies
398       // on SearchProvider to always return one.)
399       ApplyCalculatedRelevance();
400       ConvertResultsToAutocompleteMatches();
401     }
402     DCHECK(HasKeywordDefaultMatchInKeywordMode());
403     DCHECK(!IsTopMatchSearchWithURLInput());
404     DCHECK(FindTopMatch() != matches_.end());
405   }
406   UMA_HISTOGRAM_CUSTOM_COUNTS(
407       "Omnibox.SearchProviderMatches", matches_.size(), 1, 6, 7);
408 
409   const TemplateURL* keyword_url = providers_.GetKeywordProviderURL();
410   if ((keyword_url != NULL) && HasKeywordDefaultMatchInKeywordMode()) {
411     // If there is a keyword match that is allowed to be the default match,
412     // then prohibit default provider matches from being the default match lest
413     // such matches cause the user to break out of keyword mode.
414     for (ACMatches::iterator it = matches_.begin(); it != matches_.end();
415          ++it) {
416       if (it->keyword != keyword_url->keyword())
417         it->allowed_to_be_default_match = false;
418     }
419   }
420 
421   base::TimeTicks update_starred_start_time(base::TimeTicks::Now());
422   UpdateStarredStateOfMatches();
423   UMA_HISTOGRAM_TIMES("Omnibox.SearchProvider.UpdateStarredTime",
424                       base::TimeTicks::Now() - update_starred_start_time);
425   UpdateDone();
426 }
427 
Run()428 void SearchProvider::Run() {
429   // Start a new request with the current input.
430   suggest_results_pending_ = 0;
431   time_suggest_request_sent_ = base::TimeTicks::Now();
432 
433   default_fetcher_.reset(CreateSuggestFetcher(kDefaultProviderURLFetcherID,
434       providers_.GetDefaultProviderURL(), input_));
435   keyword_fetcher_.reset(CreateSuggestFetcher(kKeywordProviderURLFetcherID,
436       providers_.GetKeywordProviderURL(), keyword_input_));
437 
438   // Both the above can fail if the providers have been modified or deleted
439   // since the query began.
440   if (suggest_results_pending_ == 0) {
441     UpdateDone();
442     // We only need to update the listener if we're actually done.
443     if (done_)
444       listener_->OnProviderUpdate(false);
445   }
446 }
447 
DoHistoryQuery(bool minimal_changes)448 void SearchProvider::DoHistoryQuery(bool minimal_changes) {
449   // The history query results are synchronous, so if minimal_changes is true,
450   // we still have the last results and don't need to do anything.
451   if (minimal_changes)
452     return;
453 
454   keyword_history_results_.clear();
455   default_history_results_.clear();
456 
457   if (OmniboxFieldTrial::SearchHistoryDisable(
458       input_.current_page_classification()))
459     return;
460 
461   HistoryService* const history_service =
462       HistoryServiceFactory::GetForProfile(profile_, Profile::EXPLICIT_ACCESS);
463   history::URLDatabase* url_db = history_service ?
464       history_service->InMemoryDatabase() : NULL;
465   if (!url_db)
466     return;
467 
468   // Request history for both the keyword and default provider.  We grab many
469   // more matches than we'll ultimately clamp to so that if there are several
470   // recent multi-word matches who scores are lowered (see
471   // AddHistoryResultsToMap()), they won't crowd out older, higher-scoring
472   // matches.  Note that this doesn't fix the problem entirely, but merely
473   // limits it to cases with a very large number of such multi-word matches; for
474   // now, this seems OK compared with the complexity of a real fix, which would
475   // require multiple searches and tracking of "single- vs. multi-word" in the
476   // database.
477   int num_matches = kMaxMatches * 5;
478   const TemplateURL* default_url = providers_.GetDefaultProviderURL();
479   if (default_url) {
480     const base::TimeTicks start_time = base::TimeTicks::Now();
481     url_db->GetMostRecentKeywordSearchTerms(default_url->id(), input_.text(),
482         num_matches, &default_history_results_);
483     UMA_HISTOGRAM_TIMES(
484         "Omnibox.SearchProvider.GetMostRecentKeywordTermsDefaultProviderTime",
485         base::TimeTicks::Now() - start_time);
486   }
487   const TemplateURL* keyword_url = providers_.GetKeywordProviderURL();
488   if (keyword_url) {
489     url_db->GetMostRecentKeywordSearchTerms(keyword_url->id(),
490         keyword_input_.text(), num_matches, &keyword_history_results_);
491   }
492 }
493 
StartOrStopSuggestQuery(bool minimal_changes)494 void SearchProvider::StartOrStopSuggestQuery(bool minimal_changes) {
495   if (!IsQuerySuitableForSuggest()) {
496     StopSuggest();
497     ClearAllResults();
498     return;
499   }
500 
501   // For the minimal_changes case, if we finished the previous query and still
502   // have its results, or are allowed to keep running it, just do that, rather
503   // than starting a new query.
504   if (minimal_changes &&
505       (!default_results_.suggest_results.empty() ||
506        !default_results_.navigation_results.empty() ||
507        !keyword_results_.suggest_results.empty() ||
508        !keyword_results_.navigation_results.empty() ||
509        (!done_ && input_.want_asynchronous_matches())))
510     return;
511 
512   // We can't keep running any previous query, so halt it.
513   StopSuggest();
514 
515   // Remove existing results that cannot inline autocomplete the new input.
516   RemoveAllStaleResults();
517 
518   // Update the content classifications of remaining results so they look good
519   // against the current input.
520   UpdateMatchContentsClass(input_.text(), &default_results_);
521   if (!keyword_input_.text().empty())
522     UpdateMatchContentsClass(keyword_input_.text(), &keyword_results_);
523 
524   // We can't start a new query if we're only allowed synchronous results.
525   if (!input_.want_asynchronous_matches())
526     return;
527 
528   // To avoid flooding the suggest server, don't send a query until at
529   // least 100 ms since the last query.
530   base::TimeTicks next_suggest_time(time_suggest_request_sent_ +
531       base::TimeDelta::FromMilliseconds(kMinimumTimeBetweenSuggestQueriesMs));
532   base::TimeTicks now(base::TimeTicks::Now());
533   if (now >= next_suggest_time) {
534     Run();
535     return;
536   }
537   timer_.Start(FROM_HERE, next_suggest_time - now, this, &SearchProvider::Run);
538 }
539 
IsQuerySuitableForSuggest() const540 bool SearchProvider::IsQuerySuitableForSuggest() const {
541   // Don't run Suggest in incognito mode, if the engine doesn't support it, or
542   // if the user has disabled it.
543   const TemplateURL* default_url = providers_.GetDefaultProviderURL();
544   const TemplateURL* keyword_url = providers_.GetKeywordProviderURL();
545   if (profile_->IsOffTheRecord() ||
546       ((!default_url || default_url->suggestions_url().empty()) &&
547        (!keyword_url || keyword_url->suggestions_url().empty())) ||
548       !profile_->GetPrefs()->GetBoolean(prefs::kSearchSuggestEnabled))
549     return false;
550 
551   // If the input type might be a URL, we take extra care so that private data
552   // isn't sent to the server.
553 
554   // FORCED_QUERY means the user is explicitly asking us to search for this, so
555   // we assume it isn't a URL and/or there isn't private data.
556   if (input_.type() == metrics::OmniboxInputType::FORCED_QUERY)
557     return true;
558 
559   // Next we check the scheme.  If this is UNKNOWN/URL with a scheme that isn't
560   // http/https/ftp, we shouldn't send it.  Sending things like file: and data:
561   // is both a waste of time and a disclosure of potentially private, local
562   // data.  Other "schemes" may actually be usernames, and we don't want to send
563   // passwords.  If the scheme is OK, we still need to check other cases below.
564   // If this is QUERY, then the presence of these schemes means the user
565   // explicitly typed one, and thus this is probably a URL that's being entered
566   // and happens to currently be invalid -- in which case we again want to run
567   // our checks below.  Other QUERY cases are less likely to be URLs and thus we
568   // assume we're OK.
569   if (!LowerCaseEqualsASCII(input_.scheme(), url::kHttpScheme) &&
570       !LowerCaseEqualsASCII(input_.scheme(), url::kHttpsScheme) &&
571       !LowerCaseEqualsASCII(input_.scheme(), url::kFtpScheme))
572     return (input_.type() == metrics::OmniboxInputType::QUERY);
573 
574   // Don't send URLs with usernames, queries or refs.  Some of these are
575   // private, and the Suggest server is unlikely to have any useful results
576   // for any of them.  Also don't send URLs with ports, as we may initially
577   // think that a username + password is a host + port (and we don't want to
578   // send usernames/passwords), and even if the port really is a port, the
579   // server is once again unlikely to have and useful results.
580   // Note that we only block based on refs if the input is URL-typed, as search
581   // queries can legitimately have #s in them which the URL parser
582   // overaggressively categorizes as a url with a ref.
583   const url::Parsed& parts = input_.parts();
584   if (parts.username.is_nonempty() || parts.port.is_nonempty() ||
585       parts.query.is_nonempty() ||
586       (parts.ref.is_nonempty() &&
587        (input_.type() == metrics::OmniboxInputType::URL)))
588     return false;
589 
590   // Don't send anything for https except the hostname.  Hostnames are OK
591   // because they are visible when the TCP connection is established, but the
592   // specific path may reveal private information.
593   if (LowerCaseEqualsASCII(input_.scheme(), url::kHttpsScheme) &&
594       parts.path.is_nonempty())
595     return false;
596 
597   return true;
598 }
599 
RemoveAllStaleResults()600 void SearchProvider::RemoveAllStaleResults() {
601   if (keyword_input_.text().empty()) {
602     // User is either in keyword mode with a blank input or out of
603     // keyword mode entirely.
604     keyword_results_.Clear();
605   }
606 }
607 
ApplyCalculatedRelevance()608 void SearchProvider::ApplyCalculatedRelevance() {
609   ApplyCalculatedSuggestRelevance(&keyword_results_.suggest_results);
610   ApplyCalculatedSuggestRelevance(&default_results_.suggest_results);
611   ApplyCalculatedNavigationRelevance(&keyword_results_.navigation_results);
612   ApplyCalculatedNavigationRelevance(&default_results_.navigation_results);
613   default_results_.verbatim_relevance = -1;
614   keyword_results_.verbatim_relevance = -1;
615 }
616 
ApplyCalculatedSuggestRelevance(SuggestResults * list)617 void SearchProvider::ApplyCalculatedSuggestRelevance(SuggestResults* list) {
618   for (size_t i = 0; i < list->size(); ++i) {
619     SuggestResult& result = (*list)[i];
620     result.set_relevance(
621         result.CalculateRelevance(input_, providers_.has_keyword_provider()) +
622         (list->size() - i - 1));
623     result.set_relevance_from_server(false);
624   }
625 }
626 
ApplyCalculatedNavigationRelevance(NavigationResults * list)627 void SearchProvider::ApplyCalculatedNavigationRelevance(
628     NavigationResults* list) {
629   for (size_t i = 0; i < list->size(); ++i) {
630     NavigationResult& result = (*list)[i];
631     result.set_relevance(
632         result.CalculateRelevance(input_, providers_.has_keyword_provider()) +
633         (list->size() - i - 1));
634     result.set_relevance_from_server(false);
635   }
636 }
637 
CreateSuggestFetcher(int id,const TemplateURL * template_url,const AutocompleteInput & input)638 net::URLFetcher* SearchProvider::CreateSuggestFetcher(
639     int id,
640     const TemplateURL* template_url,
641     const AutocompleteInput& input) {
642   if (!template_url || template_url->suggestions_url().empty())
643     return NULL;
644 
645   // Bail if the suggestion URL is invalid with the given replacements.
646   TemplateURLRef::SearchTermsArgs search_term_args(input.text());
647   search_term_args.input_type = input.type();
648   search_term_args.cursor_position = input.cursor_position();
649   search_term_args.page_classification = input.current_page_classification();
650   if (OmniboxFieldTrial::EnableAnswersInSuggest())
651     search_term_args.session_token = GetSessionToken();
652   GURL suggest_url(template_url->suggestions_url_ref().ReplaceSearchTerms(
653       search_term_args,
654       providers_.template_url_service()->search_terms_data()));
655   if (!suggest_url.is_valid())
656     return NULL;
657   // Send the current page URL if user setting and URL requirements are met and
658   // the user is in the field trial.
659   if (CanSendURL(current_page_url_, suggest_url, template_url,
660                  input.current_page_classification(), profile_) &&
661       OmniboxFieldTrial::InZeroSuggestAfterTypingFieldTrial()) {
662     search_term_args.current_page_url = current_page_url_.spec();
663     // Create the suggest URL again with the current page URL.
664     suggest_url = GURL(template_url->suggestions_url_ref().ReplaceSearchTerms(
665         search_term_args,
666         providers_.template_url_service()->search_terms_data()));
667   }
668 
669   suggest_results_pending_++;
670   LogOmniboxSuggestRequest(REQUEST_SENT);
671 
672   net::URLFetcher* fetcher =
673       net::URLFetcher::Create(id, suggest_url, net::URLFetcher::GET, this);
674   fetcher->SetRequestContext(profile_->GetRequestContext());
675   fetcher->SetLoadFlags(net::LOAD_DO_NOT_SAVE_COOKIES);
676   // Add Chrome experiment state to the request headers.
677   net::HttpRequestHeaders headers;
678   chrome_variations::VariationsHttpHeaderProvider::GetInstance()->AppendHeaders(
679       fetcher->GetOriginalURL(), profile_->IsOffTheRecord(), false, &headers);
680   fetcher->SetExtraRequestHeaders(headers.ToString());
681   fetcher->Start();
682   return fetcher;
683 }
684 
ConvertResultsToAutocompleteMatches()685 void SearchProvider::ConvertResultsToAutocompleteMatches() {
686   // Convert all the results to matches and add them to a map, so we can keep
687   // the most relevant match for each result.
688   base::TimeTicks start_time(base::TimeTicks::Now());
689   MatchMap map;
690   const base::Time no_time;
691   int did_not_accept_keyword_suggestion =
692       keyword_results_.suggest_results.empty() ?
693       TemplateURLRef::NO_SUGGESTIONS_AVAILABLE :
694       TemplateURLRef::NO_SUGGESTION_CHOSEN;
695 
696   bool relevance_from_server;
697   int verbatim_relevance = GetVerbatimRelevance(&relevance_from_server);
698   int did_not_accept_default_suggestion =
699       default_results_.suggest_results.empty() ?
700       TemplateURLRef::NO_SUGGESTIONS_AVAILABLE :
701       TemplateURLRef::NO_SUGGESTION_CHOSEN;
702   if (verbatim_relevance > 0) {
703     const base::string16& trimmed_verbatim =
704         base::CollapseWhitespace(input_.text(), false);
705     SuggestResult verbatim(
706         trimmed_verbatim, AutocompleteMatchType::SEARCH_WHAT_YOU_TYPED,
707         trimmed_verbatim, base::string16(), base::string16(), base::string16(),
708         base::string16(), std::string(), std::string(), false,
709         verbatim_relevance, relevance_from_server, false,
710         trimmed_verbatim);
711     AddMatchToMap(verbatim, std::string(), did_not_accept_default_suggestion,
712                   false, &map);
713   }
714   if (!keyword_input_.text().empty()) {
715     const TemplateURL* keyword_url = providers_.GetKeywordProviderURL();
716     // We only create the verbatim search query match for a keyword
717     // if it's not an extension keyword.  Extension keywords are handled
718     // in KeywordProvider::Start().  (Extensions are complicated...)
719     // Note: in this provider, SEARCH_OTHER_ENGINE must correspond
720     // to the keyword verbatim search query.  Do not create other matches
721     // of type SEARCH_OTHER_ENGINE.
722     if (keyword_url &&
723         (keyword_url->GetType() != TemplateURL::OMNIBOX_API_EXTENSION)) {
724       bool keyword_relevance_from_server;
725       const int keyword_verbatim_relevance =
726           GetKeywordVerbatimRelevance(&keyword_relevance_from_server);
727       if (keyword_verbatim_relevance > 0) {
728         const base::string16& trimmed_verbatim =
729             base::CollapseWhitespace(keyword_input_.text(), false);
730         SuggestResult verbatim(
731             trimmed_verbatim, AutocompleteMatchType::SEARCH_OTHER_ENGINE,
732             trimmed_verbatim, base::string16(), base::string16(),
733             base::string16(), base::string16(), std::string(), std::string(),
734             true, keyword_verbatim_relevance, keyword_relevance_from_server,
735             false, trimmed_verbatim);
736         AddMatchToMap(verbatim, std::string(),
737                       did_not_accept_keyword_suggestion, false, &map);
738       }
739     }
740   }
741   AddHistoryResultsToMap(keyword_history_results_, true,
742                          did_not_accept_keyword_suggestion, &map);
743   AddHistoryResultsToMap(default_history_results_, false,
744                          did_not_accept_default_suggestion, &map);
745 
746   AddSuggestResultsToMap(keyword_results_.suggest_results,
747                          keyword_results_.metadata, &map);
748   AddSuggestResultsToMap(default_results_.suggest_results,
749                          default_results_.metadata, &map);
750 
751   ACMatches matches;
752   for (MatchMap::const_iterator i(map.begin()); i != map.end(); ++i)
753     matches.push_back(i->second);
754 
755   AddNavigationResultsToMatches(keyword_results_.navigation_results, &matches);
756   AddNavigationResultsToMatches(default_results_.navigation_results, &matches);
757 
758   // Now add the most relevant matches to |matches_|.  We take up to kMaxMatches
759   // suggest/navsuggest matches, regardless of origin.  If Instant Extended is
760   // enabled and we have server-provided (and thus hopefully more accurate)
761   // scores for some suggestions, we allow more of those, until we reach
762   // AutocompleteResult::kMaxMatches total matches (that is, enough to fill the
763   // whole popup).
764   //
765   // We will always return any verbatim matches, no matter how we obtained their
766   // scores, unless we have already accepted AutocompleteResult::kMaxMatches
767   // higher-scoring matches under the conditions above.
768   std::sort(matches.begin(), matches.end(), &AutocompleteMatch::MoreRelevant);
769   matches_.clear();
770 
771   size_t num_suggestions = 0;
772   for (ACMatches::const_iterator i(matches.begin());
773        (i != matches.end()) &&
774            (matches_.size() < AutocompleteResult::kMaxMatches);
775        ++i) {
776     // SEARCH_OTHER_ENGINE is only used in the SearchProvider for the keyword
777     // verbatim result, so this condition basically means "if this match is a
778     // suggestion of some sort".
779     if ((i->type != AutocompleteMatchType::SEARCH_WHAT_YOU_TYPED) &&
780         (i->type != AutocompleteMatchType::SEARCH_OTHER_ENGINE)) {
781       // If we've already hit the limit on non-server-scored suggestions, and
782       // this isn't a server-scored suggestion we can add, skip it.
783       if ((num_suggestions >= kMaxMatches) &&
784           (!chrome::IsInstantExtendedAPIEnabled() ||
785            (i->GetAdditionalInfo(kRelevanceFromServerKey) != kTrue))) {
786         continue;
787       }
788 
789       ++num_suggestions;
790     }
791 
792     matches_.push_back(*i);
793   }
794   UMA_HISTOGRAM_TIMES("Omnibox.SearchProvider.ConvertResultsTime",
795                       base::TimeTicks::Now() - start_time);
796 }
797 
FindTopMatch() const798 ACMatches::const_iterator SearchProvider::FindTopMatch() const {
799   ACMatches::const_iterator it = matches_.begin();
800   while ((it != matches_.end()) && !it->allowed_to_be_default_match)
801     ++it;
802   return it;
803 }
804 
HasKeywordDefaultMatchInKeywordMode() const805 bool SearchProvider::HasKeywordDefaultMatchInKeywordMode() const {
806   const TemplateURL* keyword_url = providers_.GetKeywordProviderURL();
807   // If the user is not in keyword mode, return true to say that this
808   // constraint is not violated.
809   if (keyword_url == NULL)
810     return true;
811   for (ACMatches::const_iterator it = matches_.begin(); it != matches_.end();
812        ++it) {
813     if ((it->keyword == keyword_url->keyword()) &&
814         it->allowed_to_be_default_match)
815       return true;
816   }
817   return false;
818 }
819 
IsTopMatchSearchWithURLInput() const820 bool SearchProvider::IsTopMatchSearchWithURLInput() const {
821   ACMatches::const_iterator first_match = FindTopMatch();
822   return (input_.type() == metrics::OmniboxInputType::URL) &&
823       (first_match != matches_.end()) &&
824       (first_match->relevance > CalculateRelevanceForVerbatim()) &&
825       (first_match->type != AutocompleteMatchType::NAVSUGGEST) &&
826       (first_match->type != AutocompleteMatchType::NAVSUGGEST_PERSONALIZED);
827 }
828 
AddNavigationResultsToMatches(const NavigationResults & navigation_results,ACMatches * matches)829 void SearchProvider::AddNavigationResultsToMatches(
830     const NavigationResults& navigation_results,
831     ACMatches* matches) {
832   for (NavigationResults::const_iterator it = navigation_results.begin();
833         it != navigation_results.end(); ++it) {
834     matches->push_back(NavigationToMatch(*it));
835     // In the absence of suggested relevance scores, use only the single
836     // highest-scoring result.  (The results are already sorted by relevance.)
837     if (!it->relevance_from_server())
838       return;
839   }
840 }
841 
AddHistoryResultsToMap(const HistoryResults & results,bool is_keyword,int did_not_accept_suggestion,MatchMap * map)842 void SearchProvider::AddHistoryResultsToMap(const HistoryResults& results,
843                                             bool is_keyword,
844                                             int did_not_accept_suggestion,
845                                             MatchMap* map) {
846   if (results.empty())
847     return;
848 
849   base::TimeTicks start_time(base::TimeTicks::Now());
850   bool prevent_inline_autocomplete = input_.prevent_inline_autocomplete() ||
851       (input_.type() == metrics::OmniboxInputType::URL);
852   const base::string16& input_text =
853       is_keyword ? keyword_input_.text() : input_.text();
854   bool input_multiple_words = HasMultipleWords(input_text);
855 
856   SuggestResults scored_results;
857   if (!prevent_inline_autocomplete && input_multiple_words) {
858     // ScoreHistoryResults() allows autocompletion of multi-word, 1-visit
859     // queries if the input also has multiple words.  But if we were already
860     // scoring a multi-word, multi-visit query aggressively, and the current
861     // input is still a prefix of it, then changing the suggestion suddenly
862     // feels wrong.  To detect this case, first score as if only one word has
863     // been typed, then check if the best result came from aggressive search
864     // history scoring.  If it did, then just keep that score set.  This
865     // 1200 the lowest possible score in CalculateRelevanceForHistory()'s
866     // aggressive-scoring curve.
867     scored_results = ScoreHistoryResults(results, prevent_inline_autocomplete,
868                                          false, input_text, is_keyword);
869     if ((scored_results.front().relevance() < 1200) ||
870         !HasMultipleWords(scored_results.front().suggestion()))
871       scored_results.clear();  // Didn't detect the case above, score normally.
872   }
873   if (scored_results.empty())
874     scored_results = ScoreHistoryResults(results, prevent_inline_autocomplete,
875                                          input_multiple_words, input_text,
876                                          is_keyword);
877   for (SuggestResults::const_iterator i(scored_results.begin());
878        i != scored_results.end(); ++i) {
879     AddMatchToMap(*i, std::string(), did_not_accept_suggestion, true, map);
880   }
881   UMA_HISTOGRAM_TIMES("Omnibox.SearchProvider.AddHistoryResultsTime",
882                       base::TimeTicks::Now() - start_time);
883 }
884 
ScoreHistoryResults(const HistoryResults & results,bool base_prevent_inline_autocomplete,bool input_multiple_words,const base::string16 & input_text,bool is_keyword)885 SearchProvider::SuggestResults SearchProvider::ScoreHistoryResults(
886     const HistoryResults& results,
887     bool base_prevent_inline_autocomplete,
888     bool input_multiple_words,
889     const base::string16& input_text,
890     bool is_keyword) {
891   AutocompleteClassifier* classifier =
892       AutocompleteClassifierFactory::GetForProfile(profile_);
893   SuggestResults scored_results;
894   const bool prevent_search_history_inlining =
895       OmniboxFieldTrial::SearchHistoryPreventInlining(
896           input_.current_page_classification());
897   const base::string16& trimmed_input =
898       base::CollapseWhitespace(input_text, false);
899   for (HistoryResults::const_iterator i(results.begin()); i != results.end();
900        ++i) {
901     const base::string16& trimmed_suggestion =
902         base::CollapseWhitespace(i->term, false);
903 
904     // Don't autocomplete multi-word queries that have only been seen once
905     // unless the user has typed more than one word.
906     bool prevent_inline_autocomplete = base_prevent_inline_autocomplete ||
907         (!input_multiple_words && (i->visits < 2) &&
908          HasMultipleWords(trimmed_suggestion));
909 
910     // Don't autocomplete search terms that would normally be treated as URLs
911     // when typed. For example, if the user searched for "google.com" and types
912     // "goog", don't autocomplete to the search term "google.com". Otherwise,
913     // the input will look like a URL but act like a search, which is confusing.
914     // NOTE: We don't check this in the following cases:
915     //  * When inline autocomplete is disabled, we won't be inline
916     //    autocompleting this term, so we don't need to worry about confusion as
917     //    much.  This also prevents calling Classify() again from inside the
918     //    classifier (which will corrupt state and likely crash), since the
919     //    classifier always disables inline autocomplete.
920     //  * When the user has typed the whole term, the "what you typed" history
921     //    match will outrank us for URL-like inputs anyway, so we need not do
922     //    anything special.
923     if (!prevent_inline_autocomplete && classifier &&
924         (trimmed_suggestion != trimmed_input)) {
925       AutocompleteMatch match;
926       classifier->Classify(trimmed_suggestion, false, false,
927                            input_.current_page_classification(), &match, NULL);
928       prevent_inline_autocomplete =
929           !AutocompleteMatch::IsSearchType(match.type);
930     }
931 
932     int relevance = CalculateRelevanceForHistory(
933         i->time, is_keyword, !prevent_inline_autocomplete,
934         prevent_search_history_inlining);
935     scored_results.push_back(SuggestResult(
936         trimmed_suggestion, AutocompleteMatchType::SEARCH_HISTORY,
937         trimmed_suggestion, base::string16(), base::string16(),
938         base::string16(), base::string16(), std::string(), std::string(),
939         is_keyword, relevance, false, false, trimmed_input));
940   }
941 
942   // History returns results sorted for us.  However, we may have docked some
943   // results' scores, so things are no longer in order.  Do a stable sort to get
944   // things back in order without otherwise disturbing results with equal
945   // scores, then force the scores to be unique, so that the order in which
946   // they're shown is deterministic.
947   std::stable_sort(scored_results.begin(), scored_results.end(),
948                    CompareScoredResults());
949   int last_relevance = 0;
950   for (SuggestResults::iterator i(scored_results.begin());
951        i != scored_results.end(); ++i) {
952     if ((i != scored_results.begin()) && (i->relevance() >= last_relevance))
953       i->set_relevance(last_relevance - 1);
954     last_relevance = i->relevance();
955   }
956 
957   return scored_results;
958 }
959 
AddSuggestResultsToMap(const SuggestResults & results,const std::string & metadata,MatchMap * map)960 void SearchProvider::AddSuggestResultsToMap(const SuggestResults& results,
961                                             const std::string& metadata,
962                                             MatchMap* map) {
963   for (size_t i = 0; i < results.size(); ++i)
964     AddMatchToMap(results[i], metadata, i, false, map);
965 }
966 
GetVerbatimRelevance(bool * relevance_from_server) const967 int SearchProvider::GetVerbatimRelevance(bool* relevance_from_server) const {
968   // Use the suggested verbatim relevance score if it is non-negative (valid),
969   // if inline autocomplete isn't prevented (always show verbatim on backspace),
970   // and if it won't suppress verbatim, leaving no default provider matches.
971   // Otherwise, if the default provider returned no matches and was still able
972   // to suppress verbatim, the user would have no search/nav matches and may be
973   // left unable to search using their default provider from the omnibox.
974   // Check for results on each verbatim calculation, as results from older
975   // queries (on previous input) may be trimmed for failing to inline new input.
976   bool use_server_relevance =
977       (default_results_.verbatim_relevance >= 0) &&
978       !input_.prevent_inline_autocomplete() &&
979       ((default_results_.verbatim_relevance > 0) ||
980        !default_results_.suggest_results.empty() ||
981        !default_results_.navigation_results.empty());
982   if (relevance_from_server)
983     *relevance_from_server = use_server_relevance;
984   return use_server_relevance ?
985       default_results_.verbatim_relevance : CalculateRelevanceForVerbatim();
986 }
987 
CalculateRelevanceForVerbatim() const988 int SearchProvider::CalculateRelevanceForVerbatim() const {
989   if (!providers_.keyword_provider().empty())
990     return 250;
991   return CalculateRelevanceForVerbatimIgnoringKeywordModeState();
992 }
993 
994 int SearchProvider::
CalculateRelevanceForVerbatimIgnoringKeywordModeState() const995     CalculateRelevanceForVerbatimIgnoringKeywordModeState() const {
996   switch (input_.type()) {
997     case metrics::OmniboxInputType::UNKNOWN:
998     case metrics::OmniboxInputType::QUERY:
999     case metrics::OmniboxInputType::FORCED_QUERY:
1000       return kNonURLVerbatimRelevance;
1001 
1002     case metrics::OmniboxInputType::URL:
1003       return 850;
1004 
1005     default:
1006       NOTREACHED();
1007       return 0;
1008   }
1009 }
1010 
GetKeywordVerbatimRelevance(bool * relevance_from_server) const1011 int SearchProvider::GetKeywordVerbatimRelevance(
1012     bool* relevance_from_server) const {
1013   // Use the suggested verbatim relevance score if it is non-negative (valid),
1014   // if inline autocomplete isn't prevented (always show verbatim on backspace),
1015   // and if it won't suppress verbatim, leaving no keyword provider matches.
1016   // Otherwise, if the keyword provider returned no matches and was still able
1017   // to suppress verbatim, the user would have no search/nav matches and may be
1018   // left unable to search using their keyword provider from the omnibox.
1019   // Check for results on each verbatim calculation, as results from older
1020   // queries (on previous input) may be trimmed for failing to inline new input.
1021   bool use_server_relevance =
1022       (keyword_results_.verbatim_relevance >= 0) &&
1023       !input_.prevent_inline_autocomplete() &&
1024       ((keyword_results_.verbatim_relevance > 0) ||
1025        !keyword_results_.suggest_results.empty() ||
1026        !keyword_results_.navigation_results.empty());
1027   if (relevance_from_server)
1028     *relevance_from_server = use_server_relevance;
1029   return use_server_relevance ?
1030       keyword_results_.verbatim_relevance :
1031       CalculateRelevanceForKeywordVerbatim(keyword_input_.type(),
1032                                            keyword_input_.prefer_keyword());
1033 }
1034 
CalculateRelevanceForHistory(const base::Time & time,bool is_keyword,bool use_aggressive_method,bool prevent_search_history_inlining) const1035 int SearchProvider::CalculateRelevanceForHistory(
1036     const base::Time& time,
1037     bool is_keyword,
1038     bool use_aggressive_method,
1039     bool prevent_search_history_inlining) const {
1040   // The relevance of past searches falls off over time. There are two distinct
1041   // equations used. If the first equation is used (searches to the primary
1042   // provider that we want to score aggressively), the score is in the range
1043   // 1300-1599 (unless |prevent_search_history_inlining|, in which case
1044   // it's in the range 1200-1299). If the second equation is used the
1045   // relevance of a search 15 minutes ago is discounted 50 points, while the
1046   // relevance of a search two weeks ago is discounted 450 points.
1047   double elapsed_time = std::max((base::Time::Now() - time).InSecondsF(), 0.0);
1048   bool is_primary_provider = is_keyword || !providers_.has_keyword_provider();
1049   if (is_primary_provider && use_aggressive_method) {
1050     // Searches with the past two days get a different curve.
1051     const double autocomplete_time = 2 * 24 * 60 * 60;
1052     if (elapsed_time < autocomplete_time) {
1053       int max_score = is_keyword ? 1599 : 1399;
1054       if (prevent_search_history_inlining)
1055         max_score = 1299;
1056       return max_score - static_cast<int>(99 *
1057           std::pow(elapsed_time / autocomplete_time, 2.5));
1058     }
1059     elapsed_time -= autocomplete_time;
1060   }
1061 
1062   const int score_discount =
1063       static_cast<int>(6.5 * std::pow(elapsed_time, 0.3));
1064 
1065   // Don't let scores go below 0.  Negative relevance scores are meaningful in
1066   // a different way.
1067   int base_score;
1068   if (is_primary_provider)
1069     base_score = (input_.type() == metrics::OmniboxInputType::URL) ? 750 : 1050;
1070   else
1071     base_score = 200;
1072   return std::max(0, base_score - score_discount);
1073 }
1074 
NavigationToMatch(const NavigationResult & navigation)1075 AutocompleteMatch SearchProvider::NavigationToMatch(
1076     const NavigationResult& navigation) {
1077   base::string16 input;
1078   const bool trimmed_whitespace = base::TrimWhitespace(
1079       navigation.from_keyword_provider() ?
1080           keyword_input_.text() : input_.text(),
1081       base::TRIM_TRAILING, &input) != base::TRIM_NONE;
1082   AutocompleteMatch match(this, navigation.relevance(), false,
1083                           navigation.type());
1084   match.destination_url = navigation.url();
1085   BaseSearchProvider::SetDeletionURL(navigation.deletion_url(), &match);
1086   // First look for the user's input inside the formatted url as it would be
1087   // without trimming the scheme, so we can find matches at the beginning of the
1088   // scheme.
1089   const URLPrefix* prefix =
1090       URLPrefix::BestURLPrefix(navigation.formatted_url(), input);
1091   size_t match_start = (prefix == NULL) ?
1092       navigation.formatted_url().find(input) : prefix->prefix.length();
1093   bool trim_http = !AutocompleteInput::HasHTTPScheme(input) &&
1094       (!prefix || (match_start != 0));
1095   const net::FormatUrlTypes format_types =
1096       net::kFormatUrlOmitAll & ~(trim_http ? 0 : net::kFormatUrlOmitHTTP);
1097 
1098   const std::string languages(
1099       profile_->GetPrefs()->GetString(prefs::kAcceptLanguages));
1100   size_t inline_autocomplete_offset = (prefix == NULL) ?
1101       base::string16::npos : (match_start + input.length());
1102   match.fill_into_edit +=
1103       AutocompleteInput::FormattedStringWithEquivalentMeaning(navigation.url(),
1104           net::FormatUrl(navigation.url(), languages, format_types,
1105                          net::UnescapeRule::SPACES, NULL, NULL,
1106                          &inline_autocomplete_offset));
1107   // Preserve the forced query '?' prefix in |match.fill_into_edit|.
1108   // Otherwise, user edits to a suggestion would show non-Search results.
1109   if (input_.type() == metrics::OmniboxInputType::FORCED_QUERY) {
1110     match.fill_into_edit.insert(0, base::ASCIIToUTF16("?"));
1111     if (inline_autocomplete_offset != base::string16::npos)
1112       ++inline_autocomplete_offset;
1113   }
1114   if (inline_autocomplete_offset != base::string16::npos) {
1115     DCHECK(inline_autocomplete_offset <= match.fill_into_edit.length());
1116     match.inline_autocompletion =
1117         match.fill_into_edit.substr(inline_autocomplete_offset);
1118   }
1119   // An inlineable navsuggestion can only be the default match when there
1120   // is no keyword provider active, lest it appear first and break the user
1121   // out of keyword mode.  It can also only be default if either the inline
1122   // autocompletion is empty or we're not preventing inline autocompletion.
1123   // Finally, if we have an inlineable navsuggestion with an inline completion
1124   // that we're not preventing, make sure we didn't trim any whitespace.
1125   // We don't want to claim http://foo.com/bar is inlineable against the
1126   // input "foo.com/b ".
1127   match.allowed_to_be_default_match = navigation.IsInlineable(input) &&
1128       (providers_.GetKeywordProviderURL() == NULL) &&
1129       (match.inline_autocompletion.empty() ||
1130       (!input_.prevent_inline_autocomplete() && !trimmed_whitespace));
1131 
1132   match.contents = navigation.match_contents();
1133   match.contents_class = navigation.match_contents_class();
1134   match.description = navigation.description();
1135   AutocompleteMatch::ClassifyMatchInString(input, match.description,
1136       ACMatchClassification::NONE, &match.description_class);
1137 
1138   match.RecordAdditionalInfo(
1139       kRelevanceFromServerKey,
1140       navigation.relevance_from_server() ? kTrue : kFalse);
1141   match.RecordAdditionalInfo(kShouldPrefetchKey, kFalse);
1142 
1143   return match;
1144 }
1145 
UpdateDone()1146 void SearchProvider::UpdateDone() {
1147   // We're done when the timer isn't running, there are no suggest queries
1148   // pending, and we're not waiting on Instant.
1149   done_ = !timer_.IsRunning() && (suggest_results_pending_ == 0);
1150 }
1151 
GetSessionToken()1152 std::string SearchProvider::GetSessionToken() {
1153   base::TimeTicks current_time(base::TimeTicks::Now());
1154   // Renew token if it expired.
1155   if (current_time > token_expiration_time_) {
1156     const size_t kTokenBytes = 12;
1157     std::string raw_data;
1158     base::RandBytes(WriteInto(&raw_data, kTokenBytes + 1), kTokenBytes);
1159     base::Base64Encode(raw_data, &current_token_);
1160 
1161     // Make the base64 encoded value URL and filename safe(see RFC 3548).
1162     std::replace(current_token_.begin(), current_token_.end(), '+', '-');
1163     std::replace(current_token_.begin(), current_token_.end(), '/', '_');
1164   }
1165 
1166   // Extend expiration time another 60 seconds.
1167   token_expiration_time_ = current_time + base::TimeDelta::FromSeconds(60);
1168 
1169   return current_token_;
1170 }
1171