// Copyright (c) 2006-2008 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. #include #include "base/file_util.h" #include "base/scoped_ptr.h" #include "base/string_util.h" #include "chrome/installer/setup/setup.h" #include "chrome/installer/setup/setup_constants.h" #include "chrome/installer/util/browser_distribution.h" #include "chrome/installer/util/google_update_constants.h" #include "chrome/installer/util/shell_util.h" namespace { std::wstring AppendPath(const std::wstring parent_path, const std::wstring path) { std::wstring new_path(parent_path); file_util::AppendToPath(&new_path, path); return new_path; } // This method adds work items to create (or update) Chrome uninstall entry in // Control Panel->Add/Remove Programs list. void AddUninstallShortcutWorkItems(HKEY reg_root, const std::wstring& exe_path, const std::wstring& install_path, const std::wstring& product_name, const std::wstring& new_version, WorkItemList* install_list) { std::wstring uninstall_cmd(L"\""); uninstall_cmd.append(installer::GetInstallerPathUnderChrome(install_path, new_version)); file_util::AppendToPath(&uninstall_cmd, file_util::GetFilenameFromPath(exe_path)); uninstall_cmd.append(L"\" --"); uninstall_cmd.append(installer_util::switches::kUninstall); if (reg_root == HKEY_LOCAL_MACHINE) { uninstall_cmd.append(L" --"); uninstall_cmd.append(installer_util::switches::kSystemLevel); } // Create DisplayName, UninstallString and InstallLocation keys BrowserDistribution* dist = BrowserDistribution::GetDistribution(); std::wstring uninstall_reg = dist->GetUninstallRegPath(); install_list->AddCreateRegKeyWorkItem(reg_root, uninstall_reg); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, installer_util::kUninstallDisplayNameField, product_name, true); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, installer_util::kUninstallStringField, uninstall_cmd, true); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"InstallLocation", install_path, true); // DisplayIcon, NoModify and NoRepair std::wstring chrome_icon = AppendPath(install_path, installer_util::kChromeExe); ShellUtil::GetChromeIcon(chrome_icon); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"DisplayIcon", chrome_icon, true); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"NoModify", 1, true); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"NoRepair", 1, true); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"Publisher", dist->GetPublisherName(), true); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"Version", new_version.c_str(), true); install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"DisplayVersion", new_version.c_str(), true); time_t rawtime = time(NULL); struct tm timeinfo = {0}; localtime_s(&timeinfo, &rawtime); wchar_t buffer[9]; if (wcsftime(buffer, 9, L"%Y%m%d", &timeinfo) == 8) { install_list->AddSetRegValueWorkItem(reg_root, uninstall_reg, L"InstallDate", buffer, false); } } void AddInstallerCopyTasks(const std::wstring& exe_path, const std::wstring& archive_path, const std::wstring& temp_path, const std::wstring& install_path, const std::wstring& new_version, WorkItemList* install_list) { std::wstring installer_dir(installer::GetInstallerPathUnderChrome( install_path, new_version)); install_list->AddCreateDirWorkItem(installer_dir); std::wstring exe_dst(installer_dir); std::wstring archive_dst(installer_dir); file_util::AppendToPath(&exe_dst, file_util::GetFilenameFromPath(exe_path)); file_util::AppendToPath(&archive_dst, file_util::GetFilenameFromPath(archive_path)); install_list->AddCopyTreeWorkItem(exe_path, exe_dst, temp_path, WorkItem::ALWAYS); install_list->AddCopyTreeWorkItem(archive_path, archive_dst, temp_path, WorkItem::ALWAYS); } // This method tells if we are running on 64 bit platform so that we can copy // one extra exe. If the API call to determine 64 bit fails, we play it safe // and return true anyway so that the executable can be copied. bool Is64bit() { typedef BOOL (WINAPI *WOW_FUNC)(HANDLE, PBOOL); BOOL is64 = FALSE; HANDLE handle = GetCurrentProcess(); HMODULE module = GetModuleHandle(L"kernel32.dll"); WOW_FUNC p = reinterpret_cast(GetProcAddress(module, "IsWow64Process")); if ((p != NULL) && (!(p)(handle, &is64) || (is64 != FALSE))) { return true; } return false; } } bool installer::InstallNewVersion(const std::wstring& exe_path, const std::wstring& archive_path, const std::wstring& src_path, const std::wstring& install_path, const std::wstring& temp_dir, const HKEY reg_root, const Version& new_version) { if (reg_root != HKEY_LOCAL_MACHINE && reg_root != HKEY_CURRENT_USER) return false; scoped_ptr install_list(WorkItem::CreateWorkItemList()); // A temp directory that work items need and the actual install directory. install_list->AddCreateDirWorkItem(temp_dir); install_list->AddCreateDirWorkItem(install_path); // Copy the version folder install_list->AddCopyTreeWorkItem( AppendPath(src_path, new_version.GetString()), AppendPath(install_path, new_version.GetString()), temp_dir, WorkItem::ALWAYS); // Always overwrite. // Delete any new_chrome.exe if present (we will end up create a new one // if required) and then copy chrome.exe install_list->AddDeleteTreeWorkItem( AppendPath(install_path, installer::kChromeNewExe), std::wstring()); install_list->AddCopyTreeWorkItem( AppendPath(src_path, installer_util::kChromeExe), AppendPath(install_path, installer_util::kChromeExe), temp_dir, WorkItem::RENAME_IF_IN_USE, AppendPath(install_path, installer::kChromeNewExe)); // Extra executable for 64 bit systems. if (Is64bit()) { install_list->AddCopyTreeWorkItem( AppendPath(src_path, installer::kWowHelperExe), AppendPath(install_path, installer::kWowHelperExe), temp_dir, WorkItem::ALWAYS); } // Copy the default Dictionaries only if the folder doesnt exist already install_list->AddCopyTreeWorkItem( AppendPath(src_path, installer::kDictionaries), AppendPath(install_path, installer::kDictionaries), temp_dir, WorkItem::IF_NOT_PRESENT); // Copy installer in install directory and // add shortcut in Control Panel->Add/Remove Programs. AddInstallerCopyTasks(exe_path, archive_path, temp_dir, install_path, new_version.GetString(), install_list.get()); BrowserDistribution* dist = BrowserDistribution::GetDistribution(); std::wstring product_name = dist->GetApplicationName(); AddUninstallShortcutWorkItems(reg_root, exe_path, install_path, product_name, new_version.GetString(), install_list.get()); // Delete any old_chrome.exe if present. install_list->AddDeleteTreeWorkItem( AppendPath(install_path, installer::kChromeOldExe), std::wstring()); // Create Version key (if not already present) and set the new Chrome // version as last step. std::wstring version_key = dist->GetVersionKey(); install_list->AddCreateRegKeyWorkItem(reg_root, version_key); install_list->AddSetRegValueWorkItem(reg_root, version_key, google_update::kRegNameField, product_name, true); // overwrite name also install_list->AddSetRegValueWorkItem(reg_root, version_key, google_update::kRegVersionField, new_version.GetString(), true); // overwrite version // Perform install operations. if (!install_list->Do()) { LOG(ERROR) << "Install failed, rolling back... "; install_list->Rollback(); LOG(ERROR) << "Rollback complete. "; return false; } return true; }