chrome/browser/metrics/variations/variations_service.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_
   6 #define CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_
   7
   8 #include <string>
   9
  10 #include "base/compiler_specific.h"
  11 #include "base/gtest_prod_util.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/memory/weak_ptr.h"
  14 #include "base/metrics/field_trial.h"
  15 #include "base/observer_list.h"
  16 #include "base/threading/thread_checker.h"
  17 #include "base/time/time.h"
  18 #include "components/variations/service/variations_service_client.h"
  19 #include "components/variations/variations_request_scheduler.h"
  20 #include "components/variations/variations_seed_simulator.h"
  21 #include "components/variations/variations_seed_store.h"
  22 #include "components/web_resource/resource_request_allowed_notifier.h"
  23 #include "net/url_request/url_fetcher_delegate.h"
  24 #include "url/gurl.h"
  25
  26 class PrefService;
  27 class PrefRegistrySimple;
  28
  29 namespace base {
  30 class Version;
  31 }
  32
  33 namespace metrics {
  34 class MetricsStateManager;
  35 }
  36
  37 namespace user_prefs {
  38 class PrefRegistrySyncable;
  39 }
  40
  41 namespace variations {
  42 class VariationsSeed;
  43 }
  44
  45 namespace chrome_variations {
  46
  47 // Used to setup field trials based on stored variations seed data, and fetch
  48 // new seed data from the variations server.
  49 class VariationsService
  50     : public net::URLFetcherDelegate,
  51       public web_resource::ResourceRequestAllowedNotifier::Observer {
  52  public:
  53   class Observer {
  54    public:
  55     // How critical a detected experiment change is. Whether it should be
  56     // handled on a "best-effort" basis or, for a more critical change, if it
  57     // should be given higher priority.
  58     enum Severity {
  59       BEST_EFFORT,
  60       CRITICAL,
  61     };
  62
  63     // Called when the VariationsService detects that there will be significant
  64     // experiment changes on a restart. This notification can then be used to
  65     // update UI (i.e. badging an icon).
  66     virtual void OnExperimentChangesDetected(Severity severity) = 0;
  67
  68    protected:
  69     virtual ~Observer() {}
  70   };
  71
  72   ~VariationsService() override;
  73
  74   // Creates field trials based on Variations Seed loaded from local prefs. If
  75   // there is a problem loading the seed data, all trials specified by the seed
  76   // may not be created.
  77   bool CreateTrialsFromSeed();
  78
  79   // Should be called before startup of the main message loop.
  80   void PerformPreMainMessageLoopStartup();
  81
  82   // Calls FetchVariationsSeed once and repeats this periodically. See
  83   // implementation for details on the period. Must be called after
  84   // |CreateTrialsFromSeed|.
  85   void StartRepeatedVariationsSeedFetch();
  86
  87   // Adds an observer to listen for detected experiment changes.
  88   void AddObserver(Observer* observer);
  89
  90   // Removes a previously-added observer.
  91   void RemoveObserver(Observer* observer);
  92
  93   // Called when the application enters foreground. This may trigger a
  94   // FetchVariationsSeed call.
  95   // TODO(rkaplow): Handle this and the similar event in metrics_service by
  96   // observing an 'OnAppEnterForeground' event instead of requiring the frontend
  97   // code to notify each service individually.
  98   void OnAppEnterForeground();
  99
 100   // Sets the value of the "restrict" URL param to the variations service that
 101   // should be used for variation seed requests. This takes precedence over any
 102   // value coming from policy prefs. This should be called prior to any calls
 103   // to |StartRepeatedVariationsSeedFetch|.
 104   void SetRestrictMode(const std::string& restrict_mode);
 105
 106   // Exposed for testing.
 107   void SetCreateTrialsFromSeedCalledForTesting(bool called);
 108
 109   // Returns the variations server URL, which can vary if a command-line flag is
 110   // set and/or the variations restrict pref is set in |local_prefs|. Declared
 111   // static for test purposes.
 112   static GURL GetVariationsServerURL(PrefService* local_prefs,
 113                                      const std::string& restrict_mode_override);
 114
 115   // Exposed for testing.
 116   static std::string GetDefaultVariationsServerURLForTesting();
 117
 118   // Register Variations related prefs in Local State.
 119   static void RegisterPrefs(PrefRegistrySimple* registry);
 120
 121   // Register Variations related prefs in the Profile prefs.
 122   static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry);
 123
 124   // Factory method for creating a VariationsService. Does not take ownership of
 125   // |state_manager|. Caller should ensure that |state_manager| is valid for the
 126   // lifetime of this class.
 127   static scoped_ptr<VariationsService> Create(
 128       scoped_ptr<VariationsServiceClient> client,
 129       PrefService* local_state,
 130       metrics::MetricsStateManager* state_manager);
 131
 132   // Set the PrefService responsible for getting policy-related preferences,
 133   // such as the restrict parameter.
 134   void set_policy_pref_service(PrefService* service) {
 135     DCHECK(service);
 136     policy_pref_service_ = service;
 137   }
 138
 139   // Returns the invalid variations seed signature in base64 format, or an empty
 140   // string if the signature was valid, missing, or if signature verification is
 141   // disabled.
 142   std::string GetInvalidVariationsSeedSignature() const;
 143
 144  protected:
 145   // Starts the fetching process once, where |OnURLFetchComplete| is called with
 146   // the response.
 147   virtual void DoActualFetch();
 148
 149   // Stores the seed to prefs. Set as virtual and protected so that it can be
 150   // overridden by tests.
 151   virtual bool StoreSeed(const std::string& seed_data,
 152                          const std::string& seed_signature,
 153                          const std::string& country_code,
 154                          const base::Time& date_fetched,
 155                          bool is_delta_compressed);
 156
 157   // Creates the VariationsService with the given |local_state| prefs service
 158   // and |state_manager|. This instance will take ownership of |notifier|.
 159   // Does not take ownership of |state_manager|. Caller should ensure that
 160   // |state_manager| is valid for the lifetime of this class. Use the |Create|
 161   // factory method to create a VariationsService.
 162   VariationsService(scoped_ptr<VariationsServiceClient> client,
 163                     web_resource::ResourceRequestAllowedNotifier* notifier,
 164                     PrefService* local_state,
 165                     metrics::MetricsStateManager* state_manager);
 166
 167  private:
 168   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, Observer);
 169   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedStoredWhenOKStatus);
 170   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedNotStoredWhenNonOKStatus);
 171   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedDateUpdatedOn304Status);
 172   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest,
 173                            LoadPermanentConsistencyCountry);
 174   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, CountryHeader);
 175
 176   // Set of different possible values to report for the
 177   // Variations.LoadPermanentConsistencyCountryResult histogram. This enum must
 178   // be kept consistent with its counterpart in histograms.xml.
 179   enum LoadPermanentConsistencyCountryResult {
 180     LOAD_COUNTRY_NO_PREF_NO_SEED = 0,
 181     LOAD_COUNTRY_NO_PREF_HAS_SEED,
 182     LOAD_COUNTRY_INVALID_PREF_NO_SEED,
 183     LOAD_COUNTRY_INVALID_PREF_HAS_SEED,
 184     LOAD_COUNTRY_HAS_PREF_NO_SEED_VERSION_EQ,
 185     LOAD_COUNTRY_HAS_PREF_NO_SEED_VERSION_NEQ,
 186     LOAD_COUNTRY_HAS_BOTH_VERSION_EQ_COUNTRY_EQ,
 187     LOAD_COUNTRY_HAS_BOTH_VERSION_EQ_COUNTRY_NEQ,
 188     LOAD_COUNTRY_HAS_BOTH_VERSION_NEQ_COUNTRY_EQ,
 189     LOAD_COUNTRY_HAS_BOTH_VERSION_NEQ_COUNTRY_NEQ,
 190     LOAD_COUNTRY_MAX,
 191   };
 192
 193   // Checks if prerequisites for fetching the Variations seed are met, and if
 194   // so, performs the actual fetch using |DoActualFetch|.
 195   void FetchVariationsSeed();
 196
 197   // Notify any observers of this service based on the simulation |result|.
 198   void NotifyObservers(
 199       const variations::VariationsSeedSimulator::Result& result);
 200
 201   // net::URLFetcherDelegate implementation:
 202   void OnURLFetchComplete(const net::URLFetcher* source) override;
 203
 204   // ResourceRequestAllowedNotifier::Observer implementation:
 205   void OnResourceRequestsAllowed() override;
 206
 207   // Performs a variations seed simulation with the given |seed| and |version|
 208   // and logs the simulation results as histograms.
 209   void PerformSimulationWithVersion(scoped_ptr<variations::VariationsSeed> seed,
 210                                     const base::Version& version);
 211
 212   // Record the time of the most recent successful fetch.
 213   void RecordLastFetchTime();
 214
 215   // Loads the country code to use for filtering permanent consistency studies,
 216   // updating the stored country code if the stored value was for a different
 217   // Chrome version. The country used for permanent consistency studies is kept
 218   // consistent between Chrome upgrades in order to avoid annoying the user due
 219   // to experiment churn while traveling.
 220   std::string LoadPermanentConsistencyCountry(
 221       const base::Version& version,
 222       const std::string& latest_country);
 223
 224   scoped_ptr<VariationsServiceClient> client_;
 225
 226   // The pref service used to store persist the variations seed.
 227   PrefService* local_state_;
 228
 229   // Used for instantiating entropy providers for variations seed simulation.
 230   // Weak pointer.
 231   metrics::MetricsStateManager* state_manager_;
 232
 233   // Used to obtain policy-related preferences. Depending on the platform, will
 234   // either be Local State or Profile prefs.
 235   PrefService* policy_pref_service_;
 236
 237   VariationsSeedStore seed_store_;
 238
 239   // Contains the scheduler instance that handles timing for requests to the
 240   // server. Initially NULL and instantiated when the initial fetch is
 241   // requested.
 242   scoped_ptr<VariationsRequestScheduler> request_scheduler_;
 243
 244   // Contains the current seed request. Will only have a value while a request
 245   // is pending, and will be reset by |OnURLFetchComplete|.
 246   scoped_ptr<net::URLFetcher> pending_seed_request_;
 247
 248   // The value of the "restrict" URL param to the variations server that has
 249   // been specified via |SetRestrictMode|. If empty, the URL param will be set
 250   // based on policy prefs.
 251   std::string restrict_mode_;
 252
 253   // The URL to use for querying the variations server.
 254   GURL variations_server_url_;
 255
 256   // Tracks whether |CreateTrialsFromSeed| has been called, to ensure that
 257   // it gets called prior to |StartRepeatedVariationsSeedFetch|.
 258   bool create_trials_from_seed_called_;
 259
 260   // Tracks whether the initial request to the variations server had completed.
 261   bool initial_request_completed_;
 262
 263   // Indicates that the next request to the variations service shouldn't specify
 264   // that it supports delta compression. Set to true when a delta compressed
 265   // response encountered an error.
 266   bool disable_deltas_for_next_request_;
 267
 268   // Helper class used to tell this service if it's allowed to make network
 269   // resource requests.
 270   scoped_ptr<web_resource::ResourceRequestAllowedNotifier>
 271       resource_request_allowed_notifier_;
 272
 273   // The start time of the last seed request. This is used to measure the
 274   // latency of seed requests. Initially zero.
 275   base::TimeTicks last_request_started_time_;
 276
 277   // The number of requests to the variations server that have been performed.
 278   int request_count_;
 279
 280   // List of observers of the VariationsService.
 281   base::ObserverList<Observer> observer_list_;
 282
 283   base::ThreadChecker thread_checker_;
 284
 285   base::WeakPtrFactory<VariationsService> weak_ptr_factory_;
 286
 287   DISALLOW_COPY_AND_ASSIGN(VariationsService);
 288 };
 289
 290 }  // namespace chrome_variations
 291
 292 #endif  // CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_