[Autofill/Autocomplete Feature] Substring matching instead of prefix matching.

components/autofill/core/browser/webdata/autofill_table.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/autofill/core/browser/webdata/autofill_table.h"
 
 #include <algorithm>
 #include <cmath>
 #include <limits>
 #include <map>
 #include <set>
 #include <string>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/guid.h"
 #include "base/i18n/case_conversion.h"
 #include "base/logging.h"
 #include "base/numerics/safe_conversions.h"
 #include "base/strings/string_number_conversions.h"
+#include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
 #include "components/autofill/core/browser/autofill_country.h"
 #include "components/autofill/core/browser/autofill_profile.h"
 #include "components/autofill/core/browser/autofill_type.h"
 #include "components/autofill/core/browser/credit_card.h"
 #include "components/autofill/core/browser/personal_data_manager.h"
 #include "components/autofill/core/browser/webdata/autofill_change.h"
 #include "components/autofill/core/browser/webdata/autofill_entry.h"
 #include "components/autofill/core/common/autofill_switches.h"
+#include "components/autofill/core/common/autofill_util.h"
 #include "components/autofill/core/common/form_field_data.h"
 #include "components/os_crypt/os_crypt.h"
 #include "components/webdata/common/web_database.h"
 #include "sql/statement.h"
 #include "sql/transaction.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "url/gurl.h"
 
 using base::ASCIIToUTF16;
 using base::Time;
@@ skipping 320 matching lines @@
 }
 
 CreditCard::ServerStatus ServerStatusStringToEnum(const std::string& status) {
   if (status == "EXPIRED")
     return CreditCard::EXPIRED;
 
   DCHECK_EQ("OK", status);
   return CreditCard::OK;
 }
 
+// Returns |s| with |escaper| in front of each of occurrence of a character from
+// |special_chars|. Any occurrence of |escaper| in |s| is doubled. For example,
+// Substitute("hello_world!", "_%", '!'') returns "hello!_world!!".
+base::string16 Substitute(const base::string16& s,
+                          const base::string16& special_chars,
+                          const base::char16& escaper) {
+  // Prepend |escaper| to the list of |special_chars|.
+  base::string16 escape_wildcards(special_chars);
+  escape_wildcards.insert(escape_wildcards.begin(), escaper);
+
+  // Prepend the |escaper| just before |special_chars| in |s|.
+  base::string16 result(s);
+  for (base::char16 c : escape_wildcards) {
+    for (size_t pos = 0; (pos = result.find(c, pos)) != base::string16::npos;
+         pos += 2) {
+      result.insert(result.begin() + pos, escaper);
+    }
+  }
+
+  return result;
+}
+
 }  // namespace
 
 // The maximum length allowed for form data.
 const size_t AutofillTable::kMaxDataLength = 1024;
 
 AutofillTable::AutofillTable(const std::string& app_locale)
     : app_locale_(app_locale) {
 }
 
 AutofillTable::~AutofillTable() {
@@ skipping 68 matching lines @@
                                       std::vector<AutofillChange>* changes) {
   return AddFormFieldValueTime(element, changes, Time::Now());
 }
 
 bool AutofillTable::GetFormValuesForElementName(
     const base::string16& name,
     const base::string16& prefix,
     std::vector<base::string16>* values,
     int limit) {
   DCHECK(values);
-  sql::Statement s;
+  bool succeeded = false;
 
   if (prefix.empty()) {
+    sql::Statement s;
     s.Assign(db_->GetUniqueStatement(
         "SELECT value FROM autofill "
         "WHERE name = ? "
         "ORDER BY count DESC "
         "LIMIT ?"));
     s.BindString16(0, name);
     s.BindInt(1, limit);
+
+    values->clear();
+    while (s.Step())
+      values->push_back(s.ColumnString16(0));
+
+    succeeded = s.Succeeded();
   } else {
     base::string16 prefix_lower = base::i18n::ToLower(prefix);
     base::string16 next_prefix = prefix_lower;
     next_prefix[next_prefix.length() - 1]++;
 
-    s.Assign(db_->GetUniqueStatement(
+    sql::Statement s1;
+    s1.Assign(db_->GetUniqueStatement(
         "SELECT value FROM autofill "
         "WHERE name = ? AND "
         "value_lower >= ? AND "
         "value_lower < ? "
         "ORDER BY count DESC "
         "LIMIT ?"));
-    s.BindString16(0, name);
-    s.BindString16(1, prefix_lower);
-    s.BindString16(2, next_prefix);
-    s.BindInt(3, limit);
+    s1.BindString16(0, name);
+    s1.BindString16(1, prefix_lower);
+    s1.BindString16(2, next_prefix);
+    s1.BindInt(3, limit);
+
+    values->clear();
+    while (s1.Step())
+      values->push_back(s1.ColumnString16(0));
+
+    succeeded = s1.Succeeded();
+
+    if (IsFeatureSubstringMatchEnabled()) {
+      sql::Statement s2;
+      s2.Assign(db_->GetUniqueStatement(
+          "SELECT value FROM autofill "
+          "WHERE name = ? AND ("
+          " value LIKE '% ' || :prefix || '%' ESCAPE '!' OR "
+          " value LIKE '%.' || :prefix || '%' ESCAPE '!' OR "
+          " value LIKE '%,' || :prefix || '%' ESCAPE '!' OR "
+          " value LIKE '%-' || :prefix || '%' ESCAPE '!' OR "
+          " value LIKE '%@' || :prefix || '%' ESCAPE '!' OR "
+          " value LIKE '%!_' || :prefix || '%' ESCAPE '!' ) "
+          "ORDER BY count DESC "
+          "LIMIT ?"));
+
+      s2.BindString16(0, name);
+      // escaper as L'!' -> 0x21.
+      s2.BindString16(1, Substitute(prefix_lower, ASCIIToUTF16("_%"), 0x21));
+      s2.BindInt(2, limit);
+      while (s2.Step())
+        values->push_back(s2.ColumnString16(0));
+
+      succeeded &= s2.Succeeded();
+    }
   }
 
-  values->clear();
-  while (s.Step())
-    values->push_back(s.ColumnString16(0));
-  return s.Succeeded();
+  return succeeded;
 }
 
 bool AutofillTable::HasFormElements() {
   sql::Statement s(db_->GetUniqueStatement("SELECT COUNT(*) FROM autofill"));
   if (!s.Step()) {
     NOTREACHED();
     return false;
   }
   return s.ColumnInt(0) > 0;
 }
@@ skipping 1691 matching lines @@
     insert.BindString16(index++, profile.GetRawInfo(PHONE_HOME_WHOLE_NUMBER));
     insert.BindString(index++, profile.language_code());
     insert.Run();
     insert.Reset(true);
   }
 
   return transaction.Commit();
 }
 
 }  // namespace autofill