| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/installer/setup/cf_migration.h" | |
| 6 | |
| 7 #include "base/command_line.h" | |
| 8 #include "base/files/file_path.h" | |
| 9 #include "base/logging.h" | |
| 10 #include "base/process/launch.h" | |
| 11 #include "base/win/registry.h" | |
| 12 #include "chrome/installer/setup/setup_constants.h" | |
| 13 #include "chrome/installer/setup/setup_main.h" | |
| 14 #include "chrome/installer/setup/setup_util.h" | |
| 15 #include "chrome/installer/util/browser_distribution.h" | |
| 16 #include "chrome/installer/util/install_util.h" | |
| 17 #include "chrome/installer/util/installation_state.h" | |
| 18 #include "chrome/installer/util/installer_state.h" | |
| 19 #include "chrome/installer/util/master_preferences.h" | |
| 20 #include "chrome/installer/util/util_constants.h" | |
| 21 | |
| 22 namespace installer { | |
| 23 | |
| 24 bool LaunchChromeFrameMigrationProcess( | |
| 25 const ProductState& chrome_frame_product, | |
| 26 const CommandLine& command_line, | |
| 27 const base::FilePath& installer_directory, | |
| 28 bool system_level) { | |
| 29 // Before running the migration, mutate the CF ap value to include a | |
| 30 // "-migrate" beacon. This beacon value will be cleaned up by the "ap" | |
| 31 // cleanup in MigrateGoogleUpdateStateMultiToSingle that calls | |
| 32 // ChannelInfo::RemoveAllModifiersAndSuffixes(). | |
| 33 if (chrome_frame_product.is_multi_install()) { | |
| 34 const HKEY root = system_level ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; | |
| 35 base::win::RegKey state_key; | |
| 36 installer::ChannelInfo channel_info; | |
| 37 BrowserDistribution* dist = BrowserDistribution::GetSpecificDistribution( | |
| 38 BrowserDistribution::CHROME_FRAME); | |
| 39 | |
| 40 LONG result = state_key.Open(root, dist->GetStateKey().c_str(), | |
| 41 KEY_QUERY_VALUE | KEY_SET_VALUE); | |
| 42 if (result != ERROR_SUCCESS || !channel_info.Initialize(state_key)) { | |
| 43 LOG(ERROR) << "Failed to read CF channel to store beacon."; | |
| 44 } else if (!channel_info.SetMigratingSuffix(true)) { | |
| 45 LOG(WARNING) << "CF already has migration beacon in channel."; | |
| 46 } else { | |
| 47 VLOG(1) << "Writing CF migration beacon to channel: " | |
| 48 << channel_info.value(); | |
| 49 channel_info.Write(&state_key); | |
| 50 } | |
| 51 } | |
| 52 | |
| 53 // Call the installed setup.exe using the current installer command line and | |
| 54 // adding the migration flags. This seems like it could be unsafe, here's why | |
| 55 // it's safe today: | |
| 56 // 1) MigrateChromeFrameInChildProcess is called only during a multi update. | |
| 57 // 2) Multi update processing occurs after HandleNonInstallCmdLineOptions is | |
| 58 // called. | |
| 59 // 3) Setup exits if there were any non-install command line options handled. | |
| 60 // 4) Thus, the command line being copied will have no non-install command | |
| 61 // line options at time of copying. | |
| 62 // 5) kMigrateChromeFrame is a non-install command line option. | |
| 63 // 6) Thus, it will be handled (and the child setup process will exit) before | |
| 64 // the child setup process acts on any other flags on the command line. | |
| 65 // 7) Furthermore, --uncompressed-archive takes precedence over | |
| 66 // --install-archive, so it is safe to add the former to the command line | |
| 67 // without removing the latter. | |
| 68 CommandLine setup_cmd(command_line); | |
| 69 setup_cmd.SetProgram(installer_directory.Append(installer::kSetupExe)); | |
| 70 setup_cmd.AppendSwitchPath( | |
| 71 switches::kUncompressedArchive, | |
| 72 installer_directory.Append(installer::kChromeArchive)); | |
| 73 setup_cmd.AppendSwitch(switches::kMigrateChromeFrame); | |
| 74 | |
| 75 VLOG(1) << "Running Chrome Frame migration process with command line: " | |
| 76 << setup_cmd.GetCommandLineString(); | |
| 77 | |
| 78 base::LaunchOptions options; | |
| 79 options.force_breakaway_from_job_ = true; | |
| 80 if (!base::LaunchProcess(setup_cmd, options, NULL)) { | |
| 81 PLOG(ERROR) << "Launching Chrome Frame migration process failed. " | |
| 82 << "(Command line: " << setup_cmd.GetCommandLineString() << ")"; | |
| 83 return false; | |
| 84 } | |
| 85 | |
| 86 return true; | |
| 87 } | |
| 88 | |
| 89 InstallStatus MigrateChromeFrame(const InstallationState& original_state, | |
| 90 InstallerState* installer_state) { | |
| 91 const bool system_level = installer_state->system_install(); | |
| 92 | |
| 93 // Nothing to do if multi-install Chrome Frame is not installed. | |
| 94 const ProductState* multi_chrome_frame = original_state.GetProductState( | |
| 95 system_level, BrowserDistribution::CHROME_FRAME); | |
| 96 if (!multi_chrome_frame || !multi_chrome_frame->is_multi_install()) | |
| 97 return INVALID_STATE_FOR_OPTION; | |
| 98 | |
| 99 // Install SxS Chrome Frame. | |
| 100 InstallerState install_gcf(installer_state->level()); | |
| 101 { | |
| 102 scoped_ptr<Product> chrome_frame( | |
| 103 new Product(BrowserDistribution::GetSpecificDistribution( | |
| 104 BrowserDistribution::CHROME_FRAME))); | |
| 105 install_gcf.AddProduct(&chrome_frame); | |
| 106 } | |
| 107 DCHECK(!install_gcf.is_multi_install()); | |
| 108 | |
| 109 ArchiveType archive_type = UNKNOWN_ARCHIVE_TYPE; | |
| 110 bool delegated_to_existing = false; | |
| 111 InstallStatus install_status = InstallProductsHelper( | |
| 112 original_state, *CommandLine::ForCurrentProcess(), | |
| 113 MasterPreferences::ForCurrentProcess(), install_gcf, | |
| 114 NULL, &archive_type, &delegated_to_existing); | |
| 115 | |
| 116 if (!InstallUtil::GetInstallReturnCode(install_status)) { | |
| 117 // Migration was successful. There's no turning back now. The multi-install | |
| 118 // npchrome_frame.dll and/or chrome.exe may still be in use at this point, | |
| 119 // although the user-level helper will not be. It is not safe to delete the | |
| 120 // multi-install binaries until npchrome_frame.dll and chrome.exe are no | |
| 121 // longer in use. The remaining tasks here are best-effort. Failure does not | |
| 122 // do any harm. | |
| 123 MigrateGoogleUpdateStateMultiToSingle(system_level, | |
| 124 BrowserDistribution::CHROME_FRAME, | |
| 125 original_state); | |
| 126 } | |
| 127 | |
| 128 return install_status; | |
| 129 } | |
| 130 | |
| 131 } // namespace installer | |
| OLD | NEW |