Merge 68951 - Added refreshing of elevation policy on first run of new versio...

ceee/ie/common/ceee_module_util.cc

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 //
 // CEEE module-wide utilities.
 
 #include "ceee/ie/common/ceee_module_util.h"
 
+#include <iepmapi.h>
+
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/path_service.h"
 #include "base/stringize_macros.h"
 #include "base/win/registry.h"
+#include "ceee/common/com_utils.h"
 #include "ceee/common/process_utils_win.h"
+#include "ceee/ie/common/ie_util.h"
 #include "chrome/installer/util/google_update_constants.h"
 
 #include "version.h"  // NOLINT
 
 namespace {
 
 const wchar_t* kRegistryPath = L"SOFTWARE\\Google\\CEEE";
 const wchar_t* kRegistryValueToolbandIsHidden = L"toolband_is_hidden";
 const wchar_t* kRegistryValueToolbandPlaced = L"toolband_placed";
 const wchar_t* kRegistryValueCrxInstalledPath = L"crx_installed_path";
@@ skipping 273 matching lines @@
   base::win::RegKey key(HKEY_CURRENT_USER, reg_path.c_str(), KEY_READ);
   DWORD value;
   if (!key.ReadValueDW(google_update::kRegUsageStatsField, &value)) {
     base::win::RegKey hklm_key(HKEY_LOCAL_MACHINE, reg_path.c_str(), KEY_READ);
     if (!hklm_key.ReadValueDW(google_update::kRegUsageStatsField, &value))
       return false;
   }
   return (1 == value);
 }
 
+bool RefreshElevationPolicyIfNeeded() {
+  if (ie_util::GetIeVersion() < ie_util::IEVERSION_IE7)
+    return false;
+
+  // This may access InternetRegistry instead of real one. However this is
+  // acceptable, we just refresh policy twice.
+  base::win::RegKey hkcu(HKEY_CURRENT_USER, kRegistryPath,
+                         KEY_WRITE | KEY_QUERY_VALUE);
+  LOG_IF(ERROR, !hkcu.Valid()) << "Failed to open reg key: " << kRegistryPath;
+  if (!hkcu.Valid())
+    return false;
+
+  std::wstring expected_version = TO_L_STRING(CHROME_VERSION_STRING);
+
+  static const wchar_t kValueName[] = L"last_elevation_refresh";
+  std::wstring last_elevation_refresh_version;
+  bool result = hkcu.ReadValue(kValueName, &last_elevation_refresh_version);
+  if (last_elevation_refresh_version == expected_version)
+    return false;
+
+  HRESULT hr = ::IERefreshElevationPolicy();
+  VLOG(1) << "Elevation policy refresh result: " << com::LogHr(hr);
+
+  // Write after refreshing because it's better to refresh twice, than to miss
+  // once.
+  result = hkcu.WriteValue(kValueName, expected_version.c_str());
+  LOG_IF(ERROR, !result) << "Failed to write a registry value: " << kValueName;
+
+  return true;
+}
+
 }  // namespace ceee_module_util