blob: 3d6b59b76618a819943914759ddb04d8bb382eaa [file] [log] [blame]
// 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.
#include <map>
#include <string>
#include "base/memory/memory_pressure_listener.h"
#include "base/memory/ref_counted.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 SdchObserver, public PrefStore::Observer {
static const size_t kMaxTotalDictionarySize;
static const size_t kMinSpaceForDictionaryFetch;
// Consumer must guarantee that |sdch_manager| and |context| outlive
// this object.
SdchOwner(SdchManager* sdch_manager, URLRequestContext* context);
~SdchOwner() override;
// Enables use of pref persistence. Note that |pref_store| is owned
// by the caller, but must be guaranteed to outlive SdchOwner. The
// actual mechanisms by which the PersistentPrefStore are persisted
// are the responsibility of the caller. 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 OnDictionaryAdded(const GURL& dictionary_url,
const std::string& server_hash) override;
void OnDictionaryRemoved(const std::string& server_hash) override;
void OnDictionaryUsed(const std::string& server_hash) override;
void OnGetDictionary(const GURL& request_url,
const GURL& dictionary_url) override;
void OnClearDictionaries() override;
// PrefStore::Observer implementation.
void OnPrefValueChanged(const std::string& key) override;
void OnInitializationCompleted(bool succeeded) override;
// Implementation detail--this is the function callback by the callback passed
// to the fetcher through which the fetcher informs the SdchOwner that it's
// gotten the dictionary. The first two arguments are bound locally.
// Public for testing.
void OnDictionaryFetched(base::Time last_used,
int use_count,
const std::string& dictionary_text,
const GURL& dictionary_url,
const BoundNetLog& net_log,
bool was_from_cache);
void SetClockForTesting(scoped_ptr<base::Clock> clock);
// Returns the total number of dictionaries loaded.
int GetDictionaryCountForTesting() const;
// Returns whether this SdchOwner has dictionary from |url| loaded.
bool HasDictionaryFromURLForTesting(const GURL& url) const;
void SetFetcherForTesting(scoped_ptr<SdchDictionaryFetcher> fetcher);
// For each active dictionary, stores local info.
// Indexed by the server hash of the dictionary.
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);
// Schedule loading of all dictionaries described in |persisted_info|.
// Returns false and does not schedule a load if |persisted_info| has an
// unsupported version or no dictionaries key. Skips any dictionaries that are
// malformed in |persisted_info|.
bool SchedulePersistedDictionaryLoads(
const base::DictionaryValue& persisted_info);
bool IsPersistingDictionaries() const;
enum DictionaryFate {
// A Get-Dictionary header wasn't acted on.
// A fetch was attempted, but failed.
// TODO(rdsmith): Actually record this case.
// A successful fetch was dropped on the floor, no space.
// A successful fetch was refused by the SdchManager.
// A dictionary was successfully added based on
// a Get-Dictionary header in a response.
// A dictionary was evicted by an incoming dict.
// A dictionary was evicted by memory pressure.
// A dictionary was evicted on destruction.
// A dictionary was successfully added based on
// persistence from a previous browser revision.
// A dictionary was unloaded on destruction, but is still present on disk.
void RecordDictionaryFate(DictionaryFate fate);
// Record the lifetime memory use of a specified dictionary, identified by
// server hash.
void RecordDictionaryEvictionOrUnload(
const std::string& server_hash,
size_t size,
int use_count, DictionaryFate fate);
net::SdchManager* manager_;
scoped_ptr<net::SdchDictionaryFetcher> fetcher_;
size_t total_dictionary_bytes_;
scoped_ptr<base::Clock> clock_;
size_t max_total_dictionary_size_;
size_t min_space_for_dictionary_fetch_;
base::MemoryPressureListener memory_pressure_listener_;
// Dictionary persistence machinery.
// * |in_memory_pref_store_| is created on construction and used in
// the absence of any call to EnablePersistentStorage().
// * |external_pref_store_| holds the preference store specified
// by EnablePersistentStorage() (if any), while it is being read in.
// A non-null value here signals that the SdchOwner is observing
// the pref store; when read-in completes and observation is no longer
// needed, the pointer is set to null. This is to avoid lots of
// extra irrelevant function calls; the only observer interface this
// class is interested in is OnInitializationCompleted().
// * |pref_store_| holds an unowned pointer to the currently
// active pref store (one of the preceding two).
scoped_refptr<ValueMapPrefStore> in_memory_pref_store_;
PersistentPrefStore* external_pref_store_;
WriteablePrefStore* pref_store_;
// The use counts of dictionaries when they were loaded from the persistent
// store, keyed by server hash. These are stored to avoid generating
// misleading ever-increasing use counts for dictionaries that are persisted,
// since the UMA histogram for use counts is only supposed to be since last
// load.
std::map<std::string, int> use_counts_at_load_;
// Load times for loaded dictionaries, keyed by server hash. These are used to
// track the durations that dictionaries are in memory.
std::map<std::string, base::Time> load_times_;
// Byte-seconds consumed by dictionaries that have been unloaded. These are
// stored for later uploading in the SdchOwner destructor.
std::vector<int64> consumed_byte_seconds_;
// Creation time for this SdchOwner object, used for reporting temporal memory
// pressure.
base::Time creation_time_;
} // namespace net