Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(732)

Side by Side Diff: chrome/installer/gcapi/gcapi_omaha_experiment.cc

Issue 11280067: Refactor SetOmahaExperimentLabel out of gcpai and into install_util. (Closed) Base URL: http://git.chromium.org/chromium/src.git@master
Patch Set: grt additional comments Created 7 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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/string16.h" 7 #include "base/string16.h"
8 #include "base/stringprintf.h" 8 #include "base/stringprintf.h"
9 #include "base/string_number_conversions.h"
10 #include "base/time.h" 9 #include "base/time.h"
11 #include "base/win/registry.h" 10 #include "base/win/registry.h"
12 #include "chrome/installer/util/browser_distribution.h"
13 #include "chrome/installer/util/google_update_constants.h"
14 #include "chrome/installer/gcapi/gcapi.h" 11 #include "chrome/installer/gcapi/gcapi.h"
12 #include "chrome/installer/util/google_update_settings.h"
15 13
16 using base::Time; 14 using base::Time;
17 using base::TimeDelta; 15 using base::TimeDelta;
18 using base::win::RegKey; 16 using base::win::RegKey;
19 17
20 namespace { 18 namespace {
21 19
22 const wchar_t kExperimentLabels[] = L"experiment_labels";
23
24 const wchar_t* kExperimentAppGuids[] = {
25 L"{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}",
26 L"{8A69D345-D564-463C-AFF1-A69D9E530F96}",
27 };
28
29 const wchar_t* kDays[] = 20 const wchar_t* kDays[] =
30 { L"Sun", L"Mon", L"Tue", L"Wed", L"Thu", L"Fri", L"Sat" }; 21 { L"Sun", L"Mon", L"Tue", L"Wed", L"Thu", L"Fri", L"Sat" };
31 22
32 const wchar_t* kMonths[] = {L"Jan", L"Feb", L"Mar", L"Apr", L"May", L"Jun", 23 const wchar_t* kMonths[] = {L"Jan", L"Feb", L"Mar", L"Apr", L"May", L"Jun",
33 L"Jul", L"Aug", L"Sep", L"Oct", L"Nov", L"Dec"}; 24 L"Jul", L"Aug", L"Sep", L"Oct", L"Nov", L"Dec"};
34 25
35 // Constructs a date string of the following format for the current time plus 26 // Constructs a date string of the following format for the current time plus
36 // one year: 27 // one year:
37 // "DAY, DD0 MON YYYY HH0:MI0:SE0 TZ" 28 // "DAY, DD0 MON YYYY HH0:MI0:SE0 TZ"
38 // DAY = 3 character day of week, 29 // DAY = 3 character day of week,
(...skipping 28 matching lines...) Expand all
67 // Returns the number of weeks since 2/3/2003. 58 // Returns the number of weeks since 2/3/2003.
68 int GetCurrentRlzWeek() { 59 int GetCurrentRlzWeek() {
69 Time::Exploded february_third_2003_exploded = {2003, 2, 1, 3, 0, 0, 0, 0}; 60 Time::Exploded february_third_2003_exploded = {2003, 2, 1, 3, 0, 0, 0, 0};
70 Time f = Time::FromUTCExploded(february_third_2003_exploded); 61 Time f = Time::FromUTCExploded(february_third_2003_exploded);
71 TimeDelta delta = Time::Now() - f; 62 TimeDelta delta = Time::Now() - f;
72 return delta.InDays() / 7; 63 return delta.InDays() / 7;
73 } 64 }
74 65
75 } // namespace 66 } // namespace
76 67
77 bool SetOmahaExperimentLabel(const wchar_t* brand_code, int shell_mode) { 68 bool SetReactivationExperimentLabels(const wchar_t* brand_code,
69 int shell_mode) {
78 if (!brand_code) { 70 if (!brand_code) {
79 return false; 71 return false;
80 } 72 }
81 73
82 // When this function is invoked in standard, non-elevated shell, we default
83 // to writing the experiment label to HKCU. When it is invoked in a UAC-
84 // elevated shell, we write the experiment label to HKLM.
85 HKEY registry_hive =
86 shell_mode == GCAPI_INVOKED_UAC_ELEVATION ? HKEY_LOCAL_MACHINE :
87 HKEY_CURRENT_USER;
88
89 int week_number = GetCurrentRlzWeek(); 74 int week_number = GetCurrentRlzWeek();
90 if (week_number < 0 || week_number > 999) 75 if (week_number < 0 || week_number > 999)
91 week_number = 999; 76 week_number = 999;
92 77
93 string16 experiment_label; 78 string16 experiment_labels;
94 base::SStringPrintf(&experiment_label, 79 base::SStringPrintf(&experiment_labels,
95 L"reacbrand=%ls_%d|%ls", 80 L"reacbrand=%ls_%d|%ls",
96 brand_code, 81 brand_code,
97 week_number, 82 week_number,
98 BuildOmahaExperimentDateString().c_str()); 83 BuildOmahaExperimentDateString().c_str());
99 84
100 int successful_writes = 0; 85 return GoogleUpdateSettings::SetExperimentLabels(
101 for (int i = 0; i < arraysize(kExperimentAppGuids); ++i) { 86 shell_mode == GCAPI_INVOKED_UAC_ELEVATION,
102 string16 experiment_path(google_update::kRegPathClientState); 87 experiment_labels);
103 experiment_path += L"\\";
104 experiment_path += kExperimentAppGuids[i];
105
106 RegKey client_state(registry_hive, experiment_path.c_str(),
107 KEY_SET_VALUE);
108 if (client_state.Valid()) {
109 if (client_state.WriteValue(kExperimentLabels,
110 experiment_label.c_str()) == ERROR_SUCCESS) {
111 successful_writes++;
112 }
113 }
114 }
115
116 return (successful_writes == arraysize(kExperimentAppGuids));
117 } 88 }
OLDNEW
« no previous file with comments | « chrome/installer/gcapi/gcapi_omaha_experiment.h ('k') | chrome/installer/util/browser_distribution.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698