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