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

Unified Diff: chrome_frame/chrome_tab.cc

Issue 126143005: Remove Chrome Frame code and resources. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: sync to r244038 Created 6 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « chrome_frame/chrome_protocol.rgs ('k') | chrome_frame/chrome_tab.def » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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, &reg_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,
- &reg_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)
« no previous file with comments | « chrome_frame/chrome_protocol.rgs ('k') | chrome_frame/chrome_tab.def » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698