Make SDCH dictionaries persistent across browser restart.

net/sdch/sdch_owner.h

 // Copyright 2014 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 NET_SDCH_SDCH_OWNER_H_
 #define NET_SDCH_SDCH_OWNER_H_
 
+#include <map>
 #include <string>
 
 #include "base/memory/memory_pressure_listener.h"
+#include "base/prefs/pref_store.h"
 #include "net/base/sdch_observer.h"
 #include "net/url_request/sdch_dictionary_fetcher.h"
 
 class GURL;
+class PersistentPrefStore;
+class ValueMapPrefStore;
+class WriteablePrefStore;
 
 namespace base {
 class Clock;
 }
 
 namespace net {
 class SdchManager;
 class URLRequestContext;
 
 // This class owns the SDCH objects not owned as part of URLRequestContext, and
 // exposes interface for setting SDCH policy.  It should be instantiated by
 // the net/ embedder.
 // TODO(rdsmith): Implement dictionary prioritization.
-class NET_EXPORT SdchOwner : public net::SdchObserver {
+class NET_EXPORT SdchOwner : public net::SdchObserver,
+                             public PrefStore::Observer {
  public:
   static const size_t kMaxTotalDictionarySize;
   static const size_t kMinSpaceForDictionaryFetch;
 
   // Consumer must guarantee that |sdch_manager| and |context| outlive
   // this object.
   SdchOwner(net::SdchManager* sdch_manager, net::URLRequestContext* context);
   ~SdchOwner() override;
 
+  // Enables use of pref persistence.  Dictionaries added to the SdchManager
+  // before this call is made will not be added to the pref store.
+  // Note that |pref_store| is owned by the caller, but must be guaranteed
+  // to outlive SdchOwner.
+  // This routine may only be called once per SdchOwner instance.
+  void EnablePersistentStorage(PersistentPrefStore* pref_store);
+
   // Defaults to kMaxTotalDictionarySize.
   void SetMaxTotalDictionarySize(size_t max_total_dictionary_size);
 
   // Defaults to kMinSpaceForDictionaryFetch.
   void SetMinSpaceForDictionaryFetch(size_t min_space_for_dictionary_fetch);
 
   // SdchObserver implementation.
   void OnDictionaryUsed(SdchManager* manager,
                         const std::string& server_hash) override;
   void OnGetDictionary(net::SdchManager* manager,
                        const GURL& request_url,
                        const GURL& dictionary_url) override;
   void OnClearDictionaries(net::SdchManager* manager) override;
 
+  // PrefStore::Observer implementation.
+  void OnPrefValueChanged(const std::string& key) override;
+  void OnInitializationCompleted(bool succeeded) override;
+
   // Implementation detail--this is the pathway through which the
   // fetcher informs the SdchOwner that it's gotten the dictionary.
   // Public for testing.
   void OnDictionaryFetched(const std::string& dictionary_text,
                            const GURL& dictionary_url,
+                           scoped_ptr<SdchDictionaryFetcher::Data> extra_data,
                            const net::BoundNetLog& net_log);
 
   void SetClockForTesting(scoped_ptr<base::Clock> clock);
 
  private:
   // For each active dictionary, stores local info.
   // Indexed by server hash.

[mmenke, 2015/02/02 16:35:28]

Should define server hash.  I'm assuming it's the server's hash for the
dictionary, rather than a hash of the server name.

[Randy Smith (Not in Mondays), 2015/02/04 19:29:03]

Done (though it will be moot).

   struct DictionaryInfo {
     base::Time last_used;
     int use_count;
     size_t size;
 
     DictionaryInfo() : use_count(0), size(0) {}
     DictionaryInfo(const base::Time& last_used, size_t size)
         : last_used(last_used), use_count(0), size(size) {}
     DictionaryInfo(const DictionaryInfo& rhs) = default;
     DictionaryInfo& operator=(const DictionaryInfo& rhs) = default;
   };
 
   void OnMemoryPressure(
       base::MemoryPressureListener::MemoryPressureLevel level);
 
+  // Returns false if the load fails for some reason.
+  bool LoadPersistedDictionaries(const base::DictionaryValue& persisted_info);

[mmenke, 2015/02/02 16:35:28]

These needs a lot more detail, and maybe a rename, because of how it works -
scheduling fetches, rather than loading them into a pref store directly isn't
terribly intuitive.

[Randy Smith (Not in Mondays), 2015/02/04 19:29:03]

I've rewritten the comment and renamed the function; let me know what you think.

+
   net::SdchManager* manager_;
   net::SdchDictionaryFetcher fetcher_;
 
   std::map<std::string, DictionaryInfo> local_dictionary_info_;
   size_t total_dictionary_bytes_;
 
   scoped_ptr<base::Clock> clock_;
 
   size_t max_total_dictionary_size_;
   size_t min_space_for_dictionary_fetch_;
 
+  // Dictionary persistence machinery.  |in_memory_pref_store_|
+  // is created on construction and used in the absence of any call
+  // to EnablePersistentStorage().  |persistent_pref_store_| holds the
+  // preference store specified by EnablePersistentStorage() (if any).
+  // |pref_store_| is shifted from the first to the second after the
+  // persistent pref store information is fully read in.
+  scoped_refptr<ValueMapPrefStore> in_memory_pref_store_;
+  PersistentPrefStore* persistent_pref_store_;
+
+  WriteablePrefStore* pref_store_;
+
   base::MemoryPressureListener memory_pressure_listener_;
 
   DISALLOW_COPY_AND_ASSIGN(SdchOwner);
 };
 
 }  // namespace net
 
 #endif  // NET_SDCH_SDCH_OWNER_H_