Fix credit card table migration step.

chrome/browser/webdata/web_database.cc

 // Copyright (c) 2010 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 "chrome/browser/webdata/web_database.h"
 
 #include <algorithm>
 #include <limits>
 #include <set>
 #include <string>
@@ skipping 428 matching lines @@
 
   // Version check.
   if (!meta_table_.Init(&db_, kCurrentVersionNumber, kCompatibleVersionNumber))
     return sql::INIT_FAILURE;
   if (meta_table_.GetCompatibleVersionNumber() > kCurrentVersionNumber) {
     LOG(WARNING) << "Web database is too new.";
     return sql::INIT_TOO_NEW;
   }
 
   // Initialize the tables.
+  bool credit_card_table_created = false;
   if (!InitKeywordsTable() || !InitLoginsTable() || !InitWebAppIconsTable() ||
       !InitWebAppsTable() || !InitAutofillTable() ||
       !InitAutofillDatesTable() || !InitAutoFillProfilesTable() ||
-      !InitCreditCardsTable() || !InitTokenServiceTable()) {
+      !InitCreditCardsTable(&credit_card_table_created) ||
+      !InitTokenServiceTable()) {
     LOG(WARNING) << "Unable to initialize the web database.";
     return sql::INIT_FAILURE;
   }
 
   // If the file on disk is an older database version, bring it up to date.
-  MigrateOldVersionsAsNeeded();
+  MigrateOldVersionsAsNeeded(credit_card_table_created);
 
   return transaction.Commit() ? sql::INIT_OK : sql::INIT_FAILURE;
 }
 
 bool WebDatabase::SetWebAppImage(const GURL& url, const SkBitmap& image) {
   // Don't bother with a cached statement since this will be a relatively
   // infrequent operation.
   sql::Statement s(db_.GetUniqueStatement(
       "INSERT OR REPLACE INTO web_app_icons "
       "(url, width, height, image) VALUES (?, ?, ?, ?)"));
@@ skipping 283 matching lines @@
     }
     if (!db_.Execute("CREATE INDEX autofill_profiles_label_index "
                      "ON autofill_profiles (label)")) {
       NOTREACHED();
       return false;
     }
   }
   return true;
 }
 
-bool WebDatabase::InitCreditCardsTable() {
+bool WebDatabase::InitCreditCardsTable(bool* table_was_created) {
   if (!db_.DoesTableExist("credit_cards")) {
     if (!db_.Execute("CREATE TABLE credit_cards ( "
                      "label VARCHAR, "
                      "unique_id INTEGER PRIMARY KEY, "
                      "name_on_card VARCHAR, "
                      "type VARCHAR, "
                      "card_number VARCHAR, "
                      "expiration_month INTEGER, "
                      "expiration_year INTEGER, "
                      "verification_code VARCHAR, "
                      "billing_address VARCHAR, "
                      "shipping_address VARCHAR, "
                      "card_number_encrypted BLOB, "
                      "verification_code_encrypted BLOB)")) {
       NOTREACHED();
       return false;
     }
+    *table_was_created = true;
     if (!db_.Execute("CREATE INDEX credit_cards_label_index "
                      "ON credit_cards (label)")) {
       NOTREACHED();
       return false;
     }
   }
   return true;
 }
 
 bool WebDatabase::InitWebAppIconsTable() {
@@ skipping 1073 matching lines @@
     return false;
   }
   s.BindInt64(0, pair_id);
   if (s.Run()) {
     return RemoveFormElementForTimeRange(pair_id, base::Time(), base::Time(),
                                          NULL);
   }
   return false;
 }
 
-void WebDatabase::MigrateOldVersionsAsNeeded() {
+void WebDatabase::MigrateOldVersionsAsNeeded(bool credit_card_table_created) {
   // Migrate if necessary.
   int current_version = meta_table_.GetVersionNumber();
   switch (current_version) {
     // Versions 1 - 19 are unhandled.  Version numbers greater than
     // kCurrentVersionNumber should have already been weeded out by the caller.
     default:
       // When the version is too old, we just try to continue anyway.  There
       // should not be a released product that makes a database too old for us
       // to handle.
       LOG(WARNING) << "Web database version " << current_version <<
@@ skipping 16 matching lines @@
     case 21:
       if (!ClearAutofillEmptyValueElements()) {
         NOTREACHED() << "Failed to clean-up autofill DB.";
       }
       meta_table_.SetVersionNumber(22);
       // No change in the compatibility version number.
 
       // FALL THROUGH
 
     case 22:
-      // Add the card_number_encrypted column.
-      if (!db_.Execute("ALTER TABLE credit_cards ADD COLUMN "
-                       "card_number_encrypted BLOB DEFAULT NULL")) {
-          NOTREACHED();
-          LOG(WARNING) << "Unable to update web database to version 23.";
-          return;
-      }
-      if (!db_.Execute("ALTER TABLE credit_cards ADD COLUMN "
-                       "verification_code_encrypted BLOB DEFAULT NULL")) {
-          NOTREACHED();
-          LOG(WARNING) << "Unable to update web database to version 23.";
-          return;
+      // Add the card_number_encrypted column if credit card table was not
+      // created in this build.
+      if (!credit_card_table_created) {
+        if (!db_.Execute("ALTER TABLE credit_cards ADD COLUMN "
+                         "card_number_encrypted BLOB DEFAULT NULL")) {
+            NOTREACHED();
+            LOG(WARNING) << "Unable to update web database to version 23.";
+            return;
+        }
+        if (!db_.Execute("ALTER TABLE credit_cards ADD COLUMN "
+                         "verification_code_encrypted BLOB DEFAULT NULL")) {
+            NOTREACHED();
+            LOG(WARNING) << "Unable to update web database to version 23.";
+            return;
+        }
       }
       meta_table_.SetVersionNumber(23);
 
       // FALL THROUGH
 
     case 23: {
       // One-time cleanup for Chromium bug 38364.  In the presence of
       // multi-byte UTF-8 characters, that bug could cause AutoFill strings
       // to grow larger and more corrupt with each save.  The cleanup removes
       // any row with a string field larger than a reasonable size.  The string
@@ skipping 65 matching lines @@
 
     // Add successive versions here.  Each should set the version number and
     // compatible version number as appropriate, then fall through to the next
     // case.
 
     case kCurrentVersionNumber:
       // No migration needed.
       return;
   }
 }