Reland 37913. Clear local state on exit.

webkit/database/database_tracker.cc

-// Copyright (c) 2009 The Chromium Authors. All rights reserved.
+// 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 "webkit/database/database_tracker.h"
 
 #include <vector>
 
 #include "app/sql/connection.h"
 #include "app/sql/meta_table.h"
 #include "app/sql/statement.h"
 #include "app/sql/transaction.h"
 #include "base/basictypes.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/string_util.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebSecurityOrigin.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebString.h"
 #include "webkit/database/databases_table.h"
 #include "webkit/database/quota_table.h"
+#include "webkit/glue/webkit_glue.h"
 
 namespace webkit_database {
 
 const FilePath::CharType kDatabaseDirectoryName[] =
     FILE_PATH_LITERAL("databases");
 const FilePath::CharType kTrackerDatabaseFileName[] =
     FILE_PATH_LITERAL("Databases.db");
 const int kCurrentVersion = 2;
 const int kCompatibleVersion = 1;
 const int64 kDefaultExtensionQuota = 50 * 1024 * 1024;
@@ skipping 295 matching lines @@
 int64 DatabaseTracker::UpdateCachedDatabaseFileSize(
     const string16& origin_identifier,
     const string16& database_name) {
   int64 new_size = GetDBFileSize(origin_identifier, database_name);
   CachedOriginInfo* origin_info = GetCachedOriginInfo(origin_identifier);
   if (origin_info)
     origin_info->SetDatabaseSize(database_name, new_size);
   return new_size;
 }
 
+// static
+void DatabaseTracker::ClearLocalState(const FilePath& profile_path,
+                                      const char* url_scheme_to_be_skipped) {
+  FilePath db_dir = profile_path.Append(FilePath(kDatabaseDirectoryName));
+  FilePath db_tracker = db_dir.Append(FilePath(kTrackerDatabaseFileName));
+  if (file_util::DirectoryExists(db_dir) &&
+      file_util::PathExists(db_tracker)) {
+    scoped_ptr<sql::Connection> db_(new sql::Connection);
+    if (!db_->Open(db_tracker) ||
+        !db_->DoesTableExist("Databases")) {
+      db_->Close();
+      file_util::Delete(db_dir, true);
+      return;
+    } else {
+      sql::Statement delete_statement(db_->GetCachedStatement(
+            SQL_FROM_HERE, "DELETE FROM Databases WHERE origin NOT LIKE ?"));
+      std::string filter(url_scheme_to_be_skipped);
+      filter += "_%";
+      delete_statement.BindString(0, filter);
+      if (!delete_statement.Run()) {
+        db_->Close();
+        file_util::Delete(db_dir, true);
+        return;
+      }
+    }
+  }
+  file_util::FileEnumerator file_enumerator(db_dir, false,
+      file_util::FileEnumerator::DIRECTORIES);
+  for (FilePath file_path = file_enumerator.Next(); !file_path.empty();
+       file_path = file_enumerator.Next()) {
+    if (file_path.BaseName() != FilePath(kTrackerDatabaseFileName)) {
+      scoped_ptr<WebKit::WebSecurityOrigin> web_security_origin(
+          WebKit::WebSecurityOrigin::createFromDatabaseIdentifier(
+              webkit_glue::FilePathToWebString(file_path.BaseName())));
+      if (!EqualsASCII(web_security_origin->protocol(),
+                       url_scheme_to_be_skipped))
+        file_util::Delete(file_path, true);
+    }
+  }
+}
+
 }  // namespace webkit_database