Index: chrome_frame/chrome_tab.cc |
diff --git a/chrome_frame/chrome_tab.cc b/chrome_frame/chrome_tab.cc |
deleted file mode 100644 |
index 3c3c80a84a965934d8a8fc292cd7bbcba306703b..0000000000000000000000000000000000000000 |
--- a/chrome_frame/chrome_tab.cc |
+++ /dev/null |
@@ -1,1036 +0,0 @@ |
-// Copyright (c) 2012 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. |
- |
-// chrome_tab.cc : Implementation of DLL Exports. |
- |
-// Need to include this before the ATL headers below. |
-#include "chrome_frame/chrome_tab.h" |
- |
-#include <atlsecurity.h> |
-#include <objbase.h> |
- |
-#include "base/at_exit.h" |
-#include "base/basictypes.h" |
-#include "base/command_line.h" |
-#include "base/file_util.h" |
-#include "base/file_version_info.h" |
-#include "base/i18n/icu_util.h" |
-#include "base/logging.h" |
-#include "base/logging_win.h" |
-#include "base/path_service.h" |
-#include "base/process/launch.h" |
-#include "base/strings/string16.h" |
-#include "base/strings/string_number_conversions.h" |
-#include "base/strings/string_piece.h" |
-#include "base/strings/string_util.h" |
-#include "base/strings/sys_string_conversions.h" |
-#include "base/strings/utf_string_conversions.h" |
-#include "base/win/registry.h" |
-#include "base/win/windows_version.h" |
-#include "chrome/common/chrome_constants.h" |
-#include "chrome/common/chrome_switches.h" |
-#include "chrome/installer/util/google_update_settings.h" |
-#include "chrome_frame/bho.h" |
-#include "chrome_frame/chrome_active_document.h" |
-#include "chrome_frame/chrome_frame_activex.h" |
-#include "chrome_frame/chrome_frame_automation.h" |
-#include "chrome_frame/chrome_frame_reporting.h" |
-#include "chrome_frame/chrome_launcher_utils.h" |
-#include "chrome_frame/chrome_protocol.h" |
-#include "chrome_frame/dll_redirector.h" |
-#include "chrome_frame/exception_barrier.h" |
-#include "chrome_frame/pin_module.h" |
-#include "chrome_frame/resource.h" |
-#include "chrome_frame/utils.h" |
-#include "components/variations/entropy_provider.h" |
-#include "grit/chrome_frame_resources.h" |
-#include "url/url_util.h" |
- |
-#if _ATL_VER >= 0x0C00 |
-// This was removed between the VS2010 version and the VS2013 version, and |
-// the unsuffixed version was repurposed to mean 'S'. |
-#define UpdateRegistryFromResourceS UpdateRegistryFromResource |
-#endif |
- |
-using base::win::RegKey; |
- |
-namespace { |
- |
-const wchar_t kInternetSettings[] = |
- L"Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings"; |
- |
-const wchar_t kProtocolHandlers[] = |
- L"Software\\Classes\\Protocols\\Handler"; |
- |
-const wchar_t kRunOnce[] = |
- L"Software\\Microsoft\\Windows\\CurrentVersion\\RunOnce"; |
- |
-const wchar_t kRunKeyName[] = L"ChromeFrameHelper"; |
- |
-const wchar_t kChromeFrameHelperExe[] = L"chrome_frame_helper.exe"; |
-const wchar_t kChromeFrameHelperStartupArg[] = L"--startup"; |
- |
-// Window class and window names. |
-// TODO(robertshield): These and other constants need to be refactored into |
-// a common chrome_frame_constants.h|cc and built into a separate lib |
-// (either chrome_frame_utils or make another one). |
-const wchar_t kChromeFrameHelperWindowClassName[] = |
- L"ChromeFrameHelperWindowClass"; |
-const wchar_t kChromeFrameHelperWindowName[] = |
- L"ChromeFrameHelperWindowName"; |
- |
-// {0562BFC3-2550-45b4-BD8E-A310583D3A6F} |
-const GUID kChromeFrameProvider = |
- { 0x562bfc3, 0x2550, 0x45b4, |
- { 0xbd, 0x8e, 0xa3, 0x10, 0x58, 0x3d, 0x3a, 0x6f } }; |
- |
-const wchar_t kPostPlatformUAKey[] = |
- L"Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\" |
- L"User Agent\\Post Platform"; |
-const wchar_t kChromeFramePrefix[] = L"chromeframe/"; |
- |
-// See comments in DllGetClassObject. |
-LPFNGETCLASSOBJECT g_dll_get_class_object_redir_ptr = NULL; |
- |
-// This function has the side effect of initializing an unprotected |
-// vector pointer inside GoogleUrl. If this is called during DLL loading, |
-// it has the effect of avoiding an initialization race on that pointer. |
-// TODO(siggi): fix GoogleUrl. |
-void InitGoogleUrl() { |
- static const char kDummyUrl[] = "http://www.google.com"; |
- |
- url_util::IsStandard(kDummyUrl, |
- url_parse::MakeRange(0, arraysize(kDummyUrl))); |
-} |
- |
-class ChromeTabModule : public CAtlDllModuleT<ChromeTabModule> { |
- public: |
- typedef CAtlDllModuleT<ChromeTabModule> ParentClass; |
- |
- ChromeTabModule() : do_system_registration_(true), |
- crash_reporting_(NULL), |
- icu_initialized_(false) {} |
- |
- DECLARE_LIBID(LIBID_ChromeTabLib) |
- DECLARE_REGISTRY_APPID_RESOURCEID(IDR_CHROMETAB, |
- "{FD9B1B31-F4D8-436A-8F4F-D3C2E36733D3}") |
- |
- // Override to add our SYSTIME binary value to registry scripts. |
- // See chrome_frame_activex.rgs for usage. |
- virtual HRESULT AddCommonRGSReplacements(IRegistrarBase* registrar) throw() { |
- HRESULT hr = ParentClass::AddCommonRGSReplacements(registrar); |
- |
- if (SUCCEEDED(hr)) { |
- SYSTEMTIME local_time; |
- ::GetSystemTime(&local_time); |
- std::string hex(base::HexEncode(&local_time, sizeof(local_time))); |
- base::StringPiece sp_hex(hex); |
- hr = registrar->AddReplacement(L"SYSTIME", |
- base::SysNativeMBToWide(sp_hex).c_str()); |
- DCHECK(SUCCEEDED(hr)); |
- } |
- |
- if (SUCCEEDED(hr)) { |
- base::FilePath app_path = |
- chrome_launcher::GetChromeExecutablePath().DirName(); |
- hr = registrar->AddReplacement(L"CHROME_APPPATH", |
- app_path.value().c_str()); |
- DCHECK(SUCCEEDED(hr)); |
- } |
- |
- if (SUCCEEDED(hr)) { |
- hr = registrar->AddReplacement(L"CHROME_APPNAME", |
- chrome::kBrowserProcessExecutableName); |
- DCHECK(SUCCEEDED(hr)); |
- |
- // Fill in VERSION from the VERSIONINFO stored in the DLL's resources. |
- scoped_ptr<FileVersionInfo> module_version_info( |
- FileVersionInfo::CreateFileVersionInfoForCurrentModule()); |
- DCHECK(module_version_info != NULL); |
- std::wstring file_version(module_version_info->file_version()); |
- hr = registrar->AddReplacement(L"VERSION", file_version.c_str()); |
- DCHECK(SUCCEEDED(hr)); |
- } |
- |
- if (SUCCEEDED(hr)) { |
- // Add the directory of chrome_launcher.exe. This will be the same |
- // as the directory for the current DLL. |
- std::wstring module_dir; |
- base::FilePath module_path; |
- if (PathService::Get(base::FILE_MODULE, &module_path)) { |
- module_dir = module_path.DirName().value(); |
- } else { |
- NOTREACHED(); |
- } |
- hr = registrar->AddReplacement(L"CHROME_LAUNCHER_APPPATH", |
- module_dir.c_str()); |
- DCHECK(SUCCEEDED(hr)); |
- } |
- |
- if (SUCCEEDED(hr)) { |
- // Add the filename of chrome_launcher.exe |
- hr = registrar->AddReplacement(L"CHROME_LAUNCHER_APPNAME", |
- chrome_launcher::kLauncherExeBaseName); |
- DCHECK(SUCCEEDED(hr)); |
- } |
- |
- if (SUCCEEDED(hr)) { |
- // Add the registry hive to use. |
- // Note: This is ugly as hell. I'd rather use the pMapEntries parameter |
- // to CAtlModule::UpdateRegistryFromResource, unfortunately we have a |
- // few components that are registered by calling their |
- // static T::UpdateRegistry() methods directly, which doesn't allow |
- // pMapEntries to be passed through :-( |
- if (do_system_registration_) { |
- hr = registrar->AddReplacement(L"HIVE", L"HKLM"); |
- } else { |
- hr = registrar->AddReplacement(L"HIVE", L"HKCU"); |
- } |
- DCHECK(SUCCEEDED(hr)); |
- } |
- |
- if (SUCCEEDED(hr)) { |
- // Add the Chrome Frame CLSID. |
- wchar_t cf_clsid[64]; |
- StringFromGUID2(CLSID_ChromeFrame, &cf_clsid[0], arraysize(cf_clsid)); |
- hr = registrar->AddReplacement(L"CHROME_FRAME_CLSID", &cf_clsid[0]); |
- } |
- |
- return hr; |
- } |
- |
- // The module is "locked" when an object takes a reference on it. The first |
- // time it is locked, take a reference on crash reporting to bind its lifetime |
- // to the module and initialize ICU. |
- virtual LONG Lock() throw() { |
- LONG result = ParentClass::Lock(); |
- if (result == 1) { |
- DCHECK_EQ(crash_reporting_, |
- static_cast<chrome_frame::ScopedCrashReporting*>(NULL)); |
- crash_reporting_ = new chrome_frame::ScopedCrashReporting(); |
- |
- // Initialize ICU if this is the first time the module has been locked. |
- if (!icu_initialized_) { |
- icu_initialized_ = true; |
- // Best-effort since something is better than nothing here. |
- ignore_result(base::i18n::InitializeICU()); |
- } |
- } |
- return result; |
- } |
- |
- // The module is "unlocked" when an object that had a reference on it is |
- // destroyed. The last time it is unlocked, release the reference on crash |
- // reporting. |
- virtual LONG Unlock() throw() { |
- LONG result = ParentClass::Unlock(); |
- if (!result) { |
- DCHECK_NE(crash_reporting_, |
- static_cast<chrome_frame::ScopedCrashReporting*>(NULL)); |
- delete crash_reporting_; |
- crash_reporting_ = NULL; |
- } |
- return result; |
- } |
- |
- // See comments in AddCommonRGSReplacements |
- bool do_system_registration_; |
- |
- private: |
- // A scoper created when the module is initially locked and destroyed when it |
- // is finally unlocked. This is not a scoped_ptr since that could cause |
- // reporting to shut down at exit, which would lead to problems with the |
- // loader lock. |
- chrome_frame::ScopedCrashReporting* crash_reporting_; |
- |
- // Initially false, this is flipped to true to indicate that ICU has been |
- // initialized for the module. |
- bool icu_initialized_; |
-}; |
- |
-ChromeTabModule _AtlModule; |
- |
-base::AtExitManager* g_exit_manager = NULL; |
- |
-HRESULT RefreshElevationPolicy() { |
- const wchar_t kIEFrameDll[] = L"ieframe.dll"; |
- const char kIERefreshPolicy[] = "IERefreshElevationPolicy"; |
- HRESULT hr = E_NOTIMPL; |
- |
- // Stick an SEH in the chain to prevent the VEH from picking up on first |
- // chance exceptions caused by loading ieframe.dll. Use the vanilla |
- // ExceptionBarrier to report any exceptions that do make their way to us |
- // though. |
- ExceptionBarrier barrier; |
- |
- HMODULE ieframe_module = LoadLibrary(kIEFrameDll); |
- if (ieframe_module) { |
- typedef HRESULT (__stdcall *IERefreshPolicy)(); |
- IERefreshPolicy ie_refresh_policy = reinterpret_cast<IERefreshPolicy>( |
- GetProcAddress(ieframe_module, kIERefreshPolicy)); |
- |
- if (ie_refresh_policy) { |
- hr = ie_refresh_policy(); |
- } else { |
- hr = HRESULT_FROM_WIN32(GetLastError()); |
- } |
- |
- FreeLibrary(ieframe_module); |
- } else { |
- hr = HRESULT_FROM_WIN32(GetLastError()); |
- } |
- |
- return hr; |
-} |
- |
-// Experimental boot prefetch optimization for Chrome Frame |
-// |
-// If chrome is warmed up during a single reboot, it gets paged |
-// in for subsequent reboots and the cold startup times essentially |
-// look like warm times thereafter! The 'warm up' is done by |
-// setting up a 'RunOnce' key during DLLRegisterServer of |
-// npchrome_frame.dll. |
-// |
-// This works because chrome prefetch becomes part of boot |
-// prefetch file ntosboot-b00dfaad.pf and paged in on subsequent |
-// reboots. As long as the sytem does not undergo significant |
-// memory pressure those pages remain in memory and we get pretty |
-// amazing startup times, down to about 300 ms from 1200 ms |
-// |
-// The downside is: |
-// - Whether chrome frame is used or not, there's a read penalty |
-// (1200-300 =) 900 ms for every boot. |
-// - Heavy system memory usage after reboot will nullify the benefits |
-// but the user will still pay the cost. |
-// - Overall the time saved will always be less than total time spent |
-// paging in chrome |
-// - We are not sure when the chrome 'warm up' will age out from the |
-// boot prefetch file. |
-// |
-// The idea here is to try this out on chrome frame dev channel |
-// and see if it produces a significant drift in startup numbers. |
-HRESULT SetupRunOnce() { |
- HRESULT result = E_FAIL; |
- |
- base::string16 channel_name; |
- if (base::win::GetVersion() < base::win::VERSION_VISTA && |
- GoogleUpdateSettings::GetChromeChannelAndModifiers(true, &channel_name)) { |
- std::transform(channel_name.begin(), channel_name.end(), |
- channel_name.begin(), tolower); |
- // Use this only for the dev channel. |
- if (channel_name.find(L"dev") != base::string16::npos) { |
- HKEY hive = HKEY_CURRENT_USER; |
- if (IsSystemProcess()) { |
- // For system installs, our updates will be running as SYSTEM which |
- // makes writing to a RunOnce key under HKCU not so terribly useful. |
- hive = HKEY_LOCAL_MACHINE; |
- } |
- |
- RegKey run_once; |
- LONG ret = run_once.Create(hive, kRunOnce, KEY_READ | KEY_WRITE); |
- if (ret == ERROR_SUCCESS) { |
- CommandLine run_once_cmd(chrome_launcher::GetChromeExecutablePath()); |
- run_once_cmd.AppendSwitchASCII(switches::kAutomationClientChannelID, |
- "0"); |
- run_once_cmd.AppendSwitch(switches::kChromeFrame); |
- ret = run_once.WriteValue(L"A", |
- run_once_cmd.GetCommandLineString().c_str()); |
- } |
- result = HRESULT_FROM_WIN32(ret); |
- } else { |
- result = S_FALSE; |
- } |
- } else { |
- // We're on a non-XP version of Windows or on a stable channel. Nothing |
- // needs doing. |
- result = S_FALSE; |
- } |
- |
- return result; |
-} |
- |
-// Helper method called for user-level installs where we don't have admin |
-// permissions. Starts up the long running process and registers it to get it |
-// started at next boot. |
-HRESULT SetupUserLevelHelper() { |
- HRESULT hr = S_OK; |
- |
- // Remove existing run-at-startup entry. |
- base::win::RemoveCommandFromAutoRun(HKEY_CURRENT_USER, kRunKeyName); |
- |
- // Build the chrome_frame_helper command line. |
- base::FilePath module_path; |
- base::FilePath helper_path; |
- if (PathService::Get(base::FILE_MODULE, &module_path)) { |
- module_path = module_path.DirName(); |
- helper_path = module_path.Append(kChromeFrameHelperExe); |
- if (!base::PathExists(helper_path)) { |
- // If we can't find the helper in the current directory, try looking |
- // one up (this is the layout in the build output folder). |
- module_path = module_path.DirName(); |
- helper_path = module_path.Append(kChromeFrameHelperExe); |
- DCHECK(base::PathExists(helper_path)) << |
- "Could not find chrome_frame_helper.exe."; |
- } |
- |
- // Find window handle of existing instance. |
- HWND old_window = FindWindow(kChromeFrameHelperWindowClassName, |
- kChromeFrameHelperWindowName); |
- |
- if (base::PathExists(helper_path)) { |
- std::wstring helper_path_cmd(L"\""); |
- helper_path_cmd += helper_path.value(); |
- helper_path_cmd += L"\" "; |
- helper_path_cmd += kChromeFrameHelperStartupArg; |
- |
- // Add new run-at-startup entry. |
- if (!base::win::AddCommandToAutoRun(HKEY_CURRENT_USER, kRunKeyName, |
- helper_path_cmd)) { |
- hr = E_FAIL; |
- LOG(ERROR) << "Could not add helper process to auto run key."; |
- } |
- |
- // Start new instance. |
- base::LaunchOptions options; |
- options.start_hidden = true; |
- bool launched = base::LaunchProcess(helper_path.value(), options, NULL); |
- if (!launched) { |
- hr = E_FAIL; |
- PLOG(DFATAL) << "Could not launch helper process."; |
- } |
- |
- // Kill old instance using window handle. |
- if (IsWindow(old_window)) { |
- BOOL result = PostMessage(old_window, WM_CLOSE, 0, 0); |
- if (!result) { |
- PLOG(ERROR) << "Failed to post close message to old helper process: "; |
- } |
- } |
- } else { |
- hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND); |
- } |
- } else { |
- hr = E_UNEXPECTED; |
- NOTREACHED(); |
- } |
- |
- return hr; |
-} |
- |
-// To delete the user agent, set value to NULL. |
-// The is_system parameter indicates whether this is a per machine or a per |
-// user installation. |
-HRESULT SetChromeFrameUA(bool is_system, const wchar_t* value) { |
- HRESULT hr = E_FAIL; |
- HKEY parent_hive = is_system ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; |
- |
- RegKey ua_key; |
- LONG reg_result = ua_key.Create(parent_hive, kPostPlatformUAKey, |
- KEY_READ | KEY_WRITE); |
- if (reg_result == ERROR_SUCCESS) { |
- // Make sure that we unregister ChromeFrame UA strings registered previously |
- wchar_t value_name[MAX_PATH + 1] = {}; |
- wchar_t value_data[MAX_PATH + 1] = {}; |
- |
- DWORD value_index = 0; |
- while (value_index < ua_key.GetValueCount()) { |
- DWORD name_size = arraysize(value_name); |
- DWORD value_size = arraysize(value_data); |
- DWORD type = 0; |
- LRESULT ret = ::RegEnumValue(ua_key.Handle(), value_index, value_name, |
- &name_size, NULL, &type, |
- reinterpret_cast<BYTE*>(value_data), |
- &value_size); |
- if (ret == ERROR_SUCCESS) { |
- if (StartsWith(value_name, kChromeFramePrefix, false)) { |
- ua_key.DeleteValue(value_name); |
- } else { |
- ++value_index; |
- } |
- } else { |
- break; |
- } |
- } |
- |
- std::wstring chrome_frame_ua_value_name = kChromeFramePrefix; |
- chrome_frame_ua_value_name += GetCurrentModuleVersion(); |
- if (value) { |
- ua_key.WriteValue(chrome_frame_ua_value_name.c_str(), value); |
- } |
- hr = S_OK; |
- } else { |
- DLOG(ERROR) << __FUNCTION__ << ": " << kPostPlatformUAKey |
- << ", error code = " << reg_result; |
- hr = HRESULT_FROM_WIN32(reg_result); |
- } |
- return hr; |
-} |
- |
-class SecurityDescBackup { |
- public: |
- explicit SecurityDescBackup(const std::wstring& backup_key) |
- : backup_key_name_(backup_key) {} |
- ~SecurityDescBackup() {} |
- |
- // Save given security descriptor to the backup key. |
- bool SaveSecurity(const CSecurityDesc& sd) { |
- CString str; |
- if (!sd.ToString(&str)) |
- return false; |
- |
- RegKey backup_key(HKEY_LOCAL_MACHINE, backup_key_name_.c_str(), |
- KEY_READ | KEY_WRITE); |
- if (backup_key.Valid()) { |
- return backup_key.WriteValue(NULL, str.GetString()) == ERROR_SUCCESS; |
- } |
- |
- return false; |
- } |
- |
- // Restore security descriptor from backup key to given key name. |
- bool RestoreSecurity(const wchar_t* key_name) { |
- std::wstring sddl; |
- if (!ReadBackupKey(&sddl)) |
- return false; |
- |
- // Create security descriptor from string. |
- CSecurityDesc sd; |
- if (!sd.FromString(sddl.c_str())) |
- return false; |
- |
- bool result = true; |
- // Restore DACL and Owner of the key from saved security descriptor. |
- CDacl dacl; |
- CSid owner; |
- sd.GetDacl(&dacl); |
- sd.GetOwner(&owner); |
- |
- DWORD error = ::SetNamedSecurityInfo(const_cast<wchar_t*>(key_name), |
- SE_REGISTRY_KEY, OWNER_SECURITY_INFORMATION | DACL_SECURITY_INFORMATION, |
- const_cast<SID*>(owner.GetPSID()), NULL, |
- const_cast<ACL*>(dacl.GetPACL()), NULL); |
- |
- DeleteBackupKey(); |
- return (error == ERROR_SUCCESS); |
- } |
- |
- private: |
- // Read SDDL string from backup key |
- bool ReadBackupKey(std::wstring* sddl) { |
- RegKey backup_key(HKEY_LOCAL_MACHINE, backup_key_name_.c_str(), KEY_READ); |
- if (!backup_key.Valid()) |
- return false; |
- |
- DWORD len = 0; |
- DWORD reg_type = REG_NONE; |
- if (backup_key.ReadValue(NULL, NULL, &len, ®_type) != ERROR_SUCCESS) |
- return false; |
- DCHECK_EQ(0u, len % sizeof(wchar_t)); |
- |
- if ((len == 0) || (reg_type != REG_SZ)) |
- return false; |
- |
- size_t wchar_count = 1 + len / sizeof(wchar_t); |
- if (backup_key.ReadValue(NULL, WriteInto(sddl, wchar_count), &len, |
- ®_type) != ERROR_SUCCESS) { |
- return false; |
- } |
- |
- return true; |
- } |
- |
- void DeleteBackupKey() { |
- ::RegDeleteKey(HKEY_LOCAL_MACHINE, backup_key_name_.c_str()); |
- } |
- |
- std::wstring backup_key_name_; |
-}; |
- |
-struct TokenWithPrivileges { |
- TokenWithPrivileges() { |
- token_.GetEffectiveToken(TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY); |
- token_.GetUser(&user_); |
- } |
- |
- ~TokenWithPrivileges() { |
- token_.EnableDisablePrivileges(take_ownership_); |
- token_.EnableDisablePrivileges(restore_); |
- } |
- |
- bool EnablePrivileges() { |
- if (take_ownership_.GetCount() == 0) |
- if (!token_.EnablePrivilege(L"SeTakeOwnershipPrivilege", |
- &take_ownership_)) |
- return false; |
- |
- if (restore_.GetCount() == 0) |
- if (!token_.EnablePrivilege(L"SeRestorePrivilege", &restore_)) |
- return false; |
- |
- return true; |
- } |
- |
- const CSid& GetUser() const { |
- return user_; |
- } |
- |
- private: |
- CAccessToken token_; |
- CTokenPrivileges take_ownership_; |
- CTokenPrivileges restore_; |
- CSid user_; |
-}; |
- |
-const wchar_t* const kMimeHandlerKeyValues[] = { |
- L"ChromeTab.ChromeActiveDocument", |
- L"ChromeTab.ChromeActiveDocument.1", |
-}; |
- |
-// Returns true if the values are present or absent in |root_key|'s Secure Mime |
-// Handlers key based on |for_installed|. Returns false if the values are not as |
-// expected or if an error occurred. |
-bool MimeHandlerKeyIsConfigured(bool for_install, HKEY root_key) { |
- base::string16 key_name(kInternetSettings); |
- key_name.append(L"\\Secure Mime Handlers"); |
- RegKey key(root_key, key_name.c_str(), KEY_QUERY_VALUE); |
- if (!key.Valid()) |
- return false; |
- |
- for (size_t i = 0; i < arraysize(kMimeHandlerKeyValues); ++i) { |
- DWORD value = 0; |
- LONG result = key.ReadValueDW(kMimeHandlerKeyValues[i], &value); |
- if (for_install) { |
- if (result != ERROR_SUCCESS || value != 1) |
- return false; |
- } else { |
- if (result != ERROR_FILE_NOT_FOUND) |
- return false; |
- } |
- } |
- return true; |
-} |
- |
-HRESULT SetOrDeleteMimeHandlerKey(bool set, HKEY root_key) { |
- base::string16 key_name(kInternetSettings); |
- key_name.append(L"\\Secure Mime Handlers"); |
- RegKey key(root_key, key_name.c_str(), KEY_SET_VALUE); |
- if (!key.Valid()) |
- return false; |
- |
- HRESULT result = S_OK; |
- for (size_t i = 0; i < arraysize(kMimeHandlerKeyValues); ++i) { |
- LONG intermediate = set ? |
- key.WriteValue(kMimeHandlerKeyValues[i], 1) : |
- key.DeleteValue(kMimeHandlerKeyValues[i]); |
- if (intermediate != ERROR_SUCCESS && result == S_OK) |
- result = HRESULT_FROM_WIN32(intermediate); |
- } |
- return result; |
-} |
- |
-void OnPinModule() { |
- // Pin crash reporting by leaking a reference. |
- ignore_result(new chrome_frame::ScopedCrashReporting()); |
-} |
- |
-// Chrome Frame registration functions. |
-//----------------------------------------------------------------------------- |
-HRESULT RegisterSecuredMimeHandler(bool enable, bool is_system) { |
- if (MimeHandlerKeyIsConfigured(enable, HKEY_LOCAL_MACHINE)) |
- return S_OK; |
- if (!is_system) |
- return SetOrDeleteMimeHandlerKey(enable, HKEY_CURRENT_USER); |
- if (base::win::GetVersion() < base::win::VERSION_VISTA) |
- return SetOrDeleteMimeHandlerKey(enable, HKEY_LOCAL_MACHINE); |
- |
- std::wstring mime_key = kInternetSettings; |
- mime_key.append(L"\\Secure Mime Handlers"); |
- std::wstring backup_key = kInternetSettings; |
- backup_key.append(L"\\__backup_SMH__"); |
- std::wstring object_name = L"MACHINE\\"; |
- object_name.append(mime_key); |
- |
- TokenWithPrivileges token_; |
- if (!token_.EnablePrivileges()) |
- return E_ACCESSDENIED; |
- |
- // If there is a backup key - something bad happened; try to restore |
- // security on "Secure Mime Handlers" from the backup. |
- SecurityDescBackup backup(backup_key); |
- backup.RestoreSecurity(object_name.c_str()); |
- |
- // Read old security descriptor of the Mime key first. |
- CSecurityDesc sd; |
- if (!AtlGetSecurityDescriptor(object_name.c_str(), SE_REGISTRY_KEY, &sd)) { |
- return E_FAIL; |
- } |
- |
- backup.SaveSecurity(sd); |
- HRESULT hr = E_FAIL; |
- // set new owner |
- if (AtlSetOwnerSid(object_name.c_str(), SE_REGISTRY_KEY, token_.GetUser())) { |
- // set new dacl |
- CDacl new_dacl; |
- sd.GetDacl(&new_dacl); |
- new_dacl.AddAllowedAce(token_.GetUser(), GENERIC_WRITE | GENERIC_READ); |
- if (AtlSetDacl(object_name.c_str(), SE_REGISTRY_KEY, new_dacl)) { |
- hr = SetOrDeleteMimeHandlerKey(enable, HKEY_LOCAL_MACHINE); |
- } |
- } |
- |
- backup.RestoreSecurity(object_name.c_str()); |
- return hr; |
-} |
- |
-HRESULT RegisterActiveDoc(bool reg, bool is_system) { |
- // We have to call the static T::UpdateRegistry function instead of |
- // _AtlModule.UpdateRegistryFromResourceS(IDR_CHROMEFRAME_ACTIVEDOC, reg) |
- // because there is specific OLEMISC replacement. |
- return ChromeActiveDocument::UpdateRegistry(reg); |
-} |
- |
-HRESULT RegisterActiveX(bool reg, bool is_system) { |
- // We have to call the static T::UpdateRegistry function instead of |
- // _AtlModule.UpdateRegistryFromResourceS(IDR_CHROMEFRAME_ACTIVEX, reg) |
- // because there is specific OLEMISC replacement. |
- return ChromeFrameActivex::UpdateRegistry(reg); |
-} |
- |
-HRESULT RegisterElevationPolicy(bool reg, bool is_system) { |
- HRESULT hr = S_OK; |
- if (base::win::GetVersion() >= base::win::VERSION_VISTA) { |
- // Register the elevation policy. This must succeed for Chrome Frame to |
- // be able launch Chrome when running in low-integrity IE. |
- hr = _AtlModule.UpdateRegistryFromResourceS(IDR_CHROMEFRAME_ELEVATION, reg); |
- if (SUCCEEDED(hr)) { |
- // Ignore failures since old versions of IE 7 (e.g., 7.0.6000.16386, which |
- // shipped with Vista RTM) do not export IERefreshElevationPolicy. |
- RefreshElevationPolicy(); |
- } |
- } |
- return hr; |
-} |
- |
-HRESULT RegisterProtocol(bool reg, bool is_system) { |
- return _AtlModule.UpdateRegistryFromResourceS(IDR_CHROMEPROTOCOL, reg); |
-} |
- |
-HRESULT RegisterBhoClsid(bool reg, bool is_system) { |
- return Bho::UpdateRegistry(reg); |
-} |
- |
-HRESULT RegisterBhoIE(bool reg, bool is_system) { |
- if (is_system) { |
- return _AtlModule.UpdateRegistryFromResourceS(IDR_REGISTER_BHO, reg); |
- } else { |
- if (reg) { |
- // Setup the long running process: |
- return SetupUserLevelHelper(); |
- } else { |
- // Unschedule the user-level helper. Note that we don't kill it here |
- // so that during updates we don't have a time window with no running |
- // helper. Uninstalls and updates will explicitly kill the helper from |
- // within the installer. Unregister existing run-at-startup entry. |
- return base::win::RemoveCommandFromAutoRun(HKEY_CURRENT_USER, |
- kRunKeyName) ? S_OK : E_FAIL; |
- } |
- } |
-} |
- |
-HRESULT RegisterTypeLib(bool reg, bool is_system) { |
- if (reg && !is_system) { |
- // Enables the RegisterTypeLib Function function to override default |
- // registry mappings under Windows Vista Service Pack 1 (SP1), |
- // Windows Server 2008, and later operating system versions |
- typedef void (WINAPI* OaEnablePerUserTypeLibReg)(void); |
- OaEnablePerUserTypeLibReg per_user_typelib_func = |
- reinterpret_cast<OaEnablePerUserTypeLibReg>( |
- GetProcAddress(GetModuleHandle(L"oleaut32.dll"), |
- "OaEnablePerUserTLibRegistration")); |
- if (per_user_typelib_func) { |
- (*per_user_typelib_func)(); |
- } |
- } |
- return reg ? |
- UtilRegisterTypeLib(_AtlComModule.m_hInstTypeLib, |
- NULL, !is_system) : |
- UtilUnRegisterTypeLib(_AtlComModule.m_hInstTypeLib, |
- NULL, !is_system); |
-} |
- |
-HRESULT RegisterLegacyNPAPICleanup(bool reg, bool is_system) { |
- if (!reg) { |
- _AtlModule.UpdateRegistryFromResourceS(IDR_CHROMEFRAME_NPAPI, reg); |
- UtilRemovePersistentNPAPIMarker(); |
- } |
- // Ignore failures. |
- return S_OK; |
-} |
- |
-HRESULT RegisterAppId(bool reg, bool is_system) { |
- return _AtlModule.UpdateRegistryAppId(reg); |
-} |
- |
-HRESULT RegisterUserAgent(bool reg, bool is_system) { |
- if (reg) { |
- return SetChromeFrameUA(is_system, L"1"); |
- } else { |
- return SetChromeFrameUA(is_system, NULL); |
- } |
-} |
- |
-enum RegistrationStepId { |
- kStepSecuredMimeHandler = 0, |
- kStepActiveDoc = 1, |
- kStepActiveX = 2, |
- kStepElevationPolicy = 3, |
- kStepProtocol = 4, |
- kStepBhoClsid = 5, |
- kStepBhoRegistration = 6, |
- kStepRegisterTypeLib = 7, |
- kStepNpapiCleanup = 8, |
- kStepAppId = 9, |
- kStepUserAgent = 10, |
- kStepEnd = 11 |
-}; |
- |
-enum RegistrationFlags { |
- ACTIVEX = 0x0001, |
- ACTIVEDOC = 0x0002, |
- GCF_PROTOCOL = 0x0004, |
- BHO_CLSID = 0x0008, |
- BHO_REGISTRATION = 0x0010, |
- TYPELIB = 0x0020, |
- |
- ALL = 0xFFFF |
-}; |
- |
-// Mux the failure step into the hresult. We take only the first four bits |
-// and stick those into the top four bits of the facility code. We also set the |
-// Customer bit to be polite. Graphically, we write our error code to the |
-// bits marked with ^: |
-// 3 3 2 2 2 2 2 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1 |
-// 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0 |
-// +---+-+-+-----------------------+-------------------------------+ |
-// |Sev|C|R| Facility | Code | |
-// +---+-+-+-----------------------+-------------------------------+ |
-// ^ ^ ^ ^ ^ |
-// See http://msdn.microsoft.com/en-us/library/cc231198(PROT.10).aspx for |
-// more details on HRESULTS. |
-// |
-// The resulting error can be extracted by: |
-// error_code = (fiddled_hr & 0x07800000) >> 23 |
-HRESULT MuxErrorIntoHRESULT(HRESULT hr, int error_code) { |
- DCHECK_GE(error_code, 0); |
- DCHECK_LT(error_code, kStepEnd); |
- COMPILE_ASSERT(kStepEnd <= 0xF, update_error_muxing_too_many_steps); |
- |
- // Check that our four desired bits are clear. |
- // 0xF87FFFFF == 11111000011111111111111111111111 |
- DCHECK_EQ(static_cast<HRESULT>(hr & 0xF87FFFFF), hr); |
- |
- HRESULT fiddled_hr = ((error_code & 0xF) << 23) | hr; |
- fiddled_hr |= 1 << 29; // Set the customer bit. |
- |
- return fiddled_hr; |
-} |
- |
-HRESULT CustomRegistration(uint16 reg_flags, bool reg, bool is_system) { |
- if (reg && (reg_flags & (ACTIVEDOC | ACTIVEX))) |
- reg_flags |= (TYPELIB | GCF_PROTOCOL); |
- |
- // Set the flag that gets checked in AddCommonRGSReplacements before doing |
- // registration work. |
- _AtlModule.do_system_registration_ = is_system; |
- |
- typedef HRESULT (*RegistrationFn)(bool reg, bool is_system); |
- struct RegistrationStep { |
- RegistrationStepId id; |
- uint16 condition; |
- RegistrationFn func; |
- }; |
- static const RegistrationStep registration_steps[] = { |
- { kStepSecuredMimeHandler, ACTIVEDOC, &RegisterSecuredMimeHandler }, |
- { kStepActiveDoc, ACTIVEDOC, &RegisterActiveDoc }, |
- { kStepActiveX, ACTIVEX, &RegisterActiveX }, |
- { kStepElevationPolicy, (ACTIVEDOC | ACTIVEX), &RegisterElevationPolicy }, |
- { kStepProtocol, GCF_PROTOCOL, &RegisterProtocol }, |
- { kStepBhoClsid, BHO_CLSID, &RegisterBhoClsid }, |
- { kStepBhoRegistration, BHO_REGISTRATION, &RegisterBhoIE }, |
- { kStepRegisterTypeLib, TYPELIB, &RegisterTypeLib }, |
- { kStepNpapiCleanup, ALL, &RegisterLegacyNPAPICleanup }, |
- { kStepAppId, ALL, &RegisterAppId }, |
- { kStepUserAgent, ALL, &RegisterUserAgent } |
- }; |
- |
- HRESULT hr = S_OK; |
- |
- bool rollback = false; |
- int failure_step = 0; |
- HRESULT step_hr = S_OK; |
- for (int step = 0; step < arraysize(registration_steps); ++step) { |
- if ((reg_flags & registration_steps[step].condition) != 0) { |
- step_hr = registration_steps[step].func(reg, is_system); |
- if (FAILED(step_hr)) { |
- // Store only the first failing HRESULT with the step value muxed in. |
- if (hr == S_OK) { |
- hr = MuxErrorIntoHRESULT(step_hr, step); |
- } |
- |
- // On registration if a step fails, we abort and rollback. |
- if (reg) { |
- rollback = true; |
- failure_step = step; |
- break; |
- } |
- } |
- } |
- } |
- |
- if (rollback) { |
- DCHECK(reg); |
- // Rollback the failing action and all preceding ones. |
- for (int step = failure_step; step >= 0; --step) { |
- registration_steps[step].func(!reg, is_system); |
- } |
- } |
- |
- return hr; |
-} |
- |
-} // namespace |
- |
-// DLL Entry Point |
-extern "C" BOOL WINAPI DllMain(HINSTANCE instance, |
- DWORD reason, |
- LPVOID reserved) { |
- UNREFERENCED_PARAMETER(instance); |
- if (reason == DLL_PROCESS_ATTACH) { |
-#if _ATL_VER < 0x0C00 && !defined(NDEBUG) |
- // Silence traces from the ATL registrar to reduce the log noise. |
- ATL::CTrace::s_trace.ChangeCategory(atlTraceRegistrar, 0, |
- ATLTRACESTATUS_DISABLED); |
-#endif |
- InitGoogleUrl(); |
- |
- g_exit_manager = new base::AtExitManager(); |
- CommandLine::Init(0, NULL); |
- logging::LoggingSettings settings; |
- settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG; |
- logging::InitLogging(settings); |
- |
- // Log the same items as Chrome. |
- logging::SetLogItems(true, // enable_process_id |
- true, // enable_thread_id |
- false, // enable_timestamp |
- true); // enable_tickcount |
- |
- DllRedirector* dll_redirector = DllRedirector::GetInstance(); |
- DCHECK(dll_redirector); |
- |
- if (!dll_redirector->RegisterAsFirstCFModule()) { |
- // Someone else was here first, try and get a pointer to their |
- // DllGetClassObject export: |
- g_dll_get_class_object_redir_ptr = |
- dll_redirector->GetDllGetClassObjectPtr(); |
- DCHECK(g_dll_get_class_object_redir_ptr != NULL) |
- << "Found CF module with no DllGetClassObject export."; |
- } |
- |
- // Enable trace control and transport through event tracing for Windows. |
- logging::LogEventProvider::Initialize(kChromeFrameProvider); |
- |
- // Set a callback so that crash reporting can be pinned when the module is |
- // pinned. |
- chrome_frame::SetPinModuleCallback(&OnPinModule); |
- } else if (reason == DLL_PROCESS_DETACH) { |
- DllRedirector* dll_redirector = DllRedirector::GetInstance(); |
- DCHECK(dll_redirector); |
- dll_redirector->UnregisterAsFirstCFModule(); |
- |
- g_patch_helper.UnpatchIfNeeded(); |
- |
- delete g_exit_manager; |
- g_exit_manager = NULL; |
- } |
- return _AtlModule.DllMain(reason, reserved); |
-} |
- |
-// Used to determine whether the DLL can be unloaded by OLE |
-STDAPI DllCanUnloadNow() { |
- return _AtlModule.DllCanUnloadNow(); |
-} |
- |
-// Returns a class factory to create an object of the requested type |
-STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv) { |
- chrome_frame::ScopedCrashReporting crash_reporting; |
- |
- // IE 11+ are unsupported. |
- if (GetIEVersion() > IE_10) { |
- return CLASS_E_CLASSNOTAVAILABLE; |
- } |
- |
- // If we found another module present when we were loaded, then delegate to |
- // that: |
- if (g_dll_get_class_object_redir_ptr) { |
- return g_dll_get_class_object_redir_ptr(rclsid, riid, ppv); |
- } |
- |
- // Enable sniffing and switching only if asked for BHO |
- // (we use BHO to get loaded in IE). |
- if (rclsid == CLSID_ChromeFrameBHO) { |
- g_patch_helper.InitializeAndPatchProtocolsIfNeeded(); |
- } |
- |
- return _AtlModule.DllGetClassObject(rclsid, riid, ppv); |
-} |
- |
-// DllRegisterServer - Adds entries to the system registry |
-STDAPI DllRegisterServer() { |
- chrome_frame::ScopedCrashReporting crash_reporting; |
- uint16 flags = ACTIVEX | ACTIVEDOC | TYPELIB | GCF_PROTOCOL | |
- BHO_CLSID | BHO_REGISTRATION; |
- |
- HRESULT hr = CustomRegistration(flags, true, true); |
- if (SUCCEEDED(hr)) { |
- SetupRunOnce(); |
- } |
- |
- return hr; |
-} |
- |
-// DllUnregisterServer - Removes entries from the system registry |
-STDAPI DllUnregisterServer() { |
- chrome_frame::ScopedCrashReporting crash_reporting; |
- HRESULT hr = CustomRegistration(ALL, false, true); |
- return hr; |
-} |
- |
-// DllRegisterUserServer - Adds entries to the HKCU hive in the registry. |
-STDAPI DllRegisterUserServer() { |
- chrome_frame::ScopedCrashReporting crash_reporting; |
- UINT flags = ACTIVEX | ACTIVEDOC | TYPELIB | GCF_PROTOCOL | |
- BHO_CLSID | BHO_REGISTRATION; |
- |
- HRESULT hr = CustomRegistration(flags, TRUE, false); |
- if (SUCCEEDED(hr)) { |
- SetupRunOnce(); |
- } |
- |
- return hr; |
-} |
- |
-// DllUnregisterUserServer - Removes entries from the HKCU hive in the registry. |
-STDAPI DllUnregisterUserServer() { |
- chrome_frame::ScopedCrashReporting crash_reporting; |
- HRESULT hr = CustomRegistration(ALL, FALSE, false); |
- return hr; |
-} |
- |
-// Object entries go here instead of with each object, so that we can move |
-// the objects to a lib. Also reduces magic. |
-OBJECT_ENTRY_AUTO(CLSID_ChromeFrameBHO, Bho) |
-OBJECT_ENTRY_AUTO(__uuidof(ChromeActiveDocument), ChromeActiveDocument) |
-OBJECT_ENTRY_AUTO(__uuidof(ChromeFrame), ChromeFrameActivex) |
-OBJECT_ENTRY_AUTO(__uuidof(ChromeProtocol), ChromeProtocol) |