Persist sessionStorage on disk.

webkit/dom_storage/dom_storage_context.cc

 // Copyright (c) 2012 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/dom_storage/dom_storage_context.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/file_util.h"
 #include "base/location.h"
 #include "base/time.h"
 #include "webkit/dom_storage/dom_storage_area.h"
+#include "webkit/dom_storage/dom_storage_database.h"
 #include "webkit/dom_storage/dom_storage_namespace.h"
 #include "webkit/dom_storage/dom_storage_task_runner.h"
 #include "webkit/dom_storage/dom_storage_types.h"
+#include "webkit/dom_storage/session_storage_database.h"
 #include "webkit/quota/special_storage_policy.h"
 
 using file_util::FileEnumerator;
 
 namespace dom_storage {
 
 DomStorageContext::UsageInfo::UsageInfo() : data_size(0) {}
 DomStorageContext::UsageInfo::~UsageInfo() {}
 
 DomStorageContext::DomStorageContext(
     const FilePath& localstorage_directory,
     const FilePath& sessionstorage_directory,
     quota::SpecialStoragePolicy* special_storage_policy,
     DomStorageTaskRunner* task_runner)
     : localstorage_directory_(localstorage_directory),
       sessionstorage_directory_(sessionstorage_directory),
       task_runner_(task_runner),
       is_shutdown_(false),
       clear_local_state_(false),
       save_session_state_(false),
       special_storage_policy_(special_storage_policy) {
   // AtomicSequenceNum starts at 0 but we want to start session
   // namespace ids at one since zero is reserved for the
   // kLocalStorageNamespaceId.
   session_id_sequence_.GetNext();
+  if (!sessionstorage_directory_.empty()) {
+    session_storage_database_ = new SessionStorageDatabase(
+        sessionstorage_directory_,
+        base::Bind(&DomStorageContext::NotifySessionStorageNamespaceAssociated,
+                   this));
+  }
 }
 
 DomStorageContext::~DomStorageContext() {
 }
 
 DomStorageNamespace* DomStorageContext::GetStorageNamespace(
     int64 namespace_id) {
   if (is_shutdown_)
     return NULL;
   StorageNamespaceMap::iterator found = namespaces_.find(namespace_id);
@@ skipping 29 matching lines @@
       info.origin = DomStorageArea::OriginFromDatabaseFileName(path);
       if (include_file_info) {
         FileEnumerator::FindInfo find_info;
         enumerator.GetFindInfo(&find_info);
         info.data_size = FileEnumerator::GetFilesize(find_info);
         info.last_modified = FileEnumerator::GetLastModifiedTime(find_info);
       }
       infos->push_back(info);
     }
   }
+  // FIXME(marja): Get usage infos for sessionStorage (crbug.com/123599).
 }
 
 void DomStorageContext::DeleteOrigin(const GURL& origin) {
   DCHECK(!is_shutdown_);
   DomStorageNamespace* local = GetStorageNamespace(kLocalStorageNamespaceId);
   local->DeleteOrigin(origin);
 }
 
 void DomStorageContext::DeleteDataModifiedSince(const base::Time& cutoff) {
   std::vector<UsageInfo> infos;
@@ skipping 17 matching lines @@
   if (found != namespaces_.end())
     found->second->PurgeMemory();
 }
 
 void DomStorageContext::Shutdown() {
   is_shutdown_ = true;
   StorageNamespaceMap::const_iterator it = namespaces_.begin();
   for (; it != namespaces_.end(); ++it)
     it->second->Shutdown();
 
+  // Delete data from sessionStorage. If the previous exit was unclean, the
+  // session storage backing might contain leftover data. Delete it now.
+  if (session_storage_database_.get()) {
+    bool success = task_runner_->PostShutdownBlockingTask(
+        FROM_HERE,
+        DomStorageTaskRunner::COMMIT_SEQUENCE,
+        base::Bind(&DomStorageContext::DeleteLeftoverDataInCommitSequence,
+                   this));
+    DCHECK(success);
+  }
+
+  // Delete data from localStorage.
   if (localstorage_directory_.empty())
     return;
 
   // Respect the content policy settings about what to
   // keep and what to discard.
-  if (save_session_state_)
-    return;  // Keep everything.
+  if (save_session_state_)  // Keep everything.
+    return;
 
   bool has_session_only_origins =
       special_storage_policy_.get() &&
       special_storage_policy_->HasSessionOnlyOrigins();
 
   if (clear_local_state_ || has_session_only_origins) {
     // We may have to delete something. We continue on the
     // commit sequence after area shutdown tasks have cycled
     // thru that sequence (and closed their database files).
     bool success = task_runner_->PostShutdownBlockingTask(
@@ skipping 34 matching lines @@
 }
 
 void DomStorageContext::NotifyAreaCleared(
     const DomStorageArea* area,
     const GURL& page_url) {
   FOR_EACH_OBSERVER(
       EventObserver, event_observers_,
       OnDomStorageAreaCleared(area, page_url));
 }
 
+void DomStorageContext::AddSessionStorageObserver(
+    int64 namespace_id,
+    SessionStorageObserver* observer) {
+  session_storage_observers_[namespace_id] = observer;
+}
+
+void DomStorageContext::RemoveSessionStorageObserver(int64 namespace_id) {
+  session_storage_observers_.erase(namespace_id);
+}
+
+void DomStorageContext::NotifySessionStorageNamespaceAssociated(
+    int64 namespace_id,
+    int64 real_namespace_id) {
+  std::map<int64, SessionStorageObserver*>::const_iterator it =
+      session_storage_observers_.find(namespace_id);
+  if (it != session_storage_observers_.end())
+      it->second->OnSessionStorageNamespaceAssociated(real_namespace_id);
+}
+
 void DomStorageContext::CreateSessionNamespace(
     int64 namespace_id) {
   if (is_shutdown_)
     return;
   DCHECK(namespace_id != kLocalStorageNamespaceId);
   DCHECK(namespaces_.find(namespace_id) == namespaces_.end());
   namespaces_[namespace_id] = new DomStorageNamespace(
-      namespace_id, task_runner_);
+      namespace_id, session_storage_database_.get(), task_runner_);
 }
 
 void DomStorageContext::DeleteSessionNamespace(
     int64 namespace_id) {
   DCHECK_NE(kLocalStorageNamespaceId, namespace_id);
   namespaces_.erase(namespace_id);
+  // FIXME: protect from deletion
+  if (session_storage_database_.get()) {
+    bool success = task_runner_->PostShutdownBlockingTask(
+        FROM_HERE,
+        DomStorageTaskRunner::COMMIT_SEQUENCE,
+        base::Bind(&DomStorageContext::DeleteSessionNamespaceInCommitSequence,
+                   this, namespace_id));
+    DCHECK(success);
+  }
 }
 
 void DomStorageContext::CloneSessionNamespace(
     int64 existing_id, int64 new_id) {
   if (is_shutdown_)
     return;
   DCHECK_NE(kLocalStorageNamespaceId, existing_id);
   DCHECK_NE(kLocalStorageNamespaceId, new_id);
   StorageNamespaceMap::iterator found = namespaces_.find(existing_id);
   if (found != namespaces_.end())
     namespaces_[new_id] = found->second->Clone(new_id);
   else
     CreateSessionNamespace(new_id);
 }
 
+void DomStorageContext::AssociateSessionStorage(int64 namespace_id,
+                                                int64 real_id) {
+  if (!session_storage_database_.get())
+    return;
+  bool success = task_runner_->PostShutdownBlockingTask(
+      FROM_HERE,
+      DomStorageTaskRunner::COMMIT_SEQUENCE,
+      base::Bind(&SessionStorageDatabase::AssociateNamespaceId,
+                 session_storage_database_.get(),
+                 namespace_id, real_id));
+  DCHECK(success);
+}
+
 void DomStorageContext::ClearLocalStateInCommitSequence() {
   std::vector<UsageInfo> infos;
   const bool kDontIncludeFileInfo = false;
   GetUsageInfo(&infos, kDontIncludeFileInfo);
   for (size_t i = 0; i < infos.size(); ++i) {
     const GURL& origin = infos[i].origin;
     if (special_storage_policy_ &&
         special_storage_policy_->IsStorageProtected(origin))
       continue;
     if (!clear_local_state_ &&
         !special_storage_policy_->IsStorageSessionOnly(origin))
       continue;
 
     const bool kNotRecursive = false;
     FilePath database_file_path = localstorage_directory_.Append(
         DomStorageArea::DatabaseFileNameFromOrigin(origin));
     file_util::Delete(database_file_path, kNotRecursive);
     file_util::Delete(
         DomStorageDatabase::GetJournalFilePath(database_file_path),
         kNotRecursive);
   }
 }
 
+void DomStorageContext::DeleteSessionNamespaceInCommitSequence(
+    int64 namespace_id) {
+  session_storage_database_->DeleteNamespace(namespace_id);
+}
+
+void DomStorageContext::DeleteLeftoverDataInCommitSequence() {
+  DCHECK(session_storage_database_.get());
+  // Delete all namespaces which don't have an associated DomStorageNamespace
+  // alive.
+  // FIXME: protect from deletion.
+  std::vector<int64> namespace_ids;
+  session_storage_database_->ReadNamespaceIds(&namespace_ids);
+  for (std::vector<int64>::const_iterator it = namespace_ids.begin();
+       it != namespace_ids.end(); ++it) {
+    if (namespaces_.find(*it) == namespaces_.end())
+      session_storage_database_->DeleteNamespace(*it);
+  }
+}
+
 }  // namespace dom_storage