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

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

Issue 8508060: Add functionality to the GCAPI to detect when Chrome was last run. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src/
Patch Set: Created 9 years, 1 month 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 | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2011 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 // NOTE: This code is a legacy utility API for partners to checking whether
grt (UTC plus 2) 2011/11/11 17:01:01 checking -> check
robertshield 2011/11/12 05:25:38 Done.
6 // Chrome can be installed and launched. Recent updates are being made
7 // to add new functionality. These updates use code from Chromium, the old
8 // coded against the win32 api directly. If you have an itch to shave a
9 // yak, feel free to re-write the old code too.
10
5 #include "chrome/installer/gcapi/gcapi.h" 11 #include "chrome/installer/gcapi/gcapi.h"
6 12
7 #include <atlbase.h> 13 #include <atlbase.h>
8 #include <atlcom.h> 14 #include <atlcom.h>
9 #include <windows.h>
10 #include <sddl.h> 15 #include <sddl.h>
11 #define STRSAFE_NO_DEPRECATE 16 #define STRSAFE_NO_DEPRECATE
12 #include <strsafe.h> 17 #include <strsafe.h>
13 #include <tlhelp32.h> 18 #include <tlhelp32.h>
19 #include <windows.h>
14 20
15 #include <cstdlib> 21 #include <cstdlib>
22 #include <limits>
23 #include <string>
16 24
17 #include "google_update_idl.h" 25 #include "base/basictypes.h"
26 #include "base/string_number_conversions.h"
27 #include "base/time.h"
28 #include "base/win/registry.h"
29 #include "chrome/installer/util/google_update_constants.h"
30
31 #include "google_update_idl.h" // NOLINT
18 32
19 namespace { 33 namespace {
20 34
21 const wchar_t kChromeRegClientsKey[] = 35 const wchar_t kChromeRegClientsKey[] =
22 L"Software\\Google\\Update\\Clients\\" 36 L"Software\\Google\\Update\\Clients\\"
23 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; 37 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}";
24 const wchar_t kChromeRegClientStateKey[] = 38 const wchar_t kChromeRegClientStateKey[] =
25 L"Software\\Google\\Update\\ClientState\\" 39 L"Software\\Google\\Update\\ClientState\\"
26 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; 40 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}";
27 const wchar_t kChromeRegLaunchCmd[] = L"InstallerSuccessLaunchCmdLine"; 41 const wchar_t kChromeRegLaunchCmd[] = L"InstallerSuccessLaunchCmdLine";
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after
63 memcpy(&lang, data, 4); 77 memcpy(&lang, data, 4);
64 ::StringCchPrintf(info_name, _countof(info_name), 78 ::StringCchPrintf(info_name, _countof(info_name),
65 L"\\StringFileInfo\\%02X%02X%02X%02X\\CompanyName", 79 L"\\StringFileInfo\\%02X%02X%02X%02X\\CompanyName",
66 (lang & 0xff00)>>8, (lang & 0xff), (lang & 0xff000000)>>24, 80 (lang & 0xff00)>>8, (lang & 0xff), (lang & 0xff000000)>>24,
67 (lang & 0xff0000)>>16); 81 (lang & 0xff0000)>>16);
68 82
69 data_len = 0; 83 data_len = 0;
70 if (!::VerQueryValue(file_version_info, info_name, 84 if (!::VerQueryValue(file_version_info, info_name,
71 reinterpret_cast<LPVOID *>(&data), reinterpret_cast<UINT *>(&data_len))) 85 reinterpret_cast<LPVOID *>(&data), reinterpret_cast<UINT *>(&data_len)))
72 return false; 86 return false;
73 if (data_len <= 0 || data_len >= out_len) 87 if (data_len <= 0 || data_len >= (out_len / sizeof(wchar_t)))
74 return false; 88 return false;
75 89
76 memset(buffer, 0, out_len); 90 memset(buffer, 0, out_len);
77 ::StringCchCopyN(buffer, out_len, (const wchar_t*)data, data_len); 91 ::StringCchCopyN(buffer,
92 (out_len / sizeof(wchar_t)),
93 reinterpret_cast<const wchar_t*>(data),
94 data_len);
78 return true; 95 return true;
79 } 96 }
80 97
81 // Return true if we can re-offer Chrome; false, otherwise. 98 // Return true if we can re-offer Chrome; false, otherwise.
82 // Each partner can only offer Chrome once every six months. 99 // Each partner can only offer Chrome once every six months.
83 bool CanReOfferChrome(BOOL set_flag) { 100 bool CanReOfferChrome(BOOL set_flag) {
84 wchar_t filename[MAX_PATH+1]; 101 wchar_t filename[MAX_PATH+1];
85 wchar_t company[MAX_PATH]; 102 wchar_t company[MAX_PATH];
86 103
87 // If we cannot retrieve the version info of the executable or company 104 // If we cannot retrieve the version info of the executable or company
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after
133 } 150 }
134 151
135 ::RegCloseKey(key); 152 ::RegCloseKey(key);
136 } 153 }
137 154
138 return can_re_offer; 155 return can_re_offer;
139 } 156 }
140 157
141 // Helper function to read a value from registry. Returns true if value 158 // Helper function to read a value from registry. Returns true if value
142 // is read successfully and stored in parameter value. Returns false otherwise. 159 // is read successfully and stored in parameter value. Returns false otherwise.
143 bool ReadValueFromRegistry(HKEY root_key, const wchar_t *sub_key, 160 bool ReadValueFromRegistry(HKEY root_key, const wchar_t* sub_key,
144 const wchar_t *value_name, wchar_t *value, 161 const wchar_t* value_name, wchar_t* value,
145 size_t *size) { 162 size_t* size) {
146 HKEY key; 163 HKEY key;
147 if ((::RegOpenKeyEx(root_key, sub_key, NULL, 164 if ((::RegOpenKeyEx(root_key, sub_key, NULL,
148 KEY_READ, &key) == ERROR_SUCCESS) && 165 KEY_READ, &key) == ERROR_SUCCESS) &&
149 (::RegQueryValueEx(key, value_name, NULL, NULL, 166 (::RegQueryValueEx(key, value_name, NULL, NULL,
150 reinterpret_cast<LPBYTE>(value), 167 reinterpret_cast<LPBYTE>(value),
151 reinterpret_cast<LPDWORD>(size)) == ERROR_SUCCESS)) { 168 reinterpret_cast<LPDWORD>(size)) == ERROR_SUCCESS)) {
152 ::RegCloseKey(key); 169 ::RegCloseKey(key);
153 return true; 170 return true;
154 } 171 }
155 return false; 172 return false;
156 } 173 }
157 174
158 bool IsChromeInstalled(HKEY root_key) { 175 bool IsChromeInstalled(HKEY root_key) {
159 wchar_t version[64]; 176 wchar_t version[64];
160 size_t size = _countof(version); 177 size_t size = _countof(version);
161 return ReadValueFromRegistry(root_key, kChromeRegClientsKey, 178 return ReadValueFromRegistry(root_key, kChromeRegClientsKey,
162 kChromeRegVersion, version, &size); 179 kChromeRegVersion, version, &size);
163 } 180 }
164 181
165 enum WindowsVersion { 182 enum WindowsVersion {
166 VERSION_BELOW_XP_SP2, 183 VERSION_BELOW_XP_SP2,
167 VERSION_XP_SP2_UP_TO_VISTA, // "but not including" 184 VERSION_XP_SP2_UP_TO_VISTA, // "but not including"
168 VERSION_VISTA_OR_HIGHER, 185 VERSION_VISTA_OR_HIGHER,
169 }; 186 };
170 WindowsVersion GetWindowsVersion() { 187 WindowsVersion GetWindowsVersion() {
171 OSVERSIONINFOEX version_info = { sizeof version_info }; 188 OSVERSIONINFOEX version_info = { sizeof version_info };
172 GetVersionEx(reinterpret_cast<OSVERSIONINFO*>(&version_info)); 189 GetVersionEx(reinterpret_cast<OSVERSIONINFO*>(&version_info));
173 190
174 // Windows Vista is version 6.0. 191 // Windows Vista is version 6.0.
175 if (version_info.dwMajorVersion >= 6) 192 if (version_info.dwMajorVersion >= 6)
176 return VERSION_VISTA_OR_HIGHER; 193 return VERSION_VISTA_OR_HIGHER;
177 194
(...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after
278 } 295 }
279 ::CloseHandle(process_token); 296 ::CloseHandle(process_token);
280 } 297 }
281 ::CloseHandle(process_handle); 298 ::CloseHandle(process_handle);
282 return result; 299 return result;
283 } 300 }
284 } // namespace 301 } // namespace
285 302
286 #pragma comment(linker, "/EXPORT:GoogleChromeCompatibilityCheck=_GoogleChromeCom patibilityCheck@8,PRIVATE") 303 #pragma comment(linker, "/EXPORT:GoogleChromeCompatibilityCheck=_GoogleChromeCom patibilityCheck@8,PRIVATE")
287 DLLEXPORT BOOL __stdcall GoogleChromeCompatibilityCheck(BOOL set_flag, 304 DLLEXPORT BOOL __stdcall GoogleChromeCompatibilityCheck(BOOL set_flag,
288 DWORD *reasons) { 305 DWORD* reasons) {
289 DWORD local_reasons = 0; 306 DWORD local_reasons = 0;
290 307
291 WindowsVersion windows_version = GetWindowsVersion(); 308 WindowsVersion windows_version = GetWindowsVersion();
292 // System requirements? 309 // System requirements?
293 if (windows_version == VERSION_BELOW_XP_SP2) 310 if (windows_version == VERSION_BELOW_XP_SP2)
294 local_reasons |= GCCC_ERROR_OSNOTSUPPORTED; 311 local_reasons |= GCCC_ERROR_OSNOTSUPPORTED;
295 312
296 if (IsChromeInstalled(HKEY_LOCAL_MACHINE)) 313 if (IsChromeInstalled(HKEY_LOCAL_MACHINE))
297 local_reasons |= GCCC_ERROR_SYSTEMLEVELALREADYPRESENT; 314 local_reasons |= GCCC_ERROR_SYSTEMLEVELALREADYPRESENT;
298 315
(...skipping 145 matching lines...) Expand 10 before | Expand all | Expand 10 after
444 // This loop iterates through all of the top-level Windows named 461 // This loop iterates through all of the top-level Windows named
445 // Chrome_WindowImpl_0, and looks for the first one with any children. 462 // Chrome_WindowImpl_0, and looks for the first one with any children.
446 while (handle && !FindWindowEx(handle, NULL, L"Chrome_WindowImpl_0", NULL)) { 463 while (handle && !FindWindowEx(handle, NULL, L"Chrome_WindowImpl_0", NULL)) {
447 // Get the next top-level Chrome window. 464 // Get the next top-level Chrome window.
448 handle = FindWindowEx(NULL, handle, L"Chrome_WindowImpl_0", NULL); 465 handle = FindWindowEx(NULL, handle, L"Chrome_WindowImpl_0", NULL);
449 } 466 }
450 467
451 return (handle && 468 return (handle &&
452 SetWindowPos(handle, 0, x, y, width, height, SWP_NOZORDER)); 469 SetWindowPos(handle, 0, x, y, width, height, SWP_NOZORDER));
453 } 470 }
471
472 #pragma comment(linker, "/EXPORT:LaunchGoogleChromeWithDimensions=_LaunchGoogleC hromeWithDimensions@24,PRIVATE")
grt (UTC plus 2) 2011/11/11 17:01:01 LaunchGoogleChromeWithDimensions -> GoogleChromeDa
robertshield 2011/11/12 05:25:38 heh, done
473 DLLEXPORT int __stdcall GoogleChromeDaysSinceLastRun() {
474 using base::win::RegKey;
475 using base::Time;
476 using base::TimeDelta;
477
478 int days_since_last_run = std::numeric_limits<int>::max();
479
480 HKEY hives[] = {HKEY_LOCAL_MACHINE, HKEY_CURRENT_USER};
grt (UTC plus 2) 2011/11/11 17:01:01 Chrome doesn't even try to write to HKLM since it
robertshield 2011/11/12 05:25:38 so, as discussed modified this to assume that one
481 for (int i = 0; i < arraysize(hives); ++i) {
482 RegKey client_state(hives[i],
483 kChromeRegClientStateKey,
484 KEY_QUERY_VALUE);
485 if (client_state.Valid()) {
486 std::wstring last_run;
487 int64 last_run_value = 0;
488 if (client_state.ReadValue(google_update::kRegLastRunTimeField,
489 &last_run) == ERROR_SUCCESS &&
490 base::StringToInt64(last_run, &last_run_value)) {
491 Time last_run_time = Time::FromInternalValue(last_run_value);
492 TimeDelta difference = Time::NowFromSystemTime() - last_run_time;
493 int days_elapsed = difference.InDays();
grt (UTC plus 2) 2011/11/11 17:01:01 include <algorithm> and replace these four lines w
robertshield 2011/11/12 05:25:38 Done.
494 if (days_elapsed < days_since_last_run) {
495 days_since_last_run = days_elapsed;
496 }
497 }
498 }
499 }
500
501 if (days_since_last_run == std::numeric_limits<int>::max()) {
502 days_since_last_run = -1;
503 }
504
505 return days_since_last_run;
506 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698