Adding session-only localStorage.

content/browser/in_process_webkit/dom_storage_namespace.h

 // 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.
 
 #ifndef CONTENT_BROWSER_IN_PROCESS_WEBKIT_DOM_STORAGE_NAMESPACE_H_
 #define CONTENT_BROWSER_IN_PROCESS_WEBKIT_DOM_STORAGE_NAMESPACE_H_
 #pragma once
 
 #include "base/hash_tables.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 13 matching lines @@
 // Only to be used on the WebKit thread.
 class DOMStorageNamespace {
  public:
   static DOMStorageNamespace* CreateLocalStorageNamespace(
       DOMStorageContext* dom_storage_context, const FilePath& data_dir_path);
   static DOMStorageNamespace* CreateSessionStorageNamespace(
       DOMStorageContext* dom_storage_context, int64 namespace_id);
 
   ~DOMStorageNamespace();
 
-  DOMStorageArea* GetStorageArea(const string16& origin);
+  DOMStorageArea* GetStorageArea(const string16& origin,
+                                 bool enforce_session_only_storage);
   DOMStorageNamespace* Copy(int64 clone_namespace_id);
 
   void PurgeMemory();
 
   const DOMStorageContext* dom_storage_context() const {
     return dom_storage_context_;
   }
   int64 id() const { return id_; }
   const WebKit::WebString& data_dir_path() const { return data_dir_path_; }
   DOMStorageType dom_storage_type() const { return dom_storage_type_; }
 
   // Creates a WebStorageArea for the given origin.  This should only be called
   // by an owned DOMStorageArea.
   WebKit::WebStorageArea* CreateWebStorageArea(const string16& origin);
 
  private:
   // Called by the static factory methods above.
   DOMStorageNamespace(DOMStorageContext* dom_storage_context,
                       int64 id,
                       const WebKit::WebString& data_dir_path,
                       DOMStorageType storage_type);
 
   // Creates the underlying WebStorageNamespace on demand.
   void CreateWebStorageNamespaceIfNecessary();
 
+  // Helpers for creating DOMStorageAreas.
+  DOMStorageArea* GetExistingStorageArea(const string16& origin);
+  DOMStorageArea* CreateStorageArea(const string16& origin);
+
+  // Returns true if this DOMStorageNamespace represents localStorage and should
+  // have a pointer to a session-only localStorage DOMStorageNamespace (which is
+  // lazily created). Returns false if this DOMStorageNamespace is a
+  // sessionStorage, or a session-only localStorage, or a localStorage
+  // associated with an off the record profile.
+  bool ShouldHaveSessionOnlyLocalStorage() const;
+
+  // Lazy creation of the DOMStorageNamespace representing session-only
+  // localStorage.
+  void EnsureSessionOnlyLocalStorageCreated();
+
   // All the storage areas we own.
   typedef base::hash_map<string16, DOMStorageArea*> OriginToStorageAreaMap;
   OriginToStorageAreaMap origin_to_storage_area_;
 
   // The DOMStorageContext that owns us.
   DOMStorageContext* dom_storage_context_;
 
   // The WebKit storage namespace we manage.
   scoped_ptr<WebKit::WebStorageNamespace> storage_namespace_;
 
   // Our id.  Unique to our parent WebKitContext class.
   int64 id_;
 
   // The path used to create us, so we can recreate our WebStorageNamespace on
   // demand.
   WebKit::WebString data_dir_path_;
 
   // SessionStorage vs. LocalStorage.
   const DOMStorageType dom_storage_type_;
 
+  // Session-only version of the same localStorage (lazily created).
+  scoped_ptr<DOMStorageNamespace> session_only_local_storage_;
+
   DISALLOW_IMPLICIT_CONSTRUCTORS(DOMStorageNamespace);
 };
 
 #endif  // CONTENT_BROWSER_IN_PROCESS_WEBKIT_DOM_STORAGE_NAMESPACE_H_