google_update_settings.cc revision eb525c5499e34cc9c4b825d6d9e75bb07cc06ace
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/google_update_settings.h"
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <algorithm>
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/path_service.h"
112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_number_conversions.h"
12868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h"
13868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h"
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/thread_restrictions.h"
15eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "base/time/time.h"
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/registry.h"
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_switches.h"
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/browser_distribution.h"
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/channel_info.h"
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/google_update_constants.h"
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/install_util.h"
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/installation_state.h"
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/product.h"
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::win::RegKey;
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using installer::InstallationState;
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace {
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kGoogleUpdatePoliciesKey[] =
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    L"SOFTWARE\\Policies\\Google\\Update";
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kGoogleUpdateUpdatePolicyValue[] = L"UpdateDefault";
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kGoogleUpdateUpdateOverrideValuePrefix[] = L"Update";
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GoogleUpdateSettings::UpdatePolicy kGoogleUpdateDefaultUpdatePolicy =
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(GOOGLE_CHROME_BUILD)
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    GoogleUpdateSettings::AUTOMATIC_UPDATES;
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    GoogleUpdateSettings::UPDATES_DISABLED;
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ReadGoogleUpdateStrKey(const wchar_t* const name, std::wstring* value) {
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_path = dist->GetStateKey();
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key(HKEY_CURRENT_USER, reg_path.c_str(), KEY_READ);
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (key.ReadValue(name, value) != ERROR_SUCCESS) {
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    RegKey hklm_key(HKEY_LOCAL_MACHINE, reg_path.c_str(), KEY_READ);
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return (hklm_key.ReadValue(name, value) == ERROR_SUCCESS);
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool WriteGoogleUpdateStrKeyInternal(BrowserDistribution* dist,
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                     const wchar_t* const name,
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                     const std::wstring& value) {
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(dist);
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_path(dist->GetStateKey());
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key(HKEY_CURRENT_USER, reg_path.c_str(), KEY_SET_VALUE);
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return (key.WriteValue(name, value.c_str()) == ERROR_SUCCESS);
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool WriteGoogleUpdateStrKey(const wchar_t* const name,
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                             const std::wstring& value) {
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return WriteGoogleUpdateStrKeyInternal(dist, name, value);
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool WriteGoogleUpdateStrKeyMultiInstall(BrowserDistribution* dist,
682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                         const wchar_t* const name,
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                         const std::wstring& value,
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                         bool system_level) {
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool result = WriteGoogleUpdateStrKeyInternal(dist, name, value);
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!InstallUtil::IsMultiInstall(dist, system_level))
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return result;
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // It is a multi-install distro. Must write the reg value again.
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* multi_dist =
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      BrowserDistribution::GetSpecificDistribution(
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          BrowserDistribution::CHROME_BINARIES);
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return WriteGoogleUpdateStrKeyInternal(multi_dist, name, value) && result;
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ClearGoogleUpdateStrKey(const wchar_t* const name) {
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_path = dist->GetStateKey();
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key(HKEY_CURRENT_USER, reg_path.c_str(), KEY_READ | KEY_WRITE);
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring value;
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (key.ReadValue(name, &value) != ERROR_SUCCESS)
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return (key.WriteValue(name, L"") == ERROR_SUCCESS);
895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool RemoveGoogleUpdateStrKey(const wchar_t* const name) {
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_path = dist->GetStateKey();
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key(HKEY_CURRENT_USER, reg_path.c_str(), KEY_READ | KEY_WRITE);
955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!key.HasValue(name))
965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return true;
975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return (key.DeleteValue(name) == ERROR_SUCCESS);
985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GetChromeChannelInternal(bool system_install,
1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                              bool add_multi_modifier,
1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                              string16* channel) {
1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (dist->GetChromeChannel(channel)) {
1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return true;
1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  string16 reg_path = dist->GetStateKey();
1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key(root_key, reg_path.c_str(), KEY_READ);
1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  installer::ChannelInfo channel_info;
1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!channel_info.Initialize(key)) {
1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    channel->assign(installer::kChromeChannelUnknown);
1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!channel_info.GetChannelName(channel)) {
1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    channel->assign(installer::kChromeChannelUnknown);
1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Tag the channel name if this is a multi-install.
1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (add_multi_modifier && channel_info.IsMultiInstall()) {
1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!channel->empty()) {
1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      channel->append(1, L'-');
1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    channel->append(1, L'm');
1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Populates |update_policy| with the UpdatePolicy enum value corresponding to a
1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DWORD read from the registry and returns true if |value| is within range.
1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// If |value| is out of range, returns false without modifying |update_policy|.
1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GetUpdatePolicyFromDword(
1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const DWORD value,
1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    GoogleUpdateSettings::UpdatePolicy* update_policy) {
1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  switch (value) {
1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case GoogleUpdateSettings::UPDATES_DISABLED:
1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case GoogleUpdateSettings::AUTOMATIC_UPDATES:
1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case GoogleUpdateSettings::MANUAL_UPDATES_ONLY:
143eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch    case GoogleUpdateSettings::AUTO_UPDATES_ONLY:
1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      *update_policy = static_cast<GoogleUpdateSettings::UpdatePolicy>(value);
1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return true;
1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    default:
1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      LOG(WARNING) << "Unexpected update policy override value: " << value;
1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return false;
1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace
1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::IsSystemInstall() {
1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool system_install = false;
1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  base::FilePath module_dir;
1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!PathService::Get(base::DIR_MODULE, &module_dir)) {
1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    LOG(WARNING)
1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        << "Failed to get directory of module; assuming per-user install.";
1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    system_install = !InstallUtil::IsPerUserInstall(module_dir.value().c_str());
1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return system_install;
1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetCollectStatsConsent() {
1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return GetCollectStatsConsentAtLevel(IsSystemInstall());
1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Older versions of Chrome unconditionally read from HKCU\...\ClientState\...
1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and then HKLM\...\ClientState\....  This means that system-level Chrome
1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// never checked ClientStateMedium (which has priority according to Google
1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Update) and gave preference to a value in HKCU (which was never checked by
1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Google Update).  From now on, Chrome follows Google Update's policy.
1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetCollectStatsConsentAtLevel(bool system_install) {
1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Consent applies to all products in a multi-install package.
1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (InstallUtil::IsMultiInstall(dist, system_install)) {
1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    dist = BrowserDistribution::GetSpecificDistribution(
1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        BrowserDistribution::CHROME_BINARIES);
1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key;
1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DWORD value = 0;
1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool have_value = false;
1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // For system-level installs, try ClientStateMedium first.
1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  have_value =
1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      system_install &&
1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      key.Open(HKEY_LOCAL_MACHINE, dist->GetStateMediumKey().c_str(),
1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)               KEY_QUERY_VALUE) == ERROR_SUCCESS &&
1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      key.ReadValueDW(google_update::kRegUsageStatsField,
1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                      &value) == ERROR_SUCCESS;
1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Otherwise, try ClientState.
1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!have_value) {
1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    have_value =
1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        key.Open(system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 dist->GetStateKey().c_str(),
2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 KEY_QUERY_VALUE) == ERROR_SUCCESS &&
2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        key.ReadValueDW(google_update::kRegUsageStatsField,
2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                        &value) == ERROR_SUCCESS;
2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Google Update specifically checks that the value is 1, so we do the same.
2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return have_value && value == 1;
2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::SetCollectStatsConsent(bool consented) {
2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return SetCollectStatsConsentAtLevel(IsSystemInstall(), consented);
2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::SetCollectStatsConsentAtLevel(bool system_install,
2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                                         bool consented) {
2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Google Update writes and expects 1 for true, 0 for false.
2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DWORD value = consented ? 1 : 0;
2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Consent applies to all products in a multi-install package.
2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (InstallUtil::IsMultiInstall(dist, system_install)) {
2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    dist = BrowserDistribution::GetSpecificDistribution(
2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        BrowserDistribution::CHROME_BINARIES);
2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Write to ClientStateMedium for system-level; ClientState otherwise.
2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_path =
2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      system_install ? dist->GetStateMediumKey() : dist->GetStateKey();
2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key;
2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LONG result = key.Create(root_key, reg_path.c_str(), KEY_SET_VALUE);
2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (result != ERROR_SUCCESS) {
2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    LOG(ERROR) << "Failed opening key " << reg_path << " to set "
2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)               << google_update::kRegUsageStatsField << "; result: " << result;
2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    result = key.WriteValue(google_update::kRegUsageStatsField, value);
2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    LOG_IF(ERROR, result != ERROR_SUCCESS) << "Failed setting "
2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        << google_update::kRegUsageStatsField << " in key " << reg_path
2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        << "; result: " << result;
2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return (result == ERROR_SUCCESS);
2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetMetricsId(std::wstring* metrics_id) {
2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ReadGoogleUpdateStrKey(google_update::kRegMetricsId, metrics_id);
2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::SetMetricsId(const std::wstring& metrics_id) {
2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return WriteGoogleUpdateStrKey(google_update::kRegMetricsId, metrics_id);
2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// EULA consent is only relevant for system-level installs.
2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::SetEULAConsent(
2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const InstallationState& machine_state,
2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    BrowserDistribution* dist,
2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool consented) {
2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(dist);
2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const DWORD eula_accepted = consented ? 1 : 0;
2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_path = dist->GetStateMediumKey();
2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool succeeded = true;
2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key;
2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Write the consent value into the product's ClientStateMedium key.
2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (key.Create(HKEY_LOCAL_MACHINE, reg_path.c_str(),
2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 KEY_SET_VALUE) != ERROR_SUCCESS ||
2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      key.WriteValue(google_update::kRegEULAAceptedField,
2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                     eula_accepted) != ERROR_SUCCESS) {
2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    succeeded = false;
2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // If this is a multi-install, also write it into the binaries' key.
2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // --mutli-install is not provided on the command-line, so deduce it from
2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // the product's state.
2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const installer::ProductState* product_state =
2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      machine_state.GetProductState(true, dist->GetType());
2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (product_state != NULL && product_state->is_multi_install()) {
2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    dist = BrowserDistribution::GetSpecificDistribution(
2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        BrowserDistribution::CHROME_BINARIES);
2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    reg_path = dist->GetStateMediumKey();
2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (key.Create(HKEY_LOCAL_MACHINE, reg_path.c_str(),
2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                   KEY_SET_VALUE) != ERROR_SUCCESS ||
2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        key.WriteValue(google_update::kRegEULAAceptedField,
2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       eula_accepted) != ERROR_SUCCESS) {
2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        succeeded = false;
2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return succeeded;
2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int GoogleUpdateSettings::GetLastRunTime() {
2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring time_s;
2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!ReadGoogleUpdateStrKey(google_update::kRegLastRunTimeField, &time_s))
2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return -1;
2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int64 time_i;
2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!base::StringToInt64(time_s, &time_i))
2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return -1;
2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  base::TimeDelta td =
3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Time::NowFromSystemTime() - base::Time::FromInternalValue(time_i);
3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return td.InDays();
3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::SetLastRunTime() {
3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int64 time = base::Time::NowFromSystemTime().ToInternalValue();
3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return WriteGoogleUpdateStrKey(google_update::kRegLastRunTimeField,
3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                 base::Int64ToString16(time));
3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::RemoveLastRunTime() {
3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return RemoveGoogleUpdateStrKey(google_update::kRegLastRunTimeField);
3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetBrowser(std::wstring* browser) {
3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ReadGoogleUpdateStrKey(google_update::kRegBrowserField, browser);
3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetLanguage(std::wstring* language) {
3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ReadGoogleUpdateStrKey(google_update::kRegLangField, language);
3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetBrand(std::wstring* brand) {
3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ReadGoogleUpdateStrKey(google_update::kRegRLZBrandField, brand);
3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetReactivationBrand(std::wstring* brand) {
3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ReadGoogleUpdateStrKey(google_update::kRegRLZReactivationBrandField,
3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                brand);
3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetClient(std::wstring* client) {
3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ReadGoogleUpdateStrKey(google_update::kRegClientField, client);
3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::SetClient(const std::wstring& client) {
3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return WriteGoogleUpdateStrKey(google_update::kRegClientField, client);
3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetReferral(std::wstring* referral) {
3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ReadGoogleUpdateStrKey(google_update::kRegReferralField, referral);
3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::ClearReferral() {
3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ClearGoogleUpdateStrKey(google_update::kRegReferralField);
3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::UpdateDidRunState(bool did_run,
3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                             bool system_level) {
3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return UpdateDidRunStateForDistribution(dist, did_run, system_level);
3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
3532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool GoogleUpdateSettings::UpdateDidRunStateForDistribution(
3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    BrowserDistribution* dist,
3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    bool did_run,
3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    bool system_level) {
3572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return WriteGoogleUpdateStrKeyMultiInstall(dist,
3582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                             google_update::kRegDidRunField,
3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                             did_run ? L"1" : L"0",
3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                             system_level);
3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)std::wstring GoogleUpdateSettings::GetChromeChannel(bool system_install) {
3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring channel;
3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  GetChromeChannelInternal(system_install, false, &channel);
3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return channel;
3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetChromeChannelAndModifiers(bool system_install,
3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                                        string16* channel) {
3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return GetChromeChannelInternal(system_install, true, channel);
3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void GoogleUpdateSettings::UpdateInstallStatus(bool system_install,
3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    installer::ArchiveType archive_type, int install_return_code,
3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::wstring& product_guid) {
3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(archive_type != installer::UNKNOWN_ARCHIVE_TYPE ||
3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)         install_return_code != 0);
3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  HKEY reg_root = (system_install) ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key;
3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  installer::ChannelInfo channel_info;
3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_key(google_update::kRegPathClientState);
3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  reg_key.append(L"\\");
3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  reg_key.append(product_guid);
3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LONG result = key.Open(reg_root, reg_key.c_str(),
3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                         KEY_QUERY_VALUE | KEY_SET_VALUE);
3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (result == ERROR_SUCCESS)
3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    channel_info.Initialize(key);
3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  else if (result != ERROR_FILE_NOT_FOUND)
3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    LOG(ERROR) << "Failed to open " << reg_key << "; Error: " << result;
3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (UpdateGoogleUpdateApKey(archive_type, install_return_code,
3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                              &channel_info)) {
3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // We have a modified channel_info value to write.
3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Create the app's ClientState key if it doesn't already exist.
3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!key.Valid()) {
3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      result = key.Open(reg_root, google_update::kRegPathClientState,
3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                        KEY_CREATE_SUB_KEY);
4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (result == ERROR_SUCCESS)
4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        result = key.CreateKey(product_guid.c_str(), KEY_SET_VALUE);
4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (result != ERROR_SUCCESS) {
4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        LOG(ERROR) << "Failed to create " << reg_key << "; Error: " << result;
4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        return;
4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!channel_info.Write(&key)) {
4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      LOG(ERROR) << "Failed to write to application's ClientState key "
4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 << google_update::kRegApField << " = " << channel_info.value();
4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::UpdateGoogleUpdateApKey(
4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    installer::ArchiveType archive_type, int install_return_code,
4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    installer::ChannelInfo* value) {
4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(archive_type != installer::UNKNOWN_ARCHIVE_TYPE ||
4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)         install_return_code != 0);
4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool modified = false;
4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (archive_type == installer::FULL_ARCHIVE_TYPE || !install_return_code) {
4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (value->SetFullSuffix(false)) {
4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      VLOG(1) << "Removed incremental installer failure key; "
4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 "switching to channel: "
4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              << value->value();
4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      modified = true;
4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else if (archive_type == installer::INCREMENTAL_ARCHIVE_TYPE) {
4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (value->SetFullSuffix(true)) {
4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      VLOG(1) << "Incremental installer failed; switching to channel: "
4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              << value->value();
4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      modified = true;
4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    } else {
4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      VLOG(1) << "Incremental installer failure; already on channel: "
4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              << value->value();
4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // It's okay if we don't know the archive type.  In this case, leave the
4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // "-full" suffix as we found it.
4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK_EQ(installer::UNKNOWN_ARCHIVE_TYPE, archive_type);
4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (value->SetMultiFailSuffix(false)) {
4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    VLOG(1) << "Removed multi-install failure key; switching to channel: "
4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)            << value->value();
4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    modified = true;
4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return modified;
4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int GoogleUpdateSettings::DuplicateGoogleUpdateSystemClientKey() {
4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring reg_path = dist->GetStateKey();
4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Minimum access needed is to be able to write to this key.
4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey reg_key(HKEY_LOCAL_MACHINE, reg_path.c_str(), KEY_SET_VALUE);
4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!reg_key.Valid())
4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return 0;
4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  HANDLE target_handle = 0;
4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!DuplicateHandle(GetCurrentProcess(), reg_key.Handle(),
4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       GetCurrentProcess(), &target_handle, KEY_SET_VALUE,
4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       TRUE, DUPLICATE_SAME_ACCESS)) {
4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return 0;
4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return reinterpret_cast<int>(target_handle);
4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::WriteGoogleUpdateSystemClientKey(
4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int handle, const std::wstring& key, const std::wstring& value) {
4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  HKEY reg_key = reinterpret_cast<HKEY>(reinterpret_cast<void*>(handle));
4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DWORD size = static_cast<DWORD>(value.size()) * sizeof(wchar_t);
4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LSTATUS status = RegSetValueEx(reg_key, key.c_str(), 0, REG_SZ,
4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      reinterpret_cast<const BYTE*>(value.c_str()), size);
4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return status == ERROR_SUCCESS;
4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)GoogleUpdateSettings::UpdatePolicy GoogleUpdateSettings::GetAppUpdatePolicy(
4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::wstring& app_guid,
4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool* is_overridden) {
4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool found_override = false;
4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  UpdatePolicy update_policy = kGoogleUpdateDefaultUpdatePolicy;
4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(GOOGLE_CHROME_BUILD)
4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!app_guid.empty());
4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey policy_key;
4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Google Update Group Policy settings are always in HKLM.
4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (policy_key.Open(HKEY_LOCAL_MACHINE, kGoogleUpdatePoliciesKey,
4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                      KEY_QUERY_VALUE) == ERROR_SUCCESS) {
4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    static const size_t kPrefixLen =
4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        arraysize(kGoogleUpdateUpdateOverrideValuePrefix) - 1;
4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DWORD value;
4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    std::wstring app_update_override;
4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    app_update_override.reserve(kPrefixLen + app_guid.size());
4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    app_update_override.append(kGoogleUpdateUpdateOverrideValuePrefix,
4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               kPrefixLen);
5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    app_update_override.append(app_guid);
5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // First try to read and comprehend the app-specific override.
5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    found_override = (policy_key.ReadValueDW(app_update_override.c_str(),
5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                             &value) == ERROR_SUCCESS &&
5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                      GetUpdatePolicyFromDword(value, &update_policy));
5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Failing that, try to read and comprehend the default override.
5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!found_override &&
5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        policy_key.ReadValueDW(kGoogleUpdateUpdatePolicyValue,
5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               &value) == ERROR_SUCCESS) {
5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      GetUpdatePolicyFromDword(value, &update_policy);
5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif  // defined(GOOGLE_CHROME_BUILD)
5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (is_overridden != NULL)
5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    *is_overridden = found_override;
5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return update_policy;
5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)string16 GoogleUpdateSettings::GetUninstallCommandLine(bool system_install) {
5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  string16 cmd_line;
5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey update_key;
5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (update_key.Open(root_key, google_update::kRegPathGoogleUpdate,
5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                      KEY_QUERY_VALUE) == ERROR_SUCCESS) {
5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    update_key.ReadValue(google_update::kRegUninstallCmdLine, &cmd_line);
5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return cmd_line;
5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Version GoogleUpdateSettings::GetGoogleUpdateVersion(bool system_install) {
5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  string16 version;
5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey key;
5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (key.Open(root_key,
5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)               google_update::kRegPathGoogleUpdate,
5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)               KEY_QUERY_VALUE) == ERROR_SUCCESS &&
5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      key.ReadValue(google_update::kRegGoogleUpdateVersion,
5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                    &version) == ERROR_SUCCESS) {
5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return Version(UTF16ToUTF8(version));
5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return Version();
5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::Time GoogleUpdateSettings::GetGoogleUpdateLastStartedAU(
5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool system_install) {
5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey update_key;
5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (update_key.Open(root_key, google_update::kRegPathGoogleUpdate,
5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                      KEY_QUERY_VALUE) == ERROR_SUCCESS) {
5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DWORD last_start;
5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (update_key.ReadValueDW(google_update::kRegLastStartedAUField,
5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               &last_start) == ERROR_SUCCESS) {
5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return base::Time::FromTimeT(last_start);
5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return base::Time();
5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::Time GoogleUpdateSettings::GetGoogleUpdateLastChecked(
5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool system_install) {
5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey update_key;
5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (update_key.Open(root_key, google_update::kRegPathGoogleUpdate,
5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                      KEY_QUERY_VALUE) == ERROR_SUCCESS) {
5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DWORD last_check;
5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (update_key.ReadValueDW(google_update::kRegLastCheckedField,
5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               &last_check) == ERROR_SUCCESS) {
5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return base::Time::FromTimeT(last_check);
5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return base::Time();
5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetUpdateDetailForApp(bool system_install,
5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                                 const wchar_t* app_guid,
5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                                 ProductData* data) {
5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(app_guid);
5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(data);
5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool product_found = false;
5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  string16 clientstate_reg_path(google_update::kRegPathClientState);
5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  clientstate_reg_path.append(L"\\");
5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  clientstate_reg_path.append(app_guid);
5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  RegKey clientstate;
5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (clientstate.Open(root_key, clientstate_reg_path.c_str(),
5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       KEY_QUERY_VALUE) == ERROR_SUCCESS) {
6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    string16 version;
6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DWORD dword_value;
6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if ((clientstate.ReadValueDW(google_update::kRegLastCheckSuccessField,
6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                 &dword_value) == ERROR_SUCCESS) &&
6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        (clientstate.ReadValue(google_update::kRegVersionField,
6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               &version) == ERROR_SUCCESS)) {
6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      product_found = true;
6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      data->version = WideToASCII(version);
6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      data->last_success = base::Time::FromTimeT(dword_value);
6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      data->last_result = 0;
6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      data->last_error_code = 0;
6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      data->last_extra_code = 0;
6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (clientstate.ReadValueDW(google_update::kRegLastInstallerResultField,
6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                  &dword_value) == ERROR_SUCCESS) {
6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Google Update convention is that if an installer writes an result
6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // code that is invalid, it is clamped to an exit code result.
6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        const DWORD kMaxValidInstallResult = 4;  // INSTALLER_RESULT_EXIT_CODE
6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        data->last_result = std::min(dword_value, kMaxValidInstallResult);
6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (clientstate.ReadValueDW(google_update::kRegLastInstallerErrorField,
6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                  &dword_value) == ERROR_SUCCESS) {
6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        data->last_error_code = dword_value;
6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (clientstate.ReadValueDW(google_update::kRegLastInstallerExtraField,
6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                  &dword_value) == ERROR_SUCCESS) {
6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        data->last_extra_code = dword_value;
6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return product_found;
6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetUpdateDetailForGoogleUpdate(bool system_install,
6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                                          ProductData* data) {
6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return GetUpdateDetailForApp(system_install,
6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               google_update::kGoogleUpdateUpgradeCode,
6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               data);
6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GoogleUpdateSettings::GetUpdateDetail(bool system_install,
6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                           ProductData* data) {
6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return GetUpdateDetailForApp(system_install,
6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               dist->GetAppGuid().c_str(),
6465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               data);
6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool GoogleUpdateSettings::SetExperimentLabels(
6502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    bool system_install,
6512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const string16& experiment_labels) {
6522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  HKEY reg_root = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
6532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Use the browser distribution and install level to write to the correct
6552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // client state/app guid key.
6562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  bool success = false;
6572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
6582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (dist->ShouldSetExperimentLabels()) {
6592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    string16 client_state_path(
6602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        system_install ? dist->GetStateMediumKey() : dist->GetStateKey());
6612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    RegKey client_state(
6622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        reg_root, client_state_path.c_str(), KEY_SET_VALUE);
6632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (experiment_labels.empty()) {
6642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      success = client_state.DeleteValue(google_update::kExperimentLabels)
6652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          == ERROR_SUCCESS;
6662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    } else {
6672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      success = client_state.WriteValue(google_update::kExperimentLabels,
6682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          experiment_labels.c_str()) == ERROR_SUCCESS;
6692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    }
6702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
6712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return success;
6732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
6742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool GoogleUpdateSettings::ReadExperimentLabels(
6762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    bool system_install,
6772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    string16* experiment_labels) {
6782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  HKEY reg_root = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
6792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // If this distribution does not set the experiment labels, don't bother
6812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // reading.
6822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
6832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (!dist->ShouldSetExperimentLabels())
6842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return false;
6852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  string16 client_state_path(
6872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      system_install ? dist->GetStateMediumKey() : dist->GetStateKey());
6882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RegKey client_state;
6902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  LONG result =
6912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      client_state.Open(reg_root, client_state_path.c_str(), KEY_QUERY_VALUE);
6922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result == ERROR_SUCCESS) {
6932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    result = client_state.ReadValue(google_update::kExperimentLabels,
6942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                    experiment_labels);
6952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
6962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // If the key or value was not present, return the empty string.
6982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result == ERROR_FILE_NOT_FOUND || result == ERROR_PATH_NOT_FOUND) {
6992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    experiment_labels->clear();
7002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return true;
7012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
7022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return result == ERROR_SUCCESS;
7042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
705