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/time/time.h"
  17 #include "chrome/browser/metrics/variations/variations_request_scheduler.h"
  18 #include "chrome/browser/metrics/variations/variations_seed_store.h"
  19 #include "chrome/common/chrome_version_info.h"
  20 #include "components/variations/variations_seed_simulator.h"
  21 #include "components/web_resource/resource_request_allowed_notifier.h"
  22 #include "net/url_request/url_fetcher_delegate.h"
  23 #include "url/gurl.h"
  24
  25 #if defined(OS_WIN)
  26 #include "chrome/browser/metrics/variations/variations_registry_syncer_win.h"
  27 #endif
  28
  29 class PrefService;
  30 class PrefRegistrySimple;
  31
  32 namespace base {
  33 class Version;
  34 }
  35
  36 namespace metrics {
  37 class MetricsStateManager;
  38 }
  39
  40 namespace user_prefs {
  41 class PrefRegistrySyncable;
  42 }
  43
  44 namespace variations {
  45 class VariationsSeed;
  46 }
  47
  48 namespace chrome_variations {
  49
  50 // Used to setup field trials based on stored variations seed data, and fetch
  51 // new seed data from the variations server.
  52 class VariationsService
  53     : public net::URLFetcherDelegate,
  54       public web_resource::ResourceRequestAllowedNotifier::Observer {
  55  public:
  56   class Observer {
  57    public:
  58     // How critical a detected experiment change is. Whether it should be
  59     // handled on a "best-effort" basis or, for a more critical change, if it
  60     // should be given higher priority.
  61     enum Severity {
  62       BEST_EFFORT,
  63       CRITICAL,
  64     };
  65
  66     // Called when the VariationsService detects that there will be significant
  67     // experiment changes on a restart. This notification can then be used to
  68     // update UI (i.e. badging an icon).
  69     virtual void OnExperimentChangesDetected(Severity severity) = 0;
  70
  71    protected:
  72     virtual ~Observer() {}
  73   };
  74
  75   ~VariationsService() override;
  76
  77   // Creates field trials based on Variations Seed loaded from local prefs. If
  78   // there is a problem loading the seed data, all trials specified by the seed
  79   // may not be created.
  80   bool CreateTrialsFromSeed();
  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 #if defined(OS_WIN)
 101   // Starts syncing Google Update Variation IDs with the registry.
 102   void StartGoogleUpdateRegistrySync();
 103 #endif
 104
 105   // Sets the value of the "restrict" URL param to the variations service that
 106   // should be used for variation seed requests. This takes precedence over any
 107   // value coming from policy prefs. This should be called prior to any calls
 108   // to |StartRepeatedVariationsSeedFetch|.
 109   void SetRestrictMode(const std::string& restrict_mode);
 110
 111   // Exposed for testing.
 112   void SetCreateTrialsFromSeedCalledForTesting(bool called);
 113
 114   // Returns the variations server URL, which can vary if a command-line flag is
 115   // set and/or the variations restrict pref is set in |local_prefs|. Declared
 116   // static for test purposes.
 117   static GURL GetVariationsServerURL(PrefService* local_prefs,
 118                                      const std::string& restrict_mode_override);
 119
 120   // Exposed for testing.
 121   static std::string GetDefaultVariationsServerURLForTesting();
 122
 123   // Register Variations related prefs in Local State.
 124   static void RegisterPrefs(PrefRegistrySimple* registry);
 125
 126   // Register Variations related prefs in the Profile prefs.
 127   static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry);
 128
 129   // Factory method for creating a VariationsService. Does not take ownership of
 130   // |state_manager|. Caller should ensure that |state_manager| is valid for the
 131   // lifetime of this class.
 132   static scoped_ptr<VariationsService> Create(
 133       PrefService* local_state,
 134       metrics::MetricsStateManager* state_manager);
 135
 136   // Set the PrefService responsible for getting policy-related preferences,
 137   // such as the restrict parameter.
 138   void set_policy_pref_service(PrefService* service) {
 139     DCHECK(service);
 140     policy_pref_service_ = service;
 141   }
 142
 143   // Returns the invalid variations seed signature in base64 format, or an empty
 144   // string if the signature was valid, missing, or if signature verification is
 145   // disabled.
 146   std::string GetInvalidVariationsSeedSignature() const;
 147
 148  protected:
 149   // Starts the fetching process once, where |OnURLFetchComplete| is called with
 150   // the response.
 151   virtual void DoActualFetch();
 152
 153   // Stores the seed to prefs. Set as virtual and protected so that it can be
 154   // overridden by tests.
 155   virtual void StoreSeed(const std::string& seed_data,
 156                          const std::string& seed_signature,
 157                          const base::Time& date_fetched);
 158
 159   // Creates the VariationsService with the given |local_state| prefs service
 160   // and |state_manager|. This instance will take ownership of |notifier|.
 161   // Does not take ownership of |state_manager|. Caller should ensure that
 162   // |state_manager| is valid for the lifetime of this class. Use the |Create|
 163   // factory method to create a VariationsService.
 164   VariationsService(web_resource::ResourceRequestAllowedNotifier* notifier,
 165                     PrefService* local_state,
 166                     metrics::MetricsStateManager* state_manager);
 167
 168  private:
 169   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, Observer);
 170   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedStoredWhenOKStatus);
 171   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedNotStoredWhenNonOKStatus);
 172   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedDateUpdatedOn304Status);
 173   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest,
 174                            LoadPermanentConsistencyCountry);
 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 variations::VariationsSeed& seed);
 223
 224   // The pref service used to store persist the variations seed.
 225   PrefService* local_state_;
 226
 227   // Used for instantiating entropy providers for variations seed simulation.
 228   // Weak pointer.
 229   metrics::MetricsStateManager* state_manager_;
 230
 231   // Used to obtain policy-related preferences. Depending on the platform, will
 232   // either be Local State or Profile prefs.
 233   PrefService* policy_pref_service_;
 234
 235   VariationsSeedStore seed_store_;
 236
 237   // Contains the scheduler instance that handles timing for requests to the
 238   // server. Initially NULL and instantiated when the initial fetch is
 239   // requested.
 240   scoped_ptr<VariationsRequestScheduler> request_scheduler_;
 241
 242   // Contains the current seed request. Will only have a value while a request
 243   // is pending, and will be reset by |OnURLFetchComplete|.
 244   scoped_ptr<net::URLFetcher> pending_seed_request_;
 245
 246   // The value of the "restrict" URL param to the variations server that has
 247   // been specified via |SetRestrictMode|. If empty, the URL param will be set
 248   // based on policy prefs.
 249   std::string restrict_mode_;
 250
 251   // The URL to use for querying the variations server.
 252   GURL variations_server_url_;
 253
 254   // Tracks whether |CreateTrialsFromSeed| has been called, to ensure that
 255   // it gets called prior to |StartRepeatedVariationsSeedFetch|.
 256   bool create_trials_from_seed_called_;
 257
 258   // Tracks whether the initial request to the variations server had completed.
 259   bool initial_request_completed_;
 260
 261   // Helper class used to tell this service if it's allowed to make network
 262   // resource requests.
 263   scoped_ptr<web_resource::ResourceRequestAllowedNotifier>
 264       resource_request_allowed_notifier_;
 265
 266   // The start time of the last seed request. This is used to measure the
 267   // latency of seed requests. Initially zero.
 268   base::TimeTicks last_request_started_time_;
 269
 270   // The number of requests to the variations server that have been performed.
 271   int request_count_;
 272
 273   // List of observers of the VariationsService.
 274   base::ObserverList<Observer> observer_list_;
 275
 276 #if defined(OS_WIN)
 277   // Helper that handles synchronizing Variations with the Registry.
 278   VariationsRegistrySyncer registry_syncer_;
 279 #endif
 280
 281   base::WeakPtrFactory<VariationsService> weak_ptr_factory_;
 282
 283   DISALLOW_COPY_AND_ASSIGN(VariationsService);
 284 };
 285
 286 }  // namespace chrome_variations
 287
 288 #endif  // CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_