chrome/installer/gcapi/gcapi_omaha_experiment.cc

   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 #include "chrome/installer/gcapi/gcapi_omaha_experiment.h"
   6
   7 #include "base/basictypes.h"
   8 #include "base/strings/string_split.h"
   9 #include "base/strings/string_util.h"
  10 #include "base/strings/stringprintf.h"
  11 #include "base/strings/utf_string_conversions.h"
  12 #include "base/time/time.h"
  13 #include "chrome/installer/gcapi/gcapi.h"
  14 #include "chrome/installer/util/google_update_constants.h"
  15 #include "chrome/installer/util/google_update_experiment_util.h"
  16 #include "chrome/installer/util/google_update_settings.h"
  17
  18 namespace {
  19
  20 // Returns the number of weeks since 2/3/2003.
  21 int GetCurrentRlzWeek(const base::Time& current_time) {
  22   base::Time::Exploded february_third_2003_exploded =
  23       {2003, 2, 1, 3, 0, 0, 0, 0};
  24   base::Time f = base::Time::FromUTCExploded(february_third_2003_exploded);
  25   base::TimeDelta delta = current_time - f;
  26   return delta.InDays() / 7;
  27 }
  28
  29 bool SetExperimentLabel(const wchar_t* brand_code,
  30                         const base::string16& label,
  31                         int shell_mode) {
  32   if (!brand_code) {
  33     return false;
  34   }
  35
  36   const bool system_level = shell_mode == GCAPI_INVOKED_UAC_ELEVATION;
  37
  38   base::string16 original_labels;
  39   if (!GoogleUpdateSettings::ReadExperimentLabels(system_level,
  40                                                   &original_labels)) {
  41     return false;
  42   }
  43
  44   // Split the original labels by the label separator.
  45   std::vector<base::string16> entries = base::SplitString(
  46       original_labels,
  47       base::string16(1, google_update::kExperimentLabelSeparator),
  48       base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
  49
  50   // Keep all labels, but the one we want to add/replace.
  51   base::string16 label_and_separator(label);
  52   label_and_separator.push_back('=');
  53   base::string16 new_labels;
  54   for (const base::string16& entry : entries) {
  55     if (!entry.empty() &&
  56         !base::StartsWith(entry, label_and_separator,
  57                           base::CompareCase::SENSITIVE)) {
  58       new_labels += entry;
  59       new_labels += google_update::kExperimentLabelSeparator;
  60     }
  61   }
  62
  63   new_labels.append(
  64       gcapi_internals::GetGCAPIExperimentLabel(brand_code, label));
  65
  66   return GoogleUpdateSettings::SetExperimentLabels(system_level,
  67                                                    new_labels);
  68 }
  69
  70 }  // namespace
  71
  72 namespace gcapi_internals {
  73
  74 const wchar_t kReactivationLabel[] = L"reacbrand";
  75 const wchar_t kRelaunchLabel[] = L"relaunchbrand";
  76
  77 base::string16 GetGCAPIExperimentLabel(const wchar_t* brand_code,
  78                                        const base::string16& label) {
  79   // Keeps a fixed time state for this GCAPI instance; this makes tests reliable
  80   // when crossing time boundaries on the system clock and doesn't otherwise
  81   // affect results of this short lived binary.
  82   static time_t instance_time_value = 0;
  83   if (instance_time_value == 0)
  84     instance_time_value = base::Time::Now().ToTimeT();
  85
  86   base::Time instance_time = base::Time::FromTimeT(instance_time_value);
  87
  88   base::string16 gcapi_experiment_label;
  89   base::SStringPrintf(&gcapi_experiment_label,
  90                       L"%ls=%ls_%d|%ls",
  91                       label.c_str(),
  92                       brand_code,
  93                       GetCurrentRlzWeek(instance_time),
  94                       installer::BuildExperimentDateString(
  95                           instance_time).c_str());
  96   return gcapi_experiment_label;
  97 }
  98
  99 }  // namespace gcapi_internals
 100
 101 bool SetReactivationExperimentLabels(const wchar_t* brand_code,
 102                                      int shell_mode) {
 103   return SetExperimentLabel(brand_code, gcapi_internals::kReactivationLabel,
 104                             shell_mode);
 105 }
 106
 107 bool SetRelaunchExperimentLabels(const wchar_t* brand_code, int shell_mode) {
 108   return SetExperimentLabel(brand_code, gcapi_internals::kRelaunchLabel,
 109                             shell_mode);
 110 }