Implement GetOrigins for indexeddb quota

content/browser/in_process_webkit/indexed_db_context.cc

 // Copyright (c) 2011 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 "content/browser/in_process_webkit/indexed_db_context.h"
 
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/message_loop_proxy.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "content/browser/browser_thread.h"
 #include "content/browser/in_process_webkit/indexed_db_quota_client.h"
 #include "content/browser/in_process_webkit/webkit_context.h"
+#include "content/common/url_constants.h"
 #include "googleurl/src/gurl.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebCString.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebIDBDatabase.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebIDBFactory.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebSecurityOrigin.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebString.h"
 #include "webkit/glue/webkit_glue.h"
 #include "webkit/quota/quota_manager.h"
 #include "webkit/quota/special_storage_policy.h"
 
@@ skipping 86 matching lines @@
   FilePath idb_file = GetIndexedDBFilePath(origin_id);
   DCHECK_EQ(idb_file.BaseName().value().substr(0, strlen("chrome-extension")),
             FILE_PATH_LITERAL("chrome-extension"));
   DeleteIndexedDBFile(GetIndexedDBFilePath(origin_id));
 }
 
 bool IndexedDBContext::IsUnlimitedStorageGranted(
     const GURL& origin) const {
   return special_storage_policy_->IsStorageUnlimited(origin);
 }
+
+// TODO(dgrogan): Merge this code with the similar loop in
+// BrowsingDataIndexedDBHelperImpl::FetchIndexedDBInfoInWebKitThread.
+void IndexedDBContext::GetAllOriginIdentifiers(
+    std::vector<string16>* origin_ids) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::WEBKIT));
+  file_util::FileEnumerator file_enumerator(data_path_,
+      false, file_util::FileEnumerator::DIRECTORIES);
+  for (FilePath file_path = file_enumerator.Next(); !file_path.empty();
+       file_path = file_enumerator.Next()) {
+    if (file_path.Extension() == IndexedDBContext::kIndexedDBExtension) {
+      WebKit::WebString origin_id_webstring =
+          webkit_glue::FilePathToWebString(file_path.BaseName());
+      string16 origin_id = origin_id_webstring;
+      WebSecurityOrigin web_security_origin =
+          WebSecurityOrigin::createFromDatabaseIdentifier(origin_id_webstring);
+      if (EqualsASCII(web_security_origin.protocol(),
+                      chrome::kExtensionScheme)) {

[michaeln, 2011/07/07 21:33:56]

I don't think extension origins should be filtered out here. See
DatabaseTracker::GetAllOriginIdentifiers(origin_ids) for example, similar method
used for the same purposes, but extension origins are not special cased.

[dgrogan, 2011/07/07 21:55:37]

Done.

+        // Extension state is not considered browsing data.
+        continue;
+      }
+      origin_ids->push_back(origin_id);
+    }
+  }
+}