Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // | 4 // |
| 5 // This file declares util functions for setup project. | 5 // This file declares util functions for setup project. |
| 6 | 6 |
| 7 #include "chrome/installer/setup/setup_util.h" | 7 #include "chrome/installer/setup/setup_util.h" |
| 8 | 8 |
| 9 #include <windows.h> | 9 #include <windows.h> |
| 10 #include <stddef.h> | 10 #include <stddef.h> |
| 11 | 11 |
| 12 #include <algorithm> | 12 #include <algorithm> |
| 13 #include <iterator> | 13 #include <iterator> |
| 14 #include <limits> | |
| 15 #include <memory> | |
| 14 #include <set> | 16 #include <set> |
| 15 #include <string> | 17 #include <string> |
| 16 | 18 |
| 17 #include "base/bind.h" | 19 #include "base/bind.h" |
| 18 #include "base/command_line.h" | 20 #include "base/command_line.h" |
| 19 #include "base/cpu.h" | 21 #include "base/cpu.h" |
| 20 #include "base/files/file_enumerator.h" | 22 #include "base/files/file_enumerator.h" |
| 21 #include "base/files/file_path.h" | 23 #include "base/files/file_path.h" |
| 22 #include "base/files/file_util.h" | 24 #include "base/files/file_util.h" |
| 23 #include "base/logging.h" | 25 #include "base/logging.h" |
| 24 #include "base/macros.h" | 26 #include "base/macros.h" |
| 25 #include "base/metrics/histogram.h" | 27 #include "base/metrics/histogram.h" |
| 26 #include "base/numerics/safe_conversions.h" | 28 #include "base/numerics/safe_conversions.h" |
| 27 #include "base/strings/string_number_conversions.h" | 29 #include "base/strings/string_number_conversions.h" |
| 28 #include "base/strings/string_util.h" | 30 #include "base/strings/string_util.h" |
| 29 #include "base/strings/utf_string_conversions.h" | 31 #include "base/strings/utf_string_conversions.h" |
| 30 #include "base/time/time.h" | 32 #include "base/time/time.h" |
| 31 #include "base/version.h" | 33 #include "base/version.h" |
| 32 #include "base/win/registry.h" | 34 #include "base/win/registry.h" |
| 33 #include "base/win/windows_version.h" | 35 #include "base/win/windows_version.h" |
| 34 #include "chrome/installer/setup/installer_state.h" | 36 #include "chrome/installer/setup/installer_state.h" |
| 35 #include "chrome/installer/setup/setup_constants.h" | 37 #include "chrome/installer/setup/setup_constants.h" |
| 36 #include "chrome/installer/setup/user_hive_visitor.h" | 38 #include "chrome/installer/setup/user_hive_visitor.h" |
| 37 #include "chrome/installer/util/app_registration_data.h" | 39 #include "chrome/installer/util/app_registration_data.h" |
| 38 #include "chrome/installer/util/google_update_constants.h" | 40 #include "chrome/installer/util/google_update_constants.h" |
| 41 #include "chrome/installer/util/install_util.h" | |
| 39 #include "chrome/installer/util/installation_state.h" | 42 #include "chrome/installer/util/installation_state.h" |
| 40 #include "chrome/installer/util/master_preferences.h" | 43 #include "chrome/installer/util/master_preferences.h" |
| 41 #include "chrome/installer/util/master_preferences_constants.h" | 44 #include "chrome/installer/util/master_preferences_constants.h" |
| 42 #include "chrome/installer/util/util_constants.h" | 45 #include "chrome/installer/util/util_constants.h" |
| 43 #include "courgette/courgette.h" | 46 #include "courgette/courgette.h" |
| 44 #include "courgette/third_party/bsdiff/bsdiff.h" | 47 #include "courgette/third_party/bsdiff/bsdiff.h" |
| 45 | 48 |
| 46 namespace installer { | 49 namespace installer { |
| 47 | 50 |
| 48 namespace { | 51 namespace { |
| 49 | 52 |
| 53 // Event log providers registry location. | |
| 54 constexpr wchar_t kEventLogProvidersRegPath[] = | |
| 55 L"SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\"; | |
| 56 | |
| 57 // TODO(grt): use install_static::InstallDetails::Get().install_full_name() when | |
| 58 // InstallDetails is initialized in the installer. | |
| 59 base::string16 InstallFullName() { | |
| 60 base::string16 reg_path(kEventLogProvidersRegPath); | |
|
grt (UTC plus 2)
2016/12/16 13:03:58
remove
pastarmovj
2016/12/16 14:58:12
Done.
| |
| 61 #if defined(GOOGLE_CHROME_BUILD) | |
| 62 reg_path = L"Chrome"; | |
| 63 if (InstallUtil::IsChromeSxSProcess()) | |
| 64 reg_path.append(L" SxS"); | |
| 65 #else | |
| 66 reg_path = L"Chromium"; | |
| 67 #endif | |
| 68 return reg_path; | |
| 69 } | |
| 70 | |
| 50 // Returns true if product |type| cam be meaningfully installed without the | 71 // Returns true if product |type| cam be meaningfully installed without the |
| 51 // --multi-install flag. | 72 // --multi-install flag. |
| 52 bool SupportsSingleInstall(BrowserDistribution::Type type) { | 73 bool SupportsSingleInstall(BrowserDistribution::Type type) { |
| 53 return (type == BrowserDistribution::CHROME_BROWSER || | 74 return (type == BrowserDistribution::CHROME_BROWSER || |
| 54 type == BrowserDistribution::CHROME_FRAME); | 75 type == BrowserDistribution::CHROME_FRAME); |
| 55 } | 76 } |
| 56 | 77 |
| 57 // Returns true if the "lastrun" value in |root|\|key_path| (a path to Chrome's | 78 // Returns true if the "lastrun" value in |root|\|key_path| (a path to Chrome's |
| 58 // ClientState key for a user) indicates that Chrome has been used within the | 79 // ClientState key for a user) indicates that Chrome has been used within the |
| 59 // last 28 days. | 80 // last 28 days. |
| (...skipping 593 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 653 UNPACK_STATUS_COUNT, UNPACK_STATUS_COUNT + 1, | 674 UNPACK_STATUS_COUNT, UNPACK_STATUS_COUNT + 1, |
| 654 base::HistogramBase::kUmaTargetedHistogramFlag) | 675 base::HistogramBase::kUmaTargetedHistogramFlag) |
| 655 ->Add(unpack_status); | 676 ->Add(unpack_status); |
| 656 | 677 |
| 657 base::SparseHistogram::FactoryGet( | 678 base::SparseHistogram::FactoryGet( |
| 658 std::string(kUnPackNTSTATUSMetricsName) + "_" + consumer_name, | 679 std::string(kUnPackNTSTATUSMetricsName) + "_" + consumer_name, |
| 659 base::HistogramBase::kUmaTargetedHistogramFlag) | 680 base::HistogramBase::kUmaTargetedHistogramFlag) |
| 660 ->Add(status); | 681 ->Add(status); |
| 661 } | 682 } |
| 662 | 683 |
| 684 void RegisterEventLogProvider(const base::FilePath& install_directory, | |
| 685 const base::Version& version) { | |
| 686 base::string16 reg_path(kEventLogProvidersRegPath); | |
| 687 reg_path.append(InstallFullName()); | |
| 688 VLOG(1) << "Registering Chrome's event log provider at " << reg_path; | |
| 689 | |
| 690 std::unique_ptr<WorkItemList> work_item_list(WorkItem::CreateWorkItemList()); | |
| 691 work_item_list->set_log_message("Register event log provider"); | |
| 692 | |
| 693 work_item_list->AddCreateRegKeyWorkItem(HKEY_LOCAL_MACHINE, reg_path, | |
| 694 WorkItem::kWow64Default); | |
| 695 // Speicifes the number of event categories defined in the dll. | |
| 696 work_item_list->AddSetRegValueWorkItem( | |
| 697 HKEY_LOCAL_MACHINE, reg_path, WorkItem::kWow64Default, L"CategoryCount", | |
| 698 static_cast<DWORD>(1), true); | |
| 699 // Specifies the event type emitted by this event source. | |
| 700 work_item_list->AddSetRegValueWorkItem( | |
| 701 HKEY_LOCAL_MACHINE, reg_path, WorkItem::kWow64Default, L"TypesSupported", | |
| 702 static_cast<DWORD>(EVENTLOG_ERROR_TYPE | EVENTLOG_INFORMATION_TYPE | | |
| 703 EVENTLOG_WARNING_TYPE), | |
| 704 true); | |
| 705 | |
| 706 const base::FilePath provider( | |
| 707 install_directory.AppendASCII(version.GetString()) | |
| 708 .Append(FILE_PATH_LITERAL("eventlog_provider.dll"))); | |
| 709 | |
| 710 static constexpr const wchar_t* kFileKeys[] = { | |
| 711 L"CategoryMessageFile", L"EventMessageFile", L"ParameterMessageFile", | |
| 712 }; | |
| 713 for (const wchar_t* file_key : kFileKeys) { | |
| 714 work_item_list->AddSetRegValueWorkItem(HKEY_LOCAL_MACHINE, reg_path, | |
| 715 WorkItem::kWow64Default, file_key, | |
| 716 provider.value(), true); | |
| 717 } | |
| 718 | |
| 719 // if the operation fails we log the error but still continue because none of | |
| 720 // these are critical for the proper operation of the browser. | |
| 721 if (!work_item_list->Do()) | |
| 722 work_item_list->Rollback(); | |
| 723 } | |
| 724 | |
| 725 void DeRegisterEventLogProvider() { | |
| 726 base::string16 reg_path(kEventLogProvidersRegPath); | |
| 727 reg_path.append(InstallFullName()); | |
| 728 | |
| 729 // TODO(http://crbug.com/668120): If the Event Viewer is open the provider dll | |
| 730 // will fail to get deleted. This doesn't fail the uninstallation altogether | |
| 731 // but leaves files behind. | |
| 732 InstallUtil::DeleteRegistryKey(HKEY_LOCAL_MACHINE, reg_path, | |
| 733 WorkItem::kWow64Default); | |
| 734 } | |
| 735 | |
| 663 ScopedTokenPrivilege::ScopedTokenPrivilege(const wchar_t* privilege_name) | 736 ScopedTokenPrivilege::ScopedTokenPrivilege(const wchar_t* privilege_name) |
| 664 : is_enabled_(false) { | 737 : is_enabled_(false) { |
| 665 HANDLE temp_handle; | 738 HANDLE temp_handle; |
| 666 if (!::OpenProcessToken(::GetCurrentProcess(), | 739 if (!::OpenProcessToken(::GetCurrentProcess(), |
| 667 TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, | 740 TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, |
| 668 &temp_handle)) { | 741 &temp_handle)) { |
| 669 return; | 742 return; |
| 670 } | 743 } |
| 671 token_.Set(temp_handle); | 744 token_.Set(temp_handle); |
| 672 | 745 |
| (...skipping 22 matching lines...) Expand all Loading... | |
| 695 } | 768 } |
| 696 | 769 |
| 697 ScopedTokenPrivilege::~ScopedTokenPrivilege() { | 770 ScopedTokenPrivilege::~ScopedTokenPrivilege() { |
| 698 if (is_enabled_ && previous_privileges_.PrivilegeCount != 0) { | 771 if (is_enabled_ && previous_privileges_.PrivilegeCount != 0) { |
| 699 ::AdjustTokenPrivileges(token_.Get(), FALSE, &previous_privileges_, | 772 ::AdjustTokenPrivileges(token_.Get(), FALSE, &previous_privileges_, |
| 700 sizeof(TOKEN_PRIVILEGES), NULL, NULL); | 773 sizeof(TOKEN_PRIVILEGES), NULL, NULL); |
| 701 } | 774 } |
| 702 } | 775 } |
| 703 | 776 |
| 704 } // namespace installer | 777 } // namespace installer |
| OLD | NEW |